nnet::TrainingDataAlg Class Reference

#include "PointIdAlg.h"

Inheritance diagram for nnet::TrainingDataAlg:

Classes
struct	Config
struct	WireDrift

Public Types
enum	EMask { kNone = 0, kPdgMask = 0x00000FFF, kTypeMask = 0x0000F000, kVtxMask = 0xFFFF0000 }
enum	ETrkType { kDelta = 0x1000, kMichel = 0x2000, kPriEl = 0x4000, kPriMu = 0x8000 }
enum	EVtxId {   kNuNC = 0x0010000, kNuCC = 0x0020000, kNuPri = 0x0040000, kNuE = 0x0100000,   kNuMu = 0x0200000, kNuTau = 0x0400000, kHadr = 0x1000000, kPi0 = 0x2000000,   kDecay = 0x4000000, kConv = 0x8000000, kElectronEnd = 0x10000000 }
enum	EDownscaleMode { kMax = 1, kMaxMean = 2, kMean = 3 }

Public Member Functions
	TrainingDataAlg (const fhicl::ParameterSet &pset)
	TrainingDataAlg (const Config &config)
	~TrainingDataAlg (void) override
void	reconfigure (const Config &config)
bool	saveSimInfo () const
bool	setEventData (const art::Event &event, unsigned int plane, unsigned int tpc, unsigned int cryo)
bool	setDataEventData (const art::Event &event, unsigned int plane, unsigned int tpc, unsigned int cryo)
bool	findCrop (float max_e_cut, unsigned int &w0, unsigned int &w1, unsigned int &d0, unsigned int &d1) const
double	getEdepTot (void) const
std::vector< float > const &	wireEdep (size_t widx) const
std::vector< int > const &	wirePdg (size_t widx) const
bool	setWireDriftData (const std::vector< recob::Wire > &wires, unsigned int plane, unsigned int tpc, unsigned int cryo)
std::vector< float > const &	wireData (size_t widx) const
std::vector< std::vector< float > >	getPatch (size_t wire, float drift, size_t patchSizeW, size_t patchSizeD) const
float	getPixelOrZero (int wire, int drift) const
double	getAdcSum (void) const
size_t	getAdcArea (void) const
float	poolMax (int wire, int drift, size_t r=0) const
	Pool max value in a patch around the wire/drift pixel. More...
float	poolSum (int wire, int drift, size_t r=0) const
	Pool sum of pixels in a patch around the wire/drift pixel. More...
unsigned int	Cryo (void) const
unsigned int	TPC (void) const
unsigned int	Plane (void) const
unsigned int	NWires (void) const
unsigned int	NScaledDrifts (void) const
unsigned int	NCachedDrifts (void) const
unsigned int	DriftWindow (void) const
float	ZeroLevel (void) const
	Level of zero ADC after scaling. More...
double	LifetimeCorrection (double tick) const

Protected Member Functions
void	resizeView (size_t wires, size_t drifts) override
void	downscaleMax (std::vector< float > &dst, std::vector< float > const &adc, size_t tick0) const
void	downscaleMaxMean (std::vector< float > &dst, std::vector< float > const &adc, size_t tick0) const
void	downscaleMean (std::vector< float > &dst, std::vector< float > const &adc, size_t tick0) const
void	downscale (std::vector< float > &dst, std::vector< float > const &adc, size_t tick0) const
size_t	getDriftIndex (float drift) const
bool	setWireData (std::vector< float > const &adc, size_t wireIdx)
bool	patchFromDownsampledView (size_t wire, float drift, size_t size_w, size_t size_d, std::vector< std::vector< float > > &patch) const
bool	patchFromOriginalView (size_t wire, float drift, size_t size_w, size_t size_d, std::vector< std::vector< float > > &patch) const

Protected Attributes
unsigned int	fCryo
unsigned int	fTPC
unsigned int	fPlane
unsigned int	fNWires
unsigned int	fNDrifts
unsigned int	fNScaledDrifts
unsigned int	fNCachedDrifts
std::vector< raw::ChannelID_t >	fWireChannels
std::vector< std::vector< float > >	fWireDriftData
std::vector< float >	fLifetimeCorrFactors
EDownscaleMode	fDownscaleMode
size_t	fDriftWindow
bool	fDownscaleFullView
float	fDriftWindowInv
calo::CalorimetryAlg	fCalorimetryAlg
geo::GeometryCore const *	fGeometry
detinfo::DetectorProperties const *	fDetProp

Private Member Functions
WireDrift	getProjection (const TLorentzVector &tvec, unsigned int plane) const
bool	setWireEdepsAndLabels (std::vector< float > const &edeps, std::vector< int > const &pdgs, size_t wireIdx)
void	collectVtxFlags (std::unordered_map< size_t, std::unordered_map< int, int > > &wireToDriftToVtxFlags, const std::unordered_map< int, const simb::MCParticle * > &particleMap, unsigned int plane) const
bool	isElectronEnd (const simb::MCParticle &particle, const std::unordered_map< int, const simb::MCParticle * > &particleMap) const
bool	isMuonDecaying (const simb::MCParticle &particle, const std::unordered_map< int, const simb::MCParticle * > &particleMap) const

Static Private Member Functions
static float	particleRange2 (const simb::MCParticle &particle)

Private Attributes
double	fEdepTot
std::vector< std::vector< float > >	fWireDriftEdep
std::vector< std::vector< int > >	fWireDriftPdg
art::InputTag	fWireProducerLabel
art::InputTag	fHitProducerLabel
art::InputTag	fTrackModuleLabel
art::InputTag	fSimulationProducerLabel
bool	fSaveVtxFlags
bool	fSaveSimInfo
unsigned int	fAdcDelay
std::vector< size_t >	fEventsPerBin

Detailed Description

Definition at line 187 of file PointIdAlg.h.

Member Enumeration Documentation

enum img::DataProviderAlg::EDownscaleMode

inherited

Enumerator
kMax
kMaxMean
kMean

Definition at line 49 of file DataProviderAlg.h.

{ kMax = 1, kMaxMean = 2, kMean = 3 };

enum nnet::TrainingDataAlg::EMask

Enumerator
kNone
kPdgMask
kTypeMask
kVtxMask

Definition at line 191 of file PointIdAlg.h.

    {
        kNone     = 0,
        kPdgMask  = 0x00000FFF, // pdg code mask
        kTypeMask = 0x0000F000, // track type mask
        kVtxMask  = 0xFFFF0000  // vertex flags
    };

enum nnet::TrainingDataAlg::ETrkType

Enumerator
kDelta
kMichel
kPriEl
kPriMu

Definition at line 199 of file PointIdAlg.h.

    {
        kDelta  = 0x1000,      // delta electron
        kMichel = 0x2000,      // Michel electron
        kPriEl  = 0x4000,      // primary electron
        kPriMu  = 0x8000       // primary muon
    };

enum nnet::TrainingDataAlg::EVtxId

Enumerator
kNuNC
kNuCC
kNuPri
kNuE
kNuMu
kNuTau
kHadr
kPi0
kDecay
kConv
kElectronEnd

Definition at line 207 of file PointIdAlg.h.

        {
                kNuNC  = 0x0010000, kNuCC = 0x0020000, kNuPri = 0x0040000,  // nu interaction type
                kNuE   = 0x0100000, kNuMu = 0x0200000, kNuTau = 0x0400000,  // nu flavor
                kHadr  = 0x1000000,       // hadronic inelastic scattering
                kPi0   = 0x2000000,       // pi0 produced in this vertex
                kDecay = 0x4000000,       // point of particle decay
                kConv  = 0x8000000,       // gamma conversion
                kElectronEnd = 0x10000000 // clear end of an electron
        };

Constructor & Destructor Documentation

nnet::TrainingDataAlg::TrainingDataAlg ( const fhicl::ParameterSet &  pset )

inline

Definition at line 254 of file PointIdAlg.h.

                                                       :
                TrainingDataAlg(fhicl::Table<Config>(pset, {})())
        {}

nnet::TrainingDataAlg::TrainingDataAlg

(

const Config &

config

)

Definition at line 343 of file PointIdAlg.cxx.

References fSaveSimInfo, fSimulationProducerLabel, and art::InputTag::label().

                                                         : img::DataProviderAlg(config),
        fEdepTot(0),
        fWireProducerLabel(config.WireLabel()),
        fHitProducerLabel(config.HitLabel()),
        fTrackModuleLabel(config.TrackLabel()),
        fSimulationProducerLabel(config.SimulationLabel()),
        fSaveVtxFlags(config.SaveVtxFlags()),
    fAdcDelay(config.AdcDelayTicks()),
    fEventsPerBin(100, 0)
{
    fSaveSimInfo = !fSimulationProducerLabel.label().empty();
}

nnet::TrainingDataAlg::~TrainingDataAlg ( void  )

override

Definition at line 357 of file PointIdAlg.cxx.

{
}

Member Function Documentation

void nnet::TrainingDataAlg::collectVtxFlags ( std::unordered_map< size_t, std::unordered_map< int, int > > &  wireToDriftToVtxFlags, const std::unordered_map< int, const simb::MCParticle * > &  particleMap, unsigned int  plane  ) const

