hms
/
aptdec
spogulis no https://github.com/Xerbo/aptdec
1
0
Atdalīts 0
Kods Problēmas 0 Laidieni 7 Vikivietne Aktivitāte
Nevar pievienot vairāk kā 25 tēmas Tēmai ir jāsākas ar burtu vai ciparu, tā var saturēt domu zīmes ('-') un var būt līdz 35 simboliem gara.
61 Revīzija
2 Atzari
9.4 MiB
 
 
 
 
 
Koks: e9e4bbd07d
Atzari Tagi
${ item.name }
Izveidot atzaru ${ searchTerm }
no 'e9e4bbd07d'
${ noResults }
aptdec/main.c

401 rinda
9.9 KiB
Neapstrādāts Vainot Vēsture 

  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. #include <string.h>

  26. 

  27. #ifdef WIN32

  28. #include "w32util.h"

  29. #else

  30. #include <libgen.h>

  31. #endif

  32. 

  33. #include <sndfile.h>

  34. #include <png.h>

  35. 

  36. #include "version.h"

  37. #include "offsets.h"

  38. 

  39. #include "temppalette.h"

  40. #include "gvipalette.h"

  41. 

  42. extern int getpixelrow(float *pixelv);

  43. extern int init_dsp(void);;

  44. 

  45. 

  46. 

  47. static png_text text_ptr[] = {

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

  49.     ,

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

  51.     ,

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

  53.      25}

  54. };

  55. 

  56. static int

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

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

  59. {

  60.     FILE *pngfile;

  61.     png_infop info_ptr;

  62.     png_structp png_ptr;

  63.     int n;

  64. 

  65. /* init png lib */

  66.     png_ptr =

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

  68.     if (!png_ptr) {

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

  70. 	return (1);

  71.     }

  72. 

  73.     info_ptr = png_create_info_struct(png_ptr);

  74.     if (!info_ptr) {

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

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

  77. 	return (1);

  78.     }

  79. 

  80.     if(palette==NULL)  {

  81. 	/* grey image */

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

  83. 		 8, PNG_COLOR_TYPE_GRAY, PNG_INTERLACE_NONE,

  84. 		 PNG_COMPRESSION_TYPE_DEFAULT, PNG_FILTER_TYPE_DEFAULT);

  85.     } else {

  86. 	/* palette color mage */

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

  88. 		 8, PNG_COLOR_TYPE_PALETTE, PNG_INTERLACE_NONE,

  89. 		 PNG_COMPRESSION_TYPE_DEFAULT, PNG_FILTER_TYPE_DEFAULT);

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

  91.     }

  92. 

  93.     text_ptr[1].text = chid;

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

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

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

  97. 

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

  99.     fflush(stdout);

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

  101.     if (pngfile == NULL) {

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

  103. 	return (1);

  104.     }

  105.     png_init_io(png_ptr, pngfile);

  106.     png_write_info(png_ptr, info_ptr);

  107. 

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

  109. 	float *pixelv;

  110. 	png_byte pixel[2*IMG_WIDTH];

  111. 	int i;

  112. 

  113. 	pixelv = prow[n];

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

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

  116. 	}

  117. 	png_write_row(png_ptr, pixel);

  118.     }

  119.     png_write_end(png_ptr, info_ptr);

  120.     fclose(pngfile);

  121.     printf("Done\n");

  122.     png_destroy_write_struct(&png_ptr, &info_ptr);

  123.     return (0);

  124. }

  125. 

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

  127. {

  128.     FILE *pngfile;

  129.     png_infop info_ptr;

  130.     png_structp png_ptr;

  131.     int n;

  132. 

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

  134. 			    float *b);

  135. 

  136. /* init png lib */

  137.     png_ptr =

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

  139.     if (!png_ptr) {

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

  141. 	return (1);

  142.     }

  143. 

  144.     info_ptr = png_create_info_struct(png_ptr);

  145.     if (!info_ptr) {

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

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

  148. 	return (1);

  149.     }

  150. 

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

  152. 		 8, PNG_COLOR_TYPE_RGB, PNG_INTERLACE_NONE,

  153. 		 PNG_COMPRESSION_TYPE_DEFAULT, PNG_FILTER_TYPE_DEFAULT);

  154. 

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

  156. 

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

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

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

  160. 

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

  162.     fflush(stdout);

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

  164.     if (pngfile == NULL) {

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

  166. 	return (1);

  167.     }

  168.     png_init_io(png_ptr, pngfile);

  169.     png_write_info(png_ptr, info_ptr);

  170. 

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

  172. 	png_color pix[CH_WIDTH];

  173.     	float *pixelc;

  174. 	int i;

  175. 

  176. 	pixelc = prow[n];

  177. 

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

  179. 	    float v, t;

  180.   	    float r, g, b;

  181. 

  182. 	    v = pixelc[i+CHA_OFFSET];

  183. 	    t = pixelc[i+CHB_OFFSET];

  184. 

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

  186. 

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

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

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

  190. 	}

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

  192.     }

  193.     png_write_end(png_ptr, info_ptr);

  194.     fclose(pngfile);

  195.     printf("Done\n");

  196.     png_destroy_write_struct(&png_ptr, &info_ptr);

  197.     return (0);

  198. }

  199. 

  200. 

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

  202. {

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

  204. int n;

  205. int x,y;

  206. int max=0;

  207. FILE *df;

  208. 

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

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

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

  212. 

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

  214.         float *pixelv;

  215.         int     i;

  216. 

  217.         pixelv=prow[n];

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

  219. 

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

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

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

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

  224.         }

  225. }

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

  227. 

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

  229. 

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

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

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

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

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

  235. fclose(df);

  236. }

  237. 

  238. 

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

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

  241. extern int Ngvi(float **prow, int nrow);

  242. extern void readfconf(char *file);

  243. extern int optind, opterr;

  244. extern char *optarg;

  245. int satnum = 4;

  246. int nch=0;

  247. 

  248. static void usage(void)

  249. {

  250.     fprintf(stderr, version);

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

  252.     fprintf(stderr,

  253. 	    "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|19]\tSatellite number (for temperature and false color generation)\n");

  254.    exit(1);

  255. }

  256. 

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

  258. {

  259.     char pngfilename[1024];

  260.     char name[1024];

  261.     char pngdirname[1024] = "";

  262. 

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

  264.     float *prow[3000];

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

  266.     int nrow;

  267.     int chA,chB;

  268.     int c;

  269. 

  270.     opterr = 0;

  271.     while ((c = getopt(argc, argv, "a:f:c:d:i:s:LR")) != EOF) {

  272. 	switch (c) {

  273. 	case 'd':

  274. 	    strcpy(pngdirname, optarg);

  275. 	    break;

  276. 	case 'c':

  277. 	    readfconf(optarg);

  278. 	    break;

  279. 	case 'i':

  280. 	    strcpy(imgopt, optarg);

  281. 	    break;

  282. 	case 's':

  283. 	    satnum = atoi(optarg)-15;

  284. 	    if (satnum < 0 || satnum > 4) {

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

  286. 		exit(1);

  287. 	    }

  288. 	    break;

  289. 	case 'f':

  290. 		strcpy(name, basename(optarg));

  291. 		strtok(name, ".");

  292. 		if(initsnd(optarg))

  293. 		  exit(1);

  294. 	    break;

  295. 	case 'a':

  296. 		strcpy(name,"audioin");

  297. 		if(initalsa(optarg))

  298. 		  exit(1);

  299. 	    break;

  300. 	case 'L':

  301. 		nch=0;

  302. 	    break;

  303. 	case 'R':

  304. 		nch=0;

  305. 	    break;

  306. 	default:

  307. 	    usage();

  308. 	}

  309.     }

  310. 

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

  312. 	prow[nrow] = NULL;

  313. 

  314. 	chA=chB=0;

  315. 

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

  317. 		strcpy(pngdirname,".");

  318. 	}

  319. 

  320. 

  321. /* init dsp */

  322.     if(init_dsp())

  323. 	exit(2);

  324. 

  325. /* main loop */

  326. 	printf("Decoding: %s \n", name);

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

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

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

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

  331. 		break;

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

  333. 	    fflush(stdout);

  334. 	}

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

  336. 	endsample();

  337. 

  338. 

  339. /* raw image */

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

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

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

  343. 	}

  344. 

  345. /* Channel A */

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

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

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

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

  350. 	    if (chA >= 0) {

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

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

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

  354. 		}

  355. 	    }

  356. 	}

  357. 

  358. /* Channel B */

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

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

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

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

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

  364. 	    if (chB >= 0) {

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

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

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

  368. 		}

  369. 	    }

  370. 	    if (chB > 3) {

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

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

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

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

  375. 		}

  376. 	    }

  377. 	}

  378. 

  379. /* distribution */

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

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

  382. 	    Distrib(pngfilename, prow, nrow);

  383. 	}

  384. 

  385. /* color image */

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

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

  388. 	    ImageRGBOut(pngfilename, prow, nrow);

  389. 	}

  390. 

  391. /* GVI image */

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

  393. 	    Ngvi(prow, nrow);

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

  395. 	   ImageOut(pngfilename, "GVI", prow, nrow, CH_WIDTH, CHB_OFFSET, (png_color*)GviPalette);

  396. 	}

  397. 

  398. 

  399.     exit(0);

  400. }