From 025f675a8564211ee6b91a2713c46a790969e0f5 Mon Sep 17 00:00:00 2001
From: Riccardo Maria Bianchi <riccardo.maria.bianchi@cern.ch>
Date: Sat, 22 Jun 2024 01:59:49 +0200
Subject: [PATCH] Add getters to retrieve number of shapes with the new DB
 schema

---
 .../GeoModelWrite/WriteGeoModel.h             | 19 ++++++++++++++++++-
 1 file changed, 18 insertions(+), 1 deletion(-)

diff --git a/GeoModelIO/GeoModelWrite/GeoModelWrite/WriteGeoModel.h b/GeoModelIO/GeoModelWrite/GeoModelWrite/WriteGeoModel.h
index 28f09ab3e..0d69ccb6e 100644
--- a/GeoModelIO/GeoModelWrite/GeoModelWrite/WriteGeoModel.h
+++ b/GeoModelIO/GeoModelWrite/GeoModelWrite/WriteGeoModel.h
@@ -95,6 +95,7 @@ class WriteGeoModel : public GeoNodeAction {
     /// - 0 : Default
     /// - 1 : Verbose
     /// - 2 : Debug
+    /// - 3 : Deep Debug
     void setLogLevel(unsigned loglevel) { m_loglevel = loglevel; };
 
     virtual void handlePhysVol(
@@ -146,10 +147,26 @@ class WriteGeoModel : public GeoNodeAction {
     };
     unsigned long getNFunctions() { return m_functions.size(); };
     unsigned long getNNameTags() { return m_nameTags.size(); };
-    unsigned long getNShapes() { return m_shapes.size(); };
     unsigned long getNChildrenConnections() {
         return m_childrenPositions.size();
     };
+    unsigned long getNShapes() { return m_shapes.size(); }; // TODO: to be removed later
+    unsigned long getNShapes_Box() { return m_shapes_Box.size(); };
+    unsigned long getNShapes_Cons() { return m_shapes_Cons.size(); };
+    unsigned long getNShapes_Para() { return m_shapes_Para.size(); };
+    unsigned long getNShapes_Pcon() { return m_shapes_Pcon.size(); };
+    unsigned long getNShapes_Pgon() { return m_shapes_Pgon.size(); };
+    unsigned long getNShapes_SimplePolygonBrep() { return m_shapes_SimplePolygonBrep.size(); };
+    unsigned long getNShapes_Trap() { return m_shapes_Trap.size(); };
+    unsigned long getNShapes_Trd() { return m_shapes_Trd.size(); };
+    unsigned long getNShapes_Tube() { return m_shapes_Tube.size(); };
+    unsigned long getNShapes_Tubs() { return m_shapes_Tubs.size(); };
+    unsigned long getNShapes_TwistedTrap() { return m_shapes_TwistedTrap.size(); };
+    unsigned long getNShapes_UnidentifiedShape() { return m_shapes_UnidentifiedShape.size(); };
+    unsigned long getNShapes_Shift() { return m_shapes_Shift.size(); };
+    unsigned long getNShapes_Union() { return m_shapes_Union.size(); };
+    unsigned long getNShapes_Intersection() { return m_shapes_Intersection.size(); };
+    unsigned long getNShapes_Subtraction() { return m_shapes_Subtraction.size(); };
 
    private:
     // define copy constructor, needed for the GeoModelAction subclass
-- 
GitLab