LSSTApplications  17.0+10,17.0+51,17.0+88,18.0.0+10,18.0.0+15,18.0.0+34,18.0.0+4,18.0.0+6,18.0.0-2-ge43143a+6,18.1.0-1-g0001055+2,18.1.0-1-g0896a44+10,18.1.0-1-g1349e88+9,18.1.0-1-g2505f39+7,18.1.0-1-g380d4d4+9,18.1.0-1-g5e4b7ea+2,18.1.0-1-g7e8fceb,18.1.0-1-g85f8cd4+7,18.1.0-1-g9a6769a+3,18.1.0-1-ga1a4c1a+6,18.1.0-1-gc037db8+2,18.1.0-1-gd55f500+3,18.1.0-1-ge10677a+7,18.1.0-10-g73b8679e+12,18.1.0-12-gf30922b,18.1.0-13-g451e75588,18.1.0-13-gbfe7f7f,18.1.0-2-g31c43f9+7,18.1.0-2-g9c63283+9,18.1.0-2-gdf0b915+9,18.1.0-2-gf03bb23+2,18.1.0-3-g52aa583+3,18.1.0-3-g8f4a2b1+1,18.1.0-3-g9cb968e+8,18.1.0-4-g7bbbad0,18.1.0-5-g510c42a+8,18.1.0-5-ga46117f,18.1.0-5-gaeab27e+9,18.1.0-6-gdda7f3e+11,18.1.0-8-g4084bf03+1,w.2019.34
LSSTDataManagementBasePackage
Functions
lsst.meas.algorithms.testUtils Namespace Reference

Functions

def plantSources (bbox, kwid, sky, coordList, addPoissonNoise=True)
 
def makeRandomTransmissionCurve (rng, minWavelength=4000.0, maxWavelength=7000.0, nWavelengths=200, maxRadius=80.0, nRadii=30, perturb=0.05)
 

Function Documentation

◆ makeRandomTransmissionCurve()

def lsst.meas.algorithms.testUtils.makeRandomTransmissionCurve (   rng,
  minWavelength = 4000.0,
  maxWavelength = 7000.0,
  nWavelengths = 200,
  maxRadius = 80.0,
  nRadii = 30,
  perturb = 0.05 
)
Create a random TransmissionCurve with nontrivial spatial and
wavelength variation.

Parameters
----------
rng : numpy.random.RandomState
    Random number generator.
minWavelength : float
    Average minimum wavelength for generated TransmissionCurves (will be
    randomly perturbed).
maxWavelength : float
    Average maximum wavelength for generated TransmissionCurves (will be
    randomly perturbed).
nWavelengths : int
    Number of samples in the wavelength dimension.
maxRadius : float
    Average maximum radius for spatial variation (will be perturbed).
nRadii : int
    Number of samples in the radial dimension.
perturb: float
    Fraction by which wavelength and radius bounds should be randomly
    perturbed.

Definition at line 89 of file testUtils.py.

89  maxRadius=80.0, nRadii=30, perturb=0.05):
90  """Create a random TransmissionCurve with nontrivial spatial and
91  wavelength variation.
92 
93  Parameters
94  ----------
95  rng : numpy.random.RandomState
96  Random number generator.
97  minWavelength : float
98  Average minimum wavelength for generated TransmissionCurves (will be
99  randomly perturbed).
100  maxWavelength : float
101  Average maximum wavelength for generated TransmissionCurves (will be
102  randomly perturbed).
103  nWavelengths : int
104  Number of samples in the wavelength dimension.
105  maxRadius : float
106  Average maximum radius for spatial variation (will be perturbed).
107  nRadii : int
108  Number of samples in the radial dimension.
109  perturb: float
110  Fraction by which wavelength and radius bounds should be randomly
111  perturbed.
112  """
113  dWavelength = maxWavelength - minWavelength
114 
115  def perturbed(x, s=perturb*dWavelength):
116  return x + 2.0*s*(rng.rand() - 0.5)
117 
118  wavelengths = np.linspace(perturbed(minWavelength), perturbed(maxWavelength), nWavelengths)
119  radii = np.linspace(0.0, perturbed(maxRadius, perturb*maxRadius), nRadii)
120  throughput = np.zeros(wavelengths.shape + radii.shape, dtype=float)
121  # throughput will be a rectangle in wavelength, shifting to higher wavelengths and shrinking
122  # in height with radius, going to zero at all bounds.
123  peak0 = perturbed(0.9, 0.05)
124  start0 = perturbed(minWavelength + 0.25*dWavelength)
125  stop0 = perturbed(minWavelength + 0.75*dWavelength)
126  for i, r in enumerate(radii):
127  mask = np.logical_and(wavelengths >= start0 + r, wavelengths <= stop0 + r)
128  throughput[mask, i] = peak0*(1.0 - r/1000.0)
129  return afwImage.TransmissionCurve.makeRadial(throughput, wavelengths, radii)
130 

