git.sesse.net Git - ffmpeg/blob - libavformat/fitsenc.c

   1 /*
   2  * FITS muxer
   3  * Copyright (c) 2017 Paras Chadha
   4  *
   5  * This file is part of FFmpeg.
   6  *
   7  * FFmpeg is free software; you can redistribute it and/or
   8  * modify it under the terms of the GNU Lesser General Public
   9  * License as published by the Free Software Foundation; either
  10  * version 2.1 of the License, or (at your option) any later version.
  11  *
  12  * FFmpeg 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 GNU
  15  * Lesser General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU Lesser General Public
  18  * License along with FFmpeg; if not, write to the Free Software
  19  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  20  */
  21
  22 /**
  23  * @file
  24  * FITS muxer.
  25  */
  26
  27 #include "internal.h"
  28
  29 typedef struct FITSContext {
  30     int first_image;
  31 } FITSContext;
  32
  33 static int fits_write_header(AVFormatContext *s)
  34 {
  35     FITSContext *fitsctx = s->priv_data;
  36     fitsctx->first_image = 1;
  37     return 0;
  38 }
  39
  40 /**
  41  * Write one header line comprising of keyword and value(int)
  42  * @param s AVFormat Context
  43  * @param keyword pointer to the char array in which keyword is stored
  44  * @param value the value corresponding to the keyword
  45  * @param lines_written to keep track of lines written so far
  46  * @return 0
  47  */
  48 static int write_keyword_value(AVFormatContext *s, const char *keyword, int value, int *lines_written)
  49 {
  50     int len, ret;
  51     uint8_t header[80];
  52
  53     len = strlen(keyword);
  54     memset(header, ' ', sizeof(header));
  55     memcpy(header, keyword, len);
  56
  57     header[8] = '=';
  58     header[9] = ' ';
  59
  60     ret = snprintf(header + 10, 70, "%d", value);
  61     memset(&header[ret + 10], ' ', sizeof(header) - (ret + 10));
  62
  63     avio_write(s->pb, header, sizeof(header));
  64     *lines_written += 1;
  65     return 0;
  66 }
  67
  68 static int write_image_header(AVFormatContext *s)
  69 {
  70     AVStream *st = s->streams[0];
  71     AVCodecParameters *encctx = st->codecpar;
  72     FITSContext *fitsctx = s->priv_data;
  73     uint8_t buffer[80];
  74     int bitpix, naxis, naxis3 = 1, bzero = 0, rgb = 0, lines_written = 0, lines_left;
  75
  76     switch (encctx->format) {
  77         case AV_PIX_FMT_GRAY8:
  78             bitpix = 8;
  79             naxis = 2;
  80             break;
  81         case AV_PIX_FMT_GRAY16BE:
  82             bitpix = 16;
  83             naxis = 2;
  84             bzero = 32768;
  85             break;
  86         case AV_PIX_FMT_GBRP:
  87         case AV_PIX_FMT_GBRAP:
  88             bitpix = 8;
  89             naxis = 3;
  90             rgb = 1;
  91             if (encctx->format == AV_PIX_FMT_GBRP) {
  92                 naxis3 = 3;
  93             } else {
  94                 naxis3 = 4;
  95             }
  96             break;
  97         case AV_PIX_FMT_GBRP16BE:
  98         case AV_PIX_FMT_GBRAP16BE:
  99             bitpix = 16;
 100             naxis = 3;
 101             rgb = 1;
 102             if (encctx->format == AV_PIX_FMT_GBRP16BE) {
 103                 naxis3 = 3;
 104             } else {
 105                 naxis3 = 4;
 106             }
 107             bzero = 32768;
 108             break;
 109     }
 110
 111     if (fitsctx->first_image) {
 112         memcpy(buffer, "SIMPLE  = ", 10);
 113         memset(buffer + 10, ' ', 70);
 114         buffer[29] = 'T';
 115         avio_write(s->pb, buffer, sizeof(buffer));
 116     } else {
 117         memcpy(buffer, "XTENSION= 'IMAGE   '", 20);
 118         memset(buffer + 20, ' ', 60);
 119         avio_write(s->pb, buffer, sizeof(buffer));
 120     }
 121     lines_written++;
 122
 123     write_keyword_value(s, "BITPIX", bitpix, &lines_written);         // no of bits per pixel
 124     write_keyword_value(s, "NAXIS", naxis, &lines_written);           // no of dimensions of image
 125     write_keyword_value(s, "NAXIS1", encctx->width, &lines_written);   // first dimension i.e. width
 126     write_keyword_value(s, "NAXIS2", encctx->height, &lines_written);  // second dimension i.e. height
 127
 128     if (rgb)
 129         write_keyword_value(s, "NAXIS3", naxis3, &lines_written);     // third dimension to store RGBA planes
 130
 131     if (!fitsctx->first_image) {
 132         write_keyword_value(s, "PCOUNT", 0, &lines_written);
 133         write_keyword_value(s, "GCOUNT", 1, &lines_written);
 134     } else {
 135         fitsctx->first_image = 0;
 136     }
 137
 138     /*
 139      * Since FITS does not support unsigned 16 bit integers,
 140      * BZERO = 32768 is used to store unsigned 16 bit integers as
 141      * signed integers so that it can be read properly.
 142      */
 143     if (bitpix == 16)
 144         write_keyword_value(s, "BZERO", bzero, &lines_written);
 145
 146     if (rgb) {
 147         memcpy(buffer, "CTYPE3  = 'RGB     '", 20);
 148         memset(buffer + 20, ' ', 60);
 149         avio_write(s->pb, buffer, sizeof(buffer));
 150         lines_written++;
 151     }
 152
 153     memcpy(buffer, "END", 3);
 154     memset(buffer + 3, ' ', 77);
 155     avio_write(s->pb, buffer, sizeof(buffer));
 156     lines_written++;
 157
 158     lines_left = ((lines_written + 35) / 36) * 36 - lines_written;
 159     memset(buffer, ' ', 80);
 160     while (lines_left > 0) {
 161         avio_write(s->pb, buffer, sizeof(buffer));
 162         lines_left--;
 163     }
 164     return 0;
 165 }
 166
 167 static int fits_write_packet(AVFormatContext *s, AVPacket *pkt)
 168 {
 169     write_image_header(s);
 170     avio_write(s->pb, pkt->data, pkt->size);
 171     return 0;
 172 }
 173
 174 AVOutputFormat ff_fits_muxer = {
 175     .name         = "fits",
 176     .long_name    = NULL_IF_CONFIG_SMALL("Flexible Image Transport System"),
 177     .extensions   = "fits",
 178     .priv_data_size = sizeof(FITSContext),
 179     .audio_codec  = AV_CODEC_ID_NONE,
 180     .video_codec  = AV_CODEC_ID_FITS,
 181     .write_header = fits_write_header,
 182     .write_packet = fits_write_packet,
 183 };