CellularAutomaton Class Reference

#include <CellularAutomaton.h>

Public Member Functions
	CellularAutomaton (const CAGraph &graph)
void	createAndConnectCells (const std::vector< const HitDoublets * > &, const TrackingRegion &, const float, const float, const float)
void	evolve (const unsigned int)
void	findNtuplets (std::vector< CACell::CAntuplet > &, const unsigned int)
void	findTriplets (const std::vector< const HitDoublets * > &hitDoublets, std::vector< CACell::CAntuplet > &foundTriplets, const TrackingRegion &region, const float thetaCut, const float phiCut, const float hardPtCut)

Private Attributes
CAGraph	theLayerGraph
std::vector< std::vector < CACell * > >	theNtuplets
std::vector< CACell * >	theRootCells

Detailed Description

Definition at line 8 of file CellularAutomaton.h.

Constructor & Destructor Documentation

CellularAutomaton::CellularAutomaton ( const CAGraph &  graph )

inline

Definition at line 11 of file CellularAutomaton.h.

    : theLayerGraph(graph)
    {

    }

Member Function Documentation

void CellularAutomaton::createAndConnectCells	(	const std::vector< const HitDoublets * > &	hitDoublets,
		const TrackingRegion &	region,
		const float	thetaCut,
		const float	phiCut,
		const float	hardPtCut
	)

Definition at line 3 of file CellularAutomaton.cc.

References i, HitDoublets::innerHitId(), TrackingRegion::origin(), TrackingRegion::originRBound(), HitDoublets::outerHitId(), TrackingRegion::ptMin(), ptmin, HitDoublets::size(), theLayerGraph, CAGraph::theLayerPairs, CAGraph::theLayers, CAGraph::theRootLayers, class-composition::visited, PV3DBase< T, PVType, FrameType >::x(), and PV3DBase< T, PVType, FrameType >::y().

Referenced by CAHitQuadrupletGenerator::hitQuadruplets().

{
    unsigned int cellId = 0;
    float ptmin = region.ptMin();
    float region_origin_x = region.origin().x();
    float region_origin_y = region.origin().y();
    float region_origin_radius = region.originRBound();

    std::vector<bool> alreadyVisitedLayerPairs;
    alreadyVisitedLayerPairs.resize(theLayerGraph.theLayerPairs.size());
    for (auto visited : alreadyVisitedLayerPairs)
    {
        visited = false;
    }
    for (int rootVertex : theLayerGraph.theRootLayers)
    {

        std::queue<int> LayerPairsToVisit;

        for (int LayerPair : theLayerGraph.theLayers[rootVertex].theOuterLayerPairs)
        {
            LayerPairsToVisit.push(LayerPair);

        }

        unsigned int numberOfLayerPairsToVisitAtThisDepth =
                LayerPairsToVisit.size();

        while (!LayerPairsToVisit.empty())
        {
            auto currentLayerPair = LayerPairsToVisit.front();
            auto & currentLayerPairRef = theLayerGraph.theLayerPairs[currentLayerPair];
            auto & currentInnerLayerRef = theLayerGraph.theLayers[currentLayerPairRef.theLayers[0]];
            auto & currentOuterLayerRef = theLayerGraph.theLayers[currentLayerPairRef.theLayers[1]];
            bool allInnerLayerPairsAlreadyVisited   { true };

            for (auto innerLayerPair : currentInnerLayerRef.theInnerLayerPairs)
            {
                allInnerLayerPairsAlreadyVisited &=
                        alreadyVisitedLayerPairs[innerLayerPair];
            }

            if (alreadyVisitedLayerPairs[currentLayerPair] == false
                    && allInnerLayerPairsAlreadyVisited)
            {

                const HitDoublets* doubletLayerPairId =
                        hitDoublets[currentLayerPair];
                auto numberOfDoublets = doubletLayerPairId->size();
                currentLayerPairRef.theFoundCells.reserve(numberOfDoublets);
                for (unsigned int i = 0; i < numberOfDoublets; ++i)
                {
                    currentLayerPairRef.theFoundCells.emplace_back(
                            doubletLayerPairId, i, cellId,
                            doubletLayerPairId->innerHitId(i),
                            doubletLayerPairId->outerHitId(i));
                    currentOuterLayerRef.isOuterHitOfCell[doubletLayerPairId->outerHitId(i)].push_back(
                            &(currentLayerPairRef.theFoundCells[i]));
                    cellId++;

                    for (auto neigCell : currentInnerLayerRef.isOuterHitOfCell[doubletLayerPairId->innerHitId(i)])
                    {
                        currentLayerPairRef.theFoundCells[i].checkAlignmentAndTag(
                                neigCell, ptmin, region_origin_x,
                                region_origin_y, region_origin_radius, thetaCut,
                                phiCut, hardPtCut);
                    }

                }

                for (auto outerLayerPair : currentOuterLayerRef.theOuterLayerPairs)
                {
                    LayerPairsToVisit.push(outerLayerPair);
                }

                alreadyVisitedLayerPairs[currentLayerPair] = true;
            }
            LayerPairsToVisit.pop();
            numberOfLayerPairsToVisitAtThisDepth--;
            if (numberOfLayerPairsToVisitAtThisDepth == 0)
            {
                numberOfLayerPairsToVisitAtThisDepth = LayerPairsToVisit.size();
            }

        }

    }

}

