PndLmdPairFinderTask.h

Go to the documentation of this file.

/*
 * PairFinderTask.h
 *
 *  Created on: Jul 22, 2014
 *      Author: Roman Klasen, roklasen@uni-mainz.de or klasen@kph.uni-mainz.de
 */

#ifndef PNDLMDPAIRFINDERTASK_H_
#define PNDLMDPAIRFINDERTASK_H_

#include "PndLmdAlignStructs.h"
#include <PndLmdHitPair.h>
#include <PndSdsHit.h>
#include "PndSdsTask.h"

#include "TClonesArray.h"

#include <boost/property_tree/ptree.hpp>

#include <iostream>
#include <string>
#include <vector>

class PndLmdGeometryHelper;

class PndLmdPairFinderTask: public PndSdsTask {

private:

        //for statistics
        Int_t noOfGoodPairs, noOfEvents;
        Int_t plane0, plane1, plane2, plane3;
        Int_t eventMissedAllPlanes, noOverlap;
        Int_t distanceTooHigh;
        Int_t sumOfPixelHits;
        Int_t unsuitable;
        Int_t hitsClustered, hitsSinglePixel;
        Int_t noOfCombos;

        Double_t _maxDistance;

        Bool_t _ignoreClusters;

        //data for dynamic cut
        Bool_t _findDynamicCutParameters;
        Bool_t _useDynamicCut;
        std::string _cutParameterFile;
        std::map<int, dynamicCutHandler> cutHandlers;

        pixelHit getPixelHitFromSdsHit(PndSdsHit *sdsHit);

        boost::property_tree::ptree config;

public:

        void SetParContainers();

        PndLmdPairFinderTask();
        PndLmdPairFinderTask(const char* name);
        virtual ~PndLmdPairFinderTask();

        virtual void SetBranchNames();

        virtual InitStatus Init();
        virtual InitStatus ReInit();

        virtual void Exec(Option_t* opt);

        virtual void FinishEvent();
        virtual void FinishTask();

        //find the minDistance and maxDistance for dynamic cut
        void findDynamicCutParameters(Bool_t value, std::string parameterFile) {
                _findDynamicCutParameters = value;
                _cutParameterFile = parameterFile;
        }

        //apply a dynamic cut. uses the first N pairs to decide what min and max distance should be.
        void useDynamicCut(Bool_t value, std::string parameterFile) {
                _useDynamicCut = value;
                _cutParameterFile = parameterFile;
        }

        // apply distance cut, will be ignored when using dynamic cut
        void setMaxDistance(Double_t value) {
                _maxDistance = value;
        }

        void ignoreClusters(Bool_t value) {
                _ignoreClusters = value;
        }

protected:

        //PndLmdDim* dimension;

        PndLmdGeometryHelper *helper;

        TClonesArray* digiArray;
        TClonesArray* recoArray;
        TClonesArray* hitPairArray;
        TClonesArray* clusterCandidateArray;

        TString fInRecoBranchName;
        TString fInClusterCandidates;

        std::map<int, TClonesArray*> hitPairMap;
        std::map<int, int> hitCountMap;

        void Register();
        void Reset();

        //function to return result of all checks, distance cut etc.
        void getStatistics(PndLmdHitPair &candidate);

        bool applyDynamicDistanceCut(PndLmdHitPair &candidate);
        bool applyStaticDistanceCut(PndLmdHitPair &candidate);

        //self explanatory
        bool candHitsOverlappingArea(const PndLmdHitPair &candidate);

        //means two clusters can reasonably belong to a single track
        bool candDistanceIsGood(PndLmdHitPair &candidate);

ClassDef(PndLmdPairFinderTask,19)
        ;
};

#endif /* PAIRFINDERTASK_H_ */