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
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
|
#!/usr/bin/env python
# -*- coding: utf-8 -*-
#
# Copyright (C) 2013 Tobias Klauser <tklauser@distanz.ch>
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License version 2 as
# published by the Free Software Foundation.
import os
import sys
import getopt
import numpy as np
DEFAULT_N = 1000
DEFAULT_XMIN = 0.0
DEFAULT_XMAX = 100.0
DEFAULT_YMIN = 0.0
DEFAULT_YMAX = 50.0
def usage():
print("""usage: {} [OPTION...] FILE...
-f force overwrite of existing file
-n N set number of data points to N (default: {})
-x N set minimum x value to N (default: {})
-X N set maximum x value to N (default: {})
-y N set minimum y value to N (default: {})
-Y N set maximum y value to N (default: {})
""".format(os.path.basename(sys.argv[0]), DEFAULT_N, DEFAULT_XMIN, DEFAULT_XMAX,
DEFAULT_YMIN, DEFAULT_YMAX))
def generate_activation_data(N, xmin, xmax, ymin, ymax):
X = np.array([ ((x / float(N)) * (xmax - xmin)) + xmin for x in range(0, N) ])
Y = np.zeros(N)
# Y = np.array([ ((y / float(N)) * (ymax - ymin)) + ymin for y in range(0, N) ])
act = np.array((X, Y)).transpose()
return act
def main():
try:
opts, args = getopt.getopt(sys.argv[1:], "fn:x:X:y:Y:h")
except getopt.GetoptError as err:
print(str(err))
usage()
sys.exit(-1)
overwrite = False
N = DEFAULT_N
xmin = DEFAULT_XMIN
xmax = DEFAULT_XMAX
ymin = DEFAULT_YMIN
ymax = DEFAULT_YMAX
for o, a in opts:
if o == '-f':
overwrite = True
elif o == '-n':
try:
N = int(a)
except ValueError:
print("Error: number of data points must be an integer > 0")
usage()
sys.exit(-1)
elif o == '-x':
try:
xmin = float(a)
except ValueError:
print("Error: xmin must be a float")
usage()
sys.exit(-1)
elif o == '-X':
try:
xmax = float(a)
except ValueError:
print("Error: xmax must be a float")
usage()
sys.exit(-1)
elif o == '-y':
try:
ymin = float(a)
except ValueError:
print("Error: ymin must be a float")
usage()
sys.exit(-1)
elif o == '-Y':
try:
ymax = float(a)
except ValueError:
print("Error: ymax must be a float")
usage()
sys.exit(-1)
elif o == '-h':
usage()
sys.exit(0)
if N <= 0:
print("Error: invalid number of data points ({}), must be > 0".format(N))
usage()
sys.exit(-1)
if len(args) < 1:
print("Error: no output file(s) specified")
usage()
sys.exit(-1)
for a in args:
if not overwrite and os.path.exists(a):
print("Error: output file {} already exists, use -f to overwrite".format(a))
sys.exit(-1)
for a in args:
with open(a, 'w') as f:
act = generate_activation_data(N, xmin, xmax, ymin, ymax)
f.write('x,y\n')
for x,y in act:
f.write('{},{}\n'.format(x,y))
sys.exit(0)
if __name__ == '__main__':
main()
|