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();
    Action act;
    act.fActionId = ACT_NEXTEV;
 
        if(fIteration == 1)
        {
                fOutputFilename = GetString("OutputFilename", "");
                if(fOutputFilename != "")
                {
                        bool notroot = (fOutputFilename.find(".root") == string::npos);
                        bool nottxt = (fOutputFilename.find(".txt") == string::npos);
                        if(notroot && nottxt)
                        {
                                Panic("Invalid filename for temporary data.  Filename must end in .root or .txt");
                        }
                }
        }
        else
        {
                fIndex = 0;
        act.fActionId = ACT_GOTOEV;
        act.fEventNumber = (fEventVector[fIndex]).first;
                fIndex++;
        }
        
        fStartTime=0;
 
    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();
 
    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 QHeader& header = ev.Get<QHeader>("DAQ","Header");
                const QPulseInfo& pulseInfo = ev.Get<QPulseInfo>("DAQ","PulseInfo");
                const QTime& t = header.GetTime(); 
                int chan = pulseInfo.GetChannelId();
                EventAndTime.second = t.GetFromStartRunNs();
                fEventVector.push_back(EventAndTime);
                fChannelVector.push_back(chan);
                if(fStartTime == 0)
                        fStartTime = t.GetStartRunUnix();
        }
        else
        {
                if(fIndex<fNumberOfEvents)
                {
 
            act.fActionId = ACT_GOTOEV;
            act.fEventNumber = (fEventVector[fIndex]).first;
                        fIndex++;
                }
                else
            {
            // this will stop the diana loop, otherwise it will continue til the last event in file
            act.fActionId = ACT_GOTOEV;
            act.fEventNumber = fNumberOfEvents;
        }
        }
    return act;
}
 
// Done method is called after event loop
QDriver::Action MTimeSort::Done() {
        
    Action act;
    act = ACT_NONE;
        if(fIteration == 1)
        {
                sort(fEventVector.begin(), fEventVector.end(), CompareTimes);
                std::pair<int, unsigned long long> EventAndTime;
                fNumberOfEvents = fEventVector.size();
                
                if(fOutputFilename != "")
                {
                        QVector EventList = QVector(fNumberOfEvents);
                        QVector TimeList = QVector(fNumberOfEvents); // Time in ms
                        QVector ChannelList = QVector(fNumberOfEvents);
                        for(int i=0; i<fNumberOfEvents; i++) 
                        {
                           EventAndTime = fEventVector[i];
                   EventList[i] = (double)EventAndTime.first;
                                TimeList[i] = (double)EventAndTime.second/1.0e6;
                                ChannelList[i] = (double)fChannelVector[i];
                        }               
                        GlobalData().Set("RunStartTime", QInt(fStartTime), fOutputFilename);
                        GlobalData().Set("EventList", EventList, fOutputFilename);
                        GlobalData().Set("TimeMs", TimeList, fOutputFilename);
                        GlobalData().Set("Channel", ChannelList, fOutputFilename);
                }
                if(GetBool("SortInPlace", true)) act = ACT_RERUN;
        }
   return act;
}