AlignmentIORootBase Class Reference

Base class for ROOT-based I/O of Alignment parameters etc. More...

#include <AlignmentIORootBase.h>

Inheritance diagram for AlignmentIORootBase:

Protected Member Functions
	AlignmentIORootBase ()
	constructor More...
int	closeRoot (void)
	close IO More...
virtual void	createBranches (void)=0
	create root branches More...
int	openRoot (const char *filename, int iteration, bool writemode)
	open IO More...
virtual void	setBranchAddresses (void)=0
	set root branches More...
int	testFile (const char *filename, const TString &tname)
	test if file is existing and if so, what the highest iteration is More...
TString	treeName (int iter, const TString &tname)
	compose tree name More...
virtual	~AlignmentIORootBase ()
	destructor More...

Protected Attributes
bool	bWrite
TTree *	tree
TString	treename
TString	treetxt

Static Protected Attributes
static const int	itermax = 1000
static const int	nParMax = 20

Private Attributes
TFile *	myFile

Detailed Description

Base class for ROOT-based I/O of Alignment parameters etc.

Definition at line 11 of file AlignmentIORootBase.h.

Constructor & Destructor Documentation

AlignmentIORootBase()

AlignmentIORootBase::AlignmentIORootBase ( )

inlineprotected

constructor

Definition at line 14 of file AlignmentIORootBase.h.

: tree(nullptr), myFile(nullptr) {}

~AlignmentIORootBase()

AlignmentIORootBase::~AlignmentIORootBase ( )

protectedvirtual

destructor

Definition at line 9 of file AlignmentIORootBase.cc.

References myFile.

                                          {
  delete myFile;  // tree is deleted automatically with file
}

Member Function Documentation

closeRoot()

int AlignmentIORootBase::closeRoot ( void  )

protected

close IO

Definition at line 92 of file AlignmentIORootBase.cc.

References bWrite, and myFile.

Referenced by AlignmentCorrelationsIORoot::close(), AlignableDataIORoot::close(), AlignmentParametersIORoot::close(), HIPUserVariablesIORoot::close(), and MillePedeVariablesIORoot::close().

                                       {
  if (bWrite) {  //writing
    tree->Write();
  }

  delete myFile;
  myFile = nullptr;
  tree = nullptr;  // deleted with file

  return 0;
}

createBranches()

virtual void AlignmentIORootBase::createBranches ( void  )

protectedpure virtual

create root branches

Implemented in MillePedeVariablesIORoot, HIPUserVariablesIORoot, AlignmentParametersIORoot, AlignableDataIORoot, and AlignmentCorrelationsIORoot.

Referenced by openRoot().

openRoot()

int AlignmentIORootBase::openRoot ( const char *  filename, int  iteration, bool  writemode  )

protected

open IO

Definition at line 16 of file AlignmentIORootBase.cc.

References bWrite, createBranches(), corrVsCorr::filename, align_cfg::iteration, myFile, setBranchAddresses(), testFile(), treeName(), treename, treetxt, and writeEcalDQMStatus::write.

Referenced by AlignmentCorrelationsIORoot::open(), AlignableDataIORoot::open(), AlignmentParametersIORoot::open(), HIPUserVariablesIORoot::open(), and MillePedeVariablesIORoot::open().

                                                                                 {
  bWrite = write;
  int iter;

  edm::LogInfo("AlignmentIORootBase") << "File: " << filename;

  if (bWrite) {  // writing

    int iterfile = testFile(filename, treename);
    if (iterfile == -1) {
      iter = iteration;
      edm::LogInfo("AlignmentIORootBase") << "Write to new file; first iteration: " << iter;
      myFile = TFile::Open(filename, "recreate");
    } else {
      if (iteration == -1) {
        iter = iterfile + 1;
        edm::LogInfo("AlignmentIORootBase") << "Write to existing file; highest iteration: " << iter;
      } else {
        if (iteration <= iterfile) {
          edm::LogError("AlignmentIORootBase")
              << "Iteration " << iteration << " invalid or already exists for tree " << treename;
          return -1;
        }
        iter = iteration;
        edm::LogInfo("AlignmentIORootBase") << "Write to new iteration: " << iter;
      }
      myFile = TFile::Open(filename, "update");
    }

    // create tree
    myFile->cd();
    edm::LogInfo("AlignmentIORootBase") << "Tree: " << treeName(iter, treename);
    tree = new TTree(treeName(iter, treename), treetxt);
    createBranches();

  } else {  // reading

    int iterfile = testFile(filename, treename);
    if (iterfile == -1) {
      edm::LogError("AlignmentIORootBase") << "File does not exist!";
      return -1;
    } else if (iterfile == -2) {
      edm::LogError("AlignmentIORootBase") << "Tree " << treename << " does not exist in file " << filename;
      return -1;
    } else {
      if (iteration == -1) {
        iter = iterfile;
        edm::LogInfo("AlignmentIORootBase") << "Read from highest iteration: " << iter;
      } else {
        if (iteration > iterfile) {
          edm::LogError("AlignmentIORootBase") << "Iteration " << iteration << " does not exist for tree " << treename;
          return -1;
        }
        iter = iteration;
        edm::LogInfo("AlignmentIORootBase") << "Read from specified iteration: " << iter;
      }
      myFile = TFile::Open(filename, "read");
    }

    myFile->cd();
    // set trees
    edm::LogInfo("AlignmentIORootBase") << " Tree: " << treeName(iter, treename);
    tree = (TTree*)myFile->Get(treeName(iter, treename));
    if (tree == nullptr) {
      edm::LogError("AlignmentIORootBase") << "Tree does not exist in file!";
      return -1;
    }
    setBranchAddresses();
  }

  return 0;
}

