TrackCheater_module.cc

Go to the documentation of this file.

//
// TrackCheater module
//
// brebel@fnal.gov
//
// Framework includes
#include "art/Framework/Core/ModuleMacros.h"

#ifndef TRKF_TRACKCHEATER_H
#define TRKF_TRACKCHEATER_H
#include <string>

#include <vector>

// ROOT includes
#include "TVector3.h"

// LArSoft includes
#include "larsim/MCCheater/BackTrackerService.h"
#include "larsim/MCCheater/ParticleInventoryService.h"
#include "larcore/Geometry/Geometry.h"
#include "lardataobj/RecoBase/Cluster.h"
#include "lardataobj/RecoBase/Hit.h"
#include "lardataobj/RecoBase/SpacePoint.h"
#include "lardataobj/RecoBase/Track.h"
#include "lardata/Utilities/AssociationUtil.h"
#include "nusimdata/SimulationBase/MCParticle.h"
#include "larcoreobj/SimpleTypesAndConstants/PhysicalConstants.h"

// Framework includes
#include "art/Framework/Core/EDProducer.h"
#include "canvas/Persistency/Common/FindManyP.h"
#include "art/Framework/Principal/Event.h"
#include "fhiclcpp/ParameterSet.h"
#include "art/Framework/Principal/Handle.h"
#include "art/Framework/Services/Registry/ServiceHandle.h"
#include "art/Framework/Services/Optional/TFileService.h"
#include "art/Framework/Services/Optional/TFileDirectory.h"
#include "canvas/Persistency/Common/FindManyP.h"
#include "messagefacility/MessageLogger/MessageLogger.h"


namespace trkf {
  class TrackCheater : public art::EDProducer {
  public:
    explicit TrackCheater(fhicl::ParameterSet const& pset);
    virtual ~TrackCheater();

    void produce(art::Event& evt);

    void reconfigure(fhicl::ParameterSet const& pset);

 private:

    std::string fCheatedClusterLabel; 
    std::string fG4ModuleLabel;       

  };
}

namespace trkf{

  //--------------------------------------------------------------------
  TrackCheater::TrackCheater(fhicl::ParameterSet const& pset)
  {
    this->reconfigure(pset);

    produces< std::vector<recob::Track>                   >();
    produces< std::vector<recob::SpacePoint>              >();
    produces< art::Assns<recob::Track, recob::Cluster>    >();
    produces< art::Assns<recob::Track, recob::SpacePoint> >();
    produces< art::Assns<recob::Track, recob::Hit>        >();
  }

  //--------------------------------------------------------------------
  TrackCheater::~TrackCheater()
  {
  }

  //--------------------------------------------------------------------
  void TrackCheater::reconfigure(fhicl::ParameterSet const& pset)
  {
    fCheatedClusterLabel = pset.get< std::string >("CheatedClusterLabel", "cluster" );
    fG4ModuleLabel       = pset.get< std::string >("G4ModuleLabel",       "largeant");

    return;
  }

