PDF_Office/PDF Office/PDF Master/Class/PDFTools/OCR/KMDocumentAIViewController.m

//
//  KMDocumentAIViewController.m
//  PDF Master
//
//  Created by 丁林圭 on 2023/1/30.
//

#import "KMDocumentAIViewController.h"
#import "KMOCRBox.h"
#import "KMDocumentAIManager.h"
#import "KMFileAttribute.h"
#import "KMOCRComboBox.h"
#import "NSButton+TitleColor.h"
#import <PDF_Master-Swift.h>
#import <Masonry/Masonry.h>

#define kIndicatorWidth  32.0
#define kIndicatorHeight 32.0

#pragma mark - KMBookletMaskView

@interface KMBookletMaskView : NSView

@property (nonatomic, retain) NSProgressIndicator *progressIndicator;

@end

@implementation KMBookletMaskView

- (void)drawRect:(NSRect)dirtyRect {
    [super drawRect:dirtyRect];

    self.wantsLayer = YES;
    self.layer.backgroundColor = [NSColor clearColor].CGColor;
}

- (instancetype)initWithFrame:(NSRect)frameRect {
    self = [super initWithFrame:frameRect];
    if (self) {
        NSProgressIndicator *indicator = [[NSProgressIndicator alloc] initWithFrame:CGRectMake((frameRect.size.width - kIndicatorWidth)/2, (frameRect.size.height - kIndicatorHeight)/2, kIndicatorWidth, kIndicatorHeight)];
        indicator.style = NSProgressIndicatorStyleSpinning;
        self.progressIndicator = indicator;
        [indicator startAnimation:nil];
        [self addSubview:indicator];
    }
    return self;
}

- (void)mouseDown:(NSEvent *)event {
    
}

- (void)mouseUp:(NSEvent *)event {
    
}

@end

#pragma mark - KMDocumentAIViewController

#define kDocumentAIFileSavePath [[[NSSearchPathForDirectoriesInDomains(NSApplicationSupportDirectory, NSUserDomainMask, YES) lastObject] stringByAppendingPathComponent:[NSBundle mainBundle].bundleIdentifier] stringByAppendingPathComponent:@"documentAI.pdf"]

@interface KMDocumentAIViewController ()<CPDFViewDelegate,KMDocumentAIManagerDelegate,NSComboBoxDelegate,CPDFViewDelegate>

@property(nonatomic,assign) IBOutlet KMDocumentAIPDFView *pdfPreView;

//@property(nonatomic,assign) COCRModelType modelType;

@property(nonatomic,retain) CPDFDocument *orgDocument;

@property(nonatomic,retain) CPDFDocument *preDocument;

@property (nonatomic,retain) KMBookletMaskView *posterMaskView;

@property (nonatomic,retain) NSMutableArray *datas;

@property (nonatomic,retain) NSMutableArray *resultDics;

@property (nonatomic,assign) CGRect recognitionRect;

@property (nonatomic,assign) BOOL isContinusOCR;

@property (weak) IBOutlet NSView *rightView;
@property (weak) IBOutlet NSTextField *rightTitleLabel;
@property (weak) IBOutlet KMOCRBox *pageRecognitionBox;
@property (weak) IBOutlet NSTextField *pageRecognitionTitle;
@property (weak) IBOutlet NSImageView *pageRecognitionImageView;
@property (weak) IBOutlet KMOCRBox *areaRecognitionBox;
@property (weak) IBOutlet NSTextField *areaRecognitionTitle;
@property (weak) IBOutlet NSImageView *areaRecognitionImageView;
@property (weak) IBOutlet NSTextField *textRecognitionSetLabel;
@property (weak) IBOutlet KMOCRComboBox *languagePopButton;
@property (weak) IBOutlet NSTextField *pageRecognitionSetLabel;
@property (weak) IBOutlet KMOCRComboBox *pageRangePopButton;
@property (weak) IBOutlet NSButton *OCRButton;
@property (weak) IBOutlet NSView *languageSetView;
@property (weak) IBOutlet NSView *rangeSetView;
@property (nonatomic,copy) NSString *pagesString;

@property (weak) IBOutlet NSLayoutConstraint *tipConstraint;
@property (weak) IBOutlet NSTextField *topTitleLabel;
@property (weak) IBOutlet NSView *topView;

