223031012
/
DAMASK_EICMD
mirror of https://github.com/achalhp/DAMASK_EICMD.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

simplified column check. 

fixed typos and improved help.
This commit is contained in:
Philip Eisenlohr 2015-05-20 23:21:35 +00:00
parent d70b903d1b
commit 439d778cc0
1 changed files with 10 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
processing/post/imageDataRGB.py
View File

@ -20,9 +20,9 @@ Generate PNG image from data in given column vector containing RGB tuples.
""", version = scriptID) """, version = scriptID)
parser.add_option('-l','--label', dest='label', type='string', parser.add_option('-l','--label', dest='label', type='string',
help='column containing RGB triplet)') help='column containing RGB triplet')
parser.add_option('-d','--dimension', dest='dimension', type='int', nargs=2, parser.add_option('-d','--dimension', dest='dimension', type='int', nargs=2,
help='data dimension (width height) [native]') help='data dimension (width height)')
parser.add_option('--fliplr', dest='flipLR', action='store_true', parser.add_option('--fliplr', dest='flipLR', action='store_true',
help='flip around vertical axis') help='flip around vertical axis')
parser.add_option('--flipud', dest='flipUD', action='store_true', parser.add_option('--flipud', dest='flipUD', action='store_true',
@ -77,21 +77,16 @@ for name in filenames:
labels = options.label != None) # no labels when taking 2D dataset labels = options.label != None) # no labels when taking 2D dataset
table.head_read() # read ASCII header info table.head_read() # read ASCII header info
# --------------- figure out column to process -----------------------------------------------------
columns = table.labels_index(["%i_%s"%(i,options.label) for i in [1,2,3]])
if np.any(np.array(columns) == -1):
file['croak'].write('column %s not found...\n'%options.label)
table.close(dismiss = True) # close ASCII table file handles and delete output file
continue
# ------------------------------------------ process data ------------------------------------------ # ------------------------------------------ process data ------------------------------------------
table.data_readArray(columns) missing_labels = table.data_readArray(["%i_%s"%(i,options.label) for i in [1,2,3]])
# convert data to values between 0 and 1 and arrange according to given options if len(missing_labels) > 0:
file['croak'].write('column%s %s not found...\n'%('s' if len(missing_labels) > 1 else '',', '.join(missing_labels)))
table.close(dismiss = True) # close ASCII table file handles and delete output file
continue
# convert data to shape and arrange according to given options
if options.dimension != []: table.data = table.data.reshape(options.dimension[1],options.dimension[0],3) if options.dimension != []: table.data = table.data.reshape(options.dimension[1],options.dimension[0],3)
if options.flipLR: table.data = np.fliplr(table.data) if options.flipLR: table.data = np.fliplr(table.data)
if options.flipUD: table.data = np.flipud(table.data) if options.flipUD: table.data = np.flipud(table.data)