CbmStsStation.cxx

Go to the documentation of this file.

// -------------------------------------------------------------------------
// -----                     CbmStsStation source file                 -----
// -----                  Created 31/03/05  by V. Friese               -----
// -------------------------------------------------------------------------
 
#include "CbmStsStation.h"
 
#include "CbmStsSector.h"
#include "CbmStsSectorDigiPar.h"
 
#include "CbmStsSensor.h"
 
#include "TMath.h"
 
#include <iostream>
#include <map>
 
using std::cout;
using std::endl;
using std::map;
 
 
// -----   Default constructor   -------------------------------------------
CbmStsStation::CbmStsStation()
: fName(),
  fDetectorId(0),
  fZ(0.),
  fD(0.),
  fRadLength(),
  fRmin(0.),
  fRmax(0.),
  fRotation(0.),
  fSectors(),
  fSectorMap(),
  fMcIdMap()
{
  fSectors = new TObjArray(100);
  fSectorMap.clear();
}
 
// -----   Standard constructor   ------------------------------------------
CbmStsStation::CbmStsStation(const char* name, Int_t iStation, Double_t z, 
           Double_t d, Double_t rl, Double_t rmin, 
           Double_t rmax, Double_t rotation) 
: TNamed(name, "STS station"),
  fName(),
  fDetectorId(0),
  fZ(0.),
  fD(0.),
  fRadLength(),
  fRmin(0.),
  fRmax(0.),
  fRotation(0.),
  fSectors(),
  fSectorMap(),
  fMcIdMap()
{
  fDetectorId = 2 << 24 | iStation << 16;
  fZ          = z;
  fD          = d;
  fRadLength  = rl;
  fRmin       = rmin;
  fRmax       = rmax;
  fRotation   = rotation;
  fSectors = new TObjArray(100);
  fSectorMap.clear();
 
  for ( Int_t isz = 0 ; isz < 10 ; isz++ )
    fSensorZ[isz] = -666.;
  fSensorZ[0] = z;
}
 
// -----   Destructor   ----------------------------------------------------
CbmStsStation::~CbmStsStation() {
  if ( fSectors ) {
    fSectors->Delete();
    delete fSectors;
  }
}
// -------------------------------------------------------------------------
 
 
// -----   Public method GetNChannels   ------------------------------------
Int_t CbmStsStation::GetNChannels() {
  Int_t nChan = 0;
  for (Int_t iSect=0; iSect<GetNSectors(); iSect++) 
    if ( GetSector(iSect)->GetNChannels() > 0 )
      nChan += GetSector(iSect)->GetNChannels();
  return nChan;
}
// -------------------------------------------------------------------------
 
 
 
 
// -----   Public method GetSectorByNr   -----------------------------------
CbmStsSector* CbmStsStation::GetSectorByNr(Int_t sectorNr) {
  if ( fSectorMap.find(sectorNr) != fSectorMap.end() ) {
    Int_t index = fSectorMap[sectorNr];
    return (CbmStsSector*) fSectors->At(index);
  }
  else {
    cout << "-W- CbmStsStation::GetSectorByNr: sector " << sectorNr
   << " not found (station " << GetStationNr() << ")." << endl;
    return NULL;
  }
}
// -------------------------------------------------------------------------
 
 
 
// -----   Public method AddSector   ---------------------------------------
void CbmStsStation::AddSector(CbmStsSectorDigiPar* sectorPar) {
 
  // Get digitisation parameters
  Int_t iSector   = sectorPar->GetSectorNr();
 
  // Check for previous existence of sector
  if ( fSectorMap.find(iSector) != fSectorMap.end() ) {
    cout << "-W- " << fName << "::AddSector: Sector number " << iSector
   << " already existing for station " << GetStationNr() << endl;
    return;
  }
 
  // Calculate detectorId and rotation in global c.s.
  Int_t    detId  = 2 << 24 | (GetStationNr() << 16) | (iSector << 4); 
 
  // Number of existing sectors
  Int_t nSectors = fSectors->GetEntries();
 
  // Create and add new sector
  fSectors->Add(new CbmStsSector(Form("%ssect%d",fName.Data(),iSector),detId));
 
  fSectorMap[iSector] = nSectors;
}
// -------------------------------------------------------------------------
 
