Merged changes in the trunk up to revision 51853.
[blender-staging.git] / source / blender / makesrna / intern / rna_color.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  * Contributor(s): Blender Foundation (2008).
19  *
20  * ***** END GPL LICENSE BLOCK *****
21  */
22
23 /** \file blender/makesrna/intern/rna_color.c
24  *  \ingroup RNA
25  */
26
27
28 #include <stdlib.h>
29 #include <stdio.h>
30
31 #include "RNA_define.h"
32 #include "rna_internal.h"
33
34 #include "DNA_color_types.h"
35 #include "DNA_texture_types.h"
36
37 #include "WM_api.h"
38 #include "WM_types.h"
39
40 #ifdef RNA_RUNTIME
41
42 #include "RNA_access.h"
43
44 #include "DNA_image_types.h"
45 #include "DNA_material_types.h"
46 #include "DNA_movieclip_types.h"
47 #include "DNA_node_types.h"
48
49 #include "MEM_guardedalloc.h"
50
51 #include "BKE_colortools.h"
52 #include "BKE_depsgraph.h"
53 #include "BKE_image.h"
54 #include "BKE_movieclip.h"
55 #include "BKE_node.h"
56 #include "BKE_texture.h"
57 #include "BKE_linestyle.h"
58
59 #include "ED_node.h"
60
61 #include "IMB_colormanagement.h"
62
63 static int rna_CurveMapping_curves_length(PointerRNA *ptr)
64 {
65         CurveMapping *cumap = (CurveMapping *)ptr->data;
66         int len;
67
68         for (len = 0; len < CM_TOT; len++)
69                 if (!cumap->cm[len].curve)
70                         break;
71         
72         return len;
73 }
74
75 static void rna_CurveMapping_curves_begin(CollectionPropertyIterator *iter, PointerRNA *ptr)
76 {
77         CurveMapping *cumap = (CurveMapping *)ptr->data;
78
79         rna_iterator_array_begin(iter, cumap->cm, sizeof(CurveMap), rna_CurveMapping_curves_length(ptr), 0, NULL);
80 }
81
82 static void rna_CurveMapping_clip_set(PointerRNA *ptr, int value)
83 {
84         CurveMapping *cumap = (CurveMapping *)ptr->data;
85
86         if (value) cumap->flag |= CUMA_DO_CLIP;
87         else cumap->flag &= ~CUMA_DO_CLIP;
88
89         curvemapping_changed(cumap, FALSE);
90 }
91
92 static void rna_CurveMapping_black_level_set(PointerRNA *ptr, const float *values)
93 {
94         CurveMapping *cumap = (CurveMapping *)ptr->data;
95         cumap->black[0] = values[0];
96         cumap->black[1] = values[1];
97         cumap->black[2] = values[2];
98         curvemapping_set_black_white(cumap, NULL, NULL);
99 }
100
101 static void rna_CurveMapping_white_level_set(PointerRNA *ptr, const float *values)
102 {
103         CurveMapping *cumap = (CurveMapping *)ptr->data;
104         cumap->white[0] = values[0];
105         cumap->white[1] = values[1];
106         cumap->white[2] = values[2];
107         curvemapping_set_black_white(cumap, NULL, NULL);
108 }
109
110 static void rna_CurveMapping_clipminx_range(PointerRNA *ptr, float *min, float *max, float *softmin, float *softmax)
111 {
112         CurveMapping *cumap = (CurveMapping *)ptr->data;
113
114         *min = -100.0f;
115         *max = cumap->clipr.xmax;
116 }
117
118 static void rna_CurveMapping_clipminy_range(PointerRNA *ptr, float *min, float *max, float *softmin, float *softmax)
119 {
120         CurveMapping *cumap = (CurveMapping *)ptr->data;
121
122         *min = -100.0f;
123         *max = cumap->clipr.ymax;
124 }
125
126 static void rna_CurveMapping_clipmaxx_range(PointerRNA *ptr, float *min, float *max, float *softmin, float *softmax)
127 {
128         CurveMapping *cumap = (CurveMapping *)ptr->data;
129
130         *min = cumap->clipr.xmin;
131         *max = 100.0f;
132 }
133
134 static void rna_CurveMapping_clipmaxy_range(PointerRNA *ptr, float *min, float *max, float *softmin, float *softmax)
135 {
136         CurveMapping *cumap = (CurveMapping *)ptr->data;
137
138         *min = cumap->clipr.ymin;
139         *max = 100.0f;
140 }
141
142
143 static char *rna_ColorRamp_path(PointerRNA *ptr)
144 {
145         char *path = NULL;
146         
147         /* handle the cases where a single datablock may have 2 ramp types */
148         if (ptr->id.data) {
149                 ID *id = ptr->id.data;
150                 
151                 switch (GS(id->name)) {
152                         case ID_MA: /* material has 2 cases - diffuse and specular */
153                         {
154                                 Material *ma = (Material *)id;
155                                 
156                                 if (ptr->data == ma->ramp_col)
157                                         path = BLI_strdup("diffuse_ramp");
158                                 else if (ptr->data == ma->ramp_spec)
159                                         path = BLI_strdup("specular_ramp");
160                                 break;
161                         }
162                         
163                         case ID_NT:
164                         {
165                                 bNodeTree *ntree = (bNodeTree *)id;
166                                 bNode *node;
167                                 PointerRNA node_ptr;
168                                 char *node_path;
169                                 
170                                 for (node = ntree->nodes.first; node; node = node->next) {
171                                         if (ELEM3(node->type, SH_NODE_VALTORGB, CMP_NODE_VALTORGB, TEX_NODE_VALTORGB)) {
172                                                 if (node->storage == ptr->data) {
173                                                         /* all node color ramp properties called 'color_ramp'
174                                                          * prepend path from ID to the node
175                                                          */
176                                                         RNA_pointer_create(id, &RNA_Node, node, &node_ptr);
177                                                         node_path = RNA_path_from_ID_to_struct(&node_ptr);
178                                                         path = BLI_sprintfN("%s.color_ramp", node_path);
179                                                         MEM_freeN(node_path);
180                                                 }
181                                         }
182                                 }
183                                 break;
184                         }
185                         
186                         default:
187                                 /* everything else just uses 'color_ramp' */
188                                 path = BLI_strdup("color_ramp");
189                                 break;
190
191                         case ID_LS:
192                         {
193                                 char *path = FRS_path_from_ID_to_color_ramp((FreestyleLineStyle *)id, (ColorBand *)ptr->data);
194                                 if (path)
195                                         return path;
196                         }
197                                 break;
198                 }
199         }
200         else {
201                 /* everything else just uses 'color_ramp' */
202                 path = BLI_strdup("color_ramp");
203         }
204         
205         return path;
206 }
207
208 static char *rna_ColorRampElement_path(PointerRNA *ptr)
209 {
210         PointerRNA ramp_ptr;
211         PropertyRNA *prop;
212         char *path = NULL;
213         int index;
214         
215         /* helper macro for use here to try and get the path
216          *      - this calls the standard code for getting a path to a texture...
217          */
218
219 #define COLRAMP_GETPATH                                                       \
220 {                                                                             \
221         prop = RNA_struct_find_property(&ramp_ptr, "elements");                   \
222         if (prop) {                                                               \
223                 index = RNA_property_collection_lookup_index(&ramp_ptr, prop, ptr);   \
224                 if (index >= 0) {                                                     \
225                         char *texture_path = rna_ColorRamp_path(&ramp_ptr);               \
226                         path = BLI_sprintfN("%s.elements[%d]", texture_path, index);      \
227                         MEM_freeN(texture_path);                                          \
228                 }                                                                     \
229         }                                                                         \
230 } (void)0
231
232         /* determine the path from the ID-block to the ramp */
233         /* FIXME: this is a very slow way to do it, but it will have to suffice... */
234         if (ptr->id.data) {
235                 ID *id = ptr->id.data;
236                 
237                 switch (GS(id->name)) {
238                         case ID_MA: /* 2 cases for material - diffuse and spec */
239                         {
240                                 Material *ma = (Material *)id;
241                                 
242                                 /* try diffuse first */
243                                 if (ma->ramp_col) {
244                                         RNA_pointer_create(id, &RNA_ColorRamp, ma->ramp_col, &ramp_ptr);
245                                         COLRAMP_GETPATH;
246                                 }
247                                 /* try specular if not diffuse */
248                                 if (!path && ma->ramp_spec) {
249                                         RNA_pointer_create(id, &RNA_ColorRamp, ma->ramp_spec, &ramp_ptr);
250                                         COLRAMP_GETPATH;
251                                 }
252                         }
253                         break;
254                                 
255                         case ID_NT:
256                         {
257                                 bNodeTree *ntree = (bNodeTree *)id;
258                                 bNode *node;
259                                 
260                                 for (node = ntree->nodes.first; node; node = node->next) {
261                                         if (ELEM3(node->type, SH_NODE_VALTORGB, CMP_NODE_VALTORGB, TEX_NODE_VALTORGB)) {
262                                                 RNA_pointer_create(id, &RNA_ColorRamp, node->storage, &ramp_ptr);
263                                                 COLRAMP_GETPATH;
264                                         }
265                                 }
266                         }
267                         break;
268                                 
269                         case ID_LS:
270                         {
271                                 ListBase listbase;
272                                 LinkData *link;
273
274                                 FRS_list_modifier_color_ramps((FreestyleLineStyle *)id, &listbase);
275                                 for (link = (LinkData *)listbase.first; link; link = link->next) {
276                                         RNA_pointer_create(id, &RNA_ColorRamp, link->data, &ramp_ptr);
277                                         COLRAMP_GETPATH;
278                                 }
279                                 BLI_freelistN(&listbase);
280                         }
281                                 break;
282
283                         default: /* everything else should have a "color_ramp" property */
284                         {
285                                 /* create pointer to the ID block, and try to resolve "color_ramp" pointer */
286                                 RNA_id_pointer_create(id, &ramp_ptr);
287                                 if (RNA_path_resolve(&ramp_ptr, "color_ramp", &ramp_ptr, &prop)) {
288                                         COLRAMP_GETPATH;
289                                 }
290                         }
291                 }
292         }
293         
294         /* cleanup the macro we defined */
295 #undef COLRAMP_GETPATH
296         
297         return path;
298 }
299
300 static void rna_ColorRamp_update(Main *bmain, Scene *UNUSED(scene), PointerRNA *ptr)
301 {
302         if (ptr->id.data) {
303                 ID *id = ptr->id.data;
304                 
305                 switch (GS(id->name)) {
306                         case ID_MA:
307                         {
308                                 Material *ma = ptr->id.data;
309                                 
310                                 DAG_id_tag_update(&ma->id, 0);
311                                 WM_main_add_notifier(NC_MATERIAL | ND_SHADING_DRAW, ma);
312                         }
313                         break;
314                         case ID_NT:
315                         {
316                                 bNodeTree *ntree = (bNodeTree *)id;
317                                 bNode *node;
318
319                                 for (node = ntree->nodes.first; node; node = node->next) {
320                                         if (ELEM3(node->type, SH_NODE_VALTORGB, CMP_NODE_VALTORGB, TEX_NODE_VALTORGB)) {
321                                                 ED_node_generic_update(bmain, ntree, node);
322                                         }
323                                 }
324                         }
325                         break;
326                         case ID_TE:
327                         {
328                                 Tex *tex = ptr->id.data;
329
330                                 DAG_id_tag_update(&tex->id, 0);
331                                 WM_main_add_notifier(NC_TEXTURE, tex);
332                         }
333                         break;
334                         case ID_LS:
335                         {
336                                 FreestyleLineStyle *linestyle= ptr->id.data;
337
338                                 WM_main_add_notifier(NC_LINESTYLE, linestyle);
339                         }
340                                 break;
341                         default:
342                                 break;
343                 }
344         }
345 }
346
347 static void rna_ColorRamp_eval(struct ColorBand *coba, float position, float color[4])
348 {
349         do_colorband(coba, position, color);
350 }
351
352 static CBData *rna_ColorRampElement_new(struct ColorBand *coba, ReportList *reports, float position)
353 {
354         CBData *element = colorband_element_add(coba, position);
355
356         if (element == NULL)
357                 BKE_reportf(reports, RPT_ERROR, "Unable to add element to colorband (limit %d)", MAXCOLORBAND);
358
359         return element;
360 }
361
362 static void rna_ColorRampElement_remove(struct ColorBand *coba, ReportList *reports, PointerRNA *element_ptr)
363 {
364         CBData *element = element_ptr->data;
365         int index = (int)(element - coba->data);
366         if (colorband_element_remove(coba, index) == FALSE) {
367                 BKE_report(reports, RPT_ERROR, "Element not found in element collection or last element");
368                 return;
369         }
370
371         RNA_POINTER_INVALIDATE(element_ptr);
372 }
373
374 void rna_CurveMap_remove_point(CurveMap *cuma, ReportList *reports, PointerRNA *point_ptr)
375 {
376         CurveMapPoint *point = point_ptr->data;
377         if (curvemap_remove_point(cuma, point) == FALSE) {
378                 BKE_report(reports, RPT_ERROR, "Unable to remove curve point");
379                 return;
380         }
381
382         RNA_POINTER_INVALIDATE(point_ptr);
383 }
384
385 static void rna_Scopes_update(Main *UNUSED(bmain), Scene *UNUSED(scene), PointerRNA *ptr)
386 {
387         Scopes *s = (Scopes *)ptr->data;
388         s->ok = 0;
389 }
390
391 static int rna_ColorManagedDisplaySettings_display_device_get(struct PointerRNA *ptr)
392 {
393         ColorManagedDisplaySettings *display = (ColorManagedDisplaySettings *) ptr->data;
394
395         return IMB_colormanagement_display_get_named_index(display->display_device);
396 }
397
398 static void rna_ColorManagedDisplaySettings_display_device_set(struct PointerRNA *ptr, int value)
399 {
400         ColorManagedDisplaySettings *display = (ColorManagedDisplaySettings *) ptr->data;
401         const char *name = IMB_colormanagement_display_get_indexed_name(value);
402
403         if (name) {
404                 BLI_strncpy(display->display_device, name, sizeof(display->display_device));
405         }
406 }
407
408 static EnumPropertyItem *rna_ColorManagedDisplaySettings_display_device_itemf(bContext *C, PointerRNA *ptr, PropertyRNA *UNUSED(prop), int *free)
409 {
410         EnumPropertyItem *items = NULL;
411         int totitem = 0;
412
413         IMB_colormanagement_display_items_add(&items, &totitem);
414         RNA_enum_item_end(&items, &totitem);
415
416         *free = TRUE;
417
418         return items;
419 }
420
421 static void rna_ColorManagedDisplaySettings_display_device_update(Main *UNUSED(bmain), Scene *UNUSED(scene), PointerRNA *ptr)
422 {
423         ID *id = ptr->id.data;
424
425         if (!id)
426                 return;
427
428         if (GS(id->name) == ID_SCE) {
429                 Scene *scene = (Scene *) id;
430
431                 IMB_colormanagement_validate_settings(&scene->display_settings, &scene->view_settings);
432
433                 WM_main_add_notifier(NC_SCENE | ND_SEQUENCER, NULL);
434         }
435 }
436
437 static int rna_ColorManagedViewSettings_view_transform_get(PointerRNA *ptr)
438 {
439         ColorManagedViewSettings *view = (ColorManagedViewSettings *) ptr->data;
440
441         return IMB_colormanagement_view_get_named_index(view->view_transform);
442 }
443
444 static void rna_ColorManagedViewSettings_view_transform_set(PointerRNA *ptr, int value)
445 {
446         ColorManagedViewSettings *view = (ColorManagedViewSettings *) ptr->data;
447
448         const char *name = IMB_colormanagement_view_get_indexed_name(value);
449
450         if (name) {
451                 BLI_strncpy(view->view_transform, name, sizeof(view->view_transform));
452         }
453 }
454
455 static EnumPropertyItem* rna_ColorManagedViewSettings_view_transform_itemf(bContext *C, PointerRNA *ptr, PropertyRNA *UNUSED(prop), int *free)
456 {
457         Scene *scene = CTX_data_scene(C);
458         EnumPropertyItem *items = NULL;
459         ColorManagedDisplaySettings *display_settings = &scene->display_settings;
460         int totitem = 0;
461
462         IMB_colormanagement_view_items_add(&items, &totitem, display_settings->display_device);
463         RNA_enum_item_end(&items, &totitem);
464
465         *free = TRUE;
466         return items;
467 }
468
469 static void rna_ColorManagedViewSettings_use_curves_set(PointerRNA *ptr, int value)
470 {
471         ColorManagedViewSettings *view_settings = (ColorManagedViewSettings *) ptr->data;
472
473         if (value) {
474                 view_settings->flag |= COLORMANAGE_VIEW_USE_CURVES;
475
476                 if (view_settings->curve_mapping == NULL) {
477                         view_settings->curve_mapping = curvemapping_add(4, 0.0f, 0.0f, 1.0f, 1.0f);
478                 }
479         }
480         else {
481                 view_settings->flag &= ~COLORMANAGE_VIEW_USE_CURVES;
482         }
483 }
484
485 static int rna_ColorManagedColorspaceSettings_colorspace_get(struct PointerRNA *ptr)
486 {
487         ColorManagedColorspaceSettings *colorspace = (ColorManagedColorspaceSettings *) ptr->data;
488
489         return IMB_colormanagement_colorspace_get_named_index(colorspace->name);
490 }
491
492 static void rna_ColorManagedColorspaceSettings_colorspace_set(struct PointerRNA *ptr, int value)
493 {
494         ColorManagedColorspaceSettings *colorspace = (ColorManagedColorspaceSettings *) ptr->data;
495         const char *name = IMB_colormanagement_colorspace_get_indexed_name(value);
496
497         if (name) {
498                 BLI_strncpy(colorspace->name, name, sizeof(colorspace->name));
499         }
500 }
501
502 static EnumPropertyItem *rna_ColorManagedColorspaceSettings_colorspace_itemf(bContext *C, PointerRNA *ptr, PropertyRNA *UNUSED(prop), int *free)
503 {
504         EnumPropertyItem *items = NULL;
505         int totitem = 0;
506
507         IMB_colormanagement_colorspace_items_add(&items, &totitem);
508         RNA_enum_item_end(&items, &totitem);
509
510         *free = TRUE;
511
512         return items;
513 }
514
515 static void rna_ColorManagedColorspaceSettings_reload_update(Main *UNUSED(bmain), Scene *UNUSED(scene), PointerRNA *ptr)
516 {
517         ID *id = ptr->id.data;
518
519         if (GS(id->name) == ID_IM) {
520                 Image *ima = (Image *) id;
521
522                 DAG_id_tag_update(&ima->id, 0);
523
524                 BKE_image_signal(ima, NULL, IMA_SIGNAL_COLORMANAGE);
525
526                 WM_main_add_notifier(NC_IMAGE | ND_DISPLAY, &ima->id);
527                 WM_main_add_notifier(NC_IMAGE | NA_EDITED, &ima->id);
528         }
529         else if (GS(id->name) == ID_MC) {
530                 MovieClip *clip = (MovieClip *) id;
531
532                 BKE_movieclip_reload(clip);
533
534                 WM_main_add_notifier(NC_MOVIECLIP | ND_DISPLAY, &clip->id);
535                 WM_main_add_notifier(NC_MOVIECLIP | NA_EDITED, &clip->id);
536         }
537 }
538
539 static void rna_ColorManagement_update(Main *UNUSED(bmain), Scene *UNUSED(scene), PointerRNA *ptr)
540 {
541         ID *id = ptr->id.data;
542
543         if (!id)
544                 return;
545
546         if (GS(id->name) == ID_SCE) {
547                 WM_main_add_notifier(NC_SCENE | ND_SEQUENCER, NULL);
548         }
549 }
550
551 /* this function only exists because #curvemap_evaluateF uses a 'const' qualifier */
552 static float rna_CurveMap_evaluateF(struct CurveMap *cuma, float value)
553 {
554         return curvemap_evaluateF(cuma, value);
555 }
556
557 #else
558
559 static void rna_def_curvemappoint(BlenderRNA *brna)
560 {
561         StructRNA *srna;
562         PropertyRNA *prop;
563         static EnumPropertyItem prop_handle_type_items[] = {
564                 {0, "AUTO", 0, "Auto Handle", ""},
565                 {CUMA_VECTOR, "VECTOR", 0, "Vector Handle", ""},
566                 {0, NULL, 0, NULL, NULL}
567         };
568
569         srna = RNA_def_struct(brna, "CurveMapPoint", NULL);
570         RNA_def_struct_ui_text(srna, "CurveMapPoint", "Point of a curve used for a curve mapping");
571
572         prop = RNA_def_property(srna, "location", PROP_FLOAT, PROP_XYZ);
573         RNA_def_property_float_sdna(prop, NULL, "x");
574         RNA_def_property_array(prop, 2);
575         RNA_def_property_ui_text(prop, "Location", "X/Y coordinates of the curve point");
576
577         prop = RNA_def_property(srna, "handle_type", PROP_ENUM, PROP_NONE);
578         RNA_def_property_enum_bitflag_sdna(prop, NULL, "flag");
579         RNA_def_property_enum_items(prop, prop_handle_type_items);
580         RNA_def_property_ui_text(prop, "Handle Type", "Curve interpolation at this point: Bezier or vector");
581
582         prop = RNA_def_property(srna, "select", PROP_BOOLEAN, PROP_NONE);
583         RNA_def_property_boolean_sdna(prop, NULL, "flag", CUMA_SELECT);
584         RNA_def_property_ui_text(prop, "Select", "Selection state of the curve point");
585 }
586
587 static void rna_def_curvemap_points_api(BlenderRNA *brna, PropertyRNA *cprop)
588 {
589         StructRNA *srna;
590         PropertyRNA *parm;
591         FunctionRNA *func;
592
593         RNA_def_property_srna(cprop, "CurveMapPoints");
594         srna = RNA_def_struct(brna, "CurveMapPoints", NULL);
595         RNA_def_struct_sdna(srna, "CurveMap");
596         RNA_def_struct_ui_text(srna, "Curve Map Point", "Collection of Curve Map Points");
597
598         func = RNA_def_function(srna, "new", "curvemap_insert");
599         RNA_def_function_ui_description(func, "Add point to CurveMap");
600         parm = RNA_def_float(func, "position", 0.0f, -FLT_MAX, FLT_MAX, "Position", "Position to add point", -FLT_MAX, FLT_MAX);
601         RNA_def_property_flag(parm, PROP_REQUIRED);
602         parm = RNA_def_float(func, "value", 0.0f, -FLT_MAX, FLT_MAX, "Value", "Value of point", -FLT_MAX, FLT_MAX);
603         RNA_def_property_flag(parm, PROP_REQUIRED);
604         parm = RNA_def_pointer(func, "point", "CurveMapPoint", "", "New point");
605         RNA_def_function_return(func, parm);
606
607         func = RNA_def_function(srna, "remove", "rna_CurveMap_remove_point");
608         RNA_def_function_flag(func, FUNC_USE_REPORTS);
609         RNA_def_function_ui_description(func, "Delete point from CurveMap");
610         parm = RNA_def_pointer(func, "point", "CurveMapPoint", "", "PointElement to remove");
611         RNA_def_property_flag(parm, PROP_REQUIRED | PROP_NEVER_NULL | PROP_RNAPTR);
612         RNA_def_property_clear_flag(parm, PROP_THICK_WRAP);
613 }
614
615 static void rna_def_curvemap(BlenderRNA *brna)
616 {
617         StructRNA *srna;
618         PropertyRNA *prop, *parm;
619         FunctionRNA *func;
620
621         static EnumPropertyItem prop_extend_items[] = {
622                 {0, "HORIZONTAL", 0, "Horizontal", ""},
623                 {CUMA_EXTEND_EXTRAPOLATE, "EXTRAPOLATED", 0, "Extrapolated", ""},
624                 {0, NULL, 0, NULL, NULL}
625         };
626
627         srna = RNA_def_struct(brna, "CurveMap", NULL);
628         RNA_def_struct_ui_text(srna, "CurveMap", "Curve in a curve mapping");
629
630         prop = RNA_def_property(srna, "extend", PROP_ENUM, PROP_NONE);
631         RNA_def_property_enum_bitflag_sdna(prop, NULL, "flag");
632         RNA_def_property_enum_items(prop, prop_extend_items);
633         RNA_def_property_ui_text(prop, "Extend", "Extrapolate the curve or extend it horizontally");
634
635         prop = RNA_def_property(srna, "points", PROP_COLLECTION, PROP_NONE);
636         RNA_def_property_collection_sdna(prop, NULL, "curve", "totpoint");
637         RNA_def_property_struct_type(prop, "CurveMapPoint");
638         RNA_def_property_ui_text(prop, "Points", "");
639         rna_def_curvemap_points_api(brna, prop);
640
641         func = RNA_def_function(srna, "evaluate", "rna_CurveMap_evaluateF");
642         RNA_def_function_ui_description(func, "Evaluate curve at given location");
643         parm = RNA_def_float(func, "position", 0.0f, -FLT_MAX, FLT_MAX, "Position", "Position to evaluate curve at", -FLT_MAX, FLT_MAX);
644         RNA_def_property_flag(parm, PROP_REQUIRED);
645         parm = RNA_def_float(func, "value", 0.0f, -FLT_MAX, FLT_MAX, "Value", "Value of curve at given location", -FLT_MAX, FLT_MAX);
646         RNA_def_function_return(func, parm);
647 }
648
649 static void rna_def_curvemapping(BlenderRNA *brna)
650 {
651         StructRNA *srna;
652         PropertyRNA *prop;
653         FunctionRNA *func;
654
655         srna = RNA_def_struct(brna, "CurveMapping", NULL);
656         RNA_def_struct_ui_text(srna, "CurveMapping",
657                                "Curve mapping to map color, vector and scalar values to other values using "
658                                "a user defined curve");
659         
660         prop = RNA_def_property(srna, "use_clip", PROP_BOOLEAN, PROP_NONE);
661         RNA_def_property_boolean_sdna(prop, NULL, "flag", CUMA_DO_CLIP);
662         RNA_def_property_ui_text(prop, "Clip", "Force the curve view to fit a defined boundary");
663         RNA_def_property_boolean_funcs(prop, NULL, "rna_CurveMapping_clip_set");
664
665         prop = RNA_def_property(srna, "clip_min_x", PROP_FLOAT, PROP_NONE);
666         RNA_def_property_float_sdna(prop, NULL, "clipr.xmin");
667         RNA_def_property_range(prop, -100.0f, 100.0f);
668         RNA_def_property_ui_text(prop, "Clip Min X", "");
669         RNA_def_property_float_funcs(prop, NULL, NULL, "rna_CurveMapping_clipminx_range");
670
671         prop = RNA_def_property(srna, "clip_min_y", PROP_FLOAT, PROP_NONE);
672         RNA_def_property_float_sdna(prop, NULL, "clipr.ymin");
673         RNA_def_property_range(prop, -100.0f, 100.0f);
674         RNA_def_property_ui_text(prop, "Clip Min Y", "");
675         RNA_def_property_float_funcs(prop, NULL, NULL, "rna_CurveMapping_clipminy_range");
676
677         prop = RNA_def_property(srna, "clip_max_x", PROP_FLOAT, PROP_NONE);
678         RNA_def_property_float_sdna(prop, NULL, "clipr.xmax");
679         RNA_def_property_range(prop, -100.0f, 100.0f);
680         RNA_def_property_ui_text(prop, "Clip Max X", "");
681         RNA_def_property_float_funcs(prop, NULL, NULL, "rna_CurveMapping_clipmaxx_range");
682
683         prop = RNA_def_property(srna, "clip_max_y", PROP_FLOAT, PROP_NONE);
684         RNA_def_property_float_sdna(prop, NULL, "clipr.ymax");
685         RNA_def_property_range(prop, -100.0f, 100.0f);
686         RNA_def_property_ui_text(prop, "Clip Max Y", "");
687         RNA_def_property_float_funcs(prop, NULL, NULL, "rna_CurveMapping_clipmaxy_range");
688
689         prop = RNA_def_property(srna, "curves", PROP_COLLECTION, PROP_NONE);
690         RNA_def_property_collection_funcs(prop, "rna_CurveMapping_curves_begin", "rna_iterator_array_next",
691                                           "rna_iterator_array_end", "rna_iterator_array_get",
692                                           "rna_CurveMapping_curves_length", NULL, NULL, NULL);
693         RNA_def_property_struct_type(prop, "CurveMap");
694         RNA_def_property_ui_text(prop, "Curves", "");
695
696         prop = RNA_def_property(srna, "black_level", PROP_FLOAT, PROP_COLOR);
697         RNA_def_property_float_sdna(prop, NULL, "black");
698         RNA_def_property_range(prop, -1000.0f, 1000.0f);
699         RNA_def_property_ui_text(prop, "Black Level", "For RGB curves, the color that black is mapped to");
700         RNA_def_property_float_funcs(prop, NULL, "rna_CurveMapping_black_level_set", NULL);
701
702         prop = RNA_def_property(srna, "white_level", PROP_FLOAT, PROP_COLOR);
703         RNA_def_property_float_sdna(prop, NULL, "white");
704         RNA_def_property_range(prop, -1000.0f, 1000.0f);
705         RNA_def_property_ui_text(prop, "White Level", "For RGB curves, the color that white is mapped to");
706         RNA_def_property_float_funcs(prop, NULL, "rna_CurveMapping_white_level_set", NULL);
707
708         func = RNA_def_function(srna, "update", "curvemapping_changed_all");
709         RNA_def_function_ui_description(func, "Update curve mapping after making changes");
710 }
711
712 static void rna_def_color_ramp_element(BlenderRNA *brna)
713 {
714         StructRNA *srna;
715         PropertyRNA *prop;
716
717         srna = RNA_def_struct(brna, "ColorRampElement", NULL);
718         RNA_def_struct_sdna(srna, "CBData");
719         RNA_def_struct_path_func(srna, "rna_ColorRampElement_path");
720         RNA_def_struct_ui_text(srna, "Color Ramp Element", "Element defining a color at a position in the color ramp");
721         
722         prop = RNA_def_property(srna, "color", PROP_FLOAT, PROP_COLOR);
723         RNA_def_property_float_sdna(prop, NULL, "r");
724         RNA_def_property_array(prop, 4);
725         RNA_def_property_ui_text(prop, "Color", "Set color of selected color stop");
726         RNA_def_property_update(prop, 0, "rna_ColorRamp_update");
727         
728         prop = RNA_def_property(srna, "position", PROP_FLOAT, PROP_NONE);
729         RNA_def_property_float_sdna(prop, NULL, "pos");
730         RNA_def_property_range(prop, 0, 1);
731         RNA_def_property_ui_text(prop, "Position", "Set position of selected color stop");
732         RNA_def_property_update(prop, 0, "rna_ColorRamp_update");
733 }
734
735 static void rna_def_color_ramp_element_api(BlenderRNA *brna, PropertyRNA *cprop)
736 {
737         StructRNA *srna;
738         PropertyRNA *parm;
739         FunctionRNA *func;
740
741         RNA_def_property_srna(cprop, "ColorRampElements");
742         srna = RNA_def_struct(brna, "ColorRampElements", NULL);
743         RNA_def_struct_sdna(srna, "ColorBand");
744         RNA_def_struct_path_func(srna, "rna_ColorRampElement_path");
745         RNA_def_struct_ui_text(srna, "Color Ramp Elements", "Collection of Color Ramp Elements");
746
747         /* TODO, make these functions generic in texture.c */
748         func = RNA_def_function(srna, "new", "rna_ColorRampElement_new");
749         RNA_def_function_ui_description(func, "Add element to ColorRamp");
750         RNA_def_function_flag(func, FUNC_USE_REPORTS);
751         parm = RNA_def_float(func, "position", 0.0f, 0.0f, 1.0f, "Position", "Position to add element", 0.0f, 1.0f);
752         RNA_def_property_flag(parm, PROP_REQUIRED);
753         /* return type */
754         parm = RNA_def_pointer(func, "element", "ColorRampElement", "", "New element");
755         RNA_def_function_return(func, parm);
756
757         func = RNA_def_function(srna, "remove", "rna_ColorRampElement_remove");
758         RNA_def_function_ui_description(func, "Delete element from ColorRamp");
759         RNA_def_function_flag(func, FUNC_USE_REPORTS);
760         parm = RNA_def_pointer(func, "element", "ColorRampElement", "", "Element to remove");
761         RNA_def_property_flag(parm, PROP_REQUIRED | PROP_NEVER_NULL | PROP_RNAPTR);
762         RNA_def_property_clear_flag(parm, PROP_THICK_WRAP);
763 }
764
765 static void rna_def_color_ramp(BlenderRNA *brna)
766 {
767         StructRNA *srna;
768         PropertyRNA *prop;
769         FunctionRNA *func;
770
771         static EnumPropertyItem prop_interpolation_items[] = {
772                 {1, "EASE", 0, "Ease", ""},
773                 {3, "CARDINAL", 0, "Cardinal", ""},
774                 {0, "LINEAR", 0, "Linear", ""},
775                 {2, "B_SPLINE", 0, "B-Spline", ""},
776                 {4, "CONSTANT", 0, "Constant", ""},
777                 {0, NULL, 0, NULL, NULL}
778         };
779         
780         srna = RNA_def_struct(brna, "ColorRamp", NULL);
781         RNA_def_struct_sdna(srna, "ColorBand");
782         RNA_def_struct_path_func(srna, "rna_ColorRamp_path");
783         RNA_def_struct_ui_text(srna, "Color Ramp", "Color ramp mapping a scalar value to a color");
784         
785         prop = RNA_def_property(srna, "elements", PROP_COLLECTION, PROP_COLOR);
786         RNA_def_property_collection_sdna(prop, NULL, "data", "tot");
787         RNA_def_property_struct_type(prop, "ColorRampElement");
788         RNA_def_property_ui_text(prop, "Elements", "");
789         RNA_def_property_update(prop, 0, "rna_ColorRamp_update");
790         rna_def_color_ramp_element_api(brna, prop);
791
792         prop = RNA_def_property(srna, "interpolation", PROP_ENUM, PROP_NONE);
793         RNA_def_property_enum_sdna(prop, NULL, "ipotype");
794         RNA_def_property_enum_items(prop, prop_interpolation_items);
795         RNA_def_property_ui_text(prop, "Interpolation", "");
796         RNA_def_property_update(prop, 0, "rna_ColorRamp_update");
797
798 #if 0 /* use len(elements) */
799         prop = RNA_def_property(srna, "total", PROP_INT, PROP_NONE);
800         RNA_def_property_int_sdna(prop, NULL, "tot");
801         /* needs a function to do the right thing when adding elements like colorband_add_cb() */
802         RNA_def_property_clear_flag(prop, PROP_EDITABLE);
803         RNA_def_property_range(prop, 0, 31); /* MAXCOLORBAND = 32 */
804         RNA_def_property_ui_text(prop, "Total", "Total number of elements");
805         RNA_def_property_update(prop, 0, "rna_ColorRamp_update");
806 #endif
807         
808         func = RNA_def_function(srna, "evaluate", "rna_ColorRamp_eval");
809         RNA_def_function_ui_description(func, "Evaluate ColorRamp");
810         prop = RNA_def_float(func, "position", 1.0f, 0.0f, 1.0f, "Position", "Evaluate ColorRamp at position", 0.0f, 1.0f);
811         RNA_def_property_flag(prop, PROP_REQUIRED);
812         /* return */
813         prop = RNA_def_float_color(func, "color", 4, NULL, -FLT_MAX, FLT_MAX, "Color", "Color at given position",
814                                    -FLT_MAX, FLT_MAX);
815         RNA_def_property_flag(prop, PROP_THICK_WRAP);
816         RNA_def_function_output(func, prop);
817 }
818
819 static void rna_def_histogram(BlenderRNA *brna)
820 {
821         StructRNA *srna;
822         PropertyRNA *prop;
823         
824         static EnumPropertyItem prop_mode_items[] = {
825                 {HISTO_MODE_LUMA, "LUMA", 0, "Luma", "Luma"},
826                 {HISTO_MODE_RGB, "RGB", 0, "RGB", "Red Green Blue"},
827                 {HISTO_MODE_R, "R", 0, "R", "Red"},
828                 {HISTO_MODE_G, "G", 0, "G", "Green"},
829                 {HISTO_MODE_B, "B", 0, "B", "Blue"},
830                 {HISTO_MODE_ALPHA, "A", 0, "A", "Alpha"},
831                 {0, NULL, 0, NULL, NULL}
832         };
833
834         srna = RNA_def_struct(brna, "Histogram", NULL);
835         RNA_def_struct_ui_text(srna, "Histogram", "Statistical view of the levels of color in an image");
836         
837         prop = RNA_def_property(srna, "mode", PROP_ENUM, PROP_NONE);
838         RNA_def_property_enum_sdna(prop, NULL, "mode");
839         RNA_def_property_enum_items(prop, prop_mode_items);
840         RNA_def_property_ui_text(prop, "Mode", "Channels to display when drawing the histogram");
841
842         prop = RNA_def_property(srna, "show_line", PROP_BOOLEAN, PROP_NONE);
843         RNA_def_property_boolean_sdna(prop, NULL, "flag", HISTO_FLAG_LINE);
844         RNA_def_property_ui_text(prop, "Show Line", "Display lines rather then filled shapes");
845         RNA_def_property_ui_icon(prop, ICON_IPO, 0);
846 }
847
848 static void rna_def_scopes(BlenderRNA *brna)
849 {
850         StructRNA *srna;
851         PropertyRNA *prop;
852
853         static EnumPropertyItem prop_wavefrm_mode_items[] = {
854                 {SCOPES_WAVEFRM_LUMA, "LUMA", ICON_COLOR, "Luma", ""},
855                 {SCOPES_WAVEFRM_RGB, "RGB", ICON_COLOR, "Red Green Blue", ""},
856                 {SCOPES_WAVEFRM_YCC_601, "YCBCR601", ICON_COLOR, "YCbCr (ITU 601)", ""},
857                 {SCOPES_WAVEFRM_YCC_709, "YCBCR709", ICON_COLOR, "YCbCr (ITU 709)", ""},
858                 {SCOPES_WAVEFRM_YCC_JPEG, "YCBCRJPG", ICON_COLOR, "YCbCr (Jpeg)", ""},
859                 {0, NULL, 0, NULL, NULL}
860         };
861
862         srna = RNA_def_struct(brna, "Scopes", NULL);
863         RNA_def_struct_ui_text(srna, "Scopes", "Scopes for statistical view of an image");
864         
865         prop = RNA_def_property(srna, "use_full_resolution", PROP_BOOLEAN, PROP_NONE);
866         RNA_def_property_boolean_sdna(prop, "Scopes", "sample_full", 1);
867         RNA_def_property_ui_text(prop, "Full Sample", "Sample every pixel of the image");
868         RNA_def_property_update(prop, 0, "rna_Scopes_update");
869         
870         prop = RNA_def_property(srna, "accuracy", PROP_FLOAT, PROP_PERCENTAGE);
871         RNA_def_property_float_sdna(prop, "Scopes", "accuracy");
872         RNA_def_property_range(prop, 0.0, 100.0);
873         RNA_def_property_ui_range(prop, 0.0, 100.0, 10, 1);
874         RNA_def_property_ui_text(prop, "Accuracy", "Proportion of original image source pixel lines to sample");
875         RNA_def_property_update(prop, 0, "rna_Scopes_update");
876
877         prop = RNA_def_property(srna, "histogram", PROP_POINTER, PROP_NONE);
878         RNA_def_property_pointer_sdna(prop, "Scopes", "hist");
879         RNA_def_property_struct_type(prop, "Histogram");
880         RNA_def_property_ui_text(prop, "Histogram", "Histogram for viewing image statistics");
881
882         prop = RNA_def_property(srna, "waveform_mode", PROP_ENUM, PROP_NONE);
883         RNA_def_property_enum_sdna(prop, "Scopes", "wavefrm_mode");
884         RNA_def_property_enum_items(prop, prop_wavefrm_mode_items);
885         RNA_def_property_ui_text(prop, "Waveform Mode", "");
886         RNA_def_property_update(prop, 0, "rna_Scopes_update");
887
888         prop = RNA_def_property(srna, "waveform_alpha", PROP_FLOAT, PROP_FACTOR);
889         RNA_def_property_float_sdna(prop, "Scopes", "wavefrm_alpha");
890         RNA_def_property_range(prop, 0, 1);
891         RNA_def_property_ui_text(prop, "Waveform Opacity", "Opacity of the points");
892
893         prop = RNA_def_property(srna, "vectorscope_alpha", PROP_FLOAT, PROP_FACTOR);
894         RNA_def_property_float_sdna(prop, "Scopes", "vecscope_alpha");
895         RNA_def_property_range(prop, 0, 1);
896         RNA_def_property_ui_text(prop, "Vectorscope Opacity", "Opacity of the points");
897 }
898
899 static void rna_def_colormanage(BlenderRNA *brna)
900 {
901         StructRNA *srna;
902         PropertyRNA *prop;
903
904         static EnumPropertyItem display_device_items[] = {
905                 {0, "DEFAULT", 0, "Default", ""},
906                 {0, NULL, 0, NULL, NULL}
907         };
908
909         static EnumPropertyItem view_transform_items[] = {
910                 {0, "NONE", 0, "None", "Do not perform any color transform on display, use old non-color managed technique for display"},
911                 {0, NULL, 0, NULL, NULL}
912         };
913
914         static EnumPropertyItem color_space_items[] = {
915                 {0, "NONE", 0, "None", "Do not perform any color transform on load, treat colors as in scene linear space already"},
916                 {0, NULL, 0, NULL, NULL}
917         };
918
919         /* ** Display Settings  **  */
920         srna = RNA_def_struct(brna, "ColorManagedDisplaySettings", NULL);
921         RNA_def_struct_ui_text(srna, "ColorManagedDisplaySettings", "Color management specific to display device");
922
923         prop = RNA_def_property(srna, "display_device", PROP_ENUM, PROP_NONE);
924         RNA_def_property_enum_items(prop, display_device_items);
925         RNA_def_property_enum_funcs(prop, "rna_ColorManagedDisplaySettings_display_device_get",
926                                           "rna_ColorManagedDisplaySettings_display_device_set",
927                                           "rna_ColorManagedDisplaySettings_display_device_itemf");
928         RNA_def_property_ui_text(prop, "Display Device", "Display device name");
929         RNA_def_property_update(prop, NC_WINDOW, "rna_ColorManagedDisplaySettings_display_device_update");
930
931         /* ** View Settings  **  */
932         srna = RNA_def_struct(brna, "ColorManagedViewSettings", NULL);
933         RNA_def_struct_ui_text(srna, "ColorManagedViewSettings", "Color management settings used for displaying images on the display");
934
935         prop = RNA_def_property(srna, "view_transform", PROP_ENUM, PROP_NONE);
936         RNA_def_property_enum_items(prop, view_transform_items);
937         RNA_def_property_enum_funcs(prop, "rna_ColorManagedViewSettings_view_transform_get",
938                                           "rna_ColorManagedViewSettings_view_transform_set",
939                                           "rna_ColorManagedViewSettings_view_transform_itemf");
940         RNA_def_property_ui_text(prop, "View Transform", "View used ");
941         RNA_def_property_update(prop, NC_WINDOW, "rna_ColorManagement_update");
942
943         prop = RNA_def_property(srna, "exposure", PROP_FLOAT, PROP_FACTOR);
944         RNA_def_property_float_sdna(prop, NULL, "exposure");
945         RNA_def_property_range(prop, -10.0f, 10.0f);
946         RNA_def_property_float_default(prop, 0.0f);
947         RNA_def_property_ui_text(prop, "Exposure", "Exposure (stops) applied before display transform");
948         RNA_def_property_update(prop, NC_WINDOW, "rna_ColorManagement_update");
949
950         prop = RNA_def_property(srna, "gamma", PROP_FLOAT, PROP_FACTOR);
951         RNA_def_property_float_sdna(prop, NULL, "gamma");
952         RNA_def_property_float_default(prop, 1.0f);
953         RNA_def_property_range(prop, 0.0f, 5.0f);
954         RNA_def_property_ui_text(prop, "Gamma", "Amount of gamma modification applied after display transform");
955         RNA_def_property_update(prop, NC_WINDOW, "rna_ColorManagement_update");
956
957         prop = RNA_def_property(srna, "curve_mapping", PROP_POINTER, PROP_NONE);
958         RNA_def_property_pointer_sdna(prop, NULL, "curve_mapping");
959         RNA_def_property_ui_text(prop, "Curve", "Color curve mapping applied before display transform");
960         RNA_def_property_update(prop, NC_WINDOW, "rna_ColorManagement_update");
961
962         prop = RNA_def_property(srna, "use_curve_mapping", PROP_BOOLEAN, PROP_NONE);
963         RNA_def_property_boolean_sdna(prop, NULL, "flag", COLORMANAGE_VIEW_USE_CURVES);
964         RNA_def_property_boolean_funcs(prop, NULL, "rna_ColorManagedViewSettings_use_curves_set");
965         RNA_def_property_ui_text(prop, "Use Curves", "Use RGB curved for pre-display transformation");
966         RNA_def_property_update(prop, NC_WINDOW, "rna_ColorManagement_update");
967
968         /* ** Colorspace **  */
969         srna = RNA_def_struct(brna, "ColorManagedColorspaceSettings", NULL);
970         RNA_def_struct_ui_text(srna, "ColorManagedColorspaceSettings", "Input color space settings");
971
972         prop = RNA_def_property(srna, "name", PROP_ENUM, PROP_NONE);
973         RNA_def_property_enum_items(prop, color_space_items);
974         RNA_def_property_enum_funcs(prop, "rna_ColorManagedColorspaceSettings_colorspace_get",
975                                           "rna_ColorManagedColorspaceSettings_colorspace_set",
976                                           "rna_ColorManagedColorspaceSettings_colorspace_itemf");
977         RNA_def_property_ui_text(prop, "Color Space", "Input color space name");
978         RNA_def_property_update(prop, NC_WINDOW, "rna_ColorManagedColorspaceSettings_reload_update");
979 }
980
981 void RNA_def_color(BlenderRNA *brna)
982 {
983         rna_def_curvemappoint(brna);
984         rna_def_curvemap(brna);
985         rna_def_curvemapping(brna);
986         rna_def_color_ramp_element(brna);
987         rna_def_color_ramp(brna);
988         rna_def_histogram(brna);
989         rna_def_scopes(brna);
990         rna_def_colormanage(brna);
991 }
992
993 #endif