29b63b5c8217eb268175692a5f1d52dc11e800af
[dcpomatic.git] / src / lib / config.cc
1 /*
2     Copyright (C) 2012 Carl Hetherington <cth@carlh.net>
3
4     This program is free software; you can redistribute it and/or modify
5     it under the terms of the GNU General Public License as published by
6     the Free Software Foundation; either version 2 of the License, or
7     (at your option) any later version.
8
9     This program is distributed in the hope that it will be useful,
10     but WITHOUT ANY WARRANTY; without even the implied warranty of
11     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12     GNU General Public License for more details.
13
14     You should have received a copy of the GNU General Public License
15     along with this program; if not, write to the Free Software
16     Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
17
18 */
19
20 #include <sstream>
21 #include <cstdlib>
22 #include <fstream>
23 #include <glib.h>
24 #include <boost/filesystem.hpp>
25 #include <libcxml/cxml.h>
26 #include "config.h"
27 #include "server.h"
28 #include "scaler.h"
29 #include "filter.h"
30 #include "ratio.h"
31 #include "dcp_content_type.h"
32 #include "sound_processor.h"
33
34 #include "i18n.h"
35
36 using std::vector;
37 using std::ifstream;
38 using std::string;
39 using std::ofstream;
40 using std::list;
41 using std::max;
42 using boost::shared_ptr;
43 using boost::lexical_cast;
44 using boost::optional;
45
46 Config* Config::_instance = 0;
47
48 /** Construct default configuration */
49 Config::Config ()
50         : _num_local_encoding_threads (max (2U, boost::thread::hardware_concurrency()))
51         , _server_port (6192)
52         , _tms_path (N_("."))
53         , _sound_processor (SoundProcessor::from_id (N_("dolby_cp750")))
54         , _default_still_length (10)
55         , _default_container (Ratio::from_id ("185"))
56         , _default_dcp_content_type (DCPContentType::from_dci_name ("TST"))
57         , _default_j2k_bandwidth (200000000)
58 {
59         _allowed_dcp_frame_rates.push_back (24);
60         _allowed_dcp_frame_rates.push_back (25);
61         _allowed_dcp_frame_rates.push_back (30);
62         _allowed_dcp_frame_rates.push_back (48);
63         _allowed_dcp_frame_rates.push_back (50);
64         _allowed_dcp_frame_rates.push_back (60);
65 }
66
67 void
68 Config::read ()
69 {
70         if (!boost::filesystem::exists (file (false))) {
71                 read_old_metadata ();
72                 return;
73         }
74
75         cxml::File f (file (false), "Config");
76         optional<string> c;
77
78         _num_local_encoding_threads = f.number_child<int> ("NumLocalEncodingThreads");
79         _default_directory = f.string_child ("DefaultDirectory");
80         _server_port = f.number_child<int> ("ServerPort");
81         
82         list<shared_ptr<cxml::Node> > servers = f.node_children ("Server");
83         for (list<shared_ptr<cxml::Node> >::iterator i = servers.begin(); i != servers.end(); ++i) {
84                 _servers.push_back (new ServerDescription (*i));
85         }
86
87         _tms_ip = f.string_child ("TMSIP");
88         _tms_path = f.string_child ("TMSPath");
89         _tms_user = f.string_child ("TMSUser");
90         _tms_password = f.string_child ("TMSPassword");
91
92         c = f.optional_string_child ("SoundProcessor");
93         if (c) {
94                 _sound_processor = SoundProcessor::from_id (c.get ());
95         }
96
97         _language = f.optional_string_child ("Language");
98
99         c = f.optional_string_child ("DefaultContainer");
100         if (c) {
101                 _default_container = Ratio::from_id (c.get ());
102         }
103
104         c = f.optional_string_child ("DefaultDCPContentType");
105         if (c) {
106                 _default_dcp_content_type = DCPContentType::from_dci_name (c.get ());
107         }
108
109         _dcp_metadata.issuer = f.optional_string_child ("DCPMetadataIssuer").get_value_or ("");
110         _dcp_metadata.creator = f.optional_string_child ("DCPMetadataCreator").get_value_or ("");
111
112         _default_dci_metadata = DCIMetadata (f.node_child ("DCIMetadata"));
113         _default_still_length = f.optional_number_child<int>("DefaultStillLength").get_value_or (10);
114         _default_j2k_bandwidth = f.optional_number_child<int>("DefaultJ2KBandwidth").get_value_or (200000000); 
115 }
116
117 void
118 Config::read_old_metadata ()
119 {
120         ifstream f (file(true).c_str ());
121         string line;
122         while (getline (f, line)) {
123                 if (line.empty ()) {
124                         continue;
125                 }
126
127                 if (line[0] == '#') {
128                         continue;
129                 }
130
131                 size_t const s = line.find (' ');
132                 if (s == string::npos) {
133                         continue;
134                 }
135                 
136                 string const k = line.substr (0, s);
137                 string const v = line.substr (s + 1);
138
139                 if (k == N_("num_local_encoding_threads")) {
140                         _num_local_encoding_threads = atoi (v.c_str ());
141                 } else if (k == N_("default_directory")) {
142                         _default_directory = v;
143                 } else if (k == N_("server_port")) {
144                         _server_port = atoi (v.c_str ());
145                 } else if (k == N_("server")) {
146                         _servers.push_back (ServerDescription::create_from_metadata (v));
147                 } else if (k == N_("tms_ip")) {
148                         _tms_ip = v;
149                 } else if (k == N_("tms_path")) {
150                         _tms_path = v;
151                 } else if (k == N_("tms_user")) {
152                         _tms_user = v;
153                 } else if (k == N_("tms_password")) {
154                         _tms_password = v;
155                 } else if (k == N_("sound_processor")) {
156                         _sound_processor = SoundProcessor::from_id (v);
157                 } else if (k == "language") {
158                         _language = v;
159                 } else if (k == "default_container") {
160                         _default_container = Ratio::from_id (v);
161                 } else if (k == "default_dcp_content_type") {
162                         _default_dcp_content_type = DCPContentType::from_dci_name (v);
163                 } else if (k == "dcp_metadata_issuer") {
164                         _dcp_metadata.issuer = v;
165                 } else if (k == "dcp_metadata_creator") {
166                         _dcp_metadata.creator = v;
167                 } else if (k == "dcp_metadata_issue_date") {
168                         _dcp_metadata.issue_date = v;
169                 }
170
171                 _default_dci_metadata.read_old_metadata (k, v);
172         }
173 }
174
175 /** @return Filename to write configuration to */
176 string
177 Config::file (bool old) const
178 {
179         boost::filesystem::path p;
180         p /= g_get_user_config_dir ();
181         boost::system::error_code ec;
182         boost::filesystem::create_directory (p, ec);
183         if (old) {
184                 p /= ".dvdomatic";
185         } else {
186                 p /= ".dcpomatic.xml";
187         }
188         return p.string ();
189 }
190
191 /** @return Singleton instance */
192 Config *
193 Config::instance ()
194 {
195         if (_instance == 0) {
196                 _instance = new Config;
197                 try {
198                         _instance->read ();
199                 } catch (...) {
200                         /* configuration load failed; never mind, just
201                            stick with the default.
202                         */
203                 }
204         }
205
206         return _instance;
207 }
208
209 /** Write our configuration to disk */
210 void
211 Config::write () const
212 {
213         xmlpp::Document doc;
214         xmlpp::Element* root = doc.create_root_node ("Config");
215
216         root->add_child("NumLocalEncodingThreads")->add_child_text (lexical_cast<string> (_num_local_encoding_threads));
217         root->add_child("DefaultDirectory")->add_child_text (_default_directory);
218         root->add_child("ServerPort")->add_child_text (lexical_cast<string> (_server_port));
219         
220         for (vector<ServerDescription*>::const_iterator i = _servers.begin(); i != _servers.end(); ++i) {
221                 (*i)->as_xml (root->add_child ("Server"));
222         }
223
224         root->add_child("TMSIP")->add_child_text (_tms_ip);
225         root->add_child("TMSPath")->add_child_text (_tms_path);
226         root->add_child("TMSUser")->add_child_text (_tms_user);
227         root->add_child("TMSPassword")->add_child_text (_tms_password);
228         if (_sound_processor) {
229                 root->add_child("SoundProcessor")->add_child_text (_sound_processor->id ());
230         }
231         if (_language) {
232                 root->add_child("Language")->add_child_text (_language.get());
233         }
234         if (_default_container) {
235                 root->add_child("DefaultContainer")->add_child_text (_default_container->id ());
236         }
237         if (_default_dcp_content_type) {
238                 root->add_child("DefaultDCPContentType")->add_child_text (_default_dcp_content_type->dci_name ());
239         }
240         root->add_child("DCPMetadataIssuer")->add_child_text (_dcp_metadata.issuer);
241         root->add_child("DCPMetadataCreator")->add_child_text (_dcp_metadata.creator);
242
243         _default_dci_metadata.as_xml (root->add_child ("DCIMetadata"));
244
245         root->add_child("DefaultStillLength")->add_child_text (lexical_cast<string> (_default_still_length));
246         root->add_child("DefaultJ2KBandwidth")->add_child_text (lexical_cast<string> (_default_j2k_bandwidth));
247
248         doc.write_to_file_formatted (file (false));
249 }
250
251 string
252 Config::default_directory_or (string a) const
253 {
254         if (_default_directory.empty() || !boost::filesystem::exists (_default_directory)) {
255                 return a;
256         }
257
258         return _default_directory;
259 }
260
261 void
262 Config::drop ()
263 {
264         delete _instance;
265         _instance = 0;
266 }