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 "QEvent.hh"
#include "QRawEvent.hh"
#include "QError.hh"
#include <vector>
#include <algorithm>  
#include <utility>
 
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;
}
 
// ctor
MTimeSort::MTimeSort(QSequence *s) : QModule("TimeSort",s) {
}
 
// dtor
MTimeSort::~MTimeSort() {
}
 
// Init method is called before event loop
void MTimeSort::Init() {
        
        fIteration = GetIteration();
        
        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;
                JumpToEvent((fEventVector[fIndex]).first);
                fIndex++;
        }
        
        fStartTime=0;
}
 
// Doit method is called for each event, getting the event as argument
QEvent* MTimeSort::Do( QEvent* ev) {
   
        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 QRawEvent& raw = ev->GetRawEvent();
                QTime t = raw.GetTime(); 
                int chan = raw.GetPulse().GetChannelId();
                EventAndTime.second = t.GetFromStartRunNs();
                fEventVector.push_back(EventAndTime);
                fChannelVector.push_back(chan);
                if(fStartTime == 0)
                        fStartTime = t.GetStartRunUnix();
        }
        else
        {
                if(fIndex<fNumberOfEvents)
                {
                        QError err = JumpToEvent((fEventVector[fIndex]).first);
            if(err != QERR_SUCCESS) Error("%s: %s",err.ToString().c_str(), err.GetDescription().c_str());
                        fIndex++;
                }
                else
            {
            // this will stop the diana loop, otherwise it will continue til the last event in file
                        QError err = JumpToEvent(fNumberOfEvents);
            if(err != QERR_SUCCESS) Error("%s: %s",err.ToString().c_str(), err.GetDescription().c_str());
        }
        }
        return ev;
}
 
// Done method is called after event loop
void MTimeSort::Done() {
        
        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];
                        }               
                        SeqAuxData().SetInt("RunStartTime", fStartTime, fOutputFilename);
                        SeqAuxData().SetQObject("EventList", &EventList, fOutputFilename);
                        SeqAuxData().SetQObject("TimeMs", &TimeList, fOutputFilename);
                        SeqAuxData().SetQObject("Channel", &ChannelList, fOutputFilename);
                }
                if(GetBool("SortInPlace", true)) SetRunAgain(true);
        }
        else
                SetRunAgain(false);
   return;
}