forked from AmbaPant/mantid
-
Notifications
You must be signed in to change notification settings - Fork 1
/
GenerateGroupingSNSInelastic.py
107 lines (83 loc) · 3.64 KB
/
GenerateGroupingSNSInelastic.py
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
# Mantid Repository : https://github.com/mantidproject/mantid
#
# Copyright © 2018 ISIS Rutherford Appleton Laboratory UKRI,
# NScD Oak Ridge National Laboratory, European Spallation Source,
# Institut Laue - Langevin & CSNS, Institute of High Energy Physics, CAS
# SPDX - License - Identifier: GPL - 3.0 +
#pylint: disable=no-init,invalid-name
import mantid
import mantid.api
import mantid.simpleapi
import mantid.kernel
import numpy
class GenerateGroupingSNSInelastic(mantid.api.PythonAlgorithm):
""" Class to generate grouping file
"""
def category(self):
""" Mantid required
"""
return "Inelastic\\Utility;Transforms\\Grouping"
def seeAlso(self):
return [ "GroupWorkspaces" ]
def name(self):
""" Mantid required
"""
return "GenerateGroupingSNSInelastic"
def summary(self):
""" Mantid required
"""
return "Generate grouping files for ARCS, CNCS, HYSPEC, and SEQUOIA."
def PyInit(self):
""" Python initialization: Define input parameters
"""
py = ["1", "2", "4","8","16","32","64","128"]
px = ["1", "2", "4","8"]
instrument = ["ARCS","CNCS","HYSPEC","SEQUOIA"]
self.declareProperty("AlongTubes", "1",mantid.kernel.StringListValidator(py), "Number of pixels across tubes to be grouped")
self.declareProperty("AcrossTubes", "1", mantid.kernel.StringListValidator(px), "Number of pixels across tubes to be grouped")
self.declareProperty("Instrument", instrument[0], mantid.kernel.StringListValidator(instrument),
"The instrument for wich to create grouping")
f=mantid.api.FileProperty("Filename","",mantid.api.FileAction.Save,".xml")
self.declareProperty(f,"Output filename.")
return
def PyExec(self):
""" Main execution body
"""
# 1. Get input
pixelsy = int(self.getProperty("AlongTubes").value)
pixelsx = int(self.getProperty("AcrossTubes").value)
instrument = self.getProperty("Instrument").value
filename = self.getProperty("Filename").value
IDF=mantid.api.ExperimentInfo.getInstrumentFilename(instrument)
__w = mantid.simpleapi.LoadEmptyInstrument(Filename=IDF)
i=0
spectrumInfo = __w.spectrumInfo()
while spectrumInfo.isMonitor(i):
i += 1
#i is the index of the first true detector
#now, crop the workspace of the monitors
__w = mantid.simpleapi.CropWorkspace(__w,StartWorkspaceIndex=i)
#get number of detectors (not including monitors)
y=__w.extractY()
numdet=(y[y==1]).size
spectra = numpy.arange(numdet).reshape(-1,8,128)
banks = numdet//8//128
f = open(filename,'w')
f.write('<?xml version="1.0" encoding="UTF-8" ?>\n<detector-grouping instrument="'+instrument+'">\n')
groupnum = 0
for i in range(banks):
for j in range(0,8,pixelsx):
for k in range(0,128,pixelsy):
groupname = str(groupnum)
ids = spectra[i, j:j+pixelsx, k:k+pixelsy].reshape(-1)
detids = []
for l in ids:
detids.append(__w.getDetector(int(l)).getID())
detids = str(detids).replace("[","").replace("]","")
f.write('<group name="'+groupname+'">\n <detids val="'+detids+'"/> \n</group>\n')
groupnum += 1
f.write('</detector-grouping>')
f.close()
mantid.simpleapi.DeleteWorkspace(__w.name())
return
mantid.api.AlgorithmFactory.subscribe(GenerateGroupingSNSInelastic)