/*
 * SamplerTask.cxx
 *
 *  Created on: Nov 22, 2012
 *      Author: dklein
 */

#include "SamplerTask.h"

namespace Highway
{

SamplerTask::SamplerTask(const char* name, Int_t iVerbose) :
  FairTask(name, iVerbose),
  fInput(NULL),
  fBranch(""),
  fMessageSize(32768),
  fOutput(new std::vector<Message*>)
{
}

SamplerTask::SamplerTask() :
  FairTask(
    "Abstract base task used for loading a branch from a root file into memory"),
  fInput(NULL),
  fBranch(""),
  fMessageSize(32768),
  fOutput(new std::vector<Message*>)
{
}

SamplerTask::~SamplerTask()
{
  delete fInput;

  // let fOutput in memory, because it is needed even after SamplerTask has terminated.
}

InitStatus SamplerTask::Init()
{
  FairRootManager* ioman = FairRootManager::Instance();
  fInput = (TClonesArray*) ioman->GetObject(fBranch.c_str());

  return kSUCCESS;
}

void SamplerTask::SetBranch(std::string branch)
{
  fBranch = branch;
}

void SamplerTask::SetMessageSize(int size)
{
  fMessageSize = size;
}

std::vector<Message*> *SamplerTask::GetOutput()
{
  return fOutput;
}

} /* namespace Highway */