#include "luascript.h" #include "kbd.h" #include "platform.h" #include "script.h" #include "lua.h" #include "lualib.h" #include "lauxlib.h" #include "conf.h" #include "shot_histogram.h" #include "stdlib.h" #include "raw.h" #include "raw_merge.h" #include "levent.h" #include "console.h" #include "action_stack.h" #include "motion_detector.h" #include "ptp.h" #include "core.h" #include "gui_fselect.h" #include "gui_lang.h" #include "../lib/lua/lstate.h" // for L->nCcalls, baseCcalls lua_State* L; lua_State* Lt; static int lua_script_is_ptp; static int yield_hook_enabled; static void lua_script_disable_yield_hook(void) { yield_hook_enabled = 0; } static void lua_script_enable_yield_hook(void) { yield_hook_enabled = 1; } #ifdef CAM_CHDK_PTP // create a ptp message from the given stack index // incompatible types will return a TYPE_UNSUPPORTED message static ptp_script_msg *lua_create_usb_msg( lua_State* L, int index, unsigned msgtype) { // TODO maybe we should just pass the lua type constants unsigned datatype, datasize = 4; int ivalue = 0; void *data = &ivalue; int ltype = lua_type(L,index); switch(ltype) { case LUA_TNONE: return NULL; // nothing on the stack, no message generated break; case LUA_TNIL: datatype = PTP_CHDK_TYPE_NIL; break; case LUA_TBOOLEAN: datatype = PTP_CHDK_TYPE_BOOLEAN; ivalue = lua_toboolean(L,index); break; case LUA_TNUMBER: datatype = PTP_CHDK_TYPE_INTEGER; ivalue = lua_tonumber(L,index); break; case LUA_TSTRING: datatype = PTP_CHDK_TYPE_STRING; data = (char *)lua_tolstring(L,index,&datasize); break; // TODO this uses usb_msg_table_to_string to serialize the table // the default format is described in // http://chdk.setepontos.com/index.php?topic=4338.msg62606#msg62606 // other formats can be implemented by overriding this function in your lua code case LUA_TTABLE: { int result; lua_script_disable_yield_hook(); // don't want to yield while converting lua_getglobal(L, "usb_msg_table_to_string"); // push function lua_pushvalue(L, index); // copy specified index to top of stack result = lua_pcall(L,1,1,0); // this will leave an error message as a string on the stack if call fails lua_script_enable_yield_hook(); if( result ) { // if called from lua, throw a normal error if( msgtype == PTP_CHDK_S_MSGTYPE_USER ) { luaL_error(L,lua_tostring(L,-1)); return NULL; // not reached } else { // if it's a return, convert the message to an ERR msgtype = PTP_CHDK_S_MSGTYPE_ERR; datatype = PTP_CHDK_S_ERRTYPE_RUN; data = (char *)lua_tolstring(L,-1,&datasize); break; } } // an empty table is returned as an empty string by default // a non-string should never show up here if ( !lua_isstring(L,-1) ) { return NULL; } datatype = PTP_CHDK_TYPE_TABLE; data = (char *)lua_tolstring(L,-1,&datasize); lua_pop(L,1); } break; default: datatype = PTP_CHDK_TYPE_UNSUPPORTED; data = (char *)lua_typename(L,ltype); // return type name as message data datasize = strlen(data); } return ptp_script_create_msg(msgtype,datatype,datasize,data); } void lua_script_error_ptp(int runtime, const char *err) { if(runtime) { ptp_script_write_error_msg(PTP_CHDK_S_ERRTYPE_RUN, err); script_end(); } else { ptp_script_write_error_msg(PTP_CHDK_S_ERRTYPE_COMPILE, err); lua_script_reset(); } } #endif void lua_script_reset() { lua_close( L ); L = 0; } static void lua_count_hook(lua_State *L, lua_Debug *ar) { if( L->nCcalls <= L->baseCcalls && yield_hook_enabled ) lua_yield( L, 0 ); } void lua_script_error(lua_State *Lt,int runtime) { const char *err = lua_tostring( Lt, -1 ); script_console_add_line( err ); if(lua_script_is_ptp) { #ifdef CAM_CHDK_PTP lua_script_error_ptp(runtime,err); #endif } else { if(runtime) { if(conf.debug_lua_restart_on_error) { lua_script_reset(); script_start_gui(0); } else { script_wait_and_end(); } } else { script_print_screen_end(); script_wait_and_end(); } } } // TODO more stuff from script.c should be moved here void lua_script_finish(lua_State *L) { #ifdef CAM_CHDK_PTP if(lua_script_is_ptp) { // send all return values as RET messages int i,end = lua_gettop(L); for(i=1;i<=end; i++) { ptp_script_msg *msg = lua_create_usb_msg(L,i,PTP_CHDK_S_MSGTYPE_RET); // if the queue is full return values will be silently discarded // incompatible types will be returned as TYPE_UNSUPPORTED to preserve expected number and order of return values if(msg) { ptp_script_write_msg(msg); // create_usb_msg may convert the message to an error if(msg->type != PTP_CHDK_S_MSGTYPE_RET) { break; } } else { ptp_script_write_error_msg(PTP_CHDK_S_ERRTYPE_RUN, "error creating return msg"); break; } } } #endif } int lua_script_start( char const* script, int ptp ) { lua_script_is_ptp = ptp; L = lua_open(); luaL_openlibs( L ); register_lua_funcs( L ); Lt = lua_newthread( L ); lua_setfield( L, LUA_REGISTRYINDEX, "Lt" ); if( luaL_loadstring( Lt, script ) != 0 ) { lua_script_error(Lt,0); return 0; } lua_sethook(Lt, lua_count_hook, LUA_MASKCOUNT, 1000 ); lua_script_enable_yield_hook(); return 1; } // run the "restore" function at the end of a script void lua_run_restore() { lua_getglobal(Lt, "restore"); if (lua_isfunction(Lt, -1)) { if (lua_pcall( Lt, 0, 0, 0 )) { script_console_add_line( lua_tostring( Lt, -1 ) ); } } } // get key ID of key name at arg, throw error if invalid static int lua_get_key_arg( lua_State * L, int narg ) { int k = script_keyid_by_name( luaL_checkstring( L, narg ) ); if(!k) luaL_error( L, "unknown key" ); return k; } #ifdef OPT_CURVES #include "curves.h" static int luaCB_set_curve_state( lua_State* L ) { int value; value=luaL_checknumber( L, 1 ); curve_set_mode(value); return 0; } #endif static int luaCB_set_aflock(lua_State* L) { int val = luaL_checknumber(L, 1); if (val>0) DoAFLock(); // 1: enable AFLock else UnlockAF(); // 0: disable unlock AF return 0; } static int luaCB_shoot( lua_State* L ) { action_push(AS_SHOOT); return lua_yield( L, 0 ); } static int luaCB_sleep( lua_State* L ) { action_push_delay( luaL_checknumber( L, 1 ) ); return lua_yield( L, 0 ); } // for press,release and click static int luaCB_keyfunc( lua_State* L ) { void* func = lua_touserdata( L, lua_upvalueindex(1) ); ((void(*)(long))func)( lua_get_key_arg( L, 1 ) ); return lua_yield( L, 0 ); } static int luaCB_cls( lua_State* L ) { console_clear(); return 0; } static int luaCB_set_console_layout( lua_State* L ) { console_set_layout(luaL_checknumber( L, 1 ),luaL_checknumber( L, 2 ),luaL_checknumber( L, 3 ),luaL_checknumber( L, 4 )); return 0; } static int luaCB_set_console_autoredraw( lua_State* L ) { console_set_autoredraw(luaL_checknumber(L,1)); return 0; } static int luaCB_console_redraw( lua_State* L ) { console_redraw(); return 0; } static int luaCB_get_av96( lua_State* L ) { lua_pushnumber( L, shooting_get_av96() ); return 1; } static int luaCB_get_bv96( lua_State* L ) { lua_pushnumber( L, shooting_get_bv96() ); return 1; } static int luaCB_get_day_seconds( lua_State* L ) { lua_pushnumber( L, shooting_get_day_seconds() ); return 1; } static int luaCB_get_disk_size( lua_State* L ) { lua_pushnumber( L, GetTotalCardSpaceKb() ); return 1; } static int luaCB_get_dof( lua_State* L ) { lua_pushnumber( L, shooting_get_depth_of_field() ); return 1; } static int luaCB_get_far_limit( lua_State* L ) { lua_pushnumber( L, shooting_get_far_limit_of_acceptable_sharpness() ); return 1; } static int luaCB_get_free_disk_space( lua_State* L ) { lua_pushnumber( L, GetFreeCardSpaceKb() ); return 1; } static int luaCB_get_focus( lua_State* L ) { lua_pushnumber( L, shooting_get_subject_distance() ); return 1; } static int luaCB_get_hyp_dist( lua_State* L ) { lua_pushnumber( L, shooting_get_hyperfocal_distance() ); return 1; } static int luaCB_get_iso_market( lua_State* L ) { lua_pushnumber( L, shooting_get_iso_market() ); return 1; } static int luaCB_get_iso_mode( lua_State* L ) { lua_pushnumber( L, shooting_get_iso_mode() ); return 1; } static int luaCB_get_iso_real( lua_State* L ) { lua_pushnumber( L, shooting_get_iso_real() ); return 1; } static int luaCB_get_jpg_count( lua_State* L ) { lua_pushnumber( L, GetJpgCount() ); return 1; } static int luaCB_get_near_limit( lua_State* L ) { lua_pushnumber( L, shooting_get_near_limit_of_acceptable_sharpness() ); return 1; } /* val=get_prop(id) get propcase value identified by id the propcase is read as a short and sign extended to an int */ static int luaCB_get_prop( lua_State* L ) { lua_pushnumber( L, shooting_get_prop( luaL_checknumber( L, 1 ) ) ); return 1; } /* val=get_prop_str(prop_id,length) get the value of a propertycase as a string numeric values may be extracted using string.byte or or the binstr.lua module returns the value as a string, or false if the underlying propcase call returned non-zero */ static int luaCB_get_prop_str( lua_State* L ) { void *buf; unsigned size; unsigned prop_id = luaL_checknumber( L, 1 ); size = luaL_checknumber( L, 2 ); buf = malloc(size); if(!buf) { return luaL_error( L, "malloc failed in luaCB_get_prop" ); } if(get_property_case(prop_id,buf,size) == 0) { lua_pushlstring( L, buf, size ); } else { lua_pushboolean( L, 0); } free(buf); return 1; } static int luaCB_get_raw_count( lua_State* L ) { lua_pushnumber( L, GetRawCount() ); return 1; } static int luaCB_get_sv96( lua_State* L ) { lua_pushnumber( L, shooting_get_sv96() ); return 1; } static int luaCB_get_tick_count( lua_State* L ) { lua_pushnumber( L, shooting_get_tick_count() ); return 1; } static int luaCB_get_exp_count( lua_State* L ) { lua_pushnumber( L, get_exposure_counter() ); return 1; } static int luaCB_get_tv96( lua_State* L ) { lua_pushnumber( L, shooting_get_tv96() ); return 1; } static int luaCB_get_user_av_id( lua_State* L ) { lua_pushnumber( L, shooting_get_user_av_id() ); return 1; } static int luaCB_get_user_av96( lua_State* L ) { lua_pushnumber( L, shooting_get_user_av96() ); return 1; } static int luaCB_get_user_tv_id( lua_State* L ) { lua_pushnumber( L, shooting_get_user_tv_id() ); return 1; } static int luaCB_get_user_tv96( lua_State* L ) { lua_pushnumber( L, shooting_get_user_tv96() ); return 1; } static int luaCB_get_vbatt( lua_State* L ) { lua_pushnumber( L, stat_get_vbatt() ); return 1; } static int luaCB_get_zoom( lua_State* L ) { lua_pushnumber( L, shooting_get_zoom() ); return 1; } static int luaCB_get_parameter_data( lua_State* L ) { extern long* FlashParamsTable[]; unsigned size; unsigned id = luaL_checknumber( L, 1 ); unsigned val; if (id >= get_flash_params_count()) { // return nil return 0; } size = FlashParamsTable[id][1]>>16; if (size == 0) { // return nil return 0; } if (size >= 1 && size <= 4) { val = 0; get_parameter_data( id, &val, size ); lua_pushlstring( L, (char *)&val, size ); // for convenience, params that fit in a number are returned in one as a second result lua_pushnumber( L, val ); return 2; } else { char *buf = malloc(size); if(!buf) { luaL_error( L, "malloc failed in luaCB_get_parameter_data" ); } get_parameter_data( id, buf, size ); lua_pushlstring( L, buf, size ); free(buf); return 1; } } static int luaCB_get_flash_params_count( lua_State* L ) { lua_pushnumber( L, get_flash_params_count() ); return 1; } static int luaCB_set_av96_direct( lua_State* L ) { shooting_set_av96_direct( luaL_checknumber( L, 1 ), SET_LATER ); return 0; } static int luaCB_set_av96( lua_State* L ) { shooting_set_av96( luaL_checknumber( L, 1 ), SET_LATER ); return 0; } static int luaCB_set_focus( lua_State* L ) { int to = luaL_checknumber( L, 1 ); int m=mode_get()&MODE_SHOOTING_MASK; int mode_video=MODE_IS_VIDEO(m); #if CAM_HAS_MANUAL_FOCUS if (shooting_get_focus_mode() || (mode_video)) shooting_set_focus(to, SET_NOW); else shooting_set_focus(to, SET_LATER); #else if (mode_video) shooting_set_focus(to, SET_NOW); else shooting_set_focus(to, SET_LATER); #endif return 0; } static int luaCB_set_iso_mode( lua_State* L ) { shooting_set_iso_mode( luaL_checknumber( L, 1 ) ); return 0; } static int luaCB_set_iso_real( lua_State* L ) { shooting_set_iso_real( luaL_checknumber( L, 1 ), SET_LATER); return 0; } static int luaCB_set_led( lua_State* L ) { int to, to1, to2; to = luaL_checknumber( L, 1 ); to1 = luaL_checknumber( L, 2 ); to2 = 200; if( lua_isnumber( L, 3 ) ) to2 = lua_tonumber( L, 3 ); camera_set_led(to, to1, to2); return 0; } static int luaCB_set_nd_filter( lua_State* L ) { shooting_set_nd_filter_state( luaL_checknumber( L, 1 ), SET_LATER); return 0; } /* set_prop(id,value) the value is treated as a short */ static int luaCB_set_prop( lua_State* L ) { shooting_set_prop(luaL_checknumber( L, 1 ), luaL_checknumber( L, 2 )); return 0; } /* status=set_prop_str(prop_id,value) set propertycase value as a string. Length is taken from the string numeric propcase values may be assembled by setting byte values using string.char or the binstr module status: boolean - true if the underlying propcase call returns 0, otherwise false */ static int luaCB_set_prop_str( lua_State *L ) { int prop_id; unsigned len; const char *str; prop_id = luaL_checknumber( L, 1 ); str = luaL_checklstring( L, 2, &len ); if(str && len > 0) { lua_pushboolean( L, (set_property_case(prop_id,(void *)str,len) == 0)); } else { return luaL_error( L, "invalid value"); } return 1; } static int luaCB_set_raw_nr( lua_State* L ) { camera_set_nr(luaL_checknumber( L, 1 )); return 0; } static int luaCB_get_raw_nr( lua_State* L ) { lua_pushnumber( L, camera_get_nr() ); return 1; } static int luaCB_set_raw( lua_State* L ) { camera_set_raw(luaL_checknumber( L, 1 )); return 0; } static int luaCB_get_raw( lua_State* L ) { lua_pushnumber( L, conf.save_raw ); return 1; } static int luaCB_set_sv96( lua_State* L ) { shooting_set_sv96(luaL_checknumber( L, 1 ), SET_LATER); return 0; } static int luaCB_set_tv96_direct( lua_State* L ) { shooting_set_tv96_direct(luaL_checknumber( L, 1 ), SET_LATER); return 0; } static int luaCB_set_tv96( lua_State* L ) { shooting_set_tv96(luaL_checknumber( L, 1 ), SET_LATER); return 0; } static int luaCB_set_user_av_by_id_rel( lua_State* L ) { shooting_set_user_av_by_id_rel(luaL_checknumber( L, 1 )); return 0; } static int luaCB_set_user_av_by_id( lua_State* L ) { shooting_set_user_av_by_id(luaL_checknumber( L, 1 )); return 0; } static int luaCB_set_user_av96( lua_State* L ) { shooting_set_user_av96(luaL_checknumber( L, 1 )); return 0; } static int luaCB_set_user_tv_by_id_rel( lua_State* L ) { shooting_set_user_tv_by_id_rel(luaL_checknumber( L, 1 )); return 0; } static int luaCB_set_user_tv_by_id( lua_State* L ) { shooting_set_user_tv_by_id(luaL_checknumber( L, 1 )); return 0; } static int luaCB_set_user_tv96( lua_State* L ) { shooting_set_user_tv96(luaL_checknumber( L, 1 )); return 0; } static int luaCB_set_zoom_speed( lua_State* L ) { shooting_set_zoom_speed(luaL_checknumber( L, 1 )); return 0; } static int luaCB_set_zoom_rel( lua_State* L ) { shooting_set_zoom_rel(luaL_checknumber( L, 1 )); return 0; } static int luaCB_set_zoom( lua_State* L ) { shooting_set_zoom(luaL_checknumber( L, 1 )); return 0; } static int luaCB_wait_click( lua_State* L ) { int timeout = luaL_optnumber( L, 1, 0 ); action_wait_for_click(timeout); return lua_yield( L, 0 ); } static int luaCB_is_pressed( lua_State* L ) { lua_pushboolean( L, script_key_is_pressed(lua_get_key_arg( L, 1 ))); return 1; } static int luaCB_is_key( lua_State* L ) { lua_pushboolean( L, script_key_is_clicked(lua_get_key_arg( L, 1 ))); return 1; } #if CAM_HAS_JOGDIAL static int luaCB_wheel_right( lua_State* L ) { JogDial_CW(); return 0; } static int luaCB_wheel_left( lua_State* L ) { JogDial_CCW(); return 0; } #endif static int luaCB_md_get_cell_diff( lua_State* L ) { lua_pushnumber( L, md_get_cell_diff(luaL_checknumber(L,1), luaL_checknumber(L,2))); return 1; } static int luaCB_md_detect_motion( lua_State* L ) { int columns = (luaL_optnumber(L,1,6)); int rows = (luaL_optnumber(L,2,4)); int pixel_measure_mode = (luaL_optnumber(L,3,1)); int detection_timeout = (luaL_optnumber(L,4,10000)); int measure_interval = (luaL_optnumber(L,5,7)); int threshold = (luaL_optnumber(L,6,10)); int draw_grid = (luaL_optnumber(L,7,1)); // arg 8 is the return value in ubasic. We // ignore it here. - AUJ int clipping_region_mode = (luaL_optnumber(L,9,0)); int clipping_region_column1 = (luaL_optnumber(L,10,0)); int clipping_region_row1 = (luaL_optnumber(L,11,0)); int clipping_region_column2 = (luaL_optnumber(L,12,0)); int clipping_region_row2 = (luaL_optnumber(L,13,0)); int parameters = (luaL_optnumber(L,14,1)); int pixels_step = (luaL_optnumber(L,15,6)); int msecs_before_trigger = (luaL_optnumber(L,16,0)); if(md_init_motion_detector( columns, rows, pixel_measure_mode, detection_timeout, measure_interval, threshold, draw_grid, clipping_region_mode, clipping_region_column1, clipping_region_row1, clipping_region_column2, clipping_region_row2, parameters, pixels_step, msecs_before_trigger )) return lua_yield(L, 0); else return luaL_error( L, "md_init_motion_detector failed" ); } static void file_browser_selected(const char *fn) { // Reconnect button input to script - will also signal action stack // that file browser is finished and return last selected file // to script caller state_kbd_script_run = 1; } static int luaCB_file_browser( lua_State* L ) { // Disconnect button input from script so buttons will work in file browser state_kbd_script_run = 0; // Push file browser action onto stack - will loop doing nothing until file browser exits action_push(AS_FILE_BROWSER); // Switch to file browser gui mode. Path can be supplied in call or defaults to "A" (root directory). gui_fselect_init(LANG_STR_FILE_BROWSER, luaL_optstring( L, 1, "A" ), "A", file_browser_selected); // Yield the script so that the action stack will process the AS_FILE_BROWSER action return lua_yield(L, 0); } static int luaCB_autostarted( lua_State* L ) { lua_pushboolean( L, camera_get_script_autostart() ); return 1; } static int luaCB_get_autostart( lua_State* L ) { lua_pushnumber( L, conf.script_startup ); return 1; } static int luaCB_set_autostart( lua_State* L ) { int to; to = luaL_checknumber( L, 1 ); if ( to >= 0 && to <= 2 ) conf.script_startup = to; conf_save(); return 0; } static int luaCB_get_usb_power( lua_State* L ) { if (luaL_optnumber( L, 1, 0 )) lua_pushnumber( L, get_usb_power(1) ); else lua_pushnumber( L, get_usb_power(0) ); return 1; } static int luaCB_exit_alt( lua_State* L ) { exit_alt(); return 0; } // optional parameter is 0 for soft shutdown (default) or 1 for hard/immediate static int luaCB_shut_down( lua_State* L ) { if ( luaL_optnumber(L,1,0) == 1 ) { shutdown(); } else { camera_shutdown_in_a_second(); } return 0; } static int luaCB_print_screen( lua_State* L ) { if (lua_isboolean( L, 1 )) script_print_screen_statement( lua_toboolean( L, 1 ) ); else script_print_screen_statement( luaL_checknumber( L, 1 )+10000 ); return 0; } static int luaCB_get_movie_status( lua_State* L ) { lua_pushnumber( L, movie_status ); return 1; } static int luaCB_set_movie_status( lua_State* L ) { int to; switch(luaL_checknumber( L, 1 )) { case 1: if (movie_status == 4) { movie_status = 1; } break; case 2: if (movie_status == 1) { movie_status = 4; } break; case 3: if (movie_status == 1 || movie_status == 4) { movie_status = 5; } break; } return 0; } static int luaCB_get_drive_mode( lua_State* L ) { lua_pushnumber( L, shooting_get_drive_mode() ); return 1; } static int luaCB_get_focus_mode( lua_State* L ) { lua_pushnumber( L, shooting_get_prop(PROPCASE_FOCUS_MODE) ); return 1; } static int luaCB_get_flash_mode( lua_State* L ) { lua_pushnumber( L, shooting_get_prop(PROPCASE_FLASH_MODE) ); return 1; } static int luaCB_get_shooting( lua_State* L ) { lua_pushboolean( L, shooting_get_prop(PROPCASE_SHOOTING) ); return 1; } static int luaCB_get_flash_ready( lua_State* L ) { lua_pushboolean( L, shooting_get_prop(PROPCASE_IS_FLASH_READY) ); return 1; } static int luaCB_get_IS_mode( lua_State* L ) { lua_pushnumber( L, shooting_get_prop(PROPCASE_IS_MODE) ); return 1; } static int luaCB_get_orientation_sensor( lua_State* L ) { lua_pushnumber( L, shooting_get_prop(PROPCASE_ORIENTATION_SENSOR) ); return 1; } static int luaCB_get_zoom_steps( lua_State* L ) { lua_pushnumber( L, zoom_points ); return 1; } static int luaCB_get_nd_present( lua_State* L ) { int to; #if !CAM_HAS_ND_FILTER to = 0; #endif #if CAM_HAS_ND_FILTER && !CAM_HAS_IRIS_DIAPHRAGM to = 1; #endif #if CAM_HAS_ND_FILTER && CAM_HAS_IRIS_DIAPHRAGM to = 2; #endif lua_pushnumber( L, to ); return 1; } static int luaCB_get_propset( lua_State* L ) { lua_pushnumber( L, CAM_PROPSET ); return 1; } static int luaCB_get_ev( lua_State* L ) { lua_pushnumber( L, shooting_get_prop(PROPCASE_EV_CORRECTION_1) ); return 1; } static int luaCB_set_ev( lua_State* L ) { int to; to = luaL_checknumber( L, 1 ); shooting_set_prop(PROPCASE_EV_CORRECTION_1, to); shooting_set_prop(PROPCASE_EV_CORRECTION_2, to); return 0; } static int luaCB_get_histo_range( lua_State* L ) { int from = (luaL_checknumber(L,1)); int to = (luaL_checknumber(L,2)); if (shot_histogram_isenabled()) lua_pushnumber( L, shot_histogram_get_range(from, to) ); else lua_pushnumber( L, -1 ); // TODO should probably return nil return 1; } static int luaCB_shot_histo_enable( lua_State* L ) { shot_histogram_set(luaL_checknumber( L, 1 )); return 0; } static int luaCB_play_sound( lua_State* L ) { play_sound(luaL_checknumber( L, 1 )); return 0; } static int luaCB_get_temperature( lua_State* L ) { int which = (luaL_checknumber( L, 1 )); int temp = -100; // do something insane if users passes bad value switch (which) { case 0: temp = get_optical_temp(); break; case 1: temp = get_ccd_temp(); break; case 2: temp = get_battery_temp(); break; } lua_pushnumber( L, temp ); return 1; } static int luaCB_get_time( lua_State* L ) { int r = -1; unsigned long t2 = time(NULL); static struct tm *ttm; ttm = localtime(&t2); const char *t = luaL_checkstring( L, 1 ); if (strncmp("s", t, 1)==0) r = ( L, ttm->tm_sec ); else if (strncmp("m", t, 1)==0) r = ( L, ttm->tm_min ); else if (strncmp("h", t, 1)==0) r = ( L, ttm->tm_hour ); else if (strncmp("D", t, 1)==0) r = ( L, ttm->tm_mday ); else if (strncmp("M", t, 1)==0) r = ( L, ttm->tm_mon+1 ); else if (strncmp("Y", t, 1)==0) r = ( L, 1900+ttm->tm_year ); lua_pushnumber( L, r ); return 1; } /* val=peek(address[,size]) return the value found at address in memory, or nil if address or size is invalid size is optional 1=byte 2=halfword 4=word. defaults is 4 */ static int luaCB_peek( lua_State* L ) { unsigned addr = luaL_checknumber(L,1); unsigned size = luaL_optnumber(L, 2, 4); switch(size) { case 1: lua_pushnumber( L, *(unsigned char *)(addr) ); break; case 2: if (addr & 0x1) { lua_pushnil(L); } else { lua_pushnumber( L, *(unsigned short *)(addr) ); } break; case 4: if (addr & 0x3) { lua_pushnil(L); } else { lua_pushnumber( L, *(unsigned *)(addr) ); } break; default: lua_pushnil(L); } return 1; } /* status=poke(address,value[,size]) writes value to address in memory size is optional 1=byte 2=halfword 4=word. defaults is 4 returns true, or nil if address or size is invalid */ static int luaCB_poke( lua_State* L ) { unsigned addr = luaL_checknumber(L,1); unsigned val = luaL_checknumber(L,2); unsigned size = luaL_optnumber(L, 3, 4); int status = 0; switch(size) { case 1: *(unsigned char *)(addr) = (unsigned char)val; status=1; break; case 2: if (!(addr & 0x1)) { *(unsigned short *)(addr) = (unsigned short)val; status=1; } break; case 4: if (!(addr & 0x3)) { *(unsigned *)(addr) = val; status=1; } break; } if(status) { lua_pushboolean(L,1); } else { lua_pushnil(L); } return 1; } static int luaCB_bitand( lua_State* L ) { int v1 = (luaL_checknumber(L,1)); int v2 = (luaL_checknumber(L,2)); lua_pushnumber( L, v1 & v2 ); return 1; } static int luaCB_bitor( lua_State* L ) { int v1 = (luaL_checknumber(L,1)); int v2 = (luaL_checknumber(L,2)); lua_pushnumber( L, v1 | v2 ); return 1; } static int luaCB_bitxor( lua_State* L ) { int v1 = (luaL_checknumber(L,1)); int v2 = (luaL_checknumber(L,2)); lua_pushnumber( L, v1 ^ v2 ); return 1; } static int luaCB_bitshl( lua_State* L ) { int val = (luaL_checknumber(L,1)); unsigned shift = (luaL_checknumber(L,2)); lua_pushnumber( L, val << shift ); return 1; } static int luaCB_bitshri( lua_State* L ) { int val = (luaL_checknumber(L,1)); unsigned shift = (luaL_checknumber(L,2)); lua_pushnumber( L, val >> shift ); return 1; } static int luaCB_bitshru( lua_State* L ) { unsigned val = (luaL_checknumber(L,1)); unsigned shift = (luaL_checknumber(L,2)); lua_pushnumber( L, val >> shift ); return 1; } static int luaCB_bitnot( lua_State* L ) { unsigned val = (luaL_checknumber(L,1)); lua_pushnumber( L, ~val ); return 1; } static void set_string_field(lua_State* L, const char *key, const char *val) { lua_pushstring(L, val); lua_setfield(L, -2, key); } static void set_number_field(lua_State* L, const char *key, int val) { lua_pushnumber(L, val); lua_setfield(L, -2, key); } static int luaCB_get_buildinfo( lua_State* L ) { lua_createtable(L, 0, 8); set_string_field( L,"platform", PLATFORM ); set_string_field( L,"platsub", PLATFORMSUB ); set_string_field( L,"version", HDK_VERSION ); set_string_field( L,"build_number", BUILD_NUMBER ); set_string_field( L,"build_date", __DATE__ ); set_string_field( L,"build_time", __TIME__ ); #ifndef CAM_DRYOS set_string_field( L,"os", "vxworks" ); #else set_string_field( L,"os", "dryos" ); #endif set_number_field( L, "platformid", PLATFORMID ); return 1; } static int luaCB_get_mode( lua_State* L ) { int m = mode_get(); lua_pushboolean( L, (m&MODE_MASK) != MODE_PLAY ); lua_pushboolean( L, MODE_IS_VIDEO(m) ); lua_pushnumber( L, m ); return 3; } // TODO sanity check file ? static int luaCB_set_raw_develop( lua_State* L ) { raw_prepare_develop(luaL_optstring( L, 1, NULL )); return 0; } static int luaCB_raw_merge_start( lua_State* L ) { int op = luaL_checknumber(L,1); if (op == RAW_OPERATION_SUM || op == RAW_OPERATION_AVERAGE) { raw_merge_start(op); } else { return luaL_argerror(L,1,"invalid raw merge op"); } return 0; } // TODO sanity check file ? static int luaCB_raw_merge_add_file( lua_State* L ) { raw_merge_add_file(luaL_checkstring( L, 1 )); return 0; } static int luaCB_raw_merge_end( lua_State* L ) { raw_merge_end(); return 0; } // Enable/disable LCD back light (input argument 1/0) static int luaCB_set_backlight( lua_State* L ) { int val = (luaL_checknumber(L,1)); if (val > 0) TurnOnBackLight(); else TurnOffBackLight(); return 0; } // get the string or number passed in index and return it as an event id static unsigned levent_id_from_lua_arg( lua_State* L, int index) { unsigned event_id; if (lua_type(L, index) == LUA_TSTRING) { const char *ev_name = lua_tostring(L, index); event_id = levent_id_for_name(ev_name); if (event_id == 0) { return luaL_error( L, "bad event name '%s'", ev_name ); } } // could check here if it is in the table, but even valid ones can crash // so we avoid searching the table if given a number else if (lua_type(L,index) == LUA_TNUMBER){ event_id = lua_tonumber(L,index); } else { return luaL_error( L, "expected event name or id" ); } return event_id; } /* get a value where boolean or 0/!0 are accepted for on/off. normal lua toboolean will convert 0 to true, but ubasic and c users will expect 0 to be off intentional HACK: numbers greater than 1 are returned as is */ static unsigned on_off_value_from_lua_arg( lua_State* L, int index) { if( lua_isboolean(L,index) ) { return lua_toboolean(L,index); } else { return luaL_checknumber(L,index); } } /* return the index of an event, given it's name or event id */ static unsigned levent_index_from_id_lua_arg( lua_State* L, int index ) { if (lua_type(L, index) == LUA_TSTRING) { return levent_index_for_name(lua_tostring(L, index)); } else if (lua_type(L,index) == LUA_TNUMBER){ return levent_index_for_id(lua_tonumber(L,index)); } else { return luaL_error( L, "expected string or number" ); } } /* name,id,param = get_levent_def(event) event is an event id (number) or name (string) returns nil if event is not found */ static int luaCB_get_levent_def( lua_State* L ) { unsigned event_index = levent_index_from_id_lua_arg(L,1); if (event_index == LEVENT_INVALID_INDEX) { lua_pushnil(L); return 1; } lua_pushstring(L, levent_table[event_index].name); lua_pushnumber(L, levent_table[event_index].id); lua_pushnumber(L, levent_table[event_index].param); return 3; } /* index=get_levent_index(event) event is an event id (number) or name (string) returns index or nil if not found */ static int luaCB_get_levent_index( lua_State* L ) { unsigned event_index = levent_index_from_id_lua_arg(L,1); if (event_index == LEVENT_INVALID_INDEX) { lua_pushnil(L); } else { lua_pushnumber(L, event_index); } return 1; } /* name,id,param = get_levent_def_by_index(event_index) event_index is number index into the event table returns nil if event is not found */ static int luaCB_get_levent_def_by_index( lua_State* L ) { unsigned i = luaL_checknumber(L,1); if(i >= levent_count()) { lua_pushnil(L); return 1; } lua_pushstring(L, levent_table[i].name); lua_pushnumber(L, levent_table[i].id); lua_pushnumber(L, levent_table[i].param); return 3; } /* post_levent_*(event[,unk]) post the event with PostLogicalEventToUI or PostLogicaEventForNotPowerType This sends the event. The difference between functions isn't clear. event is an event id (number) or name (string). unk is an optional number whose meaning is unknown, defaults to zero. Based on code, other values would probably be a pointer. This is NOT the 3rd item in the event table. */ static int luaCB_post_levent_to_ui( lua_State* L ) { unsigned event_id,arg; event_id = levent_id_from_lua_arg(L,1); arg = luaL_optnumber(L, 2, 0); PostLogicalEventToUI(event_id,arg); return 0; } static int luaCB_post_levent_for_npt( lua_State* L ) { unsigned event_id,arg; event_id = levent_id_from_lua_arg(L,1); arg = luaL_optnumber(L, 2, 0); PostLogicalEventForNotPowerType(event_id,arg); return 0; } /* set_levent_active(event,state) event is an event id (number) or name (string) state is a numeric or boolean state. true or non zero numbers turn on zero, false or nil turn off exact meaning is unknown, but it has something to do with the delivery of the specified event. */ static int luaCB_set_levent_active( lua_State* L ) { unsigned event_id; unsigned state; event_id = levent_id_from_lua_arg(L,1); state = on_off_value_from_lua_arg(L,2); SetLogicalEventActive(event_id,state); return 0; } /* set_levent_script_mode(state) state is numeric or boolean state. true or non zero numbers turn on zero, false or nil turn off exact meaning is unknown, but it has something to do with the behavior of events and/or SetLogicalEventActive. */ static int luaCB_set_levent_script_mode( lua_State* L ) { SetScriptMode(on_off_value_from_lua_arg(L,1)); return 0; } /* result=set_capture_mode_canon(value) where value is a valid PROPCASE_SHOOTING_MODE value for the current camera result is true if the camera is in rec mode */ static int luaCB_set_capture_mode_canon( lua_State* L ) { int modenum = luaL_checknumber(L,1); // if the value as negative, assume it is a mistakenly sign extended PROPCASE_SHOOTING_MODE value if(modenum < 0) modenum &= 0xFFFF; lua_pushboolean( L, shooting_set_mode_canon(modenum) ); return 1; } /* result=set_capture_mode(modenum) where modenum is a valid CHDK modemap value result is true if modenum is a valid modemap value, otherwise false */ static int luaCB_set_capture_mode( lua_State* L ) { int modenum = luaL_checknumber(L,1); lua_pushboolean( L, shooting_set_mode_chdk(modenum) ); return 1; } /* result=is_capture_mode_valid(modenum) where modenum is a valid CHDK modemap value result is true if modenum is a valid modemap value, otherwise false */ static int luaCB_is_capture_mode_valid( lua_State* L ) { int modenum = luaL_checknumber(L,1); lua_pushboolean( L, shooting_mode_chdk2canon(modenum) != -1 ); return 1; } /* set_record(state) if state is 0 (or false) the camera is set to play mode. If 1 or true, the camera is set to record mode. NOTE: this only begins the mode change. Script should wait until get_mode() reflects the change, before doing anything that requires the new mode. e.g. set_record(true) while not get_mode() do sleep(10) end */ static int luaCB_set_record( lua_State* L ) { if(on_off_value_from_lua_arg(L,1)) { levent_set_record(); } else { levent_set_play(); } return 0; } // switch mode (0 = playback, 1 = record) // only for when USB is connected static int luaCB_switch_mode_usb( lua_State* L ) { int mode = luaL_checknumber(L,1); if ( mode != 0 && mode != 1 ) { return 0; } return switch_mode_usb(mode); } #ifdef CAM_CHDK_PTP // PTP Live View functions // Function used to get viewport, bitmap and palette data via PTP // Address of this function sent back to client program which then // calls this with options to determine what to transfer static int handle_video_transfer(ptp_data *data, int flags, int arg2) { int total_size; // Calculated total size of data to transfer to client // Structure containing the info for the current live view frame // This information may change across calls struct { int vp_xoffset; // Viewport X offset in pixels (for cameras with variable image size) int vp_yoffset; // Viewpoer Y offset in pixels (for cameras with variable image size) int vp_width; // Actual viewport width in pixels (for cameras with variable image size) int vp_height; // Actual viewport height in pixels (for cameras with variable image size) int vp_buffer_start; // Offset in data transferred where the viewport data starts int vp_buffer_size; // Size of viewport data sent (in bytes) int bm_buffer_start; // Offset in data transferred where the bitmap data starts int bm_buffer_size; // Size of bitmap data sent (in bytes) int palette_type; // Camera palette type // (0 = no palette, 1 = 16 x 4 byte AYUV values, 2 = 16 x 4 byte AYUV values with A = 0..3, 3 = 256 x 4 byte AYUV values with A = 0..3) int palette_buffer_start; // Offset in data transferred where the palette data starts int palette_buffer_size; // Size of palette data sent (in bytes) } vid_info; // Populate the above structure with the current default details vid_info.vp_xoffset = vid_get_viewport_xoffset_proper(); vid_info.vp_yoffset = vid_get_viewport_yoffset_proper(); vid_info.vp_width = vid_get_viewport_width_proper(); vid_info.vp_height = vid_get_viewport_height_proper(); vid_info.vp_buffer_start = 0; vid_info.vp_buffer_size = 0; vid_info.bm_buffer_start = 0; vid_info.bm_buffer_size = 0; vid_info.palette_type = vid_get_palette_type(); vid_info.palette_buffer_start = 0; vid_info.palette_buffer_size = 0; total_size = sizeof(vid_info); // Add viewport details if requested if ( flags & 0x1 ) // live buffer { vid_info.vp_buffer_start = total_size; vid_info.vp_buffer_size = (vid_get_viewport_buffer_width_proper()*vid_get_viewport_max_height()*6)/4; total_size += vid_info.vp_buffer_size; } // Add bitmap details if requested if ( flags & 0x4 ) // bitmap buffer { vid_info.bm_buffer_start = total_size; vid_info.bm_buffer_size = vid_get_bitmap_buffer_width()*vid_get_bitmap_screen_height(); total_size += vid_info.bm_buffer_size; } // Add palette detals if requested if ( flags & 0x8 ) // bitmap palette { vid_info.palette_buffer_start = total_size; vid_info.palette_buffer_size = vid_get_palette_size(); total_size += vid_info.palette_buffer_size; } // Send header structure (along with total size to be sent) data->send_data(data->handle,(char*)&vid_info,sizeof(vid_info),total_size,0,0,0); // Send viewport data if requested if ( flags & 0x1 ) { data->send_data(data->handle,vid_get_viewport_active_buffer(),vid_info.vp_buffer_size,0,0,0,0); } // Send bitmap data if requested if ( flags & 0x4 ) { data->send_data(data->handle,vid_get_bitmap_active_buffer(),vid_info.bm_buffer_size,0,0,0,0); } // Send palette data if requested if ( flags & 0x8 ) { data->send_data(data->handle,vid_get_bitmap_active_palette(),vid_info.palette_buffer_size,0,0,0,0); } return 0; } // Lua function to return base info for PTP live view, including address of above transfer function static int luaCB_get_video_details( lua_State* L ) { // Structure to popualate with live view details // These details are static and only need to be retrieved once struct { int transfer_function; // Address of transfer function above int vp_max_width; // Maximum viewport width (in pixels) int vp_max_height; // Maximum viewport height (in pixels) int vp_buffer_width; // Viewport buffer width in case buffer is wider than visible viewport (in pixels) int bm_max_width; // Maximum width of bitmap (in pixels) int bm_max_height; // Maximum height of bitmap (in pixels) int bm_buffer_width; // Bitmap buffer width in case buffer is wider than visible bitmap (in pixels) int lcd_aspect_ratio; // 0 = 4:3, 1 = 16:9 } details; // Populate structure info details.transfer_function = (int) handle_video_transfer; details.vp_max_width = vid_get_viewport_max_width(); details.vp_max_height = vid_get_viewport_max_height(); details.vp_buffer_width = vid_get_viewport_buffer_width_proper(); #if CAM_USES_ASPECT_CORRECTION details.bm_max_width = ASPECT_XCORRECTION(vid_get_bitmap_screen_width()); #else details.bm_max_width = vid_get_bitmap_screen_width(); #endif details.bm_max_height = vid_get_bitmap_screen_height(); details.bm_buffer_width = vid_get_bitmap_buffer_width(); details.lcd_aspect_ratio = vid_get_aspect_ratio(); // Send data back to client lua_pushlstring( L, (char *) &details, sizeof(details) ); return 1; } #endif /* pack the lua args into a buffer to pass to the native code calling functions currently only handles strings/numbers start is the stack index of the first arg */ #ifdef OPT_LUA_CALL_NATIVE static int pack_native_args( lua_State* L, unsigned start, unsigned *argbuf) { unsigned i; unsigned end = lua_gettop(L); for(i = start; i <= end; i++,argbuf++) { if (lua_type(L, i) == LUA_TSTRING) { *argbuf=(unsigned)lua_tostring( L, i); } else if (lua_type(L, i) == LUA_TNUMBER) { *argbuf=lua_tonumber( L, i); } else { return 0; } } return 1; } /* Native function call interface. Can be used to call canon eventprocs or arbitrary pointers. NOTE: this is preliminary, interface may change in later versions! All arguments must be strings or numbers. If the function expects to modify it's arguments via a pointer, then you must provide a number that is a valid pointer. You can use the "AllocateMemory" eventproc to obtain buffers. If the function tries to write to a string passed from lua, Bad Things may happen. This is potentially dangerous, functions exist which can destroy the onboard firmware. */ /* result=call_func_ptr(ptr,...) ptr: address of a valid ARM or Thumb function, which uses the normal C calling convention. result: R0 value after the call returns */ static int luaCB_call_func_ptr( lua_State* L) { unsigned *argbuf=NULL; unsigned i; unsigned n_args = lua_gettop(L)-1; void *fptr; fptr=(void *)luaL_checknumber( L, 1 ); if (n_args) { argbuf=malloc(n_args * 4); if(!argbuf) { return luaL_error( L, "malloc fail" ); } if(!pack_native_args(L, 2, argbuf)) { free(argbuf); return luaL_error( L, "expected string or number" ); } } lua_pushnumber( L, call_func_ptr(fptr, argbuf, n_args) ); free(argbuf); return 1; } /* Call an event procedure result=call_event_proc("EventprocName",...) result is the value returned by ExecuteEventProcedure, which is -1 if the eventproc is not found, or the eventproc return value (which could also be -1) NOTE: Many eventprocs are not registered by default, but can be loaded by calling another event proc Some useful ones are SystemEventInit includes AllocateMemory, FreeMemory, sprintf, memcpy, Fut functions, log ... UI_RegistDebugEventProc includes capture mode functions, PTM_ functions and much more RegisterProductTestEvent includes PT_ functions Others: RegisterShootSeqEvent RegisterNRTableEvent */ // grab from lowlevel extern unsigned _ExecuteEventProcedure(const char *name,...); static int luaCB_call_event_proc( lua_State* L ) { const char *evpname; unsigned *argbuf; unsigned i; unsigned n_args = lua_gettop(L); evpname=luaL_checkstring( L, 1 ); argbuf=malloc(n_args * 4); if (!argbuf) { return luaL_error( L, "malloc fail" ); } // event proc name is first arg *argbuf = (unsigned)evpname; if(!pack_native_args(L,2,argbuf+1)) { free(argbuf); return luaL_error( L, "expected string or number" ); } lua_pushnumber( L, call_func_ptr(_ExecuteEventProcedure,argbuf,n_args) ); free(argbuf); return 1; } #endif // OPT_LUA_CALL_NATIVE /* result = reboot(["filename"]) returns false on failure, does not return on success see lib/armutil/reboot.c for details */ static int luaCB_reboot( lua_State* L ) { lua_pushboolean(L, reboot(luaL_optstring( L, 1, NULL ))); return 1; } static int luaCB_get_config_value( lua_State* L ) { unsigned int argc = lua_gettop(L); unsigned int id, i; int ret = 1; tConfigVal configVal; if( argc>=1 ) { id = luaL_checknumber(L, 1); switch( conf_getValue(id, &configVal) ) { case CONF_VALUE: lua_pushnumber(L, configVal.numb); break; case CONF_INT_PTR: lua_createtable(L, 0, configVal.numb); for( i=0; i=2) { //Default ret = argc-1; } else { lua_pushnil(L); } break; } } else { lua_pushnil(L); } return ret; } static int luaCB_set_config_value( lua_State* L ) { unsigned int argc = lua_gettop(L); unsigned int id, i, j; tConfigVal configVal = {0,0,0,0}; //initialize isXXX if( argc>=2 ) { id = luaL_checknumber(L, 1); for( i=2; i<=argc; i++) { switch( lua_type(L, i) ) { case LUA_TNUMBER: if( !configVal.isNumb ) { configVal.numb = luaL_checknumber(L, i); configVal.isNumb++; } switch( configVal.isPos ) { case 0: configVal.pos.x = luaL_checknumber(L, i); configVal.isPos++; break; case 1: configVal.pos.y = luaL_checknumber(L, i); configVal.isPos++; break; } break; case LUA_TSTRING: if( !configVal.isStr ) { configVal.str = (char*)luaL_checkstring(L, i); configVal.isStr++; } break; case LUA_TTABLE: if( !configVal.isPInt ) { configVal.numb = lua_objlen(L, i); if( configVal.pInt ) { free(configVal.pInt); configVal.pInt = NULL; } configVal.pInt = malloc(configVal.numb*sizeof(int)); if( configVal.pInt ) { for( j=1; j<=configVal.numb; j++) { lua_rawgeti(L, i, j); configVal.pInt[j-1] = lua_tointeger(L, -1); lua_pop(L, 1); } } configVal.isPInt++; } break; } } lua_pushboolean(L, conf_setValue(id, configVal)); if( configVal.pInt ) { free(configVal.pInt); configVal.pInt = NULL; } } else lua_pushboolean(L, 0); return 1; } #ifdef CAM_CHDK_PTP /* msg = read_usb_msg([timeout]) read a message from the CHDK ptp interface. Returns the next available message as a string, or nil if no messages are available If timeout is given and not zero, wait until a message is available or timeout expires */ static int luaCB_read_usb_msg( lua_State* L ) { int timeout = luaL_optnumber( L, 1, 0 ); if(timeout) { action_push(timeout); action_push(AS_SCRIPT_READ_USB_MSG); return lua_yield( L, 0 ); } ptp_script_msg *msg = ptp_script_read_msg(); if(msg) { lua_pushlstring(L,msg->data,msg->size); return 1; } lua_pushnil(L); return 1; } /* status = write_usb_msg(msg,[timeout]) writes a message to the CHDK ptp interface msg may be nil, boolean, number, string or table (table has some restrictions, will be converted to string) returns true if the message was queued successfully, otherwise false if timeout is set and not zero, wait until message is written or timeout expires NOTE strings will not include a terminating NULL, must be handled by recipient */ static int luaCB_write_usb_msg( lua_State* L ) { ptp_script_msg *msg; int timeout = luaL_optnumber( L, 2, 0 ); // TODO would it be better to either ignore this or return nil ? // a write_usb_msg(function_which_returns_no_value()) is an error in this case // replacing with nil might be more luaish if(lua_gettop(L) < 1) { return luaL_error(L,"missing argument"); } msg=lua_create_usb_msg(L,1,PTP_CHDK_S_MSGTYPE_USER); // for user messages, trying to create a message from an incompatible type throws an error if(msg->subtype == PTP_CHDK_TYPE_UNSUPPORTED) { free(msg); return luaL_error(L,"unsupported type"); } if(!msg) { return luaL_error(L,"failed to create message"); } if(timeout) { action_push(timeout); action_push((int)msg); action_push(AS_SCRIPT_WRITE_USB_MSG); return lua_yield( L, 0 ); } lua_pushboolean(L,ptp_script_write_msg(msg)); return 1; } #endif /* helper for meminfo to set table field only if valid */ static void set_meminfo_num( lua_State* L,const char *name, int val) { if(val != -1) { set_number_field( L, name, val ); } } /* meminfo=get_meminfo([heapname]) get camera memory information heapname="system" or "exmem" if not given, meminfo is returned for heap used by CHDK for malloc meminfo is false if the requested heapname isn't valid ("exmem" when exmem is not enabled, or unknown) otherwise, a table of the form meminfo = { name -- string "system" or "exmem" chdk_malloc -- bool, this is the heap used by CHDK for malloc chdk_start -- number, load address of CHDK chdk_size -- number, size of CHDK image -- all the following are numbers, will not be set if not available start_address end_address total_size allocated_size allocated_peak allocated_count free_size free_block_max_size free_block_count } NOTES * under vxworks and cameras without GetMemInfo only the only valid fields for the system heap will be those defined by chdk and free_block_max_size * the meaning of fields may not correspond exactly between exmem and system */ static int luaCB_get_meminfo( lua_State* L ) { // for memory info, duplicated from lowlevel extern const char _start,_end; #if defined(OPT_EXMEM_MALLOC) && !defined(OPT_EXMEM_TESTING) const char *default_heapname="exmem"; #else const char *default_heapname="system"; #endif const char *heapname = luaL_optstring( L, 1, default_heapname ); cam_meminfo meminfo; if(strcmp(heapname,"system") == 0) { #if defined(CAM_FIRMWARE_MEMINFO) GetMemInfo(&meminfo); #else memset(&meminfo,0xFF,sizeof(cam_meminfo)); meminfo.free_block_max_size = core_get_free_memory(); #endif } #if defined(OPT_EXMEM_MALLOC) && !defined(OPT_EXMEM_TESTING) else if(strcmp(heapname,"exmem") == 0) { GetExMemInfo(&meminfo); meminfo.allocated_count = -1; // not implemented in suba } #endif else { lua_pushboolean(L,0); return 1; } // adjust start and size, if CHDK is loaded at heap start if(meminfo.start_address == (int)(&_start)) { meminfo.start_address += MEMISOSIZE; meminfo.total_size -= MEMISOSIZE; } lua_createtable(L, 0, 13); // might not always use 13, but doesn't hurt set_string_field( L,"name", heapname ); lua_pushboolean( L, (strcmp(heapname,default_heapname)==0)); lua_setfield(L, -2, "chdk_malloc"); set_number_field( L, "chdk_start", (int)(&_start) ); set_number_field( L, "chdk_size", MEMISOSIZE ); set_meminfo_num( L, "start_address", meminfo.start_address ); set_meminfo_num( L, "end_address", meminfo.end_address); set_meminfo_num( L, "total_size", meminfo.total_size); set_meminfo_num( L, "allocated_size", meminfo.allocated_size); set_meminfo_num( L, "allocated_peak", meminfo.allocated_peak); set_meminfo_num( L, "allocated_count", meminfo.allocated_count); set_meminfo_num( L, "free_size", meminfo.free_size); set_meminfo_num( L, "free_block_max_size", meminfo.free_block_max_size); set_meminfo_num( L, "free_block_count", meminfo.free_block_count); return 1; } static void register_func( lua_State* L, const char *name, void *func) { lua_pushcfunction( L, func ); lua_setglobal( L, name ); } #define FUNC( X ) { #X, luaCB_##X }, static const luaL_Reg chdk_funcs[] = { FUNC(shoot) FUNC(sleep) FUNC(cls) FUNC(set_console_layout) FUNC(set_console_autoredraw) FUNC(console_redraw) FUNC(get_av96) FUNC(get_av96) FUNC(get_bv96) FUNC(get_day_seconds) FUNC(get_disk_size) FUNC(get_dof) FUNC(get_far_limit) FUNC(get_free_disk_space) FUNC(get_focus) FUNC(get_hyp_dist) FUNC(get_iso_market) FUNC(get_iso_mode) FUNC(get_iso_real) FUNC(get_jpg_count) FUNC(get_near_limit) FUNC(get_prop) FUNC(get_prop_str) FUNC(get_raw_count) FUNC(get_raw_nr) FUNC(get_raw) FUNC(get_sv96) FUNC(get_tick_count) FUNC(get_tv96) FUNC(get_user_av_id) FUNC(get_user_av96) FUNC(get_user_tv_id) FUNC(get_user_tv96) FUNC(get_vbatt) FUNC(get_zoom) FUNC(get_exp_count) FUNC(get_flash_params_count) FUNC(get_parameter_data) FUNC(set_av96_direct) FUNC(set_av96) FUNC(set_focus) FUNC(set_iso_mode) FUNC(set_iso_real) FUNC(set_led) FUNC(set_nd_filter) FUNC(set_prop) FUNC(set_prop_str) FUNC(set_raw_nr) FUNC(set_raw) FUNC(set_sv96) FUNC(set_tv96_direct) FUNC(set_tv96) FUNC(set_user_av_by_id_rel) FUNC(set_user_av_by_id) FUNC(set_user_av96) FUNC(set_user_tv_by_id_rel) FUNC(set_user_tv_by_id) FUNC(set_user_tv96) FUNC(set_zoom_speed) FUNC(set_zoom_rel) FUNC(set_zoom) FUNC(wait_click) FUNC(is_pressed) FUNC(is_key) #ifdef CAM_HAS_JOGDIAL FUNC(wheel_right) FUNC(wheel_left) #endif FUNC(md_get_cell_diff) FUNC(md_detect_motion) FUNC(autostarted) FUNC(get_autostart) FUNC(set_autostart) FUNC(get_usb_power) FUNC(exit_alt) FUNC(shut_down) FUNC(print_screen) FUNC(get_focus_mode) FUNC(get_propset) FUNC(get_zoom_steps) FUNC(get_drive_mode) FUNC(get_flash_mode) FUNC(get_shooting) FUNC(get_flash_ready) FUNC(get_IS_mode) FUNC(set_ev) FUNC(get_ev) FUNC(get_orientation_sensor) FUNC(get_nd_present) FUNC(get_movie_status) FUNC(set_movie_status) FUNC(get_histo_range) FUNC(shot_histo_enable) FUNC(play_sound) FUNC(get_temperature) FUNC(peek) FUNC(poke) FUNC(bitand) FUNC(bitor) FUNC(bitxor) FUNC(bitshl) FUNC(bitshri) FUNC(bitshru) FUNC(bitnot) FUNC(get_time) FUNC(get_buildinfo) FUNC(get_mode) FUNC(set_raw_develop) // NOTE these functions normally run in the spytask. // called from lua they will run from kbd task instead FUNC(raw_merge_start) FUNC(raw_merge_add_file) FUNC(raw_merge_end) FUNC(set_backlight) FUNC(set_aflock) #ifdef OPT_CURVES FUNC(set_curve_state) #endif // get levent definition by name or id, nil if not found FUNC(get_levent_def) // get levent definition by index, nil if out of range FUNC(get_levent_def_by_index) // get levent index from name or ID FUNC(get_levent_index) FUNC(post_levent_to_ui) FUNC(post_levent_for_npt) FUNC(set_levent_active) FUNC(set_levent_script_mode) FUNC(set_capture_mode) FUNC(set_capture_mode_canon) FUNC(is_capture_mode_valid) FUNC(set_record) FUNC(switch_mode_usb) #ifdef CAM_CHDK_PTP FUNC(get_video_details) #endif #ifdef OPT_LUA_CALL_NATIVE FUNC(call_event_proc) FUNC(call_func_ptr) #endif FUNC(reboot) FUNC(get_config_value) FUNC(set_config_value) #ifdef CAM_CHDK_PTP FUNC(read_usb_msg) FUNC(write_usb_msg) #endif FUNC(get_meminfo) FUNC(file_browser) {NULL, NULL}, }; void register_lua_funcs( lua_State* L ) { const luaL_reg *r; lua_pushlightuserdata( L, action_push_click ); lua_pushcclosure( L, luaCB_keyfunc, 1 ); lua_setglobal( L, "click" ); lua_pushlightuserdata( L, action_push_press ); lua_pushcclosure( L, luaCB_keyfunc, 1 ); lua_setglobal( L, "press" ); lua_pushlightuserdata( L, action_push_release ); lua_pushcclosure( L, luaCB_keyfunc, 1 ); lua_setglobal( L, "release" ); for(r=chdk_funcs;r->name;r++) { lua_pushcfunction( L, r->func ); lua_setglobal( L, r->name ); } #ifdef CAM_CHDK_PTP luaL_dostring(L,"function usb_msg_table_to_string(t)" " local v2s=function(v)" " local t=type(v)" " if t=='string' then return v end" " if t=='number' or t=='boolean' or t=='nil' then return tostring(v) end" " return '' end" " local r=''" " for k,v in pairs(t) do" " local s,vs=''" " if type(v)=='table' then" " for i=1,table.maxn(v) do" " s=s..'\\t'..v2s(v[i]) end" " else" " vs=v2s(v)" " if #vs then s=s..'\\t'..vs end" " end" " vs=v2s(k)" " if #vs>0 and #s>0 then r=r..vs..s..'\\n' end" " end" " return r" " end"); #endif }