MillePedeVariablesIORoot.h

Go to the documentation of this file.

#ifndef MILLEPEDEVARIABLESIOROOT_H
#define MILLEPEDEVARIABLESIOROOT_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 std::vector<Alignable*> &alivec, const char *filename,
                   int iter, bool validCheck, int &ierr);

  std::vector<AlignmentUserVariables*> readMillePedeVariables
    (const std::vector<Alignable*> &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