private

Definition at line 553 of file PointIdAlg.cxx.

References d, img::DataProviderAlg::fCryo, img::DataProviderAlg::fTPC, getProjection(), isElectronEnd(), isMuonDecaying(), kConv, kDecay, kElectronEnd, kHadr, kNone, kNuPri, and kPi0.

Referenced by setEventData().

{
        for (auto const & p : particleMap)
        {
                auto const & particle = *p.second;

                double ekStart = 1000. * (particle.E() - particle.Mass());
                double ekEnd = 1000. * (particle.EndE() - particle.Mass());

                int pdg = abs(particle.PdgCode());
                int flagsStart = nnet::TrainingDataAlg::kNone;
                int flagsEnd = nnet::TrainingDataAlg::kNone;
        
                switch (pdg)
                {
                        case 22:   // gamma
                                if ((particle.EndProcess() == "conv") &&
                                    (ekStart > 40.0)) // conversion, gamma > 40MeV
                                {
                                        //std::cout << "---> gamma conversion at " << ekStart << std::endl;
                                        flagsEnd = nnet::TrainingDataAlg::kConv;
                                }
                                break;

                        case 11:   // e+/-
                            if (isElectronEnd(particle, particleMap))
                            {
                                flagsEnd = nnet::TrainingDataAlg::kElectronEnd;
                            }
                                break;


                        case 13:   // mu+/-
                            if (isMuonDecaying(particle, particleMap))
                            {
                                //std::cout << "---> mu decay to electron" << std::endl;
                                flagsEnd = nnet::TrainingDataAlg::kDecay;
                            }
                                break;

                        case 111:  // pi0
                                //std::cout << "---> pi0" << std::endl;
                                flagsStart = nnet::TrainingDataAlg::kPi0;
                                break;

                        case 321:  // K+/-
                        case 211:  // pi+/-
                        case 2212: // proton
                                if (ekStart > 50.0)
                                {
                                        if (particle.Mother() != 0)
                                        {
                                                auto search = particleMap.find(particle.Mother());
                                                if (search != particleMap.end())
                                                {
                                                        auto const & mother = *((*search).second);
                                                        int m_pdg = abs(mother.PdgCode());
                                                        unsigned int nSec = mother.NumberDaughters();
                                                        unsigned int nVisible = 0;
                                                        if (nSec > 1)
                                                        {
                                                                for (size_t d = 0; d < nSec; ++d)
                                                                {
                                                                        auto d_search = particleMap.find(mother.Daughter(d));
                                                                        if (d_search != particleMap.end())
                                                                        {
                                                                                auto const & daughter = *((*d_search).second);
                                                                                int d_pdg = abs(daughter.PdgCode());
                                                                                if (((d_pdg == 2212) || (d_pdg == 211) || (d_pdg == 321)) &&
                                                                                    (1000. * (daughter.E() - daughter.Mass()) > 50.0))
                                                                                {
                                                                                        ++nVisible;
                                                                                }
                                                                        }
                                                                }
                                                        }
                                                        // hadron with Ek > 50MeV (so well visible) and
                                                        // produced by another hadron (but not neutron, so not single track from nothing) or
                                                        // at least secondary hadrons with Ek > 50MeV (so this is a good kink or V-like)
                                                        if (((m_pdg != pdg) && (m_pdg != 2112)) || ((m_pdg != 2112) && (nVisible > 0)) || ((m_pdg == 2112) && (nVisible > 1)))
                                                        {
                                                                // std::cout << "---> hadron at " << ekStart
                                                                //      << ", pdg: " << pdg << ", mother pdg: " << m_pdg
                                                                //      << ", vis.daughters: " << nVisible << std::endl;
                                                                flagsStart = nnet::TrainingDataAlg::kHadr;
                                                        } 
                                                }
                                                // else std::cout << "---> mother not found for tid: " << particle.Mother() << std::endl;
                                        }

                                        if (particle.EndProcess() == "FastScintillation") // potential decay at rest
                                        {
                                                unsigned int nSec = particle.NumberDaughters();
                                                for (size_t d = 0; d < nSec; ++d)
                                                {
                                                        auto d_search = particleMap.find(particle.Daughter(d));
                                                        if (d_search != particleMap.end())
                                                        {
                                                                auto const & daughter = *((*d_search).second);
                                                                int d_pdg = abs(daughter.PdgCode());
                                                                if ((pdg == 321) && (d_pdg == 13))
                                                                {
                                                                        //std::cout << "---> K decay to mu" << std::endl;
                                                                        flagsEnd = nnet::TrainingDataAlg::kDecay;
                                                                        break;
                                                                }
                                                                if ((pdg == 211) && (d_pdg == 13))
                                                                {
                                                                        //std::cout << "---> pi decay to mu" << std::endl;
                                                                        flagsEnd = nnet::TrainingDataAlg::kDecay;
                                                                        break;
                                                                }
                                                        }
                                                }
                                        }

                                        if ((particle.EndProcess() == "Decay") && (ekEnd > 200.0)) // decay in flight
                                        {
                                                unsigned int nSec = particle.NumberDaughters();
                                                for (size_t d = 0; d < nSec; ++d)
                                                {
                                                        auto d_search = particleMap.find(particle.Daughter(d));
                                                        if (d_search != particleMap.end())
                                                        {
                                                                auto const & daughter = *((*d_search).second);
                                                                int d_pdg = abs(daughter.PdgCode());
                                                                if ((pdg == 321) && (d_pdg == 13))
                                                                {
                                                                        //std::cout << "---> in-flight K decay to mu" << std::endl;
                                                                        flagsEnd = nnet::TrainingDataAlg::kHadr;
                                                                        break;
                                                                }
                                                                if ((pdg == 211) && (d_pdg == 13))
                                                                {
                                                                        //std::cout << "---> in-flight pi decay to mu" << std::endl;
                                                                        flagsEnd = nnet::TrainingDataAlg::kHadr;
                                                                        break;
                                                                }
                                                        }
                                                }
                                        }
                                }
                                break;

                        default: continue;
                }
                
                if (particle.Process() == "primary")
                {
                        flagsStart |= nnet::TrainingDataAlg::kNuPri;
                }
                
                
                if (flagsStart != nnet::TrainingDataAlg::kNone)
                {
                        auto wd = getProjection(particle.Position(), plane);
                        
                        if ((wd.TPC == (int)fTPC) && (wd.Cryo == (int)fCryo))
                        {
                                wireToDriftToVtxFlags[wd.Wire][wd.Drift] |= flagsStart;
                                // std::cout << "---> flagsStart:" << flagsStart << " plane:" << plane << " wire:" << wd.Wire << " drift:" << wd.Drift << std::endl;
                        }
                        // else std::cout << "---> not in current TPC" << std::endl;
                }
                if (flagsEnd != nnet::TrainingDataAlg::kNone)
                {
                        auto wd = getProjection(particle.EndPosition(), plane);
                        if ((wd.TPC == (int)fTPC) && (wd.Cryo == (int)fCryo))
                        {
                            //if (flagsEnd == nnet::TrainingDataAlg::kElectronEnd) { std::cout << "---> clear electron endpoint" << std::endl; }
                                wireToDriftToVtxFlags[wd.Wire][wd.Drift] |= flagsEnd;
                                //if (flagsEnd == nnet::TrainingDataAlg::kElectronEnd)
                                //    std::cout << "---> flagsEnd:" << flagsEnd << " plane:" << plane << " wire:" << wd.Wire << " drift:" << wd.Drift << std::endl;
                        }
                        // else std::cout << "---> not in current TPC" << std::endl;
                }

                //if (ekStart > 30.0)
                //{
                //      std::cout << particle.PdgCode() << ", " << ekStart << ": "
                //              << particle.Process() << " --> " << particle.EndProcess()
                //              << " " << ekEnd << std::endl;
                //}
        }
}

unsigned int img::DataProviderAlg::Cryo ( void  ) const

inlineinherited

Definition at line 157 of file DataProviderAlg.h.

References img::DataProviderAlg::fCryo.

{ return fCryo; }

void img::DataProviderAlg::downscale ( std::vector< float > &  dst, std::vector< float > const &  adc, size_t  tick0  ) const

inlineprotectedinherited

Definition at line 189 of file DataProviderAlg.h.

References img::DataProviderAlg::downscaleMax(), img::DataProviderAlg::downscaleMaxMean(), img::DataProviderAlg::downscaleMean(), img::DataProviderAlg::kMax, img::DataProviderAlg::kMaxMean, and img::DataProviderAlg::kMean.

Referenced by img::DataProviderAlg::patchFromOriginalView(), and img::DataProviderAlg::setWireData().

        {
            switch (fDownscaleMode)
            {
                case img::DataProviderAlg::kMean: downscaleMean(dst, adc, tick0); break;
                case img::DataProviderAlg::kMaxMean: downscaleMaxMean(dst, adc, tick0); break;
                case img::DataProviderAlg::kMax: downscaleMax(dst, adc, tick0); break;
                default:throw cet::exception("img::DataProviderAlg") << "Downscale mode not supported." << std::endl; break;
            }
        }

