Add wx_ptr and use it instead of ScopeGuard in a lot of places.
[dcpomatic.git] / src / wx / content_menu.cc
index fe117aac03521e9f1d198a512d62b84667d91852..5957ac06df18d0985991759bc883b083fca26f24 100644 (file)
 */
 
 
+#include "auto_crop_dialog.h"
 #include "content_advanced_dialog.h"
 #include "content_menu.h"
 #include "content_properties_dialog.h"
+#include "film_viewer.h"
 #include "repeat_dialog.h"
-#include "timeline_audio_content_view.h"
 #include "timeline_video_content_view.h"
+#include "timeline_audio_content_view.h"
+#include "wx_ptr.h"
 #include "wx_util.h"
 #include "lib/audio_content.h"
 #include "lib/config.h"
 #include "lib/content_factory.h"
+#include "lib/constants.h"
 #include "lib/copy_dcp_details_to_film.h"
 #include "lib/dcp_content.h"
 #include "lib/dcp_examiner.h"
 #include "lib/exceptions.h"
 #include "lib/ffmpeg_content.h"
 #include "lib/film.h"
+#include "lib/find_missing.h"
+#include "lib/guess_crop.h"
 #include "lib/image_content.h"
 #include "lib/job_manager.h"
 #include "lib/playlist.h"
+#include "lib/video_content.h"
 #include <dcp/cpl.h>
 #include <dcp/decrypted_kdm.h>
 #include <dcp/exceptions.h>
 #include <dcp/search.h>
+#include <dcp/warnings.h>
+LIBDCP_DISABLE_WARNINGS
 #include <wx/dirdlg.h>
 #include <wx/wx.h>
+LIBDCP_ENABLE_WARNINGS
 
 
 using std::dynamic_pointer_cast;
@@ -58,6 +68,7 @@ using boost::optional;
 #if BOOST_VERSION >= 106100
 using namespace boost::placeholders;
 #endif
