You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

190 lines
6.0 KiB
Python

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# Libraries
# https://circuitpython.readthedocs.io/projects/servokit/en/latest/
from adafruit_servokit import ServoKit
from leg import Leg
3 years ago
# python3-numpy
import numpy as np
import time
import json
from path_generator import forward_path, backward_path
3 years ago
3 years ago
SIN30 = 0.5
COS30 = 0.866
SIN45 = 0.7071
COS45 = 0.7071
SIN15 = 0.2588
COS15 = 0.9659
class Hexapod:
def __init__(self):
3 years ago
# x -> right
# y -> front
# z -> up
# origin is the center of the body
# roots are the positions of the bottom screws
# length units are in mm
# time units are in ms
with open('./config.json', 'r') as read_file:
self.config = json.load(read_file)
3 years ago
self.mount_x = np.array(self.config['legMountX'])
self.mount_y = np.array(self.config['legMountY'])
self.root_j1 = self.config['legRootToJoint1']
self.j1_j2 = self.config['legJoint1ToJoint2']
self.j2_j3 = self.config['legJoint2ToJoint3']
self.j3_tip = self.config['legJoint3ToTip']
3 years ago
self.mount_angle = np.array(self.config['legMountAngle'])/180*np.pi
3 years ago
self.mount_position = np.zeros((6, 3))
self.mount_position[:, 0] = self.mount_x
self.mount_position[:, 1] = self.mount_y
# Objects
self.pca_right = ServoKit(channels=16, address=0x40, frequency=120)
self.pca_left = ServoKit(channels=16, address=0x41, frequency=120)
# front right
self.leg_0 = Leg(0,
[self.pca_left.servo[15], self.pca_left.servo[2],
self.pca_left.servo[1]],
3 years ago
correction=[-6, 4, -6])
# center right
self.leg_1 = Leg(1,
[self.pca_left.servo[7], self.pca_left.servo[8],
self.pca_left.servo[6]],
3 years ago
correction=[3, -5, -6])
# rear right
self.leg_2 = Leg(2,
[self.pca_left.servo[0], self.pca_left.servo[14],
self.pca_left.servo[13]],
3 years ago
correction=[3, -6, -5])
# rear left
self.leg_3 = Leg(3,
[self.pca_right.servo[15], self.pca_right.servo[1],
self.pca_right.servo[2]],
3 years ago
correction=[-3, -4, 6])
# center left
self.leg_4 = Leg(4,
[self.pca_right.servo[7], self.pca_right.servo[6],
self.pca_right.servo[8]],
correction=[-6, 2, 0])
# front left
self.leg_5 = Leg(5,
[self.pca_right.servo[0], self.pca_right.servo[13],
self.pca_right.servo[14]],
3 years ago
correction=[-6, 4, 0])
3 years ago
self.standby_coordinate = np.zeros((6, 3))
self.standby()
3 years ago
time.sleep(0.1)
3 years ago
full_path = forward_path()
3 years ago
3 years ago
for mm in range(0, 30):
3 years ago
self.move(full_path, 0.005)
full_path = backward_path()
for mm in range(0, 30):
self.move(full_path, 0.005)
3 years ago
def move(self, path, interval):
for p_idx in range(0, np.shape(path)[0]):
dest = path[p_idx, :, :]+self.standby_coordinate
angles = self.inverse_kinematics(dest)
3 years ago
3 years ago
self.leg_0.move_junctions(angles[0, :])
self.leg_1.move_junctions(angles[1, :])
self.leg_2.move_junctions(angles[2, :])
self.leg_3.move_junctions(angles[3, :])
self.leg_4.move_junctions(angles[4, :])
self.leg_5.move_junctions(angles[5, :])
3 years ago
3 years ago
time.sleep(interval)
3 years ago
3 years ago
def standby(self):
3 years ago
3 years ago
self.standby_coordinate[:, 0] = np.array(self.mount_x)+(self.root_j1+self.j1_j2+(
3 years ago
self.j2_j3*COS30)+self.j3_tip*SIN15)*np.cos(self.mount_angle)
3 years ago
self.standby_coordinate[:, 1] = self.mount_y + (self.root_j1+self.j1_j2+(
3 years ago
self.j2_j3*COS30)+self.j3_tip*SIN15)*np.sin(self.mount_angle)
self.standby_coordinate[:, 2] = self.j2_j3 * \
SIN30 - self.j3_tip * COS15
3 years ago
3 years ago
self.leg_0.set_angle(0, 90)
self.leg_0.set_angle(1, 60)
self.leg_0.set_angle(2, 75)
self.leg_1.set_angle(0, 90)
self.leg_1.set_angle(1, 60)
self.leg_1.set_angle(2, 75)
self.leg_2.set_angle(0, 90)
self.leg_2.set_angle(1, 60)
self.leg_2.set_angle(2, 75)
self.leg_3.set_angle(0, 90)
self.leg_3.set_angle(1, 60)
self.leg_3.set_angle(2, 75)
self.leg_4.set_angle(0, 90)
self.leg_4.set_angle(1, 60)
self.leg_4.set_angle(2, 75)
self.leg_5.set_angle(0, 90)
self.leg_5.set_angle(1, 60)
self.leg_5.set_angle(2, 75)
3 years ago
def inverse_kinematics(self, dest):
temp_dest = dest-self.mount_position
local_dest = np.zeros_like(dest)
local_dest[:, 0] = temp_dest[:, 0] * \
np.cos(self.mount_angle) + \
temp_dest[:, 1] * np.sin(self.mount_angle)
local_dest[:, 1] = temp_dest[:, 0] * \
np.sin(self.mount_angle) - \
temp_dest[:, 1] * np.cos(self.mount_angle)
local_dest[:, 2] = temp_dest[:, 2]
3 years ago
3 years ago
angles = np.zeros((6, 3))
x = local_dest[:, 0] - self.root_j1
y = local_dest[:, 1]
3 years ago
angles[:, 0] = -(np.arctan2(y, x) * 180 / np.pi)+90
3 years ago
x = np.sqrt(x*x + y*y) - self.j1_j2
3 years ago
y = local_dest[:, 2]
3 years ago
ar = np.arctan2(y, x)
lr2 = x*x + y*y
lr = np.sqrt(lr2)
3 years ago
a1 = np.arccos((lr2 + self.j2_j3*self.j2_j3 -
self.j3_tip*self.j3_tip)/(2*self.j2_j3*lr))
a2 = np.arccos((lr2 - self.j2_j3*self.j2_j3 +
self.j3_tip*self.j3_tip)/(2*self.j3_tip*lr))
3 years ago
angles[:, 1] = 90-((ar + a1) * 180 / np.pi)
angles[:, 2] = (90 - ((a1 + a2) * 180 / np.pi))+90
return angles
3 years ago
# return [path, mir_path, path, mir_path, path, mir_path, ], "shift", 20, (0, halfsteps)
3 years ago
def main():
hexapod = Hexapod()
if __name__ == '__main__':
main()