  //--------------------------------------------------------------------
  void TrackCheater::produce(art::Event& evt)
  {
    art::ServiceHandle<cheat::ParticleInventoryService> pi_serv;
    art::ServiceHandle<cheat::BackTrackerService> bt_serv;
    art::ServiceHandle<geo::Geometry>            geo;

    // grab the clusters that have been reconstructed
    art::Handle< std::vector<recob::Cluster> > clustercol;
    evt.getByLabel(fCheatedClusterLabel, clustercol);

    art::FindManyP<recob::Hit> fmh(clustercol, evt, fCheatedClusterLabel);

    // make a vector of them - we aren't writing anything out to a file
    // so no need for a art::PtrVector here
    std::vector< art::Ptr<recob::Cluster> > clusters;
    art::fill_ptr_vector(clusters, clustercol);
    
    // loop over the clusters and figure out which particle contributed to each one
    std::vector< art::Ptr<recob::Cluster> >::iterator itr = clusters.begin();

    // make a map of vectors of art::Ptrs keyed by eveID values
    std::map< int, std::vector< std::pair<size_t, art::Ptr<recob::Cluster> > > > eveClusterMap;

    // loop over all clusters and fill in the map
    for(size_t c = 0; c < clusters.size(); ++c){

      std::pair<size_t, art::Ptr<recob::Cluster> > idxPtr(c, clusters[c]);

      // in the ClusterCheater module we set the cluster ID to be 
      // the eve particle track ID*1000 + plane*100 + tpc*10 + cryostat number.  The
      // floor function on the cluster ID / 1000 will give us
      // the eve track ID
      int eveID = floor((*itr)->ID()/1000.);

      eveClusterMap[eveID].push_back(idxPtr);
      ++itr;
    }// end loop over clusters

    // loop over the map and make prongs
    std::unique_ptr< std::vector<recob::Track> >                   trackcol(new std::vector<recob::Track>);
    std::unique_ptr< std::vector<recob::SpacePoint> >              spcol  (new std::vector<recob::SpacePoint>);
    std::unique_ptr< art::Assns<recob::Track, recob::SpacePoint> > tspassn(new art::Assns<recob::Track, recob::SpacePoint>);
    std::unique_ptr< art::Assns<recob::Track, recob::Cluster> >    tcassn (new art::Assns<recob::Track, recob::Cluster>);
    std::unique_ptr< art::Assns<recob::Track, recob::Hit> >        thassn (new art::Assns<recob::Track, recob::Hit>);

    for(auto const& clusterMapItr : eveClusterMap){

      // separate out the hits for each particle into the different views
      std::vector< std::pair<size_t, art::Ptr<recob::Cluster> > > const& eveClusters = clusterMapItr.second;

      simb::MCParticle *part = pi_serv->ParticleList()[clusterMapItr.first];

      // is this prong electro-magnetic in nature or 
      // hadronic/muonic?  EM --> shower, everything else is a track
      if( abs(part->PdgCode()) != 11  &&
          abs(part->PdgCode()) != 22  &&
          abs(part->PdgCode()) != 111 ){

        // vectors to hold the positions and directions of the track
        std::vector<TVector3> points;
        std::vector<TVector3> dirs;

        size_t nviews = geo->Nviews();
        std::vector< std::vector<double> > dQdx(nviews);

        mf::LogInfo("TrackCheater") << "G4 id " << clusterMapItr.first 
                                    << " is a track with pdg code "
                                    << part->PdgCode();

        art::PtrVector<recob::Cluster> ptrvs;
        std::vector<size_t> idxs;
        for(auto const& idxPtr : eveClusters){
          ptrvs.push_back(idxPtr.second);
          idxs.push_back(idxPtr.first);
        }

        // grab the hits associated with the collection plane
        std::vector< art::Ptr<recob::Hit> > hits;
        for(size_t p = 0; p < ptrvs.size(); ++p){
          std::vector< art::Ptr<recob::Hit> > chits = fmh.at(idxs[p]);
          if (!chits.size()) continue;
          if (chits[0]->SignalType() != geo::kCollection) continue;
          hits.insert(hits.end(), chits.begin(), chits.end());
        }

        // need at least 2 hits to make a track
        if(hits.size() < 2) continue;

        // loop over the hits to get the positions and directions
        size_t spStart = spcol->size();
        for(size_t t = 0; t < hits.size(); ++t){
          std::vector<double> xyz = bt_serv->HitToXYZ(hits[t]);
          points.push_back(TVector3(xyz[0], xyz[1], xyz[2]));

          std::vector<double> xyz1;
          double charge = hits[t]->Integral();
          double dx     = 0.;
          double sign   = 1.;

          if(t < hits.size()-1){
            xyz1 = bt_serv->HitToXYZ(hits[t+1]);
          }
          else{
            xyz1 = bt_serv->HitToXYZ(hits[t-1]);
            sign = -1.;
          }

          // dx is always positive
          dx = std::sqrt(std::pow(xyz1[0] - xyz[0], 2) + 
                         std::pow(xyz1[1] - xyz[1], 2) + 
                         std::pow(xyz1[2] - xyz[2], 2));
          
          // direction is always from the previous point along track to
          // the next point, that is why sign is there
          dirs.push_back(TVector3(sign*(xyz1[0] - xyz[0])/dx, 
                                  sign*(xyz1[1] - xyz[1])/dx, 
                                  sign*(xyz1[2] - xyz[2])/dx));

          dQdx[0].push_back(charge/dx);
          dQdx[1].push_back(charge/dx);
          dQdx[2].push_back(charge/dx);

          // make the space point and set its ID and XYZ
          double xyzerr[6] = {1.e-3};
          double chisqr    = 0.9;
          recob::SpacePoint sp(&xyz[0], xyzerr, chisqr, clusterMapItr.first*10000 + t);
          spcol->push_back(sp);
        }
        
        size_t spEnd = spcol->size();
        
        // add a track to the collection.  Make the track
        // ID be the same as the track ID for the eve particle
        std::vector<double> momentum(2);
        momentum[0] = part->P();
        momentum[1] = part->P(part->NumberTrajectoryPoints()-1);
        trackcol->push_back(recob::Track(points, dirs, dQdx, momentum, clusterMapItr.first));

        // associate the track with its clusters
        util::CreateAssn(*this, evt, *trackcol, ptrvs, *tcassn);

        // assume the input tracks were previously associated with hits
        hits.clear();
        for(size_t p = 0; p < ptrvs.size(); ++p){
          hits  = fmh.at(idxs[p]);
          util::CreateAssn(*this, evt, *trackcol, hits, *thassn);
        }

        // associate the track to the space points
        util::CreateAssn(*this, evt, *trackcol, *spcol, *tspassn, spStart, spEnd);

        mf::LogInfo("TrackCheater") << "adding track: \n" 
                                     << trackcol->back()
                                     << "\nto collection.";

      }//end if this is a track

    } // end loop over the map

    evt.put(std::move(trackcol));
    evt.put(std::move(spcol));
    evt.put(std::move(tcassn));
    evt.put(std::move(thassn));
    evt.put(std::move(tspassn));

    return;

  } // end produce

} // end namespace

namespace trkf{

  DEFINE_ART_MODULE(TrackCheater)

}

#endif