Revision ea29304b examples/yarp_icub/src/icub_data_receiver.cpp

View differences:

examples/yarp_icub/src/icub_data_receiver.cpp
1 1
#include "icub_data_receiver.h"
2 2
#include <humotion/server/joint_interface.h>
3 3
#include <yarp/os/Property.h>
4
#include <boost/format.hpp>
4 5
//using namespace yarp::dev;
5 6
//using namespace yarp::sig;
6 7
//using namespace yarp::os;
......
11 12
using yarp::dev::IEncodersTimed;
12 13
using yarp::sig::Vector;
13 14

  
15
#define ICUB_DATA_RECEIVER_USE_ENCODERSPEED 0
16

  
14 17
iCubDataReceiver::iCubDataReceiver(int period, iCubJointInterface *icub_jointinterface)
15 18
    : yarp::os::RateThread(period) {
16 19

  
17 20
    // store pointer to icub jointinterface
18 21
    icub_jointinterface_ = icub_jointinterface;
19 22

  
20
    //fetch yarp iencs view:
23
    //fetch iencs view from yarp
21 24
    yarp::dev::PolyDriver *poly_driver = icub_jointinterface->get_yarp_polydriver();
22
    bool success = poly_driver->view(yarp_iencs_);
25
    bool success = poly_driver->view(iencs_);
23 26
    if (!success) {
24 27
        cout << "ERROR: polydriver failed to init iencs view\n";
25 28
        exit(EXIT_FAILURE);
26 29
    }
27 30

  
28
    // resize data storage vectors to match the number of axes:
31
    // resize data storage vectors to match the number of axes
29 32
    int joints;
30
    yarp_iencs_->getAxes(&joints);
31
    yarp_positions_.resize(joints);
32
    yarp_timestamps_.resize(joints);
33
    iencs_->getAxes(&joints);
34
    positions_.resize(joints);
35
    velocities_.resize(joints);
36
    timestamps_.resize(joints);
33 37
}
34 38

  
35 39
bool iCubDataReceiver::threadInit() {
......
39 43
void iCubDataReceiver::threadRelease() {
40 44
}
41 45

  
46
yarp::sig::Vector iCubDataReceiver::calculate_velocities(yarp::sig::Vector positions,
47
                                                         yarp::sig::Vector timestamps) {
48
    yarp::sig::Vector velocities;
49
    velocities.resize(positions.size());
50

  
51
    cout << "\n";
52
    if (previous_positions_.size() == 0){
53
        // first run, no valid old position available, return zero velocities
54
        // by setting all all elements to zero
55
        velocities = 0.0;
56
    }else{
57
        // calculate speed based on positions:
58
        for(int i=0; i<positions.size(); i++){
59
            float diff = positions[i] - previous_positions_[i];
60
            float timediff = timestamps[i] - previous_timestamps_[i];
61
            // calc speed:
62
            velocities[i] = diff / timediff;
63
            cout << " [" << i << "]=" <<  boost::format("%6.3f") % velocities[i];
64
        }
65
    }
66
    cout << " VEL\n";
67

  
68
    previous_positions_ = positions;
69
    previous_timestamps_ = timestamps;
70

  
71
    return velocities;
72
}
73

  
42 74
void iCubDataReceiver::run() {
75
    float velocity;
76

  
43 77
    //grab pos+vel data:
44
    yarp_iencs_->getEncodersTimed(yarp_positions_.data(), yarp_timestamps_.data());
45
    //iencs->getEncoderSpeeds(velocities.data());
78
    iencs_->getEncodersTimed(positions_.data(), timestamps_.data());
46 79

  
47
    //publish data to humotion
48
    for(int i=0; i<yarp_positions_.size(); i++){
49
        store_incoming_position(i, yarp_positions_[i], yarp_timestamps_[i]);
80
    #if ICUB_DATA_RECEIVER_USE_ENCODERSPEED
81
    // fetch data from icub. NOTE: make sure to enable the vel broadcast in the ini file!
82
    iencs_->getEncoderSpeeds(velocities_.data());
83
    #else
84
    // manually calculate the speed based on old position:
85
    velocities_ = calculate_velocities(positions_, timestamps_);
86
    #endif
87

  
88
    // publish data to humotion
89
    for(int i=0; i<positions_.size(); i++){
90
        // store position values
91
        store_incoming_position(i, positions_[i], timestamps_[i]);
92
        // store velocity
93
        store_incoming_velocity(i, velocities_[i], timestamps_[i]);
50 94
    }
51 95
    
52
    printf("\n");
53

  
54 96
    //small hack to tell humotion to update the lid angle
55 97
    //fixme: use real id
56
    store_incoming_position(100, 0.0, yarp_timestamps_[0]);
98
    store_incoming_position(100, 0.0, timestamps_[0]);
57 99
}
58 100

  
59 101
void iCubDataReceiver::store_incoming_position(int icub_id, double value, double timestamp) {
60
    cout << "iCubDataReceiver::store_incoming_position(icubid=" << icub_id << ", " << value << "..)\n";
102
    cout << "iCubDataReceiver::store_incoming_position(icub=" << icub_id << ", " << value << ")\n";
61 103

  
62 104
    // store joint position in humotion backend
63 105
    if ((icub_id == iCubJointInterface::ICUB_ID_EYES_PAN) ||
......
93 135
        icub_jointinterface_->store_incoming_position(humotion_id, value, timestamp);
94 136
    }
95 137
}
138

  
139
void iCubDataReceiver::store_incoming_velocity(int icub_id, double velocity, double timestamp) {
140
    cout << "iCubDataReceiver::store_incoming_velocity(icub=" << icub_id << ", " << velocity << ")\n";
141

  
142
    // store joint position in humotion backend
143
    if ((icub_id == iCubJointInterface::ICUB_ID_EYES_PAN) ||
144
            (icub_id == iCubJointInterface::ICUB_ID_EYES_VERGENCE)) {
145
        // the icub handles eyes differently
146
        // instead of using seperate left/right pan the icub uses
147
        // a combined pan angle and vergence. therfore we have to convert this here:
148
        if (icub_id == iCubJointInterface::ICUB_ID_EYES_PAN) {
149
            target_eye_pan_velocity_ = velocity;
150
        } else {
151
            target_eye_vergence_velocity_ = -velocity;
152
        }
153

  
154
        float left  = target_eye_pan_velocity_ + target_eye_vergence_velocity_/2.0;
155
        float right = target_eye_pan_velocity_ - target_eye_vergence_velocity_/2.0;
156

  
157
        icub_jointinterface_->store_incoming_velocity(JointInterface::ID_EYES_LEFT_LR,
158
                                                     left, timestamp);
159
        icub_jointinterface_->store_incoming_velocity(JointInterface::ID_EYES_RIGHT_LR,
160
                                                     right, timestamp);
161
    } else if (icub_id == 100) {
162
        //HACK
163
        //icub_jointinterface->store_incoming_position(ID_EYES_RIGHT_LID_UPPER,
164
        //                                             lid_angle, timestamp);
165
    } else {
166
        if (icub_id == iCubJointInterface::ID_NECK_PAN) {
167
            // icub uses an inverted neck pan specification
168
            velocity = -velocity;
169
        }
170

  
171
        // known configured mapping between joint ids
172
        int humotion_id = icub_jointinterface_->convert_icub_jointid_to_humotion(icub_id);
173
        icub_jointinterface_->store_incoming_velocity(humotion_id, velocity, timestamp);
174
    }
175
}
176

  
177

  

Also available in: Unified diff