Ejemplo n.º 1
0
if __name__ == '__main__':
    info('main line')
    # 共有メモリの準備
    shmem = Value(Point, 0)
    shmem.preparingRestart = False
    shmem.relyStation = False
    shmem.soundPhase = 0
    # モータ制御インスタンスの生成
    motorController = MotorController()
    # 画像処理インスタンスの生成
    imageProcessing = ImageProcessing()
    # 慣性計測インスタンスの生成
    imu = Imu()
    # スタート時に一度ジャイロセンサの補正をしておく
    imu.calibrate()
    # websocketサーバの生成
    # server = Server(9001)
    sound = Sound()

    p_motorContl = Process(target=motorController.target, args=(shmem, ))
    p_imageProcessing = Process(target=imageProcessing.target, args=(shmem, ))
    p_imu = Process(target=imu.target, args=(shmem, ))
    # p_server = Process(target=server.target, args=())
    p_sound = Process(target=sound.target, args=(shmem, ))

    p_motorContl.start()
    DEBUG('p_motorContl started')
    p_imageProcessing.start()
    DEBUG('p_imageProcessing started')
    p_imu.start()
Ejemplo n.º 2
0
class RovMovement:
    def __init__(self,
                 xy_lf_pin,
                 xy_rf_pin,
                 xy_lb_pin,
                 xy_rb_pin,
                 z_lf_pin,
                 z_rf_pin,
                 z_lb_pin,
                 z_rb_pin,
                 arm_pin,
                 initialize_motors=True,
                 ssc_control=True):
        with open('p2t.json', 'r') as json_file:
            p2t = json.load(json_file)
        self.ssc_control = ssc_control

        self.motors = {
            "xy_lf": ContinuousRotationServo(xy_lf_pin, p2t=p2t),
            "xy_rf": ContinuousRotationServo(xy_rf_pin, p2t=p2t),
            "xy_lb": ContinuousRotationServo(xy_lb_pin, p2t=p2t),
            "xy_rb": ContinuousRotationServo(xy_rb_pin, p2t=p2t),
            "z_lf": ContinuousRotationServo(z_lf_pin, p2t=p2t),
            "z_rf": ContinuousRotationServo(z_rf_pin, p2t=p2t),
            "z_lb": ContinuousRotationServo(z_lb_pin, p2t=p2t),
            "z_rb": ContinuousRotationServo(z_rb_pin, p2t=p2t)
        }
        self.motor_prev_powers = {
            "xy_lf": 0,
            "xy_rf": 0,
            "xy_lb": 0,
            "xy_rb": 0,
            "z_lf": 0,
            "z_rf": 0,
            "z_lb": 0,
            "z_rb": 0
        }

        self.arm = StandardServo(arm_pin)
        self.arm_status = False
        self.open_arm()

        # PID
        self.old_err = np.zeros((2, 2))
        self.imu = Imu()
        self.imu.start()

        if initialize_motors:
            self.initialize_motors()
            self._initialize_imu()
        sleep(2)

    def select_motors(self, search_list):
        return {
            key: value
            for (key, value) in self.motors.items()
            if not any(s not in key for s in search_list)
        }

    def initialize_motors(self, mp=30):
        print("All motors initializing...")
        for cp in list(range(0, mp)) + list(range(mp, -mp, -1)) + list(
                range(-mp, 1)):
            print("Power:", cp)
            for motor in self.motors.values():
                motor.run_bidirectional(cp)
            sleep(0.01)
        print("All motors initialized...")

    def _initialize_imu(self, seconds=5):
        print("IMU is being calibrated...")
        self.imu.calibrate(seconds)
        print("IMU is calibrated...")

    def _get_z_balance_p(self, kp=0.0, kd=0.0, type_=int, tilt_degree=0):
        # if not kp > kd:
        #     raise Exception("Kp must be bigger than Kd. Kp: %s - Kd: %s" % (kp, kd))
        try:
            x, y, _ = self.imu.get_degree().get()
            # if abs(tilt_degree) == 90:
            #     if 45 < y < 90:
            #         tilt_degree = 90
            #     elif -90 < y < -45:
            #         tilt_degree = -90
            y = y - tilt_degree
            comp_sign = +1 if (x < 0 and y < 0) or (x > 0 and y > 0) else -1
            if -1 < x < 1: x = 0
            if -1 < y < 1: y = 0

            lf = sign_n(-y - x) * math.sqrt(
                abs(-y * -y + comp_sign * -x * -x))  # -y - x
            rf = sign_n(-y + x) * math.sqrt(
                abs(-y * -y - comp_sign * +x * +x))  # -y + x
            lb = sign_n(+y - x) * math.sqrt(
                abs(+y * +y - comp_sign * -x * -x))  # +y - x
            rb = sign_n(+y + x) * math.sqrt(
                abs(+y * +y + comp_sign * +x * +x))  # +y + x
            err = np.array([[lf, rf], [lb, rb]])

            ret = kp * err + kd * (err - self.old_err)
            self.old_err = err
            if type_ == int:
                ret = np.rint(ret)
            return ret.ravel()
        except Exception as e:
            print("Exception in _get_balance_p:", e)
            return 0, 0, 0, 0

    def _gradual_power_change(self,
                              current_powers,
                              ssc_control=True,
                              ssc_step=5,
                              ssc_sleep=0.01):
        start_powers = self.motor_prev_powers
        stop_powers = current_powers
        motor_keys = stop_powers.keys()

        diff = [
            abs(stop_powers[key] - start_powers[key]) for key in motor_keys
        ]
        steps = int(max(diff) / ssc_step)
        power_list = {
            key: np.linspace(start_powers[key],
                             stop_powers[key],
                             steps,
                             endpoint=False,
                             dtype=int)
            for key in motor_keys
        }

        if steps and ssc_control:
            for k in range(1, steps):
                for key in motor_keys:
                    self.motors[key].run_bidirectional(int(power_list[key][k]))
                    print(str(key) + ":", power_list[key][k], end="\t")
                print()
                sleep(ssc_sleep)

        for key in motor_keys:
            self.motors[key].run_bidirectional(int(stop_powers[key]))
            print(str(key) + ":", stop_powers[key], end="\t")
        print()

    def go_z_bidirectional(self, power, with_balance=True, tilt_degree=0):
        power_per_motor = int(power / 4)

        lf_p, rf_p, lb_p, rb_p = self._get_z_balance_p(
            kp=0.35, kd=0.30, tilt_degree=tilt_degree) if with_balance else (0,
                                                                             0,
                                                                             0,
                                                                             0)
        current_powers = {
            "z_lf": power_per_motor + lf_p,
            "z_rf": power_per_motor + rf_p,
            "z_lb": power_per_motor + lb_p,
            "z_rb": power_per_motor + rb_p
        }
        self._gradual_power_change(current_powers,
                                   ssc_control=self.ssc_control)
        self.motor_prev_powers.update(current_powers)

    def go_xy_and_turn(self, power, degree, turn_power):
        """
        :param power: Power sent to the vehicle's movement
        :param degree: degree of movement (0between 0-360 degree)
                       0 -> ileri
                       90 -> sağa
                       180 -> geri
                       270 -> sola
        :param turn_power: Turn power
                           Positive value -> Turn right
                           Negative value -> Turn left
        :return:
        """

        # if turn_power:
        #     turn_power = turn_power / 4
        #     turn_power_per_motor = int(turn_power / 4)
        # else:
        #     _, _, z = self.imu.get_instant_gyro().get()
        #     turn_power_per_motor = int(z)
        turn_power = turn_power / 4
        turn_power_per_motor = int(turn_power / 4)

        go_power_per_motor = int(power / 2)

        radian_rf = (45 - degree) / 180 * math.pi
        radian_lf = (135 - degree) / 180 * math.pi
        radian_lb = (225 - degree) / 180 * math.pi
        radian_rb = (315 - degree) / 180 * math.pi

        pow_rf = int(
            math.sin(radian_rf) * go_power_per_motor - turn_power_per_motor)
        pow_lf = int(
            math.sin(radian_lf) * go_power_per_motor + turn_power_per_motor)
        pow_lb = int(
            math.sin(radian_lb) * go_power_per_motor - turn_power_per_motor)
        pow_rb = int(
            math.sin(radian_rb) * go_power_per_motor + turn_power_per_motor)

        current_powers = {
            "xy_lf": pow_lf,
            "xy_rf": pow_rf,
            "xy_lb": pow_lb,
            "xy_rb": pow_rb
        }
        self._gradual_power_change(current_powers)
        self.motor_prev_powers.update(current_powers)

    def go_xyz_with_tilt(self,
                         xy_power,
                         z_power,
                         turn_power,
                         with_balance=True,
                         tilt_degree=0):
        """
        :param xy_power: between from -70 to 70
        :param z_power: between from -70 to 70
        :param turn_power: Turn power
                           Positive value -> Turn right
                           Negative value -> Turn left
        :param with_balance: Should the balance of the vehicle be achieved with pid control?
        :param tilt_degree: the inclination the vehicle will make forward
                            negative value -> leaning forward of the vehicle
                            positive value -> raising the front of the vehicle
        :return:
        """
        tilt_radian = tilt_degree * math.pi / 180
        xy_power_ = math.cos(-tilt_radian) * xy_power - math.sin(
            -tilt_radian) * z_power
        z_power_ = math.sin(-tilt_radian) * xy_power + math.cos(
            -tilt_radian) * z_power
        self.go_xy_and_turn(xy_power_, 0, turn_power)
        self.go_z_bidirectional(z_power_,
                                with_balance=with_balance,
                                tilt_degree=tilt_degree)

    def open_arm(self):
        self.arm.change_angle(100)
        self.arm_status = True

    def close_arm(self):
        self.arm.change_angle(30)
        self.arm_status = False

    def toggle_arm(self, arm_status=None):
        if self.arm_status and (arm_status is None or arm_status == False):
            self.close_arm()
        elif not self.arm_status and (arm_status is None
                                      or arm_status == True):
            self.open_arm()

    def run_all_motors_cw(self, power):
        for motor in self.motors.values():
            motor.run_clockwise(power)

    def run_all_motors_ccw(self, power):
        for motor in self.motors.values():
            motor.run_counterclockwise(power)

    def stop(self):
        print("RovMovement is terminating...")
        for motor in self.motors.values():
            motor.stop()
        self.arm.stop()
        self.imu.stop()
        print("RovMovement has been terminated.")

    def close(self):
        self.stop()
