LitStationGroupMuon.h

Go to the documentation of this file.

 
#ifndef LITSTATIONGROUP_H_
#define LITSTATIONGROUP_H_
 
#include "../LitFieldGrid.h"
#include "../LitFieldRegion.h"
#include "LitAbsorber.h"
#include "LitStationMuon.h"
 
namespace lit {
  namespace parallel {
 
    template<class T>
    class LitStationGroupMuon {
    public:
      LitStationGroupMuon()
        : fFieldGridFront()
        , fFieldGridMiddle()
        , fFieldGridBack()
        , fStations()
        , fAbsorber() {}
 
      virtual ~LitStationGroupMuon() {}
 
      const LitFieldGrid& GetFieldGridFront() const { return fFieldGridFront; }
 
      void SetFieldGridFront(const LitFieldGrid& fieldGrid) {
        fFieldGridFront = fieldGrid;
      }
 
      const LitFieldGrid& GetFieldGridMiddle() const {
        return fFieldGridMiddle;
      }
 
      void SetFieldGridMiddle(const LitFieldGrid& fieldGrid) {
        fFieldGridMiddle = fieldGrid;
      }
 
      const LitFieldGrid& GetFieldGridBack() const { return fFieldGridBack; }
 
      void SetFieldGridBack(const LitFieldGrid& fieldGrid) {
        fFieldGridBack = fieldGrid;
      }
 
      void AddStation(const LitStationMuon<T>& station) {
        fStations.push_back(station);
      }
 
      const LitStationMuon<T>& GetStation(unsigned short index) const {
        return fStations[index];
      }
 
      unsigned char GetNofStations() const { return fStations.size(); }
 
      const LitAbsorber<T>& GetAbsorber() const { return fAbsorber; }
 
      void SetAbsorber(const LitAbsorber<T>& absorber) { fAbsorber = absorber; }
 
      void GetFieldRegion(T x, T y, LitFieldRegion<T>& field) const {
        LitFieldValue<T> v1, v2, v3;
        fFieldGridFront.GetFieldValue(x, y, v1);
        fFieldGridMiddle.GetFieldValue(x, y, v2);
        fFieldGridBack.GetFieldValue(x, y, v3);
        field.Set(v1,
                  fFieldGridFront.GetZ(),
                  v2,
                  fFieldGridMiddle.GetZ(),
                  v3,
                  fFieldGridBack.GetZ());
      }
 
      std::string ToString() const {
        std::string str = "LitStationGroupMuon: nofStations="
                          + lit::parallel::ToString<int>((int) GetNofStations())
                          + "\n";
        for (unsigned char i = 0; i < GetNofStations(); i++) {
          str += "  " + lit::parallel::ToString<int>((int) i) + " "
                 + GetStation(i).ToString();
        }
        str += "  " + GetAbsorber().ToString();
        return str;
      }
 
      friend std::ostream& operator<<(std::ostream& strm,
                                      const LitStationGroupMuon& stationGroup) {
        strm << stationGroup.ToString();
        return strm;
      }
 
    private:
      // This field slices are used to construct LitFieldRegion
      // in the group of stations. We need 3 values in order to use
      // parabolic approximation. So the values are in front, middle
      // and back of the gap.
      LitFieldGrid fFieldGridFront;  // Magnetic field approximation in front
      LitFieldGrid
        fFieldGridMiddle;  // Magnetic field approximation in the middle
      LitFieldGrid fFieldGridBack;  // Magnetic field approximation in the back
 
      std::vector<LitStationMuon<T>> fStations;  // Array with stations
      LitAbsorber<T> fAbsorber;                  // Absorber
    } _fvecalignment;
 
    /* Some typedefs for convenience */
    typedef LitStationGroupMuon<fvec> LitStationGroupMuonVec;
    typedef LitStationGroupMuon<fscal> LitStationGroupMuonScal;
 
  }  // namespace parallel
}  // namespace lit
 
#endif /* LITSTATIONGROUP_H_ */