Source code for psfsim.postprocess

import numpy as np



[docs]
class ImageCube:
    """Class to make a 3D cube."""

    def __init__(self, image2D, tExp=120):

[docs]
        self.image2D = image2D


[docs]
        self.imageCube = self.makeImageCube


[docs]
        self.tFrame = 3.08


[docs]
        self.tExp = tExp


[docs]
        self.nFrames = self.tExp // self.tFrame



[docs]
    def makeImageCube(self):
        """Makes a cube."""
        rng = np.random.default_rng()
        pArray = [1 / (self.nFrames - 1)] * (self.nFrames - 1)
        rvs = rng.multinomial(self.image2D, pArray, size=(self.nFrames - 1,) + self.image2D.shape)
        return np.cumsum(rvs)