fit PEP8 bugs

.idea/.gitignore

+# 默认忽略的文件
+/shelf/
+/workspace.xml
+# 基于编辑器的 HTTP 客户端请求
+/httpRequests/
+# Datasource local storage ignored files
+/dataSources/
+/dataSources.local.xml

.idea/csst_ifs_gehong.iml

+<?xml version="1.0" encoding="UTF-8"?>
+<module type="PYTHON_MODULE" version="4">
+  <component name="NewModuleRootManager">
+    <content url="file://$MODULE_DIR$" />
+    <orderEntry type="inheritedJdk" />
+    <orderEntry type="sourceFolder" forTests="false" />
+  </component>
+</module>
\ No newline at end of file

.idea/inspectionProfiles/profiles_settings.xml

+<component name="InspectionProjectProfileManager">
+  <settings>
+    <option name="USE_PROJECT_PROFILE" value="false" />
+    <version value="1.0" />
+  </settings>
+</component>
\ No newline at end of file

.idea/modules.xml

+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="ProjectModuleManager">
+    <modules>
+      <module fileurl="file://$PROJECT_DIR$/.idea/csst_ifs_gehong.iml" filepath="$PROJECT_DIR$/.idea/csst_ifs_gehong.iml" />
+    </modules>
+  </component>
+</project>
\ No newline at end of file

.idea/vcs.xml

+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="VcsDirectoryMappings">
+    <mapping directory="" vcs="Git" />
+  </component>
+</project>
\ No newline at end of file

csst_ifs_gehong/config.py

 import numpy as np
 
+
 class config():
     """
     The configuration of spectral modeling. The default value is used for ETC calculation.
@@ -21,9 +22,9 @@ class config():
     dpix : float, optional
         Pixel size in the spatial direction, by default 0.2arcsec
     """
-    def __init__(self, wave_min = 3500.0, wave_max = 10000.0, 
-                 dlam = 2.0, inst_fwhm = 0.1,
-                 nx = 30, ny = 30, dpix = 0.2):
+    def __init__(self, wave_min=3500.0, wave_max=10000.0, 
+                 dlam=2.0, inst_fwhm=0.1,
+                 nx=30, ny=30, dpix=0.2):
         self.dlam = dlam
         self.wave = np.arange(wave_min, wave_max, dlam)
         self.wave_min = wave_min

csst_ifs_gehong/cube3d.py

-import astropy.units as u
 import numpy as np
-from scipy.interpolate import interp1d
-from .spec1d import *
+from .spec1d import StellarContinuum, HII_Region
 import astropy.wcs
+from astropy.io import fits
+
 
 class Cube3D():
     """
     Class of 3-dimentional spectral cube
     """
-    def __init__(self, config, stellar_map = None, gas_map = None):
+    def __init__(self, config, stellar_map=None, gas_map=None):
         
-        self.config= config
+        self.config = config
         
         self.nx = config.nx
         self.ny = config.ny
@@ -23,30 +23,30 @@ class Cube3D():
         self.wave0 = np.min(self.wave)
         self.inst_fwhm = config.inst_fwhm
         
-        self.flux  = np.zeros((self.nx,self.ny,self.nz))
+        self.flux = np.zeros((self.nx, self.ny, self.nz))
         
         self.stellar_map = stellar_map
         self.gas_map = gas_map
         
-    def make_cube(self, stellar_tem = None, hii_tem = None):
+    def make_cube(self, stellar_tem=None, hii_tem=None):
         
         for i in range(self.nx):
             for j in range(self.ny):
                 if self.stellar_map is not None:
-                    ss = StellarContinuum(self.config, stellar_tem, mag = self.stellar_map.mag[i,j], 
-                                          age = self.stellar_map.age[i,j], feh = self.stellar_map.feh[i,j], 
-                                          vel = self.stellar_map.vel[i,j], vdisp = self.stellar_map.vdisp[i,j], 
-                                          ebv = self.stellar_map.ebv[i,j])
+                    ss = StellarContinuum(self.config, stellar_tem, mag=self.stellar_map.mag[i, j], 
+                                          age=self.stellar_map.age[i, j], feh=self.stellar_map.feh[i, j], 
+                                          vel=self.stellar_map.vel[i, j], vdisp=self.stellar_map.vdisp[i, j], 
+                                          ebv=self.stellar_map.ebv[i, j])
                 if self.gas_map is not None:
-                    gg = HII_Region(self.config, hii_tem, halpha = self.gas_map.halpha[i,j], 
-                                    logz = self.gas_map.zh[i,j], vel = self.gas_map.vel[i,j], 
-                                    vdisp = self.gas_map.vdisp[i,j], ebv = self.gas_map.ebv[i,j])
-                    self.flux[i,j,:] = ss.flux + gg.flux
+                    gg = HII_Region(self.config, hii_tem, halpha=self.gas_map.halpha[i, j], 
+                                    logz=self.gas_map.zh[i, j], vel=self.gas_map.vel[i, j], 
+                                    vdisp=self.gas_map.vdisp[i, j], ebv=self.gas_map.ebv[i, j])
+                    self.flux[i, j, :] = ss.flux + gg.flux
                 else:
-                    self.flux[i,j,:] = ss.flux
+                    self.flux[i, j, :] = ss.flux
                     
     def wcs_info(self):
-        wcs = fits.Header()
+        # wcs = fits.Header()
         wcs_dict = {'CTYPE1': 'WAVE    ', 
                     'CUNIT1': 'Angstrom', 
                     'CDELT1': self.config.dlam, 
@@ -62,16 +62,16 @@ class Cube3D():
                     'CDELT3': self.dpix / 3600., 
                     'CRPIX3': np.round(self.nx / 2.), 
                     'CRVAL3': 1, 
-                    'BUNIT' : '10**(-17)*erg/s/cm**2/Angstrom'}
+                    'BUNIT': '10**(-17)*erg/s/cm**2/Angstrom'}
         input_wcs = astropy.wcs.WCS(wcs_dict)
         self.wcs_header = input_wcs.to_header()
         
-    def insert_spec(self, spec, dx = 0, dy = 0):
+    def insert_spec(self, spec, dx=0, dy=0):
         x0 = np.int(np.round(self.config.nx / 2.))
         y0 = np.int(np.round(self.config.ny / 2.))
         self.flux[x0 + dx, y0 + dy, :] = self.flux[x0 + dx, y0 + dy, :] + spec.flux
         
-    def savefits(self, filename, path = './'): 
+    def savefits(self, filename, path='./'): 
         
         hdr = fits.Header()
         
@@ -83,12 +83,12 @@ class Cube3D():
         hdr['RA'] = 0.0
         hdr['DEC'] = 0.0
         
-        hdu0 = fits.PrimaryHDU(header = hdr)
+        hdu0 = fits.PrimaryHDU(header=hdr)
         
         self.wcs_info()
         hdr = self.wcs_header
-        hdu1 = fits.ImageHDU(self.flux, header = hdr)
+        hdu1 = fits.ImageHDU(self.flux, header=hdr)
         
         # Output
         hdulist = fits.HDUList([hdu0, hdu1])
-        hdulist.writeto(path + filename, overwrite = True)
+        hdulist.writeto(path + filename, overwrite=True)
\ No newline at end of file