def buscaArquivo(self):
nomeArquivo=self.nome_arquivo
diretorio=self.diretorio
print "buscaArquivo::[", nomeArquivo, "]"
# flist = SU.ffind(diretorio, shellglobs=("*"+nomeArquivo+"*"))
# SU.printr(flist)
flist = SU.ffind(diretorio, shellglobs=("\*"+nomeArquivo+"\*"))
#SU.printr(flist)
flist = SU.ffind(diretorio, shellglobs=(nomeArquivo, diretorio))
# SU.printr(flist)
self.resultado=flist
return flist
def buscaArquivo(self):
nomeArquivo = self.nome_arquivo
diretorio = self.diretorio
print "buscaArquivo::[", nomeArquivo, "]"
# flist = SU.ffind(diretorio, shellglobs=("*"+nomeArquivo+"*"))
# SU.printr(flist)
flist = SU.ffind(diretorio, shellglobs=("\*" + nomeArquivo + "\*"))
#SU.printr(flist)
flist = SU.ffind(diretorio, shellglobs=(nomeArquivo, diretorio))
# SU.printr(flist)
self.resultado = flist
return flist
def read_data(mydirectory, myfilebase, myend):
myfilebaseglob = myfilebase + '*.' + myend
print myfilebaseglob
flist = SU.ffind(mydirectory, shellglobs=(myfilebaseglob, ))
#SU.printr(flist)
mydatareader = readncnr5.datareader()
Qx = []
Qy = []
Qz = []
Counts = []
Errs = []
mon0 = 240000.0
for currfile in sorted(flist):
mydata = mydatareader.readbuffer(currfile)
mon = mydata.metadata['monitor'] * mydata.metadata['monitor_prefactor']
H, K, L, I = [
np.array(mydata.data[s]) for s in 'h', 'k', 'l', 'counts'
]
idx = I >= 1
print currfile
print "exclude index", np.nonzero(
~idx)[0], "counts", I[~idx], "monitors", mon
Qx.append(H[idx])
Qy.append(K[idx])
Qz.append(L[idx])
Counts.append(I[idx] * mon0 / mon)
Errs.append(np.sqrt(I[idx]) * mon0 / mon)
return [np.concatenate(V) for V in Qx, Qy, Counts, Errs]
def get_file_list(self, model, iter, dir):
""" Get the file list from a given directory """
ls = os.listdir(dir)
ls.sort(key=str.lower)
for i in ls:
path = ospath.join(dir,i)
if ospath.isdir(path) or not self.show_only_dirs :
if (ospath.isdir(path) and len(scriptutil.ffind(path, shellglobs=self.file_types))>0) or not ospath.isdir(path):
if i[0] != '.' or (self.show_hidden and i[0] == '.'):
newiter = model.append(iter)
if ospath.isdir(path): icon = self.get_folder_closed_icon()
else: icon = self.get_file_icon()
model.set_value(newiter, 0, icon)
model.set_value(newiter, 1, i)
model.set_value(newiter, 2, path)
if ospath.isdir(path):
try: subdir = os.listdir(path)
except: subdir = []
if subdir != []:
for i in subdir:
if ospath.isdir(ospath.join(path,i)) or not self.show_only_dirs:
if i[0] != '.' or (self.show_hidden and i[0] == '.'):
self.add_empty_child(model, newiter)
break
def readmeshfiles(mydirectory,myfilebase,myend,eflag='hhl'):
myfilebaseglob=myfilebase+'*.'+myend
print myfilebaseglob
flist = SU.ffind(mydirectory, shellglobs=(myfilebaseglob,))
#SU.printr(flist)
mydatareader=readicp.datareader()
Qx=N.array([])
Qy=N.array([])
Qz=N.array([])
Counts=N.array([])
mon0=80000.0
for currfile in flist:
print currfile
mydata=mydatareader.readbuffer(currfile)
mon=mydata.header['count_info']['monitor']
Qx=N.concatenate((Qx,N.array(mydata.data['Qx'])))
Qy=N.concatenate((Qy,N.array(mydata.data['Qy'])))
Qz=N.concatenate((Qz,N.array(mydata.data['Qz'])))
Counts=N.concatenate((Counts,N.array(mydata.data['Counts'])*mon0/mon))
if eflag=='hhl':
xa,ya,za=prep_data2(Qx,Qz,Counts)
return xa,ya,za,Qx,Qz,Counts
elif eflag=='hkk':
xa,ya,za=prep_data2(Qy,Qx,Counts)
return xa,ya,za,Qx,Qy,Counts
elif eflag=='hkh':
xa,ya,za=prep_data2(Qx,Qy,Counts)
return xa,ya,za
def read_order_files(mydirectory,myfilebase,myend):
myfilebaseglob=myfilebase+'*.'+myend
print myfilebaseglob
flist = SU.ffind(mydirectory, shellglobs=(myfilebaseglob,))
#SU.printr(flist)
mydatareader=readncnr.datareader()
temp=N.array([])
I=N.array([])
Ierr=N.array([])
count=0
mon0=5.0e4
for currfile in flist:
print currfile
mydata=mydatareader.readbuffer(currfile)
if count==0:
#mon0=mydata.header['count_info']['monitor']
mon0=5.0e4
mon=mydata.header['count_info']['monitor']
#print count, mon0,mon
temp=N.concatenate((temp,N.array(mydata.data['temp'])))
It=N.array(mydata.data['counts'])
Iterr=N.sqrt(It)
It=It*mon0/mon
Iterr=Iterr*mon0/mon
I=N.concatenate((I,It))
Ierr=N.concatenate((Ierr,Iterr))
#print I
#print Iterr
#xa,ya,za=prep_data2(Qx,Qy,Counts);
count=count+1
return temp,I,Ierr
def read_order_files(mydirectory, myfilebase, myend):
myfilebaseglob = myfilebase + '*.' + myend
print myfilebaseglob
flist = SU.ffind(mydirectory, shellglobs=(myfilebaseglob, ))
#SU.printr(flist)
mydatareader = readncnr.datareader()
temp = N.array([])
I = N.array([])
Ierr = N.array([])
count = 0
mon0 = 5.0e4
for currfile in flist:
print currfile
mydata = mydatareader.readbuffer(currfile)
if count == 0:
#mon0=mydata.header['count_info']['monitor']
mon0 = 5.0e4
mon = mydata.header['count_info']['monitor']
#print count, mon0,mon
temp = N.concatenate((temp, N.array(mydata.data['temp'])))
It = N.array(mydata.data['counts'])
Iterr = N.sqrt(It)
It = It * mon0 / mon
Iterr = Iterr * mon0 / mon
I = N.concatenate((I, It))
Ierr = N.concatenate((Ierr, Iterr))
#print I
#print Iterr
#xa,ya,za=prep_data2(Qx,Qy,Counts);
count = count + 1
return temp, I, Ierr
def get_file_list(self, model, iter, dir):
""" Get the file list from a given directory """
ls = os.listdir(dir)
ls.sort(key=str.lower)
for i in ls:
path = ospath.join(dir, i)
if ospath.isdir(path) or not self.show_only_dirs:
if (ospath.isdir(path) and
len(scriptutil.ffind(path, shellglobs=self.file_types))
> 0) or not ospath.isdir(path):
if i[0] != '.' or (self.show_hidden and i[0] == '.'):
newiter = model.append(iter)
if ospath.isdir(path):
icon = self.get_folder_closed_icon()
else:
icon = self.get_file_icon()
model.set_value(newiter, 0, icon)
model.set_value(newiter, 1, i)
model.set_value(newiter, 2, path)
if ospath.isdir(path):
try:
subdir = os.listdir(path)
except:
subdir = []
if subdir != []:
for i in subdir:
if ospath.isdir(ospath.join(
path,
i)) or not self.show_only_dirs:
if i[0] != '.' or (self.show_hidden
and i[0] == '.'):
self.add_empty_child(
model, newiter)
break
def selectMap(self, prefix=''):
"""Let user select a map."""
maps = scriptutil.ffind('./maps', (prefix+'*.map',))
userFil = './maps/'+self.askUserInput('Enter map name')+'.map';
if(userFil in maps):
self.map = userFil;
else:
self.printMessageWOption(("",), title="Couldn't find the map!");
return None;
def __init__(self):
self.music = {};
musicF = scriptutil.ffind('./audio', ('music*.mp3',));
for mus in musicF:
self.music[mus] = False;
if len(self.music) == 0:
print "No music found. Music disabled!";
self.disabled = True;
else:
self.disabled = False;
def playSingleMap(self):
"""Play just a single map selected from a list."""
maps = scriptutil.ffind('./maps', ('*.map',));
maps.sort();
i = 0;
## The list lenght. Increase to show more maps on each page.
listLen = 3;
## Change this to the same number as listLen.
lastKey = K_3;
mapNo = 0;
## Outer loop to create map lists
done = False;
while(not done):
msg = [];
lastIndex = i+listLen if(len(maps) > i+listLen) else len(maps);
for j in range(i, lastIndex):
msg.append(str.format("{0} - {1}", j-i+1, os.path.basename(maps[j][0:-4])));
self.screen.fill((0,0,0));
msgSprite = self.printMessage(msg, "Select map: P(revious) N(ext)", align = Message.align_left);
### Inner loop for selection
doneKey = False;
while(not doneKey):
for event in self.getEvents(exitKey=None, allowed = (KEYDOWN, pygame.QUIT)):
if(event.type == KEYDOWN):
if(event.key == K_q):
return;
elif(event.key == K_n):
i = i+listLen if(i+listLen < len(maps)) else i;
doneKey = True;
break;
elif(event.key == K_p):
i -= listLen if(i >listLen+1) else i;
doneKey = True;
break;
elif(event.key >= K_1 and event.key <= lastKey):
mapNo = i + event.key - K_0;
doneKey = True;
self.map = maps[mapNo-1];
self.loadMap();
self.screen.fill((0,0,0));
self.walls.draw(self.screen);
pygame.event.clear();
doneKey = False;
self.selectLevel();
self.play();
self.screen.fill((0,0,0));
msgSprite.draw(self.screen);
pygame.display.flip();
self.clock.tick(50);
msgSprite.clear(self.screen, self.background);
def readmeshfiles(mydirectory,myfilebase,myend,eflag='hhl'):
myfilebaseglob=myfilebase+'*.'+myend
print myfilebaseglob
flist = SU.ffind(mydirectory, shellglobs=(myfilebaseglob,))
#SU.printr(flist)
mydatareader=readicp.datareader()
Qx=N.array([])
Qy=N.array([])
Qz=N.array([])
Counts=N.array([])
#mon0=240000.0
mon0=65000.0
for currfile in flist:
print currfile
mydata=mydatareader.readbuffer(currfile)
mon=mydata.header['count_info']['monitor']
o1=N.array([mydata.header['orient1']['h'],mydata.header['orient1']['k'],mydata.header['orient1']['l']])
o2=N.array([mydata.header['orient2']['h'],mydata.header['orient2']['k'],mydata.header['orient2']['l']])
o3=N.cross(o1,o2)
h=N.array(mydata.data['Qx'])
k=N.array(mydata.data['Qy'])
l=N.array(mydata.data['Qz'])
A=N.array([o1,o2,o3]).T
a_arr=[]
b_arr=[]
for i in range(len(h)):
hkl=N.array([h[i],k[i],l[i]])
sol=N.linalg.solve(A,hkl)
a=sol[0]
b=sol[1]
a_arr.append(a)
b_arr.append(b)
if eflag=='weird':
Qx=N.concatenate((Qx,a_arr))
Qy=N.concatenate((Qy,b_arr))
else:
Qx=N.concatenate((Qx,N.array(mydata.data['Qx'])))
Qy=N.concatenate((Qy,N.array(mydata.data['Qy'])))
Qz=N.concatenate((Qz,N.array(mydata.data['Qz'])))
Counts=N.concatenate((Counts,N.array(mydata.data['Counts'])*mon0/mon))
if eflag=='hhl':
xa,ya,za=prep_data2(Qx,Qz,Counts)
elif eflag=='hkk':
xa,ya,za=prep_data2(Qy,Qx,Counts)
elif eflag=='hkh':
xa,ya,za=prep_data2(Qx,Qy,Counts)
elif eflag=='weird':
xa,ya,za=prep_data2(Qy,Qx,Counts)
return xa,ya,za
def readmeshfiles(mydirectory,myfilebase,myend,eflag='hhl'):
myfilebaseglob=myfilebase+'*.'+myend
print myfilebaseglob
flist = SU.ffind(mydirectory, shellglobs=(myfilebaseglob,))
#SU.printr(flist)
mydatareader=readicp.datareader()
Qx=N.array([])
Qy=N.array([])
Qz=N.array([])
Counts=N.array([])
#mon0=240000.0
mon0=65000.0
for currfile in flist:
print currfile
mydata=mydatareader.readbuffer(currfile)
mon=mydata.header['count_info']['monitor']
o1=N.array([mydata.header['orient1']['h'],mydata.header['orient1']['k'],mydata.header['orient1']['l']])
o2=N.array([mydata.header['orient2']['h'],mydata.header['orient2']['k'],mydata.header['orient2']['l']])
o3=N.cross(o1,o2)
h=N.array(mydata.data['Qx'])
k=N.array(mydata.data['Qy'])
l=N.array(mydata.data['Qz'])
A=N.array([o1,o2,o3]).T
a_arr=[]
b_arr=[]
for i in range(len(h)):
hkl=N.array([h[i],k[i],l[i]])
sol=N.linalg.solve(A,hkl)
a=sol[0]
b=sol[1]
a_arr.append(a)
b_arr.append(b)
if eflag=='weird':
Qx=N.concatenate((Qx,a_arr))
Qy=N.concatenate((Qy,b_arr))
else:
Qx=N.concatenate((Qx,N.array(mydata.data['Qx'])))
Qy=N.concatenate((Qy,N.array(mydata.data['Qy'])))
Qz=N.concatenate((Qz,N.array(mydata.data['Qz'])))
Counts=N.concatenate((Counts,N.array(mydata.data['Counts'])*mon0/mon))
if eflag=='hhl':
xa,ya,za=prep_data2(Qx,Qz,Counts)
elif eflag=='hkk':
xa,ya,za=prep_data2(Qy,Qx,Counts)
elif eflag=='hkh':
xa,ya,za=prep_data2(Qx,Qy,Counts)
elif eflag=='weird':
xa,ya,za=prep_data2(Qy,Qx,Counts)
return xa,ya,za
def readmeshfiles(mydirectory, myfilebase, myend, eflag="hhl"):
myfilebaseglob = myfilebase + "*." + myend
print myfilebaseglob
flist = SU.ffind(mydirectory, shellglobs=(myfilebaseglob,))
# SU.printr(flist)
mydatareader = readicp.datareader()
Qx = N.array([])
Qy = N.array([])
Qz = N.array([])
Counts = N.array([])
# mon0=240000.0
mon0 = 65000.0
for currfile in flist:
print currfile
mydata = mydatareader.readbuffer(currfile)
mon = mydata.header["count_info"]["monitor"]
o1 = N.array([mydata.header["orient1"]["h"], mydata.header["orient1"]["k"], mydata.header["orient1"]["l"]])
o2 = N.array([mydata.header["orient2"]["h"], mydata.header["orient2"]["k"], mydata.header["orient2"]["l"]])
o3 = N.cross(o1, o2)
h = N.array(mydata.data["Qx"])
k = N.array(mydata.data["Qy"])
l = N.array(mydata.data["Qz"])
A = N.array([o1, o2, o3]).T
a_arr = []
b_arr = []
for i in range(len(h)):
hkl = N.array([h[i], k[i], l[i]])
sol = N.linalg.solve(A, hkl)
a = sol[0]
b = sol[1]
a_arr.append(a)
b_arr.append(b)
if eflag == "weird":
Qx = N.concatenate((Qx, a_arr))
Qy = N.concatenate((Qy, b_arr))
else:
Qx = N.concatenate((Qx, N.array(mydata.data["Qx"])))
Qy = N.concatenate((Qy, N.array(mydata.data["Qy"])))
Qz = N.concatenate((Qz, N.array(mydata.data["Qz"])))
Counts = N.concatenate((Counts, N.array(mydata.data["Counts"]) * mon0 / mon))
if eflag == "hhl":
xa, ya, za = prep_data2(Qx, Qz, Counts)
elif eflag == "hkk":
xa, ya, za = prep_data2(Qy, Qx, Counts)
elif eflag == "hkh":
xa, ya, za = prep_data2(Qx, Qy, Counts)
elif eflag == "weird":
xa, ya, za = prep_data2(Qy, Qx, Counts)
return xa, ya, za
def get_audio_files(path):
"""
Return a list of plain file with audio extensions.
"""
def file_filter(s):
"""True for plain files with an audio extension."""
return (
os.path.isfile(s) and not os.path.islink(s) and
audio_file_regex.search(s))
audio_files = scriptutil.ffind(path, namefs=(file_filter,))
return [os.path.basename(audio_file) for audio_file in audio_files]
def readfiles(mydirectory,myfilebase,myend):
myfilebaseglob=myfilebase+'*.'+myend
print myfilebaseglob
flist = SU.ffind(mydirectory, shellglobs=(myfilebaseglob,))
Qx=N.array([])
Qy=N.array([])
Qz=N.array([])
Counts=N.array([])
for currfile in flist:
print currfile
mydata=mydatareader.readbuffer(currfile)
Qx=N.concatenate((Qx,N.array(mydata.data['qx'])))
Qy=N.concatenate((Qy,N.array(mydata.data['qy'])))
Qz=N.concatenate((Qz,N.array(mydata.data['qz'])))
Counts=N.concatenate((Counts,N.array(mydata.data['counts'])))
def playLevels(self):
"""Play each level in turn."""
self.selectLevel();
maps = scriptutil.ffind('./maps', ('Level*.map',))
maps.sort();
for map in maps:
self.map = map;
self.loadMap();
self.screen.fill((0,0,0));
self.walls.draw(self.screen);
pygame.event.clear();
if(self.play() == False):
break;
self.printEndGame();
def read_data(mydirectory, myfilebase, myend):
myfilebaseglob = myfilebase + '*.' + myend
print myfilebaseglob
flist = SU.ffind(mydirectory, shellglobs=(myfilebaseglob, ))
#SU.printr(flist)
mydatareader = readncnr5.datareader()
datasets = []
for currfile in sorted(flist):
print currfile
mydata = mydatareader.readbuffer(currfile)
mon = mydata.metadata['monitor'] * mydata.metadata['monitor_prefactor']
H, K, L = [np.array(mydata.data[s]) for s in 'h', 'k', 'l']
I = np.array(mydata.data['counts'], 'int32')
datasets.append((H, K, I, np.ones_like(I) * mon))
return Data(*[np.concatenate(V) for V in zip(*datasets)])
def readmeshfiles(mydirectory,myfilebase,myend):
myfilebaseglob=myfilebase+'*.'+myend
print myfilebaseglob
flist = SU.ffind(mydirectory, shellglobs=(myfilebaseglob,))
#SU.printr(flist)
mydatareader=readicp.datareader()
Qx=N.array([])
Qy=N.array([])
Qz=N.array([])
Counts=N.array([])
for currfile in flist:
print currfile
mydata=mydatareader.readbuffer(currfile)
Qx=N.concatenate((Qx,N.array(mydata.data['Qx'])))
Qy=N.concatenate((Qy,N.array(mydata.data['Qy'])))
Qz=N.concatenate((Qz,N.array(mydata.data['Qz'])))
Counts=N.concatenate((Counts,N.array(mydata.data['Counts'])))
xa,ya,za=prep_data2(Qx,Qy,Counts);
return xa,ya,za
def scoreList(self):
"""Print the scorelist of each map."""
maps = scriptutil.ffind('./maps', ('*.map',))
maps.sort();
self.screen.fill((0,0,0));
for map in maps:
msgSprite = self.scores.printScores(map);
msgSprite.draw(self.screen);
pygame.display.flip();
done = False;
while(not done):
for event in self.getEvents(exitKey=None):
if(event.type == KEYDOWN):
if(event.key == K_RETURN or event.key == K_SPACE):
done = True;
elif(event.key == K_q):
msgSprite.clear(self.screen, self.background);
return;
msgSprite.clear(self.screen, self.background);
def readmeshfiles(mydirectory, myfilebase, myend):
myfilebaseglob = myfilebase + "*." + myend
print myfilebaseglob
flist = SU.ffind(mydirectory, shellglobs=(myfilebaseglob,))
# SU.printr(flist)
mydatareader = readncnr.datareader()
Qx = N.array([])
Qy = N.array([])
Qz = N.array([])
Counts = N.array([])
for currfile in flist:
print currfile
mydata = mydatareader.readbuffer(currfile)
Qx = N.concatenate((Qx, N.array(mydata.data["qx"])))
Qy = N.concatenate((Qy, N.array(mydata.data["qy"])))
Qz = N.concatenate((Qz, N.array(mydata.data["qz"])))
Counts = N.concatenate((Counts, N.array(mydata.data["detector"])))
xa, ya, za = prep_data2(Qx, Qy, Counts)
return xa, ya, za
def read_order_files(mydirectory,myfilebase,myend):
myfilebaseglob=myfilebase+'*.'+myend
# print myfilebaseglob
flist = SU.ffind(mydirectory, shellglobs=(myfilebaseglob,))
#SU.printr(flist)
mydatareader=readncnr.datareader()
temp=[]#N.array([])
I=[]#N.array([])
Ierr=[]#N.array([])
monlist=[]
count=0
mon0=5.0e4
for currfile in flist:
#print currfile
mydata=mydatareader.readbuffer(currfile)
#if count==0:
# #mon0=mydata.header['count_info']['monitor']
# mon0=5.0e4
mon=mydata.metadata['count_info']['monitor']
#print count, mon0,mon
#temp=N.concatenate((temp,N.array(mydata.data['temp'])))
temp.append(N.array(mydata.data['temp'],'float64'))
It=N.array(N.array(mydata.data['counts']),'float64')
Iterr=N.sqrt(It)
#It=It*mon0/mon
monlist.append(mon)
I.append(It)
Ierr.append(Iterr)
#Iterr=Iterr*mon0/mon
#I=N.concatenate((I,It))
#Ierr=N.concatenate((Ierr,Iterr))
#print I
#print Iterr
#xa,ya,za=prep_data2(Qx,Qy,Counts);
count=count+1
return temp,I,Ierr,monlist
def readmeshfiles(mydirectory,myfilebase,myend):
myfilebaseglob=myfilebase+'*.'+myend
print myfilebaseglob
flist = SU.ffind(mydirectory, shellglobs=(myfilebaseglob,))
#SU.printr(flist)
mydatareader=readicp.datareader()
Qx=N.array([])
Qy=N.array([])
Qz=N.array([])
Counts=N.array([])
a=N.array([3.943],'d')
b=N.array([2.779],'d')
for currfile in flist:
print currfile
mydata=mydatareader.readbuffer(currfile)
Qx=N.concatenate((Qx,N.array(mydata.data['Qx'])*2*pi/a[0]))
Qy=N.concatenate((Qy,N.array(mydata.data['Qy'])*2*pi/b[0]))
Qz=N.concatenate((Qz,N.array(mydata.data['Qz'])))
Counts=N.concatenate((Counts,N.array(mydata.data['Counts'])))
xa,ya,za=prep_data2(Qx,Qy,Counts);
print 'xa',xa.min(),xa.max()
print 'qx',Qx.min(),Qx.max()
print
return xa,ya,za
def readmeshfiles(mydirectory,myfilebase,myend):
myfilebaseglob=myfilebase+'*.'+myend
print myfilebaseglob
flist = SU.ffind(mydirectory, shellglobs=(myfilebaseglob,))
#SU.printr(flist)
mydatareader=readicp.datareader()
Qx=N.array([])
Qy=N.array([])
Qz=N.array([])
Counts=N.array([])
a=N.array([3.943],'d')
b=N.array([2.779],'d')
for currfile in flist:
print currfile
mydata=mydatareader.readbuffer(currfile)
Qx=N.concatenate((Qx,N.array(mydata.data['Qx'])*2*pi/a[0]))
Qy=N.concatenate((Qy,N.array(mydata.data['Qy'])*2*pi/b[0]))
Qz=N.concatenate((Qz,N.array(mydata.data['Qz'])))
Counts=N.concatenate((Counts,N.array(mydata.data['Counts'])))
xa,ya,za=prep_data2(Qx,Qy,Counts);
print 'xa',xa.min(),xa.max()
print 'qx',Qx.min(),Qx.max()
print
return xa,ya,za
def read_order_files(mydirectory, myfilebase, myend):
myfilebaseglob = myfilebase + '*.' + myend
# print myfilebaseglob
flist = SU.ffind(mydirectory, shellglobs=(myfilebaseglob, ))
#SU.printr(flist)
mydatareader = readncnr.datareader()
temp = [] #N.array([])
I = [] #N.array([])
Ierr = [] #N.array([])
monlist = []
count = 0
mon0 = 5.0e4
for currfile in flist:
#print currfile
mydata = mydatareader.readbuffer(currfile)
#if count==0:
# #mon0=mydata.header['count_info']['monitor']
# mon0=5.0e4
mon = mydata.metadata['count_info']['monitor']
#print count, mon0,mon
#temp=N.concatenate((temp,N.array(mydata.data['temp'])))
temp.append(N.array(mydata.data['temp'], 'float64'))
It = N.array(N.array(mydata.data['counts']), 'float64')
Iterr = N.sqrt(It)
#It=It*mon0/mon
monlist.append(mon)
I.append(It)
Ierr.append(Iterr)
#Iterr=Iterr*mon0/mon
#I=N.concatenate((I,It))
#Ierr=N.concatenate((Ierr,Iterr))
#print I
#print Iterr
#xa,ya,za=prep_data2(Qx,Qy,Counts);
count = count + 1
return temp, I, Ierr, monlist
def main():
do_java = False
do_csharp = True
# get list of test directories
test_dirs = os.listdir('./')
cpp_files = [
'Array8/Array8_cmpt_1/array8_utst.cpp',
'Array9/Array9_cmpt_1/array9_utst.cpp',
'Array10/Array10_cmpt_1/array10_utst.cpp',
'Variant2/Variant2_cmpt_1/variant2_utst.cpp',
'Variant3/Variant3_cmpt_1/variant3_utst.cpp',
'Variant4/Variant4_cmpt_1/variant4_utst.cpp'
]
java_files = [
'Array8/Array8_java_1/Array8_utst.java',
'Array9/Array9_java_1/Array9_utst.java',
'Array10/Array10_java_1/Array10_utst.java',
'Variant2/Variant2_java_1/Variant2_utst.java',
'Variant3/Variant3_java_1/Variant3_utst.java',
'Variant4/Variant4_java_1/Variant4_utst.java'
]
csharp_files = [
'Array8/Array8_csharp_1/src/Array8_utst.cs',
'Array9/Array9_csharp_1/src/Array9_utst.cs',
'Array10/Array10_csharp_1/src/Array10_utst.cs',
'Variant2/Variant2_csharp_1/src/Variant2_utst.cs',
'Variant3/Variant3_csharp_1/src/Variant3_utst.cs',
'Variant4/Variant4_csharp_1/src/Variant4_utst.cs'
]
java_filenames = [
'Array8_utst.java',
'Array9_utst.java',
'Array10_utst.java',
'Variant2_utst.java',
'Variant3_utst.java',
'Variant4_utst.java',
]
csharp_filenames = [
'Array8_utst.cs',
'Array9_utst.cs',
'Array10_utst.cs',
'Variant2_utst.cs',
'Variant3_utst.cs',
'Variant4_utst.cs',
]
if do_java:
# start with copies of the C++ files
for (cpp_file, java_file) in zip(cpp_files, java_files):
shutil.copy(cpp_file, java_file)
scriptutil.freplace(path='./',
regexl=java_exprs,
shellglobs=java_filenames)
bak_files = scriptutil.ffind(path='./', shellglobs=['*_utst.java.bak'])
if do_csharp:
# start with copies of the C++ files
for (cpp_file, csharp_file) in zip(cpp_files, csharp_files):
shutil.copy(cpp_file, csharp_file)
scriptutil.freplace(path='./',
regexl=csharp_exprs,
shellglobs=csharp_filenames)
bak_files = scriptutil.ffind(path='./', shellglobs=['*_utst.cs.bak'])
EXP['vcol']=N.array([120, 120, 120, 120],'d')
EXP['infix']=-1 #positive for fixed incident energy
EXP['efixed']=14.7
EXP['method']=0
setup=[EXP]
mydirectory=r'c:\BiFeO3xtal\Oct9_2007'
myfilebase='bfo_spinflip_he3flip51585'
myfilebase2='bfo_spinflip51579'
myfilebase3='bfo_spinflip51583'
myend='bt7'
data={}
#pm
myfilebaseglob=myfilebase+'*.'+myend
print myfilebaseglob
flist = SU.ffind(mydirectory, shellglobs=(myfilebaseglob,))
myfilestr=flist[0]
Counts=N.array([],'float64')
monlist=[]
timestamp={}
mydatareader=readncnr.datareader()
mydata=mydatareader.readbuffer(myfilestr)
S1=N.array(mydata.data['a3'][1:])
S2=N.array(mydata.data['a4'][1:])
A2=N.array(mydata.data['a5'][1:])*2
M2=N.array(mydata.data['a2'][1:])
Counts_pm=N.concatenate((Counts,N.array(mydata.data['detector'][1:])))
H_pm,K_pm,L_pm,E_pm,Q_pm,Ei_pm,Ef_pm=mylattice.SpecWhere(myradians(M2),myradians(S1),myradians(S2),myradians(A2),setup)
monlist.append(mydata.data['monitor'][0])
Counts_pm=Counts_pm[:-1]
H_pm=H_pm[:-1]
def readmeshfiles(mydirectory,myfilebase,myend):
myfilebaseglob=myfilebase+'*.'+myend
print myfilebaseglob
flist = SU.ffind(mydirectory, shellglobs=(myfilebaseglob,))
#SU.printr(flist)
mydatareader=readncnr.datareader()
temp1=N.array([])
temp2=N.array([])
temp3=N.array([])
Counts1=N.array([])
Counts2=N.array([])
Counts3=N.array([])
errors1=N.array([])
errors2=N.array([])
errors3=N.array([])
mydata=mydatareader.readbuffer(flist[0])
mon0=mydata.metadata['count_info']['monitor']
for currfile in flist:
#print currfile
mydata=mydatareader.readbuffer(currfile)
#print mydata.data.keys()
#print mydata.__dict__
#print mydata.metadata.keys()
qcenter=mydata.metadata['q_center']
hc,kc,lc=qcenter['h_center'],qcenter['k_center'],qcenter['l_center']
mon=mydata.metadata['count_info']['monitor']
curr_counts=N.array(mydata.data['counts'])
curr_error=N.sqrt(curr_counts)*mon0/mon
curr_counts=curr_counts*mon0/mon
curr_temp=N.array(mydata.data['temp'])
if hc==1.004:
Counts1=N.concatenate((Counts1,curr_counts))
temp1=N.concatenate((temp1,curr_temp))
errors1=N.concatenate((errors1,curr_error))
elif hc==1.036:
Counts2=N.concatenate((Counts2,curr_counts))
temp2=N.concatenate((temp2,curr_temp))
errors2=N.concatenate((errors2,curr_error))
elif hc==2.0:
Counts3=N.concatenate((Counts3,curr_counts))
temp3=N.concatenate((temp3,curr_temp))
errors3=N.concatenate((errors3,curr_error))
#print
#Qx=N.concatenate((Qx,N.array(mydata.data['qx'])))
#Qy=N.concatenate((Qy,N.array(mydata.data['qy'])))
#Qz=N.concatenate((Qz,N.array(mydata.data['qz'])))
#Counts=N.concatenate((Counts,N.array(mydata.data['counts'])))
#print Qx
#print Qy
#print Counts
dataset={}
dataset['Counts1']=Counts1
dataset['Counts2']=Counts2
dataset['Counts3']=Counts3
dataset['temp1']=temp1
dataset['temp2']=temp2
dataset['temp3']=temp3
dataset['errors1']=errors1
dataset['errors2']=errors2
dataset['errors3']=errors3
return dataset
def readmeshfiles(mydirectory, myfilebase, myend):
myfilebaseglob = myfilebase + '*.' + myend
print myfilebaseglob
flist = SU.ffind(mydirectory, shellglobs=(myfilebaseglob, ))
#SU.printr(flist)
mydatareader = readncnr.datareader()
temp1 = N.array([])
temp2 = N.array([])
temp3 = N.array([])
Counts1 = N.array([])
Counts2 = N.array([])
Counts3 = N.array([])
errors1 = N.array([])
errors2 = N.array([])
errors3 = N.array([])
mydata = mydatareader.readbuffer(flist[0])
mon0 = mydata.metadata['count_info']['monitor']
for currfile in flist:
#print currfile
mydata = mydatareader.readbuffer(currfile)
#print mydata.data.keys()
#print mydata.__dict__
#print mydata.metadata.keys()
qcenter = mydata.metadata['q_center']
hc, kc, lc = qcenter['h_center'], qcenter['k_center'], qcenter[
'l_center']
mon = mydata.metadata['count_info']['monitor']
curr_counts = N.array(mydata.data['counts'])
curr_error = N.sqrt(curr_counts) * mon0 / mon
curr_counts = curr_counts * mon0 / mon
curr_temp = N.array(mydata.data['temp'])
if hc == 1.004:
Counts1 = N.concatenate((Counts1, curr_counts))
temp1 = N.concatenate((temp1, curr_temp))
errors1 = N.concatenate((errors1, curr_error))
elif hc == 1.036:
Counts2 = N.concatenate((Counts2, curr_counts))
temp2 = N.concatenate((temp2, curr_temp))
errors2 = N.concatenate((errors2, curr_error))
elif hc == 2.0:
Counts3 = N.concatenate((Counts3, curr_counts))
temp3 = N.concatenate((temp3, curr_temp))
errors3 = N.concatenate((errors3, curr_error))
#print
#Qx=N.concatenate((Qx,N.array(mydata.data['qx'])))
#Qy=N.concatenate((Qy,N.array(mydata.data['qy'])))
#Qz=N.concatenate((Qz,N.array(mydata.data['qz'])))
#Counts=N.concatenate((Counts,N.array(mydata.data['counts'])))
#print Qx
#print Qy
#print Counts
dataset = {}
dataset['Counts1'] = Counts1
dataset['Counts2'] = Counts2
dataset['Counts3'] = Counts3
dataset['temp1'] = temp1
dataset['temp2'] = temp2
dataset['temp3'] = temp3
dataset['errors1'] = errors1
dataset['errors2'] = errors2
dataset['errors3'] = errors3
return dataset
def main():
do_java = False;
do_csharp = True;
# get list of test directories
test_dirs = os.listdir('./')
cpp_files = [
'Array8/Array8_cmpt_1/array8_utst.cpp',
'Array9/Array9_cmpt_1/array9_utst.cpp',
'Array10/Array10_cmpt_1/array10_utst.cpp',
'Variant2/Variant2_cmpt_1/variant2_utst.cpp',
'Variant3/Variant3_cmpt_1/variant3_utst.cpp',
'Variant4/Variant4_cmpt_1/variant4_utst.cpp'
]
java_files = [
'Array8/Array8_java_1/Array8_utst.java',
'Array9/Array9_java_1/Array9_utst.java',
'Array10/Array10_java_1/Array10_utst.java',
'Variant2/Variant2_java_1/Variant2_utst.java',
'Variant3/Variant3_java_1/Variant3_utst.java',
'Variant4/Variant4_java_1/Variant4_utst.java'
]
csharp_files = [
'Array8/Array8_csharp_1/src/Array8_utst.cs',
'Array9/Array9_csharp_1/src/Array9_utst.cs',
'Array10/Array10_csharp_1/src/Array10_utst.cs',
'Variant2/Variant2_csharp_1/src/Variant2_utst.cs',
'Variant3/Variant3_csharp_1/src/Variant3_utst.cs',
'Variant4/Variant4_csharp_1/src/Variant4_utst.cs'
]
java_filenames = [
'Array8_utst.java',
'Array9_utst.java',
'Array10_utst.java',
'Variant2_utst.java',
'Variant3_utst.java',
'Variant4_utst.java',
]
csharp_filenames = [
'Array8_utst.cs',
'Array9_utst.cs',
'Array10_utst.cs',
'Variant2_utst.cs',
'Variant3_utst.cs',
'Variant4_utst.cs',
]
if do_java:
# start with copies of the C++ files
for (cpp_file, java_file) in zip(cpp_files, java_files):
shutil.copy(cpp_file, java_file)
scriptutil.freplace(path='./', regexl=java_exprs, shellglobs=java_filenames)
bak_files = scriptutil.ffind(path='./', shellglobs=['*_utst.java.bak'])
if do_csharp:
# start with copies of the C++ files
for (cpp_file, csharp_file) in zip(cpp_files, csharp_files):
shutil.copy(cpp_file, csharp_file)
scriptutil.freplace(path='./', regexl=csharp_exprs, shellglobs=csharp_filenames)
bak_files = scriptutil.ffind(path='./', shellglobs=['*_utst.cs.bak'])
