def __init__(self, main_window, page_data, ocr_engines):
super(ImageReviewer, self).__init__()
self.configuration_manager = ConfigurationManager()
self.main_window = main_window
self.path_to_image = page_data.image_path
self.text_box_fill_color = (94, 156, 235, 150)
self.box_stroke_color = (94, 156, 235, 250)
self.image_box_fill_color = (0, 183, 0, 150)
self.selectable_boxes_area = BoxesArea(self.path_to_image)
self.selectable_boxes_area.connect('selected_box', self.selectedBox)
self.selectable_boxes_area.connect('removed_box', self.removedBox)
self.selectable_boxes_area.connect('updated_box', self.updatedBox)
self.selectable_boxes_area.connect('dragged_box', self.updatedBoxBounds)
self.selectable_boxes_area.connect('deselected_box',
self.deselectedBoxCb)
self.image_pixbuf = GdkPixbuf.Pixbuf.new_from_file(self.path_to_image)
self.set_position(500)
self.show()
self.ocr_engines = ocr_engines
self.editor = Editor(self.image_pixbuf, self.ocr_engines, self)
self.boxes_dict = {}
selectable_boxes_scrolled_window = Gtk.ScrolledWindow()
selectable_boxes_scrolled_window.get_accessible().set_name(
_('Selectable areas'))
selectable_boxes_scrolled_window.set_policy(Gtk.PolicyType.AUTOMATIC, Gtk.PolicyType.AUTOMATIC)
selectable_boxes_scrolled_window.add(self.selectable_boxes_area)
self.selectable_boxes_area.show()
selectable_boxes_scrolled_window.show()
self.pack1(selectable_boxes_scrolled_window, True, False)
self.pack2(self.editor.box_editor, True, False)
self.page = page_data
self.updatePageData(self.page)
selectable_boxes_scrolled_window.connect_after("size-allocate", self.zoomFitCb)
def __init__(self, main_window, page_data, ocr_engines):
super(ImageReviewer, self).__init__()
self.configuration_manager = ConfigurationManager()
self.main_window = main_window
self.path_to_image = page_data.image_path
self.text_box_fill_color = (94, 156, 235, 150)
self.box_stroke_color = (94, 156, 235, 250)
self.image_box_fill_color = (0, 183, 0, 150)
self.selectable_boxes_area = BoxesArea(self.path_to_image)
self.selectable_boxes_area.connect('selected_box', self.selectedBox)
self.selectable_boxes_area.connect('removed_box', self.removedBox)
self.selectable_boxes_area.connect('updated_box', self.updatedBox)
self.selectable_boxes_area.connect('dragged_box', self.updatedBoxBounds)
self.selectable_boxes_area.connect('deselected_box',
self.deselectedBoxCb)
self.image_pixbuf = GdkPixbuf.Pixbuf.new_from_file(self.path_to_image)
self.set_position(500)
self.show()
self.ocr_engines = ocr_engines
self.editor = Editor(self.image_pixbuf, self.ocr_engines, self)
self.boxes_dict = {}
selectable_boxes_scrolled_window = Gtk.ScrolledWindow()
selectable_boxes_scrolled_window.get_accessible().set_name(
_('Selectable areas'))
selectable_boxes_scrolled_window.set_policy(Gtk.PolicyType.AUTOMATIC, Gtk.PolicyType.AUTOMATIC)
selectable_boxes_scrolled_window.add(self.selectable_boxes_area)
self.selectable_boxes_area.show()
selectable_boxes_scrolled_window.show()
self.pack1(selectable_boxes_scrolled_window, True, False)
self.pack2(self.editor.box_editor, True, False)
self.page = page_data
self.updatePageData(self.page)
selectable_boxes_scrolled_window.connect_after("size-allocate", self.zoomFitCb)
class ImageReviewer(Gtk.HPaned):
def __init__(self, main_window, page_data, ocr_engines):
super(ImageReviewer, self).__init__()
self.configuration_manager = ConfigurationManager()
self.main_window = main_window
self.path_to_image = page_data.image_path
self.text_box_fill_color = (94, 156, 235, 150)
self.box_stroke_color = (94, 156, 235, 250)
self.image_box_fill_color = (0, 183, 0, 150)
self.selectable_boxes_area = BoxesArea(self.path_to_image)
self.selectable_boxes_area.connect('selected_box', self.selectedBox)
self.selectable_boxes_area.connect('removed_box', self.removedBox)
self.selectable_boxes_area.connect('updated_box', self.updatedBox)
self.selectable_boxes_area.connect('dragged_box', self.updatedBoxBounds)
self.selectable_boxes_area.connect('deselected_box',
self.deselectedBoxCb)
self.image_pixbuf = GdkPixbuf.Pixbuf.new_from_file(self.path_to_image)
self.set_position(500)
self.show()
self.ocr_engines = ocr_engines
self.editor = Editor(self.image_pixbuf, self.ocr_engines, self)
self.boxes_dict = {}
selectable_boxes_scrolled_window = Gtk.ScrolledWindow()
selectable_boxes_scrolled_window.get_accessible().set_name(
_('Selectable areas'))
selectable_boxes_scrolled_window.set_policy(Gtk.PolicyType.AUTOMATIC, Gtk.PolicyType.AUTOMATIC)
selectable_boxes_scrolled_window.add(self.selectable_boxes_area)
self.selectable_boxes_area.show()
selectable_boxes_scrolled_window.show()
self.pack1(selectable_boxes_scrolled_window, True, False)
self.pack2(self.editor.box_editor, True, False)
self.page = page_data
self.updatePageData(self.page)
selectable_boxes_scrolled_window.connect_after("size-allocate", self.zoomFitCb)
def setTextFillColor(self, color):
self.text_box_fill_color = color
self.selectable_boxes_area.setAreaFillRgba(self.text_box_fill_color)
def setBoxesStrokeColor(self, color):
self.box_stroke_color = color
self.selectable_boxes_area.setAreaStrokeRgba(self.box_stroke_color)
def setImageFillColor(self, color):
self.image_box_fill_color = color
def addBoxEditor(self, box, data_box = None):
if not data_box:
data_box = DataBox()
data_box.setLanguage(self.configuration_manager.language)
data_box.updateBoundsFromBox(box)
self.boxes_dict[box] = data_box
self.editor.setBoxes(box, data_box)
def selectedBox(self, widget, box):
data_box = self.boxes_dict.get(box)
if data_box:
self.editor.setBoxes(box, data_box)
else:
self.addBoxEditor(box)
self.updateMainWindow()
def deselectedBoxCb(self, widget, box):
self.updateMainWindow()
def updatedBox(self, widget, box):
if self.editor.box == box:
self.editor.update(box)
def updatedBoxBounds(self, widget, box):
if self.editor.box == box:
self.editor.data_box.updateBoundsFromBox(box)
def removedBox(self, widget, box):
self.updateMainWindow()
if not self.boxes_dict.has_key(box):
return False
del self.boxes_dict[box]
if self.editor.box == box:
self.editor.updateDataBox(None)
self.editor.box = None
self.selectable_boxes_area.selectNextArea()
return False
def clear(self):
self.selectable_boxes_area.clearAreas()
self.editor.updateDataBox(None)
self.editor.box = None
self.boxes_dict.clear()
def applyTextColors(self):
self.selectable_boxes_area.fill_color_rgba = self.text_box_fill_color
self.selectable_boxes_area.stroke_color_rgba = self.box_stroke_color
def applyImageColors(self):
self.selectable_boxes_area.fill_color_rgba = self.image_box_fill_color
self.selectable_boxes_area.stroke_color_rgba = self.box_stroke_color
def addNewEditorsToAllBoxes(self):
self.editor_list = []
boxes = self.selectable_boxes_area.getAllAreas()
for box in boxes:
self.addBoxEditor(box)
def performOcrForSelectedBoxes(self):
selected_boxes = self.selectable_boxes_area.getSelectedAreas()
configured_engine = None
for engine, path in self.ocr_engines:
if engine.name == self.configuration_manager.favorite_engine:
configured_engine = engine
break
if not configured_engine:
return
for box in selected_boxes:
data_box = self.boxes_dict[box]
self.performOcrForDataBox(data_box, engine)
self.editor.updateDataBox(self.editor.data_box)
self.updateMainWindow()
def performOcrForDataBox(self, data_box, engine):
if engine.hasLanguages():
engine.setLanguage(data_box.getLanguage())
pixbuf_width = self.image_pixbuf.get_width()
pixbuf_height = self.image_pixbuf.get_height()
new_pixbuf_width = min(data_box.getWidth(), pixbuf_width)
new_pixbuf_height = min(data_box.getHeight(), pixbuf_height)
subpixbuf = self.image_pixbuf.new_subpixbuf(data_box.getX(),
data_box.getY(),
new_pixbuf_width,
new_pixbuf_height)
subpixbuf.x = data_box.getX()
subpixbuf.y = data_box.getY()
subpixbuf.width = pixbuf_width
image = graphics.convertPixbufToImage(subpixbuf)
layout_analysis = LayoutAnalysis(engine,
clean_text = self.configuration_manager.clean_text)
text = layout_analysis.readImage(image)
data_box.setText(text)
self.main_window.copy_to_clipboard_menu.set_sensitive(True)
self.main_window.spellchecker_menu.set_sensitive(True)
debug('Finished reading')
text_size = layout_analysis.getTextSizeFromImage(image,
self.page.resolution[1])
if text_size:
data_box.setFontSize(text_size)
def getTextFromBoxes(self, boxes):
text = ''
#@todo: Implement a proper way to get the boxes
# in reading order
boxes.reverse()
self.editor.saveDataBox()
if boxes:
text = self.page.getTextFromBoxes(
[self.boxes_dict.get(box) for box in boxes])
else:
if self.editor.box_editor.getType() == TEXT_TYPE:
text = self.box_editor.getText()
return text
def copyTextToClipboard(self):
selected_boxes = self.selectable_boxes_area.getSelectedAreas()
text = self.getTextFromBoxes(selected_boxes)
Gtk.Clipboard.get(Gdk.Atom.intern("CLIPBOARD", True)).set_text(text, -1)
def getAllText(self):
boxes = self.selectable_boxes_area.getAllAreas()
return self.getTextFromBoxes(boxes)
def __getAllDataBoxes(self):
boxes = []
for data_box in self.boxes_dict.values():
boxes.append((data_box.y, data_box))
boxes.sort()
boxes_sorted = []
for y, box in boxes:
boxes_sorted.append(box)
boxes = boxes_sorted
return boxes
def savePageData(self):
self.editor.saveDataBox()
self.page.data_boxes = self.__getAllDataBoxes()
return self.page
def updatePageData(self, page_data):
self.page = page_data
for data_box in self.page.data_boxes:
self.addDataBox(data_box)
def addDataBox(self, data_box):
dimensions = (int(data_box.x), int(data_box.y), int(data_box.width), int(data_box.height))
box = self.selectable_boxes_area.addArea(dimensions)
self.addBoxEditor(box, data_box)
def updateBackgroundImage(self, image_path):
self.path_to_image = self.page.image_path = image_path
if not os.path.exists(self.path_to_image):
return
try:
self.image_pixbuf = GdkPixbuf.Pixbuf.new_from_file(self.path_to_image)
except Exception, exception:
debug(exception.message)
return
self.selectable_boxes_area.setBackgroundImage(self.path_to_image)
class ImageReviewer(Gtk.Paned):
def __init__(self, main_window, page_data, ocr_engines):
super(ImageReviewer,
self).__init__(orientation=Gtk.Orientation.HORIZONTAL)
self.set_wide_handle(True)
self.configuration_manager = ConfigurationManager()
self.main_window = main_window
self.path_to_image = page_data.image_path
self.text_box_fill_color = (94, 156, 235, 150)
self.box_stroke_color = (94, 156, 235, 250)
self.image_box_fill_color = (0, 183, 0, 150)
self.selectable_boxes_area = BoxesArea(self.path_to_image)
self.selectable_boxes_area.connect('selected_box', self.selectedBox)
self.selectable_boxes_area.connect('removed_box', self.removedBox)
self.selectable_boxes_area.connect('updated_box', self.updatedBox)
self.selectable_boxes_area.connect('dragged_box',
self.updatedBoxBounds)
self.selectable_boxes_area.connect('deselected_box',
self.deselectedBoxCb)
self.image_pixbuf = GdkPixbuf.Pixbuf.new_from_file(self.path_to_image)
self.show()
self.ocr_engines = ocr_engines
self.editor = Editor(self.image_pixbuf, self.ocr_engines, self)
self.boxes_dict = {}
selectable_boxes_scrolled_window = Gtk.ScrolledWindow()
selectable_boxes_scrolled_window.get_accessible().set_name(
_('Selectable areas'))
selectable_boxes_scrolled_window.set_policy(Gtk.PolicyType.AUTOMATIC,
Gtk.PolicyType.AUTOMATIC)
selectable_boxes_scrolled_window.add(self.selectable_boxes_area)
self.selectable_boxes_area.show()
selectable_boxes_scrolled_window.show()
self.pack1(selectable_boxes_scrolled_window, True, False)
self.pack2(self.editor.box_editor, False, True)
self.page = page_data
self.updatePageData(self.page)
selectable_boxes_scrolled_window.connect_after("size-allocate",
self.zoomFitCb)
def setTextFillColor(self, color):
self.text_box_fill_color = color
self.selectable_boxes_area.setAreaFillRgba(self.text_box_fill_color)
def setBoxesStrokeColor(self, color):
self.box_stroke_color = color
self.selectable_boxes_area.setAreaStrokeRgba(self.box_stroke_color)
def setImageFillColor(self, color):
self.image_box_fill_color = color
def addBoxEditor(self, box, data_box=None):
if not data_box:
data_box = DataBox()
data_box.setLanguage(self.configuration_manager.language)
data_box.updateBoundsFromBox(box)
self.boxes_dict[box] = data_box
self.editor.setBoxes(box, data_box)
def selectedBox(self, widget, box):
data_box = self.boxes_dict.get(box)
if data_box:
self.editor.setBoxes(box, data_box)
else:
self.addBoxEditor(box)
self.updateMainWindow()
def deselectedBoxCb(self, widget, box):
self.updateMainWindow()
def updatedBox(self, widget, box):
if self.editor.box == box:
self.editor.update(box)
def updatedBoxBounds(self, widget, box):
if self.editor.box == box:
self.editor.data_box.updateBoundsFromBox(box)
def removedBox(self, widget, box):
self.updateMainWindow()
if not self.boxes_dict.has_key(box):
return False
del self.boxes_dict[box]
if self.editor.box == box:
self.editor.updateDataBox(None)
self.editor.box = None
self.selectable_boxes_area.selectNextArea()
return False
def clear(self):
self.selectable_boxes_area.clearAreas()
self.editor.updateDataBox(None)
self.editor.box = None
self.boxes_dict.clear()
def applyTextColors(self):
self.selectable_boxes_area.fill_color_rgba = self.text_box_fill_color
self.selectable_boxes_area.stroke_color_rgba = self.box_stroke_color
def applyImageColors(self):
self.selectable_boxes_area.fill_color_rgba = self.image_box_fill_color
self.selectable_boxes_area.stroke_color_rgba = self.box_stroke_color
def addNewEditorsToAllBoxes(self):
self.editor_list = []
boxes = self.selectable_boxes_area.getAllAreas()
for box in boxes:
self.addBoxEditor(box)
def performOcrForSelectedBoxes(self):
selected_boxes = self.selectable_boxes_area.getSelectedAreas()
configured_engine = None
for engine, path in self.ocr_engines:
if engine.name == self.configuration_manager.favorite_engine:
configured_engine = engine
break
if not configured_engine:
return
for box in selected_boxes:
data_box = self.boxes_dict[box]
self.performOcrForDataBox(data_box, engine)
self.editor.updateDataBox(self.editor.data_box)
self.updateMainWindow()
def performOcrForDataBox(self, data_box, engine):
if engine.hasLanguages():
engine.setLanguage(data_box.getLanguage())
pixbuf_width = self.image_pixbuf.get_width()
pixbuf_height = self.image_pixbuf.get_height()
new_pixbuf_width = min(data_box.getWidth(), pixbuf_width)
new_pixbuf_height = min(data_box.getHeight(), pixbuf_height)
subpixbuf = self.image_pixbuf.new_subpixbuf(data_box.getX(),
data_box.getY(),
new_pixbuf_width,
new_pixbuf_height)
subpixbuf.x = data_box.getX()
subpixbuf.y = data_box.getY()
subpixbuf.width = pixbuf_width
image = graphics.convertPixbufToImage(subpixbuf)
layout_analysis = LayoutAnalysis(
engine, clean_text=self.configuration_manager.clean_text)
text = layout_analysis.readImage(image)
data_box.setText(text)
self.main_window.copy_to_clipboard_menu.set_sensitive(True)
self.main_window.spellchecker_menu.set_sensitive(True)
debug('Finished reading')
text_size = layout_analysis.getTextSizeFromImage(
image, self.page.resolution[1])
if text_size:
data_box.setFontSize(text_size)
def getTextFromBoxes(self, boxes):
text = ''
#@todo: Implement a proper way to get the boxes
# in reading order
boxes.reverse()
self.editor.saveDataBox()
if boxes:
text = self.page.getTextFromBoxes(
[self.boxes_dict.get(box) for box in boxes])
else:
if self.editor.box_editor.getType() == TEXT_TYPE:
text = self.box_editor.getText()
return text
def copyTextToClipboard(self):
selected_boxes = self.selectable_boxes_area.getSelectedAreas()
text = self.getTextFromBoxes(selected_boxes)
Gtk.Clipboard.get(Gdk.Atom.intern("CLIPBOARD",
True)).set_text(text, -1)
def getAllText(self):
boxes = self.selectable_boxes_area.getAllAreas()
return self.getTextFromBoxes(boxes)
def __getAllDataBoxes(self):
boxes = []
for data_box in self.boxes_dict.values():
boxes.append((data_box.y, data_box))
boxes.sort()
boxes_sorted = []
for y, box in boxes:
boxes_sorted.append(box)
boxes = boxes_sorted
return boxes
def savePageData(self):
self.editor.saveDataBox()
self.page.data_boxes = self.__getAllDataBoxes()
return self.page
def updatePageData(self, page_data):
self.page = page_data
for data_box in self.page.data_boxes:
self.addDataBox(data_box)
def addDataBox(self, data_box):
dimensions = (int(data_box.x), int(data_box.y), int(data_box.width),
int(data_box.height))
box = self.selectable_boxes_area.addArea(dimensions)
self.addBoxEditor(box, data_box)
def updateBackgroundImage(self, image_path):
self.path_to_image = self.page.image_path = image_path
if not os.path.exists(self.path_to_image):
return
try:
self.image_pixbuf = GdkPixbuf.Pixbuf.new_from_file(
self.path_to_image)
except Exception, exception:
debug(exception.message)
return
self.selectable_boxes_area.setBackgroundImage(self.path_to_image)
