PDF_Office/PDF Office/PDF Master/Class/Common/OC/Batch/KMOCTool.m

//
//  KMOCTool.m
//  PDF Reader Pro
//
//  Created by liujiajie on 2023/11/15.
//

#import "KMOCTool.h"
#import <Foundation/Foundation.h>
#import <CoreGraphics/CoreGraphics.h>
#import <Cocoa/Cocoa.h>
#import <PDF_Reader_Pro-Swift.h>

@implementation KMOCTool
+ (void)createPDFFile:(NSString *)filePath imagePaths:(NSArray *)paths results:(NSArray *)resultsArray scale:(CGFloat)scale {
    if (paths.count < 1) {
        return;
    }
    
    CFStringRef path = (__bridge CFStringRef)filePath;
    CFURLRef url = CFURLCreateWithFileSystemPath(NULL, path, kCFURLPOSIXPathStyle, 0);
    CFMutableDictionaryRef myDictionary = CFDictionaryCreateMutable(NULL,
                                                                    0,
                                                                    &kCFTypeDictionaryKeyCallBacks,
                                                                    &kCFTypeDictionaryValueCallBacks);
    CFDictionarySetValue(myDictionary, kCGPDFContextCreator, CFSTR("Kdan Mobile PDF Reader"));
    
    CGContextRef pdfContext = CGPDFContextCreateWithURL(url, &CGRectZero, myDictionary);
    CGContextSetRGBFillColor(pdfContext, 1.0, 0.0, 0.0, 0.0);
    CGContextSetTextDrawingMode(pdfContext, kCGTextFill);
    
    CFRelease(myDictionary);
    CFRelease(url);
    
    for (int i=0; i<paths.count; i++) {
        NSString *path = [paths objectAtIndex:i];
        NSImage *image = [[NSImage alloc] initWithContentsOfFile:path];
        CIImage *imageCIImage = [CIImage imageWithContentsOfURL:[NSURL fileURLWithPath:path]];
        NSSize size = [imageCIImage extent].size;
        CGRect pageRect = CGRectMake(0, 0, size.width/scale, size.height/scale);
        
        CFMutableDictionaryRef pageDictionary = CFDictionaryCreateMutable(NULL,
                                                                          0,
                                                                          &kCFTypeDictionaryKeyCallBacks,
                                                                          &kCFTypeDictionaryValueCallBacks);
        CFDataRef boxData = CFDataCreate(NULL,(const UInt8 *)&pageRect, sizeof (CGRect));
        CFDictionarySetValue(pageDictionary, kCGPDFContextMediaBox, boxData);
        CGPDFContextBeginPage (pdfContext, pageDictionary);

        NSData *imageData = [NSData dataWithContentsOfFile:path];
        CGImageSourceRef imageSource = CGImageSourceCreateWithData((CFDataRef)imageData, NULL);
        CGImageRef imageRef = CGImageSourceCreateImageAtIndex(imageSource, 0, NULL);
        CGContextSaveGState(pdfContext);
        CGContextDrawImage(pdfContext, pageRect, imageRef);
        CGContextRestoreGState(pdfContext);
        CGImageRelease(imageRef);
        
        [NSGraphicsContext saveGraphicsState];
        [NSGraphicsContext setCurrentContext:[NSGraphicsContext graphicsContextWithCGContext:pdfContext flipped:NO]];
        NSArray *results = nil;
        if (i < resultsArray.count) {
            results = resultsArray[i];
            CGFloat newScale = scale;
            if([KMGOCRManager defaultManager].OCRType == KMOCRType_Apple)
                newScale = 1;
            if (results.count == 1) {
                KMGOCRResult *result = results[0];
                NSRect bounds = NSMakeRect((result.textBounds.origin.x)/newScale,
                                           pageRect.size.height-(result.textBounds.origin.y+result.textBounds.size.height)/newScale,
                                           (result.textBounds.size.width)/newScale,
                                           (result.textBounds.size.height)/newScale);
                NSDictionary *dic = @{NSFontAttributeName:FontWithSize(result.text, CGSizeMake(result.textBounds.size.width/newScale, result.textBounds.size.height/newScale)),
                                      NSForegroundColorAttributeName:[NSColor clearColor]};
                [result.text drawInRect:bounds withAttributes:dic];
            } else {
                for (int i=1; i<results.count; i++) {
                    KMGOCRResult *result = results[i];
                    NSRect bounds = NSMakeRect((result.textBounds.origin.x)/newScale,
                                               pageRect.size.height-(result.textBounds.origin.y+result.textBounds.size.height)/newScale,
                                               (result.textBounds.size.width)/newScale,
                                               (result.textBounds.size.height)/newScale);
                    NSDictionary *dic = @{NSFontAttributeName:FontWithSize(result.text, CGSizeMake(result.textBounds.size.width/newScale, result.textBounds.size.height/newScale)),
                                          NSForegroundColorAttributeName:[NSColor clearColor]};
                    [result.text drawInRect:bounds withAttributes:dic];
                }
            }
        }
        [NSGraphicsContext restoreGraphicsState];
        
        CGPDFContextEndPage (pdfContext);
        
        CFRelease(pageDictionary);
        CFRelease(boxData);
        image = nil;
    }
    
    CGPDFContextClose(pdfContext);
    CGContextRelease (pdfContext);
}

