+#include "imageloader.h"
+#include <msp/core/algorithm.h>
#include <msp/core/refptr.h>
#include <msp/io/file.h>
#include <msp/strings/format.h>
-#include "imageloader.h"
+#include <msp/strings/utils.h>
+#include "bmploader.h"
#ifdef WITH_LIBPNG
#include "png/pngloader.h"
#endif
+#ifdef WITH_LIBJPEG
+#include "jpeg/jpegloader.h"
+#endif
#ifdef WITH_DEVIL
#include "devil/devilloader.h"
#endif
+#ifdef WITH_QUARTZ
+#include "quartz/quartzloader.h"
+#endif
using namespace std;
namespace Msp {
namespace Graphics {
-ImageLoader::ImageLoader():
- source(0)
-{ }
-
ImageLoader::~ImageLoader()
{
delete source;
}
+bool ImageLoader::detect_signature(const string &sig)
+{
+ Registry ®istry = get_registry();
+ for(const RegisterBase *r: registry.loaders)
+ if(r->detect(sig))
+ return true;
+ return false;
+}
+
ImageLoader *ImageLoader::open_file(const string &fn)
{
try
{
- RefPtr<IO::File> file = new IO::BufferedFile(fn);
+ RefPtr<IO::BufferedFile> file = new IO::BufferedFile(fn);
ImageLoader *loader = open_io(*file);
loader->source = file.release();
return loader;
}
}
-ImageLoader *ImageLoader::open_io(IO::Base &io)
+ImageLoader *ImageLoader::open_io(IO::Seekable &io)
{
- char sig_buf[8];
- unsigned sig_len = io.read(sig_buf, sizeof(sig_buf));
- string sig(sig_buf, sig_len);
-
- // TODO register loader classes automatically
- ImageLoader *loader = 0;
- if(0)
- ;
-#ifdef WITH_LIBPNG
- else if(PngLoader::detect(sig))
- loader = new PngLoader(io, sig);
-#endif
-#ifdef WITH_DEVIL
- else if(DevilLoader::detect(sig))
- loader = new DevilLoader(io, sig);
-#endif
- else
+ Registry ®istry = get_registry();
+ if(registry.changed)
+ {
+ registry.changed = false;
+ sort(registry.loaders, [](RegisterBase *r1, RegisterBase *r2){
+ return r1->get_signature_size()<r2->get_signature_size();
+ });
+ }
+
+ if(registry.loaders.empty())
+ throw unsupported_image_format("no loaders");
+
+ string signature(registry.loaders.back()->get_signature_size(), 0);
+ unsigned sig_len = io.read(&signature[0], signature.size());
+
+ ImageLoader *loader = nullptr;
+ for(auto i=registry.loaders.begin(); (!loader && i!=registry.loaders.end()); ++i)
+ if((*i)->detect(signature))
+ loader = (*i)->create(io);
+
+ io.seek(0, IO::S_BEG);
+
+ if(!loader)
{
string sig_hex;
for(unsigned i=0; i<sig_len; ++i)
- {
- if(i)
- sig_hex += ' ';
- sig_hex += format("%02X", static_cast<unsigned char>(sig_buf[i]));
- }
+ append(sig_hex, " ", format("%02X", static_cast<unsigned char>(signature[i])));
throw unsupported_image_format(sig_hex);
}
return loader;
}
+void ImageLoader::load(Image::Data &data)
+{
+ if(state>=FINISHED)
+ throw logic_error("already loaded");
+
+ if(state<HEADERS_LOADED)
+ load_headers_(data);
+ if(!data.pixels)
+ data.pixels = data.owned_pixels = new char[data.stride*data.height];
+ load_pixels_(data);
+ state = FINISHED;
+}
+
+void ImageLoader::load_headers(Image::Data &data)
+{
+ if(state>=HEADERS_LOADED)
+ throw logic_error("headers already loaded");
+
+ load_headers_(data);
+ state = HEADERS_LOADED;
+}
+
+ImageLoader::Registry &ImageLoader::get_registry()
+{
+ static Registry registry;
+ static bool initialized = false;
+ if(!initialized)
+ {
+ initialized = true;
+ register_loader<BmpLoader>();
+#ifdef WITH_LIBPNG
+ register_loader<PngLoader>();
+#endif
+#ifdef WITH_LIBJPEG
+ register_loader<JpegLoader>();
+#endif
+#ifdef WITH_DEVIL
+ register_loader<DevilLoader>();
+#endif
+#ifdef WITH_QUARTZ
+ register_loader<QuartzLoader>();
+#endif
+ }
+ return registry;
+}
+
+
+ImageLoader::Registry::~Registry()
+{
+ for(auto l: loaders)
+ delete l;
+}
+
} // namespace Graphics
} // namespace Msp