@property (weak) IBOutlet NSView *scanView;
@property (weak) IBOutlet NSTextField *scanTipLabel;
@property (weak) IBOutlet NSProgressIndicator *progressIndicator;

@property (weak) IBOutlet NSView *ocrTipView;
@property (weak) IBOutlet NSTextField *ocrTipLabel;

@end

@implementation KMDocumentAIViewController

static inline NSFont * FontWithSize(NSString *strChar, CGSize size) {
    CGFloat fontsize = 1.0;
    NSFont *font = [NSFont systemFontOfSize:fontsize];
    CGSize strSize = [strChar sizeWithAttributes:@{NSFontAttributeName:font}];
    while ((fontsize<127) && (strSize.width>size.width && strSize.height>size.height)) {
        fontsize += 1.0;
        font = [NSFont systemFontOfSize:fontsize];
        strSize = [strChar sizeWithAttributes:@{NSFontAttributeName:font}];
    }
    return [NSFont systemFontOfSize:fontsize-1];
}

- (void)dealloc {
    [KMDocumentAIManager defaultManager].delegate = nil;
}

- (instancetype)initWithPDFView:(CPDFView *)pdfView {
    if(self = [super initWithNibName:@"KMDocumentAIViewController" bundle:nil]) {
        self.orgDocument = pdfView.document;
        dispatch_async(dispatch_get_global_queue(0, 0), ^{
            [pdfView.document writeToURL:[NSURL fileURLWithPath:kDocumentAIFileSavePath]];
            
            self.preDocument = [[CPDFDocument alloc]initWithURL:[NSURL fileURLWithPath:kDocumentAIFileSavePath]];
            [self.preDocument unlockWithPassword:pdfView.document.password];
            dispatch_async(dispatch_get_main_queue(), ^{
                self.pdfPreView.delegate = self;
                self.pdfPreView.displayDirection = CPDFDisplayDirectionHorizontal;
                self.pdfPreView.scaleFactor = pdfView.scaleFactor;
                self.pdfPreView.displayTwoUp = NO;
                self.pdfPreView.autoScales = YES;
                self.pdfPreView.document = self.preDocument;
                
                NSInteger pageIndex = pdfView.currentPageIndex;
                [self.pdfPreView goToPageIndex:pageIndex animated:NO];
            });
        });
    }
    return self;
}

- (void)loadView {
    [super loadView];
    [self initUI];
}

- (void)viewDidLoad {
    [super viewDidLoad];
    
    self.datas = [NSMutableArray array];
    self.resultDics = [NSMutableArray array];
    
    __block typeof(self) blockSelf = self;
    
    [KMDocumentAIManager defaultManager].delegate = self;
    self.pdfPreView.callback = ^(NSRect recognitionRect, CPDFPage *page) {
        if (!CGRectEqualToRect(recognitionRect, CGRectZero) && blockSelf.pdfPreView.isRecognitionErea) {
            [blockSelf.datas removeAllObjects];
            [blockSelf.datas addObject:page];
            blockSelf.recognitionRect = recognitionRect;

            CPDFPage *copyPage = [page copy];
            [copyPage setBounds:recognitionRect forBox:CPDFDisplayCropBox];
            NSImage * image = [[NSImage alloc] initWithSize:recognitionRect.size];
            [image lockFocus];
            [[NSGraphicsContext currentContext] setImageInterpolation:NSImageInterpolationHigh];
            CGContextRef context = (CGContextRef)[[NSGraphicsContext currentContext] graphicsPort];
            [copyPage drawWithBox:CPDFDisplayCropBox toContext:context];
            [image unlockFocus];
            
//            [[KMDocumentAIManager defaultManager] recognitionImages:@[image] withModelType:blockSelf.modelType];
            
        }
    };
}

