diff options
Diffstat (limited to 'usrp/host/lib/inband/usrp_rx.cc')
-rw-r--r-- | usrp/host/lib/inband/usrp_rx.cc | 142 |
1 files changed, 142 insertions, 0 deletions
diff --git a/usrp/host/lib/inband/usrp_rx.cc b/usrp/host/lib/inband/usrp_rx.cc new file mode 100644 index 000000000..7344a6e89 --- /dev/null +++ b/usrp/host/lib/inband/usrp_rx.cc @@ -0,0 +1,142 @@ +/* -*- c++ -*- */ +/* + * Copyright 2007 Free Software Foundation, Inc. + * + * This file is part of GNU Radio + * + * GNU Radio is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 3, or (at your option) + * any later version. + * + * GNU Radio is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License along + * with this program; if not, write to the Free Software Foundation, Inc., + * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. + */ + +#ifdef HAVE_CONFIG_H +#include <config.h> +#endif + +#include <usrp_rx.h> + +#include <usrp_standard.h> +#include <iostream> +#include <vector> +#include <usb.h> +#include <mb_class_registry.h> +#include <usrp_inband_usb_packet.h> +#include <fpga_regs_common.h> +#include <stdio.h> + +#include <symbols_usrp_rx_cs.h> + +typedef usrp_inband_usb_packet transport_pkt; + +static const bool verbose = false; + +usrp_rx::usrp_rx(mb_runtime *rt, const std::string &instance_name, pmt_t user_arg) + : mb_mblock(rt, instance_name, user_arg), + d_disk_write(false) +{ + d_cs = define_port("cs", "usrp-rx-cs", true, mb_port::EXTERNAL); + + //d_disk_write=true; + + if(d_disk_write) { + d_ofile.open("rx_data.dat",std::ios::binary|std::ios::out); + d_cs_ofile.open("rx_cs.dat",std::ios::binary|std::ios::out); + } + +} + +usrp_rx::~usrp_rx() +{ + if(d_disk_write) { + d_ofile.close(); + d_cs_ofile.close(); + } +} + +void +usrp_rx::initial_transition() +{ + +} + +void +usrp_rx::handle_message(mb_message_sptr msg) +{ + pmt_t event = msg->signal(); + pmt_t port_id = msg->port_id(); + pmt_t data = msg->data(); + + // Theoretically only have 1 message to ever expect, but + // want to make sure its at least what we want + if(pmt_eq(port_id, d_cs->port_symbol())) { + + if(pmt_eqv(event, s_cmd_usrp_rx_start_reading)) + read_and_respond(data); + } +} + +void +usrp_rx::read_and_respond(pmt_t data) +{ + size_t ignore; + bool underrun; + unsigned int n_read; + unsigned int pkt_size = sizeof(transport_pkt); + + pmt_t invocation_handle = pmt_nth(0, data); + + // Need the handle to the RX port to send responses, this is passed + // by the USRP interface m-block + d_urx = + boost::any_cast<usrp_standard_rx *>(pmt_any_ref(pmt_nth(1, data))); + + if(verbose) + std::cout << "[usrp_rx] Waiting for packets..\n"; + + // Read by 512 which is packet size and send them back up + while(1) { + + pmt_t v_pkt = pmt_make_u8vector(pkt_size, 0); + transport_pkt *pkt = + (transport_pkt *) pmt_u8vector_writeable_elements(v_pkt, ignore); + + n_read = d_urx->read(pkt, pkt_size, &underrun); + + if(n_read != pkt_size) { + std::cerr << "[usrp_rx] Error reading packet, shutting down\n"; + d_cs->send(s_response_usrp_rx_read, + pmt_list3(PMT_NIL, PMT_F, PMT_NIL)); + return; + } + + if(underrun) + std::cout << "[usrp_rx] Underrun\n"; + + d_cs->send(s_response_usrp_rx_read, + pmt_list3(PMT_NIL, PMT_T, v_pkt)); + if(verbose) + std::cout << "[usrp_rx] Read 1 packet\n"; + + if(d_disk_write) { + if(pkt->chan() == 0x1f) + d_cs_ofile.write((const char *)pkt, transport_pkt::max_pkt_size()); + else + d_ofile.write((const char *)pkt, transport_pkt::max_pkt_size()); + + d_cs_ofile.flush(); + d_ofile.flush(); + } + } +} + +REGISTER_MBLOCK_CLASS(usrp_rx); |