BPython:
[blender.git] / source / blender / src / buttons_script.c
1 /**
2  * $Id: 
3  *
4  * ***** BEGIN GPL/BL DUAL LICENSE BLOCK *****
5  *
6  * This program is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU General Public License
8  * as published by the Free Software Foundation; either version 2
9  * of the License, or (at your option) any later version. The Blender
10  * Foundation also sells licenses for use in proprietary software under
11  * the Blender License.  See http://www.blender.org/BL/ for information
12  * about this.
13  *
14  * This program is distributed in the hope that it will be useful,
15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
17  * GNU General Public License for more details.
18  *
19  * You should have received a copy of the GNU General Public License
20  * along with this program; if not, write to the Free Software Foundation,
21  * Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
22  *
23  * The Original Code is Copyright (C) 2001-2002 by NaN Holding BV.
24  * All rights reserved.
25  *
26  * The Original Code is: all of this file.
27  *
28  * Contributor(s): none yet.
29  *
30  * ***** END GPL/BL DUAL LICENSE BLOCK *****
31  */
32
33 #include <time.h>
34 #include <math.h>
35 #include <stdlib.h>
36 #include <string.h>
37
38 #ifdef HAVE_CONFIG_H
39 #include <config.h>
40 #endif
41
42 #include "MEM_guardedalloc.h"
43 #include "DNA_action_types.h"
44 #include "DNA_armature_types.h"
45 #include "DNA_camera_types.h"
46 #include "DNA_constraint_types.h"
47 #include "DNA_curve_types.h"
48 #include "DNA_effect_types.h"
49 #include "DNA_group_types.h"
50 #include "DNA_ID.h"
51 #include "DNA_ika_types.h"
52 #include "DNA_image_types.h"
53 #include "DNA_key_types.h"
54 #include "DNA_lamp_types.h"
55 #include "DNA_lattice_types.h"
56 #include "DNA_material_types.h"
57 #include "DNA_meta_types.h"
58 #include "DNA_object_types.h"
59 #include "DNA_radio_types.h"
60 #include "DNA_screen_types.h"
61 #include "DNA_scene_types.h"
62 #include "DNA_sound_types.h"
63 #include "DNA_text_types.h"
64 #include "DNA_texture_types.h"
65 #include "DNA_userdef_types.h"
66 #include "DNA_vfont_types.h"
67 #include "DNA_view3d_types.h"
68 #include "DNA_world_types.h"
69 #include "DNA_space_types.h"
70
71 #include "BKE_anim.h"
72 #include "BKE_armature.h"
73 #include "BKE_constraint.h"
74 #include "BKE_curve.h"
75 #include "BKE_displist.h"
76 #include "BKE_effect.h"
77 #include "BKE_font.h"
78 #include "BKE_ika.h"
79 #include "BKE_image.h"
80 #include "BKE_ipo.h"
81 #include "BKE_lattice.h"
82 #include "BKE_main.h"
83 #include "BKE_material.h"
84 #include "BKE_mball.h"
85 #include "BKE_mesh.h"
86 #include "BKE_object.h"
87 #include "BKE_sound.h"
88 #include "BKE_texture.h"
89 #include "BKE_global.h"
90 #include "BKE_utildefines.h"
91
92 #include "BLI_blenlib.h"
93 #include "BLI_dynstr.h"
94
95 #include "BSE_filesel.h"
96
97 #include "BIF_gl.h"
98 #include "BIF_graphics.h"
99 #include "BIF_keyval.h"
100 #include "BIF_mainqueue.h"
101 #include "BIF_resources.h"
102 #include "BIF_screen.h"
103 #include "BIF_mywindow.h"
104 #include "BIF_space.h"
105 #include "BIF_glutil.h"
106 #include "BIF_interface.h"
107 #include "BIF_toolbox.h"
108
109 #include "BIF_butspace.h"
110
111 #include "mydevice.h"
112 #include "blendef.h"
113 #include "butspace.h"
114
115 /* ************************ function prototypes ********************** */
116 void draw_scriptlink(uiBlock *, ScriptLink *, int , int , int ) ;
117
118 /* *************************** SCRIPT ******************************** */
119
120 static void extend_scriptlink(ScriptLink *slink)
121 {
122         void *stmp, *ftmp;
123
124         if (!slink) return;
125                 
126         stmp= slink->scripts;           
127         slink->scripts= MEM_mallocN(sizeof(ID*)*(slink->totscript+1), "scriptlistL");
128         
129         ftmp= slink->flag;              
130         slink->flag= MEM_mallocN(sizeof(short*)*(slink->totscript+1), "scriptlistF");
131         
132         if (slink->totscript) {
133                 memcpy(slink->scripts, stmp, sizeof(ID*)*(slink->totscript));
134                 MEM_freeN(stmp);
135
136                 memcpy(slink->flag, ftmp, sizeof(short)*(slink->totscript));
137                 MEM_freeN(ftmp);
138         }
139
140         slink->scripts[slink->totscript]= NULL;
141         slink->flag[slink->totscript]= SCRIPT_FRAMECHANGED;
142
143         slink->totscript++;
144         slink->actscript = slink->totscript;
145 }
146
147 static void delete_scriptlink(ScriptLink *slink)
148 {
149         int i;
150         
151         if (!slink) return;
152         
153         if (slink->totscript>0) {
154                 for (i=slink->actscript-1; i<slink->totscript-1; i++) {
155                         slink->flag[i]= slink->flag[i+1];
156                         slink->scripts[i]= slink->scripts[i+1];
157                 }
158                 
159                 slink->totscript--;
160         }
161                 
162         CLAMP(slink->actscript, 1, slink->totscript);
163                 
164         if (slink->totscript==0) {
165                 if (slink->scripts) MEM_freeN(slink->scripts);
166                 if (slink->flag) MEM_freeN(slink->flag);
167
168                 slink->scripts= NULL;
169                 slink->flag= NULL;
170                 slink->totscript= slink->actscript= 0;                  
171         }
172 }
173
174 static char *scriptlinks_menu_string(void)
175 {
176         char *menu = NULL;
177         DynStr *ds = BLI_dynstr_new();
178         Text *text = G.main->text.first;
179         int txtcounter = 0;
180
181         if (text) {
182                 BLI_dynstr_append(ds, "Select Script Link%t");
183                 while (text) {
184                         BLI_dynstr_append(ds, "|");
185                         BLI_dynstr_append(ds, text->id.name+2);
186                         txtcounter += 1;
187                         text = text->id.next;
188                 }
189                 if (txtcounter) menu = BLI_dynstr_get_cstring(ds);
190         }
191         BLI_dynstr_free(ds);
192         return menu;
193 }
194
195 static void scriptlinks_pupmenu(ScriptLink *slink)
196 {
197         short menuitem;
198         char *menustr = scriptlinks_menu_string();
199
200         if (menustr) {
201                 menuitem = pupmenu_col(menustr, 20);
202                 MEM_freeN(menustr);
203                 if (menuitem > 0) {
204                         Text *text = G.main->text.first;
205                         while (--menuitem) text = text->id.next;
206                         if (text) slink->scripts[slink->totscript - 1]= (ID *)text;
207                 }
208         }
209 }
210
211 void do_scriptbuts(unsigned short event)
212 {
213         Object *ob=NULL;
214         ScriptLink *slink=NULL;
215         Material *ma;
216         
217         switch (event) {
218         case B_SSCRIPT_ADD:
219                 slink = &G.scene->scriptlink;
220                 extend_scriptlink(slink);
221                 BIF_undo_push("Add scriptlink");
222                 scriptlinks_pupmenu(slink);
223                 break;
224         case B_SSCRIPT_DEL:
225                 BIF_undo_push("Delete scriptlink");
226                 delete_scriptlink(&G.scene->scriptlink);
227                 break;
228                 
229         case B_SCRIPT_ADD:
230         case B_SCRIPT_DEL:
231                 ob= OBACT;
232
233                 if (ob && G.buts->scriptblock==ID_OB) {
234                                 slink= &ob->scriptlink;
235
236                 } else if (ob && G.buts->scriptblock==ID_MA) {
237                         ma= give_current_material(ob, ob->actcol);
238                         if (ma) slink= &ma->scriptlink;
239
240                 } else if (ob && G.buts->scriptblock==ID_CA) {
241                         if (ob->type==OB_CAMERA)
242                                 slink= &((Camera *)ob->data)->scriptlink;
243
244                 } else if (ob && G.buts->scriptblock==ID_LA) {
245                         if (ob->type==OB_LAMP)
246                                 slink= &((Lamp *)ob->data)->scriptlink;
247
248                 } else if (G.buts->scriptblock==ID_WO) {
249                         if (G.scene->world) 
250                                 slink= &(G.scene->world->scriptlink);
251                 }
252                 
253                 if (event==B_SCRIPT_ADD) {
254                         extend_scriptlink(slink);
255                         BIF_undo_push("Add scriptlink");
256                         scriptlinks_pupmenu(slink);
257                 }
258                 else {
259                         delete_scriptlink(slink);
260                         BIF_undo_push("Delete scriptlink");
261                 }
262                 break;
263         default:
264                 break;
265         }
266
267         allqueue(REDRAWBUTSSCRIPT, 0);
268 }
269
270 void draw_scriptlink(uiBlock *block, ScriptLink *script, int sx, int sy, int scene) 
271 {
272         char str[256];
273
274         if (script->totscript) {
275                 strcpy(str, "FrameChanged%x 1|");
276                 strcat(str, "Redraw%x 4|");
277                 strcat(str, "Render%x 16|");
278                 if (scene) {
279                         strcat(str, "OnLoad%x 2|");
280                         strcat(str, "OnSave%x 8");
281                 }
282
283                 uiDefButS(block, MENU, 1, str, (short)sx, (short)sy, 140, 19, &script->flag[script->actscript-1], 0, 0, 0, 0, "Script links for this event");
284
285                 uiDefIDPoinBut(block, test_scriptpoin_but, 1, "", (short)(sx+140),(short)sy, 140, 19, &script->scripts[script->actscript-1], "Name of Script to link");
286         }
287
288         sprintf(str,"%d Scr:", script->totscript);
289         uiDefButS(block, NUM, REDRAWBUTSSCRIPT, str, (short)(sx+140), (short)sy-20,60,19, &script->actscript, 1, script->totscript, 0, 0, "Total / Active Script link (LeftMouse + Drag to change)");
290
291         if (scene) {
292                 if (script->totscript<32767) 
293                         uiDefBut(block, BUT, B_SSCRIPT_ADD, "New", (short)(sx+240), (short)sy-20, 40, 19, 0, 0, 0, 0, 0, "Add a new Script link");
294                 if (script->totscript) 
295                         uiDefBut(block, BUT, B_SSCRIPT_DEL, "Del", (short)(sx+200), (short)sy-20, 40, 19, 0, 0, 0, 0, 0, "Delete the current Script link");
296
297                 uiDefBut(block, LABEL, 0, "Scene Script link",  sx,sy-20,140,20, 0, 0, 0, 0, 0, "");
298
299         } 
300         else {
301                 if (script->totscript<32767) 
302                         uiDefBut(block, BUT, B_SCRIPT_ADD, "New", (short)(sx+240), (short)sy-20, 40, 19, 0, 0, 0, 0, 0, "Add a new Script link");
303                 if (script->totscript) 
304                         uiDefBut(block, BUT, B_SCRIPT_DEL, "Del", (short)(sx+200), (short)sy-20, 40, 19, 0, 0, 0, 0, 0, "Delete the current Script link");
305
306                 uiDefBut(block, LABEL, 0, "Selected Script link",       sx,sy-20,140,20, 0, 0, 0, 0, 0, "");
307         }               
308 }
309
310 /* ************************************* */
311
312 static void  script_panel_scriptlink(void)
313 {
314         uiBlock *block;
315         Object *ob=NULL;
316         ScriptLink *script=NULL;
317         Material *ma;
318         int xco = 10;
319
320         block= uiNewBlock(&curarea->uiblocks, "script_panel_scriptlink", UI_EMBOSS, UI_HELV, curarea->win);
321         if(uiNewPanel(curarea, block, "Scriptlinks", "Script", 0, 0, 318, 204)==0) return;
322
323         uiDefButI(block, TOG|BIT|13, REDRAWBUTSSCRIPT,
324                         "Enable Script Links", xco, 200, 150, 20, &G.f, 0, 0, 0, 0,
325                         "Enable execution of all assigned Script links");
326         /* for proper alignment: */
327         uiDefBut(block, LABEL, 0, "",   160, 200,150,20, NULL, 0.0, 0.0, 0, 0, "");
328
329         if (G.f & G_DOSCRIPTLINKS) {
330                 ob= OBACT;
331                 if(ob) 
332                         uiDefIconButS(block, ROW, B_REDR, ICON_OBJECT, xco,175,25,20, &G.buts->scriptblock,  2.0, (float)ID_OB, 0, 0, "Displays Object script links");
333
334                 if(ob && give_current_material(ob, ob->actcol))
335                         uiDefIconButS(block, ROW, B_REDR, ICON_MATERIAL,        xco+=25,175,25,20, &G.buts->scriptblock, 2.0, (float)ID_MA, 0, 0, "Displays Material script links ");
336
337                 if(G.scene->world) 
338                         uiDefIconButS(block, ROW, B_REDR, ICON_WORLD,   xco+=25,175,25,20, &G.buts->scriptblock, 2.0, (float)ID_WO, 0, 0, "Displays World script links");
339
340                 if(ob && ob->type==OB_CAMERA)
341                         uiDefIconButS(block, ROW, B_REDR, ICON_CAMERA,  xco+=25,175,25,20, &G.buts->scriptblock, 2.0, (float)ID_CA, 0, 0, "Displays Camera script links");
342
343                 if(ob && ob->type==OB_LAMP)
344                         uiDefIconButS(block, ROW, B_REDR, ICON_LAMP,    xco+=25,175,25,20, &G.buts->scriptblock, 2.0, (float)ID_LA, 0, 0, "Displays Lamp script links");
345
346
347                 if (ob && G.buts->scriptblock==ID_OB) {
348                         script= &ob->scriptlink;
349                 
350                 } else if (ob && G.buts->scriptblock==ID_MA) {
351                         ma= give_current_material(ob, ob->actcol);
352                         if (ma) script= &ma->scriptlink;
353                 
354                 } else if (ob && G.buts->scriptblock==ID_CA) {
355                         if (ob->type==OB_CAMERA)
356                                 script= &((Camera *)ob->data)->scriptlink;
357                         
358                 } else if (ob && G.buts->scriptblock==ID_LA) {
359                         if (ob->type==OB_LAMP)
360                                 script= &((Lamp *)ob->data)->scriptlink;
361
362                 } else if (G.buts->scriptblock==ID_WO) {
363                         if (G.scene->world)
364                                 script= &(G.scene->world->scriptlink);
365                 }
366
367                 if (script) draw_scriptlink(block, script, 10, 140, 0);                 
368
369                 draw_scriptlink(block, &G.scene->scriptlink, 10, 80, 1);
370         }
371 }
372
373
374
375
376 void script_panels()
377 {
378         script_panel_scriptlink();
379         
380 }