addLinked can now link between vector-valued columns

This commit is contained in:
Philip Eisenlohr 2017-07-31 09:34:30 -04:00
parent 982e925f7b
commit 837b14592a
1 changed files with 15 additions and 9 deletions

View File

@ -48,17 +48,16 @@ if options.asciitable is not None and os.path.isfile(options.asciitable):
buffered = False, buffered = False,
readonly = True) readonly = True)
linkedTable.head_read() # read ASCII header info of linked table linkedTable.head_read() # read ASCII header info of linked table
if linkedTable.label_dimension(options.link[1]) != 1: linkDim = linkedTable.label_dimension(options.link[1]) # dimension of linking column
parser.error('linking column {} needs to be scalar valued.'.format(options.link[1]))
missing_labels = linkedTable.data_readArray([options.link[1]]+options.label) missing_labels = linkedTable.data_readArray([options.link[1]]+options.label) # try reading linked ASCII table
linkedTable.close() # close linked ASCII table linkedTable.close() # close linked ASCII table
if len(missing_labels) > 0: if len(missing_labels) > 0:
damask.util.croak('column{} {} not found...'.format('s' if len(missing_labels) > 1 else '',', '.join(missing_labels))) damask.util.croak('column{} {} not found...'.format('s' if len(missing_labels) > 1 else '',', '.join(missing_labels)))
index = linkedTable.data[:,0] index = linkedTable.data[:,:linkDim]
data = linkedTable.data[:,1:] data = linkedTable.data[:,linkDim:]
else: else:
parser.error('no linked ASCIItable given.') parser.error('no linked ASCIItable given.')
@ -80,8 +79,10 @@ for name in filenames:
errors = [] errors = []
linkColumn = table.label_index(options.link[0]) myLink = table.label_index (options.link[0])
if linkColumn < 0: errors.append('linking column {} not found.'.format(options.link[0])) myLinkDim = table.label_dimension(options.link[0])
if myLink < 0: errors.append('linking column {} not found.'.format(options.link[0]))
if myLinkDim != linkDim: errors.append('dimension mismatch for column {}.'.format(options.link[0]))
if errors != []: if errors != []:
damask.util.croak(errors) damask.util.croak(errors)
@ -91,7 +92,7 @@ for name in filenames:
# ------------------------------------------ assemble header -------------------------------------- # ------------------------------------------ assemble header --------------------------------------
table.info_append(scriptID + '\t' + ' '.join(sys.argv[1:])) table.info_append(scriptID + '\t' + ' '.join(sys.argv[1:]))
table.labels_append(linkedTable.labels(raw = True)[1:]) # extend with new labels (except for linked column) table.labels_append(linkedTable.labels(raw = True)[linkDim:]) # extend with new labels (except for linked column)
table.head_write() table.head_write()
@ -100,7 +101,12 @@ for name in filenames:
outputAlive = True outputAlive = True
while outputAlive and table.data_read(): # read next data line of ASCII table while outputAlive and table.data_read(): # read next data line of ASCII table
try: try:
table.data_append(data[np.argwhere(index == float(table.data[linkColumn]))[0]]) # add data from first matching line table.data_append(
data[
np.argwhere(np.all(
(map(float,table.data[myLink:myLink+myLinkDim]) - index) == 0,
axis = 1))[0]
]) # add data from first matching line
except IndexError: except IndexError:
table.data_append(np.nan*np.ones_like(data[0])) # or add NaNs table.data_append(np.nan*np.ones_like(data[0])) # or add NaNs
outputAlive = table.data_write() # output processed line outputAlive = table.data_write() # output processed line