LCDG4CalHit.cc

Go to the documentation of this file.

#include "LCDG4CalHit.hh"
#include "G4VVisManager.hh"
#include "G4Circle.hh"
#include "G4Colour.hh"
#include "G4VisAttributes.hh"
#include "LCDtowerID.h"

G4Allocator<LCDG4CalHit> LCDG4CalHitAllocator;

/*
 * Default constructor
 */
LCDG4CalHit::LCDG4CalHit()
  : G4VHit(), m_layer(-1), m_sysNo(-1), m_barend(-1), m_phi(-1), m_theta(-1)
{
  m_nmc=0;
  m_edep=0;
  m_edepabs=0;

  m_trkid=new std::vector<G4int>;
  m_trkid->clear();
  m_emc =new std::vector<G4double>;
  m_emc->clear();
  m_tmc =new std::vector<G4double>;
  m_tmc->clear();
}

/*
 * Destructor
 */
LCDG4CalHit::~LCDG4CalHit(){
  if (m_trkid) {
    m_trkid->clear();
    delete m_trkid;
    m_trkid=0;
  }
  if (m_emc) {
    m_emc->clear();
    delete m_emc;
  }
  if (m_tmc) {
    m_tmc->clear();
    delete m_tmc;
  }
}

/*
 * Copy constructor
 */
LCDG4CalHit::LCDG4CalHit(const LCDG4CalHit &right) : G4VHit() {
  m_layer  = right.m_layer;
  m_sysNo  = right.m_sysNo;
  m_barend = right.m_barend;
  m_edep   = right.m_edep;
  m_edepabs= right.m_edepabs;
  m_theta  = right.m_theta;
  m_phi    = right.m_phi;
  m_pos    = right.m_pos;

  m_trkid=new std::vector<G4int>;
  m_emc =new std::vector<G4double>;
  m_trkid->clear();
  m_emc->clear();
  int imc=0;
  m_nmc=right.m_nmc;
  for(imc=0 ; imc < right.m_nmc ; imc++) {
    m_trkid->push_back((*right.m_trkid)[imc]);
    m_emc->push_back((*right.m_emc)[imc]);
  }
}

/*
 * Assignment operator
 */
const LCDG4CalHit& LCDG4CalHit::operator=(const LCDG4CalHit &right) {
  m_layer  = right.m_layer;
  m_sysNo  = right.m_sysNo;
  m_barend = right.m_barend;
  m_edep   = right.m_edep;
  m_edepabs= right.m_edepabs;
  m_theta  = right.m_theta;
  m_phi    = right.m_phi;
  m_pos    = right.m_pos;

  int imc=0;
  m_trkid->clear();
  m_emc->clear();
  m_nmc=right.m_nmc;
  for(imc=0 ; imc < right.m_nmc ; imc++) {
    m_trkid->push_back((*right.m_trkid)[imc]);
    m_emc->push_back((*right.m_emc)[imc]);
  }
  return *this;
}

/*
 * Add a contributing MC particle to the raw G4 hit
 * takes trackID and energy as arguments
 */
G4int LCDG4CalHit::AddMcPart(G4int mcidx, G4double emc, G4double tmc) {
  int flg_same=0;
  int imc=0;
  for (imc=0 ; imc < m_nmc ; imc++) {
    if ((*m_trkid)[imc] == mcidx) {
      flg_same=imc+1;
      break;
    }
  }

  m_edep+=emc;
  if (flg_same) {
    (*m_emc)[flg_same-1] += emc;
  } else {
    m_nmc++;
    m_trkid->push_back(mcidx);
    m_emc->push_back(emc);
    m_tmc->push_back(tmc);
  }

  return m_nmc;
}

/*
 * Equality operator
 */
bool LCDG4CalHit::operator==(const LCDG4CalHit &right) const {
  if(m_layer!=right.m_layer) return false;
  if(m_sysNo!=right.m_sysNo) return false;
  if(m_barend!=m_barend) return false;
  if(m_phi!=m_phi) return false;
  if(m_theta!=m_theta) return false;
//    if(m_edep!=m_edep) return false;
//    if(m_edepabs!=m_edepabs) return false;

//    G4int    m_nmc;
//    std::vector<G4int>    *m_trkid;
//    std::vector<G4double> *m_emc;
  return true;
}

unsigned int LCDG4CalHit::GetCellID() const {
  LCDtowerID towerID(0);
  towerID.SetTheta(m_theta);
  towerID.SetPhi(m_phi);
  towerID.SetLayer(m_layer);
  towerID.SetSystem(m_sysNo);
  unsigned int sioTag = towerID.GetTag();
  // convert to LCIO format
  return sioTagToLcioTag( sioTag );
}

unsigned int LCDG4CalHit::sioTagToLcioTag(unsigned int sioTag) const {
  // Left-most four bits: ABCD (sio) -> DABC (lcio)
  unsigned lcioTag = sioTag & 0x0fffffff;
  lcioTag |= (sioTag & 0xe0000000) >> 1;
  lcioTag |= (sioTag & 0x10000000) << 3;
  return lcioTag;
}

/*
 * Draw method
 */
void LCDG4CalHit::Draw() {
  G4VVisManager* pVVisManager = G4VVisManager::GetConcreteInstance();
  if(pVVisManager) {
    G4Circle circle(m_pos);
    circle.SetScreenSize(1.0);
    //circle.SetScreenSize(0.04);
    circle.SetFillStyle(G4Circle::filled);
    G4Colour colour(1.0,0.0,1.0); // Magenta
    G4VisAttributes attribs(colour);
    circle.SetVisAttributes(attribs);
    pVVisManager->Draw(circle);
  }
}

/*
 * Print the hit information
 */
void LCDG4CalHit::Print() {
  G4cout << "Layer:" << m_layer << G4endl;
  G4cout << "Sysno:" << m_sysNo << G4endl;
  G4cout << "Bar/End:" << m_barend << G4endl;
  G4cout << "Phi:" << m_phi << G4endl;
  G4cout << "Theta:" << m_theta << G4endl;
  G4cout << "Edep:" << m_edep << G4endl;
}

---