◆ plantSources()

def lsst.meas.algorithms.testUtils.plantSources (   bbox,
  kwid,
  sky,
  coordList,
  addPoissonNoise = True 
)
Make an exposure with stars (modelled as Gaussians)

@param bbox: parent bbox of exposure
@param kwid: kernel width (and height; kernel is square)
@param sky: amount of sky background (counts)
@param coordList: a list of [x, y, counts, sigma], where:
    * x,y are relative to exposure origin
    * counts is the integrated counts for the star
    * sigma is the Gaussian sigma in pixels
@param addPoissonNoise: add Poisson noise to the exposure?

Definition at line 31 of file testUtils.py.

31 def plantSources(bbox, kwid, sky, coordList, addPoissonNoise=True):
32  """Make an exposure with stars (modelled as Gaussians)
33 
34  @param bbox: parent bbox of exposure
35  @param kwid: kernel width (and height; kernel is square)
36  @param sky: amount of sky background (counts)
37  @param coordList: a list of [x, y, counts, sigma], where:
38  * x,y are relative to exposure origin
39  * counts is the integrated counts for the star
40  * sigma is the Gaussian sigma in pixels
41  @param addPoissonNoise: add Poisson noise to the exposure?
42  """
43  # make an image with sources
44  img = afwImage.ImageD(bbox)
45  meanSigma = 0.0
46  for coord in coordList:
47  x, y, counts, sigma = coord
48  meanSigma += sigma
49 
50  # make a single gaussian psf
51  psf = SingleGaussianPsf(kwid, kwid, sigma)
52 
53  # make an image of it and scale to the desired number of counts
54  thisPsfImg = psf.computeImage(lsst.geom.PointD(x, y))
55  thisPsfImg *= counts
56 
57  # bbox a window in our image and add the fake star image
58  psfBox = thisPsfImg.getBBox()
59  psfBox.clip(bbox)
60  if psfBox != thisPsfImg.getBBox():
61  thisPsfImg = thisPsfImg[psfBox, afwImage.PARENT]
62  imgSeg = img[psfBox, afwImage.PARENT]
63  imgSeg += thisPsfImg
64  meanSigma /= len(coordList)
65 
66  img += sky
67 
68  # add Poisson noise
69  if (addPoissonNoise):
70  np.random.seed(seed=1) # make results reproducible
71  imgArr = img.getArray()
72  imgArr[:] = np.random.poisson(imgArr)
73 
74  # bundle into a maskedimage and an exposure
75  mask = afwImage.Mask(bbox)
76  var = img.convertFloat()
77  img -= sky
78  mimg = afwImage.MaskedImageF(img.convertFloat(), mask, var)
79  exposure = afwImage.makeExposure(mimg)
80 
81  # insert an approximate psf
82  psf = SingleGaussianPsf(kwid, kwid, meanSigma)
83  exposure.setPsf(psf)
84 
85  return exposure
86 
87 
def plantSources(bbox, kwid, sky, coordList, addPoissonNoise=True)
Definition: testUtils.py:31
std::shared_ptr< Exposure< ImagePixelT, MaskPixelT, VariancePixelT > > makeExposure(MaskedImage< ImagePixelT, MaskPixelT, VariancePixelT > &mimage, std::shared_ptr< geom::SkyWcs const > wcs=std::shared_ptr< geom::SkyWcs const >())
A function to return an Exposure of the correct type (cf.
Definition: Exposure.h:453
Represent a 2-dimensional array of bitmask pixels.
Definition: Mask.h:77