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
bool	initForRead (TTree &tree, const std::string &varname, int ncol, int nrow, const std::string &prefix="")
	initialize class for reading
bool	fill (const Amg::MatrixX &matrix)
	fill a vector
bool	fill (const HepGeom::Rotate3D &matrix)
	fill a vector

Public Attributes
int	m_nrows {-1}
int	m_ncols {-1}
float	m_matrix [COLMAX][ROWMAX] {}

Static Public Attributes
static const int	COLMAX = 10
	data
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.

52{};

◆ m_ncols

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

Definition at line 51 of file MatrixNtupleBranch.h.

51{-1};

◆ m_nrows

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

Definition at line 50 of file MatrixNtupleBranch.h.

50{-1};

◆ ROWMAX

const int Trk::MatrixNtupleBranch::ROWMAX = 10