/* -*- c-basic-offset: 4; tab-width: 4; indent-tabs-mode: t -*- */
/* AbiWord SVG Graphic importer employing LibRSVG and GdkPixbuf
 * Copyright (C) 2002 Dom Lachowicz
 * Copyright (C) 2002 Martin Sevior (much of the base png from pixbuf code)
 *
 * Portions from GdkPixBuf Library 
 * Copyright (C) 1999 The Free Software Foundation
 * 
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.
 * 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 * 
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  
 * 02111-1307, USA.
 */

#include <string.h>
#include <glib.h>
#include <rsvg.h>
#include <png.h>

#include "ut_debugmsg.h"
#include "ut_types.h"
#include "xap_Module.h"
#include "ie_impGraphic.h"
#include "fg_GraphicRaster.h"

//------------------------------------------------------------------------------------

/*! 
 * This class will import any graphic type supported by gdk-pixbuf
 * into a png buffer for abiword.
 */

static void _write_png( png_structp png_ptr, 
		        png_bytep data, 
		        png_size_t length )
{
	UT_ByteBuf* bb = (UT_ByteBuf*) png_get_io_ptr(png_ptr);
	bb->append(data, length);
}


static void _write_flush(png_structp png_ptr) { } // Empty Fuction.

//------------------------------------------------------------------------------------

class ABI_EXPORT IE_RSVGGraphic : public IE_ImpGraphic
{
public:
	IE_RSVGGraphic()
	{
	}
	
	virtual ~IE_RSVGGraphic()
	{
	}

	void setXPM(bool b)
	{
		m_bIsXPM = b;
	}

	void _createPNGFromPixbuf(GdkPixbuf * pixbuf)
	{
	  int colorType;

	  //
	  // OK define the PNG header from the info in GdkPixbuf
	  //
	  
	  UT_uint32 width =  gdk_pixbuf_get_width(pixbuf);
	  UT_uint32 height = gdk_pixbuf_get_height(pixbuf);
	  int rowstride = gdk_pixbuf_get_rowstride(pixbuf);
	  guchar * pBuf = gdk_pixbuf_get_pixels(pixbuf);
	  colorType =  PNG_COLOR_TYPE_RGB;
	  if(gdk_pixbuf_get_has_alpha(pixbuf))
	    {
	      colorType =  PNG_COLOR_TYPE_RGB_ALPHA;
	    }

	  //
	  // Abi only has 8 bits per sample
	  //
	  int bitsPerSampleAbi = 8;

	  png_set_IHDR ( m_pPNG,  m_pPNGInfo,
			 width,			       
			 height,			       
			 bitsPerSampleAbi,
			 colorType,
			 PNG_INTERLACE_NONE,
			 PNG_COMPRESSION_TYPE_DEFAULT,
			 PNG_FILTER_TYPE_DEFAULT );
	  
	  png_write_info(m_pPNG, m_pPNGInfo);

	  UT_Byte* pngScanline = new UT_Byte[rowstride];

	  for(UT_uint32 i =0; i < height; i++)
	    {
	      memmove(pngScanline,pBuf,rowstride);
	      png_write_row(m_pPNG, pngScanline);
	      pBuf += rowstride;
	    }

	  DELETEP (pngScanline);		
	  png_write_end(m_pPNG, m_pPNGInfo);
	}