- (void)initUI {
    self.scanView.hidden = YES;
    self.scanView.frame = CGRectMake(CGRectGetWidth(self.view.bounds),20, 226, 53);
    self.scanView.wantsLayer = YES;
    self.scanView.layer.backgroundColor = [NSColor colorWithSRGBRed:54/255.0 green:56/255.0 blue:59/255.0 alpha:1].CGColor;
    self.scanView.layer.cornerRadius = 4;
    self.scanTipLabel.textColor = [NSColor whiteColor];
    [self.view addSubview:self.scanView];
    [self.scanView mas_makeConstraints:^(MASConstraintMaker *make) {
        make.bottom.equalTo(self.view.mas_bottom).offset(-20);
        make.right.equalTo(self.view.mas_right).offset(-20);
        make.width.offset(226);
        make.height.offset(53);
    }];
    
    self.rightView.hidden = YES;
    self.rightTitleLabel.stringValue = NSLocalizedString(@"OCR Text Recognition", nil);
    self.pageRecognitionTitle.stringValue = NSLocalizedString(@"Page Recognition", nil);
    self.areaRecognitionTitle.stringValue = NSLocalizedString(@"Area Recognition", nil);
    self.textRecognitionSetLabel.stringValue = NSLocalizedString(@"Text Recognition Settings", nil);
    self.pageRecognitionSetLabel.stringValue = NSLocalizedString(@"Page Range", nil);
    self.topTitleLabel.stringValue = NSLocalizedString(@"This file is an image file, do you need to perform OCR?", nil);

    self.languagePopButton.type = KMOCRComboBoxType_None;
    self.pageRangePopButton.type = KMOCRComboBoxType_None;
    self.languagePopButton.cell.focusRingType = NSFocusRingTypeNone;
    self.pageRangePopButton.cell.focusRingType = NSFocusRingTypeNone;
    self.OCRButton.wantsLayer =
    self.topView.wantsLayer =
    self.rightView.wantsLayer = YES;
    self.rightView.layer.backgroundColor = [NSColor colorWithSRGBRed:247.0/255.0 green:248.0/255.0 blue:250.0/255.0 alpha:1].CGColor;
    self.topView.layer.backgroundColor = [NSColor colorWithSRGBRed:189.0/255.0 green:223.0/255.0 blue:253.0/255.0 alpha:1].CGColor;
    self.pageRecognitionBox.fillColor =
    self.areaRecognitionBox.fillColor = [NSColor whiteColor];
    self.pageRecognitionBox.borderColor =
    self.areaRecognitionBox.borderColor = [NSColor colorWithSRGBRed:223.0/255.0 green:225.0/255.0 blue:229.0/255.0 alpha:1];
    self.languageSetView.hidden =
    self.rangeSetView.hidden = YES;
    self.OCRButton.layer.backgroundColor = [NSColor blueColor].CGColor;
    self.OCRButton.layer.cornerRadius = 4;
    [self.OCRButton setTitle:NSLocalizedString(@"OCR", nil)];
    [self.OCRButton setTitleColor:[NSColor whiteColor]];
    self.OCRButton.enabled = NO;
    
    self.ocrTipView.hidden = YES;
    self.ocrTipView.wantsLayer = YES;
    self.ocrTipView.layer.backgroundColor = [NSColor colorWithSRGBRed:255.0/255.0 green:241.0/255.0 blue:193.0/255.0 alpha:1].CGColor;
    self.ocrTipLabel.stringValue = NSLocalizedString(@"A page will only record the last OCR result and will not display multiple results.", nil);
    
    [self.languagePopButton removeAllItems];
    NSArray *languages = @[@"Chinese Simplified",@"Chinese Traditional",@"English",@"Japanese",@"Korean"];
    for (NSString *language in languages) {
        [self.languagePopButton addItemWithObjectValue:language];
    }
    [self.languagePopButton selectItemAtIndex:0];
    [self.languagePopButton setEditable:NO];
    
    self.pageRangePopButton.delegate = nil;
    [self.pageRangePopButton removeAllItems];
    [self.pageRangePopButton addItemsWithObjectValues:@[NSLocalizedString(@"All Pages", nil),
                                                       NSLocalizedString(@"Odd Pages Only", nil),
                                                       NSLocalizedString(@"Even Pages Only",nil),
                                                       NSLocalizedString(@"e.g. 1,3-5,10",nil)]];
    [self.pageRangePopButton.cell setPlaceholderString:NSLocalizedString(@"e.g. 1,3-5,10", nil)];
    [self.pageRangePopButton selectItemAtIndex:0];
    [self.pageRangePopButton setEditable:NO];
    self.pageRangePopButton.delegate = self;
    
    self.pageRecognitionBox.mouseDownCallback = ^(BOOL downEntered, KMOCRBox *mouseBox) {
        if (downEntered) {
            if (self.languageSetView.hidden || self.rangeSetView.hidden) {
                self.pageRecognitionBox.borderColor = [NSColor colorWithSRGBRed:23.0/255.0 green:12.0/255.0 blue:244.0/255.0 alpha:1];
                self.areaRecognitionBox.borderColor = [NSColor colorWithSRGBRed:223.0/255.0 green:225.0/255.0 blue:229.0/255.0 alpha:1];
                self.pageRecognitionImageView.image = [NSImage imageNamed:@"KMImageNameOCRPageSelect"];
                self.areaRecognitionImageView.image = [NSImage imageNamed:@"KMImageNameOCRArea"];
                self.languageSetView.hidden =
                self.rangeSetView.hidden = NO;
                self.OCRButton.enabled = YES;
                self.ocrTipView.hidden = NO;
            } else {
                self.pageRecognitionBox.borderColor =
                self.areaRecognitionBox.borderColor = [NSColor colorWithSRGBRed:223.0/255.0 green:225.0/255.0 blue:229.0/255.0 alpha:1];
                self.areaRecognitionImageView.image = [NSImage imageNamed:@"KMImageNameOCRArea"];
                self.pageRecognitionImageView.image = [NSImage imageNamed:@"KMImageNameOCRPage"];
                self.languageSetView.hidden =
                self.rangeSetView.hidden = YES;
                self.OCRButton.enabled = NO;
                self.ocrTipView.hidden = YES;
            }
            self.pdfPreView.isRecognitionErea = NO;
        }
    };
    
    self.areaRecognitionBox.mouseDownCallback = ^(BOOL downEntered, KMOCRBox *mouseBox) {
        if (downEntered) {
            if(!self.languageSetView.hidden && self.rangeSetView.hidden) {
                self.pageRecognitionBox.borderColor =
                self.areaRecognitionBox.borderColor = [NSColor colorWithSRGBRed:223.0/255.0 green:225.0/255.0 blue:229.0/255.0 alpha:1];
                self.areaRecognitionImageView.image = [NSImage imageNamed:@"KMImageNameOCRArea"];
                self.pageRecognitionImageView.image = [NSImage imageNamed:@"KMImageNameOCRPage"];
                self.languageSetView.hidden =
                self.rangeSetView.hidden = YES;
                self.OCRButton.enabled = NO;
                self.ocrTipView.hidden = YES;
                self.pdfPreView.isRecognitionErea = NO;
            } else {
                self.areaRecognitionBox.borderColor = [NSColor colorWithSRGBRed:23.0/255.0 green:12.0/255.0 blue:244.0/255.0 alpha:1];
                self.pageRecognitionBox.borderColor = [NSColor colorWithSRGBRed:223.0/255.0 green:225.0/255.0 blue:229.0/255.0 alpha:1];
                self.pageRecognitionImageView.image = [NSImage imageNamed:@"KMImageNameOCRPage"];
                self.areaRecognitionImageView.image = [NSImage imageNamed:@"KMImageNameOCRAreaSelect"];
                self.languageSetView.hidden = NO;
//                [self recognitionPartModelWithModelType:self.languagePopButton.indexOfSelectedItem+1];
                self.rangeSetView.hidden = YES;
                self.OCRButton.enabled = NO;
                self.ocrTipView.hidden = NO;
            }
        }
    };

    self.pdfPreView.delegate = self;
  
    [self closeTopViewAction:nil];
}

