python/templates: add references between Objects and Collection types

AIDASoft · Aug 14, 2023 · a2493cc · a2493cc
1 parent dbf9425
commit a2493cc
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/python/templates/Collection.h.jinja2 b/python/templates/Collection.h.jinja2
@@ -48,6 +48,7 @@ A Collection is identified by an ID.
 */
 class {{ class.bare_type }}Collection : public podio::CollectionBase {
 public:
+  using object = {{ class.bare_type }};
   using const_iterator = {{ class.bare_type }}CollectionIterator;
   using iterator = {{ class.bare_type }}MutableCollectionIterator;
 

diff --git a/python/templates/Object.h.jinja2 b/python/templates/Object.h.jinja2
@@ -22,6 +22,7 @@
 
 {{ utils.namespace_open(class.namespace) }}
 class Mutable{{ class.bare_type }};
+class {{ class.bare_type }}Collection;
 
 {{ macros.class_description(class.bare_type, Description, Author) }}
 class {{ class.bare_type }} {
@@ -31,6 +32,9 @@ class {{ class.bare_type }} {
   friend class {{ class.bare_type }}CollectionIterator;
 
 public:
+  using mutable_ = Mutable{{ class.bare_type }};
+  using collection = {{ class.bare_type }}Collection;
+
 {{ macros.constructors_destructors(class.bare_type, Members) }}
 
 public: