Removed some debugging code that I didn't mean to commit.
[synfig.git] / synfig-core / trunk / src / modules / example / metaballs.cpp
1 /* === S Y N F I G ========================================================= */
2 /*!     \file metaballs.cpp
3 **      \brief Implements metaballs
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 <synfig/string.h>
33 #include <synfig/time.h>
34 #include <synfig/context.h>
35 #include <synfig/paramdesc.h>
36 #include <synfig/renddesc.h>
37 #include <synfig/surface.h>
38 #include <synfig/value.h>
39 #include <synfig/valuenode.h>
40 #include <ETL/pen>
41
42 #include "metaballs.h"
43
44 #endif
45
46 /* === U S I N G =========================================================== */
47
48 using namespace etl;
49 using namespace std;
50 using namespace synfig;
51
52 /* === G L O B A L S ======================================================= */
53
54 SYNFIG_LAYER_INIT(Metaballs);
55 SYNFIG_LAYER_SET_NAME(Metaballs,"metaballs");
56 SYNFIG_LAYER_SET_LOCAL_NAME(Metaballs,_("Metaballs"));
57 SYNFIG_LAYER_SET_CATEGORY(Metaballs,_("Default"));
58 SYNFIG_LAYER_SET_VERSION(Metaballs,"0.1");
59 SYNFIG_LAYER_SET_CVS_ID(Metaballs,"$Id$");
60
61 /* === P R O C E D U R E S ================================================= */
62
63 /* === M E T H O D S ======================================================= */
64
65 /* === E N T R Y P O I N T ================================================= */
66
67 Metaballs::Metaballs():
68         Layer_Composite(1.0,Color::BLEND_STRAIGHT),
69         color(Color::black())
70 {
71 }
72
73 bool
74 Metaballs::set_param(const String & param, const ValueBase &value)
75 {
76         if(     param=="centers" && value.same_type_as(centers))
77         {
78                 centers = value;
79                 return true;
80         }
81
82         if(     param=="weights" && value.same_type_as(weights))
83         {
84                 weights = value;
85                 return true;
86         }
87
88         if(     param=="radii" && value.same_type_as(radii))
89         {
90                 radii = value;
91                 return true;
92         }
93
94         IMPORT(color);
95         IMPORT(threshold);
96
97         return Layer_Composite::set_param(param,value);
98 }
99
100 ValueBase
101 Metaballs::get_param(const String &param)const
102 {
103         EXPORT(color);
104
105         EXPORT(radii);
106         EXPORT(weights);
107         EXPORT(centers);
108         EXPORT(threshold);
109
110         EXPORT_NAME();
111         EXPORT_VERSION();
112
113         return Layer_Composite::get_param(param);
114 }
115
116 Layer::Vocab
117 Metaballs::get_param_vocab()const
118 {
119         Layer::Vocab ret(Layer_Composite::get_param_vocab());
120
121         ret.push_back(ParamDesc("color")
122                 .set_local_name(_("Color"))
123         );
124
125         ret.push_back(ParamDesc("centers")
126                 .set_local_name(_("Points"))
127         );
128
129         ret.push_back(ParamDesc("radii")
130                 .set_local_name(_("Radii"))
131         );
132
133         ret.push_back(ParamDesc("weights")
134                 .set_local_name(_("Weights"))
135         );
136
137         ret.push_back(ParamDesc("threshold")
138                 .set_local_name(_("Threshold"))
139         );
140
141         return ret;
142 }
143
144 static inline Real densityfunc(const synfig::Point &p, const synfig::Point &c, Real R)
145 {
146         const Real dx = p[0] - c[0];
147         const Real dy = p[1] - c[1];
148
149         const Real n = (1 - (dx*dx + dy*dy)/(R*R));
150         return (n*n*n);
151
152         /*
153         f(d) = (1 - d^2)^3
154         f'(d) = -6d * (1 - d^2)^2
155
156         could use this too...
157         f(d) = (1 - d^2)^2
158         f'(d) = -6d * (1 - d^2)
159         */
160 }
161
162 Real
163 Metaballs::totaldensity(const Point &pos) const
164 {
165         Real density = 0;
166
167         //sum up weighted functions
168         for(unsigned int i=0;i<centers.size();i++)
169         {
170                 density += weights[i] * densityfunc(pos,centers[i], radii[i]);
171         }
172
173         return density;
174 }
175
176 Color
177 Metaballs::get_color(Context context, const Point &pos)const
178 {
179         Real dens = totaldensity(pos);
180
181         if(dens >= threshold)
182                 return color;
183         else
184                 return context.get_color(pos);
185 }
186
187 bool
188 Metaballs::accelerated_render(Context context,Surface *surface,int quality, const RendDesc &renddesc, ProgressCallback *cb)const
189 {
190         // Width and Height of a pixel
191         const Point br(renddesc.get_br()),
192                                 tl(renddesc.get_tl());
193
194         const int       w = renddesc.get_w(),
195                                 h = renddesc.get_h();
196
197         Real    pw = renddesc.get_pw();
198         Real    ph = renddesc.get_ph();
199
200         SuperCallback supercb(cb,0,9000,10000);
201
202         if(!context.accelerated_render(surface,quality,renddesc,&supercb))
203         {
204                 if(cb)cb->error(strprintf(__FILE__"%d: Accelerated Renderer Failure",__LINE__));
205                 return false;
206         }
207
208         Point pos(tl[0],tl[1]);
209
210         Real    dens;
211
212         if(!context.accelerated_render(surface,quality,renddesc,&supercb))
213         {
214                 if(cb)cb->error(strprintf(__FILE__"%d: Accelerated Renderer Failure",__LINE__));
215                 return false;
216         }
217
218         for(int y = 0; y < h; y++, pos[1] += ph)
219         {
220                 pos[0] = tl[0];
221                 for(int x = 0; x < w; x++, pos[0] += pw)
222                 {
223                         dens = totaldensity(pos);
224
225                         if(dens >= threshold)
226                         {
227                                 (*surface)[y][x] = Color::blend(color,(*surface)[y][x],get_amount(),get_blend_method());
228                         }
229                 }
230         }
231
232         // Mark our progress as finished
233         if(cb && !cb->amount_complete(10000,10000))
234                 return false;
235
236         return true;
237 }