4 * Copyright (C) 2011 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
33 #include "bcpbuffer.h"
34 #include "bcresources.h"
35 #include "bcsignals.h"
36 #include "bcsynchronous.h"
37 #include "bctexture.h"
38 #include "bcwindowbase.h"
40 #include "bccmodels.h"
46 static void png_read_function(png_structp png_ptr,
47 png_bytep data, png_size_t length)
49 VFrame *frame = (VFrame*)png_get_io_ptr(png_ptr);
50 if(frame->image_size - frame->image_offset < (long)length)
52 printf("PngReadFunction::png_read_function %d: overrun\n", __LINE__);
53 length = frame->image_size - frame->image_offset;
56 memcpy(data, &frame->image[frame->image_offset], length);
57 frame->image_offset += length;
67 VFrameScene::VFrameScene()
71 VFrameScene::~VFrameScene()
81 //static BCCounter counter;
83 VFramePng::VFramePng(unsigned char *png_data, double s)
86 ((long)png_data[0] << 24) | ((long)png_data[1] << 16) |
87 ((long)png_data[2] << 8) | (long)png_data[3];
88 if( !s ) s = BC_WindowBase::get_resources()->icon_scale;
89 read_png(png_data+4, image_size, s, s);
92 VFramePng::VFramePng(unsigned char *png_data, long image_size, double xs, double ys)
94 if( !xs ) xs = BC_WindowBase::get_resources()->icon_scale;
95 if( !ys ) ys = BC_WindowBase::get_resources()->icon_scale;
96 read_png(png_data, image_size, xs, ys);
99 VFramePng::~VFramePng()
103 VFrame *VFramePng::vframe_png(int fd, double xs, double ys)
106 if( fstat(fd, &st) ) return 0;
107 long len = st.st_size;
110 unsigned char *bfr = (unsigned char *)
111 ::mmap (NULL, len, PROT_READ, MAP_SHARED, fd, 0);
112 if( bfr == MAP_FAILED ) return 0;
113 VFrame *vframe = new VFramePng(bfr, len, xs, ys);
114 if( (w=vframe->get_w()) <= 0 || (h=vframe->get_h()) <= 0 ||
115 vframe->get_data() == 0 ) { delete vframe; vframe = 0; }
119 VFrame *VFramePng::vframe_png(const char *png_path, double xs, double ys)
122 int fd = ::open(png_path, O_RDONLY);
124 vframe = vframe_png(fd, xs, ys);
130 VFrame::VFrame(VFrame &frame)
133 params = new BC_Hash;
134 use_shm = frame.use_shm;
135 allocate_data(0, -1, 0, 0, 0, frame.w, frame.h,
136 frame.color_model, frame.bytes_per_line);
141 VFrame::VFrame(int w, int h, int color_model, long bytes_per_line)
144 // use bytes_per_line == 0 to allocate default unshared
145 if( !bytes_per_line ) { bytes_per_line = -1; use_shm = 0; }
146 params = new BC_Hash;
147 allocate_data(data, -1, 0, 0, 0, w, h,
148 color_model, bytes_per_line);
151 VFrame::VFrame(unsigned char *data, int shmid, int w, int h,
152 int color_model, long bytes_per_line)
155 params = new BC_Hash;
156 allocate_data(data, shmid, 0, 0, 0, w, h,
157 color_model, bytes_per_line);
160 VFrame::VFrame(unsigned char *data, int shmid,
161 long y_offset, long u_offset, long v_offset,
162 int w, int h, int color_model, long bytes_per_line)
165 params = new BC_Hash;
166 allocate_data(data, shmid, y_offset, u_offset, v_offset, w, h,
167 color_model, bytes_per_line);
170 VFrame::VFrame(BC_Bitmap *bitmap, int w, int h,
171 int color_model, long bytes_per_line)
174 params = new BC_Hash;
176 unsigned char *data = 0;
177 if( bitmap->is_shared() )
178 shmid = bitmap->get_shmid();
180 data = bitmap->get_data();
181 allocate_data(data, shmid,
182 bitmap->get_y_offset(),
183 bitmap->get_u_offset(),
184 bitmap->get_v_offset(),
185 w, h, color_model, bytes_per_line);
191 params = new BC_Hash;
192 this->color_model = BC_COMPRESSED;
200 // Delete effect stack
201 prev_effects.remove_all_objects();
202 next_effects.remove_all_objects();
207 int VFrame::equivalent(VFrame *src, int test_stacks)
209 return (src->get_color_model() == get_color_model() &&
210 src->get_w() == get_w() &&
211 src->get_h() == get_h() &&
212 src->bytes_per_line == bytes_per_line &&
213 (!test_stacks || equal_stacks(src)));
216 int VFrame::data_matches(VFrame *frame)
218 if(data && frame->get_data() &&
219 frame->params_match(get_w(), get_h(), get_color_model()) &&
220 get_data_size() == frame->get_data_size())
222 int data_size = get_data_size();
223 unsigned char *ptr1 = get_data();
224 unsigned char *ptr2 = frame->get_data();
225 for(int i = 0; i < data_size; i++)
227 if(*ptr1++ != *ptr2++) return 0;
234 // long VFrame::set_shm_offset(long offset)
236 // shm_offset = offset;
240 // long VFrame::get_shm_offset()
242 // return shm_offset;
245 int VFrame::get_memory_type()
250 int VFrame::params_match(int w, int h, int color_model)
252 return (this->w == w &&
254 this->color_model == color_model);
258 int VFrame::reset_parameters(int do_opengl)
263 memory_type = VFrame::PRIVATE;
271 compressed_allocated = 0;
272 compressed_size = 0; // Size of current image
279 sequence_number = -1;
282 pixel_rgb = 0x000000; // BLACK
283 pixel_yuv = 0x008080;
288 // By default, anything is going to be done in RAM
289 opengl_state = VFrame::RAM;
294 prev_effects.set_array_delete();
295 next_effects.set_array_delete();
299 int VFrame::clear_objects(int do_opengl)
312 if( memory_type != VFrame::SHARED )
313 printf("==del %p from %p\n", data, __builtin_return_address(0));
319 case VFrame::PRIVATE:
321 // if(this->w * this->h > 1500 * 1100)
322 // printf("VFrame::clear_objects 2 this=%p data=%p\n", this, data);
325 //printf("VFrame::clear_objects %d this=%p shmid=%p data=%p\n", __LINE__, this, shmid, data);
345 // Delete row pointers
370 int VFrame::get_field2_offset()
372 return field2_offset;
375 int VFrame::set_field2_offset(int value)
377 this->field2_offset = value;
381 void VFrame::set_keyframe(int value)
383 this->is_keyframe = value;
386 int VFrame::get_keyframe()
391 void VFrame::get_temp(VFrame *&vfrm, int w, int h, int color_model)
393 if( vfrm && ( vfrm->get_w() != w || vfrm->get_h() != h ) ) {
394 delete vfrm; vfrm = 0;
396 if( !vfrm ) vfrm = new VFrame(w, h, color_model, 0);
401 VFrameScene* VFrame::get_scene()
406 int VFrame::calculate_bytes_per_pixel(int color_model)
408 return BC_CModels::calculate_pixelsize(color_model);
411 long VFrame::get_bytes_per_line()
413 return bytes_per_line;
416 long VFrame::get_data_size()
418 return calculate_data_size(w, h, bytes_per_line, color_model) - 4;
421 long VFrame::calculate_data_size(int w, int h, int bytes_per_line, int color_model)
423 return BC_CModels::calculate_datasize(w, h, bytes_per_line, color_model);
426 void VFrame::create_row_pointers()
429 switch(color_model) {
431 if( this->v_offset ) break;
434 this->v_offset = sz + w / 4 * h / 4;
440 if( this->v_offset ) break;
443 this->v_offset = sz + sz / 4;
447 if( this->v_offset ) break;
450 this->v_offset = sz + sz / 2;
453 if( this->v_offset ) break;
456 this->v_offset = sz + sz;
459 if( this->v_offset ) break;
461 this->u_offset = sz * sizeof(uint8_t);
462 this->v_offset = 2 * sz * sizeof(uint8_t);
465 if( this->v_offset || a ) break;
466 a = this->data + 3 * sz * sizeof(float);
468 if( this->v_offset ) break;
470 this->u_offset = sz * sizeof(float);
471 this->v_offset = 2 * sz * sizeof(float);
475 rows = new unsigned char*[h];
476 for(int i = 0; i < h; i++)
477 rows[i] = &this->data[i * this->bytes_per_line];
480 y = this->data + this->y_offset;
481 u = this->data + this->u_offset;
482 v = this->data + this->v_offset;
485 int VFrame::allocate_data(unsigned char *data, int shmid,
486 long y_offset, long u_offset, long v_offset, int w, int h,
487 int color_model, long bytes_per_line)
491 this->color_model = color_model;
492 this->bytes_per_pixel = calculate_bytes_per_pixel(color_model);
493 this->y_offset = this->u_offset = this->v_offset = 0;
495 // printf("VFrame::allocate_data %d shmid == 0\n", __LINE__, shmid);
498 this->bytes_per_line = bytes_per_line >= 0 ?
499 bytes_per_line : this->bytes_per_pixel * w;
501 // Allocate data + padding for MMX
503 //printf("VFrame::allocate_data %d %p\n", __LINE__, this->data);
504 memory_type = VFrame::SHARED;
507 this->y_offset = y_offset;
508 this->u_offset = u_offset;
509 this->v_offset = v_offset;
511 else if(shmid >= 0) {
512 memory_type = VFrame::SHMGET;
513 this->data = (unsigned char*)shmat(shmid, NULL, 0);
514 //printf("VFrame::allocate_data %d shmid=%d data=%p\n", __LINE__, shmid, this->data);
516 this->y_offset = y_offset;
517 this->u_offset = u_offset;
518 this->v_offset = v_offset;
521 memory_type = VFrame::PRIVATE;
523 int size = calculate_data_size(this->w, this->h,
524 this->bytes_per_line, this->color_model);
525 if( use_shm && size >= SHM_MIN_SIZE &&
526 BC_WindowBase::get_resources()->use_vframe_shm() ) {
527 this->shmid = shmget(IPC_PRIVATE, size, IPC_CREAT | 0777);
528 if( this->shmid >= 0 ) {
529 this->data = (unsigned char*)shmat(this->shmid, NULL, 0);
530 //printf("VFrame::allocate_data %d %d %d %p\n", __LINE__, size, this->shmid, this->data);
531 // This causes it to automatically delete when the program exits.
532 shmctl(this->shmid, IPC_RMID, 0);
535 printf("VFrame::allocate_data %d could not allocate"
536 " shared memory, %dx%d (model %d) size=0x%08x\n",
537 __LINE__, w, h, color_model, size);
538 BC_Trace::dump_shm_stats(stdout);
541 // Have to use malloc for libpng
543 this->data = (unsigned char *)malloc(size);
547 // if(this->w * this->h > 1500 * 1100)
548 // printf("VFrame::allocate_data 2 this=%p w=%d h=%d this->data=%p\n",
549 // this, this->w, this->h, this->data);
552 printf("VFrame::allocate_data %dx%d: memory exhausted.\n", this->w, this->h);
554 printf("==new %p from %p sz %d\n", this->data, __builtin_return_address(0), size);
557 //printf("VFrame::allocate_data %d %p data=%p %d %d\n", __LINE__, this, this->data, this->w, this->h);
558 //if(size > 1000000) printf("VFrame::allocate_data %d\n", size);
561 // Create row pointers
562 create_row_pointers();
566 void VFrame::set_memory(unsigned char *data,
576 memory_type = VFrame::SHARED;
579 this->y_offset = y_offset;
580 this->u_offset = u_offset;
581 this->v_offset = v_offset;
586 memory_type = VFrame::SHMGET;
587 this->data = (unsigned char*)shmat(shmid, NULL, 0);
591 y = this->data + this->y_offset;
592 u = this->data + this->u_offset;
593 v = this->data + this->v_offset;
595 create_row_pointers();
598 void VFrame::set_memory(BC_Bitmap *bitmap)
601 unsigned char *data = 0;
602 if( bitmap->is_shared() && !bitmap->is_zombie() )
603 shmid = bitmap->get_shmid();
605 data = bitmap->get_data();
606 set_memory(data, shmid,
607 bitmap->get_y_offset(),
608 bitmap->get_u_offset(),
609 bitmap->get_v_offset());
612 void VFrame::set_compressed_memory(unsigned char *data,
621 memory_type = VFrame::SHARED;
628 memory_type = VFrame::SHMGET;
629 this->data = (unsigned char*)shmat(shmid, NULL, 0);
633 this->compressed_allocated = data_allocated;
634 this->compressed_size = data_size;
638 // Reallocate uncompressed buffer with or without alpha
639 int VFrame::reallocate(
650 // if(shmid == 0) printf("VFrame::reallocate %d shmid=%d\n", __LINE__, shmid);
652 // reset_parameters(0);
665 int VFrame::allocate_compressed_data(long bytes)
667 if(bytes < 1) return 1;
669 // Want to preserve original contents
670 if(data && compressed_allocated < bytes)
673 unsigned char *new_data = 0;
674 if(BC_WindowBase::get_resources()->use_vframe_shm() && use_shm)
676 new_shmid = shmget(IPC_PRIVATE,
679 new_data = (unsigned char*)shmat(new_shmid, NULL, 0);
680 shmctl(new_shmid, IPC_RMID, 0);
684 // Have to use malloc for libpng
685 new_data = (unsigned char *)malloc(bytes);
688 bcopy(data, new_data, compressed_allocated);
691 if(memory_type == VFrame::PRIVATE)
701 if(memory_type == VFrame::SHMGET)
709 compressed_allocated = bytes;
714 if(BC_WindowBase::get_resources()->use_vframe_shm() && use_shm)
716 shmid = shmget(IPC_PRIVATE,
719 data = (unsigned char*)shmat(shmid, NULL, 0);
720 shmctl(shmid, IPC_RMID, 0);
724 // Have to use malloc for libpng
725 data = (unsigned char *)malloc(bytes);
728 compressed_allocated = bytes;
735 int VFramePng::read_png(const unsigned char *data, long sz, double xscale, double yscale)
737 // Test for RAW format
738 if(data[0] == 'R' && data[1] == 'A' && data[2] == 'W' && data[3] == ' ') {
739 int new_color_model = BC_RGBA8888;
740 w = data[4] | (data[5] << 8) | (data[6] << 16) | (data[7] << 24);
741 h = data[8] | (data[9] << 8) | (data[10] << 16) | (data[11] << 24);
742 int components = data[12];
743 new_color_model = components == 3 ? BC_RGB888 : BC_RGBA8888;
744 // This shares the data directly
745 // reallocate(data + 20, 0, 0, 0, w, h, new_color_model, -1);
747 // Can't use shared data for theme since button constructions overlay the
749 reallocate(NULL, -1, 0, 0, 0, w, h, new_color_model, -1);
750 memcpy(get_data(), data + 16, w * h * components);
753 else if(data[0] == 0x89 && data[1] == 'P' && data[2] == 'N' && data[3] == 'G') {
755 png_structp png_ptr = png_create_read_struct(PNG_LIBPNG_VER_STRING, 0, 0, 0);
756 png_infop info_ptr = png_create_info_struct(png_ptr);
760 image = data; image_size = sz;
761 png_set_read_fn(png_ptr, this, PngReadFunction::png_read_function);
762 png_read_info(png_ptr, info_ptr);
764 w = png_get_image_width(png_ptr, info_ptr);
765 h = png_get_image_height(png_ptr, info_ptr);
767 int src_color_model = png_get_color_type(png_ptr, info_ptr);
769 /* tell libpng to strip 16 bit/color files down to 8 bits/color */
770 png_set_strip_16(png_ptr);
772 /* extract multiple pixels with bit depths of 1, 2, and 4 from a single
773 * byte into separate bytes (useful for paletted and grayscale images).
775 png_set_packing(png_ptr);
777 /* expand paletted colors into true RGB triplets */
778 if (src_color_model == PNG_COLOR_TYPE_PALETTE)
779 png_set_expand(png_ptr);
781 /* expand grayscale images to the full 8 bits from 1, 2, or 4 bits/pixel */
782 if (src_color_model == PNG_COLOR_TYPE_GRAY && png_get_bit_depth(png_ptr, info_ptr) < 8)
783 png_set_expand(png_ptr);
785 if (src_color_model == PNG_COLOR_TYPE_GRAY ||
786 src_color_model == PNG_COLOR_TYPE_GRAY_ALPHA)
787 png_set_gray_to_rgb(png_ptr);
789 /* expand paletted or RGB images with transparency to full alpha channels
790 * so the data will be available as RGBA quartets */
791 if (png_get_valid(png_ptr, info_ptr, PNG_INFO_tRNS)){
793 png_set_expand(png_ptr);
796 switch(src_color_model)
798 case PNG_COLOR_TYPE_GRAY:
799 case PNG_COLOR_TYPE_RGB:
800 new_color_model = BC_RGB888;
803 case PNG_COLOR_TYPE_GRAY_ALPHA:
804 case PNG_COLOR_TYPE_RGB_ALPHA:
806 new_color_model = BC_RGBA8888;
809 case PNG_COLOR_TYPE_PALETTE:
811 new_color_model = BC_RGBA8888;
813 new_color_model = BC_RGB888;
816 reallocate(NULL, -1, 0, 0, 0, w, h, new_color_model, -1);
818 //printf("VFrame::read_png %d %d %d %p\n", __LINE__, w, h, get_rows());
819 png_read_image(png_ptr, get_rows());
820 png_destroy_read_struct(&png_ptr, &info_ptr, NULL);
823 printf("VFrame::read_png %d: unknown file format"
824 " 0x%02x 0x%02x 0x%02x 0x%02x\n",
825 __LINE__, data[4], data[5], data[6], data[7]);
828 int ww = w * xscale, hh = h * yscale;
829 if( ww != w || hh != h ) {
830 VFrame vframe(*this);
831 reallocate(NULL, -1, 0, 0, 0, ww, hh, color_model, -1);
832 transfer_from(&vframe);
837 int VFrame::write_png(const char *path)
839 VFrame *vframe = this;
840 png_structp png_ptr = png_create_write_struct(PNG_LIBPNG_VER_STRING, 0, 0, 0);
841 png_infop info_ptr = png_create_info_struct(png_ptr);
842 FILE *out_fd = fopen(path, "w");
844 printf("VFrame::write_png %d %s %s\n", __LINE__, path, strerror(errno));
848 int png_cmodel = PNG_COLOR_TYPE_RGB;
849 int bc_cmodel = get_color_model();
850 switch( bc_cmodel ) {
851 case BC_RGB888: break;
852 case BC_RGBA8888: png_cmodel = PNG_COLOR_TYPE_RGB_ALPHA; break;
853 case BC_A8: png_cmodel = PNG_COLOR_TYPE_GRAY; break;
855 bc_cmodel = BC_RGB888;
856 if( BC_CModels::has_alpha(bc_cmodel) ) {
857 bc_cmodel = BC_RGBA8888;
858 png_cmodel = PNG_COLOR_TYPE_RGB_ALPHA;
860 vframe = new VFrame(get_w(), get_h(), bc_cmodel, 0);
861 vframe->transfer_from(this);
864 png_init_io(png_ptr, out_fd);
865 png_set_compression_level(png_ptr, 9);
866 png_set_IHDR(png_ptr, info_ptr, get_w(), get_h(), 8, png_cmodel,
867 PNG_INTERLACE_NONE, PNG_COMPRESSION_TYPE_DEFAULT, PNG_FILTER_TYPE_DEFAULT);
868 png_write_info(png_ptr, info_ptr);
869 png_write_image(png_ptr, vframe->get_rows());
870 png_write_end(png_ptr, info_ptr);
871 png_destroy_write_struct(&png_ptr, &info_ptr);
873 if( vframe != this ) delete vframe;
878 #define ZERO_YUV(components, type, max) \
880 for(int i = 0; i < h; i++) \
882 type *row = (type*)get_rows()[i]; \
883 for(int j = 0; j < w; j++) \
885 row[j * components] = 0; \
886 row[j * components + 1] = (max + 1) / 2; \
887 row[j * components + 2] = (max + 1) / 2; \
888 if(components == 4) row[j * components + 3] = 0; \
893 int VFrame::clear_frame()
896 //printf("VFrame::clear_frame %d\n", __LINE__);
897 switch(color_model) {
903 memset(get_u(), 0x80, w / 4 * h / 4);
904 memset(get_v(), 0x80, w / 4 * h / 4);
911 memset(get_u(), 0x80, sz / 4);
912 memset(get_v(), 0x80, sz / 4);
917 memset(get_u(), 0x80, sz / 2);
918 memset(get_v(), 0x80, sz / 2);
927 case BC_RGBA_FLOATP: if( a ) {
928 float *ap = (float *)a;
929 for( int i=sz; --i>=0; ++ap ) *ap = 0.f; }
930 case BC_RGB_FLOATP: {
931 float *rp = (float *)y;
932 for( int i=sz; --i>=0; ++rp ) *rp = 0.f;
933 float *gp = (float *)u;
934 for( int i=sz; --i>=0; ++gp ) *gp = 0.f;
935 float *bp = (float *)v;
936 for( int i=sz; --i>=0; ++bp ) *bp = 0.f;
940 memset(get_u(), 0x80, sz);
941 memset(get_v(), 0x80, sz);
945 ZERO_YUV(3, unsigned char, 0xff);
949 ZERO_YUV(4, unsigned char, 0xff);
953 ZERO_YUV(3, uint16_t, 0xffff);
956 case BC_YUVA16161616:
957 ZERO_YUV(4, uint16_t, 0xffff);
961 bzero(data, calculate_data_size(w, h, bytes_per_line, color_model));
967 void VFrame::rotate90()
969 // Allocate new frame
970 int new_w = h, new_h = w;
971 VFrame new_frame(new_w, new_h, color_model);
972 unsigned char **new_rows = new_frame.get_rows();
974 for(int in_y = 0, out_x = new_w - 1; in_y < h; in_y++, out_x--)
976 for(int in_x = 0, out_y = 0; in_x < w; in_x++, out_y++)
978 for(int k = 0; k < bytes_per_pixel; k++)
980 new_rows[out_y][out_x * bytes_per_pixel + k] =
981 rows[in_y][in_x * bytes_per_pixel + k];
988 unsigned char *new_data = new_frame.data;
989 new_frame.data = data;
992 new_rows = new_frame.rows;
993 new_frame.rows = rows;
996 int new_shmid = new_frame.shmid;
997 new_frame.shmid = shmid;
999 // swap bytes_per_line
1000 int new_bpl = new_frame.bytes_per_line;
1001 new_frame.bytes_per_line = bytes_per_line;
1002 bytes_per_line = new_bpl;
1003 new_frame.clear_objects(0);
1009 void VFrame::rotate270()
1011 // Allocate new frame
1012 int new_w = h, new_h = w;
1013 VFrame new_frame(new_w, new_h, color_model);
1014 unsigned char **new_rows = new_frame.get_rows();
1016 for(int in_y = 0, out_x = 0; in_y < h; in_y++, out_x++)
1018 for(int in_x = 0, out_y = new_h - 1; in_x < w; in_x++, out_y--)
1020 for(int k = 0; k < bytes_per_pixel; k++)
1022 new_rows[out_y][out_x * bytes_per_pixel + k] =
1023 rows[in_y][in_x * bytes_per_pixel + k];
1030 unsigned char *new_data = new_frame.data;
1031 new_frame.data = data;
1034 new_rows = new_frame.rows;
1035 new_frame.rows = rows;
1038 int new_shmid = new_frame.shmid;
1039 new_frame.shmid = shmid;
1041 // swap bytes_per_line
1042 int new_bpl = new_frame.bytes_per_line;
1043 new_frame.bytes_per_line = bytes_per_line;
1044 bytes_per_line = new_bpl;
1045 new_frame.clear_objects(0);
1051 void VFrame::flip_vert()
1053 unsigned char temp[bytes_per_line];
1054 for( int i=0, j=h; --j>i; ++i ) {
1055 memcpy(temp, rows[j], bytes_per_line);
1056 memcpy(rows[j], rows[i], bytes_per_line);
1057 memcpy(rows[i], temp, bytes_per_line);
1061 void VFrame::flip_horiz()
1063 unsigned char temp[32];
1064 for(int i = 0; i < h; i++)
1066 unsigned char *row = rows[i];
1067 for(int j = 0; j < bytes_per_line / 2; j += bytes_per_pixel)
1069 memcpy(temp, row + j, bytes_per_pixel);
1070 memcpy(row + j, row + bytes_per_line - j - bytes_per_pixel, bytes_per_pixel);
1071 memcpy(row + bytes_per_line - j - bytes_per_pixel, temp, bytes_per_pixel);
1078 int VFrame::copy_from(VFrame *frame)
1080 if(this->w != frame->get_w() ||
1081 this->h != frame->get_h())
1083 printf("VFrame::copy_from %d sizes differ src %dx%d != dst %dx%d\n",
1092 int w = MIN(this->w, frame->get_w());
1093 int h = MIN(this->h, frame->get_h());
1094 timestamp = frame->timestamp;
1096 switch(frame->color_model)
1099 allocate_compressed_data(frame->compressed_size);
1100 memcpy(data, frame->data, frame->compressed_size);
1101 this->compressed_size = frame->compressed_size;
1105 memcpy(get_y(), frame->get_y(), w * h);
1106 memcpy(get_u(), frame->get_u(), w / 4 * h / 4);
1107 memcpy(get_v(), frame->get_v(), w / 4 * h / 4);
1113 //printf("%d %d %p %p %p %p %p %p\n", w, h, get_y(), get_u(), get_v(), frame->get_y(), frame->get_u(), frame->get_v());
1114 memcpy(get_y(), frame->get_y(), w * h);
1115 memcpy(get_u(), frame->get_u(), w * h / 4);
1116 memcpy(get_v(), frame->get_v(), w * h / 4);
1120 //printf("%d %d %p %p %p %p %p %p\n", w, h, get_y(), get_u(), get_v(), frame->get_y(), frame->get_u(), frame->get_v());
1121 memcpy(get_y(), frame->get_y(), w * h);
1122 memcpy(get_u(), frame->get_u(), w * h / 2);
1123 memcpy(get_v(), frame->get_v(), w * h / 2);
1127 //printf("%d %d %p %p %p %p %p %p\n", w, h, get_y(), get_u(), get_v(), frame->get_y(), frame->get_u(), frame->get_v());
1128 memcpy(get_y(), frame->get_y(), w * h);
1129 memcpy(get_u(), frame->get_u(), w * h);
1130 memcpy(get_v(), frame->get_v(), w * h);
1133 // printf("VFrame::copy_from %d\n", calculate_data_size(w,
1136 // frame->color_model));
1137 // Copy without extra 4 bytes in case the source is a hardware device
1138 memmove(data, frame->data, get_data_size());
1145 int VFrame::transfer_from(VFrame *that, int bg_color, int in_x, int in_y, int in_w, int in_h)
1147 timestamp = that->timestamp;
1150 if( this->get_color_model() == that->get_color_model() &&
1151 this->get_w() == that->get_w() && this->get_h() == that->get_h() &&
1152 this->get_bytes_per_line() == that->get_bytes_per_line() )
1153 return this->copy_from(that);
1156 BC_CModels::transfer(
1157 this->get_rows(), that->get_rows(), // Packed data out/in
1158 this->get_y(), this->get_u(), this->get_v(), // Planar data out/in
1159 that->get_y(), that->get_u(), that->get_v(),
1160 0, 0, that->get_w(), that->get_h(), // Dimensions in/out
1161 0, 0, this->get_w(), this->get_h(),
1162 that->get_color_model(), this->get_color_model(), // Color models in/out
1163 bg_color, // alpha blend bg_color
1164 that->get_bytes_per_line(),
1165 this->get_bytes_per_line()); // rowspans (of luma for YUV)
1167 unsigned char *in_ptrs[4], *out_ptrs[4];
1168 unsigned char **inp, **outp;
1169 if( BC_CModels::is_planar(that->get_color_model()) ) {
1170 in_ptrs[0] = that->get_y();
1171 in_ptrs[1] = that->get_u();
1172 in_ptrs[2] = that->get_v();
1173 in_ptrs[3] = that->get_a();
1177 inp = that->get_rows();
1178 if( BC_CModels::is_planar(this->get_color_model()) ) {
1179 out_ptrs[0] = this->get_y();
1180 out_ptrs[1] = this->get_u();
1181 out_ptrs[2] = this->get_v();
1182 out_ptrs[3] = this->get_a();
1186 outp = this->get_rows();
1187 BC_CModels::transfer(outp, this->get_color_model(),
1188 0, 0, this->get_w(), this->get_h(),
1189 this->get_bytes_per_line(),
1190 inp, that->get_color_model(),
1191 in_x, in_y, in_w, in_h,
1192 that->get_bytes_per_line(),
1199 int VFrame::get_scale_tables(int *column_table, int *row_table,
1200 int in_x1, int in_y1, int in_x2, int in_y2,
1201 int out_x1, int out_y1, int out_x2, int out_y2)
1204 float w_in = in_x2 - in_x1;
1205 float h_in = in_y2 - in_y1;
1206 int w_out = out_x2 - out_x1;
1207 int h_out = out_y2 - out_y1;
1209 float hscale = w_in / w_out;
1210 float vscale = h_in / h_out;
1212 for(i = 0; i < w_out; i++)
1214 column_table[i] = (int)(hscale * i);
1217 for(i = 0; i < h_out; i++)
1219 row_table[i] = (int)(vscale * i) + in_y1;
1224 void VFrame::push_prev_effect(const char *name)
1227 prev_effects.append(ptr = new char[strlen(name) + 1]);
1229 if(prev_effects.total > MAX_STACK_ELEMENTS) prev_effects.remove_object(0);
1232 void VFrame::pop_prev_effect()
1234 if(prev_effects.total)
1235 prev_effects.remove_object(prev_effects.last());
1238 void VFrame::push_next_effect(const char *name)
1241 next_effects.append(ptr = new char[strlen(name) + 1]);
1243 if(next_effects.total > MAX_STACK_ELEMENTS) next_effects.remove_object(0);
1246 void VFrame::pop_next_effect()
1248 if(next_effects.total)
1249 next_effects.remove_object(next_effects.last());
1252 const char* VFrame::get_next_effect(int number)
1254 if(!next_effects.total) return "";
1256 if(number > next_effects.total - 1) number = next_effects.total - 1;
1258 return next_effects.values[next_effects.total - number - 1];
1261 const char* VFrame::get_prev_effect(int number)
1263 if(!prev_effects.total) return "";
1265 if(number > prev_effects.total - 1) number = prev_effects.total - 1;
1267 return prev_effects.values[prev_effects.total - number - 1];
1270 BC_Hash* VFrame::get_params()
1275 void VFrame::clear_stacks()
1277 next_effects.remove_all_objects();
1278 prev_effects.remove_all_objects();
1283 void VFrame::copy_params(VFrame *src)
1285 status = src->status;
1286 params->copy_from(src->params);
1289 void VFrame::copy_stacks(VFrame *src)
1293 for( int i=0; i < src->next_effects.total; ++i )
1294 next_effects.append(cstrdup(src->next_effects[i]));
1295 for( int i=0; i < src->prev_effects.total; ++i )
1296 prev_effects.append(cstrdup(src->prev_effects[i]));
1301 int VFrame::copy_vframe(VFrame *frame)
1304 return copy_from(frame);
1307 int VFrame::equal_stacks(VFrame *src)
1309 for(int i = 0; i < src->next_effects.total && i < next_effects.total; i++)
1311 if(strcmp(src->next_effects.values[i], next_effects.values[i])) return 0;
1314 for(int i = 0; i < src->prev_effects.total && i < prev_effects.total; i++)
1316 if(strcmp(src->prev_effects.values[i], prev_effects.values[i])) return 0;
1319 if(!params->equivalent(src->params)) return 0;
1323 void VFrame::dump_stacks()
1325 printf("VFrame::dump_stacks\n");
1326 printf(" next_effects:\n");
1327 for(int i = next_effects.total - 1; i >= 0; i--)
1328 printf(" %s\n", next_effects.values[i]);
1329 printf(" prev_effects:\n");
1330 for(int i = prev_effects.total - 1; i >= 0; i--)
1331 printf(" %s\n", prev_effects.values[i]);
1334 void VFrame::dump_params()
1341 printf("VFrame::dump %d this=%p\n", __LINE__, this);
1342 printf(" w=%d h=%d colormodel=%d rows=%p use_shm=%d shmid=%d\n",
1343 w, h, color_model, rows, use_shm, shmid);
1347 int VFrame::get_memory_usage()
1349 if(get_compressed_allocated()) return get_compressed_allocated();
1350 return get_h() * get_bytes_per_line();
1353 // rgb component colors (eg. from colors.h)
1354 // a (~alpha) transparency, 0x00==solid .. 0xff==transparent
1355 void VFrame::set_pixel_color(int rgb, int a)
1357 pixel_rgb = (rgb&0xffffff) | ~a<<24;
1358 int ir = 0xff & (pixel_rgb >> 16);
1359 int ig = 0xff & (pixel_rgb >> 8);
1360 int ib = 0xff & (pixel_rgb >> 0);
1361 YUV::yuv.rgb_to_yuv_8(ir, ig, ib);
1362 pixel_yuv = (~a<<24) | (ir<<16) | (ig<<8) | (ib<<0);
1365 void VFrame::set_stiple(int mask)
1370 int VFrame::draw_pixel(int x, int y)
1372 if( x < 0 || y < 0 || x >= get_w() || y >= get_h() ) return 1;
1374 #define DRAW_PIXEL(type, r, g, b, comps, a) { \
1375 type **rows = (type**)get_rows(); \
1376 type *rp = rows[y], *bp = rp + x*comps; \
1378 if( comps > 1 ) { bp[1] = g; bp[2] = b; } \
1379 if( comps == 4 ) bp[3] = a; \
1381 float fr = 0, fg = 0, fb = 0, fa = 0;
1382 int pixel_color = BC_CModels::is_yuv(color_model) ? pixel_yuv : pixel_rgb;
1383 int ir = (0xff & (pixel_color >> 16));
1384 int ig = (0xff & (pixel_color >> 8));
1385 int ib = (0xff & (pixel_color >> 0));
1386 int ia = (0xff & (pixel_color >> 24)) ^ 0xff; // transparency, not opacity
1387 if( (x+y) & stipple ) {
1388 ir = 255 - ir; ig = 255 - ig; ib = 255 - ib;
1390 int rr = (ir<<8) | ir, gg = (ig<<8) | ig, bb = (ib<<8) | ib, aa = (ia<<8) | ia;
1391 if( BC_CModels::is_float(color_model) ) {
1392 fr = rr/65535.f; fg = gg/65535.f; fb = bb/65535.f; fa = aa/65535.f;
1395 switch(get_color_model()) {
1397 DRAW_PIXEL(uint8_t, ib, 0, 0, 1, 0);
1401 DRAW_PIXEL(uint8_t, ir, ig, ib, 3, 0);
1405 DRAW_PIXEL(uint8_t, ir, ig, ib, 4, ia);
1409 DRAW_PIXEL(uint16_t, rr, gg, bb, 3, 0);
1411 case BC_RGBA16161616:
1412 case BC_YUVA16161616:
1413 DRAW_PIXEL(uint16_t, rr, gg, bb, 4, aa);
1416 DRAW_PIXEL(float, fr, fg, fb, 3, 0);
1419 DRAW_PIXEL(float, fr, fg, fb, 4, fa);
1427 void VFrame::draw_line(int x1, int y1, int x2, int y2)
1430 int tx = x1; x1 = x2; x2 = tx;
1431 int ty = y1; y1 = y2; y2 = ty;
1435 int dx = x2-x1, dy = y2-y1;
1436 int dx2 = 2*dx, dy2 = 2*dy;
1437 if( dx < 0 ) dx = -dx;
1438 int r = dx > dy ? dx : dy, n = r;
1440 if( dx2 < 0 ) dir += 1;
1442 if( dx2 >= 0 ) do { /* +Y, +X */
1444 if( (r -= dx2) < 0 ) { r += dy2; ++x; }
1445 } while( --n >= 0 );
1446 else do { /* +Y, -X */
1448 if( (r += dx2) < 0 ) { r += dy2; --x; }
1449 } while( --n >= 0 );
1452 if( dx2 >= 0 ) do { /* +X, +Y */
1454 if( (r -= dy2) < 0 ) { r += dx2; ++y; }
1455 } while( --n >= 0 );
1456 else do { /* -X, +Y */
1458 if( (r -= dy2) < 0 ) { r -= dx2; ++y; }
1459 } while( --n >= 0 );
1463 // g++ -dD -E - < /dev/null | grep DBL_EPSILON
1464 #ifndef __DBL_EPSILON__
1465 #define __DBL_EPSILON__ ((double)2.22044604925031308085e-16L)
1467 // weakest fraction * graphics integer range
1468 #define RND_EPSILON (__DBL_EPSILON__*65536)
1471 int rnd(double v) { return round(v)+RND_EPSILON; }
1474 int sx, sy, ex, ey; /* current point, end point */
1475 int xs, ys; /* x/y quadrant sign -1/1 */
1476 int64_t A, B, C; /* quadratic coefficients */
1477 int64_t r, dx, dy; /* residual, dr/dx and dr/dy */
1478 int xmxx, xmxy; /* x,y at apex */
1481 void init0(int x1,int y1, int x2,int y2, int x3,int y3, int top);
1482 void init1(int x1,int y1, int x2,int y2, int x3,int y3);
1483 int64_t rx() { return r + xs*8*dx + 4*A; }
1484 void moveX(int64_t r) {
1485 dx += xs*A; dy -= xs*B;
1486 this->r = r; sx += xs;
1488 int64_t ry() { return r + 8*dy + 4*C; }
1489 void moveY(int64_t r) {
1495 smooth_line(VFrame *vframe) { this->vframe = vframe; this->done = 0; }
1499 void smooth_line::draw()
1502 if( abs(dy) >= abs(dx) ) {
1503 if( xs*(sx-xmxx) >= 0 ) {
1504 if( ys > 0 ) { done = 1; return; }
1505 if( dy < 0 || ry() < 0 ) { moveY(ry()); goto xit; }
1506 xmxx = ex; xmxy = ey;
1511 if( abs(rr) < abs(r) )
1516 if( ys > 0 ) { done = 1; return; }
1517 xmxx = ex; xmxy = ey;
1522 if( abs(rr) < abs(r) )
1525 xit: vframe->draw_pixel(sx, sy);
1528 void VFrame::draw_smooth(int x1, int y1, int x2, int y2, int x3, int y3)
1530 if( (x1 == x2 && y1 == y2) || (x2 == x3 && y2 == y3) )
1531 draw_line(x1,y1, x3,y3);
1532 else if( x1 == x3 && y1 == y3 )
1533 draw_line(x1,y1, x2,y2);
1534 else if( (x2-x1) * (y2-y3) == (x2-x3) * (y2-y1) ) {
1535 // co-linear, draw line from min to max
1537 if( x2 < x1 ) { x1 = x2; y1 = y2; }
1538 if( x2 > x3 ) { x3 = x2; y3 = y2; }
1541 if( x2 > x1 ) { x1 = x2; y1 = y2; }
1542 if( x2 < x3 ) { x3 = x2; y3 = y2; }
1544 draw_line(x1,y1, x3,y3);
1547 smooth_draw(x1, y1, x2, y2, x3, y3);
1551 Non-Parametric Smooth Curve Generation. Don Kelly 1984
1558 Let the starting point be P. the ending point R. and the tangent vertex Q.
1559 A general point Z on the curve is then
1560 Z = (P + R - Q) + (Q - P) sin t + (Q - R) cos t
1562 Expanding the Cartesian coordinates around (P + R - Q) gives
1563 [x y] = Z - (P + R - Q)
1566 x = a*sin(t) + b*cos(t)
1567 y = c*sin(t) + d*cos(t)
1569 from which t can now be eliminated via
1570 c*x - a*y = (c*b - a*d)*cos(t)
1571 d*x - b*y = (a*d - c*b)*sin(t)
1573 giving the Cartesian equation for the ellipse as
1574 f(x, y) = (c*x - a*y)**2 + (d*x - b*y)**2 - (a*d - c*b)**2 = 0
1576 or: f(x, y) = A*x**2 - 2*B*x*y + C*y**2 + B**2 - A*C = 0
1577 where: A = c**2 + d**2, B = a*c + b*d, C = a**2 + b**2
1579 The maximum y extent of the ellipse may now be derived as follows:
1580 let df/dx = 0, 2*A*x = 2*B*y, x = y*B/A
1581 f(x, y) == B**2 * y**2 / A - 2*B**2 * y**2 / A + C*y**2 + B**2 - A*C = 0
1582 (A*C - B**2)*y = (A*C - B**2)*A
1583 max x = sqrt(C), at y = B/sqrt(C)
1584 max y = sqrt(A), at x = B/sqrt(A)
1589 /* x1,y1 = P, x2,y2 = Q, x3,y3=R,
1590 * draw from P to Q to R if top=0
1591 * or from P to (x,ymax) if top>0
1592 * or from Q to (x,ymax) if top<0
1594 void smooth_line::init0(int x1,int y1, int x2,int y2, int x3,int y3, int top)
1596 int x0 = x1+x3-x2, y0 = y1+y3-y2; // Q'
1598 int a = x2-x1, c = y2-y1;
1599 int b = x2-x3, d = y2-y3;
1600 A = c*c + d*d; C = a*a + b*b; B = a*c + b*d;
1602 sx = top >= 0 ? x1 : x3;
1603 sy = top >= 0 ? y1 : y3;
1604 xs = x2 > sx || (x2==sx && (x1+x3-sx)>=x2) ? 1 : -1;
1605 int64_t px = sx-x0, py = sy-y0;
1606 dx = A*px - B*py; dy = C*py - B*px;
1610 double ymy = sqrt(A), ymx = B/ymy;
1618 ys = a*b > 0 && (!top || top*xs*(b*c - a*d) > 0) ? -1 : 1;
1620 double xmx = xs*sqrt(C), xmy = B/xmx;
1621 xmxx = x0 + rnd(xmx);
1622 xmxy = y0 + rnd(xmy);
1625 xmxx = ex; xmxy = ey;
1629 /* x1,y1 = P, x2,y2 = Q, x3,y3=R,
1630 * draw from (x,ymax) to P
1632 void smooth_line::init1(int x1,int y1, int x2,int y2, int x3,int y3)
1634 int x0 = x1+x3-x2, y0 = y1+y3-y2; // Q'
1636 int a = x2-x1, c = y2-y1;
1637 int b = x2-x3, d = y2-y3;
1638 A = c*c + d*d; C = a*a + b*b; B = a*c + b*d;
1640 double ymy = -sqrt(A), ymx = B/ymy;
1641 int64_t px = rnd(ymx), py = rnd(ymy);
1642 sx = x0 + px; ex = x1;
1643 sy = y0 + py; ey = y1;
1644 xs = x2 > x1 || (x2==x1 && x3>=x2) ? 1 : -1;
1645 dx = A*px - B*py; dy = C*py - B*px;
1646 r = 4 * (A*px*px - 2*B*px*py + C*py*py + B*B - A*C);
1648 ys = a*b > 0 && xs*(b*c - a*d) < 0 ? -1 : 1;
1650 double xmx = xs*sqrt(C), xmy = B/xmx;
1651 xmxx = x0 + rnd(xmx);
1652 xmxy = y0 + rnd(xmy);
1656 xmxx = ex; xmxy = ey;
1659 vframe->draw_pixel(sx, sy);
1660 while( xs*(sx-xmxx) < 0 && (xs*dx < 0 || rx() < 0) ) {
1662 vframe->draw_pixel(sx, sy);
1667 void VFrame::smooth_draw(int x1, int y1, int x2, int y2, int x3, int y3)
1669 //printf("p smooth_draw( %d,%d, %d,%d, %d,%d )\n", x1,y1,x2,y2,x3,y3);
1670 if( y1 > y3 ) { // make y3 >= y1
1671 int xt = x1; x1 = x3; x3 = xt;
1672 int yt = y1; y1 = y3; y3 = yt;
1674 if( y1 > y2 && y3 > y2 ) {
1675 smooth_line lt(this), rt(this); // Q on bottom
1676 lt.init1(x1, y1, x2, y2, x3, y3);
1677 rt.init1(x3, y3, x2, y2, x1, y1);
1678 while( !lt.done || !rt.done ) {
1683 else if( y1 < y2 && y3 < y2 ) {
1684 smooth_line lt(this), rt(this); // Q on top
1685 lt.init0(x1, y1, x2, y2, x3, y3, 1);
1686 draw_pixel(lt.sx, lt.sy);
1687 rt.init0(x1, y1, x2, y2, x3, y3, -1);
1688 draw_pixel(rt.sx, rt.sy);
1689 while( !lt.done || !rt.done ) {
1695 smooth_line pt(this); // Q in between
1696 pt.init0(x1, y1, x2, y2, x3, y3, 0);
1697 draw_pixel(pt.sx, pt.sy);
1705 void VFrame::draw_rect(int x1, int y1, int x2, int y2)
1707 draw_line(x1, y1, x2, y1);
1708 draw_line(x2, y1 + 1, x2, y2);
1709 draw_line(x2 - 1, y2, x1, y2);
1710 draw_line(x1, y2 - 1, x1, y1 + 1);
1714 void VFrame::draw_oval(int x1, int y1, int x2, int y2)
1718 int center_x = (x2 + x1) / 2;
1719 int center_y = (y2 + y1) / 2;
1722 //printf("VFrame::draw_oval %d %d %d %d %d\n", __LINE__, x1, y1, x2, y2);
1724 for(int i = 0; i < h / 2; i++) {
1725 // A^2 = -(B^2) + C^2
1726 x_table[i] = (int)(sqrt(-SQR(h / 2 - i) + SQR(h / 2)) * w / h);
1727 //printf("VFrame::draw_oval %d i=%d x=%d\n", __LINE__, i, x_table[i]);
1730 for(int i = 0; i < h / 2 - 1; i++) {
1731 int x3 = x_table[i];
1732 int x4 = x_table[i + 1];
1735 for(int j = x3; j < x4; j++) {
1736 draw_pixel(center_x + j, y1 + i);
1737 draw_pixel(center_x - j, y1 + i);
1738 draw_pixel(center_x + j, y2 - i - 1);
1739 draw_pixel(center_x - j, y2 - i - 1);
1743 draw_pixel(center_x + x3, y1 + i);
1744 draw_pixel(center_x - x3, y1 + i);
1745 draw_pixel(center_x + x3, y2 - i - 1);
1746 draw_pixel(center_x - x3, y2 - i - 1);
1750 draw_pixel(center_x, y1);
1751 draw_pixel(center_x, y2 - 1);
1752 draw_pixel(x1, center_y);
1753 draw_pixel(x2 - 1, center_y);
1754 draw_pixel(x1, center_y - 1);
1755 draw_pixel(x2 - 1, center_y - 1);
1759 void VFrame::draw_arrow(int x1, int y1, int x2, int y2, int sz)
1761 double angle = atan((float)(y2 - y1) / (float)(x2 - x1));
1762 double angle1 = angle + (float)145 / 360 * 2 * M_PI;
1763 double angle2 = angle - (float)145 / 360 * 2 * M_PI;
1764 int s = x2 < x1 ? -1 : 1;
1765 int x3 = x2 + s * (int)(sz * cos(angle1));
1766 int y3 = y2 + s * (int)(sz * sin(angle1));
1767 int x4 = x2 + s * (int)(sz * cos(angle2));
1768 int y4 = y2 + s * (int)(sz * sin(angle2));
1771 draw_line(x1, y1, x2, y2);
1772 // draw_line(x1, y1 + 1, x2, y2 + 1);
1775 if(abs(y2 - y1) || abs(x2 - x1)) draw_line(x2, y2, x3, y3);
1777 if(abs(y2 - y1) || abs(x2 - x1)) draw_line(x2, y2, x4, y4);
1780 void VFrame::draw_x(int x, int y, int sz)
1782 draw_line(x-sz,y-sz, x+sz,y+sz);
1783 draw_line(x+sz,y-sz, x-sz,y+sz);
1785 void VFrame::draw_t(int x, int y, int sz)
1787 draw_line(x,y-sz, x,y+sz);
1788 draw_line(x+sz,y, x-sz,y);