- (void)updateToolState:(BOOL)isOCR {
    self.rightView.hidden = !isOCR;
    self.isContinusOCR = !self.rightView.hidden;
}

- (IBAction)closeScanViewAction:(id)sender {
    
}

- (IBAction)closeTopViewAction:(id)sender {
    self.topView.hidden = YES;
    self.tipConstraint.constant = 0;
}

- (IBAction)seleclanguageAction:(id)sender {
    if(!self.languageSetView.hidden && self.rangeSetView.hidden) {
//        [self recognitionPartModelWithModelType:self.languagePopButton.indexOfSelectedItem+1];
    }
}

- (IBAction)selectPageRangeAction:(KMOCRComboBox *)sender {
    if (sender.indexOfSelectedItem == 3) {
        self.pageRangePopButton.stringValue = @"";
        [self.pageRangePopButton setEditable:YES];
    } else {
        [self.pageRangePopButton setEditable:NO];
    }
}
- (IBAction)OCRDone:(id)sender {
    //避免范围提示弹框弹2次
    NSMutableArray * pages = [NSMutableArray array];
    if(0 == [_pageRangePopButton indexOfSelectedItem]){
        for (NSUInteger i = 0; i<self.pdfPreView.document.pageCount; i++) {
            [pages addObject:@(i+1)];
        }
    
    } else if (1 == [_pageRangePopButton indexOfSelectedItem]) {
        for (NSUInteger i = 0; i < self.pdfPreView.document.pageCount; i ++) {
            if (i % 2 == 0) {
                [pages addObject:@(i+1)];
            }
        }
        
    } else if (2 == [_pageRangePopButton indexOfSelectedItem]) {
        for (NSUInteger i = 0; i < self.pdfPreView.document.pageCount; i ++) {
            if (i % 2 != 0) {
                [pages addObject:@(i+1)];
            }
        }
    } else {
        KMFileAttribute *fileAttribute = [[KMFileAttribute alloc] init];
        fileAttribute.filePath = [self.pdfPreView.document.documentURL path];
        
        fileAttribute.bAllPage = NO;
        fileAttribute.pagesString = _pageRangePopButton.stringValue;
        
        if (!fileAttribute.selectPages) {
            NSAlert *alert = [[NSAlert alloc] init];
            [alert setAlertStyle:NSAlertStyleCritical];
            [alert setMessageText:[NSString stringWithFormat:@"%@ %@",[fileAttribute.filePath lastPathComponent],NSLocalizedString(@"Invalid page range or the page number is out of range. Please try again.", nil)]];
            [alert runModal];
            
            return;
        }
        for (NSNumber * num in fileAttribute.selectPages) {
            [pages addObject:@(num.integerValue)];
        }
    }
    
    if (pages.count >0) {
        self.pagesString = [pages componentsJoinedByString:@","];
    } else {
        self.pagesString = nil;
    }
   
    if (self.pagesString) {
//        [self recognitionPageString:self.pagesString withModelType:self.languagePopButton.indexOfSelectedItem+1];
    }
    
}