void CellularAutomaton::evolve

(

const unsigned int

minHitsPerNtuplet

)

Definition at line 94 of file CellularAutomaton.cc.

References align_cfg::iteration, theLayerGraph, CAGraph::theLayerPairs, CAGraph::theLayers, theRootCells, and CAGraph::theRootLayers.

Referenced by CAHitQuadrupletGenerator::hitQuadruplets().

{
    unsigned int numberOfIterations = minHitsPerNtuplet - 2;
    // keeping the last iteration for later
    for (unsigned int iteration = 0; iteration < numberOfIterations - 1;
            ++iteration)
    {
        for (auto& layerPair : theLayerGraph.theLayerPairs)
        {
            for (auto& cell : layerPair.theFoundCells)
            {
                cell.evolve();
            }
        }

        for (auto& layerPair : theLayerGraph.theLayerPairs)
        {
            for (auto& cell : layerPair.theFoundCells)
            {
                cell.updateState();
            }
        }

    }

    //last iteration


    for(int rootLayerId : theLayerGraph.theRootLayers)
    {
        for(int rootLayerPair: theLayerGraph.theLayers[rootLayerId].theOuterLayerPairs)
        {
            for (auto& cell : theLayerGraph.theLayerPairs[rootLayerPair].theFoundCells)
            {
                cell.evolve();
                cell.updateState();
                if (cell.isRootCell(minHitsPerNtuplet - 2))
                {
                    theRootCells.push_back(&cell);
                }
            }
        }
    }

}

void CellularAutomaton::findNtuplets	(	std::vector< CACell::CAntuplet > &	foundNtuplets,
		const unsigned int	minHitsPerNtuplet
	)

Definition at line 140 of file CellularAutomaton.cc.

References theRootCells.

Referenced by CAHitQuadrupletGenerator::hitQuadruplets().

{
    std::vector<CACell*> tmpNtuplet;
    tmpNtuplet.reserve(minHitsPerNtuplet);

    for (CACell* root_cell : theRootCells)
    {
        tmpNtuplet.clear();
        tmpNtuplet.push_back(root_cell);
        root_cell->findNtuplets(foundNtuplets, tmpNtuplet, minHitsPerNtuplet);
    }

}

void CellularAutomaton::findTriplets	(	const std::vector< const HitDoublets * > &	hitDoublets,
		std::vector< CACell::CAntuplet > &	foundTriplets,
		const TrackingRegion &	region,
		const float	thetaCut,
		const float	phiCut,
		const float	hardPtCut
	)

Definition at line 157 of file CellularAutomaton.cc.

References i, HitDoublets::innerHitId(), TrackingRegion::origin(), TrackingRegion::originRBound(), HitDoublets::outerHitId(), TrackingRegion::ptMin(), ptmin, HitDoublets::size(), theLayerGraph, CAGraph::theLayerPairs, CAGraph::theLayers, CAGraph::theRootLayers, class-composition::visited, PV3DBase< T, PVType, FrameType >::x(), and PV3DBase< T, PVType, FrameType >::y().

