DomainColoring(Toy) directly from the 2-D FFT of an image

I just discovered the DomainColoring.jl package and I
would like to use it to display the 2-D FFT of an input 2-D single channel image.

Is there a way to directly pass the Matrix of Complex64 values to be domaincolor()-ed?

Or do I have to create a fake function that takes the input FFT image and maps
the index pairs to complex numbers—only to be unwrapped to create the complex
value matrix to be shown?

Thanks,
Chris

With M being your complex matrix, you may try:

img = DomainColoring.domaincolorshader.(M; abs=true)

# then use Makie to plot the color matrix and add axes:
fig = Figure()
ax = Axis(fig[1,1])
image!(ax, img')