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 }