GeoNodePath.h

Go to the documentation of this file.

#ifndef LARCOREALG_GEOMETRY_GEONODEPATH_H
#define LARCOREALG_GEOMETRY_GEONODEPATH_H

// LArSoft libraries
#include "larcorealg/Geometry/LocalTransformation.h"

// ROOT libraries
#include "TGeoNode.h"

// C++ standard library
#include <cstddef> // std::size_t
#include <initializer_list>
#include <string>
#include <vector>

namespace geo {

  class GeoNodePath {

  public:
    // --- BEGIN Data types ----------------------------------------------------
    using Node_t = TGeoNode const;

    using Nodes_t = std::vector<Node_t const*>;

    using Depth_t = std::size_t;

    // --- END Data types ------------------------------------------------------

    // --- BEGIN Constructors and destructor -----------------------------------
    GeoNodePath() = default;

    GeoNodePath(std::initializer_list<TGeoNode const*> nodes) : fNodes(nodes) {}

    template <typename Iter>
    GeoNodePath(Iter begin, Iter end) : fNodes(begin, end)
    {}

    // --- END Constructors and destructor -------------------------------------

    // --- BEGIN Query and access ----------------------------------------------
    bool empty() const { return fNodes.empty(); }

    Depth_t depth() const { return fNodes.size(); }

    Node_t const& current() const { return *(fNodes.back()); }

    // --- END Query and access ------------------------------------------------

    // --- BEGIN Content management --------------------------------------------
    void append(Node_t const& node) { fNodes.push_back(&node); }

    void pop() { fNodes.pop_back(); }
    // --- END Content management ----------------------------------------------

    template <typename Matrix = TGeoHMatrix>
    Matrix currentTransformation() const;

    operator std::string() const;

  private:
    Nodes_t fNodes; 

  }; // class GeoNodePath

} // namespace geo

//------------------------------------------------------------------------------
//--- template implementation
//------------------------------------------------------------------------------
template <typename Matrix /* = TGeoHMatrix */>
Matrix geo::GeoNodePath::currentTransformation() const
{
  return geo::transformationFromPath<Matrix>(fNodes.begin(), fNodes.end());
} // geo::GeoNodePath::currentTransformation()

//------------------------------------------------------------------------------

#endif // LARCOREALG_GEOMETRY_GEONODEPATH_H