InfoTransfer.h

Go to the documentation of this file.

//
// Transfer hitlist and run info into a producer module.
// Do not copy this code without contacting Andrzej Szelc and Brian Rebel first.
//
// \author andrzej.szelc@yale.edu
// \author ellen.klein@yale.edu
#ifndef INFOTRANSFER_H
#define INFOTRANSFER_H
#ifndef __CINT__
#include <iostream>
#include <string>
#include <vector>

#include "nuevdb/EventDisplayBase/Reconfigurable.h"

#include "art/Framework/Services/Registry/ServiceDeclarationMacros.h"
#include "art/Persistency/Provenance/ScheduleContext.h"

namespace art {
  class ActivityRegistry;
  class Event;
}
#include "canvas/Persistency/Common/Ptr.h"

namespace fhicl {
  class ParameterSet;
}

namespace recob {
  class Hit;
}

namespace util {
  class PxPoint;
  class PxLine;
}

namespace evd {
  class InfoTransfer : public evdb::Reconfigurable {
  public:
    explicit InfoTransfer(fhicl::ParameterSet const& pset, art::ActivityRegistry& reg);
    ~InfoTransfer();

    // The Rebuild function rebuilds the various maps we need to pickup hits.
    // It is called automatically before each event is processed. For jobs involving
    // Monte Carlo generation, this is too soon. So, we'll call rebuild after those data
    // products are put into the event in LArG4.  This is the least bad way of ensuring the
    // InfoTransfer works in jobs that combine MC production and reconstruction analysis based
    // on MC truth.  Don't copy this design pattern without talking to brebel@fnal.gov first
    void Rebuild(const art::Event& evt, art::ScheduleContext);

    void reconfigure(fhicl::ParameterSet const& pset);
    void SetTestFlag(int value) { testflag = value; }
    int GetTestFlag() const { return testflag; }
    void SetRunNumber(int value) { fRun = value; }
    int GetRunNumber() const { return fRun; }
    void SetSubRunNumber(int value) { fSubRun = value; }
    int GetSubRunNumber() const { return fSubRun; }
    void SetEvtNumber(int value) { fEvt = value; }
    int GetEvtNumber() const { return fEvt; }

    void SetHitList(unsigned int p, std::vector<art::Ptr<recob::Hit>> hits_to_save)
    {
      fSelectedHitlist[p].clear();
      fSelectedHitlist[p] = hits_to_save;
    }

    std::vector<art::Ptr<recob::Hit>> const& GetHitList(unsigned int plane) const
    {
      return fRefinedHitlist[plane];
    }

    std::vector<art::Ptr<recob::Hit>> const& GetSelectedHitList(unsigned int plane) const
    {
      return fSelectedHitlist[plane];
    }

    void ClearSelectedHitList(int plane)
    {
      if (fSelectedHitlist.size() == 0) {
        return;
        std::cout << "no size" << std::endl;
      }
      fSelectedHitlist[plane].clear();
      for (unsigned int i = 0; i < fRefStartHit.size(); i++) {
        fRefStartHit[i] = NULL;
        fRefEndHit[i] = NULL;
      }
      return;
    }

    void SetStartHit(unsigned int p, recob::Hit* starthit) { fStartHit[p] = starthit; }

    recob::Hit* GetStartHit(unsigned int plane) const { return fRefStartHit[plane]; }

    void SetEndHit(unsigned int p, recob::Hit* endhit) { fEndHit[p] = endhit; }

    recob::Hit* GetEndHit(unsigned int plane) const { return fRefEndHit[plane]; }

    std::vector<double> const& GetStartHitCoords(unsigned int plane) const
    {
      return refstarthitout[plane];
    }

    std::vector<double> const& GetEndHitCoords(unsigned int plane) const
    {
      return refendhitout[plane];
    }

    void SetStartHitCoords(unsigned int plane, std::vector<double> starthitin)
    {
      starthitout[plane].clear();
      starthitout[plane].resize(2);
      starthitout[plane] = starthitin;
    }

    void SetEndHitCoords(unsigned int plane, std::vector<double> endhitin)
    {
      endhitout[plane].clear();
      endhitout[plane].resize(2);
      endhitout[plane] = endhitin;
    }

    void SetSeedList(std::vector<util::PxLine> seedlines);

    std::vector<util::PxLine> const& GetSeedList() const;

  private:
    void FillStartEndHitCoords(unsigned int plane);

    int testflag;
    int fEvt;
    int fRun;
    int fSubRun;
    std::vector<std::vector<art::Ptr<recob::Hit>>>
      fSelectedHitlist; 
    std::vector<std::vector<art::Ptr<recob::Hit>>>
      fRefinedHitlist; 
    std::vector<art::Ptr<recob::Hit>> fFullHitlist; 
    std::string fHitModuleLabel;                    

    std::vector<recob::Hit*> fStartHit;    
    std::vector<recob::Hit*> fRefStartHit; 

    std::vector<recob::Hit*> fEndHit;    
    std::vector<recob::Hit*> fRefEndHit; 

    std::vector<util::PxLine> fSeedList;

    std::vector<std::vector<double>> starthitout;
    std::vector<std::vector<double>> endhitout;

    std::vector<std::vector<double>> refstarthitout;
    std::vector<std::vector<double>> refendhitout;
  };
} //namespace
#endif // __CINT__
DECLARE_ART_SERVICE(evd::InfoTransfer, LEGACY)
#endif