def __init__(self, **kw):
super(KeywordsEditor, self).__init__()
self.config_store = QtWidgets.QApplication.instance().config_store
self.league_table = {}
for keyword, score in self.config_store.get('descriptive', 'keywords',
{}).items():
if isinstance(score, int):
# old style keyword list
self.league_table[keyword] = date.min.isoformat(), score // 50
else:
# new style keyword list
self.league_table[keyword] = score
layout = QtWidgets.QHBoxLayout()
layout.setContentsMargins(0, 0, 0, 0)
self.setLayout(layout)
# line edit box
self.edit = SingleLineEdit(**kw)
layout.addWidget(self.edit)
# favourites drop down
self.favourites = ComboBox()
self.favourites.addItem(translate('DescriptiveTab', '<favourites>'))
self.favourites.setFixedWidth(
self.favourites.minimumSizeHint().width())
self.update_favourites()
self.favourites.currentIndexChanged.connect(self.add_favourite)
layout.addWidget(self.favourites)
# adopt child widget methods and signals
self.get_value = self.edit.get_value
self.set_value = self.edit.set_value
self.set_multiple = self.edit.set_multiple
self.is_multiple = self.edit.is_multiple
self.editingFinished = self.edit.editingFinished
def __init__(self, *arg, **kw):
super(KeywordsEditor, self).__init__(*arg, **kw)
self.config_store = QtWidgets.QApplication.instance().config_store
self.league_table = defaultdict(int)
for keyword, score in eval(
self.config_store.get('descriptive', 'keywords',
'{}')).items():
self.league_table[keyword] = score
layout = QtWidgets.QHBoxLayout()
layout.setContentsMargins(0, 0, 0, 0)
self.setLayout(layout)
# line edit box
self.edit = SingleLineEdit(spell_check=True)
layout.addWidget(self.edit)
# favourites drop down
self.favourites = ComboBox()
self.update_favourites()
self.favourites.setFixedWidth(self.favourites.title_width())
self.favourites.currentIndexChanged.connect(self.add_favourite)
layout.addWidget(self.favourites)
# adopt child widget methods and signals
self.get_value = self.edit.get_value
self.set_value = self.edit.set_value
self.set_multiple = self.edit.set_multiple
self.is_multiple = self.edit.is_multiple
self.editingFinished = self.edit.editingFinished
def __init__(self, image_list, parent=None):
super(PhotiniMap, self).__init__(parent)
self.app = QtWidgets.QApplication.instance()
self.image_list = image_list
name = self.__module__.split('.')[-1]
self.script_dir = pkg_resources.resource_filename(
'photini', 'data/' + name + '/')
self.drag_icon = QtGui.QPixmap(
os.path.join(self.script_dir, '../map_pin_grey.png'))
self.drag_hotspot = 11, 35
self.search_string = None
self.map_loaded = 0 # not loaded
self.marker_info = {}
self.map_status = {}
self.dropped_images = []
self.geocoder = self.get_geocoder()
self.gpx_ids = []
self.widgets = {}
self.setLayout(QtWidgets.QHBoxLayout())
## left side
left_side = QtWidgets.QGridLayout()
# latitude & longitude
self.widgets['latlon'] = LatLongDisplay(self.image_list)
left_side.addWidget(self.widgets['latlon'].label, 0, 0)
self.widgets['latlon'].changed.connect(self.new_coords)
left_side.addWidget(self.widgets['latlon'], 0, 1)
# altitude
label = QtWidgets.QLabel(translate('MapTabsAll', 'Altitude'))
label.setAlignment(Qt.AlignRight)
left_side.addWidget(label, 1, 0)
self.widgets['altitude'] = DoubleSpinBox()
self.widgets['altitude'].setSuffix(' m')
self.widgets['altitude'].new_value.connect(self.new_altitude)
left_side.addWidget(self.widgets['altitude'], 1, 1)
if hasattr(self.geocoder, 'get_altitude'):
self.widgets['get_altitude'] = QtWidgets.QPushButton(
translate('MapTabsAll', 'Get altitude from map'))
self.widgets['get_altitude'].clicked.connect(self.get_altitude)
left_side.addWidget(self.widgets['get_altitude'], 2, 1)
else:
self.widgets['get_altitude'] = None
# search
label = QtWidgets.QLabel(translate('MapTabsAll', 'Search'))
label.setAlignment(Qt.AlignRight)
left_side.addWidget(label, 3, 0)
self.widgets['search'] = ComboBox()
self.widgets['search'].setEditable(True)
self.widgets['search'].setInsertPolicy(QtWidgets.QComboBox.NoInsert)
self.widgets['search'].lineEdit().setPlaceholderText(
translate('MapTabsAll', '<new search>'))
self.widgets['search'].lineEdit().returnPressed.connect(self.search)
self.widgets['search'].activated.connect(self.goto_search_result)
self.clear_search()
self.widgets['search'].setEnabled(False)
left_side.addWidget(self.widgets['search'], 3, 1)
# search terms and conditions
for n, widget in enumerate(self.geocoder.search_terms()):
left_side.addWidget(widget, n+4, 0, 1, 2)
left_side.setColumnStretch(1, 1)
left_side.setRowStretch(7, 1)
# GPX importer
if self.app.gpx_importer:
button = QtWidgets.QPushButton(
translate('MapTabsAll', 'Load GPX file'))
button.clicked.connect(self.load_gpx)
left_side.addWidget(button, 8, 1)
self.widgets['set_from_gpx'] = QtWidgets.QPushButton(
translate('MapTabsAll', 'Set coords from GPX'))
self.widgets['set_from_gpx'].setEnabled(False)
self.widgets['set_from_gpx'].clicked.connect(self.set_from_gpx)
left_side.addWidget(self.widgets['set_from_gpx'], 9, 1)
self.widgets['clear_gpx'] = QtWidgets.QPushButton(
translate('MapTabsAll', 'Remove GPX data'))
self.widgets['clear_gpx'].setEnabled(False)
self.widgets['clear_gpx'].clicked.connect(self.clear_gpx)
left_side.addWidget(self.widgets['clear_gpx'], 10, 1)
self.layout().addLayout(left_side)
# map
# create handler for calls from JavaScript
self.call_handler = CallHandler(parent=self)
self.widgets['map'] = MapWebView(self.call_handler)
self.widgets['map'].drop_text.connect(self.drop_text)
self.widgets['map'].setAcceptDrops(False)
self.layout().addWidget(self.widgets['map'])
self.layout().setStretch(1, 1)
# other init
self.image_list.image_list_changed.connect(self.image_list_changed)
class KeywordsEditor(QtWidgets.QWidget):
def __init__(self, *arg, **kw):
super(KeywordsEditor, self).__init__(*arg, **kw)
self.config_store = QtWidgets.QApplication.instance().config_store
self.league_table = defaultdict(int)
for keyword, score in eval(
self.config_store.get('descriptive', 'keywords',
'{}')).items():
self.league_table[keyword] = score
layout = QtWidgets.QHBoxLayout()
layout.setContentsMargins(0, 0, 0, 0)
self.setLayout(layout)
# line edit box
self.edit = SingleLineEdit(spell_check=True)
layout.addWidget(self.edit)
# favourites drop down
self.favourites = ComboBox()
self.update_favourites()
self.favourites.setFixedWidth(self.favourites.title_width())
self.favourites.currentIndexChanged.connect(self.add_favourite)
layout.addWidget(self.favourites)
# adopt child widget methods and signals
self.get_value = self.edit.get_value
self.set_value = self.edit.set_value
self.set_multiple = self.edit.set_multiple
self.is_multiple = self.edit.is_multiple
self.editingFinished = self.edit.editingFinished
def update_favourites(self):
self.favourites.clear()
self.favourites.addItem(self.tr('<favourites>'))
keywords = list(self.league_table.keys())
keywords.sort(key=lambda x: self.league_table[x], reverse=True)
# limit size of league_table by deleting lowest scoring
if len(keywords) > 100:
threshold = self.league_table[keywords[100]]
for keyword in keywords:
if self.league_table[keyword] <= threshold:
del self.league_table[keyword]
# select highest scoring for drop down list
keywords = keywords[:20]
keywords.sort(key=lambda x: x.lower())
for keyword in keywords:
self.favourites.addItem(keyword)
self.favourites.set_dropdown_width()
def update_league_table(self, images):
for image in images:
value = list(
filter(lambda x: not x.startswith('flickr:photo_id'),
image.metadata.keywords or []))
if not value:
continue
for keyword in self.league_table:
self.league_table[keyword] = max(
self.league_table[keyword] - 1, -5)
for keyword in value:
self.league_table[keyword] = min(
self.league_table[keyword] + 10, 1000)
self.config_store.set('descriptive', 'keywords',
six.text_type(dict(self.league_table)))
self.update_favourites()
@QtCore.pyqtSlot(int)
@catch_all
def add_favourite(self, idx):
if idx <= 0:
return
self.favourites.setCurrentIndex(0)
new_value = self.favourites.itemText(idx)
current_value = self.get_value()
if current_value:
new_value = current_value + '; ' + new_value
self.set_value(new_value)
self.editingFinished.emit()
class PhotiniMap(QtWidgets.QWidget):
def __init__(self, image_list, parent=None):
super(PhotiniMap, self).__init__(parent)
self.app = QtWidgets.QApplication.instance()
self.image_list = image_list
name = self.__module__.split('.')[-1]
self.script_dir = pkg_resources.resource_filename(
'photini', 'data/' + name + '/')
self.drag_icon = QtGui.QPixmap(
os.path.join(self.script_dir, '../map_pin_grey.png'))
self.drag_hotspot = 11, 35
self.search_string = None
self.map_loaded = False
self.marker_info = {}
self.map_status = {}
self.dropped_images = []
self.geocoder = self.get_geocoder()
self.setLayout(QtWidgets.QHBoxLayout())
## left side
left_side = QtWidgets.QGridLayout()
# latitude & longitude
self.coords = LatLongDisplay(self.image_list)
left_side.addWidget(self.coords.label, 0, 0)
self.coords.changed.connect(self.new_coords)
left_side.addWidget(self.coords, 0, 1)
# altitude
label = QtWidgets.QLabel(translate('MapTabsAll', 'Altitude'))
label.setAlignment(Qt.AlignRight)
left_side.addWidget(label, 1, 0)
self.altitude = DoubleSpinBox()
self.altitude.setSuffix(' m')
self.altitude.new_value.connect(self.new_altitude)
left_side.addWidget(self.altitude, 1, 1)
if hasattr(self.geocoder, 'get_altitude'):
self.altitude_button = QtWidgets.QPushButton(
translate('MapTabsAll', 'Get altitude from map'))
self.altitude_button.clicked.connect(self.get_altitude)
left_side.addWidget(self.altitude_button, 2, 1)
else:
self.altitude_button = None
# search
label = QtWidgets.QLabel(translate('MapTabsAll', 'Search'))
label.setAlignment(Qt.AlignRight)
left_side.addWidget(label, 3, 0)
self.edit_box = ComboBox()
self.edit_box.setEditable(True)
self.edit_box.setInsertPolicy(QtWidgets.QComboBox.NoInsert)
self.edit_box.lineEdit().setPlaceholderText(
translate('MapTabsAll', '<new search>'))
self.edit_box.lineEdit().returnPressed.connect(self.search)
self.edit_box.activated.connect(self.goto_search_result)
self.clear_search()
self.edit_box.setEnabled(False)
left_side.addWidget(self.edit_box, 3, 1)
# search terms and conditions
for n, widget in enumerate(self.geocoder.search_terms()):
left_side.addWidget(widget, n + 4, 0, 1, 2)
left_side.setColumnStretch(1, 1)
left_side.setRowStretch(7, 1)
self.layout().addLayout(left_side)
# map
# create handler for calls from JavaScript
self.call_handler = CallHandler(parent=self)
self.map = MapWebView(self.call_handler)
self.map.drop_text.connect(self.drop_text)
self.map.setAcceptDrops(False)
self.layout().addWidget(self.map)
self.layout().setStretch(1, 1)
# other init
self.image_list.image_list_changed.connect(self.image_list_changed)
@QtSlot()
@catch_all
def image_list_changed(self):
self.redraw_markers()
self.coords.refresh()
self.update_altitude()
self.see_selection()
@QtSlot()
@catch_all
def initialise(self):
page = '''<!DOCTYPE html>
<html>
<head>
<meta charset="utf-8" />
<meta name="viewport" content="initial-scale=1.0, user-scalable=no" />
<style type="text/css">
html, body {{ height: 100%; margin: 0; padding: 0 }}
#mapDiv {{ position: relative; width: 100%; height: 100% }}
</style>
{initialize}
{head}
<script type="text/javascript" src="script.js"></script>
</head>
<body ondragstart="return false">
<div id="mapDiv"></div>
</body>
</html>'''
lat, lng = eval(
self.app.config_store.get('map', 'centre', '(51.0, 0.0)'))
zoom = int(eval(self.app.config_store.get('map', 'zoom', '11')))
if QtWebEngineWidgets:
initialize = ''' <script type="text/javascript"
src="qrc:///qtwebchannel/qwebchannel.js">
</script>
<script type="text/javascript">
var python;
function initialize()
{{
new QWebChannel(qt.webChannelTransport, doLoadMap);
}}
function doLoadMap(channel)
{{
python = channel.objects.python;
loadMap({lat}, {lng}, {zoom});
}}
</script>'''
else:
initialize = ''' <script type="text/javascript">
function initialize()
{{
loadMap({lat}, {lng}, {zoom});
}}
</script>'''
initialize = initialize.format(lat=lat, lng=lng, zoom=zoom)
page = page.format(initialize=initialize, head=self.get_head())
QtWidgets.QApplication.setOverrideCursor(Qt.WaitCursor)
self.map.setHtml(page, QtCore.QUrl.fromLocalFile(self.script_dir))
@catch_all
def initialize_finished(self):
QtWidgets.QApplication.restoreOverrideCursor()
self.map_loaded = True
self.edit_box.setEnabled(True)
self.map.setAcceptDrops(True)
self.redraw_markers()
def refresh(self):
self.image_list.set_drag_to_map(self.drag_icon, self.drag_hotspot)
if not self.map_loaded:
self.initialise()
return
lat, lng = eval(self.app.config_store.get('map', 'centre'))
zoom = int(eval(self.app.config_store.get('map', 'zoom')))
self.JavaScript('setView({!r},{!r},{:d})'.format(lat, lng, zoom))
def do_not_close(self):
return False
@catch_all
def new_status(self, status):
self.map_status.update(status)
for key in ('centre', 'zoom'):
if key in status:
self.app.config_store.set('map', key,
repr(self.map_status[key]))
@QtSlot(int, int, str)
@catch_all
def drop_text(self, x, y, text):
self.dropped_images = eval(text)
self.JavaScript('markerDrop({:d},{:d})'.format(x, y))
@catch_all
def marker_drop(self, lat, lng):
for path in self.dropped_images:
image = self.image_list.get_image(path)
image.metadata.latlong = lat, lng
self.dropped_images = []
self.redraw_markers()
self.coords.refresh()
self.see_selection()
@QtSlot()
@catch_all
def new_coords(self):
self.redraw_markers()
self.update_altitude()
self.see_selection()
@QtSlot(object)
@catch_all
def new_altitude(self, value):
for image in self.image_list.get_selected_images():
image.metadata.altitude = value
self.update_altitude()
def update_altitude(self):
images = self.image_list.get_selected_images()
if not images:
self.altitude.set_value(None)
self.altitude.setEnabled(False)
if self.altitude_button:
self.altitude_button.setEnabled(False)
return
values = []
for image in images:
value = image.metadata.altitude
if value not in values:
values.append(value)
if len(values) > 1:
self.altitude.set_multiple(choices=filter(None, values))
else:
self.altitude.set_value(values[0])
self.altitude.setEnabled(True)
if self.altitude_button:
self.altitude_button.setEnabled(bool(self.coords.get_value()))
def see_selection(self):
locations = []
for image in self.image_list.get_selected_images():
latlong = image.metadata.latlong
if not latlong:
continue
location = [latlong['lat'], latlong['lon']]
if location not in locations:
locations.append(location)
if not locations:
return
self.JavaScript('fitPoints({})'.format(repr(locations)))
@QtSlot(list)
@catch_all
def new_selection(self, selection):
self.redraw_markers()
self.coords.refresh()
self.update_altitude()
self.see_selection()
def redraw_markers(self):
if not self.map_loaded:
return
for info in self.marker_info.values():
info['images'] = []
for image in self.image_list.get_images():
latlong = image.metadata.latlong
if not latlong:
continue
for info in self.marker_info.values():
if info['latlong'] == latlong:
info['images'].append(image)
break
else:
for i in range(len(self.marker_info) + 2):
marker_id = i
if marker_id not in self.marker_info:
break
self.marker_info[marker_id] = {
'images': [image],
'latlong': LatLon(latlong),
'selected': image.selected,
}
self.JavaScript('addMarker({:d},{!r},{!r},{:d})'.format(
marker_id, latlong['lat'], latlong['lon'], image.selected))
for marker_id in list(self.marker_info.keys()):
info = self.marker_info[marker_id]
if not info['images']:
self.JavaScript('delMarker({:d})'.format(marker_id))
del self.marker_info[marker_id]
elif info['selected'] != any([x.selected for x in info['images']]):
info['selected'] = not info['selected']
self.JavaScript('enableMarker({:d},{:d})'.format(
marker_id, info['selected']))
def plot_track(self, tracks):
latlngs = []
for t in tracks:
latlngs.append([[x[1], x[2]] for x in t])
self.JavaScript('plotTrack({!r})'.format(latlngs))
@QtSlot()
@catch_all
def get_altitude(self):
altitude = self.geocoder.get_altitude(self.coords.get_value())
if altitude is not None:
self.new_altitude(round(altitude, 1))
@QtSlot()
@catch_all
def search(self, search_string=None, bounded=True):
if not search_string:
search_string = self.edit_box.lineEdit().text()
self.edit_box.clearEditText()
if not search_string:
return
self.search_string = search_string
self.clear_search()
if bounded:
bounds = self.map_status['bounds']
else:
bounds = None
for result in self.geocoder.search(search_string, bounds=bounds):
north, east, south, west, name = result
self.edit_box.addItem(name, (north, east, south, west))
self.edit_box.set_dropdown_width()
self.edit_box.showPopup()
def clear_search(self):
self.edit_box.clear()
self.edit_box.addItem('')
if self.search_string:
self.edit_box.addItem(translate('MapTabsAll', '<widen search>'),
'widen')
self.edit_box.addItem(
translate('MapTabsAll', '<repeat search>', 'repeat'))
@QtSlot(int)
@catch_all
def goto_search_result(self, idx):
self.edit_box.setCurrentIndex(0)
self.edit_box.clearFocus()
if idx == 0:
return
data = self.edit_box.itemData(idx)
if data is None:
return
if data == 'widen':
# widen search
self.search(search_string=self.search_string, bounded=False)
elif data == 'repeat':
# repeat search
self.search(search_string=self.search_string)
elif data[-1] is None:
self.JavaScript('setView({},{},{})'.format(
data[0], data[1], self.map_status['zoom']))
else:
self.JavaScript('adjustBounds({},{},{},{})'.format(*data))
@catch_all
def marker_click(self, marker_id):
self.image_list.select_images(self.marker_info[marker_id]['images'])
@catch_all
def marker_drag(self, lat, lng):
self.coords.set_value('{:.6f}, {:.6f}'.format(lat, lng))
@catch_all
def marker_drag_end(self, lat, lng, marker_id):
info = self.marker_info[marker_id]
for image in info['images']:
image.metadata.latlong = lat, lng
info['latlong'] = LatLon((lat, lng))
self.coords.refresh()
def JavaScript(self, command):
if self.map_loaded:
self.map.page().do_java_script(command)
def __init__(self, image_list, parent=None):
super(PhotiniMap, self).__init__(parent)
self.app = QtWidgets.QApplication.instance()
self.image_list = image_list
self.geocode_cache = OrderedDict()
name = self.__module__.split('.')[-1]
self.api_key = key_store.get(name, 'api_key')
self.search_key = key_store.get('opencage', 'api_key')
self.script_dir = pkg_resources.resource_filename(
'photini', 'data/' + name + '/')
self.drag_icon = QtGui.QPixmap(
os.path.join(self.script_dir, '../map_pin_grey.png'))
self.drag_hotspot = 11, 35
self.search_string = None
self.map_loaded = False
self.marker_info = {}
self.map_status = {}
self.dropped_images = []
self.setChildrenCollapsible(False)
left_side = QtWidgets.QWidget()
self.addWidget(left_side)
left_side.setLayout(QtWidgets.QFormLayout())
left_side.layout().setContentsMargins(0, 0, 0, 0)
left_side.layout().setFieldGrowthPolicy(
QtWidgets.QFormLayout.AllNonFixedFieldsGrow)
# map
# create handler for calls from JavaScript
self.call_handler = CallHandler(parent=self)
self.map = MapWebView(self.call_handler)
self.map.drop_text.connect(self.drop_text)
self.map.setAcceptDrops(False)
self.addWidget(self.map)
# search
search_layout = QtWidgets.QFormLayout()
search_layout.setContentsMargins(0, 0, 0, 0)
search_layout.setVerticalSpacing(0)
search_layout.setFieldGrowthPolicy(
QtWidgets.QFormLayout.AllNonFixedFieldsGrow)
self.edit_box = ComboBox()
self.edit_box.setEditable(True)
self.edit_box.setInsertPolicy(QtWidgets.QComboBox.NoInsert)
self.edit_box.lineEdit().setPlaceholderText(
translate('PhotiniMap', '<new search>'))
self.edit_box.lineEdit().returnPressed.connect(self.search)
self.edit_box.activated.connect(self.goto_search_result)
self.clear_search()
self.edit_box.setEnabled(False)
search_layout.addRow(translate('PhotiniMap', 'Search'), self.edit_box)
# search terms and conditions
terms = self.search_terms()
if terms:
search_layout.addRow(*terms)
left_side.layout().addRow(search_layout)
if terms:
divider = QtWidgets.QFrame()
divider.setFrameStyle(QtWidgets.QFrame.HLine)
left_side.layout().addRow(divider)
left_side.layout().addItem(
QtWidgets.QSpacerItem(1,
1000,
vPolicy=QtWidgets.QSizePolicy.Expanding))
# latitude & longitude
layout = QtWidgets.QHBoxLayout()
self.coords = SingleLineEdit()
self.coords.editingFinished.connect(self.new_coords)
self.coords.setEnabled(False)
layout.addWidget(self.coords)
# convert lat/lng to location info
self.auto_location = QtWidgets.QPushButton(
translate('PhotiniMap',
six.unichr(0x21e8) + ' address'))
self.auto_location.setFixedHeight(self.coords.height())
self.auto_location.setEnabled(False)
self.auto_location.clicked.connect(self.get_address)
layout.addWidget(self.auto_location)
left_side.layout().addRow(translate('PhotiniMap', 'Lat, long'), layout)
# location info
self.location_widgets = []
self.location_info = QtWidgets.QTabWidget()
tab_bar = QTabBar()
self.location_info.setTabBar(tab_bar)
tab_bar.context_menu.connect(self.location_tab_context_menu)
tab_bar.tabMoved.connect(self.location_tab_moved)
self.location_info.setElideMode(Qt.ElideLeft)
self.location_info.setMovable(True)
self.location_info.setEnabled(False)
left_side.layout().addRow(self.location_info)
# address lookup (and default search) terms and conditions
layout = QtWidgets.QHBoxLayout()
if terms:
widget = CompactButton(
self.tr('Address lookup\npowered by OpenCage'))
else:
widget = CompactButton(
self.tr('Search && lookup\npowered by OpenCage'))
widget.clicked.connect(self.load_tou_opencage)
layout.addWidget(widget)
widget = CompactButton(
self.tr('Geodata © OpenStreetMap\ncontributors'))
widget.clicked.connect(self.load_tou_osm)
layout.addWidget(widget)
left_side.layout().addRow(layout)
# other init
self.image_list.image_list_changed.connect(self.image_list_changed)
self.splitterMoved.connect(self.new_split)
self.block_timer = QtCore.QTimer(self)
self.block_timer.setInterval(5000)
self.block_timer.setSingleShot(True)
self.block_timer.timeout.connect(self.enable_search)
class PhotiniMap(QtWidgets.QSplitter):
def __init__(self, image_list, parent=None):
super(PhotiniMap, self).__init__(parent)
self.app = QtWidgets.QApplication.instance()
self.image_list = image_list
self.geocode_cache = OrderedDict()
name = self.__module__.split('.')[-1]
self.api_key = key_store.get(name, 'api_key')
self.search_key = key_store.get('opencage', 'api_key')
self.script_dir = pkg_resources.resource_filename(
'photini', 'data/' + name + '/')
self.drag_icon = QtGui.QPixmap(
os.path.join(self.script_dir, '../map_pin_grey.png'))
self.drag_hotspot = 11, 35
self.search_string = None
self.map_loaded = False
self.marker_info = {}
self.map_status = {}
self.dropped_images = []
self.setChildrenCollapsible(False)
left_side = QtWidgets.QWidget()
self.addWidget(left_side)
left_side.setLayout(QtWidgets.QFormLayout())
left_side.layout().setContentsMargins(0, 0, 0, 0)
left_side.layout().setFieldGrowthPolicy(
QtWidgets.QFormLayout.AllNonFixedFieldsGrow)
# map
# create handler for calls from JavaScript
self.call_handler = CallHandler(parent=self)
self.map = MapWebView(self.call_handler)
self.map.drop_text.connect(self.drop_text)
self.map.setAcceptDrops(False)
self.addWidget(self.map)
# search
search_layout = QtWidgets.QFormLayout()
search_layout.setContentsMargins(0, 0, 0, 0)
search_layout.setVerticalSpacing(0)
search_layout.setFieldGrowthPolicy(
QtWidgets.QFormLayout.AllNonFixedFieldsGrow)
self.edit_box = ComboBox()
self.edit_box.setEditable(True)
self.edit_box.setInsertPolicy(QtWidgets.QComboBox.NoInsert)
self.edit_box.lineEdit().setPlaceholderText(
translate('PhotiniMap', '<new search>'))
self.edit_box.lineEdit().returnPressed.connect(self.search)
self.edit_box.activated.connect(self.goto_search_result)
self.clear_search()
self.edit_box.setEnabled(False)
search_layout.addRow(translate('PhotiniMap', 'Search'), self.edit_box)
# search terms and conditions
terms = self.search_terms()
if terms:
search_layout.addRow(*terms)
left_side.layout().addRow(search_layout)
if terms:
divider = QtWidgets.QFrame()
divider.setFrameStyle(QtWidgets.QFrame.HLine)
left_side.layout().addRow(divider)
left_side.layout().addItem(
QtWidgets.QSpacerItem(1,
1000,
vPolicy=QtWidgets.QSizePolicy.Expanding))
# latitude & longitude
layout = QtWidgets.QHBoxLayout()
self.coords = SingleLineEdit()
self.coords.editingFinished.connect(self.new_coords)
self.coords.setEnabled(False)
layout.addWidget(self.coords)
# convert lat/lng to location info
self.auto_location = QtWidgets.QPushButton(
translate('PhotiniMap',
six.unichr(0x21e8) + ' address'))
self.auto_location.setFixedHeight(self.coords.height())
self.auto_location.setEnabled(False)
self.auto_location.clicked.connect(self.get_address)
layout.addWidget(self.auto_location)
left_side.layout().addRow(translate('PhotiniMap', 'Lat, long'), layout)
# location info
self.location_widgets = []
self.location_info = QtWidgets.QTabWidget()
tab_bar = QTabBar()
self.location_info.setTabBar(tab_bar)
tab_bar.context_menu.connect(self.location_tab_context_menu)
tab_bar.tabMoved.connect(self.location_tab_moved)
self.location_info.setElideMode(Qt.ElideLeft)
self.location_info.setMovable(True)
self.location_info.setEnabled(False)
left_side.layout().addRow(self.location_info)
# address lookup (and default search) terms and conditions
layout = QtWidgets.QHBoxLayout()
if terms:
widget = CompactButton(
self.tr('Address lookup\npowered by OpenCage'))
else:
widget = CompactButton(
self.tr('Search && lookup\npowered by OpenCage'))
widget.clicked.connect(self.load_tou_opencage)
layout.addWidget(widget)
widget = CompactButton(
self.tr('Geodata © OpenStreetMap\ncontributors'))
widget.clicked.connect(self.load_tou_osm)
layout.addWidget(widget)
left_side.layout().addRow(layout)
# other init
self.image_list.image_list_changed.connect(self.image_list_changed)
self.splitterMoved.connect(self.new_split)
self.block_timer = QtCore.QTimer(self)
self.block_timer.setInterval(5000)
self.block_timer.setSingleShot(True)
self.block_timer.timeout.connect(self.enable_search)
def search_terms(self):
return None
@QtCore.pyqtSlot()
@catch_all
def load_tou_opencage(self):
QtGui.QDesktopServices.openUrl(
QtCore.QUrl('https://geocoder.opencagedata.com/'))
@QtCore.pyqtSlot()
@catch_all
def load_tou_osm(self):
QtGui.QDesktopServices.openUrl(
QtCore.QUrl('http://www.openstreetmap.org/copyright'))
@QtCore.pyqtSlot(int, int)
@catch_all
def new_split(self, pos, index):
self.app.config_store.set('map', 'split', str(self.sizes()))
@QtCore.pyqtSlot()
@catch_all
def image_list_changed(self):
self.redraw_markers()
self.display_coords()
self.display_location()
self.see_selection()
@QtCore.pyqtSlot()
@catch_all
def initialise(self):
page = '''
<!DOCTYPE html>
<html>
<head>
<meta charset="utf-8" />
<meta name="viewport" content="initial-scale=1.0, user-scalable=no" />
<style type="text/css">
html, body {{ height: 100%; margin: 0; padding: 0 }}
#mapDiv {{ position: relative; width: 100%; height: 100% }}
</style>
<script type="text/javascript">
var initData = {{key: "{key}", lat: {lat}, lng: {lng}, zoom: {zoom}}};
</script>
{initialize}
{head}
<script type="text/javascript" src="script.js" async></script>
</head>
<body ondragstart="return false">
<div id="mapDiv"></div>
</body>
</html>
'''
lat, lng = eval(
self.app.config_store.get('map', 'centre', '(51.0, 0.0)'))
zoom = int(eval(self.app.config_store.get('map', 'zoom', '11')))
if using_qtwebengine:
initialize = '''
<script type="text/javascript"
src="qrc:///qtwebchannel/qwebchannel.js">
</script>
<script type="text/javascript">
var python;
function initialize()
{
new QWebChannel(qt.webChannelTransport, function (channel) {
python = channel.objects.python;
loadMap();
});
}
</script>
'''
else:
initialize = '''
<script type="text/javascript">
function initialize()
{
loadMap();
}
</script>
'''
page = page.format(lat=lat,
lng=lng,
zoom=zoom,
initialize=initialize,
key=self.api_key,
head=self.get_head())
QtWidgets.QApplication.setOverrideCursor(Qt.WaitCursor)
self.map.setHtml(page, QtCore.QUrl.fromLocalFile(self.script_dir))
@catch_all
def initialize_finished(self):
QtWidgets.QApplication.restoreOverrideCursor()
self.map_loaded = True
self.edit_box.setEnabled(True)
self.map.setAcceptDrops(True)
self.redraw_markers()
self.display_coords()
def refresh(self):
self.setSizes(
eval(self.app.config_store.get('map', 'split', str(self.sizes()))))
self.image_list.set_drag_to_map(self.drag_icon, self.drag_hotspot)
if not self.map_loaded:
self.initialise()
return
lat, lng = eval(self.app.config_store.get('map', 'centre'))
zoom = int(eval(self.app.config_store.get('map', 'zoom')))
self.JavaScript('setView({!r},{!r},{:d})'.format(lat, lng, zoom))
def do_not_close(self):
return False
@catch_all
def new_status(self, status):
self.map_status.update(status)
for key in ('centre', 'zoom'):
if key in status:
self.app.config_store.set('map', key,
repr(self.map_status[key]))
@QtCore.pyqtSlot(int, int, six.text_type)
@catch_all
def drop_text(self, x, y, text):
self.dropped_images = eval(text)
self.JavaScript('markerDrop({:d},{:d})'.format(x, y))
@catch_all
def marker_drop(self, lat, lng):
for path in self.dropped_images:
image = self.image_list.get_image(path)
image.metadata.latlong = lat, lng
self.dropped_images = []
self.redraw_markers()
self.display_coords()
self.see_selection()
@QtCore.pyqtSlot()
@catch_all
def new_coords(self):
text = self.coords.get_value().strip()
if not text:
for image in self.image_list.get_selected_images():
image.metadata.latlong = None
self.redraw_markers()
return
try:
lat, lng = map(float, text.split(','))
except Exception:
self.display_coords()
return
for image in self.image_list.get_selected_images():
image.metadata.latlong = lat, lng
self.redraw_markers()
self.display_coords()
self.see_selection()
def see_selection(self):
locations = []
for image in self.image_list.get_selected_images():
latlong = image.metadata.latlong
if not latlong:
continue
location = [latlong.lat, latlong.lon]
if location not in locations:
locations.append(location)
if not locations:
return
self.JavaScript('fitPoints({})'.format(repr(locations)))
@QtCore.pyqtSlot(QtGui.QContextMenuEvent)
@catch_all
def location_tab_context_menu(self, event):
idx = self.location_info.tabBar().tabAt(event.pos())
self.location_info.setCurrentIndex(idx)
menu = QtWidgets.QMenu(self)
menu.addAction(self.tr('Duplicate location'), self.duplicate_location)
menu.addAction(self.tr('Delete location'), self.delete_location)
action = menu.exec_(event.globalPos())
@QtCore.pyqtSlot()
@catch_all
def duplicate_location(self):
idx = self.location_info.currentIndex()
for image in self.image_list.get_selected_images():
# duplicate data
location = Location(self._get_location(image, idx) or {})
# shuffle data up
location_list = list(image.metadata.location_shown or [])
location_list.insert(idx, location)
image.metadata.location_shown = location_list
# display data
self.display_location()
@QtCore.pyqtSlot()
@catch_all
def delete_location(self):
idx = self.location_info.currentIndex()
for image in self.image_list.get_selected_images():
# shuffle data down
location_list = list(image.metadata.location_shown or [])
if idx == 0:
if location_list:
location = location_list[0]
else:
location = None
image.metadata.location_taken = location
if idx <= len(location_list):
del location_list[max(idx - 1, 0)]
image.metadata.location_shown = location_list
# display data
self.display_location()
@QtCore.pyqtSlot(int, int)
@catch_all
def location_tab_moved(self, idx_a, idx_b):
self.pending_move = idx_a, idx_b
# do actual swap when idle to avoid seg fault
QtCore.QTimer.singleShot(0, self._location_tab_moved)
@QtCore.pyqtSlot()
@catch_all
def _location_tab_moved(self):
idx_a, idx_b = self.pending_move
# swap data
for image in self.image_list.get_selected_images():
temp_a = self._get_location(image, idx_a)
temp_b = self._get_location(image, idx_b)
self._set_location(image, idx_a, temp_b)
self._set_location(image, idx_b, temp_a)
# adjust tab names
for idx in range(min(idx_a, idx_b), max(idx_a, idx_b) + 1):
self.set_tab_text(idx)
# display data
self.display_location()
def _get_location(self, image, idx):
if idx == 0:
return image.metadata.location_taken
elif not image.metadata.location_shown:
return None
elif idx <= len(image.metadata.location_shown):
return image.metadata.location_shown[idx - 1]
return None
def _set_location(self, image, idx, location):
if idx == 0:
image.metadata.location_taken = location
else:
location_list = list(image.metadata.location_shown or [])
while len(location_list) < idx:
location_list.append(None)
location_list[idx - 1] = location
image.metadata.location_shown = location_list
@QtCore.pyqtSlot(object, dict)
@catch_all
def new_location(self, widget, new_value):
idx = self.location_info.indexOf(widget)
for image in self.image_list.get_selected_images():
temp = dict(self._get_location(image, idx) or {})
temp.update(new_value)
self._set_location(image, idx, temp)
# new_location can be called when changing tab, so don't delete
# tabs until later
QtCore.QTimer.singleShot(0, self.display_location)
def display_coords(self):
images = self.image_list.get_selected_images()
if not images:
self.coords.set_value(None)
self.auto_location.setEnabled(False)
return
values = []
for image in images:
value = image.metadata.latlong
if value not in values:
values.append(value)
if len(values) > 1:
self.coords.set_multiple(choices=filter(None, values))
self.auto_location.setEnabled(False)
else:
self.coords.set_value(values[0])
self.auto_location.setEnabled(
bool(values[0]) and not self.block_timer.isActive())
def set_tab_text(self, idx):
if idx == 0:
text = self.tr('camera')
else:
text = self.tr('subject {}').format(idx)
self.location_info.setTabText(idx, text)
@QtCore.pyqtSlot()
@catch_all
def display_location(self):
images = self.image_list.get_selected_images()
# get required number of tabs
count = 0
for image in images:
if image.metadata.location_shown:
count = max(count, len(image.metadata.location_shown))
count += 2
# add or remove tabs
if self.location_info.currentIndex() >= count:
self.location_info.setCurrentIndex(count - 1)
idx = self.location_info.count()
while idx < count:
if not self.location_widgets:
widget = LocationInfo()
widget.new_value.connect(self.new_location)
self.location_widgets.append(widget)
self.location_info.addTab(self.location_widgets.pop(), '')
self.set_tab_text(idx)
idx += 1
while idx > count:
idx -= 1
self.location_widgets.append(self.location_info.widget(idx))
self.location_info.removeTab(idx)
# display data
for idx in range(count):
widget = self.location_info.widget(idx)
if images:
values = defaultdict(list)
for image in images:
location = self._get_location(image, idx) or {}
for key in widget.members:
value = None
if key in location:
value = location[key]
if value not in values[key]:
values[key].append(value)
for key in widget.members:
if len(values[key]) > 1:
widget.members[key].set_multiple(
choices=filter(None, values[key]))
else:
widget.members[key].set_value(values[key][0])
else:
for key in widget.members:
widget.members[key].set_value(None)
@QtCore.pyqtSlot(list)
@catch_all
def new_selection(self, selection):
self.coords.setEnabled(bool(selection))
self.location_info.setEnabled(bool(selection))
self.redraw_markers()
self.display_coords()
self.display_location()
self.see_selection()
def redraw_markers(self):
if not self.map_loaded:
return
for info in self.marker_info.values():
info['images'] = []
for image in self.image_list.get_images():
latlong = image.metadata.latlong
if not latlong:
continue
for info in self.marker_info.values():
if info['latlong'] == latlong:
info['images'].append(image)
break
else:
for i in range(len(self.marker_info) + 2):
marker_id = i
if marker_id not in self.marker_info:
break
self.marker_info[marker_id] = {
'images': [image],
'latlong': LatLon(latlong),
'selected': image.selected,
}
self.JavaScript('addMarker({:d},{!r},{!r},{:d})'.format(
marker_id, latlong.lat, latlong.lon, image.selected))
for marker_id in list(self.marker_info.keys()):
info = self.marker_info[marker_id]
if not info['images']:
self.JavaScript('delMarker({:d})'.format(marker_id))
del self.marker_info[marker_id]
elif info['selected'] != any([x.selected for x in info['images']]):
info['selected'] = not info['selected']
self.JavaScript('enableMarker({:d},{:d})'.format(
marker_id, info['selected']))
def plot_track(self, tracks):
latlngs = []
for t in tracks:
latlngs.append([[x[1], x[2]] for x in t])
self.JavaScript('plotTrack({!r})'.format(latlngs))
@QtCore.pyqtSlot()
@catch_all
def enable_search(self):
self.block_timer.stop()
self.edit_box.lineEdit().setEnabled(self.map_loaded)
if self.search_string:
item = self.edit_box.model().item(1)
item.setFlags(Qt.ItemIsSelectable | Qt.ItemIsEnabled)
item = self.edit_box.model().item(2)
item.setFlags(Qt.ItemIsSelectable | Qt.ItemIsEnabled)
self.display_coords()
def disable_search(self):
self.edit_box.lineEdit().setEnabled(False)
if self.search_string:
item = self.edit_box.model().item(1)
item.setFlags(~(Qt.ItemIsSelectable | Qt.ItemIsEnabled))
item = self.edit_box.model().item(2)
item.setFlags(~(Qt.ItemIsSelectable | Qt.ItemIsEnabled))
self.auto_location.setEnabled(False)
self.block_timer.start()
focus = QtWidgets.QApplication.focusWidget()
if focus:
focus.clearFocus()
def do_geocode(self, params):
cache_key = params['q']
if 'bounds' in params:
cache_key += params['bounds']
if cache_key in self.geocode_cache:
return self.geocode_cache[cache_key]
self.disable_search()
params['key'] = self.search_key
params['abbrv'] = '1'
params['no_annotations'] = '1'
lang, encoding = locale.getdefaultlocale()
if lang:
params['language'] = lang
with Busy():
try:
rsp = requests.get(
'https://api.opencagedata.com/geocode/v1/json',
params=params,
timeout=5)
except Exception as ex:
logger.error(str(ex))
return []
if rsp.status_code >= 400:
logger.error('Search error %d', rsp.status_code)
return []
rsp = rsp.json()
status = rsp['status']
if status['code'] != 200:
logger.error('Search error %d: %s', status['code'],
status['message'])
return []
if rsp['total_results'] < 1:
logger.error('No results found')
return []
rate = rsp['rate']
self.block_timer.setInterval(5000 * rate['limit'] //
max(rate['remaining'], 1))
self.geocode_cache[cache_key] = rsp['results']
while len(self.geocode_cache) > 20:
self.geocode_cache.popitem(last=False)
return rsp['results']
def geocode(self, search_string, bounds=None):
params = {
'q': search_string,
'limit': '20',
}
if bounds:
north, east, south, west = bounds
w = east - west
h = north - south
if min(w, h) < 10.0:
lat, lon = self.map_status['centre']
north = min(lat + 5.0, 90.0)
south = max(lat - 5.0, -90.0)
east = lon + 5.0
west = lon - 5.0
params['bounds'] = '{!r},{!r},{!r},{!r}'.format(
west, south, east, north)
for result in self.do_geocode(params):
yield (result['bounds']['northeast']['lat'],
result['bounds']['northeast']['lng'],
result['bounds']['southwest']['lat'],
result['bounds']['southwest']['lng'], result['formatted'])
# Map OpenCage address components to IPTC address heirarchy. There
# are many possible components (user generated data) so any
# unrecognised ones are put in 'sublocation'. See
# https://github.com/OpenCageData/address-formatting/blob/master/conf/components.yaml
address_map = {
'world_region': ('continent', ),
'country_code':
('ISO_3166-1_alpha-3', 'ISO_3166-1_alpha-2', 'country_code'),
'country_name': ('country', 'country_name'),
'province_state':
('county', 'county_code', 'local_administrative_area',
'state_district', 'state', 'state_code', 'province', 'region',
'island'),
'city': ('village', 'locality', 'hamlet', 'neighbourhood',
'city_district', 'suburb', 'city', 'town', 'postcode'),
'sublocation':
('house_number', 'street_number', 'house', 'public_building',
'building', 'water', 'road', 'pedestrian', 'path', 'residential',
'street_name', 'street', 'cycleway', 'footway', 'place'),
'ignore': ('political_union', 'road_reference', 'road_reference_intl',
'road_type', '_type'),
}
@QtCore.pyqtSlot()
@catch_all
def get_address(self):
results = self.do_geocode(
{'q': self.coords.get_value().replace(' ', '')})
if not results:
return
address = dict(results[0]['components'])
if 'county_code' in address and 'county' in address:
del address['county_code']
if 'state_code' in address and 'state' in address:
del address['state_code']
self.new_location(self.location_info.currentWidget(),
Location.from_address(address, self.address_map))
@QtCore.pyqtSlot()
@catch_all
def search(self, search_string=None, bounded=True):
if not search_string:
search_string = self.edit_box.lineEdit().text()
self.edit_box.clearEditText()
if not search_string:
return
self.search_string = search_string
self.clear_search()
if bounded:
bounds = self.map_status['bounds']
else:
bounds = None
for result in self.geocode(search_string, bounds=bounds):
north, east, south, west, name = result
self.edit_box.addItem(name, (north, east, south, west))
self.edit_box.set_dropdown_width()
self.edit_box.showPopup()
def clear_search(self):
self.edit_box.clear()
self.edit_box.addItem('')
if self.search_string:
self.edit_box.addItem(translate('PhotiniMap', '<widen search>'))
self.edit_box.addItem(translate('PhotiniMap', '<repeat search>'))
@QtCore.pyqtSlot(int)
@catch_all
def goto_search_result(self, idx):
self.edit_box.setCurrentIndex(0)
self.edit_box.clearFocus()
if idx == 0:
return
if self.search_string and idx == 1:
# widen search
self.search(search_string=self.search_string, bounded=False)
return
if self.search_string and idx == 2:
# repeat search
self.search(search_string=self.search_string)
return
view = self.edit_box.itemData(idx)
if view[-1] is None:
self.JavaScript('setView({},{},{})'.format(
view[0], view[1], self.map_status['zoom']))
else:
self.JavaScript('adjustBounds({},{},{},{})'.format(*view))
@catch_all
def marker_click(self, marker_id):
self.image_list.select_images(self.marker_info[marker_id]['images'])
@catch_all
def marker_drag(self, lat, lng):
self.coords.set_value('{:.6f}, {:.6f}'.format(lat, lng))
@catch_all
def marker_drag_end(self, lat, lng, marker_id):
info = self.marker_info[marker_id]
for image in info['images']:
image.metadata.latlong = lat, lng
info['latlong'] = LatLon((lat, lng))
self.display_coords()
def JavaScript(self, command):
if self.map_loaded:
self.map.page().do_java_script(command)
class KeywordsEditor(QtWidgets.QWidget):
def __init__(self, **kw):
super(KeywordsEditor, self).__init__()
self.config_store = QtWidgets.QApplication.instance().config_store
self.league_table = {}
for keyword, score in self.config_store.get('descriptive', 'keywords',
{}).items():
if isinstance(score, int):
# old style keyword list
self.league_table[keyword] = date.min.isoformat(), score // 50
else:
# new style keyword list
self.league_table[keyword] = score
layout = QtWidgets.QHBoxLayout()
layout.setContentsMargins(0, 0, 0, 0)
self.setLayout(layout)
# line edit box
self.edit = SingleLineEdit(**kw)
layout.addWidget(self.edit)
# favourites drop down
self.favourites = ComboBox()
self.favourites.addItem(translate('DescriptiveTab', '<favourites>'))
self.favourites.setFixedWidth(
self.favourites.minimumSizeHint().width())
self.update_favourites()
self.favourites.currentIndexChanged.connect(self.add_favourite)
layout.addWidget(self.favourites)
# adopt child widget methods and signals
self.get_value = self.edit.get_value
self.set_value = self.edit.set_value
self.set_multiple = self.edit.set_multiple
self.is_multiple = self.edit.is_multiple
self.editingFinished = self.edit.editingFinished
def update_favourites(self):
self.favourites.clear()
self.favourites.addItem(translate('DescriptiveTab', '<favourites>'))
keywords = list(self.league_table.keys())
keywords.sort(key=lambda x: self.league_table[x], reverse=True)
# limit size of league_table by deleting lowest scoring
if len(keywords) > 100:
threshold = self.league_table[keywords[100]]
for keyword in keywords:
if self.league_table[keyword] <= threshold:
del self.league_table[keyword]
# select highest scoring for drop down list
keywords = keywords[:20]
keywords.sort(key=lambda x: x.lower())
for keyword in keywords:
self.favourites.addItem(keyword)
self.favourites.set_dropdown_width()
def update_league_table(self, images):
today = date.today().isoformat()
for image in images:
value = list(
filter(lambda x: not x.startswith('flickr:photo_id'),
image.metadata.keywords or []))
if not value:
continue
for keyword in value:
if keyword not in self.league_table:
self.league_table[keyword] = today, 1
elif self.league_table[keyword][0] != today:
self.league_table[keyword] = (
today, self.league_table[keyword][1] + 1)
self.config_store.set('descriptive', 'keywords', self.league_table)
self.update_favourites()
@QtSlot(int)
@catch_all
def add_favourite(self, idx):
if idx <= 0:
return
self.favourites.setCurrentIndex(0)
new_value = self.favourites.itemText(idx)
current_value = self.get_value()
if current_value:
new_value = current_value + '; ' + new_value
self.set_value(new_value)
self.editingFinished.emit()