MilleBinary.cc

Go to the documentation of this file.

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

#include "Alignment/ReferenceTrajectories/interface/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, 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]) {
          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]) {
          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);
  }
}