Bastian Bloessl
79 lines
2.2 KiB
C++
79 lines
2.2 KiB
C++
/* -*- c++ -*- */
|
|
/*
|
|
* Copyright 2021 Anton Ottosson.
|
|
*
|
|
* This 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.
|
|
*
|
|
* This software 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 software; see the file COPYING. If not, write to
|
|
* the Free Software Foundation, Inc., 51 Franklin Street,
|
|
* Boston, MA 02110-1301, USA.
|
|
*/
|
|
|
|
#ifdef HAVE_CONFIG_H
|
|
#include "config.h"
|
|
#endif
|
|
|
|
#include <gnuradio/io_signature.h>
|
|
#include "extract_csi_impl.h"
|
|
|
|
namespace gr {
|
|
namespace ieee802_11 {
|
|
|
|
extract_csi::sptr extract_csi::make() {
|
|
return gnuradio::get_initial_sptr(new extract_csi_impl());
|
|
}
|
|
|
|
|
|
extract_csi_impl::extract_csi_impl()
|
|
: gr::sync_block("extract_csi", gr::io_signature::make(0, 0, 0),
|
|
gr::io_signature::make(1, 1, 52 * sizeof(gr_complex))) {
|
|
message_port_register_in(pmt::mp("pdu in"));
|
|
}
|
|
|
|
extract_csi_impl::~extract_csi_impl() {}
|
|
|
|
int extract_csi_impl::work(int noutput_items,
|
|
gr_vector_const_void_star &input_items,
|
|
gr_vector_void_star &output_items) {
|
|
pmt::pmt_t pdu(delete_head_nowait(pmt::mp("pdu in")));
|
|
|
|
if (pdu.get() == NULL)
|
|
return 0;
|
|
|
|
if (!pmt::is_pair(pdu) || !pmt::is_dict(pmt::car(pdu)))
|
|
throw std::runtime_error("received a malformed pdu message");
|
|
|
|
d_meta = pmt::car(pdu);
|
|
|
|
if (!pmt::dict_has_key(d_meta, pmt::mp("csi")))
|
|
return 0;
|
|
|
|
d_csi = pmt::c32vector_elements(
|
|
pmt::dict_ref(d_meta, pmt::mp("csi"), pmt::PMT_NIL));
|
|
|
|
gr_complex *out = (gr_complex *)output_items[0];
|
|
|
|
std::copy(d_csi.begin(), d_csi.end(), out);
|
|
|
|
pmt::pmt_t keys(pmt::dict_keys(d_meta));
|
|
for (int i = 0; i < pmt::length(keys); i++) {
|
|
pmt::pmt_t k(pmt::nth(i, keys));
|
|
pmt::pmt_t v(pmt::dict_ref(d_meta, k, pmt::PMT_NIL));
|
|
add_item_tag(0, nitems_written(0), k, v, alias_pmt());
|
|
}
|
|
|
|
return 1;
|
|
}
|
|
|
|
} /* namespace ieee802_11 */
|
|
} /* namespace gr */
|