字符匹配字符串

标题可能表达的不太准确，需求是这样的：我们对用户语音通过TensorFlowLite识别，识别结果是26字母中的字母，然后要得到语音句子中哪些单词说对了哪些说错了，比如句子是“I am a ball”识别结果是“i a m b l l”。这样的识别结果应该是I AM BALL是对的 A是错误的。

?

#pragma mark ==========AVAudioRecorderDelegate==========

- (void) audioRecorderDidFinishRecording:(AVAudioRecorder *)avrecorder successfully:(BOOL)flag{

    NSLog(@"录音完毕。开始识别");

    NSLog(@"当前句子%@",self.sentenceModel.sentence_en);

    //去掉标点符号

    NSString * res = [[TFLiteManager shareManager] recognizeAudioWithPath:[self getRecorderPath] text:self.sentenceModel.sentence_en];

    NSLog(@"识别结果：%@",res);

    [self setupRecognizeRes:res];

     

 

}

 

- (void)setupRecognizeRes:(NSString *)res{

    self.recognizeCorrectCount = 0;

    self.totalPhomeCount = 0;

    [self.recognizeRes removeAllObjects];

    //识别结果通过空格分割后的集合

    self.recognizeRes  = [NSMutableArray arrayWithArray:[res componentsSeparatedByString:@" "]];

     

 

    //lowerString小写文本、为了查询字母在原文本中的位置

    NSString * sentence_en = [self.sentenceModel.sentence_en mutableCopy];

    self.upperString = [[NSMutableAttributedString alloc]initWithString:sentence_en];

    self.lowerString = [[NSMutableAttributedString alloc]initWithString:[sentence_en lowercaseString]];

 

    //目标文本

    NSString * orgin_text = [self.sentenceModel.sentence_en mutableCopy];

    //删除特殊字符便于后面处理

    orgin_text = [orgin_text stringByReplacingOccurrencesOfString:@"\"" withString:@""];

    orgin_text = [orgin_text stringByReplacingOccurrencesOfString:@"." withString:@""];

    orgin_text = [orgin_text stringByReplacingOccurrencesOfString:@"," withString:@""];

    orgin_text = [orgin_text stringByReplacingOccurrencesOfString:@"!" withString:@""];

    orgin_text = [orgin_text stringByReplacingOccurrencesOfString:@"?" withString:@""];

    //全部变小写

    orgin_text = [orgin_text lowercaseString];

    //先将目标文本根据空格分割

    NSArray * originWords =  [orgin_text componentsSeparatedByString:@" "];

    self.totalPhomeCount = originWords.count;

 

    for (NSString * sub in originWords) {

        [self isCorrectSub:sub];

    }

 

    //识别结果处理完毕

    [self updateDownbooks];

     

    BOOL allRecorded = [self checkBookPagesisAllRecorded];

    NSLog(@"是否全录音：%d",allRecorded);

    if (allRecorded) {

        NSLog(@"全录完，上传音频,获取最终得分");

        self.bookModel.recorderIndex = [AppSingle sharedAppSingle].record_id;

        self.bookModel.bookScore = [self getBookScore];

        self.bookModel.time = [YCUtils getTimestampString];

        [[AppSingle sharedAppSingle].recordedBooks addObject:self.bookModel];

        [AppSingle sharedAppSingle].record_id ++ ;

        [SaveCachesFile saveDataList:[AppSingle sharedAppSingle].recordedBooks fileName:RecordedList];

    }

     

}

//计算所有句子的平均分

- (float)getBookScore{

    NSInteger sentenceCount = 0;

    NSInteger sentenceScore = 0;

    for (NSInteger i = 0 ;i < self.bookModel.pages.count ; i++) {

        PageModel * page = self.bookModel.pages[i];

        for (NSInteger j = 0; j < page.sentences.count ; j ++) {

            SentenceModel * sentence = page.sentences[j];

            sentenceScore += sentence.score;

            sentenceCount ++ ;

        }

    }

    float res = (float)sentenceScore/(float)sentenceCount;

    NSLog(@"最终得分:%f",res);

    return res;

}

//更新本地数据

- (void)updateDownbooks{

    float score = (float)self.recognizeCorrectCount/(float)self.totalPhomeCount;

    self.currentSpeakCell.scoreView.score = score;

    //再找到对应的page

    for (PageModel * page in self.bookModel.pages) {

        if ([page.page_id isEqualToString:self.sentenceModel.page_id]) {

            //再找到对应的句子

            for (SentenceModel * sentence  in page.sentences) {

                if ([sentence.sentence_id isEqualToString:self.sentenceModel.sentence_id]) {

                    sentence.attributedString  = (NSMutableAttributedString *)self.currentBookCell.lb.attributedText;

                    sentence.score = score;

                    sentence.hasRecord = YES;

                }

            }

        }

    }

}

 

/**

 检查是否本书内句子是否全部录音

 全部已录音返回YES；

 */