void img::DataProviderAlg::downscaleMax ( std::vector< float > &  dst, std::vector< float > const &  adc, size_t  tick0  ) const

protectedinherited

Definition at line 182 of file DataProviderAlg.cxx.

References img::DataProviderAlg::fDriftWindow, img::DataProviderAlg::fLifetimeCorrFactors, and img::DataProviderAlg::scaleAdcSamples().

Referenced by img::DataProviderAlg::downscale().

{
        size_t kStop = dst.size();
        if (adc.size() < kStop) { kStop = adc.size(); }
        for (size_t i = 0, k0 = 0; i < kStop; ++i, k0 += fDriftWindow)
        {
                size_t k1 = k0 + fDriftWindow;

                float max_adc = adc[k0] * fLifetimeCorrFactors[k0 + tick0];
                for (size_t k = k0 + 1; k < k1; ++k)
                {
                        float ak = adc[k] * fLifetimeCorrFactors[k + tick0];
                        if (ak > max_adc) max_adc = ak;
                }

                dst[i] = max_adc;
        }
        scaleAdcSamples(dst);
}

void img::DataProviderAlg::downscaleMaxMean ( std::vector< float > &  dst, std::vector< float > const &  adc, size_t  tick0  ) const

protectedinherited

Definition at line 202 of file DataProviderAlg.cxx.

References img::DataProviderAlg::fDriftWindow, img::DataProviderAlg::fLifetimeCorrFactors, n, and img::DataProviderAlg::scaleAdcSamples().

Referenced by img::DataProviderAlg::downscale().

{
        size_t kStop = dst.size();
        if (adc.size() < kStop) { kStop = adc.size(); }
        for (size_t i = 0, k0 = 0; i < kStop; ++i, k0 += fDriftWindow)
        {
                size_t k1 = k0 + fDriftWindow;

                size_t max_idx = k0;
                float max_adc = adc[k0] * fLifetimeCorrFactors[k0 + tick0];
                for (size_t k = k0 + 1; k < k1; ++k)
                {
                        float ak = adc[k] * fLifetimeCorrFactors[k + tick0];
                        if (ak > max_adc) { max_adc = ak; max_idx = k; }
                }

                size_t n = 1;
                if (max_idx > 0) { max_adc += adc[max_idx - 1] * fLifetimeCorrFactors[max_idx - 1 + tick0]; n++; }
                if (max_idx + 1 < adc.size()) { max_adc += adc[max_idx + 1] * fLifetimeCorrFactors[max_idx + 1 + tick0]; n++; }

                dst[i] = max_adc / n;
        }
        scaleAdcSamples(dst);
}

void img::DataProviderAlg::downscaleMean ( std::vector< float > &  dst, std::vector< float > const &  adc, size_t  tick0  ) const

protectedinherited

Definition at line 227 of file DataProviderAlg.cxx.

References img::DataProviderAlg::fDriftWindow, img::DataProviderAlg::fDriftWindowInv, img::DataProviderAlg::fLifetimeCorrFactors, and img::DataProviderAlg::scaleAdcSamples().

Referenced by img::DataProviderAlg::downscale().

{
        size_t kStop = dst.size();
        if (adc.size() < kStop) { kStop = adc.size(); }
        for (size_t i = 0, k0 = 0; i < kStop; ++i, k0 += fDriftWindow)
        {
                size_t k1 = k0 + fDriftWindow;

                float sum_adc = 0;
                for (size_t k = k0; k < k1; ++k)
                {
                        sum_adc += adc[k] * fLifetimeCorrFactors[k + tick0];
                }
        dst[i] = sum_adc * fDriftWindowInv;
        }
        scaleAdcSamples(dst);
}

unsigned int img::DataProviderAlg::DriftWindow ( void  ) const

inlineinherited

Definition at line 164 of file DataProviderAlg.h.

References img::DataProviderAlg::fDriftWindow.

{ return fDriftWindow; }

bool nnet::TrainingDataAlg::findCrop	(	float	max_e_cut,
		unsigned int &	w0,
		unsigned int &	w1,
		unsigned int &	d0,
		unsigned int &	d1
	)		const

Definition at line 1063 of file PointIdAlg.cxx.

References d, and fWireDriftEdep.

{
    if (fWireDriftEdep.empty() || fWireDriftEdep.front().empty()) return false;

    float max_cut = 0.25 * max_e_cut;

    w0 = 0;
    float cut = 0;
    while (w0 < fWireDriftEdep.size())
    {
        for (auto const d : fWireDriftEdep[w0]) cut += d;
        if (cut < max_cut) w0++;
        else break;
    }
    w1 = fWireDriftEdep.size() - 1;
    cut = 0;
    while (w1 > w0)
    {
        for (auto const d : fWireDriftEdep[w1]) cut += d;
        if (cut < max_cut) w1--;
        else break;
    }
    w1++;

    d0 = 0;
    cut = 0;
    while (d0 < fWireDriftEdep.front().size())
    {
        for (size_t i = w0; i < w1; ++i) cut += fWireDriftEdep[i][d0];
        if (cut < max_cut) d0++;
        else break;
    }
    d1 = fWireDriftEdep.front().size() - 1;
    cut = 0;
    while (d1 > d0)
    {
        for (size_t i = w0; i < w1; ++i) cut += fWireDriftEdep[i][d1];
        if (cut < max_cut) d1--;
        else break;
    }
    d1++;

    unsigned int margin = 20;
    if ((w1 - w0 > 8) && (d1 - d0 > 8))
    {
        if (w0 < margin) w0 = 0;
        else w0 -= margin;

        if (w1 > fWireDriftEdep.size() - margin) w1 = fWireDriftEdep.size();
        else w1 += margin;
        
        if (d0 < margin) d0 = 0;
        else d0 -= margin;
        
        if (d1 > fWireDriftEdep.front().size() - margin) d1 = fWireDriftEdep.front().size();
        else d1 += margin;
        
        return true;
    }
    else return false;
}

size_t img::DataProviderAlg::getAdcArea ( void  ) const

inlineinherited

Definition at line 149 of file DataProviderAlg.h.

References img::DataProviderAlg::fAdcAreaOverThr, img::DataProviderAlg::poolMax(), and img::DataProviderAlg::poolSum().

{ return fAdcAreaOverThr; }

double img::DataProviderAlg::getAdcSum ( void  ) const

inlineinherited

Definition at line 148 of file DataProviderAlg.h.

References img::DataProviderAlg::fAdcSumOverThr.

{ return fAdcSumOverThr; }

size_t img::DataProviderAlg::getDriftIndex ( float  drift ) const

inlineprotectedinherited

Definition at line 200 of file DataProviderAlg.h.

References img::DataProviderAlg::fDriftWindowInv, img::DataProviderAlg::patchFromDownsampledView(), img::DataProviderAlg::patchFromOriginalView(), img::DataProviderAlg::resizeView(), img::DataProviderAlg::setWireData(), and lar::dump::vector().

Referenced by img::DataProviderAlg::getPixelOrZero(), img::DataProviderAlg::poolMax(), and img::DataProviderAlg::poolSum().

    {
        if (fDownscaleFullView) return (size_t)(drift * fDriftWindowInv);
        else return (size_t)drift;
    }

double nnet::TrainingDataAlg::getEdepTot ( void  ) const

inline

Definition at line 275 of file PointIdAlg.h.

{ return fEdepTot; } // [GeV]

std::vector< std::vector<float> > img::DataProviderAlg::getPatch ( size_t  wire, float  drift, size_t  patchSizeW, size_t  patchSizeD  ) const

inlineinherited

Return patch of data centered on the wire and drift, witht the size in (downscaled) pixels givent with patchSizeW and patchSizeD. Pad with the zero-level calue if patch extends beyond the event projection.

Definition at line 114 of file DataProviderAlg.h.

References img::DataProviderAlg::fDownscaleFullView, img::DataProviderAlg::patchFromDownsampledView(), and img::DataProviderAlg::patchFromOriginalView().

        {
                bool ok = false;
                std::vector< std::vector<float> > patch;
                if (fDownscaleFullView)
                {
                        ok = patchFromDownsampledView(wire, drift, patchSizeW, patchSizeD, patch);
                }
                else
                {
                        ok = patchFromOriginalView(wire, drift, patchSizeW, patchSizeD, patch);
                }

                if (ok) return patch;
                else
                {
                        throw cet::exception("img::DataProviderAlg") << "Patch filling failed." << std::endl;
                }
        }

float img::DataProviderAlg::getPixelOrZero ( int  wire, int  drift  ) const

inlineinherited

Return value from the ADC buffer, or zero if coordinates are out of the view; will scale the drift according to the downscale settings.

Definition at line 136 of file DataProviderAlg.h.

