hms
/
aptdec
mirror of https://github.com/Xerbo/aptdec
1
0
Fork 0
Code Issues 0 Releases 7 Wiki Activity
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
55 Commits
2 Branches
1.8 MiB
 
 
 
 
 
Tree: c7e205473c
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from 'c7e205473c'
${ noResults }
aptdec/main.c

420 lines
11 KiB
Raw Blame History 

  1. /*

  2.  *  Atpdec

  3.  *  Copyright (c) 2004-2005 by Thierry Leconte (F4DWV)

  4.  *

  5.  *      $Id$

  6.  *

  7.  *   This library is free software; you can redistribute it and/or modify

  8.  *   it under the terms of the GNU Library General Public License as

  9.  *   published by the Free Software Foundation; either version 2 of

  10.  *   the License, or (at your option) any later version.

  11.  *

  12.  *   This program is distributed in the hope that it will be useful,

  13.  *   but WITHOUT ANY WARRANTY; without even the implied warranty of

  14.  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

  15.  *   GNU Library General Public License for more details.

  16.  *

  17.  *   You should have received a copy of the GNU Library General Public

  18.  *   License along with this library; if not, write to the Free Software

  19.  *   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.

  20.  *

  21.  */

  22. 

  23. #include <stdlib.h>

  24. #include <stdio.h>

  25. #ifdef WIN32

  26. #include "w32util.h"

  27. #else

  28. #include <libgen.h>

  29. #endif

  30. #include <string.h>

  31. #include <sndfile.h>

  32. #include <png.h>

  33. 

  34. #include "version.h"

  35. #include "temppalette.h"

  36. #include "offsets.h"

  37. 

  38. extern int getpixelrow(float *pixelv);

  39. extern int init_dsp(double F);;

  40. 

  41. static SNDFILE *inwav;

  42. 

  43. static int initsnd(char *filename)

  44. {

  45.     SF_INFO infwav;

  46.     int	    res;

  47. 

  48. /* open wav input file */

  49.     infwav.format = 0;

  50.     inwav = sf_open(filename, SFM_READ, &infwav);

  51.     if (inwav == NULL) {

  52. 	fprintf(stderr, "could not open %s\n", filename);

  53. 	return (1);

  54.     }

  55. 

  56.     res=init_dsp(infwav.samplerate);

  57.     if(res<0) {

  58. 	fprintf(stderr, "Sample rate too low : %d\n", infwav.samplerate);

  59. 	return (1);

  60.     }

  61.     if(res>0) {

  62. 	fprintf(stderr, "Sample rate too hight : %d\n", infwav.samplerate);

  63. 	return (1);

  64.     }

  65.     fprintf(stderr, "Sample rate : %d\n", infwav.samplerate);

  66. 

  67.     if (infwav.channels != 1) {

  68. 	fprintf(stderr, "Too many channels in input file : %d\n", infwav.channels);

  69. 	return (1);

  70.     }

  71. 

  72.     return (0);

  73. }

  74. 

  75. int getsample(float *sample, int nb)

  76. {

  77.     return (sf_read_float(inwav, sample, nb));

  78. }

  79. 

  80. static png_text text_ptr[] = {

  81.     {PNG_TEXT_COMPRESSION_NONE, "Software", version, sizeof(version)}

  82.     ,

  83.     {PNG_TEXT_COMPRESSION_NONE, "Channel", NULL, 0}

  84.     ,

  85.     {PNG_TEXT_COMPRESSION_NONE, "Description", "NOAA POES satellite Image",

  86.      25}

  87. };

  88. 

  89. static int

  90. ImageOut(char *filename, char *chid, float **prow, int nrow, 

  91. 	 int width, int offset, png_color *palette)

  92. {

  93.     FILE *pngfile;

  94.     png_infop info_ptr;

  95.     png_structp png_ptr;

  96.     int n;

  97. 

  98. /* init png lib */

  99.     png_ptr =

  100. 	png_create_write_struct(PNG_LIBPNG_VER_STRING, NULL, NULL, NULL);

  101.     if (!png_ptr) {

  102. 	fprintf(stderr, "could not open create png_ptr\n");

  103. 	return (1);

  104.     }

  105. 

  106.     info_ptr = png_create_info_struct(png_ptr);

  107.     if (!info_ptr) {

  108. 	png_destroy_write_struct(&png_ptr, (png_infopp) NULL);

  109. 	fprintf(stderr, "could not open create info_ptr\n");

  110. 	return (1);

  111.     }

  112. 

  113.     if(palette==NULL)  {

  114. 	/* grey image */

  115.     	png_set_IHDR(png_ptr, info_ptr, width, nrow,

  116. 		 8, PNG_COLOR_TYPE_GRAY, PNG_INTERLACE_NONE,

  117. 		 PNG_COMPRESSION_TYPE_DEFAULT, PNG_FILTER_TYPE_DEFAULT);

  118.     } else {

  119. 	/* palette color mage */

  120.     	png_set_IHDR(png_ptr, info_ptr, width, nrow,

  121. 		 8, PNG_COLOR_TYPE_PALETTE, PNG_INTERLACE_NONE,

  122. 		 PNG_COMPRESSION_TYPE_DEFAULT, PNG_FILTER_TYPE_DEFAULT);

  123. 	png_set_PLTE(png_ptr, info_ptr, palette, 256);

  124.     }

  125. 

  126.     text_ptr[1].text = chid;

  127.     text_ptr[1].text_length = strlen(chid);

  128.     png_set_text(png_ptr, info_ptr, text_ptr, 3);

  129.     png_set_pHYs(png_ptr, info_ptr, 4000, 4000, PNG_RESOLUTION_METER);

  130. 

  131.     printf("Writing %s ... ", filename);

  132.     fflush(stdout);

  133.     pngfile = fopen(filename, "wb");

  134.     if (pngfile == NULL) {

  135. 	fprintf(stderr, "could not open %s\n", filename);

  136. 	return (1);

  137.     }

  138.     png_init_io(png_ptr, pngfile);

  139.     png_write_info(png_ptr, info_ptr);

  140. 

  141.     for (n = 0; n < nrow; n++) {

  142. 	float *pixelv;

  143. 	png_byte pixel[2*IMG_WIDTH];

  144. 	int i;

  145. 

  146. 	pixelv = prow[n];

  147. 	for (i = 0; i < width; i++) {

  148. 	    pixel[i] = pixelv[i + offset];

  149. 	}

  150. 	png_write_row(png_ptr, pixel);

  151.     }

  152.     png_write_end(png_ptr, info_ptr);

  153.     fclose(pngfile);

  154.     printf("Done\n");

  155.     png_destroy_write_struct(&png_ptr, &info_ptr);

  156.     return (0);

  157. }

  158. 

  159. static int ImageRGBOut(char *filename, float **prow, int nrow)

  160. {

  161.     FILE *pngfile;

  162.     png_infop info_ptr;

  163.     png_structp png_ptr;

  164.     int n;

  165. 

  166.     extern void falsecolor(double v, double t, float *r, float *g,

  167. 			    float *b);

  168. 

  169. /* init png lib */

  170.     png_ptr =

  171. 	png_create_write_struct(PNG_LIBPNG_VER_STRING, NULL, NULL, NULL);

  172.     if (!png_ptr) {

  173. 	fprintf(stderr, "could not open create png_ptr\n");

  174. 	return (1);

  175.     }

  176. 

  177.     info_ptr = png_create_info_struct(png_ptr);

  178.     if (!info_ptr) {

  179. 	png_destroy_write_struct(&png_ptr, (png_infopp) NULL);

  180. 	fprintf(stderr, "could not open create info_ptr\n");

  181. 	return (1);

  182.     }

  183. 

  184.     png_set_IHDR(png_ptr, info_ptr, CH_WIDTH , nrow ,

  185. 		 8, PNG_COLOR_TYPE_RGB, PNG_INTERLACE_NONE,

  186. 		 PNG_COMPRESSION_TYPE_DEFAULT, PNG_FILTER_TYPE_DEFAULT);

  187. 

  188.     png_set_pHYs(png_ptr, info_ptr, 4000, 4000, PNG_RESOLUTION_METER);

  189. 

  190.     text_ptr[1].text = "False Colors";

  191.     text_ptr[1].text_length = strlen(text_ptr[1].text);

  192.     png_set_text(png_ptr, info_ptr, text_ptr, 3);

  193. 

  194.     printf("Computing False colors & writing : %s ...", filename);

  195.     fflush(stdout);

  196.     pngfile = fopen(filename, "wb");

  197.     if (pngfile == NULL) {

  198. 	fprintf(stderr, "could not open %s\n", filename);

  199. 	return (1);

  200.     }

  201.     png_init_io(png_ptr, pngfile);

  202.     png_write_info(png_ptr, info_ptr);

  203. 

  204.     for (n = 0; n < nrow ; n++) {

  205. 	png_color pix[CH_WIDTH];

  206.     	float *pixelc;

  207. 	int i;

  208. 

  209. 	pixelc = prow[n];

  210. 

  211. 	for (i = 0; i < CH_WIDTH - 1; i++) {

  212. 	    float v, t;

  213.   	    float r, g, b;

  214. 

  215. 	    v = pixelc[i+CHA_OFFSET];

  216. 	    t = pixelc[i+CHB_OFFSET];

  217. 

  218. 	    falsecolor(v, t, &r, &g, &b);

  219. 

  220. 	    pix[i].red = 255.0 * r; 

  221. 	    pix[i].green = 255.0 * g;

  222. 	    pix[i].blue = 255.0 * b;

  223. 	}

  224. 	png_write_row(png_ptr, (png_bytep) pix);

  225.     }

  226.     png_write_end(png_ptr, info_ptr);

  227.     fclose(pngfile);

  228.     printf("Done\n");

  229.     png_destroy_write_struct(&png_ptr, &info_ptr);

  230.     return (0);

  231. }

  232. 

  233. 

  234. static void Distrib(char *filename,float **prow,int nrow)

  235. {

  236. unsigned int distrib[256][256];

  237. int n;

  238. int x,y;

  239. int max=0;

  240. FILE *df;

  241. 

  242. for(y=0;y<256;y++)

  243. 	for(x=0;x<256;x++)

  244.                 distrib[y][x]=0;

  245. 

  246. for(n=0;n<nrow;n++) {

  247.         float *pixelv;

  248.         int     i;

  249. 

  250.         pixelv=prow[n];

  251.         for(i=0;i<CH_WIDTH;i++) {

  252. 

  253.                 y=(int)(pixelv[i+CHA_OFFSET]);

  254.                 x=(int)(pixelv[i+CHB_OFFSET]);

  255.                 distrib[y][x]+=1;

  256. 		if(distrib[y][x]> max) max=distrib[y][x];

  257.         }

  258. }

  259. df=fopen(filename,"w");

  260. 

  261. printf("Writing %s\n",filename);

  262. 

  263. fprintf(df,"P2\n#max %d\n",max);

  264. fprintf(df,"256 256\n255\n");

  265. for(y=0;y<256;y++)

  266. 	for(x=0;x<256;x++)

  267. 		fprintf(df,"%d\n",(int)((255.0*(double)(distrib[y][x]))/(double)max));

  268. fclose(df);

  269. }

  270. 

  271. 

  272. extern int Calibrate(float **prow, int nrow, int offset);

  273. extern void Temperature(float **prow, int nrow, int ch, int offset);

  274. extern int Ngiv(float **prow, int nrow);

  275. extern void readfconf(char *file);

  276. extern int optind, opterr;

  277. extern char *optarg;

  278. int satnum = 1;

  279. 

  280. static void usage(void)

  281. {

  282.     fprintf(stderr, "atpdec [options] soundfiles ...\n");

  283.     fprintf(stderr,

  284. 	    "options:\n-d <dir>\tDestination directory\n-i [r|a|b|c|t]\tOutput image type\n\t\t\tr: Raw\n\t\t\ta: A chan.\n\t\t\tb: B chan.\n\t\t\tc: False color\n\t\t\tt: Temperature\n-c <file>\tFalse color config file\n-s [15|16|17|18]\tSatellite number (for temperature and false color generation)\n");

  285.    exit(1);

  286. }

  287. 

  288. int main(int argc, char **argv)

  289. {

  290.     char pngfilename[1024];

  291.     char name[1024];

  292.     char pngdirname[1024] = "";

  293.     char imgopt[20] = "ac";

  294.     float *prow[3000];

  295.     char *chid[] = { "?", "1", "2", "3A", "4", "5", "3B" };

  296.     int nrow;

  297.     int chA,chB;

  298.     int c;

  299. 

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

  301. 

  302.     opterr = 0;

  303.     while ((c = getopt(argc, argv, "c:d:i:s:")) != EOF) {

  304. 	switch (c) {

  305. 	case 'd':

  306. 	    strcpy(pngdirname, optarg);

  307. 	    break;

  308. 	case 'c':

  309. 	    readfconf(optarg);

  310. 	    break;

  311. 	case 'i':

  312. 	    strcpy(imgopt, optarg);

  313. 	    break;

  314. 	case 's':

  315. 	    satnum = atoi(optarg)-15;

  316. 	    if (satnum < 0 || satnum > 3) {

  317. 		fprintf(stderr, "invalid satellite number  : must be in [15-18]\n");

  318. 		exit(1);

  319. 	    }

  320. 	    break;

  321. 	default:

  322. 	    usage();

  323. 	}

  324.     }

  325. 

  326.     for (nrow = 0; nrow < 3000; nrow++)

  327. 	prow[nrow] = NULL;

  328. 

  329.     for (; optind < argc; optind++) {

  330. 

  331. 	chA=chB=0;

  332. 

  333. 	strcpy(pngfilename, argv[optind]);

  334. 	strcpy(name, basename(pngfilename));

  335. 	strtok(name, ".");

  336. 	if (pngdirname[0] == '\0') {

  337. 	    strcpy(pngfilename, argv[optind]);

  338. 	    strcpy(pngdirname, dirname(pngfilename));

  339. 	}

  340. 

  341. /* open snd input */

  342. 	if (initsnd(argv[optind]))

  343. 	    exit(1);

  344. 

  345. /* main loop */

  346. 	printf("Decoding: %s \n", argv[optind]);

  347. 	for (nrow = 0; nrow < 3000; nrow++) {

  348. 	    if (prow[nrow] == NULL)

  349. 		prow[nrow] = (float *) malloc(sizeof(float) * 2150);

  350. 	    if (getpixelrow(prow[nrow]) == 0)

  351. 		break;

  352. 	    printf("%d\r", nrow);

  353. 	    fflush(stdout);

  354. 	}

  355. 	printf("\nDone\n");

  356. 	sf_close(inwav);

  357. 

  358. 

  359. /* raw image */

  360. 	if (strchr(imgopt, (int) 'r') != NULL) {

  361. 	    sprintf(pngfilename, "%s/%s-r.png", pngdirname, name);

  362. 	    ImageOut(pngfilename, "raw", prow, nrow, IMG_WIDTH, 0,NULL);

  363. 	}

  364. 

  365. /* Channel A */

  366. 	if (((strchr(imgopt, (int) 'a') != NULL)

  367. 	     || (strchr(imgopt, (int) 'c') != NULL)

  368. 	     || (strchr(imgopt, (int) 'd') != NULL))) {

  369. 	    chA = Calibrate(prow, nrow, CHA_OFFSET);

  370. 	    if (chA >= 0) {

  371. 		if (strchr(imgopt, (int) 'a') != NULL) {

  372. 		    sprintf(pngfilename, "%s/%s-%s.png", pngdirname, name, chid[chA]);

  373. 		    ImageOut(pngfilename, chid[chA], prow, nrow, CH_WIDTH , CHA_OFFSET,NULL);

  374. 		}

  375. 	    }

  376. 	}

  377. 

  378. /* Channel B */

  379. 	if ((strchr(imgopt, (int) 'b') != NULL)

  380. 	    || (strchr(imgopt, (int) 'c') != NULL)

  381. 	    || (strchr(imgopt, (int) 't') != NULL)

  382. 	    || (strchr(imgopt, (int) 'd') != NULL)) {

  383. 	    chB = Calibrate(prow, nrow, CHB_OFFSET);

  384. 	    if (chB >= 0) {

  385. 		if (strchr(imgopt, (int) 'b') != NULL) {

  386. 		    sprintf(pngfilename, "%s/%s-%s.png", pngdirname, name, chid[chB]);

  387. 		    ImageOut(pngfilename, chid[chB], prow, nrow, CH_WIDTH , CHB_OFFSET ,NULL);

  388. 		}

  389. 	    }

  390. 	    if (chB > 3) {

  391. 		Temperature(prow, nrow, chB, CHB_OFFSET);

  392. 		if (strchr(imgopt, (int) 't') != NULL) {

  393. 		    sprintf(pngfilename, "%s/%s-t.png", pngdirname, name);

  394. 		    ImageOut(pngfilename, "Temperature", prow, nrow, CH_WIDTH, CHB_OFFSET, (png_color*)TempPalette);

  395. 		}

  396. 	    }

  397. 	}

  398. 

  399. /* distribution */

  400. 	if (chA && chB && strchr(imgopt, (int) 'd') != NULL) {

  401. 	    sprintf(pngfilename, "%s/%s-d.pnm", pngdirname, name);

  402. 	    Distrib(pngfilename, prow, nrow);

  403. 	}

  404. 

  405. /* color image */

  406. 	if (chA==2 && chB==4 && strchr(imgopt, (int) 'c') != NULL) {

  407. 	    sprintf(pngfilename, "%s/%s-c.png", pngdirname, name);

  408. 	    ImageRGBOut(pngfilename, prow, nrow);

  409. 	}

  410. 

  411. /* vegetation image */

  412. 	if (chA==1 && chB==2 && strchr(imgopt, (int) 'c') != NULL) {

  413. 	    Ngiv(prow, nrow);

  414. 	    sprintf(pngfilename, "%s/%s-c.png", pngdirname, name);

  415. 	   ImageOut(pngfilename, "Vegetation", prow, nrow, CH_WIDTH, CHB_OFFSET, (png_color*)TempPalette);

  416. 	}

  417.     }

  418.     exit(0);

  419. }