iOS 通過(guò)VisionKit從PDF中提取TXT

懶惰與近視是程序員進(jìn)步的第一生產(chǎn)力(他絕不會(huì)承認(rèn)是因?yàn)樨毟F)。

場(chǎng)景:小A想要展示一本PDF文檔鸣驱,通過(guò)WKWebView直接展示是首選方案吏口,因?yàn)楹?jiǎn)單方便。但是雕擂,因?yàn)樯頌槌绦騿T的驕傲啡邑,促使他使用了更高級(jí)的方式。
于是井赌,他這樣做:

- (void)viewDidLoad {
    [super viewDidLoad];
    // Do any additional setup after loading the view.
    [self.view addSubview:self.pdfView];
}

- (PDFView *)pdfView {
    if (!_pdfView) {
        _pdfView = [[PDFView alloc] initWithFrame:self.view.bounds];
        _pdfView.document = self.document;
        _pdfView.delegate = self;
    }
    return _pdfView;
}

- (PDFDocument *)document {
    if (!_document) {
        NSURL *url = [[NSBundle mainBundle] URLForResource:@"xxx" withExtension:@"pdf"];
        _document = [[PDFDocument alloc] initWithURL:url];
    }
    return _document;
}

一切都那么順利谤逼,他展示出了pdf文件每一頁(yè)的內(nèi)容。但是人的欲望總是難以得到滿足仇穗,雖然展示出來(lái)了數(shù)據(jù)流部,但是由于只是單純的展示PDF文件,在小小的iPhone上面纹坐,什么都顯得那么小枝冀,對(duì)于近視的小A來(lái)說(shuō),真的是螞蟻爬爬耘子。小A不禁想到果漾,如果可以使用百度小說(shuō)的閱讀器直接打開(kāi),那么就可以隨心所欲的改變字體大小以及排版結(jié)構(gòu)了谷誓,近不近視的完全不受影響嘛绒障。

所幸,PDFKit可以讓用戶獲取到每一頁(yè)的page內(nèi)容捍歪,然后通過(guò)page可以拿到內(nèi)容所展示的文字户辱,那么,就可以將每一頁(yè)的內(nèi)容進(jìn)行整合展示成txt文件糙臼,用任意你想用的程序進(jìn)行展示庐镐,做法也非常簡(jiǎn)單:

for (int i = 0; i < [self.document pageCount]; i++) {
        PDFPage *page = [self.document pageAtIndex:i];
        NSLog(@"--!!!-- content: %@", page.string);
//        NSLog(@"--!!!-- content: %@", page.attributedString);
    }

小A很開(kāi)心,因?yàn)樗梢院苁孢m的瀏覽他想要看到的內(nèi)容了弓摘,他不禁想到焚鹊,近視真的是人類進(jìn)步的階梯。他一直這樣開(kāi)心韧献,直到他看到文章出現(xiàn)了斷層末患。。

WTF锤窑?璧针?什么情況,正在生離死別的緊要關(guān)頭怎么就開(kāi)始你儂我儂了渊啰?

于是探橱,小A開(kāi)始了苦逼的數(shù)據(jù)校驗(yàn)申屹,試圖找到那只該死的『蟲子』。幸運(yùn)的是隧膏,原文件支持他很快的找到了缺失的那一章節(jié)哗讥,原來(lái)那一頁(yè)P(yáng)DF不是簡(jiǎn)單text,而是令人崩潰的image胞枕。杆煞。圖片轉(zhuǎn)文字,需要用的傳說(shuō)中的OCR技術(shù)腐泻,想到一系列的算法識(shí)別以及數(shù)據(jù)喂養(yǎng)决乎,小A頭大如斯,從現(xiàn)場(chǎng)景來(lái)看派桩,通過(guò)接入各大廠商的OCR收費(fèi)sdk是首選方案构诚,但是,貧窮的程序員怎么可以忍受嗟來(lái)之食铆惑?于是范嘱,他開(kāi)始關(guān)注相關(guān)PDF轉(zhuǎn)TXT的APP,果然天下沒(méi)有白吃的午餐鸭津,想要不花錢就進(jìn)行轉(zhuǎn)換彤侍,就有了轉(zhuǎn)換內(nèi)容大小的限制,一次只能轉(zhuǎn)化1M的內(nèi)容逆趋,小A望著47.5M的文件陷入了沉思。晒奕。

最終闻书,小A將目光投向了VisionKit,謝天謝地脑慧,強(qiáng)大的VisionKit可以直接進(jìn)行中文識(shí)別魄眉,貧窮的小A覺(jué)得一切都是最好的安排。于是誕生了下面的一段代碼:

        PDFPage *page = [self.document pageAtIndex:i];
        UIImage *image = [page thumbnailOfSize:[page boundsForBox:kPDFDisplayBoxMediaBox].size forBox:kPDFDisplayBoxMediaBox];
        CGImageRef img = image.CGImage;
        
        VNImageRequestHandler *handler = [[VNImageRequestHandler alloc] initWithCGImage:img options:nil];
        VNRecognizeTextRequest *textRequest = [[VNRecognizeTextRequest alloc] initWithCompletionHandler:^(VNRequest * _Nonnull request, NSError * _Nullable error) {
            
            if (error) {
                NSLog(@"--!!!-- error : %@", error);
            } else {
                NSArray<VNRecognizedTextObservation *> *result = request.results;
                // 處理結(jié)果
            }
        }];
        textRequest.recognitionLevel = VNRequestTextRecognitionLevelAccurate;
        textRequest.usesLanguageCorrection = NO;
        textRequest.recognitionLanguages = @[@"zh-Hans", @"en-US"];
