2.5 Texture paint:
[blender.git] / source / blender / blenkernel / intern / paint.c
index e945c218217a3511bf15fadd3bc1671aa3205110..5ae8411835609a0d7e8bdde12d0c75b534cb897c 100644 (file)
  * ***** END GPL LICENSE BLOCK *****
  */ 
 
+#include "MEM_guardedalloc.h"
+
+#include "DNA_brush_types.h"
 #include "DNA_object_types.h"
+#include "DNA_scene_types.h"
 
+#include "BKE_brush.h"
 #include "BKE_global.h"
 #include "BKE_paint.h"
 
+#include <stdlib.h>
+#include <string.h>
+
+Paint *paint_get_active(Scene *sce)
+{
+       if(sce && sce->basact && sce->basact->object) {
+               ToolSettings *ts = sce->toolsettings;
+
+               switch(sce->basact->object->mode) {
+               case OB_MODE_SCULPT:
+                       return &ts->sculpt->paint;
+               case OB_MODE_VERTEX_PAINT:
+                       return &ts->vpaint->paint;
+               case OB_MODE_WEIGHT_PAINT:
+                       return &ts->wpaint->paint;
+               case OB_MODE_TEXTURE_PAINT:
+                       break;
+                       //return &ts->imapaint->paint;
+               }
+       }
+
+       return NULL;
+}
+
+Brush *paint_brush(Paint *p)
+{
+       return p && p->brushes ? p->brushes[p->active_brush_index] : NULL;
+}
+
+void paint_brush_set(Paint *p, Brush *br)
+{
+       if(p && p->brushes) {
+               int i;
+
+               /* See if there's already a slot with the brush */
+               for(i = 0; i < p->brush_count; ++i) {
+                       if(p->brushes[i] == br) {
+                               p->active_brush_index = i;
+                               break;
+                       }
+               }
+               
+       }
+       else
+               paint_brush_slot_add(p);
+
+       /* Make sure the current slot is the new brush */
+       p->brushes[p->active_brush_index] = br;
+}
+
+static void paint_brush_slots_alloc(Paint *p, const int count)
+{
+       p->brush_count = count;
+       if(count == 0)
+               p->brushes = NULL;
+       else
+               p->brushes = MEM_callocN(sizeof(Brush*) * count, "Brush slots");
+}
+
+void paint_brush_slot_add(Paint *p)
+{
+       Brush **orig = p->brushes;
+       int orig_count = p->brushes ? p->brush_count : 0;
+
+       /* Increase size of brush slot array */
+       paint_brush_slots_alloc(p, orig_count + 1);
+       if(orig) {
+               memcpy(p->brushes, orig, sizeof(Brush*) * orig_count);
+               MEM_freeN(orig);
+       }
+
+       p->active_brush_index = orig_count;
+}
+
+void paint_brush_slot_remove(Paint *p)
+{
+       if(p->brushes) {
+               Brush **orig = p->brushes;
+               int src, dst;
+               
+               /* Decrease size of brush slot array */
+               paint_brush_slots_alloc(p, p->brush_count - 1);
+               if(p->brushes) {
+                       for(src = 0, dst = 0; dst < p->brush_count; ++src) {
+                               if(src != p->active_brush_index) {
+                                       p->brushes[dst] = orig[src];
+                                       ++dst;
+                               }
+                       }
+               }
+               MEM_freeN(orig);
+
+               if(p->active_brush_index >= p->brush_count)
+                       p->active_brush_index = p->brush_count - 1;
+               if(p->active_brush_index < 0)
+                       p->active_brush_index = 0;
+       }
+}
+
 int paint_facesel_test(Object *ob)
 {
-       return (G.f&G_FACESELECT) && ((G.f & (G_WEIGHTPAINT|G_TEXTUREPAINT)) || (ob && (ob->mode & OB_MODE_VERTEX_PAINT)));
+       return (G.f&G_FACESELECT) && (ob && (ob->mode & (OB_MODE_VERTEX_PAINT|OB_MODE_WEIGHT_PAINT|OB_MODE_TEXTURE_PAINT)));
+
+}
+
+void paint_init(Paint *p, const char *name)
+{
+       Brush *brush;
+
+       /* If there's no brush, create one */
+       brush = paint_brush(p);
+       brush_check_exists(&brush, name);
+       paint_brush_set(p, brush);
+}
 
+void free_paint(Paint *paint)
+{
+       if(paint->brushes)
+               MEM_freeN(paint->brushes);
+}
+
+void copy_paint(Paint *orig, Paint *new)
+{
+       if(orig->brushes) {
+               int i;
+               new->brushes = MEM_dupallocN(orig->brushes);
+               for(i = 0; i < orig->brush_count; ++i)
+                       id_us_plus((ID *)new->brushes[i]);
+       }
 }