MilleBinary.cpp

Go to the documentation of this file.

/*
 * MilleBinary.cpp
 *
 *  Created on: Aug 31, 2011
 *      Author: kleinwrt
 */
 
#include "MilleBinary.h"
 
namespace gbl {
 
 
MilleBinary::MilleBinary(const std::string& fileName, bool doublePrec,
                bool keepZeros, unsigned int aSize) :
                binaryFile(fileName.c_str(), std::ios::binary | std::ios::out), intBuffer(), floatBuffer(), doubleBuffer(), doublePrecision(
                                doublePrec), globalDerKeepZeros(keepZeros) {
 
        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, unsigned int numLocal,
                unsigned int* indLocal, 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 < numLocal; ++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] or globalDerKeepZeros) {
                                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 < numLocal; ++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] or globalDerKeepZeros) {
                                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);
}
}