ConcatArrays.cxx

Go to the documentation of this file.

// $Id: ConcatArrays.cxx,v 1.7 2007/01/15 20:16:28 rhatcher Exp $
//
// A series of helper functions (primarily for minossoft MCMerge pkg)
// to concatenate truth and hit-level information stored in TClonesArray
// containers, properly adjusting indices in the input arrays.
//
// K. Arms
// arms@physics.umn.edu
#include <MCMerge/ConcatArrays.h>

#ifdef EVENT_KINEMATICS_PKGS
#include <EventKinematics/NuEvtKin.h>
#include <EventKinematics/StdHepUtil.h>
using namespace StdHepUtil;
#endif

#include <Digitization/DigiScintHit.h>
#include <REROOT_Classes/REROOT_StdHep.h>
#include <REROOT_Classes/REROOT_StdHepHead.h>
#include <REROOT_Classes/REROOT_NeuKin.h>
#include <REROOT_Classes/REROOT_FluxInfo.h>
#include <REROOT_Classes/REROOT_FluxWgt.h>
#include "TParticle.h"
#include "TMath.h"

#include "Riostream.h"
#include "TClass.h"

//CVSID("$Id: ConcatArrays.cxx,v 1.7 2007/01/15 20:16:28 rhatcher Exp $");

//__________________________________________________________________________
void ConcatArrays::ConcatDigiScintHits(TClonesArray*  addList,
                                       TClonesArray*  permList,
                                       Int_t          permStdHepSz) {
  // Concatenate (permList += addList)
  // --> Call *before* ConcatStdHep

  if (!addList || !permList) return;

  const Int_t oldSize = permList->GetEntriesFast();
  const Int_t newSize = oldSize + addList->GetEntriesFast();

  for (Int_t i = oldSize; i < newSize; i++) {
    DigiScintHit* tempHit =
      dynamic_cast<DigiScintHit*>(addList->At(i-oldSize));

    if (tempHit) {
      Int_t oldId = tempHit->TrackId();
      Int_t newId = (oldId < 0)? oldId - permStdHepSz : oldId + permStdHepSz ;
      tempHit->SetTrackId( newId );

      new( (*permList)[i] ) DigiScintHit( *tempHit );
    }
  }
  
}
//__________________________________________________________________________
void ConcatArrays::ConcatFluxInfo(TClonesArray*  addList,
                                  TClonesArray*  permList){
  // Concatenate (permList += addList)

  if (!addList || !permList) return;

  const Int_t oldSize = permList->GetEntriesFast();
  const Int_t newSize = oldSize + addList->GetEntriesFast();

  for (Int_t i = oldSize; i < newSize; i++)
    new( (*permList)[i] ) REROOT_FluxInfo( *dynamic_cast<REROOT_FluxInfo*>(addList->At(i-oldSize)) );

}
//__________________________________________________________________________
void ConcatArrays::ConcatFluxWgt(TClonesArray*  addList,
                                 TClonesArray*  permList){
  // Concatenate (permList += addList)

  if (!addList || !permList) return;

  const Int_t oldSize = permList->GetEntriesFast();
  const Int_t newSize = oldSize + addList->GetEntriesFast();

  for (Int_t i = oldSize; i < newSize; i++)
    new( (*permList)[i] ) REROOT_FluxWgt( *dynamic_cast<REROOT_FluxWgt*>(addList->At(i-oldSize)) );

}
//__________________________________________________________________________
void ConcatArrays::ConcatNuEvtKin(TClonesArray*  addList,
                                  TClonesArray*  permList,
                                  TClonesArray*  stdhepList){
  // Concatenate (permList += addList)

  if (!addList || !permList || !stdhepList) return;

#ifdef EVENT_KINEMATICS_PKGS
  const Int_t oldSize = permList->GetEntriesFast();
  const Int_t newSize = oldSize + addList->GetEntriesFast();

  for (Int_t i = oldSize; i < newSize; i++)
    new( (*permList)[i] ) NuEvtKin( *dynamic_cast<NuEvtKin*>(addList->At(i-oldSize)) );

#endif  

}
//__________________________________________________________________________
void ConcatArrays::ConcatNuKin(TClonesArray*  addList,
                               TClonesArray*  permList){
  // Concatenate (permList += addList)

  if (!addList || !permList) return;

  const Int_t oldSize = permList->GetEntriesFast();
  const Int_t newSize = oldSize + addList->GetEntriesFast();

  for (Int_t i = oldSize; i < newSize; i++)
    new( (*permList)[i] ) REROOT_NeuKin( *dynamic_cast<REROOT_NeuKin*>(addList->At(i-oldSize)) );

}
//__________________________________________________________________________
void  ConcatArrays::ConcatStdHep(TClonesArray*  addList,
                                 TClonesArray*  permList) {
  // Concatenate (permList += addList) and adjust StdHep contents accordingly
  // Call another fn to adjust DigiScintHit.trkId() accordingly
  //
  // MUST BE CALLED ONLY *AFTER* ConcatDigiScintHits() !!!!
  //

  if (!addList || !permList ) return;

  const Int_t oldSize = permList->GetEntriesFast();
  const Int_t newSize = oldSize + addList->GetEntriesFast();

  for (Int_t i = oldSize; i < newSize; i++) {
    TParticle* thisPart = new( (*permList)[i] ) TParticle( *dynamic_cast<TParticle*>(addList->At(i-oldSize)) );

    if ( thisPart ) {
      // Adjust indices of this particle's (1st/last) parents & children
      for (Int_t j = 0; j < 2; j++) {
        if (thisPart->GetMother(j)   >= 0)
          thisPart->SetMother(  j, thisPart->GetMother(j)   + oldSize );
        if (thisPart->GetDaughter(j) >= 0)
          thisPart->SetDaughter(j, thisPart->GetDaughter(j) + oldSize );
      } // end loop over Mother/Daughter arrays
    } // end if stdhep TParticle is OK
  } // end loop over new TParticles

}

