tests/data/cli/params/bt_plugin_params.py

   1 #
   2 # Copyright (C) 2019 EfficiOS Inc.
   3 #
   4 # This program is free software; you can redistribute it and/or
   5 # modify it under the terms of the GNU General Public License
   6 # as published by the Free Software Foundation; only version 2
   7 # of the License.
   8 #
   9 # This program is distributed in the hope that it will be useful,
  10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12 # GNU General Public License for more details.
  13 #
  14 # You should have received a copy of the GNU General Public License
  15 # along with this program; if not, write to the Free Software
  16 # Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
  17 #
  18
  19 import bt2
  20
  21
  22 def to_string(p):
  23     # Print BT values in a predictable way (the order of map entries) and with
  24     # additional information (u suffix to differentiate unsigned integers from
  25     # signed integers).
  26
  27     if type(p) is bt2.ArrayValue:
  28         s = '[{}]'.format(', '.join([to_string(x) for x in p]))
  29     elif type(p) is bt2.MapValue:
  30         s = '{{{}}}'.format(
  31             ', '.join([k + '=' + to_string(p[k]) for k in sorted(p.keys())])
  32         )
  33     elif type(p) is bt2.UnsignedIntegerValue:
  34         s = str(p) + 'u'
  35     else:
  36         s = str(p)
  37
  38     return s
  39
  40
  41 @bt2.plugin_component_class
  42 class SinkThatPrintsParams(bt2._UserSinkComponent):
  43     def __init__(self, config, params, obj):
  44         self._add_input_port('in')
  45         print(to_string(params))
  46
  47     def _user_consume(self):
  48         raise bt2.Stop
  49
  50
  51 bt2.register_plugin(__name__, "params")