Commit 3351be22 authored by Wei Chengliang's avatar Wei Chengliang
Browse files

update codestyle-PEP8

parent 8fd9723d
...@@ -98,7 +98,7 @@ class Quasar(MockObject): ...@@ -98,7 +98,7 @@ class Quasar(MockObject):
del self.sed del self.sed
def getGSObj(self, psf, g1=0, g2=0, flux=None, filt=None, tel=None, exptime=150.): def getGSObj(self, psf, g1=0, g2=0, flux=None, filt=None, tel=None, exptime=150.):
if flux == None: if flux is None:
flux = self.getElectronFluxFilt(filt, tel, exptime) flux = self.getElectronFluxFilt(filt, tel, exptime)
qso = galsim.Gaussian(sigma=1.e-8, flux=1.) qso = galsim.Gaussian(sigma=1.e-8, flux=1.)
qso = qso.withFlux(flux) qso = qso.withFlux(flux)
......
...@@ -21,7 +21,7 @@ class Stamp(MockObject): ...@@ -21,7 +21,7 @@ class Stamp(MockObject):
del self.sed del self.sed
def drawObj_multiband(self, tel, pos_img, psf_model, bandpass_list, filt, chip, nphotons_tot=None, g1=0, g2=0, exptime=150., fd_shear=None): def drawObj_multiband(self, tel, pos_img, psf_model, bandpass_list, filt, chip, nphotons_tot=None, g1=0, g2=0, exptime=150., fd_shear=None):
if nphotons_tot == None: if nphotons_tot is None:
nphotons_tot = self.getElectronFluxFilt(filt, tel, exptime) nphotons_tot = self.getElectronFluxFilt(filt, tel, exptime)
try: try:
......
...@@ -22,7 +22,7 @@ class Star(MockObject): ...@@ -22,7 +22,7 @@ class Star(MockObject):
del self.sed del self.sed
def getGSObj(self, psf, g1=0, g2=0, flux=None, filt=None, tel=None, exptime=150.): def getGSObj(self, psf, g1=0, g2=0, flux=None, filt=None, tel=None, exptime=150.):
if flux == None: if flux is None:
flux = self.getElectronFluxFilt(filt, tel, exptime) flux = self.getElectronFluxFilt(filt, tel, exptime)
# star = galsim.Gaussian(sigma=1.e-8, flux=1.) # star = galsim.Gaussian(sigma=1.e-8, flux=1.)
star = galsim.DeltaFunction() star = galsim.DeltaFunction()
...@@ -35,7 +35,7 @@ class Star(MockObject): ...@@ -35,7 +35,7 @@ class Star(MockObject):
raise ValueError( raise ValueError(
"!!!The number of PSF profiles and the number of bandpasses must be equal.") "!!!The number of PSF profiles and the number of bandpasses must be equal.")
objs = [] objs = []
if nphotons_tot == None: if nphotons_tot is None:
nphotons_tot = self.getElectronFluxFilt(filt, tel, exptime) nphotons_tot = self.getElectronFluxFilt(filt, tel, exptime)
try: try:
......
...@@ -58,7 +58,7 @@ def seds(sedlistn, seddir="./", unit="A"): ...@@ -58,7 +58,7 @@ def seds(sedlistn, seddir="./", unit="A"):
read SEDs and save into Python dictionary read SEDs and save into Python dictionary
Parameters: Parameters:
sedlistn: filename of the sed template list and corresponding intrinsic sedlistn: filename of the sed template list and corresponding intrinsic
extinction model, see tmag_dz.py for detailes extinction model, see tmag_dz.py for detailes
listdir: directory of the list listdir: directory of the list
unit: wavelength unit of the input templates unit: wavelength unit of the input templates
...@@ -420,7 +420,7 @@ def sed2mag(mag_i, sedCat, filter_list, redshift=0.0, av=0.0, redden=0): ...@@ -420,7 +420,7 @@ def sed2mag(mag_i, sedCat, filter_list, redshift=0.0, av=0.0, redden=0):
def eObs(e1, e2, g1, g2): def eObs(e1, e2, g1, g2):
""" """
Calculate the sheared (observed) ellipticity using the Calculate the sheared (observed) ellipticity using the
intrinsic ellipticity and cosmic shear components. intrinsic ellipticity and cosmic shear components.
Parameters: Parameters:
...@@ -558,7 +558,7 @@ def getNormFactorForSpecWithABMAG(ABMag=20., spectrum=None, norm_thr=None, sWave ...@@ -558,7 +558,7 @@ def getNormFactorForSpecWithABMAG(ABMag=20., spectrum=None, norm_thr=None, sWave
eWave: the end of norm_thr eWave: the end of norm_thr
Return: Return:
the normalization factor flux of AB system(fix a band and magnitude ) /the flux of inpute spectrum(fix a band) the normalization factor flux of AB system(fix a band and magnitude ) /the flux of inpute spectrum(fix a band)
""" """
spectrumi = interpolate.interp1d(spectrum['WAVELENGTH'], spectrum['FLUX']) spectrumi = interpolate.interp1d(spectrum['WAVELENGTH'], spectrum['FLUX'])
norm_thri = interpolate.interp1d( norm_thri = interpolate.interp1d(
......
...@@ -428,8 +428,8 @@ class TestSpecDisperse(unittest.TestCase): ...@@ -428,8 +428,8 @@ class TestSpecDisperse(unittest.TestCase):
plt.xlim(6200, 10000) plt.xlim(6200, 10000)
plt.ylim(1, 4) plt.ylim(1, 4)
plt.yscale('log') plt.yscale('log')
plt.xlabel('$\lambda$') plt.xlabel(r'$\lambda$')
plt.ylabel('$F\lambda$') plt.ylabel(r'$F\lambda$')
plt.legend(['one spec', 'split in 8000 A']) plt.legend(['one spec', 'split in 8000 A'])
plt.show() plt.show()
......
...@@ -192,7 +192,7 @@ class TestStraylight(unittest.TestCase): ...@@ -192,7 +192,7 @@ class TestStraylight(unittest.TestCase):
plt.figure() plt.figure()
plt.plot(spec['WAVELENGTH'], spec['FLUX'], 'r') plt.plot(spec['WAVELENGTH'], spec['FLUX'], 'r')
plt.xlabel('WAVELENGTH') plt.xlabel('WAVELENGTH')
plt.ylabel('F$\lambda$(erg/s/cm2/A/arcsec2)') plt.ylabel(r'F$\lambda$(erg/s/cm2/A/arcsec2)')
plt.xlim(2000, 10000) plt.xlim(2000, 10000)
plt.show() plt.show()
median = np.median(sl_e_pix[0:tnum]) median = np.median(sl_e_pix[0:tnum])
......
...@@ -10,11 +10,11 @@ import yaml ...@@ -10,11 +10,11 @@ import yaml
from observation_sim.instruments import Chip, Filter, FilterParam, FocalPlane from observation_sim.instruments import Chip, Filter, FilterParam, FocalPlane
### test FUNCTION --- START ### # test FUNCTION --- START ###
def get_base_img(img, chip, read_noise, readout_time, dark_noise, exptime=150., InputDark=None): def get_base_img(img, chip, read_noise, readout_time, dark_noise, exptime=150., InputDark=None):
if InputDark == None: if InputDark is None:
# base_level = read_noise**2 + dark_noise*(exptime+0.5*readout_time) # base_level = read_noise**2 + dark_noise*(exptime+0.5*readout_time)
# base_level = dark_noise*(exptime+0.5*readout_time) # base_level = dark_noise*(exptime+0.5*readout_time)
base_level = dark_noise*(exptime) base_level = dark_noise*(exptime)
...@@ -31,7 +31,7 @@ def get_base_img(img, chip, read_noise, readout_time, dark_noise, exptime=150., ...@@ -31,7 +31,7 @@ def get_base_img(img, chip, read_noise, readout_time, dark_noise, exptime=150.,
base_img2[ny:, :] = arr[::-1, :] base_img2[ny:, :] = arr[::-1, :]
base_img2[:, :] = base_img2[:, :]*(readout_time/ny)*dark_noise base_img2[:, :] = base_img2[:, :]*(readout_time/ny)*dark_noise
return base_img1+base_img2 return base_img1+base_img2
### test FUNCTION --- END ### # test FUNCTION --- END ###
def defineCCD(iccd, config_file): def defineCCD(iccd, config_file):
......
...@@ -26,7 +26,7 @@ def focalPlaneInf(ra_target, dec_target, ra_point, dec_point, image_rot=-113.433 ...@@ -26,7 +26,7 @@ def focalPlaneInf(ra_target, dec_target, ra_point, dec_point, image_rot=-113.433
float, in unit of degrees; float, in unit of degrees;
image_rot : orientation of the camera with respect the sky; image_rot : orientation of the camera with respect the sky;
float, in unit of degrees; float, in unit of degrees;
NOTE: image_rot=-113.4333 is the default value NOTE: image_rot=-113.4333 is the default value
in current CSST image simulation; in current CSST image simulation;
figout : location of the target object in the focal plane; figout : location of the target object in the focal plane;
str str
...@@ -36,7 +36,7 @@ def focalPlaneInf(ra_target, dec_target, ra_point, dec_point, image_rot=-113.433 ...@@ -36,7 +36,7 @@ def focalPlaneInf(ra_target, dec_target, ra_point, dec_point, image_rot=-113.433
the pointing center (ra_point dec_point) of the telescope the pointing center (ra_point dec_point) of the telescope
1) open a terminal 1) open a terminal
2) type >> python TargetLocationCheck.py ra_target dec_target ra_point dec_point 2) type >> python TargetLocationCheck.py ra_target dec_target ra_point dec_point
or type >> python TargetLocationCheck.py ra_target dec_target ra_point dec_point -image_rot=floatNum or type >> python TargetLocationCheck.py ra_target dec_target ra_point dec_point -image_rot=floatNum
or type >> python TargetLocationCheck.py ra_target dec_target ra_point dec_point -image_rot=floatNum -figout=FigureName or type >> python TargetLocationCheck.py ra_target dec_target ra_point dec_point -image_rot=floatNum -figout=FigureName
""" """
print("^_^ Input target coordinate: [Ra, Dec] = [%10.6f, %10.6f]" % ( print("^_^ Input target coordinate: [Ra, Dec] = [%10.6f, %10.6f]" % (
...@@ -105,7 +105,7 @@ def ccdParam(): ...@@ -105,7 +105,7 @@ def ccdParam():
def getTanWCS(ra, dec, img_rot, pix_scale=0.074): def getTanWCS(ra, dec, img_rot, pix_scale=0.074):
""" """
Get the WCS of the image mosaic using Gnomonic/TAN projection Get the WCS of the image mosaic using Gnomonic/TAN projection
Parameter: Parameter:
ra, dec: float ra, dec: float
...@@ -221,8 +221,8 @@ def ccdLayout(xpixTar, ypixTar, figout="ccdLayout.pdf"): ...@@ -221,8 +221,8 @@ def ccdLayout(xpixTar, ypixTar, figout="ccdLayout.pdf"):
ax.text(ix0+500, iy0+1500, "%s#%s" % ax.text(ix0+500, iy0+1500, "%s#%s" %
(fType, ischip), fontsize=12, color="grey") (fType, ischip), fontsize=12, color="grey")
ax.plot(xpixTar, ypixTar, "r*", ms=12) ax.plot(xpixTar, ypixTar, "r*", ms=12)
ax.set_xlabel("$X\,[\mathrm{pixels}]$", fontsize=20) ax.set_xlabel(r"$X\,[\mathrm{pixels}]$", fontsize=20)
ax.set_ylabel("$Y\,[\mathrm{pixels}]$", fontsize=20) ax.set_ylabel(r"$Y\,[\mathrm{pixels}]$", fontsize=20)
ax.invert_yaxis() ax.invert_yaxis()
ax.axis('off') ax.axis('off')
plt.savefig(figout) plt.savefig(figout)
......
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