Merge branch 'blender2.7'
[blender.git] / release / scripts / startup / bl_ui / properties_material_gpencil.py
1 # ##### BEGIN GPL LICENSE BLOCK #####
2 #
3 #  This program is free software; you can redistribute it and/or
4 #  modify it under the terms of the GNU General Public License
5 #  as published by the Free Software Foundation; either version 2
6 #  of the License, or (at your option) any later version.
7 #
8 #  This program is distributed in the hope that it will be useful,
9 #  but WITHOUT ANY WARRANTY; without even the implied warranty of
10 #  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11 #  GNU General Public License for more details.
12 #
13 #  You should have received a copy of the GNU General Public License
14 #  along with this program; if not, write to the Free Software Foundation,
15 #  Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
16 #
17 # ##### END GPL LICENSE BLOCK #####
18
19 # <pep8 compliant>
20 import bpy
21 from bpy.types import Menu, Panel, UIList
22 from rna_prop_ui import PropertyPanel
23 from bl_ui.utils import PresetPanel
24
25 from .properties_grease_pencil_common import (
26     GreasePencilMaterialsPanel,
27 )
28
29
30 class GPENCIL_MT_color_context_menu(Menu):
31     bl_label = "Layer"
32
33     def draw(self, context):
34         layout = self.layout
35
36         layout.operator("gpencil.color_reveal", icon='RESTRICT_VIEW_OFF', text="Show All")
37         layout.operator("gpencil.color_hide", icon='RESTRICT_VIEW_ON', text="Hide Others").unselected = True
38
39         layout.separator()
40
41         layout.operator("gpencil.color_lock_all", icon='LOCKED', text="Lock All")
42         layout.operator("gpencil.color_unlock_all", icon='UNLOCKED', text="UnLock All")
43
44         layout.operator("gpencil.stroke_lock_color", text="Lock Unselected")
45         layout.operator("gpencil.lock_layer", text="Lock Unused")
46
47
48 class GPENCIL_UL_matslots(UIList):
49     def draw_item(self, context, layout, data, item, icon, active_data, active_propname, index):
50         slot = item
51         ma = slot.material
52         if (ma is not None) and (ma.grease_pencil is not None):
53             gpcolor = ma.grease_pencil
54
55             if self.layout_type in {'DEFAULT', 'COMPACT'}:
56                 if gpcolor.lock:
57                     layout.active = False
58
59                 row = layout.row(align=True)
60                 row.enabled = not gpcolor.lock
61                 row.prop(ma, "name", text="", emboss=False, icon_value=icon)
62
63                 row = layout.row(align=True)
64                 row.prop(gpcolor, "lock", text="", emboss=False)
65                 row.prop(gpcolor, "hide", text="", emboss=False)
66                 if gpcolor.ghost is True:
67                     icon = 'ONIONSKIN_OFF'
68                 else:
69                     icon = 'ONIONSKIN_ON'
70                 row.prop(gpcolor, "ghost", text="", icon=icon, emboss=False)
71
72             elif self.layout_type == 'GRID':
73                 layout.alignment = 'CENTER'
74                 layout.label(text="", icon_value=icon)
75
76
77 class GPMaterialButtonsPanel:
78     bl_space_type = 'PROPERTIES'
79     bl_region_type = 'WINDOW'
80     bl_context = "material"
81
82     @classmethod
83     def poll(cls, context):
84         ma = context.material
85         return ma and ma.grease_pencil
86
87
88 class MATERIAL_PT_gpencil_slots(GreasePencilMaterialsPanel, Panel):
89     bl_label = "Grease Pencil Material Slots"
90     bl_space_type = 'PROPERTIES'
91     bl_region_type = 'WINDOW'
92     bl_context = "material"
93     bl_options = {'HIDE_HEADER'}
94
95     @classmethod
96     def poll(cls, context):
97         ob = context.object
98         ma = context.material
99
100         return (ma and ma.grease_pencil) or (ob and ob.type == 'GPENCIL')
101
102
103 # Used as parent for "Stroke" and "Fill" panels
104 class MATERIAL_PT_gpencil_surface(GPMaterialButtonsPanel, Panel):
105     bl_label = "Surface"
106
107     def draw_header_preset(self, context):
108         MATERIAL_PT_gpencil_material_presets.draw_panel_header(self.layout)
109
110     def draw(self, context):
111         layout = self.layout
112         layout.use_property_split = True
113
114
115 class MATERIAL_PT_gpencil_strokecolor(GPMaterialButtonsPanel, Panel):
116     bl_label = "Stroke"
117     bl_parent_id = 'MATERIAL_PT_gpencil_surface'
118
119     def draw_header(self, context):
120         ma = context.material
121         if ma is not None and ma.grease_pencil is not None:
122             gpcolor = ma.grease_pencil
123             self.layout.prop(gpcolor, "show_stroke", text="")
124
125     def draw(self, context):
126         layout = self.layout
127         layout.use_property_split = True
128
129         ma = context.material
130         if ma is not None and ma.grease_pencil is not None:
131             gpcolor = ma.grease_pencil
132
133             col = layout.column()
134             col.active = not gpcolor.lock
135
136             col.prop(gpcolor, "mode")
137
138             col.prop(gpcolor, "stroke_style", text="Style")
139
140             if gpcolor.stroke_style == 'TEXTURE':
141                 row = col.row()
142                 row.enabled = not gpcolor.lock
143                 col = row.column(align=True)
144                 col.template_ID(gpcolor, "stroke_image", open="image.open")
145                 if gpcolor.mode == 'LINE':
146                     col.prop(gpcolor, "pixel_size", text="UV Factor")
147
148                 col.prop(gpcolor, "use_stroke_pattern", text="Use As Pattern")
149
150             if gpcolor.stroke_style == 'SOLID' or gpcolor.use_stroke_pattern is True:
151                 col.prop(gpcolor, "color", text="Color")
152
153             if gpcolor.mode in {'DOTS', 'BOX'}:
154                 col.prop(gpcolor, "use_follow_path", text="Follow Drawing Path")
155
156
157 class MATERIAL_PT_gpencil_fillcolor(GPMaterialButtonsPanel, Panel):
158     bl_label = "Fill"
159     bl_parent_id = 'MATERIAL_PT_gpencil_surface'
160
161     def draw_header(self, context):
162         ma = context.material
163         gpcolor = ma.grease_pencil
164         self.layout.prop(gpcolor, "show_fill", text="")
165
166     def draw(self, context):
167         layout = self.layout
168         layout.use_property_split = True
169
170         ma = context.material
171         gpcolor = ma.grease_pencil
172
173         # color settings
174         col = layout.column()
175         col.active = not gpcolor.lock
176         col.prop(gpcolor, "fill_style", text="Style")
177
178         if gpcolor.fill_style == 'GRADIENT':
179             col.prop(gpcolor, "gradient_type")
180
181         if gpcolor.fill_style != 'TEXTURE':
182             col.prop(gpcolor, "fill_color", text="Color")
183
184             if gpcolor.fill_style in {'GRADIENT', 'CHESSBOARD'}:
185                 col.prop(gpcolor, "mix_color", text="Secondary Color")
186
187             if gpcolor.fill_style == 'GRADIENT':
188                 col.prop(gpcolor, "mix_factor", text="Mix Factor", slider=True)
189
190             if gpcolor.fill_style in {'GRADIENT', 'CHESSBOARD'}:
191                 col.prop(gpcolor, "flip", text="Flip Colors")
192
193                 col.prop(gpcolor, "pattern_shift", text="Location")
194                 col.prop(gpcolor, "pattern_scale", text="Scale")
195
196             if gpcolor.gradient_type == 'RADIAL' and gpcolor.fill_style not in {'SOLID', 'CHESSBOARD'}:
197                 col.prop(gpcolor, "pattern_radius", text="Radius")
198             else:
199                 if gpcolor.fill_style != 'SOLID':
200                     col.prop(gpcolor, "pattern_angle", text="Angle")
201
202             if gpcolor.fill_style == 'CHESSBOARD':
203                 col.prop(gpcolor, "pattern_gridsize", text="Box Size")
204
205         # Texture
206         if gpcolor.fill_style == 'TEXTURE' or (gpcolor.texture_mix is True and gpcolor.fill_style == 'SOLID'):
207             col.template_ID(gpcolor, "fill_image", open="image.open")
208
209             if gpcolor.fill_style == 'TEXTURE':
210                 col.prop(gpcolor, "use_fill_pattern", text="Use As Pattern")
211                 if gpcolor.use_fill_pattern is True:
212                     col.prop(gpcolor, "fill_color", text="Color")
213
214             col.prop(gpcolor, "texture_offset", text="Offset")
215             col.prop(gpcolor, "texture_scale", text="Scale")
216             col.prop(gpcolor, "texture_angle")
217             col.prop(gpcolor, "texture_opacity")
218             col.prop(gpcolor, "texture_clamp", text="Clip Image")
219
220             if gpcolor.use_fill_pattern is False:
221                 col.prop(gpcolor, "texture_mix", text="Mix With Color")
222
223                 if gpcolor.texture_mix is True:
224                     col.prop(gpcolor, "fill_color", text="Mix Color")
225                     col.prop(gpcolor, "mix_factor", text="Mix Factor", slider=True)
226
227
228 class MATERIAL_PT_gpencil_preview(GPMaterialButtonsPanel, Panel):
229     bl_label = "Preview"
230     COMPAT_ENGINES = {'BLENDER_EEVEE'}
231     bl_options = {'DEFAULT_CLOSED'}
232
233     def draw(self, context):
234         ma = context.material
235         self.layout.label(text=ma.name)
236         self.layout.template_preview(ma)
237
238
239 class MATERIAL_PT_gpencil_custom_props(GPMaterialButtonsPanel, PropertyPanel, Panel):
240     COMPAT_ENGINES = {'BLENDER_EEVEE', 'BLENDER_WORKBENCH'}
241     _context_path = "object.active_material"
242     _property_type = bpy.types.Material
243
244
245 class MATERIAL_PT_gpencil_options(GPMaterialButtonsPanel, Panel):
246     bl_label = "Options"
247     bl_options = {'DEFAULT_CLOSED'}
248
249     def draw(self, context):
250         layout = self.layout
251         layout.use_property_split = True
252
253         ma = context.material
254         gpcolor = ma.grease_pencil
255         layout.prop(gpcolor, "pass_index")
256
257
258 class MATERIAL_PT_gpencil_material_presets(PresetPanel, Panel):
259     """Material settings"""
260     bl_label = "Material Presets"
261     preset_subdir = "gpencil_material"
262     preset_operator = "script.execute_preset"
263     preset_add_operator = "scene.gpencil_material_preset_add"
264
265
266 classes = (
267     GPENCIL_UL_matslots,
268     GPENCIL_MT_color_context_menu,
269     MATERIAL_PT_gpencil_slots,
270     MATERIAL_PT_gpencil_preview,
271     MATERIAL_PT_gpencil_material_presets,
272     MATERIAL_PT_gpencil_surface,
273     MATERIAL_PT_gpencil_strokecolor,
274     MATERIAL_PT_gpencil_fillcolor,
275     MATERIAL_PT_gpencil_options,
276     MATERIAL_PT_gpencil_custom_props,
277 )
278
279 if __name__ == "__main__":  # only for live edit.
280     from bpy.utils import register_class
281     for cls in classes:
282         register_class(cls)