PndMvdMQFileSink Class Reference

#include <PndMvdMQFileSink.h>

Inheritance diagram for PndMvdMQFileSink:

Public Types
enum	{ OutputFileName = FairMQDevice::Last, Last }

Public Member Functions
	PndMvdMQFileSink ()
virtual	~PndMvdMQFileSink ()
void	SetProperty (const int key, const std::string &value)
void	SetProperty (const int key, const int value)
std::string	GetProperty (const int key, const std::string &default_="")
int	GetProperty (const int key, const int default_=0)
void	SetOutputFileName (std::string tempString)
void	AddOutputBranch (std::string classString, std::string branchString)
std::string	GetOutputFileName ()

Protected Member Functions
virtual void	Init ()
virtual void	Run ()

Private Attributes
std::string	fFileName
std::string	fTreeName
std::vector< std::string >	fBranchNames
std::vector< std::string >	fClassNames
std::string	fFileOption
bool	fFlowMode
bool	fWrite
TFile *	fOutFile
TTree *	fTree
unsigned int	fNObjects
TObject **	fOutputObjects
TFolder *	fFolder

Detailed Description

PndMvdMQFileSink.h

Since

2016-03-08

Author

R. Karabowicz

Definition at line 26 of file PndMvdMQFileSink.h.

Member Enumeration Documentation

anonymous enum

Enumerator
OutputFileName
Last

Definition at line 29 of file PndMvdMQFileSink.h.

    {
        OutputFileName = FairMQDevice::Last,
        Last
    };

Constructor & Destructor Documentation

PndMvdMQFileSink::PndMvdMQFileSink

(

)

Definition at line 37 of file PndMvdMQFileSink.cxx.

  : FairMQDevice()
  , fFileName()
  , fTreeName()

  , fBranchNames()
  , fClassNames()
  , fFileOption()
  , fFlowMode(false)
  , fWrite(false)

  , fOutFile(NULL)
  , fTree(NULL)
  , fNObjects(0)
  , fOutputObjects(new TObject*[1000])
  , fFolder(NULL)
{
}

PndMvdMQFileSink::~PndMvdMQFileSink ( )

virtual

Definition at line 175 of file PndMvdMQFileSink.cxx.

References fOutFile, and fTree.

{
  if (fTree)
    {
      fTree->Write();
      delete fTree;
    }

  if (fOutFile)
    {
      if (fOutFile->IsOpen())
        fOutFile->Close();
      delete fOutFile;
    }
}

Member Function Documentation

void PndMvdMQFileSink::AddOutputBranch ( std::string  classString, std::string  branchString  )

inline

Definition at line 44 of file PndMvdMQFileSink.h.

References fBranchNames, and fClassNames.

Referenced by main().

    { 
      fClassNames .push_back(classString); 
      fBranchNames.push_back(branchString); 
    }

std::string PndMvdMQFileSink::GetOutputFileName ( )

inline

Definition at line 49 of file PndMvdMQFileSink.h.

References fFileName.

Referenced by GetProperty().

{ return fFileName;}

std::string PndMvdMQFileSink::GetProperty	(	const int	key,
		const std::string &	default_ = ""
	)

Definition at line 158 of file PndMvdMQFileSink.cxx.

References GetOutputFileName(), and OutputFileName.

{
    switch (key)
    {
        case OutputFileName :
          return GetOutputFileName();

        default:
            return FairMQDevice::GetProperty(key, default_);
    }
}

int PndMvdMQFileSink::GetProperty	(	const int	key,
		const int	default_ = 0
	)

Definition at line 170 of file PndMvdMQFileSink.cxx.

{
    return FairMQDevice::GetProperty(key, value);
}

void PndMvdMQFileSink::Init ( )

protectedvirtual

Definition at line 56 of file PndMvdMQFileSink.cxx.

References fBranchNames, fClassNames, fFileName, fFileOption, fFolder, fNObjects, fOutFile, fOutputObjects, fTree, and fTreeName.

{
  LOG(INFO) << "SHOULD CREATE THE FILE AND TREE";
  //  fFileName = "/Users/karabowi/fairroot/pixel9_dev/FairRoot/examples/MQ/9-PixelDetector/macros/tmpOut.root";
  fFileOption = "RECREATE";
  fTreeName = "pndsim";

  // fBranchNames.push_back("EventHeader.");
  // fClassNames .push_back("FairEventHeader");
  // fBranchNames.push_back("PixelHits");
  // fClassNames .push_back("TClonesArray(PixelHit)");

  fOutFile = TFile::Open(fFileName.c_str(),fFileOption.c_str());

  fTree = new TTree(fTreeName.c_str(), "/cbmout");

  fFolder = new TFolder("cbmout", "Main Output Folder");
  TFolder* foldEventHeader = fFolder->AddFolder("EvtHeader","EvtHeader");
  TFolder* foldPixel       = fFolder->AddFolder("Pixel","Pixel");

  TList* BranchNameList = new TList();

  LOG(INFO) << "PndMvdMQFileSink: " << fBranchNames.size();

  for ( fNObjects = 0 ; fNObjects < fBranchNames.size() ; fNObjects++ ) {
    if      ( fClassNames[fNObjects].find("TClonesArray(") == 0 ) {
      fClassNames   [fNObjects] = fClassNames[fNObjects].substr(13,fClassNames[fNObjects].length()-12-2);
      fOutputObjects            [fNObjects] = new    TClonesArray(fClassNames[fNObjects].c_str());
      fTree->Branch(fBranchNames[fNObjects].c_str(),"TClonesArray", &fOutputObjects[fNObjects]);
      foldPixel->Add(fOutputObjects[fNObjects]);
      LOG(INFO) << fNObjects << " : " << fBranchNames[fNObjects];
      BranchNameList->AddLast(new TObjString(fBranchNames[fNObjects].c_str()));
    }
    else if ( fClassNames[fNObjects].find("FairEventHeader") == 0 ) {
      fOutputObjects            [fNObjects] = new    FairEventHeader();
      fTree->Branch(fBranchNames[fNObjects].c_str(),"FairEventHeader", &fOutputObjects[fNObjects]);
      foldEventHeader->Add(fOutputObjects[fNObjects]);
      BranchNameList->AddLast(new TObjString(fBranchNames[fNObjects].c_str()));
    }
    else {
      LOG(ERROR) << "!!! Unknown output object \"" << fClassNames[fNObjects] << "\" !!!";
    }
  }

  fFolder->Write();
  BranchNameList->Write("BranchList", TObject::kSingleKey);
  BranchNameList->Delete();
  delete BranchNameList;

}

