/*
 * \file xformsGImage.C
 * Copyright 2002 the LyX Team
 * Read the file COPYING
 *
 * \author Angus Leeming, a.leeming@ic.ac.uk
 */

#include <config.h>

#ifdef __GNUG__
#pragma implementation
#endif

#include "xformsGImage.h"
#include "graphics/GraphicsParams.h"
#include "LColor.h"
#include "converter.h"              // formats
#include "debug.h"
#include "frontends/GUIRunTime.h"   // x11Display, x11Screen
#include "support/LAssert.h"
#include "support/lyxfunctional.h"  // compare_memfun

using std::find_if;

namespace {

void init_graphics();

unsigned int packedcolor(LColor::color c);

} // namespace anon


namespace grfx {

/// Access to this class is through this static method.
ImagePtr xformsGImage::newImage()
{
	init_graphics();

	ImagePtr ptr;
	ptr.reset(new xformsGImage());
	return ptr;
}


/// Return the list of loadable formats.
GImage::FormatList xformsGImage::loadableFormats()
{
	static FormatList fmts;
	if (!fmts.empty())
		return fmts;

	init_graphics();

	// The formats recognised by LyX
	Formats::const_iterator begin = formats.begin();
	Formats::const_iterator end   = formats.end();

	lyxerr[Debug::GRAPHICS]
		<< "\nThe image loader can load the following directly:\n";

	// Don't forget the Fortran numbering used by xforms!
	int const nformats = flimage_get_number_of_formats();
	for (int i = 1; i <= nformats; ++i) {

		FLIMAGE_FORMAT_INFO const * info = flimage_get_format_info(i);
		string const formal_name =
			info->formal_name ? info->formal_name : string();
		string ext =
			info->extension   ? info->extension   : string();

		if (ext.empty() || ext == "gz")
			continue;

		if (ext == "rgb") ext = "sgi";

		lyxerr[Debug::GRAPHICS]
			<< formal_name << ", extension \"" << ext << "\"\n";

		Formats::const_iterator it =
			find_if(begin, end,
				lyx::compare_memfun(&Format::extension, ext));
		if (it != end)
			fmts.push_back(it->name());
	}

	lyxerr[Debug::GRAPHICS]
		<< "\nOf these, LyX recognises the following formats:\n";

	FormatList::const_iterator fbegin = fmts.begin();
	FormatList::const_iterator fend   = fmts.end();
	for (FormatList::const_iterator fit = fbegin; fit != fend; ++fit) {
		if (fit != fbegin)
			lyxerr[Debug::GRAPHICS] << ", ";
		lyxerr[Debug::GRAPHICS] << *fit;
	}
	lyxerr[Debug::GRAPHICS] << '\n' << std::endl;

	return fmts;
}


xformsGImage::xformsGImage()
	: image_(0),
	  pixmap_(0),
	  pixmap_status_(PIXMAP_UNINITIALISED)
{}


xformsGImage::xformsGImage(xformsGImage const & other)
	: GImage(other),
	  image_(0),
	  pixmap_(0),
	  pixmap_status_(PIXMAP_UNINITIALISED)
{
	if (other.image_) {
		image_ = flimage_dup(other.image_);
		image_->u_vdata = this;
	}
}


xformsGImage::~xformsGImage()
{
	if (image_)
		flimage_free(image_);
	if (pixmap_)
		XFreePixmap(GUIRunTime::x11Display(), pixmap_);
}


GImage * xformsGImage::clone() const
{
	return new xformsGImage(*this);
}


unsigned int xformsGImage::getWidth() const
{
	if (!image_)
		return 0;
	return image_->w;
}


unsigned int xformsGImage::getHeight() const
{
	if (!image_)
		return 0;
	return image_->h;
}


Pixmap xformsGImage::getPixmap() const
{
	if (!pixmap_status_ == PIXMAP_SUCCESS)
		return 0;
	return pixmap_;
}


void xformsGImage::load(string const & filename, SignalTypePtr on_finish)
{
	if (image_) {
		lyxerr[Debug::GRAPHICS]
			<< "Image is loaded already!" << std::endl;
		on_finish->emit(false);
		return;
	}

	image_ = flimage_open(filename.c_str());
	if (!image_) {
		lyxerr[Debug::GRAPHICS]
			<< "Unable to open image" << std::endl;
		on_finish->emit(false);
		return;
	}

	// Store the Signal that will be emitted once the image is loaded.
	on_finish_ = on_finish;

	// Set this now and we won't need to bother again.
	image_->fill_color = packedcolor(LColor::graphicsbg);

	// Used by the callback routines to return to this
	image_->u_vdata = this;

	// Begin the reading process.
	flimage_read(image_);
}


bool xformsGImage::setPixmap(GParams const & params)
{
	if (!image_ || params.display == GParams::NONE)
		return false;

	Display * display = GUIRunTime::x11Display();

	if (pixmap_ && pixmap_status_ == PIXMAP_SUCCESS)
		XFreePixmap(display, pixmap_);

	int color_key;
	switch (params.display) {
	case GParams::MONOCHROME:
		color_key = FL_IMAGE_MONO;
		break;
	case GParams::GRAYSCALE:
		color_key = FL_IMAGE_GRAY;
		break;
	case GParams::COLOR:
	default: // NONE cannot happen!
		color_key = FL_IMAGE_RGB;
		break;
	}

	if (color_key != FL_IMAGE_RGB) {
		flimage_convert(image_, color_key, 0);
	}

	unsigned int fill = packedcolor(LColor::graphicsbg);
	if (fill != image_->fill_color) {
		// the background color has changed.
		// Note that in grayscale/monochrome images the background is
		// grayed also, so this call will have no visible effect. Sorry!
		flimage_replace_pixel(image_, image_->fill_color, fill);
		image_->fill_color = fill;
	}

	image_->xdisplay = display;
	Screen * screen  = ScreenOfDisplay(display, GUIRunTime::x11Screen());

	pixmap_ = flimage_to_pixmap(image_, XRootWindowOfScreen(screen));
	pixmap_status_ = pixmap_ ? PIXMAP_SUCCESS : PIXMAP_FAILED;

	return pixmap_status_ == PIXMAP_SUCCESS;
}


void xformsGImage::clip(GParams const & params)
{
	if (!image_)
		return;

	if (params.bb.empty())
		// No clipping is necessary.
		return;

	int const new_width  = params.bb.xr - params.bb.xl;
	int const new_height = params.bb.yt - params.bb.yb;

	// No need to check if the width, height are > 0 because the
	// Bounding Box would be empty() in this case.
	if (new_width > image_->w || new_height > image_->h)
		// Bounds are invalid.
		return;

	if (new_width == image_->w && new_height == image_->h)
		// Bounds are unchanged.
		return;

	int const xoffset_l = params.bb.xl;
	int const xoffset_r = image_->w - params.bb.xr;
	int const yoffset_t = image_->h - params.bb.yt;
	int const yoffset_b = params.bb.yb;

	flimage_crop(image_, xoffset_l, yoffset_t, xoffset_r, yoffset_b);
}


void xformsGImage::rotate(GParams const & params)
{
	if (!image_)
		return ;

	if (!params.angle)
		// No rotation is necessary.
		return;

	// The angle passed to flimage_rotate is the angle in one-tenth of a
	// degree units.
	flimage_rotate(image_, params.angle * 10, FLIMAGE_SUBPIXEL);
}


void xformsGImage::scale(GParams const & params)
{
	if (!image_)
		return;

	// boost::tie produces horrible compilation errors on my machine
	// Angus 25 Feb 2002
	std::pair<unsigned int, unsigned int> d = getScaledDimensions(params);
	unsigned int const width  = d.first;
	unsigned int const height = d.second;

	if (width == getWidth() && height == getHeight())
		// No scaling needed
		return;

	flimage_scale(image_, width, height, FLIMAGE_SUBPIXEL);
}


void xformsGImage::statusCB(string const & status_message)
{
	if (status_message.empty() || !on_finish_.get())
		return;

	if (prefixIs(status_message, "Done Reading")) {
		if (image_) {
			flimage_close(image_);
		}

		if (on_finish_.get()) {
			on_finish_->emit(true);
			on_finish_.reset();
		}
	}
}


void xformsGImage::errorCB(string const & error_message)
{
	if (error_message.empty() || !on_finish_.get())
		return;

	if (image_) {
		flimage_close(image_);
	}

	if (on_finish_.get()) {
		on_finish_->emit(false);
		on_finish_.reset();
	}
}

} // namespace grfx


