MLECO-2835: Remove magic number for ASR-KWS
Replaced ctx.set/get<uint32>(keywordindex) with
keyword itself as const std::string&
Change-Id: I1811d93548105d6db58e57b88675f9b41e66d914
Signed-off-by: Liam Barry <liam.barry@arm.com>
diff --git a/source/use_case/kws_asr/src/MainLoop.cc b/source/use_case/kws_asr/src/MainLoop.cc
index 30cb084..c7e977f 100644
--- a/source/use_case/kws_asr/src/MainLoop.cc
+++ b/source/use_case/kws_asr/src/MainLoop.cc
@@ -136,8 +136,15 @@
caseContext.Set<const std::vector <std::string>&>("asrlabels", asrLabels);
caseContext.Set<const std::vector <std::string>&>("kwslabels", kwsLabels);
- /* Index of the kws outputs we trigger ASR on. */
- caseContext.Set<uint32_t>("keywordindex", 9 );
+ /* KWS keyword that triggers ASR and associated checks */
+ std::string triggerKeyword = std::string("yes");
+ if (std::find(kwsLabels.begin(), kwsLabels.end(), triggerKeyword) != kwsLabels.end()) {
+ caseContext.Set<const std::string &>("triggerkeyword", triggerKeyword);
+ }
+ else {
+ printf_err("Selected trigger keyword not found in labels file\n");
+ return;
+ }
/* Loop. */
bool executionSuccessful = true;