# -*- coding: utf-8 -*-
from __future__ import absolute_import, division, print_function
import utool as ut
import plottool as pt
print, print_, printDBG, rrr, profile = ut.inject(__name__, '[user_dialogs]')
[docs]def convert_name_suggestion_to_aids(ibs, choicetup, name_suggest_tup):
"""
Example:
>>> # DISABLE_DOCTEST
>>> from ibeis.model.hots.user_dialogs import * # NOQA
>>> import ibeis
>>> # build test data
>>> ibs = ibeis.opendb('testdb1')
>>> comp_aids = [2, 3, 4]
>>> comp_names = ['fred', 'sue', 'alice']
>>> chosen_names = ['fred']
>>> # execute function
>>> result = convert_name_suggestion_to_aids(ibs, choicetup, name_suggest_tup)
>>> # verify results
>>> print(result)
"""
num_top = 3
autoname_msg, chosen_names, name_confidence = name_suggest_tup
comp_aids_all = ut.get_list_column(choicetup.sorted_aids, 0)
comp_aids = ut.listclip(comp_aids_all, num_top)
comp_names = ibs.get_annot_names(comp_aids)
issuggested = ut.list_cover(comp_names, chosen_names)
suggest_aids = ut.filter_items(comp_aids, issuggested)
return comp_aids, suggest_aids
[docs]def wait_for_user_name_decision(ibs, cm, qreq_, choicetup, name_suggest_tup, incinfo=None):
r"""
Prompts the user for input
hooks into to some method of getting user input for names
Args:
ibs (IBEISController):
cm (QueryResult): object of feature correspondences and scores
autoname_func (function):
CommandLine:
python -m ibeis.model.hots.user_dialogs --test-wait_for_user_name_decision --show
Example:
>>> # DISABLE_DOCTEST
>>> from ibeis.model.hots.user_dialogs import * # NOQA
>>> import ibeis
>>> # build test data
>>> ibs = ibeis.opendb('testdb1')
>>> qaids = [1]
>>> daids = [2, 3, 4, 5]
>>> cm, qreq_ = ibs.query_chips(qaids, daids, cfgdict=dict(), return_request=True, return_cm=True)[0]
>>> choicetup = '?'
>>> name_suggest_tup = '?'
>>> incinfo = None
>>> # execute function
>>> result = wait_for_user_name_decision(ibs, cm, qreq_, choicetup, name_suggest_tup, incinfo)
>>> # verify results
>>> print(result)
>>> ut.show_if_requested()
"""
if cm is None:
print('WARNING: qres is None')
new_mplshow = True and cm is not None
mplshowtop = False and cm is not None
qtinspect = False and cm is not None
if new_mplshow:
from ibeis.viz.interact import interact_query_decision
print('Showing matplotlib window')
# convert name choices into data for gui
comp_aids, suggest_aids = convert_name_suggestion_to_aids(ibs, choicetup, name_suggest_tup)
# Update names tree callback
# Let the harness do these callbacks
#backend_callback = incinfo.get('backend_callback', None)
#update_callback = incinfo.get('update_callback', None)
name_decision_callback = incinfo['name_decision_callback']
progress_current = incinfo['count']
progress_total = incinfo['nTotal']
fnum = incinfo['fnum']
qvi = interact_query_decision.QueryVerificationInteraction(
qreq_, cm, comp_aids, suggest_aids,
name_decision_callback=name_decision_callback,
#update_callback=update_callback,
#backend_callback=backend_callback,
progress_current=progress_current, progress_total=progress_total,
fnum=fnum)
qvi.fig.show()
pt.bring_to_front(qvi.fig)
if mplshowtop:
import guitool
fnum = 513
pt.figure(fnum=fnum, pnum=(2, 3, 1), doclf=True, docla=True)
fig = cm.ishow_top(qreq_, fnum=fnum, in_image=False, annot_mode=0,
sidebyside=False, show_query=True)
fig.show()
#fig.canvas.raise_()
#from plottool import fig_presenter
#fig_presenter.bring_to_front(fig)
newname = ibs.make_next_name()
newname_prefix = 'New Name:\n'
# FIXME or remoev
name = None
#if chosen_names is None:
# name = newname_prefix + newname
aid_list = ut.get_list_column(choicetup.sorted_aids, 0)
name_options = ibs.get_annot_names(aid_list) + [newname_prefix + newname]
msg = 'Decide on query name. System suggests; ' + str(name)
title = 'name decision'
options = name_options[::-1]
user_chosen_name = guitool.user_option(None, msg, title, options) # NOQA
if user_chosen_name is None:
raise AssertionError('User Canceled Query')
user_chosen_name = user_chosen_name.replace(newname_prefix, '')
# TODO: Make the old interface use the correct sorted_aids format
#name_decision_callback(user_chosen_name)
if qtinspect:
print('Showing qt inspect window')
qres_wgt = cm.qt_inspect_gui(qreq_)
qres_wgt.show()
qres_wgt.raise_()
#if qreq_ is not None:
# if qreq_.normalizer is None:
# print('normalizer is None!!')
# else:
# qreq_.normalizer.visualize(update=False, fnum=2)
# Prompt the user (this could be swaped out with a qt or web interface)
#if qtinspect:
# qres_wgt.close()
#return user_chosen_name
[docs]def wait_for_user_exemplar_decision(autoexemplar_msg, exemplar_decision,
exemplar_condience, incinfo=None):
r""" hooks into to some method of getting user input for exemplars
TODO: really good interface
Args:
autoexemplar_msg (?):
exemplar_decision (?):
exemplar_condience (?):
Returns:
?: True
CommandLine:
python -m ibeis.model.hots.automated_matcher --test-get_user_exemplar_decision
Example:
>>> # DISABLE_DOCTEST
>>> from ibeis.model.hots.automated_matcher import * # NOQA
>>> import ibeis # NOQA
>>> # build test data
>>> autoexemplar_msg = '?'
>>> exemplar_decision = '?'
>>> exemplar_condience = '?'
>>> True = get_user_exemplar_decision(autoexemplar_msg, exemplar_decision, exemplar_condience)
>>> # verify results
>>> result = str(True)
>>> print(result)
"""
import guitool
options = ['No', 'Yes']
msg = 'Add query as new exemplar?. IBEIS suggests: ' + options[exemplar_decision]
title = 'exemplar decision'
responce = guitool.user_option(None, msg, title, options) # NOQA
if responce is None:
raise AssertionError('User Canceled Query')
if responce == 'Yes':
exemplar_decision = True
elif responce == 'No':
exemplar_decision = False
else:
raise AssertionError('answer yes or no')
# TODO CALLBACK HERE
exemplar_decision_callback = incinfo['exemplar_decision_callback']
exemplar_decision_callback(exemplar_decision)