X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Fgraphics%2Fimageloader.cpp;h=859a34c12894a1a0460b439e230a0fe9154d1385;hb=7302a061c57602203895b616bf54d96269c677c6;hp=6cdaef912e65062740c284559b3851c742bd102b;hpb=0466fc5fe92ae03189f45f8872fb070ef7022290;p=libs%2Fgui.git

diff --git a/source/graphics/imageloader.cpp b/source/graphics/imageloader.cpp
index 6cdaef9..859a34c 100644
--- a/source/graphics/imageloader.cpp
+++ b/source/graphics/imageloader.cpp
@@ -1,13 +1,21 @@
 #include <msp/core/refptr.h>
 #include <msp/io/file.h>
 #include <msp/strings/format.h>
+#include <msp/strings/utils.h>
+#include "bmploader.h"
 #include "imageloader.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;
 
@@ -15,7 +23,8 @@ namespace Msp {
 namespace Graphics {
 
 ImageLoader::ImageLoader():
-	source(0)
+	source(0),
+	state(INITIAL)
 { }
 
 ImageLoader::~ImageLoader()
@@ -23,6 +32,15 @@ ImageLoader::~ImageLoader()
 	delete source;
 }
 
+bool ImageLoader::detect_signature(const std::string &sig)
+{
+	Registry &registry = get_registry();
+	for(list<RegisterBase *>::const_iterator i=registry.loaders.begin(); i!=registry.loaders.end(); ++i)
+		if((*i)->detect(sig))
+			return true;
+	return false;
+}
+
 ImageLoader *ImageLoader::open_file(const string &fn)
 {
 	try
@@ -40,37 +58,98 @@ ImageLoader *ImageLoader::open_file(const string &fn)
 
 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);
-	io.seek(0, IO::S_BEG);
+	Registry &registry = get_registry();
+	if(registry.changed)
+	{
+		registry.changed = false;
+		registry.loaders.sort(signature_size_compare);
+	}
+
+	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());
 
-	// TODO register loader classes automatically
 	ImageLoader *loader = 0;
-	if(0)
-		;
-#ifdef WITH_LIBPNG
-	else if(PngLoader::detect(sig))
-		loader = new PngLoader(io);
-#endif
-#ifdef WITH_DEVIL
-	else if(DevilLoader::detect(sig))
-		loader = new DevilLoader(io);
-#endif
-	else
+	for(list<RegisterBase *>::const_iterator 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;
+}
+
+bool ImageLoader::signature_size_compare(RegisterBase *r1, RegisterBase *r2)
+{
+	return r1->get_signature_size()<r2->get_signature_size();
+}
+
+
+ImageLoader::Registry::Registry():
+	changed(false)
+{ }
+
+ImageLoader::Registry::~Registry()
+{
+	for(list<RegisterBase *>::iterator i=loaders.begin(); i!=loaders.end(); ++i)
+		delete *i;
+}
+
 } // namespace Graphics
 } // namespace Msp