References img::DataProviderAlg::fNCachedDrifts, img::DataProviderAlg::fWireDriftData, and img::DataProviderAlg::getDriftIndex().

    {
        size_t didx = getDriftIndex(drift), widx = (size_t)wire;

        if ((widx < fWireDriftData.size()) &&
            (didx < fNCachedDrifts))
        {
            return fWireDriftData[widx][didx];
        }
        else { return 0; }
    }

nnet::TrainingDataAlg::WireDrift nnet::TrainingDataAlg::getProjection ( const TLorentzVector &  tvec, unsigned int  plane  ) const

private

Definition at line 427 of file PointIdAlg.cxx.

References detinfo::DetectorProperties::ConvertXToTicks(), geo::CryostatID::Cryostat, geo::TPCGeo::DetectDriftDirection(), e, img::DataProviderAlg::fDetProp, img::DataProviderAlg::fGeometry, geo::GeometryCore::FindTPCAtPosition(), geo::CryostatID::isValid, geo::GeometryCore::NearestWire(), geo::TPCID::TPC, geo::GeometryCore::TPC(), and nnet::TrainingDataAlg::WireDrift::Wire.

Referenced by collectVtxFlags().

{
        auto const* detprop = lar::providerFrom<detinfo::DetectorPropertiesService>();
        nnet::TrainingDataAlg::WireDrift wd;
        wd.Wire = 0; wd.Drift = 0; wd.TPC = -1; wd.Cryo = -1;

    try
    {
        double vtx[3] = {tvec.X(), tvec.Y(), tvec.Z()};
            if (fGeometry->FindTPCAtPosition(vtx).isValid)
            {
                geo::TPCID tpcid = fGeometry->FindTPCAtPosition(vtx);
                unsigned int tpc = tpcid.TPC, cryo = tpcid.Cryostat;

                // correct for the time offset
                float dx = tvec.T() * 1.e-3 * detprop->DriftVelocity();
                int driftDir = fGeometry->TPC(tpcid).DetectDriftDirection();
                        if (driftDir == 1) { dx *= -1; }
                        else if (driftDir != -1)
                        {
                            throw cet::exception("nnet::TrainingDataAlg") << "drift direction is not X." << std::endl;
                        }
                        vtx[0] = tvec.X() + dx;
                                
                wd.Wire = fGeometry->NearestWire(vtx, plane, tpc, cryo);
                wd.Drift = fDetProp->ConvertXToTicks(vtx[0], plane, tpc, cryo);
                wd.TPC = tpc; wd.Cryo = cryo;
            }
        }
        catch (const geo::InvalidWireIDError & e)
        {
            mf::LogWarning("TrainingDataAlg") << "Vertex projection out of wire planes, just skipping this vertex.";
        }
        catch (...)
        {
            mf::LogWarning("TrainingDataAlg") << "Vertex projection out of wire planes, skip MC vertex.";
        }
        return wd;
}

bool nnet::TrainingDataAlg::isElectronEnd ( const simb::MCParticle &  particle, const std::unordered_map< int, const simb::MCParticle * > &  particleMap  ) const

private

Definition at line 468 of file PointIdAlg.cxx.

References d, simb::MCParticle::Daughter(), simb::MCParticle::NumberDaughters(), particleRange2(), and simb::MCParticle::PdgCode().

Referenced by collectVtxFlags().

{
    const float minElectronLength2 = 2.5*2.5;
    const float maxDeltaLength2 = 0.15*0.15;

    int pdg = abs(particle.PdgCode());
        if (pdg != 11) return false; // should be applied only to electrons

        size_t nSec = particle.NumberDaughters();
        for (size_t d = 0; d < nSec; ++d)
        {
                auto d_search = particleMap.find(particle.Daughter(d));
                if (d_search != particleMap.end())
                {
                        auto const & daughter = *((*d_search).second);
                        int d_pdg = abs(daughter.PdgCode());
                        if (d_pdg != 22) { return false; } // not the end of the shower
                }
        }

    float trkLength2 = 0;
        auto const * p = &particle;
        bool branching = false;
        while (!branching)
        {
        trkLength2 += particleRange2(*p);
        auto m_search = particleMap.find(p->Mother());
                if (m_search != particleMap.end())
        {
                        p = (*m_search).second;
                        int m_pdg = abs(p->PdgCode());
                        if (m_pdg == 11)
                        {
                            nSec = p->NumberDaughters();
                            size_t ne = 0;
                            for (size_t d = 0; d < nSec; ++d)
                            {
                                auto d_search = particleMap.find(p->Daughter(d));
                                if (d_search != particleMap.end())
                                {
                                    auto const & daughter = *((*d_search).second);
                                    int d_pdg = abs(daughter.PdgCode());
                                    if (d_pdg == 11)
                                    {
                                if (particleRange2(daughter) > maxDeltaLength2) { ne++; }
                                    }
                                }
                            }
                            if (ne > 1) { branching = true; }
                        }
                        else break;
        }
        else break;
    }

    return (trkLength2 > minElectronLength2);
}

bool nnet::TrainingDataAlg::isMuonDecaying ( const simb::MCParticle &  particle, const std::unordered_map< int, const simb::MCParticle * > &  particleMap  ) const

private

Definition at line 527 of file PointIdAlg.cxx.

References d, simb::MCParticle::Daughter(), simb::MCParticle::EndProcess(), simb::MCParticle::NumberDaughters(), and simb::MCParticle::PdgCode().

Referenced by collectVtxFlags(), and setEventData().

{
    bool hasElectron = false, hasNuMu = false, hasNuE = false;

    int pdg = abs(particle.PdgCode());
        if ((pdg == 13) && (particle.EndProcess() == "FastScintillation")) // potential muon decay at rest
        {
                unsigned int nSec = particle.NumberDaughters();
                for (size_t d = 0; d < nSec; ++d)
                {
                        auto d_search = particleMap.find(particle.Daughter(d));
                        if (d_search != particleMap.end())
                        {
                                auto const & daughter = *((*d_search).second);
                                int d_pdg = abs(daughter.PdgCode());
                                if (d_pdg == 11) hasElectron = true;
                                else if (d_pdg == 14) hasNuMu = true;
                                else if (d_pdg == 12) hasNuE = true;
                        }
                }
        }

        return (hasElectron && hasNuMu && hasNuE);
}

double img::DataProviderAlg::LifetimeCorrection ( double  tick ) const

inlineinherited

Definition at line 169 of file DataProviderAlg.h.

References img::DataProviderAlg::fCalorimetryAlg, and calo::CalorimetryAlg::LifetimeCorrection().

{ return fCalorimetryAlg.LifetimeCorrection(tick); }

unsigned int img::DataProviderAlg::NCachedDrifts ( void  ) const

inlineinherited

Definition at line 163 of file DataProviderAlg.h.

References img::DataProviderAlg::fNCachedDrifts.

{ return fNCachedDrifts; }

unsigned int img::DataProviderAlg::NScaledDrifts ( void  ) const

inlineinherited

Definition at line 162 of file DataProviderAlg.h.

References img::DataProviderAlg::fNScaledDrifts.

{ return fNScaledDrifts; }

unsigned int img::DataProviderAlg::NWires ( void  ) const

inlineinherited

Definition at line 161 of file DataProviderAlg.h.

References img::DataProviderAlg::fNWires.

{ return fNWires; }

static float nnet::TrainingDataAlg::particleRange2 ( const simb::MCParticle &  particle )

inlinestaticprivate

Definition at line 305 of file PointIdAlg.h.

References simb::MCParticle::EndX(), simb::MCParticle::EndY(), simb::MCParticle::EndZ(), simb::MCParticle::Vx(), simb::MCParticle::Vy(), and simb::MCParticle::Vz().

Referenced by isElectronEnd().

    {
        float dx = particle.EndX() - particle.Vx();
        float dy = particle.EndY() - particle.Vy();
        float dz = particle.EndZ() - particle.Vz();
        return dx*dx + dy*dy + dz*dz;
    }

bool img::DataProviderAlg::patchFromDownsampledView ( size_t  wire, float  drift, size_t  size_w, size_t  size_d, std::vector< std::vector< float > > &  patch  ) const

protectedinherited

Definition at line 400 of file DataProviderAlg.cxx.

References d, img::DataProviderAlg::fAdcZero, img::DataProviderAlg::fDriftWindow, trkf::fill(), img::DataProviderAlg::fWireDriftData, and w.

Referenced by img::DataProviderAlg::getDriftIndex(), and img::DataProviderAlg::getPatch().

{
        int halfSizeW = size_w / 2;
        int halfSizeD = size_d / 2;

        int w0 = wire - halfSizeW;
        int w1 = wire + halfSizeW;

        size_t sd = (size_t)(drift / fDriftWindow);
        int d0 = sd - halfSizeD;
        int d1 = sd + halfSizeD;

        int wsize = fWireDriftData.size();
        for (int w = w0, wpatch = 0; w < w1; ++w, ++wpatch)
        {
                auto & dst = patch[wpatch];
                if ((w >= 0) && (w < wsize))
                {
                        auto & src = fWireDriftData[w];
                        int dsize = src.size();
                        for (int d = d0, dpatch = 0; d < d1; ++d, ++dpatch)
                        {
                                if ((d >= 0) && (d < dsize))
                                {
                                        dst[dpatch] = src[d];
                                }
                                else
                                {
                                        dst[dpatch] = fAdcZero;
                                }
                        }
                }
                else
                {
                        std::fill(dst.begin(), dst.end(), fAdcZero);
                }
        }

        return true;
}

