usrp.h

Go to the documentation of this file.

#ifndef USRP_H
#define USRP_H

#include <uhd/usrp/multi_usrp.hpp>
#include <uhd/convert.hpp>
#include <uhd/stream.hpp>
#include <semaphore.h>

namespace fun
{
    struct usrp_params
    {
        double freq;                
        double rate;                
        double tx_gain;             
        double rx_gain;             
        double tx_amp;              
        std::string device_addr;    

        usrp_params(double freq = 5.72e9, double rate = 5e6, double tx_gain=20, double rx_gain=20, double tx_amp=1.0, std::string device_addr="") :
            freq(freq),
            rate(rate),
            tx_gain(tx_gain),
            rx_gain(rx_gain),
            tx_amp(tx_amp),
            device_addr(device_addr)
        {
        }
    };

    class usrp
    {
    public:

        usrp(usrp_params params);


        // Send a burst of samples, and block until the burst has finished
        void send_burst_sync(std::vector<std::complex<double> > samples);

        void send_burst(std::vector<std::complex<double> > samples);

        // Get some samples from the USRP
        void get_samples(int num_samples, std::vector<std::complex<double> > & buffer);


    private:


        usrp_params m_params; 

        boost::shared_ptr<uhd::usrp::multi_usrp> m_usrp; 
        boost::shared_ptr<uhd::device> m_device;         
        uhd::rx_streamer::sptr m_rx_streamer;            
        uhd::tx_streamer::sptr m_tx_streamer;            

        sem_t m_tx_sem;                                  
    };

}


#endif // USRP_H