Referenced by CAHitTripletGenerator::hitTriplets().

{
    unsigned int cellId = 0;
    float ptmin = region.ptMin();
    float region_origin_x = region.origin().x();
    float region_origin_y = region.origin().y();
    float region_origin_radius = region.originRBound();

    std::vector<bool> alreadyVisitedLayerPairs;
    alreadyVisitedLayerPairs.resize(theLayerGraph.theLayerPairs.size());
    for (auto visited : alreadyVisitedLayerPairs)
    {
        visited = false;
    }
    for (int rootVertex : theLayerGraph.theRootLayers)
    {

        std::queue<int> LayerPairsToVisit;

        for (int LayerPair : theLayerGraph.theLayers[rootVertex].theOuterLayerPairs)
        {
            LayerPairsToVisit.push(LayerPair);

        }

        unsigned int numberOfLayerPairsToVisitAtThisDepth =
                LayerPairsToVisit.size();

        while (!LayerPairsToVisit.empty())
        {
            auto currentLayerPair = LayerPairsToVisit.front();
            auto & currentLayerPairRef = theLayerGraph.theLayerPairs[currentLayerPair];
            auto & currentInnerLayerRef = theLayerGraph.theLayers[currentLayerPairRef.theLayers[0]];
            auto & currentOuterLayerRef = theLayerGraph.theLayers[currentLayerPairRef.theLayers[1]];
            bool allInnerLayerPairsAlreadyVisited   { true };

            for (auto innerLayerPair : currentInnerLayerRef.theInnerLayerPairs)
            {
                allInnerLayerPairsAlreadyVisited &=
                        alreadyVisitedLayerPairs[innerLayerPair];
            }

            if (alreadyVisitedLayerPairs[currentLayerPair] == false
                    && allInnerLayerPairsAlreadyVisited)
            {

                const HitDoublets* doubletLayerPairId =
                        hitDoublets[currentLayerPair];
                auto numberOfDoublets = doubletLayerPairId->size();
                currentLayerPairRef.theFoundCells.reserve(numberOfDoublets);
                for (unsigned int i = 0; i < numberOfDoublets; ++i)
                {
                    currentLayerPairRef.theFoundCells.emplace_back(
                            doubletLayerPairId, i, cellId,
                            doubletLayerPairId->innerHitId(i),
                            doubletLayerPairId->outerHitId(i));
                    currentOuterLayerRef.isOuterHitOfCell[doubletLayerPairId->outerHitId(i)].push_back(
                            &(currentLayerPairRef.theFoundCells[i]));
                    cellId++;

                    for (auto neigCell : currentInnerLayerRef.isOuterHitOfCell[doubletLayerPairId->innerHitId(i)])
                    {
                        currentLayerPairRef.theFoundCells[i].checkAlignmentAndPushTriplet(
                                neigCell, foundTriplets, ptmin, region_origin_x,
                                region_origin_y, region_origin_radius, thetaCut,
                                phiCut, hardPtCut);
                    }

                }

                for (auto outerLayerPair : currentOuterLayerRef.theOuterLayerPairs)
                {
                    LayerPairsToVisit.push(outerLayerPair);
                }

                alreadyVisitedLayerPairs[currentLayerPair] = true;
            }
            LayerPairsToVisit.pop();
            numberOfLayerPairsToVisitAtThisDepth--;
            if (numberOfLayerPairsToVisitAtThisDepth == 0)
            {
                numberOfLayerPairsToVisitAtThisDepth = LayerPairsToVisit.size();
            }

        }

    }

}

Member Data Documentation

CAGraph CellularAutomaton::theLayerGraph

private

Definition at line 26 of file CellularAutomaton.h.

Referenced by createAndConnectCells(), evolve(), and findTriplets().

std::vector<std::vector<CACell*> > CellularAutomaton::theNtuplets

private

Definition at line 28 of file CellularAutomaton.h.

std::vector<CACell*> CellularAutomaton::theRootCells

private

Definition at line 27 of file CellularAutomaton.h.

Referenced by evolve(), and findNtuplets().