2.5 Paint:
[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_library.h"
37 #include "BKE_paint.h"
38
39 #include <stdlib.h>
40 #include <string.h>
41
42 Paint *paint_get_active(Scene *sce)
43 {
44         if(sce && sce->basact && sce->basact->object) {
45                 ToolSettings *ts = sce->toolsettings;
46
47                 switch(sce->basact->object->mode) {
48                 case OB_MODE_SCULPT:
49                         return &ts->sculpt->paint;
50                 case OB_MODE_VERTEX_PAINT:
51                         return &ts->vpaint->paint;
52                 case OB_MODE_WEIGHT_PAINT:
53                         return &ts->wpaint->paint;
54                 case OB_MODE_TEXTURE_PAINT:
55                         break;
56                         //return &ts->imapaint->paint;
57                 }
58         }
59
60         return NULL;
61 }
62
63 Brush *paint_brush(Paint *p)
64 {
65         return p && p->brushes ? p->brushes[p->active_brush_index] : NULL;
66 }
67
68 void paint_brush_set(Paint *p, Brush *br)
69 {
70         if(!br) {
71                 /* Setting to NULL removes the current slot */
72                 paint_brush_slot_remove(p);
73         }
74         else {
75                 int found = 0;
76         
77                 if(p && p->brushes) {
78                         int i;
79                         
80                         /* See if there's already a slot with the brush */
81                         for(i = 0; i < p->brush_count; ++i) {
82                                 if(p->brushes[i] == br) {
83                                         p->active_brush_index = i;
84                                         found = 1;
85                                         break;
86                                 }
87                         }
88                         
89                 }
90                 
91                 if(!found)
92                         paint_brush_slot_add(p);
93                 
94                 /* Make sure the current slot is the new brush */
95                 p->brushes[p->active_brush_index] = br;
96         }
97 }
98
99 static void paint_brush_slots_alloc(Paint *p, const int count)
100 {
101         p->brush_count = count;
102         if(count == 0)
103                 p->brushes = NULL;
104         else
105                 p->brushes = MEM_callocN(sizeof(Brush*) * count, "Brush slots");
106 }
107
108 void paint_brush_slot_add(Paint *p)
109 {
110         if(p) {
111                 Brush **orig = p->brushes;
112                 int orig_count = p->brushes ? p->brush_count : 0;
113
114                 /* Increase size of brush slot array */
115                 paint_brush_slots_alloc(p, orig_count + 1);
116                 if(orig) {
117                         memcpy(p->brushes, orig, sizeof(Brush*) * orig_count);
118                         MEM_freeN(orig);
119                 }
120
121                 p->active_brush_index = orig_count;
122         }
123 }
124
125 void paint_brush_slot_remove(Paint *p)
126 {
127         if(p && p->brushes) {
128                 Brush **orig = p->brushes;
129                 int src, dst;
130                 
131                 /* Decrease size of brush slot array */
132                 paint_brush_slots_alloc(p, p->brush_count - 1);
133                 if(p->brushes) {
134                         for(src = 0, dst = 0; dst < p->brush_count; ++src) {
135                                 if(src != p->active_brush_index) {
136                                         p->brushes[dst] = orig[src];
137                                         ++dst;
138                                 }
139                         }
140                 }
141                 MEM_freeN(orig);
142
143                 if(p->active_brush_index >= p->brush_count)
144                         p->active_brush_index = p->brush_count - 1;
145                 if(p->active_brush_index < 0)
146                         p->active_brush_index = 0;
147         }
148 }
149
150 int paint_facesel_test(Object *ob)
151 {
152         return (G.f&G_FACESELECT) && (ob && (ob->mode & (OB_MODE_VERTEX_PAINT|OB_MODE_WEIGHT_PAINT|OB_MODE_TEXTURE_PAINT)));
153
154 }
155
156 void paint_init(Paint *p, const char *name)
157 {
158         Brush *brush;
159
160         /* If there's no brush, create one */
161         brush = paint_brush(p);
162         brush_check_exists(&brush, name);
163         paint_brush_set(p, brush);
164 }
165
166 void free_paint(Paint *paint)
167 {
168         if(paint->brushes)
169                 MEM_freeN(paint->brushes);
170 }
171
172 void copy_paint(Paint *orig, Paint *new)
173 {
174         if(orig->brushes) {
175                 int i;
176                 new->brushes = MEM_dupallocN(orig->brushes);
177                 for(i = 0; i < orig->brush_count; ++i)
178                         id_us_plus((ID *)new->brushes[i]);
179         }
180 }