PndMvdMQFileSink.h

Go to the documentation of this file.

/********************************************************************************
 *    Copyright (C) 2014 GSI Helmholtzzentrum fuer Schwerionenforschung GmbH    *
 *                                                                              *
 *              This software is distributed under the terms of the             *
 *         GNU Lesser General Public Licence version 3 (LGPL) version 3,        *
 *                  copied verbatim in the file "LICENSE"                       *
 ********************************************************************************/
#ifndef PndMvdMQFileSink_H_
#define PndMvdMQFileSink_H_

#include "TClonesArray.h"
#include "TFile.h"
#include "TFolder.h"
#include "TTree.h"
#include "FairEventHeader.h"

#include "FairMQDevice.h"

class PndMvdMQFileSink : public FairMQDevice {
 public:
  enum { OutputFileName = FairMQDevice::Last, Last };

  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) { fFileName = tempString; }
  void AddOutputBranch(std::string classString, std::string branchString)
  {
    fClassNames.push_back(classString);
    fBranchNames.push_back(branchString);
  }
  std::string GetOutputFileName() { return fFileName; }

 protected:
  virtual void Init();
  virtual void Run();

 private:
  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;
  /* FairEventHeader* fEventHeader; */
  /* TClonesArray*    fOutput; */
  TFolder *fFolder;
};

#endif /* PndMvdMQFileSink_H_ */