def get_texture(smap, dpi):
figure_args=dict(figsize=(2,2), facecolor='green')
a = dtu.CreateImageFromPylab(dpi=dpi, figure_args=figure_args)
frames = list(set(_.id_frame for _ in smap.points.values()))
id_frame = frames[0]
# print('frames: %s choose %s' % (frames, id_frame))
with a as pylab:
_plot_map_segments(smap, pylab, id_frame, plot_ref_segments=False)
pylab.axis('equal')
turn_all_axes_off(pylab)
pylab.tight_layout()
png = a.get_png()
return png
def draw_node_graph(filename,
G,
pos_func,
color_func=lambda _: 0.5,
label_func=lambda x: str(x),
cmap=None):
from reprep.plot_utils.axes import turn_all_axes_off
import matplotlib.pyplot as plt # XXX: use pylab argument
import networkx as nx
import os
import collections
nodes = list(G)
pos = dict((n, pos_func(n)) for n in nodes)
all_positions = [tuple(pos_func(n)) for n in nodes]
if len(all_positions) != len(set(all_positions)):
print('Warning, overlapping nodes')
y = collections.Counter(all_positions)
for p, num in y.items():
if num > 1:
print('- %d for %s' % (num, p))
for node, node_pos in pos.items():
if tuple(node_pos) == p:
print(' - %s ' % str(node))
pass
node_color = map(color_func, G)
labels = dict((n, label_func(n)) for n in G)
fig = plt.figure()
nx.draw_networkx(G,
with_labels=True,
pos=pos,
labels=labels,
node_color=node_color,
cmap=cmap)
plt.colorbar()
turn_all_axes_off(plt)
dirname = os.path.dirname(filename)
if not os.path.exists(dirname):
os.makedirs(dirname)
plt.savefig(filename)
plt.close(fig)
def report_tension(mdp, resample_res, resolution=0.33):
tension = resample_res['tension']
density_sf = resample_res['density_sf']
r = Report()
f = r.figure()
with f.plot('tension1') as pylab:
mdp.display_neigh_field_value(pylab, tension)
turn_all_axes_off(pylab)
with f.plot('density') as pylab:
display_sf_field_cont(mdp.get_grid(),
pylab,
density_sf,
res=density_sf)
turn_all_axes_off(pylab)
return r
def report_agent(res, pomdp):
agent = res['agent']
r = Report()
f = r.figure()
p_p0 = pomdp.get_start_dist_dist()
for i, (p0, _) in enumerate(p_p0.items()):
with f.plot('p0-%d' % i) as pylab:
pomdp.display_state_dist(pylab, p0)
turn_all_axes_off(pylab)
with r.subsection('states') as sub:
agent.report_states(sub)
with r.subsection('transitions') as sub:
agent.report_transitions(sub)
return r
def draw_node_graph(filename, G, pos_func,
color_func=lambda _: 0.5,
label_func=lambda x: str(x),
cmap=None):
from reprep.plot_utils.axes import turn_all_axes_off
import matplotlib.pyplot as plt # XXX: use pylab argument
import networkx as nx
import os
import collections
nodes = list(G)
pos = dict((n, pos_func(n)) for n in nodes)
all_positions = [tuple(pos_func(n)) for n in nodes]
if len(all_positions) != len(set(all_positions)):
print('Warning, overlapping nodes')
y = collections.Counter(all_positions)
for p, num in y.items():
if num > 1:
print('- %d for %s' % (num, p))
for node, node_pos in pos.items():
if tuple(node_pos) == p:
print(' - %s ' % str(node))
pass
node_color = map(color_func, G)
labels = dict((n, label_func(n)) for n in G)
fig = plt.figure()
nx.draw_networkx(G, with_labels=True, pos=pos, labels=labels,
node_color=node_color, cmap=cmap)
plt.colorbar()
turn_all_axes_off(plt)
dirname = os.path.dirname(filename)
if not os.path.exists(dirname):
os.makedirs(dirname)
plt.savefig(filename)
plt.close(fig)
def report_agent(res, pomdp):
agent = res['agent']
r = Report()
f = r.figure()
p_p0 = pomdp.get_start_dist_dist()
for i, (p0, _) in enumerate(p_p0.items()):
with f.plot('p0-%d' % i) as pylab:
pomdp.display_state_dist(pylab, p0)
turn_all_axes_off(pylab)
with r.subsection('states') as sub:
agent.report_states(sub)
with r.subsection('transitions') as sub:
agent.report_transitions(sub)
return r