4 * Copyright (C) 2008 Adam Williams <broadcast at earthling dot net>
6 * This program 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.
11 * This program 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.
16 * You should have received a copy of the GNU General Public License
17 * along with this program; if not, write to the Free Software
18 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
25 #include "bccmodels.h"
35 extern char dcraw_info[1024];
36 extern float **dcraw_data;
37 extern int dcraw_alpha;
38 extern float dcraw_matrix[9];
39 int dcraw_main (int argc, const char **argv);
42 static int dcraw_run(int argc, const char **argv)
44 static Mutex dcraw_lock;
45 dcraw_lock.lock("dcraw_run");
46 int result = dcraw_main(argc, argv);
51 FileCR2::FileCR2(Asset *asset, File *file)
52 : FileList(asset, file, "CR2LIST", ".cr2", FILE_CR2, FILE_CR2_LIST)
55 if(asset->format == FILE_UNKNOWN)
56 asset->format = FILE_CR2;
61 //printf("FileCR2::~FileCR2\n");
70 int FileCR2::check_sig(Asset *asset)
72 char *ptr = strstr(asset->path, ".pcm");
74 //printf("FileCR2::check_sig %d\n", __LINE__);
75 FILE *stream = fopen(asset->path, "rb");
80 (void)fread(test, 10, 1, stream);
83 if(test[0] == 'C' && test[1] == 'R' && test[2] == '2' &&
84 test[3] == 'L' && test[4] == 'I' && test[5] == 'S' && test[6] == 'T')
86 //printf("FileCR2::check_sig %d\n", __LINE__);
91 //printf("FileCR2::check_sig %d\n", __LINE__);
94 char string[BCTEXTLEN];
97 strcpy(string, asset->path);
107 int result = dcraw_run(argc, argv);
109 //printf("FileCR2::check_sig %d %d\n", __LINE__, result);
114 // int FileCR2::open_file(int rd, int wr)
118 // const char *argv[4] =
126 // int result = dcraw_run(argc, argv);
127 // if(!result) format_to_asset();
132 int FileCR2::read_frame_header(char *path)
135 printf("FileCR2::read_frame_header %d\n", __LINE__);
136 const char *argv[4] =
144 int result = dcraw_run(argc, argv);
145 if(!result) format_to_asset();
147 printf("FileCR2::read_frame_header %d %d\n", __LINE__, result);
154 // int FileCR2::close_file()
159 void FileCR2::format_to_asset()
161 asset->video_data = 1;
163 sscanf(dcraw_info, "%d %d", &asset->width, &asset->height);
167 int FileCR2::read_frame(VFrame *frame, char *path)
169 //printf("FileCR2::read_frame\n");
171 if(frame->get_color_model() == BC_RGBA_FLOAT)
176 // Want to disable interpolation if an interpolation plugin is on, but
177 // this is impractical because of the amount of caching. The interpolation
178 // could not respond to a change in the plugin settings and it could not
179 // reload the frame after the plugin was added. Also, since an 8 bit
180 // PBuffer would be required, it could never have enough resolution.
181 // int interpolate = 0;
182 // if(!strcmp(frame->get_next_effect(), "Interpolate Pixels"))
186 // printf("FileCR2::read_frame %d\n", interpolate);
187 // frame->dump_stacks();
190 char *argv[] = { 0, 0, 0, 0, 0, 0, 0, 0, 0 };
191 argv[argc++] = (char*)"dcraw";
193 argv[argc++] = (char*)"-c";
195 argv[argc++] = (char*)"-j";
197 // printf("FileCR2::read_frame %d interpolate=%d white_balance=%d\n",
199 // file->interpolate_raw,
200 // file->white_balance_raw);
202 // Use camera white balance.
203 // Before 2006, DCraw had no Canon white balance.
204 // In 2006 DCraw seems to support Canon white balance.
205 // Still no gamma support.
206 // Need to toggle this in preferences because it defeats dark frame subtraction.
207 if(file->white_balance_raw)
208 argv[argc++] = (char*)"-w";
211 if(!file->interpolate_raw)
213 // Trying to do everything but interpolate doesn't work because convert_to_rgb
214 // doesn't work with bayer patterns.
215 // Use document mode and hack dcraw to apply white balance in the write_ function.
216 argv[argc++] = (char*)"-d";
219 printf("FileCR2::read_frame %d %s\n", __LINE__, path);
222 dcraw_data = (float**)frame->get_rows();
225 int result = dcraw_run(argc, (const char**) argv);
227 // This was only used by the bayer interpolate plugin, which itself created
228 // too much complexity to use effectively.
229 // It required bypassing the cache any time a plugin parameter changed
230 // to store the color matrix from dcraw in the frame stack along with the new
231 // plugin parameters. The cache couldn't know if a parameter in the stack came
232 // from dcraw or a plugin & replace it.
233 char string[BCTEXTLEN];
235 "%f %f %f %f %f %f %f %f %f\n",
247 frame->get_params()->update("DCRAW_MATRIX", string);
249 // frame->dump_params();
254 int FileCR2::colormodel_supported(int colormodel)
256 if(colormodel == BC_RGB_FLOAT ||
257 colormodel == BC_RGBA_FLOAT)
263 // Be sure to add a line to File::get_best_colormodel
264 int FileCR2::get_best_colormodel(Asset *asset, int driver)
266 //printf("FileCR2::get_best_colormodel %d\n", __LINE__);
270 // int64_t FileCR2::get_memory_usage()
272 // int64_t result = asset->width * asset->height * sizeof(float) * 3;
273 // //printf("FileCR2::get_memory_usage %d " _LD "\n", __LINE__, result);
278 int FileCR2::use_path()