Commit 874cfe5f authored by BO ZHANG's avatar BO ZHANG 🏀
Browse files

added some docstring

parent 2c14e4c3
...@@ -20,9 +20,21 @@ class CsstData: ...@@ -20,9 +20,21 @@ class CsstData:
_l2data = OrderedDict() # _l2data = OrderedDict() #
_auxdata = OrderedDict() _auxdata = OrderedDict()
def __init__(self, primaryHDU, imgHDU, instrument=None, detector=None): def __init__(self, priHDU, imgHDU, instrument=None, detector=None):
# print('create CsstData') """
self._primary_hdu = primaryHDU
Parameters
----------
priHDU:
primary HDU
imgHDU:
image HDU
instrument:
instrument
detector:
detector
"""
self._primary_hdu = priHDU
self._l0data = imgHDU self._l0data = imgHDU
self.instrument = instrument self.instrument = instrument
self.detector = detector self.detector = detector
......
...@@ -17,7 +17,7 @@ class CsstProcStatus(Enum): ...@@ -17,7 +17,7 @@ class CsstProcStatus(Enum):
class CsstProcessor(metaclass=ABCMeta): class CsstProcessor(metaclass=ABCMeta):
def __init__(self, **kwargs): def __init__(self, **kwargs):
pass self._status = CsstProcStatus()
@abstractmethod @abstractmethod
def prepare(self, **kwargs): def prepare(self, **kwargs):
......
...@@ -35,34 +35,44 @@ class CsstMscData(CsstData): ...@@ -35,34 +35,44 @@ class CsstMscData(CsstData):
self._auxdata['flat'] = flat self._auxdata['flat'] = flat
def set_bias(self, biasimg): def set_bias(self, biasimg):
""" set bias """
self._auxdata['bias'] = biasimg self._auxdata['bias'] = biasimg
def set_dark(self, darkimg): def set_dark(self, darkimg):
""" set dark """
self._auxdata['dark'] = darkimg self._auxdata['dark'] = darkimg
def set_badpixel(self, badpixelimg): def set_badpixel(self, badpixelimg):
""" set badpixel """
self._auxdata['badpixel'] = badpixelimg self._auxdata['badpixel'] = badpixelimg
def get_flat(self): def get_flat(self):
""" get flat """
return self._auxdata['flat'] return self._auxdata['flat']
def get_bias(self): def get_bias(self):
""" get bias """
return self._auxdata['bias'] return self._auxdata['bias']
def get_dark(self): def get_dark(self):
""" get dark """
return self._auxdata['dark'] return self._auxdata['dark']
def get_badpixel(self): def get_badpixel(self):
""" get badpixel """
return self._auxdata['badpixel'] return self._auxdata['badpixel']
def init_l0data(self): def init_l0data(self):
""" initialize L0 data """
pass pass
def set_l1keyword(self, key, value, comment=''): def set_l1keyword(self, key, value, comment=''):
""" set L1 keyword """
print('check out whether ' + key + " is a valid key and " + value + " is valid value") print('check out whether ' + key + " is a valid key and " + value + " is valid value")
self._l1hdr_global.set(key, value, comment) self._l1hdr_global.set(key, value, comment)
def set_l1data(self, imgtype, img): def set_l1data(self, imgtype, img):
""" set L1 data """
try: try:
if self._l1img_types[imgtype]: if self._l1img_types[imgtype]:
self._l1data[imgtype].data = img.copy() self._l1data[imgtype].data = img.copy()
...@@ -71,6 +81,7 @@ class CsstMscData(CsstData): ...@@ -71,6 +81,7 @@ class CsstMscData(CsstData):
print('save image data to l1data') print('save image data to l1data')
def save_l1data(self, imgtype, filename): def save_l1data(self, imgtype, filename):
""" save L1 data """
print('check ' + imgtype + ' is validate') print('check ' + imgtype + ' is validate')
try: try:
if self._l1img_types[imgtype]: if self._l1img_types[imgtype]:
...@@ -89,7 +100,7 @@ class CsstMscImgData(CsstMscData): ...@@ -89,7 +100,7 @@ class CsstMscImgData(CsstMscData):
@staticmethod @staticmethod
def read(fp): def read(fp):
""" read from fits file """ read data from fits file
Parameters Parameters
---------- ----------
...@@ -100,6 +111,16 @@ class CsstMscImgData(CsstMscData): ...@@ -100,6 +111,16 @@ class CsstMscImgData(CsstMscData):
------- -------
CsstMscImgData CsstMscImgData
Example
-------
>>> fp = "MSC_MS_210527171000_100000279_16_raw.fits"
>>> from csst.msc import CsstMscImgData
>>> data = CsstMscImgData.read(fp)
>>> # print some info
>>> print("data: ", data)
>>> print("instrument: ", data.get_l0keyword("pri", "INSTRUME"))
>>> print("object: ", data.get_l0keyword("pri", "OBJECT"))
""" """
try: try:
......
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