BmnGemStripLayer.h

Go to the documentation of this file.

#ifndef BMNGEMSTRIPLAYER_H
#define BMNGEMSTRIPLAYER_H
 
#include "BmnMatch.h"
#include "BmnStripData.h"
#include "TMath.h"
 
#include <iostream>
#include <vector>
 
using namespace TMath;
using namespace std;
 
class BmnGemStripLayer
{
 
  public:
    // Constructors -------------------------------------------------------------
    BmnGemStripLayer();
 
    BmnGemStripLayer(Int_t zone_number,
                     StripLayerType layer_type,
                     Double_t xsize,
                     Double_t ysize,
                     Double_t xorig,
                     Double_t yorig,
                     Double_t pitch,
                     Double_t adeg);
    //--------------------------------------------------------------------------
 
    // Destructor
    virtual ~BmnGemStripLayer();
 
    void InitializeLayer();
    void ResetLayer();   // clear all data and rebuild the layer in accordance with new parameters
 
    Int_t GetNStrips();
 
    void SetVerbosity(Bool_t verb) { Verbosity = verb; }
    void SetPitch(Double_t pitch);
    void SetLayerSizes(Double_t xsize, Double_t ysize, Double_t xorig = 0.0, Double_t yorig = 0.0);
    void SetAngleDeg(Double_t deg);   // plus - clockwise from vertical
 
    // never called
    int GetFirstStripNumber() { return 0; }
    int GetLastStripNumber() { return 0; }
 
    Bool_t GetVerbosity() { return Verbosity; }
    Int_t GetZoneNumber() { return LayerZoneNumber; }
    StripLayerType GetType() { return LayerType; }
    Double_t GetPitch() { return Pitch; }
    Double_t GetAngleDeg() { return AngleDeg; }
    Double_t GetAngleRad() { return AngleRad; }
    Double_t GetXSize() { return XMaxLayer - XMinLayer; }
    Double_t GetYSize() { return YMaxLayer - YMinLayer; }
    Double_t GetXMinLayer() { return XMinLayer; }
    Double_t GetXMaxLayer() { return XMaxLayer; }
    Double_t GetYMinLayer() { return YMinLayer; }
    Double_t GetYMaxLayer() { return YMaxLayer; }
 
    // Strip numbering methods --------------------------------------------------
    Bool_t SetStripNumberingBorders(Double_t x_left,
                                    Double_t y_left,
                                    Double_t x_right,
                                    Double_t y_right);   // or method below
    Bool_t SetStripNumberingBorders(StripBorderPoint left, StripBorderPoint right);
    Bool_t SetStripNumberingOrder(StripNumberingDirection strip_direction);
 
    StripNumberingDirection GetStripNumberingOrder() { return StripOrder; }
    Double_t GetXLeftStripBorderPoint() { return XLeftPointOfStripNumbering; }
    Double_t GetYLeftStripBorderPoint() { return YLeftPointOfStripNumbering; }
    Double_t GetXRightStripBorderPoint() { return XRightPointOfStripNumbering; }
    Double_t GetYRightStripBorderPoint() { return YRightPointOfStripNumbering; }
    //--------------------------------------------------------------------------
 
    // Dead zones methods -------------------------------------------------------
    Bool_t AddDeadZone(Int_t n_points, Double_t* x_points, Double_t* y_points);
    Bool_t AddDeadZone(DeadZoneOfStripLayer dead_zone);
    Bool_t IsPointInsideDeadZones(Double_t x, Double_t y);
    Bool_t IsPointInsideStripLayer(Double_t x, Double_t y);
 
    vector<DeadZoneOfStripLayer> GetDeadZones() { return DeadZones; }
    void ResetAllDeadZones() { DeadZones.clear(); }
    //--------------------------------------------------------------------------
 
    // Direct strip signal manipulation -----------------------------------------
    Bool_t SetStripSignal(Int_t strip_num, Double_t signal);   // replace signal
    Bool_t AddStripSignal(Int_t strip_num, Double_t signal);   // add signal
 
    Double_t GetStripSignal(Int_t strip_num);   // signal of strip_num-strip
    //--------------------------------------------------------------------------
 
    // Strip matches controls ---------------------------------------------------
    Bool_t SetStripMatch(Int_t strip_num, BmnMatch mc_match);                     // set a MC-match for a strip
    Bool_t AddLinkToStripMatch(Int_t strip_num, Double_t weight, Int_t mc_num);   // add a mc-number to a MC-match
 
    Bool_t SetStripDigitNumberMatch(Int_t strip_num,
                                    BmnMatch digit_num_match);   // set a digit number match for a strip
    Bool_t AddLinkToStripDigitNumberMatch(Int_t strip_num,
                                          Double_t weight,
                                          Int_t digit_num);   // add a digit number match for a strip
 
