root/core/script.c

DEFINITIONS

1. script_print_screen_init
2. script_print_screen_end
3. script_print_screen_statement
4. script_console_add_line
5. script_console_add_error
6. action_stack_AS_SCRIPT_RUN
7. script_stack_start
8. script_is_running
9. script_set_terminate_key
10. script_get_alt_text
11. script_check_terminate
12. gui_script_kbd_process
13. gui_script_draw
14. script_wait_terminate
15. script_end
16. script_start_gui

   1 #include "camera_info.h"
   2 #include "keyboard.h"
   3 #include "modes.h"
   4 #include "viewport.h"
   5 #include "conf.h"
   6 #include "script.h"
   7 #include "console.h"
   8 #include "action_stack.h"
   9 #include "shot_histogram.h"
  10 #include "lang.h"
  11 #include "gui.h"
  12 #include "gui_lang.h"
  13 #include "ptp.h"
  14 #include "clock.h"
  15 #include "usb_remote.h"
  16 #include "script_api.h"
  17 #include "motion_detector.h"
  19 //-------------------------------------------------------------------
  21 static AS_ID running_script_stack_name = 0;     // ID of action_stack, which used to control script processing
  23 //-------------------------------------------------------------------
  24 // script terminate key, may be set from script
  25 static int script_terminate_key = KEY_SHOOT_FULL;
  26 static char script_terminate_key_name[20]; // TODO this is only here because there's no easy way to map key number back to name
  28 static int script_terminate_request = 0; // script abort requested from another task (ptp)
  30 // Forward references
  31 void script_end();
  33 //=======================================================
  34 //                 SCRIPT CONSOLE FUNCTIONS
  35 //=======================================================
  37 //-------------------------------------------------------------------
  38 static int  print_screen_p;             // print_screen predicate: 0=off, else is log file number. negative=append, postive=overwrite 
  39 static int  print_screen_d = -1;        // print_screen file descriptor.
  40 char print_screen_file[25];
  42 static void script_print_screen_init()
  43 {
print_screen_p = 0;
  45     if (print_screen_d >= 0) {
close(print_screen_d);
print_screen_d = -1;
  48     }
  49 }
  51 static void script_print_screen_end()
  52 {
  53     if (print_screen_d >= 0) {
close(print_screen_d);
print_screen_d = -1;
print_screen_p = 0;
  57     }
  58 }
  60 void script_print_screen_statement(int val)
  61 {
  62   // Negative values for 'val' parameter will append to log file,
  63   // positive values will truncate the log file
  64   int flag_trunc = O_TRUNC;
print_screen_p = val;
  67   if (val) {
  68     if (print_screen_d>=0) close(print_screen_d);
  69     if (val<0) {
flag_trunc = 0;
val = -val;
  72     }
  73     while (val > 9999) val -= 10000;
sprintf(print_screen_file, "A/CHDK/LOGS/LOG_%04d.TXT", val);
print_screen_d = open(print_screen_file, O_WRONLY|O_CREAT|flag_trunc, 0777);
  76     if (print_screen_d>=0) lseek(print_screen_d,0,SEEK_END);
  77   }
  78   else script_print_screen_end() ;
  79 }
  81 void script_console_add_line(long str_id)
  82 {
  83     const char* str = lang_str(str_id);
console_add_line(str);
  86     if (print_screen_p && (print_screen_d >= 0)) {
  87         char nl = '\n';
  88         // TODO this should be uncached memory
write(print_screen_d, str, strlen(str) );
write(print_screen_d, &nl, 1);
  91     }
  92 }
  94 void script_console_add_error(long str_id)
  95 {
console_set_autoredraw(1);          // Force console display on
script_console_add_line(str_id);
  98 }
 100 //=======================================================
 101 //                 SCRIPT FUNCTIONS
 102 //=======================================================
 104 // Stack process function for running current script
 105 static int action_stack_AS_SCRIPT_RUN()
 106 {
 107     if (camera_info.state.state_kbd_script_run)
 108     {
 109         int rv = libscriptapi->script_run();
 110         if (rv != SCRIPT_RUN_RUNNING)
 111         {
 112             // Script language is responsible for displaying 'Finished' or error
 113             // messages so all we need to do is shutdown the script engine
script_end();
 115             return 1;
 116         }
 117     }
 118     else
 119     {
action_pop_func(0);
 121         return 1;
 122     }
 123     return 0;
 124 }
 126 long script_stack_start()
 127 {
script_terminate_request = 0; // ensure any previous terminate request is cleared
camera_info.state.state_kbd_script_run = SCRIPT_STATE_RAN;
running_script_stack_name = action_stack_create(&action_stack_AS_SCRIPT_RUN);
 132     return running_script_stack_name;
 133 }
 135 int script_is_running()
 136 {
 137     return !action_stack_is_finished(running_script_stack_name);
 138 }
 140 //-------------------------------------------------------------------
 141 void script_set_terminate_key(int key, const char *keyname)
 142 {
script_terminate_key = key;
strncpy(script_terminate_key_name,keyname,sizeof(script_terminate_key_name));
script_terminate_key_name[sizeof(script_terminate_key_name)-1] = 0;
 146 }
 148 void script_get_alt_text(char *buf)
 149 {
 150     if ((script_terminate_key != KEY_SHOOT_FULL) && camera_info.state.state_kbd_script_run)
 151     {
sprintf(buf,"<EXIT=%s>",script_terminate_key_name);
 153     }
 154     else
 155     {
strcpy(buf,"<ALT>");
 157     }
 158 }
 160 // check if an external terminate request has been set
 161 // this is called directly from kbd_process, so it will be processed whether or not in alt mode
 162 void script_check_terminate(void)
 163 {
 164     if(camera_info.state.state_kbd_script_run && script_terminate_request) {
script_console_add_error(LANG_CONSOLE_TEXT_TERMINATED);
script_end();
 167     }
 168 }
 170 // Main button processing for CHDK Script mode
 172 static int gui_script_kbd_process()
 173 {
 174     // Stop a script if the shutter button pressed in Script mode
 175     if (kbd_is_key_clicked(script_terminate_key))
 176     {
script_console_add_error(LANG_CONSOLE_TEXT_INTERRUPTED);
 178         if (camera_info.state.state_kbd_script_run == SCRIPT_STATE_INTERRUPTED)
script_end();
 180         else
 181         {
camera_info.state.state_kbd_script_run = SCRIPT_STATE_INTERRUPTED;
 183             if (libscriptapi->run_restore() == 0)
script_end();
 185         }
 186     }
 188     return 0;
 189 }
 191 //-------------------------------------------------------------------
 192 void gui_script_draw()
 193 {
 194     extern void gui_chdk_draw();
gui_chdk_draw();
 197     if (camera_info.state.mode_rec || camera_info.state.mode_play)
 198     {
 199         static int show_md_grid=0;
 200         if (camera_info.state.state_kbd_script_run) show_md_grid=5;
 201         if (show_md_grid)
 202         {
 203             --show_md_grid;
libmotiondetect->md_draw_grid();
 205         }
 206     }
 207 }
 209 // GUI handler for Script mode
