source: flair-src/trunk/lib/FlairSensorActuator/src/SimuImu.cpp@ 10

// %flair:license{
// This file is part of the Flair framework distributed under the
// CECILL-C License, Version 1.0.
// %flair:license}
//  created:    2014/02/07
//  filename:   SimuImu.cpp
//
//  author:     Guillaume Sanahuja
//              Copyright Heudiasyc UMR UTC/CNRS 7253
//
//  version:    $Id: $
//
//  purpose:    Class for a simulation imu
//
//
/*********************************************************************/

#include "SimuImu.h"
#include <FrameworkManager.h>
#include <ImuData.h>
#include <SpinBox.h>
#include <GroupBox.h>
#include <cvmatrix.h>
#include <SharedMem.h>
#include <AhrsData.h>
#include <sstream>

using std::string;
using std::ostringstream;
using namespace flair::core;
using namespace flair::gui;

namespace flair { namespace sensor {

SimuImu::SimuImu(const FrameworkManager* parent,string name,uint32_t dev_id,uint8_t priority) :Imu(parent,name),Thread(parent,name,priority) {
    data_rate=new SpinBox(GetGroupBox()->NewRow(),"data rate"," Hz",1,500,1,200);
    ahrsData=new AhrsData((Imu*)this);

    ostringstream dev_name;
    dev_name << "simu_imu_" << dev_id;
    shmem=new SharedMem((Thread*)this,dev_name.str().c_str(),sizeof(imu_states_t));
}

SimuImu::SimuImu(const IODevice* parent,string name,uint32_t dev_id) :Imu(parent,name),Thread(parent,name,0) {
    data_rate=NULL;

    ostringstream dev_name;
    dev_name << "simu_imu_" << dev_id;
    shmem=new SharedMem((Thread*)this,dev_name.str().c_str(),sizeof(imu_states_t));
}

SimuImu::~SimuImu() {
    SafeStop();
    Join();
}

void SimuImu::UpdateFrom(const io_data *data) {
    if(data!=NULL) {
        cvmatrix *input=(cvmatrix*)data;
        imu_states_t state;

        input->GetMutex();
        state.q0=input->ValueNoMutex(0,0);
        state.q1=input->ValueNoMutex(1,0);
        state.q2=input->ValueNoMutex(2,0);
        state.q3=input->ValueNoMutex(3,0);
        state.wx=input->ValueNoMutex(7,0);
        state.wy=input->ValueNoMutex(8,0);
        state.wz=input->ValueNoMutex(9,0);
        input->ReleaseMutex();

        shmem->Write((char*)&state,sizeof(imu_states_t));
    }
}

void SimuImu::Run(void) {
    imu_states_t state;
    ImuData* imuData;
    GetDatas(&imuData);

    if(data_rate==NULL) {
        Thread::Err("not applicable for simulation part.\n");
        return;
    }

    SetPeriodUS((uint32_t)(1000000./data_rate->Value()));

    while(!ToBeStopped()) {
        WaitPeriod();

        if(data_rate->ValueChanged()==true) {
            SetPeriodUS((uint32_t)(1000000./data_rate->Value()));
        }

        shmem->Read((char*)&state,sizeof(imu_states_t));
        ahrsData->SetQuaternionAndAngularRates(Quaternion(state.q0,state.q1,state.q2,state.q3)
                                               ,Vector3D(state.wx,state.wy,state.wz));

        imuData->SetDataTime(GetTime());
        ahrsData->SetDataTime(GetTime());

        UpdateImu();
        ProcessUpdate(ahrsData);
    }
}

} // end namespace sensor
} // end namespace flair