File: igorpackedexperiment.py

package info (click to toggle)
python-igor 0.3-4
  • links: PTS, VCS
  • area: main
  • in suites: bookworm, forky, sid, trixie
  • size: 528 kB
  • sloc: python: 3,461; makefile: 6
file content (54 lines) | stat: -rwxr-xr-x 1,738 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
#!/usr/bin/env python
#
# Copyright (C) 2012 W. Trevor King <wking@tremily.us>
#
# This file is part of igor.
#
# igor is free software: you can redistribute it and/or modify it under the
# terms of the GNU Lesser General Public License as published by the Free
# Software Foundation, either version 3 of the License, or (at your option) any
# later version.
#
# igor is distributed in the hope that it will be useful, but WITHOUT ANY
# WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
# A PARTICULAR PURPOSE.  See the GNU Lesser General Public License for more
# details.
#
# You should have received a copy of the GNU Lesser General Public License
# along with igor.  If not, see <http://www.gnu.org/licenses/>.

"PXP -> ASCII conversion"

import pprint

from igor.packed import load, walk
from igor.record.wave import WaveRecord
from igor.script import Script


class PackedScript (Script):
    def _run(self, args):
        self.args = args
        records,filesystem = load(args.infile)
        if hasattr(args.outfile, 'write'):
            f = args.outfile  # filename is actually a stream object
        else:
            f = open(args.outfile, 'w')
        try:
            f.write(pprint.pformat(records))
            f.write('\n')
        finally:
            if f != args.outfile:
                f.close()
        if args.verbose > 0:
            pprint.pprint(filesystem)
        walk(filesystem, self._plot_wave_callback)

    def _plot_wave_callback(self, dirpath, key, value):
        if isinstance(value, WaveRecord):
            self.plot_wave(self.args, value.wave, title=dirpath + [key])


s = PackedScript(
    description=__doc__, filetype='IGOR Packed Experiment (.pxp) file')
s.run()