keyframes

scripts/Manager.py

@@ -1,4 +1,5 @@
 import mathutils
+import bge
 
 def main():
 
@@ -564,58 +565,153 @@ def main():
                     print('new key!')
                     freecam['pg'] = 0
                 else:
-                    freecam['pg'] += 1    
-                freecam['last_key'] = freecam['cur_key']    
-
-            
+                    
+                    num = own['objIndex'] - own['last_objIndex']
+                       
+                    #print(num)
+                    #freecam['clock'] += num
+                    #print(num)
+                    if num > 4:
+                        num = 2
+                    if num < -4:
+                        num = -2    
+                        #print('-------************num is biiig')
+
+                    freecam['pg'] += num    
+                freecam['last_key'] = freecam['cur_key'] 
+
+            def draw_keys(keyframes, dict):
+                scene_list = bge.logic.getSceneList()
+                if 'replay_HUD' in scene_list:
+                    hud = scene_list['replay_HUD']
+                    ph = hud.objects['Circle']
+                    for x in hud.objects:
+                        if 'keyframe' in x:
+                            x.endObject()
+                    for y in keyframes:
+                        place = (y / round((dict['replay_record_length']), 4) * 1000)
+                        print('action frame', place)
+                        added = hud.addObject('kf', ph)
+                        added.scaling = [1.25,1,1]
+                        added.playAction("replay_playhead", place, place, layer=1, play_mode=1, speed=0)
+                else:
+                    print('no hud scene')
             
-            keyframes = [190, 480,1050, 1800, 2500, 4000, 6000, 7000]
-            positions = [mathutils.Vector([0,0,-1]), mathutils.Vector([-3,.5,-1]), mathutils.Vector([-4,1,-1]), mathutils.Vector([-5,2,-1]), mathutils.Vector([-5,-2,-1]), mathutils.Vector([-2,-2,-1]), mathutils.Vector([2,-2,-1]), mathutils.Vector([5,-3,-1])]
-            kl = len(keyframes) - 1
+            keyframes = dict['rp_keyframes']
+            positions = dict['rp_positions']
+            orientations = dict['rp_orientations']
             
             if 'inited' not in freecam:
                 freecam['inited'] = True
                 freecam['clock'] = 0
                 freecam['cur_key'] = 0
-                freecam['last_key'] = kl
+                freecam['last_key'] = -1
                 freecam['cur_length'] = 0
                 freecam['pg'] = 0
                 freecam['last_position'] = 0
                 freecam['playback'] = True
             
-            #clock(freecam)
-            
-            freecam['cur_length'] = transLength(freecam, keyframes, freecam['cur_key'], kl)
-            getProgress(freecam)
-            skip = False
-            if own["objIndex"] > keyframes[kl]:
-                skip = True
-            if keyframes[freecam['cur_key']] > own["objIndex"]:
-                pass
-            else:
-                if skip == False:
-                    try:
-                        freecam['last_position'] = freecam['cur_key'] 
-                        freecam['cur_key'] += 1 
-                        temp = keyframes[freecam['cur_key']] 
-                        freecam['pg'] = 0
-                        
-                    except:
-                        print('excepting')
-                        freecam['cur_key'] = 0
-                        freecam['last_position'] = kl
-                        freecam['pg'] = 0
-
-            time = round(freecam['pg'] / freecam['cur_length'], 4) * 2
-
-            #if own['playback'] == True:
-            freecam.worldPosition = (positions[freecam['last_position']].lerp(positions[freecam['cur_key']], time))
-                
+            if len(positions) > 1:
+                kl = len(keyframes) - 1
+                freecam['cur_length'] = transLength(freecam, keyframes, freecam['cur_key'], kl)
+                getProgress(freecam)
+                skip = False
+                if own["objIndex"] > keyframes[kl]:
+                    skip = True
+                    #skip = False
+                    #print('skipping')
+                if freecam['cur_key'] != 0:    
+                    if keyframes[(freecam['cur_key'] - 1)] > own['objIndex']:
+                        freecam['cur_key'] -= 1 
+                        freecam['last_position'] -= 1
+                        if freecam['last_position'] == -1:
+                            freecam['last_position'] = keyframes.len() - 1
+                        print('backwards')
+                if keyframes[freecam['cur_key']] > own["objIndex"]:
+                    
+                    pass
+                else:
+                    if skip == False:
+                        try:
+                            freecam['last_position'] = freecam['cur_key'] 
+                            freecam['cur_key'] += 1 
+                            temp = keyframes[freecam['cur_key']]
+                            if freecam['cur_key'] != 0: 
+                                freecam['pg'] = 0
+                                print('setting pg to 0')
+                            
+                        except:
+                            print('excepting')
+                            freecam['cur_key'] = 0
+                            freecam['last_position'] = kl
+                            freecam['pg'] = 0
+
+                time = abs(round(freecam['pg'] / freecam['cur_length'], 6) )
                 
