Cleanup: Adhere to our naming convention for BKE_linestyle.h API
[blender.git] / 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_linestyle_new("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 }