Skip to content
GitLab
Projects
Groups
Snippets
/
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
Bo Zhang
csst
Commits
874cfe5f
Commit
874cfe5f
authored
Mar 03, 2022
by
BO ZHANG
🏀
Browse files
added some docstring
parent
2c14e4c3
Changes
3
Hide whitespace changes
Inline
Side-by-side
csst/core/data.py
View file @
874cfe5f
...
@@ -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
...
...
csst/core/processor.py
View file @
874cfe5f
...
@@ -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
):
...
...
csst/msc/data.py
View file @
874cfe5f
...
@@ -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
:
...
...
Write
Preview
Supports
Markdown
0%
Try again
or
attach a new file
.
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment