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 ( )

inlineprotected

constructor

Definition at line 14 of file AlignmentIORootBase.h.

References closeRoot(), createBranches(), corrVsCorr::filename, align_cfg::iteration, openRoot(), setBranchAddresses(), testFile(), cond::persistency::GLOBAL_TAG::tname, treeName(), and ~AlignmentIORootBase().

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

AlignmentIORootBase::~AlignmentIORootBase ( )

protectedvirtual

destructor

Definition at line 9 of file AlignmentIORootBase.cc.

References myFile.

Referenced by AlignmentIORootBase().

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

Member Function Documentation

int AlignmentIORootBase::closeRoot ( void  )

protected

close IO

Definition at line 92 of file AlignmentIORootBase.cc.

References bWrite, and myFile.

Referenced by AlignmentIORootBase(), 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;
}

virtual void AlignmentIORootBase::createBranches ( void  )

protectedpure virtual

create root branches

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

Referenced by AlignmentIORootBase(), and 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 TriggerAnalyzer::write().

Referenced by AlignmentIORootBase(), 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;
}

virtual void AlignmentIORootBase::setBranchAddresses ( void  )

protectedpure virtual

set root branches

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

Referenced by AlignmentIORootBase(), and openRoot().

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 itermax, and treeName().

Referenced by AlignmentIORootBase(), 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;
  }
}

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

protected

compose tree name

Definition at line 129 of file AlignmentIORootBase.cc.

Referenced by AlignmentIORootBase(), openRoot(), and testFile().

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

Member Data Documentation

bool AlignmentIORootBase::bWrite

protected

Definition at line 41 of file AlignmentIORootBase.h.

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

const int AlignmentIORootBase::itermax = 1000

staticprotected

Definition at line 44 of file AlignmentIORootBase.h.

Referenced by testFile().

TFile* AlignmentIORootBase::myFile

private

Definition at line 47 of file AlignmentIORootBase.h.

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

const int AlignmentIORootBase::nParMax = 20

staticprotected

Definition at line 43 of file AlignmentIORootBase.h.

Referenced by AlignmentCorrelationsIORoot::read().

TTree* AlignmentIORootBase::tree

protected

Definition at line 38 of file AlignmentIORootBase.h.

Referenced by eventstfile.Events::__iter__(), svgfig.SVG::__str__(), ZJetsTreeAnalyzer.ZJetsTreeAnalyzer::beginLoop(), MetTreeProducer.MetTreeProducer::declareVariables(), core.AutoFillTreeProducer.AutoFillTreeProducer::declareVariables(), core.AutoFillTreeProducer.AutoFillTreeProducer::fillTree(), ZJetsTreeAnalyzer.ZJetsTreeAnalyzer::process(), MetTreeProducer.MetTreeProducer::process(), python.cmstools.EventTree::SetAlias(), and eventstfile.Events::to().

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().

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().