Revision: 18664
          
http://projects.blender.org/plugins/scmsvn/viewcvs.php?view=rev&root=bf-blender&revision=18664
Author:   nicholasbishop
Date:     2009-01-25 08:28:11 +0100 (Sun, 25 Jan 2009)

Log Message:
-----------
Did some refactoring with brush and sculpt to make radial control easier to add 
to other modes.

Modified Paths:
--------------
    branches/blender2.5/blender/source/blender/blenkernel/BKE_brush.h
    branches/blender2.5/blender/source/blender/blenkernel/intern/brush.c
    branches/blender2.5/blender/source/blender/editors/sculpt/sculpt.c
    
branches/blender2.5/blender/source/blender/makesdna/DNA_windowmanager_types.h
    branches/blender2.5/blender/source/blender/windowmanager/WM_types.h

Modified: branches/blender2.5/blender/source/blender/blenkernel/BKE_brush.h
===================================================================
--- branches/blender2.5/blender/source/blender/blenkernel/BKE_brush.h   
2009-01-25 04:02:31 UTC (rev 18663)
+++ branches/blender2.5/blender/source/blender/blenkernel/BKE_brush.h   
2009-01-25 07:28:11 UTC (rev 18664)
@@ -34,6 +34,7 @@
 struct ID;
 struct Brush;
 struct ImBuf;
+struct wmOperator;
 
 /* datablock functions */
 struct Brush *add_brush(char *name);
@@ -58,6 +59,7 @@
        BRUSH_PRESET_MAX
 } BrushCurvePreset;
 void brush_curve_preset(struct Brush *b, BrushCurvePreset preset);
+float brush_curve_strength(struct Brush *br, float p, const float len);
 
 /* sampling */
 float brush_sample_falloff(struct Brush *brush, float dist);
@@ -79,5 +81,12 @@
 void brush_painter_break_stroke(BrushPainter *painter);
 void brush_painter_free(BrushPainter *painter);
 
+/* texture */
+unsigned int *brush_gen_texture_cache(struct Brush *br, int half_side);
+
+/* radial control */
+void brush_radial_control_invoke(struct wmOperator *op, struct Brush *br);
+int brush_radial_control_exec(struct wmOperator *op, struct Brush *br);
+
 #endif
 

Modified: branches/blender2.5/blender/source/blender/blenkernel/intern/brush.c
===================================================================
--- branches/blender2.5/blender/source/blender/blenkernel/intern/brush.c        
2009-01-25 04:02:31 UTC (rev 18663)
+++ branches/blender2.5/blender/source/blender/blenkernel/intern/brush.c        
2009-01-25 07:28:11 UTC (rev 18664)
@@ -28,6 +28,7 @@
  */
 
 #include <math.h>
+#include <string.h>
 
 #include "MEM_guardedalloc.h"
 
@@ -36,13 +37,17 @@
 #include "DNA_image_types.h"
 #include "DNA_texture_types.h"
 #include "DNA_scene_types.h"
+#include "DNA_windowmanager_types.h"
 
+#include "RNA_access.h"
+
 #include "BLI_arithb.h"
 #include "BLI_blenlib.h"
 
 #include "BKE_brush.h"
 #include "BKE_colortools.h"
 #include "BKE_global.h"
+#include "BKE_image.h"
 #include "BKE_library.h"
 #include "BKE_main.h"
 #include "BKE_texture.h"
@@ -52,6 +57,7 @@
 #include "IMB_imbuf_types.h"
 
 #include "RE_render_ext.h" /* externtex */
+#include "RE_shader_ext.h"
 
 /* Datablock add/copy/free/make_local */
 
@@ -935,4 +941,123 @@
        return totpaintops;
 }
 
+/* Uses the brush curve control to find a strength value between 0 and 1 */
+float brush_curve_strength(Brush *br, float p, const float len)
+{
+       if(p > len) p= len;
+       return curvemapping_evaluateF(br->curve, 0, p/len);
+}
 
