MvdMQ/src/devices/PndMvdMQFileSampler.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 PndMvdMQFileSampler_H_
#define PndMvdMQFileSampler_H_

#include <string>

#include "TClonesArray.h"

#include "FairFileSource.h"
#include "FairRunAna.h"

#include "FairMQDevice.h"

class PndMvdMQFileSampler : public FairMQDevice {
 public:
  enum { InputFileName = FairMQDevice::Last, Last };

  PndMvdMQFileSampler();
  virtual ~PndMvdMQFileSampler();

  void AddInputFileName(std::string tempString) { fFileNames.push_back(tempString); }
  void AddInputChannelBranchName(std::pair<std::string, std::string> tempString)
  {
    fBranchNames.push_back(tempString);
    fPorts.insert(tempString.first);
  }

  void SetMaxIndex(int64_t tempInt) { fMaxIndex = tempInt; }

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

 private:
  FairRunAna *fRunAna;
  FairFileSource *fSource;
  std::multimap<std::string, TObject *> fInputObjects; //< multimap of port and object
  std::map<std::string, TObject *> fInputBranches;
  TObject *fEventHeaderPointer;
  std::set<std::string> fPorts;
  int fNObjects;
  int64_t fMaxIndex;
  std::vector<std::pair<std::string, std::string>> fBranchNames;
  std::vector<std::string> fFileNames;
};

#endif /* PndMvdMQFileSampler_H_ */