rawIOpgm.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 __RAWIOPGM_H
  30 #define __RAWIOPGM_H
  31
  32 #include <stdio.h>
  33 #include "rawFrameIO.h"
  34
  35 class RawWriterpgm : public RawWriter {
  36 public:
  37         RawWriterpgm(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: /* fall through */
  48                 case RawFrame::Cr420:
  49                         writeCr42x(f);
  50                         break;
  51                 case RawFrame::YOnly:
  52                         writeYOnly(f);
  53                         break;
  54                 default:
  55                         throw /* invalid frame type */;
  56                 }
  57         }
  58
  59 private:
  60         void writeCr42x(const RawFrame&) const;
  61         void writeYOnly(const RawFrame&) const;
  62
  63         FILE *outfile;
  64         int depth;
  65 };
  66
  67 class RawReaderpgm : public RawReader {
  68 public:
  69         RawReaderpgm(FILE *fd) :
  70                 infile(fd)
  71         {
  72                 readHeader();
  73         }
  74         ;
  75
  76         RawFrame& read(RawFrame&);
  77         RawFrame& read();
  78
  79 private:
  80         void readHeader();
  81
  82         int _maxval;
  83         FILE *infile;
  84 };
  85
  86 #endif