Merge pull request #40 from padma18-vb/main

Producing variable snapshots of AGN disk + convolving with microlensing maps

Author: Henry-Best-01

amoeba/Notebooks/AmoebaExamples.ipynb

@@ -99,6 +99,11 @@ def __init__(
             * 2
             / np.size(self.temp_array, 0)
         )
+        print('pixel_size: ',self.pixel_size)
+        print('gravitational_radius given bh mass: ',self.rg)
+        print('maximum radius of accretion disk in r_g: ',self.r_out_in_gravitational_radii)
+        print('np.size(self.temp_array, 0): ',np.size(self.temp_array, 0))
+
         self.corona_height = corona_height
 
     def calculate_surface_intensity_map(
@@ -323,15 +328,14 @@ def generate_snapshots(
         :return: a list of snapshots of the accretion disk at each time step. Note that
             this is an experimental method.
         """
-
         rest_frame_wavelength_in_nm = (
             observer_frame_wavelength_in_nm / (1 + self.redshift_source) / self.g_array
         )
 
         if corona_height is None:
             corona_height = self.corona_height
 
-        radiation_patterns = generate_snapshots_of_radiation_pattern(
+        radiation_patterns, tl_arr = generate_snapshots_of_radiation_pattern(
             rest_frame_wavelength_in_nm,
             time_stamps,
             self.temp_array,
@@ -365,7 +369,7 @@ def generate_snapshots(
 
             radiation_patterns_flux_projections.append(current_projection)
 
-        return radiation_patterns_flux_projections
+        return radiation_patterns_flux_projections, tl_arr
 
     def get_plotting_axes(self):
         """Method to get plotting axes for the accretion disk. Useful for plotting any

amoeba/Notebooks/clean.ipynb

@@ -98,6 +98,7 @@ def __init__(
             * self.total_microlens_einstein_radii
             / self.resolution
         )
+        # print('pixel_size: ',self.pixel_size)
         self.pixel_shift = 0
 
     def convolve_with_flux_projection(
@@ -152,6 +153,8 @@ def pull_light_curve(
         phi_travel_direction=None,
         weight_by_macromag=False,
         return_track_coords=False,
+        return_sub_grids=False,
+        grid_length=10,
         random_seed=None,
     ):
         """Calculate the values of a light curve. By default, the light curve will be
@@ -187,6 +190,8 @@ def pull_light_curve(
             y_start_position=y_start_position,
             phi_travel_direction=phi_travel_direction,
             return_track_coords=return_track_coords,
+            return_sub_grids=return_sub_grids,
+            grid_length=grid_length,
             random_seed=random_seed,
         )
         if weight_by_macromag:
@@ -215,6 +220,7 @@ def calculate_microlensed_transfer_function(
         return_descaled_response_array_and_lags=False,
         return_magnification_map_crop=False,
         random_seed=None,
+        disk_tf = None,
     ):
         """Calculate the transfer function of an accretion disk when the response map is
         microlensed at a particular location.
@@ -282,6 +288,7 @@ def calculate_microlensed_transfer_function(
             return_descaled_response_array_and_lags=return_descaled_response_array_and_lags,
             return_magnification_map_crop=return_magnification_map_crop,
             random_seed=random_seed,
+            disk_tf=disk_tf
         )