...

1 files changed, 5 insertions, 1 deletions

diff --git a/tnet_io/cwrapper.cpp b/tnet_io/cwrapper.cpp
index e82f3f8..4149557 100644
--- a/tnet_io/cwrapper.cpp
+++ b/tnet_io/cwrapper.cpp
@@ -2,13 +2,13 @@
 #include "KaldiLib/Labels.h"
 #include "KaldiLib/Common.h"
 #include "KaldiLib/UserInterface.h"
-#include "../../common.h"
 #include <string>
 #define SNAME "TNET"
 
 extern "C" {
 #include "cwrapper.h"
 #include "string.h"
+#include "../../common.h"
 
     extern Matrix *nerv_matrix_host_float_new_(lua_State *L, long nrow, long ncol);
 
@@ -77,6 +77,10 @@ extern "C" {
         repo->feature_repo.MoveNext();
     }
 
+    int tnet_feature_repo_is_end(TNetFeatureRepo *repo) {
+        return repo->feature_repo.EndOfList();
+    }
+
     size_t tnet_feature_repo_current_samplerate(TNetFeatureRepo *repo) {
         return repo->feature_repo.CurrentHeader().mSamplePeriod;
     }