MuonSegmentCombination.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2019 CERN for the benefit of the ATLAS collaboration
*/

// MdtSegment.h, (c) ATLAS Detector software
 
#ifndef MUONSEGMENT_MUONSEGMENTCOMBINATION_H
#define MUONSEGMENT_MUONSEGMENTCOMBINATION_H
 
namespace Trk {
  class Segment;
}
 
#include <vector>
#include <memory>
#include "MuonSegment/MuonSegment.h"
 
namespace Muon {

  class MuonSegmentCombination {
  public:
    typedef std::vector< std::unique_ptr<MuonSegment> > SegmentVec;
    typedef std::vector< std::unique_ptr<SegmentVec> >  SegmentVecVec;
  public:
    MuonSegmentCombination();
    
    ~MuonSegmentCombination();

    MuonSegmentCombination( const MuonSegmentCombination& );

    MuonSegmentCombination& operator=( const MuonSegmentCombination& );

    bool addSegments( std::unique_ptr<SegmentVec> );

    unsigned int numberOfStations() const;

    SegmentVec* stationSegments( unsigned int index ) const;

    unsigned int numberOfAmbiguities() const;
 
    void setNGoodCscLayers(int nEta, int nPhi){m_nGood[0]=nPhi; m_nGood[1]=nEta;}
 
    int getNGoodCscLayers(int isEta) const; //isEta=0 means phi
 
    bool use2LayerSegments(int isEta) const; //isEta=0 means phi
 
    bool useStripsInSegment(int isEta) const; //isEta=0 means phi
 
  private:
    void clear();

    void copy( const MuonSegmentCombination& segc );
 
    SegmentVecVec         m_segmentsPerStation;
 
    //if the station is a CSC station, this tells us how many good eta and phi layers it has
    int m_nGood[2]{};
  };
 
  //if there are only 2 good layers for eta or phi those segments may be 2-layer segments
  inline bool MuonSegmentCombination::use2LayerSegments(int isEta) const
    {
      return m_nGood[isEta]==2;
    }
 
  //if there are fewer than 2 good layers for eta or phi we don't try to use eta or phi to build segments
  inline bool MuonSegmentCombination::useStripsInSegment(int isEta) const
    {
      return m_nGood[isEta]>1;
    }
 
  inline int MuonSegmentCombination::getNGoodCscLayers(int isEta) const
    {
      return m_nGood[isEta];
    }
 
  inline  bool MuonSegmentCombination::addSegments( std::unique_ptr<SegmentVec> segs )
  {
    m_segmentsPerStation.push_back( std::move(segs) );
    return true;
  }
 
  inline unsigned int MuonSegmentCombination::numberOfStations() const 
  {
    return m_segmentsPerStation.size();
  }
  
  inline MuonSegmentCombination::SegmentVec* 
    MuonSegmentCombination::stationSegments( unsigned int index ) const
  {
    if( index < numberOfStations() ) return m_segmentsPerStation[index].get();
    return 0;
  }
 
  inline unsigned int MuonSegmentCombination::numberOfAmbiguities() const
  {
    unsigned int solutions(1);
    SegmentVecVec::const_iterator it = m_segmentsPerStation.begin();
    SegmentVecVec::const_iterator it_end = m_segmentsPerStation.end();
    for( ; it!=it_end; ++it ) solutions *= (*it)->size();
    return solutions;
  }
 
}
 
#endif