+using namespace dcpomatic;
 
 
 enum {
@@ -68,6 +79,7 @@ enum {
        ID_properties,
        ID_advanced,
        ID_re_examine,
+       ID_auto_crop,
        ID_kdm,
        ID_ov,
        ID_choose_cpl,
@@ -76,17 +88,19 @@ enum {
 };
 
 
-ContentMenu::ContentMenu (wxWindow* p)
+ContentMenu::ContentMenu(wxWindow* p, FilmViewer& viewer)
        : _menu (new wxMenu)
        , _parent (p)
        , _pop_up_open (false)
+       , _viewer (viewer)
 {
        _repeat = _menu->Append (ID_repeat, _("Repeat..."));
        _join = _menu->Append (ID_join, _("Join"));
        _find_missing = _menu->Append (ID_find_missing, _("Find missing..."));
+       _re_examine = _menu->Append (ID_re_examine, _("Re-examine..."));
+       _auto_crop = _menu->Append (ID_auto_crop, _("Auto-crop..."));
        _properties = _menu->Append (ID_properties, _("Properties..."));
        _advanced = _menu->Append (ID_advanced, _("Advanced settings..."));
-       _re_examine = _menu->Append (ID_re_examine, _("Re-examine..."));
        _menu->AppendSeparator ();
        _kdm = _menu->Append (ID_kdm, _("Add KDM..."));
        _ov = _menu->Append (ID_ov, _("Add OV..."));
@@ -102,6 +116,7 @@ ContentMenu::ContentMenu (wxWindow* p)
        _parent->Bind (wxEVT_MENU, boost::bind (&ContentMenu::properties, this), ID_properties);
        _parent->Bind (wxEVT_MENU, boost::bind (&ContentMenu::advanced, this), ID_advanced);
        _parent->Bind (wxEVT_MENU, boost::bind (&ContentMenu::re_examine, this), ID_re_examine);
+       _parent->Bind (wxEVT_MENU, boost::bind (&ContentMenu::auto_crop, this), ID_auto_crop);
        _parent->Bind (wxEVT_MENU, boost::bind (&ContentMenu::kdm, this), ID_kdm);
        _parent->Bind (wxEVT_MENU, boost::bind (&ContentMenu::ov, this), ID_ov);
        _parent->Bind (wxEVT_MENU, boost::bind (&ContentMenu::set_dcp_settings, this), ID_set_dcp_settings);
@@ -136,6 +151,7 @@ ContentMenu::popup (weak_ptr<Film> film, ContentList c, TimelineContentViewList
        _properties->Enable (_content.size() == 1);
        _advanced->Enable (_content.size() == 1);
        _re_examine->Enable (!_content.empty ());
+       _auto_crop->Enable (_content.size() == 1);
 
        if (_content.size() == 1) {
                auto dcp = dynamic_pointer_cast<DCPContent> (_content.front());
@@ -329,46 +345,22 @@ ContentMenu::find_missing ()
        boost::filesystem::path path;
 
        if ((ic && !ic->still ()) || dc) {
-               auto d = new wxDirDialog (0, _("Choose a folder"), wxT (""), wxDD_DIR_MUST_EXIST);
+               auto d = new wxDirDialog (nullptr, _("Choose a folder"), wxT (""), wxDD_DIR_MUST_EXIST);
                r = d->ShowModal ();
-               path = wx_to_std (d->GetPath ());
+               path = wx_to_std (d->GetPath());
                d->Destroy ();
        } else {
-               auto d = new wxFileDialog (0, _("Choose a file"), wxT (""), wxT (""), wxT ("*.*"));
+               auto d = new wxFileDialog (nullptr, _("Choose a file"), wxT (""), wxT (""), wxT ("*.*"));
                r = d->ShowModal ();
-               path = wx_to_std (d->GetPath ());
+               path = wx_to_std (d->GetPath());
                d->Destroy ();
        }
 
-       list<shared_ptr<Content>> content;
-
-       if (r == wxID_OK) {
-               if (dc) {
-                       content.push_back (make_shared<DCPContent>(path));
-               } else {
-                       content = content_factory (path);
-               }
-       }
-
-       if (content.empty ()) {
+       if (r == wxID_CANCEL) {
                return;
        }
 
-       for (auto i: content) {
-               auto j = make_shared<ExamineContentJob>(film, i);
-
-               j->Finished.connect (
-                       bind (
-                               &ContentMenu::maybe_found_missing,
-                               this,
-                               std::weak_ptr<Job> (j),
-                               std::weak_ptr<Content> (_content.front ()),
-                               std::weak_ptr<Content> (i)
-                               )
-                       );
-
-               JobManager::instance()->add (j);
-       }
+       dcpomatic::find_missing (film->content(), path);
 }
 
 void
@@ -380,30 +372,10 @@ ContentMenu::re_examine ()
        }
 
        for (auto i: _content) {
-               JobManager::instance()->add (shared_ptr<Job> (new ExamineContentJob (film, i)));
+               JobManager::instance()->add(make_shared<ExamineContentJob>(film, i));
        }
 }
 
-void
-ContentMenu::maybe_found_missing (weak_ptr<Job> j, weak_ptr<Content> oc, weak_ptr<Content> nc)
-{
-       auto job = j.lock ();
-       if (!job || !job->finished_ok ()) {
-               return;
-       }
-
-       auto old_content = oc.lock ();
-       auto new_content = nc.lock ();
-       DCPOMATIC_ASSERT (old_content);
-       DCPOMATIC_ASSERT (new_content);
-
-       if (new_content->digest() != old_content->digest()) {
-               error_dialog (0, _("The content file(s) you specified are not the same as those that are missing.  Either try again with the correct content file or remove the missing content."));
-               return;
-       }
-
-       old_content->set_paths (new_content->paths());
-}
 
 void
 ContentMenu::kdm ()
@@ -491,9 +463,32 @@ ContentMenu::properties ()
 void
 ContentMenu::advanced ()
 {
-       auto d = new ContentAdvancedDialog (_parent, _content.front());
-       d->ShowModal ();
-       d->Destroy ();
+       DCPOMATIC_ASSERT(!_content.empty());
+
+       auto content = _content.front();
+       auto dialog = make_wx<ContentAdvancedDialog>(_parent, content);
+
+       if (dialog->ShowModal() == wxID_CANCEL) {
+               return;
+       }
+
+       if (content->video) {
+               content->video->set_use(!dialog->ignore_video());
+               content->video->set_burnt_subtitle_language(dialog->burnt_subtitle_language());
+       }
+
+       auto ffmpeg = dynamic_pointer_cast<FFmpegContent>(content);
+       if (ffmpeg) {
+               ffmpeg->set_filters(dialog->filters());
+       }
+
+       if (dialog->video_frame_rate()) {
+               auto film = _film.lock();
+               DCPOMATIC_ASSERT(film);
+               content->set_video_frame_rate(film, *dialog->video_frame_rate());
+       } else {
+               content->unset_video_frame_rate();
+       }
 }
 
 
@@ -522,3 +517,80 @@ ContentMenu::cpl_selected (wxCommandEvent& ev)
        DCPOMATIC_ASSERT (film);
        JobManager::instance()->add (make_shared<ExamineContentJob>(film, dcp));
 }
+
+
+void
+ContentMenu::auto_crop ()
+{
+       DCPOMATIC_ASSERT (_content.size() == 1);
+
+       auto film = _film.lock ();
+       DCPOMATIC_ASSERT (film);
+
+       auto update_viewer = [this](Crop crop) {
+               auto film = _film.lock();
+               DCPOMATIC_ASSERT (film);
+               auto const content = _content.front();
+               auto const current_crop = content->video->actual_crop();
+               _viewer.set_crop_guess(
+                       dcpomatic::Rect<float>(
+                               static_cast<float>(std::max(0, crop.left - current_crop.left)) / content->video->size().width,
+                               static_cast<float>(std::max(0, crop.top - current_crop.top)) / content->video->size().height,
+                               1.0f - (static_cast<float>(std::max(0, crop.left - current_crop.left + crop.right - current_crop.right)) / content->video->size().width),
+                               1.0f - (static_cast<float>(std::max(0, crop.top - current_crop.top + crop.bottom - current_crop.bottom)) / content->video->size().height)
+                               ));
+       };
+
+       auto guess_crop_for_content = [this, film]() {
+               auto position = _viewer.position_in_content(_content.front()).get_value_or(
+                       ContentTime::from_frames(_content.front()->video->length(), _content.front()->video_frame_rate().get_value_or(24))
+                       );
+               return guess_crop(film, _content.front(), Config::instance()->auto_crop_threshold(), position);
+       };
+
+       /* Make an initial guess in the view and open the dialog */
+
+       auto const crop = guess_crop_for_content ();
+       update_viewer (crop);
+
+       if (_auto_crop_dialog) {
+               _auto_crop_dialog->Destroy();
+               _auto_crop_dialog = nullptr;
+       }
+       _auto_crop_dialog = new AutoCropDialog (_parent, crop);
+       _auto_crop_dialog->Show ();
+
+       /* Update the dialog and view when the crop threshold changes */
+       _auto_crop_config_connection = Config::instance()->Changed.connect([this, guess_crop_for_content, update_viewer](Config::Property property) {
+               auto film = _film.lock();
+               DCPOMATIC_ASSERT (film);
+               if (property == Config::AUTO_CROP_THRESHOLD) {
+                       auto const crop = guess_crop_for_content();
+                       _auto_crop_dialog->set(crop);
+                       update_viewer(crop);
+               }
+       });
+
+       /* Also update the dialog and view when we're looking at a different frame */
+       _auto_crop_viewer_connection = _viewer.ImageChanged.connect([this, guess_crop_for_content, update_viewer](shared_ptr<PlayerVideo>) {
+               auto const crop = guess_crop_for_content();
+               _auto_crop_dialog->set(crop);
+               update_viewer(crop);
+       });
+
+       /* Handle the user closing the dialog (with OK or cancel) */
+       _auto_crop_dialog->Bind(wxEVT_BUTTON, [this](wxCommandEvent& ev) {
+               _auto_crop_config_connection.disconnect ();
+               _auto_crop_viewer_connection.disconnect ();
+               if (ev.GetId() == wxID_OK) {
+                       _content.front()->video->set_crop(_auto_crop_dialog->get());
+               }
+               _auto_crop_dialog->Show (false);
+               _viewer.unset_crop_guess ();
+       });
+
+       /* Update the view when something in the dialog is changed */
+       _auto_crop_dialog->Changed.connect([update_viewer](Crop crop) {
+               update_viewer (crop);
+       });
+}