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 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 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'])