KVGroup.cpp

#include "Riostream.h"
#include "KVGroup.h"
#include "KVNucleus.h"
#include "KVList.h"
#include "KVDetector.h"
#include "KVLayer.h"
#include "TROOT.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();
}
 
 
 
 
UInt_t KVGroup::GetNumberOfDetectorLayers()
{
   // The number of detector layers is the maximum number of detectors in the
   // group which are placed one in front of the other, i.e. we interrogate
   // each detector as to how many detectors there are in front of it
 
   UInt_t max = 0;
   TIter next(GetDetectors());
   KVDetector* d;
   while ((d = (KVDetector*)next())) {
      UInt_t e = d->GetAlignedDetectors()->GetEntries();
      if (e > max) max = e;
   }
   return max;
}
 
 
 
 
TList* KVGroup::GetDetectorsInLayer(UInt_t lay)
{
   // lay=1 : create and fill list with detectors closest to target
   // lay=GetNumberOfDetectorLayers() : detectors furthest from target
   //
   // Delete TList after use!
 
   TList* dets = new TList;
   TIter next(GetDetectors());
   KVDetector* d;
 
   while ((d = (KVDetector*)next())) {
      if (lay == (UInt_t)d->GetAlignedDetectors()->GetEntries()) dets->Add(d);
   }
   return dets;
}
 
 
 
 
 
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");
   }
}
 
 
 
 
TList* KVGroup::GetAlignedDetectors(KVDetector* det, UChar_t dir)
{
   //Fill TList with all detectors aligned with "det" which are closer to the target.
   //These are the detectors through which any particle stopping in "det" will have
   //to pass. By default (dir=KVGroup::kBackwards) the list starts with "det" and
   //goes towards the target. Use dir=KVGroup::kForwards to have the list in the
   //order seen by an impinging particle.
   //
   //Delete TList after use.
 
   TList* tmp = new TList;
 
   while (det) {
      tmp->Add(det);
      KVGeoDetectorNode* node = det->GetNode();
      KVSeqCollection* infront = node->GetDetectorsInFront();
      if (!infront) break;
      if (infront->GetEntries() > 1) {
         Warning("GetAlignedDetectors",
                 "No unique solution. There are %d detectors in front of %s.",
                 infront->GetEntries(), det->GetName());
         infront->ls();
      }
      det = (KVDetector*)infront->First();
   }
 
   if (dir == kForwards) {
      TIter next(tmp, kIterBackward);
      TList* tmp2 = new TList;
      while ((det = (KVDetector*)next())) tmp2->Add(det);
      delete tmp;
      tmp = tmp2;
   }
   return tmp;
}
 
 
 
 
 
 
 
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");
}