summaryrefslogtreecommitdiffstats
path: root/include/usrp_uhd/utils.hpp
blob: ef236245311e97a9dd6211e379a53325ee6792f7 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
//
// Copyright 2010 Ettus Research LLC
//

#include <usrp_uhd/wax.hpp>
#include <boost/foreach.hpp>
#include <boost/format.hpp>
#include <boost/function.hpp>
#include <stdexcept>
#include <algorithm>
#include <vector>
#include <map>

#ifndef INCLUDED_USRP_UHD_UTILS_HPP
#define INCLUDED_USRP_UHD_UTILS_HPP

namespace usrp_uhd{

template <class Key, class T> //TODO template this better
std::vector<Key> get_map_keys(const std::map<Key, T> &m){
    std::vector<Key> v;
    std::pair<Key, T> p;
    BOOST_FOREACH(p, m){
        v.push_back(p.first);
    }
    return v;
}

template<typename T> T signum(T n){
    if (n < 0) return -1;
    if (n > 0) return 1;
    return 0;
}

inline void tune(
    freq_t target_freq,
    freq_t lo_offset,
    wax::proxy subdev_freq_proxy,
    bool subdev_quadrature,
    bool subdev_spectrum_inverted,
    bool subdev_is_tx,
    wax::proxy dsp_freq_proxy,
    freq_t dsp_sample_rate
){
    // Ask the d'board to tune as closely as it can to target_freq+lo_offset
    subdev_freq_proxy = target_freq + lo_offset;
    freq_t inter_freq = wax::cast<freq_t>(subdev_freq_proxy);

    // Calculate the DDC setting that will downconvert the baseband from the
    // daughterboard to our target frequency.
    freq_t delta_freq = target_freq - inter_freq;
    int delta_sign = signum(delta_freq);
    delta_freq *= delta_sign;
    delta_freq = fmod(delta_freq, dsp_sample_rate);
    bool inverted = delta_freq > dsp_sample_rate/2.0;
    freq_t dxc_freq = inverted? (delta_freq - dsp_sample_rate) : (-delta_freq);
    dxc_freq *= delta_sign;

    // If the spectrum is inverted, and the daughterboard doesn't do
    // quadrature downconversion, we can fix the inversion by flipping the
    // sign of the dxc_freq...  (This only happens using the basic_rx board)
    if (subdev_spectrum_inverted){
        inverted = not inverted;
    }
    if (inverted and not subdev_quadrature){
        dxc_freq = -dxc_freq;
        inverted = not inverted;
    }
    if (subdev_is_tx){
        dxc_freq = -dxc_freq;	// down conversion versus up conversion
    }

    dsp_freq_proxy = dxc_freq;
    //freq_t actual_dxc_freq = wax::cast<freq_t>(dsp_freq_proxy);

    //return some kind of tune result tuple/struct
}

} //namespace usrp_uhd

/*!
 * Useful templated functions and classes that I like to pretend are part of stl
 */
namespace std{

    class assert_error : public std::logic_error{
    public:
        explicit assert_error(const string& what_arg) : logic_error(what_arg){
            /* NOP */
        }
    };

    #define ASSERT_THROW(_x) if (not (_x)) { \
        throw std::assert_error("Assertion Failed: " + std::string(#_x)); \
    }

    template<class T, class InputIterator, class Function>
    T reduce(InputIterator first, InputIterator last, Function fcn, T init = 0){
        T tmp = init;
        for ( ; first != last; ++first ){
            tmp = fcn(tmp, *first);
        }
        return tmp;
    }

    template<class T, class InputIterator>
    bool has(InputIterator first, InputIterator last, const T &elem){
        return last != std::find(first, last, elem);
    }

    template<class T>
    T sum(const T &a, const T &b){
        return a + b;
    }

}//namespace std

#endif /* INCLUDED_USRP_UHD_UTILS_HPP */