// -----   Public method AddSector   ---------------------------------------
void CbmStsStation::AddSector(CbmStsSector* sector) {
 
  Int_t iSector  = sector->GetSectorNr();
  Int_t nSectors = fSectors->GetEntries();
 
  fSectors->Add(sector);
 
  fSectorMap[iSector] = nSectors;
 
}
// -------------------------------------------------------------------------
 
// -------------------------------------------------------------------------
Double_t CbmStsStation::GetZ(Int_t it)
{
  if ( fSensorZ[it] < -665. ) {
    Int_t knownZPos = 0;
    for ( knownZPos = 0 ; knownZPos < 10 ; knownZPos++ ) {
      if ( fSensorZ[knownZPos] < -665. ) break;
    }
 
    for (Int_t iSect=GetNSectors(); iSect > 0 ; iSect--) {
      CbmStsSector* sector = (CbmStsSector*)GetSector(iSect-1);
      for (Int_t iSens=sector->GetNSensors() ; iSens > 0 ; iSens--) {
  CbmStsSensor* sensor = (CbmStsSensor*)sector->GetSensor(iSens-1);
  Bool_t knownAlready = kFALSE;
  for ( Int_t isz = 0 ; isz < knownZPos ; isz++ )
    if ( TMath::Abs(sensor->GetZ0()-fSensorZ[isz]) < 0.0001 ) {
      knownAlready = kTRUE;
      break;
    }
 
  if ( knownAlready ) continue;
 
  fSensorZ[knownZPos] = sensor->GetZ0();
  knownZPos++;
      }
    }
  }
  return fSensorZ[it];
}
// -------------------------------------------------------------------------
 
// -------------------------------------------------------------------------
Int_t CbmStsStation::GetNofZ()
{
  Int_t knownZPos = 0;
  for ( knownZPos = 0 ; knownZPos < 10 ; knownZPos++ ) {
    if ( fSensorZ[knownZPos] < -665. ) break;
  }
 
  for (Int_t iSect=GetNSectors(); iSect > 0 ; iSect--) {
    CbmStsSector* sector = (CbmStsSector*)GetSector(iSect-1);
    for (Int_t iSens=sector->GetNSensors() ; iSens > 0 ; iSens--) {
      CbmStsSensor* sensor = (CbmStsSensor*)sector->GetSensor(iSens-1);
      Bool_t knownAlready = kFALSE;
      for ( Int_t isz = 0 ; isz < knownZPos ; isz++ )
  if ( TMath::Abs(sensor->GetZ0()-fSensorZ[isz]) < 0.0001 ) {
    knownAlready = kTRUE;
    break;
  }
      
      if ( knownAlready ) continue;
      
      fSensorZ[knownZPos] = sensor->GetZ0();
      knownZPos++;
    }
  }
  return knownZPos;
}
// -------------------------------------------------------------------------
 
 
// -----   Public method Reset   -------------------------------------------
void CbmStsStation::Reset() {
  for (Int_t iSector=0; iSector<GetNSectors(); iSector++) 
    GetSector(iSector)->Reset();
}
// -------------------------------------------------------------------------
 
 
 
// -----   Public method Print   -------------------------------------------
void CbmStsStation::Print(Bool_t kLong) {
  cout << "Station Nr. ";
  cout.width(2);
  cout << GetStationNr() << ", z =  ";
  cout.width(3);
  cout << fZ << " cm, Rmin = ";
  cout.width(5);
  cout << fRmin << " cm, Rmax = " << fRmax << " cm, angle = ";
  cout.width(3);
  cout << fRotation * 180 / TMath::Pi() << " deg., sectors: ";
  cout.width(4);
  cout << GetNSectors() << ", channels: ";
  cout.width(8);
  cout << GetNChannels() << endl;
  if ( kLong ) 
    for (Int_t iSec=0; iSec<GetNSectors(); iSec++) 
      GetSector(iSec)->Print();
}
// -------------------------------------------------------------------------