FairMQ/fairmq/FairMQSamplerTask.cxx

/**
 * FairMQSamplerTask.cxx
 *
 * @since 2012-11-22
 * @author D. Klein, A. Rybalchenko
 */

#include "FairMQSamplerTask.h"


FairMQSamplerTask::FairMQSamplerTask(const Text_t* name, int iVerbose) :
  FairTask(name, iVerbose),
  fInput(NULL),
  fBranch(""),
  fOutput(NULL)
{
}

FairMQSamplerTask::FairMQSamplerTask() :
  FairTask( "Abstract base task used for loading a branch from a root file into memory"),
  fInput(NULL),
  fBranch(""),
  fOutput(NULL)
{
}

FairMQSamplerTask::~FairMQSamplerTask()
{
  delete fInput;
  //delete fOutput; // leave fOutput in memory, because it is needed even after FairMQSamplerTask is terminated. ClearOutput will clean it when it is no longer needed.
}

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

  return kSUCCESS;
}

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

FairMQMessage* FairMQSamplerTask::GetOutput()
{
  return fOutput;
}

void FairMQSamplerTask::SetTransport(FairMQTransportFactory* factory)
{
  fTransportFactory = factory;
}