namespace {

extern "C" {

int status_report(FL_IMAGE * ob, const char *s)
{
	lyx::Assert(ob && ob->u_vdata);

	string const str = s ? strip(s) : string();
	if (str.empty())
		return 0;

	lyxerr[Debug::GRAPHICS]
		<< "xforms image loader. Status : " << str << std::endl;

	grfx::xformsGImage * ptr =
		static_cast<grfx::xformsGImage *>(ob->u_vdata);
	ptr->statusCB(str);

	return 0;
}


static void error_report(FL_IMAGE * ob, const char *s)
{
	lyx::Assert(ob && ob->u_vdata);

	string const str = s ? strip(s) : string();
	if (str.empty())
		return;

	lyxerr[Debug::GRAPHICS]
		<< "xforms image loader. Error : " << str << std::endl;

	grfx::xformsGImage * ptr =
		static_cast<grfx::xformsGImage *>(ob->u_vdata);
	ptr->errorCB(str);
}

} // extern "C"


void init_graphics()
{
	// Paranoia check
	static bool initialised = false;
	if (initialised)
		return;
	initialised = true;

	flimage_enable_bmp();
	flimage_enable_fits();
	flimage_enable_gif();
	flimage_enable_jpeg();

	// xforms itself uses pngtopnm to convert to a loadable format.
	// We prefer to use our own conversion mechanism, therefore.
	// flimage_enable_png();

	flimage_enable_pnm();

#ifdef HAVE_FLIMAGE_ENABLE_PS
	// xforms recognises PS but not EPS
	flimage_enable_ps();
#endif

	flimage_enable_sgi();
	flimage_enable_tiff();
	flimage_enable_xbm();
	flimage_enable_xwd();
	flimage_enable_xpm();

	// xforms stores this permanently (does not make a copy) so
	// this should never be destroyed.
	static FLIMAGE_SETUP setup;
	setup.visual_cue    = status_report;
	setup.error_message = error_report;
	flimage_setup(&setup);
}


unsigned int packedcolor(LColor::color c)
{
	string const x11color = lcolor.getX11Name(c);

	Display * display = GUIRunTime::x11Display();
	Colormap cmap     = GUIRunTime::x11Colormap();
	XColor xcol;
	XColor ccol;
	if (XLookupColor(display, cmap, x11color.c_str(), &xcol, &ccol) == 0)
		// Unable to parse x11color.
		return FL_PACK(255,255,255);

	// Note that X stores the RGB values in the range 0 - 65535
	// whilst we require them in the range 0 - 255.
	unsigned int const r = xcol.red   / 256;
	unsigned int const g = xcol.green / 256;
	unsigned int const b = xcol.blue  / 256;

	return FL_PACK(r, g, b);
}

} // namespace anon