Ejemplo n.º 3
0
class RovMovement:
    def __init__(self,
                 xy_lf_pin,
                 xy_rf_pin,
                 xy_lb_pin,
                 xy_rb_pin,
                 z_lf_pin,
                 z_rf_pin,
                 z_lb_pin,
                 z_rb_pin,
                 arm_pin,
                 initialize_motors=True,
                 ssc_control=True):
        with open('p2t.json', 'r') as json_file:
            p2t = json.load(json_file)
        self.xy_lf = ContinuousRotationServo(xy_lf_pin,
                                             ssc_control=ssc_control,
                                             p2t=p2t)
        self.xy_rf = ContinuousRotationServo(xy_rf_pin,
                                             ssc_control=ssc_control,
                                             p2t=p2t)
        self.xy_lb = ContinuousRotationServo(xy_lb_pin,
                                             ssc_control=ssc_control,
                                             p2t=p2t)
        self.xy_rb = ContinuousRotationServo(xy_rb_pin,
                                             ssc_control=ssc_control,
                                             p2t=p2t)
        self.z_lf = ContinuousRotationServo(z_lf_pin,
                                            ssc_control=ssc_control,
                                            p2t=p2t)
        self.z_rf = ContinuousRotationServo(z_rf_pin,
                                            ssc_control=ssc_control,
                                            p2t=p2t)
        self.z_lb = ContinuousRotationServo(z_lb_pin,
                                            ssc_control=ssc_control,
                                            p2t=p2t)
        self.z_rb = ContinuousRotationServo(z_rb_pin,
                                            ssc_control=ssc_control,
                                            p2t=p2t)
        self.arm = StandardServo(arm_pin)
        self.imu = Imu()
        self.z_motors_list = [self.z_lf, self.z_rf, self.z_lb, self.z_rb]
        self.xy_motors_list = [self.xy_lf, self.xy_rf, self.xy_lb, self.xy_rb]
        self.all_motors_list = self.z_motors_list + self.xy_motors_list
        self.arm_status = False
        self.open_arm()

        # PID
        self.old_err = np.zeros((2, 2))

        self.imu.start()
        if initialize_motors:
            self.initialize_motors()
            self._initialize_imu()
        sleep(2)

    def initialize_motors(self, mp=30):
        print("All motors initializing...")
        for cp in list(range(0, mp)) + list(range(mp, -mp, -1)) + list(
                range(-mp, 1)):
            print("Power:", cp)
            for motor in self.all_motors_list:
                motor.run_bidirectional(cp)
            sleep(0.01)
        print("All motors initialized...")

    def _initialize_imu(self, seconds=5):
        print("IMU is being calibrated...")
        self.imu.calibrate(seconds)
        print("IMU is calibrated...")

    def _get_z_balance_p(self, kp=0.0, kd=0.0, type_=int):
        # if not kp > kd:
        #     raise Exception("Kp must be bigger than Kd. Kp: %s - Kd: %s" % (kp, kd))
        try:
            x, y, _ = self.imu.get_degree().get()
            comp_sign = +1 if (x < 0 and y < 0) or (x > 0 and y > 0) else -1
            if -1 < x < 1: x = 0
            if -1 < y < 1: y = 0

            lf = sign_n(-y - x) * math.sqrt(
                abs(-y * -y + comp_sign * -x * -x))  # -y - x
            rf = sign_n(-y + x) * math.sqrt(
                abs(-y * -y - comp_sign * +x * +x))  # -y + x
            lb = sign_n(+y - x) * math.sqrt(
                abs(+y * +y - comp_sign * -x * -x))  # +y - x
            rb = sign_n(+y + x) * math.sqrt(
                abs(+y * +y + comp_sign * +x * +x))  # +y + x
            err = np.array([[lf, rf], [lb, rb]])

            ret = kp * err + kd * (err - self.old_err)
            self.old_err = err
            if type_ == int:
                ret = np.rint(ret)
            return ret.ravel()
        except Exception as e:
            print("Exception in _get_balance_p:", e)
            return 0, 0, 0, 0

    def go_z_bidirectional(self, power, with_balance=True):
        power_per_motor = int(power / 4)

        lf_p, rf_p, lb_p, rb_p = self._get_z_balance_p(
            kp=0.3, kd=0.15) if with_balance else (0, 0, 0, 0)
        self.z_lf.run_bidirectional(power_per_motor + int(lf_p))
        self.z_rf.run_bidirectional(power_per_motor + int(rf_p))
        self.z_lb.run_bidirectional(power_per_motor + int(lb_p))
        self.z_rb.run_bidirectional(power_per_motor + int(rb_p))

    def go_up(self, power):
        power_per_motor = int(power / 4)
        for motor in self.z_motors_list:
            motor.run_clockwise(power_per_motor)

    def go_down(self, power):
        power_per_motor = int(power / 4)
        for motor in self.z_motors_list:
            motor.run_counterclockwise(power_per_motor)

    def turn_left(self, power):
        power = power / 4
        power_per_motor = int(power / 4)
        self.xy_rf.run_clockwise(power_per_motor)
        self.xy_lf.run_counterclockwise(power_per_motor)
        self.xy_lb.run_clockwise(power_per_motor)
        self.xy_rb.run_counterclockwise(power_per_motor)

    def turn_right(self, power):
        power = power / 4
        power_per_motor = int(power / 4)
        self.xy_rf.run_counterclockwise(power_per_motor)
        self.xy_lf.run_clockwise(power_per_motor)
        self.xy_lb.run_counterclockwise(power_per_motor)
        self.xy_rb.run_clockwise(power_per_motor)

    def go_xy(self, power, degree):
        """
        :param power: Power sent to the vehicle's movement
        :param degree: degree of movement (0between 0-360 degree)
                       0 -> ileri
                       90 -> sağa
                       180 -> geri
                       270 -> sola
        :return:
        """
        power_per_motor = int(power / 2)

        radian_rf = (45 - degree) / 180 * math.pi
        radian_lf = (135 - degree) / 180 * math.pi
        radian_lb = (225 - degree) / 180 * math.pi
        radian_rb = (315 - degree) / 180 * math.pi

        pow_rf = int(math.sin(radian_rf) * power_per_motor)
        pow_lf = int(math.sin(radian_lf) * power_per_motor)
        pow_lb = int(math.sin(radian_lb) * power_per_motor)
        pow_rb = int(math.sin(radian_rb) * power_per_motor)

        self.xy_rf.run_bidirectional(pow_rf)
        self.xy_lf.run_bidirectional(pow_lf)
        self.xy_lb.run_bidirectional(pow_lb)
        self.xy_rb.run_bidirectional(pow_rb)

    def go_xy_and_turn(self, power, degree, turn_power):
        """
        :param power: Power sent to the vehicle's movement
        :param degree: degree of movement (0between 0-360 degree)
                       0 -> ileri
                       90 -> sağa
                       180 -> geri
                       270 -> sola
        :param turn_power: Turn power
                           Positive value -> Turn right
                           Negative value -> Turn left
        :return:
        """
        turn_power = turn_power / 4
        turn_power_per_motor = int(turn_power / 4)
        go_power_per_motor = int(power / 2)

        radian_rf = (45 - degree) / 180 * math.pi
        radian_lf = (135 - degree) / 180 * math.pi
        radian_lb = (225 - degree) / 180 * math.pi
        radian_rb = (315 - degree) / 180 * math.pi

        pow_rf = int(
            math.sin(radian_rf) * go_power_per_motor - turn_power_per_motor)
        pow_lf = int(
            math.sin(radian_lf) * go_power_per_motor + turn_power_per_motor)
        pow_lb = int(
            math.sin(radian_lb) * go_power_per_motor - turn_power_per_motor)
        pow_rb = int(
            math.sin(radian_rb) * go_power_per_motor + turn_power_per_motor)

        self.xy_rf.run_bidirectional(pow_rf)
        self.xy_lf.run_bidirectional(pow_lf)
        self.xy_lb.run_bidirectional(pow_lb)
        self.xy_rb.run_bidirectional(pow_rb)

    def open_arm(self):
        self.arm.change_angle(100)
        self.arm_status = True

    def close_arm(self):
        self.arm.change_angle(30)
        self.arm_status = False

    def toggle_arm(self, arm_status=None):
        if self.arm_status and (arm_status is None or arm_status == False):
            self.close_arm()
        elif not self.arm_status and (arm_status is None
                                      or arm_status == True):
            self.open_arm()

    def run_all_motors_cw(self, power):
        for motor in self.all_motors_list:
            motor.run_clockwise(power)

    def run_all_motors_ccw(self, power):
        for motor in self.all_motors_list:
            motor.run_counterclockwise(power)

    def stop(self):
        print("RovMovement is terminating...")
        for motor in self.all_motors_list:
            motor.stop()
        self.arm.stop()
        self.imu.stop()
        print("RovMovement has been terminated.")

    def close(self):
        self.stop()