| 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._ArrayValueConst: |
| 28 | s = '[{}]'.format(', '.join([to_string(x) for x in p])) |
| 29 | elif type(p) is bt2._MapValueConst: |
| 30 | s = '{{{}}}'.format( |
| 31 | ', '.join([k + '=' + to_string(p[k]) for k in sorted(p.keys())]) |
| 32 | ) |
| 33 | elif type(p) is bt2._UnsignedIntegerValueConst: |
| 34 | s = str(p) + 'u' |
| 35 | elif ( |
| 36 | type(p) |
| 37 | in ( |
| 38 | bt2._StringValueConst, |
| 39 | bt2._SignedIntegerValueConst, |
| 40 | bt2._RealValueConst, |
| 41 | bt2._BoolValueConst, |
| 42 | ) |
| 43 | or p is None |
| 44 | ): |
| 45 | s = str(p) |
| 46 | |
| 47 | else: |
| 48 | raise TypeError('Unexpected type', type(p)) |
| 49 | |
| 50 | return s |
| 51 | |
| 52 | |
| 53 | @bt2.plugin_component_class |
| 54 | class SinkThatPrintsParams(bt2._UserSinkComponent): |
| 55 | def __init__(self, config, params, obj): |
| 56 | self._add_input_port('in') |
| 57 | print(to_string(params)) |
| 58 | |
| 59 | def _user_consume(self): |
| 60 | raise bt2.Stop |
| 61 | |
| 62 | |
| 63 | bt2.register_plugin(__name__, "params") |