source/graphics/imageloader.cpp

   1 #include <msp/core/refptr.h>
   2 #include <msp/io/file.h>
   3 #include <msp/strings/format.h>
   4 #include "imageloader.h"
   5 #ifdef WITH_LIBPNG
   6 #include "png/pngloader.h"
   7 #endif
   8 #ifdef WITH_DEVIL
   9 #include "devil/devilloader.h"
  10 #endif
  11
  12 using namespace std;
  13
  14 namespace Msp {
  15 namespace Graphics {
  16
  17 ImageLoader::ImageLoader():
  18         source(0)
  19 { }
  20
  21 ImageLoader::~ImageLoader()
  22 {
  23         delete source;
  24 }
  25
  26 ImageLoader *ImageLoader::open_file(const string &fn)
  27 {
  28         try
  29         {
  30                 RefPtr<IO::BufferedFile> file = new IO::BufferedFile(fn);
  31                 ImageLoader *loader = open_io(*file);
  32                 loader->source = file.release();
  33                 return loader;
  34         }
  35         catch(const unsupported_image_format &e)
  36         {
  37                 throw unsupported_image_format(format("%s: %s", fn, e.what()));
  38         }
  39 }
  40
  41 ImageLoader *ImageLoader::open_io(IO::Seekable &io)
  42 {
  43         char sig_buf[8];
  44         unsigned sig_len = io.read(sig_buf, sizeof(sig_buf));
  45         string sig(sig_buf, sig_len);
  46         io.seek(0, IO::S_BEG);
  47
  48         // TODO register loader classes automatically
  49         ImageLoader *loader = 0;
  50         if(0)
  51                 ;
  52 #ifdef WITH_LIBPNG
  53         else if(PngLoader::detect(sig))
  54                 loader = new PngLoader(io);
  55 #endif
  56 #ifdef WITH_DEVIL
  57         else if(DevilLoader::detect(sig))
  58                 loader = new DevilLoader(io);
  59 #endif
  60         else
  61         {
  62                 string sig_hex;
  63                 for(unsigned i=0; i<sig_len; ++i)
  64                 {
  65                         if(i)
  66                                 sig_hex += ' ';
  67                         sig_hex += format("%02X", static_cast<unsigned char>(sig_buf[i]));
  68                 }
  69                 throw unsupported_image_format(sig_hex);
  70         }
  71
  72         return loader;
  73 }
  74
  75 } // namespace Graphics
  76 } // namespace Msp