source/texture2d.cpp

   1 #include "bindable.h"
   2 #include "buffer.h"
   3 #include "error.h"
   4 #include "pixelstore.h"
   5 #include "resources.h"
   6 #include "texture2d.h"
   7
   8 using namespace std;
   9
  10 namespace Msp {
  11 namespace GL {
  12
  13 class Texture2D::AsyncLoader: public Resource::AsyncLoader
  14 {
  15 private:
  16         Texture2D &texture;
  17         IO::Seekable &io;
  18         bool srgb_conversion;
  19         Buffer pixel_buffer;
  20         char *mapped_address;
  21         Graphics::Image image;
  22         unsigned n_bytes;
  23         int phase;
  24
  25 public:
  26         AsyncLoader(Texture2D &, IO::Seekable &);
  27
  28         void set_srgb_conversion(bool);
  29         virtual bool needs_sync() const;
  30         virtual bool process();
  31 };
  32
  33
  34 Texture2D::Texture2D(ResourceManager *m):
  35         Texture(GL_TEXTURE_2D, m),
  36         ifmt(RGB),
  37         width(0),
  38         height(0),
  39         allocated(0)
  40 { }
  41
  42 Texture2D::~Texture2D()
  43 {
  44         set_manager(0);
  45 }
  46
  47 void Texture2D::storage(PixelFormat fmt, unsigned wd, unsigned ht)
  48 {
  49         if(width>0)
  50                 throw invalid_operation("Texture2D::storage");
  51         if(wd==0 || ht==0)
  52                 throw invalid_argument("Texture2D::storage");
  53
  54         if(MSP_sized_internal_formats)
  55                 fmt = get_sized_pixelformat(fmt);
  56         require_pixelformat(fmt);
  57
  58         ifmt = fmt;
  59         width = wd;
  60         height = ht;
  61 }
  62
  63 void Texture2D::allocate(unsigned level)
  64 {
  65         if(allocated&(1<<level))
  66                 return;
  67
  68         PixelFormat base_fmt = get_base_pixelformat(ifmt);
  69         image(level, base_fmt, get_alloc_type(base_fmt), 0);
  70 }
  71
  72 void Texture2D::image(unsigned level, PixelFormat fmt, DataType type, const void *data)
  73 {
  74         if(width==0 || height==0)
  75                 throw invalid_operation("Texture2D::image");
  76
  77         unsigned w = width;
  78         unsigned h = height;
  79         get_level_size(level, w, h);
  80
  81         BindRestore _bind(this);
  82         glTexImage2D(target, level, ifmt, w, h, 0, fmt, type, data);
  83
  84         allocated |= 1<<level;
  85         if(gen_mipmap && level==0)
  86         {
  87                 auto_generate_mipmap();
  88                 for(; (w || h); w>>=1, h>>=1, ++level) ;
  89                 allocated |= (1<<level)-1;
  90         }
  91 }
  92
  93 void Texture2D::sub_image(unsigned level, int x, int y, unsigned wd, unsigned ht, PixelFormat fmt, DataType type, const void *data)
  94 {
  95         if(width==0 || height==0)
  96                 throw invalid_operation("Texture2D::sub_image");
  97
  98         allocate(level);
  99
 100         BindRestore _bind(this);
 101         glTexSubImage2D(target, level, x, y, wd, ht, fmt, type, data);
 102 }
 103
 104 void Texture2D::image(const Graphics::Image &img, bool srgb)
 105 {
 106         image(img, srgb, false);
 107 }
 108
 109 void Texture2D::image(const Graphics::Image &img, bool srgb, bool from_buffer)
 110 {
 111         unsigned w = img.get_width();
 112         unsigned h = img.get_height();
 113         PixelFormat fmt = pixelformat_from_graphics(img.get_format());
 114         if(width==0)
 115                 storage(storage_pixelformat_from_graphics(img.get_format(), srgb), w, h);
 116         else if(w!=width || h!=height)
 117                 throw incompatible_data("Texture2D::image");
 118
 119         PixelStore pstore = PixelStore::from_image(img);
 120         BindRestore _bind_ps(pstore);
 121
 122         image(0, fmt, UNSIGNED_BYTE, from_buffer ? 0 : img.get_data());
 123 }
 124
 125 void Texture2D::get_level_size(unsigned level, unsigned &w, unsigned &h)
 126 {
 127         w >>= level;
 128         h >>= level;
 129
 130         if(!w && h)
 131                 w = 1;
 132         else if(!h && w)
 133                 h = 1;
 134 }
 135
 136 Resource::AsyncLoader *Texture2D::load(IO::Seekable &io, const Resources *res)
 137 {
 138         AsyncLoader *ldr = new AsyncLoader(*this, io);
 139         if(res)
 140                 ldr->set_srgb_conversion(res->get_srgb_conversion());
 141         return ldr;
 142 }
 143
 144 UInt64 Texture2D::get_data_size() const
 145 {
 146         return id ? width*height*get_component_count(ifmt) : 0;
 147 }
 148
 149 void Texture2D::unload()
 150 {
 151         glDeleteTextures(1, &id);
 152         id = 0;
 153         // TODO check which params actually need refreshing
 154         dirty_params = -1;
 155 }
 156
 157
 158 Texture2D::Loader::Loader(Texture2D &t):
 159         DataFile::DerivedObjectLoader<Texture2D, Texture::Loader>(t)
 160 {
 161         init();
 162 }
 163
 164 Texture2D::Loader::Loader(Texture2D &t, Collection &c):
 165         DataFile::DerivedObjectLoader<Texture2D, Texture::Loader>(t, c)
 166 {
 167         init();
 168 }
 169
 170 void Texture2D::Loader::init()
 171 {
 172         add("raw_data", &Loader::raw_data);
 173         add("storage", &Loader::storage);
 174 }
 175
 176 void Texture2D::Loader::raw_data(const string &data)
 177 {
 178         obj.image(0, get_base_pixelformat(obj.ifmt), UNSIGNED_BYTE, data.data());
 179 }
 180
 181 void Texture2D::Loader::storage(PixelFormat fmt, unsigned w, unsigned h)
 182 {
 183         if(srgb)
 184                 fmt = get_srgb_pixelformat(fmt);
 185         obj.storage(fmt, w, h);
 186 }
 187
 188
 189 Texture2D::AsyncLoader::AsyncLoader(Texture2D &t, IO::Seekable &i):
 190         texture(t),
 191         io(i),
 192         srgb_conversion(false),
 193         pixel_buffer(PIXEL_UNPACK_BUFFER),
 194         mapped_address(0),
 195         phase(0)
 196 { }
 197
 198 void Texture2D::AsyncLoader::set_srgb_conversion(bool c)
 199 {
 200         srgb_conversion = c;
 201 }
 202
 203 bool Texture2D::AsyncLoader::needs_sync() const
 204 {
 205         return phase%2;
 206 }
 207
 208 bool Texture2D::AsyncLoader::process()
 209 {
 210         if(phase==0)
 211         {
 212                 /* TODO Enhance the ImageLoader system so that the image can be loaded
 213                 directly to the buffer */
 214                 image.load_io(io);
 215                 n_bytes = image.get_stride()*image.get_height();
 216         }
 217         else if(phase==1)
 218         {
 219                 pixel_buffer.data(n_bytes, 0);
 220                 mapped_address = reinterpret_cast<char *>(pixel_buffer.map(WRITE_ONLY));
 221         }
 222         else if(phase==2)
 223         {
 224                 const char *data = reinterpret_cast<const char *>(image.get_data());
 225                 copy(data, data+n_bytes, mapped_address);
 226         }
 227         else if(phase==3)
 228         {
 229                 Bind _bind_buf(pixel_buffer, PIXEL_UNPACK_BUFFER);
 230                 if(!pixel_buffer.unmap())
 231                 {
 232                         phase = 1;
 233                         return false;
 234                 }
 235
 236                 if(!texture.id)
 237                         glGenTextures(1, &texture.id);
 238                 texture.image(image, srgb_conversion, true);
 239         }
 240
 241         ++phase;
 242         return phase>3;
 243 }
 244
 245 } // namespace GL
 246 } // namespace Msp