cluster::ClusterCheater Class Reference

Inheritance diagram for cluster::ClusterCheater:

Public Member Functions
	ClusterCheater (fhicl::ParameterSet const &pset)
Public Member Functions inherited from art::EDProducer
	EDProducer (fhicl::ParameterSet const &pset)
template<typename Config >
	EDProducer (Table< Config > const &config)
std::string	workerType () const
Public Member Functions inherited from art::detail::Producer
virtual	~Producer () noexcept
	Producer (fhicl::ParameterSet const &)
	Producer (Producer const &)=delete
	Producer (Producer &&)=delete
Producer &	operator= (Producer const &)=delete
Producer &	operator= (Producer &&)=delete
void	doBeginJob (SharedResources const &resources)
void	doEndJob ()
void	doRespondToOpenInputFile (FileBlock const &fb)
void	doRespondToCloseInputFile (FileBlock const &fb)
void	doRespondToOpenOutputFiles (FileBlock const &fb)
void	doRespondToCloseOutputFiles (FileBlock const &fb)
bool	doBeginRun (RunPrincipal &rp, ModuleContext const &mc)
bool	doEndRun (RunPrincipal &rp, ModuleContext const &mc)
bool	doBeginSubRun (SubRunPrincipal &srp, ModuleContext const &mc)
bool	doEndSubRun (SubRunPrincipal &srp, ModuleContext const &mc)
bool	doEvent (EventPrincipal &ep, ModuleContext const &mc, std::atomic< std::size_t > &counts_run, std::atomic< std::size_t > &counts_passed, std::atomic< std::size_t > &counts_failed)
Public Member Functions inherited from art::Modifier
	~Modifier () noexcept
	Modifier ()
	Modifier (Modifier const &)=delete
	Modifier (Modifier &&)=delete
Modifier &	operator= (Modifier const &)=delete
Modifier &	operator= (Modifier &&)=delete
Public Member Functions inherited from art::ModuleBase
virtual	~ModuleBase () noexcept
	ModuleBase ()
ModuleDescription const &	moduleDescription () const
void	setModuleDescription (ModuleDescription const &)
std::array< std::vector< ProductInfo >, NumBranchTypes > const &	getConsumables () const
void	sortConsumables (std::string const &current_process_name)
template<typename T , BranchType BT>
ViewToken< T >	consumesView (InputTag const &tag)
template<typename T , BranchType BT>
ViewToken< T >	mayConsumeView (InputTag const &tag)

Private Member Functions
void	produce (art::Event &evt) override

Private Attributes
std::string	fMCGeneratorLabel
	label for module to get MC truth information More...
std::string	fHitModuleLabel
	label for module creating recob::Hit objects More...
std::string	fG4ModuleLabel
	label for module running G4 and making particles, etc More...
unsigned int	fMinHits
	minimum number of hits to make a cluster More...

Additional Inherited Members
Public Types inherited from art::EDProducer
using	ModuleType = EDProducer
using	WorkerType = WorkerT< EDProducer >
Public Types inherited from art::detail::Producer
template<typename UserConfig , typename KeysToIgnore = void>
using	Table = Modifier::Table< UserConfig, KeysToIgnore >
Public Types inherited from art::Modifier
template<typename UserConfig , typename UserKeysToIgnore = void>
using	Table = ProducerTable< UserConfig, detail::ModuleConfig, UserKeysToIgnore >
Static Public Member Functions inherited from art::EDProducer
static void	commitEvent (EventPrincipal &ep, Event &e)
Protected Member Functions inherited from art::ModuleBase
ConsumesCollector &	consumesCollector ()
template<typename T , BranchType = InEvent>
ProductToken< T >	consumes (InputTag const &)
template<typename Element , BranchType = InEvent>
ViewToken< Element >	consumesView (InputTag const &)
template<typename T , BranchType = InEvent>
void	consumesMany ()
template<typename T , BranchType = InEvent>
ProductToken< T >	mayConsume (InputTag const &)
template<typename Element , BranchType = InEvent>
ViewToken< Element >	mayConsumeView (InputTag const &)
template<typename T , BranchType = InEvent>
void	mayConsumeMany ()

Detailed Description

Definition at line 40 of file ClusterCheater_module.cc.

Constructor & Destructor Documentation

cluster::ClusterCheater::ClusterCheater ( fhicl::ParameterSet const &  pset )

