class Studio:
EXPORT_FORMATS = ['HTML', 'ODT']
def __init__(self):
# i18n
languages = []
lc, encoding = locale.getdefaultlocale()
if lc:
languages = [lc]
languages += DEFAULT_LANGUAGES
gettext.bindtextdomain(OCRFEEDER_COMPACT_NAME, LOCALE_DIR)
gettext.textdomain(OCRFEEDER_COMPACT_NAME)
language = gettext.translation(OCRFEEDER_COMPACT_NAME, LOCALE_DIR,
languages = languages, fallback = True)
_ = language.gettext
self.title = OCRFEEDER_STUDIO_NAME
self.main_window = widgetPresenter.MainWindow()
self.main_window.setTitle(self.title)
cli_command_retriever = ArgsRetriever(sys.argv)
imgs = cli_command_retriever.getParams('--images')
self.configuration_manager = ConfigurationManager()
self.ocr_engines_manager = OcrEnginesManager(self.configuration_manager)
self.ocr_engines_manager.makeEnginesFromFolder(self.configuration_manager.user_engines_folder)
self.ocr_engines = self.ocr_engines_manager.ocr_engines
self.configuration_manager.loadConfiguration()
self.source_images_selector = SourceImagesSelector(imgs)
self.source_images_selector.connect('selection_changed', self.selectionChanged)
self.source_images_icon_view = SourceImagesSelectorIconView(self.source_images_selector)
self.source_images_icon_view.setDeleteCurrentPageFunction(self.deleteCurrentPage)
self.source_images_icon_view.show()
self.main_window.main_area_left.add_with_viewport(self.source_images_icon_view)
self.images_selectable_area = {}
self.images_dict = create_images_dict_from_liststore(self.source_images_selector.list_store)
self.source_images_controler = ImageReviewer_Controler(self.main_window.notebook, self.images_dict, self.source_images_icon_view, self.ocr_engines, self.configuration_manager, self.main_window.tripple_statusbar)
self.project_name = None
toolbar_callback_dict = {'detection': self.source_images_controler.performBoxDetection,
'export_to_odt': self.exportToOdt}
menubar_callback_dict = {'exit': self.quit, 'add_image': self.addImage, 'export_to_odt': self.exportToOdt, 'edit_page': self.choosePageSize,
'delete_page': self.deleteCurrentPage, 'export_dialog': self.exportDialog, 'add_folder': self.addFolder,
'import_pdf': self.importPdf, 'save_project': self.saveProject, 'save_project_as': self.saveProjectAs,
'open_project': self.openProject, 'append_project': self.appendProject,'clear': self.clear,
'unpaper': self.unpaper, 'preferences': self.preferences, 'about': self.about,
'ocr_engines': self.ocrEngines, 'zoom_in': self.zoomIn, 'zoom_out': self.zoomOut,
'zoom_fit': self.zoomFit, 'reset_zoom': self.resetZoom}
self.main_window.setHeader(menubar_callback_dict, toolbar_callback_dict)
self.main_window.setDestroyEvent(self.quit)
dirs = cli_command_retriever.getParams('--dir')
if dirs:
self.__addImagesToReviewer(lib.getImagesFromFolder(dirs[0]))
self.main_window.setHasImages(not self.source_images_selector.isEmpty())
def run(self):
gtk.main()
def addImage(self, widget):
file_open_dialog = widgetPresenter.FileDialog('open', file_filters = [(_('Images'), ['image/*'], [])])
response = file_open_dialog.run()
if response == gtk.RESPONSE_OK:
for file_name in file_open_dialog.get_filenames():
self.__addImagesToReviewer([file_name])
file_open_dialog.destroy()
def importPdf(self, widget):
file_open_dialog = widgetPresenter.FileDialog('open', file_filters = [(_('PDF'), ['application/pdf'], [])])
response = file_open_dialog.run()
if response == gtk.RESPONSE_OK:
for file_name in file_open_dialog.get_filenames():
folder = lib.convertPdfToImages(file_name, self.configuration_manager.getTemporaryDir())
self.__addImagesToReviewer(lib.getImagesFromFolder(folder))
file_open_dialog.destroy()
def addFolder(self, widget):
file_open_dialog = widgetPresenter.FileDialog('select-folder')
response = file_open_dialog.run()
if response == gtk.RESPONSE_OK:
for folder in file_open_dialog.get_filenames():
self.__addImagesToReviewer(lib.getImagesFromFolder(folder))
file_open_dialog.destroy()
def exportToHtml(self, widget = None):
self.source_images_controler.exportPagesToHtml(self.source_images_selector.getPixbufsSorted())
def exportToOdt(self, widget = None):
self.source_images_controler.exportPagesToOdt(self.source_images_selector.getPixbufsSorted())
def exportDialog(self, widget):
export_dialog = widgetPresenter.ExportDialog(_('Export pages'), self.EXPORT_FORMATS)
response = export_dialog.run()
if response == gtk.RESPONSE_ACCEPT:
format = export_dialog.getSelectedFormat().capitalize()
export_dialog.destroy()
exec('self.exportTo%s()' % format)
return None
export_dialog.destroy()
return None
def choosePageSize(self, widget):
self.source_images_controler.choosePageSize()
def deleteCurrentPage(self, widget):
delete_dialog = widgetPresenter.QuestionDialog(_('Are you sure you want to delete the current image?'))
response = delete_dialog.run()
if response == gtk.RESPONSE_YES:
self.source_images_controler.deleteCurrentPage()
self.source_images_icon_view.deleteCurrentSelection()
delete_dialog.destroy()
def __addImagesToReviewer(self, images):
for image in images:
pixbuf, image, iter = self.source_images_selector.addImage(image)
self.source_images_controler.addImage(pixbuf, image)
tree_path = self.source_images_selector.list_store.get_path(iter)
self.source_images_icon_view.select_path(tree_path)
def setProjectName(self, project_name):
self.project_name = project_name
project_title = os.path.splitext(os.path.basename(self.project_name))[0]
self.main_window.setTitle('%s - %s' % (self.title, project_title))
def saveProjectAs(self, widget = None):
file_name = self.source_images_controler.saveProjectAs()
if file_name:
self.setProjectName(file_name)
self.saveProject()
def saveProject(self, widget = None):
if self.project_name:
self.source_images_controler.saveProject(self.project_name)
else:
self.saveProjectAs()
def appendProject(self, widget = None):
self.__loadProject(False)
def openProject(self, widget = None):
self.__loadProject()
def __loadProject(self, clear_current = True):
project_title = self.source_images_controler.openProject(clear_current)
if project_title:
self.setProjectName(project_title)
def clear(self, widget = None):
self.source_images_controler.clear()
def unpaper(self, widget = None):
self.source_images_controler.unpaperTool()
def preferences(self, widget = None):
preferences_dialog = widgetPresenter.PreferencesDialog(self.configuration_manager, self.ocr_engines_manager.ocr_engines)
if preferences_dialog.run() == gtk.RESPONSE_ACCEPT:
preferences_dialog.saveToManager()
self.source_images_controler.updateFromConfiguration()
preferences_dialog.destroy()
def ocrEngines(self, widget = None):
ocr_dialog = widgetPresenter.OcrManagerDialog(self.ocr_engines_manager)
ocr_dialog.run()
ocr_dialog.destroy()
def enginesTool(self, widget = None):
pass
def selectionChanged(self, selector, is_empty):
self.main_window.setHasImages(not is_empty)
def about(self, widget = None):
from gnome import url_show
gtk.about_dialog_set_url_hook(lambda x,y,z: url_show(y), "")
about_dialog = widgetPresenter.CustomAboutDialog()
if about_dialog.run():
about_dialog.destroy()
def zoomIn(self, widget = None):
self.source_images_controler.zoomIn()
def zoomOut(self, widget = None):
self.source_images_controler.zoomOut()
def resetZoom(self, widget = None):
self.source_images_controler.resetZoom()
def zoomFit(self, widget = None):
self.source_images_controler.zoomFit()
def quit(self, widget = None, data = None):
if not self.project_name and not self.source_images_selector.isEmpty():
quit_dialog = widgetPresenter.QuestionDialog('<b>' + _("The project hasn't been saved.") + '</b>', gtk.BUTTONS_NONE)
quit_dialog.format_secondary_text(_('Do you want to save it before closing?'))
quit_dialog.add_buttons(_('Close anyway'), gtk.RESPONSE_NO, gtk.STOCK_CANCEL, gtk.RESPONSE_CANCEL, gtk.STOCK_SAVE_AS, gtk.RESPONSE_YES)
response = quit_dialog.run()
quit_dialog.destroy()
if response == gtk.RESPONSE_YES:
self.saveProjectAs()
self.__main_quit()
elif response == gtk.RESPONSE_NO:
quit_dialog.destroy()
self.__main_quit()
else:
self.__main_quit()
def __main_quit(self):
self.configuration_manager.configurationToXml()
gtk.main_quit()
class Studio:
EXPORT_FORMATS = ['HTML', 'ODT']
def __init__(self):
# i18n
languages = []
lc, encoding = locale.getdefaultlocale()
if lc:
languages = [lc]
languages += DEFAULT_LANGUAGES
gettext.bindtextdomain(OCRFEEDER_COMPACT_NAME, LOCALE_DIR)
gettext.textdomain(OCRFEEDER_COMPACT_NAME)
language = gettext.translation(OCRFEEDER_COMPACT_NAME,
LOCALE_DIR,
languages=languages,
fallback=True)
_ = language.gettext
self.title = OCRFEEDER_STUDIO_NAME
self.main_window = widgetPresenter.MainWindow()
self.main_window.setTitle(self.title)
cli_command_retriever = ArgsRetriever(sys.argv)
imgs = cli_command_retriever.getParams('--images')
self.configuration_manager = ConfigurationManager()
self.ocr_engines_manager = OcrEnginesManager(
self.configuration_manager)
self.ocr_engines_manager.makeEnginesFromFolder(
self.configuration_manager.user_engines_folder)
self.ocr_engines = self.ocr_engines_manager.ocr_engines
self.configuration_manager.loadConfiguration()
self.source_images_selector = SourceImagesSelector(imgs)
self.source_images_selector.connect('selection_changed',
self.selectionChanged)
self.source_images_icon_view = SourceImagesSelectorIconView(
self.source_images_selector)
self.source_images_icon_view.setDeleteCurrentPageFunction(
self.deleteCurrentPage)
self.source_images_icon_view.show()
self.main_window.main_area_left.add_with_viewport(
self.source_images_icon_view)
self.images_selectable_area = {}
self.images_dict = create_images_dict_from_liststore(
self.source_images_selector.list_store)
self.source_images_controler = ImageReviewer_Controler(
self.main_window.notebook, self.images_dict,
self.source_images_icon_view, self.ocr_engines,
self.configuration_manager, self.main_window.tripple_statusbar)
self.project_name = None
toolbar_callback_dict = {
'detection': self.source_images_controler.performBoxDetection,
'export_to_odt': self.exportToOdt
}
menubar_callback_dict = {
'exit': self.quit,
'add_image': self.addImage,
'export_to_odt': self.exportToOdt,
'edit_page': self.choosePageSize,
'delete_page': self.deleteCurrentPage,
'export_dialog': self.exportDialog,
'add_folder': self.addFolder,
'import_pdf': self.importPdf,
'save_project': self.saveProject,
'save_project_as': self.saveProjectAs,
'open_project': self.openProject,
'append_project': self.appendProject,
'clear': self.clear,
'unpaper': self.unpaper,
'preferences': self.preferences,
'about': self.about,
'ocr_engines': self.ocrEngines,
'zoom_in': self.zoomIn,
'zoom_out': self.zoomOut,
'zoom_fit': self.zoomFit,
'reset_zoom': self.resetZoom
}
self.main_window.setHeader(menubar_callback_dict,
toolbar_callback_dict)
self.main_window.setDestroyEvent(self.quit)
dirs = cli_command_retriever.getParams('--dir')
if dirs:
self.__addImagesToReviewer(lib.getImagesFromFolder(dirs[0]))
self.main_window.setHasImages(
not self.source_images_selector.isEmpty())
def run(self):
gtk.main()
def addImage(self, widget):
file_open_dialog = widgetPresenter.FileDialog('open',
file_filters=[
(_('Images'),
['image/*'], [])
])
response = file_open_dialog.run()
if response == gtk.RESPONSE_OK:
for file_name in file_open_dialog.get_filenames():
self.__addImagesToReviewer([file_name])
file_open_dialog.destroy()
def importPdf(self, widget):
file_open_dialog = widgetPresenter.FileDialog(
'open', file_filters=[(_('PDF'), ['application/pdf'], [])])
response = file_open_dialog.run()
if response == gtk.RESPONSE_OK:
for file_name in file_open_dialog.get_filenames():
folder = lib.convertPdfToImages(
file_name, self.configuration_manager.getTemporaryDir())
self.__addImagesToReviewer(lib.getImagesFromFolder(folder))
file_open_dialog.destroy()
def addFolder(self, widget):
file_open_dialog = widgetPresenter.FileDialog('select-folder')
response = file_open_dialog.run()
if response == gtk.RESPONSE_OK:
for folder in file_open_dialog.get_filenames():
self.__addImagesToReviewer(lib.getImagesFromFolder(folder))
file_open_dialog.destroy()
def exportToHtml(self, widget=None):
self.source_images_controler.exportPagesToHtml(
self.source_images_selector.getPixbufsSorted())
def exportToOdt(self, widget=None):
self.source_images_controler.exportPagesToOdt(
self.source_images_selector.getPixbufsSorted())
def exportDialog(self, widget):
export_dialog = widgetPresenter.ExportDialog(_('Export pages'),
self.EXPORT_FORMATS)
response = export_dialog.run()
if response == gtk.RESPONSE_ACCEPT:
format = export_dialog.getSelectedFormat().capitalize()
export_dialog.destroy()
exec('self.exportTo%s()' % format)
return None
export_dialog.destroy()
return None
def choosePageSize(self, widget):
self.source_images_controler.choosePageSize()
def deleteCurrentPage(self, widget):
delete_dialog = widgetPresenter.QuestionDialog(
_('Are you sure you want to delete the current image?'))
response = delete_dialog.run()
if response == gtk.RESPONSE_YES:
self.source_images_controler.deleteCurrentPage()
self.source_images_icon_view.deleteCurrentSelection()
delete_dialog.destroy()
def __addImagesToReviewer(self, images):
for image in images:
pixbuf, image, iter = self.source_images_selector.addImage(image)
self.source_images_controler.addImage(pixbuf, image)
tree_path = self.source_images_selector.list_store.get_path(iter)
self.source_images_icon_view.select_path(tree_path)
def setProjectName(self, project_name):
self.project_name = project_name
project_title = os.path.splitext(os.path.basename(
self.project_name))[0]
self.main_window.setTitle('%s - %s' % (self.title, project_title))
def saveProjectAs(self, widget=None):
file_name = self.source_images_controler.saveProjectAs()
if file_name:
self.setProjectName(file_name)
self.saveProject()
def saveProject(self, widget=None):
if self.project_name:
self.source_images_controler.saveProject(self.project_name)
else:
self.saveProjectAs()
def appendProject(self, widget=None):
self.__loadProject(False)
def openProject(self, widget=None):
self.__loadProject()
def __loadProject(self, clear_current=True):
project_title = self.source_images_controler.openProject(clear_current)
if project_title:
self.setProjectName(project_title)
def clear(self, widget=None):
self.source_images_controler.clear()
def unpaper(self, widget=None):
self.source_images_controler.unpaperTool()
def preferences(self, widget=None):
preferences_dialog = widgetPresenter.PreferencesDialog(
self.configuration_manager, self.ocr_engines_manager.ocr_engines)
if preferences_dialog.run() == gtk.RESPONSE_ACCEPT:
preferences_dialog.saveToManager()
self.source_images_controler.updateFromConfiguration()
preferences_dialog.destroy()
def ocrEngines(self, widget=None):
ocr_dialog = widgetPresenter.OcrManagerDialog(self.ocr_engines_manager)
ocr_dialog.run()
ocr_dialog.destroy()
def enginesTool(self, widget=None):
pass
def selectionChanged(self, selector, is_empty):
self.main_window.setHasImages(not is_empty)
def about(self, widget=None):
from gnome import url_show
gtk.about_dialog_set_url_hook(lambda x, y, z: url_show(y), "")
about_dialog = widgetPresenter.CustomAboutDialog()
if about_dialog.run():
about_dialog.destroy()
def zoomIn(self, widget=None):
self.source_images_controler.zoomIn()
def zoomOut(self, widget=None):
self.source_images_controler.zoomOut()
def resetZoom(self, widget=None):
self.source_images_controler.resetZoom()
def zoomFit(self, widget=None):
self.source_images_controler.zoomFit()
def quit(self, widget=None, data=None):
if not self.project_name and not self.source_images_selector.isEmpty():
quit_dialog = widgetPresenter.QuestionDialog(
'<b>' + _("The project hasn't been saved.") + '</b>',
gtk.BUTTONS_NONE)
quit_dialog.format_secondary_text(
_('Do you want to save it before closing?'))
quit_dialog.add_buttons(_('Close anyway'), gtk.RESPONSE_NO,
gtk.STOCK_CANCEL, gtk.RESPONSE_CANCEL,
gtk.STOCK_SAVE_AS, gtk.RESPONSE_YES)
response = quit_dialog.run()
quit_dialog.destroy()
if response == gtk.RESPONSE_YES:
self.saveProjectAs()
self.__main_quit()
elif response == gtk.RESPONSE_NO:
quit_dialog.destroy()
self.__main_quit()
else:
self.__main_quit()
def __main_quit(self):
self.configuration_manager.configurationToXml()
gtk.main_quit()