From f3808df0cdc4dfd5bdd0d558f404de5aeb911427 Mon Sep 17 00:00:00 2001 From: Andrey Golovizin Date: Thu, 28 Aug 2014 13:05:56 +0200 Subject: [PATCH] Add --skip option. --- pixelocr/gui/__init__.py | 9 ++++++++- pixelocr/gui/ocrengine.py | 4 ++-- 2 files changed, 10 insertions(+), 3 deletions(-) diff --git a/pixelocr/gui/__init__.py b/pixelocr/gui/__init__.py index 339a681..ec603c3 100644 --- a/pixelocr/gui/__init__.py +++ b/pixelocr/gui/__init__.py @@ -17,6 +17,7 @@ import sys import signal from threading import Thread +from argparse import ArgumentParser import sip sip.setapi('QString', 2) @@ -33,13 +34,19 @@ from .window import MainWindow from .ocrengine import OCREngine +parser = ArgumentParser(description='PixelOCR') +parser.add_argument('--skip', metavar='N', type=int, default=0) +parser.add_argument('filename') + + def main(): app = QApplication(sys.argv) QApplication.setOrganizationName("PixelOCR"); QApplication.setApplicationName("PixelOCR"); - ocr = OCREngine(sys.argv[1]) + args = parser.parse_args() + ocr = OCREngine(args.filename, skip=args.skip) app.aboutToQuit.connect(ocr.save_glyphdb) win = MainWindow(ocr) diff --git a/pixelocr/gui/ocrengine.py b/pixelocr/gui/ocrengine.py index f94b7f4..a2fb696 100644 --- a/pixelocr/gui/ocrengine.py +++ b/pixelocr/gui/ocrengine.py @@ -33,10 +33,10 @@ class OCREngine(QThread): unknownGlyph = signal([Glyph]) pageChanged = signal([Page]) - def __init__(self, dirname): + def __init__(self, dirname, skip=0): super().__init__() self.dirname = dirname - self.filenames = sorted(glob(path.join(dirname, '*.png'))) + self.filenames = sorted(glob(path.join(dirname, '*.png')))[skip:] self.glyphdb = GlyphDB(path.join(self.dirname, 'glyphdb.pickle')) self.help_queue = Queue()