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 16 of file AlignmentIORootBase.h.

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

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

int AlignmentIORootBase::closeRoot ( void  )

protected

close IO

Definition at line 99 of file AlignmentIORootBase.cc.

References bWrite, myFile, and tree.

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

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

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

  return 0;
}

virtual void AlignmentIORootBase::createBranches ( void  )

protectedpure virtual

create root branches

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

Referenced by openRoot().

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

protected

open IO

Definition at line 17 of file AlignmentIORootBase.cc.

References bWrite, createBranches(), lut2db_cfg::filename, align_cfg::iteration, myFile, NULL, setBranchAddresses(), testFile(), tree, treeName(), treename, treetxt, and TablePrint::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==NULL) {
      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, HIPUserVariablesIORoot, AlignmentParametersIORoot, AlignableDataIORoot, and AlignmentCorrelationsIORoot.

Referenced by 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 116 of file AlignmentIORootBase.cc.

References itermax, NULL, and treeName().

Referenced by openRoot().

{
  FILE* testFILE;
  testFILE = fopen(filename,"r");
  if (testFILE == NULL) {
    return -1;
  } else {
    fclose(testFILE);
    int ihighest=-2;
    TFile *aFile = TFile::Open(filename,"read");
    for (int iter=0; iter<itermax; iter++) {
      if ((0 != (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 138 of file AlignmentIORootBase.cc.

Referenced by openRoot(), and testFile().

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

Member Data Documentation

bool AlignmentIORootBase::bWrite

protected

Definition at line 43 of file AlignmentIORootBase.h.

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

const int AlignmentIORootBase::itermax = 1000

staticprotected

Definition at line 46 of file AlignmentIORootBase.h.

Referenced by testFile().

TFile* AlignmentIORootBase::myFile

private

Definition at line 49 of file AlignmentIORootBase.h.

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

const int AlignmentIORootBase::nParMax = 20

staticprotected

Definition at line 45 of file AlignmentIORootBase.h.

Referenced by AlignmentCorrelationsIORoot::read().

TTree* AlignmentIORootBase::tree

protected

Definition at line 40 of file AlignmentIORootBase.h.

Referenced by eventstfile.Events::__iter__(), svgfig.SVG::__str__(), AlignmentParametersIORoot::close(), closeRoot(), AlignmentCorrelationsIORoot::createBranches(), AlignableDataIORoot::createBranches(), AlignmentParametersIORoot::createBranches(), HIPUserVariablesIORoot::createBranches(), MillePedeVariablesIORoot::createBranches(), MetTreeProducer.MetTreeProducer::declareVariables(), core.AutoFillTreeProducer.AutoFillTreeProducer::declareVariables(), core.AutoFillTreeProducer.AutoFillTreeProducer::fillTree(), AlignableDataIORoot::findEntry(), HIPUserVariablesIORoot::findEntry(), openRoot(), MetTreeProducer.MetTreeProducer::process(), AlignmentCorrelationsIORoot::read(), AlignableDataIORoot::readAbsRaw(), AlignmentParametersIORoot::readOne(), HIPUserVariablesIORoot::readOne(), MillePedeVariablesIORoot::readOne(), AlignableDataIORoot::readRelRaw(), python.cmstools.EventTree::SetAlias(), AlignmentCorrelationsIORoot::setBranchAddresses(), AlignableDataIORoot::setBranchAddresses(), AlignmentParametersIORoot::setBranchAddresses(), HIPUserVariablesIORoot::setBranchAddresses(), MillePedeVariablesIORoot::setBranchAddresses(), eventstfile.Events::to(), AlignmentCorrelationsIORoot::write(), AlignableDataIORoot::writeAbsRaw(), AlignmentParametersIORoot::writeOne(), HIPUserVariablesIORoot::writeOne(), MillePedeVariablesIORoot::writeOne(), and AlignableDataIORoot::writeRelRaw().

TString AlignmentIORootBase::treename

protected

Definition at line 41 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 42 of file AlignmentIORootBase.h.

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