git.sesse.net Git - nageru/blob - ycbcr_converter.cpp

   1 #include "ycbcr_converter.h"
   2
   3 #include <movit/mix_effect.h>
   4 #include <movit/ycbcr_input.h>
   5
   6 #include "jpeg_frame.h"
   7
   8 using namespace std;
   9 using namespace movit;
  10
  11 YCbCrConverter::YCbCrConverter(YCbCrConverter::OutputMode output_mode, ResourcePool *resource_pool)
  12 {
  13         ImageFormat inout_format;
  14         inout_format.color_space = COLORSPACE_sRGB;
  15         inout_format.gamma_curve = GAMMA_sRGB;
  16
  17         ycbcr_format.luma_coefficients = YCBCR_REC_709;
  18         ycbcr_format.full_range = false;
  19         ycbcr_format.num_levels = 256;
  20         ycbcr_format.chroma_subsampling_x = 2;
  21         ycbcr_format.chroma_subsampling_y = 1;
  22         ycbcr_format.cb_x_position = 0.0f;  // H.264 -- _not_ JPEG, even though our input is MJPEG-encoded
  23         ycbcr_format.cb_y_position = 0.5f;  // Irrelevant.
  24         ycbcr_format.cr_x_position = 0.0f;
  25         ycbcr_format.cr_y_position = 0.5f;
  26
  27         YCbCrFormat ycbcr_output_format = ycbcr_format;
  28         ycbcr_output_format.chroma_subsampling_x = 1;
  29
  30         // Planar Y'CbCr decoding chain.
  31         planar_chain.reset(new EffectChain(1280, 720, resource_pool));
  32         ycbcr_planar_input = (YCbCrInput *)planar_chain->add_input(new YCbCrInput(inout_format, ycbcr_format, 1280, 720, YCBCR_INPUT_PLANAR));
  33         if (output_mode == OUTPUT_TO_RGBA) {
  34                 planar_chain->add_output(inout_format, OUTPUT_ALPHA_FORMAT_POSTMULTIPLIED);
  35                 planar_chain->set_output_origin(OUTPUT_ORIGIN_BOTTOM_LEFT);
  36         } else {
  37                 assert(output_mode == OUTPUT_TO_DUAL_YCBCR);
  38
  39                 // One full Y'CbCr texture (for interpolation), one that's just Y (throwing away the
  40                 // Cb and Cr channels). The second copy is sort of redundant, but it's the easiest way
  41                 // of getting the gray data into a layered texture.
  42                 planar_chain->add_ycbcr_output(inout_format, OUTPUT_ALPHA_FORMAT_POSTMULTIPLIED, ycbcr_output_format);
  43                 planar_chain->add_ycbcr_output(inout_format, OUTPUT_ALPHA_FORMAT_POSTMULTIPLIED, ycbcr_output_format);
  44                 planar_chain->set_output_origin(OUTPUT_ORIGIN_TOP_LEFT);
  45         }
  46         planar_chain->set_dither_bits(8);
  47         planar_chain->finalize();
  48
  49         // Semiplanar Y'CbCr decoding chain (for images coming from VA-API).
  50         semiplanar_chain.reset(new EffectChain(1280, 720, resource_pool));
  51         ycbcr_semiplanar_input = (YCbCrInput *)semiplanar_chain->add_input(new YCbCrInput(inout_format, ycbcr_format, 1280, 720, YCBCR_INPUT_SPLIT_Y_AND_CBCR));
  52         if (output_mode == OUTPUT_TO_RGBA) {
  53                 semiplanar_chain->add_output(inout_format, OUTPUT_ALPHA_FORMAT_POSTMULTIPLIED);
  54                 semiplanar_chain->set_output_origin(OUTPUT_ORIGIN_BOTTOM_LEFT);
  55         } else {
  56                 // See above.
  57                 semiplanar_chain->add_ycbcr_output(inout_format, OUTPUT_ALPHA_FORMAT_POSTMULTIPLIED, ycbcr_output_format);
  58                 semiplanar_chain->add_ycbcr_output(inout_format, OUTPUT_ALPHA_FORMAT_POSTMULTIPLIED, ycbcr_output_format);
  59                 semiplanar_chain->set_output_origin(OUTPUT_ORIGIN_TOP_LEFT);
  60         }
  61         semiplanar_chain->set_dither_bits(8);
  62         semiplanar_chain->finalize();
  63 }
  64
  65 EffectChain *YCbCrConverter::prepare_chain_for_conversion(shared_ptr<Frame> frame)
  66 {
  67         if (frame->is_semiplanar) {
  68                 setup_input_for_frame(frame, ycbcr_format, ycbcr_semiplanar_input);
  69                 return semiplanar_chain.get();
  70         } else {
  71                 setup_input_for_frame(frame, ycbcr_format, ycbcr_planar_input);
  72                 return planar_chain.get();
  73         }
  74 }
  75
  76 void setup_input_for_frame(shared_ptr<Frame> frame, const YCbCrFormat &ycbcr_format, YCbCrInput *input)
  77 {
  78         YCbCrFormat format_copy = ycbcr_format;
  79         format_copy.chroma_subsampling_x = frame->chroma_subsampling_x;
  80         format_copy.chroma_subsampling_y = frame->chroma_subsampling_y;
  81         input->change_ycbcr_format(format_copy);
  82
  83         input->set_width(frame->width);
  84         input->set_height(frame->height);
  85         input->set_pixel_data(0, frame->y.get());
  86         input->set_pitch(0, frame->pitch_y);
  87         if (frame->is_semiplanar) {
  88                 input->set_pixel_data(1, frame->cbcr.get());
  89                 input->set_pitch(1, frame->pitch_chroma);
  90         } else {
  91                 input->set_pixel_data(1, frame->cb.get());
  92                 input->set_pixel_data(2, frame->cr.get());
  93                 input->set_pitch(1, frame->pitch_chroma);
  94                 input->set_pitch(2, frame->pitch_chroma);
  95         }
  96 }