From 2468ff7c6dacf991e9373782edfdb8f307a66982 Mon Sep 17 00:00:00 2001
From: xin <zhangxinbjfu@gmail.com>
Date: Tue, 27 Sep 2022 23:43:25 +0800
Subject: [PATCH] fix bug no parameter

---
 ObservationSim/MockObject/MockObject.py | 6 ++++--
 ObservationSim/ObservationSim.py        | 2 +-
 2 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/ObservationSim/MockObject/MockObject.py b/ObservationSim/MockObject/MockObject.py
index 99a72ed..bbb1b6e 100755
--- a/ObservationSim/MockObject/MockObject.py
+++ b/ObservationSim/MockObject/MockObject.py
@@ -171,6 +171,8 @@ class MockObject(object):
         dy = y - y_nominal
         offset = galsim.PositionD(dx, dy)
 
+        real_wcs_local = self.real_wcs.local(real_pos)
+
         for i in range(len(bandpass_list)):
             bandpass = bandpass_list[i]
             try:
@@ -205,7 +207,7 @@ class MockObject(object):
 
 
 
-            stamp = star.drawImage(wcs=self.real_wcs, method='phot', offset=offset, save_photons=True)
+            stamp = star.drawImage(wcs=real_wcs_local, method='phot', offset=offset, save_photons=True)
             xmax = max(xmax, stamp.xmax)
             ymax = max(ymax, stamp.ymax)
             photons = stamp.photons
@@ -214,7 +216,7 @@ class MockObject(object):
             photons_list.append(photons)
 
         stamp = galsim.ImageF(int(xmax * 1.1), int(ymax * 1.1))
-        stamp.wcs = self.real_wcs.local(real_pos)
+        stamp.wcs = self.real_wcs_local
         stamp.setCenter(x_nominal, y_nominal)
         bounds = stamp.bounds & galsim.BoundsD(0, chip.npix_x-1, 0, chip.npix_y-1)
         chip.img.setOrigin(0, 0)
diff --git a/ObservationSim/ObservationSim.py b/ObservationSim/ObservationSim.py
index a661b22..d80b0dc 100755
--- a/ObservationSim/ObservationSim.py
+++ b/ObservationSim/ObservationSim.py
@@ -273,7 +273,7 @@ class Observation(object):
                     col_num=chip.colID,
                     extName='raw')
 
-                pos_img, offset, local_wcs, real_wcs = obj.getPosImg_Offset_WCS(img=chip.img, fdmodel=self.fd_model, chip=chip, verbose=False)
+                pos_img, offset, local_wcs, real_wcs = obj.getPosImg_Offset_WCS(img=chip.img, fdmodel=self.fd_model, chip=chip, verbose=False, img_header=h_ext)
                 if pos_img.x == -1 or pos_img.y == -1:
                     # Exclude object which is outside the chip area (after field distortion)
                     # print("obj missed!!")
-- 
GitLab