X-Git-Url: https://main.carlh.net/gitweb/?a=blobdiff_plain;f=src%2Fwx%2Ffilm_viewer.cc;h=3d8198457db1a3c342290087fe938408d63a11c1;hb=21d3f3889ede9bd4b2c48424715e44ad2c405500;hp=85632075c0d4c90ea02a2dcfa186d7c379080960;hpb=f78ce22dc46e42e4f041ebf877fd7199232e4856;p=dcpomatic.git diff --git a/src/wx/film_viewer.cc b/src/wx/film_viewer.cc index 85632075c..3d8198457 100644 --- a/src/wx/film_viewer.cc +++ b/src/wx/film_viewer.cc @@ -18,360 +18,361 @@ */ /** @file src/film_viewer.cc - * @brief A wx widget to view `thumbnails' of a Film. + * @brief A wx widget to view a preview of a Film. */ #include #include +#include #include "lib/film.h" #include "lib/format.h" #include "lib/util.h" -#include "lib/thumbs_job.h" #include "lib/job_manager.h" -#include "lib/film_state.h" #include "lib/options.h" +#include "lib/subtitle.h" +#include "lib/image.h" +#include "lib/scaler.h" +#include "lib/exceptions.h" +#include "lib/examine_content_job.h" #include "film_viewer.h" #include "wx_util.h" +#include "video_decoder.h" -using namespace std; -using namespace boost; +using std::string; +using std::pair; +using std::max; +using std::cout; +using std::list; +using boost::shared_ptr; -class ThumbPanel : public wxPanel +FilmViewer::FilmViewer (shared_ptr f, wxWindow* p) + : wxPanel (p) + , _panel (new wxPanel (this)) + , _slider (new wxSlider (this, wxID_ANY, 0, 0, 4096)) + , _play_button (new wxToggleButton (this, wxID_ANY, wxT ("Play"))) + , _got_frame (false) + , _out_width (0) + , _out_height (0) + , _panel_width (0) + , _panel_height (0) + , _clear_required (false) { -public: - ThumbPanel (wxPanel* parent, Film* film) - : wxPanel (parent) - , _film (film) - , _current_index (-1) - , _pending_index (-1) - , _current_subtitle_offset (0) - , _pending_subtitle_offset (0) - { - } - - /** Handle a paint event */ - void paint_event (wxPaintEvent& ev) - { - if (_current_index != _pending_index) { - _image.reset (new wxImage (std_to_wx (_film->thumb_file (_pending_index)))); - _current_index = _pending_index; + _panel->SetDoubleBuffered (true); +#if wxMAJOR_VERSION == 2 && wxMINOR_VERSION >= 9 + _panel->SetBackgroundStyle (wxBG_STYLE_PAINT); +#endif + + _v_sizer = new wxBoxSizer (wxVERTICAL); + SetSizer (_v_sizer); - _subtitles.clear (); + _v_sizer->Add (_panel, 1, wxEXPAND); - list > s = _film->thumb_subtitles (_pending_index); - for (list >::iterator i = s.begin(); i != s.end(); ++i) { - _subtitles.push_back (SubtitleView (i->first, std_to_wx (i->second))); - } + wxBoxSizer* h_sizer = new wxBoxSizer (wxHORIZONTAL); + h_sizer->Add (_play_button, 0, wxEXPAND); + h_sizer->Add (_slider, 1, wxEXPAND); - setup (); - } - - if (_current_crop != _pending_crop) { - _current_crop = _pending_crop; - setup (); - } + _v_sizer->Add (h_sizer, 0, wxEXPAND); - if (_current_subtitle_offset != _pending_subtitle_offset) { - _current_subtitle_offset = _pending_subtitle_offset; - setup (); - } + _panel->Connect (wxID_ANY, wxEVT_PAINT, wxPaintEventHandler (FilmViewer::paint_panel), 0, this); + _panel->Connect (wxID_ANY, wxEVT_SIZE, wxSizeEventHandler (FilmViewer::panel_sized), 0, this); + _slider->Connect (wxID_ANY, wxEVT_SCROLL_THUMBTRACK, wxScrollEventHandler (FilmViewer::slider_moved), 0, this); + _slider->Connect (wxID_ANY, wxEVT_SCROLL_PAGEUP, wxScrollEventHandler (FilmViewer::slider_moved), 0, this); + _slider->Connect (wxID_ANY, wxEVT_SCROLL_PAGEDOWN, wxScrollEventHandler (FilmViewer::slider_moved), 0, this); + _play_button->Connect (wxID_ANY, wxEVT_COMMAND_TOGGLEBUTTON_CLICKED, wxCommandEventHandler (FilmViewer::play_clicked), 0, this); + _timer.Connect (wxID_ANY, wxEVT_TIMER, wxTimerEventHandler (FilmViewer::timer), 0, this); - wxPaintDC dc (this); - if (_bitmap) { - dc.DrawBitmap (*_bitmap, 0, 0, false); - } + set_film (f); - if (_film->with_subtitles ()) { - for (list::iterator i = _subtitles.begin(); i != _subtitles.end(); ++i) { - dc.DrawBitmap (*i->bitmap, i->cropped_position.x, i->cropped_position.y, true); - } - } - } + JobManager::instance()->ActiveJobsChanged.connect ( + bind (&FilmViewer::active_jobs_changed, this, _1) + ); +} - /** Handle a size event */ - void size_event (wxSizeEvent &) +void +FilmViewer::film_changed (Film::Property p) +{ + switch (p) { + case Film::FORMAT: + calculate_sizes (); + update_from_raw (); + break; + case Film::CONTENT: { - if (!_image) { - return; - } - - setup (); - Refresh (); + shared_ptr o (new DecodeOptions); + o->decode_audio = false; + o->decode_subtitles = true; + o->video_sync = false; + _decoders = decoder_factory (_film, o, 0); + _decoders.video->Video.connect (bind (&FilmViewer::process_video, this, _1, _2, _3)); + _decoders.video->OutputChanged.connect (boost::bind (&FilmViewer::decoder_changed, this)); + _decoders.video->set_subtitle_stream (_film->subtitle_stream()); + calculate_sizes (); + get_frame (); + _panel->Refresh (); + _slider->Show (_film->content_type() == VIDEO); + _play_button->Show (_film->content_type() == VIDEO); + _v_sizer->Layout (); + break; } - - /** @param n Thumbnail index */ - void set (int n) - { - _pending_index = n; - Refresh (); + case Film::WITH_SUBTITLES: + case Film::SUBTITLE_OFFSET: + case Film::SUBTITLE_SCALE: + case Film::SCALER: + update_from_raw (); + break; + case Film::SUBTITLE_STREAM: + _decoders.video->set_subtitle_stream (_film->subtitle_stream ()); + break; + default: + break; } +} - void set_crop (Crop c) - { - _pending_crop = c; - Refresh (); +void +FilmViewer::set_film (shared_ptr f) +{ + if (_film == f) { + return; } + + _film = f; - void set_subtitle_offset (int o) - { - _pending_subtitle_offset = o; - Refresh (); + if (!_film) { + return; } - void set_film (Film* f) - { - _film = f; - if (!_film) { - clear (); - Refresh (); - } else { - setup (); - Refresh (); - } - } + _film->Changed.connect (boost::bind (&FilmViewer::film_changed, this, _1)); - /** Clear our thumbnail image */ - void clear () - { - _bitmap.reset (); - _image.reset (); - _subtitles.clear (); - } + film_changed (Film::CONTENT); + film_changed (Film::CROP); + film_changed (Film::FORMAT); + film_changed (Film::WITH_SUBTITLES); + film_changed (Film::SUBTITLE_OFFSET); + film_changed (Film::SUBTITLE_SCALE); + film_changed (Film::SUBTITLE_STREAM); +} - void refresh () - { - setup (); - Refresh (); +void +FilmViewer::decoder_changed () +{ + if (_decoders.video->seek_to_last ()) { + return; } - DECLARE_EVENT_TABLE (); - -private: - - void setup () - { - if (!_film || !_image) { - return; - } - - /* Size of the view */ - int vw, vh; - GetSize (&vw, &vh); - - /* Cropped rectangle */ - Rectangle cropped ( - _current_crop.left, - _current_crop.top, - _image->GetWidth() - (_current_crop.left + _current_crop.right), - _image->GetHeight() - (_current_crop.top + _current_crop.bottom) - ); - - /* Target ratio */ - float const target = _film->format() ? _film->format()->ratio_as_float (_film) : 1.78; + get_frame (); + _panel->Refresh (); + _panel->Update (); +} - _cropped_image = _image->GetSubImage (wxRect (cropped.x, cropped.y, cropped.w, cropped.h)); +void +FilmViewer::timer (wxTimerEvent &) +{ + if (!_film) { + return; + } + + _panel->Refresh (); + _panel->Update (); - float x_scale = 1; - float y_scale = 1; + get_frame (); - if ((float (vw) / vh) > target) { - /* view is longer (horizontally) than the ratio; fit height */ - _cropped_image.Rescale (vh * target, vh, wxIMAGE_QUALITY_HIGH); - x_scale = vh * target / cropped.w; - y_scale = float (vh) / cropped.h; - } else { - /* view is shorter (horizontally) than the ratio; fit width */ - _cropped_image.Rescale (vw, vw / target, wxIMAGE_QUALITY_HIGH); - x_scale = float (vw) / cropped.w; - y_scale = (vw / target) / cropped.h; + if (_film->length()) { + int const new_slider_position = 4096 * _decoders.video->last_source_time() / (_film->length().get() / _film->frames_per_second()); + if (new_slider_position != _slider->GetValue()) { + _slider->SetValue (new_slider_position); } + } +} - _bitmap.reset (new wxBitmap (_cropped_image)); - - for (list::iterator i = _subtitles.begin(); i != _subtitles.end(); ++i) { - - /* Area of the subtitle graphic within the (uncropped) picture frame */ - Rectangle sub_rect (i->position.x, i->position.y + _current_subtitle_offset, i->image.GetWidth(), i->image.GetHeight()); - /* Hence the subtitle graphic after it has been cropped */ - Rectangle cropped_sub_rect = sub_rect.intersection (cropped); - - /* Get the cropped version of the subtitle image */ - i->cropped_image = i->image.GetSubImage ( - wxRect ( - cropped_sub_rect.x - sub_rect.x, - cropped_sub_rect.y - sub_rect.y, - cropped_sub_rect.w, - cropped_sub_rect.h - ) - ); - - i->cropped_image.Rescale (cropped_sub_rect.w * x_scale, cropped_sub_rect.h * y_scale, wxIMAGE_QUALITY_HIGH); - i->cropped_position = Position ( - cropped_sub_rect.x * x_scale, - (cropped_sub_rect.y - _current_crop.top) * y_scale - ); +void +FilmViewer::paint_panel (wxPaintEvent &) +{ + wxPaintDC dc (_panel); - i->bitmap.reset (new wxBitmap (i->cropped_image)); - } + if (_clear_required) { + dc.Clear (); + _clear_required = false; } - Film* _film; - shared_ptr _image; - wxImage _cropped_image; - /** currently-displayed thumbnail index */ - int _current_index; - int _pending_index; - shared_ptr _bitmap; - Crop _current_crop; - Crop _pending_crop; - int _current_subtitle_offset; - int _pending_subtitle_offset; - - struct SubtitleView - { - SubtitleView (Position p, wxString const & i) - : position (p) - , image (i) - {} - - Position position; - wxImage image; - Position cropped_position; - wxImage cropped_image; - shared_ptr bitmap; - }; - - list _subtitles; -}; - -BEGIN_EVENT_TABLE (ThumbPanel, wxPanel) -EVT_PAINT (ThumbPanel::paint_event) -EVT_SIZE (ThumbPanel::size_event) -END_EVENT_TABLE () - -FilmViewer::FilmViewer (Film* f, wxWindow* p) - : wxPanel (p) - , _film (0) -{ - _sizer = new wxBoxSizer (wxVERTICAL); - SetSizer (_sizer); - - _thumb_panel = new ThumbPanel (this, f); - _sizer->Add (_thumb_panel, 1, wxEXPAND); - - int const max = f ? f->num_thumbs() - 1 : 0; - _slider = new wxSlider (this, wxID_ANY, 0, 0, max); - _sizer->Add (_slider, 0, wxEXPAND | wxLEFT | wxRIGHT); - set_thumbnail (0); + if (!_display_frame || !_film || !_out_width || !_out_height) { + dc.Clear (); + return; + } - _slider->Connect (wxID_ANY, wxEVT_COMMAND_SLIDER_UPDATED, wxCommandEventHandler (FilmViewer::slider_changed), 0, this); + wxImage frame (_out_width, _out_height, _display_frame->data()[0], true); + wxBitmap frame_bitmap (frame); + dc.DrawBitmap (frame_bitmap, 0, 0); - set_film (_film); + if (_film->with_subtitles() && _display_sub) { + wxImage sub (_display_sub->size().width, _display_sub->size().height, _display_sub->data()[0], _display_sub->alpha(), true); + wxBitmap sub_bitmap (sub); + dc.DrawBitmap (sub_bitmap, _display_sub_position.x, _display_sub_position.y); + } } + void -FilmViewer::set_thumbnail (int n) +FilmViewer::slider_moved (wxScrollEvent &) { - if (_film == 0 || _film->num_thumbs() <= n) { + if (!_film || !_film->length()) { return; } - - _thumb_panel->set (n); + + if (_decoders.video->seek (_slider->GetValue() * _film->length().get() / (4096 * _film->frames_per_second()))) { + return; + } + + get_frame (); + _panel->Refresh (); + _panel->Update (); } void -FilmViewer::slider_changed (wxCommandEvent &) +FilmViewer::panel_sized (wxSizeEvent& ev) { - set_thumbnail (_slider->GetValue ()); + _panel_width = ev.GetSize().GetWidth(); + _panel_height = ev.GetSize().GetHeight(); + calculate_sizes (); + update_from_raw (); } void -FilmViewer::film_changed (Film::Property p) +FilmViewer::update_from_raw () { - switch (p) { - case Film::CROP: - _thumb_panel->set_crop (_film->crop ()); - break; - case Film::THUMBS: - if (_film && _film->num_thumbs() > 1) { - _slider->SetRange (0, _film->num_thumbs () - 1); - } else { - _thumb_panel->clear (); - _slider->SetRange (0, 1); - } - - _slider->SetValue (0); - set_thumbnail (0); - break; - case Film::FORMAT: - _thumb_panel->refresh (); - break; - case Film::CONTENT: - setup_visibility (); - _film->examine_content (); - update_thumbs (); - break; - case Film::WITH_SUBTITLES: - _thumb_panel->Refresh (); - break; - case Film::SUBTITLE_OFFSET: - _thumb_panel->set_subtitle_offset (_film->subtitle_offset ()); - break; - default: - break; + if (!_raw_frame) { + return; } + + raw_to_display (); + + _panel->Refresh (); + _panel->Update (); } void -FilmViewer::set_film (Film* f) +FilmViewer::raw_to_display () { - if (_film == f) { + if (!_raw_frame || _out_width < 64 || _out_height < 64 || !_film) { return; } - - _film = f; - _thumb_panel->set_film (_film); + Size old_size; + if (_display_frame) { + old_size = _display_frame->size(); + } + + /* Get a compacted image as we have to feed it to wxWidgets */ + _display_frame = _raw_frame->scale_and_convert_to_rgb (Size (_out_width, _out_height), 0, _film->scaler(), false); + + if (old_size != _display_frame->size()) { + _clear_required = true; + } + + if (_raw_sub) { + Rect tx = subtitle_transformed_area ( + float (_out_width) / _film->size().width, + float (_out_height) / _film->size().height, + _raw_sub->area(), _film->subtitle_offset(), _film->subtitle_scale() + ); + + _display_sub.reset (new RGBPlusAlphaImage (_raw_sub->image()->scale (tx.size(), _film->scaler(), false))); + _display_sub_position = tx.position(); + } else { + _display_sub.reset (); + } +} + +void +FilmViewer::calculate_sizes () +{ if (!_film) { return; } + + float const panel_ratio = static_cast (_panel_width) / _panel_height; + float const film_ratio = _film->format() ? _film->format()->ratio_as_float(_film) : 1.78; + if (panel_ratio < film_ratio) { + /* panel is less widscreen than the film; clamp width */ + _out_width = _panel_width; + _out_height = _out_width / film_ratio; + } else { + /* panel is more widescreen than the film; clamp heignt */ + _out_height = _panel_height; + _out_width = _out_height * film_ratio; + } +} - _film->Changed.connect (sigc::mem_fun (*this, &FilmViewer::film_changed)); - film_changed (Film::CROP); - film_changed (Film::THUMBS); - _thumb_panel->refresh (); - setup_visibility (); +void +FilmViewer::play_clicked (wxCommandEvent &) +{ + check_play_state (); } void -FilmViewer::update_thumbs () +FilmViewer::check_play_state () { if (!_film) { return; } + + if (_play_button->GetValue()) { + _timer.Start (1000 / _film->frames_per_second()); + } else { + _timer.Stop (); + } +} - _film->update_thumbs_pre_gui (); +void +FilmViewer::process_video (shared_ptr image, bool, shared_ptr sub) +{ + _raw_frame = image; + _raw_sub = sub; - shared_ptr s = _film->state_copy (); - shared_ptr o (new Options (s->dir ("thumbs"), ".png", "")); - o->out_size = _film->size (); - o->apply_crop = false; - o->decode_audio = false; - o->decode_video_frequency = 128; - - shared_ptr j (new ThumbsJob (s, o, _film->log(), shared_ptr ())); - j->Finished.connect (sigc::mem_fun (_film, &Film::update_thumbs_post_gui)); - JobManager::instance()->add (j); + raw_to_display (); + + _got_frame = true; } void -FilmViewer::setup_visibility () +FilmViewer::get_frame () { - if (!_film) { - return; + /* Clear our raw frame in case we don't get a new one */ + _raw_frame.reset (); + + try { + _got_frame = false; + while (!_got_frame) { + if (_decoders.video->pass ()) { + /* We didn't get a frame before the decoder gave up, + so clear our display frame. + */ + _display_frame.reset (); + break; + } + } + } catch (DecodeError& e) { + error_dialog (this, String::compose ("Could not decode video for view (%1)", e.what())); } +} - ContentType const c = _film->content_type (); - _slider->Show (c == VIDEO); +void +FilmViewer::active_jobs_changed (bool a) +{ + if (a) { + list > jobs = JobManager::instance()->get (); + list >::iterator i = jobs.begin (); + while (i != jobs.end() && boost::dynamic_pointer_cast (*i) == 0) { + ++i; + } + + if (i == jobs.end() || (*i)->finished()) { + /* no examine content job running, so we're ok to use the viewer */ + a = false; + } + } + + _slider->Enable (!a); + _play_button->Enable (!a); } +