89 lines
2.9 KiB
Python
89 lines
2.9 KiB
Python
import bpy
|
|
from random import random
|
|
from mathutils import Matrix, Quaternion
|
|
|
|
|
|
class ArmGraspController:
|
|
def __init__(self, armature_name):
|
|
self.arm = bpy.data.objects[armature_name]
|
|
self.open = False
|
|
self.finger_grab_bones = ["index_3", "middle_3", "ring_3", "pinky_3"]
|
|
self.thumb_rot_bone = "thumb_3"
|
|
|
|
def deselect_all_bones(self):
|
|
for bone in self.arm.pose.bones: # Deselect all selected bones
|
|
bone.bone.select = False
|
|
|
|
def open_fingers(self):
|
|
self.grab_movement(0.8)
|
|
|
|
def close_fingers(self):
|
|
self.grab_movement(-0.8)
|
|
|
|
def move_fingers(self):
|
|
if self.open:
|
|
self.open_fingers()
|
|
else:
|
|
self.close_fingers()
|
|
self.open = not self.open
|
|
|
|
def grab_movement(self, angle):
|
|
# select the 4 base finger bones
|
|
# Then:
|
|
# To rotate around local axis (will cause to grab)
|
|
# Then do the same with thumb, but only around local Z axis!
|
|
bpy.ops.object.mode_set(mode='OBJECT')
|
|
bpy.ops.object.select_all(action='DESELECT') # Deselect all objects
|
|
bpy.context.view_layer.objects.active = self.arm # Make the Armature the active object
|
|
bpy.ops.object.mode_set(mode='POSE')
|
|
|
|
# Deselect all fingers
|
|
self.deselect_all_bones()
|
|
|
|
# Rotate fingers
|
|
for name in self.finger_grab_bones:
|
|
bone = self.arm.pose.bones.get(name)
|
|
# Bones need to be selected when adding keyframes, otherwise blender will throw an error:
|
|
# https://blender.stackexchange.com/questions/1828/what-constitutes-a-context-in-pose-mode
|
|
bone.bone.select = True
|
|
# rotate
|
|
bone.rotation_mode = 'XYZ'
|
|
bone.rotation_euler.rotate_axis('X', angle)
|
|
|
|
# bpy.ops.transform.rotate(value=angle, orient_axis='X', orient_type='GLOBAL')
|
|
# Add keyframe
|
|
bpy.ops.anim.keying_set_active_set(type='Rotation')
|
|
bpy.ops.anim.keyframe_insert(type='Rotation')
|
|
|
|
# Now rotate the thumb too
|
|
self.deselect_all_bones()
|
|
bone = self.arm.pose.bones.get(self.thumb_rot_bone).bone
|
|
if bone: bone.select = True
|
|
bpy.ops.transform.rotate(value=angle, orient_axis='Z', orient_type='LOCAL')
|
|
bpy.ops.anim.keyframe_insert(type='Rotation')
|
|
|
|
|
|
ob = bpy.data.objects['Sphere']
|
|
frame_number = 0
|
|
|
|
max_dist = 0.5
|
|
#start_location = ob.location.copy()
|
|
start_location = (1.6766993999481201, 0.3146645724773407, -1.3483989238739014)
|
|
|
|
# Script start
|
|
controller = ArmGraspController("Armature")
|
|
|
|
for i in range(12):
|
|
x = random()*max_dist
|
|
y = random()*max_dist
|
|
z = random()*max_dist
|
|
|
|
bpy.context.scene.frame_set(frame_number)
|
|
ob.location = (
|
|
start_location[0] + x,
|
|
start_location[1] + y,
|
|
start_location[2] + z
|
|
)
|
|
ob.keyframe_insert(data_path='location', index=-1)
|
|
controller.move_fingers()
|
|
frame_number += 20 |