Add python bindings for the datamodel classes

CMakeLists.txt

@@ -108,6 +108,7 @@ add_subdirectory(edm4hep)
 add_subdirectory(utils)
 add_subdirectory(test)
 add_subdirectory(tools)
+add_subdirectory(python)
 
 #--- create uninstall target ---------------------------------------------------
 include(cmake/EDM4HEPUninstall.cmake)

README.md

@@ -72,10 +72,24 @@ make test
 
 The library files and dictionaries (`libedm4hep.so`, ...) must be put somewhere in `LD_LIBRARY_PATH`.
 
-
-
-
-
+## Python bindings
+There are python bindings for all the classes in the datamodel. Make sure that
+`install/lib` is in `LD_LIBRARY_PATH` and `install/python` is in `PYTHONPATH`:
+```python
+from edm4hep import Datamodel
+model = Datamodel() # factory class to provide dynamically loaded classes
+particle = model.MCParticle() # default initialized particle
+particle.getCharge() # 0.0
+
+series = model.TimeSeries(1, 2, 3) # classes can be constructed with non-default parameters
+series.getCellID() # 1
+series.getTime() # 2.0
+series.getInterval() # 3.0
+
+mc = model.MutableMCParticle() # mutable classes can also be modified
+mc.setPDG(2212)
+mc.getPDG() # 2212
+```
 
 ## Contributing
 

python/CMakeLists.txt

@@ -0,0 +1 @@
+install(DIRECTORY edm4hep DESTINATION python)

python/edm4hep/__init__.py

@@ -0,0 +1 @@
+from .datamodel import Datamodel

python/edm4hep/datamodel.py

@@ -0,0 +1,14 @@
+import importlib
+import ROOT
+ROOT.gSystem.Load('libedm4hepDict.so')
+
+
+class Datamodel:
+    def __init__(self):
+        self.edm4hep = importlib.import_module('ROOT').edm4hep
+
+    def __getattr__(self, name):
+        try:
+            return getattr(self.edm4hep, name)
+        except AttributeError:
+            raise AttributeError(f'EDM4hep has no class named {name}')