LitCudaDetectorGeometry.h

Go to the documentation of this file.

 
#ifndef LITCUDADETECTORGEOMETRY_H_
#define LITCUDADETECTORGEOMETRY_H_
 
//#include "LitCudaTypes.h"
#include "LitCudaField.cu"
#include "LitCudaMaterialInfo.h"
//#include "base/CbmLitEnums.h"
 
#include <iostream>
 
const unsigned char MAX_NOF_STATION_GROUPS = 6;
const unsigned char MAX_NOF_STATIONS       = 4;
const unsigned char MAX_NOF_SUBSTATIONS    = 2;
 
struct LitCudaSubstation {
  float Z;
  LitCudaMaterialInfo material;
  LitCudaFieldSlice fieldSlice;
 
  friend std::ostream& operator<<(std::ostream& strm,
                                  const LitCudaSubstation& substation) {
    strm << "LitCudaSubstation: "
         << "Z=" << substation.Z << ", material=" << substation.material;
    //    strm << ", fieldSlice=" << substation.fieldSlice;
    return strm;
  }
};
 
 
struct LitCudaStation {
  LitCudaStation() : nofSubstations(0) {}
 
  void AddSubstation(const LitCudaSubstation* substation) {
    substations[nofSubstations++] = *substation;
  }
 
  // unsigned char GetNofSubstations() const {
  //    return nofSubstations;
  // }
 
  // Type of hits on the station
  // LitHitType type;
  unsigned char type;
  // array with substations in the station
  LitCudaSubstation substations[MAX_NOF_SUBSTATIONS];
  // number of substations
  unsigned char nofSubstations;
 
  friend std::ostream& operator<<(std::ostream& strm,
                                  const LitCudaStation& station) {
    strm << "LitCudaStation: type" << station.type
         << ", nofSubstations=" << station.nofSubstations << std::endl;
    for (int i = 0; i < station.nofSubstations; i++) {
      strm << "    " << i << station.substations[i];
    }
    return strm;
  }
};
 
 
struct LitCudaAbsorber {
  float Z;
  LitCudaMaterialInfo material;
  LitCudaFieldSlice fieldSliceFront;
  LitCudaFieldSlice fieldSliceBack;
 
  friend std::ostream& operator<<(std::ostream& strm,
                                  const LitCudaAbsorber& absorber) {
    strm << "LitCudaAbsorber: Z" << absorber.Z
         << ", material=" << absorber.material;
    //    strm << "fieldSliceFront=" << absorber.fieldSliceFront
    //       << " fieladSliceBack=" << absorber.fieldSliceBack;
    return strm;
  }
};
 
 
struct LitCudaStationGroup {
  LitCudaStationGroup() : nofStations(0) {}
 
  // virtual ~LitCudaStationGroup() {}
 
  void AddStation(const LitCudaStation* station) {
    stations[nofStations++] = *station;
  }
 
  // unsigned char GetNofStations() const {
  //    return nofStations;
  // }
 
  // array with stations in the station group
  LitCudaStation stations[MAX_NOF_STATIONS];
  // number of stations in the station group
  unsigned char nofStations;
  // absorber
  LitCudaAbsorber absorber;
 
  friend std::ostream& operator<<(std::ostream& strm,
                                  const LitCudaStationGroup& stationGroup) {
    strm << "LitCudaStationGroup: "
         << "nofStations=" << stationGroup.nofStations << std::endl;
    for (unsigned char i = 0; i < stationGroup.nofStations; i++) {
      strm << "  " << i << stationGroup.stations[i];
    }
    strm << "  " << stationGroup.absorber;
    return strm;
  }
};
 
 
struct LitCudaDetectorLayout {
  LitCudaDetectorLayout() : nofStationGroups(0) {};
 
  void AddStationGroup(const LitCudaStationGroup& stationGroup) {
    stationGroups[nofStationGroups++] = stationGroup;
  }
 
  // unsigned char GetNofStationGroups() const {
  //    return nofStationGroups;
  // }
 
  // unsigned char GetNofStations(unsigned char stationGroup) const {
  //    return stationGroups[stationGroup].GetNofStations();
  // }
  //
  // unsigned char GetNofSubstations(unsigned char stationGroup, unsigned char station) const {
  //    return stationGroups[stationGroup].stations[station].GetNofSubstations();
  // }
  //
  // const LitCudaStationGroup& GetStationGroup(unsigned char stationGroup) {
  //    return stationGroups[stationGroup];
  // }
  //
  // const LitCudaStation& GetStation(unsigned char stationGroup, unsigned char station) {
  //    return stationGroups[stationGroup].stations[station];
  // }
  //
  // const LitCudaSubstation& GetSubstation(unsigned char stationGroup, unsigned char station, unsigned char substation){
  //    return stationGroups[stationGroup].stations[station].substations[substation];
  // }
 
  // array with station groups
  LitCudaStationGroup stationGroups[MAX_NOF_STATION_GROUPS];
  //number of station groups
  unsigned char nofStationGroups;
 
  friend std::ostream& operator<<(std::ostream& strm,
                                  const LitCudaDetectorLayout& layout) {
    strm << "LitCudaDetectorLayout: "
         << "nofStationGroups=" << layout.nofStationGroups << std::endl;
    for (unsigned char i = 0; i < layout.nofStationGroups; i++) {
      strm << i << layout.stationGroups[i];
    }
    return strm;
  }
};
 
#endif