nnet::SPMultiTpcDump Class Reference

Inheritance diagram for nnet::SPMultiTpcDump:

Classes
struct	Config

Public Types
using	Parameters = art::EDAnalyzer::Table< Config >
Public Types inherited from art::EDAnalyzer
using	WorkerType = WorkerT< EDAnalyzer >
using	ModuleType = EDAnalyzer

Public Member Functions
	SPMultiTpcDump (Parameters const &config)
void	beginJob () override
void	analyze (const art::Event &event) override
Public Member Functions inherited from art::EDAnalyzer
	EDAnalyzer (fhicl::ParameterSet const &pset)
template<typename Config >
	EDAnalyzer (Table< Config > const &config)
std::string	workerType () const
Public Member Functions inherited from art::detail::Analyzer
virtual	~Analyzer () noexcept
	Analyzer (fhicl::ParameterSet const &pset)
template<typename Config >
	Analyzer (Table< Config > const &config)
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::Observer
	~Observer () noexcept
	Observer (Observer const &)=delete
	Observer (Observer &&)=delete
Observer &	operator= (Observer const &)=delete
Observer &	operator= (Observer &&)=delete
void	registerProducts (ProductDescriptions &, ModuleDescription const &)
void	fillDescriptions (ModuleDescription const &)
fhicl::ParameterSetID	selectorConfig () const
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	ResetVars ()
bool	prepareEv (const art::Event &event, detinfo::DetectorPropertiesData const &detProp)
bool	InsideFidVol (TVector3 const &pvtx) const
void	CorrOffset (detinfo::DetectorPropertiesData const &detProp, TVector3 &vec, const simb::MCParticle &particle)
TVector2	GetProjVtx (detinfo::DetectorPropertiesData const &detProp, TVector3 const &vtx3d, const size_t cryo, const size_t tpc, const size_t plane) const

Private Attributes
nnet::TrainingDataAlg	fTrainingDataAlg
art::InputTag	fGenieGenLabel
TTree *	fTree
TTree *	fTree2D
int	fEvent
	number of the event being processed More...
int	fRun
	number of the run being processed More...
int	fSubRun
	number of the sub-run being processed More...
bool	fSaveDepositMap
bool	fSavePdgMap
double	fFidVolCut
NUVTX	fPointid
int	fCryo
int	fTpc
int	fPlane
int	fPdg
int	fInteraction
int	fPixX
int	fPixY
float	fPosX
float	fPosY
geo::GeometryCore const *	fGeometry

Additional Inherited Members
Protected Member Functions inherited from art::Observer
std::string const &	processName () const
bool	wantAllEvents () const noexcept
bool	wantEvent (ScheduleID id, Event const &e) const
Handle< TriggerResults >	getTriggerResults (Event const &e) const
	Observer (fhicl::ParameterSet const &config)
	Observer (std::vector< std::string > const &select_paths, std::vector< std::string > const &reject_paths, fhicl::ParameterSet const &config)
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 66 of file SPMultiTpcDump_module.cc.

Member Typedef Documentation

using nnet::SPMultiTpcDump::Parameters = art::EDAnalyzer::Table<Config>

Definition at line 80 of file SPMultiTpcDump_module.cc.

Constructor & Destructor Documentation

nnet::SPMultiTpcDump::SPMultiTpcDump ( SPMultiTpcDump::Parameters const &  config )

explicit

Definition at line 124 of file SPMultiTpcDump_module.cc.

                                                                       : art::EDAnalyzer(config),
        fTrainingDataAlg(config().TrainingDataAlg()),
        fGenieGenLabel(config().GenModuleLabel()),
        fSaveDepositMap(config().SaveDepositMap()),
        fSavePdgMap(config().SavePdgMap()),
        fFidVolCut(config().FidVolCut())
  {
    fGeometry = &*(art::ServiceHandle<geo::Geometry>());
  }

Member Function Documentation

void nnet::SPMultiTpcDump::analyze ( const art::Event &  event )

override

