SmartWheels self-driving race car. Designed for NXP Cup. Uses FRDM-KL25Z, area-scan camera, and simple image processing to detect and navigate any NXP spec track.

Dependencies:   TSI USBDevice mbed-dev

Fork of SmartWheels by haofan Zheng

main.cpp

Committer:
Bobymicjohn
Date:
2017-04-13
Revision:
67:1b5c8add3d01
Parent:
61:bdc31e4fa3c4
Parent:
66:c9b64f5337cc
Child:
69:34e97694f0ef

File content as of revision 67:1b5c8add3d01:

#include "mbed.h"

#include <math.h>
#define PI 3.14159265f

//#define SW_DEBUG
#include "SWCommon.h"
#include "GlobalVariable.h"

#include "Motor.h"
#include "Servo.h"
#include "WheelEncoder.h"
#include "Core.h"
#include "SWUSBServer.h"

#include "ArduCAM.h"
#include "ArduUTFT.h"
#include "IMUManager.h"

#include "PinAssignment.h"

SPI g_spi_port(PIN_SPI_MOSI, PIN_SPI_MISO, PIN_SPI_SCK);
SW::Core g_core;

#ifdef SW_DEBUG
#include <rtos.h>
Mutex g_sw_spi_lock;
#endif 

int main(void) {
    
    Timer timer;
    
    g_spi_port.frequency(5000000);
    //g_spi_port.format(8, 0);
    
    //SW::Core core;
    
    //Motor motor(g_core);
    //Servo servo(g_core);
    //WheelEncoder wheelEncoder(core);
    //Camera cam(core);
    motor_init();
    servo_init();
    
    bool isRegRead = false;
    
    
    ardu_utft_init();
    
    
    ardu_cam_init();
    
    
    //uint8_t IMUInitResult = imu_manager_init();
    //LOGI("IMU Init: %#x", IMUInitResult);
    //imu_manager_calibrate();
    //imu_manager_begin_tick();
    //wait(0.5);
    
    //timer.reset();
    timer.start();
    float timeWas = timer.read();
    
    servo_set_angle(0.0f);
    
    float speedRatio = 1.0f;
    float lastAngle = 0.0f;
    float cornerRatio = 1.0f;
    
    DebugCounter counter(10, PTE5);
    
    while (1) 
    {
        float deltaTime = timeWas;
        timeWas = timer.read();
        deltaTime = timeWas - deltaTime;
        
        //g_core.Update(deltaTime);
        
        //if(!isRegRead && g_core.GetUSBServer().GetStatus() == SER_STAT_RUNNING && timer.read() > 2.5f && ardu_cam_is_capture_finished())
        //{
            //ardu_cam_print_debug();
        //    isRegRead = true;
        //}
        
        //ardu_cam_display_img_utft();
        image_processing();
        uint8_t shouldTerminate = ardu_cam_get_is_encounter_terminate();
        
        if(shouldTerminate)
        {
            ardu_utft_print("F", 230, 100);
            servo_set_angle(0.0f);
            motor_set_right_speed(-1.0f);
            motor_set_left_speed(-1.0f);
            wait(0.5f);
            motor_set_right_speed(0.0f);
            motor_set_left_speed(0.0f);
            return 0;
        }
        //else
        //{
        //    ardu_utft_print(" ", 230, 100);
        //}
        
        const volatile uint8_t* centerLine = ardu_cam_get_center_array();
        
        /////////////Calculate the curvature:
        float srcY = (0.0f + 1.0f) / 2.0f;
        float srcX = static_cast<float>(centerLine[0] + centerLine[1]) / 2.0f;
        
        float destY = static_cast<float>(CAM_ROI_UPPER_LIMIT - 1 + (CAM_ROI_UPPER_LIMIT - 2)) / 2.0f;
        float destX = static_cast<float>(centerLine[CAM_ROI_UPPER_LIMIT - 1] + centerLine[CAM_ROI_UPPER_LIMIT - 2]) / 2.0f;
        
        float disY = destY - srcY;
        float disX = destX - srcX;
        
        float angleRadians = -1.0f * static_cast<float>(atan(static_cast<double>(disX / disY)));
        float angleDegrees = (angleRadians * (180.0f / PI));
        
        ///////////////////////////////////////
        
        /////////////Calculate the offset:
        float centerPos = static_cast<float>(RESOLUTION_WIDTH / 2);
        float offsetDegrees = ((srcX - centerPos) / centerPos) * SERVO_MAX_ANGLE;
        //////////////////////////////////////
        

        float totalAngleDegrees = (angleDegrees * 0.50f) + (offsetDegrees * (centerLine[2 * CAM_ROI_UPPER_LIMIT - 1] != BOTH_FOUND ? 0.90f : 0.35f));
        float totalAngleDegreesAbs = abs(totalAngleDegrees);
        if((totalAngleDegrees < 0 && centerLine[CAM_ROI_UPPER_LIMIT] == LEFT_FOUND) ||
            (totalAngleDegrees > 0 && centerLine[CAM_ROI_UPPER_LIMIT] == RIGHT_FOUND))
        {
            totalAngleDegrees = totalAngleDegrees * -1.0f;
        }
        /*
        if(totalAngleDegreesAbs >= 1.0f && totalAngleDegreesAbs <= 4.0)
        {   
            continue;
        }
        */
        servo_set_angle(totalAngleDegrees);
        
        
        if(totalAngleDegrees > SERVO_MAX_ANGLE)
            totalAngleDegrees = SERVO_MAX_ANGLE;
        else if(totalAngleDegrees < -SERVO_MAX_ANGLE)
            totalAngleDegrees = -SERVO_MAX_ANGLE;
        
        if(totalAngleDegrees > lastAngle)
        {
            cornerRatio =  cornerRatio * (lastAngle / totalAngleDegrees);
        }
        else if(totalAngleDegrees < lastAngle)
        {
            cornerRatio =  cornerRatio * (lastAngle / totalAngleDegrees);    
        }
        else
        {
            cornerRatio = 1.0f;    
        }
        
        if(totalAngleDegreesAbs > 18.0f)
        {
            speedRatio = 0.90f;
        }
        else if(totalAngleDegreesAbs > 15.0f)
        {
            speedRatio = 0.94f;
        }
        else if(totalAngleDegreesAbs > 12.0f)
        {
            speedRatio = 0.98f;
        }
        else
        {
            speedRatio = 1.0f;
        }
        
   
        if(totalAngleDegrees < 0.0f)
        {
            motor_set_left_speed(speedRatio * cornerRatio * ((MOTOR_DIFF_MIN_SPEED) + ((1.0f - MOTOR_DIFF_MIN_SPEED) * ((SERVO_MAX_ANGLE - abs(totalAngleDegrees)) / SERVO_MAX_ANGLE))));
            motor_set_right_speed(speedRatio * cornerRatio * 1.0f);    
        }
        else
        {
            motor_set_right_speed(speedRatio * cornerRatio * ((MOTOR_DIFF_MIN_SPEED) + ((1.0f - MOTOR_DIFF_MIN_SPEED) * ((SERVO_MAX_ANGLE - abs(totalAngleDegrees)) / SERVO_MAX_ANGLE))));
            motor_set_left_speed(speedRatio * cornerRatio * 1.0f);
        }

        
        lastAngle = totalAngleDegrees;
        
        //LOGI("FPS: %f", 1 / deltaTime);
        
        //imu_manager_init();
        //imu_manager_update();
        //float imuTemp = imu_manager_get_temp();
        
        //const volatile struct imu_vec3* AccelV = imu_manager_get_accl();
        
        //LOGI("A: %5.3f, %5.3f, %5.3f.T%5.2f ", AccelV->x, AccelV->y, AccelV->z, imuTemp);
        
        //const volatile struct imu_vec3* VelocityV = imu_manager_get_velocity();
        
        //LOGI("V: %5.3f, %5.3f, %5.3f.T%5.2f ", VelocityV->x, VelocityV->y, VelocityV->z, imuTemp);
        
        //const volatile struct imu_vec3* PositionV = imu_manager_get_position();
        
        //LOGI("P: %5.3f, %5.3f, %5.3f ", PositionV->x, PositionV->y, PositionV->z);
        
        counter.Update();
        
        /*
        //////// Steer Vehicle / Adjust Speed for Differential //////////
        servo_set_angle((angleDegrees * -0.3f) + (offsetPercent * SERVO_MAX_ANGLE * 0.3f));
        
        if(angleRadians > 0.366)
            angleRadians = 0.366;
        else if(angleRadians < -0.366)
            angleRadians = -0.366;
        
        if(angleRadians < 0)
        {
            motor_set_left_speed((0.5) + (0.5) * ((0.366 - abs(angleRadians))/0.366));
            motor_set_right_speed(1.0);    
        }
        else
        {
            motor_set_right_speed((0.5) + (0.5) * ((0.366 - abs(angleRadians))/0.366));
            motor_set_left_speed(1.0);
        }
        */
        
        
        //char buf[20];
        //sprintf(buf, "angle %f", angle);
        //g_core.GetUSBServer().PushUnreliableMsg('D', buf);
        /*
        std::string tempStr = "XX";
        for(uint8_t i = 0; i < CAM_ROI_UPPER_LIMIT; ++i)
        {
            tempStr[0] = i;
            tempStr[1] = centerLine[i];
            g_core.GetUSBServer().PushUnreliableMsg('L', tempStr);
        }
        */
        
        /*
        if(offsetPercent > 0.1)
        {
            motor_set_speeds(0.05f, 0.05f);
        }
        else
        {
            motor_set_speeds(0.13f, 0.13f);
        }
        */
        //wait(0.01);
    }
}

/*
PwmOut servo(PTE20);
 
int main() {
    servo.period(0.020);          // servo requires a 20ms period
    
    while (1) {
        for(float offset=0.0; offset<0.001; offset+=0.0001) {
            servo.pulsewidth(0.001 + offset); // servo position determined by a pulsewidth between 1-2ms
            wait(0.25);
        }
    }
    
}
*/

/*  //code for accelerometer sensor.
    const char regAddr = 0x0D;
    char readValue = 0;
    int result1 = m_sccbCtrl.write(0x1D<<1, &regAddr, 1, true);
    int result2 = m_sccbCtrl.read(0x1D<<1, &readValue, 1, false);
    char buf[20];
    sprintf(buf, "%#x-%#x-%d-%d", regAddr, readValue, result1, result2);
    m_core.GetUSBServer().PushUnreliableMsg('D', buf);
*/