MillePedeVariablesIORoot.h

Go to the documentation of this file.

#ifndef MILLEPEDEVARIABLESIOROOT_H
#define MILLEPEDEVARIABLESIOROOT_H


#include "Alignment/CommonAlignment/interface/Utilities.h"
#include "Alignment/CommonAlignmentAlgorithm/interface/AlignmentIORootBase.h"
#include "Alignment/CommonAlignmentAlgorithm/interface/AlignmentUserVariablesIO.h"

#include <string>
#include <vector>

// ROOT types:
#include "Rtypes.h"

class Alignable;
class AlignmentUserVariables;

//__________________________________________________________________________________________________

class MillePedeVariablesIORoot : public AlignmentIORootBase, public AlignmentUserVariablesIO {
public:
  MillePedeVariablesIORoot();
  ~MillePedeVariablesIORoot() override {}

  void writeMillePedeVariables(
      const align::Alignables &alivec, const char *filename, int iter, bool validCheck, int &ierr);

  std::vector<AlignmentUserVariables *> readMillePedeVariables(const align::Alignables &alivec,
                                                               const char *filename,
                                                               int iter,
                                                               int &ierr);

protected:
  int writeOne(Alignable *ali) override;  // inherited from AlignmentUserVariablesIO

  AlignmentUserVariables *readOne(Alignable *ali, int &ierr) override;
  // previous inherited from AlignmentUserVariablesIO
  // inherited from AlignmentUserVariablesIO
  int open(const char *filename, int iteration, bool writemode) override {
    return this->openRoot(filename, iteration, writemode);
  }

  int close() override { return this->closeRoot(); }  // inherited from AlignmentUserVariablesIO

  void createBranches() override;  // inherited from AlignmentIORootBase
  void setBranchAddresses() override;  // inherited from AlignmentIORootBase

private:
  // variables for ROOT tree
  enum { kMaxNumPar = 20 };  // slighly above 'two bowed surfaces' limit

  unsigned int myId;
  int myObjId;

  unsigned int myNumPar;
  Byte_t myIsValid[kMaxNumPar];
  Float_t myDiffBefore[kMaxNumPar];
  Float_t myGlobalCor[kMaxNumPar];
  Float_t myPreSigma[kMaxNumPar];
  Float_t myParameter[kMaxNumPar];
  Float_t mySigma[kMaxNumPar];
  UInt_t myHitsX;
  UInt_t myHitsY;
  UInt_t myLabel;
  std::string myName;
  std::string *myNamePtr;  // needed for ROOT IO
};

#endif