store requested data in ASCII file

This commit is contained in:
Martin Diehl 2019-05-07 15:18:12 +02:00
parent 1eaf67ad82
commit 0cc2b1214d
1 changed files with 16 additions and 9 deletions

View File

@ -56,14 +56,6 @@ for filename in options.filenames:
data = np.concatenate((data,coords),1) data = np.concatenate((data,coords),1)
header+=' 1_pos 2_pos 3_pos' header+=' 1_pos 2_pos 3_pos'
dirname = os.path.abspath(os.path.join(os.path.dirname(filename),options.dir))
try:
os.mkdir(dirname)
except FileExistsError:
pass
file_out = '{}_inc{:04d}.txt'.format(filename.split('.')[0],i)
np.savetxt(os.path.join(dirname,file_out),data,header=header,comments='')
results.active['increments'] = [inc] results.active['increments'] = [inc]
for label in options.labels: for label in options.labels:
for o in results.c_output_types: for o in results.c_output_types:
@ -73,4 +65,19 @@ for filename in options.filenames:
x = results.get_dataset_location(label) x = results.get_dataset_location(label)
if len(x) == 0: if len(x) == 0:
continue continue
label = x[0].split('/')[-1]
array = results.read_dataset(x,0) array = results.read_dataset(x,0)
d = np.product(np.shape(array)[1:])
array = np.reshape(array,[np.product(results.grid),d])
data = np.concatenate((data,array),1)
header+= ''.join([' {}_{}'.format(j+1,label) for j in range(d)])
dirname = os.path.abspath(os.path.join(os.path.dirname(filename),options.dir))
try:
os.mkdir(dirname)
except FileExistsError:
pass
file_out = '{}_inc{:04d}.txt'.format(filename.split('.')[0],i)
np.savetxt(os.path.join(dirname,file_out),data,header=header,comments='')