- (void)enteredIncreaseAllPage {
    NSMutableArray *images = [NSMutableArray array];
    
    for(NSInteger i = 0;i<self.orgDocument.pageCount;i++) {
        CPDFPage *page = [self.orgDocument pageAtIndex:i];
        NSImage *image = [page thumbnailOfSize:page.bounds.size];
        [images addObject:image];
        
        [self.datas addObject:page];
    }
    self.isContinusOCR = NO;
    [[KMDocumentAIManager defaultManager] increaseImages:images];
    
    self.pdfPreView.isRecognitionErea = NO;
}

- (void)resetIncreaseAllPage {
    self.pdfPreView.document = self.orgDocument;
    [self.pdfPreView layoutDocumentView];
    
    self.pdfPreView.isRecognitionErea = NO;
}

//- (void)recognitionPageString:(NSString *)pageString withModelType:(COCRModelType)modelType {
//    self.pdfPreView.isRecognitionErea = NO;
//
//    KMFileAttribute *fileAttribute = [[KMFileAttribute alloc] init];
//    fileAttribute.pdfDocument = self.orgDocument;
//    fileAttribute.bAllPage = NO;
//    fileAttribute.filePath = self.orgDocument.documentURL.path;
//    fileAttribute.pagesString = pageString;
//    fileAttribute.password = self.orgDocument.password;
//    NSMutableArray *images = [NSMutableArray array];
//    self.datas = [NSMutableArray array];
//    self.resultDics = [NSMutableArray array];
//    for (NSNumber * num in fileAttribute.selectPages) {
//        CPDFPage *page = [self.preDocument pageAtIndex:num.integerValue-1];
//        NSImage *image = [page thumbnailOfSize:page.bounds.size];
//        if(image) {
//            [images addObject:image];
//            [self.datas addObject:page];
//        }else {
//            NSLog(@"111");
//        }
//    }
//
//    [KMDocumentAIManager defaultManager].delegate = self;
//    [[KMDocumentAIManager defaultManager] recognitionImages:images withModelType:modelType];
//    self.modelType = modelType;
//}
//
//- (void)recognitionPartModelWithModelType:(COCRModelType)modelType {
//    self.pdfPreView.isRecognitionErea = YES;
//    self.modelType = modelType;
//}

