test/make_black_test.cc

   1 /*
   2     Copyright (C) 2012 Carl Hetherington <cth@carlh.net>
   3
   4     This program is free software; you can redistribute it and/or modify
   5     it under the terms of the GNU General Public License as published by
   6     the Free Software Foundation; either version 2 of the License, or
   7     (at your option) any later version.
   8
   9     This program is distributed in the hope that it will be useful,
  10     but WITHOUT ANY WARRANTY; without even the implied warranty of
  11     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12     GNU General Public License for more details.
  13
  14     You should have received a copy of the GNU General Public License
  15     along with this program; if not, write to the Free Software
  16     Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  17
  18 */
  19
  20 /** @file  test/make_black_test.cc
  21  *  @brief Check that Image::make_black works, and doesn't use values which crash
  22  *  sws_scale().
  23  *
  24  *  @see test/image_test.cc
  25  */
  26
  27 #include <boost/test/unit_test.hpp>
  28 #include <dcp/util.h>
  29 extern "C" {
  30 #include <libavutil/pixfmt.h>
  31 }
  32 #include "lib/image.h"
  33
  34 using std::list;
  35
  36 BOOST_AUTO_TEST_CASE (make_black_test)
  37 {
  38         dcp::Size in_size (512, 512);
  39         dcp::Size out_size (1024, 1024);
  40
  41         list<AVPixelFormat> pix_fmts;
  42         pix_fmts.push_back (AV_PIX_FMT_RGB24); // 2
  43         pix_fmts.push_back (AV_PIX_FMT_ARGB);
  44         pix_fmts.push_back (AV_PIX_FMT_RGBA);
  45         pix_fmts.push_back (AV_PIX_FMT_ABGR);
  46         pix_fmts.push_back (AV_PIX_FMT_BGRA);
  47         pix_fmts.push_back (AV_PIX_FMT_YUV420P); // 0
  48         pix_fmts.push_back (AV_PIX_FMT_YUV411P);
  49         pix_fmts.push_back (AV_PIX_FMT_YUV422P10LE);
  50         pix_fmts.push_back (AV_PIX_FMT_YUV422P16LE);
  51         pix_fmts.push_back (AV_PIX_FMT_YUV444P9LE);
  52         pix_fmts.push_back (AV_PIX_FMT_YUV444P9BE);
  53         pix_fmts.push_back (AV_PIX_FMT_YUV444P10LE);
  54         pix_fmts.push_back (AV_PIX_FMT_YUV444P10BE);
  55         pix_fmts.push_back (AV_PIX_FMT_UYVY422);
  56         pix_fmts.push_back (AV_PIX_FMT_YUVJ420P);
  57         pix_fmts.push_back (AV_PIX_FMT_YUVJ422P);
  58         pix_fmts.push_back (AV_PIX_FMT_YUVJ444P);
  59         pix_fmts.push_back (AV_PIX_FMT_YUVA420P9BE);
  60         pix_fmts.push_back (AV_PIX_FMT_YUVA422P9BE);
  61         pix_fmts.push_back (AV_PIX_FMT_YUVA444P9BE);
  62         pix_fmts.push_back (AV_PIX_FMT_YUVA420P9LE);
  63         pix_fmts.push_back (AV_PIX_FMT_YUVA422P9LE);
  64         pix_fmts.push_back (AV_PIX_FMT_YUVA444P9LE);
  65         pix_fmts.push_back (AV_PIX_FMT_YUVA420P10BE);
  66         pix_fmts.push_back (AV_PIX_FMT_YUVA422P10BE);
  67         pix_fmts.push_back (AV_PIX_FMT_YUVA444P10BE);
  68         pix_fmts.push_back (AV_PIX_FMT_YUVA420P10LE);
  69         pix_fmts.push_back (AV_PIX_FMT_YUVA422P10LE);
  70         pix_fmts.push_back (AV_PIX_FMT_YUVA444P10LE);
  71         pix_fmts.push_back (AV_PIX_FMT_YUVA420P16BE);
  72         pix_fmts.push_back (AV_PIX_FMT_YUVA422P16BE);
  73         pix_fmts.push_back (AV_PIX_FMT_YUVA444P16BE);
  74         pix_fmts.push_back (AV_PIX_FMT_YUVA420P16LE);
  75         pix_fmts.push_back (AV_PIX_FMT_YUVA422P16LE);
  76         pix_fmts.push_back (AV_PIX_FMT_YUVA444P16LE);
  77         pix_fmts.push_back (AV_PIX_FMT_RGB555LE); // 46
  78
  79         int N = 0;
  80         for (list<AVPixelFormat>::const_iterator i = pix_fmts.begin(); i != pix_fmts.end(); ++i) {
  81                 boost::shared_ptr<Image> foo (new Image (*i, in_size, true));
  82                 foo->make_black ();
  83                 boost::shared_ptr<Image> bar = foo->scale (out_size, dcp::YUV_TO_RGB_REC601, PIX_FMT_RGB24, true);
  84
  85                 uint8_t* p = bar->data()[0];
  86                 for (int y = 0; y < bar->size().height; ++y) {
  87                         uint8_t* q = p;
  88                         for (int x = 0; x < bar->line_size()[0]; ++x) {
  89                                 if (*q != 0) {
  90                                         std::cerr << "x=" << x << ", (x%3)=" << (x%3) << "\n";
  91                                 }
  92                                 BOOST_CHECK_EQUAL (*q++, 0);
  93                         }
  94                         p += bar->stride()[0];
  95                 }
  96
  97                 ++N;
  98         }
  99 }