[Python][UHI] Refactor the UHI module

Author: siliataider

bindings/pyroot/pythonizations/CMakeLists.txt

@@ -127,7 +127,10 @@ set(py_sources
   ROOT/_pythonization/_ttree.py
   ROOT/_pythonization/_tvector3.py
   ROOT/_pythonization/_tvectort.py
-  ROOT/_pythonization/_uhi.py
+  ROOT/_pythonization/_uhi/main.py
+  ROOT/_pythonization/_uhi/tags.py
+  ROOT/_pythonization/_uhi/indexing.py
+  ROOT/_pythonization/_uhi/plotting.py
   ${PYROOT_EXTRA_PYTHON_SOURCES}
 )
 

bindings/pyroot/pythonizations/python/ROOT/_facade.py

@@ -479,7 +479,7 @@ def uhi(self):
         uhi_module.__file__ = "<module ROOT>"
         uhi_module.__package__ = self
         try:
-            from ._pythonization._uhi import _add_module_level_uhi_helpers
+            from ._pythonization._uhi.main import _add_module_level_uhi_helpers
 
             _add_module_level_uhi_helpers(uhi_module)
         except ImportError:

bindings/pyroot/pythonizations/python/ROOT/_pythonization/_th1.py

@@ -241,7 +241,7 @@ def _FillWithArrayTH1(self, *args):
 for klass in _th1_derived_classes_to_pythonize:
     pythonization(klass)(inject_constructor_releasing_ownership)
 
-    from ROOT._pythonization._uhi import _add_plotting_features
+    from ROOT._pythonization._uhi.main import _add_plotting_features
 
     # Add UHI plotting features
     pythonization(klass)(_add_plotting_features)
@@ -257,7 +257,7 @@ def _enable_numpy_fill(klass):
 def pythonize_th1(klass):
     # Parameters:
     # klass: class to be pythonized
-    from ROOT._pythonization._uhi import _add_indexing_features
+    from ROOT._pythonization._uhi.main import _add_indexing_features
 
     # Support hist *= scalar
     klass.__imul__ = _imul

bindings/pyroot/pythonizations/python/ROOT/_pythonization/_th2.py

@@ -78,7 +78,7 @@ def _FillWithArrayTH2(self, *args):
 for klass in _th2_derived_classes_to_pythonize:
     pythonization(klass)(inject_constructor_releasing_ownership)
 
-    from ROOT._pythonization._uhi import _add_plotting_features
+    from ROOT._pythonization._uhi.main import _add_plotting_features
 
     # Add UHI plotting features
     pythonization(klass)(_add_plotting_features)

bindings/pyroot/pythonizations/python/ROOT/_pythonization/_th3.py

@@ -27,7 +27,7 @@
 for klass in _th3_derived_classes_to_pythonize:
     pythonization(klass)(inject_constructor_releasing_ownership)
 
-    from ROOT._pythonization._uhi import _add_plotting_features
+    from ROOT._pythonization._uhi.main import _add_plotting_features
 
     # Add UHI plotting features
     pythonization(klass)(_add_plotting_features)