//
// Copyright 2010-2011,2014 Ettus Research LLC
//
// This program 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 of the License, or
// (at your option) any later version.
//
// This program 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, see <http://www.gnu.org/licenses/>.
//
//----------------------------------------------------------------------
// JITC Testing
//
// Allen-Vanguard Corporation, 2016
//
// 2016-02-25
// 2018-02-08 updated by KSK
//
// An example of packet scheduling based on the Ettus Research example
//   tx_timed_samples.cpp
//

#include <uhd/utils/thread_priority.hpp>
#include <uhd/utils/safe_main.hpp>
#include <uhd/usrp/multi_usrp.hpp>
#include <boost/program_options.hpp>
#include <boost/thread/thread.hpp>
#include <boost/format.hpp>
#include <boost/date_time/posix_time/posix_time.hpp>
#include <iostream>
#include <complex>

namespace po = boost::program_options;

int UHD_SAFE_MAIN(int argc, char *argv[]){
  uhd::set_thread_priority_safe();

  //variables to be set by po
  std::string args;
  std::string wire;
  std::string timesrc;
  std::string clksrc;
  double seconds_in_future;
  double interval;
  double gain;
  size_t total_num_samps;
  double rate;
  float ampl;

  //setup the program options
  po::options_description desc("Allowed options");
  desc.add_options()
    ("help", "help message")
    ("args", po::value<std::string>(&args)->default_value(""), "single uhd device address args")
    ("wire", po::value<std::string>(&wire)->default_value(""), "the over the wire type, sc16, sc8, etc")
    ("gain", po::value<double>(&gain)->default_value(20), "gain for the RF chain [0-89.5]")
    ("secs", po::value<double>(&seconds_in_future)->default_value(4.0), "number of seconds in the future to start transmitting")
    ("interval", po::value<double>(&interval)->default_value(0.01), "period of timed transmit in seconds")
    ("nsamps", po::value<size_t>(&total_num_samps)->default_value(10000), "total number of samples to transmit")
    ("rate", po::value<double>(&rate)->default_value(100e6/16), "rate of outgoing samples")
    ("ampl", po::value<float>(&ampl)->default_value(float(0.3)), "amplitude of each sample")
    ("dilv", "specify to disable inner-loop verbose")
    ("timesrc", po::value<std::string>(&timesrc)->default_value("internal"), "specify the time source [internal(default),external]")
    ("clksrc", po::value<std::string>(&clksrc)->default_value("internal"), "specify the clock source [internal(default),external,MIMO]")
    ;
  po::variables_map vm;
  po::store(po::parse_command_line(argc, argv, desc), vm);
  po::notify(vm);

  //print the help message
  if (vm.count("help")){
    std::cout << boost::format("UHD TX Timed Samples %s") % desc << std::endl;
    return ~0;
  }

  bool verbose = vm.count("dilv") == 0;

  //create a usrp device
  std::cout << std::endl;
  std::cout << boost::format("Creating the usrp device with: %s...") % args << std::endl;
  uhd::usrp::multi_usrp::sptr usrp = uhd::usrp::multi_usrp::make(args);
  std::cout << boost::format("Using Device: %s") % usrp->get_pp_string() << std::endl;

  //set the rf gain
  if (vm.count("gain")){
    std::cout << boost::format("Setting TX Gain: %f ...") % gain << std::endl;
    usrp->set_tx_gain(gain);
    std::cout << boost::format("Actual TX Gain: %f ...") % usrp->get_tx_gain() << std::endl << std::endl;
  }

  //set the tx sample rate
  std::cout << boost::format("Setting TX Rate: %f Msps...") % (rate/1e6) << std::endl;
  usrp->set_tx_rate(rate);
  std::cout << boost::format("Actual TX Rate: %f Msps...") % (usrp->get_tx_rate()/1e6) << std::endl << std::endl;


  // set the 1 PPS source
  usrp->set_time_source(timesrc);
  // set the clock source
  usrp->set_clock_source(clksrc);
  // set the time source
  usrp->set_time_source(timesrc);
  std::cout << "clock source : " << usrp->get_clock_source(0) << std::endl;
  std::cout << "time source : " << usrp->get_time_source(0) << std::endl;
  std::cout << boost::format("Setting device timestamp to 0...") << std::endl;
  usrp->set_time_unknown_pps(uhd::time_spec_t(0.0));

  uhd::tune_request_t tune_request(100e6);
  usrp->set_tx_freq(tune_request);
  std::cout << "tx freq	 : " << usrp->get_tx_freq(0) << std::endl;

  //create a transmit streamer
  uhd::stream_args_t stream_args("fc32", wire); //complex floats
  uhd::tx_streamer::sptr tx_stream = usrp->get_tx_stream(stream_args);

  //allocate buffer with data to send
  //    std::vector<std::complex<float> > buff(tx_stream->get_max_num_samps(), std::complex<float>(ampl, ampl));
  std::vector<std::complex<float> > buff(total_num_samps, std::complex<float>(ampl, ampl));

  uhd::time_spec_t ts = usrp->get_time_now();

  //setup metadata for the first packet
  uhd::tx_metadata_t md;
  uhd::async_metadata_t async_md;
  bool got_async_burst_ack = false;

  //the first call to send() will block this many seconds before sending:
  double timeout = seconds_in_future + interval; //timeout (delay before transmit + padding)
  double secsFromNow = seconds_in_future;
  size_t num_tx_samps = 0; //number of accumulated samples

  md.start_of_burst = false;
  md.end_of_burst = false;

  boost::posix_time::ptime mst1; // = boost::posix_time::microsec_clock::local_time();
  boost::posix_time::time_duration secs_since_epoch;
  boost::posix_time::ptime time_t_epoch(boost::gregorian::date(1970,1,1));
  std::cout << time_t_epoch << std::endl;

  while (1)
  {
    md.start_of_burst = true;
    md.end_of_burst = false;
    md.has_time_spec = true;

    md.time_spec = uhd::time_spec_t(secsFromNow);
    std::cout << "transmitting " << buff.size() << " sample packet at " << secsFromNow;
    num_tx_samps = tx_stream->send (&buff.front (),
      buff.size (), md, timeout);

    // send a length zero packet to end the burst
    md.start_of_burst = false;
    md.end_of_burst = true;
    md.has_time_spec = false;
    num_tx_samps = tx_stream->send ("", 0, md, timeout);

    got_async_burst_ack = false;
    //loop through all messages for the ACK packet (may have underflow messages in queue)
    while (not got_async_burst_ack and tx_stream->recv_async_msg(async_md, timeout)){
      got_async_burst_ack = (async_md.event_code == uhd::async_metadata_t::EVENT_CODE_BURST_ACK);
    }
    if (!got_async_burst_ack)
      std::cout << " ------------- ack failed" << std::endl;
    timeout = interval+interval/10;;
    ts = usrp->get_time_now();
    std::cout << " completed at " << ts.get_real_secs() << " s" << std::endl;

    secsFromNow += interval;
  }

  //finished; never gets here...
  std::cout << std::endl << "Done!" << std::endl << std::endl;

  return EXIT_SUCCESS;
}