Definition at line 212 of file SPMultiTpcDump_module.cc.

  {
    fEvent  = event.id().event();
    fRun    = event.run();
    fSubRun = event.subRun();
    ResetVars();
    
    auto const clockData = art::ServiceHandle<detinfo::DetectorClocksService const>()->DataFor(event);
    auto const detProp = art::ServiceHandle<detinfo::DetectorPropertiesService const>()->DataFor(event, clockData);
    prepareEv(event, detProp);

        std::ostringstream os;
        os << "event_" << fEvent << "_run_" << fRun << "_subrun_" << fSubRun;
        std::cout << "analyze " << os.str() << std::endl;

    size_t cryo = 0;
    size_t plane_align0[3] = { 1, 0, 2 }; // offsets to align signals in drift direction between planes
    size_t plane_align1[3] = { 2, 1, 0 };
    size_t max_align = 0;
    for (size_t i = 0; i < 3; ++i)
    {
        if (plane_align0[i] > max_align) max_align = plane_align0[i];
        if (plane_align1[i] > max_align) max_align = plane_align1[i];
    }

    size_t weff[3] = { 404, 404, 485 }; // effective offset in wires between consecutive tpc's, incl. dead spaces, per plane
    size_t tpcs[3] = { 2, 2, 6 };       // FD workspace dimension in #tpc's
    //size_t tpcs[3] = { 4, 1, 3 };     // the same for ProtoDUNE
    size_t apa_gap = 29;                // APA thickness (dead space in drift direction) in pixels

    bool goodEvent = false;
    unsigned int gd0 = 0, gd1 = 0;
        for (int p = 2; p >= 0; --p) // loop over planes and make global images
        {
            size_t drift_size = detProp.NumberTimeSamples() / fTrainingDataAlg.DriftWindow(); // tpc size in drift direction
            size_t wire_size = fGeometry->Nwires(p, 0, cryo);                                   // tpc size in wire direction
            size_t max_offset = (p < 2) ? 752 : 0;                                              // top-bottom projection max offset

        size_t ntotw = (tpcs[2] - 1) * weff[p] + wire_size + tpcs[1] * max_offset;          // global image size in wire direction
        size_t ntotd = tpcs[0] * drift_size + (apa_gap + max_align) * (tpcs[0] / 2);        // global image size in drift direction

            std::cout << "Plane: " << p << ", wires: " << ntotw << " drift: " << ntotd << std::endl;
            EventImageData fullimg(ntotw, ntotd, fSaveDepositMap);

        size_t a, maxArea = 0;
            for (size_t t = 0; t < fGeometry->NTPC(cryo); ++t) // loop over tpc's
            {
            size_t tpc_z = t / (tpcs[0] * tpcs[1]);
            size_t tpc_y = (t / tpcs[0]) % tpcs[1];
            size_t tpc_x = t % tpcs[0];

            int dir = fGeometry->TPC(t, cryo).DetectDriftDirection();
            bool flip_d = (dir > 0); // need the flip in drift direction?

                bool flip_w = false;     // need the flip in wire direction?
                size_t eff_p = p;        // global plane
                int offset = 0;
            int p_align = 0;

            if (p < 2) // no wire flip nor top-bottom alignments in Z (collection) plane
            {
                    if ((t % 4 == 0) || (t % 4 == 2))
                    {
                        eff_p = (p == 1) ? 0 : 1;     // swap U and V
                    }

                if ((t % 4 == 0) || (t % 4 == 1)) // bottom tpc's
                {
                    flip_w = (dir > 0) ? (eff_p == 1) : (eff_p == 0);
                }
                else                              // top tpc's
                {
                    flip_w = (dir < 0) ? (eff_p == 1) : (eff_p == 0);
                }

                offset = (p == 0) ? 48 : 752;     // top-bottopm offsets
            }
            
            if ((t % 4 == 0) || (t % 4 == 2))
            {
                p_align = plane_align0[p];
            }
            else
            {
                p_align = plane_align1[p];
            }

            size_t gw = tpc_z * weff[p] + tpc_y * offset;                          // global wire
            size_t gd = tpc_x * drift_size + apa_gap * (1 + tpc_x) / 2 + p_align;  // global drift

            fTrainingDataAlg.setEventData(event, clockData, detProp, eff_p, t, cryo); // prepare ADC and truth projections for 1 plane in 1 tpc
                    std::cout << "   TPC: " << t << " wires: " << fTrainingDataAlg.NWires() << std::endl;
                    a = fTrainingDataAlg.getAdcArea();
                    if (a > 150)
                    {
                        fullimg.addTpc(fTrainingDataAlg, gw, flip_w, gd, flip_d);
                        if (a > maxArea)
                        {
                    TVector2 vtxProj = GetProjVtx(detProp, fPointid.vtx3d, cryo, t, p);
                    fullimg.setProjXY(fTrainingDataAlg, vtxProj.X(), vtxProj.Y(), gw, flip_w, gd, flip_d);
                            a = maxArea;
                        }
                    }
                }

        std::cout << "   find crop..." << std::endl;
        unsigned int w0, w1, d0, d1;
                if (fullimg.findCrop(80, w0, w1, d0, d1))
                {
                    if (goodEvent)
                    {
                        d0 = gd0; d1 = gd1;
                    }
                    else
                    {
                        goodEvent = true;
                        gd0 = d0; gd1 = d1;
                    }
                        std::cout << "   crop: " << w0 << " " << w1 << " " << d0 << " " << d1 << std::endl;
                }
            else { std::cout << "   skip empty event" << std::endl; break; }

                std::ostringstream ss1;
                ss1 << os.str() << "_plane_" << p; // TH2's name

                art::ServiceHandle<art::TFileService> tfs;
                TH2C* rawHist = tfs->make<TH2C>((ss1.str() + "_raw").c_str(), "ADC",
                    (int)(w1 - w0), (double)w0, (double)w1, (int)(d1 - d0), (double)d0, (double)d1);
                
                float zero = fTrainingDataAlg.ZeroLevel();
        for (size_t w = w0; w < w1; ++w)
        {
            auto const & raw = fullimg.wireAdc(w);
            for (size_t d = d0; d < d1; ++d)
            {
                rawHist->Fill(w, d, (char)(raw[d] + zero));
            }
                }

                if (fSaveDepositMap)
                {
                TH2F* depHist = tfs->make<TH2F>((ss1.str() + "_deposit").c_str(), "Deposit",
                    (int)(w1 - w0), (double)w0, (double)w1, (int)(d1 - d0), (double)d0, (double)d1);
            for (size_t w = w0; w < w1; ++w)
            {
                auto const & edep = fullimg.wireDep(w);
                for (size_t d = d0; d < d1; ++d) { depHist->Fill(w, d, edep[d]); }
                }
        }

        if (fSavePdgMap)
        {
                TH2I* pdgHist = tfs->make<TH2I>((ss1.str() + "_pdg").c_str(), "PDG",
                    (int)(w1 - w0), (double)w0, (double)w1, (int)(d1 - d0), (double)d0, (double)d1);
            for (size_t w = w0; w < w1; ++w)
            {
                auto const & pdg = fullimg.wirePdg(w);
                for (size_t d = d0; d < d1; ++d) { pdgHist->Fill(w, d, pdg[d]); }
                }
                }

        if (goodEvent)
        {
                fPlane = p;

                if (fullimg.getVtxX() > -9999) fPixX = fullimg.getVtxX() - w0;
                if (fullimg.getVtxY() > -9999) fPixY = fullimg.getVtxY() - d0;

                if (fullimg.getProjX() > -9999) fPosX = fullimg.getProjX() - w0;
                if (fullimg.getProjY() > -9999) fPosY = fullimg.getProjY() - d0;

            std::cout << " *** plane:" << p << std::endl;
            std::cout << " ***   w0:" << w0 << ", w1:" << w1 << std::endl;
            std::cout << " ***   d0:" << d0 << ", d1:" << d1 << std::endl;
            std::cout << " ***  pix *** x:" << fPixX << ", y:" << fPixY << std::endl;
            std::cout << " *** proj *** x:" << fPosX << ", y:" << fPosY << std::endl;
            std::cout << " *** zero:" << zero << std::endl;

                fTree2D->Fill();
        }
        }
        if (goodEvent)
        {
                fPdg = fPointid.nupdg;
                fInteraction = fPointid.interaction;
            fTree->Fill();
        }
  }

