fakecbed 0.2.1__py3-none-any.whl → 0.3.0__py3-none-any.whl

fakecbed/discretized.py

@@ -412,6 +412,41 @@ def _de_pre_serialize_apply_shot_noise(serializable_rep):
 
 
 
+def _check_and_convert_rng_seed(params):
+    obj_name = "rng_seed"
+    obj = params[obj_name]
+
+    current_func_name = "_check_and_convert_rng_seed"
+    
+    if obj is not None:
+        kwargs = {"obj": obj, "obj_name": obj_name}
+        try:
+            rng_seed = czekitout.convert.to_nonnegative_int(**kwargs)
+        except:
+            err_msg = globals()[current_func_name+"_err_msg_1"]
+            raise TypeError(err_msg)
+    else:
+        rng_seed = obj
+
+    return rng_seed
+
+
+
+def _pre_serialize_rng_seed(rng_seed):
+    obj_to_pre_serialize = rng_seed
+    serializable_rep = obj_to_pre_serialize
+    
+    return serializable_rep
+
+
+
+def _de_pre_serialize_rng_seed(serializable_rep):
+    rng_seed = serializable_rep
+
+    return rng_seed
+
+
+
 def _check_and_convert_detector_partition_width_in_pixels(params):
     obj_name = "detector_partition_width_in_pixels"
     func_alias = czekitout.convert.to_nonnegative_int
@@ -562,6 +597,8 @@ _default_num_pixels_across_pattern = \
     512
 _default_apply_shot_noise = \
     False
+_default_rng_seed = \
+    None
 _default_detector_partition_width_in_pixels = \
     0
 _default_cold_pixels = \
@@ -1035,6 +1072,9 @@ class CBEDPattern(fancytypes.PreSerializableAndUpdatable):
         If ``apply_shot_noise`` is set to ``True``, then shot noise is applied
         to the image of the fake CBED pattern. Otherwise, no shot noise is
         applied.
+    rng_seed : `int` | `None`, optional
+        ``rng_seed`` specifies the seed used in the random number generator used
+        to apply shot noise.
     detector_partition_width_in_pixels : `int`, optional
         The detector partition width in units of pixels,
         :math:`N_{\text{DPW}}`. Must be nonnegative.
@@ -1057,6 +1097,7 @@ class CBEDPattern(fancytypes.PreSerializableAndUpdatable):
                         "num_pixels_across_pattern",
                         "distortion_model",
                         "apply_shot_noise",
+                        "rng_seed",
                         "detector_partition_width_in_pixels",
                         "cold_pixels",
                         "mask_frame")
@@ -1091,6 +1132,8 @@ class CBEDPattern(fancytypes.PreSerializableAndUpdatable):
                  _default_distortion_model,
                  apply_shot_noise=\
                  _default_apply_shot_noise,
+                 rng_seed=\
+                 _default_rng_seed,
                  detector_partition_width_in_pixels=\
                  _default_detector_partition_width_in_pixels,
                  cold_pixels=\
@@ -1657,7 +1700,11 @@ class CBEDPattern(fancytypes.PreSerializableAndUpdatable):
         image = method_alias(u_x, u_y)
 
         apply_shot_noise = self._apply_shot_noise
-        image = torch.poisson(image) if (apply_shot_noise == True) else image
+        if apply_shot_noise == True:
+            np_rng = np.random.default_rng(self._rng_seed)
+            torch_rng_seed = np_rng.integers(low=0, high=2**32-1).item()
+            torch_rng = torch.Generator().manual_seed(torch_rng_seed)
+            image = torch.poisson(image, torch_rng)
 
         image = self._apply_detector_partition_inpainting(input_image=image)
             
@@ -2522,6 +2569,10 @@ _check_and_convert_distortion_model_err_msg_1 = \
      "grid, specified by the object ``distortion_model``,  must be divisible "
      "by the object ``num_pixels_across_pattern``.")
 
