/*
 *  Atpdec
 *  Copyright (c) 2003 by Thierry Leconte (F4DWV)
 *
 *      $Id$
 *
 *   This library is free software; you can redistribute it and/or modify
 *   it under the terms of the GNU Library 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 Library General Public License for more details.
 *
 *   You should have received a copy of the GNU Library General Public
 *   License along with this library; if not, write to the Free Software
 *   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 *
 */

#include <stdio.h>
#ifdef WIN32
#include "w32util.h"
#else
#include <libgen.h>
#endif
#include <string.h>
#include <sndfile.h>
#include <png.h>

#include "version.h"

extern getpixelrow(float *pixelv);

#define CHA_OFFSET  86
#define CHB_OFFSET  1126	 
#define CH_WIDTH  909	 
#define IMG_WIDTH  2080

static SNDFILE *inwav;
static int initsnd(char *filename)
{
SF_INFO infwav;

/* open wav input file */
infwav.format=0;
inwav=sf_open(filename,SFM_READ,&infwav);
if (inwav==NULL) {
	fprintf(stderr,"could not open %s\n",filename);
	return(1);
}
if(infwav.samplerate !=11025) {
	fprintf(stderr,"Bad Input File sample rate: %d. Must be 11025\n",infwav.samplerate);
	return(1);
}
if(infwav.channels !=1) {
	fprintf(stderr,"Too many channels in input file : %d\n",infwav.channels);
	return(1);
}

return(0);
}

int getsample(float *sample,int nb)
{
return(sf_read_float(inwav,sample,nb));
}

png_text text_ptr[]={
{ PNG_TEXT_COMPRESSION_NONE, "Software", version ,sizeof(version) },
{ PNG_TEXT_COMPRESSION_NONE, "Channel", NULL ,0 },
{ PNG_TEXT_COMPRESSION_NONE, "Description", "NOAA POES satellite Image" ,25 }
};
static int ImageOut(char *filename,char* chid,float **prow,int nrow,int depth,int width,int offset)
{
FILE *pngfile;
png_infop info_ptr;
png_structp png_ptr;
int n;

/* init png lib */
png_ptr = png_create_write_struct(PNG_LIBPNG_VER_STRING, NULL, NULL, NULL);
if (!png_ptr) {
	fprintf(stderr,"could not open create png_ptr\n");
	return(1);
}

info_ptr = png_create_info_struct(png_ptr);
    if (!info_ptr) {
        png_destroy_write_struct(&png_ptr, (png_infopp)NULL);
	fprintf(stderr,"could not open create info_ptr\n");
	return(1);
    }

png_set_IHDR(png_ptr, info_ptr, width, nrow,
       depth, PNG_COLOR_TYPE_GRAY, PNG_INTERLACE_NONE,
       PNG_COMPRESSION_TYPE_DEFAULT, PNG_FILTER_TYPE_DEFAULT);

text_ptr[1].text=chid;
text_ptr[1].text_length=strlen(chid);
png_set_text(png_ptr, info_ptr, text_ptr, 3);
png_set_pHYs(png_ptr, info_ptr, 4000,4000,PNG_RESOLUTION_METER);

printf("Writing %s ... ",filename);fflush(stdout);
pngfile=fopen(filename,"w");
if (pngfile==NULL) {
	fprintf(stderr,"could not open %s\n",filename);
	return(1);
}
png_init_io(png_ptr,pngfile);
png_write_info(png_ptr,info_ptr);

for(n=0;n<nrow;n++) {
	float *pixelv;
	png_byte pixel[2*IMG_WIDTH];
	int	i;

	pixelv=prow[n];
	for(i=0;i<width;i++) {
		float pv;

		pv=pixelv[i+offset];
		switch(depth) {
		case 8 :
			pixel[i]=(png_byte)pv;
			break;
		case 16 :
			((unsigned short*)pixel)[i]=htons((unsigned short)(pv*255.0));
			break;
		}
	}
	png_write_row(png_ptr,pixel);
}
png_write_end(png_ptr,info_ptr);
fclose(pngfile);
printf("Done\n");
png_destroy_write_struct(&png_ptr,&info_ptr);
return(0);
} 

