rawIOYV12.cpp

   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 #include <assert.h>
  30 #include "rawIOYV12.h"
  31
  32 // yv12 is 4:2:0 planar Y, Cr, Cb
  33
  34 RawFrame& RawReaderYV12::read(RawFrame& f)
  35 {
  36         assert(f.chroma == RawFrame::Cr420);
  37
  38         size_t linesRead;
  39         unsigned char *inLine = new unsigned char[f.luma.width()];
  40
  41         //luminance
  42         for (int y = 0; y < f.luma.height(); y++) {
  43
  44                 linesRead = fread(inLine, f.luma.width(), 1, infile);
  45
  46                 for (int x = 0; x < f.luma.width(); x++)
  47                         f.luma[y][x] = inLine[x];
  48
  49                 if (linesRead < 1)
  50                         goto exit;
  51         }
  52
  53         //Cr
  54         for (int y = 0; y < f.cr.height(); y++) {
  55
  56                 linesRead = fread(inLine, f.cr.width(), 1, infile);
  57
  58                 for (int x = 0; x < f.cr.width(); x++)
  59                         f.cr[y][x] = inLine[x];
  60
  61                 if (linesRead < 1)
  62                         goto exit;
  63         }
  64
  65         //Cb
  66         for (int y = 0; y < f.cb.height(); y++) {
  67
  68                 linesRead = fread(inLine, f.cb.width(), 1, infile);
  69
  70                 for (int x = 0; x < f.cb.width(); x++)
  71                         f.cb[y][x] = inLine[x];
  72
  73                 if (linesRead < 1)
  74                         goto exit;
  75         }
  76
  77         exit:
  78
  79         free(inLine);
  80         return f;
  81 }
  82
  83 RawFrame& RawReaderYV12::read()
  84 {
  85         throw /* unknown size */;
  86 }
  87
  88 void RawWriterYV12::write(const RawFrame& f) const
  89 {
  90         assert(f.chroma == RawFrame::Cr420);
  91         for (int y = 0; y < f.luma.height(); y++)
  92                 for (int x = 0; x < f.luma.width(); x++)
  93                         fputc(f.luma[y][x], outfile);
  94
  95         for (int y = 0; y < f.cr.height(); y++)
  96                 for (int x = 0; x < f.cr.width(); x++)
  97                         fputc(f.cr[y][x], outfile);
  98
  99         for (int y = 0; y < f.cb.height(); y++)
 100                 for (int x = 0; x < f.cb.width(); x++)
 101                         fputc(f.cb[y][x], outfile);
 102 }