def tryout_web_graphs(self, infr):
"""
https://plot.ly/python/
http://bokeh.pydata.org/en/latest/
pip install bokeh
Notes:
http://www.coppelia.io/2014/07/an-a-to-z-of-extra-features-for-the-d3-force-layout/
http://andrewmellor.co.uk/blog/articles/2014/12/14/d3-networks/
pip install plotly # eww need to sign up and get a key
http://igraph.org/
import mpld3
mpld3.save_html(fig, open('fig.html', 'w'))
mpld3.save_json(fig, open('fig.json', 'w'))
fig = pt.gcf()
"""
#import plottool as pt
# http://andrewmellor.co.uk/blog/articles/2014/12/14/d3-networks/
from networkx.readwrite import json_graph
G = infr.graph
data = json_graph.node_link_data(G)
json_text = ut.to_json(data, pretty=True)
ut.writeto('graph.json', json_text)
ut.editfile('graph.json')
ut.startfile('d3_example.html')
def generate_all():
r"""
CommandLine:
python -m ibeis.scripts.gen_cand_expts --exec-generate_all --vim
python -m ibeis.scripts.gen_cand_expts --exec-generate_all
python -m ibeis.scripts.gen_cand_expts --exec-generate_all --full
./experiments_overnight.sh
Example:
>>> from ibeis.scripts.gen_cand_expts import * # NOQA
>>> generate_all()
"""
#script_names = ['sh ' + func()[0] for func in TEST_GEN_FUNCS]
script_lines = ut.flatten([
['\n\n### ' + ut.get_funcname(func),
'# python -m ibeis.scripts.gen_cand_expts --exec-' +
ut.get_funcname(func)] + make_standard_test_scripts(func())[2]
for func in TEST_GEN_FUNCS])
fname, script, line_list = write_script_lines(script_lines, 'experiments_overnight.sh')
if ut.get_argflag('--vim'):
ut.editfile(fname)
return fname, script, line_list
def generate_all():
r"""
CommandLine:
python -m ibeis.scripts.gen_cand_expts --exec-generate_all --vim
python -m ibeis.scripts.gen_cand_expts --exec-generate_all
python -m ibeis.scripts.gen_cand_expts --exec-generate_all --full
./experiments_overnight.sh
Example:
>>> from ibeis.scripts.gen_cand_expts import * # NOQA
>>> generate_all()
"""
#script_names = ['sh ' + func()[0] for func in TEST_GEN_FUNCS]
script_lines = ut.flatten([
['\n\n### ' + ut.get_funcname(func),
'# python -m ibeis.scripts.gen_cand_expts --exec-' +
ut.get_funcname(func)] + make_standard_test_scripts(func())[2]
for func in TEST_GEN_FUNCS])
fname, script, line_list = write_script_lines(script_lines, 'experiments_overnight.sh')
if ut.get_argflag('--vim'):
ut.editfile(fname)
return fname, script, line_list
def postload_commands(ibs, back):
"""
Postload commands deal with a specific ibeis database
ibeis --db PZ_MTEST --occur "*All Images" --query 1
ibeis --db PZ_MTEST --occur "*All Images" --query-intra
"""
if ut.NOT_QUIET:
print('\n[main_cmd] postload_commands')
if params.args.view_database_directory:
print('got arg --vdd')
vdd(ibs)
if params.args.set_default_dbdir:
sysres.set_default_dbdir(ibs.get_dbdir())
if params.args.update_query_cfg is not None:
# Set query parameters from command line using the --cfg flag
cfgdict = ut.parse_cfgstr_list(params.args.update_query_cfg)
print('Custom cfgdict specified')
print(ut.repr2(cfgdict))
ibs.update_query_cfg(**cfgdict)
if params.args.edit_notes:
ut.editfile(ibs.get_dbnotes_fpath(ensure=True))
if params.args.delete_cache:
ibs.delete_cache()
if params.args.delete_cache_complete:
ibs.delete_cache(delete_imagesets=True)
if params.args.delete_query_cache:
ibs.delete_qres_cache()
if params.args.set_all_species is not None:
ibs._overwrite_all_annot_species_to(params.args.set_all_species)
if params.args.dump_schema:
ibs.db.print_schema()
if ut.get_argflag('--ipynb'):
back.launch_ipy_notebook()
select_imgsetid = ut.get_argval(('--select-imgsetid', '--imgsetid', '--occur', '--gsid'), None)
if select_imgsetid is not None:
print('\n+ --- CMD SELECT IMGSETID=%r ---' % (select_imgsetid,))
# Whoa: this doesnt work. weird.
#back.select_imgsetid(select_imgsetid)
# This might be the root of gui problems
#back.front._change_imageset(select_imgsetid)
back.front.select_imageset_tab(select_imgsetid)
print('L ___ CMD SELECT IMGSETID=%r ___\n' % (select_imgsetid,))
# Send commands to GUIBack
if params.args.select_aid is not None:
if back is not None:
try:
ibsfuncs.assert_valid_aids(ibs, (params.args.select_aid,))
except AssertionError:
print('Valid RIDs are: %r' % (ibs.get_valid_aids(),))
raise
back.select_aid(params.args.select_aid)
if params.args.select_gid is not None:
back.select_gid(params.args.select_gid)
if params.args.select_nid is not None:
back.select_nid(params.args.select_nid)
select_name = ut.get_argval('--select-name')
if select_name is not None:
import ibeis.gui.guiheaders as gh
back.ibswgt.select_table_indicies_from_text(gh.NAMES_TREE, select_name,
allow_table_change=True)
if ut.get_argflag(('--intra-occur-query', '--query-intra-occur', '--query-intra')):
back.special_query_funcs['intra_occurrence'](cfgdict={'use_k_padding': False})
qaid_list = ut.get_argval(('--query-aid', '--query'), type_=list, default=None)
if qaid_list is not None:
#qaid_list = params.args.query_aid
# fix stride case
if len(qaid_list) == 1 and isinstance(qaid_list[0], tuple):
qaid_list = list(qaid_list[0])
daids_mode = ut.get_argval('--daids-mode', type_=str, default=const.VS_EXEMPLARS_KEY)
back.compute_queries(qaid_list=qaid_list, daids_mode=daids_mode, ranks_top=10)
if ut.get_argflag('--inc-query'):
back.incremental_query()
if ut.get_argflag(('--dbinfo', '--display_dbinfo')):
back.display_dbinfo()
pass
aidcmd = ut.get_argval('--aidcmd', default=None)
aid = ut.get_argval('--aid', type_=int, default=1)
if aidcmd:
#aidcmd = 'Interact image'
metadata = ibs.get_annot_lazy_dict(aid)
annot_context_options = metadata['annot_context_options']
aidcmd_dict = dict(annot_context_options)
print('aidcmd_dict = %s' % (ut.repr3(aidcmd_dict),))
command = aidcmd_dict[aidcmd]
command()
#import utool
#utool.embed()
#back.start_web_server_parallel()
if ut.get_argflag('--start-web'):
back.start_web_server_parallel()
if ut.get_argflag('--name-tab'):
from ibeis.gui.guiheaders import NAMES_TREE
back.front.set_table_tab(NAMES_TREE)
view = back.front.views[NAMES_TREE]
model = view.model()
view._set_sort(model.col_name_list.index('nAids'), col_sort_reverse=True)
if ut.get_argflag('--graph'):
back.make_qt_graph_interface()
screengrab_fpath = ut.get_argval('--screengrab')
if screengrab_fpath:
from guitool.__PYQT__.QtGui import QPixmap
from PyQt4.QtTest import QTest
from PyQt4.QtCore import Qt
fpath = ut.truepath(screengrab_fpath)
import guitool
#ut.embed()
timer2 = guitool.__PYQT__.QtCore.QTimer()
done = [1000]
def delayed_screenshot_func():
if done[0] == 500:
#back.mainwin.menubar.triggered.emit(back.mainwin.menuFile)
print('Mouseclick')
QTest.mouseClick(back.mainwin.menuFile, Qt.LeftButton)
# This works
#QTest.mouseClick(back.front.import_button, Qt.LeftButton)
if done[0] == 1:
timer2.stop()
print('screengrab to %r' % (fpath,))
screenimg = QPixmap.grabWindow(back.mainwin.winId())
screenimg.save(fpath, 'jpg')
ut.startfile(fpath)
print('lub dub2')
done[0] -= 1
return None
CLICK_FILE_MENU = True
if CLICK_FILE_MENU:
#ut.embed()
#QTest::keyClick(menu, Qt::Key_Down)
pass
timer2.delayed_screenshot_func = delayed_screenshot_func
timer2.timeout.connect(timer2.delayed_screenshot_func)
timer2.start(1)
back.mainwin.timer2 = timer2
guitool.activate_qwindow(back.mainwin)
#QPixmap.grabWindow(back.mainwin.winId()).save(fpath, 'jpg')
#ut.startfile(fpath)
#ut.embed()
pass
if params.args.postload_exit:
print('[main_cmd] postload exit')
sys.exit(0)
def postload_commands(ibs, back):
"""
Postload commands deal with a specific wbia database
wbia --db PZ_MTEST --occur "*All Images" --query 1
wbia --db PZ_MTEST --occur "*All Images" --query-intra
"""
params.parse_args()
if ut.NOT_QUIET:
logger.info('\n[main_cmd] postload_commands')
if params.args.view_database_directory:
logger.info('got arg --vdd')
vdd(ibs)
if params.args.set_default_dbdir:
sysres.set_default_dbdir(ibs.get_dbdir())
if params.args.update_query_cfg is not None:
# Set query parameters from command line using the --cfg flag
cfgdict = ut.parse_cfgstr_list(params.args.update_query_cfg)
logger.info('Custom cfgdict specified')
logger.info(ut.repr2(cfgdict))
ibs.update_query_cfg(**cfgdict)
if params.args.edit_notes:
ut.editfile(ibs.get_dbnotes_fpath(ensure=True))
if params.args.delete_cache:
ibs.delete_cache()
if params.args.delete_cache_complete:
ibs.delete_cache(delete_imagesets=True)
if params.args.delete_query_cache:
ibs.delete_qres_cache()
if params.args.set_all_species is not None:
ibs._overwrite_all_annot_species_to(params.args.set_all_species)
if params.args.dump_schema:
ibs.db.print_schema()
if ut.get_argflag('--ipynb'):
back.launch_ipy_notebook()
select_imgsetid = ut.get_argval(
('--select-imgsetid', '--imgsetid', '--occur', '--gsid'), None)
if select_imgsetid is not None:
logger.info('\n+ --- CMD SELECT IMGSETID=%r ---' % (select_imgsetid, ))
# Whoa: this doesnt work. weird.
# back.select_imgsetid(select_imgsetid)
# This might be the root of gui problems
# back.front._change_imageset(select_imgsetid)
back.front.select_imageset_tab(select_imgsetid)
logger.info('L ___ CMD SELECT IMGSETID=%r ___\n' % (select_imgsetid, ))
# Send commands to GUIBack
if params.args.select_aid is not None:
if back is not None:
try:
ibsfuncs.assert_valid_aids(ibs, (params.args.select_aid, ))
except AssertionError:
logger.info('Valid RIDs are: %r' % (ibs.get_valid_aids(), ))
raise
back.select_aid(params.args.select_aid)
if params.args.select_gid is not None:
back.select_gid(params.args.select_gid)
if params.args.select_nid is not None:
back.select_nid(params.args.select_nid)
select_name = ut.get_argval('--select-name')
if select_name is not None:
import wbia.gui.guiheaders as gh
back.ibswgt.select_table_indicies_from_text(gh.NAMES_TREE,
select_name,
allow_table_change=True)
if ut.get_argflag(
('--intra-occur-query', '--query-intra-occur', '--query-intra')):
back.special_query_funcs['intra_occurrence'](cfgdict={
'use_k_padding': False
})
qaid_list = ut.get_argval(('--query-aid', '--query'),
type_=list,
default=None)
if qaid_list is not None:
# qaid_list = params.args.query_aid
# fix stride case
if len(qaid_list) == 1 and isinstance(qaid_list[0], tuple):
qaid_list = list(qaid_list[0])
daids_mode = ut.get_argval('--daids-mode',
type_=str,
default=const.VS_EXEMPLARS_KEY)
back.compute_queries(qaid_list=qaid_list,
daids_mode=daids_mode,
ranks_top=10)
if ut.get_argflag('--inc-query'):
back.incremental_query()
if ut.get_argflag(('--dbinfo', '--display_dbinfo')):
back.display_dbinfo()
pass
aidcmd = ut.get_argval('--aidcmd', default=None)
aid = ut.get_argval('--aid', type_=int, default=1)
if aidcmd:
# aidcmd = 'Interact image'
metadata = ibs.get_annot_lazy_dict(aid)
annot_context_options = metadata['annot_context_options']
aidcmd_dict = dict(annot_context_options)
logger.info('aidcmd_dict = %s' % (ut.repr3(aidcmd_dict), ))
command = aidcmd_dict[aidcmd]
command()
# import utool
# utool.embed()
# back.start_web_server_parallel()
if ut.get_argflag('--start-web'):
back.start_web_server_parallel()
if ut.get_argflag('--name-tab'):
from wbia.gui.guiheaders import NAMES_TREE
back.front.set_table_tab(NAMES_TREE)
view = back.front.views[NAMES_TREE]
model = view.model()
view._set_sort(model.col_name_list.index('nAids'),
col_sort_reverse=True)
if ut.get_argflag('--graph'):
back.make_qt_graph_interface()
if params.args.postload_exit:
logger.info('[main_cmd] postload exit')
sys.exit(0)
def postload_commands(ibs, back):
""" Postload commands deal with a specific ibeis database """
if ut.NOT_QUIET:
print('\n[main_cmd] postload_commands')
if params.args.view_database_directory:
print('got arg --vdd')
vdd(ibs)
if params.args.set_default_dbdir:
sysres.set_default_dbdir(ibs.get_dbdir())
if params.args.update_query_cfg is not None:
# Set query parameters from command line using the --cfg flag
cfgdict = ut.parse_cfgstr_list(params.args.update_query_cfg)
print('Custom cfgdict specified')
print(ut.dict_str(cfgdict))
ibs.update_query_cfg(**cfgdict)
#print(ibs.cfg.query_cfg.get_cfgstr())
if params.args.edit_notes:
ut.editfile(ibs.get_dbnotes_fpath(ensure=True))
if params.args.delete_cache:
ibs.delete_cache()
if params.args.delete_cache_complete:
ibs.delete_cache(delete_chips=True, delete_encounters=True)
if params.args.delete_query_cache:
ibs.delete_qres_cache()
if params.args.set_notes is not None:
ibs.set_dbnotes(params.args.set_notes)
if params.args.set_aids_as_hard is not None:
aid_list = params.args.set_aids_as_hard
ibs.set_annot_is_hard(aid_list, [True] * len(aid_list))
if params.args.set_all_species is not None:
ibs._overwrite_all_annot_species_to(params.args.set_all_species)
if params.args.dump_schema:
ibs.db.print_schema()
if ut.get_argflag('--ipynb'):
back.launch_ipy_notebook()
select_eid = ut.get_argval(('--select-eid', '--eid',), None)
if select_eid is not None:
print('\n+ --- CMD SELECT EID=%r ---' % (select_eid,))
# Whoa: this doesnt work. weird.
#back.select_eid(select_eid)
# This might be the root of gui problems
#back.front._change_enc(select_eid)
back.front.select_encounter_tab(select_eid)
print('L ___ CMD SELECT EID=%r ___\n' % (select_eid,))
# Send commands to GUIBack
if params.args.select_aid is not None:
if back is not None:
try:
ibsfuncs.assert_valid_aids(ibs, (params.args.select_aid,))
except AssertionError:
print('Valid RIDs are: %r' % (ibs.get_valid_aids(),))
raise
back.select_aid(params.args.select_aid)
if params.args.select_gid is not None:
back.select_gid(params.args.select_gid)
if params.args.select_nid is not None:
back.select_nid(params.args.select_nid)
select_name = ut.get_argval('--select-name')
if select_name is not None:
import ibeis.gui.guiheaders as gh
back.ibswgt.select_table_indicies_from_text(gh.NAMES_TREE, select_name,
allow_table_change=True)
qaid_list = ut.get_argval(('--query-aid', '--query'), type_=list, default=None)
if qaid_list is not None:
#qaid_list = params.args.query_aid
# fix stride case
if len(qaid_list) == 1 and isinstance(qaid_list[0], tuple):
qaid_list = list(qaid_list[0])
daids_mode = ut.get_argval('--daids-mode', type_=str, default=const.VS_EXEMPLARS_KEY)
back.compute_queries(qaid_list=qaid_list,
daids_mode=daids_mode, ranks_lt=10)
if ut.get_argflag('--inc-query'):
back.incremental_query()
if ut.get_argflag(('--dbinfo', '--display_dbinfo')):
back.display_dbinfo()
pass
if ut.get_argflag('--start-web'):
back.start_web_server_parallel()
#screengrab_fpath = ut.get_argval('--screengrab')
#if screengrab_fpath:
# from guitool.__PYQT__.QtGui import QPixmap
# from PyQt4.QtTest import QTest
# from PyQt4.QtCore import Qt
# fpath = ut.truepath(screengrab_fpath)
# import guitool
# #ut.embed()
# timer2 = guitool.__PYQT__.QtCore.QTimer()
# done = [1000]
# def delayed_screenshot_func():
# if done[0] == 500:
# #back.mainwin.menubar.triggered.emit(back.mainwin.menuFile)
# print('Mouseclick')
# QTest.mouseClick(back.mainwin.menuFile, Qt.LeftButton)
# # This works
# #QTest.mouseClick(back.front.import_button, Qt.LeftButton)
# if done[0] == 1:
# timer2.stop()
# print('screengrab to %r' % (fpath,))
# screenimg = QPixmap.grabWindow(back.mainwin.winId())
# screenimg.save(fpath, 'jpg')
# ut.startfile(fpath)
# print('lub dub2')
# done[0] -= 1
# return None
# CLICK_FILE_MENU = True
# if CLICK_FILE_MENU:
# #ut.embed()
# #QTest::keyClick(menu, Qt::Key_Down)
# pass
# timer2.delayed_screenshot_func = delayed_screenshot_func
# timer2.timeout.connect(timer2.delayed_screenshot_func)
# timer2.start(1)
# back.mainwin.timer2 = timer2
# guitool.activate_qwindow(back.mainwin)
# #QPixmap.grabWindow(back.mainwin.winId()).save(fpath, 'jpg')
# #ut.startfile(fpath)
# #ut.embed()
# pass
if params.args.postload_exit:
print('[main_cmd] postload exit')
sys.exit(1)
