ATLAS Offline Software
Namespaces | Functions
ExaTrkXUtils.cxx File Reference
#include "ExaTrkXUtils.hpp"
#include <queue>
#include <boost/graph/topological_sort.hpp>
#include <boost/graph/graphviz.hpp>
#include <boost/property_map/property_map.hpp>
#include <boost/graph/graph_utility.hpp>
#include <boost/graph/graph_traits.hpp>

Go to the source code of this file.

Namespaces

 ExaTrkXUtils
 

Functions

void ExaTrkXUtils::buildEdges (const std::vector< float > &embedFeatures, std::vector< int64_t > &senders, std::vector< int64_t > &receivers, int64_t numSpacepoints, int embeddingDim, float rVal, int kVal)
 
void ExaTrkXUtils::CCandWalk (vertex_t numSpacepoints, const std::vector< int64_t > &rowIndices, const std::vector< int64_t > &colIndices, const std::vector< weight_t > &edgeWeights, std::vector< std::vector< uint32_t > > &tracks, float ccCut, float walkMin, float walkMax)
 
std::vector< std::vector< vertex_t > > ExaTrkXUtils::getSimplePath (const UndirectedGraph &G)
 
std::vector< vertex_t > ExaTrkXUtils::findNextNode (const Graph &G, vertex_t current_hit, float th_min, float th_add)
 
std::vector< std::vector< vertex_t > > ExaTrkXUtils::buildRoads (const Graph &G, vertex_t starting_node, std::function< std::vector< vertex_t >(const Graph &, vertex_t)> next_node_fn, std::map< vertex_t, bool > &used_hits)
 
Graph ExaTrkXUtils::cleanupGraph (const Graph &G, float ccCut)
 
void ExaTrkXUtils::calculateEdgeFeatures (const std::vector< float > &gNodeFeatures, int64_t numSpacepoints, const std::vector< int64_t > &rowIndices, const std::vector< int64_t > &colIndices, std::vector< float > &edgeFeatures)