rawIOppm.h

   1 /* ***** BEGIN LICENSE BLOCK *****
   2  *
   3  * $Id$
   4  *
   5  * The MIT License
   6  *
   7  * Copyright (c) 2008 BBC Research
   8  *
   9  * Permission is hereby granted, free of charge, to any person obtaining a copy
  10  * of this software and associated documentation files (the "Software"), to deal
  11  * in the Software without restriction, including without limitation the rights
  12  * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  13  * copies of the Software, and to permit persons to whom the Software is
  14  * furnished to do so, subject to the following conditions:
  15  *
  16  * The above copyright notice and this permission notice shall be included in
  17  * all copies or substantial portions of the Software.
  18  *
  19  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  20  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  21  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  22  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  23  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  24  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  25  * THE SOFTWARE.
  26  *
  27  * ***** END LICENSE BLOCK ***** */
  28
  29 #ifndef __RAWIOPPM_H
  30 #define __RAWIOPPM_H
  31
  32 #include <stdio.h>
  33 #include "rawFrameIO.h"
  34
  35 class RawWriterppm : public RawWriter {
  36 public:
  37         RawWriterppm(FILE *fd, int d) :
  38                 outfile(fd), depth(d)
  39         {
  40                 /* pgm may only have a maxval of 65535 */
  41                 assert(d <= 16);
  42         };
  43
  44         void write(const RawFrame& f) const
  45         {
  46                 switch (f.chroma) {
  47                 case RawFrame::Cr422:
  48                         writeCr422(f);
  49                         break; /* 422 is done */
  50                 case RawFrame::Cr420:
  51                         throw;
  52                         break; /* 420 is not yet implemented */
  53                 case RawFrame::YOnly:
  54                         writeYOnly(f);
  55                         break; /* Y-only is done */
  56                 default:
  57                         throw /* invalid frame type */;
  58                 }
  59         }
  60
  61 private:
  62         void writeCr422(const RawFrame&) const;
  63         void writeYOnly(const RawFrame&) const;
  64         void
  65             chroma422to444(const Array2d<int>& in, const int line, Array1d<int>& out) const;
  66
  67         void yuv2rgb(const Array2d<int>& Y, const int lineNum,
  68                      const Array1d<int>& ULine, const Array1d<int>& VLine,
  69                      Array1d<int>& RLine, Array1d<int>& GLine, Array1d<int>& BLine,
  70                      int depth) const;
  71
  72         FILE *outfile;
  73         int depth;
  74 };
  75
  76 class RawReaderppm : public RawReader {
  77 public:
  78         RawReaderppm(FILE *fd) :
  79                 infile(fd)
  80         {
  81                 readHeader(); /* get the dimensions */
  82         }
  83         ;
  84
  85         RawFrame& read(RawFrame&);
  86         RawFrame& read();
  87
  88 private:
  89         void readHeader();
  90
  91         int _maxval;
  92         FILE *infile;
  93 };
  94
  95 #endif