MilleBinary.cc

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,
    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);
}
}