explicit

Definition at line 78 of file ClusterCheater_module.cc.

                                                              : EDProducer{pset}
  {
    fMCGeneratorLabel = pset.get<std::string>("MCGeneratorLabel", "generator");
    fHitModuleLabel = pset.get<std::string>("HitModuleLabel", "hit");
    fG4ModuleLabel = pset.get<std::string>("G4ModuleLabel", "largeant");
    fMinHits = pset.get<unsigned int>("MinHits", 1);

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

Member Function Documentation

void cluster::ClusterCheater::produce ( art::Event &  evt )

overrideprivatevirtual

Todo:

: The above encoding of the ID probably won't work for DUNE and should be revisited

Implements art::EDProducer.

Definition at line 91 of file ClusterCheater_module.cc.

  {
    art::ServiceHandle<geo::Geometry const> geo;
    art::ServiceHandle<cheat::BackTrackerService const> bt_serv;
    art::ServiceHandle<cheat::ParticleInventoryService> pi_serv;

    // grab the hits that have been reconstructed
    art::Handle<std::vector<recob::Hit>> hitcol;
    evt.getByLabel(fHitModuleLabel, hitcol);

    // 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::Hit>> hits;
    art::fill_ptr_vector(hits, hitcol);

    // adopt an EmEveIdCalculator to find the eve ID.
    // will return a primary particle if it doesn't find
    // a responsible particle for an EM process
    pi_serv->SetEveIdCalculator(new sim::EmEveIdCalculator);

    MF_LOG_DEBUG("ClusterCheater") << pi_serv->ParticleList();

    // make a map of vectors of art::Ptrs keyed by eveID values and
    // location in cryostat, TPC, plane coordinates of where the hit originated
    std::map<eveLoc, std::vector<art::Ptr<recob::Hit>>> eveHitMap;

    auto const clockData = art::ServiceHandle<detinfo::DetectorClocksService const>()->DataFor(evt);

    // loop over all hits and fill in the map
    for (auto const& itr : hits) {

      std::vector<sim::TrackIDE> eveides = bt_serv->HitToEveTrackIDEs(clockData, itr);

      // loop over all eveides for this hit
      for (size_t e = 0; e < eveides.size(); ++e) {

        // don't worry about eve particles that contribute less than 10% of the
        // energy in the current hit
        if (eveides[e].energyFrac < 0.1) continue;

        eveLoc el(eveides[e].trackID, itr->WireID().planeID());

        eveHitMap[el].push_back(itr);

      } // end loop over eve IDs for this hit

    } // end loop over hits

    // loop over the map and make clusters
    std::unique_ptr<std::vector<recob::Cluster>> clustercol(new std::vector<recob::Cluster>);
    std::unique_ptr<art::Assns<recob::Cluster, recob::Hit>> assn(
      new art::Assns<recob::Cluster, recob::Hit>);

    auto const detProp =
      art::ServiceHandle<detinfo::DetectorPropertiesService const>()->DataFor(evt, clockData);
    util::GeometryUtilities const gser{*geo, clockData, detProp};

    // prepare the algorithm to compute the cluster characteristics;
    // we use the "standard" one here; configuration would happen here,
    // but we are using the default configuration for that algorithm
    ClusterParamsImportWrapper<StandardClusterParamsAlg> ClusterParamAlgo;

    for (auto hitMapItr : eveHitMap) {

      // ================================================================================
      // === Only keeping clusters with fMinHits
      // ================================================================================
      if (hitMapItr.second.size() < fMinHits) continue;

      // get the center of this plane in world coordinates
      double xyz[3] = {0.};
      double xyz2[3] = {0.};
      double local[3] = {0.};
      unsigned int cryostat = hitMapItr.first.planeID.Cryostat;
      unsigned int tpc = hitMapItr.first.planeID.TPC;
      unsigned int plane = hitMapItr.first.planeID.Plane;
      geo->Cryostat(cryostat).TPC(tpc).Plane(plane).LocalToWorld(local, xyz);

      MF_LOG_DEBUG("ClusterCheater")
        << "make cluster for eveID: " << hitMapItr.first.eveID << " in cryostat: " << cryostat
        << " tpc: " << tpc << " plane: " << plane << " view: " << hitMapItr.second.at(0)->View();

      // get the direction of this particle in the current cryostat, tpc and plane
      const simb::MCParticle* part = pi_serv->TrackIdToParticle_P(hitMapItr.first.eveID);
      if (!part) continue;

      // now set the y and z coordinates of xyz to be the first point on the particle
      // trajectory and use the initial directions to determine the dT/dW
      // multiply the direction cosine by 10 to give a decent lever arm for determining
      // dW
      xyz[1] = part->Vy();
      xyz[2] = part->Vz();
      xyz2[0] = xyz[0];
      xyz2[1] = xyz[1] + 10. * part->Py() / part->P();
      xyz2[2] = xyz[2] + 10. * part->Pz() / part->P();

      // convert positions to wire and time
      unsigned int w1 = 0;
      unsigned int w2 = 0;

      try {
        w1 = geo->NearestWire(xyz, plane, tpc, cryostat);
      }
      catch (cet::exception& e) {
        w1 = atoi(e.explain_self().substr(e.explain_self().find("#") + 1, 5).c_str());
      }
      try {
        w2 = geo->NearestWire(xyz2, plane, tpc, cryostat);
      }
      catch (cet::exception& e) {
        w2 = atoi(e.explain_self().substr(e.explain_self().find("#") + 1, 5).c_str());
      }

      // sort the vector of hits with respect to the directionality of the wires determined by
      if (w2 < w1)
        std::sort(hitMapItr.second.rbegin(), hitMapItr.second.rend(), sortHitsByWire);
      else
        std::sort(hitMapItr.second.begin(), hitMapItr.second.end(), sortHitsByWire);

      // set the start and end wires and times
      double startWire = hitMapItr.second.front()->WireID().Wire;
      double startTime = hitMapItr.second.front()->PeakTimeMinusRMS();
      double endWire = hitMapItr.second.back()->WireID().Wire;
      double endTime = hitMapItr.second.back()->PeakTimePlusRMS();

      // add a cluster to the collection.  Make the ID be the eve particle
      // trackID*1000 + plane number*100 + tpc*10 + cryostat that the current hits are from
      ///\todo: The above encoding of the ID probably won't work for DUNE and should be revisited
      const geo::PlaneID& planeID = hitMapItr.first.planeID;
      recob::Cluster::ID_t clusterID =
        (((hitMapItr.first.eveID) * 10 + planeID.Plane) * 10 +
         planeID.TPC // 10 is weird choice for DUNE FD... should be 1000! FIXME
         ) *
          10 +
        planeID.Cryostat;

      // feed the algorithm with all the cluster hits
      ClusterParamAlgo.ImportHits(gser, hitMapItr.second);

      // create the recob::Cluster directly in the vector
      ClusterCreator cluster(gser,
                             ClusterParamAlgo,               // algo
                             startWire,                      // start_wire
                             0.,                             // sigma_start_wire
                             startTime,                      // start_tick
                             0.,                             // sigma_start_tick
                             endWire,                        // end_wire
                             0.,                             // sigma_end_wire
                             endTime,                        // end_tick
                             0.,                             // sigma_end_tick
                             clusterID,                      // ID
                             hitMapItr.second.at(0)->View(), // view
                             planeID,                        // plane
                             recob::Cluster::Sentry          // sentry
      );

      clustercol->emplace_back(cluster.move());

      // association the hits to this cluster
      util::CreateAssn(evt, *clustercol, hitMapItr.second, *assn);

      mf::LogInfo("ClusterCheater") << "adding cluster: \n"
                                    << clustercol->back() << "\nto collection.";

    } // end loop over the map

    evt.put(std::move(clustercol));
    evt.put(std::move(assn));
  } // end produce

Member Data Documentation

std::string cluster::ClusterCheater::fG4ModuleLabel

private

label for module running G4 and making particles, etc

Definition at line 49 of file ClusterCheater_module.cc.

std::string cluster::ClusterCheater::fHitModuleLabel

private

label for module creating recob::Hit objects

Definition at line 48 of file ClusterCheater_module.cc.

std::string cluster::ClusterCheater::fMCGeneratorLabel

private

label for module to get MC truth information

Definition at line 47 of file ClusterCheater_module.cc.

unsigned int cluster::ClusterCheater::fMinHits

private

minimum number of hits to make a cluster

Definition at line 50 of file ClusterCheater_module.cc.

---

The documentation for this class was generated from the following file:

• larreco/larreco/ClusterFinder/ClusterCheater_module.cc