git.sesse.net Git - mlt/blob - src/modules/opengl/mlt_movit_input.cpp

   1 /*
   2  * mlt_movit_input.cpp
   3  * Copyright (C) 2013 Dan Dennedy <dan@dennedy.org>
   4  *
   5  * This program is free software; you can redistribute it and/or modify
   6  * it under the terms of the GNU General Public License as published by
   7  * the Free Software Foundation; either version 2 of the License, or
   8  * (at your option) any later version.
   9  *
  10  * This program is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13  * GNU General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU General Public License
  16  * along with this program; if not, write to the Free Software Foundation,
  17  * Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
  18  */
  19
  20 #include "mlt_movit_input.h"
  21
  22 using namespace movit;
  23
  24 MltInput::MltInput( mlt_image_format format )
  25         : m_format(format)
  26         , input(0)
  27         , isRGB(true)
  28 {
  29 }
  30
  31 MltInput::~MltInput()
  32 {
  33 }
  34
  35 void MltInput::useFlatInput(MovitPixelFormat pix_fmt, unsigned width, unsigned height)
  36 {
  37         if (!input) {
  38                 m_width = width;
  39                 m_height = height;
  40                 ImageFormat image_format;
  41                 image_format.color_space = COLORSPACE_sRGB;
  42                 image_format.gamma_curve = GAMMA_sRGB;
  43                 input = new FlatInput(image_format, pix_fmt, GL_UNSIGNED_BYTE, width, height);
  44         }
  45 }
  46
  47 void MltInput::useYCbCrInput(const ImageFormat& image_format, const YCbCrFormat& ycbcr_format, unsigned width, unsigned height)
  48 {
  49         if (!input) {
  50                 m_width = width;
  51                 m_height = height;
  52                 input = new YCbCrInput(image_format, ycbcr_format, width, height);
  53                 isRGB = false;
  54                 m_ycbcr_format = ycbcr_format;
  55         }
  56 }
  57
  58 void MltInput::set_pixel_data(const unsigned char* data)
  59 {
  60         assert(input);
  61         if (isRGB) {
  62                 FlatInput* flat = (FlatInput*) input;
  63                 flat->set_pixel_data(data);
  64         } else {
  65                 YCbCrInput* ycbcr = (YCbCrInput*) input;
  66                 ycbcr->set_pixel_data(0, data);
  67                 ycbcr->set_pixel_data(1, &data[m_width * m_height]);
  68                 ycbcr->set_pixel_data(2, &data[m_width * m_height + (m_width / m_ycbcr_format.chroma_subsampling_x * m_height / m_ycbcr_format.chroma_subsampling_y)]);
  69         }
  70 }
  71
  72 void MltInput::invalidate_pixel_data()
  73 {
  74         assert(input);
  75         if (isRGB) {
  76                 FlatInput* flat = (FlatInput*) input;
  77                 flat->invalidate_pixel_data();
  78         } else {
  79                 YCbCrInput* ycbcr = (YCbCrInput*) input;
  80                 ycbcr->invalidate_pixel_data();
  81         }
  82 }