def _create_explicit(l_tech):
v_fmt = 'p%s_v%s'
nattr = 'color="%s", href="%s"' % (commodity_color, url_fmt)
vattr = 'color="%s", href="model.%s"' % (tech_color, ffmt)
etattr = 'color="%s", sametail="%%s"' % arrowheadin_color
efattr = 'color="%s", samehead="%%s"' % arrowheadout_color
if show_capacity:
vattr = 'color="%s", label="p%%(p)s_v%%(v)s\\n' \
'Capacity = %%(val).2f", href="model.%s"' % (tech_color, ffmt)
# begin/end/vintage nodes
bnodes, enodes, vnodes, edges = set(), set(), set(), set()
for l_per, tmp, l_vin in g_processInputs:
if tmp != l_tech:
continue
for l_inp in ProcessInputs(l_per, l_tech, l_vin):
for l_out in ProcessOutputsByInput(l_per, l_tech, l_vin,
l_inp):
bnodes.add((l_inp, nattr % l_inp))
enodes.add((l_out, nattr % l_out))
attr_args = dict()
if show_capacity:
val = value(VintageCap[l_tech, l_vin])
attr_args.update(p=l_per, v=l_vin, val=val)
vnodes.add((v_fmt % (l_per, l_vin), vattr % attr_args))
edges.add((l_inp, v_fmt % (l_per, l_vin), etattr % l_inp))
edges.add((v_fmt % (l_per, l_vin), l_out, efattr % l_out))
if not (bnodes and enodes and vnodes and edges):
# apparently specified a technology in tech_resource, but never used
# it in efficiency.
return None
bnodes = create_text_nodes(bnodes, indent=2)
enodes = create_text_nodes(enodes, indent=2)
vnodes = create_text_nodes(vnodes)
edges = create_text_edges(edges)
dot_fname = fname % (l_tech, 'dot')
with open(fname, 'w') as f:
f.write(model_dot_fmt % dict(
tech=l_tech,
images_dir=images_dir,
home_color=home_color,
usedfont_color=usedfont_color,
dummy=dummystr,
bnodes=bnodes,
enodes=enodes,
vnodes=vnodes,
edges=edges,
))
return dot_fname
def CreatePartialSegmentsDiagram(**kwargs):
from temoa_lib import g_activeCapacityAvailable_pt, g_processInputs, \
ProcessVintages, ProcessInputs, ProcessOutputsByInput
M = kwargs.get('model')
ffmt = kwargs.get('image_format')
arrowheadin_color = kwargs.get('arrowheadin_color')
arrowheadout_color = kwargs.get('arrowheadout_color')
sb_vpbackg_color = kwargs.get('sb_vpbackg_color')
sb_vp_color = kwargs.get('sb_vp_color')
commodity_color = kwargs.get('commodity_color')
home_color = kwargs.get('home_color')
tech_color = kwargs.get('tech_color')
usedfont_color = kwargs.get('usedfont_color')
options = kwargs.get('options')
splinevar = options.splinevar
os.chdir('results')
slice_dot_fmt = """\
strict digraph model {
label = "Activity split of process %(tech)s, %(vintage)s in year %(period)s" ;
compound = "True" ;
concentrate = "True";
rankdir = "LR" ;
splines = "%(splinevar)s" ;
node [ style="filled" ] ;
edge [ arrowhead="vee" ] ;
subgraph cluster_slices {
label = "%(vintage)s Capacity: %(total_cap).2f" ;
color = "%(vintage_cluster_color)s" ;
rank = "same" ;
style = "filled" ;
node [ color="%(vintage_color)s", shape="box" ] ;
%(snodes)s
}
subgraph energy_carriers {
node [
color = "%(commodity_color)s",
fontcolor = "%(usedfont_color)s",
shape = "circle"
] ;
%(enodes)s
}
subgraph inputs {
edge [ color="%(input_color)s" ] ;
%(iedges)s
}
subgraph outputs {
edge [ color="%(output_color)s" ] ;
%(oedges)s
}
}
"""
enode_attr_fmt = 'href="../commodities/rc_%%s_%%s.%s"' % ffmt
for p, t in g_activeCapacityAvailable_pt:
total_cap = value(M.V_CapacityAvailableByPeriodAndTech[p, t])
for v in ProcessVintages(p, t):
if not M.V_ActivityByPeriodAndProcess[p, t, v]:
continue
cap = M.V_Capacity[t, v]
vnode = str(v)
for i in ProcessInputs(p, t, v):
for o in ProcessOutputsByInput(p, t, v, i):
# energy/vintage nodes, in/out edges
snodes, enodes, iedges, oedges = set(), set(), set(), set()
for s in M.time_season:
for d in M.time_of_day:
flowin = value(M.V_FlowIn[p, s, d, i, t, v, o])
if not flowin: continue
flowout = value(M.V_FlowOut[p, s, d, i, t, v, o])
snode = "%s, %s" % (s, d)
snodes.add((snode, None))
enodes.add((i, enode_attr_fmt % (i, p)))
enodes.add((o, enode_attr_fmt % (o, p)))
iedges.add((i, snode, 'label="%.2f"' % flowin))
oedges.add((snode, o, 'label="%.2f"' % flowout))
if not snodes: continue
snodes = create_text_nodes(snodes, indent=2)
enodes = create_text_nodes(enodes, indent=2)
iedges = create_text_edges(iedges, indent=2)
oedges = create_text_edges(oedges, indent=2)
fname = 'results_%s_p%sv%s_segments.' % (t, p, v)
with open(fname + 'dot', 'w') as f:
f.write(slice_dot_fmt % dict(
period=p,
tech=t,
vintage=v,
ffmt=ffmt,
commodity_color=commodity_color,
usedfont_color=usedfont_color,
home_color=home_color,
input_color=arrowheadin_color,
output_color=arrowheadout_color,
vintage_cluster_color=sb_vpbackg_color,
vintage_color=sb_vp_color,
splinevar=splinevar,
total_cap=total_cap,
snodes=snodes,
enodes=enodes,
iedges=iedges,
oedges=oedges,
))
cmd = ('dot', '-T' + ffmt, '-o' + fname + ffmt,
fname + 'dot')
call(cmd)
os.chdir('..')
def CreateTechResultsDiagrams(**kwargs):
from temoa_lib import g_activeCapacityAvailable_pt, g_processInputs, \
ProcessVintages, ProcessInputs, ProcessOutputsByInput
M = kwargs.get('model')
ffmt = kwargs.get('image_format')
images_dir = kwargs.get('images_dir')
arrowheadin_color = kwargs.get('arrowheadin_color')
arrowheadout_color = kwargs.get('arrowheadout_color')
sb_vpbackg_color = kwargs.get('sb_vpbackg_color')
sb_vp_color = kwargs.get('sb_vp_color')
commodity_color = kwargs.get('commodity_color')
home_color = kwargs.get('home_color')
tech_color = kwargs.get('tech_color')
usedfont_color = kwargs.get('usedfont_color')
options = kwargs.get('options')
splinevar = options.splinevar
os.chdir('results')
model_dot_fmt = """\
strict digraph model {
label = "Results for %(tech)s in %(period)s" ;
compound = "True" ;
concentrate = "True";
rankdir = "LR" ;
splines = "%(splinevar)s" ;
node [ style="filled" ] ;
edge [ arrowhead="vee" ] ;
subgraph cluster_vintages {
label = "Vintages\\nCapacity: %(total_cap).2f" ;
href = "results%(period)s.%(ffmt)s" ;
style = "filled"
color = "%(vintage_cluster_color)s"
node [ color="%(vintage_color)s", shape="box" ] ;
%(vnodes)s
}
subgraph energy_carriers {
node [
color = "%(commodity_color)s",
fontcolor = "%(usedfont_color)s",
shape = "circle"
] ;
%(enodes)s
}
subgraph inputs {
edge [ color="%(input_color)s" ] ;
%(iedges)s
}
subgraph outputs {
edge [ color="%(output_color)s" ] ;
%(oedges)s
}
}
"""
enode_attr_fmt = 'href="../commodities/rc_%%s_%%s.%s"' % ffmt
vnode_attr_fmt = 'href="results_%%s_p%%sv%%s_segments.%s", ' % ffmt
vnode_attr_fmt += 'label="%s\\nCap: %.2f"'
for per, tech in g_activeCapacityAvailable_pt:
total_cap = value(M.V_CapacityAvailableByPeriodAndTech[per, tech])
# energy/vintage nodes, in/out edges
enodes, vnodes, iedges, oedges = set(), set(), set(), set()
for l_vin in ProcessVintages(per, tech):
if not M.V_ActivityByPeriodAndProcess[per, tech, l_vin]:
continue
cap = M.V_Capacity[tech, l_vin]
vnode = str(l_vin)
for l_inp in ProcessInputs(per, tech, l_vin):
for l_out in ProcessOutputsByInput(per, tech, l_vin, l_inp):
flowin = sum(
value(M.V_FlowIn[per, ssn, tod, l_inp, tech, l_vin,
l_out]) for ssn in M.time_season
for tod in M.time_of_day)
flowout = sum(
value(M.V_FlowOut[per, ssn, tod, l_inp, tech, l_vin,
l_out]) for ssn in M.time_season
for tod in M.time_of_day)
index = (per, l_inp, tech, l_vin)
vnodes.add(
(vnode,
vnode_attr_fmt % (tech, per, l_vin, l_vin, cap)))
enodes.add((l_inp, enode_attr_fmt % (l_inp, per)))
enodes.add((l_out, enode_attr_fmt % (l_out, per)))
iedges.add((l_inp, vnode, 'label="%.2f"' % flowin))
oedges.add((vnode, l_out, 'label="%.2f"' % flowout))
if not vnodes: continue
enodes = create_text_nodes(enodes, indent=2)
vnodes = create_text_nodes(vnodes, indent=2)
iedges = create_text_edges(iedges, indent=2)
oedges = create_text_edges(oedges, indent=2)
fname = 'results_%s_%s.' % (tech, per)
with open(fname + 'dot', 'w') as f:
f.write(model_dot_fmt % dict(
images_dir=images_dir,
tech=tech,
period=per,
ffmt=ffmt,
commodity_color=commodity_color,
usedfont_color=usedfont_color,
home_color=home_color,
input_color=arrowheadin_color,
output_color=arrowheadout_color,
vintage_cluster_color=sb_vpbackg_color,
vintage_color=sb_vp_color,
splinevar=splinevar,
total_cap=total_cap,
vnodes=vnodes,
enodes=enodes,
iedges=iedges,
oedges=oedges,
))
cmd = ('dot', '-T' + ffmt, '-o' + fname + ffmt, fname + 'dot')
call(cmd)
os.chdir('..')
def CreateMainModelDiagram(**kwargs):
from temoa_lib import g_processInputs, ProcessInputs, ProcessOutputsByInput
M = kwargs.get('model')
ffmt = kwargs.get('image_format')
images_dir = kwargs.get('images_dir')
arrowheadin_color = kwargs.get('arrowheadin_color')
arrowheadout_color = kwargs.get('arrowheadout_color')
commodity_color = kwargs.get('commodity_color')
home_color = kwargs.get('home_color')
tech_color = kwargs.get('tech_color')
usedfont_color = kwargs.get('usedfont_color')
fname = 'simple_model.'
model_dot_fmt = """\
strict digraph model {
rankdir = "LR" ;
// Default node and edge attributes
node [ style="filled" ] ;
edge [ arrowhead="vee", labelfontcolor="lightgreen" ] ;
// Define individual nodes
subgraph techs {
node [ color="%(tech_color)s", shape="box" ] ;
%(tnodes)s
}
subgraph energy_carriers {
node [ color="%(commodity_color)s", shape="circle" ] ;
%(enodes)s
}
// Define edges and any specific edge attributes
subgraph inputs {
edge [ color="%(arrowheadin_color)s" ] ;
%(iedges)s
}
subgraph outputs {
edge [ color="%(arrowheadout_color)s" ] ;
%(oedges)s
}
}
"""
tech_attr_fmt = 'href="processes/process_%%s.%s"' % ffmt
carrier_attr_fmt = 'href="commodities/commodity_%%s.%s"' % ffmt
# edge/tech nodes, in/out edges
enodes, tnodes, iedges, oedges = set(), set(), set(), set()
for l_per, l_tech, l_vin in g_processInputs:
tnodes.add((l_tech, tech_attr_fmt % l_tech))
for l_inp in ProcessInputs(l_per, l_tech, l_vin):
enodes.add((l_inp, carrier_attr_fmt % l_inp))
for l_out in ProcessOutputsByInput(l_per, l_tech, l_vin, l_inp):
enodes.add((l_out, carrier_attr_fmt % l_out))
iedges.add((l_inp, l_tech, None))
oedges.add((l_tech, l_out, None))
enodes = create_text_nodes(enodes, indent=2)
tnodes = create_text_nodes(tnodes, indent=2)
iedges = create_text_edges(iedges, indent=2)
oedges = create_text_edges(oedges, indent=2)
with open(fname + 'dot', 'w') as f:
f.write(model_dot_fmt % dict(
images_dir=images_dir,
arrowheadin_color=arrowheadin_color,
arrowheadout_color=arrowheadout_color,
commodity_color=commodity_color,
home_color=home_color,
tech_color=tech_color,
usedfont_color=usedfont_color,
enodes=enodes,
tnodes=tnodes,
iedges=iedges,
oedges=oedges,
))
del enodes, tnodes, iedges, oedges
cmd = ('dot', '-T' + ffmt, '-o' + fname + ffmt, fname + 'dot')
call(cmd)
def _create_separate(l_tech):
# begin/end/period/vintage nodes
bnodes, enodes, pnodes, vnodes = set(), set(), set(), set()
eedges, vedges = set(), set()
periods = set() # used to obtain the first vintage/period, so that
vintages = set() # all connections can point to a common point
for l_per, tmp, l_vin in g_processInputs:
if tmp != l_tech: continue
periods.add(l_per)
vintages.add(l_vin)
if not (periods and vintages):
# apparently specified a technology in tech_resource, but never used
# it in efficiency.
return None
mid_period = sorted(periods)[len(periods) // 2] # // is 'floordiv'
mid_vintage = sorted(vintages)[len(vintages) // 2]
del periods, vintages
p_fmt = 'p_%s'
v_fmt = 'v_%s'
niattr = 'color="%s", href="%s"' % (sb_incom_color, url_fmt
) # inp node
noattr = 'color="%s", href="%s"' % (sb_outcom_color, url_fmt
) # out node
eattr = 'color="%s"' # edge attribute
pattr = None # period node attribute
vattr = None # vintage node attribute
# "cluster-in attribute", "cluster-out attribute"
ciattr = 'color="%s", lhead="cluster_vintage"' % arrowheadin_color
coattr = 'color="%s", ltail="cluster_period"' % arrowheadout_color
if show_capacity:
pattr_fmt = 'label="p%s\\nTotal Capacity: %.2f"'
vattr_fmt = 'label="v%s\\nCapacity: %.2f"'
j = 0
for l_per, tmp, l_vin in g_processInputs:
if tmp != l_tech: continue
if show_capacity:
pattr = pattr_fmt % (l_per, value(PeriodCap[l_per, l_tech]))
vattr = vattr_fmt % (l_vin, value(VintageCap[l_tech, l_vin]))
pnodes.add((p_fmt % l_per, pattr))
vnodes.add((v_fmt % l_vin, vattr))
for l_inp in ProcessInputs(l_per, l_tech, l_vin):
for l_out in ProcessOutputsByInput(l_per, l_tech, l_vin,
l_inp):
# use color_list for the option 1 subgraph arrows 1, so as to
# more easily delineate the connections in the graph.
rainbow = color_list[j]
j = (j + 1) % len(color_list)
enodes.add((l_inp, niattr % l_inp))
bnodes.add((l_out, noattr % l_out))
eedges.add((l_inp, v_fmt % mid_vintage, ciattr))
vedges.add((v_fmt % l_vin, p_fmt % l_per, eattr % rainbow))
eedges.add((p_fmt % mid_period, '%s' % l_out, coattr))
bnodes = create_text_nodes(bnodes, indent=2) # beginning nodes
enodes = create_text_nodes(enodes, indent=2) # ending nodes
pnodes = create_text_nodes(pnodes, indent=2) # period nodes
vnodes = create_text_nodes(vnodes, indent=2) # vintage nodes
eedges = create_text_edges(eedges, indent=2) # external edges
vedges = create_text_edges(vedges, indent=2) # vintage edges
dot_fname = fname % (l_tech, 'dot')
with open(dot_fname, 'w') as f:
f.write(model_dot_fmt % dict(
cluster_url='../simple_model.%s' % ffmt,
graph_label=l_tech,
dummy=dummystr,
images_dir=images_dir,
splinevar=splinevar,
clusternode_color=sb_vp_color,
period_color=sb_vpbackg_color,
vintage_color=sb_vpbackg_color,
usedfont_color=usedfont_color,
home_color=home_color,
bnodes=bnodes,
enodes=enodes,
pnodes=pnodes,
vnodes=vnodes,
eedges=eedges,
vedges=vedges,
))
return dot_fname
def CreateMainResultsDiagram(**kwargs):
from temoa_lib import ProcessVintages, ProcessInputs, ProcessOutputs, \
ValidActivity
M = kwargs.get('model')
images_dir = kwargs.get('images_dir')
ffmt = kwargs.get('image_format')
options = kwargs.get('options')
arrowheadin_color = kwargs.get('arrowheadin_color')
arrowheadout_color = kwargs.get('arrowheadout_color')
commodity_color = kwargs.get('commodity_color')
tech_color = kwargs.get('tech_color')
unused_color = kwargs.get('unused_color')
unusedfont_color = kwargs.get('unusedfont_color')
usedfont_color = kwargs.get('usedfont_color')
splinevar = options.splinevar
os.chdir('results')
results_dot_fmt = """\
strict digraph model {
label = "Results for %(period)s"
rankdir = "LR" ;
smoothtype = "power_dist" ;
splines = "%(splinevar)s" ;
node [ style="filled" ] ;
edge [ arrowhead="vee" ] ;
subgraph unused_techs {
node [
color = "%(unused_color)s",
fontcolor = "%(unusedfont_color)s",
shape = "box"
] ;
%(dtechs)s
}
subgraph unused_energy_carriers {
node [
color = "%(unused_color)s",
fontcolor = "%(unusedfont_color)s",
shape = "circle"
] ;
%(dcarriers)s
}
subgraph unused_emissions {
node [
color = "%(unused_color)s",
fontcolor = "%(unusedfont_color)s",
shape = "circle"
]
%(demissions)s
}
subgraph in_use_techs {
node [
color = "%(tech_color)s",
fontcolor = "%(usedfont_color)s",
shape = "box"
] ;
%(etechs)s
}
subgraph in_use_energy_carriers {
node [
color = "%(commodity_color)s",
fontcolor = "%(usedfont_color)s",
shape = "circle"
] ;
%(ecarriers)s
}
subgraph in_use_emissions {
node [
color = "%(commodity_color)s",
fontcolor = "%(usedfont_color)s",
shape = "circle"
] ;
%(eemissions)s
}
subgraph unused_flows {
edge [ color="%(unused_color)s" ]
%(dflows)s
}
subgraph in_use_flows {
subgraph inputs {
edge [ color="%(arrowheadin_color)s" ] ;
%(eflowsi)s
}
subgraph outputs {
edge [ color="%(arrowheadout_color)s" ] ;
%(eflowso)s
}
}
}
"""
tech_attr_fmt = 'label="%%s\\nCapacity: %%.2f", href="results_%%s_%%s.%s"'
tech_attr_fmt %= ffmt
commodity_fmt = 'href="../commodities/rc_%%s_%%s.%s"' % ffmt
flow_fmt = 'label="%.2f"'
V_Cap = M.V_CapacityAvailableByPeriodAndTech
FI = M.V_FlowIn
FO = M.V_FlowOut
EI = M.V_EnergyConsumptionByPeriodInputAndTech # Energy In
EO = M.V_ActivityByPeriodTechAndOutput # Energy Out
EmiO = M.V_EmissionActivityByPeriodAndTech
epsilon = 0.005 # we only care about last two decimals
# but perhaps this should be configurable? Not until we can do this
# both after the fact (i.e. not synchronous with a solve), and via a
# configuration file.
for pp in M.time_optimize:
# enabled/disabled techs/carriers/emissions/flows in/out
etechs, dtechs, ecarriers = set(), set(), set()
eemissions = set()
eflowsi, eflowso, dflows = set(), set(), set() # edges
usedc, usede = set(), set() # used carriers, used emissions
for tt in M.tech_all:
if (pp, tt) not in V_Cap: continue
cap = V_Cap[pp, tt]
if cap:
etechs.add((tt, tech_attr_fmt % (tt, cap, tt, pp)))
else:
dtechs.add((tt, None))
for vv in ProcessVintages(pp, tt):
for ii in ProcessInputs(pp, tt, vv):
inp = value(EI[pp, ii, tt])
if inp >= epsilon:
eflowsi.add((ii, tt, flow_fmt % inp))
ecarriers.add((ii, commodity_fmt % (ii, pp)))
usedc.add(ii)
else:
dflows.add((ii, tt, None))
for oo in ProcessOutputs(pp, tt, vv):
out = value(EO[pp, tt, oo])
if out >= epsilon:
eflowso.add((tt, oo, flow_fmt % out))
ecarriers.add((oo, commodity_fmt % (oo, pp)))
usedc.add(oo)
else:
dflows.add((tt, oo, None))
for ee, ii, tt, vv, oo in M.EmissionActivity.sparse_keys():
if ValidActivity(pp, tt, vv):
amt = value(EmiO[ee, pp, tt])
if amt < epsilon: continue
eflowso.add((tt, ee, flow_fmt % amt))
eemissions.add((ee, None))
usede.add(ee)
dcarriers = set(
(cc, None) for cc in M.commodity_carrier if cc not in usedc)
demissions = set(
(ee, None) for ee in M.commodity_emissions if ee not in usede)
dtechs = create_text_nodes(dtechs, indent=2)
etechs = create_text_nodes(etechs, indent=2)
dcarriers = create_text_nodes(dcarriers, indent=2)
ecarriers = create_text_nodes(ecarriers, indent=2)
demissions = create_text_nodes(demissions, indent=2)
eemissions = create_text_nodes(eemissions, indent=2)
dflows = create_text_edges(dflows, indent=2)
eflowsi = create_text_edges(eflowsi, indent=3)
eflowso = create_text_edges(eflowso, indent=3)
# link to periods: Do we want to include this in the SVG, or just let the
# modeler use the standard browser buttons? I think the latter.
# for randstr in l_perset:
# url = 'results%s.%s' % (randstr, ffmt)
# l_file.write (str(randstr) + '[URL="' + url + '",fontcolor=' + usedfont_color + ', rank="min", style=filled, shape=box, color=' + menu_color + '];\n')
# l_file.write( images + '[URL="..",shape=house, style=filled, fontcolor=' + usedfont_color + ', color=' + home_color + '];\n')
# l_file.write ("}\n");
# l_file.close()
fname = 'results%s.' % pp
with open(fname + 'dot', 'w') as f:
f.write(results_dot_fmt % dict(
period=pp,
splinevar=splinevar,
arrowheadin_color=arrowheadin_color,
arrowheadout_color=arrowheadout_color,
commodity_color=commodity_color,
tech_color=tech_color,
unused_color=unused_color,
unusedfont_color=unusedfont_color,
usedfont_color=usedfont_color,
dtechs=dtechs,
etechs=etechs,
dcarriers=dcarriers,
ecarriers=ecarriers,
demissions=demissions,
eemissions=eemissions,
dflows=dflows,
eflowsi=eflowsi,
eflowso=eflowso,
))
cmd = ('dot', '-T' + ffmt, '-o' + fname + ffmt, fname + 'dot')
call(cmd)
os.chdir('..')
def CreateCompleteEnergySystemDiagram(**kwargs):
"""\
These first couple versions of CreateModelDiagram do not fully work, and should
be thought of merely as "proof of concept" code. They create Graphviz DOT files
and equivalent PDFs, but the graphics are not "correct" representations of the
model. Specifically, there are currently a few artifacts and missing pieces:
Artifacts:
* Though the graph is "roughly" a left-right DAG, certain pieces currently a
swapped around, especially on the left-hand side of the image. This makes
the graph a bit harder to visually follow.
* Especially with the birth of energy, there are a few cycles. For example,
with the way the model currently creates energy, the graph makes it seem as
if 'imp_coal' also receives coal, when it should only export coal.
Initially known missing pieces:
* How should the graph represent the notion of periods?
* How should the graph represent the vintages?
* Should the graph include time slices? (e.g. day, season)
Notes:
* For _any_ decently sized system, displaying this type of graph of the entire
model will be infeasible, or effectively unusable. We need a way to
dynamically look at only subsections of the graph, while still giving a 10k'
foot view of the overall system.
* We need to create a system that puts results into a database, or common result
format, such that we can archive them for later. In this manner, directly
creating graphs at the point of model instantiation and running is not the
right place. Creating graphs needs to be a post processing action, and less
tightly coupled (not coupled at all!) to the internal Pyomo data structure.
"""
from temoa_lib import g_activeActivity_ptv, ProcessInputs, ProcessOutputs
M = kwargs.get('model')
ffmt = kwargs.get('image_format')
commodity_color = kwargs.get('commodity_color')
input_color = kwargs.get('arrowheadin_color')
output_color = kwargs.get('arrowheadout_color')
tech_color = kwargs.get('tech_color')
data = """\
// This file is generated by the --graph_format option of the Temoa model. It
// is a Graphviz DOT language text description of a Temoa model instance. For
// the curious, Graphviz will read this file to create an equivalent image in
// a number of formats, including SVG, PNG, GIF, and PDF. For example, here
// is how one might invoke Graphviz to create an SVG image from the dot file.
//
// dot -Tsvg -o model.svg model.dot
//
// For more information, see the Graphviz homepage: http://graphviz.org/
strict digraph TemoaModel {
rankdir = "LR"; // The direction of the graph goes from Left to Right
node [ style="filled" ] ;
edge [ arrowhead="vee", label=" " ] ;
subgraph technologies {
node [ color="%(tech_color)s", shape="box" ] ;
%(techs)s
}
subgraph energy_carriers {
node [ color="%(carrier_color)s", shape="circle" ] ;
%(carriers)s
}
subgraph inputs {
edge [ color="%(input_color)s" ] ;
%(inputs)s
}
subgraph outputs {
edge [ color="%(output_color)s" ];
%(outputs)s
}
}
"""
carriers, techs = set(), set()
inputs, outputs = set(), set()
p_fmt = '%s, %s, %s' # "Process format"
for l_per, l_tech, l_vin in g_activeActivity_ptv:
techs.add((p_fmt % (l_per, l_tech, l_vin), None))
for l_inp in ProcessInputs(l_per, l_tech, l_vin):
carriers.add((l_inp, None))
inputs.add((l_inp, p_fmt % (l_per, l_tech, l_vin), None))
for l_out in ProcessOutputs(l_per, l_tech, l_vin):
carriers.add((l_out, None))
outputs.add((p_fmt % (l_per, l_tech, l_vin), l_out, None))
techs = create_text_nodes(techs, indent=2)
carriers = create_text_nodes(carriers, indent=2)
inputs = create_text_edges(inputs, indent=2)
outputs = create_text_edges(outputs, indent=2)
fname = 'all_vintages_model.'
with open(fname + 'dot', 'w') as f:
f.write(data % dict(
input_color='forestgreen',
output_color='firebrick',
carrier_color='lightsteelblue',
tech_color='darkseagreen',
techs=techs,
carriers=carriers,
inputs=inputs,
outputs=outputs,
))
# Outsource to Graphviz via the old Unix standby: temporary files
cmd = ('dot', '-T' + ffmt, '-o' + fname + ffmt, fname + 'dot')
call(cmd)
def CreateCommodityPartialResults(**kwargs):
from temoa_lib import g_processInputs, g_processOutputs, \
ProcessInputs, ProcessOutputsByInput, ProcessesByInput, ProcessesByOutput
M = kwargs.get('model')
ffmt = kwargs.get('image_format')
images_dir = kwargs.get('images_dir')
sb_arrow_color = kwargs.get('sb_arrow_color')
commodity_color = kwargs.get('commodity_color')
home_color = kwargs.get('home_color')
tech_color = kwargs.get('tech_color')
usedfont_color = kwargs.get('usedfont_color')
unused_color = kwargs.get('unused_color')
options = kwargs.get('options')
splinevar = options.splinevar
os.chdir('commodities')
commodity_dot_fmt = """\
strict digraph result_commodity_%(commodity)s {
label = "%(commodity)s - %(period)s" ;
compound = "True" ;
concentrate = "True" ;
rankdir = "LR" ;
splines = "True" ;
node [ shape="box", style="filled" ] ;
edge [
arrowhead = "vee",
fontsize = "8",
label = " ",
labelfloat = "False",
labelfontcolor = "lightgreen"
len = "2",
weight = "0.5",
] ;
%(resource_node)s
subgraph used_techs {
node [ color="%(tech_color)s" ] ;
%(used_nodes)s
}
subgraph used_techs {
node [ color="%(unused_color)s" ] ;
%(unused_nodes)s
}
subgraph in_use_flows {
edge [ color="%(sb_arrow_color)s" ] ;
%(used_edges)s
}
subgraph unused_flows {
edge [ color="%(unused_color)s" ] ;
%(unused_edges)s
}
}
"""
FI = M.V_FlowIn
FO = M.V_FlowOut
used_carriers, used_techs = set(), set()
for p, t, v in g_processInputs:
for i in ProcessInputs(p, t, v):
for o in ProcessOutputsByInput(p, t, v, i):
flowin = sum(
value(FI[p, s, d, i, t, v, o]) for s in M.time_season
for d in M.time_of_day)
if flowin:
flowout = sum(
value(FO[p, s, d, i, t, v, o]) for s in M.time_season
for d in M.time_of_day)
used_carriers.update(g_processInputs[p, t, v])
used_carriers.update(g_processOutputs[p, t, v])
used_techs.add(t)
period_results_url_fmt = '../results/results%%s.%s' % ffmt
node_attr_fmt = 'href="../results/results_%%s_%%s.%s"' % ffmt
rc_node_fmt = 'color="%s", href="%s", shape="circle"'
for l_per in M.time_future:
url = period_results_url_fmt % l_per
for l_carrier in used_carriers:
# enabled/disabled nodes/edges
enodes, dnodes, eedges, dedges = set(), set(), set(), set()
rcnode = ((l_carrier, rc_node_fmt % (commodity_color, url)), )
for l_tech, l_vin in ProcessesByInput(l_carrier):
if l_tech in used_techs:
enodes.add((l_tech, node_attr_fmt % (l_tech, l_per)))
eedges.add((l_carrier, l_tech, None))
else:
dnodes.add((l_tech, None))
dedges.add((l_carrier, l_tech, None))
for l_tech, l_vin in ProcessesByOutput(l_carrier):
if l_tech in used_techs:
enodes.add((l_tech, node_attr_fmt % (l_tech, l_per)))
eedges.add((l_tech, l_carrier, None))
else:
dnodes.add((l_tech, None))
dedges.add((l_tech, l_carrier, None))
rcnode = create_text_nodes(rcnode)
enodes = create_text_nodes(enodes, indent=2)
dnodes = create_text_nodes(dnodes, indent=2)
eedges = create_text_edges(eedges, indent=2)
dedges = create_text_edges(dedges, indent=2)
fname = 'rc_%s_%s.' % (l_carrier, l_per)
with open(fname + 'dot', 'w') as f:
f.write(commodity_dot_fmt % dict(
images_dir=images_dir,
home_color=home_color,
usedfont_color=usedfont_color,
sb_arrow_color=sb_arrow_color,
tech_color=tech_color,
commodity=l_carrier,
period=l_per,
unused_color=unused_color,
resource_node=rcnode,
used_nodes=enodes,
unused_nodes=dnodes,
used_edges=eedges,
unused_edges=dedges,
))
cmd = ('dot', '-T' + ffmt, '-o' + fname + ffmt, fname + 'dot')
call(cmd)
os.chdir('..')
