add detection of utterances without alignment, and skip them

1 files changed, 6 insertions, 2 deletions

diff --git a/kaldi_io/src/cwrapper_kaldi.cpp b/kaldi_io/src/cwrapper_kaldi.cpp
index d16f316..542f1d0 100644
--- a/kaldi_io/src/cwrapper_kaldi.cpp
+++ b/kaldi_io/src/cwrapper_kaldi.cpp
@@ -150,8 +150,12 @@ extern "C" {
     Matrix *kaldi_label_repo_read_utterance(KaldiLabelRepo *repo, KaldiFeatureRepo *frepo, int nframes,
                                             lua_State *L,
                                             int debug) {
-        Matrix *mat;
-        kaldi::Posterior targets = repo->targets_reader->Value(frepo->utt);
+        Matrix *mat = NULL;
+        /* check if the alignment of the utterance exists, otherwise return NULL */
+        kaldi::Posterior targets;
+        if (repo->targets_reader->HasKey(frepo->utt))
+            targets = repo->targets_reader->Value(frepo->utt);
+        else return mat;
 
         int n = targets.size() < nframes ? targets.size() : nframes;
         int m = (int)targets[0].size();