TrackDrivenSeedingTool.h

Go to the documentation of this file.

/*
 *
 * TrackDrivenSeedingTool.h header template automatically generated by a class generator
 * Creation date : mer. f�vr. 24 2016
 *
 * This file is part of APRILContent libraries.
 * 
 * APRILContent is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 * based upon these libraries are permitted. Any copy of these libraries
 * must include this copyright notice.
 * 
 * APRILContent is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 * 
 * You should have received a copy of the GNU General Public License
 * along with APRILContent.  If not, see <http://www.gnu.org/licenses/>.
 * 
 * @author Remi Ete
 * @copyright CNRS , IPNL
 */


#ifndef TRACKDRIVENSEEDINGTOOL_H
#define TRACKDRIVENSEEDINGTOOL_H

#include "APRILTools/ConnectorAlgorithmTool.h"

namespace april_content
{

class CaloHit;
class Connector;

typedef std::map<const pandora::Track* const, pandora::CaloHitVector> TrackCaloHitVector;

class TrackDrivenSeedingTool : public ConnectorAlgorithmTool
{
public:
    class Factory : public pandora::AlgorithmToolFactory
    {
    public:
        pandora::AlgorithmTool *CreateAlgorithmTool() const;
    };

    ~TrackDrivenSeedingTool() { /* nop */ }

    pandora::StatusCode Process(const pandora::Algorithm &algorithm, const pandora::CaloHitList *const pCaloHitList = nullptr);

    pandora::StatusCode ReadSettings(const pandora::TiXmlHandle xmlHandle);

    static inline TrackCaloHitVector& GetTrackAndInitCaloHits() { return m_trackHitVector; }

private:
    pandora::StatusCode FindInitialCaloHits(const pandora::Algorithm &algorithm, const pandora::Track *pTrack, const pandora::OrderedCaloHitList& orderedCaloHitList,
            pandora::CaloHitVector &caloHitVector);

    const pandora::CartesianVector GetFromHitDirection(const april_content::CaloHit *const pCaloHit);

    pandora::StatusCode ConnectCaloHits(const pandora::Algorithm &algorithm, const pandora::Track *pTrack,
            pandora::CaloHitVector &caloHitVector);

    void CheckInitialTrackHit() const;

    void CleanTrackInitHitsAssociation(const pandora::Algorithm &algorithm) const;

    bool IsSiblingTrack(const pandora::Track *const trackA, const pandora::Track *const trackB) const;


private:
    float                           m_initialHitSearchRange;
    float                           m_hitSearchRangeFine;
    float                           m_hitSearchRangeCoarse;
    float                           m_hitSearchRangeAtBoundary;
    unsigned int                    m_maxInitialPseudoLayer;
    float                           m_maxInitialTrackDistance;
    float                           m_maxInitialTrackAngle;

    float                           m_maxConnectionPseudoLayer;
    float                           m_maxTransverseDistanceFine;
    float                           m_maxTransverseDistanceCoarse;
    float                           m_maxDistanceToTrackFine;
    float                           m_maxDistanceToTrackCoarse;

    bool                            m_shouldUseIsolatedHits;
    unsigned int                    m_connectorCreationStage;

    static TrackCaloHitVector       m_trackHitVector;
};

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

inline pandora::AlgorithmTool *TrackDrivenSeedingTool::Factory::CreateAlgorithmTool() const
{
    return new TrackDrivenSeedingTool();
}

} 

#endif  //  TRACKDRIVENSEEDINGTOOL_H