src/cvec.{c,h}: add cvec_set, cvec_zeros and cvec_ones

diff --git a/src/cvec.c b/src/cvec.c
index 3ba8104aa39d0df9d0d9dc37cc444c14f5258f7c..877cca31387159cbcd96db37c243ce4f10575979 100644 (file)
--- a/src/cvec.c
+++ b/src/cvec.c
@@ -99,3 +99,20 @@ void cvec_print(cvec_t *s) {
   }
 }
 
+void cvec_set(cvec_t *s, smpl_t val) {
+  uint_t i,j;
+  for (i=0; i< s->channels; i++) {
+    for (j=0; j< s->length; j++) {
+      s->norm[i][j] = val;
+    }
+  }
+}
+
+void cvec_zeros(cvec_t *s) {
+  cvec_set(s, 0.);
+}
+
+void cvec_ones(cvec_t *s) {
+  cvec_set(s, 1.);
+}
+

diff --git a/src/cvec.h b/src/cvec.h
index 8709d8b662c2978c163731904b3c5b35efb243c7..9ff948bcf117d51c7c6f053d95747bfc6c14d175 100644 (file)
--- a/src/cvec.h
+++ b/src/cvec.h
@@ -188,6 +188,28 @@ smpl_t ** cvec_get_phas(cvec_t *s);
 */
 void cvec_print(cvec_t *s);
 
+/** set all elements to a given value
+
+  \param s vector to modify
+  \param val value to set elements to
+
+*/
+void cvec_set(cvec_t *s, smpl_t val);
+
+/** set all elements to zero 
+
+  \param s vector to modify
+
+*/
+void cvec_zeros(cvec_t *s);
+
+/** set all elements to ones 
+
+  \param s vector to modify
+
+*/
+void cvec_ones(cvec_t *s);
+
 #ifdef __cplusplus
 }
 #endif