def main():
from bootstrapping_olympics.extra.reprep import ReprepPublisher
usage = ""
parser = OptionParser(usage=usage)
parser.disable_interspersed_args()
parser.add_option("-o",
dest='outdir',
default='diffeo_analysis',
help="Output directory [%default].")
parser.add_option("-p", dest='pickle', help="Saved agent state")
parser.add_option("-t", dest='templates', help="Directory with templates")
(options, args) = parser.parse_args()
if args:
raise Exception('Extra args')
print('Loading %r' % options.pickle)
with open(options.pickle) as f:
data = pickle.load(f)
print('(done)')
state = data.agent_state
confid = '%s-%s' % (data.id_robot, data.id_agent)
publisher = ReprepPublisher(confid)
dd = state['diffeo_dynamics']
actions = []
for cmd_index, de in dd.commands2dynamics.items():
original_cmd = dd.commands2label[cmd_index]
print('Summarizing %s' % original_cmd)
diffeo = de.summarize()
a = Action(
diffeo=diffeo,
label="u%s" % cmd_index, # index=cmd_index,
invertible=False,
primitive=True,
original_cmd=original_cmd)
actions.append(a)
# actions = [actions[1], actions[4]]
templates = list(load_templates(options.templates, shape=diffeo.d.shape))
for template in templates[:1]:
image = template.image
name = template.name
for action in actions:
section_name = '%s-%s_%s_%s' % (name, action, action.label,
action.original_cmd)
print(section_name)
S = publisher.section(section_name)
compute_effects(S, action, image)
filename = os.path.join(options.outdir, "%s-preds.html" % confid)
publisher.r.to_html(filename)
def main():
from bootstrapping_olympics.extra.reprep import ReprepPublisher
usage = ""
parser = OptionParser(usage=usage)
parser.disable_interspersed_args()
parser.add_option("-o", dest='outdir', default='diffeo_analysis',
help="Output directory [%default].")
parser.add_option("-p", dest='pickle',
help="Saved agent state")
parser.add_option("-t", dest='templates', help="Directory with templates")
(options, args) = parser.parse_args()
if args:
raise Exception('Extra args')
print('Loading %r' % options.pickle)
with open(options.pickle) as f:
data = pickle.load(f)
print('(done)')
state = data.agent_state
confid = '%s-%s' % (data.id_robot, data.id_agent)
publisher = ReprepPublisher(confid)
dd = state['diffeo_dynamics']
actions = []
for cmd_index, de in dd.commands2dynamics.items():
original_cmd = dd.commands2label[cmd_index]
print('Summarizing %s' % original_cmd)
diffeo = de.summarize()
a = Action(diffeo=diffeo, label="u%s" % cmd_index, # index=cmd_index,
invertible=False, primitive=True, original_cmd=original_cmd)
actions.append(a)
# actions = [actions[1], actions[4]]
templates = list(load_templates(options.templates, shape=diffeo.d.shape))
for template in templates[:1]:
image = template.image
name = template.name
for action in actions:
section_name = '%s-%s_%s_%s' % (name, action,
action.label, action.original_cmd)
print(section_name)
S = publisher.section(section_name)
compute_effects(S, action, image)
filename = os.path.join(options.outdir, "%s-preds.html" % confid)
publisher.r.to_html(filename)
