/*
 * Custom widget Demo
 * 
 *
 * File         :   img2res.cpp
 * Purpose      :   Convert severals GIF images into a C++ resource file
 * Author       :   JLV
 * Creation     :   10/19/04
 * Last Update  :   11/01/04
 *
 * (C) hexaZen 2004
 *
 * This file is part of a demo application for Zinzala. It can be
 * used, distributed and modified without limitation.
 *
 */

#include <stdio.h>
#include <string.h>

#include <Cincinella/File.h>
#include <Cincinella/StrTools.h>
#include <Farfalla/Bitmap.h>

#include <photon/imaging/PiIo.h>

using namespace zSDK::Cincinella;
using namespace zSDK::Farfalla;

/*
 * Method       : LoadBitmap
 * Purpose      : Load a given image
 * Inputs       : 
 *
 * const tChar *aFilename, filename of the image to load
 *
 * Output       : PhImage_t *, NULL if failed
 * Side Effects : none
 */
cBitmap *LoadBitmap(const tChar *aFilename)
{
   PhImage_t*     lImage = NULL;
   PiLoadInfo_t   lInfo;
   PiIoHandler_t* lHandler;
   const tChar*   lExt;
   tUint16        lPos;

   // extract extension of the image
   if(sStrTools::FindLast(aFilename,'.',&lPos))
   {
      // get the handler for the image format
      lExt     = &aFilename[lPos+1];
      lHandler = PiIoGetHandlerByExt(lExt);

      if(lHandler)
      {
         memset(&lInfo,NULL,sizeof(PiLoadInfo_t));

         lInfo.flags             = Pi_IO_SHMEM;
         lInfo.shmem_threshold   = 100;
         lInfo.handler           = lHandler;
         lInfo.filename          = aFilename;

         // load the image
         lImage = PiLoadImage(&lInfo);
         if(lImage)
         {
            cBitmap* lBitmap;

            // create a cBitmap and return it
            lBitmap = new cBitmap(lImage);
            if(!cBitmap::VerifyD(lBitmap))
               return lBitmap;
            else
               return NULL;
         }
         else
            return NULL;
      }
      else
         return NULL;
   }
   else
      return NULL;
}

/*
 * Function     : DumpBitmapToFile
 * Purpose      : Dump a given bitmap into a .CPP & .H
 * Inputs       : 
 *
 * cFile &aHeader,      open header file
 * cFile &aCode,        open code file
 * cBitmap *aBitmap,    bitmap to dump in the files
 * tColor *aPalette,    bitmap's palette of colors
 * tUint16 aColors,     number of colors in the palette
 * const tChar *aName,  resource name to be used for this bitmap
 *
 * Output       : none
 * Side Effects : none
 */
void DumpBitmapToFile(cFile &aHeader,cFile &aCode,cBitmap *aBitmap,tColor *aPalette,
                      tUint16 aColors,const char *aName)
{
   tUint16     lWidth,lHeight;
   tUint8      lIndex;
   tUint32     lLength;
   tUint8*     lBits;
   PgColor_t   lColor;

   aBitmap->GetTransparent(lIndex);
   aBitmap->GetSize(lWidth,lHeight);
   lLength  = aBitmap->BitsLength();
   lBits    = aBitmap->GetBits();

   aHeader.Write("extern const tBitmapResource kRscBitmap%s;\n\n",aName);

   aCode.Write("tUint8 kBitmap%sBits [] = {\n\n\t",aName);

   for(tUint16 k=0;k<lLength;k++)
   {
      lIndex = lBits[k];
	
      if(k + 1 == lLength)
         aCode.Write("0x%02x",lIndex);
      else
         aCode.Write("0x%02x,",lIndex);

      if(!((1+k) % 16))
         aCode.Write("\n\t");
   }

   aCode.Write("};\n\n");

   aCode.Write("tUint32 kBitmap%sPalette[%d] = {\n\n\t",aName,aColors);

   for(tUint16 i=0;i<aColors;i++)
   {
      lColor = PgARGB(aPalette[i].iAlpha,aPalette[i].iRed,aPalette[i].iGreen,aPalette[i].iBlue);
      aCode.Write("0x%06x,",(tUint32)lColor);
      if(!((1+i) % 8))
         aCode.Write("\n\t");
   }

   aCode.Write("\n};\n\n");

   aCode.Write("const tBitmapResource kRscBitmap%s\t= {\n",aName);
   aCode.Write("\t%d,%d,0x%02x,%d,%d,kBitmap%sBits,kBitmap%sPalette\n};\n",lWidth,lHeight,lIndex,
               aColors,lLength,aName,aName);
}

int main(int argc,char **argv)
{
   tColor   lPalette[256];
   tChar    lFilename[256];
   cBitmap* lBitmap;
   cFile    lHeader;
   cFile    lCode;
   tChar*   lName;
   tErr     lErr;

   if(argc<3)
   {
      printf("Usage : <filename> (<image> <label>)+\n");
      return 1;
   }

   lName = argv[1];

   // create header file
   sprintf(lFilename,"%s.h",lName);
   lErr = lHeader.Open(lFilename,kFileWrite | kFileTrunc);
   if(lErr)
      return 1;
	
   // create code file
   sprintf(lFilename,"%s.cpp",lName);
   lErr = lCode.Open(lFilename,kFileWrite | kFileTrunc);
   if(lErr)
   {
      lHeader.Close();
      return 1;
   }	

   lHeader.Write("#ifndef _RES_%s\n",lName);
   lHeader.Write("#define _RES_%s\n\n",lName);
   lHeader.Write("#include <Cincinella/Types.h>\n\n");
   lHeader.Write("using namespace zSDK::Cincinella;\n\n");
   lHeader.Write("typedef struct tBitmapResource {\n");
   lHeader.Write("\ttUint32  iWidth;\n");
   lHeader.Write("\ttUint32  iHeight;\n");
   lHeader.Write("\ttUint32  iIndex;\n");
   lHeader.Write("\ttUint16  iColors;\n");
   lHeader.Write("\ttUint32  iLength;\n");
   lHeader.Write("\ttUint8*  iBits;\n");
   lHeader.Write("\ttUint32* iPalette;\n");
   lHeader.Write("} tBitmapResource;\n\n");

   lCode.Write("#include <Cincinella/Types.h>\n\n");
   lCode.Write("#include <%s.h>\n\n",lName);
   lCode.Write("using namespace zSDK::Cincinella;\n\n");

   for(int i=2;i<argc;i+=2)
   {
      printf("loading %s ...\n",argv[i]);
      lBitmap = LoadBitmap(argv[i]);
      if(lBitmap)
      {
         if(lBitmap->GetColorSpace() == eSpaceColor8Bit)
         {
            tUint16 lWidth,lHeight;
            tUint16 lColors = lBitmap->CountColors();

            lBitmap->GetSize(lWidth,lHeight);
	
            printf("bitmap is %dx%d pixels and have %d colors\n",lWidth,lHeight,lColors);

            if(!lBitmap->GetPalette(lPalette))
            {
               DumpBitmapToFile(lHeader,lCode,lBitmap,lPalette,lColors,argv[i+1]);
               printf("dumped\n");
            }
            else
               printf("failed to get the colors palette\n");
         }
         else
            printf("sorry this isn't a palette based image!\n");
      }
      else
         printf("failed\n");	
   }

   // fill files with closing stuff
   lHeader.Write("\n");
   lHeader.Write("#endif\n");

   // and close them
   lHeader.Close();
   lCode.Close();

   return 0;
}