src/lib/tiff_decoder.cc

   1 /*
   2     Copyright (C) 2012 Carl Hetherington <cth@carlh.net>
   3
   4     This program is free software; you can redistribute it and/or modify
   5     it under the terms of the GNU General Public License as published by
   6     the Free Software Foundation; either version 2 of the License, or
   7     (at your option) any later version.
   8
   9     This program is distributed in the hope that it will be useful,
  10     but WITHOUT ANY WARRANTY; without even the implied warranty of
  11     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12     GNU General Public License for more details.
  13
  14     You should have received a copy of the GNU General Public License
  15     along with this program; if not, write to the Free Software
  16     Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  17
  18 */
  19
  20 /** @file src/tiff_decoder.cc
  21  *  @brief A decoder which reads a numbered set of TIFF files, one per frame.
  22  */
  23
  24 #include <vector>
  25 #include <string>
  26 #include <iostream>
  27 #include <stdint.h>
  28 #include <boost/shared_ptr.hpp>
  29 #include <boost/filesystem.hpp>
  30 #include <tiffio.h>
  31 extern "C" {
  32 #include <libavformat/avformat.h>
  33 }
  34 #include "util.h"
  35 #include "tiff_decoder.h"
  36 #include "exceptions.h"
  37 #include "image.h"
  38 #include "options.h"
  39 #include "film.h"
  40
  41 using namespace std;
  42 using namespace boost;
  43
  44 /** @param f Our Film.
  45  *  @param o Options.
  46  *  @param j Job that we are associated with, or 0.
  47  */
  48 TIFFDecoder::TIFFDecoder (boost::shared_ptr<Film> f, boost::shared_ptr<const Options> o, Job* j)
  49         : Decoder (f, o, j)
  50 {
  51         string const dir = _film->content_path ();
  52
  53         if (!filesystem::is_directory (dir)) {
  54                 throw DecodeError ("TIFF content must be in a directory");
  55         }
  56
  57         for (filesystem::directory_iterator i = filesystem::directory_iterator (dir); i != filesystem::directory_iterator(); ++i) {
  58                 /* Aah, the sweet smell of progress */
  59 #if BOOST_FILESYSTEM_VERSION == 3
  60                 string const ext = filesystem::path(*i).extension().string();
  61                 string const l = filesystem::path(*i).leaf().generic_string();
  62 #else
  63                 string const ext = filesystem::path(*i).extension();
  64                 string const l = i->leaf ();
  65 #endif
  66                 if (ext == ".tif" || ext == ".tiff") {
  67                         _files.push_back (l);
  68                 }
  69         }
  70
  71         _files.sort ();
  72
  73         _iter = _files.begin ();
  74
  75 }
  76
  77 float
  78 TIFFDecoder::frames_per_second () const
  79 {
  80         /* We don't know */
  81         return 0;
  82 }
  83
  84 Size
  85 TIFFDecoder::native_size () const
  86 {
  87         if (_files.empty ()) {
  88                 throw DecodeError ("no TIFF files found");
  89         }
  90
  91         TIFF* t = TIFFOpen (file_path (_files.front()).c_str (), "r");
  92         if (t == 0) {
  93                 throw DecodeError ("could not open TIFF file");
  94         }
  95
  96         uint32_t width;
  97         TIFFGetField (t, TIFFTAG_IMAGEWIDTH, &width);
  98         uint32_t height;
  99         TIFFGetField (t, TIFFTAG_IMAGELENGTH, &height);
 100
 101         return Size (width, height);
 102 }
 103
 104 int
 105 TIFFDecoder::audio_channels () const
 106 {
 107         /* No audio */
 108         return 0;
 109 }
 110
 111 int
 112 TIFFDecoder::audio_sample_rate () const
 113 {
 114         return 0;
 115 }
 116
 117 int64_t
 118 TIFFDecoder::audio_channel_layout () const
 119 {
 120         return 0;
 121 }
 122
 123 bool
 124 TIFFDecoder::pass ()
 125 {
 126         if (_iter == _files.end ()) {
 127                 return true;
 128         }
 129
 130         TIFF* t = TIFFOpen (file_path (*_iter).c_str (), "r");
 131         if (t == 0) {
 132                 throw DecodeError ("could not open TIFF file");
 133         }
 134
 135         uint32_t width;
 136         TIFFGetField (t, TIFFTAG_IMAGEWIDTH, &width);
 137         uint32_t height;
 138         TIFFGetField (t, TIFFTAG_IMAGELENGTH, &height);
 139
 140         int const num_pixels = width * height;
 141         uint32_t * raster = (uint32_t *) _TIFFmalloc (num_pixels * sizeof (uint32_t));
 142         if (raster == 0) {
 143                 throw DecodeError ("could not allocate memory to decode TIFF file");
 144         }
 145
 146         if (TIFFReadRGBAImage (t, width, height, raster, 0) < 0) {
 147                 throw DecodeError ("could not read TIFF data");
 148         }
 149
 150         RGBFrameImage image (Size (width, height));
 151
 152         uint8_t* p = image.data()[0];
 153         for (uint32_t y = 0; y < height; ++y) {
 154                 for (uint32_t x = 0; x < width; ++x) {
 155                         uint32_t const i = (height - y - 1) * width + x;
 156                         *p++ = raster[i] & 0xff;
 157                         *p++ = (raster[i] & 0xff00) >> 8;
 158                         *p++ = (raster[i] & 0xff0000) >> 16;
 159                 }
 160         }
 161
 162         _TIFFfree (raster);
 163         TIFFClose (t);
 164
 165         process_video (image.frame ());
 166
 167         ++_iter;
 168         return false;
 169 }
 170
 171 /** @param file name within our content directory
 172  *  @return full path to the file
 173  */
 174 string
 175 TIFFDecoder::file_path (string f) const
 176 {
 177         stringstream s;
 178         s << _film->content_path() << "/" << f;
 179         return _film->file (s.str ());
 180 }
 181
 182 PixelFormat
 183 TIFFDecoder::pixel_format () const
 184 {
 185         return PIX_FMT_RGB24;
 186 }
 187
 188 int
 189 TIFFDecoder::time_base_numerator () const
 190 {
 191         return dcp_frame_rate(_film->frames_per_second()).frames_per_second;
 192 }
 193
 194
 195 int
 196 TIFFDecoder::time_base_denominator () const
 197 {
 198         return 1;
 199 }
 200
 201 int
 202 TIFFDecoder::sample_aspect_ratio_numerator () const
 203 {
 204         /* XXX */
 205         return 1;
 206 }
 207
 208 int
 209 TIFFDecoder::sample_aspect_ratio_denominator () const
 210 {
 211         /* XXX */
 212         return 1;
 213 }