cinelerra-5.0/cinelerra/filepng.C

   1
   2 /*
   3  * CINELERRA
   4  * Copyright (C) 2008 Adam Williams <broadcast at earthling dot net>
   5  *
   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.
  10  *
  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.
  15  *
  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
  19  *
  20  */
  21
  22 #include "asset.h"
  23 #include "edit.h"
  24 #include "file.h"
  25 #include "filepng.h"
  26 #include "language.h"
  27 #include "mwindow.inc"
  28 #include "quicktime.h"
  29 #include "vframe.h"
  30 #include "videodevice.inc"
  31 #include "mainerror.h"
  32
  33 #include <png.h>
  34 #include <string.h>
  35
  36 FilePNG::FilePNG(Asset *asset, File *file)
  37  : FileList(asset, file, "PNGLIST", ".png", FILE_PNG, FILE_PNG_LIST)
  38 {
  39 }
  40
  41 FilePNG::~FilePNG()
  42 {
  43 }
  44
  45
  46
  47 int FilePNG::check_sig(Asset *asset)
  48 {
  49         FILE *stream = fopen(asset->path, "rb");
  50
  51         if(stream)
  52         {
  53
  54 //printf("FilePNG::check_sig 1\n");
  55                 char test[16];
  56                 (void)fread(test, 16, 1, stream);
  57                 fclose(stream);
  58
  59 //              if(png_sig_cmp((unsigned char*)test, 0, 8))
  60                 if(png_check_sig((unsigned char*)test, 8))
  61                 {
  62 //printf("FilePNG::check_sig 1\n");
  63                         return 1;
  64                 }
  65                 else
  66                 if(test[0] == 'P' && test[1] == 'N' && test[2] == 'G' &&
  67                         test[3] == 'L' && test[4] == 'I' && test[5] == 'S' && test[6] == 'T')
  68                 {
  69 //printf("FilePNG::check_sig 1\n");
  70                         return 1;
  71                 }
  72         }
  73         return 0;
  74 }
  75
  76
  77
  78 void FilePNG::get_parameters(BC_WindowBase *parent_window,
  79         Asset *asset,
  80         BC_WindowBase* &format_window,
  81         int audio_options,
  82         int video_options)
  83 {
  84         if(video_options)
  85         {
  86                 PNGConfigVideo *window = new PNGConfigVideo(parent_window, asset);
  87                 format_window = window;
  88                 window->create_objects();
  89                 window->run_window();
  90                 delete window;
  91         }
  92 }
  93
  94
  95
  96
  97 int FilePNG::can_copy_from(Asset *asset, int64_t position)
  98 {
  99         if(asset->format == FILE_MOV)
 100         {
 101                 if(match4(asset->vcodec, QUICKTIME_PNG)) return 1;
 102         }
 103         else
 104         if(asset->format == FILE_PNG ||
 105                 asset->format == FILE_PNG_LIST)
 106                 return 1;
 107
 108         return 0;
 109 }
 110
 111
 112 int FilePNG::colormodel_supported(int colormodel)
 113 {
 114         if (((colormodel == BC_RGBA8888) && (native_cmodel == BC_RGBA16161616))
 115             || ((colormodel == BC_RGB161616) && (native_cmodel == BC_RGBA16161616))
 116             || (colormodel == BC_RGB888))
 117         {
 118             return colormodel;
 119         }
 120         else if ((colormodel == BC_RGB161616) && (native_cmodel == BC_RGBA8888))
 121         {
 122             return BC_RGB888;
 123         }
 124         else
 125         {
 126             return native_cmodel;
 127         }
 128 }
 129
 130
 131 int FilePNG::get_best_colormodel(Asset *asset, int driver)
 132 {
 133         if(asset->png_use_alpha)
 134                 return BC_RGBA8888;
 135         else
 136                 return BC_RGB888;
 137 }
 138
 139
 140
 141
 142 int FilePNG::read_frame_header(char *path)
 143 {
 144         int result = 0;
 145         int color_type;
 146         int color_depth;
 147         int num_trans = 0;
 148
 149         FILE *stream;
 150
 151         if(!(stream = fopen(path, "rb")))
 152         {
 153                 eprintf("Error while opening \"%s\" for reading. \n%m\n", asset->path);
 154                 return 1;
 155         }
 156
 157         png_structp png_ptr;
 158         png_infop info_ptr;
 159         png_infop end_info = 0;
 160         png_ptr = png_create_read_struct(PNG_LIBPNG_VER_STRING, 0, 0, 0);
 161         info_ptr = png_create_info_struct(png_ptr);
 162         png_init_io(png_ptr, stream);
 163
 164
 165         png_read_info(png_ptr, info_ptr);
 166
 167         asset->width = png_get_image_width(png_ptr, info_ptr);
 168         asset->height = png_get_image_height(png_ptr, info_ptr);
 169         color_type = png_get_color_type(png_ptr, info_ptr);
 170         color_depth = png_get_bit_depth(png_ptr,info_ptr);
 171
 172         png_get_tRNS(png_ptr, info_ptr, NULL, &num_trans, NULL);
 173
 174         if (color_depth == 16)
 175         {
 176             if (color_type & PNG_COLOR_MASK_ALPHA)
 177             {
 178                 native_cmodel = BC_RGBA16161616;
 179             }
 180             else
 181             {
 182                 native_cmodel = BC_RGB161616;
 183             }
 184         }
 185         else
 186         if ((color_type & PNG_COLOR_MASK_ALPHA)
 187             || (num_trans > 0))
 188         {
 189             native_cmodel = BC_RGBA8888;
 190         }
 191         else
 192         {
 193             native_cmodel = BC_RGB888;
 194         }
 195
 196
 197         png_destroy_read_struct(&png_ptr, &info_ptr, &end_info);
 198         fclose(stream);
 199
 200
 201
 202         return result;
 203 }
 204
 205
 206
 207
 208 static void read_function(png_structp png_ptr,
 209         png_bytep data,
 210         png_uint_32 length)
 211 {
 212         VFrame *input = (VFrame*)png_get_io_ptr(png_ptr);
 213
 214         memcpy(data, input->get_data() + input->get_compressed_size(), length);
 215         input->set_compressed_size(input->get_compressed_size() + length);
 216 }
 217
 218 static void write_function(png_structp png_ptr, png_bytep data, png_uint_32 length)
 219 {
 220         VFrame *output = (VFrame*)png_get_io_ptr(png_ptr);
 221         long total_length = output->get_compressed_size() + length;
 222         if(output->get_compressed_allocated() < total_length) {
 223                 long new_length = 2 * (output->get_compressed_allocated() + length);
 224                 output->allocate_compressed_data(new_length);
 225         }
 226         memcpy(output->get_data() + output->get_compressed_size(), data, length);
 227         output->set_compressed_size(total_length);
 228 }
 229
 230 static void flush_function(png_structp png_ptr)
 231 {
 232 }
 233
 234
 235 int FilePNG::write_frame(VFrame *frame, VFrame *data, FrameWriterUnit *unit)
 236 {
 237         PNGUnit *png_unit = (PNGUnit*)unit;
 238         png_structp png_ptr = 0;
 239         png_infop info_ptr = 0;
 240         VFrame *output_frame;
 241         int result = 1;
 242         data->set_compressed_size(0);
 243 //printf("FilePNG::write_frame 1\n");
 244         native_cmodel = asset->png_use_alpha ? BC_RGBA8888 : BC_RGB888;
 245         if(frame->get_color_model() != native_cmodel)
 246         {
 247                 if(!png_unit->temp_frame) png_unit->temp_frame = new VFrame(0,
 248                         -1, asset->width, asset->height, native_cmodel, -1);
 249
 250                 png_unit->temp_frame->transfer_from(frame);
 251                 output_frame = png_unit->temp_frame;
 252         }
 253         else
 254                 output_frame = frame;
 255
 256 //printf("FilePNG::write_frame 1\n");
 257         png_ptr = png_create_write_struct(PNG_LIBPNG_VER_STRING, 0, 0, 0);
 258         if( png_ptr && !setjmp(png_jmpbuf(png_ptr)) ) {
 259                 info_ptr = png_create_info_struct(png_ptr);
 260                 png_set_write_fn(png_ptr, data,
 261                         (png_rw_ptr)write_function, (png_flush_ptr)flush_function);
 262                 png_set_compression_level(png_ptr, 5);
 263
 264                 png_set_IHDR(png_ptr, info_ptr, asset->width, asset->height, 8,
 265                         asset->png_use_alpha ?  PNG_COLOR_TYPE_RGB_ALPHA : PNG_COLOR_TYPE_RGB,
 266                         PNG_INTERLACE_NONE, PNG_COMPRESSION_TYPE_DEFAULT, PNG_FILTER_TYPE_DEFAULT);
 267                 png_write_info(png_ptr, info_ptr);
 268                 png_write_image(png_ptr, output_frame->get_rows());
 269                 png_write_end(png_ptr, info_ptr);
 270                 png_destroy_write_struct(&png_ptr, &info_ptr);
 271                 result = 0;
 272         }
 273         else {
 274                 char error[256];  png_error(png_ptr, error);
 275                 fprintf(stderr, "FilePNG::write_frame failed: %s\n", error);
 276         }
 277
 278         png_destroy_write_struct(&png_ptr, &info_ptr);
 279 //printf("FilePNG::write_frame 3 %d\n", data->get_compressed_size());
 280         return result;
 281 }
 282
 283 int FilePNG::read_frame(VFrame *output, VFrame *input)
 284 {
 285         png_structp png_ptr;
 286         png_infop info_ptr;
 287         png_infop end_info = 0;
 288         int result = 0;
 289         int color_type;
 290         int color_depth;
 291         int colormodel;
 292         int size = input->get_compressed_size();
 293         input->set_compressed_size(0);
 294
 295
 296         //printf("FilePNG::read_frame 1 %d %d\n", native_cmodel, output->get_color_model());
 297         png_ptr = png_create_read_struct(PNG_LIBPNG_VER_STRING, 0, 0, 0);
 298         info_ptr = png_create_info_struct(png_ptr);
 299         png_set_read_fn(png_ptr, input, (png_rw_ptr)read_function);
 300         png_read_info(png_ptr, info_ptr);
 301
 302         int png_color_type = png_get_color_type(png_ptr, info_ptr);
 303         if (png_color_type == PNG_COLOR_TYPE_GRAY ||
 304                 png_color_type == PNG_COLOR_TYPE_GRAY_ALPHA)
 305         {
 306                 png_set_gray_to_rgb(png_ptr);
 307         }
 308
 309         colormodel = output->get_color_model();
 310         color_type = png_get_color_type(png_ptr, info_ptr);
 311         color_depth = png_get_bit_depth(png_ptr,info_ptr);
 312
 313         if (((native_cmodel == BC_RGBA16161616)||(native_cmodel == BC_RGB161616))
 314             && ((colormodel == BC_RGBA8888)||(colormodel == BC_RGB888)))
 315         {
 316             png_set_strip_16(png_ptr);
 317         }
 318
 319         /* If we're dropping the alpha channel, use the background color of the image
 320            otherwise, use black */
 321         if (((native_cmodel == BC_RGBA16161616)||(native_cmodel == BC_RGBA8888))
 322             && ((colormodel == BC_RGB161616)||(colormodel == BC_RGB888)))
 323         {
 324             png_color_16 my_background;
 325             png_color_16p image_background;
 326
 327             memset(&my_background,0,sizeof(png_color_16));
 328
 329             if (png_get_bKGD(png_ptr, info_ptr, &image_background))
 330             {
 331                 png_set_background(png_ptr, image_background, PNG_BACKGROUND_GAMMA_FILE, 1, 1.0);
 332             }
 333             else
 334             {
 335                 png_set_background(png_ptr, &my_background, PNG_BACKGROUND_GAMMA_SCREEN, 0, 1.0);
 336             }
 337         }
 338
 339         /* Little endian */
 340         if ((color_depth == 16)
 341             &&((colormodel == BC_RGBA16161616)||(colormodel == BC_RGB161616)))
 342         {
 343             png_set_swap(png_ptr);
 344         }
 345
 346         if (!(color_type & PNG_COLOR_MASK_COLOR))
 347         {
 348             png_set_gray_to_rgb(png_ptr);
 349         }
 350
 351         if (color_type & PNG_COLOR_MASK_PALETTE)
 352         {
 353             png_set_palette_to_rgb(png_ptr);
 354         }
 355
 356         if (color_depth <= 8)
 357         {
 358             png_set_expand(png_ptr);
 359         }
 360
 361 /* read the image */
 362         png_read_image(png_ptr, output->get_rows());
 363 //printf("FilePNG::read_frame 3\n");
 364         png_destroy_read_struct(&png_ptr, &info_ptr, &end_info);
 365
 366         input->set_compressed_size(size);
 367
 368 //printf("FilePNG::read_frame 4\n");
 369         return result;
 370 }
 371
 372 FrameWriterUnit* FilePNG::new_writer_unit(FrameWriter *writer)
 373 {
 374         return new PNGUnit(this, writer);
 375 }
 376
 377
 378
 379
 380
 381
 382
 383
 384
 385
 386
 387
 388 PNGUnit::PNGUnit(FilePNG *file, FrameWriter *writer)
 389  : FrameWriterUnit(writer)
 390 {
 391         this->file = file;
 392         temp_frame = 0;
 393 }
 394 PNGUnit::~PNGUnit()
 395 {
 396         if(temp_frame) delete temp_frame;
 397 }
 398
 399
 400
 401
 402
 403
 404
 405
 406
 407 PNGConfigVideo::PNGConfigVideo(BC_WindowBase *parent_window, Asset *asset)
 408  : BC_Window(PROGRAM_NAME ": Video Compression",
 409         parent_window->get_abs_cursor_x(1),
 410         parent_window->get_abs_cursor_y(1),
 411         200,
 412         100)
 413 {
 414         this->parent_window = parent_window;
 415         this->asset = asset;
 416 }
 417
 418 PNGConfigVideo::~PNGConfigVideo()
 419 {
 420 }
 421
 422 void PNGConfigVideo::create_objects()
 423 {
 424         lock_window("PNGConfigVideo::create_objects()");
 425         int x = 10, y = 10;
 426         add_subwindow(new PNGUseAlpha(this, x, y));
 427         add_subwindow(new BC_OKButton(this));
 428         show_window(1);
 429         unlock_window();
 430 }
 431
 432 int PNGConfigVideo::close_event()
 433 {
 434         set_done(0);
 435         return 1;
 436 }
 437
 438
 439 PNGUseAlpha::PNGUseAlpha(PNGConfigVideo *gui, int x, int y)
 440  : BC_CheckBox(x, y, gui->asset->png_use_alpha, _("Use alpha"))
 441 {
 442         this->gui = gui;
 443 }
 444
 445 int PNGUseAlpha::handle_event()
 446 {
 447         gui->asset->png_use_alpha = get_value();
 448         return 1;
 449 }
 450
 451
 452