  /*!
   * Convert an image data buffer into PNG image buffer.
   */
  virtual UT_Error importGraphic(UT_ByteBuf * pBB, FG_Graphic ** ppfg)
  {
    GdkPixbuf * pixbuf = NULL;
    
    GError * err = g_error_new (G_FILE_ERROR, G_FILE_ERROR_NOENT, "foobar");
    
    RsvgHandle * rsvg = rsvg_handle_new ();
    rsvg_handle_write ( rsvg, (const guchar *)pBB->getPointer (0),
			(gsize)pBB->getLength (), &err ) ;

    rsvg_handle_close ( rsvg, &err ) ;
    rsvg_handle_free ( rsvg ) ;

    pixbuf = rsvg_handle_get_pixbuf ( rsvg ) ;

    g_error_free (err);

    if (!pixbuf)
      {
	return false;
      }
    
    gdk_pixbuf_ref (pixbuf);

    // Initialize stuff to create our PNG.
    UT_Error err =Initialize_PNG();
    if (err)
      {
	return err;
      }
    
    if (setjmp(m_pPNG->jmpbuf))
      {
	png_destroy_write_struct(&m_pPNG, &m_pPNGInfo);
	return UT_ERROR;
      }

    //
    // Build the png member variables.
    //
    _createPNGFromPixbuf(pixbuf);
    FG_GraphicRaster * pFGR = new FG_GraphicRaster();
    if(pFGR == NULL)
      {
	return UT_IE_NOMEMORY;
      }
    if(!pFGR->setRaster_PNG(m_pPngBB)) 
      {
	DELETEP(pFGR);
	
	return UT_IE_FAKETYPE;
      }
    
    *ppfg = (FG_Graphic *) pFGR;
    return UT_OK;
  }
  
  /*!
   * Convert the data contained in the file to into a PNG-based 
   * FG_Graphic type
   */
  UT_Error importGraphic(const char * szFilename, FG_Graphic ** ppfg)
  {
    //
    // Construct a pixbuf.
    //
    GError * err = g_error_new (G_FILE_ERROR, G_FILE_ERROR_NOENT, "foobar");
    GdkPixbuf * pixbuf = rsvg_pixbuf_from_file (szFilename, &err);
    g_error_free (err);

    if(pixbuf == NULL)
      {
	return UT_ERROR;
      }
    
    // Initialize stuff to create our PNG.
    UT_Error err =Initialize_PNG();
    if (err)
      {
	return err;
      }
    
    if (setjmp(m_pPNG->jmpbuf))
      {
	png_destroy_write_struct(&m_pPNG, &m_pPNGInfo);
	return UT_ERROR;
      }

    //
    // Build the png member variables.
    //
    _createPNGFromPixbuf(pixbuf);
    FG_GraphicRaster * pFGR = new FG_GraphicRaster();
    if(pFGR == NULL)
      {
	return UT_IE_NOMEMORY;
      }
    if(!pFGR->setRaster_PNG(m_pPngBB)) 
      {
	DELETEP(pFGR);
	
	return UT_IE_FAKETYPE;
      }
    
    *ppfg = (FG_Graphic *) pFGR;
    return UT_OK;
  }
  
  /*!
   * Convert an image byte buffer into a PNG byte buffer
   */
  UT_Error convertGraphic(UT_ByteBuf* pBB,
			  UT_ByteBuf** ppBB)
  {
    GdkPixbuf * pixbuf = NULL;
    
    GError * err = g_error_new (G_FILE_ERROR, G_FILE_ERROR_NOENT, "foobar");
    
    pixbuf = gdk_pixbuf_loader_get_pixbuf (ldr);

    if (!pixbuf)
      {	
	return false;
      }
    
    gdk_pixbuf_ref (pixbuf);

    g_error_free (err);

    // Initialize stuff to create our PNG.
    UT_Error err =Initialize_PNG();
    if (err)
      {
	return err;
      }
    
    if (setjmp(m_pPNG->jmpbuf))
      {
	png_destroy_write_struct(&m_pPNG, &m_pPNGInfo);
	return UT_ERROR;
      }
    //
    // Build the png member variables.
    //
    _createPNGFromPixbuf(pixbuf);
    *ppBB =  m_pPngBB;
    return UT_OK;
  }
  
private:

  // PNG structures used
  png_structp m_pPNG;				// libpng structure for the PNG Object
  png_infop   m_pPNGInfo;			// libpng structure for info on the PNG Object
  UT_ByteBuf*  m_pPngBB;				// pBB Converted to PNG File
  