+(NSData*)convertStringsToPDFWithString:(NSArray *)strings{
    return convertStringsToPDF(strings);
}

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];
}

static NSData *convertStringsToPDF(NSArray<NSAttributedString *> *strings)
{
    NSMutableData *pdfData = [[NSMutableData alloc] init];
    CGDataConsumerRef consumer = CGDataConsumerCreateWithCFData((CFMutableDataRef)pdfData);

    CFMutableDictionaryRef myDictionary = CFDictionaryCreateMutable(NULL,
                                                                    0,
                                                                    &kCFTypeDictionaryKeyCallBacks,
                                                                    &kCFTypeDictionaryValueCallBacks);
    CFDictionarySetValue(myDictionary, kCGPDFContextCreator, CFSTR("PDF Reader Pro"));
    
    CGRect mediaBox = CGRectMake(0, 0, 595, 842);
    CGContextRef pdfContext = CGPDFContextCreate(consumer, &mediaBox, myDictionary);
    CGContextSetRGBFillColor(pdfContext, 1.0, 0.0, 0.0, 0.0);
    CGContextSetTextDrawingMode(pdfContext, kCGTextFill);
    
    CFRelease(myDictionary);
    
    for (NSAttributedString *attributeString in strings) {
        if (floor(NSAppKitVersionNumber) >= NSAppKitVersionNumber10_10) {
            NSTextStorage *textStorage = [[NSTextStorage alloc] initWithAttributedString:attributeString];
            NSLayoutManager *layoutManager = [[NSLayoutManager alloc] init];
            [textStorage addLayoutManager:layoutManager];
            while (YES) {
                NSSize contentSize = NSMakeSize(mediaBox.size.width-40,
                                                mediaBox.size.height-40);
                NSTextContainer *textContainer = [[NSTextContainer alloc] initWithSize:contentSize];
                [layoutManager addTextContainer:textContainer];
                NSRange rang = [layoutManager glyphRangeForTextContainer:textContainer];
                if (rang.length <= 0 ||
                    rang.length + rang.location > attributeString.length) {
                    break;
                }
                NSAttributedString *attStr = [attributeString attributedSubstringFromRange:rang];
                
                [NSGraphicsContext saveGraphicsState];
//                [NSGraphicsContext setCurrentContext:[NSGraphicsContext graphicsContextWithGraphicsPort:pdfContext flipped:NO]];
                [NSGraphicsContext setCurrentContext:[NSGraphicsContext graphicsContextWithCGContext:pdfContext flipped:NO]];
                CGContextBeginPage(pdfContext, &mediaBox);
                [attStr drawInRect:CGRectMake(20, 20,
                                              mediaBox.size.width-40,
                                              mediaBox.size.height-40)];
                CGContextEndPage(pdfContext);
                [NSGraphicsContext restoreGraphicsState];
            }
        } else {
            [NSGraphicsContext saveGraphicsState];
            [NSGraphicsContext setCurrentContext:[NSGraphicsContext graphicsContextWithCGContext:pdfContext flipped:NO]];
            CGContextBeginPage(pdfContext, &mediaBox);
            [attributeString drawInRect:CGRectMake(20, 20,
                                                   mediaBox.size.width-40,
                                                   mediaBox.size.height-40)];
            CGContextEndPage(pdfContext);
            [NSGraphicsContext restoreGraphicsState];
        }
    }
    
    CGPDFContextClose(pdfContext);
    CGContextRelease (pdfContext);
    
    return pdfData;
}

@end