This class provides a simple interface to write Matricis to a root tree The size of the Matrix cannot exceed COLMAX x ROWMAX. More...

#include <MatrixNtupleBranch.h>

Collaboration diagram for Trk::MatrixNtupleBranch:

Public Member Functions
bool	initForWrite (TTree &tree, const std::string &varname, int ncol, int nrow, const std::string &prefix="")
	initialize class for writing More...

bool	initForRead (TTree &tree, const std::string &varname, int ncol, int nrow, const std::string &prefix="")
	initialize class for reading More...

bool	fill (const Amg::MatrixX &matrix)
	fill a vector More...

bool	fill (const HepGeom::Rotate3D &matrix)
	fill a vector More...

Public Attributes
int	m_nrows {-1}

int	m_ncols {-1}

float	m_matrix [COLMAX][ROWMAX] {}

Static Public Attributes
static const int	COLMAX = 10
	data More...

static const int	ROWMAX = 10

Detailed Description

This class provides a simple interface to write Matricis to a root tree The size of the Matrix cannot exceed COLMAX x ROWMAX.

The class requires all matrices to have the same size

Usage:

TTree* myTree = new TTree("data","data"); MatrixNtupleBranch branch; branch.initForWrite(*myTree,"vec",3,"");

for(int i=0;i<n;++i){ branch.fill( ); myTree->Fill(); }

Definition at line 33 of file MatrixNtupleBranch.h.

Member Function Documentation

◆ fill() [1/2]

bool Trk::MatrixNtupleBranch::fill ( const Amg::MatrixX & matrix )

fill a vector

Definition at line 44 of file MatrixNtupleBranch.cxx.

                                                           {
     if( m_ncols == -1 ) return false;
     if( matrix.rows() > m_nrows && matrix.cols() > m_ncols ) return false; 
  
     for( int i=0;i<m_nrows;++i ){
       for( int j=0;j<m_ncols;++j ){
     if( i < matrix.rows() && j < matrix.cols() ) m_matrix[i][j] = matrix(i,j);
     else                                         m_matrix[i][j] = 0.;
       }
     }
     return true;
   }

◆ fill() [2/2]

bool Trk::MatrixNtupleBranch::fill ( const HepGeom::Rotate3D & matrix )

fill a vector

Definition at line 57 of file MatrixNtupleBranch.cxx.

                                                                {
     if( m_ncols == -1 ) return false;
     if(  m_nrows != 3  && m_ncols != 3 ) return false; 
  
     for( int i=0;i<m_nrows;++i ){
       for( int j=0;j<m_ncols;++j ){
     if( i < 3 && j < 3 ) m_matrix[i][j] = matrix(i,j);
     else                 m_matrix[i][j] = 0.;
       }
     }
     return true;
   }

◆ initForRead()

bool Trk::MatrixNtupleBranch::initForRead	(	TTree &	tree,
		const std::string &	varname,
		int	ncol,
		int	nrow,
		const std::string &	prefix = `""`
	)

initialize class for reading

Definition at line 27 of file MatrixNtupleBranch.cxx.

                                                                                                                           { 
     if( ncol >= COLMAX ) return false;
     if( nrow >= ROWMAX ) return false;
     m_ncols = ncol;
     m_nrows = nrow;
     for( int i=0;i<nrow;++i ){
       for( int j=0;j<ncol;++j ){
     TString bname = prefix.c_str();
     bname += varname;
     bname += i;
     bname += j;
     tree.SetBranchAddress(bname,&m_matrix[i][j]);
       }
     }
     return true;
   }

◆ initForWrite()

bool Trk::MatrixNtupleBranch::initForWrite	(	TTree &	tree,
		const std::string &	varname,
		int	ncol,
		int	nrow,
		const std::string &	prefix = `""`
	)

initialize class for writing

Definition at line 10 of file MatrixNtupleBranch.cxx.

                                                                                                                            {
     if( ncol >= COLMAX ) return false;
     if( nrow >= ROWMAX ) return false;
     m_ncols = ncol;
     m_nrows = nrow;
     for( int i=0;i<nrow;++i ){
       for( int j=0;j<ncol;++j ){
     TString bname = prefix.c_str();
     bname += varname;
     bname += i;
     bname += j;
     tree.Branch(bname,&m_matrix[i][j]);
       }
     }
     return true;
   }

Member Data Documentation

◆ COLMAX

const int Trk::MatrixNtupleBranch::COLMAX = 10

static

data

Definition at line 48 of file MatrixNtupleBranch.h.

◆ m_matrix

float Trk::MatrixNtupleBranch::m_matrix[COLMAX][ROWMAX] {}

Definition at line 52 of file MatrixNtupleBranch.h.

◆ m_ncols

int Trk::MatrixNtupleBranch::m_ncols {-1}

Definition at line 51 of file MatrixNtupleBranch.h.

◆ m_nrows

int Trk::MatrixNtupleBranch::m_nrows {-1}

Definition at line 50 of file MatrixNtupleBranch.h.

◆ ROWMAX

const int Trk::MatrixNtupleBranch::ROWMAX = 10