59 lines
1.8 KiB
Python
59 lines
1.8 KiB
Python
from ..Qt import QtGui, QtCore
|
|
from .Exporter import Exporter
|
|
from ..parametertree import Parameter
|
|
from .. import PlotItem
|
|
|
|
import numpy
|
|
try:
|
|
import h5py
|
|
HAVE_HDF5 = True
|
|
except ImportError:
|
|
HAVE_HDF5 = False
|
|
|
|
__all__ = ['HDF5Exporter']
|
|
|
|
|
|
class HDF5Exporter(Exporter):
|
|
Name = "HDF5 Export: plot (x,y)"
|
|
windows = []
|
|
allowCopy = False
|
|
|
|
def __init__(self, item):
|
|
Exporter.__init__(self, item)
|
|
self.params = Parameter(name='params', type='group', children=[
|
|
{'name': 'Name', 'type': 'str', 'value': 'Export',},
|
|
{'name': 'columnMode', 'type': 'list', 'values': ['(x,y) per plot', '(x,y,y,y) for all plots']},
|
|
])
|
|
|
|
def parameters(self):
|
|
return self.params
|
|
|
|
def export(self, fileName=None):
|
|
if not HAVE_HDF5:
|
|
raise RuntimeError("This exporter requires the h5py package, "
|
|
"but it was not importable.")
|
|
|
|
if not isinstance(self.item, PlotItem):
|
|
raise Exception("Must have a PlotItem selected for HDF5 export.")
|
|
|
|
if fileName is None:
|
|
self.fileSaveDialog(filter=["*.h5", "*.hdf", "*.hd5"])
|
|
return
|
|
dsname = self.params['Name']
|
|
fd = h5py.File(fileName, 'a') # forces append to file... 'w' doesn't seem to "delete/overwrite"
|
|
data = []
|
|
|
|
appendAllX = self.params['columnMode'] == '(x,y) per plot'
|
|
for i,c in enumerate(self.item.curves):
|
|
d = c.getData()
|
|
if appendAllX or i == 0:
|
|
data.append(d[0])
|
|
data.append(d[1])
|
|
|
|
fdata = numpy.array(data).astype('double')
|
|
dset = fd.create_dataset(dsname, data=fdata)
|
|
fd.close()
|
|
|
|
if HAVE_HDF5:
|
|
HDF5Exporter.register()
|