|
- /*
- * 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 directory\n-i [a|b|c|t|d]\n-c \n-p\n-s [0|1]\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);
- }
|