Skip to content

Commit

Permalink
examples: Apply black
Browse files Browse the repository at this point in the history
  • Loading branch information
jorisv committed Feb 2, 2024
1 parent 1c997ad commit 3d40f7c
Showing 1 changed file with 10 additions and 9 deletions.
19 changes: 10 additions & 9 deletions examples/floating-base-velocity-viewer.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,14 +5,15 @@
from pinocchio.visualize import MeshcatVisualizer
from time import sleep

def create_pin_cube_model(j0='freeflyer'):

def create_pin_cube_model(j0="freeflyer"):
model = pin.Model()

if j0 == 'freeflyer':
if j0 == "freeflyer":
j0 = pin.JointModelFreeFlyer()
elif j0 == 'spherical':
elif j0 == "spherical":
j0 = pin.JointModelSpherical()
elif j0 == 'sphericalzyx':
elif j0 == "sphericalzyx":
j0 = pin.JointModelSphericalZYX()
else:
raise ValueError("Unknown joint type")
Expand All @@ -22,6 +23,7 @@ def create_pin_cube_model(j0='freeflyer'):
model.appendBodyToJoint(jointCube, pin.Inertia.FromBox(1, 0.8, 0.4, 0.2), M)
return model


def create_pin_geometry_cube_model(model):
jointCube = model.getFrameId("joint0")
geom_model = pin.GeometryModel()
Expand All @@ -33,7 +35,8 @@ def create_pin_geometry_cube_model(model):
geom_model.addGeometryObject(cube)
return geom_model

def create_model(joint0_type='freeflyer'):

def create_model(joint0_type="freeflyer"):
model = create_pin_cube_model(joint0_type)
geom_model = create_pin_geometry_cube_model(model)
return model, geom_model
Expand All @@ -42,7 +45,7 @@ def create_model(joint0_type='freeflyer'):
def pin_step(model, vizer, v_index_increment, dt=0.1):
q = pin.neutral(model)
v = np.zeros(model.nv)
v[v_index_increment] += 1.
v[v_index_increment] += 1.0

vizer.display(q)
print(f"{q=}, {v=}")
Expand All @@ -56,11 +59,9 @@ def pin_step(model, vizer, v_index_increment, dt=0.1):


if __name__ == "__main__":

list_joint0 = ['freeflyer', 'sphericalzyx', 'spherical']
list_joint0 = ["freeflyer", "sphericalzyx", "spherical"]

for joint0_name in list_joint0:

print(f"\njoint0_name = {joint0_name}")

model, geom_model = create_model(joint0_name)
Expand Down

0 comments on commit 3d40f7c

Please sign in to comment.