From b8b1cbbdb253e1fae2ae457c8a7581d4a846c17b Mon Sep 17 00:00:00 2001 From: Giovanni Di Grezia Date: Thu, 1 Sep 2016 19:22:15 +0200 Subject: [PATCH] Added xg 2d curve editor in single version --- .../xg_2d_curve_editor_adj.py | 796 ++++++++++++++++++ 1 file changed, 796 insertions(+) create mode 100644 single_versions_for_3de_online_db_scripts/xg_2d_curve_editor_adj.py diff --git a/single_versions_for_3de_online_db_scripts/xg_2d_curve_editor_adj.py b/single_versions_for_3de_online_db_scripts/xg_2d_curve_editor_adj.py new file mode 100644 index 0000000..d0a645c --- /dev/null +++ b/single_versions_for_3de_online_db_scripts/xg_2d_curve_editor_adj.py @@ -0,0 +1,796 @@ +# +# +# 3DE4.script.name: Adjust 2D Curve Local Filter +# +# 3DE4.script.version: v1.1 +# +# 3DE4.script.comment: Adjustment of 2d Points positions using 1D Fourier filter on different frames ranges +# +# 3DE4.script.gui: Main Window::Adjustment +# +# 3DE4.script.hide: false +# 3DE4.script.startup: false +# +# Giovanni Di Grezia 2016 +# http://www.xgiovio.com +# + +if tde4.getCurrentScriptCallHint() != "CALL_ADJUSTMENT": + + #### ensure that the script can find itself when it create a parameter adjustment + + import os, platform + if platform.system() == "Windows": + script_base = os.getenv("APPDATA") + "/.3dequalizer/py_scripts/" + else: + script_base = os.path.expanduser("~") + "/.3dequalizer/py_scripts/" + + found = False + if os.path.exists(script_base): + files = os.listdir(script_base) + for file in files: + if ".py" in file : + searchfile = open(script_base + file, "r") + for line in searchfile: + if "Adjust 2D Curve Local Filter" in line: + adj_script_name = file + found = True + break + searchfile.close() + if found : + break + if not found : + tde4.postQuestionRequester("Error","Script not found. Please copy 'Adjust 2D Curve Local Filter' script to " + script_base + " and close/reopen this window","OK") + + #### end check + + editor_width = 1200 + editor_height = 800 + + curvex = 0 + curvey = 0 + curvex_filtered_min = 0 + curvex_filtered_max = 0 + curvey_filtered_min = 0 + curvey_filtered_max = 0 + curvex_reloaded = 0 + curvey_reloaded = 0 + loaded_point = "" + loaded_cam = "" + import_status = False + + adj_parameters_list = [] + + first_time_range_adj = True + + + def callback_editor (req): + cam = tde4.getCurrentCamera() + tde4.setCurveAreaWidgetCursorPosition(req,"CurveAreaWidgetX",tde4.getCurrentFrame(cam),1) + tde4.setCurveAreaWidgetCursorPosition(req,"CurveAreaWidgetY",tde4.getCurrentFrame(cam),1) + + def list_adjs (req): + tde4.removeAllListWidgetItems(req,"ListAdj") + for i in range(len(adj_parameters_list)): + tde4.insertListWidgetItem(req,"ListAdj",adj_parameters_list[i],i) + + + def callback (req,label,event): + + global adj_parameters_list + global loaded_point + global loaded_cam + global import_status + global first_time_range_adj + + if label == "LoadCurve": + + cam = tde4.getCurrentCamera() + current_point_group = tde4.getCurrentPGroup() + selected_point = tde4.getPointList (current_point_group,1) + playback = tde4.getCameraPlaybackRange(cam) + firstframe = playback[0] + lastframe = playback[1] + + if first_time_range_adj: + tde4.setWidgetValue(req,"Range_Min",str(firstframe)) + tde4.setWidgetValue(req,"Range_Max",str(lastframe)) + first_time_range_adj = False + + + if len(selected_point) != 1 : + tde4.postQuestionRequester("Error","You need to select 1 2D point only","OK") + tde4.setWidgetSensitiveFlag(req,"FilterAllX",0) + tde4.setWidgetSensitiveFlag(req,"FilterAllY",0) + tde4.setWidgetSensitiveFlag(req,"FilterAllXAdd",0) + tde4.setWidgetSensitiveFlag(req,"FilterAllYAdd",0) + tde4.setWidgetSensitiveFlag(req,"FilterLocX",0) + tde4.setWidgetSensitiveFlag(req,"FilterLocY",0) + tde4.setWidgetSensitiveFlag(req,"FilterLocXAdd",0) + tde4.setWidgetSensitiveFlag(req,"FilterLocYAdd",0) + elif lastframe - firstframe < 2 : + tde4.postQuestionRequester("Error","You need a camera with at least 3 frames","OK") + tde4.setWidgetSensitiveFlag(req,"FilterAllX",0) + tde4.setWidgetSensitiveFlag(req,"FilterAllY",0) + tde4.setWidgetSensitiveFlag(req,"FilterAllXAdd",0) + tde4.setWidgetSensitiveFlag(req,"FilterAllYAdd",0) + tde4.setWidgetSensitiveFlag(req,"FilterLocX",0) + tde4.setWidgetSensitiveFlag(req,"FilterLocY",0) + tde4.setWidgetSensitiveFlag(req,"FilterLocXAdd",0) + tde4.setWidgetSensitiveFlag(req,"FilterLocYAdd",0) + else : + tde4.setWidgetSensitiveFlag(req,"FilterAllX",1) + tde4.setWidgetSensitiveFlag(req,"FilterAllY",1) + tde4.setWidgetSensitiveFlag(req,"FilterAllXAdd",1) + tde4.setWidgetSensitiveFlag(req,"FilterAllYAdd",1) + tde4.setWidgetSensitiveFlag(req,"FilterLocX",1) + tde4.setWidgetSensitiveFlag(req,"FilterLocY",1) + tde4.setWidgetSensitiveFlag(req,"FilterLocXAdd",1) + tde4.setWidgetSensitiveFlag(req,"FilterLocYAdd",1) + + name = tde4.getPointName(current_point_group,selected_point[0]) + tde4.setWidgetValue(req,"PointName",name) + tde4.setWidgetValue(req,"PointID",selected_point[0]) + + selected_point_positions = [] + for frame in range (firstframe, lastframe + 1): + if tde4.isPointPos2DValid(current_point_group,selected_point[0],cam,frame): + selected_point_position = tde4.getPointPosition2D(current_point_group,selected_point[0],cam,frame) + selected_point_positions.append([frame,selected_point_position[0],selected_point_position[1]]) + + + global curvex + global curvey + global curvex_filtered_min + global curvex_filtered_max + global curvey_filtered_min + global curvey_filtered_max + global curvex_reloaded + global curvey_reloaded + + + if curvex == 0 or loaded_point != selected_point[0] or loaded_cam!= cam: + + tde4.detachCurveAreaWidgetAllCurves(req,"CurveAreaWidgetX") + tde4.detachCurveAreaWidgetAllCurves(req,"CurveAreaWidgetY") + + + curvex = tde4.createCurve() + curvey = tde4.createCurve() + curvex_filtered_min = tde4.createCurve() + curvex_filtered_max = tde4.createCurve() + curvey_filtered_min = tde4.createCurve() + curvey_filtered_max = tde4.createCurve() + curvex_reloaded = 0 + curvey_reloaded = 0 + loaded_point = selected_point[0] + loaded_cam= cam + + tde4.removeAllListWidgetItems(req,"ListAdj") + adj_parameters_list = [] + + frame_min = 0 + frame_max = 0 + x_y_min = 0 + x_y_max = 0 + y_y_min = 0 + y_y_max = 0 + + first_x = True + first_y = True + for i in selected_point_positions: + c = tde4.createCurveKey(curvex,[i[0],i[1]]) + #tde4.setCurveKeyMode(curvex,c,"LINEAR") + #tde4.setCurveKeyFixedXFlag(curvex,c,1) + if first_x: + frame_min = i[0] + frame_max = i[0] + x_y_min = i[1] + x_y_max = i[1] + first_x = False + else: + if i[0] < frame_min: + frame_min = i[0] + if i[0] > frame_max: + frame_max = i[0] + if i[1] < x_y_min: + x_y_min = i[1] + if i[1] > x_y_max: + x_y_max = i[1] + + + c = tde4.createCurveKey(curvey,[i[0],i[2]]) + #tde4.setCurveKeyMode(curvey,c,"LINEAR") + #tde4.setCurveKeyFixedXFlag(curvey,c,1) + if first_y: + y_y_min = i[2] + y_y_max = i[2] + first_y = False + else: + if i[2] < y_y_min: + y_y_min = i[2] + if i[2] > y_y_max: + y_y_max = i[2] + + + tde4.attachCurveAreaWidgetCurve(req,"CurveAreaWidgetX",curvex,1,1,1,0) + tde4.attachCurveAreaWidgetCurve(req,"CurveAreaWidgetY",curvey,1,1,1,0) + + if not import_status: + if frame_min == frame_max: + tde4.setCurveAreaWidgetDimensions (req,"CurveAreaWidgetX",firstframe,lastframe,0,1) + tde4.setCurveAreaWidgetFOV (req,"CurveAreaWidgetX",firstframe,lastframe,0,1) + tde4.setCurveAreaWidgetDimensions (req,"CurveAreaWidgetY",firstframe,lastframe,0,1) + tde4.setCurveAreaWidgetFOV (req,"CurveAreaWidgetY",firstframe,lastframe,0,1) + else: + tde4.setCurveAreaWidgetDimensions (req,"CurveAreaWidgetX",frame_min,frame_max,x_y_min,x_y_max) + tde4.setCurveAreaWidgetFOV (req,"CurveAreaWidgetX",frame_min,frame_max,x_y_min,x_y_max) + tde4.setCurveAreaWidgetDimensions (req,"CurveAreaWidgetY",frame_min,frame_max,y_y_min,y_y_max) + tde4.setCurveAreaWidgetFOV (req,"CurveAreaWidgetY",frame_min,frame_max,y_y_min,y_y_max) + + else: + import_status = False + + tde4.setCurveAreaWidgetCursorPosition(req,"CurveAreaWidgetX",tde4.getCurrentFrame(cam),1) + tde4.setCurveAreaWidgetCursorPosition(req,"CurveAreaWidgetY",tde4.getCurrentFrame(cam),1) + else: + + curvex_reloaded_temp = tde4.createCurve() + curvey_reloaded_temp = tde4.createCurve() + + for i in selected_point_positions: + tde4.createCurveKey(curvex_reloaded_temp,[i[0],i[1]]) + tde4.createCurveKey(curvey_reloaded_temp,[i[0],i[2]]) + + + if curvex_reloaded!= 0 : + tde4.detachCurveAreaWidgetCurve(req,"CurveAreaWidgetX",curvex_reloaded) + curvex_reloaded = 0 + tde4.setWidgetSensitiveFlag(req,"RestoreCurve",0) + tde4.setWidgetSensitiveFlag(req,"ImportCurve",0) + + + if curvey_reloaded!= 0 : + tde4.detachCurveAreaWidgetCurve(req,"CurveAreaWidgetY",curvey_reloaded) + curvey_reloaded = 0 + tde4.setWidgetSensitiveFlag(req,"RestoreCurve",0) + tde4.setWidgetSensitiveFlag(req,"ImportCurve",0) + + pos_curvex = [] + pos_curvey = [] + pos_curvex_reloaded_temp=[] + pos_curvey_reloaded_temp=[] + + curvex_keys= tde4.getCurveKeyList(curvex,0) + curvey_keys= tde4.getCurveKeyList(curvey,0) + curvex_reloaded_temp_keys= tde4.getCurveKeyList(curvex_reloaded_temp,0) + curvey_reloaded_temp_keys= tde4.getCurveKeyList(curvey_reloaded_temp,0) + + for k in curvex_keys: + pos_curvex.append(tde4.getCurveKeyPosition(curvex,k)) + for k in curvey_keys: + pos_curvey.append(tde4.getCurveKeyPosition(curvey,k)) + for k in curvex_reloaded_temp_keys: + pos_curvex_reloaded_temp.append(tde4.getCurveKeyPosition(curvex_reloaded_temp,k)) + for k in curvey_reloaded_temp_keys: + pos_curvey_reloaded_temp.append(tde4.getCurveKeyPosition(curvey_reloaded_temp,k)) + + status = True + if len(pos_curvex) == len(pos_curvex_reloaded_temp): + for i in range(len(pos_curvex) ): + if pos_curvex[i] != pos_curvex_reloaded_temp[i]: + status = False + break + else: + status = False + if status == False: + curvex_reloaded = curvex_reloaded_temp + tde4.attachCurveAreaWidgetCurve(req,"CurveAreaWidgetX",curvex_reloaded,0,1,0,0) + tde4.setWidgetSensitiveFlag(req,"RestoreCurve",1) + tde4.setWidgetSensitiveFlag(req,"ImportCurve",1) + + status = True + if len(pos_curvey) == len(pos_curvey_reloaded_temp): + for i in range(len(pos_curvey) ): + if pos_curvey[i] != pos_curvey_reloaded_temp[i]: + status = False + break + else: + status = False + if status == False: + curvey_reloaded = curvey_reloaded_temp + tde4.attachCurveAreaWidgetCurve(req,"CurveAreaWidgetY",curvey_reloaded,0,1,0,0) + tde4.setWidgetSensitiveFlag(req,"RestoreCurve",1) + tde4.setWidgetSensitiveFlag(req,"ImportCurve",1) + + + + + elif label == "FilterAllX": + + if float(tde4.getWidgetValue(req,"FilterAll_Min")) > float(tde4.getWidgetValue(req,"FilterAll_Max")) : + tde4.postQuestionRequester("Error","Filter Min should be less than Filter Max","OK") + else: + if curvex_filtered_min!= 0 and curvex_filtered_max!= 0: + tde4.detachCurveAreaWidgetCurve(req,"CurveAreaWidgetX",curvex_filtered_min) + tde4.detachCurveAreaWidgetCurve(req,"CurveAreaWidgetX",curvex_filtered_max) + curvex_filtered_min = tde4.createCurve() + curvex_filtered_max = tde4.createCurve() + curvex_y_to_filter = [] + curvex_x = [] + curvex_keys= tde4.getCurveKeyList(curvex,0) + for k in curvex_keys: + pos = tde4.getCurveKeyPosition(curvex,k) + curvex_y_to_filter.append(pos[1]) + curvex_x.append(pos[0]) + + if len(curvex_y_to_filter) > 3: + curvex_y_filtered_min = tde4.filter1D(curvex_y_to_filter,float(tde4.getWidgetValue(req,"FilterAll_Min")),"FOURIER_FILTER") + curvex_y_filtered_max = tde4.filter1D(curvex_y_to_filter,float(tde4.getWidgetValue(req,"FilterAll_Max")),"FOURIER_FILTER") + else: + curvex_y_filtered_min = curvex_y_to_filter + curvex_y_filtered_max = curvex_y_to_filter + + for i in range(len(curvex_x)): + tde4.createCurveKey(curvex_filtered_min ,[curvex_x[i],curvex_y_filtered_min[i]]) + tde4.createCurveKey(curvex_filtered_max ,[curvex_x[i],curvex_y_filtered_max[i]]) + + + tde4.attachCurveAreaWidgetCurve(req,"CurveAreaWidgetX",curvex_filtered_min,1,0,0,0) + tde4.attachCurveAreaWidgetCurve(req,"CurveAreaWidgetX",curvex_filtered_max,0,0,1,0) + + elif label == "FilterAllY": + if float(tde4.getWidgetValue(req,"FilterAll_Min")) > float(tde4.getWidgetValue(req,"FilterAll_Max")) : + tde4.postQuestionRequester("Error","Filter Min should be less than Filter Max","OK") + else: + if curvey_filtered_min!= 0 and curvey_filtered_max!= 0: + tde4.detachCurveAreaWidgetCurve(req,"CurveAreaWidgetY",curvey_filtered_min) + tde4.detachCurveAreaWidgetCurve(req,"CurveAreaWidgetY",curvey_filtered_max) + + curvey_filtered_min = tde4.createCurve() + curvey_filtered_max = tde4.createCurve() + curvey_y_to_filter = [] + curvey_x = [] + curvey_keys= tde4.getCurveKeyList(curvey,0) + for k in curvey_keys: + pos = tde4.getCurveKeyPosition(curvey,k) + curvey_y_to_filter.append(pos[1]) + curvey_x.append(pos[0]) + + + if len(curvey_y_to_filter) > 3: + curvey_y_filtered_min = tde4.filter1D(curvey_y_to_filter,float(tde4.getWidgetValue(req,"FilterAll_Min")),"FOURIER_FILTER") + curvey_y_filtered_max = tde4.filter1D(curvey_y_to_filter,float(tde4.getWidgetValue(req,"FilterAll_Max")),"FOURIER_FILTER") + else: + curvey_y_filtered_min = curvey_y_to_filter + curvey_y_filtered_max = curvey_y_to_filter + + + for i in range(len(curvey_x)): + tde4.createCurveKey(curvey_filtered_min ,[curvey_x[i],curvey_y_filtered_min[i]]) + tde4.createCurveKey(curvey_filtered_max ,[curvey_x[i],curvey_y_filtered_max[i]]) + + tde4.attachCurveAreaWidgetCurve(req,"CurveAreaWidgetY",curvey_filtered_min,1,0,0,0) + tde4.attachCurveAreaWidgetCurve(req,"CurveAreaWidgetY",curvey_filtered_max,0,0,1,0) + + elif label == "FilterAllXAdd": + if float(tde4.getWidgetValue(req,"FilterAll_Min")) > float(tde4.getWidgetValue(req,"FilterAll_Max")) : + tde4.postQuestionRequester("Error","Filter Min should be less than Filter Max","OK") + else: + curvex_x = [] + curvex_keys= tde4.getCurveKeyList(curvex,0) + for k in curvex_keys: + pos = tde4.getCurveKeyPosition(curvex,k) + curvex_x.append(pos[0]) + adj_parameters_list.append(loaded_point + " " + "x" + " " + str(min(curvex_x)) + " " + str(max(curvex_x)) + " " + tde4.getWidgetValue(req,"FilterAll_Min") + " " + tde4.getWidgetValue(req,"FilterAll_Max") ) + list_adjs (req) + elif label == "FilterAllYAdd": + if float(tde4.getWidgetValue(req,"FilterAll_Min")) > float(tde4.getWidgetValue(req,"FilterAll_Max")) : + tde4.postQuestionRequester("Error","Filter Min should be less than Filter Max","OK") + else: + curvey_x = [] + curvey_keys= tde4.getCurveKeyList(curvey,0) + for k in curvey_keys: + pos = tde4.getCurveKeyPosition(curvey,k) + curvey_x.append(pos[0]) + adj_parameters_list.append(loaded_point + " " + "y" + " " + str(min(curvey_x)) + " " + str(max(curvey_x)) + " " + tde4.getWidgetValue(req,"FilterAll_Min") + " " + tde4.getWidgetValue(req,"FilterAll_Max") ) + list_adjs (req) + + elif label == "FilterLocX": + if float(tde4.getWidgetValue(req,"FilterLoc_Min")) > float(tde4.getWidgetValue(req,"FilterLoc_Max")) : + tde4.postQuestionRequester("Error","Filter Min should be less than Filter Max","OK") + elif float(tde4.getWidgetValue(req,"Range_Max")) - float(tde4.getWidgetValue(req,"Range_Min")) < 2: + tde4.postQuestionRequester("Error","Range Min Max should be at least 3 frames","OK") + else: + if curvex_filtered_min!= 0 and curvex_filtered_max!= 0: + tde4.detachCurveAreaWidgetCurve(req,"CurveAreaWidgetX",curvex_filtered_min) + tde4.detachCurveAreaWidgetCurve(req,"CurveAreaWidgetX",curvex_filtered_max) + curvex_filtered_min = tde4.createCurve() + curvex_filtered_max = tde4.createCurve() + curvex_y_to_filter = [] + curvex_x = [] + curvex_keys= tde4.getCurveKeyList(curvex,0) + for k in curvex_keys: + pos = tde4.getCurveKeyPosition(curvex,k) + if pos[0] >= float(tde4.getWidgetValue(req,"Range_Min")) and pos[0] <= float(tde4.getWidgetValue(req,"Range_Max")): + curvex_y_to_filter.append(pos[1]) + curvex_x.append(pos[0]) + + if len(curvex_y_to_filter) > 3: + curvex_y_filtered_min = tde4.filter1D(curvex_y_to_filter,float(tde4.getWidgetValue(req,"FilterLoc_Min")),"FOURIER_FILTER") + curvex_y_filtered_max = tde4.filter1D(curvex_y_to_filter,float(tde4.getWidgetValue(req,"FilterLoc_Max")),"FOURIER_FILTER") + else: + curvex_y_filtered_min = curvex_y_to_filter + curvex_y_filtered_max = curvex_y_to_filter + + for i in range(len(curvex_x)): + tde4.createCurveKey(curvex_filtered_min ,[curvex_x[i],curvex_y_filtered_min[i]]) + tde4.createCurveKey(curvex_filtered_max ,[curvex_x[i],curvex_y_filtered_max[i]]) + + + tde4.attachCurveAreaWidgetCurve(req,"CurveAreaWidgetX",curvex_filtered_min,1,0,0,0) + tde4.attachCurveAreaWidgetCurve(req,"CurveAreaWidgetX",curvex_filtered_max,0,0,1,0) + + elif label == "FilterLocY": + if float(tde4.getWidgetValue(req,"FilterLoc_Min")) > float(tde4.getWidgetValue(req,"FilterLoc_Max")) : + tde4.postQuestionRequester("Error","Filter Min should be less than Filter Max","OK") + elif float(tde4.getWidgetValue(req,"Range_Max")) - float(tde4.getWidgetValue(req,"Range_Min")) < 2: + tde4.postQuestionRequester("Error","Range Min Max should be at least 3 frames","OK") + else: + if curvey_filtered_min!= 0 and curvey_filtered_max!= 0: + tde4.detachCurveAreaWidgetCurve(req,"CurveAreaWidgetY",curvey_filtered_min) + tde4.detachCurveAreaWidgetCurve(req,"CurveAreaWidgetY",curvey_filtered_max) + + curvey_filtered_min = tde4.createCurve() + curvey_filtered_max = tde4.createCurve() + curvey_y_to_filter = [] + curvey_x = [] + curvey_keys= tde4.getCurveKeyList(curvey,0) + for k in curvey_keys: + pos = tde4.getCurveKeyPosition(curvey,k) + if pos[0] >= float(tde4.getWidgetValue(req,"Range_Min")) and pos[0] <= float(tde4.getWidgetValue(req,"Range_Max")): + curvey_y_to_filter.append(pos[1]) + curvey_x.append(pos[0]) + + if len(curvey_y_to_filter) > 3: + curvey_y_filtered_min = tde4.filter1D(curvey_y_to_filter,float(tde4.getWidgetValue(req,"FilterLoc_Min")),"FOURIER_FILTER") + curvey_y_filtered_max = tde4.filter1D(curvey_y_to_filter,float(tde4.getWidgetValue(req,"FilterLoc_Max")),"FOURIER_FILTER") + else: + curvey_y_filtered_min = curvey_y_to_filter + curvey_y_filtered_max = curvey_y_to_filter + + for i in range(len(curvey_x)): + tde4.createCurveKey(curvey_filtered_min ,[curvey_x[i],curvey_y_filtered_min[i]]) + tde4.createCurveKey(curvey_filtered_max ,[curvey_x[i],curvey_y_filtered_max[i]]) + + tde4.attachCurveAreaWidgetCurve(req,"CurveAreaWidgetY",curvey_filtered_min,1,0,0,0) + tde4.attachCurveAreaWidgetCurve(req,"CurveAreaWidgetY",curvey_filtered_max,0,0,1,0) + + elif label == "FilterLocXAdd": + if float(tde4.getWidgetValue(req,"FilterLoc_Min")) > float(tde4.getWidgetValue(req,"FilterLoc_Max")) : + tde4.postQuestionRequester("Error","Filter Min should be less than Filter Max","OK") + elif float(tde4.getWidgetValue(req,"Range_Max")) - float(tde4.getWidgetValue(req,"Range_Min")) < 2: + tde4.postQuestionRequester("Error","Range Min Max should be at least 3 frames","OK") + else: + adj_parameters_list.append(loaded_point + " " + "x" + " " + tde4.getWidgetValue(req,"Range_Min") + " " + tde4.getWidgetValue(req,"Range_Max") + " " + tde4.getWidgetValue(req,"FilterLoc_Min") + " " + tde4.getWidgetValue(req,"FilterLoc_Max") ) + list_adjs (req) + elif label == "FilterLocYAdd": + if float(tde4.getWidgetValue(req,"FilterLoc_Min")) > float(tde4.getWidgetValue(req,"FilterLoc_Max")) : + tde4.postQuestionRequester("Error","Filter Min should be less than Filter Max","OK") + elif float(tde4.getWidgetValue(req,"Range_Max")) - float(tde4.getWidgetValue(req,"Range_Min")) < 2: + tde4.postQuestionRequester("Error","Range Min Max should be at least 3 frames","OK") + else: + adj_parameters_list.append(loaded_point + " " + "y" + " " + tde4.getWidgetValue(req,"Range_Min") + " " +tde4.getWidgetValue(req,"Range_Max") + " " + tde4.getWidgetValue(req,"FilterLoc_Min") + " " + tde4.getWidgetValue(req,"FilterLoc_Max") ) + list_adjs (req) + elif label == "DeleteAdj": + for i in range(tde4.getListWidgetNoItems(req,"ListAdj") -1 , -1 , -1 ): + if tde4.getListWidgetItemSelectionFlag(req,"ListAdj",i) == 1: + userdata = tde4.getListWidgetItemUserdata(req,"ListAdj",i) + tde4.removeListWidgetItem(req,"ListAdj",i) + del adj_parameters_list [userdata] + elif label == "DeleteAdjAll": + tde4.removeAllListWidgetItems(req,"ListAdj") + adj_parameters_list = [] + elif label == "SendAdj": + + for i in range(tde4.getListWidgetNoItems(req,"ListAdj")): + if tde4.getListWidgetItemSelectionFlag(req,"ListAdj",i) == 1: + userdata = tde4.getListWidgetItemUserdata(req,"ListAdj",i) + param = adj_parameters_list [userdata].split() + pointid_curve_start_end = str(param[0]) + " " + str(param[1]) + " " + str(param[2]) + " " + str(param[3]) + filter_start = param[4] + filter_end = param[5] + tde4.createParameterAdjustScript(script_base + adj_script_name,"filter adj " + pointid_curve_start_end,pointid_curve_start_end ,float(filter_start),float(filter_end)) + elif label == "SendAdjAll": + for i in range(tde4.getListWidgetNoItems(req,"ListAdj")): + userdata = tde4.getListWidgetItemUserdata(req,"ListAdj",i) + param = adj_parameters_list [userdata].split() + pointid_curve_start_end = str(param[0]) + " " + str(param[1]) + " " + str(param[2]) + " " + str(param[3]) + filter_start = param[4] + filter_end = param[5] + tde4.createParameterAdjustScript(script_base + adj_script_name,"filter adj " + pointid_curve_start_end,pointid_curve_start_end ,float(filter_start),float(filter_end)) + + elif label == "RestoreCurve": + cam = tde4.getCurrentCamera() + current_point_group = tde4.getCurrentPGroup() + playback = tde4.getCameraPlaybackRange(cam) + firstframe = playback[0] + lastframe = playback[1] + + + pos_x = [] + pos_y = [] + curvex_keys= tde4.getCurveKeyList(curvex,0) + for k in curvex_keys: + pos_x.append(tde4.getCurveKeyPosition(curvex,k)) + curvey_keys= tde4.getCurveKeyList(curvey,0) + for k in curvey_keys: + pos_y.append(tde4.getCurveKeyPosition(curvey,k)) + + for i in range(len(pos_x)): + tde4.setPointPosition2D(current_point_group,loaded_point,cam,int(pos_x[i][0]),[pos_x[i][1],pos_y[i][1]]) + + callback(req,"ImportCurve","") + tde4.setWidgetSensitiveFlag(req,"RestoreCurve",0) + tde4.setWidgetSensitiveFlag(req,"ImportCurve",0) + + elif label == "ImportCurve": + loaded_point = "" + import_status = True + callback(req,"LoadCurve","") + tde4.setWidgetSensitiveFlag(req,"RestoreCurve",0) + tde4.setWidgetSensitiveFlag(req,"ImportCurve",0) + + + + + editor = tde4.createCustomRequester() + + + tde4.addCurveAreaWidget(editor,"CurveAreaWidgetX","X",0) + tde4.setWidgetAttachModes(editor,"CurveAreaWidgetX","ATTACH_WINDOW","ATTACH_WINDOW","ATTACH_WINDOW","ATTACH_WINDOW") + tde4.setWidgetOffsets(editor,"CurveAreaWidgetX",25,280,0,450) + + tde4.addCurveAreaWidget(editor,"CurveAreaWidgetY","Y",0) + tde4.setWidgetLinks(editor,"CurveAreaWidgetY","","","CurveAreaWidgetX","") + tde4.setWidgetAttachModes(editor,"CurveAreaWidgetY","ATTACH_WINDOW","ATTACH_WINDOW","ATTACH_WIDGET","ATTACH_WINDOW") + tde4.setWidgetOffsets(editor,"CurveAreaWidgetY",25,280,5,100) + + tde4.addListWidget(editor,"ListAdj","",1,0) + tde4.setWidgetLinks(editor,"ListAdj","CurveAreaWidgetX","","","") + tde4.setWidgetAttachModes(editor,"ListAdj","ATTACH_WIDGET","ATTACH_WINDOW","ATTACH_WINDOW","ATTACH_WINDOW") + tde4.setWidgetOffsets(editor,"ListAdj",5,5,0,155) + + + tde4.addButtonWidget(editor,"DeleteAdj","Delete",100,10) + tde4.setWidgetSize(editor,"DeleteAdj",50,50) + tde4.setWidgetLinks(editor,"DeleteAdj","CurveAreaWidgetY","","ListAdj","") + tde4.setWidgetAttachModes(editor,"DeleteAdj","ATTACH_WIDGET","ATTACH_AS_IS","ATTACH_WIDGET","ATTACH_AS_IS") + tde4.setWidgetCallbackFunction(editor,"DeleteAdj","callback") + + tde4.addButtonWidget(editor,"DeleteAdjAll","Delete All",100,10) + tde4.setWidgetSize(editor,"DeleteAdjAll",50,50) + tde4.setWidgetLinks(editor,"DeleteAdjAll","DeleteAdj","","ListAdj","") + tde4.setWidgetAttachModes(editor,"DeleteAdjAll","ATTACH_WIDGET","ATTACH_AS_IS","ATTACH_WIDGET","ATTACH_AS_IS") + tde4.setWidgetCallbackFunction(editor,"DeleteAdjAll","callback") + + tde4.addButtonWidget(editor,"SendAdj","Send Adj",100,10) + tde4.setWidgetSize(editor,"SendAdj",50,50) + tde4.setWidgetLinks(editor,"SendAdj","DeleteAdjAll","","ListAdj","") + tde4.setWidgetAttachModes(editor,"SendAdj","ATTACH_WIDGET","ATTACH_AS_IS","ATTACH_WIDGET","ATTACH_AS_IS") + tde4.setWidgetCallbackFunction(editor,"SendAdj","callback") + + tde4.addButtonWidget(editor,"SendAdjAll","Send Adj All",100,10) + tde4.setWidgetSize(editor,"SendAdjAll",80,50) + tde4.setWidgetLinks(editor,"SendAdjAll","SendAdj","","ListAdj","") + tde4.setWidgetAttachModes(editor,"SendAdjAll","ATTACH_WIDGET","ATTACH_AS_IS","ATTACH_WIDGET","ATTACH_AS_IS") + tde4.setWidgetCallbackFunction(editor,"SendAdjAll","callback") + + + + + tde4.addButtonWidget(editor,"LoadCurve","LoadCurve",100,10) + tde4.setWidgetSize(editor,"LoadCurve",100,50) + tde4.setWidgetLinks(editor,"LoadCurve","","","CurveAreaWidgetY","") + tde4.setWidgetAttachModes(editor,"LoadCurve","ATTACH_WINDOW","ATTACH_AS_IS","ATTACH_WIDGET","ATTACH_AS_IS") + tde4.setWidgetCallbackFunction(editor,"LoadCurve","callback") + + tde4.addButtonWidget(editor,"RestoreCurve","Restore",100,10) + tde4.setWidgetSize(editor,"RestoreCurve",100,50) + tde4.setWidgetLinks(editor,"RestoreCurve","LoadCurve","","CurveAreaWidgetY","") + tde4.setWidgetAttachModes(editor,"RestoreCurve","ATTACH_WIDGET","ATTACH_AS_IS","ATTACH_WIDGET","ATTACH_AS_IS") + tde4.setWidgetSensitiveFlag(editor,"RestoreCurve",0) + tde4.setWidgetCallbackFunction(editor,"RestoreCurve","callback") + + tde4.addButtonWidget(editor,"ImportCurve","Import",100,10) + tde4.setWidgetSize(editor,"ImportCurve",100,50) + tde4.setWidgetLinks(editor,"ImportCurve","RestoreCurve","","CurveAreaWidgetY","") + tde4.setWidgetAttachModes(editor,"ImportCurve","ATTACH_WIDGET","ATTACH_AS_IS","ATTACH_WIDGET","ATTACH_AS_IS") + tde4.setWidgetSensitiveFlag(editor,"ImportCurve",0) + tde4.setWidgetCallbackFunction(editor,"ImportCurve","callback") + + + tde4.addTextFieldWidget(editor,"FilterAll_Min","","0.1") + tde4.setWidgetLinks(editor,"FilterAll_Min","ImportCurve","ImportCurve","CurveAreaWidgetY","") + tde4.setWidgetAttachModes(editor,"FilterAll_Min","ATTACH_WIDGET","ATTACH_OPPOSITE_WIDGET","ATTACH_WIDGET","ATTACH_AS_IS") + tde4.setWidgetSize(editor,"FilterAll_Min",50,50) + tde4.setWidgetOffsets(editor,"FilterAll_Min",50,-80,5,0) + + tde4.addTextFieldWidget(editor,"FilterAll_Max","","2") + tde4.setWidgetLinks(editor,"FilterAll_Max","FilterAll_Min","FilterAll_Min","CurveAreaWidgetY","") + tde4.setWidgetAttachModes(editor,"FilterAll_Max","ATTACH_WIDGET","ATTACH_OPPOSITE_WIDGET","ATTACH_WIDGET","ATTACH_AS_IS") + tde4.setWidgetSize(editor,"FilterAll_Max",50,50) + tde4.setWidgetOffsets(editor,"FilterAll_Max",10,-40,5,0) + + + tde4.addButtonWidget(editor,"FilterAllX","Global Preview X",100,10) + tde4.setWidgetSize(editor,"FilterAllX",150,50) + tde4.setWidgetLinks(editor,"FilterAllX","FilterAll_Max","","CurveAreaWidgetY","") + tde4.setWidgetAttachModes(editor,"FilterAllX","ATTACH_WIDGET","ATTACH_AS_IS","ATTACH_WIDGET","ATTACH_AS_IS") + tde4.setWidgetSensitiveFlag(editor,"FilterAllX",0) + tde4.setWidgetCallbackFunction(editor,"FilterAllX","callback") + + tde4.addButtonWidget(editor,"FilterAllY","Y",100,10) + tde4.setWidgetSize(editor,"FilterAllY",20,50) + tde4.setWidgetLinks(editor,"FilterAllY","FilterAllX","","CurveAreaWidgetY","") + tde4.setWidgetAttachModes(editor,"FilterAllY","ATTACH_WIDGET","ATTACH_AS_IS","ATTACH_WIDGET","ATTACH_AS_IS") + tde4.setWidgetSensitiveFlag(editor,"FilterAllY",0) + tde4.setWidgetCallbackFunction(editor,"FilterAllY","callback") + + tde4.addButtonWidget(editor,"FilterAllXAdd","+X",100,10) + tde4.setWidgetSize(editor,"FilterAllXAdd",25,50) + tde4.setWidgetLinks(editor,"FilterAllXAdd","FilterAllY","","CurveAreaWidgetY","") + tde4.setWidgetAttachModes(editor,"FilterAllXAdd","ATTACH_WIDGET","ATTACH_AS_IS","ATTACH_WIDGET","ATTACH_AS_IS") + tde4.setWidgetSensitiveFlag(editor,"FilterAllXAdd",0) + tde4.setWidgetCallbackFunction(editor,"FilterAllXAdd","callback") + + tde4.addButtonWidget(editor,"FilterAllYAdd","+Y",100,10) + tde4.setWidgetSize(editor,"FilterAllYAdd",25,50) + tde4.setWidgetLinks(editor,"FilterAllYAdd","FilterAllXAdd","","CurveAreaWidgetY","") + tde4.setWidgetAttachModes(editor,"FilterAllYAdd","ATTACH_WIDGET","ATTACH_AS_IS","ATTACH_WIDGET","ATTACH_AS_IS") + tde4.setWidgetSensitiveFlag(editor,"FilterAllYAdd",0) + tde4.setWidgetCallbackFunction(editor,"FilterAllYAdd","callback") + + + + + tde4.addTextFieldWidget(editor,"FilterLoc_Min","","0.1") + tde4.setWidgetLinks(editor,"FilterLoc_Min","FilterAllYAdd","FilterAllYAdd","CurveAreaWidgetY","") + tde4.setWidgetAttachModes(editor,"FilterLoc_Min","ATTACH_WIDGET","ATTACH_OPPOSITE_WIDGET","ATTACH_WIDGET","ATTACH_AS_IS") + tde4.setWidgetSize(editor,"FilterLoc_Min",50,50) + tde4.setWidgetOffsets(editor,"FilterLoc_Min",60,-90,5,0) + + tde4.addTextFieldWidget(editor,"FilterLoc_Max","","2") + tde4.setWidgetLinks(editor,"FilterLoc_Max","FilterLoc_Min","FilterLoc_Min","CurveAreaWidgetY","") + tde4.setWidgetAttachModes(editor,"FilterLoc_Max","ATTACH_WIDGET","ATTACH_OPPOSITE_WIDGET","ATTACH_WIDGET","ATTACH_AS_IS") + tde4.setWidgetSize(editor,"FilterLoc_Max",50,50) + tde4.setWidgetOffsets(editor,"FilterLoc_Max",10,-40,5,0) + + + tde4.addButtonWidget(editor,"FilterLocX","Local Preview X",100,10) + tde4.setWidgetSize(editor,"FilterLocX",150,50) + tde4.setWidgetLinks(editor,"FilterLocX","FilterLoc_Max","","CurveAreaWidgetY","") + tde4.setWidgetAttachModes(editor,"FilterLocX","ATTACH_WIDGET","ATTACH_AS_IS","ATTACH_WIDGET","ATTACH_AS_IS") + tde4.setWidgetSensitiveFlag(editor,"FilterLocX",0) + tde4.setWidgetCallbackFunction(editor,"FilterLocX","callback") + + tde4.addButtonWidget(editor,"FilterLocY","Y",100,10) + tde4.setWidgetSize(editor,"FilterLocY",20,50) + tde4.setWidgetLinks(editor,"FilterLocY","FilterLocX","","CurveAreaWidgetY","") + tde4.setWidgetAttachModes(editor,"FilterLocY","ATTACH_WIDGET","ATTACH_AS_IS","ATTACH_WIDGET","ATTACH_AS_IS") + tde4.setWidgetSensitiveFlag(editor,"FilterLocY",0) + tde4.setWidgetCallbackFunction(editor,"FilterLocY","callback") + + tde4.addButtonWidget(editor,"FilterLocXAdd","+X",100,10) + tde4.setWidgetSize(editor,"FilterLocXAdd",25,50) + tde4.setWidgetLinks(editor,"FilterLocXAdd","FilterLocY","","CurveAreaWidgetY","") + tde4.setWidgetAttachModes(editor,"FilterLocXAdd","ATTACH_WIDGET","ATTACH_AS_IS","ATTACH_WIDGET","ATTACH_AS_IS") + tde4.setWidgetSensitiveFlag(editor,"FilterLocXAdd",0) + tde4.setWidgetCallbackFunction(editor,"FilterLocXAdd","callback") + + tde4.addButtonWidget(editor,"FilterLocYAdd","+Y",100,10) + tde4.setWidgetSize(editor,"FilterLocYAdd",25,50) + tde4.setWidgetLinks(editor,"FilterLocYAdd","FilterLocXAdd","","CurveAreaWidgetY","") + tde4.setWidgetAttachModes(editor,"FilterLocYAdd","ATTACH_WIDGET","ATTACH_AS_IS","ATTACH_WIDGET","ATTACH_AS_IS") + tde4.setWidgetSensitiveFlag(editor,"FilterLocYAdd",0) + tde4.setWidgetCallbackFunction(editor,"FilterLocYAdd","callback") + + + tde4.addTextFieldWidget(editor,"Range_Min","","1") + tde4.setWidgetLinks(editor,"Range_Min","FilterLocYAdd","FilterLocYAdd","CurveAreaWidgetY","") + tde4.setWidgetAttachModes(editor,"Range_Min","ATTACH_WIDGET","ATTACH_OPPOSITE_WIDGET","ATTACH_WIDGET","ATTACH_AS_IS") + tde4.setWidgetSize(editor,"Range_Min",50,50) + tde4.setWidgetOffsets(editor,"Range_Min",10,-40,5,0) + + tde4.addTextFieldWidget(editor,"Range_Max","","3") + tde4.setWidgetLinks(editor,"Range_Max","Range_Min","Range_Min","CurveAreaWidgetY","") + tde4.setWidgetAttachModes(editor,"Range_Max","ATTACH_WIDGET","ATTACH_OPPOSITE_WIDGET","ATTACH_WIDGET","ATTACH_AS_IS") + tde4.setWidgetSize(editor,"Range_Max",50,50) + tde4.setWidgetOffsets(editor,"Range_Max",10,-40,5,0) + + + + + + tde4.addTextFieldWidget(editor,"PointName","Loaded Point Name","") + tde4.setWidgetSensitiveFlag(editor,"PointName",0) + tde4.setWidgetAttachModes(editor,"PointName","ATTACH_WINDOW","ATTACH_WINDOW","ATTACH_NONE","ATTACH_WINDOW") + tde4.setWidgetSize(editor,"PointName",20,20) + tde4.setWidgetOffsets(editor,"PointName",150,400,0,10) + + + tde4.addTextFieldWidget(editor,"PointID","Loaded Point ID","") + tde4.setWidgetSensitiveFlag(editor,"PointID",0) + tde4.setWidgetLinks(editor,"PointID","PointName","","","") + tde4.setWidgetAttachModes(editor,"PointID","ATTACH_WIDGET","ATTACH_WINDOW","ATTACH_NONE","ATTACH_WINDOW") + tde4.setWidgetSize(editor,"PointID",20,20) + tde4.setWidgetOffsets(editor,"PointID",150,50,0,10) + + ''' + + tde4.addScaleWidget(editor,"scale1","scale1label","INT",-100,100,0,1) + tde4.setWidgetCallbackFunction(editor,"scale1","callback_scale") + + tde4.addTextFieldWidget(editor,"text","textlabel","0 0 0 0") + + ''' + + + + tde4.postCustomRequesterAndContinue (editor,"2D Curve Local Filter Editor",editor_width,editor_height,"callback_editor") + +else: + + import math + + #init + current_value = tde4.getParameterAdjustCurrentValue() + get_custom_data = tde4.getParameterAdjustCustomParameters() + running = tde4.getParameterAdjustPhaseRunningFlag() + + # script specific + params = get_custom_data.split() + point_id = params[0] + curve = params[1] + start = math.floor(float(params[2])) + end = math.floor(float(params[3])) + + + cam = tde4.getCurrentCamera() + current_point_group = tde4.getCurrentPGroup() + playback = tde4.getCameraPlaybackRange(cam) + firstframe = playback[0] + lastframe = playback[1] + + + + selected_point_positions = [] + points_to_filter = [] + points_filtered = [] + for frame in range (firstframe, lastframe + 1): + if frame >= start and frame <= end and tde4.isPointPos2DValid(current_point_group,point_id,cam,frame): + selected_point_position = tde4.getPointPosition2D(current_point_group,point_id,cam,frame) + selected_point_positions.append([frame,selected_point_position[0],selected_point_position[1]]) + if curve =="x": + points_to_filter.append(selected_point_position[0]) + else: + points_to_filter.append(selected_point_position[1]) + if len(points_to_filter) >= 3 : + points_filtered = tde4.filter1D(points_to_filter,current_value,"FOURIER_FILTER") + else: + points_filtered = points_to_filter + + if (running): + #calculating + for i in range(len(selected_point_positions)): + if curve =="x": + tde4.applyParameterAdjustDataModification("project.pgroups.@" + current_point_group +".points.@" + point_id + ".cameras.@" + cam +".frame." + str(selected_point_positions[i][0]) + ".position_2d[0]",points_filtered[i]) + else: + tde4.applyParameterAdjustDataModification("project.pgroups.@" + current_point_group +".points.@" + point_id + ".cameras.@" + cam +".frame." + str(selected_point_positions[i][0]) + ".position_2d[1]",points_filtered[i]) + else: + #transfer_to_3de + for i in range(len(selected_point_positions)): + if curve =="x": + tde4.setPointPosition2D(current_point_group,point_id,cam, selected_point_positions[i][0], [ points_filtered[i],selected_point_positions[i][2]] ) + else: + tde4.setPointPosition2D(current_point_group,point_id,cam, selected_point_positions[i][0], [ selected_point_positions[i][1],points_filtered[i]] )