Base API
Index
ComradeBase.AbstractModelComradeBase.AbstractPolarizedModelComradeBase.DensityAnalyticComradeBase.IntensityMapComradeBase.IntensityMapComradeBase.IsAnalyticComradeBase.MinimalHeaderComradeBase.NoHeaderComradeBase.NotAnalyticComradeBase.RectiGridComradeBase.SerialComradeBase.ThreadsExComradeBase.UnstructuredDomainComradeBase.UnstructuredMapComradeBase._visibilitymapComradeBase._visibilitymap!ComradeBase.allocate_imgmapComradeBase.allocate_vismapComradeBase.amplitudeComradeBase.amplitudemapComradeBase.axisdimsComradeBase.baseimageComradeBase.bispectrumComradeBase.bispectrummapComradeBase.centroidComradeBase.closure_phaseComradeBase.closure_phasemapComradeBase.create_imgmapComradeBase.create_mapComradeBase.create_vismapComradeBase.domainpointsComradeBase.executorComradeBase.fieldofviewComradeBase.fluxComradeBase.headerComradeBase.imagepixelsComradeBase.imanalyticComradeBase.intensity_pointComradeBase.intensitymapComradeBase.intensitymap!ComradeBase.intensitymap_analyticComradeBase.intensitymap_analytic!ComradeBase.intensitymap_numericComradeBase.intensitymap_numeric!ComradeBase.ispolarizedComradeBase.loadComradeBase.logclosure_amplitudeComradeBase.logclosure_amplitudemapComradeBase.named_dimsComradeBase.phasecenterComradeBase.pixelsizesComradeBase.radialextentComradeBase.saveComradeBase.second_momentComradeBase.stokesComradeBase.visanalyticComradeBase.visibilityComradeBase.visibility_pointComradeBase.visibilitymapComradeBase.visibilitymap!ComradeBase.visibilitymap_analyticComradeBase.visibilitymap_analytic!ComradeBase.visibilitymap_numericComradeBase.visibilitymap_numeric!DimensionalData.Dimensions.dims
Model API
visibility(mimg, p)Computes the complex visibility of model m at coordinates p. p corresponds to the coordinates of the model. These need to have the properties U, V and sometimes Ti for time and Fr for frequency.
Notes
If you want to compute the visibilities at a large number of positions consider using the visibilitymap.
Warn
This is only defined for analytic models. If you want to compute the visibility for a single point for a non-analytic model, please use the visibilitymap function and create an UnstructuredDomain with a single point.
visibilitymap(model::AbstractModel, p)Computes the complex visibilities at the locations p.
visibilitymap!(vis::AbstractArray, model::AbstractModel, p)Computes the complex visibilities vis in place at the locations p
intensitymap(model::AbstractModel, p::AbstractSingleDomain)Computes the intensity map of model. For the inplace version see intensitymap!
intensitymap!(buffer::AbstractDimArray, model::AbstractModel)Computes the intensity map of model by modifying the buffer
allocate_vismap(m::AbstractModel, g::AbstractSingleDomain)Allocate the default map specialized by the grid g
allocate_imgmap(m::AbstractModel, g::AbstractSingleDomain)Allocate the default map specialized by the grid g
create_imgmap(array, g::AbstractSingleDomain)Create a map of values specialized by the grid g in the image domain. The default is to call create_map with the same arguments.
create_vismap(array, g::AbstractSingleDomain)Create a map of values specialized by the grid g in the visibility domain. The default is to call create_map with the same arguments.
amplitude(model, p)Computes the visibility amplitude of model m at the coordinate p. The coordinate p is expected to have the properties U, V, and sometimes Ti and Fr.
If you want to compute the amplitudemap at a large number of positions consider using the amplitudemap function.
amplitudemap(m::AbstractModel, p)Computes the visibility amplitudemap of the model m at the coordinates p. The coordinates p are expected to have the properties U, V, and sometimes Ti and Fr.
bispectrum(model, p1, p2, p3)Computes the complex bispectrum of model m at the uv-triangle p1 -> p2 -> p3
If you want to compute the bispectrum over a number of triangles consider using the bispectrummap function.
bispectrummap(m, p1, p2, p3)Computes the closure phases of the model m at the triangles p1, p2, p3, where pi are coordinates.
closure_phase(model, p1, p2, p3, p4)Computes the closure phase of model m at the uv-triangle u1,v1 -> u2,v2 -> u3,v3
If you want to compute closure phases over a number of triangles consider using the closure_phasemap function.
closure_phasemap(m,
p1::AbstractArray
p2::AbstractArray
p3::AbstractArray
)Computes the closure phases of the model m at the triangles p1, p2, p3, where pi are coordinates.
logclosure_amplitude(model, p1, p2, p3, p4)Computes the log-closure amplitude of model m at the uv-quadrangle u1,v1 -> u2,v2 -> u3,v3 -> u4,v4 using the formula
If you want to compute log closure amplitudemap over a number of triangles consider using the logclosure_amplitudemap function.
logclosure_amplitudemap(m::AbstractModel,
p1,
p2,
p3,
p4
)Computes the log closure amplitudemap of the model m at the quadrangles p1, p2, p3, p4.
Model Interface
AbstractModelThe Comrade abstract model type. To instantiate your own model type you should subtybe from this model. Additionally you need to implement the following methods to satify the interface:
Mandatory Methods
visanalytic: defines whether the model visibilities can be computed analytically. If yes then this should returnIsAnalytic()and the user must to definevisibility_point. If not analytic thenvisanalyticshould returnNotAnalytic().imanalytic: defines whether the model intensities can be computed pointwise. If yes then this should returnIsAnalytic()and the user must to defineintensity_point. If not analytic thenimanalyticshould returnNotAnalytic().radialextent: Provides a estimate of the radial extent of the model in the image domain. This is used for estimating the size of the image, and for plotting.flux: Returns the total flux of the model.intensity_point: Defines how to compute model intensities pointwise. Note this is must be defined ifimanalytic(::Type{YourModel})==IsAnalytic().visibility_point: Defines how to compute model visibilties pointwise. Note this is must be defined ifvisanalytic(::Type{YourModel})==IsAnalytic().
Optional Methods:
ispolarized: Specified whether a model is intrinsically polarized (returnsIsPolarized()) or is not (returnsNotPolarized()), by default a model isNotPolarized()visibilitymap_analytic: Vectorized version ofvisibility_pointfor models wherevisanalyticreturnsIsAnalytic()visibilitymap_numeric: Vectorized version ofvisibility_pointfor models wherevisanalyticreturnsNotAnalytic()typically these are numerical FT'sintensitymap_analytic: Computes the entire image for models whereimanalyticreturnsIsAnalytic()intensitymap_numeric: Computes the entire image for models whereimanalyticreturnsNotAnalytic()intensitymap_analytic!: Inplace version ofintensitymapintensitymap_numeric!: Inplace version ofintensitymap
abstract type AbstractPolarizedModel <: ComradeBase.AbstractModelA generic polarized model. To implement the use needs to follow the AbstractModel implementation instructions. In addtion there is an optional method stokes(model, p::Symbol) which extracts the specific stokes parameter of the model. The default that the different stokes parameters are stored as fields of the model. To overwrite this behavior overload the function.
visanalytic(::Type{<:AbstractModel})Determines whether the model is pointwise analytic in Fourier domain, i.e. we can evaluate its fourier transform at an arbritrary point.
If IsAnalytic() then it will try to call visibility_point to calculate the complex visibilities. Otherwise it fallback to using the FFT that works for all models that can compute an image.
imanalytic(::Type{<:AbstractModel})Determines whether the model is pointwise analytic in the image domain, i.e. we can evaluate its intensity at an arbritrary point.
If IsAnalytic() then it will try to call intensity_point to calculate the intensity.
ispolarized(::Type)Trait function that defines whether a model is polarized or not.
radialextent(model::AbstractModel)Provides an estimate of the radial size/extent of the model. This is used internally to estimate image size when plotting and using modelimage
DensityAnalyticInternal type for specifying the nature of the model functions. Whether they can be easily evaluated pointwise analytic. This is an internal type that may change.
struct IsAnalytic <: ComradeBase.DensityAnalyticDefines a trait that a states that a model is analytic. This is usually used with an abstract model where we use it to specify whether a model has a analytic fourier transform and/or image.
struct NotAnalytic <: ComradeBase.DensityAnalyticDefines a trait that a states that a model is analytic. This is usually used with an abstract model where we use it to specify whether a model has does not have a easy analytic fourier transform and/or intensity function.
visibility_point(model::AbstractModel, p)Function that computes the pointwise visibility. This must be implemented in the model interface if visanalytic(::Type{MyModel}) == IsAnalytic()
visibilties_analytic(model, p)Computes the visibilties of a model using using the analytic visibility expression given by visibility_point.
visibilties_analytic!(vis, model)Computes the visibilties of a model in-place, using using the analytic visibility expression given by visibility_point.
visibilties_numeric(model, p)Computes the visibilties of a model using a numerical fourier transform. Note that none of these are implemented in ComradeBase. For implementations please see Comrade.
visibilties_numeric!(vis, model)Computes the visibilties of a model in-place using a numerical fourier transform. Note that none of these are implemented in ComradeBase. For implementations please see Comrade.
intensity_point(model::AbstractModel, p)Function that computes the pointwise intensity if the model has the trait in the image domain IsAnalytic(). Otherwise it will use construct the image in visibility space and invert it.
intensitymap_analytic(m::AbstractModel, p::AbstractSingleDomain)Computes the IntensityMap of a model m using the image dimensions p by broadcasting over the analytic intensity_point method.
intensitymap_analytic!(img::IntensityMap, m::AbstractModel)Updates the img using the model m by broadcasting over the analytic intensity_point method.
intensitymap_numeric(m::AbstractModel, p::AbstractSingleDomain)Computes the IntensityMap of a model m at the image positions p using a numerical method. This has to be specified uniquely for every model m if imanalytic(typeof(m)) === NotAnalytic(). See Comrade.jl for example implementations.
intensitymap_numeric!(img::IntensityMap, m::AbstractModel)Updates the img using the model m using a numerical method. This has to be specified uniquely for every model m if imanalytic(typeof(m)) === NotAnalytic(). See Comrade.jl for example implementations.
Image Domain
imagepixels(fovx, fovy, nx, ny; x0=0, y0=0, executor=Serial(), header=NoHeader())Construct a grid of pixels with a field of view fovx and fovy and nx and ny pixels. This points are the pixel centers and the field of view goes from the edge of the first pixel to the edge of the last pixel. The x0, y0 offsets shift the image origin over by (x0, y0) in the image plane.
Arguments:
fovx::Real: The field of view in the x-directionfovy::Real: The field of view in the y-directionnx::Integer: The number of pixels in the x-directionny::Integer: The number of pixels in the y-direction
Keyword Arguments:
x0::Real=0: The x-offset of the imagey0::Real=0: The y-offset of the imageexecutor=Serial(): The executor to use for the grid, default is serial executionheader=NoHeader(): The header to use for the grid
RectiGrid(dims::NamedTuple{Na}; executor=Serial(), header=ComradeBase.NoHeader())
RectiGrid(dims::NTuple{N, <:DimensionalData.Dimension}; executor=Serial(), header=ComradeBase.NoHeader())Creates a rectilinear grid of pixels with the dimensions dims. The dims can either be a named tuple of dimensions or a tuple of dimensions. The dimensions can be in any order however the standard orders are:
(:X, :Y, :Ti, :Fr)
(:X, :Y, :Fr, :Ti)
(:X, :Y) # spatial only
where X,Y are the RA and DEC spatial dimensions respectively, Ti is the time dimension and Fr is the frequency dimension.
Note that the majority of the time users should just call imagepixels to create a spatial grid.
Optional Arguments
executor: specifies how different models are executed. The default isSerialwhich mean serial CPU computations. For threaded computations useThreadsEx()or loadOhMyThreads.jlto uses their schedulers.header: specified underlying header information for the grid. This is used to store information about the image such as the source, RA and DEC, MJD.
Examples
dims = RectiGrid((X(-5.0:0.1:5.0), Y(-4.0:0.1:4.0), Ti([1.0, 1.5, 1.75]), Fr([230, 345])); executor=ThreadsEx())
dims = RectiGrid((X = -5.0:0.1:5.0, Y = -4.0:0.1:4.0, Ti = [1.0, 1.5, 1.75], Fr = [230, 345]); executor=ThreadsEx()))UnstructuredDomain(dims::AbstractArray; executor=Serial(), header=ComradeBase.NoHeader)Builds an unstructured grid (really a vector of points) from the dimensions dims. The executor is used controls how the grid is computed when calling visibilitymap or intensitymap. The default is Serial which mean regular CPU computations. For threaded execution use ThreadsEx() or load OhMyThreads.jl to uses their schedulers.
Note that unlike RectiGrid which assigns dimensions to the grid points, UnstructuredDomain does not. This is becuase the grid is unstructured the points are a cloud in a space
dims(g::AbstractSingleDomain)Returns a tuple containing the dimensions of g. For a named version see ComradeBase.named_dims
named_dims(g::AbstractSingleDomain)Returns a named tuple containing the dimensions of g. For a unnamed version see dims
axisdims(img::IntensityMap)
axisdims(img::IntensityMap, p::Symbol)Returns the keys of the IntensityMap as the actual internal AbstractRectiGrid object. Optionall the user can ask for a specific dimension with p
domainpoints(g::AbstractSingleDomain)Create a grid iterator that can be used to iterate through different points. All grid methods must implement this method.
fieldofview(img::IntensityMap)
fieldofview(img::IntensityMap)Returns a named tuple with the field of view of the image.
pixelsizes(img::IntensityMap)
pixelsizes(img::AbstractRectiGrid)Returns a named tuple with the spatial pixel sizes of the image.
phasecenter(img::IntensityMap)Computes the phase center of an intensity map. Note this is the pixels that is in the middle of the image.
executor(g::AbstractSingleDomain)Returns the executor used to compute the intensitymap or visibilitymap
Serial()Uses serial execution when computing the intensitymap or visibilitymap
ThreadsEx(;scheduler::Symbol = :dynamic)Uses Julia's Threads @threads macro when computing the intensitymap or visibilitymap. You can choose from Julia's various schedulers by passing the scheduler as a parameter. The default is :dynamic, but it isn't considered part of the stable API and may change at any moment.
header(g::AbstractSingleDomain)Returns the headerinformation of the dimensions g
header(img::IntensityMap)Retrieves the header of an IntensityMap
MinimalHeader{T}A minimal header type for ancillary image information.
Fields
source: Common source namera: Right ascension of the image in degrees (J2000)dec: Declination of the image in degrees (J2000)mjd: Modified Julian Date in daysfrequency: Frequency of the image in Hz
Image Types
struct IntensityMap{T, N, D, G<:(ComradeBase.AbstractRectiGrid{D}), A<:AbstractArray{T, N}, R<:Tuple, Na} <: AbstractDimArray{T, N, D, A<:AbstractArray{T, N}}This type is the basic array type for all images and models that obey the ComradeBase interface. The type is a subtype of DimensionalData.AbstractDimArray however, we make a few changes to support the Comrade API.
The dimensions should be specified by an
AbstractRectiGridinterface. Usually users just need theRectiGridgrid, for rectilinear grids.There are two ways to access the dimensions of the array.
dims(img)will return the usualDimArraydimensions, i.e. aTuple{DimensionalData.Dim, ...}. The other way to access the array dimensions is using thegetproperty, e.g.,img.Xwill return the RA/X grid locations but stripped of the usualDimensionalData.Dimensionmaterial. Thisgetpropertybehavior is NOT CONSIDERED* part of the stable API and may be changed in the future.Metadata is stored in the
AbstractRectiGridtype through theheaderproperty and can be accessed throughmetadataorheader
The most common way to create a IntensityMap is to use the function definitions
julia> g = imagepixels(10.0, 10.0, 128, 128; header=NoHeader())
julia> X = g.X; Y = g.Y
julia> data = rand(128, 128)
julia> img1 = IntensityMap(data, g)
julia> img2 = IntensityMap(data, (;X, Y); header=header(g))
julia> img1 == img2
true
julia> img3 = IntensityMap(data, 10.0, 10.0; header=NoHeader())Broadcasting, map, and reductions should all just obey the DimensionalData interface.
IntensityMap(data::AbstractArray, g::AbstractRectiGrid; refdims=(), name=Symbol(""))Creates a IntensityMap with the pixel fluxes data on the grid g. Optionally, you can specify a set of reference dimensions refdims as a tuple and a name for array name.
UnstructuredMap(data::AbstractVector, dims::UnstructuredDomain)A map that is defined on an unstructured domain. This is typically just a vector of values. The vector of locations of the visibilities are stored in dims. Otherwise this behaves very similarly to IntensityMap, except that is isn't a grid.
For instance the locations of the visibilities can be accessed with axisdims, as well as the usual getproperty and propertynames functions. Like with IntensityMap during execution the executor is used to determine the execution context.
Returns the base image of a intensity map type objectcentroid(im::AbstractIntensityMap)Computes the image centroid aka the center of light of the image.
For polarized maps we return the centroid for Stokes I only.
second_moment(im::AbstractIntensityMap; center=true)Computes the image second moment tensor of the image. By default we really return the second cumulant or centered second moment, which is specified by the center argument.
For polarized maps we return the second moment for Stokes I only.
second_moment(im::AbstractIntensityMap; center=true)Computes the image second moment tensor of the image. By default we really return the second cumulant or centered second moment, which is specified by the center argument.
ComradeBase.load(fitsfile::String, IntensityMap)This loads in a fits file that is more robust to the various imaging algorithms in the EHT, i.e. is works with clean, smili, eht-imaging. The function returns an tuple with an intensitymap and a second named tuple with ancillary information about the image, like the source name, location, mjd, and radio frequency.
ComradeBase.save(file::String, img::IntensityMap, obs)Saves an image to a fits file. You can optionally pass an EHTObservation so that ancillary information will be added.
stokes(m::AbstractPolarizedModel, p::Symbol)Extract the specific stokes component p from the polarized model m
stokes(m::AbstractArray{<:StokesParams}, p::Symbol)Extract the specific stokes component p from the polarized image m.
Internal Methods not part of public API
_visibilitymap(model::AbstractModel, p)Internal method used for trait dispatch and unpacking of args arguments in visibilities
Warn
Not part of the public API so it may change at any moment.
_visibilitymap!(model::AbstractModel, p)Internal method used for trait dispatch and unpacking of args arguments in visibilities!
Warn
Not part of the public API so it may change at any moment.
create_map(array, g::AbstractSingleDomain)Create a map of values specialized by the grid g.