Files
PythonRobotics/ArmNavigation/n_joint_arm_3d/random_forward_kinematics.py
2020-06-08 21:43:37 +09:00

34 lines
1011 B
Python

"""
Forward Kinematics for an n-link arm in 3D
Author: Takayuki Murooka (takayuki5168)
"""
import math
from NLinkArm3d import NLinkArm
import random
def random_val(min_val, max_val):
return min_val + random.random() * (max_val - min_val)
def main():
print("Start solving Forward Kinematics 10 times")
# init NLinkArm with Denavit-Hartenberg parameters of PR2
n_link_arm = NLinkArm([[0., -math.pi / 2, .1, 0.],
[math.pi / 2, math.pi / 2, 0., 0.],
[0., -math.pi / 2, 0., .4],
[0., math.pi / 2, 0., 0.],
[0., -math.pi / 2, 0., .321],
[0., math.pi / 2, 0., 0.],
[0., 0., 0., 0.]])
# execute FK 10 times
for _ in range(10):
n_link_arm.set_joint_angles(
[random_val(-1, 1) for _ in range(len(n_link_arm.link_list))])
n_link_arm.forward_kinematics(plot=True)
if __name__ == "__main__":
main()