//        textRequest.revision = VNRecognizeTextRequestRevision2;
        
        [handler performRequests:@[textRequest] error:nil];

在應(yīng)用過(guò)程中闷袒,小A發(fā)現(xiàn)可以通過(guò)recognitionLevel屬性來(lái)選擇識(shí)別的級(jí)別坑律,是快速識(shí)別或是精準(zhǔn)識(shí)別。usesLanguageCorrection用來(lái)確認(rèn)是否需要在識(shí)別過(guò)程中使用語(yǔ)言更正囊骤,如果關(guān)閉此選項(xiàng)晃择,對(duì)性能有較大的正向影響,但可能會(huì)影響識(shí)別結(jié)果的正確性也物。但是基本已經(jīng)不影響閱讀了宫屠。

至此,小A終于可以開(kāi)開(kāi)心心地隨意閱讀他想要閱讀的pdf文件了滑蚯。

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
  • 序言:七十年代末浪蹂,一起剝皮案震驚了整個(gè)濱河市抵栈,隨后出現(xiàn)的幾起案子,更是在濱河造成了極大的恐慌坤次,老刑警劉巖古劲,帶你破解...
    沈念sama閱讀 206,126評(píng)論 6 481
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現(xiàn)場(chǎng)離奇詭異缰猴,居然都是意外死亡产艾,警方通過(guò)查閱死者的電腦和手機(jī),發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 88,254評(píng)論 2 382
  • 文/潘曉璐 我一進(jìn)店門洛波,熙熙樓的掌柜王于貴愁眉苦臉地迎上來(lái)胰舆,“玉大人,你說(shuō)我怎么就攤上這事蹬挤「苛” “怎么了?”我有些...
    開(kāi)封第一講書人閱讀 152,445評(píng)論 0 341
  • 文/不壞的土叔 我叫張陵焰扳,是天一觀的道長(zhǎng)倦零。 經(jīng)常有香客問(wèn)我,道長(zhǎng)吨悍,這世上最難降的妖魔是什么扫茅? 我笑而不...
    開(kāi)封第一講書人閱讀 55,185評(píng)論 1 278
  • 正文 為了忘掉前任,我火速辦了婚禮育瓜,結(jié)果婚禮上葫隙,老公的妹妹穿的比我還像新娘。我一直安慰自己躏仇,他們只是感情好恋脚,可當(dāng)我...
    茶點(diǎn)故事閱讀 64,178評(píng)論 5 371
  • 文/花漫 我一把揭開(kāi)白布。 她就那樣靜靜地躺著焰手,像睡著了一般糟描。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上书妻,一...
    開(kāi)封第一講書人閱讀 48,970評(píng)論 1 284
  • 那天船响,我揣著相機(jī)與錄音,去河邊找鬼躲履。 笑死见间,一個(gè)胖子當(dāng)著我的面吹牛,可吹牛的內(nèi)容都是我干的崇呵。 我是一名探鬼主播缤剧,決...
    沈念sama閱讀 38,276評(píng)論 3 399
  • 文/蒼蘭香墨 我猛地睜開(kāi)眼,長(zhǎng)吁一口氣:“原來(lái)是場(chǎng)噩夢(mèng)啊……” “哼域慷!你這毒婦竟也來(lái)了荒辕?” 一聲冷哼從身側(cè)響起汗销,我...
    開(kāi)封第一講書人閱讀 36,927評(píng)論 0 259
  • 序言:老撾萬(wàn)榮一對(duì)情侶失蹤,失蹤者是張志新(化名)和其女友劉穎抵窒,沒(méi)想到半個(gè)月后弛针,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體,經(jīng)...
    沈念sama閱讀 43,400評(píng)論 1 300
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡李皇,尸身上長(zhǎng)有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 35,883評(píng)論 2 323
  • 正文 我和宋清朗相戀三年削茁,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片掉房。...
    茶點(diǎn)故事閱讀 37,997評(píng)論 1 333
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡茧跋,死狀恐怖,靈堂內(nèi)的尸體忽然破棺而出卓囚,到底是詐尸還是另有隱情瘾杭,我是刑警寧澤,帶...
    沈念sama閱讀 33,646評(píng)論 4 322
  • 正文 年R本政府宣布哪亿,位于F島的核電站粥烁,受9級(jí)特大地震影響,放射性物質(zhì)發(fā)生泄漏蝇棉。R本人自食惡果不足惜讨阻,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 39,213評(píng)論 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望篡殷。 院中可真熱鬧钝吮,春花似錦、人聲如沸板辽。這莊子的主人今日做“春日...
    開(kāi)封第一講書人閱讀 30,204評(píng)論 0 19
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽(yáng)戳气。三九已至,卻和暖如春巧鸭,著一層夾襖步出監(jiān)牢的瞬間瓶您,已是汗流浹背。 一陣腳步聲響...
    開(kāi)封第一講書人閱讀 31,423評(píng)論 1 260
  • 我被黑心中介騙來(lái)泰國(guó)打工纲仍, 沒(méi)想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留呀袱,地道東北人。 一個(gè)月前我還...
    沈念sama閱讀 45,423評(píng)論 2 352
  • 正文 我出身青樓郑叠,卻偏偏與公主長(zhǎng)得像夜赵,于是被迫代替她去往敵國(guó)和親。 傳聞我的和親對(duì)象是個(gè)殘疾皇子乡革,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 42,722評(píng)論 2 345

推薦閱讀更多精彩內(nèi)容