diff options
| author | Ben Reynwar <ben@reynwar.net> | 2012-03-09 18:00:20 -0700 |
|---|---|---|
| committer | Ben Reynwar <ben@reynwar.net> | 2012-03-09 18:00:20 -0700 |
| commit | 1c15d2ceefeacb261dcce2f70c3543f70862f4c0 (patch) | |
| tree | de1fd6ea9184895eb8954403ed37b323bf5146d7 /gr-digital/swig | |
| parent | 5fbdd919e9a32710d118fa29018dfb3ffff43ed5 (diff) | |
Adding __repr__ to constellations.
Diffstat (limited to 'gr-digital/swig')
| -rw-r--r-- | gr-digital/swig/digital_constellation.i | 17 |
1 files changed, 17 insertions, 0 deletions
diff --git a/gr-digital/swig/digital_constellation.i b/gr-digital/swig/digital_constellation.i index 7e0ad6afec..acc7aad1a0 100644 --- a/gr-digital/swig/digital_constellation.i +++ b/gr-digital/swig/digital_constellation.i @@ -187,3 +187,20 @@ class digital_constellation_8psk : public digital_constellation public: digital_constellation_8psk (); }; + +/* + We want print(constellation) in python to produce nice useful output so + we include code at the end of the generated python file that overrides + the SWIG-generated __repr__ method. + */ +%pythoncode %{ + +digital_constellation_calcdist_sptr.__repr__ = lambda self: '<constellation calcdist (m=%s)>' % str(len(self.points())) +digital_constellation_rect_sptr.__repr__ = lambda self: '<constellation rect (m=%s)>' % str(len(self.points())) +digital_constellation_psk_sptr.__repr__ = lambda self: '<constellation psk (m=%s)>' % str(len(self.points())) +digital_constellation_bpsk_sptr.__repr__ = lambda self: '<constellation bpsk>' +digital_constellation_qpsk_sptr.__repr__ = lambda self: '<constellation qpsk>' +digital_constellation_dqpsk_sptr.__repr__ = lambda self: '<constellation dqpsk>' +digital_constellation_8psk_sptr.__repr__ = lambda self: '<constellation 8psk>' + +%} |