|
- /*
- * Atpdec
- * Copyright (c) 2004-2005 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 int getpixelrow(float *pixelv);
-
- #define SYNC_WIDTH 39
- #define SPC_WIDTH 47
- #define TELE_WIDTH 45
- #define CH_WIDTH 909
- #define CH_OFFSET (SYNC_WIDTH+SPC_WIDTH+CH_WIDTH+TELE_WIDTH)
- #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));
- }
-
- static 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 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,
- 8, 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, "wb");
- 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++) {
- pixel[i] = pixelv[i + offset];
- }
- 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;
- float *pixelc;
-
- extern void falsecolor(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);
-
- png_set_pHYs(png_ptr, info_ptr, 4000, 4000, PNG_RESOLUTION_METER);
-
- 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);
-
- printf("Computing False colors & writing : %s ...", filename);
- fflush(stdout);
- pngfile = fopen(filename, "wb");
- 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++) {
- png_color pix[CH_WIDTH];
- int i;
-
- pixelc = prow[n];
-
- for (i = 0; i < CH_WIDTH - 1; i++) {
- float v, t;
- float r, g, b;
-
- v = pixelc[i+SYNC_WIDTH + SPC_WIDTH];
- t = pixelc[i+SYNC_WIDTH + SPC_WIDTH + CH_OFFSET];
-
- falsecolor(v, t, &r, &g, &b);
-
- pix[i].red = 255.0 * r;
- pix[i].green = 255.0 * g;
- pix[i].blue = 255.0 * b;
- }
- png_write_row(png_ptr, (png_bytep) pix);
- }
- png_write_end(png_ptr, info_ptr);
- fclose(pngfile);
- printf("Done\n");
- png_destroy_write_struct(&png_ptr, &info_ptr);
- return (0);
- }
-
- extern int Calibrate(float **prow, int nrow, int offset);
- extern void 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 [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");
- exit(1);
- }
-
- int 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 nrow;
- int ch;
- int c;
-
- printf("%s\n", version);
-
- opterr = 0;
- while ((c = getopt(argc, argv, "c:d:i:s:")) != EOF) {
- switch (c) {
- case 'd':
- strcpy(pngdirname, optarg);
- break;
- case 'c':
- readfconf(optarg);
- break;
- case 'i':
- strcpy(imgopt, optarg);
- break;
- case 's':
- satnum = atoi(optarg)-15;
- if (satnum < 0 || satnum > 3) {
- fprintf(stderr, "invalid satellite number : must be in [15-18]\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");
- sf_close(inwav);
-
- /* raw image */
- if (strchr(imgopt, (int) 'r') != NULL) {
- sprintf(pngfilename, "%s/%s-r.png", pngdirname, name);
- ImageOut(pngfilename, "raw", prow, nrow, 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, SYNC_WIDTH);
- if (ch >= 0) {
- if (strchr(imgopt, (int) 'a') != NULL) {
- sprintf(pngfilename, "%s/%s-%s.png", pngdirname, name,
- chid[ch]);
- ImageOut(pngfilename, chid[ch], prow, nrow,
- SPC_WIDTH + CH_WIDTH + TELE_WIDTH,
- SYNC_WIDTH);
- }
- }
- if (ch < 2)
- a = 1;
- }
-
- /* Channel B */
- if ((strchr(imgopt, (int) 'b') != NULL)
- || (strchr(imgopt, (int) 'c') != NULL)
- || (strchr(imgopt, (int) 't') != NULL)
- || (strchr(imgopt, (int) 'd') != NULL)) {
- ch = Calibrate(prow, nrow, CH_OFFSET + SYNC_WIDTH);
- 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,
- SPC_WIDTH + CH_WIDTH + TELE_WIDTH,
- CH_OFFSET + SYNC_WIDTH);
- }
- }
- if (ch > 2) {
- b = 1;
- Temperature(prow, nrow, ch, CH_OFFSET + SYNC_WIDTH);
- if (strchr(imgopt, (int) 't') != NULL) {
- sprintf(pngfilename, "%s/%s-t.png", pngdirname, name);
- ImageOut(pngfilename, "Temperature", prow, nrow,
- CH_WIDTH, CH_OFFSET + SYNC_WIDTH + SPC_WIDTH);
- }
- }
- }
-
- /* distribution */
- if (a && b && strchr(imgopt, (int) 'd') != NULL) {
- sprintf(pngfilename, "%s/%s-d.pnm", pngdirname, name);
- }
-
- /* color image */
- if (a && b && strchr(imgopt, (int) 'c') != NULL) {
- sprintf(pngfilename, "%s/%s-c.png", pngdirname, name);
- ImageColorOut(pngfilename, prow, nrow);
- }
-
- }
- exit(0);
- }
|