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.
46 Commits
2 Branches
2.7 MiB
 
 
 
 
 
Tree: 2e9a550316
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '2e9a550316'
${ noResults }
aptdec/main.c

408 lines
11 KiB
Raw Blame History 

  1. /*

  2.  *  Atpdec

  3.  *  Copyright (c) 2004 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 <stdio.h>

  24. #ifdef WIN32

  25. #include "w32util.h"

  26. #else

  27. #include <libgen.h>

  28. #endif

  29. #include <string.h>

  30. #include <sndfile.h>

  31. #include <png.h>

  32. 

  33. #include "version.h"

  34. 

  35. extern int getpixelrow(float *pixelv);

  36. 

  37. #define SYNC_WIDTH 39

  38. #define SPC_WIDTH 47

  39. #define TELE_WIDTH 45

  40. #define CH_WIDTH  909

  41. #define CH_OFFSET  (SYNC_WIDTH+SPC_WIDTH+CH_WIDTH+TELE_WIDTH)

  42. #define IMG_WIDTH  2080

  43. 

  44. static SNDFILE *inwav;

  45. 

  46. static int initsnd(char *filename)

  47. {

  48.     SF_INFO infwav;

  49. 

  50. /* open wav input file */

  51.     infwav.format = 0;

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

  53.     if (inwav == NULL) {

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

  55. 	return (1);

  56.     }

  57.     if (infwav.samplerate != 11025) {

  58. 	fprintf(stderr, "Bad Input File sample rate: %d. Must be 11025\n",

  59. 		infwav.samplerate);

  60. 	return (1);

  61.     }

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

  63. 	fprintf(stderr, "Too many channels in input file : %d\n",

  64. 		infwav.channels);

  65. 	return (1);

  66.     }

  67. 

  68.     return (0);

  69. }

  70. 

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

  72. {

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

  74. }

  75. 

  76. static png_text text_ptr[] = {

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

  78.     ,

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

  80.     ,

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

  82.      25}

  83. };

  84. 

  85. static int

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

  87. 	 int width, int offset)

  88. {

  89.     FILE *pngfile;

  90.     png_infop info_ptr;

  91.     png_structp png_ptr;

  92.     int n;

  93. 

  94. /* init png lib */

  95.     png_ptr =

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

  97.     if (!png_ptr) {

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

  99. 	return (1);

  100.     }

  101. 

  102.     info_ptr = png_create_info_struct(png_ptr);

  103.     if (!info_ptr) {

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

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

  106. 	return (1);

  107.     }

  108. 

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

  110. 		 8, PNG_COLOR_TYPE_GRAY, PNG_INTERLACE_NONE,

  111. 		 PNG_COMPRESSION_TYPE_DEFAULT, PNG_FILTER_TYPE_DEFAULT);

  112. 

  113.     text_ptr[1].text = chid;

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

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

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

  117. 

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

  119.     fflush(stdout);

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

  121.     if (pngfile == NULL) {

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

  123. 	return (1);

  124.     }

  125.     png_init_io(png_ptr, pngfile);

  126.     png_write_info(png_ptr, info_ptr);

  127. 

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

  129. 	float *pixelv;

  130. 	png_byte pixel[2*IMG_WIDTH];

  131. 	int i;

  132. 

  133. 	pixelv = prow[n];

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

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

  136. 	}

  137. 	png_write_row(png_ptr, pixel);

  138.     }

  139.     png_write_end(png_ptr, info_ptr);

  140.     fclose(pngfile);

  141.     printf("Done\n");

  142.     png_destroy_write_struct(&png_ptr, &info_ptr);

  143.     return (0);

  144. }

  145. 

  146. int ImageColorOut(char *filename, float **prow, int nrow)

  147. {

  148.     FILE *pngfile;

  149.     png_infop info_ptr;

  150.     png_structp png_ptr;

  151.     int n;

  152.     float *pixelc, *pixeln;

  153. 

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

  155. 			    float *b);

  156. 

  157. /* init png lib */

  158.     png_ptr =

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

  160.     if (!png_ptr) {

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

  162. 	return (1);

  163.     }

  164. 

  165.     info_ptr = png_create_info_struct(png_ptr);

  166.     if (!info_ptr) {

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

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

  169. 	return (1);

  170.     }

  171. 

  172.     png_set_IHDR(png_ptr, info_ptr, 2 * (CH_WIDTH - 1), 2 * (nrow - 1),

  173. 		 8, PNG_COLOR_TYPE_RGB, PNG_INTERLACE_NONE,

  174. 		 PNG_COMPRESSION_TYPE_DEFAULT, PNG_FILTER_TYPE_DEFAULT);

  175. 

  176.     png_set_pHYs(png_ptr, info_ptr, 2000, 2000, PNG_RESOLUTION_METER);

  177. 

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

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

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

  181. 

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

  183.     fflush(stdout);

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

  185.     if (pngfile == NULL) {

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

  187. 	return (1);

  188.     }

  189.     png_init_io(png_ptr, pngfile);

  190.     png_write_info(png_ptr, info_ptr);

  191. 

  192.     for (n = 0; n < nrow - 1; n++) {

  193. 	png_color pix[2 * CH_WIDTH];

  194. 	float r, g, b;

  195. 	float v, vs, t, ts;

  196. 	int i;

  197. 

  198. 	pixelc = prow[n];

  199. 	pixeln = prow[n + 1];

  200. 

  201. 	v = (pixelc[SYNC_WIDTH + SPC_WIDTH] +

  202. 	     pixeln[SYNC_WIDTH + SPC_WIDTH]) / 2.0;

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

  204. 

  205. 	    t = pixelc[i + SYNC_WIDTH + SPC_WIDTH + CH_OFFSET];

  206. 

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

  208. 	    pix[2 * i].red = 255.0 * r;

  209. 	    pix[2 * i].green = 255.0 * g;

  210. 	    pix[2 * i].blue = 255.0 * b;

  211. 

  212. 	    vs = (pixelc[i + 1 + SYNC_WIDTH + SPC_WIDTH] +

  213. 		  pixeln[i + 1 + SYNC_WIDTH + SPC_WIDTH]) / 2.0;

  214. 	    v = (v + vs) / 2.0;

  215. 	    t = (pixelc[i + SYNC_WIDTH + SPC_WIDTH + CH_OFFSET] +

  216. 		 pixelc[i + 1 + SYNC_WIDTH + SPC_WIDTH + CH_OFFSET]) / 2.0;

  217. 

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

  219. 	    pix[2 * i + 1].red = 255.0 * r;

  220. 	    pix[2 * i + 1].green = 255.0 * g;

  221. 	    pix[2 * i + 1].blue = 255.0 * b;

  222. 

  223. 	    v = vs;

  224. 	}

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

  226. 

  227. 	t = (pixelc[SYNC_WIDTH + SPC_WIDTH + CH_OFFSET] +

  228. 	     pixeln[SYNC_WIDTH + SPC_WIDTH + CH_OFFSET]) / 2.0;

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

  230. 

  231. 	    v = pixeln[i + SYNC_WIDTH + SPC_WIDTH];

  232. 

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

  234. 	    pix[2 * i].red = 255.0 * r;

  235. 	    pix[2 * i].green = 255.0 * g;

  236. 	    pix[2 * i].blue = 255.0 * b;

  237. 

  238. 	    v = (pixeln[i + SYNC_WIDTH + SPC_WIDTH] +

  239. 		 pixeln[i + 1 + SYNC_WIDTH + SPC_WIDTH]) / 2.0;

  240. 	    ts = (pixelc[i + SYNC_WIDTH + SPC_WIDTH + CH_OFFSET] +

  241. 		  pixeln[i + 1 + SYNC_WIDTH + SPC_WIDTH +

  242. 			 CH_OFFSET]) / 2.0;

  243. 	    t = (t + ts) / 2.0;

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

  245. 	    pix[2 * i + 1].red = 255.0 * r;

  246. 	    pix[2 * i + 1].green = 255.0 * g;

  247. 	    pix[2 * i + 1].blue = 255.0 * b;

  248. 

  249. 	    t = ts;

  250. 	}

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

  252.     }

  253.     png_write_end(png_ptr, info_ptr);

  254.     fclose(pngfile);

  255.     printf("Done\n");

  256.     png_destroy_write_struct(&png_ptr, &info_ptr);

  257.     return (0);

  258. }

  259. 

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

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

  262. extern void readfconf(char *file);

  263. extern int optind, opterr;

  264. extern char *optarg;

  265. int satnum = 1;

  266. 

  267. static void usage(void)

  268. {

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

  270.     fprintf(stderr,

  271. 	    "options:\n-d <dir>\tDestination directory\n-i [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 [0|1]\tSatellite id (for temperature and false color generation)\n\t\t\t0:NOAA15\n\t\t\t1:NOAA17\n");

  272.    exit(1);

  273. }

  274. 

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

  276. {

  277.     char pngfilename[1024];

  278.     char name[1024];

  279.     char pngdirname[1024] = "";

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

  281.     float *prow[3000];

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

  283.     int nrow;

  284.     int ch;

  285.     int c;

  286. 

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

  288. 

  289.     opterr = 0;

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

  291. 	switch (c) {

  292. 	case 'd':

  293. 	    strcpy(pngdirname, optarg);

  294. 	    break;

  295. 	case 'c':

  296. 	    readfconf(optarg);

  297. 	    break;

  298. 	case 'i':

  299. 	    strcpy(imgopt, optarg);

  300. 	    break;

  301. 	case 's':

  302. 	    satnum = atoi(optarg);

  303. 	    if (satnum < 0 || satnum > 1) {

  304. 		fprintf(stderr, "invalid satellite\n");

  305. 		exit(1);

  306. 	    }

  307. 	    break;

  308. 	default:

  309. 	    usage();

  310. 	}

  311.     }

  312. 

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

  314. 	prow[nrow] = NULL;

  315. 

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

  317. 	int a = 0, b = 0;

  318. 

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

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

  321. 	strtok(name, ".");

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

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

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

  325. 	}

  326. 

  327. /* open snd input */

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

  329. 	    exit(1);

  330. 

  331. /* main loop */

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

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

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

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

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

  337. 		break;

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

  339. 	    fflush(stdout);

  340. 	}

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

  342. 	sf_close(inwav);

  343. 

  344. /* raw image */

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

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

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

  348. 	}

  349. 

  350. /* Channel A */

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

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

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

  354. 	    ch = Calibrate(prow, nrow, SYNC_WIDTH);

  355. 	    if (ch >= 0) {

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

  357. 		    sprintf(pngfilename, "%s/%s-%s.png", pngdirname, name,

  358. 			    chid[ch]);

  359. 		    ImageOut(pngfilename, chid[ch], prow, nrow, 

  360. 			     SPC_WIDTH + CH_WIDTH + TELE_WIDTH,

  361. 			     SYNC_WIDTH);

  362. 		}

  363. 	    }

  364. 	    if (ch < 2)

  365. 		a = 1;

  366. 	}

  367. 

  368. /* Channel B */

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

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

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

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

  373. 	    ch = Calibrate(prow, nrow, CH_OFFSET + SYNC_WIDTH);

  374. 	    if (ch >= 0) {

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

  376. 		    sprintf(pngfilename, "%s/%s-%s.png", pngdirname, name,

  377. 			    chid[ch]);

  378. 		    ImageOut(pngfilename, chid[ch], prow, nrow, 

  379. 			     SPC_WIDTH + CH_WIDTH + TELE_WIDTH,

  380. 			     CH_OFFSET + SYNC_WIDTH);

  381. 		}

  382. 	    }

  383. 	    if (ch > 2) {

  384. 		b = 1;

  385. 		Temperature(prow, nrow, ch, CH_OFFSET + SYNC_WIDTH);

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

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

  388. 		    ImageOut(pngfilename, "Temperature", prow, nrow,

  389. 			     CH_WIDTH, CH_OFFSET + SYNC_WIDTH + SPC_WIDTH);

  390. 		}

  391. 	    }

  392. 	}

  393. 

  394. /* distribution */

  395. 	if (a && b && strchr(imgopt, (int) 'd') != NULL) {

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

  397. 	}

  398. 

  399. /* color image */

  400. 	if (a && b && strchr(imgopt, (int) 'c') != NULL) {

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

  402. 	    ImageColorOut(pngfilename, prow, nrow);

  403. 	}

  404. 

  405.     }

  406.     exit(0);

  407. }