Partially fix 1702760: stop compiler warning about trgt_png.cpp.
[synfig.git] / synfig-core / trunk / src / synfig / surfacenew.cpp
1 /* === S Y N F I G ========================================================= */
2 /*!     \file surfacenew.cpp
3 **      \brief Template File
4 **
5 **      $Id$
6 **
7 **      \legal
8 **      Copyright (c) 2002-2005 Robert B. Quattlebaum Jr., Adrian Bentley
9 **
10 **      This package is free software; you can redistribute it and/or
11 **      modify it under the terms of the GNU General Public License as
12 **      published by the Free Software Foundation; either version 2 of
13 **      the License, or (at your option) any later version.
14 **
15 **      This package is distributed in the hope that it will be useful,
16 **      but WITHOUT ANY WARRANTY; without even the implied warranty of
17 **      MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
18 **      General Public License for more details.
19 **      \endlegal
20 */
21 /* ========================================================================= */
22
23 /* === H E A D E R S ======================================================= */
24
25 #ifdef USING_PCH
26 #       include "pch.h"
27 #else
28 #ifdef HAVE_CONFIG_H
29 #       include <config.h>
30 #endif
31
32 #include "surfacenew.h"
33 #include <ETL/ref_count>
34 #include "mutex.h"
35
36 #endif
37
38 /* === U S I N G =========================================================== */
39
40 using namespace std;
41 using namespace etl;
42 using namespace synfig;
43
44 /* === M A C R O S ========================================================= */
45
46 /* === G L O B A L S ======================================================= */
47
48 /* === P R O C E D U R E S ================================================= */
49
50 /* === C L A S S E S & S T R U C T S ======================================= */
51
52 class SurfaceNew::ChannelData
53 {
54 private:
55         etl::reference_counter ref_count_;
56
57         float* data_;
58         float* origin_;
59
60         int w_,h_,stride_;
61
62 public:
63
64         RWLock rw_lock;
65
66         ChannelData():
67                 is_defined_(false),
68                 data_(0),
69                 origin_(0),
70                 w_(0),
71                 h_(0)
72         {
73         }
74
75         ~ChannelData()
76         {
77                 if(ref_count_.unique())
78                         delete [] data_;
79         }
80
81         void set_wh(int w, int h)
82         {
83                 w_=w;
84                 h_=h;
85                 stride_=w;
86
87                 if(data_&&ref_count_.is_unique())
88                         delete [] data_;
89
90                 ref_count.make_unique();
91                 data_=new float [w_*h_];
92                 origin_=data_;
93                 clear();
94         }
95
96         void crop(int x, int y, int w, int h)
97         {
98                 origin_=origin+y*stride_+x;
99                 w_=w;
100                 h_=h;
101         }
102
103         int get_stride()const
104         {
105                 return stride_;
106         }
107
108         void clear()
109         {
110                 for(int i=0;i<h;i++)
111                         bzero(origin_+stride_*i,w_*sizeof(float));
112         }
113
114         void fill(float v)
115         {
116                 float* ptr(get_data());
117
118                 for(int y=0;y<h;y++,ptr+=stride_)
119                         for(int i=0;i<w_;i++)
120                                 ptr[i]=v;
121         }
122
123         float* get_data() { return origin_; }
124
125         void make_unique()
126         {
127                 if(!ref_count_.unique())
128                 {
129                         ref_count_.make_unique();
130                         float* old_data(origin_);
131                         int old_stride;
132
133                         data_=new float [w_*h_];
134                         origin_=data_;
135                         stride_=w_;
136
137                         for(int i=0;i<h;i++)
138                                 memcpy(data_+i*stride_,old_data+i*old_stride,sizeof(float)*w_);
139                 }
140         }
141 }; // END of class ChannelData
142
143 /* === M E T H O D S ======================================================= */
144
145 SurfaceNew::SurfaceNew():
146         w_(0),
147         h_(0),
148         color_system_(COLORSYS_RGB),
149         premult_flag_(false)
150 {
151 }
152
153 SurfaceNew~SurfaceNew()
154 {
155 }
156
157 SurfaceNew::Handle
158 SurfaceNew::create(int w, int h, ColorSystem sys=COLORSYS_RGB)
159 {
160         Handle ret(new SurfaceNew);
161
162         ret.set_wh(w,h);
163         ret.set_color_system(sys);
164
165         return ret;
166 }
167
168 SurfaceNew::Handle
169 SurfaceNew::create(const Surface&)
170 {
171         // ***WRITEME***
172         return 0;
173 }
174
175 SurfaceNew::Handle
176 SurfaceNew::create(HandleConst orig)
177 {
178         Lock lock(orig);
179
180         Handle ret(new SurfaceNew);
181
182         ret.w_=orig.w_;
183         ret.h_=orig.h_;
184         ret.color_system_=orig.color_system_;
185         ret.premult_flag_=orig.premult_flag_;
186         ret.channel_map_=orig.channel_map_;
187
188         return ret;
189 }
190
191 Handle
192 SurfaceNew::crop(HandleConst, int x, int y, int w, int h)
193 {
194         Lock lock(orig);
195
196         Handle ret(new SurfaceNew);
197
198         ret.w_=orig.w_;
199         ret.h_=orig.h_;
200         ret.color_system_=orig.color_system_;
201         ret.premult_flag_=orig.premult_flag_;
202         ret.channel_map_=orig.channel_map_;
203
204         std::map<Channel,ChannelData>::iterator iter;
205         for(iter=ret.channel_map_.begin();iter!=ret.channel_map_.end();++iter)
206                 iter->crop(x,y,w,h);
207
208         return ret;
209 }
210
211 int
212 SurfaceNew::get_w()const
213 {
214         return w_;
215 }
216
217 int
218 SurfaceNew::get_h()const
219 {
220         return h_;
221 }
222
223 void
224 SurfaceNew::set_wh(int w, int h)
225 {
226         if(w!=w_ || h!=h_)
227         {
228                 w_=w;
229                 h_=h;
230                 channel_map_.clear();
231         }
232 }
233
234 SurfaceNew::ColorSystem
235 SurfaceNew::get_color_system()const
236 {
237         return color_system_;
238 }
239
240 void
241 SurfaceNew::set_color_system(SurfaceNew::ColorSystem x)
242 {
243         color_system_=x;
244 }
245
246 Color
247 SurfaceNew::get_color(int x, int y)const
248 {
249         // This operation is rather expensive, as it should be.
250         // I want to discurage people from using it all over the place.
251
252         Color ret(
253                 lock_channel_const(CHAN_R).get_value(x,y),
254                 lock_channel_const(CHAN_G).get_value(x,y),
255                 lock_channel_const(CHAN_B).get_value(x,y),
256                 lock_channel_const(CHAN_A).get_value(x,y)
257         );
258
259         if(get_premult())
260         {
261                 ret=ret.demult_alpha();
262         }
263
264         return ret;
265 }
266
267 void
268 SurfaceNew::lock()
269 {
270         mutex_.lock();
271 }
272
273 void
274 SurfaceNew::unlock()
275 {
276         mutex_.unlock();
277 }
278
279 bool
280 SurfaceNew::trylock()
281 {
282         return mutex_.trylock();
283 }
284
285 SurfaceNew::ChannelLock
286 SurfaceNew::lock_channel(SurfaceNew::Channel chan)
287 {
288         if(!is_channel_defined(chan)
289                 channel_map_[chan].set_wh(get_w(),get_h());
290         else
291                 channel_map_[chan].make_unique();
292
293         ChannelLockConst channel_lock;
294
295         channel_lock.surface_=this;
296         channel_lock.channel_=chan;
297
298         channel_map_[chan].rw_lock.writer_lock();
299
300         return channel_lock;
301 }
302
303 SurfaceNew::ChannelLockConst
304 SurfaceNew::lock_channel_const(SurfaceNew::Channel chan)const
305 {
306         if(!is_channel_defined(chan)
307                 channel_map_[chan].set_wh(get_w(),get_h());
308
309         ChannelLockConst channel_lock;
310
311         channel_lock.surface_=this;
312         channel_lock.channel_=chan;
313
314         channel_map_[chan].rw_lock.reader_lock();
315
316         return channel_lock;
317 }
318
319 SurfaceNew::ChannelLock
320 SurfaceNew::lock_channel_alpha(SurfaceNew::Channel chan)
321 {
322         // Change this when per-channel alpha
323         // is implemented
324         return lock_channel(CHAN_A);
325 }
326
327 SurfaceNew::ChannelLockConst
328 SurfaceNew::lock_channel_alpha_const(SurfaceNew::Channel chan)const
329 {
330         // Change this when per-channel alpha
331         // is implemented
332         return lock_channel_const(CHAN_A);
333 }
334
335 bool
336 SurfaceNew::is_channel_defined(Channel chan)const
337 {
338         return channel_map_.count(chan);
339 }
340
341 bool
342 SurfaceNew::get_premult()const
343 {
344         return premult_flag_;
345 }
346
347 void
348 SurfaceNew::set_premult(bool x)
349 {
350         if(x==premult_flag_)
351                 return;
352
353         premult_flag_=x;
354
355         for(int i=0;i<3;i++)
356         {
357                 Channel chan;
358                 if(get_color_system()==COLORSYS_RGB)switch(i)
359                 {
360                         case 0: chan=CHAN_R;
361                         case 1: chan=CHAN_G;
362                         case 2: chan=CHAN_B;
363                 }
364                 else
365                 if(get_color_system()==COLORSYS_YUV)switch(i)
366                 {
367                         case 0: chan=CHAN_Y;
368                         case 1: chan=CHAN_U;
369                         case 2: chan=CHAN_V;
370                 }
371
372                 // If this channel isn't defined, then
373                 // skip it and move on to the next one
374                 if(!is_channel_defined(chan))
375                         continue;
376
377                 ChannelLock color_channel(lock_channel(chan));
378                 ChannelLockConst alpha_channel(lock_channel_alpha_const(chan));
379                 const int w(get_w());
380                 const int h(get_h());
381
382                 float* color_ptr(color_channel.get_data_ptr());
383                 const float* alpha_ptr(alpha_channel.get_data_ptr());
384
385                 const int color_pitch(color_channel.get_data_ptr_stride()-w);
386                 const int alpha_pitch(alpha_channel.get_data_ptr_stride()-w);
387
388                 if(premult_flag_)
389                 {
390                         for(int y=0;y<h;y++,color_ptr+=color_pitch,alpha_ptr+=alpha_pitch)
391                                 for(int x=0;x<w;x++,color_ptr++,alpha_ptr++)
392                                         *color_ptr *= *alpha_ptr;
393                 }
394                 else
395                 {
396                         for(int y=0;y<h;y++,color_ptr+=color_pitch,alpha_ptr+=alpha_pitch)
397                                 for(int x=0;x<w;x++,color_ptr++,alpha_ptr++)
398                                         *color_ptr /= *alpha_ptr;
399                 }
400         }
401 }
402
403 void
404 SurfaceNew::blit(
405         Handle dest, int x_dest, int y_dest,
406         HandleConst src, int x_src, int y_src, int w_src, int h_src,
407         float amount=1.0, Color::BlendMethod bm=Color::BLEND_COMPOSITE
408 )
409 {
410         blit(
411                 dest,
412                 x_dest,
413                 y_dest,
414                 crop(
415                         src,
416                         x,
417                         y,
418                         w,
419                         h
420                 ),
421                 amount,
422                 bm
423         );
424 }
425
426 void
427 SurfaceNew::blit(
428         Handle dest, int x_dest, int y_dest,
429         HandleConst src
430         float amount=1.0, Color::BlendMethod bm=Color::BLEND_COMPOSITE
431 )
432 {
433         int w(src->get_w()), h(src->get_h);
434
435         // Clip
436         {
437                 int x(0), y(0);
438
439                 if(x_dest+w>dest.get_w())
440                         w=dest.get_w()-x_dest;
441                 if(y_dest+h>dest.get_h())
442                         h=dest.get_h()-y_dest;
443                 if(x_dest<0)
444                 {
445                         x-=x_dest;
446                         w+=x_dest;
447                 }
448                 if(y_dest<0)
449                 {
450                         y-=y_dest;
451                         h+=y_dest;
452                 }
453                 src=crop(src,x,y,w,h);
454         }
455
456         dest=crop(dest,x_dest,y_dest,w,h);
457
458         if(bm==Color::BLEND_STRAIGHT)
459         {
460                 chan_mlt(dest,amount/(1.0-amount));
461                 chan_add(dest,src);
462                 chan_mlt(dest,(1.0-amount)/amount);
463         }
464
465         if(bm==Color::BLEND_COMPOSITE)
466         {
467
468         }
469 }
470
471
472
473 // -----------------------------------------------------------------------------------
474
475 SurfaceChannelLockConst::SurfaceChannelLockConst():
476         data_ptr_checked_out_(false)
477 {
478 }
479
480 SurfaceChannelLockConst::~SurfaceChannelLockConst()
481 {
482         if(data_ptr_checked_out_)
483                 release_data_ptr();
484
485         if(surface_ && ref_count_.is_unique())
486                 return surface->channel_map_[channel_].rw_lock.reader_unlock();
487         surface=0;
488 }
489
490 SurfaceChannel
491 SurfaceChannelLockConst::get_channel()const
492 {
493         return channel_;
494 }
495
496 int
497 SurfaceChannelLockConst::get_w()const
498 {
499         return surface_->get_w();
500 }
501
502 int
503 SurfaceChannelLockConst::get_h()const
504 {
505         return surface_->get_h();
506 }
507
508 float
509 SurfaceChannelLockConst::get_value(int x, int y)
510 {
511         // WOW! CRAZY SLOW!
512         const ChannelData& channel_data(surface_->channel_map_[channel_]);
513         return *(channel_data.get_data()+y*channel_data.get_stride()+x);
514 }
515
516 const float*
517 SurfaceChannelLockConst::get_data_ptr()const
518 {
519         data_ptr_checked_out_=true;
520
521         // WOW! CRAZY SLOW!
522         return surface_->channel_map_[channel_].get_data();
523 }
524
525 int
526 SurfaceChannelLockConst::get_data_ptr_stride()const
527 {
528         return surface_->channel_map_[channel_].get_stride();
529 }
530
531 void
532 SurfaceChannelLockConst::release_data_ptr()const
533 {
534         data_ptr_checked_out_=false;
535 }
536
537 SurfaceChannelLockConst::operator bool()const
538 {
539         return static_cast<bool>(surface_);
540 }
541
542 // -----------------------------------------------------------------------------------
543
544 SurfaceChannelLock::SurfaceChannelLock()
545 {
546 }
547
548 SurfaceChannelLock::~SurfaceChannelLock()
549 {
550         if(data_ptr_checked_out_)
551                 release_data_ptr();
552
553         if(surface_ && ref_count_.is_unique())
554                 return surface_->channel_map_[channel_].rw_lock.writer_unlock();
555         surface=0;
556 }
557
558 void
559 SurfaceChannelLock::clear()
560 {
561         return surface_->channel_map_[channel_].clear();
562 }
563
564 void
565 SurfaceChannelLock::fill(float v)
566 {
567         return surface_->channel_map_[channel_].fill(v);
568 }
569
570 void
571 SurfaceChannelLock::set_value(int x, int y, float v)
572 {
573         // WOW! CRAZY SLOW!
574         const ChannelData& channel_data(surface_->channel_map_[channel_]);
575         *(channel_data.get_data()+y*channel_data.get_stride()+x)=v;
576 }
577
578 float*
579 SurfaceChannelLock::get_data_ptr()
580 {
581         data_ptr_checked_out_=true;
582
583         // WOW! CRAZY SLOW!
584         return surface_->channel_map_[channel_].get_data();
585 }
586
587
588
589 // -----------------------------------------------------------------------------------
590
591
592
593 void
594 SurfaceNew::chan_mlt(ChannelLock& dest, float x)
595 {
596         float* ptr(dest.get_data_ptr());
597         const int w(dest.get_w());
598         const int h(dest.get_h());
599         const int pitch(dest.get_data_pitch()-w);
600
601         int(y=0;y<h;y++,ptr+=pitch)
602                 int(x=0;x<w;x++,ptr++)
603                         *ptr*=x;
604 }
605
606 void
607 SurfaceNew::chan_mlt(ChannelLock& dest, const ChannelLockConst& x)
608 {
609         float* d_ptr(dest.get_data_ptr());
610         const float* s_ptr(x.get_data_ptr());
611         const int w(dest.get_w());
612         const int h(dest.get_h());
613         const int d_pitch(dest.get_data_stride()-w);
614         const int s_pitch(x.get_data_stride()-w);
615
616         int(y=0;y<h;y++,d_ptr+=d_pitch,s_ptr+=s_pitch)
617                 int(x=0;x<w;x++,d_ptr++,s_ptr++)
618                         *d_ptr *= *s_ptr;
619 }
620
621 void
622 SurfaceNew::chan_div(ChannelLock& dest, float x)
623 {
624         float* ptr(dest.get_data_ptr());
625         const int w(dest.get_w());
626         const int h(dest.get_h());
627         const int pitch(dest.get_data_pitch()-w);
628
629         int(y=0;y<h;y++,ptr+=pitch)
630                 int(x=0;x<w;x++,ptr++)
631                         *ptr/=x;
632 }
633
634 void
635 SurfaceNew::chan_div(ChannelLock& dest, const ChannelLockConst& x)
636 {
637         float* d_ptr(dest.get_data_ptr());
638         const float* s_ptr(x.get_data_ptr());
639         const int w(dest.get_w());
640         const int h(dest.get_h());
641         const int d_pitch(dest.get_data_stride()-w);
642         const int s_pitch(x.get_data_stride()-w);
643
644         int(y=0;y<h;y++,d_ptr+=d_pitch,s_ptr+=s_pitch)
645                 int(x=0;x<w;x++,d_ptr++,s_ptr++)
646                         *d_ptr /= *s_ptr;
647 }
648
649 void
650 SurfaceNew::chan_add(ChannelLock& dest, float x)
651 {
652         float* ptr(dest.get_data_ptr());
653         const int w(dest.get_w());
654         const int h(dest.get_h());
655         const int pitch(dest.get_data_pitch()-w);
656
657         int(y=0;y<h;y++,ptr+=pitch)
658                 int(x=0;x<w;x++,ptr++)
659                         *ptr+=x;
660 }
661
662 void
663 SurfaceNew::chan_add(ChannelLock& dest, const ChannelLockConst& x)
664 {
665         float* d_ptr(dest.get_data_ptr());
666         const float* s_ptr(x.get_data_ptr());
667         const int w(dest.get_w());
668         const int h(dest.get_h());
669         const int d_pitch(dest.get_data_stride()-w);
670         const int s_pitch(x.get_data_stride()-w);
671
672         int(y=0;y<h;y++,d_ptr+=d_pitch,s_ptr+=s_pitch)
673                 int(x=0;x<w;x++,d_ptr++,s_ptr++)
674                         *d_ptr += *s_ptr;
675 }
676
677 void
678 SurfaceNew::chan_sub(ChannelLock& dest, float x)
679 {
680         float* ptr(dest.get_data_ptr());
681         const int w(dest.get_w());
682         const int h(dest.get_h());
683         const int pitch(dest.get_data_pitch()-w);
684
685         int(y=0;y<h;y++,ptr+=pitch)
686                 int(x=0;x<w;x++,ptr++)
687                         *ptr-=x;
688 }
689
690 void
691 SurfaceNew::chan_sub(ChannelLock& dest, const ChannelLockConst& x)
692 {
693         float* d_ptr(dest.get_data_ptr());
694         const float* s_ptr(x.get_data_ptr());
695         const int w(dest.get_w());
696         const int h(dest.get_h());
697         const int d_pitch(dest.get_data_stride()-w);
698         const int s_pitch(x.get_data_stride()-w);
699
700         int(y=0;y<h;y++,d_ptr+=d_pitch,s_ptr+=s_pitch)
701                 int(x=0;x<w;x++,d_ptr++,s_ptr++)
702                         *d_ptr -= *s_ptr;
703 }
704
705
706