/*
Copyright (c) 2016, Cryst.
All rights reserved.

Redistribution and use in source and binary forms, with or without modification, are permitted
provided that the following conditions are met:

    * Redistributions of source code must retain the above copyright notice, this list of
      conditions and the following disclaimer.
    * Redistributions in binary form must reproduce the above copyright notice, this list of
      conditions and the following disclaimer in the documentation and/or other materials
      provided with the distribution.
    * Neither the name of the Cryst nor the names of its contributors may be used to endorse or
      promote products derived from this software without specific prior written permission.

THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR
IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY
AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR
CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER
IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/
/*
Copyright (c)  2016, Cryst.
All rights reserved.

  ソースコード形式かバイナリ形式か、変更するかしないかを問わず、以下の条件を満たす場合に
  限り、再頒布および使用を許可します。

      * ソースコードを再頒布する場合、上記の著作権表示、本条件一覧、および 下記免責条項を
         含めること。
      * バイナリ形式で再頒布する場合、頒布物に付属のドキュメント等の資料に、上記の著作権
         表示、本条件一覧、および下記免責条項を含めること。
      * 書面による特別の許可なしに、本ソフトウェアから派生した製品の宣伝または販売促進に、
         Cryst の名前またはコントリビューターの名前を使用してはならない。

  本ソフトウェアは、著作権者およびコントリビューターによって「現状のまま」提供されており、
  明示黙示を問わず、商業的な使用可能性、および特定の目的に対する適合性に関する暗黙の保証
  も含め、またそれに限定されない、いかなる保証もありません。著作権者もコントリビューター
  も、事由のいかんを問わず、損害発生の原因いかんを問わず、かつ責任の根拠が契約であるか
  厳格責任であるか(過失その他の)不法行為であるかを問わず、仮にそのような損害が発生する
  可能性を知らされていたとしても、本ソフトウェアの使用によって発生した(代替品または代用
  サービスの調達、使用の喪失、データの喪失、利益の喪失、業務の中断も含め、またそれに限定
  されない)直接損害、間接損害、偶発的な損害、特別損害、懲罰的損害、または結果損害につい
  て、一切責任を負わないものとします。
*/
////////////////////////////////////////////////////////////////////////////////
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
/* OBJECT 形式のデータを画像化するプログラム
gcc -O3 -W -Wall -o object2bmp object2bmp.c
*/
#pragma pack(push,1)
typedef struct PIXEL { unsigned char B, G, R, A; } PIXEL ;
#pragma pack(pop)

typedef struct OBJECT { int value, x, y, lx, ly, group_id, random; } OBJECT;

extern int int2PseudoColor(int);
#define saveArray2bmp32(name,a,w,h) saveDataAsBMP32((name),(&(a)[0][0]),(w),(h))
#define FINALFNAMEIS(foname) #foname                  // "MyFunction.bmp"
#define EXTRACTING(foname)   FINALFNAMEIS(foname.bmp) // MyFunction.bmp
#define FNAMEIS(foname)      EXTRACTING(foname)       // IMAGEPROCESSING->MyFunction
#define DEFAULT_INPUT_FILE   "source.obj"
#define IMAGEPROCESSING      object2bmp

