igor-public/pearl/pearl-area-import.ipf
matthias muntwiler 80a01f2bdb updates: pshell import, angle-scans, elog
- pshell import: fix units and data scaling.
- pshell import: support new multi-region scans.
- angle scans: add trim function.
- angle scans: update import_tpi_scan function.
- angle scans: fix scales of check waves in normalization.
- area display: new cursor mode for background selection.
- elog: bugfixes (attachment list, check existing logbook).
2017-07-04 11:06:49 +02:00

1755 lines
49 KiB
Igor

#pragma rtGlobals=3 // Use modern global access method and strict wave access.
#pragma IgorVersion = 6.2
#pragma ModuleName = PearlAreaImport
#pragma version = 1.06
#include <HDF5 Browser>
#include "pearl-gui-tools"
// copyright (c) 2013-16 Paul Scherrer Institut
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
// http:///www.apache.org/licenses/LICENSE-2.0
/// @file
/// @brief HDF5 file import from EPICS area detectors
/// @ingroup ArpesPackage
///
///
/// HDF5 file import from EPICS area detectors
/// such as CCD cameras, 2D electron analysers
///
/// as of Igor 6.3, Igor can open datasets of up to rank 4.
/// i.e. the extra dimension Y of the file plugin cannot be used.
/// the extra dimensions N and X are supported.
/// @namespace PearlAreaImport
/// @brief HDF5 file import from EPICS area detectors
///
/// PearlAreaImport is declared in @ref pearl-area-import.ipf.
/// callback function for drag&drop of HDF5 files into Igor.
///
static function BeforeFileOpenHook(refNum,fileName,path,type,creator,kind)
variable refNum, kind
string fileName, path, type, creator
variable handledOpen = 0
//PathInfo $path
//string FilePath = s_path + filename
string NickName = CleanupName(ParseFilePath(3, FileName, ":", 0, 0), 0)
string FileExt = LowerStr(ParseFilePath(4, FileName, ":", 0, 0))
string result = ""
// override nickname with custom setting
svar /z cnn = gsCustomNickName
if (svar_exists(cnn))
if (exists("gvNickNameIndex") != 2)
variable/g gvNickNameIndex = 1
endif
nvar nni = gvNickNameIndex
NickName = cnn + num2str(nni)
nni += 1
endif
if (stringmatch(FileExt, "h5") == 1)
result = adh5_load_complete(NickName, path, FileName)
endif
string/g s_latest_datafile = result
string/g s_latest_nickname = nickname
handledOpen = strlen(result) > 0
if (handledOpen)
close refnum
endif
return handledOpen // 1 tells Igor not to open the file
End
/// generate the name of a data folder based on a file name.
///
/// if the file name follows the naming convention source-date-index.extension,
/// the function tries to generate the nick name as source_date_index.
/// otherwise it's just a cleaned up version of the file name.
///
/// date must be in yyyymmdd or yymmdd format and is clipped to the short yymmdd format.
/// index should be a running numeric index of up to 6 digits, or the time encoded as hhmmss.
/// however, in the current version index can be any string that can be a valid Igor folder name.
///
/// @param filename file name, including extension. can also include a folder path (which is ignored).
/// the extension is currently ignored, but may be used to select the parent folder in a later version.
/// @param ignoredate if non-zero, the nick name will not include the date part.
/// defaults to zero.
/// @param sourcename nick name of the data source.
/// by default, the function tries to detect the source from the file name.
/// this option can be used to override auto-detection.
/// the automatic source names are:
/// sci (scienta by area detector),
/// psh (pshell),
/// sl (optics slit camera by area detector),
/// es (end station camera by area detector),
/// xy (unidentified).
/// @param unique if non-zero, the resulting name is made a unique data folder name in the current data folder
/// defaults to zero.
///
function /s ad_suggest_foldername(filename, [ignoredate,sourcename,unique])
string filename
variable ignoredate
string sourcename
variable unique
if (ParamIsDefault(ignoredate))
ignoredate = 0
endif
if (ParamIsDefault(unique))
unique = 0
endif
string basename = ParseFilePath(3, filename, ":", 0, 0)
string extension = ParseFilePath(4, filename, ":", 0, 0)
string nickname
string autosource
if (strsearch(basename, "scienta", 0, 2) >= 0)
autosource = "sci"
elseif (strsearch(basename, "pshell", 0, 2) >= 0)
autosource = "psh"
elseif (strsearch(basename, "OP-SL", 0, 2) >= 0)
autosource = "sl"
elseif (strsearch(basename, "ES-PS", 0, 2) >= 0)
autosource = "es"
else
autosource = "xy"
endif
if (ParamIsDefault(sourcename))
sourcename = autosource
endif
variable nparts = ItemsInList(basename, "-")
if (nparts >= 3)
string datepart = StringFromList(1, basename, "-")
variable l_datepart = strlen(datepart)
if (l_datepart == 8)
datepart = datepart[l_datepart-6, l_datepart-1]
endif
string indexpart = StringFromList(2, basename, "-")
if (ignoredate)
sprintf nickname, "%s_%s", sourcename, indexpart
else
sprintf nickname, "%s_%s_%s", sourcename, datepart, indexpart
endif
else
nickname = CleanupName(basename, 0)
endif
if (unique && CheckName(nickname, 11))
nickname = UniqueName(nickname + "_", 11, 0)
endif
return nickname
end
/// load area detector data files selected in a file dialog window
///
/// @param APathName Igor symbolic path name.
/// if empty, Igor will choose a folder on its own
function ad_load_dialog(APathName)
string APathName
variable refNum
string message = "Select data files"
string filepaths
string filefilters = "Area Detector HDF5 Files (*.h5):.h5;"
filefilters += "All Files:.*;"
PathInfo /S $APathName
Open /D /R /F=filefilters /M=message /MULT=1 refNum
filepaths = S_fileName
dfref saveDF = GetDataFolderDFR()
setdatafolder root:
if (strlen(filepaths) > 0)
variable nfiles = ItemsInList(filepaths, "\r")
variable ifile
for(ifile = 0; ifile < nfiles; ifile += 1)
String path = StringFromList(ifile, filepaths, "\r")
string nickname = ad_suggest_foldername(path)
adh5_load_complete(nickname, "", path)
endfor
endif
setdatafolder saveDF
end
/// import everything from a HDF5 file created by the Area Detector software.
///
/// if the data is from the electron analyser driver and some special attributes are included,
/// the function will set the scales of the image dimensions.
///
/// @param ANickName destination folder name (top level under root)
/// @param APathName igor symbolic path name. can be empty if the path is specified in FileName or a dialog box should be displayed
/// @param AFileName if empty a dialog box shows up
/// @param load_data 1 (default): load data; 0: do not load data
/// @param load_attr 1 (default): load attributes; 0: do not load attributes
/// for proper wave scaling, the attributes must be loaded
function /s adh5_load_complete(ANickName, APathName, AFileName, [load_data, load_attr])
string ANickName
string APathName
string AFileName
variable load_data
variable load_attr
if (ParamIsDefault(load_data))
load_data = 1
endif
if (ParamIsDefault(load_attr))
load_attr = 1
endif
dfref saveDF = GetDataFolderDFR()
setdatafolder root:
newdatafolder /s/o $("root:" + ANickName)
// open file
variable fileID
string instrumentpath = "/entry/instrument/"
string detectorpath = instrumentpath + "detector/"
string attributespath = instrumentpath + "NDAttributes/"
string datasetname
string datawavename
// performance monitoring
variable timerRefNum
variable /g adh5_perf_secs
timerRefNum = startMSTimer
// avoid compilation error if HDF5 XOP has not been loaded
#if Exists("HDF5OpenFile")
HDF5OpenFile /P=$APathName/R fileID as AFileName
if (v_flag == 0)
AFileName = s_path + s_filename
print "loading " + s_filename + "\r"
if (load_data)
adh5_load_detector_slabs(fileID, detectorpath)
endif
if (load_attr)
newdatafolder /o/s attr
adh5_loadattr_all(fileID, attributespath)
setdatafolder ::
endif
wave /z data
if (waveexists(data))
//adh5_redim(data) // not to be used with adh5_load_detector_slabs
adh5_scale(data)
endif
HDF5CloseFile fileID
else
AFileName = ""
endif
#else
Abort "HDF5 XOP not loaded."
#endif
if (timerRefNum >= 0)
adh5_perf_secs = stopMSTimer(timerRefNum) / 1e6
endif
setdatafolder saveDF
return AFileName
end
/// load and reduce a dataset from a HDF5 file created by the Area Detector software.
///
/// the resulting dataset is reduced in one image dimension by a user-defined reduction function,
/// e.g. by region-of-interest integration, curve fitting, etc.
///
/// the function loads the dataset image by image using the hyperslab option
/// and applies a custom reduction function to each image.
/// the results from the reduction function are composed into one result wave.
/// the raw data are discarded.
///
/// if the data is from the electron analyser driver and some special attributes are included,
/// the function will set the scales of the image dimensions.
///
/// @param ANickName destination folder name (top level under root)
/// @param APathName igor symbolic path name. can be empty if the path is specified in FileName or a dialog box should be displayed
/// @param AFileName if empty a dialog box shows up
///
/// @param reduction_func custom reduction function
/// (any user-defined function which has the same parameters as adh5_default_reduction())
/// @param reduction_param parameter string for the reduction function
///
/// @param load_data 1 (default): load data; 0: do not load data
/// @param load_attr 1 (default): load attributes; 0: do not load attributes
/// for proper wave scaling, the attributes must be loaded
/// @param progress 1 (default): show progress window; 0: do not show progress window
///
function /s adh5_load_reduced(ANickName, APathName, AFileName, reduction_func, reduction_param, [load_data, load_attr, progress])
string ANickName
string APathName
string AFileName
funcref adh5_default_reduction reduction_func
string reduction_param
variable load_data
variable load_attr
variable progress
if (ParamIsDefault(load_data))
load_data = 1
endif
if (ParamIsDefault(load_attr))
load_attr = 1
endif
if (ParamIsDefault(progress))
progress = 1
endif
dfref saveDF = GetDataFolderDFR()
setdatafolder root:
newdatafolder /s/o $("root:" + ANickName)
// open file
variable fileID
string instrumentpath = "/entry/instrument/"
string detectorpath = instrumentpath + "detector/"
string attributespath = instrumentpath + "NDAttributes/"
string datasetname
string datawavename
// performance monitoring
variable timerRefNum
variable /g adh5_perf_secs
timerRefNum = startMSTimer
// avoid compilation error if HDF5 XOP has not been loaded
#if Exists("HDF5OpenFile")
HDF5OpenFile /P=$APathName/R fileID as AFileName
if (v_flag == 0)
AFileName = s_path + s_filename
print "loading " + s_filename + "\r"
if (load_attr)
newdatafolder /o/s attr
adh5_loadattr_all(fileID, attributespath)
setdatafolder ::
endif
if (load_data)
adh5_load_reduced_detector(fileID, detectorpath, reduction_func, reduction_param, progress=progress)
endif
HDF5CloseFile fileID
else
AFileName = ""
endif
#else
Abort "HDF5 XOP not loaded."
#endif
if (timerRefNum >= 0)
adh5_perf_secs = stopMSTimer(timerRefNum) / 1e6
endif
setdatafolder saveDF
return AFileName
end
/// load a single image from a HDF5 file created by the Area Detector software.
///
/// the data wave is loaded into the current data folder.
/// attributes are loaded into the attr subfolder. existing waves in attr are deleted.
///
/// @warning EXPERIMENTAL
/// this function uses the root:pearl_area:preview data folder. existing data there may be deleted!
///
/// @param ANickName destination wave name. the wave is created in the current data folder.
/// @param APathName igor symbolic path name. can be empty if the path is specified in FileName or a dialog box should be displayed
/// @param AFileName if empty a dialog box shows up
/// @param load_data 1 (default): load data; 0: do not load data
/// @param load_attr 1 (default): load attributes; 0: do not load attributes
/// note: for correct scaling of the image, the attributes need to be loaded
function /s adh5_load_preview(ANickName, APathName, AFileName, [load_data, load_attr])
string ANickName
string APathName
string AFileName
variable load_data
variable load_attr
if (ParamIsDefault(load_data))
load_data = 1
endif
if (ParamIsDefault(load_attr))
load_attr = 1
endif
dfref saveDF = GetDataFolderDFR()
setdatafolder root:
newdatafolder /o/s pearl_area
newdatafolder /o/s preview
// open file
variable fileID
string instrumentpath = "/entry/instrument/"
string detectorpath = instrumentpath + "detector/"
string attributespath = instrumentpath + "NDAttributes/"
string datasetname
string datawavename
// performance monitoring
variable timerRefNum
variable /g adh5_perf_secs
timerRefNum = startMSTimer
// avoid compilation error if HDF5 XOP has not been loaded
#if Exists("HDF5OpenFile")
HDF5OpenFile /P=$APathName/R/Z fileID as AFileName
if (v_flag == 0)
AFileName = s_path + s_filename
// detector data
datasetname = detectorpath + "data"
STRUCT HDF5DataInfo di // Defined in HDF5 Browser.ipf.
InitHDF5DataInfo(di)
variable err = HDF5DatasetInfo(fileID, datasetname, 0, di)
if (err != 0)
print "error accessing detector/data"
return ""
endif
if (di.ndims < 2)
print "error: rank of dataset < 2"
return ""
endif
variable dim2start = 0, dim2count = 1, dim3start = 0, dim3count = 1
if (di.ndims >= 3)
dim2start = floor(di.dims[di.ndims - 3] / 2)
dim2count = 1
endif
if (di.ndims >= 4)
dim3start = floor(di.dims[di.ndims - 4] / 2)
dim3count = 1
endif
if (load_data)
adh5_load_detector_image(fileID, detectorpath, dim2start, dim2count, dim3start, dim3count)
wave /z data
string destpath = GetDataFolder(1, saveDF) + ANickName
if (waveexists(data))
duplicate /o data, $destpath
wave /z data = $destpath
endif
endif
if (load_attr)
setdatafolder saveDF
newdatafolder /o/s attr
killwaves /a/z
adh5_loadattr_all(fileID, attributespath)
setdatafolder ::
if (waveexists(data))
adh5_scale(data)
endif
endif
HDF5CloseFile fileID
else
print "error opening file " + AFileName
AFileName = ""
endif
#else
Abort "HDF5 XOP not loaded."
#endif
if (timerRefNum >= 0)
adh5_perf_secs = stopMSTimer(timerRefNum) / 1e6
endif
setdatafolder saveDF
return AFileName
end
/// load descriptive info from a HDF5 file created by the Area Detector software.
///
/// the information returned is the array size and active scans
///
/// @attention EXPERIMENTAL
/// this function should be merged with adh5_load_preview
///
/// @param APathName igor symbolic path name. can be empty if the path is specified in FileName or a dialog box should be displayed
/// @param AFileName if empty a dialog box shows up
///
function /s adh5_load_info(APathName, AFileName)
string APathName
string AFileName
dfref saveDF = GetDataFolderDFR()
// open file
variable fileID
string instrumentpath = "/entry/instrument/"
string detectorpath = instrumentpath + "detector/"
string attributespath = instrumentpath + "NDAttributes/"
string datasetname
string datawavename
string s_info = ""
string s
variable idim
// avoid compilation error if HDF5 XOP has not been loaded
#if Exists("HDF5OpenFile")
HDF5OpenFile /P=$APathName/R/Z fileID as AFileName
if (v_flag == 0)
AFileName = s_path + s_filename
// detector data
datasetname = detectorpath + "data"
STRUCT HDF5DataInfo di // Defined in HDF5 Browser.ipf.
InitHDF5DataInfo(di)
variable err = HDF5DatasetInfo(fileID, datasetname, 0, di)
if (err != 0)
print "error accessing detector/data"
return ""
endif
for (idim = 0; idim < di.ndims; idim += 1)
sprintf s, "dim %u: %u points", idim, di.dims[idim]
if (strlen(s_info) > 0)
s_info = s_info + "\r" + s
else
s_info = s
endif
endfor
dfref df = NewFreeDataFolder()
setdatafolder df
adh5_loadattr_all(fileID, attributespath)
for (idim = 1; idim < 5; idim += 1)
sprintf s, "Scan%uActive", idim
wave /z w = $s
if (waveexists(w) && (numpnts(w) > 0) && (w[0] > 0))
sprintf s, "Scan%uPositioner1", idim
wave /t wt = $s
sprintf s, "scan %u: %s", idim, wt[0]
if (strlen(s_info) > 0)
s_info = s_info + "\r" + s
else
s_info = s
endif
endif
endfor
HDF5CloseFile fileID
else
print "error opening file " + AFileName
AFileName = ""
endif
#else
Abort "HDF5 XOP not loaded."
#endif
setdatafolder saveDF
return s_info
end
/// load the detector dataset from the open HDF5 file.
///
/// the function loads the whole dataset at once
/// and redimensions it so that the image dimensions are X and Y
///
/// @param fileID ID of open HDF5 file from HDF5OpenFile
/// @param detectorpath path to detector group in the HDF5 file
///
function adh5_load_detector(fileID, detectorpath)
variable fileID
string detectorpath
// avoid compilation error if HDF5 XOP has not been loaded
#if Exists("HDF5LoadData")
string datasetname
string datawavename
// detector data
datasetname = detectorpath + "data"
STRUCT HDF5DataInfo di // Defined in HDF5 Browser.ipf.
InitHDF5DataInfo(di)
variable err = HDF5DatasetInfo(fileID, datasetname, 0, di)
if (err != 0)
print "error accessing detector/data"
return -1
endif
if (di.ndims < 2)
print "error: rank of dataset < 2"
return -2
endif
HDF5LoadData /O /Q /Z fileID, datasetname
wave data
#else
Abort "HDF5 XOP not loaded."
#endif
end
/// redimension a multi-dimensional area detector array loaded from HDF5.
///
/// so that the image dimensions are X and Y
/// singleton dimensions are removed (most common cases only)
///
/// in the redimensioned array, the original dimension type is noted in the dimension label:
/// AD_Dim0 = first image dimension
/// AD_Dim1 = second image dimension
/// AD_DimN = frame sequence
/// AD_DimX = extra dimension X
/// AD_DimY = extra dimension Y (cannot be loaded in Igor)
///
/// @param data area detector data loaded from HDF5 to be redimensioned
///
function adh5_redim(data)
wave data
duplicate /free data, tempdata
variable nd = wavedims(tempdata)
variable nx = dimsize(tempdata, nd - 1)
variable ny = dimsize(tempdata, nd - 2)
variable nz = dimsize(tempdata, nd - 3)
variable nt = dimsize(tempdata, nd - 4)
switch (nd)
case 2:
if (nx <= 1)
redimension /n=(ny) data
setdimlabel 0, -1, AD_Dim1, data
data = tempdata[p][0]
elseif (ny <= 1)
redimension /n=(nx) data
setdimlabel 0, -1, AD_Dim0, data
data = tempdata[0][p]
else
redimension /n=(nx,ny) data
setdimlabel 0, -1, AD_Dim0, data
setdimlabel 1, -1, AD_Dim1, data
data = tempdata[q][p]
endif
break
case 3:
if (nx <= 1)
redimension /n=(ny,nz) data
setdimlabel 0, -1, AD_Dim1, data
setdimlabel 1, -1, AD_DimN, data
multithread data = tempdata[q][p][0]
elseif (ny <= 1)
redimension /n=(nx,nz) data
setdimlabel 0, -1, AD_Dim0, data
setdimlabel 1, -1, AD_DimN, data
multithread data = tempdata[q][0][p]
elseif (nz <= 1)
redimension /n=(nx,ny) data
setdimlabel 0, -1, AD_Dim0, data
setdimlabel 1, -1, AD_Dim1, data
multithread data = tempdata[0][q][p]
else
redimension /n=(nx,ny,nz) data
setdimlabel 0, -1, AD_Dim0, data
setdimlabel 1, -1, AD_Dim1, data
setdimlabel 2, -1, AD_DimN, data
multithread data = tempdata[r][q][p]
endif
break
case 4:
if (nz <= 1)
// singleton "frame number" dimension
redimension /n=(nx,ny,nt) data
setdimlabel 0, -1, AD_Dim0, data
setdimlabel 1, -1, AD_Dim1, data
setdimlabel 2, -1, AD_DimX, data
multithread data = tempdata[r][0][q][p]
else
redimension /n=(nx,ny,nz,nt) data
setdimlabel 0, -1, AD_Dim0, data
setdimlabel 1, -1, AD_Dim1, data
setdimlabel 2, -1, AD_DimN, data
setdimlabel 3, -1, AD_DimX, data
multithread data = tempdata[s][r][q][p]
endif
break
endswitch
end
/// find the attributes data folder of an area detector dataset.
///
/// since version 1.04 attributes should be stored in a subfolder named attr.
/// earlier versions had the attributes in the same data folder as the actual dataset.
///
/// @param data wave containing the main dataset.
///
/// @return data folder reference of the attributes folder.
/// the reference may be invalid (and default to root) if the folder cannot be found,
/// cf. built-in DataFolderRefStatus function.
static function /DF GetAttrDataFolderDFR(data)
wave data
dfref dataDF = GetWavesDataFolderDFR(data)
dfref attrDF = dataDF:attr
if (DataFolderRefStatus(attrDF) == 0)
attrDF = dataDF
endif
return attrDF
end
/// set the dimension scales of an area detector dataset.
///
/// the intrinsic dimensions 0 and 1 are scaled according to the data source
/// (currently supported: Prosilica cameras, Scienta electron analyser).
/// the extra dimensions are scaled according to the scan.
/// the latter requires that the positioner names and position values are available.
///
function adh5_scale(data,[source])
wave data
string source
dfref saveDF = GetDataFolderDFR()
dfref dataDF = GetWavesDataFolderDFR(data)
dfref attrDF = GetAttrDataFolderDFR(data)
if (ParamIsDefault(source))
// is the source a Scienta analyser?
wave /SDFR=attrDF /Z AcquisitionMode
wave /SDFR=attrDF /T /Z Manufacturer
source = "unknown"
if (waveexists(Manufacturer) && (numpnts(Manufacturer) >= 1))
strswitch(Manufacturer[0])
case "VG Scienta":
source = "scienta"
break
case "Prosilica":
source = "prosilica"
break
endswitch
elseif (waveexists(AcquisitionMode) && (numpnts(AcquisitionMode) >= 1))
if (stringmatch(note(AcquisitionMode), "*SCIENTA*"))
source = "scienta"
endif
endif
endif
strswitch(source)
case "prosilica":
// pixel scale - nothing to do
break
case "scienta":
adh5_scale_scienta(data)
break
endswitch
setdatafolder saveDF
end
/// load the detector dataset from the open HDF5 file.
///
/// the function loads the dataset image by image using the hyperslab option.
/// this function gives the same result as adh5_load_detector.
/// it is about 5% slower, and it depends on HDF5 Browser code.
/// but it does not choke on large datasets (as long as the final wave fits into memory).
///
/// @param fileID ID of open HDF5 file from HDF5OpenFile.
/// @param detectorpath path to detector group in the HDF5 file.
/// @param progress 1 (default): show progress window; 0: do not show progress window.
///
/// @return 0 if successful, non-zero if an error occurred.
///
function adh5_load_detector_slabs(fileID, detectorpath, [progress])
variable fileID
string detectorpath
variable progress
if (ParamIsDefault(progress))
progress = 1
endif
variable result = 0
// avoid compilation error if HDF5 XOP has not been loaded
#if Exists("HDF5LoadData")
string datasetname
string datawavename
// detector data
datasetname = detectorpath + "data"
STRUCT HDF5DataInfo di // Defined in HDF5 Browser.ipf.
InitHDF5DataInfo(di)
variable err = HDF5DatasetInfo(fileID, datasetname, 0, di)
if (err != 0)
print "error accessing detector/data"
return -1
endif
if (di.ndims < 2)
print "error: rank of dataset < 2"
return -2
endif
// nx and nz are the image dimensions
variable idx, idy, idz, idt, izt
idx = di.ndims - 1
idy = di.ndims - 2
idz = -1
idt = -1
variable nx, ny, nz, nt, nzt
nx = di.dims[idx]
ny = di.dims[idy]
nz = 1
nt = 1
make /n=(nx,ny,nz,nt) /o data
string dim_labels = "AD_Dim0;AD_Dim1;AD_DimN;AD_DimX;AD_DimY"
string dim_label
dim_label = StringFromList(0, dim_labels, ";")
setdimlabel 0, -1, $dim_label, data
dim_labels = RemoveFromList(dim_label, dim_labels, ";")
dim_label = StringFromList(0, dim_labels, ";")
setdimlabel 1, -1, $dim_label, data
dim_labels = RemoveFromList(dim_label, dim_labels, ";")
// find additional dimensions, ignore singletons
variable id
for (id = idy - 1; (id >= 0) && (nz == 1); id -= 1)
if (di.dims[id] > 1)
idz = id
nz = di.dims[id]
dim_label = StringFromList(0, dim_labels, ";")
setdimlabel 2, -1, $dim_label, data
endif
dim_labels = RemoveListItem(0, dim_labels, ";")
endfor
for (id = idz - 1; (id >= 0) && (nt == 1); id -= 1)
if (di.dims[id] > 1)
idt = id
nt = di.dims[id]
dim_label = StringFromList(0, dim_labels, ";")
setdimlabel 3, -1, $dim_label, data
endif
dim_labels = RemoveListItem(0, dim_labels, ";")
endfor
redimension /n=(nx,ny,nz,nt) data
// default values if dimensions are not present in dataset
if (idz < 0)
idz = idx + 1
idt = idz + 1
elseif (idt < 0)
idt = idx + 1
endif
nzt = nz * nt
izt = 0
if (progress)
display_progress_panel("HDF5 Import", "Loading data...", nzt)
endif
// load data image by image
HDF5MakeHyperslabWave(GetDataFolder(1) + "slab", max(di.ndims, 4))
wave slab
slab[][%Start] = 0
slab[][%Stride] = 1
slab[][%Count] = 1
slab[][%Block] = 1
slab[idx][%Block] = nx
slab[idy][%Block] = ny
variable iz, it
for (iz = 0; iz < nz; iz += 1)
for (it = 0; it < nt; it += 1)
slab[idz][%Start] = iz
slab[idt][%Start] = it
HDF5LoadData /O /Q /Z /SLAB=slab /N=slabdata fileID, datasetname
wave slabdata // 2D, 3D, or 4D with singletons
switch (WaveDims(slabdata))
case 2:
data[][][iz][it] = slabdata[q][p]
break
case 3:
data[][][iz][it] = slabdata[0][q][p]
break
case 4:
data[][][iz][it] = slabdata[0][0][q][p]
break
endswitch
// progress window
izt += 1
if (progress)
if (update_progress_panel(izt))
result = -4 // user abort
break
endif
endif
endfor
if (result < 0)
break
endif
endfor
if (nz == 1)
redimension /n=(nx,ny) data
elseif (nt == 1)
redimension /n=(nx,ny,nz) data
endif
if (progress)
kill_progress_panel()
endif
#else
Abort "HDF5 XOP not loaded."
#endif
return result
end
/// load a single image from the detector dataset of the open HDF5 file
///
/// the function can average over a region in the extra dimensions.
///
/// @param fileID ID of open HDF5 file from HDF5OpenFile
/// @param detectorpath path to detector group in the HDF5 file
/// @param dim2start 2nd dimension coordinate of the first image
/// note that the order of dimensions is reversed in the file
/// 2nd dimension = N dimension in area detector = dimension 0 of the three-dimensional HDF dataset
/// set to 0 if dimension may not be present
/// @param dim2count number of subsequent images to average
/// set to 1 if dimension may not be present
/// @param dim3start 3rd dimension coordinate of the first image
/// note that the order of dimensions is reversed in the file
/// 3rd dimension = extra X dimension in area detector = dimension 0 of the four-dimensional HDF dataset
/// set to 0 if dimension may not be present
/// @param dim3count number of subsequent images to average
/// set to 1 if dimension may not be present
///
function adh5_load_detector_image(fileID, detectorpath, dim2start, dim2count, dim3start, dim3count)
variable fileID
string detectorpath
variable dim2start
variable dim2count
variable dim3start
variable dim3count
// avoid compilation error if HDF5 XOP has not been loaded
#if Exists("HDF5LoadData")
string datasetname
string datawavename
// detector data
datasetname = detectorpath + "data"
STRUCT HDF5DataInfo di // Defined in HDF5 Browser.ipf.
InitHDF5DataInfo(di)
variable err = HDF5DatasetInfo(fileID, datasetname, 0, di)
if (err != 0)
print "error accessing detector/data"
return -1
endif
if (di.ndims < 1)
print "error: rank of dataset < 1"
return -2
endif
// nx and nz are the image dimensions
variable idx, idy, idz, idt
idx = di.ndims - 1
idy = di.ndims >= 2 ? di.ndims - 2 : 1
idz = di.ndims >= 3 ? di.ndims - 3 : 2
idt = di.ndims >= 4 ? di.ndims - 4 : 3
variable nx, ny
nx = di.dims[idx]
ny = di.ndims >= 2 ? di.dims[idy] : 1
variable dim2end = dim2start + dim2count - 1
variable dim3end = dim3start + dim3count - 1
// the slab wave is at least 4-dimensional
// it will also load lower-dimensional datasets
HDF5MakeHyperslabWave(GetDataFolder(1) + "slab", max(di.ndims, 4))
wave slab
slab[][%Start] = 0
slab[][%Stride] = 1
slab[][%Count] = 1
slab[][%Block] = 1
slab[idx][%Block] = nx
slab[idy][%Block] = ny
make /n=(nx,ny)/o/d data
data = 0
variable iz, it
variable navg = 0
for (iz = dim2start; iz <= dim2end; iz += 1)
for (it = dim3start; it <= dim3end; it += 1)
slab[idz][%Start] = iz
slab[idt][%Start] = it
HDF5LoadData /O /Q /Z /SLAB=slab /N=slabdata fileID, datasetname
wave slabdata // 2D, 3D, or 4D with singletons
switch (WaveDims(slabdata))
case 1:
data += slabdata[p]
navg += 1
break
case 2:
data += slabdata[q][p]
navg += 1
break
case 3:
data += slabdata[0][q][p]
navg += 1
break
case 4:
data += slabdata[0][0][q][p]
navg += 1
break
endswitch
endfor
endfor
data /= navg
setdimlabel 0, -1, AD_Dim0, data
setdimlabel 1, -1, AD_Dim1, data
#else
Abort "HDF5 XOP not loaded."
#endif
end
/// get a list of functions which can be used as reduction functions.
///
/// the function evaluates only the function arguments,
/// it may thus include functions which are not suitable as reduction functions.
///
function /s adh5_list_reduction_funcs()
string all_funcs = FunctionList("*", ";", "KIND:6,NPARAMS:4,VALTYPE:1")
string result = ""
variable ii
variable nn = ItemsInList(all_funcs, ";")
string funcname
string info
variable nparams
variable accept
for (ii = 0; ii < nn; ii += 1)
funcname = StringFromList(ii, all_funcs, ";")
info = FunctionInfo(funcname)
accept = (NumberByKey("RETURNTYPE", info, ":", ";") == 0x0004)
accept = accept && (cmpstr(StringByKey("THREADSAFE", info, ":", ";"), "yes") == 0)
accept = accept && (NumberByKey("N_PARAMS", info, ":", ";") == 4)
accept = accept && (NumberByKey("N_OPT_PARAMS", info, ":", ";") == 0)
if (accept)
// 3 numeric waves and one pass-by-reference string
accept = accept && (NumberByKey("PARAM_0_TYPE", info, ":", ";") == 0x4002)
accept = accept && (NumberByKey("PARAM_1_TYPE", info, ":", ";") == 0x4002)
accept = accept && (NumberByKey("PARAM_2_TYPE", info, ":", ";") == 0x4002)
accept = accept && (NumberByKey("PARAM_3_TYPE", info, ":", ";") == 0x3000)
endif
if (accept)
result = AddListItem(funcname, result, ";")
endif
endfor
result = SortList(result, ";", 4)
return result
end
/// function prototype for adh5_load_reduced_detector
///
/// derived functions reduce a two-dimensional dataset to a one-dimensional dataset,
/// e.g. by ROI-integration, curve fitting, etc.
// the resulting wave must have the same size as either dimension of the source image.
///
/// each destination wave is a one-dimensional intensity distribution.
/// the function must redimension each of these waves to one of the image dimensions
/// by calling the adh5_setup_profile() function.
/// this function will also copy the scale information and dimension labels,
/// which is important for the proper scaling of the result.
///
/// the meaning of the data in dest1 and dest2 is up to the particular function,
/// e.g. dest1 could hold the mean value and dest2 the one-sigma error,
/// or dest1 could hold the X-profile, and dest2 the Y-profile.
///
/// @param source source wave
/// two-dimensional intensity distribution (image)
/// @param dest1, dest2 destination waves
/// @param param string with optional parameters, shared between calls.
/// this is a pass-by-reference argument,
/// the function may modify the string
///
/// @return zero if successful, non-zero if an error occurs.
///
threadsafe function adh5_default_reduction(source, dest1, dest2, param)
wave source
wave dest1, dest2
string &param
// demo code
// integrate along the dimensions
adh5_setup_profile(source, dest1, 0)
ad_profile_x_w(source, 0, -1, dest1)
adh5_setup_profile(source, dest2, 1)
ad_profile_y_w(source, 0, -1, dest2)
return 0
end
/// set up a one-dimensional wave for a line profile based on a 2D original wave.
///
/// redimensions the profile wave to the given dimension.
/// copies the scale and dimension label of the given dimension.
///
threadsafe function adh5_setup_profile(image, profile, dim)
wave image // prototype
wave profile // destination wave
variable dim // which dimension to keep: 0 = X, 1 = Y
redimension /n=(dimsize(image, dim)) profile
setscale /p x dimoffset(image, dim), dimdelta(image, dim), waveunits(image, dim), profile
setscale d 0, 0, waveunits(image, -1), profile
setdimlabel 0, -1, $getdimlabel(image, dim, -1), profile
end
/// wrapper function for testing reduction functions from the command line.
///
/// Igor does not allow global variables as pass-by-reference parameter for reduction_param.
///
function /s adh5_test_reduction_func(source, dest1, dest2, reduction_func, reduction_param)
wave source
wave dest1
wave dest2
funcref adh5_default_reduction reduction_func
string reduction_param
reduction_func(source, dest1, dest2, reduction_param)
return reduction_param
end
/// load a reduced detector dataset from the open HDF5 file.
///
/// the function loads the dataset image by image using the hyperslab option
/// and applies a custom reduction function to each image.
/// the results from the reduction function are composed into one result wave.
/// the raw data are discarded.
///
/// by default, the reduction function is called in separate threads to reduce the total loading time.
/// (see the global variable adh5_perf_secs which reports the total run time of the function.)
/// the effect varies depending on the balance between file loading (image size)
/// and data processing (complexity of the reduction function).
/// for debugging the reduction function, multi-threading can be disabled.
///
/// @param fileID ID of open HDF5 file from HDF5OpenFile
/// @param detectorpath path to detector group in the HDF5 file
/// @param reduction_func custom reduction function
/// (any user-defined function which has the same parameters as adh5_default_reduction())
/// @param reduction_param parameter string for the reduction function
/// @param progress 1 (default): show progress window; 0: do not show progress window
/// @param nthreads -1 (default): use as many threads as there are processor cores (in addition to main thread)
/// 0: use main thread only (e.g. for debugging the reduction function)
/// >= 1: use a fixed number of (additional) threads
function adh5_load_reduced_detector(fileID, detectorpath, reduction_func, reduction_param, [progress, nthreads])
variable fileID
string detectorpath
funcref adh5_default_reduction reduction_func
string reduction_param
variable progress
variable nthreads
if (ParamIsDefault(progress))
progress = 1
endif
if (ParamIsDefault(nthreads))
nthreads = -1
endif
variable result = 0
// avoid compilation error if HDF5 XOP has not been loaded
#if Exists("HDF5LoadData")
string datasetname
string datawavename
// detector data
datasetname = detectorpath + "data"
STRUCT HDF5DataInfo di // Defined in HDF5 Browser.ipf.
InitHDF5DataInfo(di)
variable err = HDF5DatasetInfo(fileID, datasetname, 0, di)
if (err != 0)
print "error accessing detector/data"
return -1
endif
if (di.ndims < 2)
print "error: rank of dataset < 2"
return -2
endif
// nx and nz are the image dimensions
variable idx, idy, idz, idt
idx = di.ndims - 1
idy = di.ndims - 2
idz = -1
idt = -1
variable nx, ny, nz, nt
nx = di.dims[idx]
ny = di.dims[idy]
nz = 1
nt = 1
// find additional dimensions, ignore singletons
variable id
for (id = idy - 1; (id >= 0) && (nz == 1); id -= 1)
if (di.dims[id] > 1)
idz = id
nz = di.dims[id]
endif
endfor
for (id = idz - 1; (id >= 0) && (nt == 1); id -= 1)
if (di.dims[id] > 1)
idt = id
nt = di.dims[id]
endif
endfor
// default values if dimensions are not present in dataset
if (idz < 0)
idz = idx + 1
idt = idz + 1
elseif (idt < 0)
idt = idx + 1
endif
variable nzt = nz * nt
variable izt
// load data image by image
HDF5MakeHyperslabWave(GetDataFolder(1) + "slab", max(di.ndims, 4))
wave slab
slab[][%Start] = 0
slab[][%Stride] = 1
slab[][%Count] = 1
slab[][%Block] = 1
slab[idx][%Block] = nx
slab[idy][%Block] = ny
// set up multi threading
if (nthreads < 0)
nthreads = ThreadProcessorCount
endif
if (nthreads > 0)
variable threadGroupID = ThreadGroupCreate(nthreads)
variable ithread
for (ithread = 0; ithread < nthreads; ithread += 1)
ThreadStart threadGroupID, ithread, reduce_slab_worker(reduction_func)
endfor
else
make /n=(nzt) /df /free processing_folders
endif
if (progress)
display_progress_panel("HDF5 Import", "Loading data (step 1 of 2)...", nzt)
endif
make /n=(nx,ny)/d image_template
setdimlabel 0, -1, AD_Dim0, image_template
setdimlabel 1, -1, AD_Dim1, image_template
adh5_scale(image_template)
variable iz, it
string dfname
izt = 0
for (iz = 0; iz < nz; iz += 1)
for (it = 0; it < nt; it += 1)
// load hyperslab
slab[idz][%Start] = iz
slab[idt][%Start] = it
dfname = "processing_" + num2str(izt)
newdatafolder /s $dfname
HDF5LoadData /O /Q /Z /SLAB=slab /N=slabdata fileID, datasetname
// send to processing queue
duplicate image_template, image
variable /g r_index = iz
variable /g s_index = it
string /g func_param = reduction_param
if (nthreads > 0)
WaveClear image
ThreadGroupPutDF threadGroupID, :
else
processing_folders[izt] = GetDataFolderDFR()
make /n=1/d profile1, profile2
wave slabdata
variable /g func_result
func_result = reduce_slab_image(slabdata, image, profile1, profile2, reduction_func, func_param)
WaveClear slabdata, image, profile1, profile2
setdatafolder ::
endif
izt += 1
// progress window
if (progress)
if (update_progress_panel(izt))
result = -4 // user abort
break
endif
endif
endfor
endfor
killwaves /z slab, image_template
if (progress)
update_progress_panel(0, message="Processing data (step 2 of 2)...")
endif
dfref dfr
for (izt = 0; (izt < nzt) && (result == 0); izt += 1)
if (nthreads > 0)
do
dfr = ThreadGroupGetDFR(threadGroupID, 1000)
if (DatafolderRefStatus(dfr) != 0)
break
endif
if (progress)
if (update_progress_panel(izt))
result = -4 // user abort
break
endif
endif
while (1)
else
dfr = processing_folders[izt]
if (progress)
if (update_progress_panel(izt))
result = -4 // user abort
break
endif
endif
endif
if (result != 0)
break
endif
nvar rr = dfr:r_index
nvar ss = dfr:s_index
nvar func_result = dfr:func_result
wave profile1 = dfr:profile1
wave profile2 = dfr:profile2
if (func_result == 0)
if (izt == 0)
make /n=(dimsize(profile1, 0), nz, nt)/d/o data1
make /n=(dimsize(profile2, 0), nz, nt)/d/o data2
setdimlabel 0, -1, $getdimlabel(profile1, 0, -1), data1
setdimlabel 0, -1, $getdimlabel(profile2, 0, -1), data2
setscale /p x dimoffset(profile1, 0), dimdelta(profile1, 0), waveunits(profile1, 0), data1
setscale /p x dimoffset(profile2, 0), dimdelta(profile2, 0), waveunits(profile2, 0), data2
setscale d 0, 0, waveunits(profile1, -1), data1
setscale d 0, 0, waveunits(profile2, -1), data2
endif
data1[][rr][ss] = profile1[p]
data2[][rr][ss] = profile2[p]
else
result = -3 // dimension reduction error
break
endif
endfor
if (nthreads > 0)
variable tstatus = ThreadGroupRelease(threadGroupID)
if (tstatus == -2)
result = -5 // thread did not terminate properly
endif
else
for (izt = 0; izt < nzt; izt += 1)
KillDataFolder /Z processing_folders[izt]
endfor
endif
if (result == 0)
if (nz == 1)
redimension /n=(dimsize(data1,0)) data1
redimension /n=(dimsize(data2,0)) data2
elseif (nt == 1)
redimension /n=(dimsize(data1,0),nz) data1
redimension /n=(dimsize(data2,0),nz) data2
setdimlabel 1, -1, AD_DimN, data1
setdimlabel 1, -1, AD_DimN, data2
else
setdimlabel 1, -1, AD_DimN, data1
setdimlabel 1, -1, AD_DimN, data2
setdimlabel 2, -1, AD_DimX, data1
setdimlabel 2, -1, AD_DimX, data2
endif
endif
if (progress)
kill_progress_panel()
endif
#else
Abort "HDF5 XOP not loaded."
#endif
return result
end
threadsafe static function reduce_slab_worker(reduction_func)
funcref adh5_default_reduction reduction_func
do
// wait for job from main thread
do
dfref dfr = ThreadGroupGetDFR(0, 1000)
if (DataFolderRefStatus(dfr) == 0)
if (GetRTError(2))
return 0 // no more jobs
endif
else
break
endif
while (1)
// get input data
wave slabdata = dfr:slabdata
wave image = dfr:image
svar func_param = dfr:func_param
nvar rr = dfr:r_index
nvar ss = dfr:s_index
// do the work
newdatafolder /s outDF
make /n=1/d profile1, profile2
variable /g r_index = rr
variable /g s_index = ss
variable /g func_result
func_result = reduce_slab_image(slabdata, image, profile1, profile2, reduction_func, func_param)
// send output to queue and clean up
WaveClear slabdata, image, profile1, profile2
ThreadGroupPutDF 0, :
KillDataFolder dfr
while (1)
return 0
end
threadsafe static function reduce_slab_image(slabdata, image, profile1, profile2, reduction_func, reduction_param)
wave slabdata
wave image
wave profile1
wave profile2
funcref adh5_default_reduction reduction_func
string reduction_param
switch (WaveDims(slabdata))
case 2:
image = slabdata[q][p]
break
case 3:
image = slabdata[0][q][p]
break
case 4:
image = slabdata[0][0][q][p]
break
endswitch
return reduction_func(image, profile1, profile2, reduction_param)
end
/// load an NDAttributes group from an open HDF5 file into the current data folder.
///
/// datasets contained in the group are loaded as waves.
/// if a dataset contains only one data point, it is added to the IN, ID, IV, IU waves,
/// where IN = EPICS channel name, ID = attribute name, IV = value, IU = unit
/// (units are left empty as they are not saved in HDF5).
/// attributes of the NDAttributes group are added to the IN, ID, IV, IU waves,
/// however, IN and IU are left empty as this information is not saved in the HDF5 file.
///
/// @param fileID ID of open HDF5 file from HDF5OpenFile
/// @param attributespath path to NDAttributes group in the HDF5 file
///
function adh5_loadattr_all(fileID, attributespath)
variable fileID
string attributespath
string datasetname
string datawavename
// avoid compilation error if HDF5 XOP has not been loaded
#if Exists("HDF5LoadData")
// datasets in NDAttributes group
HDF5ListGroup /F /TYPE=2 fileID, attributespath
string h5datasets = S_HDF5ListGroup
HDF5ListAttributes /TYPE=1 /Z fileID, attributespath
string h5attributes = S_HDF5ListAttributes
variable nds = ItemsInList(h5datasets, ";")
variable na = ItemsInList(h5attributes, ";")
variable ids
variable idest = 0
variable n_attr
string s_attr
string s_source
make /n=(nds+na) /t /o IN, ID, IV, IU
for (ids = 0; ids < nds; ids += 1)
datasetname = StringFromList(ids, h5datasets, ";")
HDF5LoadData /O/Q fileID, datasetname
if (v_flag == 0)
datawavename = StringFromList(0, s_wavenames)
else
datawavename = ""
endif
HDF5LoadData /A="source"/O/Q/TYPE=2 fileID, datasetname
if (v_flag == 0)
wave /t source
s_source = source[0]
else
s_source = ""
endif
read_attribute_info(datawavename, s_source, idest)
endfor
// attributes of NDAttributes group
if (v_flag == 0)
nds = ItemsInList(h5attributes, ";")
else
nds = 0
endif
for (ids = 0; ids < nds; ids += 1)
datasetname = StringFromList(ids, h5attributes, ";")
HDF5LoadData /A=datasetname/O/Q/TYPE=1 fileID, attributespath
if (v_flag == 0)
datawavename = StringFromList(0, s_wavenames)
read_attribute_info(datawavename, "", idest) // we don't get the source of these attributes
endif
endfor
redimension /n=(idest) IN, ID, IV, IU
sort {IN, ID}, IN, ID, IV, IU
killwaves /z source
#else
Abort "HDF5 XOP not loaded."
#endif
end
/// sub-function of adh5_loadattr_all.
///
/// reads one attribute from a wave which was loaded from an HDF5 file into the info waves IN, ID, IV, IU.
/// the attribute is read only if the input wave contains exactly one item,
/// i.e. either the measurement is a single image, or the attribute has string type.
///
/// @param datawavename name of the attribute wave in the current folder.
/// can be text or numeric.
/// @param source source identifier (EPICS name) of the attribute.
/// @param idest destination index in IN, ID, IV, IU where the results are written.
/// the variable is incremented if data was written, otherwise it is left unchanged.
/// make sure IN, ID, IV, IU have at least idest + 1 elements.
///
static function read_attribute_info(datawavename, source, idest)
string datawavename // name of the attribute wave in the current folder.
// can be text or numeric.
string source
// source identifier (EPICS name) of the attribute.
variable &idest
// destination index in IN, ID, IV, IU where the results are written.
// the variable is incremented if data was written, otherwise it is left unchanged.
// make sure IN, ID, IV, IU have at least idest + 1 elements.
wave /t IN
wave /t ID
wave /t IV
wave /t IU
variable n_attr
string s_attr
if (exists(datawavename) == 1)
if (strlen(source) > 0)
Note $datawavename, "PV=" + source
endif
switch(WaveType($datawavename, 1))
case 1: // numeric
wave w_attr = $datawavename
n_attr = numpnts(w_attr)
sprintf s_attr, "%.12g", w_attr[0]
break
case 2: // text
wave /t wt_attr = $datawavename
n_attr = numpnts(wt_attr)
s_attr = wt_attr[0]
break
default: // unknown
n_attr = 0
endswitch
if (n_attr == 1)
IN[idest] = source
ID[idest] = datawavename
IV[idest] = s_attr
IU[idest] = "" // we don't get the units
idest += 1
endif
endif
end
/// set the energy and angle scales of an area detector dataset from the Scienta analyser.
///
/// the dimension labels of the energy and angle scales must be set correctly:
/// AD_Dim0 = energy dimension; AD_Dim1 = angle dimension.
/// these dimensions must be the first two dimensions of a multi-dimensional dataset.
/// normally, AD_Dim0 is the X dimension, and AD_Dim1 the Y dimension.
///
function adh5_scale_scienta(data)
wave data
dfref saveDF = GetDataFolderDFR()
dfref dataDF = GetWavesDataFolderDFR(data)
dfref attrDF = GetAttrDataFolderDFR(data)
wave /SDFR=attrDF LensMode
wave /SDFR=attrDF /Z ChannelBegin, ChannelEnd
wave /SDFR=attrDF /Z SliceBegin, SliceEnd
variable EDim, ADim
variable ELow, EHigh, ALow, AHigh
string EUnit, AUnit
// which dimension is angle and which one is energy?
strswitch(GetDimLabel(data, 0, -1))
case "AD_Dim0":
EDim = 0
break
case "AD_Dim1":
EDim = 1
break
default:
EDim = -1
endswitch
strswitch(GetDimLabel(data, 1, -1))
case "AD_Dim0":
ADim = 0
break
case "AD_Dim1":
ADim = 1
break
default:
ADim = -1
endswitch
// defaults (point scaling)
if (EDim >= 0)
ELow = dimoffset(data, EDim)
EHigh = dimoffset(data, EDim) + dimdelta(data, EDim) * (dimsize(data, EDim) - 1)
EUnit = "eV"
endif
if (ADim >= 0)
ALow = dimoffset(data, ADim)
AHigh = dimoffset(data, ADim) + dimdelta(data, ADim) * (dimsize(data, ADim) - 1)
AUnit = "arb."
endif
// lens mode can give more detail
if (waveexists(LensMode) && (numpnts(LensMode) >= 1))
switch(LensMode[0])
case 1: // Angular45
ALow = -45/2
AHigh = +45/2
AUnit = "°"
break
case 2: // Angular60
ALow = -60/2
AHigh = +60/2
AUnit = "°"
break
endswitch
endif
// best option if scales are explicit in separate waves
if (waveexists(ChannelBegin) && waveexists(ChannelEnd) && (numpnts(ChannelBegin) >= 1) && (numpnts(ChannelEnd) >= 1))
ELow = ChannelBegin[0]
EHigh = ChannelEnd[0]
endif
if (waveexists(SliceBegin) && waveexists(SliceEnd) && (numpnts(SliceBegin) >= 1) && (numpnts(SliceEnd) >= 1))
ALow = SliceBegin[0]
AHigh = SliceEnd[0]
endif
// apply new scales
switch(EDim)
case 0:
setscale /i x ELow, EHigh, EUnit, data
break
case 1:
setscale /i y ELow, EHigh, EUnit, data
break
endswitch
switch(ADim)
case 0:
setscale /i x ALow, AHigh, AUnit, data
break
case 1:
setscale /i y ALow, AHigh, AUnit, data
break
endswitch
setscale d 0, 0, "arb.", data
setdatafolder saveDF
end
/// scales the extra dimensions of an area detector dataset according to the EPICS scan
///
/// the scan positioner name and its values must be available
///
/// @todo incomplete
///
function adh5_scale_scan(data)
wave data
dfref saveDF = GetDataFolderDFR()
dfref dataDF = GetWavesDataFolderDFR(data)
wave /SDFR=dataDF AcquisitionMode, DetectorMode, EnergyMode
wave /SDFR=dataDF /z Scan1Active, Scan2Active
wave /SDFR=dataDF /t /z Scan1Positioner1, Scan1Readback1
wave /SDFR=dataDF /t /z Scan1Positioner2, Scan1Readback2
wave /SDFR=dataDF /t /z Scan2Positioner1, Scan2Readback1
wave /SDFR=dataDF /t /z Scan2Positioner2, Scan2Readback2
// TODO : search the data folder for positioner waves,
// i.e. waves with the PV name corresponding to Scan1Positioner1 in their wave note.
wave /z zscale
strswitch(GetDimLabel(data, 0, -1))
case "AD_DimN":
setscale /i x zscale[0], zscale[numpnts(zscale)-1], "", data
break
endswitch
strswitch(GetDimLabel(data, 1, -1))
case "AD_DimN":
setscale /i y zscale[0], zscale[numpnts(zscale)-1], "", data
break
endswitch
strswitch(GetDimLabel(data, 2, -1))
case "AD_DimN":
setscale /i z zscale[0], zscale[numpnts(zscale)-1], "", data
break
endswitch
setdatafolder saveDF
end