In This Package:

DaqFormatContext.cc

Go to the documentation of this file.

/*
 *  DaqFormatContext.cc
 *  DaqFormatModules
 *
 *  Created by Simon Patton on 7/22/10.
 *  Copyright 2010 DayaBay Collaboration. All rights reserved.
 *
 */
#include "DaqFormatContext.h"

#include <sstream>

#include "DaqFormatModules/DaqReadoutSvc.h"
#include "DaqXml/DaqXmlStream.h"
#include "EventReadoutFormat/EventHeader.h"
#include "EventReadoutFormat/EventReadout.h"
#include "FileReadoutFormat/DataSeparatorRecord.h"
#include "FileReadoutFormat/FileBuffer.h"
#include "FileReadoutFormat/DaqFromFStream.h"
#include "FileReadoutFormat/FileTraits.h"
#include "GaudiKernel/MsgStream.h"

using DybDaq::DaqFormatContext;
using DybDaq::DaqXmlStream;
using DybDaq::DataSeparatorRecord;
using DybDaq::EventHeader;
using DybDaq::EventReadout;
using DybDaq::FileBuffer;
using DybDaq::DaqFromFStream;
using DybDaq::FileTraits;
using std::vector;
using std::string;

const unsigned int DaqFormatContext::kExitOnInvalid = 0; 
const unsigned int DaqFormatContext::kSkipInvalid = DaqFormatContext::kExitOnInvalid + 1;
const unsigned int DaqFormatContext::kProcessInvalid = DaqFormatContext::kSkipInvalid + 1;

static const string kTwoSpaces("  ");

DaqFormatContext::DaqFormatContext(const std::vector<std::string>& inputFiles,
                                   const unsigned int actionOnInvalid,
                                   IMessageSvc& msgSvc,
                                   IDaqReadoutSvc& daqSvc) :
m_messageSvc(msgSvc),
m_daqSvc(dynamic_cast<DaqReadoutSvc*>(&daqSvc)),
m_expiredEndRecord(true),
m_reader(0),
m_event(0),
m_nextFile(inputFiles.begin()),
m_endFile(inputFiles.end()),
m_warnedAboutSkip(false),
m_actionOnInvalid(actionOnInvalid) {
}

DaqFormatContext::~DaqFormatContext() {
}

void* DaqFormatContext::identifier() const {
    return 0;
}

const EventReadout* DaqFormatContext::event() const {
    return m_event;
}

StatusCode DaqFormatContext::jump(const int distance) {
    if (0 > distance) {
        MsgStream log(&m_messageSvc,
                       "DaqFormatContext");
        log << MSG::ERROR << "Can not move backward in a file" << endreq;
        return StatusCode::FAILURE;
    }
    for (int count = 0;
         distance != count;
         ++count) {
        StatusCode status = moveToNextEvent();
        if (status.isFailure()) {
            return status;
        }
    }
    if ((kProcessInvalid != m_actionOnInvalid) && (m_event->header().invalidData())) {
        MsgStream log(&m_messageSvc,
                      "DaqFormatContext");
        if (kExitOnInvalid == m_actionOnInvalid) {
            delete m_event;
            m_event = 0;
            log << MSG::ERROR << "Exiting as Invalid Event has been detected" << endreq;
            return StatusCode::FAILURE;
        }
        if (kSkipInvalid == m_actionOnInvalid) {
            if (!m_warnedAboutSkip) {
                log << MSG::WARNING << "One or more Invalid Event has been detected, they will be skipped" << endreq;
                m_warnedAboutSkip = true;
            }
            do {
                delete m_event;
                StatusCode status = moveToNextEvent();
                if (status.isFailure()) {
                    return status;
                }
            } while (m_event->header().invalidData());
        }
    }
    return StatusCode::SUCCESS;
}

StatusCode DaqFormatContext::isGoodEvent() const {
    if (0 == m_event) {
        return StatusCode::FAILURE;
    }
    return StatusCode::SUCCESS;
}

StatusCode DaqFormatContext::moveToNextEvent() {
    m_event = 0;
    while (0 == m_event) {
        if(0 == m_reader){
            m_reader = openNextFile();
            if(0 == m_reader){
                return StatusCode::FAILURE;
            }
        }
        
        const FileBuffer* record = m_reader->nextRecord();
        if(0 == record){
            MsgStream log(&m_messageSvc,
                          "DaqFormatContext");
            log << MSG::ERROR << "Failed to read next record from file: " << (*m_nextFile) << endreq;
            return StatusCode::FAILURE;
        }
        
        // Check the record type

        if (m_expiredEndRecord && (0 != m_daqSvc)) {
            m_daqSvc->setFileRecord((const FileEndRecord*)0);
        }
        if(record->isMarked(FileTraits::kDataSeparatorRecord)) {
            const DataSeparatorRecord* dataSeparator = dynamic_cast<const DataSeparatorRecord*>(record);
            m_event = dataSeparator->extractBlock();
            delete record;
            m_expiredEndRecord = true;
        } else {
            if (0 != m_daqSvc) {
                m_daqSvc->setFileRecord(record);
            }
            if(record->isMarked(FileTraits::kFileEndRecord)) {
                m_expiredEndRecord = false;
            }
            std::stringstream message;
            message << '\n';
            DaqXmlStream xmlStream(message,
                                   kTwoSpaces,
                                   true);
            xmlStream.dumpElement(*record);
            MsgStream log(&m_messageSvc,
                          "DaqFormatContext");
            log << MSG::INFO << message.str() << endreq;        
            if(record->isMarked(FileTraits::kFileEndRecord)) {
                delete m_reader;
                m_reader = 0;
                ++m_nextFile;
            }
        }
    }
    return StatusCode::SUCCESS;
}

DybDaq::DaqFromFStream* DaqFormatContext::openNextFile() {
    DaqFromFStream* result = 0;
    
    if (m_nextFile == m_endFile) {
        MsgStream log(&m_messageSvc,
                      "DaqFormatContext");
        log << MSG::INFO << "End of input file list" << endreq;
        return result;
    }
    
    result = new DaqFromFStream();
    if(0 == result){
            MsgStream log(&m_messageSvc,
                          "DaqFormatContext");
            log << MSG::ERROR << "Failed to create new file reader" << endreq;
        return result;
    }
    
    const bool opened = result->openFile((*m_nextFile).c_str());
    if (!opened) {
        MsgStream log(&m_messageSvc,
                      "DaqFormatContext");
        log << MSG::ERROR << "Failed to open input file: " << (*m_nextFile) << endreq;
        delete result;
        result = 0;
    }
    return result;
}

---