Skip to content

Commit 3b4d044

Browse files
authored
[src] Fix return condition and add resize code in online-audio-source.cc (kaldi-asr#4047)
1 parent 5968b4c commit 3b4d044

File tree

2 files changed

+4
-1
lines changed

2 files changed

+4
-1
lines changed

src/online/online-audio-source.cc

+3-1
Original file line numberDiff line numberDiff line change
@@ -177,12 +177,14 @@ bool OnlineVectorSource::Read(Vector<BaseFloat> *data) {
177177
if (data->Dim() == subsrc.Dim()) {
178178
data->CopyFromVec(subsrc);
179179
} else {
180+
data->Resize(n_elem);
180181
for (int32 i = 0; i < subsrc.Dim(); ++i)
181182
(*data)(i) = subsrc(i);
182183
}
183184
pos_ += n_elem;
185+
return true;
184186
}
185-
return (pos_ < src_.Dim());
187+
return false;
186188
}
187189

188190
} // namespace kaldi

src/online/online-tcp-source.cc

+1
Original file line numberDiff line numberDiff line change
@@ -140,6 +140,7 @@ bool OnlineTcpVectorSource::Read(Vector<BaseFloat> *data) {
140140
int32 n_read = b_read / 2;
141141

142142
short* s_frame = (short*) frame;
143+
data->Resize(n_read);
143144
for (int32 i = 0; i < n_read; i++)
144145
(*data)(i) = s_frame[i];
145146

0 commit comments

Comments
 (0)