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
|
#!/usr/bin/env python2
# $Id: invert_python.py 22454 2019-08-31 19:53:40Z yeti-dn $
# A very simple Gwyddion plug-in example in Python.
# Written by Yeti <yeti@gwyddion.net>. Public domain.
import sys, os
errmsg = "Plug-in has to be called from Gwyddion plugin-proxy."
if os.environ.has_key('GWYPLUGINLIB'):
sys.path.append(os.path.join(os.environ['GWYPLUGINLIB'], 'python'))
try:
import Gwyddion
except ImportError:
sys.stderr.write(errmsg + '\n')
sys.exit(1)
# Plug-in information.
run_modes = 'noninteractive', 'with_defaults'
plugin_info = """\
invert_python
/_Test/Value Invert (Python)
""" + ' '.join(run_modes)
def register(args):
print plugin_info
def run(args):
run_mode = args.pop(0)
assert run_mode in run_modes
filename = args.pop(0)
data = Gwyddion.dump.read(filename)
dfield = data['/0/data']
a = dfield['data']
n = len(a)
mirror = min(a) + max(a)
for i in range(n):
a[i] = mirror - a[i]
Gwyddion.dump.write(data, filename)
try:
args = sys.argv[1:]
function = globals()[args.pop(0)]
assert callable(function)
except (IndexError, KeyError, AssertionError):
sys.stderr.write(errmsg + '\n')
sys.exit(1)
function(args)
|