setBranchAddresses()

virtual void AlignmentIORootBase::setBranchAddresses ( void  )

protectedpure virtual

set root branches

Implemented in MillePedeVariablesIORoot, AlignmentParametersIORoot, HIPUserVariablesIORoot, AlignableDataIORoot, and AlignmentCorrelationsIORoot.

Referenced by openRoot().

testFile()

int AlignmentIORootBase::testFile ( const char *  filename, const TString &  tname  )

protected

test if file is existing and if so, what the highest iteration is

Definition at line 108 of file AlignmentIORootBase.cc.

References corrVsCorr::filename, itermax, tname(), and treeName().

Referenced by openRoot(), and edmIntegrityCheck.IntegrityCheck::testFileTimeOut().

                                                                            {
  FILE* testFILE;
  testFILE = fopen(filename, "r");
  if (testFILE == nullptr) {
    return -1;
  } else {
    fclose(testFILE);
    int ihighest = -2;
    TFile* aFile = TFile::Open(filename, "read");
    for (int iter = 0; iter < itermax; iter++) {
      if ((nullptr != (TTree*)aFile->Get(treeName(iter, tname))) && (iter > ihighest))
        ihighest = iter;
    }
    delete aFile;
    return ihighest;
  }
}

treeName()

TString AlignmentIORootBase::treeName ( int  iter, const TString &  tname  )

protected

compose tree name

Definition at line 129 of file AlignmentIORootBase.cc.

References tname().

Referenced by openRoot(), and testFile().

{ return TString(tname + Form("_%i", iter)); }

Member Data Documentation

bWrite

bool AlignmentIORootBase::bWrite

protected

Definition at line 41 of file AlignmentIORootBase.h.

Referenced by AlignmentParametersIORoot::close(), closeRoot(), and openRoot().

itermax

const int AlignmentIORootBase::itermax = 1000

staticprotected

Definition at line 44 of file AlignmentIORootBase.h.

Referenced by testFile().

myFile

TFile* AlignmentIORootBase::myFile

private

Definition at line 47 of file AlignmentIORootBase.h.

Referenced by closeRoot(), openRoot(), and ~AlignmentIORootBase().

nParMax

const int AlignmentIORootBase::nParMax = 20

staticprotected

Definition at line 43 of file AlignmentIORootBase.h.

Referenced by AlignmentCorrelationsIORoot::read().

tree

TTree* AlignmentIORootBase::tree

protected

Definition at line 38 of file AlignmentIORootBase.h.

Referenced by eventstfile.Events::__iter__(), svgfig.SVG::__str__(), ZJetsTreeAnalyzer.ZJetsTreeAnalyzer::beginLoop(), tree.Tree::branch_(), MetTreeProducer.MetTreeProducer::declareVariables(), core.AutoFillTreeProducer.AutoFillTreeProducer::declareVariables(), TkAlMap.TkAlMap::detect_tracker_version(), core.AutoFillTreeProducer.AutoFillTreeProducer::fillTree(), SimpleTreeProducer.SimpleTreeProducer::process(), ZJetsTreeAnalyzer.ZJetsTreeAnalyzer::process(), MetTreeProducer.MetTreeProducer::process(), python.cmstools.EventTree::SetAlias(), eventstfile.Events::size(), eventstfile.Events::to(), tree.Tree::var(), and tree.Tree::vector().

treename

TString AlignmentIORootBase::treename

protected

Definition at line 39 of file AlignmentIORootBase.h.

Referenced by AlignableDataIORoot::AlignableDataIORoot(), AlignmentCorrelationsIORoot::AlignmentCorrelationsIORoot(), AlignmentParametersIORoot::AlignmentParametersIORoot(), HIPUserVariablesIORoot::HIPUserVariablesIORoot(), MillePedeVariablesIORoot::MillePedeVariablesIORoot(), and openRoot().

treetxt

TString AlignmentIORootBase::treetxt

protected

Definition at line 40 of file AlignmentIORootBase.h.

Referenced by AlignableDataIORoot::AlignableDataIORoot(), AlignmentCorrelationsIORoot::AlignmentCorrelationsIORoot(), AlignmentParametersIORoot::AlignmentParametersIORoot(), HIPUserVariablesIORoot::HIPUserVariablesIORoot(), MillePedeVariablesIORoot::MillePedeVariablesIORoot(), and openRoot().