/* Streaming data from STDIN/OUT to FPGA.
 *
 * Author: Daniel Krebs <github@daniel-krebs.net>
 * Author: Niklas Eiling <niklas.eiling@eonerc.rwth-aachen.de>
 * SPDX-FileCopyrightText: 2017 Steffen Vogel <post@steffenvogel.de>
 * SPDX-FileCopyrightText: 2022-2023 Niklas Eiling <niklas.eiling@eonerc.rwth-aachen.de>
 * SPDX-License-Identifier: Apache-2.0
 */

#include <algorithm>
#include <iostream>
#include <jansson.h>
#include <string>
#include <thread>
#include <vector>

#include <CLI11.hpp>
#include <rang.hpp>

#include <villas/exceptions.hpp>
#include <villas/log.hpp>
#include <villas/utils.hpp>

#include <villas/fpga/card.hpp>
#include <villas/fpga/core.hpp>
#include <villas/fpga/ips/aurora_xilinx.hpp>
#include <villas/fpga/ips/dino.hpp>
#include <villas/fpga/ips/dma.hpp>
#include <villas/fpga/ips/i2c.hpp>
#include <villas/fpga/ips/rtds.hpp>
#include <villas/fpga/ips/switch.hpp>
#include <villas/fpga/utils.hpp>
#include <villas/fpga/vlnv.hpp>

using namespace villas;

static std::shared_ptr<kernel::devices::PciDeviceList> pciDevices;
static auto logger = villas::Log::get("ctrl");

void writeToDmaFromStdIn(std::shared_ptr<villas::fpga::ip::Dma> dma) {
  auto &alloc = villas::HostRam::getAllocator();
  const std::shared_ptr<villas::MemoryBlock> block =
      alloc.allocateBlock(0x200 * sizeof(float));
  villas::MemoryAccessor<float> mem = *block;
  dma->makeAccesibleFromVA(block);

  logger->info("Please enter values to write to the device, separated by ';'");

  while (true) {
    // Read values from stdin
    std::string line;
    std::getline(std::cin, line);
    auto values = villas::utils::tokenize(line, ";");

    size_t i = 0;
    for (auto &value : values) {
      if (value.empty())
        continue;

      const float number = std::stof(value);
      mem[i++] = number;
    }

    // Initiate write transfer
    bool state = dma->write(*block, i * sizeof(float));
    if (!state)
      logger->error("Failed to write to device");

    auto writeComp = dma->writeComplete();
    logger->debug("Wrote {} bytes", writeComp.bytes);
  }
}

void readFromDmaToStdOut(
    std::shared_ptr<villas::fpga::ip::Dma> dma,
    std::unique_ptr<fpga::BufferedSampleFormatter> formatter) {

  auto &alloc = villas::HostRam::getAllocator();

  const std::shared_ptr<villas::MemoryBlock> block[] = {
      alloc.allocateBlock(0x200 * sizeof(uint32_t)),
      alloc.allocateBlock(0x200 * sizeof(uint32_t))};
  villas::MemoryAccessor<int32_t> mem[] = {*block[0], *block[1]};

  for (auto b : block) {
    dma->makeAccesibleFromVA(b);
  }

  size_t cur = 0, next = 1;
  std::ios::sync_with_stdio(false);

  // Setup read transfer
  dma->read(*block[0], block[0]->getSize());

  int cnt = 0;
  while (true) {
    logger->trace("Read from stream and write to address {}:{:#x}",
                  block[next]->getAddrSpaceId(), block[next]->getOffset());
    // We could use the number of interrupts to determine if we missed a chunk of data
    dma->read(*block[next], block[next]->getSize());
    auto c = dma->readComplete();

    if (c.interrupts > 1) {
      logger->warn("Missed {} interrupts", c.interrupts - 1);
    }

    logger->trace("bytes: {}, intrs: {}, bds: {}", c.bytes, c.interrupts,
                  c.bds);
    try {
      for (size_t i = 0; i * 4 < c.bytes; i++) {
        int32_t ival = mem[cur][i];
#pragma GCC diagnostic push
#pragma GCC diagnostic ignored "-Wstrict-aliasing"
        float fval =
            *((float *)(&ival)); // cppcheck-suppress invalidPointerCast
#pragma GCC diagnostic pop
        formatter->format(fval);
        printf("%d: %#x\n", cnt++, ival);
      }
      formatter->output(std::cout);
    } catch (const std::exception &e) {
      logger->warn("Failed to output data: {}", e.what());
    }

    cur = next;
    next = (next + 1) % (sizeof(mem) / sizeof(mem[0]));
  }
}

