LCDG4LumSD.cc

Go to the documentation of this file.

// $Id: LCDG4LumSD.cc,v 1.11 2004/08/17 23:24:03 uid561 Exp $

#include "LCDG4LumSD.hh"

#include "G4Step.hh"
#include "G4HCofThisEvent.hh"
#include "G4TouchableHistory.hh"
#include "G4ios.hh"
#include "G4Track.hh"

/*
 * Constructor
 */
LCDG4LumSD::LCDG4LumSD(G4String name):G4VSensitiveDetector(name) {
  G4String HCname;
  collectionName.insert(HCname="lumCollection");
  m_cellhitb.clear();
  m_cellhite.clear();
  m_phiseg  [0]=1; m_phiseg  [1]=1;
  m_thetaseg[0]=1; m_thetaseg[1]=1;
}

/*
 * Destructor
 */
LCDG4LumSD::~LCDG4LumSD(){
}


/*
 * Intialize the hit collection for this event
 */
void LCDG4LumSD::Initialize(G4HCofThisEvent*HCE) {
  static int HCID = -1;
  m_lumCollection = new LCDG4CalHitsCollection
    (SensitiveDetectorName,collectionName[0]);
  if(HCID < 0) {
    HCID = GetCollectionID(0);
  }
  HCE->AddHitsCollection(HCID,m_lumCollection);
}


/*
 * Process step information into G4 Hit
 */
G4bool LCDG4LumSD::ProcessHits(G4Step* aStep, G4TouchableHistory* ROhist) {

  G4double edep = aStep->GetTotalEnergyDeposit();

  if(edep==0.) return false;

  const G4VPhysicalVolume* physVol
    = aStep->GetPreStepPoint()->GetPhysicalVolume();
  if( ROhist!=NULL ) {
    G4VPhysicalVolume* phyVol = ROhist->GetVolume();
    if(phyVol != physVol) G4cout<<"phy,phys="<< phyVol<<' '<<physVol <<G4endl;
  }

  G4LogicalVolume* logVol=physVol->GetLogicalVolume();

  G4Track* mclum = aStep->GetTrack();
  G4int    trkID  = mclum->GetTrackID();
  G4double tdep   = mclum->GetGlobalTime();
  G4int    layno  = m_maplayer[logVol];
  G4int    sysNo  = m_mapsys[logVol];
  G4int    barend = m_mapbe[logVol];

  G4ThreeVector pos = 0.5*(aStep->GetPreStepPoint()->GetPosition() +
                           aStep->GetPostStepPoint()->GetPosition() );
  G4double dphi     = 360.0*degree/m_phiseg[barend];
  G4double dtheta   = 180.0*degree/m_thetaseg[barend];
  G4double trkphi   = pos.phi();
  if (trkphi < 0.0) {
    trkphi += 360.0*degree;
  }
  G4double trkthe = pos.theta();
  G4int    sphi   = int(trkphi/dphi);
  G4int    sthe   = int(trkthe/dtheta);
  G4int    icell  = sthe + m_thetaseg[barend]*sphi
    + m_thetaseg[barend]*m_phiseg[barend]*layno;

  std::map<G4int,LCDG4CalHit*>* pcellhit=0;
  if (barend == 0) {
    pcellhit=&m_cellhitb;
  } else {
    pcellhit=&m_cellhite;
  }
  LCDG4CalHit *hit=(*pcellhit)[icell];

  if(hit == 0) {
    G4double r = pos.mag();
    G4ThreeVector hpos(r*sin((sthe+0.5)*dtheta)*cos((sphi+0.5)*dphi),
                       r*sin((sthe+0.5)*dtheta)*sin((sphi+0.5)*dphi),
                       r*cos((sthe+0.5)*dtheta));

    (*pcellhit).erase(icell);
    hit = new LCDG4CalHit();
    hit->SetLayer(layno);
    hit->SetSystem(sysNo);
    hit->SetBarEnd(barend);
    hit->SetPhi(sphi);
    hit->SetTheta(sthe);
    hit->SetPos(hpos);
    m_lumCollection->insert(hit);
    pcellhit->insert(std::pair<G4int,LCDG4CalHit*>(icell,hit));
  }

  if (m_mapsens[logVol]) {
    hit->AddMcPart(trkID,edep,tdep);
  } else {
    hit->AddMcPartAbs(edep);
  }
  return true;
}

/*
 * EndOfEvent method clear cell hit maps for barrel and endcaps
 */
void LCDG4LumSD::EndOfEvent(G4HCofThisEvent*HCE) {
  if(HCE) ;  // make compiler happy about unused vars
  m_cellhitb.clear();
  m_cellhite.clear();
}

/*
 * Clear method, not used
 */
void LCDG4LumSD::clear() {
}

/*
 * Draw method, not used
 */
void LCDG4LumSD::DrawAll() {
}

/*
 * Print method, not used
 */
void LCDG4LumSD::PrintAll() {
}

/*
 * Add layer number, sysNo, barrel/endcap, N/S, and sens info to logical volume maps
 */
G4int LCDG4LumSD::AddLogVolInfo
(G4LogicalVolume* g4lumlog,G4int startlayerNo,G4int sysNo,G4int be, G4int ns,
 G4int sens) {
  m_maplayer.insert(std::pair<G4LogicalVolume*,G4int>(g4lumlog,startlayerNo));
  m_mapsys.insert(std::pair<G4LogicalVolume*,G4int>(g4lumlog,sysNo));
  m_mapbe.insert(std::pair<G4LogicalVolume*,G4int>(g4lumlog,be));
  m_mapns.insert(std::pair<G4LogicalVolume*,G4int>(g4lumlog,ns));
  m_mapsens.insert(std::pair<G4LogicalVolume*,G4int>(g4lumlog,sens));
  return m_maplayer.size();
}

---