-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathCRY_load.py
91 lines (78 loc) · 3.07 KB
/
CRY_load.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
from mantid.simpleapi import *
from types import *
import math
oldversion=False
# First a function definition for the Loading algorithms which
# loads the data and immediately aligns the detectors
def Load(outputArea,pathtofile,expt,add=False):
sum=outputArea
if add:
outputArea=outputArea+"add"
try:
fin=open(pathtofile,"r")
except:
print 'WARNING!!!!!!! '+pathtofile+' NOT FOUND'
return 0.0
else:
fin.close()
if oldversion:
LoadRaw(Filename=pathtofile,OutputWorkspace=outputArea)
else:
LoadRaw(Filename=pathtofile,OutputWorkspace=outputArea,LoadLogFiles=False)
if expt.instr=="hrpd":
removeallpromptpulses(outputArea)
uamps = mtd[outputArea].getRun().getProtonCharge()
if add:
Plus(LHSWorkspace=outputArea,RHSWorkspace=sum,OutputWorkspace=sum)
#sum=outputArea+sum
mtd.remove(outputArea)
uampstot = mtd[sum].getRun().getProtonCharge()
return uamps,uampstot
def Align(outputArea,expt):
AlignDetectors(InputWorkspace=outputArea, OutputWorkspace=outputArea,CalibrationFile=expt.Path2OffFile)
def SplitBank(InputArea,bankList, Del=True):
for i in bankList:
CropWorkspace(InputWorkspace=InputArea, OutputWorkspace=InputArea+"-"+str(i), StartWorkspaceIndex=i-1, EndWorkspaceIndex=i-1)
if Del:
mtd.remove(InputArea)
def BinBank(InputArea,bankList,Drange):
for i in bankList:
Rebin(InputWorkspace=InputArea+"-"+str(i), OutputWorkspace=InputArea+"-"+str(i), Params =Drange[i-1])
def setsDrange(wkspc,expt):
datamatrix = mtd[wkspc]
for i in range(0,expt.Nbank):
x_data = datamatrix.readX(i)
last=len(x_data)-1
CropRange=expt.CropRange[i].rstrip().split()
xbegin=str(x_data[0]*(1+float(CropRange[0])))
xend=str(x_data[last]*float(CropRange[1]))
datbin=math.exp(math.log(x_data[last]/x_data[0])/last)-1
if datbin>float(expt.Bining[i]):
print 'WARNING: Rebining in *pref file '+expt.Bining[i]+' is lower than diffraction focusing rebining step'
print 'WARNING: Rebining Kept to be '+str(datbin)+' for bank '+str(i+1)
expt.Bining[i]=str(datbin)
Drange=xbegin+",-"+expt.Bining[i]+","+xend
expt.Drange.append(Drange)
expt.dataRangeSet = True
return setsDrange
def ScaleWspc(InputArea,scale):
CreateSingleValuedWorkspace(OutputWorkspace="scale",DataValue= str(scale))
Multiply(LHSWorkspace=InputArea,RHSWorkspace= "scale",OutputWorkspace= InputArea)
mtd.remove("scale")
def removepromptpulse(wkspace,midle,leftcrop,rightcrop):
mincrop=str(midle-leftcrop)
maxcrop=str(midle+rightcrop)
#RemoveBins(InputWorkspace=wkspace, OutputWorkspace=wkspace, XMin=mincrop, XMax=maxcrop)
MaskBins(InputWorkspace=wkspace, OutputWorkspace=wkspace, XMin=mincrop, XMax=maxcrop)
def removeallpromptpulses(wkspace):
raw_workspace = mtd[wkspace]
x_data = raw_workspace.readX(1)
ppulsemin=int(round(x_data[0])/20000.0)+1
ppulsemax=int(round(x_data[-1])/20000.0)+1
for i in range(ppulsemin,ppulsemax):
removepromptpulse(wkspace,20000.0*float(i),30.0,140.0)
if __name__ == '__main__':
print getSampleList("1000 1245-1268 1308-1400-10")
#CorrectVana(VanFile,EmptyFile,AcalibFile,1)
#normalizeall(SampleDatFile,AcalibFile)
#rearrangeallbanks(OutputFile,"")