int main(int argc, char *argv[]) {
  // Command Line Parser
  CLI::App app{"VILLASfpga data output"};

  try {
    std::string configFile;
    app.add_option("-c,--config", configFile, "Configuration file")
        ->check(CLI::ExistingFile);

    std::string fpgaName = "vc707";
    app.add_option("--fpga", fpgaName, "Which FPGA to use");
    std::vector<std::string> connectStr;
    app.add_option("-x,--connect", connectStr,
                   "Connect a FPGA port with another or stdin/stdout");
    bool noDma = false;
    app.add_flag("--no-dma", noDma,
                 "Do not setup DMA, only setup FPGA and Crossbar links");
    std::string outputFormat = "short";
    app.add_option("--output-format", outputFormat,
                   "Output format (short, long)");
    bool dumpGraph = false;
    app.add_flag("--dump-graph", dumpGraph,
                 "Dumps the graph of memory regions into \"graph.dot\"");
    bool dumpAuroraChannels = false;
    app.add_flag("--dump-aurora", dumpAuroraChannels,
                 "Dumps the detected Aurora channels.");
    double timestep = 10e-3;
    app.add_option("--timestep", timestep, "Timestep generated in the FPGA");
    app.parse(argc, argv);

    // Logging setup

    Log::getInstance().setLevel(spdlog::level::trace);
    logger->set_level(spdlog::level::trace);
    fpga::setupColorHandling();

    if (configFile.empty()) {
      logger->error(
          "No configuration file provided/ Please use -c/--config argument");
      return 1;
    }

    auto card = fpga::setupFpgaCard(configFile, fpgaName);

    if (dumpGraph) {
      auto &mm = MemoryManager::get();
      mm.getGraph().dump("graph.dot");
    }

    if (dumpAuroraChannels) {
      auto aurora_channels = getAuroraChannels(card);
      for (auto aurora : *aurora_channels)
        aurora->dump();
    }
    bool writeToStdout = false;
    bool readFromStdin = false;
    // Configure Crossbar switch
    for (std::string str : connectStr) {
      const fpga::ConnectString parsedConnectString(str);
      if (!parsedConnectString.configCrossBar(card)) {
        logger->error("Failed to configure crossbar");
        return -1;
      }
      if (parsedConnectString.isSrcStdin()) {
        readFromStdin = true;
        if (parsedConnectString.isBidirectional()) {
          writeToStdout = true;
        }
      }
      if (parsedConnectString.isDstStdout()) {
        writeToStdout = true;
        if (parsedConnectString.isBidirectional()) {
          readFromStdin = true;
        }
      }
    }
    auto reg = std::dynamic_pointer_cast<fpga::ip::Register>(
        card->lookupIp(fpga::Vlnv("xilinx.com:module_ref:registerif:")));

    if (reg != nullptr &&
        card->lookupIp(fpga::Vlnv("xilinx.com:module_ref:dinoif_fast:"))) {
      fpga::ip::DinoAdc::setRegisterConfigTimestep(reg, timestep);
    }

    if (writeToStdout || readFromStdin) {
      auto dma = std::dynamic_pointer_cast<fpga::ip::Dma>(
          card->lookupIp(fpga::Vlnv("xilinx.com:ip:axi_dma:")));
      if (dma == nullptr) {
        logger->error("No DMA found on FPGA ");
        throw std::runtime_error("No DMA found on FPGA");
      }
      std::unique_ptr<std::thread> stdInThread = nullptr;
      if (!noDma && writeToStdout) {
        auto formatter = fpga::getBufferedSampleFormatter(outputFormat, 16);
        // We copy the dma shared ptr but move the fomatter unqiue ptr as we don't need it
        // in this thread anymore
        stdInThread = std::make_unique<std::thread>(readFromDmaToStdOut, dma,
                                                    std::move(formatter));
      }
      if (!noDma && readFromStdin) {
        writeToDmaFromStdIn(dma);
      }

      if (stdInThread) {
        stdInThread->join();
      }
    }
  } catch (const RuntimeError &e) {
    logger->error("Error: {}", e.what());
    return -1;
  } catch (const CLI::ParseError &e) {
    return app.exit(e);
  } catch (const std::exception &e) {
    logger->error("Error: {}", e.what());
    return -1;
  } catch (...) {
    logger->error("Unknown error");
    return -1;
  }

  return 0;
}