-            print('cur_key', freecam['cur_key'], freecam['last_position'], 'time: ', own["objIndex"], freecam['cur_length'], freecam['pg'], time)
-            
+
+
+                if own['objIndex'] != own['last_objIndex']:
+                    freecam.worldPosition = (positions[freecam['last_position']].lerp(positions[freecam['cur_key']], time))
+                    freecam.worldOrientation = (orientations[freecam['last_position']].lerp(orientations[freecam['cur_key']], time))
+                #print(positions)
+                #print(positions[freecam['last_position']], positions[freecam['cur_key']])
+                #print('cur_key', freecam['cur_key'], freecam['last_position'], 'time: ', own["objIndex"], freecam['cur_length'], freecam['pg'], time)
+            else:
+                pass
+                #print('not enough keys')
             
+            if dict['xBut'] == 1 and dict['last_xBut'] == 0:
+                print('chekcing for keyframe on', own['objIndex'])
+                if len(keyframes) > 1:
+                    close = min(keyframes, key=lambda x:abs(x-own['objIndex']))               
+                    keys_away = abs(close - own['objIndex'])
+                else:
+                    close = None
+                    keys_away = 100000 
+
+                print('--', close, keys_away)       
+                if keys_away < 60:
+                    print('index already exists')
+                    ind = keyframes.index(close)
+                    positions.pop(ind)
+                    orientations.pop(ind)
+                    keyframes.pop(ind)
+                    dict['rp_keyframes'] = keyframes
+                    dict['rp_positions'] = positions
+                    dict['rp_orientations'] = orientations     
+                    if freecam['cur_key'] != 0:
+                        freecam['cur_key'] -= 1
+                    else:
+                        freecam['cur_key'] = len(keyframes) -1               
+                    draw_keys(keyframes, dict)
+
+
+                else:
+                    keyframes.append(own['objIndex']) 
+                    positions.append(freecam.worldPosition.copy()) 
+                    orientations.append(freecam.worldOrientation.copy())
+                    tk = keyframes
+                    z = [x for _,x in sorted(zip(keyframes, positions))]
+                    v = [y for _,y in sorted(zip(tk, orientations))]
+                    positions = z
+                    orientations = v
+                    keyframes.sort()
+                    print('added keyframe on', own['objIndex'])
+                    dict['rp_keyframes'] = keyframes
+                    if own['objIndex'] in positions:
+                        freecam['cur_key'] = positions.index(own['objIndex'])
+                        # freecam['cur_length'] = freecam['cur_key']
+                        # try:
+                        #     freecam['last_position'] = positions.index(own['objIndex']) -1
+                        # except:
+                        #     freecam['last_position'] = len(positions) - 1   
+
 
+                    dict['rp_keyframes'] = keyframes
+                    dict['rp_positions'] = positions
+                    dict['rp_orientations'] = orientations                       
+                draw_keys(keyframes, dict)
+            #print('lp', dict['temp_list']) 
 
 
 
@@ -638,7 +734,12 @@ def main():
     ########################################                 
         if own['camnum'] == 0:
             cam = camList["Camera.003"]
-            scene.active_camera = cam          
+            scene.active_camera = cam 
+            dict['rp_keyframes'] = []    
+            dict['rp_positions'] = [] 
+            dict['rp_orientations'] = []
+            freecam['cur_key'] = 0
+            freecam.worldOrientation = cam.worldOrientation    
         valueIndex = own['valueIndex']
         n = (valueIndex / recording_cutoff) * 1000
         n = int(round(n))
@@ -654,4 +755,5 @@ def main():
         own['last_ltsBut'] = ltsBut
         own['last_rtsBut'] = rtsBut
         own['last_playback'] = own['playback']
+        own['last_objIndex'] = own['objIndex']
         dict['playback'] = own['playback']