Added further functionality to CalibrationAssembler

resolve/re/sugar.py

 import jax.numpy as jnp
 import nifty8.re as jft
+import matplotlib.pyplot as plt
 
 from .calibration import CalibrationDistribution
+from .likelihood_models import ModelCalibrationLikelihoodFixedCovariance
 
 from ..data.observation import Observation
 
@@ -168,7 +170,19 @@ class CalibrationAssembler:
                 scaling_op = None
 
             return cls(observation,phase_field,logflux_field,dt,init_flux_field,scaling_op,log_inv_cov,lh_label)
-        
+
+    def prior_realization(self,rng_key):
+        data_model = ModelCalibrationLikelihoodFixedCovariance(self.scaled_cop,self.model_vis,jnp.asarray(self.obs.mask.val))
+        data_model_realization = data_model(data_model.init(rng_key))
+
+        plt.scatter(self.obs.vis.val.imag,self.obs.vis.val.real,label="Data")
+        plt.scatter(data_model_realization.imag,data_model_realization.real, alpha=0.01, label="Prior sample")
+        plt.legend()
+        plt.show()
+
+    def get_lh_domain(self):
+        dom = self.scaled_cop.domain if self.log_inv_cov is None else self.scaled_cop.domain | self.log_inv_cov.domain
+        return dom
 
     def __repr__(self):
         return(