|
- /*
- * This file is part of Aptdec.
- * Copyright (c) 2004-2009 Thierry Leconte (F4DWV), Xerbo (xerbo@protonmail.com) 2019-2020
- *
- * Aptdec is free software: you can redistribute it and/or modify
- * it under the terms of the GNU 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 General Public License for more details.
- *
- * You should have received a copy of the GNU General Public License
- * along with this program. If not, see <https://www.gnu.org/licenses/>.
- *
- */
-
- #include <png.h>
- #include <stdlib.h>
- #include <stdio.h>
- #include <string.h>
- #include <stdint.h>
-
- #include "common.h"
- #include "offsets.h"
-
- extern int zenith;
- extern char PrecipPalette[256*3];
- extern rgb_t applyPalette(char *palette, int val);
- extern rgb_t RGBcomposite(rgb_t top, float top_a, rgb_t bottom, float bottom_a);
-
- int mapOverlay(char *filename, rgb_t **crow, int nrow, int zenith, int MCIR) {
- FILE *fp = fopen(filename, "rb");
-
- // Create reader
- png_structp png = png_create_read_struct(PNG_LIBPNG_VER_STRING, NULL, NULL, NULL);
- if(!png) return 0;
- png_infop info = png_create_info_struct(png);
- if(!info) return 0;
- png_init_io(png, fp);
-
- // Read info from header
- png_read_info(png, info);
- int width = png_get_image_width(png, info);
- int height = png_get_image_height(png, info);
- png_byte color_type = png_get_color_type(png, info);
- png_byte bit_depth = png_get_bit_depth(png, info);
-
- // Check the image
- if(width != 1040){
- fprintf(stderr, "Map must be 1040px wide.\n");
- return 0;
- }else if(bit_depth != 16){
- fprintf(stderr, "Map must be 16 bit color.\n");
- return 0;
- }else if(color_type != PNG_COLOR_TYPE_RGB){
- fprintf(stderr, "Map must be RGB.\n");
- return 0;
- }else if(zenith > height/2 || nrow-zenith > height/2){
- fprintf(stderr, "WARNING: Map is too short to cover entire image\n");
- }
-
- // Create row buffers
- png_bytep *mapRows = NULL;
- mapRows = (png_bytep *) malloc(sizeof(png_bytep) * height);
- for(int y = 0; y < height; y++)
- mapRows[y] = (png_byte *) malloc(png_get_rowbytes(png, info));
-
- // Read image
- png_read_image(png, mapRows);
-
- // Tidy up
- fclose(fp);
- png_destroy_read_struct(&png, &info, NULL);
-
- printf("Adding map overlay\n");
-
- // Map overlay / MCIR / Precipitation
- int mapOffset = (height/2)-zenith;
- for(int y = 0; y < nrow; y++) {
- for(int x = 49; x < width - 82; x++){
- // Maps are 16 bit / channel
- png_bytep px = &mapRows[CLIP(y + mapOffset, 0, height)][x * 6];
- rgb_t map = {
- (px[0] << 8) | px[1],
- (px[2] << 8) | px[3],
- (px[4] << 8) | px[5]
- };
-
- // Pixel offsets
- int chb = x + CHB_OFFSET - 49;
- int cha = x + 36;
-
- // Fill in map
- if(MCIR){
- if(map.b < 128 && map.g > 128){
- // Land
- float green = CLIP((map.g-256)/32.0, 0, 1);
- float blue = 1 - CLIP((map.b-32)/64.0, 0, 1);
- crow[y][cha] = (rgb_t){blue*127, 30+green*80, 40};
- }else{
- // Sea
- crow[y][cha] = (rgb_t){12, 30, 85};
- }
- }
-
- // Color -> alpha: composite
- int composite = map.r + map.g + map.b;
- // Color -> alpha: flattern and convert to 8 bits / channel
- float factor = (255 * 255 * 2.0) / composite;
- map.r *= factor/257.0; map.g *= factor/257.0; map.b *= factor/257.0;
- // Color -> alpha: convert black to alpha
- float alpha = CLIP(composite / 65535.0, 0, 1);
-
- // Map overlay on channel A
- crow[y][cha] = RGBcomposite(map, alpha, crow[y][cha], 1);
- // Map overlay on channel B
- if(!MCIR)
- crow[y][chb] = RGBcomposite(map, alpha, crow[y][chb], 1);
-
- // Cloud overlay on channel A
- if(MCIR){
- float cloud = CLIP((crow[y][chb].r - 115) / 107, 0, 1);
- crow[y][cha] = RGBcomposite((rgb_t){240, 250, 255}, cloud, crow[y][cha], 1);
- }
- }
- }
-
- return 1;
- }
-
- int readRawImage(char *filename, float **prow, int *nrow) {
- FILE *fp = fopen(filename, "r");
-
- // Create reader
- png_structp png = png_create_read_struct(PNG_LIBPNG_VER_STRING, NULL, NULL, NULL);
- if(!png) return 0;
- png_infop info = png_create_info_struct(png);
- if(!info) return 0;
- png_init_io(png, fp);
-
- // Read info from header
- png_read_info(png, info);
- int width = png_get_image_width(png, info);
- int height = png_get_image_height(png, info);
- png_byte color_type = png_get_color_type(png, info);
- png_byte bit_depth = png_get_bit_depth(png, info);
-
- // Check the image
- if(width != IMG_WIDTH){
- fprintf(stderr, "Raw image must be %ipx wide.\n", IMG_WIDTH);
- return 0;
- }else if(bit_depth != 8){
- fprintf(stderr, "Raw image must have 8 bit color.\n");
- return 0;
- }else if(color_type != PNG_COLOR_TYPE_GRAY){
- fprintf(stderr, "Raw image must be grayscale.\n");
- return 0;
- }
-
- // Create row buffers
- png_bytep *PNGrows = NULL;
- PNGrows = (png_bytep *) malloc(sizeof(png_bytep) * height);
- for(int y = 0; y < height; y++) PNGrows[y] = (png_byte *)
- malloc(png_get_rowbytes(png, info));
-
- // Read image
- png_read_image(png, PNGrows);
-
- // Tidy up
- fclose(fp);
- png_destroy_read_struct(&png, &info, NULL);
-
- // Put into prow
- *nrow = height;
- for(int y = 0; y < height; y++) {
- prow[y] = (float *) malloc(sizeof(float) * width);
-
- for(int x = 0; x < width; x++)
- prow[y][x] = (float)PNGrows[y][x];
- }
-
- return 1;
- }
-
- png_text meta[] = {
- {PNG_TEXT_COMPRESSION_NONE, "Software", VERSION},
- {PNG_TEXT_COMPRESSION_NONE, "Channel", "Unknown", 7},
- {PNG_TEXT_COMPRESSION_NONE, "Description", "NOAA satellite image", 20}
- };
-
- int ImageOut(options_t *opts, image_t *img, int offset, int width, char *desc, char *chid, char *palette){
- char outName[384];
- sprintf(outName, "%s/%s-%s.png", opts->path, img->name, chid);
-
- meta[1].text = desc;
- meta[1].text_length = sizeof(desc);
-
- FILE *pngfile;
-
- // Reduce the width of the image to componsate for the missing telemetry
- int fc = strcmp(desc, "False Color") == 0;
- int greyscale = 0;
- int skiptele = 0;
- if(opts->effects != NULL && CONTAINS(opts->effects, 't')){
- width -= TOTAL_TELE;
- skiptele = 1;
- }
-
- // Create writer
- png_structp png_ptr = png_create_write_struct(PNG_LIBPNG_VER_STRING, NULL, NULL, NULL);
- if (!png_ptr) {
- png_destroy_write_struct(&png_ptr, (png_infopp) NULL);
- fprintf(stderr, "Could not create a PNG writer\n");
- return 0;
- }
- png_infop info_ptr = png_create_info_struct(png_ptr);
- if (!info_ptr) {
- png_destroy_write_struct(&png_ptr, (png_infopp) NULL);
- fprintf(stderr, "Could not create a PNG writer\n");
- return 0;
- }
-
- if(palette == NULL && !CONTAINS(opts->effects, 'p') && !fc && opts->map[0] == '\0' && strcmp(chid, "MCIR") != 0){
- greyscale = 1;
-
- // Greyscale image
- png_set_IHDR(png_ptr, info_ptr, width, img->nrow,
- 8, PNG_COLOR_TYPE_GRAY, PNG_INTERLACE_NONE,
- PNG_COMPRESSION_TYPE_DEFAULT, PNG_FILTER_TYPE_DEFAULT);
- }else{
- // 8 bit RGB image
- png_set_IHDR(png_ptr, info_ptr, width, img->nrow,
- 8, PNG_COLOR_TYPE_RGB, PNG_INTERLACE_NONE,
- PNG_COMPRESSION_TYPE_DEFAULT, PNG_FILTER_TYPE_DEFAULT);
- }
-
- png_set_text(png_ptr, info_ptr, meta, 3);
- png_set_pHYs(png_ptr, info_ptr, 3636, 3636, PNG_RESOLUTION_METER);
-
- // Init I/O
- pngfile = fopen(outName, "wb");
- if (!pngfile) {
- fprintf(stderr, "Could not open %s for writing\n", outName);
- return 1;
- }
- png_init_io(png_ptr, pngfile);
- png_write_info(png_ptr, info_ptr);
-
- // Move prow into crow, crow ~ color rows
- rgb_t *crow[img->nrow];
- if(!greyscale && !fc){
- for(int y = 0; y < img->nrow; y++){
- crow[y] = (rgb_t *) malloc(sizeof(rgb_t) * IMG_WIDTH);
-
- for(int x = 0; x < IMG_WIDTH; x++){
- if(palette == NULL)
- crow[y][x].r = crow[y][x].g = crow[y][x].b = img->prow[y][x];
- else
- crow[y][x] = applyPalette(palette, img->prow[y][x]);
- }
- }
- }
-
- // Precipitation
- // TODO: use temperature calibration
- if(CONTAINS(opts->effects, 'p')){
- for(int y = 0; y < img->nrow; y++){
- for(int x = 0; x < CH_WIDTH; x++){
- if(img->prow[y][x + CHB_OFFSET] > 191)
- crow[y][x + CHB_OFFSET] = applyPalette(PrecipPalette, img->prow[y][x + CHB_OFFSET]);
- }
- }
- }
-
- // Map stuff
- if(opts->map != NULL && opts->map[0] != '\0'){
- if(mapOverlay(opts->map, crow, img->nrow, zenith, strcmp(desc, "MCIR") == 0) == 0){
- fprintf(stderr, "Skipping MCIR generation; see above.\n");
- return 0;
- }
- }else if(strcmp(chid, "MCIR") == 0){
- fprintf(stderr, "Skipping MCIR generation; no map provided.\n");
- return 0;
- }
-
- printf("Writing %s", outName);
-
- // Build image
- for (int y = 0; y < img->nrow; y++) {
- png_color pix[width]; // Color
- png_byte mpix[width]; // Mono
-
- int skip = 0;
- for (int x = 0; x < width; x++) {
- if(skiptele){
- switch (x) {
- case 0:
- skip += SYNC_WIDTH + SPC_WIDTH;
- break;
- case CH_WIDTH:
- skip += TELE_WIDTH + SYNC_WIDTH + SPC_WIDTH;
- break;
- }
- }
-
- if(greyscale){
- mpix[x] = img->prow[y][x + skip + offset];
- }else if(fc){
- pix[x] = (png_color){
- CLIP(img->prow[y][x + CHA_OFFSET], 0, 255),
- CLIP(img->prow[y][x + CHA_OFFSET], 0, 255),
- CLIP(img->prow[y][x + CHB_OFFSET], 0, 255)
- };
- }else{
- pix[x] = (png_color){
- crow[y][x + skip + offset].r,
- crow[y][x + skip + offset].g,
- crow[y][x + skip + offset].b
- };
- }
- }
-
- if(greyscale){
- png_write_row(png_ptr, (png_bytep) mpix);
- }else{
- png_write_row(png_ptr, (png_bytep) pix);
- }
- }
-
- // Tidy up
- png_write_end(png_ptr, info_ptr);
- fclose(pngfile);
- printf("\nDone\n");
- png_destroy_write_struct(&png_ptr, &info_ptr);
-
- return 1;
- }
-
- // TODO: clean up everthing below this comment
- png_structp rt_png_ptr;
- png_infop rt_info_ptr;
- FILE *rt_pngfile;
-
- int initWriter(options_t *opts, image_t *img, int width, int height, char *desc, char *chid){
- char outName[384];
- sprintf(outName, "%s/%s-%s.png", opts->path, img->name, chid);
-
- meta[1].text = desc;
- meta[1].text_length = sizeof(desc);
-
- // Create writer
- rt_png_ptr = png_create_write_struct(PNG_LIBPNG_VER_STRING, NULL, NULL, NULL);
- if (!rt_png_ptr) {
- png_destroy_write_struct(&rt_png_ptr, (png_infopp) NULL);
- fprintf(stderr, "Could not create a PNG writer\n");
- return 0;
- }
- rt_info_ptr = png_create_info_struct(rt_png_ptr);
- if (!rt_info_ptr) {
- png_destroy_write_struct(&rt_png_ptr, (png_infopp) NULL);
- fprintf(stderr, "Could not create a PNG writer\n");
- return 0;
- }
-
- // Greyscale image
- png_set_IHDR(rt_png_ptr, rt_info_ptr, width, height,
- 8, PNG_COLOR_TYPE_GRAY, PNG_INTERLACE_NONE,
- PNG_COMPRESSION_TYPE_DEFAULT, PNG_FILTER_TYPE_DEFAULT);
-
- png_set_text(rt_png_ptr, rt_info_ptr, meta, 3);
-
- // Channel = 25cm wide
- png_set_pHYs(rt_png_ptr, rt_info_ptr, 3636, 3636, PNG_RESOLUTION_METER);
-
- // Init I/O
- rt_pngfile = fopen(outName, "wb");
- if (!rt_pngfile) {
- fprintf(stderr, "Could not open %s for writing\n", outName);
- return 0;
- }
- png_init_io(rt_png_ptr, rt_pngfile);
- png_write_info(rt_png_ptr, rt_info_ptr);
-
- return 1;
- }
-
- void pushRow(float *row, int width){
- png_byte pix[width];
- for(int i = 0; i < width; i++)
- pix[i] = row[i];
-
- png_write_row(rt_png_ptr, (png_bytep) pix);
- }
-
- void closeWriter(){
- png_write_end(rt_png_ptr, rt_info_ptr);
- fclose(rt_pngfile);
- png_destroy_write_struct(&rt_png_ptr, &rt_info_ptr);
- }
|