sky.py: Add "add_header()" method

Signed-off-by: Aaron LI <aly@aaronly.me>

1 files changed, 9 insertions, 0 deletions

diff --git a/fg21sim/sky.py b/fg21sim/sky.py
index e019b2a..b100a8d 100644
--- a/fg21sim/sky.py
+++ b/fg21sim/sky.py
@@ -186,6 +186,15 @@ class SkyBase:
                        "File creation date")
         return hdr
 
+    def add_header(self, key, value, comment=None):
+        """
+        Add/update a key to the FITS header.
+        """
+        if comment is None:
+            self.header_[key] = value
+        else:
+            self.header_[key] = (value, comment)
+
     def copy(self):
         """
         Return a (deep) copy of this instance.