package/gonum
1
0
Fork 0
mirror of https://github.com/gonum/gonum.git synced 2025-10-31 10:36:30 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

dsp/window: add Values.TransformTo window functions and example

Browse Source
This commit is contained in:
Dan Kortschak
2021-01-08 18:18:09 +10:30
parent 1c2011e56d
commit 676e41577b
3 changed files with 100 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

37
dsp/window/window_parametric.go
View File

@@ -147,6 +147,24 @@ func (v Values) Transform(seq []float64) []float64 {
return seq
}
// TransformTo applies the weights in the receiver to src placing the result
// in dst. If v is nil, TransformTo is a no-op, otherwise the length of v must
// match the length of src and dst.
func (v Values) TransformTo(dst, src []float64) {
if v == nil {
return
}
if len(v) != len(src) {
panic("window: seq length mismatch")
}
if len(v) != len(dst) {
panic("window: dst length mismatch")
}
for i, w := range v {
dst[i] = w * src[i]
}
}
// TransformComplex applies the weights in the receiver to seq in place, returning
// the result. If v is nil, TransformComplex is a no-op, otherwise the length of v
// must match the length of seq.
@@ -163,3 +181,22 @@ func (v Values) TransformComplex(seq []complex128) []complex128 {
}
return seq
}
// TransformComplexTo applies the weights in the receiver to src placing the
// result in dst. If v is nil, TransformComplexTo is a no-op, otherwise the
// length of v must match the length of src and dst.
func (v Values) TransformComplexTo(dst, src []complex128) {
if v == nil {
return
}
if len(v) != len(src) {
panic("window: seq length mismatch")
}
if len(v) != len(dst) {
panic("window: dst length mismatch")
}
for i, w := range v {
sv := src[i]
dst[i] = complex(w*real(sv), w*imag(sv))
}
}