/home/bes3soft/bes3soft/Boss/7.0.2/dist/7.0.2/Muc/MucRecEvent/MucRecEvent-00-02-52/src/MucRecHitContainer.cxx

Go to the documentation of this file.

//$id$
//
//$log$

/*
 *    2003/12/13   Zhengyun You     Peking University
 * 
 *    2004/09/12   Zhengyun You     Peking University
 *                 transplanted to Gaudi framework
 */

#include "MucRecEvent/MucRecHitContainer.h"

// Constructor.
MucRecHitContainer::MucRecHitContainer()
{ }

// Destructor.
MucRecHitContainer::~MucRecHitContainer()
{
  //Clear();
}

// Initializes the MucRecHitContainer.
void 
MucRecHitContainer::Init()
{
  //  cout << "MucRecHitContainer::Init-I1  hello world" << endl;

  Clear();

  //cout << "*********** Hit Container Initialized! *********** " << endl;
}

void 
MucRecHitContainer::Clear()
{
  for (int part = 0; part < (int)MucID::getPartNum(); part++) {
    for (int seg = 0; seg < (int)MucID::getSegNum(part); seg++) {
      for (int gap = 0; gap < (int)MucID::getGapNum(part); gap++) {
        MucRecHitID gapID(part, seg, gap);
        m_GapHitCount[gapID] = 0;
      }
    }
  }

  //m_pMucRecHitCol->clear();
  m_pHit.clear();
}

int 
MucRecHitContainer::GetHitCount()
{
  int count = 0;
  for (int part = 0; part < (int)MucID::getPartNum(); part++) {
    for (int seg = 0; seg < (int)MucID::getSegNum(part); seg++) {
      for (int gap = 0; gap < (int)MucID::getGapNum(part); gap++) {
        MucRecHitID gapID(part, seg, gap);
        count += m_GapHitCount[gapID];
      }
    }
  }
  
  return count;
}

int 
MucRecHitContainer::GetGapHitCount(const MucRecHitID gapID)
{
  if (m_GapHitCount.count(gapID)) return m_GapHitCount[gapID];
  return 0;
}

int 
MucRecHitContainer::GetGapHitCount(const int part,
                                   const int seg,
                                   const int gap)
{
  MucRecHitID gapID(part, seg, gap);
  return m_GapHitCount[gapID];
}

MucRecHit* 
MucRecHitContainer::GetHit(const MucRecHitID id)
{
  if (m_pHit.count(id)) return m_pHit[id];
  else return 0;
}

MucRecHit* 
MucRecHitContainer::GetHit(const int part,
                           const int seg,
                           const int gap,
                           const int hit)
{
  MucRecHitID id(part, seg, gap, hit);
  return m_pHit[id];
}

MucRecHit* 
MucRecHitContainer::GetHitByIdentifier(const Identifier id)
{
  int part  = MucID::part(id);
  int seg   = MucID::seg(id);
  int gap   = MucID::gap(id);
  int strip = MucID::strip(id);
  
  return GetHitByIdentifier(part, seg, gap, strip);
}

MucRecHit* 
MucRecHitContainer::GetHitByIdentifier(const int part,
                                       const int seg,
                                       const int gap,
                                       const int strip)
{
  Identifier id = MucID::channel_id(part, seg, gap, strip);
  for(int i = 0; i < GetGapHitCount(part, seg, gap); i++) {
    MucRecHit *hit = GetHit(part, seg, gap, i);
    if(id == hit->GetID()) {
      return hit;
    }
  }
  
  return 0;
}

// Place a new hit object in the container.
void
MucRecHitContainer::AddHit(const Identifier id)
{
  int part  = MucID::part(id);
  int seg   = MucID::seg(id);
  int gap   = MucID::gap(id);
  int strip = MucID::strip(id);

  AddHit(part, seg, gap, strip);
}

// Place a new hit object in the container.
void
MucRecHitContainer::AddHit(const int part,
                           const int seg,
                           const int gap,
                           const int strip)
{
  if ( (part  >= 0) && (part  < (int)MucID::getPartNum()) &&
       (seg   >= 0) && (seg   < (int)MucID::getSegNum(part)) &&
       (gap   >= 0) && (gap   < (int)MucID::getGapNum(part)) &&
       (strip >= 0) && (strip < (int)MucID::getStripNum(part, seg, gap)) )
    {
      MucRecHitID gapID(part, seg, gap);
      //       cout << "MucRecHitContainer::AddHit-I1  " << idGap << endl;
      
      int hitCount = m_GapHitCount[gapID];
      MucRecHitID id(part, seg, gap, hitCount);
      //       cout << "MucRecHitContainer::AddHit-I2  hit id = " << id << endl;
      
      MucRecHit *pHit = new MucRecHit(part, seg, gap, strip);
      m_pMucRecHitCol->push_back(pHit);

      m_pHit[id] = pHit;
      if(!m_pHit[id]) {
        cout << "MucRecHitContainer::AddHit-F1  failed to create new MucRecHit" << endl;
      }
      else {
        m_GapHitCount[gapID]++;
      }
    }
}

void
MucRecHitContainer::SetMucRecHitCol(MucRecHitCol* p)
{
  m_pMucRecHitCol = p;

  MucRecHitCol::iterator imuchit;
  for(imuchit = p->begin();
      imuchit != p->end();
      imuchit++){
        int part = (*imuchit)->Part();
        int seg  = (*imuchit)->Seg();
        int gap  = (*imuchit)->Gap();
        int strip= (*imuchit)->Strip();
        
        MucRecHitID gapID(part, seg, gap);
        int hitCount = m_GapHitCount[gapID];
        MucRecHitID id(part, seg, gap, hitCount);

        m_pHit[id] = *imuchit;
        if(!m_pHit[id]) {
                cout << "MucRecHitContainer::AddHit-F1  failed to create new MucRecHit" << endl;
        }
        else {
                m_GapHitCount[gapID]++;
        }


  }

}

---