test/video_content_scale_test.cc

   1 /*
   2     Copyright (C) 2015 Carl Hetherington <cth@carlh.net>
   3
   4     This file is part of DCP-o-matic.
   5
   6     DCP-o-matic is free software; you can redistribute it and/or modify
   7     it under the terms of the GNU General Public License as published by
   8     the Free Software Foundation; either version 2 of the License, or
   9     (at your option) any later version.
  10
  11     DCP-o-matic is distributed in the hope that it will be useful,
  12     but WITHOUT ANY WARRANTY; without even the implied warranty of
  13     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14     GNU General Public License for more details.
  15
  16     You should have received a copy of the GNU General Public License
  17     along with DCP-o-matic.  If not, see <http://www.gnu.org/licenses/>.
  18
  19 */
  20
  21 #include <boost/test/unit_test.hpp>
  22 #include "lib/ffmpeg_content.h"
  23 #include "lib/ratio.h"
  24 #include "lib/video_content.h"
  25
  26 using std::list;
  27 using std::string;
  28 using std::cerr;
  29 using boost::shared_ptr;
  30 using boost::optional;
  31
  32 static
  33 void
  34 test (dcp::Size content_size, dcp::Size display_size, dcp::Size film_size, Crop crop, Ratio const * ratio, bool scale, dcp::Size correct)
  35 {
  36         shared_ptr<Film> film;
  37         string s = "<?xml version=\"1.0\" encoding=\"UTF-8\"?>"
  38                 "<Content>"
  39                 "<Type>FFmpeg</Type>"
  40                 "<Path>/home/c.hetherington/DCP/prophet_clip.mkv</Path>"
  41                 "<Digest>f3f23663da5bef6d2cbaa0db066f3351314142710</Digest>"
  42                 "<Position>0</Position>"
  43                 "<TrimStart>0</TrimStart>"
  44                 "<TrimEnd>0</TrimEnd>"
  45                 "<VideoLength>2879</VideoLength>"
  46                 "<VideoWidth>" + raw_convert<string>(content_size.width) + "</VideoWidth>"
  47                 "<VideoHeight>" + raw_convert<string>(content_size.height) + "</VideoHeight>"
  48                 "<VideoFrameRate>23.97602462768555</VideoFrameRate>"
  49                 "<OriginalVideoFrameRate>23.97602462768555</OriginalVideoFrameRate>"
  50                 "<VideoFrameType>0</VideoFrameType>"
  51                 "<SampleAspectRatio>1</SampleAspectRatio>"
  52                 "<BitsPerPixel>12</BitsPerPixel>"
  53                 "<LeftCrop>" + raw_convert<string>(crop.left) + "</LeftCrop>"
  54                 "<RightCrop>" + raw_convert<string>(crop.right) + "</RightCrop>"
  55                 "<TopCrop>" + raw_convert<string>(crop.top) + "</TopCrop>"
  56                 "<BottomCrop>" + raw_convert<string>(crop.bottom) + "</BottomCrop>"
  57                 "<Scale>";
  58
  59         if (ratio) {
  60                 s += "<Ratio>" + ratio->id() + "</Ratio>";
  61         } else {
  62                 s += "<Scale>" + string(scale ? "1" : "0") + "</Scale>";
  63         }
  64
  65         s += "</Scale>"
  66                 "<ColourConversion>"
  67                 "<InputGamma>2.4</InputGamma>"
  68                 "<InputGammaLinearised>1</InputGammaLinearised>"
  69                 "<Matrix i=\"0\" j=\"0\">0.4124564</Matrix>"
  70                 "<Matrix i=\"0\" j=\"1\">0.3575761</Matrix>"
  71                 "<Matrix i=\"0\" j=\"2\">0.1804375</Matrix>"
  72                 "<Matrix i=\"1\" j=\"0\">0.2126729</Matrix>"
  73                 "<Matrix i=\"1\" j=\"1\">0.7151522</Matrix>"
  74                 "<Matrix i=\"1\" j=\"2\">0.072175</Matrix>"
  75                 "<Matrix i=\"2\" j=\"0\">0.0193339</Matrix>"
  76                 "<Matrix i=\"2\" j=\"1\">0.119192</Matrix>"
  77                 "<Matrix i=\"2\" j=\"2\">0.9503041</Matrix>"
  78                 "<OutputGamma>2.6</OutputGamma>"
  79                 "</ColourConversion>"
  80                 "<AudioGain>0</AudioGain>"
  81                 "<AudioDelay>0</AudioDelay>"
  82                 "<SubtitleXOffset>0</SubtitleXOffset>"
  83                 "<SubtitleYOffset>0</SubtitleYOffset>"
  84                 "<SubtitleXScale>0</SubtitleXScale>"
  85                 "<SubtitleYScale>0</SubtitleYScale>"
  86                 "</Content>";
  87
  88         shared_ptr<cxml::Document> doc (new cxml::Document ());
  89         doc->read_string (s);
  90
  91         list<string> notes;
  92         shared_ptr<FFmpegContent> vc (new FFmpegContent (film, doc, 10, notes));
  93
  94         optional<VideoContentScale> sc;
  95         if (ratio) {
  96                 sc = VideoContentScale (ratio);
  97         } else {
  98                 sc = VideoContentScale (scale);
  99         }
 100
 101         dcp::Size answer = sc.get().size (vc->video, display_size, film_size);
 102         if (answer != correct) {
 103                 cerr << "Testing " << vc->video->size().width << "x" << vc->video->size().height << "\n";
 104                 cerr << "Testing " << display_size.width << "x" << display_size.height << "\n";
 105                 cerr << answer.width << "x" << answer.height << " instead of " << correct.width << "x" << correct.height << "\n";
 106         }
 107         BOOST_CHECK (answer == correct);
 108 }
 109
 110 /* Test scale and stretch to specified ratio */
 111 BOOST_AUTO_TEST_CASE (video_content_scale_test_to_ratio)
 112 {
 113         /* To DCP */
 114
 115         // Flat in flat container
 116         test (
 117                 dcp::Size (400, 200),
 118                 dcp::Size (1998, 1080),
 119                 dcp::Size (1998, 1080),
 120                 Crop (0, 0, 0, 0),
 121                 Ratio::from_id ("185"),
 122                 true,
 123                 dcp::Size (1998, 1080)
 124                 );
 125
 126         // Scope in flat container
 127         test (
 128                 dcp::Size (400, 200),
 129                 dcp::Size (1998, 1080),
 130                 dcp::Size (1998, 1080),
 131                 Crop (0, 0, 0, 0),
 132                 Ratio::from_id ("239"),
 133                 true,
 134                 dcp::Size (1998, 837)
 135                 );
 136
 137         // Flat in scope container
 138         test (
 139                 dcp::Size (400, 200),
 140                 dcp::Size (2048, 858),
 141                 dcp::Size (2048, 858),
 142                 Crop (0, 0, 0, 0),
 143                 Ratio::from_id ("185"),
 144                 true,
 145                 dcp::Size (1587, 858)
 146                 );
 147
 148
 149         /* To player */
 150
 151         // Flat in flat container
 152         test (
 153                 dcp::Size (400, 200),
 154                 dcp::Size (185, 100),
 155                 dcp::Size (1998, 1080),
 156                 Crop (0, 0, 0, 0),
 157                 Ratio::from_id ("185"),
 158                 true,
 159                 dcp::Size (185, 100)
 160                 );
 161
 162         // Scope in flat container
 163         test (
 164                 dcp::Size (400, 200),
 165                 dcp::Size (185, 100),
 166                 dcp::Size (1998, 1080),
 167                 Crop (0, 0, 0, 0),
 168                 Ratio::from_id ("239"),
 169                 true,
 170                 dcp::Size (185, 78)
 171                 );
 172
 173         // Flat in scope container
 174         test (
 175                 dcp::Size (400, 200),
 176                 dcp::Size (239, 100),
 177                 dcp::Size (2048, 858),
 178                 Crop (0, 0, 0, 0),
 179                 Ratio::from_id ("185"),
 180                 true,
 181                 dcp::Size (185, 100)
 182                 );
 183 }
 184
 185 /* Test no scale */
 186 BOOST_AUTO_TEST_CASE (video_content_scale_no_scale)
 187 {
 188        /* No scale where the content is bigger than even the film container */
 189        test (
 190                dcp::Size (1920, 1080),
 191                dcp::Size (887, 371),
 192                dcp::Size (2048, 858),
 193                Crop (),
 194                0,
 195                false,
 196                dcp::Size (659, 371)
 197                );
 198 }