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);
43 FileCR2::FileCR2(Asset *asset, File *file)
44 : FileList(asset, file, "CR2LIST", ".cr2", FILE_CR2, FILE_CR2_LIST)
47 if(asset->format == FILE_UNKNOWN)
48 asset->format = FILE_CR2;
53 //printf("FileCR2::~FileCR2\n");
62 int FileCR2::check_sig(Asset *asset)
64 char *ptr = strstr(asset->path, ".pcm");
66 //printf("FileCR2::check_sig %d\n", __LINE__);
67 FILE *stream = fopen(asset->path, "rb");
72 (void)fread(test, 10, 1, stream);
75 if(test[0] == 'C' && test[1] == 'R' && test[2] == '2' &&
76 test[3] == 'L' && test[4] == 'I' && test[5] == 'S' && test[6] == 'T')
78 //printf("FileCR2::check_sig %d\n", __LINE__);
83 //printf("FileCR2::check_sig %d\n", __LINE__);
86 char string[BCTEXTLEN];
89 strcpy(string, asset->path);
99 int result = dcraw_main(argc, argv);
101 //printf("FileCR2::check_sig %d %d\n", __LINE__, result);
106 // int FileCR2::open_file(int rd, int wr)
110 // const char *argv[4] =
118 // int result = dcraw_main(argc, argv);
119 // if(!result) format_to_asset();
124 int FileCR2::read_frame_header(char *path)
127 printf("FileCR2::read_frame_header %d\n", __LINE__);
128 const char *argv[4] =
136 int result = dcraw_main(argc, argv);
137 if(!result) format_to_asset();
139 printf("FileCR2::read_frame_header %d %d\n", __LINE__, result);
146 // int FileCR2::close_file()
151 void FileCR2::format_to_asset()
153 asset->video_data = 1;
155 sscanf(dcraw_info, "%d %d", &asset->width, &asset->height);
159 int FileCR2::read_frame(VFrame *frame, char *path)
161 //printf("FileCR2::read_frame\n");
163 if(frame->get_color_model() == BC_RGBA_FLOAT)
168 // Want to disable interpolation if an interpolation plugin is on, but
169 // this is impractical because of the amount of caching. The interpolation
170 // could not respond to a change in the plugin settings and it could not
171 // reload the frame after the plugin was added. Also, since an 8 bit
172 // PBuffer would be required, it could never have enough resolution.
173 // int interpolate = 0;
174 // if(!strcmp(frame->get_next_effect(), "Interpolate Pixels"))
178 // printf("FileCR2::read_frame %d\n", interpolate);
179 // frame->dump_stacks();
182 char *argv[] = { 0, 0, 0, 0, 0, 0, 0, 0, 0 };
183 argv[argc++] = (char*)"dcraw";
185 argv[argc++] = (char*)"-c";
187 argv[argc++] = (char*)"-j";
189 // printf("FileCR2::read_frame %d interpolate=%d white_balance=%d\n",
191 // file->interpolate_raw,
192 // file->white_balance_raw);
194 // Use camera white balance.
195 // Before 2006, DCraw had no Canon white balance.
196 // In 2006 DCraw seems to support Canon white balance.
197 // Still no gamma support.
198 // Need to toggle this in preferences because it defeats dark frame subtraction.
199 if(file->white_balance_raw)
200 argv[argc++] = (char*)"-w";
203 if(!file->interpolate_raw)
205 // Trying to do everything but interpolate doesn't work because convert_to_rgb
206 // doesn't work with bayer patterns.
207 // Use document mode and hack dcraw to apply white balance in the write_ function.
208 argv[argc++] = (char*)"-d";
211 printf("FileCR2::read_frame %d %s\n", __LINE__, path);
214 dcraw_data = (float**)frame->get_rows();
217 int result = dcraw_main(argc, (const char**) argv);
219 // This was only used by the bayer interpolate plugin, which itself created
220 // too much complexity to use effectively.
221 // It required bypassing the cache any time a plugin parameter changed
222 // to store the color matrix from dcraw in the frame stack along with the new
223 // plugin parameters. The cache couldn't know if a parameter in the stack came
224 // from dcraw or a plugin & replace it.
225 char string[BCTEXTLEN];
227 "%f %f %f %f %f %f %f %f %f\n",
239 frame->get_params()->update("DCRAW_MATRIX", string);
241 // frame->dump_params();
246 int FileCR2::colormodel_supported(int colormodel)
248 if(colormodel == BC_RGB_FLOAT ||
249 colormodel == BC_RGBA_FLOAT)
255 // Be sure to add a line to File::get_best_colormodel
256 int FileCR2::get_best_colormodel(Asset *asset, int driver)
258 //printf("FileCR2::get_best_colormodel %d\n", __LINE__);
262 // int64_t FileCR2::get_memory_usage()
264 // int64_t result = asset->width * asset->height * sizeof(float) * 3;
265 // //printf("FileCR2::get_memory_usage %d " _LD "\n", __LINE__, result);
270 int FileCR2::use_path()