2 Copyright (C) 2015-2021 Carl Hetherington <cth@carlh.net>
4 This file is part of DCP-o-matic.
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.
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.
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/>.
22 #include "film_viewer.h"
23 #include "video_waveform_plot.h"
25 #include "lib/dcp_video.h"
27 #include "lib/image.h"
28 #include "lib/player_video.h"
29 #include <dcp/locale_convert.h>
30 #include <dcp/openjpeg_image.h>
31 #include <dcp/warnings.h>
32 LIBDCP_DISABLE_WARNINGS
33 #include <wx/graphics.h>
34 #include <wx/rawbmp.h>
35 LIBDCP_ENABLE_WARNINGS
36 #include <boost/bind/bind.hpp>
39 using std::make_shared;
42 using std::shared_ptr;
45 #if BOOST_VERSION >= 106100
46 using namespace boost::placeholders;
48 using dcp::locale_convert;
51 int const VideoWaveformPlot::_vertical_margin = 8;
52 int const VideoWaveformPlot::_pixel_values = 4096;
53 int const VideoWaveformPlot::_x_axis_width = 52;
56 VideoWaveformPlot::VideoWaveformPlot (wxWindow* parent, weak_ptr<const Film> film, weak_ptr<FilmViewer> viewer)
57 : wxPanel (parent, wxID_ANY, wxDefaultPosition, wxDefaultSize, wxFULL_REPAINT_ON_RESIZE)
61 SetDoubleBuffered (true);
64 auto fv = viewer.lock ();
65 DCPOMATIC_ASSERT (fv);
67 _viewer_connection = fv->ImageChanged.connect (boost::bind(&VideoWaveformPlot::set_image, this, _1));
69 Bind (wxEVT_PAINT, boost::bind(&VideoWaveformPlot::paint, this));
70 Bind (wxEVT_SIZE, boost::bind(&VideoWaveformPlot::sized, this, _1));
71 Bind (wxEVT_MOTION, boost::bind(&VideoWaveformPlot::mouse_moved, this, _1));
73 SetMinSize (wxSize (640, 512));
74 SetBackgroundColour (wxColour (0, 0, 0));
79 VideoWaveformPlot::paint ()
92 auto gc = wxGraphicsContext::Create (dc);
97 int const height = _waveform->size().height;
99 gc->SetPen (wxPen (wxColour (255, 255, 255), 1, wxPENSTYLE_SOLID));
101 gc->SetFont (gc->CreateFont (*wxSMALL_FONT, wxColour (255, 255, 255)));
104 double label_descent;
105 double label_leading;
106 gc->GetTextExtent (wxT ("1024"), &label_width, &label_height, &label_descent, &label_leading);
110 gc->GetTextExtent (wxT ("0"), &w, &label_height, &label_descent, &label_leading);
111 extra[0] = label_width - w;
112 gc->GetTextExtent (wxT ("64"), &w, &label_height, &label_descent, &label_leading);
113 extra[1] = label_width - w;
114 gc->GetTextExtent (wxT ("512"), &w, &label_height, &label_descent, &label_leading);
115 extra[2] = label_width - w;
118 while (height / label_gaps > 64) {
122 for (int i = 0; i < label_gaps + 1; ++i) {
123 auto p = gc->CreatePath ();
124 int const y = _vertical_margin + height - (i * height / label_gaps) - 1;
125 p.MoveToPoint (label_width + 8, y);
126 p.AddLineToPoint (_x_axis_width - 4, y);
129 int const n = i * _pixel_values / label_gaps;
132 } else if (n < 100) {
134 } else if (n < 1000) {
137 gc->DrawText (std_to_wx(locale_convert<string>(n)), x, y - (label_height / 2));
140 wxImage waveform (_waveform->size().width, height, _waveform->data()[0], true);
141 wxBitmap bitmap (waveform);
142 gc->DrawBitmap (bitmap, _x_axis_width, _vertical_margin, _waveform->size().width, height);
149 VideoWaveformPlot::create_waveform ()
157 auto const image_size = _image->size();
158 int const waveform_height = GetSize().GetHeight() - _vertical_margin * 2;
159 _waveform = make_shared<Image>(AV_PIX_FMT_RGB24, dcp::Size (image_size.width, waveform_height), Image::Alignment::PADDED);
161 for (int x = 0; x < image_size.width; ++x) {
163 /* Work out one vertical `slice' of waveform pixels. Each value in
164 strip is the number of samples in image with the corresponding group of
167 int strip[waveform_height];
168 memset (strip, 0, waveform_height * sizeof(int));
170 int* ip = _image->data (_component) + x;
171 for (int y = 0; y < image_size.height; ++y) {
172 strip[*ip * waveform_height / _pixel_values]++;
173 ip += image_size.width;
176 /* Copy slice into the waveform */
177 uint8_t* wp = _waveform->data()[0] + x * 3;
178 for (int y = waveform_height - 1; y >= 0; --y) {
179 wp[0] = wp[1] = wp[2] = min (255, (strip[y] * 255 / waveform_height) * _contrast);
180 wp += _waveform->stride()[0];
184 _waveform = _waveform->scale (
185 dcp::Size (GetSize().GetWidth() - _x_axis_width, waveform_height),
186 dcp::YUVToRGB::REC709, AV_PIX_FMT_RGB24, Image::Alignment::COMPACT, false
192 VideoWaveformPlot::set_image (shared_ptr<PlayerVideo> image)
198 /* We must copy the PlayerVideo here as we will call ::image() on it, potentially
199 with a different pixel_format than was used when ::prepare() was called.
201 _image = DCPVideo::convert_to_xyz (image->shallow_copy(), [](dcp::NoteType, string) {});
208 VideoWaveformPlot::sized (wxSizeEvent &)
215 VideoWaveformPlot::set_enabled (bool e)
222 VideoWaveformPlot::set_component (int c)
230 /** Set `contrast', i.e. a fudge multiplication factor to make low-level signals easier to see,
234 VideoWaveformPlot::set_contrast (int b)
243 VideoWaveformPlot::mouse_moved (wxMouseEvent& ev)
254 auto film = _film.lock ();
259 auto const full = film->frame_size ();
261 auto const xs = static_cast<double> (full.width) / _waveform->size().width;
262 int const x1 = max (0, min (full.width - 1, int (floor (ev.GetPosition().x - _x_axis_width - 0.5) * xs)));
263 int const x2 = max (0, min (full.width - 1, int (floor (ev.GetPosition().x - _x_axis_width + 0.5) * xs)));
265 auto const ys = static_cast<double> (_pixel_values) / _waveform->size().height;
266 int const fy = _waveform->size().height - (ev.GetPosition().y - _vertical_margin);
267 int const y1 = max (0, min (_pixel_values - 1, int (floor (fy - 0.5) * ys)));
268 int const y2 = max (0, min (_pixel_values - 1, int (floor (fy + 0.5) * ys)));
270 MouseMoved (x1, x2, y1, y2);