bool img::DataProviderAlg::patchFromOriginalView ( size_t  wire, float  drift, size_t  size_w, size_t  size_d, std::vector< std::vector< float > > &  patch  ) const

protectedinherited

Definition at line 442 of file DataProviderAlg.cxx.

References d, img::DataProviderAlg::downscale(), img::DataProviderAlg::fAdcZero, img::DataProviderAlg::fDriftWindow, trkf::fill(), img::DataProviderAlg::fWireDriftData, tmp, and w.

Referenced by img::DataProviderAlg::getDriftIndex(), and img::DataProviderAlg::getPatch().

{
        int dsize = fDriftWindow * size_d;
        int halfSizeW = size_w / 2;
        int halfSizeD = dsize / 2;

        int w0 = wire - halfSizeW;
        int w1 = wire + halfSizeW;

        int d0 = drift - halfSizeD;
        int d1 = drift + halfSizeD;

        std::vector<float> tmp(dsize);
        int wsize = fWireDriftData.size();
        for (int w = w0, wpatch = 0; w < w1; ++w, ++wpatch)
        {
                if ((w >= 0) && (w < wsize))
                {
                        auto & src = fWireDriftData[w];
                        int src_size = src.size();
                        for (int d = d0, dpatch = 0; d < d1; ++d, ++dpatch)
                        {
                                if ((d >= 0) && (d < src_size))
                                {
                                        tmp[dpatch] = src[d];
                                }
                                else
                                {
                                        tmp[dpatch] = fAdcZero;
                                }
                        }
                }
                else
                {
                        std::fill(tmp.begin(), tmp.end(), fAdcZero);
                }

                downscale(patch[wpatch], tmp, d0);
        }

        return true;
}

unsigned int img::DataProviderAlg::Plane ( void  ) const

inlineinherited

Definition at line 159 of file DataProviderAlg.h.

References img::DataProviderAlg::fPlane.

Referenced by img::DataProviderAlg::setWireDriftData(), pma::ProjectionMatchingAlg::validate_on_adc(), and pma::ProjectionMatchingAlg::validate_on_adc_test().

{ return fPlane; }

float img::DataProviderAlg::poolMax ( int  wire, int  drift, size_t  r = 0  ) const

inherited

Pool max value in a patch around the wire/drift pixel.

Definition at line 133 of file DataProviderAlg.cxx.

References col, d, img::DataProviderAlg::fDownscaleFullView, img::DataProviderAlg::fDriftWindow, img::DataProviderAlg::fNCachedDrifts, img::DataProviderAlg::fNWires, img::DataProviderAlg::fWireDriftData, img::DataProviderAlg::getDriftIndex(), and w.

Referenced by img::DataProviderAlg::getAdcArea(), pma::ProjectionMatchingAlg::validate_on_adc(), and pma::ProjectionMatchingAlg::validate_on_adc_test().

{
    size_t rw = r, rd = r;
    if (!fDownscaleFullView) { rd *= fDriftWindow; }

    size_t didx = getDriftIndex(drift);
    int d0 = didx - rd; if (d0 < 0) { d0 = 0; }
    int d1 = didx + rd; if (d1 >= (int)fNCachedDrifts) { d1 = fNCachedDrifts - 1; }

    int w0 = wire - rw; if (w0 < 0) { w0 = 0; }
    int w1 = wire + rw; if (w1 >= (int)fNWires) { w1 = fNWires - 1; }

    float adc, max_adc = 0;
    for (int w = w0; w <= w1; ++w)
    {
        auto const * col = fWireDriftData[w].data();
        for (int d = d0; d <= d1; ++d)
        {
            adc = col[d]; if (adc > max_adc) { max_adc = adc; }
        }
    }

    return max_adc;
}

float img::DataProviderAlg::poolSum ( int  wire, int  drift, size_t  r = 0  ) const

inherited

Pool sum of pixels in a patch around the wire/drift pixel.

Definition at line 159 of file DataProviderAlg.cxx.

References col, d, img::DataProviderAlg::fDownscaleFullView, img::DataProviderAlg::fDriftWindow, img::DataProviderAlg::fNCachedDrifts, img::DataProviderAlg::fNWires, img::DataProviderAlg::fWireDriftData, img::DataProviderAlg::getDriftIndex(), and w.

Referenced by img::DataProviderAlg::getAdcArea().

{
    size_t rw = r, rd = r;
    if (!fDownscaleFullView) { rd *= fDriftWindow; }

    size_t didx = getDriftIndex(drift);
    int d0 = didx - rd; if (d0 < 0) { d0 = 0; }
    int d1 = didx + rd; if (d1 >= (int)fNCachedDrifts) { d1 = fNCachedDrifts - 1; }

    int w0 = wire - rw; if (w0 < 0) { w0 = 0; }
    int w1 = wire + rw; if (w1 >= (int)fNWires) { w1 = fNWires - 1; }

    float sum = 0;
    for (int w = w0; w <= w1; ++w)
    {
        auto const * col = fWireDriftData[w].data();
        for (int d = d0; d <= d1; ++d) { sum += col[d]; }
    }

    return sum;
}

void nnet::TrainingDataAlg::reconfigure

(

const Config &

config

)

void nnet::TrainingDataAlg::resizeView ( size_t  wires, size_t  drifts  )

overrideprotectedvirtual

Reimplemented from img::DataProviderAlg.

Definition at line 362 of file PointIdAlg.cxx.

References trkf::fill(), img::DataProviderAlg::fNCachedDrifts, fWireDriftEdep, fWireDriftPdg, img::DataProviderAlg::resizeView(), and w.

{
        img::DataProviderAlg::resizeView(wires, drifts);

        fWireDriftEdep.resize(wires);
        for (auto & w : fWireDriftEdep)
        {
                w.resize(fNCachedDrifts);
                std::fill(w.begin(), w.end(), 0.0F);
        }

        fWireDriftPdg.resize(wires);
        for (auto & w : fWireDriftPdg)
        {
                w.resize(fNCachedDrifts);
                std::fill(w.begin(), w.end(), 0);
        }
}

bool nnet::TrainingDataAlg::saveSimInfo ( ) const

inline

Definition at line 264 of file PointIdAlg.h.

{ return fSaveSimInfo; }

bool nnet::TrainingDataAlg::setDataEventData	(	const art::Event &	event,
		unsigned int	plane,
		unsigned int	tpc,
		unsigned int	cryo
	)

Definition at line 743 of file PointIdAlg.cxx.

References fEventsPerBin, fHitProducerLabel, art::fill_ptr_vector(), img::DataProviderAlg::fNDrifts, fTrackModuleLabel, fWireProducerLabel, art::DataViewImpl::getByLabel(), img::DataProviderAlg::setWireDriftData(), and setWireEdepsAndLabels().