gui_handler scriptGuiHandler = { GUI_MODE_SCRIPT, gui_script_draw, gui_script_kbd_process, 0, 0, 0 };
 212 static gui_handler *old_gui_handler = 0;
 214 //-------------------------------------------------------------------
 215 // request the script end from an external task
 216 // NOTE this must NOT be called from kbd_task
 217 void script_wait_terminate(void)
 218 {
 219     if(camera_info.state.state_kbd_script_run == SCRIPT_STATE_INACTIVE) {
 220         return;
 221     }
script_terminate_request = 1;
 223     // TODO timeout?
 224     while(camera_info.state.state_kbd_script_run != SCRIPT_STATE_INACTIVE) {
msleep(10);
 226     }
 227 }
 229 // Terminate a script, either because of error, or the script finished
 230 void script_end()
 231 {
 232     // Tell other code that script has ended
camera_info.state.state_kbd_script_run = SCRIPT_STATE_INACTIVE;
camera_info.state.osd_title_line = 1 ;
 236     // reset the script terminate key
script_terminate_key = KEY_SHOOT_FULL ;
script_print_screen_end();
console_set_autoredraw(1);          // Force console display on in case script turned it off
 242     // Restore old handler - prevent calling MD draw after module unloaded
 243     if (old_gui_handler)
 244     {
 245         extern gui_handler defaultGuiHandler; 
 246         // if script switched in or out of alt, try to preserve
 247         // TODO is there ever a situation where it should be something other than default or alt?
 248         if(old_gui_handler == &altGuiHandler && camera_info.state.gui_mode_none) {
gui_set_mode(&defaultGuiHandler);
 250         } else if(old_gui_handler == &defaultGuiHandler && camera_info.state.gui_mode_alt) {
gui_set_mode(&altGuiHandler);
 252         } else {
gui_set_mode(old_gui_handler);
 254         }
old_gui_handler = 0;
 256     }
 258     // Reset script language module
libscriptapi->script_reset();
 261     // Kill off the action_stack for the script, since we've just reset the script
 262     // language and unloaded the MD module, we don't need to let the stack empty
 263     // itself.
action_stack_kill(running_script_stack_name);
running_script_stack_name = -1;
libshothisto->shot_histogram_set(0);
kbd_key_release_all();
conf_setAutosave(1);    // Turn on autosave of config file in conf_setValue in case script turned it off
conf_update_prevent_shutdown();
 272 }
 274 long script_start_gui( int autostart )
 275 {
 276     if (conf.script_file[0] == 0) return 0;
libshothisto->shot_histogram_set(0);
camera_info.state.auto_started = autostart;
 281     // Kill high speed USB timer if running
stop_usb_HPtimer();
 284     // Keyboard init
camera_info.state.kbd_last_clicked = 0;
camera_info.state.kbd_last_checked_time = get_tick_count();
kbd_key_release_all();
 289     // Close old console, will be re-opened when first line added
console_close();
script_print_screen_init();
save_params_values(0);
script_console_add_line((autostart)?LANG_CONSOLE_TEXT_AUTOSTARTED:LANG_CONSOLE_TEXT_STARTED);
module_set_script_lang(conf.script_file);
 298     if ( !libscriptapi->script_start_file(conf.script_file) )
 299     {
 300         return -1;
 301     }
sc_param *p = script_params;
 304     while (p)
 305     {
 306         if (p->name != 0)
 307         {
libscriptapi->set_variable(p->name, p->val, (p->range == 1), (p->data_type == DTYPE_TABLE), p->option_count, p->options);
 309         }
p = p->next;
 311     }
conf_update_prevent_shutdown();
old_gui_handler = gui_set_mode(&scriptGuiHandler);
 317     return script_stack_start();
 318 }
 320 //-------------------------------------------------------------------