Cleanup: remove some unnecessary includes.
[dcpomatic.git] / src / wx / content_menu.cc
index 3b49a85cc3fe5575616cc56b6775d8945b7fa690..a92d505d237b597fc2c845120cfeacee42013eba 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_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>
@@ -61,6 +67,7 @@ using boost::optional;
 #if BOOST_VERSION >= 106100
 using namespace boost::placeholders;
 #endif
+using namespace dcpomatic;
 
 
 enum {
@@ -71,6 +78,7 @@ enum {
        ID_properties,
        ID_advanced,
        ID_re_examine,
+       ID_auto_crop,
        ID_kdm,
        ID_ov,
        ID_choose_cpl,
@@ -79,17 +87,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..."));
@@ -105,6 +115,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);
@@ -139,6 +150,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());
@@ -210,9 +222,8 @@ ContentMenu::repeat ()
                return;
        }
 
-       auto d = new RepeatDialog (_parent);
+       auto d = make_wx<RepeatDialog>(_parent);
        if (d->ShowModal() != wxID_OK) {
-               d->Destroy ();
                return;
        }
 
@@ -222,7 +233,6 @@ ContentMenu::repeat ()
        }
 
        film->repeat_content (_content, d->number ());
-       d->Destroy ();
 
        _content.clear ();
        _views.clear ();
@@ -332,46 +342,20 @@ 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 = make_wx<wxDirDialog>(nullptr, _("Choose a folder"), wxT (""), wxDD_DIR_MUST_EXIST);
                r = d->ShowModal ();
-               path = wx_to_std (d->GetPath ());
-               d->Destroy ();
+               path = wx_to_std (d->GetPath());
        } else {
-               auto d = new wxFileDialog (0, _("Choose a file"), wxT (""), wxT (""), wxT ("*.*"));
+               auto d = make_wx<wxFileDialog>(nullptr, _("Choose a file"), wxT (""), wxT (""), wxT ("*.*"));
                r = d->ShowModal ();
-               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);
-               }
+               path = wx_to_std (d->GetPath());
        }
 
-       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
@@ -383,30 +367,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 ()
@@ -415,50 +379,49 @@ ContentMenu::kdm ()
        auto dcp = dynamic_pointer_cast<DCPContent> (_content.front());
        DCPOMATIC_ASSERT (dcp);
 
-       auto d = new wxFileDialog (_parent, _("Select KDM"));
-
-       if (d->ShowModal() == wxID_OK) {
-               optional<dcp::EncryptedKDM> kdm;
-               try {
-                       kdm = dcp::EncryptedKDM (dcp::file_to_string(wx_to_std(d->GetPath()), MAX_KDM_SIZE));
-               } catch (exception& e) {
-                       error_dialog (_parent, _("Could not load KDM"), std_to_wx(e.what()));
-                       d->Destroy ();
-                       return;
-               }
+       auto d = make_wx<wxFileDialog>(_parent, _("Select KDM"));
 
-               /* Try to decrypt it to get an early preview of any errors */
-               try {
-                       decrypt_kdm_with_helpful_error (*kdm);
-               } catch (KDMError& e) {
-                       error_dialog (_parent, std_to_wx(e.summary()), std_to_wx(e.detail()));
-                       return;
-               } catch (exception& e) {
-                       error_dialog (_parent, e.what());
-                       return;
-               }
+       if (d->ShowModal() != wxID_OK) {
+               return;
+       }
 
-               auto cpls = dcp::find_and_resolve_cpls (dcp->directories(), true);
-               bool const kdm_matches_any_cpl = std::any_of(cpls.begin(), cpls.end(), [kdm](shared_ptr<const dcp::CPL> cpl) { return cpl->id() == kdm->cpl_id(); });
-               bool const kdm_matches_selected_cpl = dcp->cpl() || kdm->cpl_id() == dcp->cpl().get();
+       optional<dcp::EncryptedKDM> kdm;
+       try {
+               kdm = dcp::EncryptedKDM (dcp::file_to_string(wx_to_std(d->GetPath()), MAX_KDM_SIZE));
+       } catch (exception& e) {
+               error_dialog (_parent, _("Could not load KDM"), std_to_wx(e.what()));
+               return;
+       }
 
-               if (!kdm_matches_any_cpl) {
-                       error_dialog (_parent, _("This KDM was not made for this DCP.  You will need a different one."));
-                       return;
-               }
+       /* Try to decrypt it to get an early preview of any errors */
+       try {
+               decrypt_kdm_with_helpful_error (*kdm);
+       } catch (KDMError& e) {
+               error_dialog (_parent, std_to_wx(e.summary()), std_to_wx(e.detail()));
+               return;
+       } catch (exception& e) {
+               error_dialog (_parent, e.what());
+               return;
+       }
 
-               if (!kdm_matches_selected_cpl && kdm_matches_any_cpl) {
-                       message_dialog (_parent, _("This KDM was made for one of the CPLs in this DCP, but not the currently selected one.  To play the currently-selected CPL you will need a different KDM."));
-               }
+       auto cpls = dcp::find_and_resolve_cpls (dcp->directories(), true);
+       bool const kdm_matches_any_cpl = std::any_of(cpls.begin(), cpls.end(), [kdm](shared_ptr<const dcp::CPL> cpl) { return cpl->id() == kdm->cpl_id(); });
+       bool const kdm_matches_selected_cpl = dcp->cpl() || kdm->cpl_id() == dcp->cpl().get();
 
-               dcp->add_kdm (*kdm);
+       if (!kdm_matches_any_cpl) {
+               error_dialog (_parent, _("This KDM was not made for this DCP.  You will need a different one."));
+               return;
+       }
 
-               auto film = _film.lock ();
-               DCPOMATIC_ASSERT (film);
-               JobManager::instance()->add (make_shared<ExamineContentJob>(film, dcp));
+       if (!kdm_matches_selected_cpl && kdm_matches_any_cpl) {
+               message_dialog (_parent, _("This KDM was made for one of the CPLs in this DCP, but not the currently selected one.  To play the currently-selected CPL you will need a different KDM."));
        }
 
-       d->Destroy ();
+       dcp->add_kdm (*kdm);
+
+       auto film = _film.lock ();
+       DCPOMATIC_ASSERT (film);
+       JobManager::instance()->add (make_shared<ExamineContentJob>(film, dcp));
 }
 
 void
@@ -468,7 +431,7 @@ ContentMenu::ov ()
        auto dcp = dynamic_pointer_cast<DCPContent> (_content.front());
        DCPOMATIC_ASSERT (dcp);
 
-       auto d = new wxDirDialog (_parent, _("Select OV"));
+       auto d = make_wx<wxDirDialog>(_parent, _("Select OV"));
 
        if (d->ShowModal() == wxID_OK) {
                dcp->add_ov (wx_to_std (d->GetPath()));
@@ -476,8 +439,6 @@ ContentMenu::ov ()
                DCPOMATIC_ASSERT (film);
                JobManager::instance()->add (make_shared<ExamineContentJob>(film, dcp));
        }
-
-       d->Destroy ();
 }
 
 void
@@ -485,18 +446,40 @@ ContentMenu::properties ()
 {
        auto film = _film.lock ();
        DCPOMATIC_ASSERT (film);
-       auto d = new ContentPropertiesDialog (_parent, film, _content.front());
+       auto d = make_wx<ContentPropertiesDialog>(_parent, film, _content.front());
        d->ShowModal ();
-       d->Destroy ();
 }
 
 
 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();
+       }
 }
 
 
@@ -525,3 +508,76 @@ 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);
+
+       _auto_crop_dialog.reset(_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);
+       });
+}