PndMQFileSinkHits Class Reference

#include <PndMQFileSinkHits.h>

Inheritance diagram for PndMQFileSinkHits:

Public Member Functions
	PndMQFileSinkHits (std::string outputFileName="FilesSinkHits.root")
virtual	~PndMQFileSinkHits ()
virtual void	SetFileName (std::string fileName)
virtual void	InitOutputFile (TString defaultId="100")
template<class Archive >
void	serialize (Archive &ar, const unsigned int version)

Protected Member Functions
virtual void	Run ()

Private Attributes
TFile *	fOutFile
TTree *	fTree
TClonesArray *	fOutput
TList *	fBranchNameList
std::string	fOutputFileName
vector< vector< PndSdsHit > >	fHitVector
bool	fHasBoostSerialization

Friends
class	boost::serialization::access

Detailed Description

Definition at line 55 of file PndMQFileSinkHits.h.

Constructor & Destructor Documentation

PndMQFileSinkHits::PndMQFileSinkHits ( std::string  outputFileName = "FilesSinkHits.root" )

inline

Definition at line 58 of file PndMQFileSinkHits.h.

        : fOutFile(NULL)
        , fTree(NULL)
        , fOutput(NULL)
        , fHitVector()
        , fHasBoostSerialization(false)
                , fOutputFileName(outputFileName)
                , fBranchNameList(0)
    {
        gSystem->ResetSignal(kSigInterrupt);
        gSystem->ResetSignal(kSigTermination);

        // Check if boost serialization is available if it is chosen
        using namespace baseMQ::tools::resolve;
        // coverity[pointless_expression]: suppress coverity warnings on apparant if(const).
        if (is_same<boost::archive::binary_iarchive, boost::archive::binary_iarchive>::value || is_same<boost::archive::binary_iarchive, boost::archive::text_iarchive>::value)
        {
            if (has_BoostSerialization<PndSdsHit, void(boost::archive::binary_iarchive&, const unsigned int)>::value == 1)
            {
                fHasBoostSerialization = true;
            }
        }
    }

virtual PndMQFileSinkHits::~PndMQFileSinkHits ( )

inlinevirtual

Definition at line 82 of file PndMQFileSinkHits.h.

    {
        fBranchNameList->Write("BranchList", TObject::kSingleKey);
        //fTree->Write();
        //fOutFile->Close();
        if (fHitVector.size() > 0)
        {
            fHitVector.clear();
        }

    }

Member Function Documentation

virtual void PndMQFileSinkHits::InitOutputFile ( TString  defaultId = "100" )

inlinevirtual

Definition at line 98 of file PndMQFileSinkHits.h.

References fOutFile.

Referenced by main().

    {
        fOutput = new TClonesArray("PndSdsHit");

        std::stringstream ss;
      //  ss << "/Volumes/RAMDisk/";
        ss << fOutputFileName.c_str();


        fOutFile = new TFile(ss.str().c_str(), "recreate");
        fTree = new TTree("pndsim", "Test output");
        fTree->Branch("Output", "TClonesArray", &fOutput, 64000, 99);
        fBranchNameList = new TList();
        fBranchNameList->AddLast(new TObjString("Output"));
    }

void PndMQFileSinkHits::Run ( )

protectedvirtual

Definition at line 12 of file PndMQFileSinkHits.cxx.

References fHasBoostSerialization, fHitVector, fOutFile, fOutput, fTree, i, status, and PndMQStatus::STOP.