- (BOOL)checkBookPagesisAllRecorded{

    for (PageModel * page in self.bookModel.pages) {

        for (SentenceModel * sentence in page.sentences) {

            if (sentence.hasRecord  == NO) {

                return NO;

            }

        }

    }

    return YES;

}

- (void)isCorrectSub:(NSString *)sub{

    self.right = 0;

    self.wrong = 0;

    //sub是单词，分割为字母

    for (NSInteger i = 0; i < sub.length; i++) {

        NSString * phomes = [sub substringWithRange:NSMakeRange(i, 1)];

        NSInteger index = [self getIndexWithTemp:phomes];

        if (index == -1) {

            //没有找到对应的，开始进行下一次的遍历

            //字母错误

            self.wrong ++;

        }else{

            //字母正确

            self.right ++;

        }

        //正确与否都降遍历后的删除掉，进行后续遍历

        [self.recognizeRes removeObjectsInRange:NSMakeRange(0, index+1)];

        //单词遍历完成

        if (i == sub.length - 1) {

            //查找单词在文本中的位置

            NSMutableArray *locationArr = [self calculateSubStringCount:self.lowerString.string str:sub];

            //筛选正确位置

            NSInteger  loc = [self getCorrectLoc:locationArr sub:sub];

            if (loc == -1) {

                NSLog(@"loction error");

                return;

            }

             

            float score = (float)self.right/(float)sub.length;

            if (score >= 0.5) {

                self.recognizeCorrectCount ++;

                //变绿色

                [self.upperString addAttribute:NSForegroundColorAttributeName value:greenColor range:NSMakeRange(loc, sub.length)];

            }else{

                //变红色

                [self.upperString addAttribute:NSForegroundColorAttributeName value:redColor range:NSMakeRange(loc, sub.length)];

            }

            self.currentBookCell.lb.attributedText = self.upperString;

        }

    }

}

 

 

- (NSInteger)getCorrectLoc:(NSMutableArray *)locationArr sub:(NSString *)sub{

    //筛选正确的location  比如this is 中匹配 is。mom o

    for (NSInteger i = 0; i < locationArr.count; i ++) {

        NSInteger  loc = [locationArr[i] integerValue];

        if (loc == 0) {

            return 0;

        }else{

            //前一位必须是空格

            NSMutableAttributedString * fontLocString =(NSMutableAttributedString *)[self.lowerString attributedSubstringFromRange:NSMakeRange(loc - 1, 1)];

            //字符匹配

            NSMutableAttributedString * other_sub = (NSMutableAttributedString *)[self.lowerString attributedSubstringFromRange:NSMakeRange(loc , sub.length)];

            //后一位不能是字母

            NSMutableAttributedString * afterLocString = (NSMutableAttributedString *)[self.lowerString attributedSubstringFromRange:NSMakeRange(loc + sub.length, 1)];

            //同时满足这三个情况才是正确的loc

            if ([@" " isEqualToString:fontLocString.string] && [sub isEqualToString:other_sub.string] && ![self inputIsPhome:afterLocString.string]) {

                return loc;

            }

        }

    }

    return  -1;

}

//判断字符是否是26个字母中的一个

- (BOOL)inputIsPhome:(NSString *)input{

    for (NSString * sub in self.phomes) {

        if ([input isEqualToString:sub]) {

            return YES;

        }

    }

    return NO;

}

/**

 查找子字符串在父字符串中的所有位置

 @param content 父字符串

 @param tab 子字符串

 @return 返回位置数组

 */

- (NSMutableArray*)calculateSubStringCount:(NSString *)content str:(NSString *)tab {

    int location = 0;

    NSMutableArray *locationArr = [NSMutableArray new];

    NSRange range = [content rangeOfString:tab];

    if (range.location == NSNotFound){

        return locationArr;

    }

    //声明一个临时字符串,记录截取之后的字符串

    NSString * subStr = content;

    while (range.location != NSNotFound) {

        if (location == 0) {

            location += range.location;

        } else {

            location += range.location + tab.length;

        }

        //记录位置

        NSNumber *number = [NSNumber numberWithUnsignedInteger:location];

        [locationArr addObject:number];

        //每次记录之后,把找到的字串截取掉

        subStr = [subStr substringFromIndex:range.location + range.length];

//        NSLog(@"subStr %@",subStr);

        range = [subStr rangeOfString:tab];

//        NSLog(@"rang %@",NSStringFromRange(range));

    }

    return locationArr;

}

 

- (NSInteger )getIndexWithTemp:(NSString *)temp{

    NSInteger index;

    for (NSInteger i = 0; i < self.recognizeRes.count; i ++) {

        NSString *index_str = self.recognizeRes[i];

        if ([index_str isEqualToString:temp]) {

            index = i;

            return index;

        }

    }

    //如果都没有匹配到返回-1

    index = - 1;

    return index;

}

　　核心代码是isCorrectSub。注释挺多的，有疑问可以评论。

那个获取location的最优解应该有更好的方法。有更好的方法希望不吝赐教。