centralized functionality for ang import
This commit is contained in:
parent
b4ae91f817
commit
48c21045d7
|
@ -1,8 +1,10 @@
|
||||||
#!/usr/bin/env python2.7
|
#!/usr/bin/env python3
|
||||||
# -*- coding: UTF-8 no BOM -*-
|
|
||||||
|
|
||||||
import os
|
import os
|
||||||
|
import sys
|
||||||
|
from io import StringIO
|
||||||
from optparse import OptionParser
|
from optparse import OptionParser
|
||||||
|
|
||||||
import damask
|
import damask
|
||||||
|
|
||||||
scriptName = os.path.splitext(os.path.basename(__file__))[0]
|
scriptName = os.path.splitext(os.path.basename(__file__))[0]
|
||||||
|
@ -19,47 +21,10 @@ Convert TSL/EDAX *.ang file to ASCIItable
|
||||||
""", version = scriptID)
|
""", version = scriptID)
|
||||||
|
|
||||||
(options, filenames) = parser.parse_args()
|
(options, filenames) = parser.parse_args()
|
||||||
|
|
||||||
# --- loop over input files -------------------------------------------------------------------------
|
|
||||||
|
|
||||||
if filenames == []: filenames = [None]
|
if filenames == []: filenames = [None]
|
||||||
|
|
||||||
for name in filenames:
|
for name in filenames:
|
||||||
try:
|
damask.util.report(scriptName,name)
|
||||||
table = damask.ASCIItable(name = name,
|
|
||||||
outname = os.path.splitext(name)[0]+'.txt' if name else name,
|
|
||||||
buffered = False, labeled = False)
|
|
||||||
except: continue
|
|
||||||
damask.util.report(scriptName,name)
|
|
||||||
|
|
||||||
# --- interpret header -----------------------------------------------------------------------------
|
table = damask.Table.from_ang(StringIO(''.join(sys.stdin.read())) if name is None else name)
|
||||||
|
table.to_ASCII(sys.stdout if name is None else os.path.splitext(name)[0]+'.txt')
|
||||||
table.head_read()
|
|
||||||
|
|
||||||
# --- read comments --------------------------------------------------------------------------------
|
|
||||||
|
|
||||||
table.info_clear()
|
|
||||||
while table.data_read(advance = False) and table.line.startswith('#'): # cautiously (non-progressing) read header
|
|
||||||
table.info_append(table.line) # add comment to info part
|
|
||||||
table.data_read() # wind forward
|
|
||||||
|
|
||||||
table.labels_clear()
|
|
||||||
table.labels_append(['1_Euler','2_Euler','3_Euler',
|
|
||||||
'1_pos','2_pos',
|
|
||||||
'IQ','CI','PhaseID','Intensity','Fit',
|
|
||||||
], # OIM Analysis 7.2 Manual, p 403 (of 517)
|
|
||||||
reset = True)
|
|
||||||
|
|
||||||
# ------------------------------------------ assemble header ---------------------------------------
|
|
||||||
|
|
||||||
table.head_write()
|
|
||||||
|
|
||||||
#--- write remainder of data file ------------------------------------------------------------------
|
|
||||||
|
|
||||||
outputAlive = True
|
|
||||||
while outputAlive and table.data_read():
|
|
||||||
outputAlive = table.data_write()
|
|
||||||
|
|
||||||
# ------------------------------------------ finalize output ---------------------------------------
|
|
||||||
|
|
||||||
table.close()
|
|
||||||
|
|
|
@ -3,6 +3,8 @@ import re
|
||||||
import pandas as pd
|
import pandas as pd
|
||||||
import numpy as np
|
import numpy as np
|
||||||
|
|
||||||
|
from . import version
|
||||||
|
|
||||||
class Table():
|
class Table():
|
||||||
"""Store spreadsheet-like data."""
|
"""Store spreadsheet-like data."""
|
||||||
|
|
||||||
|
@ -20,7 +22,7 @@ class Table():
|
||||||
Additional, human-readable information.
|
Additional, human-readable information.
|
||||||
|
|
||||||
"""
|
"""
|
||||||
self.comments = [] if comments is None else [c for c in comments]
|
self.comments = ['table.py v {}'.format(version)] if not comments else [c for c in comments]
|
||||||
self.data = pd.DataFrame(data=data)
|
self.data = pd.DataFrame(data=data)
|
||||||
self.shapes = shapes
|
self.shapes = shapes
|
||||||
self.__label_condensed()
|
self.__label_condensed()
|
||||||
|
@ -69,13 +71,16 @@ class Table():
|
||||||
f = open(fname)
|
f = open(fname)
|
||||||
except TypeError:
|
except TypeError:
|
||||||
f = fname
|
f = fname
|
||||||
|
f.seek(0)
|
||||||
|
|
||||||
header,keyword = f.readline().split()
|
header,keyword = f.readline().split()
|
||||||
if keyword == 'header':
|
if keyword == 'header':
|
||||||
header = int(header)
|
header = int(header)
|
||||||
else:
|
else:
|
||||||
raise Exception
|
raise Exception
|
||||||
comments = [f.readline()[:-1] for i in range(1,header)]
|
|
||||||
|
comments = ['table.py:from_ASCII v {}'.format(version)]
|
||||||
|
comments+= [f.readline()[:-1] for i in range(1,header)]
|
||||||
labels = f.readline().split()
|
labels = f.readline().split()
|
||||||
|
|
||||||
shapes = {}
|
shapes = {}
|
||||||
|
@ -95,6 +100,47 @@ class Table():
|
||||||
|
|
||||||
return Table(data,shapes,comments)
|
return Table(data,shapes,comments)
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def from_ang(fname):
|
||||||
|
"""
|
||||||
|
Create table from TSL ang file.
|
||||||
|
|
||||||
|
A valid TSL ang file needs to contains the following columns:
|
||||||
|
* Euler angles (Bunge notation) in radians, 3 floats, label 'eu'.
|
||||||
|
* Spatial position in meters, 2 floats, label 'pos'.
|
||||||
|
* Image quality, 1 float, label 'IQ'.
|
||||||
|
* Confidence index, 1 float, label 'CI'.
|
||||||
|
* Phase ID, 1 int, label 'ID'.
|
||||||
|
* SEM signal, 1 float, label 'intensity'.
|
||||||
|
* Fit, 1 float, label 'fit'.
|
||||||
|
|
||||||
|
Parameters
|
||||||
|
----------
|
||||||
|
fname : file, str, or pathlib.Path
|
||||||
|
Filename or file for reading.
|
||||||
|
|
||||||
|
"""
|
||||||
|
shapes = {'eu':(3,), 'pos':(2,),
|
||||||
|
'IQ':(1,), 'CI':(1,), 'ID':(1,), 'intensity':(1,), 'fit':(1,)}
|
||||||
|
try:
|
||||||
|
f = open(fname)
|
||||||
|
except TypeError:
|
||||||
|
f = fname
|
||||||
|
f.seek(0)
|
||||||
|
|
||||||
|
content = f.readlines()
|
||||||
|
|
||||||
|
comments = ['table.py:from_ang v {}'.format(version)]
|
||||||
|
for line in content:
|
||||||
|
if line.startswith('#'):
|
||||||
|
comments.append(line.strip())
|
||||||
|
else:
|
||||||
|
break
|
||||||
|
|
||||||
|
data = np.loadtxt(content)
|
||||||
|
|
||||||
|
return Table(data,shapes,comments)
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def labels(self):
|
def labels(self):
|
||||||
return list(self.shapes.keys())
|
return list(self.shapes.keys())
|
||||||
|
|
|
@ -0,0 +1,138 @@
|
||||||
|
# TEM_PIXperUM 1.000000
|
||||||
|
# x-star 240.000000
|
||||||
|
# y-star 240.000000
|
||||||
|
# z-star 240.000000
|
||||||
|
# WorkingDistance 20.000000
|
||||||
|
#
|
||||||
|
# Phase 1
|
||||||
|
# MaterialName Iron(Alpha)
|
||||||
|
# Formula
|
||||||
|
# Info
|
||||||
|
# Symmetry 43
|
||||||
|
# LatticeConstants 2.870 2.870 2.870 90.000 90.000 90.000
|
||||||
|
# NumberFamilies 100
|
||||||
|
# hklFamilies 9223440 0 2 32763 0.000000 32763
|
||||||
|
# hklFamilies 0 0 0 9218712 0.000000 9218712
|
||||||
|
# hklFamilies 0 0 3801155 0 0.000000 0
|
||||||
|
# hklFamilies 5570652 6619251 7536754 -1203738484 0.000000 -1203738484
|
||||||
|
# hklFamilies 7143516 5111900 7864421 32763 0.000000 32763
|
||||||
|
# hklFamilies 6488180 7274604 6553717 9220480 0.000000 9220480
|
||||||
|
# hklFamilies 3145820 2949169 3145777 0 0.000000 0
|
||||||
|
# hklFamilies 3014704 7209057 103 9220488 0.000000 9220488
|
||||||
|
# hklFamilies 0 0 0 0 0.000000 0
|
||||||
|
# hklFamilies 0 0 0 9220032 0.000000 9220032
|
||||||
|
# hklFamilies 0 0 0 0 0.000000 0
|
||||||
|
# hklFamilies 0 0 0 -1203728363 0.000000 -1203728363
|
||||||
|
# hklFamilies 0 0 0 32763 0.000000 32763
|
||||||
|
# hklFamilies 0 0 0 9218628 0.000000 9218628
|
||||||
|
# hklFamilies 0 0 0 0 0.000000 0
|
||||||
|
# hklFamilies 0 0 0 9218504 0.000000 9218504
|
||||||
|
# hklFamilies 0 0 0 0 0.000000 0
|
||||||
|
# hklFamilies 0 0 0 9219904 0.000000 9219904
|
||||||
|
# hklFamilies 0 0 0 0 0.000000 0
|
||||||
|
# hklFamilies 0 0 0 0 -0.000046 0
|
||||||
|
# hklFamilies 0 0 0 0 0.000000 0
|
||||||
|
# hklFamilies 0 0 0 256 0.000000 256
|
||||||
|
# hklFamilies 0 0 0 0 0.000000 0
|
||||||
|
# hklFamilies 0 0 0 -1203753636 0.000000 -1203753636
|
||||||
|
# hklFamilies 0 0 0 32763 0.000000 32763
|
||||||
|
# hklFamilies 0 0 0 9220576 0.000000 9220576
|
||||||
|
# hklFamilies 0 0 0 0 0.000000 0
|
||||||
|
# hklFamilies 0 0 0 9218736 0.000000 9218736
|
||||||
|
# hklFamilies 0 0 0 0 0.000000 0
|
||||||
|
# hklFamilies 0 0 0 103219574 0.000000 103219574
|
||||||
|
# hklFamilies 0 0 0 0 0.000000 0
|
||||||
|
# hklFamilies 0 0 0 9220576 0.000000 9220576
|
||||||
|
# hklFamilies 0 0 0 0 0.000000 0
|
||||||
|
# hklFamilies 0 0 0 9220692 0.000000 9220692
|
||||||
|
# hklFamilies 1434293657 0 0 0 0.000000 0
|
||||||
|
# hklFamilies 0 0 0 9218584 0.000000 9218584
|
||||||
|
# hklFamilies 0 0 0 0 0.000000 0
|
||||||
|
# hklFamilies 0 0 0 9219976 0.000000 9219976
|
||||||
|
# hklFamilies 0 0 0 0 0.000000 0
|
||||||
|
# hklFamilies 0 0 0 0 0.000000 0
|
||||||
|
# hklFamilies 0 0 0 0 0.000000 0
|
||||||
|
# hklFamilies 0 0 0 256 0.000000 256
|
||||||
|
# hklFamilies 0 0 69473872 0 0.000000 0
|
||||||
|
# hklFamilies 0 1889785611 -1546188227 -1203753636 -0.000046 -1203753636
|
||||||
|
# hklFamilies 9224144 0 1434294456 32763 0.000000 32763
|
||||||
|
# hklFamilies 0 9224160 0 9220672 0.000000 9220672
|
||||||
|
# hklFamilies -1168390977 32763 851982 0 0.000000 0
|
||||||
|
# hklFamilies 0 304 0 9218816 0.000000 9218816
|
||||||
|
# hklFamilies 27030208 0 1434297593 0 0.000000 0
|
||||||
|
# hklFamilies 0 9224160 0 101654020 0.000000 101654020
|
||||||
|
# hklFamilies 9224064 0 0 0 0.000000 0
|
||||||
|
# hklFamilies 0 25563456 0 9220672 0.000000 9220672
|
||||||
|
# hklFamilies 9224544 0 25559040 0 0.000000 0
|
||||||
|
# hklFamilies 0 25559788 0 9220788 0.000000 9220788
|
||||||
|
# hklFamilies 176 0 304 24 0.000000 24
|
||||||
|
# hklFamilies 0 25562304 0 4 0.000000 4
|
||||||
|
# hklFamilies 9224208 0 0 0 0.000000 0
|
||||||
|
# hklFamilies 0 281 0 9220032 0.000000 9220032
|
||||||
|
# hklFamilies 0 0 0 0 0.000000 0
|
||||||
|
# hklFamilies 0 -1168390977 32763 9220660 0.000000 9220660
|
||||||
|
# hklFamilies 21 0 -1168390977 8 0.000000 8
|
||||||
|
# hklFamilies 32763 2490388 0 24 0.000000 24
|
||||||
|
# hklFamilies 48 0 69650048 25 0.000000 25
|
||||||
|
# hklFamilies 0 -1216995621 32763 65535 -0.000046 65535
|
||||||
|
# hklFamilies 0 0 25562688 1 0.000000 1
|
||||||
|
# hklFamilies 0 0 21776 0 -0.000058 0
|
||||||
|
# hklFamilies 25562688 0 25559724 0 0.000000 0
|
||||||
|
# hklFamilies 0 25559040 0 1179652 0.000000 1179652
|
||||||
|
# hklFamilies 25559724 0 25562304 32763 0.000000 32763
|
||||||
|
# hklFamilies 0 48 0 9219904 0.000000 9219904
|
||||||
|
# hklFamilies 25562304 0 28 0 0.000000 0
|
||||||
|
# hklFamilies 0 0 0 8781958 0.000000 8781958
|
||||||
|
# hklFamilies 31 0 0 0 0.000000 0
|
||||||
|
# hklFamilies 0 0 0 103304392 0.000000 103304392
|
||||||
|
# hklFamilies 3 0 48 0 0.000000 0
|
||||||
|
# hklFamilies 0 9224505 0 103219694 -0.000046 103219694
|
||||||
|
# hklFamilies 27000832 0 -1168393705 0 0.000000 0
|
||||||
|
# hklFamilies 32763 25559040 0 9220192 0.000000 9220192
|
||||||
|
# hklFamilies 0 32763 31 0 0.000000 0
|
||||||
|
# hklFamilies 0 0 0 9219872 0.000000 9219872
|
||||||
|
# hklFamilies 69729712 0 9224640 0 0.000000 0
|
||||||
|
# hklFamilies 0 69729904 0 1397706823 0.000000 1397706823
|
||||||
|
# hklFamilies 69911504 0 0 59 0.000000 59
|
||||||
|
# hklFamilies 0 27007968 0 103219200 0.000000 103219200
|
||||||
|
# hklFamilies 0 0 -1216843775 0 0.000000 0
|
||||||
|
# hklFamilies 32763 69911504 0 0 0.000000 0
|
||||||
|
# hklFamilies -1168296496 32763 9225328 0 0.000000 0
|
||||||
|
# hklFamilies 0 1434343267 0 9632160 0.000000 9632160
|
||||||
|
# hklFamilies 69908840 0 -1216995621 0 0.000000 0
|
||||||
|
# hklFamilies 32763 256 0 9632112 0.000000 9632112
|
||||||
|
# hklFamilies 0 0 399376220 0 0.000000 0
|
||||||
|
# hklFamilies 21776 1966087 4456474 262148 0.000000 262148
|
||||||
|
# hklFamilies 9224704 0 1434198234 0 0.000000 0
|
||||||
|
# hklFamilies 0 0 0 9704044 0.000000 9704044
|
||||||
|
# hklFamilies -1168373699 32763 1 0 0.000000 0
|
||||||
|
# hklFamilies 0 69911504 0 94961568 -0.000046 94961568
|
||||||
|
# hklFamilies 1 0 69911504 0 0.000000 0
|
||||||
|
# hklFamilies 0 10 0 9220016 0.000000 9220016
|
||||||
|
# hklFamilies -1 0 27030208 0 0.000000 0
|
||||||
|
# hklFamilies 0 1434488087 18 9219992 -0.000046 9219992
|
||||||
|
# ElasticConstants 0.000000 0.000000 0.000000 0.000000 0.000000 0.000000
|
||||||
|
# ElasticConstants 0.000000 0.000000 0.000000 0.000000 0.000000 0.000000
|
||||||
|
# ElasticConstants 0.000000 0.000000 0.000000 0.000000 0.000000 0.000000
|
||||||
|
# ElasticConstants 0.000000 0.000000 0.000000 0.000000 0.000000 0.000000
|
||||||
|
# ElasticConstants 0.000000 0.000000 0.000000 0.000000 0.000000 0.000000
|
||||||
|
# ElasticConstants 0.000000 0.000000 0.000000 0.000000 0.000000 0.000000
|
||||||
|
# Categories1 1 1 1 1
|
||||||
|
#
|
||||||
|
# GRID: SqrGrid
|
||||||
|
# XSTEP: 0.050000
|
||||||
|
# YSTEP: 0.050000
|
||||||
|
# NCOLS_ODD: 2
|
||||||
|
# NCOLS_EVEN: 2
|
||||||
|
# NROWS: 2
|
||||||
|
#
|
||||||
|
# OPERATOR:
|
||||||
|
#
|
||||||
|
# SAMPLEID:
|
||||||
|
#
|
||||||
|
# SCANID:
|
||||||
|
#
|
||||||
|
0.0 0.0 0.0 0.00 0.00 60.0 20.0 1 2.0 1.5
|
||||||
|
0.0 2.0 0.0 0.05 0.00 60.0 20.0 1 2.0 1.5
|
||||||
|
0.0 2.0 0.0 0.00 0.05 60.0 20.0 1 2.0 1.5
|
||||||
|
0.0 0.0 1.0 0.05 0.05 60.0 20.0 1 2.0 1.5
|
|
@ -47,6 +47,17 @@ class TestTable:
|
||||||
new = Table.from_ASCII(f)
|
new = Table.from_ASCII(f)
|
||||||
assert all(default.data==new.data)
|
assert all(default.data==new.data)
|
||||||
|
|
||||||
|
def test_read_ang_str(self,reference_dir):
|
||||||
|
new = Table.from_ang(os.path.join(reference_dir,'simple.ang'))
|
||||||
|
assert new.data.shape == (4,10) and \
|
||||||
|
new.labels == ['eu', 'pos', 'IQ', 'CI', 'ID', 'intensity', 'fit']
|
||||||
|
|
||||||
|
def test_read_ang_file(self,reference_dir):
|
||||||
|
f = open(os.path.join(reference_dir,'simple.ang'))
|
||||||
|
new = Table.from_ang(f)
|
||||||
|
assert new.data.shape == (4,10) and \
|
||||||
|
new.labels == ['eu', 'pos', 'IQ', 'CI', 'ID', 'intensity', 'fit']
|
||||||
|
|
||||||
@pytest.mark.parametrize('fname',['datatype-mix.txt','whitespace-mix.txt'])
|
@pytest.mark.parametrize('fname',['datatype-mix.txt','whitespace-mix.txt'])
|
||||||
def test_read_strange(self,reference_dir,fname):
|
def test_read_strange(self,reference_dir,fname):
|
||||||
with open(os.path.join(reference_dir,fname)) as f:
|
with open(os.path.join(reference_dir,fname)) as f:
|
||||||
|
|
Loading…
Reference in New Issue