1 /*
2  *  This file is part of RawTherapee.
3  *
4  *  Copyright (c) 2004-2010 Gabor Horvath <hgabor@rawtherapee.com>
5  *
6  *  RawTherapee 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 3 of the License, or
9  *  (at your option) any later version.
10  *
11  *  RawTherapee 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 RawTherapee.  If not, see <http://www.gnu.org/licenses/>.
18  */
19 #include "darkframe.h"
20 #include "options.h"
21 #include "guiutils.h"
22 #include <sstream>
23 #include "rtimage.h"
24 #include "../rtengine/refreshmap.h"
25 
26 using namespace rtengine;
27 using namespace rtengine::procparams;
28 
DarkFrame()29 DarkFrame::DarkFrame () :
30     FoldableToolPanel(this, "darkframe", M("TP_DARKFRAME_LABEL"), false, true, true),
31     dfChanged(false), lastDFauto(false), dfp(nullptr), israw(true)
32 {
33     EvToolEnabled.set_action(DARKFRAME);
34     EvToolReset.set_action(DARKFRAME);
35 
36     hbdf = Gtk::manage(new Gtk::HBox());
37     hbdf->set_spacing(4);
38     darkFrameFile = Gtk::manage(new MyFileChooserButton(M("TP_DARKFRAME_LABEL"), Gtk::FILE_CHOOSER_ACTION_OPEN));
39     bindCurrentFolder (*darkFrameFile, options.lastDarkframeDir);
40     dfLabel = Gtk::manage(new Gtk::Label(M("GENERAL_FILE")));
41     btnReset = Gtk::manage(new Gtk::Button());
42     btnReset->set_image (*Gtk::manage(new RTImage ("cancel-small.png")));
43     hbdf->pack_start(*dfLabel, Gtk::PACK_SHRINK, 0);
44     hbdf->pack_start(*darkFrameFile);
45     hbdf->pack_start(*btnReset, Gtk::PACK_SHRINK, 0);
46     dfAuto = Gtk::manage(new Gtk::CheckButton((M("TP_DARKFRAME_AUTOSELECT"))));
47     dfInfo = Gtk::manage(new Gtk::Label(""));
48     dfInfo->set_alignment(0, 0); //left align
49 
50     pack_start( *hbdf, Gtk::PACK_SHRINK, 0);
51     pack_start( *dfAuto, Gtk::PACK_SHRINK, 0);
52     pack_start( *dfInfo, Gtk::PACK_SHRINK, 0);
53 
54     dfautoconn = dfAuto->signal_toggled().connect ( sigc::mem_fun(*this, &DarkFrame::dfAutoChanged), true);
55     dfFile = darkFrameFile->signal_file_set().connect ( sigc::mem_fun(*this, &DarkFrame::darkFrameChanged)); //, true);
56     btnReset->signal_clicked().connect( sigc::mem_fun(*this, &DarkFrame::darkFrameReset), true );
57 
58     // Set filename filters
59     b_filter_asCurrent = false;
60     Glib::RefPtr<Gtk::FileFilter> filter_any = Gtk::FileFilter::create();
61     filter_any->add_pattern("*");
62     filter_any->set_name(M("FILECHOOSER_FILTER_ANY"));
63     darkFrameFile->add_filter (filter_any);
64 
65     // filters for all supported non-raw extensions
66     for (size_t i = 0; i < options.parseExtensions.size(); i++) {
67         if (options.parseExtensionsEnabled[i] && options.parseExtensions[i].uppercase() != "JPG" && options.parseExtensions[i].uppercase() != "JPEG" && options.parseExtensions[i].uppercase() != "PNG" && options.parseExtensions[i].uppercase() != "TIF" && options.parseExtensions[i].uppercase() != "TIFF"  ) {
68             Glib::RefPtr<Gtk::FileFilter> filter_df = Gtk::FileFilter::create();
69             filter_df->add_pattern("*." + options.parseExtensions[i]);
70             filter_df->add_pattern("*." + options.parseExtensions[i].uppercase());
71             filter_df->set_name(options.parseExtensions[i].uppercase());
72             darkFrameFile->add_filter (filter_df);
73             //printf("adding filter %s \n",options.parseExtensions[i].uppercase().c_str());
74         }
75     }
76 }
77 
read(const rtengine::procparams::ProcParams * pp)78 void DarkFrame::read(const rtengine::procparams::ProcParams* pp)
79 {
80     disableListener ();
81     dfautoconn.block(true);
82 
83     setEnabled(pp->raw.enable_darkframe);
84 
85     dfAuto->set_active( pp->raw.df_autoselect );
86 
87     if (Glib::file_test (pp->raw.dark_frame, Glib::FILE_TEST_EXISTS)) {
88         darkFrameFile->set_filename (pp->raw.dark_frame);
89     } else {
90         darkFrameReset();
91     }
92 
93     hbdf->set_sensitive( !pp->raw.df_autoselect );
94 
95     lastDFauto = pp->raw.df_autoselect;
96 
97     if( pp->raw.df_autoselect  && dfp) {
98         // retrieve the auto-selected df filename
99         rtengine::RawImage *img = dfp->getDF();
100 
101         if( img ) {
102             dfInfo->set_text( Glib::ustring::compose("%1: %2ISO %3s", Glib::path_get_basename(img->get_filename()), img->get_ISOspeed(), img->get_shutter()) );
103         } else {
104             dfInfo->set_text(Glib::ustring(M("TP_PREPROCESS_NO_FOUND")));
105         }
106     } else {
107         dfInfo->set_text("");
108     }
109 
110     dfChanged = false;
111 
112     dfautoconn.block(false);
113     enableListener ();
114 
115     // Add filter with the current file extension if the current file is raw
116     if (dfp) {
117 
118         if (b_filter_asCurrent) {
119             //First, remove last filter_asCurrent if it was set for a raw file
120             std::vector< Glib::RefPtr<Gtk::FileFilter> > filters = darkFrameFile->list_filters();
121             darkFrameFile->remove_filter(*(filters.end() - 1));
122             b_filter_asCurrent = false;
123         }
124 
125         Glib::ustring fname = Glib::path_get_basename(dfp->GetCurrentImageFilePath());
126         Glib::ustring filetype;
127 
128         if (fname != "") {
129             // get image filetype, set filter to the same as current image's filetype
130             std::string::size_type idx;
131             idx = fname.rfind('.');
132 
133             if(idx != std::string::npos) {
134                 filetype = fname.substr(idx + 1);
135                 israw = filetype.uppercase() != "JPG" && filetype.uppercase() != "JPEG" && filetype.uppercase() != "PNG" && filetype.uppercase() != "TIF" && filetype.uppercase() != "TIFF";
136 
137                 //exclude non-raw
138                 if (israw) {
139                     b_filter_asCurrent = true;
140                     Glib::RefPtr<Gtk::FileFilter> filter_asCurrent = Gtk::FileFilter::create();
141                     filter_asCurrent->add_pattern("*." + filetype);
142                     filter_asCurrent->set_name(M("FILECHOOSER_FILTER_SAME") + " (" + filetype + ")");
143                     darkFrameFile->add_filter (filter_asCurrent);
144                     darkFrameFile->set_filter (filter_asCurrent);
145                 }
146             }
147         }
148     }
149 
150 }
151 
write(rtengine::procparams::ProcParams * pp)152 void DarkFrame::write( rtengine::procparams::ProcParams* pp)
153 {
154     pp->raw.enable_darkframe = getEnabled();
155     pp->raw.dark_frame = darkFrameFile->get_filename();
156     if (!Glib::file_test(pp->raw.dark_frame, Glib::FILE_TEST_EXISTS)) {
157         pp->raw.dark_frame = "";
158     }
159     pp->raw.df_autoselect = dfAuto->get_active();
160 }
161 
dfAutoChanged()162 void DarkFrame::dfAutoChanged()
163 {
164     if(dfAuto->get_active() && dfp) {
165         // retrieve the auto-selected df filename
166         rtengine::RawImage *img = dfp->getDF();
167 
168         if( img ) {
169             dfInfo->set_text( Glib::ustring::compose("%1: %2ISO %3s", Glib::path_get_basename(img->get_filename()), img->get_ISOspeed(), img->get_shutter()) );
170         } else {
171             dfInfo->set_text(Glib::ustring(M("TP_PREPROCESS_NO_FOUND")));
172         }
173     } else {
174         dfInfo->set_text("");
175     }
176 
177     hbdf->set_sensitive( !dfAuto->get_active() );
178 
179     if (listener && getEnabled()) {
180         listener->panelChanged (EvPreProcessAutoDF, dfAuto->get_active() ? M("GENERAL_ENABLED") : M("GENERAL_DISABLED"));
181     }
182 }
183 
darkFrameChanged()184 void DarkFrame::darkFrameChanged()
185 {
186     dfChanged = true;
187 
188     if (listener && getEnabled()) {
189         listener->panelChanged (EvPreProcessDFFile, Glib::path_get_basename(darkFrameFile->get_filename()));
190     }
191 }
192 
darkFrameReset()193 void DarkFrame::darkFrameReset()
194 {
195     dfChanged = true;
196 
197 // caution: I had to make this hack, because set_current_folder() doesn't work correctly!
198 //          Because szeva doesn't exist since he was committed to happy hunting ground in Issue 316
199 //          we can use him now for this hack
200     darkFrameFile->set_filename (options.lastDarkframeDir + "/szeva");
201 // end of the hack
202 
203     if (!options.lastDarkframeDir.empty()) {
204         darkFrameFile->set_current_folder(options.lastDarkframeDir);
205     }
206 
207     dfInfo->set_text("");
208 
209     if (listener && getEnabled()) {
210         listener->panelChanged (EvPreProcessDFFile, M("GENERAL_NONE"));
211     }
212 
213 }
214 
215 
setDefaults(const ProcParams * def)216 void DarkFrame::setDefaults(const ProcParams *def)
217 {
218     initial_params = def->raw;
219 }
220 
221 
toolReset(bool to_initial)222 void DarkFrame::toolReset(bool to_initial)
223 {
224     ProcParams pp;
225     if (to_initial) {
226         pp.raw = initial_params;
227     }
228     pp.raw.enable_darkframe = getEnabled();
229     read(&pp);
230 }
231