Inheritance diagram for trkf::SpacePointCheater:

Public Member Functions
	SpacePointCheater (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

void	endJob () override

Private Attributes
SpacePointAlg	fSptalg

std::string	fClusterModuleLabel

unsigned int	fMinHits

bool	fClusterAssns

int	fNumEvent

int	fNumSpt2

int	fNumSpt3

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 36 of file SpacePointCheater_module.cc.

Constructor & Destructor Documentation

trkf::SpacePointCheater::SpacePointCheater ( fhicl::ParameterSet const & pset )

explicit

Definition at line 61 of file SpacePointCheater_module.cc.

     : EDProducer{pset}
     , fSptalg(pset.get<fhicl::ParameterSet>("SpacePointAlg"))
     , fMinHits(0)
     , fClusterAssns(false)
     , fNumEvent(0)
     , fNumSpt2(0)
     , fNumSpt3(0)
   {
     fClusterModuleLabel = pset.get<std::string>("ClusterModuleLabel");
     fMinHits = pset.get<unsigned int>("MinHits");
     fClusterAssns = pset.get<bool>("ClusterAssns");
 
     produces<std::vector<art::PtrVector<recob::SpacePoint>>>();
     produces<std::vector<recob::SpacePoint>>();
     produces<art::Assns<recob::SpacePoint, recob::Hit>>();
     if (fClusterAssns) produces<art::Assns<recob::SpacePoint, recob::Cluster>>();
 
     mf::LogInfo("SpacePointCheater")
       << "SpacePointCheater configured with the following parameters:\n"
       << "  ClusterModuleLabel = " << fClusterModuleLabel << "\n"
       << "  Minimum Hits per Cluster = " << fMinHits << "\n"
       << "  Cluster associations = " << fClusterAssns;
   }

Member Function Documentation

void trkf::SpacePointCheater::endJob ( )

overrideprivatevirtual

Reimplemented from art::EDProducer.

Definition at line 315 of file SpacePointCheater_module.cc.

   {
     mf::LogInfo("SpacePointCheater")
       << "SpacePointCheater statistics:\n"
       << "  Number of events = " << fNumEvent << "\n"
       << "  Number of 2-view space points created = " << fNumSpt2 << "\n"
       << "  Number of 3-view space points created = " << fNumSpt3;
   }

void trkf::SpacePointCheater::produce ( art::Event & evt )

overrideprivatevirtual

Implements art::EDProducer.

Definition at line 93 of file SpacePointCheater_module.cc.

   {
     ++fNumEvent;
 
     // Get Services.
 
     art::ServiceHandle<geo::Geometry const> geom;
 
     // Get clusters.
 
     art::Handle<std::vector<recob::Cluster>> clusterh;
     evt.getByLabel(fClusterModuleLabel, clusterh);
 
     // Make a double or triple loop over clusters in distinct views
     // (depending on minimum number of views configured in SpacePointAlg).
 
     if (clusterh.isValid()) {
 
       // Make a collection of space points that will be inserted into the event.
 
       auto sptvecs = std::make_unique<std::vector<art::PtrVector<recob::SpacePoint>>>();
       auto spts = std::make_unique<std::vector<recob::SpacePoint>>();
       auto sphitassn = std::make_unique<art::Assns<recob::SpacePoint, recob::Hit>>();
       auto spclassn = std::make_unique<art::Assns<recob::SpacePoint, recob::Cluster>>();
 
       // Make a hit vector which will be used to store hits to be passed
       // to SpacePointAlg.
 
       art::PtrVector<recob::Hit> hits;
       art::FindManyP<recob::Hit> fm(clusterh, evt, fClusterModuleLabel);
 
       // Loop over first cluster.
 
       auto const clockData =
         art::ServiceHandle<detinfo::DetectorClocksService const>()->DataFor(evt);
       auto const detProp =
         art::ServiceHandle<detinfo::DetectorPropertiesService const>()->DataFor(evt, clockData);
 
       int nclus = clusterh->size();
       for (int iclus = 0; iclus < nclus; ++iclus) {
         art::Ptr<recob::Cluster> piclus(clusterh, iclus);
         geo::View_t iview = piclus->View();
 
         std::vector<art::Ptr<recob::Hit>> ihits = fm.at(iclus);
 
         // Test first view.
 
         if (ihits.size() >= fMinHits &&
             ((iview == geo::kU && fSptalg.enableU()) || (iview == geo::kV && fSptalg.enableV()) ||
              (iview == geo::kZ && fSptalg.enableW()))) {
 
           // Store hits from first view into hit vector.
 
           unsigned int nihits = ihits.size();
           hits.clear();
           hits.reserve(nihits);
           for (std::vector<art::Ptr<recob::Hit>>::const_iterator i = ihits.begin();
                i != ihits.end();
                ++i)
             hits.push_back(*i);
 
           // Loop over second cluster.
 
           for (int jclus = 0; jclus < iclus; ++jclus) {
             art::Ptr<recob::Cluster> pjclus(clusterh, jclus);
             geo::View_t jview = pjclus->View();
 
             std::vector<art::Ptr<recob::Hit>> jhits = fm.at(jclus);
 
             // Test second view.
 
             if (jhits.size() >= fMinHits &&
                 ((jview == geo::kU && fSptalg.enableU()) ||
                  (jview == geo::kV && fSptalg.enableV()) ||
                  (jview == geo::kZ && fSptalg.enableW())) &&
                 jview != iview) {
 
               // Store hits from second view into hit vector.
 
               unsigned int njhits = jhits.size();
               assert(hits.size() >= nihits);
               //hits.resize(nihits);
               while (hits.size() > nihits)
                 hits.pop_back();
               assert(hits.size() == nihits);
               hits.reserve(nihits + njhits);
               for (std::vector<art::Ptr<recob::Hit>>::const_iterator j = jhits.begin();
                    j != jhits.end();
                    ++j)
                 hits.push_back(*j);
 
               // If two-view space points are allowed, make them here.
 
               if (fSptalg.minViews() <= 2) {
                 std::vector<recob::SpacePoint> new_spts;
                 fSptalg.makeMCTruthSpacePoints(clockData, detProp, hits, new_spts);
 
                 // If we found some space points, insert them into the event.
 
                 if (new_spts.size() > 0) {
                   fNumSpt2 += new_spts.size();
                   art::PtrVector<recob::Cluster> clusters;
                   clusters.reserve(2);
                   clusters.push_back(piclus);
                   clusters.push_back(pjclus);
 
                   // Insert newly found space points into event collection.
 
                   int nspt = spts->size();
                   spts->insert(spts->end(), new_spts.begin(), new_spts.end());
 
                   // Associate space points with hits and clusters.
 
                   art::PtrVector<recob::SpacePoint> sptvec;
                   for (unsigned int ispt = nspt; ispt < spts->size(); ++ispt) {
                     const recob::SpacePoint& spt = (*spts)[ispt];
                     const art::PtrVector<recob::Hit>& hits = fSptalg.getAssociatedHits(spt);
                     util::CreateAssn(evt, *spts, hits, *sphitassn, ispt);
                     if (fClusterAssns) util::CreateAssn(evt, *spts, clusters, *spclassn, ispt);
 
                     // make the PtrVector for this collection of space points
                     // Do not reproduce the following lines
                     // Contact brebel@fnal.gov if you think you need to reproduce these lines.
                     art::ProductID spid = evt.getProductID<std::vector<recob::SpacePoint>>();
                     art::Ptr<recob::SpacePoint> spptr(spid, ispt, evt.productGetter(spid));
                     sptvec.push_back(spptr);
                   }
                   sptvecs->push_back(sptvec);
                 }
               }
 
               // Loop over third cluster.
 
               for (int kclus = 0; kclus < jclus; ++kclus) {
                 art::Ptr<recob::Cluster> pkclus(clusterh, kclus);
                 geo::View_t kview = pkclus->View();
 
                 std::vector<art::Ptr<recob::Hit>> khits = fm.at(kclus);
 
                 // Test third view.
 
                 if (khits.size() >= fMinHits &&
                     ((kview == geo::kU && fSptalg.enableU()) ||
                      (kview == geo::kV && fSptalg.enableV()) ||
                      (kview == geo::kZ && fSptalg.enableW())) &&
                     kview != iview && kview != jview) {
 
                   // Store hits from third view into hit vector.
 
                   unsigned int nkhits = khits.size();
                   assert(hits.size() >= nihits + njhits);
                   //hits.resize(nihits + njhits);
                   while (hits.size() > nihits + njhits)
                     hits.pop_back();
                   assert(hits.size() == nihits + njhits);
                   hits.reserve(nihits + njhits + nkhits);
                   for (std::vector<art::Ptr<recob::Hit>>::const_iterator k = khits.begin();
                        k != khits.end();
                        ++k)
                     hits.push_back(*k);
 
                   // Make three-view space points.
 
                   std::vector<recob::SpacePoint> new_spts;
                   fSptalg.makeMCTruthSpacePoints(clockData, detProp, hits, new_spts);
 
                   // If we found some space points, insert them into the event.
 
                   if (new_spts.size() > 0) {
                     fNumSpt3 += new_spts.size();
                     art::PtrVector<recob::Cluster> clusters;
                     clusters.reserve(3);
                     clusters.push_back(piclus);
                     clusters.push_back(pjclus);
                     clusters.push_back(pkclus);
 
                     // Insert newly found space points into event collection.
 
                     int nspt = spts->size();
                     spts->insert(spts->end(), new_spts.begin(), new_spts.end());
 
                     // Associate space points with hits and clusters.
 
                     art::PtrVector<recob::SpacePoint> sptvec;
                     for (unsigned int ispt = nspt; ispt < spts->size(); ++ispt) {
                       const recob::SpacePoint& spt = (*spts)[ispt];
                       const art::PtrVector<recob::Hit>& hits = fSptalg.getAssociatedHits(spt);
                       util::CreateAssn(evt, *spts, hits, *sphitassn, ispt);
                       if (fClusterAssns) util::CreateAssn(evt, *spts, clusters, *spclassn, ispt);
 
                       // make the PtrVector for this collection of space points
                       // Do not reproduce the following lines
                       // Contact brebel@fnal.gov if you think you need to reproduce these lines.
                       art::ProductID spid = evt.getProductID<std::vector<recob::SpacePoint>>();
                       art::Ptr<recob::SpacePoint> spptr(spid, ispt, evt.productGetter(spid));
                       sptvec.push_back(spptr);
                     }
                     sptvecs->push_back(sptvec);
                   }
                 }
               }
             }
           }
         }
       }
 
       // Add space points and associations to event.
 
       evt.put(std::move(spts));
       evt.put(std::move(sptvecs));
       evt.put(std::move(sphitassn));
       if (fClusterAssns) evt.put(std::move(spclassn));
     }
   }