From 3fc61db9424f6612624baad1df1df4fbd5aba2cb Mon Sep 17 00:00:00 2001
From: Tobias Winchen <tobias.winchen@rwth-aachen.de>
Date: Tue, 7 May 2019 13:10:39 +0000
Subject: [PATCH] Added VLBI application
---
psrdada_cpp/effelsberg/edd/src/VLBI_cli.cu | 139 +++++++++++++++++++++
1 file changed, 139 insertions(+)
create mode 100644 psrdada_cpp/effelsberg/edd/src/VLBI_cli.cu
diff --git a/psrdada_cpp/effelsberg/edd/src/VLBI_cli.cu b/psrdada_cpp/effelsberg/edd/src/VLBI_cli.cu
new file mode 100644
index 00000000..f7613eed
--- /dev/null
+++ b/psrdada_cpp/effelsberg/edd/src/VLBI_cli.cu
@@ -0,0 +1,139 @@
+#include "boost/program_options.hpp"
+#include "psrdada_cpp/cli_utils.hpp"
+#include "psrdada_cpp/common.hpp"
+#include "psrdada_cpp/dada_client_base.hpp"
+#include "psrdada_cpp/dada_input_stream.hpp"
+#include "psrdada_cpp/dada_null_sink.hpp"
+#include "psrdada_cpp/dada_output_stream.hpp"
+#include "psrdada_cpp/multilog.hpp"
+#include "psrdada_cpp/simple_file_writer.hpp"
+
+#include "psrdada_cpp/effelsberg/edd/VLBI.cuh"
+
+#include <ctime>
+#include <iostream>
+#include <time.h>
+
+
+using namespace psrdada_cpp;
+
+
+namespace {
+const size_t ERROR_IN_COMMAND_LINE = 1;
+const size_t SUCCESS = 0;
+const size_t ERROR_UNHANDLED_EXCEPTION = 2;
+} // namespace
+
+
+int main(int argc, char **argv) {
+ try {
+ key_t input_key;
+ unsigned int nbits;
+
+ size_t speadHeapSize;
+
+ std::time_t now = std::time(NULL);
+ std::tm *ptm = std::localtime(&now);
+ char buffer[32];
+ std::strftime(buffer, 32, "%Y-%m-%d-%H:%M:%S.bp", ptm);
+ std::string filename(buffer);
+ std::string output_type = "file";
+
+ /** Define and parse the program options
+ */
+ namespace po = boost::program_options;
+ po::options_description desc("Options");
+
+ desc.add_options()("help,h", "Print help messages");
+ desc.add_options()(
+ "input_key,i",
+ po::value<std::string>()->default_value("dada")->notifier(
+ [&input_key](std::string in) { input_key = string_to_key(in); }),
+ "The shared memory key for the dada buffer to connect to (hex "
+ "string)");
+ desc.add_options()(
+ "output_type", po::value<std::string>(&output_type)->default_value(output_type),
+ "output type [dada, file]. Default is file."
+ );
+ desc.add_options()(
+ "output_key,o", po::value<std::string>(&filename)->default_value(filename),
+ "The key of the output bnuffer / name of the output file to write spectra "
+ "to");
+ desc.add_options()("nbits,b", po::value<unsigned int>(&nbits)->required(),
+ "The number of bits per sample in the "
+ "packetiser output (8 or 12)");
+ desc.add_options()("speadheap_size",
+ po::value<size_t>()->default_value(4096)->notifier(
+ [&speadHeapSize](size_t in) { speadHeapSize = in; }),
+ "size of the spead data heaps. The number of the "
+ "heaps in the dada block depends on the number of "
+ "side channel items.");
+
+ desc.add_options()(
+ "log_level", po::value<std::string>()->default_value("info")->notifier(
+ [](std::string level) { set_log_level(level); }),
+ "The logging level to use "
+ "(debug, info, warning, "
+ "error)");
+
+ po::variables_map vm;
+ try {
+ po::store(po::parse_command_line(argc, argv, desc), vm);
+ if (vm.count("help")) {
+ std::cout << "VLBI -- Read EDD data from a DADA buffer "
+ "and convert it to 2 bit VLBI data in VDIF format"
+ << std::endl
+ << desc << std::endl;
+ return SUCCESS;
+ }
+
+ po::notify(vm);
+ if (vm.count("output_type") && (!(output_type == "dada" || output_type == "file") ))
+ {
+ throw po::validation_error(po::validation_error::invalid_option_value, "output_type", output_type);
+ }
+
+ } catch (po::error &e) {
+ std::cerr << "ERROR: " << e.what() << std::endl << std::endl;
+ std::cerr << desc << std::endl;
+ return ERROR_IN_COMMAND_LINE;
+ }
+
+ MultiLog log("edd::VLBI");
+ DadaClientBase client(input_key, log);
+ std::size_t buffer_bytes = client.data_buffer_size();
+
+ std::cout << "Running with output_type: " << output_type << std::endl;
+ if (output_type == "file")
+ {
+ SimpleFileWriter sink(filename);
+ effelsberg::edd::VLBI<decltype(sink)> vlbi(
+ buffer_bytes, nbits,
+ speadHeapSize, 5000, sink);
+
+ DadaInputStream<decltype(vlbi)> istream(input_key, log, vlbi);
+ istream.start();
+ }
+ else if (output_type == "dada")
+ {
+ DadaOutputStream sink(string_to_key(filename), log);
+ effelsberg::edd::VLBI<decltype(sink)> vlbi(
+ buffer_bytes, nbits,
+ speadHeapSize, 5000, sink);
+ DadaInputStream<decltype(vlbi)> istream(input_key, log, vlbi);
+ istream.start();
+ }
+ else
+ {
+ throw std::runtime_error("Unknown oputput-type");
+ }
+
+
+ } catch (std::exception &e) {
+ std::cerr << "Unhandled Exception reached the top of main: " << e.what()
+ << ", application will now exit" << std::endl;
+ return ERROR_UNHANDLED_EXCEPTION;
+ }
+ return SUCCESS;
+}
+
--
GitLab