Compare Pastes
Differences between the pastes
#162878 (31.01.2021 14:42)
and
#276991 (27.08.2024 23:38).
1 | import std.stdio; | |
2 | ||
3 | extern(C) __gshared string[] rt_options = | |
4 | [ "gcopt=minPoolSize:2K maxPoolSize:1M incPoolSize:4K" ]; | |
5 | ||
6 | auto fastAbs(float x) @system | |
7 | { | |
8 | uint p = *(cast(uint*) &x); | |
9 | p &=0x7fffffff; | |
10 | return *(cast(float*) &p); | |
11 | } | |
12 | ||
13 | auto renormalize(Range)(Range r) | |
14 | { | |
15 | import std.algorithm; | |
16 | ||
17 | auto maximalAbsoluteValue = r | |
18 | .map!(a => fastAbs(a)) | |
19 | .fold!max; | |
20 | return r | |
21 | .map!(a => a / maximalAbsoluteValue); | |
22 | } | |
23 | ||
24 | void main() | |
25 | { | |
26 | import std.algorithm; | |
27 | import std.complex; | |
28 | import std.math; | |
29 | import std.numeric; | |
30 | import std.random; | |
31 | import std.range; | |
32 | ||
33 | import ppmformats; | |
34 | ||
35 | auto rng = Random(unpredictableSeed); | |
36 | ||
37 | auto signalWithNoise(float x) | |
38 | { | |
39 | return (3.3 * sin(2*x)) + uniform(-0.3, 0.3, rng); | |
40 | } | |
41 | ||
42 | alias squareWave = delegate(float x) { | |
43 | return | |
44 | ((4 * 10) / (PI)) * (sin(x) + (sin(3 * x) / (3 * x)) + (sin(5 * x) /(5 * x)) + (sin(7 * x) / (7 * x)) + (sin(9 * x) / (9 * x)) + (sin(11 * x) / (11 * x))) + (sin(13 * x) / (13 * x)) + (sin(15 * x) / (15 * x)) + (sin(17 * x) / (17 * x)); | |
45 | }; | |
46 | ||
47 | auto signal = iota(0, 2_048) | |
48 | .map!squareWave; | |
49 | ||
50 | auto normalizedFFT(Range)(Range r) | |
51 | { | |
52 | return | |
53 | r | |
54 | .fft | |
55 | .map!(a => 20.0f * log10(abs(a ^^ 2))) | |
56 | .renormalize; | |
57 | } | |
58 | ||
59 | auto pd = signal | |
60 | .slide(1_024) | |
61 | .map!(a => normalizedFFT(a)); | |
62 | ||
63 | auto img = new P6Image(1024, 1024); | |
64 | ||
65 | foreach (i, e; pd.enumerate) | |
66 | { | |
67 | foreach (j, s; e.enumerate) | |
68 | { | |
69 | img[i, j] = new RGBColor( | |
70 | cast(int) (0.8 * 255.0 * abs(s)), | |
71 | 0, | |
72 | 255 - cast(int) (255 * abs(s)) | |
73 | ); | |
74 | } | |
75 | } | |
76 | img.save(`fft.ppm`); | |
77 | } |