def __init__(self, app, parent_textEdit, dialog_list):
""" Show list of image files.
On select, Show a scaleable and scrollable image.
Can add a memo to image
The slider values range from 9 to 99 with intervals of 3.
"""
super(DialogCodeImage, self).__init__()
sys.excepthook = exception_handler
self.app = app
self.dialog_list = dialog_list
self.parent_textEdit = parent_textEdit
self.codes = []
self.categories = []
self.files = []
self.file_ = None
self.log = ""
self.scale = 1.0
self.selection = None
self.get_image_files()
self.get_codes_and_categories()
self.get_coded_areas()
QtWidgets.QDialog.__init__(self)
self.ui = Ui_Dialog_code_image()
self.ui.setupUi(self)
self.ui.splitter.setSizes([100, 300])
self.scene = QtWidgets.QGraphicsScene()
self.ui.graphicsView.setScene(self.scene)
self.ui.graphicsView.setDragMode(
QtWidgets.QGraphicsView.RubberBandDrag)
# Need this otherwise small images are centred on screen, and affect context menu position points
self.ui.graphicsView.setAlignment(QtCore.Qt.AlignLeft
| QtCore.Qt.AlignTop)
self.scene.installEventFilter(self)
font = 'font: ' + str(self.app.settings['fontsize']) + 'pt '
font += '"' + self.app.settings['font'] + '";'
self.setStyleSheet(font)
font = 'font: ' + str(self.app.settings['treefontsize']) + 'pt '
font += '"' + self.app.settings['font'] + '";'
self.ui.treeWidget.setStyleSheet(font)
self.ui.label_coder.setText("Coder: " + self.app.settings['codername'])
self.setWindowTitle("Image coding")
self.ui.horizontalSlider.valueChanged[int].connect(self.change_scale)
self.ui.pushButton_memo.setEnabled(False)
self.ui.pushButton_memo.pressed.connect(self.image_memo)
self.ui.pushButton_select.pressed.connect(self.select_image)
self.ui.checkBox_show_coders.stateChanged.connect(
self.show_or_hide_coders)
self.ui.treeWidget.setDragEnabled(True)
self.ui.treeWidget.setAcceptDrops(True)
self.ui.treeWidget.setDragDropMode(
QtWidgets.QAbstractItemView.InternalMove)
self.ui.treeWidget.viewport().installEventFilter(self)
self.ui.treeWidget.setContextMenuPolicy(Qt.CustomContextMenu)
self.ui.treeWidget.customContextMenuRequested.connect(self.tree_menu)
self.ui.treeWidget.itemClicked.connect(self.fill_code_label)
self.fill_tree()
def __init__(self, settings):
''' Show list of image files.
On select, Show a scaleable and scrollable image.
Can add a memo to image
The slider values range from 9 to 99 with intervals of 3
'''
sys.excepthook = exception_handler
self.settings = settings
self.codes = []
self.categories = []
self.files = []
self.file_ = None
self.log = ""
self.scale = 1.0
self.selection = None
self.get_image_files()
self.get_codes_categories()
self.get_coded_areas()
QtWidgets.QDialog.__init__(self)
self.ui = Ui_Dialog_code_image()
self.ui.setupUi(self)
self.ui.splitter.setSizes([30, 70])
self.scene = QtWidgets.QGraphicsScene()
self.ui.graphicsView.setScene(self.scene)
self.ui.graphicsView.setDragMode(
QtWidgets.QGraphicsView.RubberBandDrag)
# need this otherwise small images are centred on screen, and may affect context menu position points
self.ui.graphicsView.setAlignment(QtCore.Qt.AlignLeft
| QtCore.Qt.AlignTop)
self.scene.installEventFilter(self)
newfont = QtGui.QFont(settings['font'], settings['fontsize'],
QtGui.QFont.Normal)
self.setFont(newfont)
treefont = QtGui.QFont(settings['font'], settings['treefontsize'],
QtGui.QFont.Normal)
self.ui.treeWidget.setFont(treefont)
self.ui.label_coder.setText("Coder: " + settings['codername'])
self.setWindowTitle("Image coding")
self.ui.horizontalSlider.valueChanged[int].connect(self.change_scale)
self.ui.pushButton_memo.setEnabled(False)
self.ui.pushButton_memo.pressed.connect(self.image_memo)
self.ui.pushButton_select.pressed.connect(self.select_image)
self.ui.checkBox_show_coders.stateChanged.connect(
self.show_or_hide_coders)
self.ui.treeWidget.setDragEnabled(True)
self.ui.treeWidget.setAcceptDrops(True)
self.ui.treeWidget.setDragDropMode(
QtWidgets.QAbstractItemView.InternalMove)
self.ui.treeWidget.viewport().installEventFilter(self)
self.ui.treeWidget.setContextMenuPolicy(Qt.CustomContextMenu)
self.ui.treeWidget.customContextMenuRequested.connect(self.tree_menu)
self.ui.treeWidget.itemClicked.connect(self.fill_code_label)
self.fill_tree()
class DialogCodeImage(QtWidgets.QDialog):
''' View and code images. Create codes and categories. '''
settings = None
filename = None
pixmap = None
scene = None
files = []
file_ = None
codes = []
categories = []
selection = None # initial code rectangle point
scale = 1.0
code_areas = []
def __init__(self, settings):
''' Show list of image files.
On select, Show a scaleable and scrollable image.
Can add a memo to image
The slider values range from 9 to 99 with intervals of 3
'''
sys.excepthook = exception_handler
self.settings = settings
self.codes = []
self.categories = []
self.files = []
self.file_ = None
self.log = ""
self.scale = 1.0
self.selection = None
self.get_image_files()
self.get_codes_categories()
self.get_coded_areas()
QtWidgets.QDialog.__init__(self)
self.ui = Ui_Dialog_code_image()
self.ui.setupUi(self)
self.ui.splitter.setSizes([30, 70])
self.scene = QtWidgets.QGraphicsScene()
self.ui.graphicsView.setScene(self.scene)
self.ui.graphicsView.setDragMode(
QtWidgets.QGraphicsView.RubberBandDrag)
# need this otherwise small images are centred on screen, and may affect context menu position points
self.ui.graphicsView.setAlignment(QtCore.Qt.AlignLeft
| QtCore.Qt.AlignTop)
self.scene.installEventFilter(self)
newfont = QtGui.QFont(settings['font'], settings['fontsize'],
QtGui.QFont.Normal)
self.setFont(newfont)
treefont = QtGui.QFont(settings['font'], settings['treefontsize'],
QtGui.QFont.Normal)
self.ui.treeWidget.setFont(treefont)
self.ui.label_coder.setText("Coder: " + settings['codername'])
self.setWindowTitle("Image coding")
self.ui.horizontalSlider.valueChanged[int].connect(self.change_scale)
self.ui.pushButton_memo.setEnabled(False)
self.ui.pushButton_memo.pressed.connect(self.image_memo)
self.ui.pushButton_select.pressed.connect(self.select_image)
self.ui.checkBox_show_coders.stateChanged.connect(
self.show_or_hide_coders)
self.ui.treeWidget.setDragEnabled(True)
self.ui.treeWidget.setAcceptDrops(True)
self.ui.treeWidget.setDragDropMode(
QtWidgets.QAbstractItemView.InternalMove)
self.ui.treeWidget.viewport().installEventFilter(self)
self.ui.treeWidget.setContextMenuPolicy(Qt.CustomContextMenu)
self.ui.treeWidget.customContextMenuRequested.connect(self.tree_menu)
self.ui.treeWidget.itemClicked.connect(self.fill_code_label)
self.fill_tree()
def get_codes_categories(self):
''' Called from init, delete category/code '''
self.categories = []
cur = self.settings['conn'].cursor()
cur.execute(
"select name, catid, owner, date, memo, supercatid from code_cat")
result = cur.fetchall()
for row in result:
self.categories.append({
'name': row[0],
'catid': row[1],
'owner': row[2],
'date': row[3],
'memo': row[4],
'supercatid': row[5]
})
self.codes = []
cur = self.settings['conn'].cursor()
cur.execute(
"select name, memo, owner, date, cid, catid, color from code_name")
result = cur.fetchall()
for row in result:
self.codes.append({
'name': row[0],
'memo': row[1],
'owner': row[2],
'date': row[3],
'cid': row[4],
'catid': row[5],
'color': row[6]
})
def get_coded_areas(self):
''' Get the coded area details for the rectangles.
Called by init and by unmark '''
self.code_areas = []
sql = "select imid,id,x1, y1, width, height, memo, date, owner, cid from code_image"
cur = self.settings['conn'].cursor()
cur.execute(sql)
results = cur.fetchall()
for row in results:
self.code_areas.append({
'imid': row[0],
'id': row[1],
'x1': row[2],
'y1': row[3],
'width': row[4],
'height': row[5],
'memo': row[6],
'date': row[7],
'owner': row[8],
'cid': row[9]
})
def get_image_files(self):
self.files = []
cur = self.settings['conn'].cursor()
cur.execute(
"select name, id, memo, owner, date from source where imagepath is not Null order by name"
)
result = cur.fetchall()
for row in result:
self.files.append({
'name': row[0],
'id': row[1],
'memo': row[2],
'owner': row[3],
'date': row[4]
})
def fill_tree(self):
''' Fill tree widget, tope level items are main categories and unlinked codes '''
cats = deepcopy(self.categories)
codes = deepcopy(self.codes)
self.ui.treeWidget.clear()
self.ui.treeWidget.setColumnCount(3)
self.ui.treeWidget.setHeaderLabels(["Name", "Id", "Memo"])
self.ui.treeWidget.setColumnHidden(1, True)
self.ui.treeWidget.header().setSectionResizeMode(
QtWidgets.QHeaderView.ResizeToContents)
self.ui.treeWidget.header().setStretchLastSection(False)
# add top level categories
remove_list = []
for c in cats:
if c['supercatid'] is None:
memo = ""
if c['memo'] != "":
memo = "Memo"
top_item = QtWidgets.QTreeWidgetItem(
[c['name'], 'catid:' + str(c['catid']), memo])
top_item.setIcon(0, QtGui.QIcon("GUI/icon_cat.png"))
top_item.setToolTip(0, c['owner'] + "\n" + c['date'])
self.ui.treeWidget.addTopLevelItem(top_item)
remove_list.append(c)
for item in remove_list:
#try:
cats.remove(item)
#except Exception as e:
# print(e, item)
''' Add child categories. Look at each unmatched category, iterate through tree
to add as child, then remove matched categories from the list. '''
count = 0
while len(cats) > 0 or count < 10000:
remove_list = []
#logger.debug("cats:" + str(cats))
for c in cats:
it = QtWidgets.QTreeWidgetItemIterator(self.ui.treeWidget)
item = it.value()
while item: # while there is an item in the list
if item.text(1) == 'catid:' + str(c['supercatid']):
memo = ""
if c['memo'] != "":
memo = "Memo"
child = QtWidgets.QTreeWidgetItem(
[c['name'], 'catid:' + str(c['catid']), memo])
child.setIcon(0, QtGui.QIcon("GUI/icon_cat.png"))
child.setToolTip(0, c['owner'] + "\n" + c['date'])
item.addChild(child)
remove_list.append(c)
it += 1
item = it.value()
for item in remove_list:
cats.remove(item)
count += 1
# add unlinked codes as top level items
remove_items = []
for c in codes:
if c['catid'] is None:
memo = ""
if c['memo'] != "":
memo = "Memo"
top_item = QtWidgets.QTreeWidgetItem(
[c['name'], 'cid:' + str(c['cid']), memo])
top_item.setIcon(0, QtGui.QIcon("GUI/icon_code.png"))
top_item.setToolTip(0, c['owner'] + "\n" + c['date'])
top_item.setBackground(
0, QBrush(QtGui.QColor(c['color']), Qt.SolidPattern))
top_item.setFlags(Qt.ItemIsSelectable | Qt.ItemIsUserCheckable
| Qt.ItemIsEnabled | Qt.ItemIsDragEnabled)
self.ui.treeWidget.addTopLevelItem(top_item)
remove_items.append(c)
for item in remove_items:
codes.remove(item)
# add codes as children
for c in codes:
it = QtWidgets.QTreeWidgetItemIterator(self.ui.treeWidget)
item = it.value()
while item:
if item.text(1) == 'catid:' + str(c['catid']):
memo = ""
if c['memo'] != "":
memo = "Memo"
child = QtWidgets.QTreeWidgetItem(
[c['name'], 'cid:' + str(c['cid']), memo])
child.setBackground(
0, QBrush(QtGui.QColor(c['color']), Qt.SolidPattern))
child.setIcon(0, QtGui.QIcon("GUI/icon_code.png"))
child.setToolTip(0, c['owner'] + "\n" + c['date'])
child.setFlags(Qt.ItemIsSelectable | Qt.ItemIsUserCheckable
| Qt.ItemIsEnabled | Qt.ItemIsDragEnabled)
item.addChild(child)
c['catid'] = -1 # make unmatchable
it += 1
item = it.value()
self.ui.treeWidget.expandAll()
def select_image(self):
''' dialog of filenames is presented to the user.
The selected image file is then displayed for coding '''
ui = DialogSelectFile(self.files, "Select file to view", "single")
ok = ui.exec_()
if ok:
self.file_ = ui.get_selected()
self.load_image()
def load_image(self):
''' Add image to scene if it exists '''
#try:
source = self.settings['path'] + "/images/" + self.file_['name']
#except Exception as e:
# QtWidgets.QMessageBox.warning(None, "Image error", "Image file not found. %s\n" + str(e), self.file_['name'])
# logger.warning(str(e) + ". " + source)
image = QtGui.QImage(source)
if image.isNull():
QtWidgets.QMessageBox.warming(None, "Image Error",
"Cannot open %s." % source)
self.close()
logger.warning("Cannot open image: " + source)
return
items = list(self.scene.items())
for i in range(items.__len__()):
self.scene.removeItem(items[i])
self.setWindowTitle("Image: " + self.file_['name'])
self.ui.pushButton_memo.setEnabled(True)
self.pixmap = QtGui.QPixmap.fromImage(image)
pixmap_item = QtWidgets.QGraphicsPixmapItem(
QtGui.QPixmap.fromImage(image))
pixmap_item.setPos(0, 0)
self.scene.setSceneRect(
QtCore.QRectF(0, 0, self.pixmap.width(), self.pixmap.height()))
self.scene.addItem(pixmap_item)
self.ui.horizontalSlider.setValue(99)
self.draw_coded_areas()
def change_scale(self):
''' Resize image.
Also called by unmark, as all items need to be redrawn '''
if self.pixmap is None:
return
self.scale = (self.ui.horizontalSlider.value() + 1) / 100
height = self.scale * self.pixmap.height()
pixmap = self.pixmap.scaledToHeight(height,
QtCore.Qt.FastTransformation)
pixmap_item = QtWidgets.QGraphicsPixmapItem(pixmap)
pixmap_item.setPos(0, 0)
self.scene.clear()
self.scene.addItem(pixmap_item)
self.draw_coded_areas()
def show_or_hide_coders(self):
''' When checked call on draw_coded_areas to either show all coders codings,
otherwise only show current coder.
Change scale is called becuase all items need to be removed and then added to the
scene. pixmap being the firs item added then the codings.
'''
self.change_scale()
def draw_coded_areas(self):
''' draw coded areas with scaling.
Remove items first, as this is called after a coded area is unmarked '''
'''items = list(self.scene.items())
for i in range(items.__len__()):
if type(i) == QtWidgets.QGraphicsRectItem:
self.scene.removeItem(items[i])
self.ui.graphicsView.viewport().update()
self.scene.clear()'''
for item in self.code_areas:
if item['id'] == self.file_['id']:
tooltip = ""
for c in self.codes:
if c['cid'] == item['cid']:
tooltip = c['name'] + " (" + item['owner'] + ")"
x = item['x1'] * self.scale
y = item['y1'] * self.scale
width = item['width'] * self.scale
height = item['height'] * self.scale
rect_item = QtWidgets.QGraphicsRectItem(x, y, width, height)
rect_item.setPen(
QtGui.QPen(QtCore.Qt.red, 2, QtCore.Qt.DashLine))
rect_item.setToolTip(tooltip)
if self.ui.checkBox_show_coders.isChecked():
self.scene.addItem(rect_item)
else:
if item['owner'] == self.settings['codername']:
self.scene.addItem(rect_item)
def fill_code_label(self):
''' Fill code label with curently selected item's code name '''
current = self.ui.treeWidget.currentItem()
if current.text(1)[0:3] == 'cat':
self.ui.label_code.setText("NO CODE SELECTED")
return
self.ui.label_code.setText("Code: " + current.text(0))
def image_memo(self):
''' Create a memo for the image file '''
ui = DialogMemo(self.settings, "Memo for image " + self.file_['name'],
self.file_['memo']) # "id=" + str(self.file_['id']))
ui.exec_()
cur = self.settings['conn'].cursor()
cur.execute('update source set memo=? where id=?',
(ui.memo, self.file_['id']))
self.settings['conn'].commit()
self.file_['memo'] = ui.memo
def tree_menu(self, position):
''' Context menu for treewidget items.
Add, rename, memo, move or delete code or category. Change code color. '''
menu = QtWidgets.QMenu()
selected = self.ui.treeWidget.currentItem()
#print(selected.parent())
#index = self.ui.treeWidget.currentIndex()
ActionItemAddCode = menu.addAction("Add a new code")
ActionItemAddCategory = menu.addAction("Add a new category")
ActionItemRename = menu.addAction("Rename")
ActionItemEditMemo = menu.addAction("View or edit memo")
ActionItemDelete = menu.addAction("Delete")
if selected is not None and selected.text(1)[0:3] == 'cid':
ActionItemChangeColor = menu.addAction("Change code color")
action = menu.exec_(self.ui.treeWidget.mapToGlobal(position))
if selected is not None and selected.text(
1)[0:3] == 'cid' and action == ActionItemChangeColor:
self.change_code_color(selected)
if action == ActionItemAddCategory:
self.add_category()
if action == ActionItemAddCode:
self.add_code()
if selected is not None and action == ActionItemRename:
self.rename_category_or_code(selected)
if selected is not None and action == ActionItemEditMemo:
self.add_edit_code_memo(selected)
if selected is not None and action == ActionItemDelete:
self.delete_category_or_code(selected)
def eventFilter(self, object, event):
''' Using this event filter to identfiy treeWidgetItem drop events.
http://doc.qt.io/qt-5/qevent.html#Type-enum
QEvent::Drop 63 A drag and drop operation is completed (QDropEvent).
https://stackoverflow.com/questions/28994494/why-does-qtreeview-not-fire-a-drop-or-move-event-during-drag-and-drop
Also use eventFilter for QGraphicsView.
'''
if object is self.ui.treeWidget.viewport():
if event.type() == QtCore.QEvent.Drop:
item = self.ui.treeWidget.currentItem()
parent = self.ui.treeWidget.itemAt(event.pos())
self.item_moved_update_data(item, parent)
self.get_codes_categories()
self.fill_tree()
if object is self.scene:
#logger.debug(event.type(), type(event))
if type(event
) == QtWidgets.QGraphicsSceneMouseEvent and event.button(
) == 1: # left mouse
#logger.debug(event.type(), type(event))
if event.type() == QtCore.QEvent.GraphicsSceneMousePress:
p0 = event.buttonDownScenePos(1) # left mouse button
#logger.debug("rectangle press:" + str(p0.x()) + ", " + str(p0.y()))
self.selection = p0
return True
if event.type() == QtCore.QEvent.GraphicsSceneMouseRelease:
p1 = event.lastScenePos()
#logger.debug("rectangle release: " + str(p1.x()) +", " + str(p1.y()))
self.code_area(p1)
return True
if type(event
) == QtWidgets.QGraphicsSceneMouseEvent and event.button(
) == 2: # right mouse
if event.type() == QtCore.QEvent.GraphicsSceneMousePress:
p = event.buttonDownScenePos(2)
self.scene_context_menu(p)
return True
return False
def scene_context_menu(self, pos):
''' Scene context menu for unmarking coded areas and adding memos '''
# outside image area, no context menu
for item in self.scene.items():
if type(item) == QtWidgets.QGraphicsPixmapItem:
if pos.x() > item.boundingRect().width() or pos.y(
) > item.boundingRect().height():
self.selection = None
return
menu = QtWidgets.QMenu()
menu.addAction('Memo')
menu.addAction('Unmark')
global_pos = QtGui.QCursor.pos()
item = self.find_coded_areas_for_pos(pos)
# no coded area item in this mouse position
if item is None:
return
action = menu.exec_(global_pos)
if action is None:
return
if action.text() == 'Memo':
self.coded_area_memo(item)
if action.text() == 'Unmark':
self.unmark(item)
def find_coded_areas_for_pos(self, pos):
''' Find any coded areas for this position '''
for item in self.code_areas:
if item['id'] == self.file_['id']:
#print(pos, item['x1'], item['y1'], item['width'], item['height'])
if pos.x() >= item['x1'] * self.scale and pos.x() <= (item['x1'] + item['width']) * self.scale \
and pos.y() >= item['y1'] * self.scale and pos.y() <= (item['y1'] + item['height']) * self.scale:
#print(pos, item['x1'] * self.scale, item['y1'] * self.scale, item['width'] * self.scale, item['height'] * self.scale)
return item
return None
def coded_area_memo(self, item):
''' Add memo to this coded area '''
ui = DialogMemo(self.settings,
"Memo for coded area of " + self.file_['name'],
item['memo'])
ui.exec_()
memo = ui.memo
if memo != item['memo']:
item['memo'] = memo
cur = self.settings['conn'].cursor()
cur.execute('update code_image set memo=? where id=?',
(ui.memo, item['id']))
self.settings['conn'].commit()
def unmark(self, item):
''' Remove coded area '''
cur = self.settings['conn'].cursor()
cur.execute("delete from code_image where imid=?", [
item['imid'],
])
self.settings['conn'].commit()
self.get_coded_areas()
self.change_scale()
def code_area(self, p1):
''' Created coded area coordinates from mouse release.
The point and width and height mush be based on the original image size,
so add in scale factor '''
code_ = self.ui.treeWidget.currentItem()
if code_ is None:
return
if code_.text(1)[0:3] == 'cat':
return
cid = code_.text(1)[4:]
x = self.selection.x()
y = self.selection.y()
#print("x", x, "y", y, "scale", self.scale)
width = p1.x() - x
height = p1.y() - y
if width < 0:
x = x + width
width = abs(width)
if height < 0:
y = y + height
height = abs(height)
#print("SCALED x", x, "y", y, "w", width, "h", height)
# outside image area, do not code
for item in self.scene.items():
if type(item) == QtWidgets.QGraphicsPixmapItem:
if x + width > item.boundingRect().width(
) or y + height > item.boundingRect().height():
self.selection = None
return
x_unscaled = x / self.scale
y_unscaled = y / self.scale
width_unscaled = width / self.scale
height_unscaled = height / self.scale
#print("UNSCALED x", x, "y", y, "w", width, "h", height)
item = {
'imid': None,
'id': self.file_['id'],
'x1': x_unscaled,
'y1': y_unscaled,
'width': width_unscaled,
'height': height_unscaled,
'owner': self.settings['codername'],
'date': datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
'cid': cid,
'memo': ''
}
cur = self.settings['conn'].cursor()
cur.execute(
"insert into code_image (id,x1,y1,width,height,cid,memo,date,owner) values(?,?,?,?,?,?,?,?,?)",
(item['id'], item['x1'], item['y1'], item['width'], item['height'],
cid, item['memo'], item['date'], item['owner']))
self.settings['conn'].commit()
cur.execute("select last_insert_rowid()")
imid = cur.fetchone()[0]
item['imid'] = imid
self.code_areas.append(item)
rect_item = QtWidgets.QGraphicsRectItem(x, y, width, height)
rect_item.setPen(QtGui.QPen(QtCore.Qt.red, 2, QtCore.Qt.DashLine))
rect_item.setToolTip(code_.text(0))
self.scene.addItem(rect_item)
#self.scene.addRect(x, y, width, height, QtGui.QPen(QtCore.Qt.red, 2, QtCore.Qt.DashLine))
self.selection = None
def item_moved_update_data(self, item, parent):
''' called from drop event in treeWidget view port.
identify code or category to move.
Also merge codes if one code is dropped on another code. '''
# find the category in the list
if item.text(1)[0:3] == 'cat':
found = -1
for i in range(0, len(self.categories)):
if self.categories[i]['catid'] == int(item.text(1)[6:]):
found = i
if found == -1:
return
if parent is None:
self.categories[found]['supercatid'] = None
else:
if parent.text(1).split(':')[0] == 'cid':
# parent is code (leaf) cannot add child
return
supercatid = int(parent.text(1).split(':')[1])
if supercatid == self.categories[found]['catid']:
# something went wrong
logger.debug("supercatid== self.categories[found][catid]")
return
self.categories[found]['supercatid'] = supercatid
cur = self.settings['conn'].cursor()
cur.execute("update code_cat set supercatid=? where catid=?", [
self.categories[found]['supercatid'],
self.categories[found]['catid']
])
self.settings['conn'].commit()
# find the code in the list
if item.text(1)[0:3] == 'cid':
found = -1
for i in range(0, len(self.codes)):
if self.codes[i]['cid'] == int(item.text(1)[4:]):
found = i
if found == -1:
return
if parent is None:
self.codes[found]['catid'] = None
else:
if parent.text(1).split(':')[0] == 'cid':
# parent is code (leaf) cannot add child, but can merge
self.merge_codes(self.codes[found], parent)
return
catid = int(parent.text(1).split(':')[1])
self.codes[found]['catid'] = catid
cur = self.settings['conn'].cursor()
cur.execute("update code_name set catid=? where cid=?",
[self.codes[found]['catid'], self.codes[found]['cid']])
self.settings['conn'].commit()
def merge_codes(self, item, parent):
''' Merge code or category with another code or category.
Called by item_moved_update_data when a code is moved onto another code'''
msg = "Merge code: " + item['name'] + "\ninto code: " + parent.text(0)
reply = QtWidgets.QMessageBox.question(None, 'Merge codes', msg,
QtWidgets.QMessageBox.Yes,
QtWidgets.QMessageBox.No)
if reply == QtWidgets.QMessageBox.No:
return
cur = self.settings['conn'].cursor()
old_cid = item['cid']
new_cid = int(parent.text(1).split(':')[1])
try:
cur.execute("update code_text set cid=? where cid=?",
[new_cid, old_cid])
self.settings['conn'].commit()
except Exception as e:
e = str(e)
msg = "cannot merge codes, unmark overlapping text first.\n" + e
QtWidgets.QInformationDialog(None, "Cannot merge", msg)
return
cur.execute("delete from code_name where cid=?", [
old_cid,
])
self.settings['conn'].commit()
def add_code(self):
''' Use add_item dialog to get new code text.
Add_code_name dialog checks for duplicate code name.
New code is added to data and database '''
ui = DialogAddItemName(self.codes, "Add new code")
ui.exec_()
newCodeText = ui.get_new_name()
if newCodeText is None:
return
item = {
'name': newCodeText,
'memo': "",
'owner': self.settings['codername'],
'date': datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
'catid': None,
'color': '#F8E0E0'
}
cur = self.settings['conn'].cursor()
cur.execute(
"insert into code_name (name,memo,owner,date,catid,color) values(?,?,?,?,?,?)",
(item['name'], item['memo'], item['owner'], item['date'],
item['catid'], item['color']))
self.settings['conn'].commit()
cur.execute("select last_insert_rowid()")
cid = cur.fetchone()[0]
item['cid'] = cid
self.codes.append(item)
top_item = QtWidgets.QTreeWidgetItem(
[item['name'], 'cid:' + str(item['cid']), ""])
top_item.setIcon(0, QtGui.QIcon("GUI/icon_code.png"))
color = item['color']
top_item.setBackground(0, QBrush(QtGui.QColor(color), Qt.SolidPattern))
self.ui.treeWidget.addTopLevelItem(top_item)
self.ui.treeWidget.setCurrentItem(top_item)
def add_category(self):
''' When button pressed, add a new category.
Note: the addItem dialog does the checking for duplicate category names
Add the new category as a top level item '''
ui = DialogAddItemName(self.categories, "Category")
ui.exec_()
newCatText = ui.get_new_name()
if newCatText is None:
return
# add to database
item = {
'name': newCatText,
'cid': None,
'memo': "",
'owner': self.settings['codername'],
'date': datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
}
cur = self.settings['conn'].cursor()
cur.execute(
"insert into code_cat (name, memo, owner, date, supercatid) values(?,?,?,?,?)",
(item['name'], item['memo'], item['owner'], item['date'], None))
self.settings['conn'].commit()
cur.execute("select last_insert_rowid()")
catid = cur.fetchone()[0]
item['catid'] = catid
self.categories.append(item)
# update widget
top_item = QtWidgets.QTreeWidgetItem(
[item['name'], 'catid:' + str(item['catid']), ""])
top_item.setIcon(0, QtGui.QIcon("GUI/icon_cat.png"))
self.ui.treeWidget.addTopLevelItem(top_item)
def delete_category_or_code(self, selected):
''' '''
if selected.text(1)[0:3] == 'cat':
self.delete_category(selected)
return # avoid error as selected is now None
if selected.text(1)[0:3] == 'cid':
self.delete_code(selected)
def delete_code(self, selected):
''' Find code, remove from database, refresh and code_name data and fill treeWidget '''
# find the code_in the list, check to delete
found = -1
for i in range(0, len(self.codes)):
if self.codes[i]['cid'] == int(selected.text(1)[4:]):
found = i
if found == -1:
return
code_ = self.codes[found]
ui = DialogConfirmDelete("Code: " + selected.text(0))
ok = ui.exec_()
if not ok:
return
cur = self.settings['conn'].cursor()
cur.execute("delete from code_name where cid=?", [
code_['cid'],
])
cur.execute("delete from code_text where cid=?", [
code_['cid'],
])
self.settings['conn'].commit()
selected = None
self.get_codes_categories()
self.fill_tree()
def delete_category(self, selected):
''' Find category, remove from database, refresh categories and code data
and fill treeWidget '''
found = -1
for i in range(0, len(self.categories)):
if self.categories[i]['catid'] == int(selected.text(1)[6:]):
found = i
if found == -1:
return
category = self.categories[found]
ui = DialogConfirmDelete("Category: " + selected.text(0))
ok = ui.exec_()
if not ok:
return
cur = self.settings['conn'].cursor()
cur.execute("update code_name set catid=null where catid=?", [
category['catid'],
])
cur.execute("update code_cat set supercatid=null where catid = ?", [
category['catid'],
])
cur.execute("delete from code_cat where catid = ?", [
category['catid'],
])
self.settings['conn'].commit()
selected = None
self.get_codes_categories()
self.fill_tree()
def add_edit_code_memo(self, selected):
''' View and edit a memo '''
if selected.text(1)[0:3] == 'cid':
# find the code in the list
found = -1
for i in range(0, len(self.codes)):
if self.codes[i]['cid'] == int(selected.text(1)[4:]):
found = i
if found == -1:
return
ui = DialogMemo(self.settings,
"Memo for Code " + self.codes[found]['name'],
self.codes[found]['memo'])
ui.exec_()
memo = ui.memo
if memo == "":
selected.setData(2, QtCore.Qt.DisplayRole, "")
else:
selected.setData(2, QtCore.Qt.DisplayRole, "Memo")
# update codes list and database
if memo != self.codes[found]['memo']:
self.codes[found]['memo'] = memo
cur = self.settings['conn'].cursor()
cur.execute("update code_name set memo=? where cid=?",
(memo, self.codes[found]['cid']))
self.settings['conn'].commit()
if selected.text(1)[0:3] == 'cat':
# find the category in the list
found = -1
for i in range(0, len(self.categories)):
if self.categories[i]['catid'] == int(selected.text(1)[6:]):
found = i
if found == -1:
return
ui = DialogMemo(
self.settings,
"Memo for Category " + self.categories[found]['name'],
self.categories[found]['memo'])
ui.exec_()
memo = ui.memo
if memo == "":
selected.setData(2, QtCore.Qt.DisplayRole, "")
else:
selected.setData(2, QtCore.Qt.DisplayRole, "Memo")
# update codes list and database
if memo != self.categories[found]['memo']:
self.categories[found]['memo'] = memo
cur = self.settings['conn'].cursor()
cur.execute("update code_cat set memo=? where catid=?",
(memo, self.categories[found]['catid']))
self.settings['conn'].commit()
def rename_category_or_code(self, selected):
''' Rename a code or category. Checks that the proposed code or category name is
not currently in use. '''
if selected.text(1)[0:3] == 'cid':
new_text, ok = QtWidgets.QInputDialog.getText(
self, "Rename code", "New code name:",
QtWidgets.QLineEdit.Normal, selected.text(0))
if not ok or new_text == '':
return
# check that no other code has this text
for c in self.codes:
if c['name'] == new_text:
QtWidgets.QMessageBox.warning(
None, "Name in use",
new_text + " is already in use, choose another name ",
QtWidgets.QMessageBox.Ok)
return
# find the code in the list
found = -1
for i in range(0, len(self.codes)):
if self.codes[i]['cid'] == int(selected.text(1)[4:]):
found = i
if found == -1:
return
# update codes list and database
cur = self.settings['conn'].cursor()
cur.execute("update code_name set name=? where cid=?",
(new_text, self.codes[found]['cid']))
self.settings['conn'].commit()
self.codes[found]['name'] = new_text
selected.setData(0, QtCore.Qt.DisplayRole, new_text)
return
if selected.text(1)[0:3] == 'cat':
new_text, ok = QtWidgets.QInputDialog.getText(
self, "Rename category", "New category name:",
QtWidgets.QLineEdit.Normal, selected.text(0))
if not ok or new_text == '':
return
# check that no other category has this text
for c in self.categories:
if c['name'] == new_text:
msg = "This code name is already in use"
QtWidgets.QMessageBox.warning(None, "Duplicate code name",
msg,
QtWidgets.QMessageBox.Ok)
return
# find the category in the list
found = -1
for i in range(0, len(self.categories)):
if self.categories[i]['catid'] == int(selected.text(1)[6:]):
found = i
if found == -1:
return
# update category list and database
cur = self.settings['conn'].cursor()
cur.execute("update code_cat set name=? where catid=?",
(new_text, self.categories[found]['catid']))
self.settings['conn'].commit()
self.categories[found]['name'] = new_text
selected.setData(0, QtCore.Qt.DisplayRole, new_text)
def change_code_color(self, selected):
''' change the color of the currently selected code '''
cid = int(selected.text(1)[4:])
found = -1
for i in range(0, len(self.codes)):
if self.codes[i]['cid'] == cid:
found = i
if found == -1:
return
ui = DialogColorSelect(self.codes[found]['color'])
ok = ui.exec_()
if not ok:
return
new_color = ui.get_color()
if new_color is None:
return
#print(new_color)
selected.setBackground(
0, QBrush(QtGui.QColor(new_color), Qt.SolidPattern))
#update codes list and database
self.codes[found]['color'] = new_color
cur = self.settings['conn'].cursor()
cur.execute("update code_name set color=? where cid=?",
(self.codes[found]['color'], self.codes[found]['cid']))
self.settings['conn'].commit()
class DialogCodeImage(QtWidgets.QDialog):
""" View and code images. Create codes and categories. """
app = None
dialog_list = None
parent_textEdit = None
filename = None
pixmap = None
scene = None
files = []
file_ = None
codes = []
categories = []
selection = None # Initial code rectangle point
scale = 1.0
code_areas = []
def __init__(self, app, parent_textEdit, dialog_list):
""" Show list of image files.
On select, Show a scaleable and scrollable image.
Can add a memo to image
The slider values range from 9 to 99 with intervals of 3.
"""
super(DialogCodeImage, self).__init__()
sys.excepthook = exception_handler
self.app = app
self.dialog_list = dialog_list
self.parent_textEdit = parent_textEdit
self.codes = []
self.categories = []
self.files = []
self.file_ = None
self.log = ""
self.scale = 1.0
self.selection = None
self.get_image_files()
self.get_codes_and_categories()
self.get_coded_areas()
QtWidgets.QDialog.__init__(self)
self.ui = Ui_Dialog_code_image()
self.ui.setupUi(self)
self.setWindowFlags(self.windowFlags()
& ~QtCore.Qt.WindowContextHelpButtonHint)
self.ui.splitter.setSizes([100, 300])
self.scene = QtWidgets.QGraphicsScene()
self.ui.graphicsView.setScene(self.scene)
self.ui.graphicsView.setDragMode(
QtWidgets.QGraphicsView.RubberBandDrag)
# Need this otherwise small images are centred on screen, and affect context menu position points
self.ui.graphicsView.setAlignment(QtCore.Qt.AlignLeft
| QtCore.Qt.AlignTop)
self.scene.installEventFilter(self)
font = 'font: ' + str(self.app.settings['fontsize']) + 'pt '
font += '"' + self.app.settings['font'] + '";'
self.setStyleSheet(font)
font = 'font: ' + str(self.app.settings['treefontsize']) + 'pt '
font += '"' + self.app.settings['font'] + '";'
self.ui.treeWidget.setStyleSheet(font)
self.ui.label_coder.setText("Coder: " + self.app.settings['codername'])
self.setWindowTitle(_("Image coding"))
self.ui.horizontalSlider.valueChanged[int].connect(self.change_scale)
self.ui.pushButton_memo.setEnabled(False)
self.ui.pushButton_memo.pressed.connect(self.image_memo)
self.ui.pushButton_select.setContextMenuPolicy(Qt.CustomContextMenu)
self.ui.pushButton_select.customContextMenuRequested.connect(
self.select_image_menu)
self.ui.pushButton_select.pressed.connect(self.select_image)
self.ui.checkBox_show_coders.stateChanged.connect(
self.show_or_hide_coders)
self.ui.treeWidget.setDragEnabled(True)
self.ui.treeWidget.setAcceptDrops(True)
self.ui.treeWidget.setDragDropMode(
QtWidgets.QAbstractItemView.InternalMove)
self.ui.treeWidget.viewport().installEventFilter(self)
self.ui.treeWidget.setContextMenuPolicy(Qt.CustomContextMenu)
self.ui.treeWidget.customContextMenuRequested.connect(self.tree_menu)
self.ui.treeWidget.itemClicked.connect(self.fill_code_label)
try:
s0 = int(self.app.settings['dialogcodeimage_splitter0'])
s1 = int(self.app.settings['dialogcodeimage_splitter1'])
if s0 > 10 and s1 > 10:
self.ui.splitter.setSizes([s0, s1])
except:
pass
self.fill_tree()
def closeEvent(self, event):
""" Save dialog and splitter dimensions. """
self.app.settings['dialogcodeimage_w'] = self.size().width()
self.app.settings['dialogcodeimage_h'] = self.size().height()
sizes = self.ui.splitter.sizes()
self.app.settings['dialogcodeimage_splitter0'] = sizes[0]
self.app.settings['dialogcodeimage_splitter1'] = sizes[1]
def get_codes_and_categories(self):
""" Called from init, delete category/code, event_filter """
self.codes, self.categories = self.app.get_data()
def get_coded_areas(self):
""" Get the coded area details for the rectangles.
Called by init and by unmark. """
self.code_areas = []
sql = "select imid,id,x1, y1, width, height, memo, date, owner, cid from code_image"
cur = self.app.conn.cursor()
cur.execute(sql)
results = cur.fetchall()
for row in results:
self.code_areas.append({
'imid': row[0],
'id': row[1],
'x1': row[2],
'y1': row[3],
'width': row[4],
'height': row[5],
'memo': row[6],
'date': row[7],
'owner': row[8],
'cid': row[9]
})
def get_image_files(self):
""" Load the image file data. """
self.files = []
cur = self.app.conn.cursor()
sql = "select name, id, memo, owner, date, mediapath from source where "
sql += "substr(mediapath,1,7) = '/images' order by name"
cur.execute(sql)
result = cur.fetchall()
for row in result:
self.files.append({
'name': row[0],
'id': row[1],
'memo': row[2],
'owner': row[3],
'date': row[4],
'mediapath': row[5]
})
def fill_tree(self):
""" Fill tree widget, top level items are main categories and unlinked codes. """
cats = deepcopy(self.categories)
codes = deepcopy(self.codes)
self.ui.treeWidget.clear()
self.ui.treeWidget.setColumnCount(4)
self.ui.treeWidget.setHeaderLabels(
[_("Name"), _("Id"), _("Memo"),
_("Count")])
if self.app.settings['showids'] == 'False':
self.ui.treeWidget.setColumnHidden(1, True)
else:
self.ui.treeWidget.setColumnHidden(1, False)
self.ui.treeWidget.header().setSectionResizeMode(
QtWidgets.QHeaderView.ResizeToContents)
self.ui.treeWidget.header().setStretchLastSection(False)
# add top level categories
remove_list = []
for c in cats:
if c['supercatid'] is None:
memo = ""
if c['memo'] != "" and c['memo'] is not None:
memo = "Memo"
top_item = QtWidgets.QTreeWidgetItem(
[c['name'], 'catid:' + str(c['catid']), memo])
top_item.setToolTip(2, c['memo'])
self.ui.treeWidget.addTopLevelItem(top_item)
remove_list.append(c)
for item in remove_list:
#try:
cats.remove(item)
#except Exception as e:
# print(e, item)
''' Add child categories. Look at each unmatched category, iterate through tree
to add as child, then remove matched categories from the list. '''
count = 0
while len(cats) > 0 or count < 10000:
remove_list = []
#logger.debug("cats:" + str(cats))
for c in cats:
it = QtWidgets.QTreeWidgetItemIterator(self.ui.treeWidget)
item = it.value()
count2 = 0
while item and count2 < 10000: # while there is an item in the list
if item.text(1) == 'catid:' + str(c['supercatid']):
memo = ""
if c['memo'] != "" and c['memo'] is not None:
memo = "Memo"
child = QtWidgets.QTreeWidgetItem(
[c['name'], 'catid:' + str(c['catid']), memo])
child.setToolTip(2, c['memo'])
item.addChild(child)
remove_list.append(c)
it += 1
item = it.value()
count2 += 1
for item in remove_list:
cats.remove(item)
count += 1
# Add unlinked codes as top level items
remove_items = []
for c in codes:
if c['catid'] is None:
memo = ""
if c['memo'] != "" and c['memo'] is not None:
memo = "Memo"
top_item = QtWidgets.QTreeWidgetItem(
[c['name'], 'cid:' + str(c['cid']), memo])
top_item.setToolTip(2, c['memo'])
top_item.setBackground(
0, QBrush(QtGui.QColor(c['color']), Qt.SolidPattern))
top_item.setFlags(Qt.ItemIsSelectable | Qt.ItemIsUserCheckable
| Qt.ItemIsEnabled | Qt.ItemIsDragEnabled)
self.ui.treeWidget.addTopLevelItem(top_item)
remove_items.append(c)
for item in remove_items:
codes.remove(item)
# Add codes as children to categories
for c in codes:
it = QtWidgets.QTreeWidgetItemIterator(self.ui.treeWidget)
item = it.value()
count = 0
while item and count < 10000:
if item.text(1) == 'catid:' + str(c['catid']):
memo = ""
if c['memo'] != "" and c['memo'] is not None:
memo = "Memo"
child = QtWidgets.QTreeWidgetItem(
[c['name'], 'cid:' + str(c['cid']), memo])
child.setBackground(
0, QBrush(QtGui.QColor(c['color']), Qt.SolidPattern))
child.setToolTip(2, c['memo'])
child.setFlags(Qt.ItemIsSelectable | Qt.ItemIsUserCheckable
| Qt.ItemIsEnabled | Qt.ItemIsDragEnabled)
item.addChild(child)
c['catid'] = -1 # Make unmatchable
it += 1
item = it.value()
count += 1
self.ui.treeWidget.expandAll()
def fill_code_counts_in_tree(self):
""" Count instances of each code for current coder and in the selected file. """
if self.file_ is None:
return
cur = self.app.conn.cursor()
sql = "select count(cid) from code_image where cid=? and id=? and owner=?"
it = QtWidgets.QTreeWidgetItemIterator(self.ui.treeWidget)
item = it.value()
count = 0
while item and count < 10000:
if item.text(1)[0:4] == "cid:":
cid = str(item.text(1)[4:])
cur.execute(
sql,
[cid, self.file_['id'], self.app.settings['codername']])
result = cur.fetchone()
if result[0] > 0:
item.setText(3, str(result[0]))
else:
item.setText(3, "")
it += 1
item = it.value()
count += 1
def select_image_menu(self, position):
""" Context menu to select the next image alphabetically, or
to select the image that was most recently coded """
if len(self.files) < 2:
return
menu = QtWidgets.QMenu()
menu.setStyleSheet("QMenu {font-size:" +
str(self.app.settings['fontsize']) + "pt} ")
action_next = menu.addAction(_("Next file"))
action_latest = menu.addAction(_("File with latest coding"))
action = menu.exec_(self.ui.pushButton_select.mapToGlobal(position))
if action == action_next:
if self.file_ is None:
self.file_ = self.files[0]
self.load_image()
return
for i in range(0, len(self.files) - 1):
if self.file_ == self.files[i]:
found = self.files[i + 1]
self.file_ = found
self.load_image()
return
if action == action_latest:
sql = "SELECT id FROM code_image where owner=? order by date desc limit 1"
cur = self.app.conn.cursor()
cur.execute(sql, [
self.app.settings['codername'],
])
result = cur.fetchone()
if result is None:
return
for f in self.files:
if f['id'] == result[0]:
self.file_ = f
self.load_image()
return
def select_image(self):
""" A dialog of filenames is presented to the user.
The selected image file is then displayed for coding. """
if self.files == []:
return
ui = DialogSelectItems(self.app, self.files, _("Select file to view"),
"single")
ok = ui.exec_()
if ok:
self.file_ = ui.get_selected()
self.load_image()
def load_image(self):
""" Add image to scene if it exists. """
source = self.app.project_path + self.file_['mediapath']
image = QtGui.QImage(source)
if image.isNull():
QtWidgets.QMessageBox.warning(None, _("Image Error"),
_("Cannot open: ") + source)
self.close()
logger.warning("Cannot open image: " + source)
return
items = list(self.scene.items())
for i in range(items.__len__()):
self.scene.removeItem(items[i])
self.setWindowTitle(_("Image: ") + self.file_['name'])
self.ui.pushButton_memo.setEnabled(True)
self.pixmap = QtGui.QPixmap.fromImage(image)
pixmap_item = QtWidgets.QGraphicsPixmapItem(
QtGui.QPixmap.fromImage(image))
pixmap_item.setPos(0, 0)
self.scene.setSceneRect(
QtCore.QRectF(0, 0, self.pixmap.width(), self.pixmap.height()))
self.scene.addItem(pixmap_item)
self.ui.horizontalSlider.setValue(99)
self.draw_coded_areas()
self.fill_code_counts_in_tree()
def update_dialog_codes_and_categories(self):
""" Update code and category tree in DialogCodeImage, DialogCodeAV,
DialogCodeText, DialogReportCodes.
Not using isinstance for other classes as could not import the classes to test
against. There was an import error.
Using try except blocks for each instance, as instance may have been deleted.
"""
for d in self.dialog_list:
if str(type(d)) == "<class 'code_text.DialogCodeText'>":
try:
d.get_codes_and_categories()
d.fill_tree()
d.unlight()
d.highlight()
d.get_coded_text_update_eventfilter_tooltips()
except RuntimeError as e:
pass
if str(type(d)) == "<class 'view_av.DialogCodeAV'>":
try:
d.get_codes_and_categories()
d.fill_tree()
d.load_segments()
d.unlight()
d.highlight()
d.get_coded_text_update_eventfilter_tooltips()
except RuntimeError as e:
pass
if isinstance(d, DialogCodeImage):
try:
d.get_codes_and_categories()
d.fill_tree()
d.get_coded_areas()
d.draw_coded_areas()
except RuntimeError as e:
pass
if str(type(d)) == "<class 'reports.DialogReportCodes'>":
try:
d.get_data()
d.fill_tree()
except RuntimeError as e:
pass
def change_scale(self):
""" Resize image. Triggered by user change in slider.
Also called by unmark, as all items need to be redrawn. """
if self.pixmap is None:
return
self.scale = (self.ui.horizontalSlider.value() + 1) / 100
height = self.scale * self.pixmap.height()
pixmap = self.pixmap.scaledToHeight(height,
QtCore.Qt.FastTransformation)
pixmap_item = QtWidgets.QGraphicsPixmapItem(pixmap)
pixmap_item.setPos(0, 0)
self.scene.clear()
self.scene.addItem(pixmap_item)
self.draw_coded_areas()
def show_or_hide_coders(self):
""" When checked call on draw_coded_areas to either show all coders codings,
otherwise only show current coder.
Change scale is called becuase all items need to be removed and then added to the
scene. pixmap being the first item added then the codings.
"""
self.change_scale()
def draw_coded_areas(self):
""" Draw coded areas with scaling. This coder is shown in dashed rectangles.
Other coders are shown via dotline rectangles.
Remove items first, as this is called after a coded area is unmarked. """
if self.file_ is None:
return
for item in self.code_areas:
if item['id'] == self.file_['id']:
color = QtGui.QColor('#AA0000') # Default color
color = None
tooltip = ""
for c in self.codes:
if c['cid'] == item['cid']:
tooltip = c['name'] + " (" + item['owner'] + ")"
tooltip += "\nMemo: " + item['memo']
color = QtGui.QColor(c['color'])
x = item['x1'] * self.scale
y = item['y1'] * self.scale
width = item['width'] * self.scale
height = item['height'] * self.scale
rect_item = QtWidgets.QGraphicsRectItem(x, y, width, height)
rect_item.setPen(QtGui.QPen(color, 2, QtCore.Qt.DashLine))
rect_item.setToolTip(tooltip)
if item['owner'] == self.app.settings['codername']:
self.scene.addItem(rect_item)
if self.ui.checkBox_show_coders.isChecked(
) and item['owner'] != self.app.settings['codername']:
rect_item.setPen(QtGui.QPen(color, 2, QtCore.Qt.DotLine))
self.scene.addItem(rect_item)
def fill_code_label(self):
""" Fill code label with currently selected item's code name. """
current = self.ui.treeWidget.currentItem()
if current.text(1)[0:3] == 'cat':
self.ui.label_code.setText(_("NO CODE SELECTED"))
return
self.ui.label_code.setText(_("Code: ") + current.text(0))
# update background colour of label
for c in self.codes:
if current.text(0) == c['name']:
palette = self.ui.label_code.palette()
code_color = QtGui.QColor(c['color'])
palette.setColor(QtGui.QPalette.Window, code_color)
self.ui.label_code.setPalette(palette)
self.ui.label_code.setAutoFillBackground(True)
break
def image_memo(self):
""" Create a memo for the image file. """
ui = DialogMemo(self.app,
_("Memo for image ") + self.file_['name'],
self.file_['memo'])
ui.exec_()
cur = self.app.conn.cursor()
cur.execute('update source set memo=? where id=?',
(ui.memo, self.file_['id']))
self.app.conn.commit()
self.file_['memo'] = ui.memo
self.app.delete_backup = False
def tree_menu(self, position):
""" Context menu for treewidget items.
Add, rename, memo, move or delete code or category. Change code color. """
menu = QtWidgets.QMenu()
menu.setStyleSheet("QMenu {font-size:" +
str(self.app.settings['fontsize']) + "pt} ")
selected = self.ui.treeWidget.currentItem()
#print(selected.parent())
#index = self.ui.treeWidget.currentIndex()
ActionItemAddCode = menu.addAction(_("Add a new code"))
ActionItemAddCategory = menu.addAction(_("Add a new category"))
ActionItemRename = menu.addAction(_("Rename"))
ActionItemEditMemo = menu.addAction(_("View or edit memo"))
ActionItemDelete = menu.addAction(_("Delete"))
if selected is not None and selected.text(1)[0:3] == 'cid':
ActionItemChangeColor = menu.addAction(_("Change code color"))
action = menu.exec_(self.ui.treeWidget.mapToGlobal(position))
if selected is not None and selected.text(
1)[0:3] == 'cid' and action == ActionItemChangeColor:
self.change_code_color(selected)
if action == ActionItemAddCategory:
self.add_category()
if action == ActionItemAddCode:
self.add_code()
if selected is not None and action == ActionItemRename:
self.rename_category_or_code(selected)
if selected is not None and action == ActionItemEditMemo:
self.add_edit_code_memo(selected)
if selected is not None and action == ActionItemDelete:
self.delete_category_or_code(selected)
def eventFilter(self, object, event):
""" Using this event filter to identfiy treeWidgetItem drop events.
http://doc.qt.io/qt-5/qevent.html#Type-enum
QEvent::Drop 63 A drag and drop operation is completed (QDropEvent).
https://stackoverflow.com/questions/28994494/why-does-qtreeview-not-fire-a-drop-or-move-event-during-drag-and-drop
Also use eventFilter for QGraphicsView.
"""
if object is self.ui.treeWidget.viewport():
if event.type() == QtCore.QEvent.Drop:
item = self.ui.treeWidget.currentItem()
parent = self.ui.treeWidget.itemAt(event.pos())
self.item_moved_update_data(item, parent)
self.update_dialog_codes_and_categories()
if object is self.scene:
#logger.debug(event.type(), type(event))
if type(event
) == QtWidgets.QGraphicsSceneMouseEvent and event.button(
) == 1: # left mouse
#logger.debug(event.type(), type(event))
if event.type() == QtCore.QEvent.GraphicsSceneMousePress:
p0 = event.buttonDownScenePos(1) # left mouse button
#logger.debug("rectangle press:" + str(p0.x()) + ", " + str(p0.y()))
self.selection = p0
return True
if event.type() == QtCore.QEvent.GraphicsSceneMouseRelease:
p1 = event.lastScenePos()
#logger.debug("rectangle release: " + str(p1.x()) +", " + str(p1.y()))
self.code_area(p1)
return True
if type(event
) == QtWidgets.QGraphicsSceneMouseEvent and event.button(
) == 2: # right mouse
if event.type() == QtCore.QEvent.GraphicsSceneMousePress:
p = event.buttonDownScenePos(2)
self.scene_context_menu(p)
return True
return False
def scene_context_menu(self, pos):
""" Scene context menu for unmarking coded areas and adding memos. """
# outside image area, no context menu
for item in self.scene.items():
if type(item) == QtWidgets.QGraphicsPixmapItem:
if pos.x() > item.boundingRect().width() or pos.y(
) > item.boundingRect().height():
self.selection = None
return
menu = QtWidgets.QMenu()
menu.setStyleSheet("QMenu {font-size:" +
str(self.app.settings['fontsize']) + "pt} ")
menu.addAction(_('Memo'))
menu.addAction(_('Unmark'))
global_pos = QtGui.QCursor.pos()
item = self.find_coded_areas_for_pos(pos)
# no coded area item in this mouse position
if item is None:
return
action = menu.exec_(global_pos)
if action is None:
return
if action.text() == _('Memo'):
self.coded_area_memo(item)
self.app.delete_backup = False
if action.text() == _('Unmark'):
self.unmark(item)
self.app.delete_backup = False
def find_coded_areas_for_pos(self, pos):
""" Find any coded areas for this position AND for this coder.
param: pos
returns: None or coded item
"""
for item in self.code_areas:
if item['id'] == self.file_['id'] and item[
'owner'] == self.app.settings['codername']:
#print(pos, item['x1'], item['y1'], item['width'], item['height'])
if pos.x() >= item['x1'] * self.scale and pos.x() <= (item['x1'] + item['width']) * self.scale \
and pos.y() >= item['y1'] * self.scale and pos.y() <= (item['y1'] + item['height']) * self.scale:
#print(pos, item['x1'] * self.scale, item['y1'] * self.scale, item['width'] * self.scale, item['height'] * self.scale)
return item
return None
def coded_area_memo(self, item):
""" Add memo to this coded area. """
ui = DialogMemo(self.app,
_("Memo for coded area of ") + self.file_['name'],
item['memo'])
ui.exec_()
memo = ui.memo
if memo != item['memo']:
item['memo'] = memo
cur = self.app.conn.cursor()
cur.execute('update code_image set memo=? where imid=?',
(ui.memo, item['imid']))
self.app.conn.commit()
# re-draw to update memos in tooltips
self.draw_coded_areas()
def unmark(self, item):
""" Remove coded area. """
cur = self.app.conn.cursor()
cur.execute("delete from code_image where imid=?", [
item['imid'],
])
self.app.conn.commit()
self.get_coded_areas()
self.change_scale()
self.fill_code_counts_in_tree()
def code_area(self, p1):
""" Create coded area coordinates from mouse release.
The point and width and height mush be based on the original image size,
so add in scale factor. """
code_ = self.ui.treeWidget.currentItem()
if code_ is None:
return
if code_.text(1)[0:3] == 'cat':
return
cid = int(code_.text(1)[4:]) # must be integer
x = self.selection.x()
y = self.selection.y()
#print("x", x, "y", y, "scale", self.scale)
width = p1.x() - x
height = p1.y() - y
if width < 0:
x = x + width
width = abs(width)
if height < 0:
y = y + height
height = abs(height)
#print("SCALED x", x, "y", y, "w", width, "h", height)
# outside image area, do not code
for item in self.scene.items():
if type(item) == QtWidgets.QGraphicsPixmapItem:
if x + width > item.boundingRect().width(
) or y + height > item.boundingRect().height():
self.selection = None
return
x_unscaled = x / self.scale
y_unscaled = y / self.scale
width_unscaled = width / self.scale
height_unscaled = height / self.scale
#print("UNSCALED x", x, "y", y, "w", width, "h", height)
item = {
'imid': None,
'id': self.file_['id'],
'x1': x_unscaled,
'y1': y_unscaled,
'width': width_unscaled,
'height': height_unscaled,
'owner': self.app.settings['codername'],
'date': datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
'cid': cid,
'memo': ''
}
cur = self.app.conn.cursor()
cur.execute(
"insert into code_image (id,x1,y1,width,height,cid,memo,date,owner) values(?,?,?,?,?,?,?,?,?)",
(item['id'], item['x1'], item['y1'], item['width'], item['height'],
cid, item['memo'], item['date'], item['owner']))
self.app.conn.commit()
cur.execute("select last_insert_rowid()")
imid = cur.fetchone()[0]
item['imid'] = imid
self.code_areas.append(item)
rect_item = QtWidgets.QGraphicsRectItem(x, y, width, height)
color = None
for i in range(0, len(self.codes)):
if self.codes[i]['cid'] == int(cid):
color = QtGui.QColor(self.codes[i]['color'])
if color is None:
print("ERROR")
return
rect_item.setPen(QtGui.QPen(color, 2, QtCore.Qt.DashLine))
rect_item.setToolTip(code_.text(0))
self.scene.addItem(rect_item)
self.selection = None
self.app.delete_backup = False
self.fill_code_counts_in_tree()
def item_moved_update_data(self, item, parent):
""" Called from drop event in treeWidget view port.
identify code or category to move.
Also merge codes if one code is dropped on another code. """
# find the category in the list
if item.text(1)[0:3] == 'cat':
found = -1
for i in range(0, len(self.categories)):
if self.categories[i]['catid'] == int(item.text(1)[6:]):
found = i
if found == -1:
return
if parent is None:
self.categories[found]['supercatid'] = None
else:
if parent.text(1).split(':')[0] == 'cid':
# parent is code (leaf) cannot add child
return
supercatid = int(parent.text(1).split(':')[1])
if supercatid == self.categories[found]['catid']:
# something went wrong
logger.debug("supercatid== self.categories[found][catid]")
return
self.categories[found]['supercatid'] = supercatid
cur = self.app.conn.cursor()
cur.execute("update code_cat set supercatid=? where catid=?", [
self.categories[found]['supercatid'],
self.categories[found]['catid']
])
self.app.conn.commit()
self.update_dialog_codes_and_categories()
self.app.delete_backup = False
return
# Find the code in the list
if item.text(1)[0:3] == 'cid':
found = -1
for i in range(0, len(self.codes)):
if self.codes[i]['cid'] == int(item.text(1)[4:]):
found = i
if found == -1:
return
if parent is None:
self.codes[found]['catid'] = None
else:
if parent.text(1).split(':')[0] == 'cid':
# parent is code (leaf) cannot add child, but can merge
self.merge_codes(self.codes[found], parent)
return
catid = int(parent.text(1).split(':')[1])
self.codes[found]['catid'] = catid
cur = self.app.conn.cursor()
cur.execute("update code_name set catid=? where cid=?",
[self.codes[found]['catid'], self.codes[found]['cid']])
self.app.conn.commit()
self.update_dialog_codes_and_categories()
self.app.delete_backup = False
def merge_codes(self, item, parent):
""" Merge code or category with another code or category.
Called by item_moved_update_data when a code is moved onto another code. """
msg = _("Merge code: ") + item['name'] + " ==> " + parent.text(0)
#TODO might need to add font size
reply = QtWidgets.QMessageBox.question(None, _('Merge codes'), msg,
QtWidgets.QMessageBox.Yes,
QtWidgets.QMessageBox.No)
if reply == QtWidgets.QMessageBox.No:
return
cur = self.app.conn.cursor()
old_cid = item['cid']
new_cid = int(parent.text(1).split(':')[1])
try:
cur.execute("update code_image set cid=? where cid=?",
[new_cid, old_cid])
cur.execute("update code_av set cid=? where cid=?",
[new_cid, old_cid])
cur.execute("update code_text set cid=? where cid=?",
[new_cid, old_cid])
self.app.conn.commit()
except Exception as e:
e = str(e)
msg = _("Cannot merge codes. Unmark overlapping text.") + "\n" + e
QtWidgets.QInformationDialog(None, "Cannot merge", msg)
return
cur.execute("delete from code_name where cid=?", [
old_cid,
])
self.app.conn.commit()
self.parent_textEdit.append(msg)
self.update_dialog_codes_and_categories()
self.app.delete_backup = False
def add_code(self):
""" Use add_item dialog to get new code text.
Add_code_name dialog checks for duplicate code name.
New code is added to data and database. """
ui = DialogAddItemName(self.app, self.codes, _("Add new code"),
_("Code name"))
ui.exec_()
new_code_name = ui.get_new_name()
if new_code_name is None:
return
code_color = colors[randint(0, len(colors) - 1)]
item = {
'name': new_code_name,
'memo': "",
'owner': self.app.settings['codername'],
'date': datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
'catid': None,
'color': code_color
}
cur = self.app.conn.cursor()
cur.execute(
"insert into code_name (name,memo,owner,date,catid,color) values(?,?,?,?,?,?)",
(item['name'], item['memo'], item['owner'], item['date'],
item['catid'], item['color']))
self.app.conn.commit()
self.update_dialog_codes_and_categories()
self.parent_textEdit.append(_("New code: ") + item['name'])
self.app.delete_backup = False
def add_category(self):
""" Add a new category.
Note: the addItem dialog does the checking for duplicate category names
Add the new category as a top level item. """
ui = DialogAddItemName(self.app, self.categories, _("Category"),
_("Category name"))
ui.exec_()
newCatText = ui.get_new_name()
if newCatText is None:
return
# add to database
item = {
'name': newCatText,
'cid': None,
'memo': "",
'owner': self.app.settings['codername'],
'date': datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
}
cur = self.app.conn.cursor()
cur.execute(
"insert into code_cat (name, memo, owner, date, supercatid) values(?,?,?,?,?)",
(item['name'], item['memo'], item['owner'], item['date'], None))
self.app.conn.commit()
self.update_dialog_codes_and_categories()
self.parent_textEdit.append(_("New category: ") + item['name'])
self.app.delete_backup = False
def delete_category_or_code(self, selected):
""" Delete the selected category or code.
If category deleted, sub-level items are retained. """
if selected.text(1)[0:3] == 'cat':
self.delete_category(selected)
return # avoid error as selected is now None
if selected.text(1)[0:3] == 'cid':
self.delete_code(selected)
def delete_code(self, selected):
""" Find code, remove from database, refresh and code_name data and fill
treeWidget. """
# find the code_in the list, check to delete
found = -1
for i in range(0, len(self.codes)):
if self.codes[i]['cid'] == int(selected.text(1)[4:]):
found = i
if found == -1:
return
code_ = self.codes[found]
ui = DialogConfirmDelete(self.app, _("Code: ") + selected.text(0))
ok = ui.exec_()
if not ok:
return
self.parent_textEdit.append(_("Code deleted: ") + code_['name'])
cur = self.app.conn.cursor()
cur.execute("delete from code_name where cid=?", [
code_['cid'],
])
cur.execute("delete from code_image where cid=?", [
code_['cid'],
])
cur.execute("delete from code_av where cid=?", [
code_['cid'],
])
cur.execute("delete from code_text where cid=?", [
code_['cid'],
])
self.app.conn.commit()
selected = None
self.update_dialog_codes_and_categories()
self.app.delete_backup = False
def delete_category(self, selected):
""" Find category, remove from database, refresh categories and code data
and fill treeWidget. Sub-level items are retained. """
found = -1
for i in range(0, len(self.categories)):
if self.categories[i]['catid'] == int(selected.text(1)[6:]):
found = i
if found == -1:
return
category = self.categories[found]
ui = DialogConfirmDelete(self.app, _("Category: ") + selected.text(0))
ok = ui.exec_()
if not ok:
return
self.parent_textEdit.append(_("Category deleted: ") + category['name'])
cur = self.app.conn.cursor()
cur.execute("update code_name set catid=null where catid=?", [
category['catid'],
])
cur.execute("update code_cat set supercatid=null where catid = ?", [
category['catid'],
])
cur.execute("delete from code_cat where catid = ?", [
category['catid'],
])
self.app.conn.commit()
selected = None
self.update_dialog_codes_and_categories()
self.app.delete_backup = False
def add_edit_code_memo(self, selected):
""" View and edit a memo. """
if selected.text(1)[0:3] == 'cid':
found = -1
for i in range(0, len(self.codes)):
if self.codes[i]['cid'] == int(selected.text(1)[4:]):
found = i
if found == -1:
return
ui = DialogMemo(self.app,
_("Memo for Code ") + self.codes[found]['name'],
self.codes[found]['memo'])
ui.exec_()
memo = ui.memo
if memo == "":
selected.setData(2, QtCore.Qt.DisplayRole, "")
else:
selected.setData(2, QtCore.Qt.DisplayRole, _("Memo"))
# update codes list and database
if memo != self.codes[found]['memo']:
self.codes[found]['memo'] = memo
cur = self.app.conn.cursor()
cur.execute("update code_name set memo=? where cid=?",
(memo, self.codes[found]['cid']))
self.app.conn.commit()
self.app.delete_backup = False
if selected.text(1)[0:3] == 'cat':
# find the category in the list
found = -1
for i in range(0, len(self.categories)):
if self.categories[i]['catid'] == int(selected.text(1)[6:]):
found = i
if found == -1:
return
ui = DialogMemo(
self.app,
_("Memo for Category: ") + self.categories[found]['name'],
self.categories[found]['memo'])
ui.exec_()
memo = ui.memo
if memo == "":
selected.setData(2, QtCore.Qt.DisplayRole, "")
else:
selected.setData(2, QtCore.Qt.DisplayRole, _("Memo"))
# update codes list and database
if memo != self.categories[found]['memo']:
self.categories[found]['memo'] = memo
cur = self.app.conn.cursor()
cur.execute("update code_cat set memo=? where catid=?",
(memo, self.categories[found]['catid']))
self.app.conn.commit()
self.app.delete_backup = False
self.update_dialog_codes_and_categories()
def rename_category_or_code(self, selected):
""" Rename a code or category. Checks that the proposed code or category name is
not currently in use. """
if selected.text(1)[0:3] == 'cid':
new_name, ok = QtWidgets.QInputDialog.getText(
self, _("Rename code"), _("New code name:"),
QtWidgets.QLineEdit.Normal, selected.text(0))
if not ok or new_name == '':
return
# check that no other code has this text
for c in self.codes:
if c['name'] == new_name:
mb = QtWidgets.QMessageBox()
mb.setStyleSheet("* {font-size:" +
str(self.app.settings['fontsize']) +
"pt} ")
mb.setWindowTitle(_("Name in use"))
mb.setText(new_name + _(" Choose another name"))
mb.exec_()
return
# Find the code in the list
found = -1
for i in range(0, len(self.codes)):
if self.codes[i]['cid'] == int(selected.text(1)[4:]):
found = i
if found == -1:
return
# Update codes list and database
cur = self.app.conn.cursor()
cur.execute("update code_name set name=? where cid=?",
(new_name, self.codes[found]['cid']))
self.app.conn.commit()
old_name = self.codes[found]['name']
self.update_dialog_codes_and_categories()
#self.codes[found]['name'] = new_name
#selected.setData(0, QtCore.Qt.DisplayRole, new_name)
self.parent_textEdit.append(_("Code renamed: ") + \
old_name + " ==> " + new_name)
self.app.delete_backup = False
return
if selected.text(1)[0:3] == 'cat':
new_name, ok = QtWidgets.QInputDialog.getText(
self, _("Rename category"), _("New category name:"),
QtWidgets.QLineEdit.Normal, selected.text(0))
if not ok or new_name == '':
return
# check that no other category has this text
for c in self.categories:
if c['name'] == new_name:
msg = _("This category name is already in use")
mb = QtWidgets.QMessageBox()
mb.setStyleSheet("* {font-size:" +
str(self.app.settings['fontsize']) +
"pt} ")
mb.setWindowTitle(_("Duplicate category name"))
mb.setText(msg)
mb.exec_()
return
# find the category in the list
found = -1
for i in range(0, len(self.categories)):
if self.categories[i]['catid'] == int(selected.text(1)[6:]):
found = i
if found == -1:
return
# update category list and database
cur = self.app.conn.cursor()
cur.execute("update code_cat set name=? where catid=?",
(new_name, self.categories[found]['catid']))
self.app.conn.commit()
old_name = self.categories[found]['name']
#self.categories[found]['name'] = new_name
#selected.setData(0, QtCore.Qt.DisplayRole, new_name)
self.parent_textEdit.append(_("Category renamed from: ") + \
old_name + " ==> " + new_name)
self.update_dialog_codes_and_categories()
self.app.delete_backup = False
def change_code_color(self, selected):
""" Change the color of the currently selected code. """
cid = int(selected.text(1)[4:])
found = -1
for i in range(0, len(self.codes)):
if self.codes[i]['cid'] == cid:
found = i
if found == -1:
return
ui = DialogColorSelect(self.codes[found]['color'])
ok = ui.exec_()
if not ok:
return
new_color = ui.get_color()
if new_color is None:
return
selected.setBackground(
0, QBrush(QtGui.QColor(new_color), Qt.SolidPattern))
#update codes list and database
self.codes[found]['color'] = new_color
cur = self.app.conn.cursor()
cur.execute("update code_name set color=? where cid=?",
(self.codes[found]['color'], self.codes[found]['cid']))
self.app.conn.commit()
self.update_dialog_codes_and_categories()
self.app.delete_backup = False
