KVGroup.cpp

#include "KVGroup.h"
#include "KVNucleus.h"
#include "KVList.h"
#include "KVDetector.h"
#include "KVNameValueList.h"
 
using namespace std;
 
ClassImp(KVGroup)
 
 
 
 
KVGroup::KVGroup()
{
   init();
}
 
 
 
 
 
void KVGroup::init()
{
   // Default initialisation
   // KVGroup does not own the structures which it groups together
 
   fReconstructedNuclei = 0;
   SetType("GROUP");
   SetOwnsDaughters(kFALSE);
   fReconTraj.SetOwner();
}
 
 
 
 
 
KVGroup::~KVGroup()
{
   if (fReconstructedNuclei && fReconstructedNuclei->TestBit(kNotDeleted)) {
      fReconstructedNuclei->Clear();
      delete fReconstructedNuclei;
      fReconstructedNuclei = 0;
   }
   fReconstructedNuclei = 0;
   fTrajectories.Clear();
}
 
 
 
 
 
void KVGroup::Reset(Option_t* opt)
{
   //Reset the group, i.e. wipe the list of reconstructed nuclei and call "Reset" method of
   //each and every detector in the group.
   // The option 'opt' is passed to KVDetector::Reset
 
   //Info("Reset","Call %s",GetName());
   if (fReconstructedNuclei && fReconstructedNuclei->GetSize()) {
      fReconstructedNuclei->Clear();
   }
   //reset energy loss and KVDetector::IsAnalysed() state
   //plus ACQParams set to zero if opt!="N" (for No Reset)
   const_cast<KVSeqCollection*>(GetDetectors())->R__FOR_EACH(KVDetector, Reset)(opt);
}
 
 
 
 
 
void KVGroup::AddHit(KVNucleus* kvd)
{
   if (!fReconstructedNuclei) {
      fReconstructedNuclei = new KVList(kFALSE);
      fReconstructedNuclei->SetCleanup();
   }
   fReconstructedNuclei->Add(kvd);
}
 
 
 
 
 
void KVGroup::RemoveHit(KVNucleus* kvd)
{
   //Remove reconstructed nucleus from group's list of reconstructed
   //particles.
   if (fReconstructedNuclei) {
      fReconstructedNuclei->Remove(kvd);
      if (fReconstructedNuclei->GetSize() == 0) {
         delete fReconstructedNuclei;
         fReconstructedNuclei = 0;
      }
   }
   else {
      Warning("RemoveHit", "No reconstructed nuclei in this group");
   }
}
 
 
 
 
void KVGroup::ClearHitDetectors()
{
   // Loop over all detectors in group and clear their list of 'hits'
   // i.e. the lists of particles which hit each detector
   const_cast<KVSeqCollection*>(GetDetectors())->R__FOR_EACH(KVDetector, ClearHits)();
}
 
 
 
 
Int_t KVGroup::CalculateReconstructionTrajectories()
{
   // Calculate all possible (sub-)trajectories
   // for particle reconstruction (GetReconTrajectories())
   //
   // Returns number of (unique) trajectories in group
 
   fReconTraj.Clear();
   TIter next_traj(GetTrajectories());
   KVGeoDNTrajectory* traj;
   while ((traj = (KVGeoDNTrajectory*)next_traj())) {   // loop over all trajectories
 
      traj->IterateFrom();   // from furthest-out to closest-in detector
 
      KVGeoDetectorNode* N;
      while ((N = traj->GetNextNode())) {
         fReconTraj.Add(KVGeoDNTrajectory::Factory("KVReconNucTrajectory", traj, N));
      }
   }
 
   // There may be trajectories with different names but identical titles
   // (=physically same trajectories)
   // We find the duplicates, delete them, and set up a map between the names of the
   // duplicates and the name of the one remaining trajectory in the list
   TList toRemove;
   KVUniqueNameList unique_trajectories(kFALSE);//no replace
   unique_trajectories.SetOwner();
   fReconTrajMap.Clear();
   TIter nxtRT(GetReconTrajectories());
   KVGeoDNTrajectory* rnt;
   while ((rnt = (KVGeoDNTrajectory*)nxtRT())) {
 
      TNamed* n = new TNamed(rnt->GetTitle(), rnt->GetName());
      unique_trajectories.Add(n);
      TNamed* orig = n;
      if (!unique_trajectories.ObjectAdded()) {
         orig = (TNamed*)unique_trajectories.FindObject(rnt->GetTitle());
         toRemove.Add(rnt);
         delete n;
      }
      // set up mapping from duplicate trajectory name to orginal trajectory name
      fReconTrajMap.SetValue(rnt->GetName(), orig->GetTitle());
 
   }
 
   // now remove & delete the duplicates
   TIter nxtDel(&toRemove);
   while ((rnt = (KVGeoDNTrajectory*)nxtDel())) {
      fReconTraj.Remove(rnt);
   }
   toRemove.Delete();
 
   return fReconTraj.GetEntries();
}
 
 
 
 
void KVGroup::PrintData() const
{
   // list detectors of group with any data they contain
   KVDetector* det;
   TIter it(GetDetectors());
   while ((det = (KVDetector*)it())) det->Print("data");
}