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
23 #include "bcsignals.h"
27 #include "jpegwrapper.h"
30 #include "mwindow.inc"
31 #include "quicktime.h"
33 #include "videodevice.inc"
34 #include "mainerror.h"
40 FileJPEG::FileJPEG(Asset *asset, File *file)
41 : FileList(asset, file, "JPEGLIST", ".jpg", FILE_JPEG, FILE_JPEG_LIST)
48 if(decompressor) mjpeg_delete((mjpeg_t*)decompressor);
52 int FileJPEG::check_sig(Asset *asset)
54 FILE *stream = fopen(asset->path, "rb");
59 (void)fread(test, 10, 1, stream);
62 if(test[6] == 'J' && test[7] == 'F' && test[8] == 'I' && test[9] == 'F')
67 if(test[0] == 'J' && test[1] == 'P' && test[2] == 'E' && test[3] == 'G' &&
68 test[4] == 'L' && test[5] == 'I' && test[6] == 'S' && test[7] == 'T')
74 if(strlen(asset->path) > 4)
76 int len = strlen(asset->path);
77 if(!strncasecmp(asset->path + len - 4, ".jpg", 4)) return 1;
84 void FileJPEG::get_parameters(BC_WindowBase *parent_window,
86 BC_WindowBase* &format_window,
92 JPEGConfigVideo *window = new JPEGConfigVideo(parent_window, asset);
93 format_window = window;
94 window->create_objects();
101 int FileJPEG::can_copy_from(Asset *asset, int64_t position)
103 //printf("FileJPEG::can_copy_from %d %s\n", asset->format, asset->vcodec);
104 if(asset->format == FILE_MOV)
106 if(match4(asset->vcodec, QUICKTIME_JPEG)) return 1;
109 if(asset->format == FILE_JPEG ||
110 asset->format == FILE_JPEG_LIST)
116 int FileJPEG::colormodel_supported(int colormodel)
122 int FileJPEG::get_best_colormodel(Asset *asset, int driver)
129 case PLAYBACK_X11_XV:
130 case PLAYBACK_DV1394:
131 case PLAYBACK_FIREWIRE:
132 case PLAYBACK_ASYNCHRONOUS:
135 case PLAYBACK_X11_GL:
148 case VIDEO4LINUX2JPEG:
151 case CAPTURE_FIREWIRE:
152 case CAPTURE_IEC61883:
156 case VIDEO4LINUX2MPEG:
164 int FileJPEG::write_frame(VFrame *frame, VFrame *data, FrameWriterUnit *unit)
167 JPEGUnit *jpeg_unit = (JPEGUnit*)unit;
169 if(!jpeg_unit->compressor)
170 jpeg_unit->compressor = mjpeg_new(asset->width,
174 mjpeg_set_quality((mjpeg_t*)jpeg_unit->compressor, asset->jpeg_quality);
177 mjpeg_compress((mjpeg_t*)jpeg_unit->compressor,
182 frame->get_color_model(),
185 data->allocate_compressed_data(mjpeg_output_size((mjpeg_t*)jpeg_unit->compressor));
186 data->set_compressed_size(mjpeg_output_size((mjpeg_t*)jpeg_unit->compressor));
187 memcpy(data->get_data(),
188 mjpeg_output_buffer((mjpeg_t*)jpeg_unit->compressor),
189 mjpeg_output_size((mjpeg_t*)jpeg_unit->compressor));
203 int FileJPEG::read_frame_header(char *path)
210 if(!(stream = fopen(path, "rb")))
212 eprintf("FileJPEG::read_frame_header %s: %m\n", path);
217 unsigned char test[2];
218 (void)fread(test, 2, 1, stream);
219 if(test[0] != 0xff || test[1] != 0xd8)
221 eprintf("FileJPEG::read_frame_header %s bad header %02x%02x\n",
222 path, test[0], test[1]);
226 fseek(stream, 0, SEEK_SET);
228 struct jpeg_decompress_struct jpeg_decompress;
229 struct jpeg_error_mgr jpeg_error;
231 jpeg_decompress.err = jpeg_std_error(&jpeg_error);
232 jpeg_create_decompress(&jpeg_decompress);
234 jpeg_stdio_src(&jpeg_decompress, stream);
235 jpeg_read_header(&jpeg_decompress, TRUE);
237 asset->width = jpeg_decompress.image_width;
238 asset->height = jpeg_decompress.image_height;
240 jpeg_destroy((j_common_ptr)&jpeg_decompress);
248 int FileJPEG::read_frame(VFrame *output, VFrame *input)
250 if(input->get_compressed_size() < 2 ||
251 input->get_data()[0] != 0xff ||
252 input->get_data()[1] != 0xd8)
255 if(!decompressor) decompressor = mjpeg_new(asset->width,
258 // printf("FileJPEG::read_frame %d %p %d %d %d %p %p %p %p %d\n",
260 // input->get_data(),
261 // input->get_compressed_size(),
264 // output->get_rows(),
268 // output->get_color_model());
269 mjpeg_decompress((mjpeg_t*)decompressor,
271 input->get_compressed_size(),
277 output->get_color_model(),
281 //printf("FileJPEG::read_frame %d\n", __LINE__);
285 FrameWriterUnit* FileJPEG::new_writer_unit(FrameWriter *writer)
287 return new JPEGUnit(this, writer);
295 JPEGUnit::JPEGUnit(FileJPEG *file, FrameWriter *writer)
296 : FrameWriterUnit(writer)
301 JPEGUnit::~JPEGUnit()
303 if(compressor) mjpeg_delete((mjpeg_t*)compressor);
312 JPEGConfigVideo::JPEGConfigVideo(BC_WindowBase *parent_window, Asset *asset)
313 : BC_Window(PROGRAM_NAME ": Video Compression",
314 parent_window->get_abs_cursor_x(1),
315 parent_window->get_abs_cursor_y(1),
319 this->parent_window = parent_window;
323 JPEGConfigVideo::~JPEGConfigVideo()
327 void JPEGConfigVideo::create_objects()
330 lock_window("JPEGConfigVideo::create_objects");
331 add_subwindow(new BC_Title(x, y, _("Quality:")));
332 add_subwindow(new BC_ISlider(x + 80,
342 &asset->jpeg_quality));
344 add_subwindow(new BC_OKButton(this));
349 int JPEGConfigVideo::close_event()