void nnet::SPMultiTpcDump::beginJob ( )

overridevirtual

Reimplemented from art::EDAnalyzer.

Definition at line 135 of file SPMultiTpcDump_module.cc.

  {
                art::ServiceHandle<art::TFileService> tfs;

                fTree = tfs->make<TTree>("event","Event level info");
                fTree->Branch("fRun", &fRun, "fRun/I");
                fTree->Branch("fEvent", &fEvent, "fEvent/I");
                fTree->Branch("fCryo", &fCryo, "fCryo/I");
                fTree->Branch("fTpc", &fTpc, "fTpc/I");
                fTree->Branch("fPdg", &fPdg, "fPdg/I");
                fTree->Branch("fInteraction", &fInteraction, "fInteraction/I");
                
                fTree2D = tfs->make<TTree>("plane","Vertex 2D info");
                fTree2D->Branch("fRun", &fRun, "fRun/I");
                fTree2D->Branch("fEvent", &fEvent, "fEvent/I");
                fTree2D->Branch("fPlane", &fPlane, "fPlane/I");
                fTree2D->Branch("fPixX", &fPixX, "fPixX/I");
                fTree2D->Branch("fPixY", &fPixY, "fPixY/I");
                fTree2D->Branch("fPosX", &fPosX, "fPosX/F");
                fTree2D->Branch("fPosY", &fPosY, "fPosY/F");
  }

