X-Git-Url: https://main.carlh.net/gitweb/?a=blobdiff_plain;f=src%2Fwx%2Ffilm_viewer.cc;h=3d8198457db1a3c342290087fe938408d63a11c1;hb=21d3f3889ede9bd4b2c48424715e44ad2c405500;hp=8c1ce34e0486df31703a53cf2ecf17c28453bb21;hpb=cc8973de75b9c484decac5f6e13fcba62658b8e6;p=dcpomatic.git diff --git a/src/wx/film_viewer.cc b/src/wx/film_viewer.cc index 8c1ce34e0..3d8198457 100644 --- a/src/wx/film_viewer.cc +++ b/src/wx/film_viewer.cc @@ -18,304 +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/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) - , _frame_rebuild_needed (false) - , _composition_needed (false) - {} - - /** Handle a paint event */ - void paint_event (wxPaintEvent& ev) - { - if (!_film || _film->thumbs().size() == 0) { - wxPaintDC dc (this); - return; - } + _panel->SetDoubleBuffered (true); +#if wxMAJOR_VERSION == 2 && wxMINOR_VERSION >= 9 + _panel->SetBackgroundStyle (wxBG_STYLE_PAINT); +#endif + + _v_sizer = new wxBoxSizer (wxVERTICAL); + SetSizer (_v_sizer); - if (_frame_rebuild_needed) { - _image.reset (new wxImage (std_to_wx (_film->thumb_file (_index)))); + _v_sizer->Add (_panel, 1, wxEXPAND); - _subtitle.reset (); - pair s = _film->thumb_subtitle (_index); - if (!s.second.empty ()) { - _subtitle.reset (new SubtitleView (s.first, std_to_wx (s.second))); - } + wxBoxSizer* h_sizer = new wxBoxSizer (wxHORIZONTAL); + h_sizer->Add (_play_button, 0, wxEXPAND); + h_sizer->Add (_slider, 1, wxEXPAND); - _frame_rebuild_needed = false; - compose (); - } + _v_sizer->Add (h_sizer, 0, wxEXPAND); - if (_composition_needed) { - compose (); - } + _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() && _subtitle) { - dc.DrawBitmap (*_subtitle->bitmap, _subtitle->transformed_area.x, _subtitle->transformed_area.y, true); - } - } - - /** Handle a size event */ - void size_event (wxSizeEvent &) - { - if (!_image) { - return; - } - - recompose (); - } + JobManager::instance()->ActiveJobsChanged.connect ( + bind (&FilmViewer::active_jobs_changed, this, _1) + ); +} - /** @param n Thumbnail index */ - void set (int n) +void +FilmViewer::film_changed (Film::Property p) +{ + switch (p) { + case Film::FORMAT: + calculate_sizes (); + update_from_raw (); + break; + case Film::CONTENT: { - _index = n; - _frame_rebuild_needed = true; - 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; } - - void set_film (Film* f) - { - _film = f; - if (!_film) { - clear (); - _frame_rebuild_needed = true; - Refresh (); - } else { - _frame_rebuild_needed = true; - 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; } +} - /** Clear our thumbnail image */ - void clear () - { - _bitmap.reset (); - _image.reset (); - _subtitle.reset (); +void +FilmViewer::set_film (shared_ptr f) +{ + if (_film == f) { + return; } + + _film = f; - void recompose () - { - _composition_needed = true; - Refresh (); + if (!_film) { + return; } - DECLARE_EVENT_TABLE (); + _film->Changed.connect (boost::bind (&FilmViewer::film_changed, this, _1)); -private: - - void compose () - { - _composition_needed = false; - - if (!_film || !_image) { - return; - } - - /* Size of the view */ - int vw, vh; - GetSize (&vw, &vh); - - Crop const fc = _film->crop (); + 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); +} - /* Cropped rectangle */ - Rect cropped_area ( - fc.left, - fc.top, - _image->GetWidth() - (fc.left + fc.right), - _image->GetHeight() - (fc.top + fc.bottom) - ); +void +FilmViewer::decoder_changed () +{ + if (_decoders.video->seek_to_last ()) { + return; + } - /* Target ratio */ - float const target = _film->format() ? _film->format()->ratio_as_float (_film) : 1.78; + get_frame (); + _panel->Refresh (); + _panel->Update (); +} - _transformed_image = _image->GetSubImage (wxRect (cropped_area.x, cropped_area.y, cropped_area.width, cropped_area.height)); +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 */ - _transformed_image.Rescale (vh * target, vh, wxIMAGE_QUALITY_HIGH); - x_scale = vh * target / cropped_area.width; - y_scale = float (vh) / cropped_area.height; - } else { - /* view is shorter (horizontally) than the ratio; fit width */ - _transformed_image.Rescale (vw, vw / target, wxIMAGE_QUALITY_HIGH); - x_scale = float (vw) / cropped_area.width; - y_scale = (vw / target) / cropped_area.height; + 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 (_transformed_image)); - - if (_subtitle) { - _subtitle->transformed_area = subtitle_transformed_area ( - x_scale, y_scale, _subtitle->base_area, _film->subtitle_offset(), _film->subtitle_scale() - ); +void +FilmViewer::paint_panel (wxPaintEvent &) +{ + wxPaintDC dc (_panel); - _subtitle->transformed_image = _subtitle->base_image; - _subtitle->transformed_image.Rescale (_subtitle->transformed_area.width, _subtitle->transformed_area.height, wxIMAGE_QUALITY_HIGH); - _subtitle->transformed_area.x -= rint (_film->crop().left * x_scale); - _subtitle->transformed_area.y -= rint (_film->crop().top * y_scale); - _subtitle->bitmap.reset (new wxBitmap (_subtitle->transformed_image)); - } + if (_clear_required) { + dc.Clear (); + _clear_required = false; } - Film* _film; - shared_ptr _image; - wxImage _transformed_image; - /** currently-displayed thumbnail index */ - int _index; - shared_ptr _bitmap; - bool _frame_rebuild_needed; - bool _composition_needed; - - struct SubtitleView - { - SubtitleView (Position p, wxString const & i) - : base_image (i) - { - base_area.x = p.x; - base_area.y = p.y; - base_area.width = base_image.GetWidth (); - base_area.height = base_image.GetHeight (); - } + if (!_display_frame || !_film || !_out_width || !_out_height) { + dc.Clear (); + return; + } - Rect base_area; - Rect transformed_area; - wxImage base_image; - wxImage transformed_image; - shared_ptr bitmap; - }; + wxImage frame (_out_width, _out_height, _display_frame->data()[0], true); + wxBitmap frame_bitmap (frame); + dc.DrawBitmap (frame_bitmap, 0, 0); - shared_ptr _subtitle; -}; + 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); + } +} -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) +void +FilmViewer::slider_moved (wxScrollEvent &) { - _sizer = new wxBoxSizer (wxVERTICAL); - SetSizer (_sizer); + if (!_film || !_film->length()) { + return; + } - _thumb_panel = new ThumbPanel (this, f); - _sizer->Add (_thumb_panel, 1, wxEXPAND); - - int const m = max ((size_t) 1, f ? f->thumbs().size() - 1 : 0); - _slider = new wxSlider (this, wxID_ANY, 0, 0, m); - _sizer->Add (_slider, 0, wxEXPAND | wxLEFT | wxRIGHT); - set_thumbnail (0); - - _slider->Connect (wxID_ANY, wxEVT_COMMAND_SLIDER_UPDATED, wxCommandEventHandler (FilmViewer::slider_changed), 0, this); + if (_decoders.video->seek (_slider->GetValue() * _film->length().get() / (4096 * _film->frames_per_second()))) { + return; + } + + get_frame (); + _panel->Refresh (); + _panel->Update (); +} - set_film (_film); +void +FilmViewer::panel_sized (wxSizeEvent& ev) +{ + _panel_width = ev.GetSize().GetWidth(); + _panel_height = ev.GetSize().GetHeight(); + calculate_sizes (); + update_from_raw (); } void -FilmViewer::set_thumbnail (int n) +FilmViewer::update_from_raw () { - if (_film == 0 || int (_film->thumbs().size()) <= n) { + if (!_raw_frame) { return; } - _thumb_panel->set (n); + raw_to_display (); + + _panel->Refresh (); + _panel->Update (); } void -FilmViewer::slider_changed (wxCommandEvent &) +FilmViewer::raw_to_display () { - set_thumbnail (_slider->GetValue ()); -} + if (!_raw_frame || _out_width < 64 || _out_height < 64 || !_film) { + return; + } -void -FilmViewer::film_changed (FilmState::Property p) -{ - switch (p) { - case FilmState::THUMBS: - if (_film && _film->thumbs().size() > 1) { - _slider->SetRange (0, _film->thumbs().size() - 1); - } else { - _thumb_panel->clear (); - _slider->SetRange (0, 1); - } + 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() + ); - _slider->SetValue (0); - set_thumbnail (0); - break; - case FilmState::CONTENT: - setup_visibility (); - break; - case FilmState::CROP: - case FilmState::FORMAT: - case FilmState::WITH_SUBTITLES: - case FilmState::SUBTITLE_OFFSET: - case FilmState::SUBTITLE_SCALE: - _thumb_panel->recompose (); - break; - default: - break; + _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::set_film (Film* f) +FilmViewer::calculate_sizes () { - if (_film == f) { + if (!_film) { return; } - _film = f; - _thumb_panel->set_film (_film); + 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; + } +} +void +FilmViewer::play_clicked (wxCommandEvent &) +{ + check_play_state (); +} + +void +FilmViewer::check_play_state () +{ if (!_film) { return; } + + if (_play_button->GetValue()) { + _timer.Start (1000 / _film->frames_per_second()); + } else { + _timer.Stop (); + } +} - _film->Changed.connect (sigc::mem_fun (*this, &FilmViewer::film_changed)); - film_changed (Film::CROP); - film_changed (Film::THUMBS); - setup_visibility (); +void +FilmViewer::process_video (shared_ptr image, bool, shared_ptr sub) +{ + _raw_frame = image; + _raw_sub = sub; + + 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); } +