BucketGraphUtils.h File Reference

#include <vector>
#include <cstdint>
#include <algorithm>
#include <cmath>
#include <unordered_set>
#include <ActsGeometryInterfaces/GeometryContext.h>
#include "MuonSpacePoint/SpacePointContainer.h"
#include "MuonSpacePoint/SpacePointPerLayerSorter.h"
#include "GeoPrimitives/GeoPrimitives.h"

Include dependency graph for BucketGraphUtils.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
struct	MuonML::BucketGraphUtils::NodeAux

Namespaces
namespace	MuonML
namespace	MuonML::BucketGraphUtils

Functions
double	MuonML::BucketGraphUtils::bucket_size_mm (const MuonR4::SpacePointBucket &b)
bool	MuonML::BucketGraphUtils::keepBucketAsTrainingNode (const MuonR4::SpacePointBucket &b)
	bucket_size == 0, where bucket_size is bucket_max - bucket_min.
void	MuonML::BucketGraphUtils::buildNodesAndFeatures (const MuonR4::SpacePointContainer &buckets, const ActsTrk::GeometryContext &gctx, std::vector< NodeAux > &nodes, std::vector< float > &featuresLeaves, std::vector< int64_t > &spInBucket)
	Build nodes + flat features (N,6) and number of SPs per kept bucket.
void	MuonML::BucketGraphUtils::buildSparseEdges (const std::vector< NodeAux > &nodes, int minLayers, int maxChamberDelta, int maxSectorDelta, double maxDistXY, double maxAbsDz, std::vector< int64_t > &srcEdges, std::vector< int64_t > &dstEdges)
size_t	MuonML::BucketGraphUtils::packEdgeIndex (const std::vector< int64_t > &srcEdges, const std::vector< int64_t > &dstEdges, std::vector< int64_t > &edgeIndexPacked)