  UT_Error	Initialize_PNG(void)
  {
    /* Set up png structures for writing */
    m_pPNG = png_create_write_struct( PNG_LIBPNG_VER_STRING, 
				      (void*) NULL,
				      NULL, 
				      NULL );
    if( m_pPNG == NULL )
      {
	return UT_ERROR;
      }
    
    m_pPNGInfo = png_create_info_struct(m_pPNG);
    if ( m_pPNGInfo == NULL )
      {
	png_destroy_write_struct(&m_pPNG, (png_infopp) NULL);
	return UT_ERROR;
      }
    
    /* Set error handling if you are using the setjmp/longjmp method (this is
     * the normal method of doing things with libpng).  REQUIRED unless you
     * set up your own error handlers in the png_create_read_struct() earlier.
     */
    if (setjmp(m_pPNG->jmpbuf))
      {
	/* Free all of the memory associated with the png_ptr and info_ptr */
	png_destroy_write_struct(&m_pPNG, &m_pPNGInfo);
	
	/* If we get here, we had a problem reading the file */
	return UT_ERROR;
      }
    m_pPngBB = new UT_ByteBuf;  /* Byte Buffer for Converted Data */
    
    /* Setting up the Data Writing Function */
    png_set_write_fn(m_pPNG, (void *)m_pPngBB, (png_rw_ptr)_write_png, (png_flush_ptr)_write_flush);
    
    return UT_OK;
  }
};

//------------------------------------------------------------------------------------

class ABI_EXPORT IE_RSVGGraphic_Sniffer : public IE_ImpGraphicSniffer
{
public:

  IE_RSVGGraphic_Sniffer() {}
  ~IE_RSVGGraphic_Sniffer() {}

  virtual UT_Confidence_t recognizeContents(const char * szBuf, UT_uint32 iNum)
  {
    // todo: make me better
    if ( strstr ( szBuf, "svg" ) != NULL )
      return UT_CONFIDENCE_PERFECT;
    return UT_CONFIDENCE_ZILCH;
  }

  virtual UT_Confidence_t recognizeSuffix(const char * szSuffix)
	{
	  if ( !UT_stricmp(szSuffix, ".svg") )
	    return UT_CONFIDENCE_PERFECT ;
	  return UT_CONFIDENCE_ZILCH;
	}

  bool getDlgLabels(const char ** pszDesc,
		    const char ** pszSuffixList,
		    IEGraphicFileType * ft)
	{
		// TODO add a more complete list of suffixes
		*pszDesc = "KSVG Images";
		*pszSuffixList = "*.svg"
		*ft = getType ();
		return true;
	}

  UT_Error constructImporter(IE_ImpGraphic **ppieg)
	{
	  *ppieg = new IE_RSVGGraphic();
	  if (*ppieg == NULL)
	    return UT_IE_NOMEMORY;
	  return UT_OK;
	}

} ;

// -----------------------------------------------------------------------

ABI_PLUGIN_DECLARE("AbiRSVG")


IE_RSVGGraphic_Sniffer * m_sniffer = 0;


ABI_FAR_CALL  extern "C"
int abi_plugin_register (XAP_ModuleInfo * mi)
{
  mi->name = "LibRSVG SVG image loader plugin";
    mi->desc = "This will enable AbiWord to read SVG files";
    mi->version = ABI_VERSION_STRING;
    mi->author = "Dom Lachowicz <cinamod@hotmail.com>";
    mi->usage = "No Usage";

    if (!m_sniffer)
    {
      // create a new sniffer
      m_sniffer = new IE_RSVGGraphic_Sniffer ();
	}
    else
	{
		// else just reference the one we've got
		m_sniffer->ref();
	}

    IE_ImpGraphic::registerImporter ( m_sniffer );

    return 1;
};

ABI_FAR_CALL extern "C" 
int abi_plugin_unregister (XAP_ModuleInfo * mi)
{
    mi->name = 0;
    mi->desc = 0;
    mi->version = 0;
    mi->author = 0;
    mi->usage = 0;

    if ( m_sniffer )
	{
		IE_ImpGraphic::unregisterImporter ( m_sniffer );
	}
    if (!m_sniffer->unref())
    {
		m_sniffer = 0;
	}

    return 1;
};


ABI_FAR_CALL extern "C"
int abi_plugin_supports_version (UT_uint32 major, UT_uint32 minor, UT_uint32 release)
{
  // true if >= 0.9.5
  if ( major >= 1 ) // >= 1.0 is true
    return 1;
  else if ( major == 0 && minor == 9 && release >= 5 )
    return 1;
  return 0;
};