Use a vector rather than a list when returning from content_factory().
[dcpomatic.git] / src / wx / content_view.cc
1 /*
2     Copyright (C) 2018-2021 Carl Hetherington <cth@carlh.net>
3
4     This file is part of DCP-o-matic.
5
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.
10
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.
15
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/>.
18
19 */
20
21
22 #include "content_view.h"
23 #include "wx_util.h"
24 #include "lib/config.h"
25 #include "lib/content_factory.h"
26 #include "lib/cross.h"
27 #include "lib/dcp_content.h"
28 #include "lib/dcpomatic_assert.h"
29 #include "lib/examine_content_job.h"
30 #include "lib/job_manager.h"
31 #include <dcp/exceptions.h>
32 #include <dcp/warnings.h>
33 #include <boost/filesystem.hpp>
34 #include <boost/optional.hpp>
35 LIBDCP_DISABLE_WARNINGS
36 #include <wx/progdlg.h>
37 LIBDCP_ENABLE_WARNINGS
38
39
40 using std::cout;
41 using std::dynamic_pointer_cast;
42 using std::list;
43 using std::make_shared;
44 using std::shared_ptr;
45 using std::string;
46 using std::weak_ptr;
47 using boost::optional;
48 using namespace dcpomatic;
49
50
51 ContentView::ContentView (wxWindow* parent)
52         : wxListCtrl (parent, wxID_ANY, wxDefaultPosition, wxDefaultSize, wxLC_REPORT | wxLC_NO_HEADER)
53 {
54         AppendColumn (wxT(""), wxLIST_FORMAT_LEFT, 80);
55         /* type */
56         AppendColumn (wxT(""), wxLIST_FORMAT_LEFT, 80);
57         /* annotation text */
58         AppendColumn (wxT(""), wxLIST_FORMAT_LEFT, 580);
59 }
60
61
62 shared_ptr<Content>
63 ContentView::selected () const
64 {
65         long int s = GetNextItem (-1, wxLIST_NEXT_ALL, wxLIST_STATE_SELECTED);
66         if (s == -1) {
67                 return {};
68         }
69
70         DCPOMATIC_ASSERT (s < int(_content.size()));
71         return _content[s];
72 }
73
74
75 void
76 ContentView::update ()
77 {
78         using namespace boost::filesystem;
79
80         DeleteAllItems ();
81         _content.clear ();
82         auto dir = Config::instance()->player_content_directory();
83         if (!dir || !boost::filesystem::is_directory(*dir)) {
84                 dir = home_directory ();
85         }
86
87         wxProgressDialog progress (_("DCP-o-matic"), _("Reading content directory"));
88         auto jm = JobManager::instance ();
89
90         list<shared_ptr<ExamineContentJob>> jobs;
91
92         for (auto i: directory_iterator(*dir)) {
93                 try {
94                         progress.Pulse ();
95
96                         shared_ptr<Content> content;
97                         if (is_directory(i) && (is_regular_file(i / "ASSETMAP") || is_regular_file(i / "ASSETMAP.xml"))) {
98                                 content = make_shared<DCPContent>(i);
99                         } else if (i.path().extension() == ".mp4") {
100                                 auto all_content = content_factory(i);
101                                 if (!all_content.empty()) {
102                                         content = all_content[0];
103                                 }
104                         }
105
106                         if (content) {
107                                 auto job = make_shared<ExamineContentJob>(shared_ptr<Film>(), content);
108                                 jm->add (job);
109                                 jobs.push_back (job);
110                         }
111                 } catch (boost::filesystem::filesystem_error& e) {
112                         /* Never mind */
113                 } catch (dcp::ReadError& e) {
114                         /* Never mind */
115                 }
116         }
117
118         while (jm->work_to_do()) {
119                 if (!progress.Pulse()) {
120                         /* user pressed cancel */
121                         for (auto i: jm->get()) {
122                                 i->cancel();
123                         }
124                         return;
125                 }
126                 dcpomatic_sleep_seconds (1);
127         }
128
129         /* Add content from successful jobs and report errors */
130         for (auto i: jobs) {
131                 if (i->finished_in_error()) {
132                         error_dialog(this, std_to_wx(i->error_summary()) + ".\n", std_to_wx(i->error_details()));
133                 } else {
134                         add (i->content());
135                         _content.push_back (i->content());
136                 }
137         }
138 }
139
140
141 void
142 ContentView::add (shared_ptr<Content> content)
143 {
144         int const N = GetItemCount();
145
146         wxListItem it;
147         it.SetId(N);
148         it.SetColumn(0);
149         auto length = content->approximate_length ();
150         auto const hmsf = length.split (24);
151         it.SetText(wxString::Format("%02d:%02d:%02d", hmsf.h, hmsf.m, hmsf.s));
152         InsertItem(it);
153
154         auto dcp = dynamic_pointer_cast<DCPContent>(content);
155         if (dcp && dcp->content_kind()) {
156                 it.SetId(N);
157                 it.SetColumn(1);
158                 it.SetText(std_to_wx(dcp::content_kind_to_string(*dcp->content_kind())));
159                 SetItem(it);
160         }
161
162         it.SetId(N);
163         it.SetColumn(2);
164         it.SetText(std_to_wx(content->summary()));
165         SetItem(it);
166 }
167
168
169 shared_ptr<Content>
170 ContentView::get (string digest) const
171 {
172         for (auto i: _content) {
173                 if (i->digest() == digest) {
174                         return i;
175                 }
176         }
177
178         return {};
179 }