Commit dec10f3e authored by Wei Chengliang's avatar Wei Chengliang
Browse files

update style-PEP8

parent f9c38528
Pipeline #7094 failed with stage
in 0 seconds
......@@ -19,7 +19,7 @@ NPSF = 900 # ***# 30*30
PixSizeInMicrons = 5. # ***# in microns
### find neighbors-KDtree###
# find neighbors-KDtree #
def findNeighbors(tx, ty, px, py, dr=0.1, dn=1, OnlyDistance=True):
"""
find nearest neighbors by 2D-KDTree
......@@ -40,9 +40,9 @@ def findNeighbors(tx, ty, px, py, dr=0.1, dn=1, OnlyDistance=True):
dataq = []
rr = dr
if OnlyDistance == True:
if OnlyDistance is True:
dataq = tree.query_ball_point([tx, ty], rr)
if OnlyDistance == False:
if OnlyDistance is False:
while len(dataq) < dn:
dataq = tree.query_ball_point([tx, ty], rr)
rr += dr
......@@ -51,7 +51,7 @@ def findNeighbors(tx, ty, px, py, dr=0.1, dn=1, OnlyDistance=True):
dataq = np.array(dataq)[ddSortindx[0:dn]]
return dataq
### find neighbors-hoclist###
# find neighbors-hoclist#
def hocBuild(partx, party, nhocx, nhocy, dhocx, dhocy):
......@@ -139,7 +139,7 @@ def findNeighbors_hoclist(px, py, tx=None, ty=None, dn=4, hoc=None, hoclist=None
return neigh
### PSF-IDW###
# PSF-IDW#
def psfMaker_IDW(px, py, PSFMat, cen_col, cen_row, IDWindex=2, OnlyNeighbors=True, hoc=None, hoclist=None, PSFCentroidWgt=False):
"""
psf interpolation by IDW
......@@ -163,7 +163,7 @@ def psfMaker_IDW(px, py, PSFMat, cen_col, cen_row, IDWindex=2, OnlyNeighbors=Tru
npsf = PSFMat[:, :, :].shape[0]
psfWeight = np.zeros([npsf])
if OnlyNeighbors == True:
if OnlyNeighbors is True:
if hoc is None:
neigh = findNeighbors(px, py, cen_col, cen_row,
dr=5., dn=4, OnlyDistance=False)
......@@ -175,7 +175,7 @@ def psfMaker_IDW(px, py, PSFMat, cen_col, cen_row, IDWindex=2, OnlyNeighbors=Tru
neighFlag[neigh] = 1
for ipsf in range(npsf):
if OnlyNeighbors == True:
if OnlyNeighbors is True:
if neighFlag[ipsf] != 1:
continue
......@@ -195,7 +195,7 @@ def psfMaker_IDW(px, py, PSFMat, cen_col, cen_row, IDWindex=2, OnlyNeighbors=Tru
psfMaker = np.zeros([ngy, ngx], dtype=np.float32)
for ipsf in range(npsf):
if OnlyNeighbors == True:
if OnlyNeighbors is True:
if neighFlag[ipsf] != 1:
continue
......@@ -208,7 +208,7 @@ def psfMaker_IDW(px, py, PSFMat, cen_col, cen_row, IDWindex=2, OnlyNeighbors=Tru
return psfMaker
### define PSFInterp###
# define PSFInterp#
class PSFInterp(PSFModel):
def __init__(self, chip, npsf=NPSF, PSF_data=None, PSF_data_file=None, PSF_data_prefix="", sigSpin=0, psfRa=0.15, HocBuild=False, LOG_DEBUG=False):
self.LOG_DEBUG = LOG_DEBUG
......@@ -223,7 +223,7 @@ class PSFInterp(PSFModel):
self.iccd = int(chip.getChipLabel(chipID=chip.chipID))
# self.iccd = chip.chip_name
if PSF_data_file == None:
if PSF_data_file is None:
print('Error - PSF_data_file is None')
sys.exit()
......@@ -402,7 +402,7 @@ class PSFInterp(PSFModel):
Spinned PSF: g1, g2 and axis ratio 'a/b'
"""
a2Rad = np.pi/(60.0*60.0*180.0)
ff = self.sigGauss * 0.107 * (1000.0/10.0) # in unit of [pixels]
rc = np.sqrt(x*x + y*y)
cpix = rc*(self.sigSpin*a2Rad)
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment