MTimeSort.cc

Go to the documentation of this file.

/*
 * Author: Laura Kogler <lkogler@berkeley.edu>
 *
 * Class MTimeSort: Sorts events by time
 *
*/
 
 
#include "MTimeSort.hh"
#include "QEventAssembler.hh"
#include "QRawEvent.hh"
#include "QVector.hh"
#include "QError.hh"
#include "QBaseType.hh"
#include <vector>
#include <algorithm>  
#include <utility>
 
using namespace Diana;
 
REGISTER_MODULE(MTimeSort)
 
using namespace std;
 
bool CompareTimes(const std::pair<int, unsigned long long> &EventAndTime1, 
                  const std::pair<int, unsigned long long> &EventAndTime2)
{
  return EventAndTime1.second < EventAndTime2.second;
}
 
 
// Init method is called before event loop
QDriver::Action MTimeSort::Init(QEventAssembler& eva) {
    QEvent& ev = eva.GetEvent();
    fIteration = GetIteration();
    fSortByRun = GetBool("SortByRun",true,false);
 
    Action act;
    act.fActionId = ACT_NEXTEV;
 
    eva.Add<QInt>("TimeSort","SortedIndex");
 
    if(fIteration >1){
        //Go to first event in the first run
        fIndex = 0;
        act.fActionId = ACT_GOTOEV;
        act.fEventNumber = (fEventVector.begin()->second[fIndex]).first;
        fIndex++;
    }
    
    fStartTime.clear();
 
    ev.Require<QHeader>("DAQ","Header");
    ev.Require<QPulseInfo>("DAQ","PulseInfo");
    return act;
}
 
// Doit method is called for each event, getting the event as argument
QDriver::Action MTimeSort::Do( QEventAssembler& eva) 
{
    Action act;
    act.fActionId = ACT_NEXTEV;
 
    QEvent& ev = eva.GetEvent();
    const QHeader& header = ev.Get<QHeader>("DAQ","Header");
    int run=1;
    if(fSortByRun){
        run = header.GetRun();
    }
 
    if(fIteration==1){
        // We will make a vector containing the tree index and time for each event
        std::pair<int, unsigned long long> EventAndTime;
        EventAndTime.first = ev.GetReadNumber(); // This returns the tree index
        const QPulseInfo& pulseInfo = ev.Get<QPulseInfo>("DAQ","PulseInfo");
        const QTime& t = header.GetTime(); 
        int chan = pulseInfo.GetChannelId();
        EventAndTime.second = t.GetFromStartRunNs();
        if(fEventVector.find(run) == fEventVector.end()){
            //If run not in the list create it
            fEventVector[run] = std::vector< std::pair<int, unsigned long long>>();
            fChannelVector[run] = std::vector<int>();
            fStartTime[run] = 0;
        }
        fEventVector[run].push_back(EventAndTime);
        fChannelVector[run].push_back(chan);
        if(fStartTime[run] == 0) fStartTime[run] = t.GetStartRunUnix();
 
    }else{
        void * vtmp = (void*) &header;
        QHeader * tmph = (QHeader*) vtmp;
        QTime& t = tmph->GetTime();
        t.SetStartRunUnix(fStartTime[run]);
 
        eva.Get<QInt>("TimeSort","SortedIndex") = fIndex;
        if(fIndex<fNumberOfEvents){
            act.fActionId = ACT_GOTOEV;
            size_t index=fIndex;
            
            std::map<int,std::vector< std::pair<int, unsigned long long>>>::iterator it;
            for(it=fEventVector.begin();it!=fEventVector.end();it++){
                if(index>=it->second.size()){
                    index -= it->second.size();
                }
            }
 
            act.fEventNumber = (fEventVector[run][index]).first;
            fIndex++;
        }else{
            // this will stop the diana loop, otherwise it will continue til the last event in file
            act.fActionId = ACT_QUIT;
        }
    }
    return act;
}
 
// Done method is called after event loop
QDriver::Action MTimeSort::Done() {
    
    Action act;
    act = ACT_NONE;
    if(fIteration == 1){
        std::map<int,std::vector< std::pair<int, unsigned long long>>>::iterator it;
        fNumberOfEvents=0;
        for(it=fEventVector.begin();it!=fEventVector.end();it++){
            sort(it->second.begin(), it->second.end(), CompareTimes);
            fNumberOfEvents+=it->second.size();
        }
        if(GetBool("SortInPlace", true)) act = ACT_RERUN;
    }
   return act;
}