int IMAGEPROCESSING( OBJECT *o, int pseudocolor, int restoration, PIXEL **d, int lx, int ly )
{   int  **w=(int**)d, x, y, i, st=1;  // sizeof(int) MUST be same as sizeof(PIXEL).
    for( i=y=0; y<ly; y++ ) {
        for( x=0; x<lx; x++, i++ ) {
            if( ( o[i].lx != lx ) || ( o[i].ly != ly ) ) goto GOOD_BY;
            if( restoration ) { // オブジェクト中の座標情報を用いた復元処理
                if( o[i].x<0 || o[i].x>=lx || o[i].y<0 || o[i].y>=ly ) goto GOOD_BY;
                if( pseudocolor ) w[o[i].y][o[i].x] = int2PseudoColor(o[i].group_id);
                else {
                    int  n = 0xFF & o[i].value;
                    w[o[i].y][o[i].x] = n<<16 | n<<8 | n;
                }   // 通常作成時は、オブジェクト中の、value(濃淡データ:0-255)を画像化
            } else {// pseudocolor オプション指定時は、group_id を疑似カラー化し画像化
                if( pseudocolor ) w[y][x] = int2PseudoColor(o[i].group_id);
                else {
                    int  n = 0xFF & o[i].value;
                    w[y][x] = n<<16 | n<<8 | n;
                }
            } 
        }   // restoration 指定が無い時は、オブジェクトの順に 画像の横,縦に単純に配置
    }
    st = 0;
GOOD_BY:
    return st;
}
////////////////////////////////////////////////////////////////////////////////
// 以下 上記関数を実際に利用する為に必要な処理の記載部
#include <stdio.h>
#include <stdlib.h>
#include <sys/stat.h>
#pragma pack(push,2)
typedef struct HEADER {
    unsigned short bfType;          // 0x4D42 BMP identification 0x42='B', 0x4D='M'
    unsigned int   bfSize;          // ?? 
    unsigned short bfReserved1;     //  0 
    unsigned short bfReserved2;     //  0 
    unsigned int   bfOffBits;       // 54  sizeof(HEADER)
    /////////////////////////////////////////////////////////////////////////////
    unsigned int   biSize;          // 40 sizeof(biSize ~ biClrImportant)
    signed int     biWidth;         // lx
    signed int     biHeight;        // ly
    unsigned short biPlanes;        //  1
    unsigned short biBitCount;      // 32
    unsigned int   biCompression;   //  0
    unsigned int   biSizeImage;     // ??
    signed int     biXPelsPerMeter; //  0
    signed int     biYPelsPerMeter; //  0
    unsigned int   biClrUsed;       //  0
    unsigned int   biClrImportant;  //  0
    /////////////////////////////////////////////////////////////////////////////
    unsigned char  lut[];           //  Symbol name only, Size 0
} HEADER ;
#pragma pack(pop)

#define BMPID      0x4D42
#define BMPIFHSIZE 40

void **alloc2D( void *data, int s, int lx, int ly )
{
void **d, *base=data ;
long long i;

    if( (data==NULL) || (s<1) || (lx<1) || (ly<1) ) return NULL ;
    if( (d=(void **)malloc(ly*sizeof(void *))) == NULL ) return NULL ;
    for( i=0; i<ly; i++ ) d[i] = (void *)((unsigned char *)base + lx*s*i) ;
    return d ;
}

#define calloc2Darray alloc2Darray
void **alloc2Darray( int s, int lx, int ly )
{
void **d, *base ;
long long i;

    if( (s<1) || (lx<1) || (ly<1) ) return NULL ;
    if( (d=(void **)calloc(1,ly*sizeof(void *)+(lx*ly*s))) == NULL ) return NULL ;
    base = (unsigned char *)d + ly*sizeof(void *) ;
    for( i=0; i<ly; i++ ) d[i] = (void *)((unsigned char *)base + lx*s*i) ;
    return d ;
}

int int2PseudoColor(int v)
{// ラベルデータを、擬似カラーデータに変換します。 この変換は可逆変換なので、
///  擬似カラーデータから、元のラベルデータを逆算して求める事が出来ます。
unsigned int r=0, g=0, b=0 ;

    if( (v&0x00ffffff) == 0 ) return 0 ;
    if( v & 0x000001 ) b |= 0x80 ;
    if( v & 0x000002 ) g |= 0x80 ;
    if( v & 0x000004 ) r |= 0x80 ;
    if( v & 0x000008 ) b |= 0x40 ;
    if( v & 0x000010 ) g |= 0x40 ;
    if( v & 0x000020 ) r |= 0x40 ;
    if( v & 0x000040 ) b |= 0x20 ;
    if( v & 0x000080 ) g |= 0x20 ;
    if( v & 0x000100 ) r |= 0x20 ;
    if( v & 0x000200 ) b |= 0x10 ;
    if( v & 0x000400 ) g |= 0x10 ;
    if( v & 0x000800 ) r |= 0x10 ;
    if( v & 0x001000 ) b |= 0x08 ;
    if( v & 0x002000 ) g |= 0x08 ;
    if( v & 0x004000 ) r |= 0x08 ;
    if( v & 0x008000 ) b |= 0x04 ;
    if( v & 0x010000 ) g |= 0x04 ;
    if( v & 0x020000 ) r |= 0x04 ;
    if( v & 0x040000 ) b |= 0x02 ;
    if( v & 0x080000 ) g |= 0x02 ;
    if( v & 0x100000 ) r |= 0x02 ;
    if( v & 0x200000 ) b |= 0x01 ;
    if( v & 0x400000 ) g |= 0x01 ;
    if( v & 0x800000 ) r |= 0x01 ;
    return (r <<16) | (g << 8) | b ;
}