+/* TODO: should probably be unified with BrushPainter stuff? */
+unsigned int *brush_gen_texture_cache(Brush *br, int half_side)
+{
+       unsigned int *texcache = NULL;
+       MTex *mtex = br->mtex[br->texact];
+       TexResult texres;
+       int hasrgb, ix, iy;
+       int side = half_side * 2;
+
+       memset(&texres, 0, sizeof(TexResult));
+       
+       if(mtex && mtex->tex) {
+               float x, y, step = 2.0 / side, co[3];
+
+               texcache = MEM_callocN(sizeof(int) * side * side, "Brush 
texture cache");
+
+               BKE_image_get_ibuf(mtex->tex->ima, NULL);
+               
+               /*do normalized cannonical view coords for texture*/
+               for (y=-1.0, iy=0; iy<side; iy++, y += step) {
+                       for (x=-1.0, ix=0; ix<side; ix++, x += step) {
+                               co[0]= x;
+                               co[1]= y;
+                               co[2]= 0.0f;
+                               
+                               /* This is copied from displace modifier code */
+                               hasrgb = multitex_ext(mtex->tex, co, NULL, 
NULL, 1, &texres);
+                       
+                               /* if the texture gave an RGB value, we assume 
it didn't give a valid
+                                * intensity, so calculate one (formula from 
do_material_tex).
+                                * if the texture didn't give an RGB value, 
copy the intensity across
+                                */
+                               if(hasrgb & TEX_RGB)
+                                       texres.tin = (0.35 * texres.tr + 0.45 *
+                                                     texres.tg + 0.2 * 
texres.tb);
+
+                               texres.tin = texres.tin * 255.0;
+                               ((char*)texcache)[(iy*side+ix)*4] = 
(char)texres.tin;
+                               ((char*)texcache)[(iy*side+ix)*4+1] = 
(char)texres.tin;
+                               ((char*)texcache)[(iy*side+ix)*4+2] = 
(char)texres.tin;
+                               ((char*)texcache)[(iy*side+ix)*4+3] = 
(char)texres.tin;
+                       }
+               }
+       }
+
+       return texcache;
+}
+
+/**** Radial Control ****/
+static struct ImBuf *brush_gen_radial_control_imbuf(Brush *br)
+{
+       ImBuf *im = MEM_callocN(sizeof(ImBuf), "radial control texture");
+       unsigned int *texcache;
+       int side = 128;
+       int half = side / 2;
+       int i, j;
+
+       texcache = brush_gen_texture_cache(br, half);
+       im->rect_float = MEM_callocN(sizeof(float) * side * side, "radial 
control rect");
+       im->x = im->y = side;
+
+       for(i=0; i<side; ++i) {
+               for(j=0; j<side; ++j) {
+                       float magn= sqrt(pow(i - half, 2) + pow(j - half, 2));
+                       im->rect_float[i*side + j]= brush_curve_strength(br, 
magn, half);
+               }
+       }
+
+       /* Modulate curve with texture */
+       if(texcache) {
+               for(i=0; i<side; ++i)
+                       for(j=0; j<side; ++j) {
+                               const int col= texcache[i*side+j];
+                               im->rect_float[i*side+j]*= 
(((char*)&col)[0]+((char*)&col)[1]+((char*)&col)[2])/3.0f/255.0f;
+                       }
+       }
+
+       MEM_freeN(texcache);
+
+       return im;
+}
+
+void brush_radial_control_invoke(wmOperator *op, Brush *br)
+{
+       int mode = RNA_int_get(op->ptr, "mode");
+       float original_value;
+
+       if(mode == WM_RADIALCONTROL_SIZE)
+               original_value = br->size;
+       else if(mode == WM_RADIALCONTROL_STRENGTH)
+               original_value = br->alpha;
+       else if(mode == WM_RADIALCONTROL_ANGLE)
+               original_value = br->rot;
+
+       RNA_float_set(op->ptr, "initial_value", original_value);
+       op->customdata = brush_gen_radial_control_imbuf(br);
+}
+
+int brush_radial_control_exec(wmOperator *op, Brush *br)
+{
+       int mode = RNA_int_get(op->ptr, "mode");
+       float new_value = RNA_float_get(op->ptr, "new_value");
+       const float conv = 0.017453293;
+
+       if(mode == WM_RADIALCONTROL_SIZE)
+               br->size = new_value;
+       else if(mode == WM_RADIALCONTROL_STRENGTH)
+               br->alpha = new_value;
+       else if(mode == WM_RADIALCONTROL_ANGLE)
+               br->rot = new_value * conv;
+
+       return OPERATOR_FINISHED;
+}

Modified: branches/blender2.5/blender/source/blender/editors/sculpt/sculpt.c
===================================================================
--- branches/blender2.5/blender/source/blender/editors/sculpt/sculpt.c  
2009-01-25 04:02:31 UTC (rev 18663)
+++ branches/blender2.5/blender/source/blender/editors/sculpt/sculpt.c  
2009-01-25 07:28:11 UTC (rev 18664)
@@ -585,12 +585,6 @@
        }
 }
 
-/* Uses the brush curve control to find a strength value between 0 and 1 */
-static float curve_strength(CurveMapping *cumap, float p, const float len)
-{
-       if(p > len) p= len;
-       return curvemapping_evaluateF(cumap, 0, p/len);
-}
 
 /* Uses symm to selectively flip any axis of a coordinate. */
 static void flip_coord(float out[3], float in[3], const char symm)
@@ -724,7 +718,7 @@
                }
        }
 
-       avg*= curve_strength(sd->brush->curve, len, ss->cache->radius); /* 
Falloff curve */
+       avg*= brush_curve_strength(sd->brush, len, ss->cache->radius); /* 
Falloff curve */
 
        return avg;
 }
@@ -990,62 +984,21 @@
                ss->projverts[i].inside = 0;
 }
 