- (void)creatRecognitionDocument {
    for (NSDictionary *dic in self.resultDics) {
        NSNumber *num = [dic objectForKey:@"page"];
        if(num.integerValue < self.orgDocument.pageCount) {
            CPDFPage *page = [self.orgDocument pageAtIndex:num.integerValue];
            
            NSArray *results = [dic objectForKey:@"results"];
//            for (COCRResult *result in results) {
//                CGRect rect =  result.textBounds;
//
//                NSString *strChar = result.text;
//                CGFloat fontsize = 1.0;
//
//                NSFont *font = [NSFont systemFontOfSize:fontsize];
//                CGSize strSize = [strChar sizeWithAttributes:@{NSFontAttributeName:font}];
//                while ((fontsize<127) && !(strSize.width>=result.textBounds.size.width || strSize.height>=result.textBounds.size.height)) {
//                    fontsize += 1.0;
//                    font = [NSFont systemFontOfSize:fontsize];
//                    strSize = [strChar sizeWithAttributes:@{NSFontAttributeName:font}];
//                }
//                NSFont *drawFont = [NSFont systemFontOfSize:fontsize-1];
//
//                NSDictionary *dic = @{NSFontAttributeName:drawFont};
//
//                [page addStringBounds:result.textBounds withString:result.text withAttributes:dic];
//            }

        }
    }
}

- (void)controlTextDidEndEditing:(NSNotification *)obj {
    if ([obj.object isEqual:self.pageRangePopButton]) {
        if (self.pageRangePopButton.indexOfSelectedItem == -1) {
            if (![self checkPageRangeValidate:self.pageRangePopButton.stringValue]) {
                NSAlert *alert = [[NSAlert alloc] init];
                [alert setAlertStyle:NSAlertStyleCritical];
                [alert setMessageText:[NSString stringWithFormat:@"%@ %@",[self.pdfPreView.document.documentURL.lastPathComponent lastPathComponent],NSLocalizedString(@"Invalid page range or the page number is out of range. Please try again.", nil)]];
                [alert runModal];
                
                [self.view.window makeFirstResponder:self.pageRangePopButton];
                return;
            } else {
                self.pagesString = self.pageRangePopButton.stringValue;
            }
        }
    }
}

- (BOOL)checkPageRangeValidate:(NSString *)pageRangeString {
    KMFileAttribute *fileAttribute = [[KMFileAttribute alloc] init];
    fileAttribute.filePath = [self.pdfPreView.document.documentURL path];
    fileAttribute.bAllPage = NO;
    fileAttribute.pagesString = self.pageRangePopButton.stringValue;
    
    if (!fileAttribute.selectPages) {
        return NO;
    }
    return YES;
}
    
#pragma mark - CPDFViewDelegate

- (void)PDFViewDocumentDidLoaded:(CPDFView *)pdfView {
    dispatch_async(dispatch_get_global_queue(DISPATCH_QUEUE_PRIORITY_DEFAULT, 0), ^{
        BOOL isImageDocument = [self.pdfPreView.document isImageDocument];
        if(isImageDocument) {
            dispatch_async(dispatch_get_main_queue(), ^{
                self.topView.hidden = NO;
                self.tipConstraint.constant = 32;
            });
        }
    });
}

#pragma mark - KMDocumentAIManagerDelegate

- (void)documentAIManagerDidStart:(KMDocumentAIManager *)manager {
    if(self.isContinusOCR) {
        self.rightView.hidden = YES;
    }
    self.scanView.hidden = NO;
    [self.pdfPreView clearOCRResult];
    [self showWaitting];
}

- (void)documentAIDidFinish:(KMDocumentAIManager *)manager {
   if(self.isContinusOCR) {
        self.rightView.hidden = NO;
    }
    self.scanView.hidden = YES;
    [self hideWaitting];

}

- (void)documentAIManager:(KMDocumentAIManager *)manager didStartImageAtIndex:(NSInteger)index {
    
}

