From d87b70fe4958f0bb14fd3ad775c7c94783adfeac Mon Sep 17 00:00:00 2001 From: Martin Diehl Date: Sat, 9 May 2015 21:33:17 +0000 Subject: [PATCH] filterTable can do the job --- processing/post/deleteColumn.py | 104 -------------------------------- 1 file changed, 104 deletions(-) delete mode 100755 processing/post/deleteColumn.py diff --git a/processing/post/deleteColumn.py b/processing/post/deleteColumn.py deleted file mode 100755 index 1879afac1..000000000 --- a/processing/post/deleteColumn.py +++ /dev/null @@ -1,104 +0,0 @@ -#!/usr/bin/env python -# -*- coding: UTF-8 no BOM -*- - -import os,sys,string -from optparse import OptionParser -import damask - -scriptID = string.replace('$Id$','\n','\\n') -scriptName = os.path.splitext(scriptID.split()[1])[0] - -# -------------------------------------------------------------------- -# MAIN -# -------------------------------------------------------------------- - -parser = OptionParser(option_class=damask.extendableOption, usage='%prog options [file[s]]', description = """ -Remove column(s) containing scalar, vectorial, and/or tensorial data. - -""", version = scriptID) - -parser.add_option('-v','--vector', dest='vector', action='extend', metavar='', - help='heading of columns containing 3x1 vector field values') -parser.add_option('-t','--tensor', dest='tensor', action='extend', metavar='', - help='heading of columns containing 3x3 tensor field values') -parser.add_option('-s','--special', dest='special', action='extend', metavar='', - help='heading of columns containing field values of special dimension') -parser.add_option('-d','--dimension', dest='N', type='int', metavar='int', - help='dimension of special field values [%default]') - -parser.set_defaults(vector = []) -parser.set_defaults(tensor = []) -parser.set_defaults(special = []) -parser.set_defaults(N = 1) - -(options,filenames) = parser.parse_args() - -if len(options.vector) + len(options.tensor) + len(options.special)== 0: - parser.error('no data column specified...') - -datainfo = { # list of requested labels per datatype - 'vector': {'len':3, - 'label':[]}, - 'tensor': {'len':9, - 'label':[]}, - 'special': {'len':options.N, - 'label':[]}, - } - -if options.vector != None: datainfo['vector']['label'] += options.vector -if options.tensor != None: datainfo['tensor']['label'] += options.tensor -if options.special != None: datainfo['special']['label'] += options.special - -# ------------------------------------------ setup file handles ------------------------------------ -files = [] -if filenames == []: - files.append({'name':'STDIN', 'input':sys.stdin, 'output':sys.stdout, 'croak':sys.stderr}) -else: - for name in filenames: - if os.path.exists(name): - files.append({'name':name, 'input':open(name), 'output':open(name+'_tmp','w'), 'croak':sys.stderr}) - -#--- loop over input files ------------------------------------------------------------------------- -for file in files: - if file['name'] != 'STDIN': file['croak'].write('\033[1m'+scriptName+'\033[0m: '+file['name']+'\n') - else: file['croak'].write('\033[1m'+scriptName+'\033[0m\n') - - table = damask.ASCIItable(file['input'],file['output'],False) # make unbuffered ASCII_table - table.head_read() # read ASCII header info - table.info_append(scriptID + '\t' + ' '.join(sys.argv[1:])) - -# --------------- figure out columns to delete ---------------------------------------------------- - columns = [] - for datatype,info in datainfo.items(): - for label in info['label']: - key = {True :'1_%s', - False:'%s' }[info['len']>1]%label - if key not in table.labels: - sys.stderr.write('column %s not found...\n'%key) - else: - columns.append([table.labels.index(key),info['len']]) # remember column and extent of requested data - if (info['len'] == 1): - table.labels.remove(label) # remove single column head - else: - for i in xrange(info['len']): - table.labels.remove('%i_%s'%(i+1,label)) # remove multidimensional column head - - columns.sort(key=lambda x:x[0],reverse=True) # sort from highest column to delete backwards - -# ------------------------------------------ assemble header --------------------------------------- - table.head_write() - -# ------------------------------------------ process data ------------------------------------------ - outputAlive = True - while outputAlive and table.data_read(): # read next data line of ASCII table - for col,len in columns: # loop over removal candidates - del table.data[col:col+len] # remove each associated entry - outputAlive = table.data_write() # output processed line - -# ------------------------------------------ output result ----------------------------------------- - outputAlive and table.output_flush() # just in case of buffered ASCII table - - table.input_close() # close input ASCII table (works for stdin) - table.output_close() # close output ASCII table (works for stdout) - if file['name'] != 'STDIN': - os.rename(file['name']+'_tmp',file['name']) # overwrite old one with tmp new