-static void sculpt_update_tex(Sculpt *sd, int half_size)
+static void sculpt_update_tex(Sculpt *sd)
 {
        SculptSession *ss= sd->session;
-       Brush *br = sd->brush;
-       MTex *mtex = br->mtex[br->texact];
-       TexResult texres;
-       int hasrgb, ix, iy;
 
-       memset(&texres, 0, sizeof(TexResult));
-       
-       /* Skip Default brush shape and non-textures */
-       if(br->texact == -1 || !br->mtex[br->texact]) return;
-
        if(ss->texcache) {
                MEM_freeN(ss->texcache);
                ss->texcache= NULL;
        }
 
        /* Need to allocate a bigger buffer for bigger brush size */
-       ss->texcache_side = half_size * 2;
+       ss->texcache_side = sd->brush->size * 2;
        if(!ss->texcache || ss->texcache_side > ss->texcache_actual) {
-               ss->texcache = MEM_callocN(sizeof(int) * ss->texcache_side * 
ss->texcache_side, "Sculpt Texture cache");
+               ss->texcache = brush_gen_texture_cache(sd->brush, 
sd->brush->size);
                ss->texcache_actual = ss->texcache_side;
        }
-
-       if(mtex && mtex->tex) {
-               float x, y, step = 2.0 / ss->texcache_side, co[3];
-
-               BKE_image_get_ibuf(br->mtex[br->texact]->tex->ima, NULL);
-               
-               /*do normalized cannonical view coords for texture*/
-               for (y=-1.0, iy=0; iy<ss->texcache_side; iy++, y += step) {
-                       for (x=-1.0, ix=0; ix<ss->texcache_side; ix++, x += 
step) {
-                               co[0]= x;
-                               co[1]= y;
-                               co[2]= 0.0f;
-                               
-                               /* This is copied from displace modifier code */
-                               hasrgb = multitex_ext(mtex->tex, co, NULL, 
NULL, 1, &texres);
-                       
-                               /* if the texture gave an RGB value, we assume 
it didn't give a valid
-                                * intensity, so calculate one (formula from 
do_material_tex).
-                                * if the texture didn't give an RGB value, 
copy the intensity across
-                                */
-                               if(hasrgb & TEX_RGB)
-                                       texres.tin = (0.35 * texres.tr + 0.45 *
-                                                     texres.tg + 0.2 * 
texres.tb);
-
-                               texres.tin = texres.tin * 255.0;
-                               
((char*)ss->texcache)[(iy*ss->texcache_side+ix)*4] = (char)texres.tin;
-                               
((char*)ss->texcache)[(iy*ss->texcache_side+ix)*4+1] = (char)texres.tin;
-                               
((char*)ss->texcache)[(iy*ss->texcache_side+ix)*4+2] = (char)texres.tin;
-                               
((char*)ss->texcache)[(iy*ss->texcache_side+ix)*4+3] = (char)texres.tin;
-                       }
-               }
-       }
 }
 
 void sculptmode_selectbrush_menu(void)
@@ -1284,57 +1237,11 @@
        }
 }
 
-static ImBuf *sculpt_radial_control_texture(bContext *C)
-{
-       Sculpt *s = CTX_data_scene(C)->toolsettings->sculpt;
-       ImBuf *im = MEM_callocN(sizeof(ImBuf), "radial control texture");
-       unsigned int *texcache;
-       int side = 256;
-       int half = side / 2;
-       int i, j;
-
-       sculpt_update_tex(s, half);
-       texcache = s->session->texcache;
-       im->rect_float = MEM_callocN(sizeof(float) * side * side, "radial 
control rect");
-       im->x = im->y = side;
-
-       for(i=0; i<side; ++i) {
-               for(j=0; j<side; ++j) {
-                       float magn= sqrt(pow(i - half, 2) + pow(j - half, 2));
-                       im->rect_float[i*side + j]= 
curve_strength(s->brush->curve, magn, half);
-               }
-       }
-
-       /* Modulate curve with texture */
-       if(texcache) {
-               for(i=0; i<side; ++i)
-                       for(j=0; j<side; ++j) {
-                               const int col= texcache[i*side+j];
-                               im->rect_float[i*side+j]*= 
(((char*)&col)[0]+((char*)&col)[1]+((char*)&col)[2])/3.0f/255.0f;
-                       }
-       }
-
-       return im;
-}
-
+/**** Radial control ****/
 static int sculpt_radial_control_invoke(bContext *C, wmOperator *op, wmEvent 
*event)
 {
-       Brush *br = CTX_data_scene(C)->toolsettings->sculpt->brush;
-       int mode = RNA_int_get(op->ptr, "mode");
-       float original_value;
-
-       if(mode == WM_RADIALCONTROL_SIZE)
-               original_value = br->size;
-       else if(mode == WM_RADIALCONTROL_STRENGTH)
-               original_value = br->alpha;
-       else if(mode == WM_RADIALCONTROL_ANGLE)
-               original_value = br->rot;
-
        toggle_paint_cursor(C);
-
-       RNA_float_set(op->ptr, "initial_value", original_value);

@@ Diff output truncated at 10240 characters. @@

_______________________________________________
Bf-blender-cvs mailing list
[email protected]
http://lists.blender.org/mailman/listinfo/bf-blender-cvs

Reply via email to