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