//- (void)documentAIManager:(KMDocumentAIManager *)manager didFinishOCRImageAtIndex:(NSInteger)index results:(NSArray<COCRResult *> *)results {
//    CPDFPage *page = self.datas[index];
//    NSInteger pageIndex = [self.preDocument indexForPage:page];
//
//    NSMutableDictionary *dic = [NSMutableDictionary dictionary];
//    [dic setValue:@(pageIndex) forKey:@"page"];
//    NSMutableDictionary *recognitionDic = [NSMutableDictionary dictionary];
//
//    CGRect recognitionRect = page.bounds;
//    if(self.pdfPreView.isRecognitionErea) {
//        NSMutableArray *tResults = [NSMutableArray array];
//        for (COCRResult * result  in results) {
//            CGRect rect = result.textBounds;
//            rect.origin.x += self.recognitionRect.origin.x;
//            rect.origin.y += self.recognitionRect.origin.y;
//            result.textBounds = rect;
//            [tResults addObject:result];
//        }
//        recognitionRect = self.recognitionRect;
//        results = tResults;
//    }
//
//    NSValue *value = [NSValue valueWithRect:recognitionRect];
//    [dic setValue:value forKey:@"recognitionRect"];
//    [dic setValue:results forKey:@"results"];
//
//    [self.resultDics addObject:dic];
//    self.pdfPreView.resultDics = self.resultDics;
//    if(pageIndex == self.pdfPreView.currentPageIndex) {
//        [self.pdfPreView updateResult:results recognitionRect:recognitionRect];
//    }
//    float allPages = [[NSString stringWithFormat:@"%lu",(unsigned long)self.datas.count] floatValue];
//    self.progressIndicator.doubleValue = (index / allPages) * 100 ;
//    [self hideWaitting];
//}

- (void)imageEngineManager:(KMDocumentAIManager *)manager didFinishCIImageAtIndex:(NSInteger)index outImage:(NSString *)filePath {
    NSImage *image = [[NSImage alloc]initWithContentsOfURL:[NSURL fileURLWithPath:filePath]];
    BOOL success = [self.preDocument insertPage:image.size withImage:filePath atIndex:index];
    CPDFPage *page = [self.preDocument pageAtIndex:index];
    [self.datas replaceObjectAtIndex:index withObject:page];
    if(success) {
        [self.preDocument removePageAtIndex:index+1];
    }
    self.pdfPreView.document = self.preDocument;
    [self.pdfPreView layoutDocumentView];
    [self hideWaitting];
}

- (void)documentAIManager:(KMDocumentAIManager *)manager didFailureImageAtIndex:(NSInteger)index error:(NSError *)error {
    
    [self hideWaitting];
}

#pragma mark - CPDFViewDelegate

- (void)PDFViewCurrentPageDidChanged:(CPDFView *)pdfView {
    if ([self.documentAIViewDelegate respondsToSelector:@selector(documentAIViewController:currentPageDidChanged:)]) {
        [self.documentAIViewDelegate documentAIViewController:self currentPageDidChanged:pdfView];
    }
    for (NSDictionary *dic in self.resultDics) {
        NSNumber *num = [dic objectForKey:@"page"];
        if(num.integerValue == self.pdfPreView.currentPageIndex) {
            NSValue *value = [dic objectForKey:@"recognitionRect"];
//            [self.pdfPreView updateResult:[dic objectForKey:@"results"] recognitionRect:value.rectValue];
        }
    }
}

- (void)PDFViewScaleDidChanged:(CPDFView *)pdfView {
    if ([self.documentAIViewDelegate respondsToSelector:@selector(documentAIViewController:scaleDidChanged:)]) {
        [self.documentAIViewDelegate documentAIViewController:self scaleDidChanged:pdfView];
    }
}

#pragma mark - show / hide Waiting

- (void)showWaitting {
    if (!_posterMaskView) {
        _posterMaskView = [[KMBookletMaskView alloc] initWithFrame:NSMakeRect(0, 0, self.view.window.frame.size.width, self.view.window.frame.size.height)];
    }
    [self.view.window.contentView addSubview:self.posterMaskView];
}

- (void)hideWaitting{
    [self.posterMaskView removeFromSuperview];
}

@end