1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
|
#/*##########################################################################
# Copyright (C) 2019 European Synchrotron Radiation Facility
#
# This file is part of the PyMca X-ray Fluorescence Toolkit developed at
# the ESRF by the Software group.
#
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the "Software"), to deal
# in the Software without restriction, including without limitation the rights
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
# copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included in
# all copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
# THE SOFTWARE.
#
#############################################################################*/
__author__ = "V.A. Sole - ESRF Data Analysis"
__contact__ = "sole@esrf.fr"
__license__ = "MIT"
__copyright__ = "European Synchrotron Radiation Facility, Grenoble, France"
__doc__ = """This module provides tools to read configurations from ini files
and from their representation in an HDF5 file"""
import sys
if sys.version_info < (3,):
from StringIO import StringIO
else:
from io import StringIO
from PyMca5.PyMcaIO import ConfigDict
from PyMca5.PyMcaGui import PyMcaQt as qt
from PyMca5.PyMcaGui.io import PyMcaFileDialogs
try:
from h5py import is_hdf5
from PyMca5.PyMcaGui.io.hdf5.HDF5Widget import getDatasetUri
HAS_H5PY = True
except ImportError:
HAS_H5PY = False
_HDF5_EXTENSIONS = [".h5", ".hdf5", ".hdf", ".nxs", ".nx"]
def getFitConfigurationFilePath(parent=None, filetypelist=None, message=None,
currentdir=None, mode="OPEN", getfilter=None,
single=True, currentfilter=None, native=None):
"""
Returns a list of fit configuration files or URIs of the form filename::dataset
if an HDF5 dataset is selected.
"""
if filetypelist is None:
filetypelist = ["Fit configuration files (*.cfg)"]
if HAS_H5PY:
filetypelist.append(
"Fit results file (*%s)" % " *".join(_HDF5_EXTENSIONS))
filetypelist.append("All files (*)")
if message is None:
message = "Choose fit configuration file"
return getConfigurationFilePath(parent=parent,
filetypelist=filetypelist,
message=message,
currentdir=currentdir,
mode=mode,
getfilter=getfilter,
single=single,
currentfilter=currentfilter,
native=native)
def getConfigurationFilePath(parent=None, filetypelist=None, message=None,
currentdir=None, mode="OPEN", getfilter=None,
single=True, currentfilter=None, native=None):
if filetypelist is None:
filetypelist = ["Configuration from .ini files (*.ini)"]
if HAS_H5PY:
filetypelist.append(
"Configuration from HDF5 file (*%s)" % " *".join(_HDF5_EXTENSIONS))
filetypelist.append("All files (*)")
if message is None:
message = "Choose configuration file"
fileList = PyMcaFileDialogs.getFileList(parent=parent,
filetypelist=filetypelist, message=message,
currentdir=currentdir,
mode="OPEN", # input ignored
getfilter=getfilter,
single=single,
currentfilter=currentfilter,
native=native)
if getfilter:
fileList, usedfilter = fileList
if HAS_H5PY:
newList = []
for filename in fileList:
if is_hdf5(filename):
# we have to select a dataset
msg = 'Select the configuration dataset by a double click'
uri = getDatasetUri(parent=parent, filename=filename, message=msg)
if uri:
newList.append(uri)
else:
newList.append(filename)
fileList = newList
if getfilter:
return fileList, usedfilter
else:
return fileList
def getFitConfigurationDict(*var, **kw):
selection = getFitConfigurationFilePath(*var, **kw)
if selection:
return ConfigDict.getDictFromPathOrUri(selection)
def getConfigurationDict(*var, **kw):
selection = getConfigurationFilePath(*var, **kw)
if selection:
return ConfigDict.getDictFromPathOrUri(selection)
if __name__ == "__main__":
app = qt.QApplication([])
if len(sys.argv) > 1:
config = ConfigDict.ConfigDict(filelist=sys.argv[1])
else:
config = getFitConfigurationDict()
ConfigDict.prtdict(config)
app = None
|
