From 7166ef8f3c2c07ea2fe4a6b6b540aac96509e56d Mon Sep 17 00:00:00 2001
From: Paul Brossier <piem@piem.org>
Date: Sat, 5 Sep 2015 11:05:43 +0200
Subject: [PATCH] src/fvec.{c,h}: add fvec_weighted_copy

---
 src/fvec.c | 12 ++++++++++++
 src/fvec.h |  9 +++++++++
 2 files changed, 21 insertions(+)

diff --git a/src/fvec.c b/src/fvec.c
index e2361cc2..619e542d 100644
--- a/src/fvec.c
+++ b/src/fvec.c
@@ -121,6 +121,18 @@ void fvec_weight(fvec_t *s, fvec_t *weight) {
 #endif /* HAVE_ACCELERATE */
 }
 
+void fvec_weighted_copy(fvec_t *in, fvec_t *weight, fvec_t *out) {
+#ifndef HAVE_ACCELERATE
+  uint_t j;
+  uint_t length = MIN(s->length, weight->length);
+  for (j=0; j< length; j++) {
+    out->data[j] = in->data[j] * weight->data[j];
+  }
+#else
+  aubio_vDSP_vmul(in->data, 1, weight->data, 1, out->data, 1, out->length);
+#endif /* HAVE_ACCELERATE */
+}
+
 void fvec_copy(fvec_t *s, fvec_t *t) {
   if (s->length != t->length) {
     AUBIO_ERR("trying to copy %d elements to %d elements \n",
diff --git a/src/fvec.h b/src/fvec.h
index c79295de..ceaec224 100644
--- a/src/fvec.h
+++ b/src/fvec.h
@@ -162,6 +162,15 @@ void fvec_weight(fvec_t *s, fvec_t *weight);
 */
 void fvec_copy(fvec_t *s, fvec_t *t);
 
+/** make a copy of a vector, applying weights to each element
+
+  \param in input vector
+  \param weight weights vector
+  \param out output vector
+
+*/
+void fvec_weighted_copy(fvec_t *in, fvec_t *weight, fvec_t *out);
+
 #ifdef __cplusplus
 }
 #endif
-- 
2.11.0