    BmnMatch GetStripMatch(Int_t strip_num);              // mc-match for a strip
    BmnMatch GetStripDigitNumberMatch(Int_t strip_num);   // digit number match for a strip
 
    void ResetStripMatches();
    void ResetStripDigitNumberMatches();
    //--------------------------------------------------------------------------
 
    // Strip hits ---------------------------------------------------------------
    Int_t GetNStripHits() { return StripHits.size(); }   // number of hits at the layer
    Double_t GetStripHitPos(Int_t num);                  // hit position
    Double_t GetStripHitTotalSignal(Int_t num);          // sum signal of the hit
    Double_t GetStripHitError(Int_t num);                // hit error
    Int_t GetStripHitClusterSize(Int_t num);             // number of strips in the cluster for the num-hit
    void ResetStripHits();
    //--------------------------------------------------------------------------
 
    // Strip Clusters -----------------------------------------------------------
    vector<StripCluster> GetStripClusters() { return StripClusters; };
    static void SetLowerUniqueID(Int_t id) { fUniqueIdL = id; }
    static void SetUpperUniqueID(Int_t id) { fUniqueIdU = id; }
    //--------------------------------------------------------------------------
 
    // Methods to convert point coordinates into strip system -------------------
    Double_t ConvertNormalPointToStripX(Double_t x, Double_t y);
    Double_t ConvertNormalPointToStripY(Double_t x, Double_t y);
 
    Double_t ConvertPointToStripPosition(Double_t x, Double_t y);
 
    Double_t CalculateStripEquationB(Double_t strip_pos);
    //--------------------------------------------------------------------------
 
    // Clustering and strip-hit finding methods ---------------------------------------
    void FindClustersAndStripHits();
    void MakeStripHit(StripCluster& cluster, vector<Double_t>& AnalyzableStrips, Int_t& curcnt);
    void SmoothStripSignal(vector<Double_t>& AnalyzableStrips, Int_t NIterations, Int_t SmoothWindow, Double_t Weight);
 
    void SetClusterFindingThreshold(Double_t threshold) { ClusterFindingThreshold = threshold; }
    Double_t GetClusterFindingThreshold() { return ClusterFindingThreshold; }
 
    void SetTotalClusterThreshold(Double_t th) { TotalClusterThreshold = th; }
 
    vector<Double_t> GetSmoothStrips() { return SmoothStrips; }   // for test
    //--------------------------------------------------------------------------
 
  private:
    // BmnGemStripLayer(const BmnGemStripLayer&) = delete;
    // BmnGemStripLayer& operator=(const BmnGemStripLayer&) = delete;
 
  private:
    Bool_t Verbosity;
 
    Int_t LayerZoneNumber;      // zone number of the strip layer
    StripLayerType LayerType;   // type of the strip layer: lower or upper
 
    Double_t Pitch;
    Double_t AngleDeg;   // Angle between two sets of parellel strips: angle from vertical
    Double_t AngleRad;
 
    Double_t XMinLayer;
    Double_t XMaxLayer;
    Double_t YMinLayer;
    Double_t YMaxLayer;
 
    vector<DeadZoneOfStripLayer> DeadZones;
 
    // Strip arrangement
    StripNumberingDirection StripOrder;   // strip numbering order (LeftToRight or RightToLeft)
    Double_t XLeftPointOfStripNumbering;
    Double_t XRightPointOfStripNumbering;
    Double_t YLeftPointOfStripNumbering;
    Double_t YRightPointOfStripNumbering;
 
    // Strip layer implementation
    vector<Double_t> Strips;
 
    // Strip matches
    vector<BmnMatch> StripMatches;              // MC-point matches for all strips
    vector<BmnMatch> StripDigitNumberMatches;   // Digit number matches for all strips
 
    // Strip hits (It is filled after cluster finding and fitting) --------------
    vector<Double_t> StripHits;   // positions of strip hits
    vector<Double_t> StripHitsTotalSignal;
    vector<Double_t> StripHitsErrors;
    vector<Int_t> StripHitsClusterSize;   // cluster size (number of strips) for each strip hit
    vector<Double_t> SmoothStrips;        // smoothed signal strips (It is filled after smoothing algorithm)
    Double_t ClusterFindingThreshold;     // min. threshold cut for strip to be included in cluster
    Double_t TotalClusterThreshold;       // min. threshold cut for total cluster signal to be written into TClonesArray
    //--------------------------------------------------------------------------
 
    // Strip Clusters -----------------------------------------------------------
    vector<StripCluster> StripClusters;
 
    static Int_t fUniqueIdU;
    static Int_t fUniqueIdL;
    //--------------------------------------------------------------------------
};
//------------------------------------------------------------------------------
 
#endif /* BMNGEMSTRIPLAYER_H */