Fix some RNA/scene issuess with unified paint settings.
[blender.git] / source / blender / blenkernel / intern / brush.c
1 /*
2  * ***** BEGIN GPL LICENSE BLOCK *****
3  *
4  * This program is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU General Public License
6  * as published by the Free Software Foundation; either version 2
7  * of the License, or (at your option) any later version.
8  *
9  * This program is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  * GNU General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License
15  * along with this program; if not, write to the Free Software Foundation,
16  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
17  *
18  * The Original Code is Copyright (C) 2001-2002 by NaN Holding BV.
19  * All rights reserved.
20  *
21  * The Original Code is: all of this file.
22  *
23  * Contributor(s): none yet.
24  *
25  * ***** END GPL LICENSE BLOCK *****
26  */
27
28 /** \file blender/blenkernel/intern/brush.c
29  *  \ingroup bke
30  */
31
32
33 #include <math.h>
34 #include <string.h>
35
36 #include "MEM_guardedalloc.h"
37
38 #include "DNA_brush_types.h"
39 #include "DNA_color_types.h"
40 #include "DNA_scene_types.h"
41 #include "DNA_object_types.h"
42 #include "DNA_windowmanager_types.h"
43
44 #include "WM_types.h"
45
46 #include "RNA_access.h"
47
48 #include "BLI_bpath.h"
49 #include "BLI_math.h"
50 #include "BLI_blenlib.h"
51 #include "BLI_rand.h"
52 #include "BLI_utildefines.h"
53
54 #include "BKE_brush.h"
55 #include "BKE_colortools.h"
56 #include "BKE_global.h"
57 #include "BKE_image.h"
58 #include "BKE_library.h"
59 #include "BKE_main.h"
60 #include "BKE_paint.h"
61 #include "BKE_texture.h"
62 #include "BKE_icons.h"
63
64 #include "IMB_imbuf.h"
65 #include "IMB_imbuf_types.h"
66
67 #include "RE_render_ext.h" /* externtex */
68 #include "RE_shader_ext.h"
69
70 static void brush_set_defaults(Brush *brush)
71 {
72         brush->blend = 0;
73         brush->flag = 0;
74
75         brush->ob_mode = OB_MODE_ALL_PAINT;
76
77         /* BRUSH SCULPT TOOL SETTINGS */
78         brush->size= 35; /* radius of the brush in pixels */
79         brush->alpha= 0.5f; /* brush strength/intensity probably variable should be renamed? */
80         brush->autosmooth_factor= 0.0f;
81         brush->crease_pinch_factor= 0.5f;
82         brush->sculpt_plane = SCULPT_DISP_DIR_AREA;
83         brush->plane_offset= 0.0f; /* how far above or below the plane that is found by averaging the faces */
84         brush->plane_trim= 0.5f;
85         brush->clone.alpha= 0.5f;
86         brush->normal_weight= 0.0f;
87         brush->flag |= BRUSH_ALPHA_PRESSURE;
88
89         /* BRUSH PAINT TOOL SETTINGS */
90         brush->rgb[0]= 1.0f; /* default rgb color of the brush when painting - white */
91         brush->rgb[1]= 1.0f;
92         brush->rgb[2]= 1.0f;
93
94         /* BRUSH STROKE SETTINGS */
95         brush->flag |= (BRUSH_SPACE|BRUSH_SPACE_ATTEN);
96         brush->spacing= 10; /* how far each brush dot should be spaced as a percentage of brush diameter */
97
98         brush->smooth_stroke_radius= 75;
99         brush->smooth_stroke_factor= 0.9f;
100
101         brush->rate= 0.1f; /* time delay between dots of paint or sculpting when doing airbrush mode */
102
103         brush->jitter= 0.0f;
104
105         /* BRUSH TEXTURE SETTINGS */
106         default_mtex(&brush->mtex);
107
108         brush->texture_sample_bias= 0; /* value to added to texture samples */
109         brush->texture_overlay_alpha= 33;
110
111         /* brush appearance  */
112
113         brush->add_col[0]= 1.00; /* add mode color is light red */
114         brush->add_col[1]= 0.39;
115         brush->add_col[2]= 0.39;
116
117         brush->sub_col[0]= 0.39; /* subtract mode color is light blue */
118         brush->sub_col[1]= 0.39;
119         brush->sub_col[2]= 1.00;
120 }
121
122 /* Datablock add/copy/free/make_local */
123
124 Brush *add_brush(const char *name)
125 {
126         Brush *brush;
127
128         brush= alloc_libblock(&G.main->brush, ID_BR, name);
129
130         /* enable fake user by default */
131         brush->id.flag |= LIB_FAKEUSER;
132
133         brush_set_defaults(brush);
134
135         brush->sculpt_tool = SCULPT_TOOL_DRAW; /* sculpting defaults to the draw tool for new brushes */
136
137          /* the default alpha falloff curve */
138         brush_curve_preset(brush, CURVE_PRESET_SMOOTH);
139
140         return brush;
141 }
142
143 Brush *copy_brush(Brush *brush)
144 {
145         Brush *brushn;
146         
147         brushn= copy_libblock(&brush->id);
148
149         if (brush->mtex.tex)
150                 id_us_plus((ID*)brush->mtex.tex);
151
152         if (brush->icon_imbuf)
153                 brushn->icon_imbuf= IMB_dupImBuf(brush->icon_imbuf);
154
155         brushn->preview = NULL;
156
157         brushn->curve= curvemapping_copy(brush->curve);
158
159         /* enable fake user by default */
160         if (!(brushn->id.flag & LIB_FAKEUSER)) {
161                 brushn->id.flag |= LIB_FAKEUSER;
162                 brushn->id.us++;
163         }
164         
165         return brushn;
166 }
167
168 /* not brush itself */
169 void free_brush(Brush *brush)
170 {
171         if (brush->mtex.tex)
172                 brush->mtex.tex->id.us--;
173
174         if (brush->icon_imbuf)
175                 IMB_freeImBuf(brush->icon_imbuf);
176
177         BKE_previewimg_free(&(brush->preview));
178
179         curvemapping_free(brush->curve);
180 }
181
182 static void extern_local_brush(Brush *brush)
183 {
184         id_lib_extern((ID *)brush->mtex.tex);
185         id_lib_extern((ID *)brush->clone.image);
186 }
187
188 void make_local_brush(Brush *brush)
189 {
190
191         /* - only lib users: do nothing
192          * - only local users: set flag
193          * - mixed: make copy
194          */
195
196         Main *bmain= G.main;
197         Scene *scene;
198         int is_local= FALSE, is_lib= FALSE;
199
200         if(brush->id.lib==NULL) return;
201
202         if(brush->clone.image) {
203                 /* special case: ima always local immediately. Clone image should only
204                    have one user anyway. */
205                 id_clear_lib_data(bmain, &brush->clone.image->id);
206                 extern_local_brush(brush);
207         }
208
209         for(scene= bmain->scene.first; scene && ELEM(0, is_lib, is_local); scene=scene->id.next) {
210                 if(paint_brush(&scene->toolsettings->imapaint.paint)==brush) {
211                         if(scene->id.lib) is_lib= TRUE;
212                         else is_local= TRUE;
213                 }
214         }
215
216         if(is_local && is_lib == FALSE) {
217                 id_clear_lib_data(bmain, &brush->id);
218                 extern_local_brush(brush);
219
220                 /* enable fake user by default */
221                 if (!(brush->id.flag & LIB_FAKEUSER)) {
222                         brush->id.flag |= LIB_FAKEUSER;
223                         brush->id.us++;
224                 }
225         }
226         else if(is_local && is_lib) {
227                 Brush *brush_new= copy_brush(brush);
228                 brush_new->id.us= 1; /* only keep fake user */
229                 brush_new->id.flag |= LIB_FAKEUSER;
230
231                 /* Remap paths of new ID using old library as base. */
232                 BKE_id_lib_local_paths(bmain, brush->id.lib, &brush_new->id);
233                 
234                 for(scene= bmain->scene.first; scene; scene=scene->id.next) {
235                         if(paint_brush(&scene->toolsettings->imapaint.paint)==brush) {
236                                 if(scene->id.lib==NULL) {
237                                         paint_brush_set(&scene->toolsettings->imapaint.paint, brush_new);
238                                 }
239                         }
240                 }
241         }
242 }
243
244 void brush_debug_print_state(Brush *br)
245 {
246         /* create a fake brush and set it to the defaults */
247         Brush def= {{NULL}};
248         brush_set_defaults(&def);
249         
250 #define BR_TEST(field, t)                                       \
251         if(br->field != def.field)                              \
252                 printf("br->" #field " = %" #t ";\n", br->field)
253
254 #define BR_TEST_FLAG(_f)                                \
255         if((br->flag & _f) && !(def.flag & _f))         \
256                 printf("br->flag |= " #_f ";\n");       \
257         else if(!(br->flag & _f) && (def.flag & _f))    \
258                 printf("br->flag &= ~" #_f ";\n")
259         
260
261         /* print out any non-default brush state */
262         BR_TEST(normal_weight, f);
263
264         BR_TEST(blend, d);
265         BR_TEST(size, d);
266
267         /* br->flag */
268         BR_TEST_FLAG(BRUSH_AIRBRUSH);
269         BR_TEST_FLAG(BRUSH_TORUS);
270         BR_TEST_FLAG(BRUSH_ALPHA_PRESSURE);
271         BR_TEST_FLAG(BRUSH_SIZE_PRESSURE);
272         BR_TEST_FLAG(BRUSH_JITTER_PRESSURE);
273         BR_TEST_FLAG(BRUSH_SPACING_PRESSURE);
274         BR_TEST_FLAG(BRUSH_FIXED_TEX);
275         BR_TEST_FLAG(BRUSH_RAKE);
276         BR_TEST_FLAG(BRUSH_ANCHORED);
277         BR_TEST_FLAG(BRUSH_DIR_IN);
278         BR_TEST_FLAG(BRUSH_SPACE);
279         BR_TEST_FLAG(BRUSH_SMOOTH_STROKE);
280         BR_TEST_FLAG(BRUSH_PERSISTENT);
281         BR_TEST_FLAG(BRUSH_ACCUMULATE);
282         BR_TEST_FLAG(BRUSH_LOCK_ALPHA);
283         BR_TEST_FLAG(BRUSH_ORIGINAL_NORMAL);
284         BR_TEST_FLAG(BRUSH_OFFSET_PRESSURE);
285         BR_TEST_FLAG(BRUSH_SPACE_ATTEN);
286         BR_TEST_FLAG(BRUSH_ADAPTIVE_SPACE);
287         BR_TEST_FLAG(BRUSH_LOCK_SIZE);
288         BR_TEST_FLAG(BRUSH_TEXTURE_OVERLAY);
289         BR_TEST_FLAG(BRUSH_EDGE_TO_EDGE);
290         BR_TEST_FLAG(BRUSH_RESTORE_MESH);
291         BR_TEST_FLAG(BRUSH_INVERSE_SMOOTH_PRESSURE);
292         BR_TEST_FLAG(BRUSH_RANDOM_ROTATION);
293         BR_TEST_FLAG(BRUSH_PLANE_TRIM);
294         BR_TEST_FLAG(BRUSH_FRONTFACE);
295         BR_TEST_FLAG(BRUSH_CUSTOM_ICON);
296
297         BR_TEST(jitter, f);
298         BR_TEST(spacing, d);
299         BR_TEST(smooth_stroke_radius, d);
300         BR_TEST(smooth_stroke_factor, f);
301         BR_TEST(rate, f);
302
303         BR_TEST(alpha, f);
304
305         BR_TEST(sculpt_plane, d);
306
307         BR_TEST(plane_offset, f);
308
309         BR_TEST(autosmooth_factor, f);
310
311         BR_TEST(crease_pinch_factor, f);
312
313         BR_TEST(plane_trim, f);
314
315         BR_TEST(texture_sample_bias, f);
316         BR_TEST(texture_overlay_alpha, d);
317
318         BR_TEST(add_col[0], f);
319         BR_TEST(add_col[1], f);
320         BR_TEST(add_col[2], f);
321         BR_TEST(sub_col[0], f);
322         BR_TEST(sub_col[1], f);
323         BR_TEST(sub_col[2], f);
324
325         printf("\n");
326         
327 #undef BR_TEST
328 #undef BR_TEST_FLAG
329 }
330
331 void brush_reset_sculpt(Brush *br)
332 {
333         /* enable this to see any non-default
334            settings used by a brush:
335
336         brush_debug_print_state(br);
337         */
338
339         brush_set_defaults(br);
340         brush_curve_preset(br, CURVE_PRESET_SMOOTH);
341
342         switch(br->sculpt_tool) {
343         case SCULPT_TOOL_CLAY:
344                 br->flag |= BRUSH_FRONTFACE;
345                 break;
346         case SCULPT_TOOL_CREASE:
347                 br->flag |= BRUSH_DIR_IN;
348                 br->alpha = 0.25;
349                 break;
350         case SCULPT_TOOL_FILL:
351                 br->add_col[1] = 1;
352                 br->sub_col[0] = 0.25;
353                 br->sub_col[1] = 1;
354                 break;
355         case SCULPT_TOOL_FLATTEN:
356                 br->add_col[1] = 1;
357                 br->sub_col[0] = 0.25;
358                 br->sub_col[1] = 1;
359                 break;
360         case SCULPT_TOOL_INFLATE:
361                 br->add_col[0] = 0.750000;
362                 br->add_col[1] = 0.750000;
363                 br->add_col[2] = 0.750000;
364                 br->sub_col[0] = 0.250000;
365                 br->sub_col[1] = 0.250000;
366                 br->sub_col[2] = 0.250000;
367                 break;
368         case SCULPT_TOOL_NUDGE:
369                 br->add_col[0] = 0.250000;
370                 br->add_col[1] = 1.000000;
371                 br->add_col[2] = 0.250000;
372                 break;
373         case SCULPT_TOOL_PINCH:
374                 br->add_col[0] = 0.750000;
375                 br->add_col[1] = 0.750000;
376                 br->add_col[2] = 0.750000;
377                 br->sub_col[0] = 0.250000;
378                 br->sub_col[1] = 0.250000;
379                 br->sub_col[2] = 0.250000;
380                 break;
381         case SCULPT_TOOL_SCRAPE:
382                 br->add_col[1] = 1.000000;
383                 br->sub_col[0] = 0.250000;
384                 br->sub_col[1] = 1.000000;
385                 break;
386         case SCULPT_TOOL_ROTATE:
387                 br->alpha = 1.0;
388                 break;
389         case SCULPT_TOOL_SMOOTH:
390                 br->flag &= ~BRUSH_SPACE_ATTEN;
391                 br->spacing = 5;
392                 br->add_col[0] = 0.750000;
393                 br->add_col[1] = 0.750000;
394                 br->add_col[2] = 0.750000;
395                 break;
396         case SCULPT_TOOL_GRAB:
397         case SCULPT_TOOL_SNAKE_HOOK:
398         case SCULPT_TOOL_THUMB:
399                 br->size = 75;
400                 br->flag &= ~BRUSH_ALPHA_PRESSURE;
401                 br->flag &= ~BRUSH_SPACE;
402                 br->flag &= ~BRUSH_SPACE_ATTEN;
403                 br->add_col[0] = 0.250000;
404                 br->add_col[1] = 1.000000;
405                 br->add_col[2] = 0.250000;
406                 break;
407         default:
408                 break;
409         }
410 }
411
412 /* Library Operations */
413 void brush_curve_preset(Brush *b, /*CurveMappingPreset*/int preset)
414 {
415         CurveMap *cm = NULL;
416
417         if(!b->curve)
418                 b->curve = curvemapping_add(1, 0, 0, 1, 1);
419
420         cm = b->curve->cm;
421         cm->flag &= ~CUMA_EXTEND_EXTRAPOLATE;
422
423         b->curve->preset = preset;
424         curvemap_reset(cm, &b->curve->clipr, b->curve->preset, CURVEMAP_SLOPE_NEGATIVE);
425         curvemapping_changed(b->curve, 0);
426 }
427
428 int brush_texture_set_nr(Brush *brush, int nr)
429 {
430         ID *idtest, *id=NULL;
431
432         id= (ID *)brush->mtex.tex;
433
434         idtest= (ID*)BLI_findlink(&G.main->tex, nr-1);
435         if(idtest==NULL) { /* new tex */
436                 if(id) idtest= (ID *)copy_texture((Tex *)id);
437                 else idtest= (ID *)add_texture("Tex");
438                 idtest->us--;
439         }
440         if(idtest!=id) {
441                 brush_texture_delete(brush);
442
443                 brush->mtex.tex= (Tex*)idtest;
444                 id_us_plus(idtest);
445
446                 return 1;
447         }
448
449         return 0;
450 }
451
452 int brush_texture_delete(Brush *brush)
453 {
454         if(brush->mtex.tex)
455                 brush->mtex.tex->id.us--;
456
457         return 1;
458 }
459
460 int brush_clone_image_set_nr(Brush *brush, int nr)
461 {
462         if(brush && nr > 0) {
463                 Image *ima= (Image*)BLI_findlink(&G.main->image, nr-1);
464
465                 if(ima) {
466                         brush_clone_image_delete(brush);
467                         brush->clone.image= ima;
468                         id_us_plus(&ima->id);
469                         brush->clone.offset[0]= brush->clone.offset[1]= 0.0f;
470
471                         return 1;
472                 }
473         }
474
475         return 0;
476 }
477
478 int brush_clone_image_delete(Brush *brush)
479 {
480         if (brush && brush->clone.image) {
481                 brush->clone.image->id.us--;
482                 brush->clone.image= NULL;
483                 return 1;
484         }
485
486         return 0;
487 }
488
489 /* Brush Sampling */
490 void brush_sample_tex(Brush *brush, float *xy, float *rgba, const int thread)
491 {
492         MTex *mtex= &brush->mtex;
493
494         if (mtex && mtex->tex) {
495                 float co[3], tin, tr, tg, tb, ta;
496                 int hasrgb;
497                 const int radius= brush_size(brush);
498
499                 co[0]= xy[0]/radius;
500                 co[1]= xy[1]/radius;
501                 co[2]= 0.0f;
502
503                 hasrgb= externtex(mtex, co, &tin, &tr, &tg, &tb, &ta, thread);
504
505                 if (hasrgb) {
506                         rgba[0]= tr;
507                         rgba[1]= tg;
508                         rgba[2]= tb;
509                         rgba[3]= ta;
510                 }
511                 else {
512                         rgba[0]= tin;
513                         rgba[1]= tin;
514                         rgba[2]= tin;
515                         rgba[3]= 1.0f;
516                 }
517         }
518         else if (rgba)
519                 rgba[0]= rgba[1]= rgba[2]= rgba[3]= 1.0f;
520 }
521
522
523 void brush_imbuf_new(Brush *brush, short flt, short texfall, int bufsize, ImBuf **outbuf, int use_color_correction)
524 {
525         ImBuf *ibuf;
526         float xy[2], dist, rgba[4], *dstf;
527         int x, y, rowbytes, xoff, yoff, imbflag;
528         const int radius= brush_size(brush);
529         char *dst, crgb[3];
530         const float alpha= brush_alpha(brush);
531         float brush_rgb[3];
532     
533         imbflag= (flt)? IB_rectfloat: IB_rect;
534         xoff = -bufsize/2.0f + 0.5f;
535         yoff = -bufsize/2.0f + 0.5f;
536         rowbytes= bufsize*4;
537
538         if (*outbuf)
539                 ibuf= *outbuf;
540         else
541                 ibuf= IMB_allocImBuf(bufsize, bufsize, 32, imbflag);
542
543         if (flt) {
544                 copy_v3_v3(brush_rgb, brush->rgb);
545                 if(use_color_correction){
546                         srgb_to_linearrgb_v3_v3(brush_rgb, brush_rgb);
547                 }
548
549                 for (y=0; y < ibuf->y; y++) {
550                         dstf = ibuf->rect_float + y*rowbytes;
551
552                         for (x=0; x < ibuf->x; x++, dstf+=4) {
553                                 xy[0] = x + xoff;
554                                 xy[1] = y + yoff;
555
556                                 if (texfall == 0) {
557                                         dist = sqrt(xy[0]*xy[0] + xy[1]*xy[1]);
558
559                                         copy_v3_v3(dstf, brush_rgb);
560                                         dstf[3]= alpha*brush_curve_strength_clamp(brush, dist, radius);
561                                 }
562                                 else if (texfall == 1) {
563                                         brush_sample_tex(brush, xy, dstf, 0);
564                                 }
565                                 else {
566                                         dist = sqrt(xy[0]*xy[0] + xy[1]*xy[1]);
567
568                                         brush_sample_tex(brush, xy, rgba, 0);
569                                         mul_v3_v3v3(dstf, rgba, brush_rgb);
570                                         dstf[3] = rgba[3]*alpha*brush_curve_strength_clamp(brush, dist, radius);
571                                 }
572                         }
573                 }
574         }
575         else {
576                 crgb[0]= FTOCHAR(brush->rgb[0]);
577                 crgb[1]= FTOCHAR(brush->rgb[1]);
578                 crgb[2]= FTOCHAR(brush->rgb[2]);
579
580                 for (y=0; y < ibuf->y; y++) {
581                         dst = (char*)ibuf->rect + y*rowbytes;
582
583                         for (x=0; x < ibuf->x; x++, dst+=4) {
584                                 xy[0] = x + xoff;
585                                 xy[1] = y + yoff;
586
587                                 if (texfall == 0) {
588                                         dist = sqrt(xy[0]*xy[0] + xy[1]*xy[1]);
589
590                                         dst[0]= crgb[0];
591                                         dst[1]= crgb[1];
592                                         dst[2]= crgb[2];
593                                         dst[3]= FTOCHAR(alpha*brush_curve_strength(brush, dist, radius));
594                                 }
595                                 else if (texfall == 1) {
596                                         brush_sample_tex(brush, xy, rgba, 0);
597                                         dst[0]= FTOCHAR(rgba[0]);
598                                         dst[1]= FTOCHAR(rgba[1]);
599                                         dst[2]= FTOCHAR(rgba[2]);
600                                         dst[3]= FTOCHAR(rgba[3]);
601                                 }
602                                 else if (texfall == 2) {
603                                         dist = sqrt(xy[0]*xy[0] + xy[1]*xy[1]);
604
605                                         brush_sample_tex(brush, xy, rgba, 0);
606                                         dst[0] = FTOCHAR(rgba[0]*brush->rgb[0]);
607                                         dst[1] = FTOCHAR(rgba[1]*brush->rgb[1]);
608                                         dst[2] = FTOCHAR(rgba[2]*brush->rgb[2]);
609                                         dst[3] = FTOCHAR(rgba[3]*alpha*brush_curve_strength_clamp(brush, dist, radius));
610                                 } else {
611                                         dist = sqrt(xy[0]*xy[0] + xy[1]*xy[1]);
612
613                                         brush_sample_tex(brush, xy, rgba, 0);
614                                         dst[0]= crgb[0];
615                                         dst[1]= crgb[1];
616                                         dst[2]= crgb[2];
617                                         dst[3] = FTOCHAR(rgba[3]*alpha*brush_curve_strength_clamp(brush, dist, radius));
618                                 }
619                         }
620                 }
621         }
622
623         *outbuf= ibuf;
624 }
625
626 /* Brush Painting */
627
628 typedef struct BrushPainterCache {
629         short enabled;
630
631         int size;                       /* size override, if 0 uses 2*brush_size(brush) */
632         short flt;                      /* need float imbuf? */
633         short texonly;          /* no alpha, color or fallof, only texture in imbuf */
634
635         int lastsize;
636         float lastalpha;
637         float lastjitter;
638
639         ImBuf *ibuf;
640         ImBuf *texibuf;
641         ImBuf *maskibuf;
642 } BrushPainterCache;
643
644 struct BrushPainter {
645         Scene *scene;
646         Brush *brush;
647
648         float lastmousepos[2];  /* mouse position of last paint call */
649
650         float accumdistance;    /* accumulated distance of brush since last paint op */
651         float lastpaintpos[2];  /* position of last paint op */
652         float startpaintpos[2]; /* position of first paint */
653
654         double accumtime;               /* accumulated time since last paint op (airbrush) */
655         double lasttime;                /* time of last update */
656
657         float lastpressure;
658
659         short firsttouch;               /* first paint op */
660
661         float startsize;
662         float startalpha;
663         float startjitter;
664         float startspacing;
665
666         BrushPainterCache cache;
667 };
668
669 BrushPainter *brush_painter_new(Scene *scene, Brush *brush)
670 {
671         BrushPainter *painter= MEM_callocN(sizeof(BrushPainter), "BrushPainter");
672
673         painter->brush= brush;
674         painter->scene= scene;
675         painter->firsttouch= 1;
676         painter->cache.lastsize= -1; /* force ibuf create in refresh */
677
678         painter->startsize = brush_size(brush);
679         painter->startalpha = brush_alpha(brush);
680         painter->startjitter = brush->jitter;
681         painter->startspacing = brush->spacing;
682
683         return painter;
684 }
685
686 void brush_painter_require_imbuf(BrushPainter *painter, short flt, short texonly, int size)
687 {
688         if ((painter->cache.flt != flt) || (painter->cache.size != size) ||
689                 ((painter->cache.texonly != texonly) && texonly)) {
690                 if (painter->cache.ibuf) IMB_freeImBuf(painter->cache.ibuf);
691                 if (painter->cache.maskibuf) IMB_freeImBuf(painter->cache.maskibuf);
692                 painter->cache.ibuf= painter->cache.maskibuf= NULL;
693                 painter->cache.lastsize= -1; /* force ibuf create in refresh */
694         }
695
696         if (painter->cache.flt != flt) {
697                 if (painter->cache.texibuf) IMB_freeImBuf(painter->cache.texibuf);
698                 painter->cache.texibuf= NULL;
699                 painter->cache.lastsize= -1; /* force ibuf create in refresh */
700         }
701
702         painter->cache.size= size;
703         painter->cache.flt= flt;
704         painter->cache.texonly= texonly;
705         painter->cache.enabled= 1;
706 }
707
708 void brush_painter_free(BrushPainter *painter)
709 {
710         Brush *brush = painter->brush;
711
712         brush_set_size(brush, painter->startsize);
713         brush_set_alpha(brush, painter->startalpha);
714         brush->jitter = painter->startjitter;
715         brush->spacing = painter->startspacing;
716
717         if (painter->cache.ibuf) IMB_freeImBuf(painter->cache.ibuf);
718         if (painter->cache.texibuf) IMB_freeImBuf(painter->cache.texibuf);
719         if (painter->cache.maskibuf) IMB_freeImBuf(painter->cache.maskibuf);
720         MEM_freeN(painter);
721 }
722
723 static void brush_painter_do_partial(BrushPainter *painter, ImBuf *oldtexibuf, int x, int y, int w, int h, int xt, int yt, float *pos)
724 {
725         Brush *brush= painter->brush;
726         ImBuf *ibuf, *maskibuf, *texibuf;
727         float *bf, *mf, *tf, *otf=NULL, xoff, yoff, xy[2], rgba[4];
728         char *b, *m, *t, *ot= NULL;
729         int dotexold, origx= x, origy= y;
730         const int radius= brush_size(brush);
731
732         xoff = -radius + 0.5f;
733         yoff = -radius + 0.5f;
734         xoff += (int)pos[0] - (int)painter->startpaintpos[0];
735         yoff += (int)pos[1] - (int)painter->startpaintpos[1];
736
737         ibuf = painter->cache.ibuf;
738         texibuf = painter->cache.texibuf;
739         maskibuf = painter->cache.maskibuf;
740
741         dotexold = (oldtexibuf != NULL);
742
743         /* not sure if it's actually needed or it's a mistake in coords/sizes
744            calculation in brush_painter_fixed_tex_partial_update(), but without this
745            limitation memory gets corrupted at fast strokes with quite big spacing (sergey) */
746         w = MIN2(w, ibuf->x);
747         h = MIN2(h, ibuf->y);
748
749         if (painter->cache.flt) {
750                 for (; y < h; y++) {
751                         bf = ibuf->rect_float + (y*ibuf->x + origx)*4;
752                         tf = texibuf->rect_float + (y*texibuf->x + origx)*4;
753                         mf = maskibuf->rect_float + (y*maskibuf->x + origx)*4;
754
755                         if (dotexold)
756                                 otf = oldtexibuf->rect_float + ((y - origy + yt)*oldtexibuf->x + xt)*4;
757
758                         for (x=origx; x < w; x++, bf+=4, mf+=4, tf+=4) {
759                                 if (dotexold) {
760                                         copy_v3_v3(tf, otf);
761                                         tf[3] = otf[3];
762                                         otf += 4;
763                                 }
764                                 else {
765                                         xy[0] = x + xoff;
766                                         xy[1] = y + yoff;
767
768                                         brush_sample_tex(brush, xy, tf, 0);
769                                 }
770
771                                 bf[0] = tf[0]*mf[0];
772                                 bf[1] = tf[1]*mf[1];
773                                 bf[2] = tf[2]*mf[2];
774                                 bf[3] = tf[3]*mf[3];
775                         }
776                 }
777         }
778         else {
779                 for (; y < h; y++) {
780                         b = (char*)ibuf->rect + (y*ibuf->x + origx)*4;
781                         t = (char*)texibuf->rect + (y*texibuf->x + origx)*4;
782                         m = (char*)maskibuf->rect + (y*maskibuf->x + origx)*4;
783
784                         if (dotexold)
785                                 ot = (char*)oldtexibuf->rect + ((y - origy + yt)*oldtexibuf->x + xt)*4;
786
787                         for (x=origx; x < w; x++, b+=4, m+=4, t+=4) {
788                                 if (dotexold) {
789                                         t[0] = ot[0];
790                                         t[1] = ot[1];
791                                         t[2] = ot[2];
792                                         t[3] = ot[3];
793                                         ot += 4;
794                                 }
795                                 else {
796                                         xy[0] = x + xoff;
797                                         xy[1] = y + yoff;
798
799                                         brush_sample_tex(brush, xy, rgba, 0);
800                                         t[0]= FTOCHAR(rgba[0]);
801                                         t[1]= FTOCHAR(rgba[1]);
802                                         t[2]= FTOCHAR(rgba[2]);
803                                         t[3]= FTOCHAR(rgba[3]);
804                                 }
805
806                                 b[0] = t[0]*m[0]/255;
807                                 b[1] = t[1]*m[1]/255;
808                                 b[2] = t[2]*m[2]/255;
809                                 b[3] = t[3]*m[3]/255;
810                         }
811                 }
812         }
813 }
814
815 static void brush_painter_fixed_tex_partial_update(BrushPainter *painter, float *pos)
816 {
817         Brush *brush= painter->brush;
818         BrushPainterCache *cache= &painter->cache;
819         ImBuf *oldtexibuf, *ibuf;
820         int imbflag, destx, desty, srcx, srcy, w, h, x1, y1, x2, y2;
821         const int diameter= 2*brush_size(brush);
822
823         imbflag= (cache->flt)? IB_rectfloat: IB_rect;
824         if (!cache->ibuf)
825                 cache->ibuf= IMB_allocImBuf(diameter, diameter, 32, imbflag);
826         ibuf= cache->ibuf;
827
828         oldtexibuf= cache->texibuf;
829         cache->texibuf= IMB_allocImBuf(diameter, diameter, 32, imbflag);
830
831         if (oldtexibuf) {
832                 srcx= srcy= 0;
833                 destx= (int)painter->lastpaintpos[0] - (int)pos[0];
834                 desty= (int)painter->lastpaintpos[1] - (int)pos[1];
835                 w= oldtexibuf->x;
836                 h= oldtexibuf->y;
837
838                 IMB_rectclip(cache->texibuf, oldtexibuf, &destx, &desty, &srcx, &srcy, &w, &h);
839         }
840         else {
841                 srcx= srcy= 0;
842                 destx= desty= 0;
843                 w= h= 0;
844         }
845         
846         x1= destx;
847         y1= desty;
848         x2= destx+w;
849         y2= desty+h;
850
851         /* blend existing texture in new position */
852         if ((x1 < x2) && (y1 < y2))
853                 brush_painter_do_partial(painter, oldtexibuf, x1, y1, x2, y2, srcx, srcy, pos);
854
855         if (oldtexibuf)
856                 IMB_freeImBuf(oldtexibuf);
857
858         /* sample texture in new areas */
859         if ((0 < x1) && (0 < ibuf->y))
860                 brush_painter_do_partial(painter, NULL, 0, 0, x1, ibuf->y, 0, 0, pos);
861         if ((x2 < ibuf->x) && (0 < ibuf->y))
862                 brush_painter_do_partial(painter, NULL, x2, 0, ibuf->x, ibuf->y, 0, 0, pos);
863         if ((x1 < x2) && (0 < y1))
864                 brush_painter_do_partial(painter, NULL, x1, 0, x2, y1, 0, 0, pos);
865         if ((x1 < x2) && (y2 < ibuf->y))
866                 brush_painter_do_partial(painter, NULL, x1, y2, x2, ibuf->y, 0, 0, pos);
867 }
868
869 static void brush_painter_refresh_cache(BrushPainter *painter, float *pos, int use_color_correction)
870 {
871         Brush *brush= painter->brush;
872         BrushPainterCache *cache= &painter->cache;
873         MTex *mtex= &brush->mtex;
874         int size;
875         short flt;
876         const int diameter= 2*brush_size(brush);
877         const float alpha= brush_alpha(brush);
878
879         if (diameter != cache->lastsize ||
880                 alpha != cache->lastalpha ||
881                 brush->jitter != cache->lastjitter)
882         {
883                 if (cache->ibuf) {
884                         IMB_freeImBuf(cache->ibuf);
885                         cache->ibuf= NULL;
886                 }
887                 if (cache->maskibuf) {
888                         IMB_freeImBuf(cache->maskibuf);
889                         cache->maskibuf= NULL;
890                 }
891
892                 flt= cache->flt;
893                 size= (cache->size)? cache->size: diameter;
894
895                 if (brush->flag & BRUSH_FIXED_TEX) {
896                         brush_imbuf_new(brush, flt, 3, size, &cache->maskibuf, use_color_correction);
897                         brush_painter_fixed_tex_partial_update(painter, pos);
898                 }
899                 else
900                         brush_imbuf_new(brush, flt, 2, size, &cache->ibuf, use_color_correction);
901
902                 cache->lastsize= diameter;
903                 cache->lastalpha= alpha;
904                 cache->lastjitter= brush->jitter;
905         }
906         else if ((brush->flag & BRUSH_FIXED_TEX) && mtex && mtex->tex) {
907                 int dx = (int)painter->lastpaintpos[0] - (int)pos[0];
908                 int dy = (int)painter->lastpaintpos[1] - (int)pos[1];
909
910                 if ((dx != 0) || (dy != 0))
911                         brush_painter_fixed_tex_partial_update(painter, pos);
912         }
913 }
914
915 void brush_painter_break_stroke(BrushPainter *painter)
916 {
917         painter->firsttouch= 1;
918 }
919
920 static void brush_apply_pressure(BrushPainter *painter, Brush *brush, float pressure)
921 {
922         if (brush_use_alpha_pressure(painter->scene, brush)) 
923                 brush_set_alpha(brush, MAX2(0.0f, painter->startalpha*pressure));
924         if (brush_use_size_pressure(painter->scene, brush))
925                 brush_set_size(brush, MAX2(1.0f, painter->startsize*pressure));
926         if (brush->flag & BRUSH_JITTER_PRESSURE)
927                 brush->jitter = MAX2(0.0f, painter->startjitter*pressure);
928         if (brush->flag & BRUSH_SPACING_PRESSURE)
929                 brush->spacing = MAX2(1.0f, painter->startspacing*(1.5f-pressure));
930 }
931
932 void brush_jitter_pos(Brush *brush, float pos[2], float jitterpos[2])
933 {
934         int use_jitter= brush->jitter != 0;
935
936         /* jitter-ed brush gives weird and unpredictable result for this
937            kinds of stroke, so manyally disable jitter usage (sergey) */
938         use_jitter&= (brush->flag & (BRUSH_RESTORE_MESH|BRUSH_ANCHORED)) == 0;
939
940         if(use_jitter){
941                 float rand_pos[2];
942                 const int radius= brush_size(brush);
943                 const int diameter= 2*radius;
944
945                 // find random position within a circle of diameter 1
946                 do {
947                         rand_pos[0] = BLI_frand()-0.5f;
948                         rand_pos[1] = BLI_frand()-0.5f;
949                 } while (len_v2(rand_pos) > 0.5f);
950
951                 jitterpos[0] = pos[0] + 2*rand_pos[0]*diameter*brush->jitter;
952                 jitterpos[1] = pos[1] + 2*rand_pos[1]*diameter*brush->jitter;
953         }
954         else {
955                 copy_v2_v2(jitterpos, pos);
956         }
957 }
958
959 int brush_painter_paint(BrushPainter *painter, BrushFunc func, float *pos, double time, float pressure, void *user, int use_color_correction)
960 {
961         Brush *brush= painter->brush;
962         int totpaintops= 0;
963
964         if (pressure == 0.0f) {
965                 if(painter->lastpressure) // XXX - hack, operator misses
966                         pressure= painter->lastpressure;
967                 else
968                         pressure = 1.0f;        /* zero pressure == not using tablet */
969         }
970         if (painter->firsttouch) {
971                 /* paint exactly once on first touch */
972                 painter->startpaintpos[0]= pos[0];
973                 painter->startpaintpos[1]= pos[1];
974
975                 brush_apply_pressure(painter, brush, pressure);
976                 if (painter->cache.enabled)
977                         brush_painter_refresh_cache(painter, pos, use_color_correction);
978                 totpaintops += func(user, painter->cache.ibuf, pos, pos);
979                 
980                 painter->lasttime= time;
981                 painter->firsttouch= 0;
982                 painter->lastpaintpos[0]= pos[0];
983                 painter->lastpaintpos[1]= pos[1];
984         }
985 #if 0
986         else if (painter->brush->flag & BRUSH_AIRBRUSH) {
987                 float spacing, step, paintpos[2], dmousepos[2], len;
988                 double starttime, curtime= time;
989
990                 /* compute brush spacing adapted to brush size */
991                 spacing= brush->rate; //radius*brush->spacing*0.01f;
992
993                 /* setup starting time, direction vector and accumulated time */
994                 starttime= painter->accumtime;
995                 sub_v2_v2v2(dmousepos, pos, painter->lastmousepos);
996                 len= normalize_v2(dmousepos);
997                 painter->accumtime += curtime - painter->lasttime;
998
999                 /* do paint op over unpainted time distance */
1000                 while (painter->accumtime >= spacing) {
1001                         step= (spacing - starttime)*len;
1002                         paintpos[0]= painter->lastmousepos[0] + dmousepos[0]*step;
1003                         paintpos[1]= painter->lastmousepos[1] + dmousepos[1]*step;
1004
1005                         if (painter->cache.enabled)
1006                                 brush_painter_refresh_cache(painter);
1007                         totpaintops += func(user, painter->cache.ibuf,
1008                                 painter->lastpaintpos, paintpos);
1009
1010                         painter->lastpaintpos[0]= paintpos[0];
1011                         painter->lastpaintpos[1]= paintpos[1];
1012                         painter->accumtime -= spacing;
1013                         starttime -= spacing;
1014                 }
1015                 
1016                 painter->lasttime= curtime;
1017         }
1018 #endif
1019         else {
1020                 float startdistance, spacing, step, paintpos[2], dmousepos[2], finalpos[2];
1021                 float t, len, press;
1022                 const int radius= brush_size(brush);
1023
1024                 /* compute brush spacing adapted to brush radius, spacing may depend
1025                    on pressure, so update it */
1026                 brush_apply_pressure(painter, brush, painter->lastpressure);
1027                 spacing= MAX2(1.0f, radius)*brush->spacing*0.01f;
1028
1029                 /* setup starting distance, direction vector and accumulated distance */
1030                 startdistance= painter->accumdistance;
1031                 sub_v2_v2v2(dmousepos, pos, painter->lastmousepos);
1032                 len= normalize_v2(dmousepos);
1033                 painter->accumdistance += len;
1034
1035                 if (brush->flag & BRUSH_SPACE) {
1036                         /* do paint op over unpainted distance */
1037                         while ((len > 0.0f) && (painter->accumdistance >= spacing)) {
1038                                 step= spacing - startdistance;
1039                                 paintpos[0]= painter->lastmousepos[0] + dmousepos[0]*step;
1040                                 paintpos[1]= painter->lastmousepos[1] + dmousepos[1]*step;
1041
1042                                 t = step/len;
1043                                 press= (1.0f-t)*painter->lastpressure + t*pressure;
1044                                 brush_apply_pressure(painter, brush, press);
1045                                 spacing= MAX2(1.0f, radius)*brush->spacing*0.01f;
1046
1047                                 brush_jitter_pos(brush, paintpos, finalpos);
1048
1049                                 if (painter->cache.enabled)
1050                                         brush_painter_refresh_cache(painter, finalpos, use_color_correction);
1051
1052                                 totpaintops +=
1053                                         func(user, painter->cache.ibuf, painter->lastpaintpos, finalpos);
1054
1055                                 painter->lastpaintpos[0]= paintpos[0];
1056                                 painter->lastpaintpos[1]= paintpos[1];
1057                                 painter->accumdistance -= spacing;
1058                                 startdistance -= spacing;
1059                         }
1060                 } else {
1061                         brush_jitter_pos(brush, pos, finalpos);
1062
1063                         if (painter->cache.enabled)
1064                                 brush_painter_refresh_cache(painter, finalpos, use_color_correction);
1065
1066                         totpaintops += func(user, painter->cache.ibuf, pos, finalpos);
1067
1068                         painter->lastpaintpos[0]= pos[0];
1069                         painter->lastpaintpos[1]= pos[1];
1070                         painter->accumdistance= 0;
1071                 }
1072
1073                 /* do airbrush paint ops, based on the number of paint ops left over
1074                    from regular painting. this is a temporary solution until we have
1075                    accurate time stamps for mouse move events */
1076                 if (brush->flag & BRUSH_AIRBRUSH) {
1077                         double curtime= time;
1078                         double painttime= brush->rate*totpaintops;
1079
1080                         painter->accumtime += curtime - painter->lasttime;
1081                         if (painter->accumtime <= painttime)
1082                                 painter->accumtime= 0.0;
1083                         else
1084                                 painter->accumtime -= painttime;
1085
1086                         while (painter->accumtime >= (double)brush->rate) {
1087                                 brush_apply_pressure(painter, brush, pressure);
1088
1089                                 brush_jitter_pos(brush, pos, finalpos);
1090
1091                                 if (painter->cache.enabled)
1092                                         brush_painter_refresh_cache(painter, finalpos, use_color_correction);
1093
1094                                 totpaintops +=
1095                                         func(user, painter->cache.ibuf, painter->lastmousepos, finalpos);
1096                                 painter->accumtime -= (double)brush->rate;
1097                         }
1098
1099                         painter->lasttime= curtime;
1100                 }
1101         }
1102
1103         painter->lastmousepos[0]= pos[0];
1104         painter->lastmousepos[1]= pos[1];
1105         painter->lastpressure= pressure;
1106
1107         brush_set_alpha(brush, painter->startalpha);
1108         brush_set_size(brush, painter->startsize);
1109         brush->jitter = painter->startjitter;
1110         brush->spacing = painter->startspacing;
1111
1112         return totpaintops;
1113 }
1114
1115 /* Uses the brush curve control to find a strength value between 0 and 1 */
1116 float brush_curve_strength_clamp(Brush *br, float p, const float len)
1117 {
1118         if(p >= len)    return 0;
1119         else                    p= p/len;
1120
1121         p= curvemapping_evaluateF(br->curve, 0, p);
1122         if(p < 0.0f)            p= 0.0f;
1123         else if(p > 1.0f)       p= 1.0f;
1124         return p;
1125 }
1126 /* same as above but can return negative values if the curve enables
1127  * used for sculpt only */
1128 float brush_curve_strength(Brush *br, float p, const float len)
1129 {
1130         if(p >= len)
1131                 p= 1.0f;
1132         else
1133                 p= p/len;
1134
1135         return curvemapping_evaluateF(br->curve, 0, p);
1136 }
1137
1138 /* TODO: should probably be unified with BrushPainter stuff? */
1139 unsigned int *brush_gen_texture_cache(Brush *br, int half_side)
1140 {
1141         unsigned int *texcache = NULL;
1142         MTex *mtex = &br->mtex;
1143         TexResult texres= {0};
1144         int hasrgb, ix, iy;
1145         int side = half_side * 2;
1146         
1147         if(mtex->tex) {
1148                 float x, y, step = 2.0 / side, co[3];
1149
1150                 texcache = MEM_callocN(sizeof(int) * side * side, "Brush texture cache");
1151
1152                 BKE_image_get_ibuf(mtex->tex->ima, NULL);
1153                 
1154                 /*do normalized cannonical view coords for texture*/
1155                 for (y=-1.0, iy=0; iy<side; iy++, y += step) {
1156                         for (x=-1.0, ix=0; ix<side; ix++, x += step) {
1157                                 co[0]= x;
1158                                 co[1]= y;
1159                                 co[2]= 0.0f;
1160                                 
1161                                 /* This is copied from displace modifier code */
1162                                 hasrgb = multitex_ext(mtex->tex, co, NULL, NULL, 0, &texres);
1163                         
1164                                 /* if the texture gave an RGB value, we assume it didn't give a valid
1165                                  * intensity, so calculate one (formula from do_material_tex).
1166                                  * if the texture didn't give an RGB value, copy the intensity across
1167                                  */
1168                                 if(hasrgb & TEX_RGB)
1169                                         texres.tin = (0.35f * texres.tr + 0.45f *
1170                                                                   texres.tg + 0.2f * texres.tb);
1171
1172                                 texres.tin = texres.tin * 255.0f;
1173                                 ((char*)texcache)[(iy*side+ix)*4] = (char)texres.tin;
1174                                 ((char*)texcache)[(iy*side+ix)*4+1] = (char)texres.tin;
1175                                 ((char*)texcache)[(iy*side+ix)*4+2] = (char)texres.tin;
1176                                 ((char*)texcache)[(iy*side+ix)*4+3] = (char)texres.tin;
1177                         }
1178                 }
1179         }
1180
1181         return texcache;
1182 }
1183
1184 /**** Radial Control ****/
1185 struct ImBuf *brush_gen_radial_control_imbuf(Brush *br)
1186 {
1187         ImBuf *im = MEM_callocN(sizeof(ImBuf), "radial control texture");
1188         unsigned int *texcache;
1189         int side = 128;
1190         int half = side / 2;
1191         int i, j;
1192
1193         texcache = brush_gen_texture_cache(br, half);
1194         im->rect_float = MEM_callocN(sizeof(float) * side * side, "radial control rect");
1195         im->x = im->y = side;
1196
1197         for(i=0; i<side; ++i) {
1198                 for(j=0; j<side; ++j) {
1199                         float magn= sqrt(pow(i - half, 2) + pow(j - half, 2));
1200                         im->rect_float[i*side + j]= brush_curve_strength_clamp(br, magn, half);
1201                 }
1202         }
1203
1204         /* Modulate curve with texture */
1205         if(texcache) {
1206                 for(i=0; i<side; ++i) {
1207                         for(j=0; j<side; ++j) {
1208                                 const int col= texcache[i*side+j];
1209                                 im->rect_float[i*side+j]*= (((char*)&col)[0]+((char*)&col)[1]+((char*)&col)[2])/3.0f/255.0f;
1210                         }
1211                 }
1212
1213                 MEM_freeN(texcache);
1214         }
1215
1216         return im;
1217 }
1218
1219 /* Unified Size and Strength */
1220
1221 /* XXX, wouldnt it be better to only pass the active scene?
1222  * this can return any old scene! - campbell*/
1223
1224 static short unified_settings(Brush *brush)
1225 {
1226         Scene *sce;
1227         for (sce= G.main->scene.first; sce; sce= sce->id.next) {
1228                 if (sce->toolsettings && 
1229                         ELEM4(brush,
1230                             paint_brush(&(sce->toolsettings->imapaint.paint)),
1231                             paint_brush(&(sce->toolsettings->vpaint->paint)),
1232                             paint_brush(&(sce->toolsettings->wpaint->paint)),
1233                             paint_brush(&(sce->toolsettings->sculpt->paint))))
1234                 {
1235                         return sce->toolsettings->unified_paint_settings.flag;
1236                 }
1237         }
1238
1239         return 0;
1240 }
1241
1242 // XXX: be careful about setting size and unprojected radius
1243 // because they depend on one another
1244 // these functions do not set the other corresponding value
1245 // this can lead to odd behavior if size and unprojected
1246 // radius become inconsistent.
1247 // the biggest problem is that it isn't possible to change
1248 // unprojected radius because a view context is not
1249 // available.  my ussual solution to this is to use the
1250 // ratio of change of the size to change the unprojected
1251 // radius.  Not completely convinced that is correct.
1252 // In anycase, a better solution is needed to prevent
1253 // inconsistency.
1254
1255 static void set_unified_size(Brush *brush, int value)
1256 {
1257         Scene *sce;
1258         for (sce= G.main->scene.first; sce; sce= sce->id.next) {
1259                 if (sce->toolsettings && 
1260                         ELEM4(brush,
1261                             paint_brush(&(sce->toolsettings->imapaint.paint)),
1262                             paint_brush(&(sce->toolsettings->vpaint->paint)),
1263                             paint_brush(&(sce->toolsettings->wpaint->paint)),
1264                             paint_brush(&(sce->toolsettings->sculpt->paint))))
1265                 {
1266                         sce->toolsettings->unified_paint_settings.size= value;
1267                 }
1268         }
1269 }
1270
1271 static int unified_size(Brush *brush)
1272 {
1273         Scene *sce;
1274         for (sce= G.main->scene.first; sce; sce= sce->id.next) {
1275                 if (sce->toolsettings && 
1276                         ELEM4(brush,
1277                             paint_brush(&(sce->toolsettings->imapaint.paint)),
1278                             paint_brush(&(sce->toolsettings->vpaint->paint)),
1279                             paint_brush(&(sce->toolsettings->wpaint->paint)),
1280                             paint_brush(&(sce->toolsettings->sculpt->paint))))
1281                 {
1282                         return sce->toolsettings->unified_paint_settings.size;
1283                 }
1284         }
1285
1286         return 35; // XXX magic number
1287 }
1288
1289 static void set_unified_alpha(Brush *brush, float value)
1290 {
1291         Scene *sce;
1292         for (sce= G.main->scene.first; sce; sce= sce->id.next) {
1293                 if (sce->toolsettings && 
1294                         ELEM4(brush,
1295                             paint_brush(&(sce->toolsettings->imapaint.paint)),
1296                             paint_brush(&(sce->toolsettings->vpaint->paint)),
1297                             paint_brush(&(sce->toolsettings->wpaint->paint)),
1298                             paint_brush(&(sce->toolsettings->sculpt->paint))))
1299                 {
1300                         sce->toolsettings->unified_paint_settings.alpha= value;
1301                 }
1302         }
1303 }
1304
1305 static float unified_alpha(Brush *brush)
1306 {
1307         Scene *sce;
1308         for (sce= G.main->scene.first; sce; sce= sce->id.next) {
1309                 if (sce->toolsettings && 
1310                         ELEM4(brush,
1311                             paint_brush(&(sce->toolsettings->imapaint.paint)),
1312                             paint_brush(&(sce->toolsettings->vpaint->paint)),
1313                             paint_brush(&(sce->toolsettings->wpaint->paint)),
1314                             paint_brush(&(sce->toolsettings->sculpt->paint))))
1315                 {
1316                         return sce->toolsettings->unified_paint_settings.alpha;
1317                 }
1318         }
1319
1320         return 0.5f; // XXX magic number
1321 }
1322
1323 static void set_unified_unprojected_radius(Brush *brush, float value)
1324 {
1325         Scene *sce;
1326         for (sce= G.main->scene.first; sce; sce= sce->id.next) {
1327                 if (sce->toolsettings && 
1328                         ELEM4(brush,
1329                             paint_brush(&(sce->toolsettings->imapaint.paint)),
1330                             paint_brush(&(sce->toolsettings->vpaint->paint)),
1331                             paint_brush(&(sce->toolsettings->wpaint->paint)),
1332                             paint_brush(&(sce->toolsettings->sculpt->paint))))
1333                 {
1334                         sce->toolsettings->unified_paint_settings.unprojected_radius= value;
1335                 }
1336         }
1337 }
1338
1339 static float unified_unprojected_radius(Brush *brush)
1340 {
1341         Scene *sce;
1342         for (sce= G.main->scene.first; sce; sce= sce->id.next) {
1343                 if (sce->toolsettings && 
1344                         ELEM4(brush,
1345                             paint_brush(&(sce->toolsettings->imapaint.paint)),
1346                             paint_brush(&(sce->toolsettings->vpaint->paint)),
1347                             paint_brush(&(sce->toolsettings->wpaint->paint)),
1348                             paint_brush(&(sce->toolsettings->sculpt->paint))))
1349                 {
1350                         return sce->toolsettings->unified_paint_settings.unprojected_radius;
1351                 }
1352         }
1353
1354         return 0.125f; // XXX magic number
1355 }
1356 void brush_set_size(Brush *brush, int size)
1357 {
1358         const short us_flag = unified_settings(brush);
1359
1360         if (us_flag & UNIFIED_PAINT_SIZE)
1361                 set_unified_size(brush, size);
1362         else
1363                 brush->size= size;
1364
1365         //WM_main_add_notifier(NC_BRUSH|NA_EDITED, brush);
1366 }
1367
1368 int brush_size(Brush *brush)
1369 {
1370         const short us_flag = unified_settings(brush);
1371
1372         return (us_flag & UNIFIED_PAINT_SIZE) ? unified_size(brush) : brush->size;
1373 }
1374
1375 int brush_use_locked_size(const Scene *scene, Brush *brush)
1376 {
1377         const short us_flag = scene->toolsettings->unified_paint_settings.flag;
1378
1379         return (us_flag & UNIFIED_PAINT_SIZE) ?
1380                 (us_flag & UNIFIED_PAINT_BRUSH_LOCK_SIZE) :
1381                 (brush->flag & BRUSH_LOCK_SIZE);
1382 }
1383
1384 int brush_use_size_pressure(const Scene *scene, Brush *brush)
1385 {
1386         const short us_flag = scene->toolsettings->unified_paint_settings.flag;
1387
1388         return (us_flag & UNIFIED_PAINT_SIZE) ?
1389                 (us_flag & UNIFIED_PAINT_BRUSH_SIZE_PRESSURE) :
1390                 (brush->flag & BRUSH_SIZE_PRESSURE);
1391 }
1392
1393 int brush_use_alpha_pressure(const Scene *scene, Brush *brush)
1394 {
1395         const short us_flag = scene->toolsettings->unified_paint_settings.flag;
1396
1397         return (us_flag & UNIFIED_PAINT_ALPHA) ?
1398                 (us_flag & UNIFIED_PAINT_BRUSH_ALPHA_PRESSURE) :
1399                 (brush->flag & BRUSH_ALPHA_PRESSURE);
1400 }
1401
1402 void brush_set_unprojected_radius(Brush *brush, float unprojected_radius)
1403 {
1404         const short us_flag = unified_settings(brush);
1405
1406         if (us_flag & UNIFIED_PAINT_SIZE)
1407                 set_unified_unprojected_radius(brush, unprojected_radius);
1408         else
1409                 brush->unprojected_radius= unprojected_radius;
1410
1411         //WM_main_add_notifier(NC_BRUSH|NA_EDITED, brush);
1412 }
1413
1414 float brush_unprojected_radius(Brush *brush)
1415 {
1416         const short us_flag = unified_settings(brush);
1417
1418         return (us_flag & UNIFIED_PAINT_SIZE) ?
1419                 unified_unprojected_radius(brush) :
1420                 brush->unprojected_radius;
1421 }
1422
1423 void brush_set_alpha(Brush *brush, float alpha)
1424 {
1425         const short us_flag = unified_settings(brush);
1426
1427         if (us_flag & UNIFIED_PAINT_ALPHA)
1428                 set_unified_alpha(brush, alpha);
1429         else
1430                 brush->alpha= alpha;
1431
1432         //WM_main_add_notifier(NC_BRUSH|NA_EDITED, brush);
1433 }
1434
1435 float brush_alpha(Brush *brush)
1436 {
1437         const short us_flag = unified_settings(brush);
1438
1439         return (us_flag & UNIFIED_PAINT_ALPHA) ?
1440                 unified_alpha(brush) :
1441                 brush->alpha;
1442 }