void nnet::SPMultiTpcDump::CorrOffset ( detinfo::DetectorPropertiesData const &  detProp, TVector3 &  vec, const simb::MCParticle &  particle  )

private

Definition at line 402 of file SPMultiTpcDump_module.cc.

  {
        double vtx[3] = {vec.X(), vec.Y(), vec.Z()};
        geo::TPCID tpcid = fGeometry->FindTPCAtPosition(vtx);

        if (tpcid.isValid)
          {
                float corrt0x = particle.T() * 1.e-3 * detProp.DriftVelocity();
                if (fGeometry->TPC(tpcid).DetectDriftDirection() == 1) { corrt0x = corrt0x*(-1); }
                
                vtx[0] = vec.X() + corrt0x;
          }

        vec.SetX(vtx[0]);
  }

TVector2 nnet::SPMultiTpcDump::GetProjVtx ( detinfo::DetectorPropertiesData const &  detProp, TVector3 const &  vtx3d, const size_t  cryo, const size_t  tpc, const size_t  plane  ) const

private

Definition at line 466 of file SPMultiTpcDump_module.cc.

  {

                TVector2 vtx2d = pma::GetProjectionToPlane(vtx3d, plane, tpc, cryo);
    TVector2 vtxwd = pma::CmToWireDrift(detProp, vtx2d.X(), vtx2d.Y(), plane, tpc, cryo);
        
                return vtxwd;
  }

bool nnet::SPMultiTpcDump::InsideFidVol ( TVector3 const &  pvtx ) const

private

Definition at line 420 of file SPMultiTpcDump_module.cc.

  {
                double vtx[3] = {pvtx.X(), pvtx.Y(), pvtx.Z()};
                bool inside = false;

                if (!fGeometry->FindTPCAtPosition(vtx).isValid) return false;

                geo::TPCID idtpc = fGeometry->FindTPCAtPosition(vtx);

                if (fGeometry->HasTPC(idtpc))
                {               
                        const geo::TPCGeo& tpcgeo = fGeometry->GetElement(idtpc);
                        double minx = tpcgeo.MinX(); double maxx = tpcgeo.MaxX();
                        double miny = tpcgeo.MinY(); double maxy = tpcgeo.MaxY();
                        double minz = tpcgeo.MinZ(); double maxz = tpcgeo.MaxZ();

                        //x
                        double dista = fabs(minx - pvtx.X());
                        double distb = fabs(pvtx.X() - maxx); 

                        if ((pvtx.X() > minx) && (pvtx.X() < maxx) &&
                        (dista > fFidVolCut) && (distb > fFidVolCut))
                        { 
                                inside = true;
                        }
                        else { inside = false; }

                        //y
                        dista = fabs(maxy - pvtx.Y());
                        distb = fabs(pvtx.Y() - miny);
                        if (inside && (pvtx.Y() > miny) && (pvtx.Y() < maxy) &&
                                (dista > fFidVolCut) && (distb > fFidVolCut)) inside = true;
                        else inside = false;

                        //z
                        dista = fabs(maxz - pvtx.Z());
                        distb = fabs(pvtx.Z() - minz);
                        if (inside && (pvtx.Z() > minz) && (pvtx.Z() < maxz) &&
                                (dista > fFidVolCut) && (distb > fFidVolCut)) inside = true;
                        else inside = false;
                }
                
                return inside;
  }

bool nnet::SPMultiTpcDump::prepareEv ( const art::Event &  event, detinfo::DetectorPropertiesData const &  detProp  )

private