{

  art::Handle< std::vector<recob::Wire> > wireHandle;
  std::vector< art::Ptr<recob::Wire> > Wirelist;

  if(event.getByLabel(fWireProducerLabel, wireHandle))
    art::fill_ptr_vector(Wirelist, wireHandle);

  if(!setWireDriftData(*wireHandle, plane, tpc, cryo)) {
    mf::LogError("TrainingDataAlg") << "Wire data not set.";
    return false;
  }

  // Hit info
  art::Handle< std::vector<recob::Hit> > HitHandle;
  std::vector< art::Ptr<recob::Hit> > Hitlist;

  if(event.getByLabel(fHitProducerLabel, HitHandle))
    art::fill_ptr_vector(Hitlist, HitHandle);

  // Track info
  art::Handle< std::vector<recob::Track> > TrackHandle;
  std::vector< art::Ptr<recob::Track> > Tracklist;

  if(event.getByLabel(fTrackModuleLabel, TrackHandle))
    art::fill_ptr_vector(Tracklist, TrackHandle);

  art::FindManyP<recob::Track> ass_trk_hits(HitHandle,   event, fTrackModuleLabel);

  // Loop over wires (sorry about hard coded value) to fill in 1) pdg and 2) charge depo
  for (size_t widx = 0; widx < 240; ++widx) {
    
    std::vector< float > labels_deposit(fNDrifts, 0);  // full-drift-length buffers
    std::vector< int > labels_pdg(fNDrifts, 0);

    // First, the charge depo
    for(size_t subwidx = 0; subwidx < Wirelist.size(); ++subwidx) {
      if(widx+240 == Wirelist[subwidx]->Channel()) {    
        labels_deposit = Wirelist[subwidx]->Signal();
        break;
      }
    }
   
    // Second, the pdg code
    // This code finds the angle of the track and records
    //  events based on its angle to try to get an isometric sample
    //  instead of just a bunch of straight tracks

    // Meta code:
    // For each hit:
    //  find farthest hit from point
    //  then find farthest hit from THAT one
    //  should be start and end of track, then just use trig

    for(size_t iHit = 0; iHit < Hitlist.size(); ++iHit) {

      if(Hitlist[iHit]->Channel() != widx+240) { continue; }
      if(Hitlist[iHit]->View() != 1) { continue; }

      // Make sure there is a track association
      if(ass_trk_hits.at(iHit).size() == 0) { continue; }
      
      // Not sure about this
      // Cutting on length to not just get a bunch of shower stubs
      // Might add a lot of bias though
      if(ass_trk_hits.at(iHit)[0]->Length() < 5) { continue; }

      // Search for farest hit from this one
      int far_index = 0;
      double far_dist = 0;

      for(size_t jHit = 0; jHit < Hitlist.size(); ++jHit) {
        if(jHit == iHit) { continue; }
        if(Hitlist[jHit]->View() != 1) { continue; }

        if(ass_trk_hits.at(jHit).size() == 0) { continue; }
        if(ass_trk_hits.at(jHit)[0]->ID() != 
           ass_trk_hits.at(iHit)[0]->ID()) { continue; }

        double dist = sqrt((Hitlist[iHit]->Channel()-Hitlist[jHit]->Channel()) *
                           (Hitlist[iHit]->Channel()-Hitlist[jHit]->Channel()) +
                           (Hitlist[iHit]->PeakTime()-Hitlist[jHit]->PeakTime()) * 
                           (Hitlist[iHit]->PeakTime()-Hitlist[jHit]->PeakTime()));

        if(far_dist < dist){
          far_dist = dist;
          far_index = jHit;
        }
      }

      // Search for the other end of the track
      int other_end = 0;
      int other_dist = 0;

      for(size_t jHit = 0; jHit < Hitlist.size(); ++jHit) {
        if(jHit == iHit or int(jHit) == far_index) { continue; }
        if(Hitlist[jHit]->View() != 1) { continue; }

        if(ass_trk_hits.at(jHit).size() == 0) { continue; }
        if(ass_trk_hits.at(jHit)[0]->ID() != 
           ass_trk_hits.at(iHit)[0]->ID()) { continue; }

        double dist = sqrt((Hitlist[far_index]->Channel()-Hitlist[jHit]->Channel()) *
                           (Hitlist[far_index]->Channel()-Hitlist[jHit]->Channel()) +
                           (Hitlist[far_index]->PeakTime()-Hitlist[jHit]->PeakTime()) * 
                           (Hitlist[far_index]->PeakTime()-Hitlist[jHit]->PeakTime()));

        if(other_dist < dist){
          other_dist = dist;
          other_end = jHit;
        }
      }

      // We have the end points now
      double del_wire = double(Hitlist[other_end]->Channel() - Hitlist[far_index]->Channel());
      double del_time = double(Hitlist[other_end]->PeakTime() - Hitlist[far_index]->PeakTime());
      double hypo = sqrt(del_wire * del_wire + del_time * del_time);      

      if(hypo == 0) { continue; } // Should never happen, but doing it anyway

      double cosser = TMath::Abs(del_wire / hypo);
      double norm_ang = TMath::ACos(cosser) * 2 / TMath::Pi();

      // Using fEventsPerBin to keep track of number of hits per angle (normalized to 0 to 1)

      int binner = int(norm_ang * fEventsPerBin.size());
      if(binner >= (int)fEventsPerBin.size()) { binner = fEventsPerBin.size() - 1; } // Dealing with rounding errors

      // So we should get a total of 5000 * 100 = 50,000 if we use the whole set
      if(fEventsPerBin[binner] > 5000) { continue; }
      fEventsPerBin[binner]++;
      
      // If survives everything, saves the pdg
      labels_pdg[Hitlist[iHit]->PeakTime()] = 211; // Same as pion for now

    }

    setWireEdepsAndLabels(labels_deposit, labels_pdg, widx);

  } // for each Wire

  /*
  for(size_t i = 0; i < fEventsPerBin.size(); i ++) {
    std::cout << i << ") " << fEventsPerBin[i] << " - ";
  }
  */

  return true;

}

bool nnet::TrainingDataAlg::setEventData	(	const art::Event &	event,
		unsigned int	plane,
		unsigned int	tpc,
		unsigned int	cryo
	)

Definition at line 896 of file PointIdAlg.cxx.

References collectVtxFlags(), energy, fAdcDelay, fEdepTot, img::DataProviderAlg::fNDrifts, img::DataProviderAlg::fNWires, fSaveSimInfo, fSaveVtxFlags, fSimulationProducerLabel, img::DataProviderAlg::fWireChannels, fWireProducerLabel, sim::LArG4Parameters::GeVToElectrons(), raw::InvalidChannelID, isMuonDecaying(), art::Event::isRealData(), kDelta, kMichel, kPdgMask, kPriEl, kPriMu, kTypeMask, img::DataProviderAlg::setWireDriftData(), and setWireEdepsAndLabels().

{
        art::ValidHandle< std::vector<recob::Wire> > wireHandle
                = event.getValidHandle< std::vector<recob::Wire> >(fWireProducerLabel);

        if (!setWireDriftData(*wireHandle, plane, tpc, cryo))
        {
                mf::LogError("TrainingDataAlg") << "Wire data not set.";
                return false;
        }

    if (!fSaveSimInfo || event.isRealData())
    {
        mf::LogInfo("TrainingDataAlg") << "Skip MC simulation info.";
        return true;
    }

        art::ServiceHandle<sim::LArG4Parameters> larParameters;
        double electronsToGeV = 1. / larParameters->GeVToElectrons();

        auto particleHandle = event.getValidHandle< std::vector<simb::MCParticle> >(fSimulationProducerLabel);

        auto simChannelHandle = event.getValidHandle< std::vector<sim::SimChannel> >(fSimulationProducerLabel);

    std::unordered_map< int, const simb::MCParticle* > particleMap;
        for (auto const & particle : *particleHandle)
    {
                particleMap[particle.TrackId()] = &particle;
        }

        std::unordered_map< size_t, std::unordered_map< int, int > > wireToDriftToVtxFlags;
        if (fSaveVtxFlags) collectVtxFlags(wireToDriftToVtxFlags, particleMap, plane);

        fEdepTot = 0;

        std::map< int, int > trackToPDG;
        for (size_t widx = 0; widx < fNWires; ++widx)
        {
                auto wireChannelNumber = fWireChannels[widx];
                if (wireChannelNumber == raw::InvalidChannelID) continue;

                std::vector< float > labels_deposit(fNDrifts, 0);         // full-drift-length buffers,
                std::vector< int > labels_pdg(labels_deposit.size(), 0);  // both of the same size,
                int labels_size = labels_deposit.size();                  // cached as int for comparisons below

                std::map< int, std::map< int, double > > timeToTrackToCharge;
                for (auto const & channel : *simChannelHandle)
                {
                        if (channel.Channel() != wireChannelNumber) continue;           

                        auto const & timeSlices = channel.TDCIDEMap();
                        for (auto const & timeSlice : timeSlices)
                        {
                                int time = timeSlice.first;

                                auto const & energyDeposits = timeSlice.second;
                                for (auto const & energyDeposit : energyDeposits)
                                {
                                        int pdg = 0;
                                        int tid = energyDeposit.trackID;
                                        if (tid < 0) // negative tid means it is EM activity, and -tid is the mother
                                        {
                                            pdg = 11; tid = -tid;
                                            
                                                auto search = particleMap.find(tid);
                                            if (search == particleMap.end())
                                            {
                                                    mf::LogWarning("TrainingDataAlg") << "PARTICLE NOT FOUND";
                                                    continue;
                                            }
                                            auto const & mother = *((*search).second); // mother particle of this EM
                                            int mPdg = abs(mother.PdgCode());
                                            if ((mPdg == 13) || (mPdg == 211) || (mPdg == 2212))
                                            {
                                                if (energyDeposit.numElectrons > 10) pdg |= nnet::TrainingDataAlg::kDelta; // tag delta ray
                                            }
                                        }
                                        else
                                        {
                                          auto search = particleMap.find(tid);
                                          if (search == particleMap.end())
                                          {
                                                   mf::LogWarning("TrainingDataAlg") << "PARTICLE NOT FOUND";
                                                   continue;
                                           }
                                           auto const & particle = *((*search).second);
                                           pdg = abs(particle.PdgCode());
                                           
                                           if (particle.Process() == "primary")
                                           {
                                                        if (pdg == 11)
                                                        {
                                                                pdg |= nnet::TrainingDataAlg::kPriEl; // tag primary
                                                        }
                                                        else if (pdg == 13)
                                                        {
                                                                pdg |= nnet::TrainingDataAlg::kPriMu; // tag primary
                                                        }
                                           }

            auto msearch = particleMap.find(particle.Mother());
                                if (msearch != particleMap.end())
                                {
                                        auto const & mother = *((*msearch).second);
              if (pdg == 11) // electron, check if it is Michel
              {
                                if (nnet::TrainingDataAlg::isMuonDecaying(mother, particleMap))
                              {
                        pdg |= nnet::TrainingDataAlg::kMichel; // tag Michel
                              }
              }
            }
                                        }

                                        trackToPDG[energyDeposit.trackID] = pdg;

                                        double energy = energyDeposit.numElectrons * electronsToGeV;
                                        timeToTrackToCharge[time][energyDeposit.trackID] += energy;
                                        fEdepTot += energy;

                        } // loop over energy deposits
                } // loop over time slices
        } // for each SimChannel

        int type_pdg_mask = nnet::TrainingDataAlg::kTypeMask | nnet::TrainingDataAlg::kPdgMask;
                for (auto const & ttc : timeToTrackToCharge)
                {
                        float max_deposit = 0.0;
                        int max_pdg = 0;
                        for (auto const & tc : ttc.second) {

                                if( tc.second > max_deposit ) 
                                {
                                        max_deposit = tc.second;
                                        max_pdg = trackToPDG[tc.first];
                                }                       
                        }

                        if (ttc.first < labels_size)
                        {
                            int tick_idx = ttc.first + fAdcDelay;
                                if (tick_idx < labels_size)
                                {
                                    labels_deposit[tick_idx] = max_deposit;
                                    labels_pdg[tick_idx] = max_pdg & type_pdg_mask;
                                }
                        }
                }

                for (auto const & drift_flags : wireToDriftToVtxFlags[widx])
                {
                        int drift = drift_flags.first, flags = drift_flags.second;
                        if ((drift >= 0) && (drift < labels_size))
                        {
                                labels_pdg[drift] |= flags;
                        }
                }

                setWireEdepsAndLabels(labels_deposit, labels_pdg, widx);

        } // for each Wire

        return true;
}

