2.5 Sculpt:
[blender.git] / source / blender / blenkernel / intern / paint.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., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
17  *
18  * The Original Code is Copyright (C) 2009 by Nicholas Bishop
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 #include "MEM_guardedalloc.h"
29
30 #include "DNA_brush_types.h"
31 #include "DNA_object_types.h"
32 #include "DNA_scene_types.h"
33
34 #include "BKE_brush.h"
35 #include "BKE_global.h"
36 #include "BKE_paint.h"
37
38 #include <stdlib.h>
39 #include <string.h>
40
41 Paint *paint_get_active(Scene *sce)
42 {
43         if(sce && sce->basact && sce->basact->object) {
44                 switch(sce->basact->object->mode) {
45                 case OB_MODE_SCULPT:
46                         return &sce->toolsettings->sculpt->paint;
47                 }
48         }
49         /*else if(G.f & G_VERTEXPAINT)
50                 return &sce->toolsettings->vpaint->paint;
51         else if(G.f & G_WEIGHTPAINT)
52                 return &sce->toolsettings->wpaint->paint;
53         else if(G.f & G_TEXTUREPAINT)
54         return &sce->toolsettings->imapaint.paint;*/
55
56         return NULL;
57 }
58
59 Brush *paint_brush(Paint *p)
60 {
61         return p && p->brushes ? p->brushes[p->active_brush_index] : NULL;
62 }
63
64 void paint_brush_set(Paint *p, Brush *br)
65 {
66         if(p && p->brushes) {
67                 int i;
68
69                 /* See if there's already a slot with the brush */
70                 for(i = 0; i < p->brush_count; ++i) {
71                         if(p->brushes[i] == br) {
72                                 p->active_brush_index = i;
73                                 break;
74                         }
75                 }
76                 
77         }
78         else
79                 paint_brush_slot_add(p);
80
81         /* Make sure the current slot is the new brush */
82         p->brushes[p->active_brush_index] = br;
83 }
84
85 static void paint_brush_slots_alloc(Paint *p, const int count)
86 {
87         p->brush_count = count;
88         if(count == 0)
89                 p->brushes = NULL;
90         else
91                 p->brushes = MEM_callocN(sizeof(Brush*) * count, "Brush slots");
92 }
93
94 void paint_brush_slot_add(Paint *p)
95 {
96         Brush **orig = p->brushes;
97         int orig_count = p->brushes ? p->brush_count : 0;
98
99         /* Increase size of brush slot array */
100         paint_brush_slots_alloc(p, orig_count + 1);
101         if(orig) {
102                 memcpy(p->brushes, orig, sizeof(Brush*) * orig_count);
103                 MEM_freeN(orig);
104         }
105
106         p->active_brush_index = orig_count;
107 }
108
109 void paint_brush_slot_remove(Paint *p)
110 {
111         if(p->brushes) {
112                 Brush **orig = p->brushes;
113                 int src, dst;
114                 
115                 /* Decrease size of brush slot array */
116                 paint_brush_slots_alloc(p, p->brush_count - 1);
117                 if(p->brushes) {
118                         for(src = 0, dst = 0; dst < p->brush_count; ++src) {
119                                 if(src != p->active_brush_index) {
120                                         p->brushes[dst] = orig[src];
121                                         ++dst;
122                                 }
123                         }
124                 }
125                 MEM_freeN(orig);
126
127                 if(p->active_brush_index >= p->brush_count)
128                         p->active_brush_index = p->brush_count - 1;
129                 if(p->active_brush_index < 0)
130                         p->active_brush_index = 0;
131         }
132 }
133
134 int paint_facesel_test(Object *ob)
135 {
136         return (G.f&G_FACESELECT) && (ob && (ob->mode & (OB_MODE_VERTEX_PAINT|OB_MODE_WEIGHT_PAINT|OB_MODE_TEXTURE_PAINT)));
137
138 }
139
140 void free_paint(Paint *paint)
141 {
142         if(paint->brushes)
143                 MEM_freeN(paint->brushes);
144 }
145
146 void copy_paint(Paint *orig, Paint *new)
147 {
148         if(orig->brushes)
149                 new->brushes = MEM_dupallocN(orig->brushes);
150 }