MilleBinary.cc

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file MilleBinary.cc

/*
 * MilleBinary.cpp
 *
 *  Created on: Aug 31, 2011
 *      Author: kleinwrt
 */

#include "MilleBinary.h"

namespace gbl {


MilleBinary::MilleBinary(const std::string &fileName, bool doublePrec,
                unsigned int aSize) :
                binaryFile(fileName.c_str(), std::ios::binary | std::ios::out), intBuffer(), floatBuffer(), doubleBuffer(), doublePrecision(
                                doublePrec) {
        intBuffer.reserve(aSize);
        intBuffer.push_back(0); // first word is error counter
        if (doublePrecision) {
                doubleBuffer.reserve(aSize);
                doubleBuffer.push_back(0.);

        } else {
                floatBuffer.reserve(aSize);
                floatBuffer.push_back(0.);
        }
}

MilleBinary::~MilleBinary() {
        binaryFile.close();
}


void MilleBinary::addData(double aMeas, double aErr,
                const std::vector<unsigned int> &indLocal,
                const std::vector<double> &derLocal, const std::vector<int> &labGlobal,
                const std::vector<double> &derGlobal) {

        if (doublePrecision) {
                // double values
                intBuffer.push_back(0);
                doubleBuffer.push_back(aMeas);
                for (unsigned int i = 0; i < indLocal.size(); ++i) {
                        intBuffer.push_back(indLocal[i]);
                        doubleBuffer.push_back(derLocal[i]);
                }
                intBuffer.push_back(0);
                doubleBuffer.push_back(aErr);
                for (unsigned int i = 0; i < labGlobal.size(); ++i) {
                        if (derGlobal[i]) {
                                intBuffer.push_back(labGlobal[i]);
                                doubleBuffer.push_back(derGlobal[i]);
                        }
                }
        } else {
                // float values
                intBuffer.push_back(0);
                floatBuffer.push_back(aMeas);
                for (unsigned int i = 0; i < indLocal.size(); ++i) {
                        intBuffer.push_back(indLocal[i]);
                        floatBuffer.push_back(derLocal[i]);
                }
                intBuffer.push_back(0);
                floatBuffer.push_back(aErr);
                for (unsigned int i = 0; i < labGlobal.size(); ++i) {
                        if (derGlobal[i]) {
                                intBuffer.push_back(labGlobal[i]);
                                floatBuffer.push_back(derGlobal[i]);
                        }
                }
        }
}

void MilleBinary::writeRecord() {

        const int recordLength =
                        (doublePrecision) ? -intBuffer.size() * 2 : intBuffer.size() * 2;
        binaryFile.write(reinterpret_cast<const char*>(&recordLength),
                        sizeof(recordLength));
        if (doublePrecision)
                binaryFile.write(reinterpret_cast<char*>(&doubleBuffer[0]),
                                doubleBuffer.size() * sizeof(doubleBuffer[0]));
        else
                binaryFile.write(reinterpret_cast<char*>(&floatBuffer[0]),
                                floatBuffer.size() * sizeof(floatBuffer[0]));
        binaryFile.write(reinterpret_cast<char*>(&intBuffer[0]),
                        intBuffer.size() * sizeof(intBuffer[0]));
// start with new record
        intBuffer.resize(1);
        if (doublePrecision)
                doubleBuffer.resize(1);
        else
                floatBuffer.resize(1);
}
}