source/blender/blenkernel/intern/freestyle.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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
  17  *
  18  * The Original Code is Copyright (C) 2013 Blender Foundation
  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 /** \file blender/blenkernel/intern/freestyle.c
  29  *  \ingroup bke
  30  */
  31
  32 #include "MEM_guardedalloc.h"
  33
  34 #include "DNA_freestyle_types.h"
  35 #include "DNA_group_types.h"
  36
  37 #include "BKE_freestyle.h"
  38 #include "BKE_linestyle.h"
  39
  40 #include "BLI_blenlib.h"
  41 #include "BLI_math.h"
  42
  43 // function declarations
  44 static FreestyleLineSet *alloc_lineset(void);
  45 static void copy_lineset(FreestyleLineSet *new_lineset, FreestyleLineSet *lineset);
  46 static FreestyleModuleConfig *alloc_module(void);
  47 static void copy_module(FreestyleModuleConfig *new_module, FreestyleModuleConfig *module);
  48
  49 void BKE_freestyle_config_init(FreestyleConfig *config)
  50 {
  51         config->mode = FREESTYLE_CONTROL_EDITOR_MODE;
  52
  53         BLI_listbase_clear(&config->modules);
  54         config->flags = 0;
  55         config->sphere_radius = 0.1f;
  56         config->dkr_epsilon = 0.0f;
  57         config->crease_angle = DEG2RADF(134.43f);
  58
  59         BLI_listbase_clear(&config->linesets);
  60 }
  61
  62 void BKE_freestyle_config_free(FreestyleConfig *config)
  63 {
  64         FreestyleLineSet *lineset;
  65
  66         for (lineset = (FreestyleLineSet *)config->linesets.first; lineset; lineset = lineset->next) {
  67                 if (lineset->group) {
  68                         lineset->group->id.us--;
  69                         lineset->group = NULL;
  70                 }
  71                 if (lineset->linestyle) {
  72                         lineset->linestyle->id.us--;
  73                         lineset->linestyle = NULL;
  74                 }
  75         }
  76         BLI_freelistN(&config->linesets);
  77         BLI_freelistN(&config->modules);
  78 }
  79
  80 void BKE_freestyle_config_copy(FreestyleConfig *new_config, FreestyleConfig *config)
  81 {
  82         FreestyleLineSet *lineset, *new_lineset;
  83         FreestyleModuleConfig *module, *new_module;
  84
  85         new_config->mode = config->mode;
  86         new_config->flags = config->flags;
  87         new_config->sphere_radius = config->sphere_radius;
  88         new_config->dkr_epsilon = config->dkr_epsilon;
  89         new_config->crease_angle = config->crease_angle;
  90
  91         BLI_listbase_clear(&new_config->linesets);
  92         for (lineset = (FreestyleLineSet *)config->linesets.first; lineset; lineset = lineset->next) {
  93                 new_lineset = alloc_lineset();
  94                 copy_lineset(new_lineset, lineset);
  95                 BLI_addtail(&new_config->linesets, (void *)new_lineset);
  96         }
  97
  98         BLI_listbase_clear(&new_config->modules);
  99         for (module = (FreestyleModuleConfig *)config->modules.first; module; module = module->next) {
 100                 new_module = alloc_module();
 101                 copy_module(new_module, module);
 102                 BLI_addtail(&new_config->modules, (void *)new_module);
 103         }
 104 }
 105
 106 static void copy_lineset(FreestyleLineSet *new_lineset, FreestyleLineSet *lineset)
 107 {
 108         new_lineset->linestyle = lineset->linestyle;
 109         if (new_lineset->linestyle)
 110                 new_lineset->linestyle->id.us++;
 111         new_lineset->flags = lineset->flags;
 112         new_lineset->selection = lineset->selection;
 113         new_lineset->qi = lineset->qi;
 114         new_lineset->qi_start = lineset->qi_start;
 115         new_lineset->qi_end = lineset->qi_end;
 116         new_lineset->edge_types = lineset->edge_types;
 117         new_lineset->exclude_edge_types = lineset->exclude_edge_types;
 118         new_lineset->group = lineset->group;
 119         if (new_lineset->group) {
 120                 new_lineset->group->id.us++;
 121         }
 122         strcpy(new_lineset->name, lineset->name);
 123 }
 124
 125 static FreestyleModuleConfig *alloc_module(void)
 126 {
 127         return (FreestyleModuleConfig *)MEM_callocN(sizeof(FreestyleModuleConfig), "style module configuration");
 128 }
 129
 130 FreestyleModuleConfig *BKE_freestyle_module_add(FreestyleConfig *config)
 131 {
 132         FreestyleModuleConfig *module_conf = alloc_module();
 133         BLI_addtail(&config->modules, (void *)module_conf);
 134         module_conf->script = NULL;
 135         module_conf->is_displayed = 1;
 136         return module_conf;
 137 }
 138
 139 static void copy_module(FreestyleModuleConfig *new_module, FreestyleModuleConfig *module)
 140 {
 141         new_module->script = module->script;
 142         new_module->is_displayed = module->is_displayed;
 143 }
 144
 145 bool BKE_freestyle_module_delete(FreestyleConfig *config, FreestyleModuleConfig *module_conf)
 146 {
 147         if (BLI_findindex(&config->modules, module_conf) == -1)
 148                 return false;
 149         BLI_freelinkN(&config->modules, module_conf);
 150         return true;
 151 }
 152
 153 bool BKE_freestyle_module_move_up(FreestyleConfig *config, FreestyleModuleConfig *module_conf)
 154 {
 155         if (BLI_findindex(&config->modules, module_conf) == -1)
 156                 return false;
 157         BLI_remlink(&config->modules, module_conf);
 158         BLI_insertlinkbefore(&config->modules, module_conf->prev, module_conf);
 159         return true;
 160 }
 161
 162 bool BKE_freestyle_module_move_down(FreestyleConfig *config, FreestyleModuleConfig *module_conf)
 163 {
 164         if (BLI_findindex(&config->modules, module_conf) == -1)
 165                 return false;
 166         BLI_remlink(&config->modules, module_conf);
 167         BLI_insertlinkafter(&config->modules, module_conf->next, module_conf);
 168         return true;
 169 }
 170
 171 void BKE_freestyle_lineset_unique_name(FreestyleConfig *config, FreestyleLineSet *lineset)
 172 {
 173         BLI_uniquename(&config->linesets, lineset, "FreestyleLineSet", '.', offsetof(FreestyleLineSet, name),
 174                        sizeof(lineset->name));
 175 }
 176
 177 static FreestyleLineSet *alloc_lineset(void)
 178 {
 179         return (FreestyleLineSet *)MEM_callocN(sizeof(FreestyleLineSet), "Freestyle line set");
 180 }
 181
 182 FreestyleLineSet *BKE_freestyle_lineset_add(FreestyleConfig *config, const char *name)
 183 {
 184         int lineset_index = BLI_countlist(&config->linesets);
 185
 186         FreestyleLineSet *lineset = alloc_lineset();
 187         BLI_addtail(&config->linesets, (void *)lineset);
 188         BKE_freestyle_lineset_set_active_index(config, lineset_index);
 189
 190         lineset->linestyle = BKE_new_linestyle("LineStyle", NULL);
 191         lineset->flags |= FREESTYLE_LINESET_ENABLED;
 192         lineset->selection = FREESTYLE_SEL_VISIBILITY | FREESTYLE_SEL_EDGE_TYPES | FREESTYLE_SEL_IMAGE_BORDER;
 193         lineset->qi = FREESTYLE_QI_VISIBLE;
 194         lineset->qi_start = 0;
 195         lineset->qi_end = 100;
 196         lineset->edge_types = FREESTYLE_FE_SILHOUETTE | FREESTYLE_FE_BORDER | FREESTYLE_FE_CREASE;
 197         lineset->exclude_edge_types = 0;
 198         lineset->group = NULL;
 199         if (name) {
 200                 BLI_strncpy(lineset->name, name, sizeof(lineset->name));
 201         }
 202         else if (lineset_index > 0) {
 203                 sprintf(lineset->name, "LineSet %i", lineset_index + 1);
 204         }
 205         else {
 206                 strcpy(lineset->name, "LineSet");
 207         }
 208         BKE_freestyle_lineset_unique_name(config, lineset);
 209
 210         return lineset;
 211 }
 212
 213 bool BKE_freestyle_lineset_delete(FreestyleConfig *config, FreestyleLineSet *lineset)
 214 {
 215         if (BLI_findindex(&config->linesets, lineset) == -1)
 216                 return false;
 217         if (lineset->group) {
 218                 lineset->group->id.us--;
 219         }
 220         if (lineset->linestyle) {
 221                 lineset->linestyle->id.us--;
 222         }
 223         BLI_remlink(&config->linesets, lineset);
 224         MEM_freeN(lineset);
 225         BKE_freestyle_lineset_set_active_index(config, 0);
 226         return true;
 227 }
 228
 229 FreestyleLineSet *BKE_freestyle_lineset_get_active(FreestyleConfig *config)
 230 {
 231         FreestyleLineSet *lineset;
 232
 233         for (lineset = (FreestyleLineSet *)config->linesets.first; lineset; lineset = lineset->next) {
 234                 if (lineset->flags & FREESTYLE_LINESET_CURRENT)
 235                         return lineset;
 236         }
 237         return NULL;
 238 }
 239
 240 short BKE_freestyle_lineset_get_active_index(FreestyleConfig *config)
 241 {
 242         FreestyleLineSet *lineset;
 243         short i;
 244
 245         for (lineset = (FreestyleLineSet *)config->linesets.first, i = 0; lineset; lineset = lineset->next, i++) {
 246                 if (lineset->flags & FREESTYLE_LINESET_CURRENT)
 247                         return i;
 248         }
 249         return 0;
 250 }
 251
 252 void BKE_freestyle_lineset_set_active_index(FreestyleConfig *config, short index)
 253 {
 254         FreestyleLineSet *lineset;
 255         short i;
 256
 257         for (lineset = (FreestyleLineSet *)config->linesets.first, i = 0; lineset; lineset = lineset->next, i++) {
 258                 if (i == index)
 259                         lineset->flags |= FREESTYLE_LINESET_CURRENT;
 260                 else
 261                         lineset->flags &= ~FREESTYLE_LINESET_CURRENT;
 262         }
 263 }