int ImageColorOut(char *filename,float **prow,int nrow)
{
FILE *pngfile;
png_infop info_ptr;
png_structp png_ptr;
int n;

extern void false_color(double v, double t, float *r, float *g, float *b);

/* init png lib */
png_ptr = png_create_write_struct(PNG_LIBPNG_VER_STRING, NULL, NULL, NULL);
if (!png_ptr) {
	fprintf(stderr,"could not open create png_ptr\n");
	return(1);
}

info_ptr = png_create_info_struct(png_ptr);
    if (!info_ptr) {
        png_destroy_write_struct(&png_ptr, (png_infopp)NULL);
	fprintf(stderr,"could not open create info_ptr\n");
	return(1);
    }

png_set_IHDR(png_ptr, info_ptr, CH_WIDTH, nrow,
       8, PNG_COLOR_TYPE_RGB, PNG_INTERLACE_NONE,
       PNG_COMPRESSION_TYPE_DEFAULT, PNG_FILTER_TYPE_DEFAULT);

text_ptr[1].text="False Colors";
text_ptr[1].text_length=strlen(text_ptr[1].text);
png_set_text(png_ptr, info_ptr, text_ptr, 3);
png_set_pHYs(png_ptr, info_ptr, 4000,4000,PNG_RESOLUTION_METER);

printf("Writing %s ...",filename);fflush(stdout);
pngfile=fopen(filename,"w");
if (pngfile==NULL) {
	fprintf(stderr,"could not open %s\n",filename);
	return(1);
}
png_init_io(png_ptr,pngfile);
png_write_info(png_ptr,info_ptr);

for(n=0;n<nrow;n++) {
	float *pixelv;
	png_color pixel[CH_WIDTH];
	int	i;

	pixelv=prow[n];
	for(i=0;i<CH_WIDTH;i++) {
		double v,t;
		float r,g,b;

		v=pixelv[i+CHA_OFFSET];
		t=pixelv[i+CHB_OFFSET];
		falsecolor(v,t,&r,&g,&b);
		pixel[i].red=( unsigned int)(255.0*r);
		pixel[i].green=( unsigned int)(255.0*g);
		pixel[i].blue=( unsigned int)(255.0*b);
	}
	png_write_row(png_ptr,(png_bytep)pixel);
}
png_write_end(png_ptr,info_ptr);
fclose(pngfile);
printf("Done\n");
png_destroy_write_struct(&png_ptr,&info_ptr);
return(0);
} 

static int Distrib(char *filename,float **prow,int nrow)
{
unsigned int distrib[256][256];
int n;
int x,y;
int max=0;
FILE *df;

for(y=0;y<256;y++)
	for(x=0;x<256;x++)
                distrib[y][x]=0;

for(n=0;n<nrow;n++) {
        float *pixelv;
        int     i;

        pixelv=prow[n];
        for(i=0;i<909;i++) {

                y=(int)(pixelv[i+CHA_OFFSET]);
                x=(int)(pixelv[i+CHB_OFFSET]);
                distrib[y][x]+=1;
		if(distrib[y][x]> max) max=distrib[y][x];
        }
}
df=fopen(filename,"w");

printf("Writing %s\n",filename);fflush(stdout);

fprintf(df,"P2\n#max %d\n",max);
fprintf(df,"256 256\n255\n");
for(y=0;y<256;y++)
	for(x=0;x<256;x++)
		fprintf(df,"%d\n",(int)((255.0*(double)(distrib[y][x]))/(double)max));
fclose(df);
}

extern int Calibrate(float **prow,int nrow,int offset);
extern int Temperature(float **prow,int nrow,int ch, int offset);
extern void readfconf(char *file);
extern int optind,opterr;
extern char *optarg;
int satnum=1;

static void usage(void)
{
	fprintf(stderr,"atpdec [options] soundfiles ...\n");
	fprintf(stderr,"options:\n-d <dir>\tDestination directory\n-i [a|b|c|t]\tOutput image type r: Raw a: A chan. b: B chan. c: False color t: Temperature\n-c <file>\tFalse color config file\n-p\t\t16bits output\n-s [0|1]\tSatellite number\n");
}

