Added 3D Renderer, First UI and First Settings

src/assets/blenderScript.py

@@ -0,0 +1,164 @@
+import csv
+import logging
+import math
+import sys
+import bpy
+import xml.etree.ElementTree as ET
+
+settings = ET.parse("./src/settings.xml")
+
+logger = logging.getLogger('simple_example')
+logger.setLevel(logging.INFO)
+formatter = logging.Formatter('%(message)s')
+console_handler = logging.StreamHandler(sys.stdout)
+console_handler.setFormatter(formatter)
+console_handler.setLevel(logging.INFO)
+logger.addHandler(console_handler)
+
+def _map(x, in_min, in_max, out_min, out_max):
+    return (x - in_min) * (out_max - out_min) / (in_max - in_min) + out_min
+
+settingsRoot = settings.getroot()
+
+FPS = int(settingsRoot[0].text)
+Width = int(settingsRoot[1].text)
+StickDistance = _map(int(settingsRoot[2].text), 0, 100, 5, 105)
+StickMode = settingsRoot[3].text
+if(StickMode == "true"):
+    StickMode = 2
+else:
+    StickMode = 1
+    
+lyMax = 0.436
+lyMin = -0.436
+lxMax = -0.436
+lxMin = 0.436
+ryMax = -0.436
+ryMin = 0.436
+rxMax = 0.436
+rxMin = -0.436
+    
+logs = settingsRoot[4].text[1:][:-1].split("\"\"")
+
+logCount = len(logs)
+logNumber = 1
+
+for log in logs:
+    logger.info("Lognr:" + ((str)(logNumber)) + ":")
+    
+    logTime = []
+    rud = []
+    ele = []
+    thr = []
+    ail = []
+    
+    try:
+        with open(log, newline='') as csvFile:
+            reader = csv.DictReader(csvFile)
+            for row in reader:
+                logTime.append(row['Time'].split(":").pop(2).replace(".", ""))
+                rud.append(int(row['Rud']))
+                ele.append(int(row['Ele']))
+                thr.append(int(row['Thr']))
+                ail.append(int(row['Ail']))
+        
+        meanTime = []
+        i = 0
+        while i < len(logTime)-1:
+            if int(logTime[i]) > int(logTime[i+1]):
+                meanTime.append(60000 - int(logTime[i]) + int(logTime[i+1]))
+            else:
+                meanTime.append(int(logTime[i+1]) - int(logTime[i]))
+            i+=1
+        
+        totalTime = 0
+        for e in meanTime:
+            totalTime+=e
+        
+        frameCount = math.floor(totalTime/1000*FPS-1)
+        FPSxxx = 1000/FPS
+    except Exception as e:
+        print("Can't read Log!")
+        exit()
+    
+    GimbalL = bpy.data.objects["GimbalL"]
+    StickL = bpy.data.objects["StickL"]
+    GimbalR = bpy.data.objects["GimbalR"]
+    StickR = bpy.data.objects["StickR"]
+    GimbalCoverR = bpy.data.objects["GimbalCoverR"]
+    TrailR = bpy.data.objects["TrailR"]
+    Camera = bpy.data.objects["Camera"]
+    Plane = bpy.data.objects["Plane.001"]
+    scn = bpy.context.scene
+    
+    scn.render.resolution_x = Width
+    GimbalCoverR.location[0] = StickDistance
+    GimbalR.location[0] = StickDistance
+    TrailR.location[0] = StickDistance
+    Plane.location[0] = StickDistance
+    Camera.location[0] = StickDistance/2
+    Camera.data.ortho_scale = StickDistance+5
+    scn.render.resolution_y = int(Width/_map(StickDistance, 5, 105, 2, 21.6))
+    bpy.context.scene.render.filepath = settingsRoot[5].text + "\\" + log.split("/")[-1].split("\\")[-1].replace(".csv", ".mov")
+    
+    scn.render.fps = 1000
+    scn.render.fps_base = FPSxxx
+    
+    scn.frame_start = 0
+    scn.frame_end = frameCount+1
+    logger.info("Frames:" + str(frameCount+1) + ":")
+    
+    frame = 0
+    log = 0
+    pastTime = 0
+    while frame <= frameCount:
+        currentTime = math.floor(FPSxxx*frame)
+        while currentTime >= pastTime+meanTime[log]:
+            pastTime+=meanTime[log]
+            log+=1
+            
+        multiplier = (currentTime-pastTime)/meanTime[log]
+        
+        ailP = _map(ail[log]+(ail[log+1]-ail[log])*multiplier, -1024, 1024, rxMin, rxMax)
+        eleP = _map(ele[log]+(ele[log+1]-ele[log])*multiplier, -1024, 1024, ryMin, ryMax)
+        rudP = _map(rud[log]+(rud[log+1]-rud[log])*multiplier, -1024, 1024, lyMin, lyMax)
+        thrP = _map(thr[log]+(thr[log+1]-thr[log])*multiplier, -1024, 1024, lxMin, lxMax)
+        
+        bpy.context.scene.frame_set(frame)
+        
+        if StickMode == "1":
+            StickL.rotation_euler=[0,0,0]
+            StickL.rotation_euler.rotate_axis("Y", ailP)
+            StickL.keyframe_insert(data_path="rotation_euler", index=-1)
+            GimbalL.rotation_euler=[0,0,0]
+            GimbalL.rotation_euler.rotate_axis("X", eleP)
+            GimbalL.keyframe_insert(data_path="rotation_euler", index=-1)
+            StickR.rotation_euler=[0,0,0]
+            StickR.rotation_euler.rotate_axis("Y", rudP)
+            StickR.keyframe_insert(data_path="rotation_euler", index=-1)
+            GimbalR.rotation_euler=[0,0,0]
+            GimbalR.rotation_euler.rotate_axis("X", thrP)
+            GimbalR.keyframe_insert(data_path="rotation_euler", index=-1)
+        else:
+            StickL.rotation_euler=[0,0,0]
+            StickL.rotation_euler.rotate_axis("Y", rudP)
+            StickL.keyframe_insert(data_path="rotation_euler", index=-1)
+            GimbalL.rotation_euler=[0,0,0]
+            GimbalL.rotation_euler.rotate_axis("X", thrP)
+            GimbalL.keyframe_insert(data_path="rotation_euler", index=-1)
+            StickR.rotation_euler=[0,0,0]
+            StickR.rotation_euler.rotate_axis("Y", ailP)
+            StickR.keyframe_insert(data_path="rotation_euler", index=-1)
+            GimbalR.rotation_euler=[0,0,0]
+            GimbalR.rotation_euler.rotate_axis("X", eleP)
+            GimbalR.keyframe_insert(data_path="rotation_euler", index=-1)
+        
+        logger.info("Init:" + ((str)(frame)) + ":")
+        frame+=1
+    
+    bpy.ops.render.render(animation=True)
+    
+    if(logCount <= logNumber):
+        logger.info("Finished")
+    
+    logNumber+=1