{
    if (fHasBoostSerialization)
    {
        int receivedMsgs = 0;

        // store the channel references to avoid traversing the map on every loop iteration
        FairMQChannel& dataInChannel = fChannels.at("data-in").at(0);

        while (CheckCurrentState(RUNNING))
        {
                std::unique_ptr<FairMQMessage> header(fTransportFactory->CreateMessage());
                        std::unique_ptr<FairMQMessage> msg(fTransportFactory->CreateMessage());
                        if (dataInChannel.Receive(header) > 0)
                        {
                                int status = *(static_cast<int*>(header->GetData()));

                                if (dataInChannel.ExpectsAnotherPart())
                                {
                                        receivedMsgs++;
                                        if (dataInChannel.Receive(msg)) {
                                                string msgStr(static_cast<char*>(msg->GetData()), msg->GetSize());
                                                istringstream ibuffer(msgStr);
                                                boost::archive::binary_iarchive InputArchive(ibuffer);
                                                //LOG(INFO) << "Received Message: " << receivedMsgs;
                                                try
                                                {
                                                        InputArchive >> fHitVector;
                                                }
                                                catch (boost::archive::archive_exception& e)
                                                {
                                                        LOG(ERROR) << e.what();
                                                }


                                                PndSdsHit* myHit = 0;
                                                bool dataAboveTimeThreshold = false;
                                                double timeStampThreshold = 0;
                                                for (auto& eventIter : fHitVector){
                                                        fOutput->Delete();
                                                        int numData = eventIter.size();
                                                        for (Int_t i = 0; i < numData; ++i)
                                                        {
                                                                if (eventIter[i].GetTimeStamp() > timeStampThreshold){
                                                                        myHit = new ((*fOutput)[fOutput->GetEntriesFast()]) PndSdsHit(eventIter[i]);
                                                                        dataAboveTimeThreshold = true;
                                                                }
                         //                   LOG(INFO) << "Data: " << i << " " << fHitVector.at(i).GetTimeStamp();
                                                        }
                                                        if (dataAboveTimeThreshold){
                                                                if (fOutput->IsEmpty())
                                                                {
                                                                        LOG(ERROR) << "PndMQFileSinkHits::Run(): No Output array!";
                                                                } else {

                                                                        fTree->Fill();
                                                                }
                                                        }
                                                }
                                                if (receivedMsgs % 1000 == 0 && myHit != 0){
                                                        LOG(INFO) << receivedMsgs << " : " << myHit->GetTimeStamp();
                                                }
                                        }
                                }
                                if (status == PndMQStatus::STOP){
                                        LOG(INFO) << "STOP-Signal Received!";
                                        fTree->AutoSave();
                                    LOG(INFO) << "AutoSave called!";
                                        fTree->Write();
                                        fOutFile->Close();
                                }

            }

            if (fHitVector.size() > 0)
            {
                fHitVector.clear();
            }
        }

        LOG(INFO) << "I've received " << receivedMsgs << " messages!";
    }
    else
    {
        LOG(ERROR) << " Boost Serialization not ok";
    }
}

template<class Archive >

void PndMQFileSinkHits::serialize ( Archive &  ar, const unsigned int  version  )

inline

Definition at line 115 of file PndMQFileSinkHits.h.

    {
        ar& fHitVector;
    }

virtual void PndMQFileSinkHits::SetFileName ( std::string  fileName )

inlinevirtual

Definition at line 94 of file PndMQFileSinkHits.h.

Referenced by main().

                                                {
        fOutputFileName = fileName;
    }

Friends And Related Function Documentation

friend class boost::serialization::access

friend

Definition at line 131 of file PndMQFileSinkHits.h.

Member Data Documentation

TList* PndMQFileSinkHits::fBranchNameList

private

Definition at line 127 of file PndMQFileSinkHits.h.

bool PndMQFileSinkHits::fHasBoostSerialization

private

Definition at line 133 of file PndMQFileSinkHits.h.

Referenced by Run().

vector<vector<PndSdsHit> > PndMQFileSinkHits::fHitVector

private

Definition at line 132 of file PndMQFileSinkHits.h.

Referenced by Run().

TFile* PndMQFileSinkHits::fOutFile

private

Definition at line 124 of file PndMQFileSinkHits.h.

Referenced by Run().

TClonesArray* PndMQFileSinkHits::fOutput

private

Definition at line 126 of file PndMQFileSinkHits.h.

Referenced by Run().

std::string PndMQFileSinkHits::fOutputFileName

private

Definition at line 128 of file PndMQFileSinkHits.h.

TTree* PndMQFileSinkHits::fTree

private

Definition at line 125 of file PndMQFileSinkHits.h.

Referenced by Run().

---

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

• PndMQFileSinkHits.h
• PndMQFileSinkHits.cxx