source: src/fmat.c @ c666a18

feature/autosinkfeature/constantqfeature/pitchshiftfeature/pydocstringsfeature/timestretchpitchshiftsamplertimestretchyinfft+
Last change on this file since c666a18 was c666a18, checked in by Paul Brossier <piem@piem.org>, 6 years ago

src/fmat.c: use memset for each columns

  • Property mode set to 100644
File size: 3.3 KB
Line 
1/*
2  Copyright (C) 2009 Paul Brossier <piem@aubio.org>
3
4  This file is part of aubio.
5
6  aubio is free software: you can redistribute it and/or modify
7  it under the terms of the GNU General Public License as published by
8  the Free Software Foundation, either version 3 of the License, or
9  (at your option) any later version.
10
11  aubio is distributed in the hope that it will be useful,
12  but WITHOUT ANY WARRANTY; without even the implied warranty of
13  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14  GNU General Public License for more details.
15
16  You should have received a copy of the GNU General Public License
17  along with aubio.  If not, see <http://www.gnu.org/licenses/>.
18
19*/
20
21#include "aubio_priv.h"
22#include "fmat.h"
23
24fmat_t * new_fmat (uint_t length, uint_t height) {
25  fmat_t * s = AUBIO_NEW(fmat_t);
26  uint_t i,j;
27  s->height = height;
28  s->length = length;
29  s->data = AUBIO_ARRAY(smpl_t*,s->height);
30  for (i=0; i< s->height; i++) {
31    s->data[i] = AUBIO_ARRAY(smpl_t, s->length);
32    for (j=0; j< s->length; j++) {
33      s->data[i][j]=0.;
34    }
35  }
36  return s;
37}
38
39void del_fmat (fmat_t *s) {
40  uint_t i;
41  for (i=0; i<s->height; i++) {
42    AUBIO_FREE(s->data[i]);
43  }
44  AUBIO_FREE(s->data);
45  AUBIO_FREE(s);
46}
47
48void fmat_write_sample(fmat_t *s, smpl_t data, uint_t channel, uint_t position) {
49  s->data[channel][position] = data;
50}
51smpl_t fmat_read_sample(fmat_t *s, uint_t channel, uint_t position) {
52  return s->data[channel][position];
53}
54void fmat_put_channel(fmat_t *s, smpl_t * data, uint_t channel) {
55  s->data[channel] = data;
56}
57void fmat_get_channel(fmat_t *s, uint_t channel, fvec_t *output) {
58  output->data = s->data[channel];
59  output->length = s->length;
60  return;
61}
62
63smpl_t ** fmat_get_data(fmat_t *s) {
64  return s->data;
65}
66
67/* helper functions */
68
69void fmat_print(fmat_t *s) {
70  uint_t i,j;
71  for (i=0; i< s->height; i++) {
72    for (j=0; j< s->length; j++) {
73      AUBIO_MSG(AUBIO_SMPL_FMT " ", s->data[i][j]);
74    }
75    AUBIO_MSG("\n");
76  }
77}
78
79void fmat_set(fmat_t *s, smpl_t val) {
80  uint_t i,j;
81  for (i=0; i< s->height; i++) {
82    for (j=0; j< s->length; j++) {
83      s->data[i][j] = val;
84    }
85  }
86}
87
88void fmat_zeros(fmat_t *s) {
89#if HAVE_MEMCPY_HACKS
90  uint_t i;
91  for (i=0; i< s->height; i++) {
92    memset(s->data[i], 0, s->length * sizeof(smpl_t));
93  }
94#else
95  fmat_set(s, 0.);
96#endif
97}
98
99void fmat_ones(fmat_t *s) {
100  fmat_set(s, 1.);
101}
102
103void fmat_rev(fmat_t *s) {
104  uint_t i,j;
105  for (i=0; i< s->height; i++) {
106    for (j=0; j< FLOOR(s->length/2); j++) {
107      ELEM_SWAP(s->data[i][j], s->data[i][s->length-1-j]);
108    }
109  }
110}
111
112void fmat_weight(fmat_t *s, fmat_t *weight) {
113  uint_t i,j;
114  uint_t length = MIN(s->length, weight->length);
115  for (i=0; i< s->height; i++) {
116    for (j=0; j< length; j++) {
117      s->data[i][j] *= weight->data[0][j];
118    }
119  }
120}
121
122void fmat_copy(fmat_t *s, fmat_t *t) {
123  if (s->height != t->height) {
124    AUBIO_ERR("trying to copy %d rows to %d rows \n",
125            s->height, t->height);
126    return;
127  }
128  if (s->length != t->length) {
129    AUBIO_ERR("trying to copy %d columns to %d columns\n",
130            s->length, t->length);
131    return;
132  }
133#if HAVE_MEMCPY_HACKS
134  memcpy(t->data, s->data, t->height * t->length * sizeof(smpl_t));
135#else
136  uint_t i,j;
137  for (i=0; i< t->height; i++) {
138    for (j=0; j< t->length; j++) {
139      t->data[i][j] = s->data[i][j];
140    }
141  }
142#endif
143}
144
Note: See TracBrowser for help on using the repository browser.