void PndMvdMQFileSink::Run ( )

protectedvirtual

Definition at line 107 of file PndMvdMQFileSink.cxx.

References fBranchNames, fOutputObjects, and fTree.

{
  while (CheckCurrentState(RUNNING))
    {
      FairMQParts parts;

      if (Receive(parts, "data-in") >= 0)
        {

          TObject* tempObjects[10];
          for ( int ipart = 0 ; ipart < parts.Size() ; ipart++ )
            {
              Ex9TMessage tm(parts.At(ipart)->GetData(), parts.At(ipart)->GetSize());
              tempObjects[ipart] = (TObject*)tm.ReadObject(tm.GetClass());
              for ( unsigned int ibr = 0 ; ibr < fBranchNames.size() ; ibr++ )
                {
                  if ( strcmp(tempObjects[ipart]->GetName(),fBranchNames[ibr].c_str()) == 0 )
                    {
                      fOutputObjects[ibr] = tempObjects[ipart];
                      fTree->SetBranchAddress(fBranchNames[ibr].c_str(),&fOutputObjects[ibr]);
                    }
                }
            }
          fTree->Fill();
        }
      else
        {
          LOG(INFO) << "oops!";
        }
    }
}

void PndMvdMQFileSink::SetOutputFileName ( std::string  tempString )

inline

Definition at line 43 of file PndMvdMQFileSink.h.

References fFileName.

Referenced by SetProperty().

{ fFileName = tempString; }

void PndMvdMQFileSink::SetProperty	(	const int	key,
		const std::string &	value
	)

Definition at line 139 of file PndMvdMQFileSink.cxx.

References OutputFileName, and SetOutputFileName().

Referenced by main().

{
    switch (key)
    {
        case OutputFileName :
          SetOutputFileName(value);
            break;

        default:
            FairMQDevice::SetProperty(key, value);
            break;
    }
}

void PndMvdMQFileSink::SetProperty	(	const int	key,
		const int	value
	)

Definition at line 153 of file PndMvdMQFileSink.cxx.

{
    FairMQDevice::SetProperty(key, value);
}

Member Data Documentation

std::vector<std::string> PndMvdMQFileSink::fBranchNames

private

Definition at line 59 of file PndMvdMQFileSink.h.

Referenced by AddOutputBranch(), Init(), and Run().

std::vector<std::string> PndMvdMQFileSink::fClassNames

private

Definition at line 60 of file PndMvdMQFileSink.h.

Referenced by AddOutputBranch(), and Init().

std::string PndMvdMQFileSink::fFileName

private

Definition at line 56 of file PndMvdMQFileSink.h.

Referenced by GetOutputFileName(), Init(), and SetOutputFileName().

std::string PndMvdMQFileSink::fFileOption

private

Definition at line 62 of file PndMvdMQFileSink.h.

Referenced by Init().

bool PndMvdMQFileSink::fFlowMode

private

Definition at line 63 of file PndMvdMQFileSink.h.

TFolder* PndMvdMQFileSink::fFolder

private

Definition at line 72 of file PndMvdMQFileSink.h.

Referenced by Init().

unsigned int PndMvdMQFileSink::fNObjects

private

Definition at line 68 of file PndMvdMQFileSink.h.

Referenced by Init().

TFile* PndMvdMQFileSink::fOutFile

private

Definition at line 66 of file PndMvdMQFileSink.h.

Referenced by Init(), and ~PndMvdMQFileSink().

TObject** PndMvdMQFileSink::fOutputObjects

private

Definition at line 69 of file PndMvdMQFileSink.h.

Referenced by Init(), and Run().

TTree* PndMvdMQFileSink::fTree

private

Definition at line 67 of file PndMvdMQFileSink.h.

Referenced by Init(), Run(), and ~PndMvdMQFileSink().

std::string PndMvdMQFileSink::fTreeName

private

Definition at line 57 of file PndMvdMQFileSink.h.

Referenced by Init().

bool PndMvdMQFileSink::fWrite

private

Definition at line 64 of file PndMvdMQFileSink.h.

---

The documentation for this class was generated from the following files:

• PndMvdMQFileSink.h
• PndMvdMQFileSink.cxx