int saveDataAsBMP32( char *fname, void *b, int lx, int ly )
{// 処理結果を 32ビット形式の BMPファイルとして書き出す。
HEADER h={BMPID, lx*abs(ly)*sizeof(PIXEL)+sizeof(HEADER), 0, 0, sizeof(HEADER),
          BMPIFHSIZE, lx, ly, 1, 32, 0, lx*abs(ly)*sizeof(PIXEL), 0, 0, 0, 0} ;
FILE *fp ;

    if( (fp=fopen(fname, "wb")) == NULL ) {
        fprintf( stderr, "File '%s' cannot create.\n", fname );
        return -1;
    } else if ( fwrite( &h, sizeof(HEADER), 1, fp ) != 1 ) {
        fclose(fp) ; return -1;
    } else if ( fwrite( b, h.biSizeImage, 1, fp ) != 1 ) {
        fclose(fp) ; return -1;
    } else fclose(fp) ;
    return 0;
}

void argumentChecker( int argc, char *argv[], int *p, int *r, char **iname, char **oname )
{   int  i;
    for( i=1; i<argc; i++ ) {
        if( argv[i][0]=='-' ) {
            if( argv[i][2] ) goto GOOD_BY;
            switch(argv[i][1]) {
                case 'p': *p=1; break;
                case 'r': *r=1; break;
                default : goto GOOD_BY;
            }
        }
        else if( *iname==NULL ) *iname = argv[i];
        else if( *oname==NULL ) *oname = argv[i];
        else goto GOOD_BY;
    }
    return ;
GOOD_BY: // USAGE & EXIT
    fprintf( stderr, "\nusage: %s [-p] [-r] [ input_obj [ output_bmp ]]\n", argv[0]);
    fprintf( stderr, "\t\t -p: pseudocolor option\n");
    fprintf( stderr, "\t\t -r: restoration option\n");
    fprintf( stderr, "\t\t input_obj : name of a object file for input data.\n");
    fprintf( stderr, "\t\t\t\tdefault name: %s\n", DEFAULT_INPUT_FILE );
    fprintf( stderr, "\t\t output_bmp: name of a BMP file for output image.\n");
    fprintf( stderr, "\t\t\t\tdefault name: %s\n", FNAMEIS(IMAGEPROCESSING));
    exit(1);
}

int main(int argc, char *argv[])
{
struct stat fileInfo;
OBJECT  *source=NULL;
PIXEL   **destination=NULL;
FILE    *fp;
char    *ifname=NULL, *ofname=NULL;
int     pseudocolor=0, restoration=0 ;
int     lx, ly, size, st=1;
#define OBJECT_SIZE ((int)sizeof(OBJECT))

    argumentChecker( argc, argv, &pseudocolor, &restoration, &ifname, &ofname );
    if( ifname==NULL ) ifname = DEFAULT_INPUT_FILE;
    if( ofname==NULL ) ofname = FNAMEIS(IMAGEPROCESSING);

    if( (fp=fopen(ifname,"rb")) == NULL ){
        fprintf(stderr,"'%s' file open error.\n",ifname);   goto GOOD_BY;
    } else if( fstat(fileno(fp), &fileInfo) != 0 ) {
        perror("stat error.\n");                fclose(fp); goto GOOD_BY;
    } else if( (size=fileInfo.st_size)<OBJECT_SIZE || size%OBJECT_SIZE ) {
        perror("Bad object size.\n" );          fclose(fp); goto GOOD_BY;
    } else if( (source=(OBJECT *)malloc(fileInfo.st_size)) == NULL ) {
        perror("Couldn't allocate memory.\n" ); fclose(fp); goto GOOD_BY;
    } else if( fread(source, fileInfo.st_size, 1, fp) != 1 ) {
        perror("read error.\n");                fclose(fp); goto GOOD_BY;
    } else fclose(fp) ;
    
    size /= OBJECT_SIZE;
    lx = source[0].lx;
    ly = source[0].ly;
    if( lx<1 || ly<1 || lx*ly!=size ) {
        fprintf( stderr, "lx = %d, ly = %d, size = %d (%d)\n",lx,ly,size,lx*ly);
        goto GOOD_BY;
    } else if((destination=(PIXEL**)alloc2Darray(sizeof(PIXEL),lx,ly))==NULL) {
        fprintf( stderr, "Couldn't allocate memory. destination\n" );
        goto GOOD_BY;
    }
    st = IMAGEPROCESSING(source, pseudocolor, restoration, destination, lx, ly);
    if( st==0 ) saveArray2bmp32( ofname, destination, lx, ly) ;
    else if( st>0 ) fprintf(stderr,"Image processing is not completed.\n");
    
GOOD_BY: // 後始末とプログラムの終了
    free(destination) ;
    free(source) ;
    return st ;
}