PndMvdMQFileSamplerBursts.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 PndMvdMQFileSamplerBursts_H_
#define PndMvdMQFileSamplerBursts_H_

#include <string>

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

#include "FairMQDevice.h"

#include "PndBurstVectorBuilder.h"

#include "TClonesArray.h"

struct BurstHeader
{
        unsigned int fRunID;
        unsigned int fBurstID;
        std::string fBranchName;

        template<class Archive>
        void serialize(Archive & ar, const unsigned int version)
        {
                ar & fRunID;
                ar & fBurstID;
                ar & fBranchName;
        }

        #ifndef __CINT__ // for BOOST serialization
                friend class boost::serialization::access;
        #endif // for BOOST serialization

};

struct BurstData
{
        BurstData() : fHeader(), fData(){};
        virtual ~BurstData(){
                for (auto itr : fData){
                        for (auto dataItr : itr){
                                delete(dataItr);
                        }
                        itr.clear();
                }
                fData.clear();
        };
        BurstHeader fHeader;
        std::vector<std::vector<FairTimeStamp*> > fData;

        virtual void Reset()
        {

        }

        template<class Archive>
        void serialize(Archive & ar, const unsigned int version)
        {
                ar & fHeader;
                ar & fData;
        }

        #ifndef __CINT__ // for BOOST serialization
                friend class boost::serialization::access;
        #endif

};

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

    PndMvdMQFileSamplerBursts();
    virtual ~PndMvdMQFileSamplerBursts();

    void AddInputFileName  (std::string tempString) { fFileNames.push_back(tempString);}
    void AddInputChannelBranchName(std::pair<std::string, std::string> tempString) {
        fPortBranchNameMap.insert(std::pair<std::string, std::string>(tempString.first, tempString.second));
        fBranchNames.push_back(tempString);
        fPorts.insert(tempString.first);
    }
    
    void SetMaxIndex(int64_t tempInt) {fMaxIndex=tempInt;}

    template<class Archive>
        void serialize(Archive & ar, const unsigned int version)
        {
                ar & fHitVector;
        }

        #ifndef __CINT__ // for BOOST serialization
                friend class boost::serialization::access;
        #endif

 protected:
    virtual void Run();
    virtual void InitTask();
    
    virtual void InitBurstBuilder(std::string branchName);

 private: 
    FairRunAna*     fRunAna;
    FairFileSource* fSource;
    std::set<std::string> fPorts;                               //< set of output ports
    std::multimap<std::string, std::string>                             fPortBranchNameMap;
    std::vector<std::pair<std::string, std::string> >    fBranchNames;                  //< vector of output ports and branch names. One port can have multiple branches send to it
    std::map<std::string, std::vector< std::vector<FairTimeStamp*> > >        fOutputData;  //< map of branch and data object to be send
    std::map<std::string, TObject* >                            fInputBranches;                                                                 //< branch name and TClonesArray in root tree
    std::map<std::string, PndBurstVectorBuilderBase*>           fBurstBuilder;                                                  //< branch name and associated burst builder
    std::vector<FairTimeStamp*> fHitVector;
    FairEventHeader* fEventHeader;


    int             fNObjects;
    int64_t         fMaxIndex;

    std::vector<std::string>     fFileNames;
};

#endif /* PndMvdMQFileSamplerBursts_H_ */