In This Package:

DaqRootWriter.cc

Go to the documentation of this file.

/*
 *  DaqRootWriter.cc
 *  DaqFormatModules
 *
 *  Created by Simon Patton on 8/8/10.
 *  Copyright 2010 DayaBay Collaboration. All rights reserved.
 *
 */

#include "DaqReadoutFormat/ByteBuffer.h"
#include "DybKernel/IDybStorageSvc.h"
#include "Event/DaqBlock.h"
#include "Event/DaqBlockHeader.h"
#include "EventReadoutFormat/EventReadout.h"
#include "FileReadoutFormat/DaqToStrStream.h"
#include "GaudiKernel/RegistryEntry.h"
#include "DaqRootWriter.h"
#include <cstring>

using DayaBay::DaqCrate;
using DayaBay::DaqBlock;
using DayaBay::DaqBlockHeader;
using DybDaq::ByteBuffer;
using DybDaq::DaqFormatWriter;
using DybDaq::DaqToOStream;
using DybDaq::EventReadout;
using DybDaq::FileTraits;
using DybDaq::DaqToStrStream;
using std::string;

DaqRootWriter::DaqRootWriter(const std::string& name,
                             ISvcLocator* svc) :
    DaqFormatWriter(name,
                    svc),
    m_storageSvc(0) {
    declareProperty("DaqBlockLocation",
                    m_daqBlockLocation=DayaBay::DaqBlockHeaderLocation::Default,
                    "Location in file where DAQ block will be saved.");
}

DaqRootWriter::~DaqRootWriter() {
}

StatusCode DaqRootWriter::initialize()
{
    StatusCode sc = DaqFormatWriter::initialize();
    if(sc.isFailure()) {
        return sc;
    }

    // Initialize the necessary services
    sc = service("DybStorageSvc",
                 m_storageSvc,
                 true);
    if(sc.isFailure()) {
        m_storageSvc = 0;
        warning() << "No Storage Service was found, so this module will be disabled" << endreq;
    }
    return StatusCode::SUCCESS;
}

StatusCode DaqRootWriter::finalize()
{
    if(0 != m_storageSvc) {
        m_storageSvc->release();
    }
    return DaqFormatWriter::finalize();
}

DaqToOStream* DaqRootWriter::createDaqToOStream(const FileTraits& traits) {
    m_strStream = new DaqToStrStream(traits);
    return  m_strStream;
}

StatusCode DaqRootWriter::writeData(const DaqCrate* daqCrate) {
  if (0 == m_strStream) {
      return StatusCode::FAILURE;
    }
    const EventReadout& eventReadout = daqCrate->eventReadout();
    m_strStream->eventSaved(eventReadout.bufferSize());
    return StatusCode::SUCCESS;
}

void DaqRootWriter::wroteFileHead() {
  writeDaqBlock(createDaqBlock(),
                m_strStream->getFileBlockCount());
}

void DaqRootWriter::wroteData() {
  // Never called
}

void DaqRootWriter::wroteFileFoot() {
  writeDaqBlock(createDaqBlock(),
                m_strStream->getFileBlockCount());
}

DaqBlock* DaqRootWriter::createDaqBlock() {
    string content = m_strStream->str();
    m_strStream->splitStream();
    unsigned int length = content.length();
    DaqBlock::RawData* rawData = new DaqBlock::RawData(length / ByteBuffer::BYTES_IN_INT);
    std::memcpy(&((*rawData)[0]),
                content.data(),
                length);
    DaqBlock* result = new DaqBlock();
    result->setBytes(rawData);
    return result;
}

StatusCode DaqRootWriter::writeDaqBlock(DaqBlock* daqBlock,
                                        unsigned int deltaEvents) {
    if (0 == m_storageSvc) {
        return StatusCode::SUCCESS;;
    }

    DaqBlockHeader* daqHeader = new DaqBlockHeader();
    daqHeader->setDeltaEvents(deltaEvents);
    daqHeader->setDaqBlock(daqBlock);

    DataSvcHelpers::RegistryEntry regEntry = DataSvcHelpers::RegistryEntry(m_daqBlockLocation);
    regEntry.setObject(daqHeader);
    debug() << "Storing DAQ Block" << endreq;
    StatusCode sc = m_storageSvc->store(daqHeader,
                                        m_daqBlockLocation);
    if(sc.isFailure()){
        error() << "Failed to store Run Header to output file" << endreq;
        return sc;
    }
    debug() << "Stored DAQ Block" << endreq;

    return StatusCode::SUCCESS;
}

---