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

#include "FairMQSamplerTask.h"


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

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

FairMQSamplerTask::~FairMQSamplerTask()
{
  delete fInput;

  // leave fOutput in memory, because it is needed even after FairMQSamplerTask is terminated.
}

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

  return kSUCCESS;
}

void FairMQSamplerTask::SetBranch(TString branch)
{
  fBranch = branch;
}

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

std::vector<FairMQMessage*> *FairMQSamplerTask::GetOutput()
{
  return fOutput;
}