input.cpp

/******************************************************************************
 * This file is part of the ethos project.
 *
 * Copyright (c) 2023-2024 by the authors listed in the file AUTHORS
 * in the top-level source directory and their institutional affiliations.
 * All rights reserved.  See the file COPYING in the top-level source
 * directory for licensing information.
 ******************************************************************************/
#include "input.h"

#include <fstream>
#include <iostream>

#include "base/check.h"

namespace ethos {

/** File input class */
class FileInput : public Input
{
 public:
  FileInput(const std::string& filename) : Input()
  {
    d_fs.open(filename, std::fstream::in);
    if (!d_fs.is_open())
    {
      EO_FATAL() << "Couldn't open file: " << filename;
    }
  }
  std::istream* getStream() override { return &d_fs; }
 private:
  /** File stream */
  std::ifstream d_fs;
};
/** Stream reference input class */
class StreamInput : public Input
{
 public:
  StreamInput(std::istream& input) : Input(), d_input(input) {}
  std::istream* getStream() override { return &d_input; }
  bool isInteractive() const override { return true; }

 private:
  /** Reference to stream */
  std::istream& d_input;
};
/** String input class */
class StringInput : public Input
{
 public:
  StringInput(const std::string& input) : Input()
  {
    d_ss << input;
  }
  std::istream* getStream() override { return &d_ss; }
 private:
  /** String stream */
  std::stringstream d_ss;
};

Input::Input() {}

bool Input::isInteractive() const { return false; }

std::unique_ptr<Input> Input::mkFileInput(const std::string& filename)
{
  return std::unique_ptr<Input>(new FileInput(filename));
}
std::unique_ptr<Input> Input::mkStreamInput(std::istream& input)
{
  return std::unique_ptr<Input>(new StreamInput(input));
}
std::unique_ptr<Input> Input::mkStringInput(const std::string& input)
{
  return std::unique_ptr<Input>(new StringInput(input));
}

}  // namespace ethos