bool img::DataProviderAlg::setWireData ( std::vector< float > const &  adc, size_t  wireIdx  )

protectedinherited

Definition at line 245 of file DataProviderAlg.cxx.

References img::DataProviderAlg::downscale(), img::DataProviderAlg::fDownscaleFullView, and img::DataProviderAlg::fWireDriftData.

Referenced by img::DataProviderAlg::getDriftIndex(), and img::DataProviderAlg::setWireDriftData().

{
        if (wireIdx >= fWireDriftData.size()) return false;
        auto & wData = fWireDriftData[wireIdx];

    if (fDownscaleFullView)
    {
        if (!adc.empty()) { downscale(wData, adc, 0); }
        else { return false; }
    }
    else
    {
        if (adc.empty()) { return false; }
        else if (adc.size() <= wData.size()) { std::copy(adc.begin(), adc.end(), wData.begin()); }
        else { std::copy(adc.begin(), adc.begin()+wData.size(), wData.begin()); }
    }
    return true;
}

bool img::DataProviderAlg::setWireDriftData ( const std::vector< recob::Wire > &  wires, unsigned int  plane, unsigned int  tpc, unsigned int  cryo  )

inherited

Definition at line 265 of file DataProviderAlg.cxx.

References img::DataProviderAlg::addCoherentNoise(), img::DataProviderAlg::addWhiteNoise(), img::DataProviderAlg::applyBlur(), geo::GeometryCore::ChannelToWire(), img::DataProviderAlg::fAdcAreaOverThr, img::DataProviderAlg::fAdcSumOverThr, img::DataProviderAlg::fAdcSumThr, img::DataProviderAlg::fCryo, img::DataProviderAlg::fDetProp, img::DataProviderAlg::fGeometry, img::DataProviderAlg::fPlane, img::DataProviderAlg::fTPC, img::DataProviderAlg::fWireChannels, detinfo::DetectorProperties::NumberTimeSamples(), geo::GeometryCore::Nwires(), img::DataProviderAlg::Plane(), img::DataProviderAlg::resizeView(), and img::DataProviderAlg::setWireData().

Referenced by img::DataProviderAlg::DataProviderAlg(), setDataEventData(), and setEventData().

{
    mf::LogInfo("DataProviderAlg") << "Create image for cryo:"
        << cryo << " tpc:" << tpc << " plane:" << plane;

        fCryo = cryo; fTPC = tpc; fPlane = plane;

    fAdcSumOverThr = 0;
    fAdcAreaOverThr = 0;

        size_t nwires = fGeometry->Nwires(plane, tpc, cryo);
        size_t ndrifts = fDetProp->NumberTimeSamples();

        resizeView(nwires, ndrifts);

    auto const & channelStatus = art::ServiceHandle< lariov::ChannelStatusService >()->GetProvider();

    bool allWrong = true;
    for (auto const & wire : wires)
        {
                auto wireChannelNumber = wire.Channel();
                if (!channelStatus.IsGood(wireChannelNumber)) { continue; }

                size_t w_idx = 0;
                for (auto const& id : fGeometry->ChannelToWire(wireChannelNumber))
                {
                        if ((id.Plane == plane) && (id.TPC == tpc) && (id.Cryostat == cryo))
                        {
                            w_idx = id.Wire;

                            auto adc = wire.Signal();
                            if (adc.size() < ndrifts)
                            {
                                mf::LogWarning("DataProviderAlg") << "Wire ADC vector size lower than NumberTimeSamples.";
                                continue; // not critical, maybe other wires are OK, so continue
                            }

                            if (!setWireData(adc, w_idx))
                            {
                                mf::LogWarning("DataProviderAlg") << "Wire data not set.";
                                continue; // also not critical, try to set other wires
                            }

                for (auto v : adc) { if (v >= fAdcSumThr) { fAdcSumOverThr += v; fAdcAreaOverThr++; } }

                            fWireChannels[w_idx] = wireChannelNumber;
                            allWrong = false;
                        }
                }
        }
        if (allWrong)
        {
            mf::LogError("DataProviderAlg") << "Wires data not set in the cryo:"
                << cryo << " tpc:" << tpc << " plane:" << plane;
            return false;
        }
        
    applyBlur();
    addWhiteNoise();
    addCoherentNoise();

        return true;
}

bool nnet::TrainingDataAlg::setWireEdepsAndLabels ( std::vector< float > const &  edeps, std::vector< int > const &  pdgs, size_t  wireIdx  )

private

Definition at line 382 of file PointIdAlg.cxx.

References img::DataProviderAlg::fDownscaleFullView, img::DataProviderAlg::fDriftWindow, img::DataProviderAlg::fNCachedDrifts, fSaveVtxFlags, fWireDriftEdep, fWireDriftPdg, kPdgMask, kTypeMask, and kVtxMask.

Referenced by setDataEventData(), and setEventData().

{
        if ((wireIdx >= fWireDriftEdep.size()) || (edeps.size() != pdgs.size())) { return false; }

        size_t dstep = 1;
        if (fDownscaleFullView) { dstep = fDriftWindow; }

        if (edeps.size() / dstep > fNCachedDrifts) { return false; }

        auto & wEdep = fWireDriftEdep[wireIdx];
        auto & wPdg = fWireDriftPdg[wireIdx];

        for (size_t i = 0; i < fNCachedDrifts; ++i)
        {
                size_t i0 = i * dstep;
                size_t i1 = (i + 1) * dstep;

                int best_pdg = pdgs[i0] & nnet::TrainingDataAlg::kPdgMask;
                int vtx_flags = pdgs[i0] & nnet::TrainingDataAlg::kVtxMask;
                int type_flags = pdgs[i0] & nnet::TrainingDataAlg::kTypeMask;
                float max_edep = edeps[i0];
                for (size_t k = i0 + 1; k < i1; ++k)
                {
                        float ek = edeps[k];
                        if (ek > max_edep)
                        {
                                max_edep = ek;
                                best_pdg = pdgs[k] & nnet::TrainingDataAlg::kPdgMask; // remember best matching pdg
                        }
                        type_flags |= pdgs[k] & nnet::TrainingDataAlg::kTypeMask; // accumulate track type flags
                        vtx_flags |= pdgs[k] & nnet::TrainingDataAlg::kVtxMask;   // accumulate all vtx flags
                }

                wEdep[i] = max_edep;

        best_pdg |= type_flags;
                if (fSaveVtxFlags) best_pdg |= vtx_flags;
                wPdg[i] = best_pdg;
        }

        return true;
}

unsigned int img::DataProviderAlg::TPC ( void  ) const

inlineinherited

Definition at line 158 of file DataProviderAlg.h.

References img::DataProviderAlg::fTPC.

{ return fTPC; }

std::vector<float> const& img::DataProviderAlg::wireData ( size_t  widx ) const

inlineinherited

Definition at line 109 of file DataProviderAlg.h.

References img::DataProviderAlg::fWireDriftData.

{ return fWireDriftData[widx]; }

std::vector<float> const& nnet::TrainingDataAlg::wireEdep ( size_t  widx ) const

inline

Definition at line 276 of file PointIdAlg.h.

{ return fWireDriftEdep[widx]; }

std::vector<int> const& nnet::TrainingDataAlg::wirePdg ( size_t  widx ) const

inline

Definition at line 277 of file PointIdAlg.h.

{ return fWireDriftPdg[widx]; }

float img::DataProviderAlg::ZeroLevel ( void  ) const

inlineinherited

Level of zero ADC after scaling.

Definition at line 167 of file DataProviderAlg.h.

References img::DataProviderAlg::fAdcZero.

{ return fAdcZero; }

Member Data Documentation

unsigned int nnet::TrainingDataAlg::fAdcDelay

private

Definition at line 331 of file PointIdAlg.h.

Referenced by setEventData().

calo::CalorimetryAlg img::DataProviderAlg::fCalorimetryAlg

protectedinherited

Definition at line 214 of file DataProviderAlg.h.

Referenced by img::DataProviderAlg::DataProviderAlg(), img::DataProviderAlg::LifetimeCorrection(), and img::DataProviderAlg::resizeView().

unsigned int img::DataProviderAlg::fCryo

protectedinherited

Definition at line 172 of file DataProviderAlg.h.

Referenced by collectVtxFlags(), img::DataProviderAlg::Cryo(), and img::DataProviderAlg::setWireDriftData().

detinfo::DetectorProperties const* img::DataProviderAlg::fDetProp

protectedinherited

Definition at line 218 of file DataProviderAlg.h.

Referenced by getProjection(), and img::DataProviderAlg::setWireDriftData().

bool img::DataProviderAlg::fDownscaleFullView

protectedinherited

Definition at line 183 of file DataProviderAlg.h.

Referenced by img::DataProviderAlg::addCoherentNoise(), img::DataProviderAlg::addWhiteNoise(), img::DataProviderAlg::DataProviderAlg(), img::DataProviderAlg::getPatch(), nnet::PointIdAlg::isCurrentPatch(), nnet::PointIdAlg::isSamePatch(), img::DataProviderAlg::poolMax(), img::DataProviderAlg::poolSum(), img::DataProviderAlg::resizeView(), img::DataProviderAlg::setWireData(), and setWireEdepsAndLabels().

EDownscaleMode img::DataProviderAlg::fDownscaleMode

protectedinherited

Definition at line 179 of file DataProviderAlg.h.

Referenced by img::DataProviderAlg::DataProviderAlg().

size_t img::DataProviderAlg::fDriftWindow

protectedinherited

Definition at line 182 of file DataProviderAlg.h.

Referenced by img::DataProviderAlg::addCoherentNoise(), img::DataProviderAlg::addWhiteNoise(), img::DataProviderAlg::DataProviderAlg(), img::DataProviderAlg::downscaleMax(), img::DataProviderAlg::downscaleMaxMean(), img::DataProviderAlg::downscaleMean(), img::DataProviderAlg::DriftWindow(), nnet::PointIdAlg::isCurrentPatch(), nnet::PointIdAlg::isInsideFiducialRegion(), nnet::PointIdAlg::isSamePatch(), img::DataProviderAlg::patchFromDownsampledView(), img::DataProviderAlg::patchFromOriginalView(), img::DataProviderAlg::poolMax(), img::DataProviderAlg::poolSum(), img::DataProviderAlg::resizeView(), and setWireEdepsAndLabels().

float img::DataProviderAlg::fDriftWindowInv

protectedinherited

Definition at line 184 of file DataProviderAlg.h.

Referenced by img::DataProviderAlg::DataProviderAlg(), img::DataProviderAlg::downscaleMean(), and img::DataProviderAlg::getDriftIndex().

double nnet::TrainingDataAlg::fEdepTot

private

Definition at line 320 of file PointIdAlg.h.

Referenced by setEventData().

std::vector<size_t> nnet::TrainingDataAlg::fEventsPerBin

private

Definition at line 333 of file PointIdAlg.h.

Referenced by setDataEventData().

geo::GeometryCore const* img::DataProviderAlg::fGeometry

protectedinherited

Definition at line 217 of file DataProviderAlg.h.

Referenced by img::DataProviderAlg::DataProviderAlg(), getProjection(), and img::DataProviderAlg::setWireDriftData().

art::InputTag nnet::TrainingDataAlg::fHitProducerLabel

private

Definition at line 325 of file PointIdAlg.h.

Referenced by setDataEventData().

std::vector<float> img::DataProviderAlg::fLifetimeCorrFactors

protectedinherited

Definition at line 177 of file DataProviderAlg.h.

Referenced by img::DataProviderAlg::downscaleMax(), img::DataProviderAlg::downscaleMaxMean(), img::DataProviderAlg::downscaleMean(), and img::DataProviderAlg::resizeView().

unsigned int img::DataProviderAlg::fNCachedDrifts

protectedinherited

Definition at line 173 of file DataProviderAlg.h.

Referenced by img::DataProviderAlg::addCoherentNoise(), img::DataProviderAlg::addWhiteNoise(), img::DataProviderAlg::getPixelOrZero(), img::DataProviderAlg::NCachedDrifts(), img::DataProviderAlg::poolMax(), img::DataProviderAlg::poolSum(), img::DataProviderAlg::resizeView(), resizeView(), and setWireEdepsAndLabels().

unsigned int img::DataProviderAlg::fNDrifts

protectedinherited

Definition at line 173 of file DataProviderAlg.h.

Referenced by img::DataProviderAlg::resizeView(), setDataEventData(), and setEventData().

unsigned int img::DataProviderAlg::fNScaledDrifts

protectedinherited

Definition at line 173 of file DataProviderAlg.h.

Referenced by nnet::PointIdAlg::isInsideFiducialRegion(), img::DataProviderAlg::NScaledDrifts(), and img::DataProviderAlg::resizeView().

unsigned int img::DataProviderAlg::fNWires

protectedinherited

Definition at line 173 of file DataProviderAlg.h.

Referenced by nnet::PointIdAlg::isInsideFiducialRegion(), img::DataProviderAlg::NWires(), img::DataProviderAlg::poolMax(), img::DataProviderAlg::poolSum(), img::DataProviderAlg::resizeView(), and setEventData().

unsigned int img::DataProviderAlg::fPlane

protectedinherited

Definition at line 172 of file DataProviderAlg.h.

Referenced by img::DataProviderAlg::Plane(), img::DataProviderAlg::scaleAdcSample(), img::DataProviderAlg::scaleAdcSamples(), and img::DataProviderAlg::setWireDriftData().

bool nnet::TrainingDataAlg::fSaveSimInfo

private

Definition at line 329 of file PointIdAlg.h.

Referenced by setEventData(), and TrainingDataAlg().

bool nnet::TrainingDataAlg::fSaveVtxFlags

private

Definition at line 328 of file PointIdAlg.h.

Referenced by setEventData(), and setWireEdepsAndLabels().

art::InputTag nnet::TrainingDataAlg::fSimulationProducerLabel

private

Definition at line 327 of file PointIdAlg.h.

Referenced by setEventData(), and TrainingDataAlg().

unsigned int img::DataProviderAlg::fTPC

protectedinherited

Definition at line 172 of file DataProviderAlg.h.

Referenced by collectVtxFlags(), img::DataProviderAlg::setWireDriftData(), and img::DataProviderAlg::TPC().

art::InputTag nnet::TrainingDataAlg::fTrackModuleLabel

private

Definition at line 326 of file PointIdAlg.h.

Referenced by setDataEventData().

std::vector< raw::ChannelID_t > img::DataProviderAlg::fWireChannels

protectedinherited

Definition at line 175 of file DataProviderAlg.h.

Referenced by img::DataProviderAlg::resizeView(), setEventData(), and img::DataProviderAlg::setWireDriftData().

std::vector< std::vector<float> > img::DataProviderAlg::fWireDriftData

protectedinherited

Definition at line 176 of file DataProviderAlg.h.

Referenced by img::DataProviderAlg::addCoherentNoise(), img::DataProviderAlg::addWhiteNoise(), img::DataProviderAlg::applyBlur(), img::DataProviderAlg::getPixelOrZero(), img::DataProviderAlg::patchFromDownsampledView(), img::DataProviderAlg::patchFromOriginalView(), img::DataProviderAlg::poolMax(), img::DataProviderAlg::poolSum(), img::DataProviderAlg::resizeView(), img::DataProviderAlg::setWireData(), and img::DataProviderAlg::wireData().

std::vector< std::vector<float> > nnet::TrainingDataAlg::fWireDriftEdep

private

Definition at line 321 of file PointIdAlg.h.

Referenced by findCrop(), resizeView(), and setWireEdepsAndLabels().

std::vector< std::vector<int> > nnet::TrainingDataAlg::fWireDriftPdg

private

Definition at line 322 of file PointIdAlg.h.

Referenced by resizeView(), and setWireEdepsAndLabels().

art::InputTag nnet::TrainingDataAlg::fWireProducerLabel

private

Definition at line 324 of file PointIdAlg.h.

Referenced by setDataEventData(), and setEventData().

---

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

• larreco/v06_64_02/source/larreco/RecoAlg/ImagePatternAlgs/Tensorflow/PointIdAlg/PointIdAlg.h
• larreco/v06_64_02/source/larreco/RecoAlg/ImagePatternAlgs/Tensorflow/PointIdAlg/PointIdAlg.cxx