//__________________________________________________________________________

void ConcatArrays::OffsetTime(TClonesArray* tcarr, Float_t offsetns){

  if ( !tcarr ) {
    return;
  }

  // DigiScintHit TClonesArray
  if ( tcarr->GetClass()->InheritsFrom("DigiScintHit") ) {

    UInt_t tempSize = tcarr->GetEntriesFast();
    
    for ( UInt_t i = 0; i < tempSize ; i++) {
      DigiScintHit* tempHit = dynamic_cast<DigiScintHit*>(tcarr->At(i));
      
      Float_t newT1 = tempHit->T1() + offsetns*1.e-9; // time is stored as sec
      Float_t newT2 = tempHit->T2() + offsetns*1.e-9; // time is stored as sec

      tempHit->SetT1(newT1);
      tempHit->SetT2(newT2);
    } // end loop over DigiScintHits

    return;
  } // end if DigiScintHit TClonesArray

  // StdHep TClonesArray
  if ( tcarr->GetClass()->InheritsFrom("TParticle") ) {
    for ( Int_t i = 0; i < tcarr->GetEntriesFast() ; i++) {
      TParticle* oldPart = dynamic_cast<TParticle*>(tcarr->At(i));
      oldPart->SetProductionVertex(oldPart->Vx(), oldPart->Vy(),
                                   oldPart->Vz(), oldPart->T()+offsetns*1.e-9);
    } // end loop over StdHep TParticles
    return;
  } // end if StdHep TClonesArray

  // NeuKin, NuEvtKin, FluxInfo, FluxWgt TClonesArray(s)
  if ( ( tcarr->GetClass()->InheritsFrom("REROOT_NeuKin"  ) ||
         tcarr->GetClass()->InheritsFrom("NeuKin"         )  ) ||
       ( tcarr->GetClass()->InheritsFrom("NuEvtKin"       )  ) ||
       ( tcarr->GetClass()->InheritsFrom("REROOT_FluxInfo") ||
         tcarr->GetClass()->InheritsFrom("FluxInfo"       )  ) ||
       ( tcarr->GetClass()->InheritsFrom("REROOT_FluxWgt" ) ||
         tcarr->GetClass()->InheritsFrom("FluxWgt"        )  )  ) {
    // Do nothing
    return;
  } // end if TClonesArray contains no time info

  
  
  cout << "=E= ConcatArrays: "
       << std::string(tcarr->At(0)->ClassName())
       << " is not a recognized SimSnarlRecord component: "
       << "this should really *never ever* happen, so you "
       << "probably broke something!"
       << endl; // really and truly

}
//__________________________________________________________________________

---