rapl 0.3.0

A crate that makes numerical scripting with Rust simple and enjoyable.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28

use image::ImageFormat;
use rapl::utils::rapl_img::open_lumaf32;
use rapl::*;

fn main() {
    //open image as lumaf32 (gray scale) where 0.0 is white 1.0 is black.
    let img = open_lumaf32("graphics/peppers.png").unwrap();
    //transform to complex and take the 2D FFT
    let fft = img.to_complex().fft2d();

    //initialize kernel for convolution
    let mut kernel: Ndarr<f32, _> = Ndarr::zeros(&fft.dim);
    let (m, n) = (img.shape()[0], img.shape()[1]);
    let mid_x = m / 2;
    let mid_y = n / 2;
    kernel[[mid_x, mid_y]] = 4.;
    kernel[[mid_x + 1, mid_y]] = -1.;
    kernel[[mid_x - 1, mid_y]] = -1.;
    kernel[[mid_x, mid_y + 1]] = -1.;
    kernel[[mid_x, mid_y - 1]] = -1.;
    //FFT the kernell
    let kernell = kernel.to_complex().fft2d();
    //multiply the image FFT to the kernel fft to then do the inverse transform to ger the convolution of the
    //image and the kernel
    let out = (fft * kernell).ifft2().fftshif().re();
    //save output image
    out.save_as_luma("graphics/pepper_edges.png", ImageFormat::Png)
}