gen, templates: generate prefix/default stream definitions

Signed-off-by: Philippe Proulx <eeppeliteloop@gmail.com>

diff --git a/barectf/gen.py b/barectf/gen.py
index 0e71e81545e379fb110fc641bec0b24b14ab6415..6d136e1cd0160cf68866e022b2ec6b074e61f754 100644 (file)
--- a/barectf/gen.py
+++ b/barectf/gen.py
@@ -343,11 +343,23 @@ class CCodeGenerator:
         self._cg.reset()
         dt = datetime.datetime.now().isoformat()
         bh_filename = self.get_bitfield_header_filename()
+        prefix_def = ''
+        default_stream_def = ''
+
+        if self._cfg.options.gen_prefix_def:
+            prefix_def = '#define _BARECTF_PREFIX {}'.format(self._cfg.prefix)
+
+        if self._cfg.options.gen_default_stream_def:
+            s_name = self._cfg.options.gen_default_stream_def
+            default_stream_def = '#define _BARECTF_DEFAULT_STREAM {}'.format(s_name)
+
         tmpl = templates._HEADER_BEGIN
         self._cg.add_lines(tmpl.format(prefix=self._cfg.prefix,
                                        ucprefix=self._cfg.prefix.upper(),
                                        bitfield_header_filename=bh_filename,
-                                       version=barectf.__version__, date=dt))
+                                       version=barectf.__version__, date=dt,
+                                       prefix_def=prefix_def,
+                                       default_stream_def=default_stream_def))
         self._cg.add_empty_line()
 
         # platform callbacks structure
@@ -1000,7 +1012,6 @@ class CCodeGenerator:
                 self._generate_func_trace(stream, ev)
                 self._cg.add_empty_line()
 
-
         return self._cg.code
 
     def get_header_filename(self):

diff --git a/barectf/templates.py b/barectf/templates.py
index b4d901f7eec989ec9863f2daf232a7683a53c801..1a242fd0534c0683e5019b8e125b39e61a25085a 100644 (file)
--- a/barectf/templates.py
+++ b/barectf/templates.py
@@ -236,6 +236,9 @@ _HEADER_BEGIN = '''#ifndef _{ucprefix}H
 
 #include "{bitfield_header_filename}"
 
+{prefix_def}
+{default_stream_def}
+
 struct {prefix}ctx;
 
 uint32_t {prefix}packet_size(void *ctx);