Add Sparse Cluster reading, haven't tested extensively yet.

DeepLearnPhysics · Mar 4, 2019 · 9e7d6b7 · 9e7d6b7
1 parent 408384a
commit 9e7d6b7
Show file tree

Hide file tree

Showing 4 changed files with 496 additions and 0 deletions.
diff --git a/larcv/app/SBNDImageMod/ParticleBoundaries.cxx b/larcv/app/SBNDImageMod/ParticleBoundaries.cxx
@@ -0,0 +1,115 @@
+// #ifndef __PARTICLEBOUNDARIES_CXX__
+// #define __PARTICLEBOUNDARIES_CXX__
+
+// #include "ParticleBoundaries.h"
+// #include "larcv/core/DataFormat/EventParticle.h"
+// #include "larcv/core/DataFormat/EventVoxel3D.h"
+// namespace larcv {
+
+//   static ParticleBoundariesProcessFactory
+//   __global_ParticleBoundariesProcessFactory__;
+
+//   ParticleBoundaries::ParticleBoundaries(const std::string name)
+//     : ProcessBase(name) {}
+
+//   void ParticleBoundaries::configure(const PSet& cfg) {
+//     _cluster_producer = cfg.get<std::string>("ClusterProducer");
+//     _particle_producer = cfg.get<std::string>("ParticleProducer");
+//     _output_producer = _particle_producer + "_bbox";
+//     _output_producer = cfg.get<std::string>("OutputProducer", _output_producer);
+//     _voxel_threshold = cfg.get<float>("Threshold", 0.0);
+
+//   }
+
+//   void ParticleBoundaries::initialize() {}
+
+//   bool ParticleBoundaries::process(IOManager& mgr) {
+//     auto const& ev_cluster3d =
+//       mgr.get_data<larcv::EventClusterVoxel3D>(_cluster_producer);
+
+//     auto const& ev_cluster2d = 
+//       mgr.get_data<larcv::EventClusterPixel2D>(_cluster_producer);
+//     auto const& meta3d = ev_cluster3d.meta();
+
+//     auto const& ev_part_input =
+//       mgr.get_data<larcv::EventParticle>(_particle_producer);
+
+//     // There should be N particles, and N + 1 clusters
+//     // The last cluster corresponds to all of the untagged 
+
+//     if (ev_cluster3d.as_vector().size()  != ev_part_input.as_vector().size() + 1) {
+//       LARCV_CRITICAL() << "Can not match " << ev_cluster3d.as_vector().size()
+//                        << " particles to " << ev_part_input.as_vector().size()
+//                        << " cluster3d objects." << std::endl;
+//       throw larbys();
+//     }
+
+//     std::vector<larcv::Particle> particle_v;
+
+//     for (size_t i_part = 0; i_part < ev_part_input.as_vector().size(); i_part++) {
+//       auto new_particle = ev_part_input.as_vector().at(i_part);
+
+//       // Get the set of voxels matched to this particle, and find the
+//       // min/max in each coordinate.
+//       auto const& voxel_v = ev_cluster3d.as_vector().at(i_part);
+//       LARCV_INFO() << "Particle " << i_part << " corresponding ClusterVoxel3D contains " << voxel_v.as_vector().size() << " voxels..." << std::endl;
+//       // BBox3D _3d_box;
+//       double x_min, y_min, z_min;
+//       double x_max, y_max, z_max;
+//       x_min = meta3d.max_x();
+//       y_min = meta3d.max_y();
+//       z_min = meta3d.max_z();
+//       x_max = meta3d.min_x();
+//       y_max = meta3d.min_y();
+//       z_max = meta3d.min_z();
+//       for (auto const& vox : voxel_v.as_vector()) {
+//         if (vox.value() > _voxel_threshold) {
+//           auto const pos = meta3d.position(vox.id());
+//           if (pos.x < x_min) x_min = pos.x;
+//           if (pos.y < y_min) y_min = pos.y;
+//           if (pos.z < z_min) z_min = pos.z;
+//           if (pos.x > x_max) x_max = pos.x;
+//           if (pos.y > y_max) y_max = pos.y;
+//           if (pos.z > z_max) z_max = pos.z;
+//         }
+//       }
+//       // Check to make sure values were actually filled:
+//       if (x_max < x_min || y_max < y_min || z_max < z_min) {
+//         particle_v.emplace_back(std::move(new_particle));
+//         LARCV_INFO() << "Particle " << i_part << " no bounding box found... " << std::endl;
+//         continue;
+//       }
+
+//       // Now create the 3D and 2D bounding boxes:
+//       new_particle.boundingbox_3d(BBox3D(x_min, y_min, z_min, x_max, y_max, z_max));
+//       LARCV_INFO() << "Particle " << i_part << " 3D BBox: " << new_particle.boundingbox_3d().dump();
+
+//       // XY projection:
+//       if (_xy) {
+//         new_particle.boundingbox_2d(BBox2D(x_min, y_min, x_max, y_max), 0);
+//         LARCV_INFO() << "Particle " << i_part << " Projection 0 BBox: " << new_particle.boundingbox_2d((ProjectionID_t)0).dump();
+//       }
+//       // YZ projection:
+//       if (_yz) {
+//         new_particle.boundingbox_2d(BBox2D(y_min, z_min, y_max, z_max), 1);
+//         LARCV_INFO() << "Particle " << i_part << " Projection 1 BBox: " << new_particle.boundingbox_2d((ProjectionID_t)1).dump();
+//       }
+//       // ZX projection:
+//       if (_zx) {
+//         new_particle.boundingbox_2d(BBox2D(z_min, x_min, z_max, x_max), 2);
+//         LARCV_INFO() << "Particle " << i_part << " Projection 2 BBox: " << new_particle.boundingbox_2d((ProjectionID_t)2).dump();
+//       }
+//       LARCV_INFO() << std::endl;
+
+//       particle_v.emplace_back(std::move(new_particle));
+//     }
+
+//     auto& ev_part_output = mgr.get_data<larcv::EventParticle>(_output_producer);
+//     ev_part_output.emplace(std::move(particle_v));
+
+//     return true;
+//   }
+
+//   void ParticleBoundaries::finalize() {}
+// }
+// #endif
diff --git a/larcv/app/SBNDImageMod/ParticleBoundaries.h b/larcv/app/SBNDImageMod/ParticleBoundaries.h
@@ -0,0 +1,71 @@
+// /**
+//  * \file ParticleBoundaries.h
+//  *
+//  * \ingroup ImageMod
+//  * 
+//  * \brief Class def header for a class ParticleBoundaries
+//  *
+//  * @author kazuhiro
+//  */
+
+// /** \addtogroup ImageMod
+
+//     @{*/
+// #ifndef __PARTICLEBOUNDARIES_H__
+// #define __PARTICLEBOUNDARIES_H__
+
+// #include "larcv/core/Processor/ProcessBase.h"
+// #include "larcv/core/Processor/ProcessFactory.h"
+// namespace larcv {
+
+//   /**
+//      \class ProcessBase
+//      User defined class ParticleBoundaries ... these comments are used to generate
+//      doxygen documentation!
+//   */
+//   class ParticleBoundaries : public ProcessBase {
+
+//   public:
+
+//     /// Default constructor
+//     ParticleBoundaries(const std::string name="ParticleBoundaries");
+
+//     /// Default destructor
+//     ~ParticleBoundaries(){}
+
+//     void configure(const PSet&);
+
+//     void initialize();
+
+//     bool process(IOManager& mgr);
+
+//     void finalize();
+
+//   private:
+
+//     std::string _cluster_producer;
+//     std::string _particle_producer;
+//     std::string _output_producer;
+//     float _voxel_threshold;
+
+//   };
+
+//   /**
+//      \class larcv::BBoxFromCluster3DFactory
+//      \brief A concrete factory class for larcv::ParticleBoundaries
+//   */
+//   class ParticleBoundariesProcessFactory : public ProcessFactoryBase {
+//   public:
+//     /// ctor
+//     ParticleBoundariesProcessFactory() { ProcessFactory::get().add_factory("ParticleBoundaries",this); }
+//     /// dtor
+//     ~ParticleBoundariesProcessFactory() {}
+//     /// creation method
+//     ProcessBase* create(const std::string instance_name) { return new ParticleBoundaries(instance_name); }
+//   };
+
+// }
+
+// #endif
+// /** @} */ // end of doxygen group 
+