LitStationGroupMuon.h

Go to the documentation of this file.

/* Copyright (C) 2009-2012 GSI/JINR-LIT, Darmstadt/Dubna
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Andrey Lebedev [committer] */
 
#ifndef LITSTATIONGROUP_H_
#define LITSTATIONGROUP_H_
 
#include "LitAbsorber.h"
#include "LitFieldGrid.h"
#include "LitFieldRegion.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_ */