VectorNtupleBranch.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2017 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef TRK_VECTORNTUPLEBRANCH_H
#define TRK_VECTORNTUPLEBRANCH_H
 
#include <string>
class TTree;
 
#include "EventPrimitives/EventPrimitives.h"
#include "CLHEP/Geometry/Vector3D.h"
#include "CLHEP/Geometry/Point3D.h"
#include "CLHEP/Matrix/Vector.h"
 
namespace Trk {

  struct VectorNtupleBranch {

    bool initForWrite(TTree& tree, const std::string& varname, int nrow, const std::string& prefix = ""  );

    bool initForRead(TTree& tree, const std::string& varname, int nrow, const std::string& prefix = "" );
    
    bool fill( const Amg::VectorX& matrix );

    bool fill( const CLHEP::HepVector& vec );

    bool fill( const CLHEP::Hep3Vector& vec );

    bool fill( const HepGeom::Point3D<double>& vec );

    bool fill( const HepGeom::Vector3D<double>& vec );

    static const int ROWMAX = 10;
    int m_nrows {-1};
    float  m_vector[ROWMAX]{}; 
  };   
 
}
 
#endif