+_check_and_convert_rng_seed_err_msg_1 = \
+    ("The object ``rng_seed`` must be either a nonnegative integer or of the "
+     "type `NoneType`.")
+
 _check_and_convert_cold_pixels_err_msg_1 = \
     ("The object ``cold_pixels`` must be a sequence of integer pairs, where "
      "each integer pair specifies valid pixel coordinates (i.e. row and column "

fakecbed/version.py

@@ -12,5 +12,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.2.1'
-__version_tuple__ = version_tuple = (0, 2, 1)
+__version__ = version = '0.3.0'
+__version_tuple__ = version_tuple = (0, 3, 0)

{fakecbed-0.2.1.dist-info → fakecbed-0.3.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: fakecbed
-Version: 0.2.1
+Version: 0.3.0
 Summary: A Python library for generating quickly images that imitate convergent beam electron diffraction patterns.
 Author-email: Matthew Fitzpatrick <matthew.rc.fitzpatrick@gmail.com>
 Project-URL: Homepage, https://mrfitzpa.github.io/fakecbed

fakecbed-0.3.0.dist-info/RECORD

@@ -0,0 +1,10 @@
+fakecbed/__init__.py,sha256=iP73WNV_HZgxiVUVdesbJqlrhDGESooh1hlBs8kox9M,1342
+fakecbed/discretized.py,sha256=vDV1kU2Q-JYDMp_WEAYCCLGXLN7urBCTfej-ggYC19M,99328
+fakecbed/shapes.py,sha256=TiqJBZI4JZib0Q6zfO0e-dh7ACjYhvAYUvEtIedoo74,116471
+fakecbed/tds.py,sha256=mk5359ErIhdpCoTdHq9r1-foKb8t3qCOHqNoabx3bQw,8066
+fakecbed/version.py,sha256=Jk2iAU7m-7Vx9XV1TtdD9ZoJraIncDq_4_Wd-qtUotg,411
+fakecbed-0.3.0.dist-info/LICENSE,sha256=N0P3pKtRMvfb64jmgJdlerg3TnS0a2QC776AzPsbZIg,35128
+fakecbed-0.3.0.dist-info/METADATA,sha256=Yh3j9fCD-N_AXT5nUkdRHDdOGsQquCLbxJbW9fLsLng,3006
+fakecbed-0.3.0.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+fakecbed-0.3.0.dist-info/top_level.txt,sha256=l0jL8P0CjRiCSznYLnq7riaAnOmYaV3zjjjIJx9vjSQ,9
+fakecbed-0.3.0.dist-info/RECORD,,

fakecbed-0.2.1.dist-info/RECORD

@@ -1,10 +0,0 @@
-fakecbed/__init__.py,sha256=iP73WNV_HZgxiVUVdesbJqlrhDGESooh1hlBs8kox9M,1342
-fakecbed/discretized.py,sha256=bM_K0VJtSTxlIUdbsbPl2eiC6jJjoLn2y2iYy1XvBZE,97939
-fakecbed/shapes.py,sha256=TiqJBZI4JZib0Q6zfO0e-dh7ACjYhvAYUvEtIedoo74,116471
-fakecbed/tds.py,sha256=mk5359ErIhdpCoTdHq9r1-foKb8t3qCOHqNoabx3bQw,8066
-fakecbed/version.py,sha256=MxUhzLJIZQfEpDTTcKSxciTGrMLd5v2VmMlHa2HGeo0,411
-fakecbed-0.2.1.dist-info/LICENSE,sha256=N0P3pKtRMvfb64jmgJdlerg3TnS0a2QC776AzPsbZIg,35128
-fakecbed-0.2.1.dist-info/METADATA,sha256=Oee7NOEdAU8tX0G00o0EiVf-ldtnubLkivCBYCq_2RY,3006
-fakecbed-0.2.1.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-fakecbed-0.2.1.dist-info/top_level.txt,sha256=l0jL8P0CjRiCSznYLnq7riaAnOmYaV3zjjjIJx9vjSQ,9
-fakecbed-0.2.1.dist-info/RECORD,,