-
-
Notifications
You must be signed in to change notification settings - Fork 1.2k
feat: add fft/base/fftpack/rffti
#11467
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
base: develop
Are you sure you want to change the base?
Changes from 2 commits
3195757
64909df
274ca8e
e7f37ce
9ca9dbf
cba0e7f
81286ff
c27a175
500b84b
daef1f3
5c8446b
2eb785a
40a0606
7b69d84
2bc1374
File filter
Filter by extension
Conversations
Jump to
Diff view
Diff view
There are no files selected for viewing
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,168 @@ | ||
| <!-- | ||
|
|
||
| @license Apache-2.0 | ||
|
|
||
| Copyright (c) 2026 The Stdlib Authors. | ||
|
|
||
| Licensed under the Apache License, Version 2.0 (the "License"); | ||
| you may not use this file except in compliance with the License. | ||
| You may obtain a copy of the License at | ||
|
|
||
| http://www.apache.org/licenses/LICENSE-2.0 | ||
|
|
||
| Unless required by applicable law or agreed to in writing, software | ||
| distributed under the License is distributed on an "AS IS" BASIS, | ||
| WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | ||
| See the License for the specific language governing permissions and | ||
| limitations under the License. | ||
|
|
||
| --> | ||
|
|
||
| # rffti | ||
|
|
||
| > Initialize a workspace array for performing a real-valued Fourier transform. | ||
|
|
||
| <!-- Section to include introductory text. Make sure to keep an empty line after the intro `section` element and another before the `/section` close. --> | ||
|
|
||
| <section class="intro"> | ||
|
|
||
| </section> | ||
|
|
||
| <!-- /.intro --> | ||
|
|
||
| <!-- Package usage documentation. --> | ||
|
|
||
| <section class="usage"> | ||
|
|
||
| ## Usage | ||
|
|
||
| ```javascript | ||
| var rffti = require( '@stdlib/fft/base/fftpack/rffti' ); | ||
| ``` | ||
|
|
||
| #### rffti( N, workspace, strideW, offsetW ) | ||
|
|
||
| Initializes a workspace array for performing a real-valued Fourier transform. | ||
|
|
||
| ```javascript | ||
| var Float64Array = require( '@stdlib/array/float64' ); | ||
|
|
||
| var N = 8; | ||
| var workspace = new Float64Array( ( 2*N ) + 34 ); | ||
|
|
||
| rffti( N, workspace, 1, 0 ); | ||
|
|
||
| var twiddleFactors = workspace.slice( N, 2*N ); | ||
| // returns <Float64Array>[ 0, ~0.707, ~0.707, 0, 0, 0, 0, 0 ] | ||
|
|
||
| var factors = workspace.slice( 2*N, ( 2*N ) + 4 ); | ||
| // returns <Float64Array>[ 8, 2, 2, 4 ] | ||
| ``` | ||
|
|
||
| The function accepts the following arguments: | ||
|
|
||
| - **N**: length of the sequence to transform. | ||
| - **workspace**: workspace array. | ||
| - **strideW**: stride length for `workspace`. | ||
| - **offsetW**: starting index for `workspace`. | ||
|
|
||
| </section> | ||
|
|
||
| <!-- /.usage --> | ||
|
|
||
| <!-- Package usage notes. Make sure to keep an empty line after the `section` element and another before the `/section` close. --> | ||
|
|
||
| <section class="notes"> | ||
|
|
||
| ## Notes | ||
|
|
||
| - The workspace array is divided into three sections: | ||
|
|
||
| ```text | ||
| size = N N 2+ceil(log2(N)/2) | ||
| ↓ ↓ ↓ | ||
| | scratch / workspace | twiddle factors | radix factor table | | ||
| ↑ ↑ ↑ | ||
| i = 0 ... N ... 2N ... | ||
| ``` | ||
|
|
||
| - **scratch/workspace**: used as a scratch space when performing transforms. This section is not updated during initialization. | ||
| - **twiddle factors**: a table of reusable complex-exponential constants stored as cosine/sine pairs. | ||
| - **radix factor table**: a table containing the sequence length `N`, the number of factors into which `N` was decomposed, and the individual integer radix factors. | ||
|
|
||
| - In general, a workspace array should have `2N + 34` indexed elements (as `log2(N)/2 ≤ 32` for all `2^64`). During initialization, only the sections for storing twiddle factors and the factorization of `N` are updated. | ||
|
|
||
| - The radix factor table is comprised as follows: | ||
|
|
||
| ```text | ||
| | sequence_length | number_of_factors | integer_factors | | ||
| ``` | ||
|
|
||
| - If `N` equals `1`, the function returns early without modifying the workspace, as a single data point is its own Fourier transform. | ||
|
|
||
| </section> | ||
|
|
||
| <!-- /.notes --> | ||
|
|
||
| <section class="examples"> | ||
|
|
||
| ## Examples | ||
|
|
||
| <!-- eslint no-undef: "error" --> | ||
|
|
||
| ```javascript | ||
| var Float64Array = require( '@stdlib/array/float64' ); | ||
| var rffti = require( '@stdlib/fft/base/fftpack/rffti' ); | ||
|
|
||
| var workspace; | ||
| var N; | ||
| var nf; | ||
| var i; | ||
| var j; | ||
|
|
||
| N = 8; | ||
| workspace = new Float64Array( ( 2*N ) + 34 ); | ||
|
|
||
| rffti( N, workspace, 1, 0 ); | ||
|
|
||
| console.log( 'Sequence length: %d', N ); | ||
| console.log( 'Twiddle factors:' ); | ||
| for ( i = N; i < 2*N; i++ ) { | ||
| console.log( ' workspace[%d] = %d', i, workspace[ i ] ); | ||
| } | ||
|
|
||
| console.log( 'Factorization:' ); | ||
| nf = workspace[ ( 2*N ) + 1 ]; | ||
| console.log( ' number of factors: %d', nf ); | ||
| for ( j = 0; j < nf; j++ ) { | ||
| console.log( ' factor[%d]: %d', j, workspace[ ( 2*N ) + 2 + j ] ); | ||
| } | ||
|
gunjjoshi marked this conversation as resolved.
Outdated
|
||
| ``` | ||
|
|
||
| </section> | ||
|
|
||
| <!-- /.examples --> | ||
|
|
||
| <!-- Section to include cited references. If references are included, add a horizontal rule *before* the section. Make sure to keep an empty line after the `section` element and another before the `/section` close. --> | ||
|
|
||
| <section class="references"> | ||
|
|
||
| </section> | ||
|
|
||
| <!-- /.references --> | ||
|
|
||
| <!-- Section for related `stdlib` packages. Do not manually edit this section, as it is automatically populated. --> | ||
|
|
||
| <section class="related"> | ||
|
|
||
| </section> | ||
|
|
||
| <!-- /.related --> | ||
|
|
||
| <!-- Section for all links. Make sure to keep an empty line after the `section` element and another before the `/section` close. --> | ||
|
|
||
| <section class="links"> | ||
|
|
||
| </section> | ||
|
|
||
| <!-- /.links --> | ||
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,101 @@ | ||
| /* | ||
| * @license Apache-2.0 | ||
| * | ||
| * Copyright (c) 2026 The Stdlib Authors. | ||
| * | ||
| * Licensed under the Apache License, Version 2.0 (the "License"); | ||
| * you may not use this file except in compliance with the License. | ||
| * You may obtain a copy of the License at | ||
| * | ||
| * http://www.apache.org/licenses/LICENSE-2.0 | ||
| * | ||
| * Unless required by applicable law or agreed to in writing, software | ||
| * distributed under the License is distributed on an "AS IS" BASIS, | ||
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | ||
| * See the License for the specific language governing permissions and | ||
| * limitations under the License. | ||
| */ | ||
|
|
||
| 'use strict'; | ||
|
|
||
| // MODULES // | ||
|
|
||
| var bench = require( '@stdlib/bench' ); | ||
| var format = require( '@stdlib/string/format' ); | ||
| var isnan = require( '@stdlib/math/base/assert/is-nan' ); | ||
| var Float64Array = require( '@stdlib/array/float64' ); | ||
| var pkg = require( './../package.json' ).name; | ||
| var rffti = require( './../lib' ); | ||
|
|
||
|
|
||
| // FUNCTIONS // | ||
|
|
||
| /** | ||
| * Creates a benchmark function. | ||
| * | ||
| * @private | ||
| * @param {PositiveInteger} N - sequence length | ||
| * @returns {Function} benchmark function | ||
| */ | ||
| function createBenchmark( N ) { | ||
| var workspace = new Float64Array( ( 2*N ) + 34 ); | ||
| return benchmark; | ||
|
|
||
| /** | ||
| * Benchmark function. | ||
| * | ||
| * @private | ||
| * @param {Benchmark} b - benchmark instance | ||
| */ | ||
| function benchmark( b ) { | ||
| var i; | ||
|
|
||
| b.tic(); | ||
| for ( i = 0; i < b.iterations; i++ ) { | ||
| rffti( N, workspace, 1, 0 ); | ||
| if ( isnan( workspace[ N+1 ] ) ) { | ||
| b.fail( 'should not return NaN' ); | ||
| } | ||
| } | ||
| b.toc(); | ||
| if ( isnan( workspace[ N+1 ] ) ) { | ||
| b.fail( 'should not return NaN' ); | ||
| } | ||
| b.pass( 'benchmark finished' ); | ||
| b.end(); | ||
| } | ||
| } | ||
|
|
||
|
|
||
| // MAIN // | ||
|
|
||
| /** | ||
| * Main execution sequence. | ||
| * | ||
| * @private | ||
| */ | ||
| function main() { | ||
| var lengths; | ||
| var N; | ||
| var f; | ||
| var i; | ||
|
|
||
| lengths = [ | ||
| 8, | ||
| 16, | ||
| 32, | ||
| 64, | ||
| 128, | ||
| 256, | ||
| 512, | ||
| 1024 | ||
| ]; | ||
|
|
||
| for ( i = 0; i < lengths.length; i++ ) { | ||
| N = lengths[ i ]; | ||
| f = createBenchmark( N ); | ||
| bench( format( '%s:N=%d', pkg, N ), f ); | ||
| } | ||
| } | ||
|
|
||
| main(); |
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,45 @@ | ||
|
|
||
| {{alias}}( N, workspace, strideW, offsetW ) | ||
| Initializes a workspace array for performing a real-valued Fourier | ||
| transform. | ||
|
|
||
| The workspace array is divided into three sections: scratch/workspace | ||
| (indices 0 to N-1), twiddle factors (indices N to 2N-1), and radix factor | ||
| table (indices 2N onwards). | ||
|
kgryte marked this conversation as resolved.
Outdated
|
||
|
|
||
| The scratch/workspace section is used while performing transforms and is | ||
| not updated during initialization. The twiddle factors section stores a | ||
| table of reusable complex exponential constants as cosine/sine pairs. The | ||
| radix factor table stores the sequence length N, the number of factors into | ||
| which N was decomposed, and the individual integer radix factors. | ||
|
|
||
| Any remaining array space remains as unused storage. | ||
|
|
||
| Parameters | ||
| ---------- | ||
| N: integer | ||
| Length of the sequence. | ||
|
|
||
| workspace: ArrayLikeObject<number> | ||
| Workspace array. | ||
|
|
||
| strideW: integer | ||
| Stride length for `workspace`. | ||
|
|
||
| offsetW: integer | ||
| Starting index for `workspace`. | ||
|
kgryte marked this conversation as resolved.
|
||
|
|
||
| Examples | ||
| -------- | ||
| > var {{alias:@stdlib/array/float64}} = require( '@stdlib/array/float64' ); | ||
| > var N = 8; | ||
| > var workspace = new {{alias:@stdlib/array/float64}}( ( 2*N ) + 34 ); | ||
| > {{alias}}( N, workspace, 1, 0 ); | ||
| > var twiddleFactors = workspace.slice( N, 2*N ) | ||
| <Float64Array>[ 0, ~0.707, ~0.707, 0, 0, 0, 0, 0 ] | ||
| > var factors = workspace.slice( 2*N, ( 2*N ) + 4 ) | ||
| <Float64Array>[ 8, 2, 2, 4 ] | ||
|
|
||
| See Also | ||
| -------- | ||
|
|
||
| Original file line number | Diff line number | Diff line change | ||||
|---|---|---|---|---|---|---|
| @@ -0,0 +1,51 @@ | ||||||
| /* | ||||||
| * @license Apache-2.0 | ||||||
| * | ||||||
| * Copyright (c) 2026 The Stdlib Authors. | ||||||
| * | ||||||
| * Licensed under the Apache License, Version 2.0 (the "License"); | ||||||
| * you may not use this file except in compliance with the License. | ||||||
| * You may obtain a copy of the License at | ||||||
| * | ||||||
| * http://www.apache.org/licenses/LICENSE-2.0 | ||||||
| * | ||||||
| * Unless required by applicable law or agreed to in writing, software | ||||||
| * distributed under the License is distributed on an "AS IS" BASIS, | ||||||
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | ||||||
| * See the License for the specific language governing permissions and | ||||||
| * limitations under the License. | ||||||
| */ | ||||||
|
|
||||||
| // TypeScript Version: 4.1 | ||||||
|
|
||||||
| /// <reference types="@stdlib/types"/> | ||||||
|
|
||||||
| import { Collection } from '@stdlib/types/array'; | ||||||
|
|
||||||
| /** | ||||||
| * Initializes a workspace array for performing a real-valued Fourier transform. | ||||||
| * | ||||||
| * @param N - length of the sequence | ||||||
| * @param workspace - workspace array | ||||||
| * @param strideW - stride length for `workspace` | ||||||
| * @param offsetW - starting index for `workspace` | ||||||
|
kgryte marked this conversation as resolved.
|
||||||
| * | ||||||
| * @example | ||||||
| * var Float64Array = require( '@stdlib/array/float64' ); | ||||||
| * var N = 8; | ||||||
|
kgryte marked this conversation as resolved.
|
||||||
| * var workspace = new Float64Array( ( 2*N ) + 34 ); | ||||||
| * | ||||||
| * rffti( N, workspace, 1, 0 ); | ||||||
| * | ||||||
| * var twiddleFactors = workspace.slice( N, 2*N ); | ||||||
| * // returns <Float64Array>[ 0, ~0.707, ~0.707, 0, 0, 0, 0, 0 ] | ||||||
| * | ||||||
| * var factors = workspace.slice( 2*N, ( 2*N ) + 4 ); | ||||||
| * // returns <Float64Array>[ 8, 2, 2, 4 ] | ||||||
| */ | ||||||
| declare function rffti( N: number, workspace: Collection<number>, strideW: number, offsetW: number ): void; | ||||||
|
Member
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more.
Suggested change
I suggest going ahead and just returning the workspace array. Will require some minor edits here and elsewhere in this PR.
Member
Author
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. Thanks for making the updates @kgryte. Should we also update all the examples to show the usage of this as a returned workspace array? For example: Changing this: To this:
Member
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. Yes, that works. And then you can also add var bool = ( out === workspace);
// returns truewhich is something we do elsewhere in similar situations. |
||||||
|
|
||||||
|
|
||||||
| // EXPORTS // | ||||||
|
|
||||||
| export = rffti; | ||||||
Uh oh!
There was an error while loading. Please reload this page.