Definition at line 158 of file SPMultiTpcDump_module.cc.

  {
        auto mctruthHandle = event.getHandle< std::vector<simb::MCTruth> >(fGenieGenLabel);
        if (!mctruthHandle) { return false; }

        for (auto const & mc : (*mctruthHandle))
        {
                if (mc.Origin() == simb::kBeamNeutrino)
                {
                        const TLorentzVector& pvtx = mc.GetNeutrino().Nu().Position();
                        TVector3 vtx(pvtx.X(), pvtx.Y(), pvtx.Z());

            CorrOffset(detProp, vtx, mc.GetNeutrino().Nu());

                        fPointid.nupdg = mc.GetNeutrino().Nu().PdgCode();
                        fPointid.interaction = mc.GetNeutrino().CCNC();
                        fPointid.vtx3d = vtx;

                        if (InsideFidVol(vtx)) 
                        {       
                                double v[3] = {vtx.X(), vtx.Y(), vtx.Z()}; 
                                unsigned int cryo = fGeometry->FindCryostatAtPosition(v);
                                unsigned int tpc = fGeometry->FindTPCAtPosition(v).TPC;
                                
                                for (size_t i = 0; i < 3; ++i)
                                {
                                        if (fGeometry->TPC(tpc, cryo).HasPlane(i))
                      { fPointid.position[i] = GetProjVtx(detProp, vtx, cryo, tpc, i); }
                                        else
                                        { fPointid.position[i] = TVector2(0, 0); }
                                }

                                fPointid.cryo = cryo;
                                fPointid.tpc = tpc;
                                
                                fCryo = cryo;
                                fTpc = tpc;
                                
                                return true;    
                        }
                        else
                        {
                                fPointid.cryo = -1;
                                fPointid.tpc = -1;
                                fCryo = -1; fTpc = -1;
                            return false;
                        }
                }
        }
        return false;
  }  

void nnet::SPMultiTpcDump::ResetVars ( )

private

Definition at line 476 of file SPMultiTpcDump_module.cc.

        {
                fCryo = 0;
                fTpc = 0;
                fPlane = 0;
                fPdg = 0;
                fInteraction = 0;
                fPosX = 0.0;
                fPosY = 0.0;
                fPixX = 0;
                fPixY = 0;
        }

Member Data Documentation

int nnet::SPMultiTpcDump::fCryo

private

Definition at line 115 of file SPMultiTpcDump_module.cc.

int nnet::SPMultiTpcDump::fEvent

private

number of the event being processed

Definition at line 106 of file SPMultiTpcDump_module.cc.

double nnet::SPMultiTpcDump::fFidVolCut

private

Definition at line 112 of file SPMultiTpcDump_module.cc.

art::InputTag nnet::SPMultiTpcDump::fGenieGenLabel

private

Definition at line 102 of file SPMultiTpcDump_module.cc.

geo::GeometryCore const* nnet::SPMultiTpcDump::fGeometry

private

Definition at line 120 of file SPMultiTpcDump_module.cc.

int nnet::SPMultiTpcDump::fInteraction

private

Definition at line 116 of file SPMultiTpcDump_module.cc.

int nnet::SPMultiTpcDump::fPdg

private

Definition at line 116 of file SPMultiTpcDump_module.cc.

int nnet::SPMultiTpcDump::fPixX

private

Definition at line 117 of file SPMultiTpcDump_module.cc.

int nnet::SPMultiTpcDump::fPixY

private

Definition at line 117 of file SPMultiTpcDump_module.cc.

int nnet::SPMultiTpcDump::fPlane

private

Definition at line 115 of file SPMultiTpcDump_module.cc.

NUVTX nnet::SPMultiTpcDump::fPointid

private

Definition at line 114 of file SPMultiTpcDump_module.cc.

float nnet::SPMultiTpcDump::fPosX

private

Definition at line 118 of file SPMultiTpcDump_module.cc.

float nnet::SPMultiTpcDump::fPosY

private

Definition at line 118 of file SPMultiTpcDump_module.cc.

int nnet::SPMultiTpcDump::fRun

private

number of the run being processed

Definition at line 107 of file SPMultiTpcDump_module.cc.

bool nnet::SPMultiTpcDump::fSaveDepositMap

private

Definition at line 110 of file SPMultiTpcDump_module.cc.

bool nnet::SPMultiTpcDump::fSavePdgMap

private

Definition at line 110 of file SPMultiTpcDump_module.cc.

int nnet::SPMultiTpcDump::fSubRun

private

number of the sub-run being processed

Definition at line 108 of file SPMultiTpcDump_module.cc.

int nnet::SPMultiTpcDump::fTpc

private

Definition at line 115 of file SPMultiTpcDump_module.cc.

nnet::TrainingDataAlg nnet::SPMultiTpcDump::fTrainingDataAlg

private

Definition at line 101 of file SPMultiTpcDump_module.cc.

TTree* nnet::SPMultiTpcDump::fTree

private

Definition at line 104 of file SPMultiTpcDump_module.cc.

TTree* nnet::SPMultiTpcDump::fTree2D

private

Definition at line 105 of file SPMultiTpcDump_module.cc.

---

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

• dunereco/dunereco/CVN/adcutils/SPMultiTpcDump_module.cc