main(int argc, char **argv)
{
char pngfilename[1024];
char name[1024];
char pngdirname[1024]="";
char imgopt[20]="ac";
float *prow[3000];
char *chid[6]={ "1","2","3A","4","5","3B"};
int depth=8;
int n,nrow;
int ch;
int c;

printf("%s\n",version);

opterr=0;
while ((c=getopt(argc,argv,"c:d:i:ps:"))!=EOF) {
	switch(c) {
		case 'd':
			strcpy(pngdirname,optarg);
			break;
		case 'c':
			readfconf(optarg);	
			break;
		case 'i':
			strcpy(imgopt,optarg);
			break;
		case 'p':
			depth=16;
			break;
		case 's':
			satnum=atoi(optarg);
			if(satnum<0 || satnum > 1) {
				fprintf(stderr,"invalid satellite\n");
				exit(1);
			}
			break;
		default:
			usage();
	}
}

for(nrow=0;nrow<3000;nrow++) prow[nrow]=NULL;

for (;optind<argc;optind++) {
int a=0,b=0;

strcpy(pngfilename,argv[optind]);
strcpy(name,basename(pngfilename));
strtok(name,".");
if(pngdirname[0]=='\0') {
	strcpy(pngfilename,argv[optind]);
	strcpy(pngdirname,dirname(pngfilename));
}

/* open snd input */
if(initsnd(argv[optind]))
	exit(1);

/* main loop */
printf("Decoding: %s \n",argv[optind]);
for(nrow=0;nrow<3000;nrow++) {
	if(prow[nrow]==NULL) prow[nrow]=(float*)malloc(sizeof(float)*2150);
	if(getpixelrow(prow[nrow])==0)
		break;
	printf("%d\r",nrow);fflush(stdout);
} 
printf("\nDone\n",nrow); 
sf_close(inwav);

/* raw image */
if(strchr(imgopt,(int)'r')!=NULL){
sprintf(pngfilename,"%s/%s-r.png",pngdirname,name);
ImageOut(pngfilename,"raw",prow,nrow,depth,IMG_WIDTH,0);
}

/* Channel A */
if(((strchr(imgopt,(int)'a')!=NULL) || (strchr(imgopt,(int)'c')!=NULL) || (strchr(imgopt,(int)'d')!=NULL))) {
	ch=Calibrate(prow,nrow,CHA_OFFSET);
	if(ch>0) {
		a=1;
		if(strchr(imgopt,(int)'a')!=NULL) {
		sprintf(pngfilename,"%s/%s-%s.png",pngdirname,name,chid[ch]);
		ImageOut(pngfilename,chid[ch],prow,nrow,depth,CH_WIDTH,CHA_OFFSET);
		}
	}
}

/* Channel B */
if((strchr(imgopt,(int)'b')!=NULL) || (strchr(imgopt,(int)'c')!=NULL) || (strchr(imgopt,(int)'t')!=NULL)) {
	ch=Calibrate(prow,nrow,CHB_OFFSET);
	if(ch>0) {
		if(strchr(imgopt,(int)'b')!=NULL) {
			sprintf(pngfilename,"%s/%s-%s.png",pngdirname,name,chid[ch]);
			ImageOut(pngfilename,chid[ch],prow,nrow,depth,CH_WIDTH,CHB_OFFSET);
		}
	}
	if(ch>2) {
		b=1;
		Temperature(prow,nrow,ch,CHB_OFFSET);
		if(strchr(imgopt,(int)'t')!=NULL) {
			sprintf(pngfilename,"%s/%s-t.png",pngdirname,name);
			ImageOut(pngfilename,"Temperature", prow,nrow,depth,CH_WIDTH,CHB_OFFSET);
		}
	}
}

/* distribution */
if(a && b  && strchr(imgopt,(int)'d')!=NULL){
	sprintf(pngfilename,"%s/%s-d.pnm",pngdirname,name);
	Distrib(pngfilename,prow,nrow);
}

/* color image */
if(a && b && strchr(imgopt,(int)'c')!=NULL){
	sprintf(pngfilename,"%s/%s-c.png",pngdirname,name);
	ImageColorOut(pngfilename,prow,nrow);
}

}
exit (0);
}