In This Package:

DaqToFStream.cc

Go to the documentation of this file.

/*
 *  DaqToFStream.cc
 *  FileReadoutFormat
 *
 *  Created by Simon Patton on 7/18/10.
 *  Copyright 2010 DayaBay Collaboration. All rights reserved.
 *
 */
#include "FileReadoutFormat/DaqToFStream.h"

#include "DaqReadoutFormat/DaqBuffer.h"
#include "FileReadoutFormat/FileEndRecord.h"
#include "FileReadoutFormat/FileStartRecord.h"
#include "FileReadoutFormat/FileTraits.h"

using DybDaq::FileEndRecord;
using DybDaq::FileStartRecord;
using DybDaq::FileTraits;
using DybDaq::DaqToFStream;
using std::fstream;
using std::ios;
using std::ofstream;
using std::ostream;
using std::string;

DaqToFStream::DaqToFStream(const FileTraits& traits) :
    DaqToOStream(traits),
    m_ofstream(0),
    m_fileDir(0),
    m_fileName(0) {
}

DaqToFStream::DaqToFStream(const FileTraits& traits,
                       std::string& fileDir) :
    DaqToOStream(traits),
    m_ofstream(0),
    m_fileDir(0),
    m_fileName(0) {
    if (0 != &fileDir) {
        m_fileDir = new string(".");
    }
}

DaqToFStream::~DaqToFStream() {
    if (0 != m_fileDir) {
        delete m_fileDir;
    }
    if (0 != m_fileName) {
        delete m_fileName;
    }
    if (hasOStream()) {
        if (m_ofstream->is_open()) {
            m_ofstream->close();
        }
    }
}

bool DaqToFStream::openFile(const string& fileName,
                          const FileStartRecord& record) {
    if (hasFileBase()) {
        return false;
    }
    if (0 != m_fileName) {
        delete m_fileName;
    }
    m_fileName = new string(fileName);
    return open(record);
}

const FileStartRecord* DaqToFStream::openFile(const string& fileName,
                                            const unsigned int fileNumber,
                                            const FileTraits::FileDateTime& dateTime,
                                            const unsigned int dataBlockLimit,
                                            const unsigned int megabyteLimit) {
    if (hasFileBase()) {
        return 0;
    }
    if (0 != m_fileName) {
        delete m_fileName;
    }
    m_fileName = new string(fileName);
    return openElement(fileNumber,
                       dateTime,
                       dataBlockLimit,
                       megabyteLimit);
}

void DaqToFStream::prepareNext() {
    // If file base is not set used provided name.
    const string& fileBase = getFileBase();
    if (0 == &fileBase) {
      return;
    }

    unsigned int segmentCount = getFileCount() + 1;
    char* segmentSuffix = new char[11];
    sprintf(segmentSuffix,
            "_%.4d.data",
            segmentCount);
    segmentSuffix[10] = 0;
    string fileName = getFileBase() + segmentSuffix;
    delete[] segmentSuffix;
    if (0 != m_fileDir) {
        fileName = *m_fileDir + "/" + fileName;
    }
    if (0 != m_fileName) {
        delete m_fileName;
    }
    m_fileName = new string(fileName);
}

ostream* DaqToFStream::openStream() {
    if (0 == m_fileName) {
        prepareNext();
    }
    m_ofstream = new std::ofstream(m_fileName->c_str(),
                                   fstream::binary | fstream::out | fstream::trunc);
    if (0 != m_fileName) {
        delete m_fileName;
    }
    m_fileName = 0;
    return m_ofstream;
}

void DaqToFStream::closeStream(std::ostream& stream) {
    ofstream& fileStream = dynamic_cast<ofstream&> (stream);
    fileStream.close();
    m_ofstream = 0;
}

---