p61743098
/
sagecal
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Switched to Python 3

This commit is contained in:
Hanno Spreeuw 2018-01-11 16:45:48 +01:00
parent 3e842bae73
commit de385db778
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
src/buildsky/create_clusters.py
View File

@ -145,7 +145,7 @@ def read_lsm_sky(infilename):
mdec=mysign*(abs(float(v.group('col5')))+float(v.group('col6'))/60.0+float(v.group('col7'))/3600.0)*math.pi/180.0 mdec=mysign*(abs(float(v.group('col5')))+float(v.group('col6'))/60.0+float(v.group('col7'))/3600.0)*math.pi/180.0
SR[str(v.group('col1'))]=(v.group('col1'),mra,mdec,float(v.group('col8'))) SR[str(v.group('col1'))]=(v.group('col1'),mra,mdec,float(v.group('col8')))
print 'Read %d sources'%len(SR) print('Read %d sources'%len(SR))
return SR return SR
@ -217,13 +217,13 @@ def cluster_this(skymodel,Q,outfile,max_iterations):
X=numpy.zeros([K,3]) X=numpy.zeros([K,3])
# iterate over sources # iterate over sources
ci=0; ci=0;
for val in SKY.itervalues(): for val in SKY.values():
X[ci,0]=val[1] X[ci,0]=val[1]
X[ci,1]=val[2] X[ci,1]=val[2]
X[ci,2]=val[3] X[ci,2]=val[3]
ci=ci+1 ci=ci+1
# source names # source names
sources=SKY.keys() sources=list(SKY.keys())
# centroids of Q clusters # centroids of Q clusters
C=numpy.zeros([Q,2]) C=numpy.zeros([Q,2])
# 1: select the Q brightest sources, initialize cluster centroids as their locations # 1: select the Q brightest sources, initialize cluster centroids as their locations
@ -258,7 +258,7 @@ def cluster_this(skymodel,Q,outfile,max_iterations):
#print "src %d closest %d"%(ci,closest) #print "src %d closest %d"%(ci,closest)
CL[ci]=closest CL[ci]=closest
# add this source to dict # add this source to dict
if D.has_key(closest): if closest in D:
D[closest].append(ci) D[closest].append(ci)
else: else:
D[closest]=list() D[closest]=list()
@ -272,7 +272,7 @@ def cluster_this(skymodel,Q,outfile,max_iterations):
CLold=numpy.copy(CL) CLold=numpy.copy(CL)
# 3: update the cluster centroids # 3: update the cluster centroids
for (clusid,sourcelist) in D.items(): for (clusid,sourcelist) in list(D.items()):
# update centroid of cluster id 'clusid' # update centroid of cluster id 'clusid'
# project soure ra,dec coordinates to l,m with center of projection # project soure ra,dec coordinates to l,m with center of projection
# taken as current centroid, then take the weighted average # taken as current centroid, then take the weighted average
@ -307,11 +307,11 @@ def cluster_this(skymodel,Q,outfile,max_iterations):
niter=niter+1 niter=niter+1
if no_more_cluster_changes: if no_more_cluster_changes:
print "Stopping after "+str(niter)+" iterations because cluster geometry did not change." print("Stopping after "+str(niter)+" iterations because cluster geometry did not change.")
# write output # write output
outF=open(outfile,'w+') outF=open(outfile,'w+')
outF.write('# Cluster file\n') outF.write('# Cluster file\n')
for (clusid,sourcelist) in D.items(): for (clusid,sourcelist) in list(D.items()):
outF.write(str(clusid+1)+' 1') outF.write(str(clusid+1)+' 1')
for sourceid in sourcelist: for sourceid in sourcelist:
outF.write(' '+sources[sourceid]) outF.write(' '+sources[sourceid])