#include "FairTestOutputHandler.h"

#include <fstream>

FairTestOutputHandler::FairTestOutputHandler(std::string filename)
  : fFileName(filename),
    fFileContent(),
    fFileContentIterator(),
    fNumberOfLines(0),
    fFileAlreadyRead(false)
{
}

FairTestOutputHandler::FairTestOutputHandler()
  : fFileName(),
    fFileContent(),
    fFileContentIterator(),
    fNumberOfLines(0),
    fFileAlreadyRead(false)
{
}

FairTestOutputHandler::~FairTestOutputHandler()
{
  remove(fFileName.c_str());
}

void FairTestOutputHandler::SetFileName(std::string filename)
{
  fFileName = filename;
  fFileContent.clear();
  fNumberOfLines = 0;
  fFileAlreadyRead =false;
}

int FairTestOutputHandler::GetNumberOfLines()
{
  ReadContentFromFile();
  return fNumberOfLines;
}

void FairTestOutputHandler::ReadContentFromFile()
{
  if ( !fFileAlreadyRead ) {
    fFileContent.clear();
    fNumberOfLines = 0;

    std::ifstream ifs(fFileName.c_str());
    std::string line;

    while( getline( ifs, line ) ) {
      fFileContent.push_back( line );
      ++fNumberOfLines;
    }
    fFileAlreadyRead = true;
  }
}

std::string FairTestOutputHandler::GetCaptureLine(int line)
{

  ReadContentFromFile();

  return fFileContent[line];

}
std::string FairTestOutputHandler::GetCapture()
{

  ReadContentFromFile();

  std::string retVal;

  for ( fFileContentIterator = fFileContent.begin();
        fFileContentIterator != fFileContent.end();
        fFileContentIterator++) {
    retVal += *fFileContentIterator;
    retVal += "\n";
  }

  return retVal;
}