source/graphics/imageloader.cpp

   1 #include <msp/core/refptr.h>
   2 #include <msp/io/file.h>
   3 #include <msp/strings/format.h>
   4 #include "bmploader.h"
   5 #include "imageloader.h"
   6 #ifdef WITH_LIBPNG
   7 #include "png/pngloader.h"
   8 #endif
   9 #ifdef WITH_LIBJPEG
  10 #include "jpeg/jpegloader.h"
  11 #endif
  12 #ifdef WITH_DEVIL
  13 #include "devil/devilloader.h"
  14 #endif
  15 #ifdef WITH_QUARTZ
  16 #include "quartz/quartzloader.h"
  17 #endif
  18
  19 using namespace std;
  20
  21 namespace Msp {
  22 namespace Graphics {
  23
  24 ImageLoader::ImageLoader():
  25         source(0)
  26 { }
  27
  28 ImageLoader::~ImageLoader()
  29 {
  30         delete source;
  31 }
  32
  33 ImageLoader *ImageLoader::open_file(const string &fn)
  34 {
  35         try
  36         {
  37                 RefPtr<IO::BufferedFile> file = new IO::BufferedFile(fn);
  38                 ImageLoader *loader = open_io(*file);
  39                 loader->source = file.release();
  40                 return loader;
  41         }
  42         catch(const unsupported_image_format &e)
  43         {
  44                 throw unsupported_image_format(format("%s: %s", fn, e.what()));
  45         }
  46 }
  47
  48 ImageLoader *ImageLoader::open_io(IO::Seekable &io)
  49 {
  50         Registry &registry = get_registry();
  51         if(registry.changed)
  52         {
  53                 registry.changed = false;
  54                 registry.loaders.sort(signature_size_compare);
  55         }
  56
  57         if(registry.loaders.empty())
  58                 throw unsupported_image_format("no loaders");
  59
  60         vector<char> sig_buf(registry.loaders.back()->get_signature_size());
  61         unsigned sig_len = io.read(&sig_buf[0], sig_buf.size());
  62         string signature(sig_buf.begin(), sig_buf.end());
  63
  64         ImageLoader *loader = 0;
  65         for(list<RegisterBase *>::const_iterator i=registry.loaders.begin(); (!loader && i!=registry.loaders.end()); ++i)
  66                 if((*i)->detect(signature))
  67                         loader = (*i)->create(io);
  68
  69         io.seek(0, IO::S_BEG);
  70
  71         if(!loader)
  72         {
  73                 string sig_hex;
  74                 for(unsigned i=0; i<sig_len; ++i)
  75                 {
  76                         if(i)
  77                                 sig_hex += ' ';
  78                         sig_hex += format("%02X", static_cast<unsigned char>(sig_buf[i]));
  79                 }
  80                 throw unsupported_image_format(sig_hex);
  81         }
  82
  83         return loader;
  84 }
  85
  86 ImageLoader::Registry &ImageLoader::get_registry()
  87 {
  88         static Registry registry;
  89         static bool initialized = false;
  90         if(!initialized)
  91         {
  92                 initialized = true;
  93                 register_loader<BmpLoader>();
  94 #ifdef WITH_LIBPNG
  95                 register_loader<PngLoader>();
  96 #endif
  97 #ifdef WITH_LIBJPEG
  98                 register_loader<JpegLoader>();
  99 #endif
 100 #ifdef WITH_DEVIL
 101                 register_loader<DevilLoader>();
 102 #endif
 103 #ifdef WITH_QUARTZ
 104                 register_loader<QuartzLoader>();
 105 #endif
 106         }
 107         return registry;
 108 }
 109
 110 bool ImageLoader::signature_size_compare(RegisterBase *r1, RegisterBase *r2)
 111 {
 112         return r1->get_signature_size()<r2->get_signature_size();
 113 }
 114
 115
 116 ImageLoader::Registry::Registry():
 117         changed(false)
 118 { }
 119
 120 ImageLoader::Registry::~Registry()
 121 {
 122         for(list<RegisterBase *>::iterator i=loaders.begin(); i!=loaders.end(); ++i)
 123                 delete *i;
 124 }
 125
 126 } // namespace Graphics
 127 } // namespace Msp