| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190 |
- <!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN"
- "http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd">
- <html xmlns="http://www.w3.org/1999/xhtml">
- <head>
- <meta http-equiv="X-UA-Compatible" content="IE=Edge" />
- <meta http-equiv="Content-Type" content="text/html; charset=utf-8" />
- <title>pypdevsbbl.tracers.PlotTracer — Python(P)DEVS-BBL documentation</title>
- <link rel="stylesheet" href="../../../_static/nature.css" type="text/css" />
- <link rel="stylesheet" href="../../../_static/pygments.css" type="text/css" />
- <link rel="stylesheet" type="text/css" href="../../../_static/custom.css" />
- <script type="text/javascript" id="documentation_options" data-url_root="../../../" src="../../../_static/documentation_options.js"></script>
- <script type="text/javascript" src="../../../_static/jquery.js"></script>
- <script type="text/javascript" src="../../../_static/underscore.js"></script>
- <script type="text/javascript" src="../../../_static/doctools.js"></script>
- <script type="text/javascript" src="../../../_static/language_data.js"></script>
- <script async="async" type="text/javascript" src="https://cdnjs.cloudflare.com/ajax/libs/mathjax/2.7.5/latest.js?config=TeX-AMS-MML_HTMLorMML"></script>
- <link rel="index" title="Index" href="../../../genindex.html" />
- <link rel="search" title="Search" href="../../../search.html" />
- </head><body>
- <div class="related" role="navigation" aria-label="related navigation">
- <h3>Navigation</h3>
- <ul>
- <li class="right" style="margin-right: 10px">
- <a href="../../../genindex.html" title="General Index"
- accesskey="I">index</a></li>
- <li class="right" >
- <a href="../../../py-modindex.html" title="Python Module Index"
- >modules</a> |</li>
- <li class="nav-item nav-item-0"><a href="../../../index.html">Python(P)DEVS-BBL documentation</a> »</li>
- <li class="nav-item nav-item-1"><a href="../../index.html" >Module code</a> »</li>
- <li class="nav-item nav-item-2"><a href="../tracers.html" accesskey="U">pypdevsbbl.tracers</a> »</li>
- </ul>
- </div>
- <div class="document">
- <div class="documentwrapper">
- <div class="bodywrapper">
- <div class="body" role="main">
-
- <h1>Source code for pypdevsbbl.tracers.PlotTracer</h1><div class="highlight"><pre>
- <span></span><span class="c1"># Copyright 2020 Modelling, Simulation and Design Lab (MSDL) at</span>
- <span class="c1"># McGill University and the University of Antwerp (http://msdl.cs.mcgill.ca/)</span>
- <span class="c1">#</span>
- <span class="c1"># Licensed under the Apache License, Version 2.0 (the "License");</span>
- <span class="c1"># you may not use this file except in compliance with the License.</span>
- <span class="c1"># You may obtain a copy of the License at</span>
- <span class="c1">#</span>
- <span class="c1"># http://www.apache.org/licenses/LICENSE-2.0</span>
- <span class="c1">#</span>
- <span class="c1"># Unless required by applicable law or agreed to in writing, software</span>
- <span class="c1"># distributed under the License is distributed on an "AS IS" BASIS,</span>
- <span class="c1"># WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.</span>
- <span class="c1"># See the License for the specific language governing permissions and</span>
- <span class="c1"># limitations under the License.</span>
- <span class="sd">"""This file contains the tracer that can provide plots of the :mod:`pypdevsbbl.generic.collectors`."""</span>
- <span class="kn">from</span> <span class="nn">pypdevsbbl.generic.collectors</span> <span class="k">import</span> <span class="n">BaseCollector</span>
- <span class="kn">from</span> <span class="nn">pypdevsbbl.tracers</span> <span class="k">import</span> <span class="n">BaseTracer</span>
- <span class="kn">import</span> <span class="nn">matplotlib.pyplot</span> <span class="k">as</span> <span class="nn">plt</span>
- <span class="kn">import</span> <span class="nn">inspect</span>
- <div class="viewcode-block" id="PlotTracer"><a class="viewcode-back" href="../../../pypdevsbbl.tracers.PlotTracer.html#pypdevsbbl.tracers.PlotTracer.PlotTracer">[docs]</a><span class="k">class</span> <span class="nc">PlotTracer</span><span class="p">(</span><span class="n">BaseTracer</span><span class="p">):</span>
- <span class="sd">"""Plots data from the provided collectors.</span>
- <span class="sd"> Note:</span>
- <span class="sd"> Because of how :mod:`matplotlib` and :mod:`pypdevs` work, this</span>
- <span class="sd"> tracer only seems to be working within a `Jupyter` notebook.</span>
- <span class="sd"> See :file:`src/notebook/PlotTracer.ipynb` for an example on how</span>
- <span class="sd"> to do this.</span>
- <span class="sd"> Args:</span>
- <span class="sd"> uid: A unique identifier for distributed simulation.</span>
- <span class="sd"> Will be set automatically.</span>
- <span class="sd"> server: The server that runs the distributed simulation.</span>
- <span class="sd"> Will be set automatically.</span>
- <span class="sd"> *collectors: The collectors to plot. This is a pair in the</span>
- <span class="sd"> form of ``(BaseCollector, Axis)`` or a triple in</span>
- <span class="sd"> the form of ``(BaseDEVS, Axis, func)``, where</span>
- <span class="sd"> ``BaseCollector`` or ``BaseDEVS`` are instances</span>
- <span class="sd"> of those classes, ``Axis`` a :class:`matplotlib.Axes`</span>
- <span class="sd"> and ``func`` a callable that takes the first two</span>
- <span class="sd"> members and plots the data on the axis.</span>
- <span class="sd"> Additionally, the tuples/lists may be extended</span>
- <span class="sd"> with a thrird/fourth member, which is an additional</span>
- <span class="sd"> argument to pass to the plot function. What this is</span>
- <span class="sd"> for each collector is described in their docs.</span>
- <span class="sd"> """</span>
- <span class="k">def</span> <span class="nf">__init__</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">uid</span><span class="p">,</span> <span class="n">server</span><span class="p">,</span> <span class="o">*</span><span class="n">collectors</span><span class="p">):</span>
- <span class="nb">super</span><span class="p">(</span><span class="n">PlotTracer</span><span class="p">,</span> <span class="bp">self</span><span class="p">)</span><span class="o">.</span><span class="fm">__init__</span><span class="p">(</span><span class="n">uid</span><span class="p">,</span> <span class="n">server</span><span class="p">,</span> <span class="kc">None</span><span class="p">)</span>
- <span class="bp">self</span><span class="o">.</span><span class="n">blocks</span> <span class="o">=</span> <span class="p">{}</span>
- <span class="k">for</span> <span class="n">ix</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="nb">len</span><span class="p">(</span><span class="n">collectors</span><span class="p">)):</span>
- <span class="bp">cls</span> <span class="o">=</span> <span class="n">collectors</span><span class="p">[</span><span class="n">ix</span><span class="p">]</span>
- <span class="k">if</span> <span class="mi">2</span> <span class="o"><=</span> <span class="nb">len</span><span class="p">(</span><span class="bp">cls</span><span class="p">)</span> <span class="o"><=</span> <span class="mi">4</span><span class="p">:</span>
- <span class="n">col</span> <span class="o">=</span> <span class="bp">cls</span><span class="p">[</span><span class="mi">0</span><span class="p">]</span>
- <span class="n">ax</span> <span class="o">=</span> <span class="bp">cls</span><span class="p">[</span><span class="mi">1</span><span class="p">]</span>
- <span class="n">func</span> <span class="o">=</span> <span class="n">PlotTracer</span><span class="o">.</span><span class="n">defPlot</span>
- <span class="n">arg</span> <span class="o">=</span> <span class="n">inspect</span><span class="o">.</span><span class="n">Parameter</span><span class="o">.</span><span class="n">empty</span>
- <span class="k">if</span> <span class="nb">len</span><span class="p">(</span><span class="bp">cls</span><span class="p">)</span> <span class="o">==</span> <span class="mi">3</span><span class="p">:</span>
- <span class="k">if</span> <span class="n">callable</span><span class="p">(</span><span class="bp">cls</span><span class="p">[</span><span class="mi">2</span><span class="p">]):</span>
- <span class="n">func</span> <span class="o">=</span> <span class="bp">cls</span><span class="p">[</span><span class="mi">2</span><span class="p">]</span>
- <span class="k">else</span><span class="p">:</span>
- <span class="n">arg</span> <span class="o">=</span> <span class="bp">cls</span><span class="p">[</span><span class="mi">2</span><span class="p">]</span>
- <span class="k">if</span> <span class="nb">len</span><span class="p">(</span><span class="bp">cls</span><span class="p">)</span> <span class="o">==</span> <span class="mi">4</span><span class="p">:</span>
- <span class="n">func</span> <span class="o">=</span> <span class="bp">cls</span><span class="p">[</span><span class="mi">2</span><span class="p">]</span>
- <span class="k">if</span> <span class="ow">not</span> <span class="n">callable</span><span class="p">(</span><span class="n">func</span><span class="p">):</span>
- <span class="k">raise</span> <span class="ne">ValueError</span><span class="p">(</span><span class="s2">"Third argument must be callable."</span><span class="p">)</span>
- <span class="n">arg</span> <span class="o">=</span> <span class="bp">cls</span><span class="p">[</span><span class="mi">3</span><span class="p">]</span>
- <span class="bp">self</span><span class="o">.</span><span class="n">blocks</span><span class="o">.</span><span class="n">setdefault</span><span class="p">(</span><span class="n">col</span><span class="p">,</span> <span class="p">[])</span><span class="o">.</span><span class="n">append</span><span class="p">((</span><span class="n">ax</span><span class="p">,</span> <span class="n">func</span><span class="p">,</span> <span class="n">arg</span><span class="p">))</span>
- <span class="k">else</span><span class="p">:</span>
- <span class="k">raise</span> <span class="ne">TypeError</span><span class="p">(</span><span class="s2">"Cannot unpack collector info. Invalid number of arguments."</span><span class="p">)</span>
- <div class="viewcode-block" id="PlotTracer.defPlot"><a class="viewcode-back" href="../../../pypdevsbbl.tracers.PlotTracer.html#pypdevsbbl.tracers.PlotTracer.PlotTracer.defPlot">[docs]</a> <span class="nd">@staticmethod</span>
- <span class="k">def</span> <span class="nf">defPlot</span><span class="p">(</span><span class="n">collector</span><span class="p">,</span> <span class="n">axis</span><span class="p">,</span> <span class="n">extra</span><span class="o">=</span><span class="n">inspect</span><span class="o">.</span><span class="n">Parameter</span><span class="o">.</span><span class="n">empty</span><span class="p">):</span>
- <span class="sd">"""Default plotting function of the collectors."""</span>
- <span class="k">if</span> <span class="ow">not</span> <span class="nb">issubclass</span><span class="p">(</span><span class="nb">type</span><span class="p">(</span><span class="n">collector</span><span class="p">),</span> <span class="n">BaseCollector</span><span class="p">):</span>
- <span class="k">raise</span> <span class="ne">ValueError</span><span class="p">(</span><span class="s2">"No collector to plot found. Please specify a plotting function."</span><span class="p">)</span>
- <span class="n">sig</span> <span class="o">=</span> <span class="n">inspect</span><span class="o">.</span><span class="n">signature</span><span class="p">(</span><span class="n">collector</span><span class="o">.</span><span class="n">plot</span><span class="p">)</span>
- <span class="n">dft</span> <span class="o">=</span> <span class="kc">None</span>
- <span class="k">for</span> <span class="n">p</span> <span class="ow">in</span> <span class="n">sig</span><span class="o">.</span><span class="n">parameters</span><span class="p">:</span>
- <span class="k">if</span> <span class="n">p</span> <span class="o">!=</span> <span class="s1">'axis'</span><span class="p">:</span>
- <span class="n">dft</span> <span class="o">=</span> <span class="n">sig</span><span class="o">.</span><span class="n">parameters</span><span class="p">[</span><span class="n">p</span><span class="p">]</span><span class="o">.</span><span class="n">default</span>
- <span class="k">break</span>
- <span class="k">if</span> <span class="n">extra</span> <span class="ow">is</span> <span class="ow">not</span> <span class="n">inspect</span><span class="o">.</span><span class="n">Parameter</span><span class="o">.</span><span class="n">empty</span><span class="p">:</span>
- <span class="n">dft</span> <span class="o">=</span> <span class="n">extra</span>
- <span class="n">collector</span><span class="o">.</span><span class="n">plot</span><span class="p">(</span><span class="n">axis</span><span class="p">,</span> <span class="n">dft</span><span class="p">)</span></div>
- <div class="viewcode-block" id="PlotTracer.traceExternal"><a class="viewcode-back" href="../../../pypdevsbbl.tracers.PlotTracer.html#pypdevsbbl.tracers.PlotTracer.PlotTracer.traceExternal">[docs]</a> <span class="k">def</span> <span class="nf">traceExternal</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">aDEVS</span><span class="p">):</span>
- <span class="n">dvs</span> <span class="o">=</span> <span class="n">aDEVS</span><span class="o">.</span><span class="n">model</span>
- <span class="k">if</span> <span class="n">dvs</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">blocks</span><span class="p">:</span>
- <span class="n">ll</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">blocks</span><span class="p">[</span><span class="n">dvs</span><span class="p">]</span>
- <span class="k">for</span> <span class="n">ax</span><span class="p">,</span> <span class="n">func</span><span class="p">,</span> <span class="n">arg</span> <span class="ow">in</span> <span class="n">ll</span><span class="p">:</span>
- <span class="k">if</span> <span class="n">arg</span> <span class="ow">is</span> <span class="n">inspect</span><span class="o">.</span><span class="n">Parameter</span><span class="o">.</span><span class="n">empty</span><span class="p">:</span>
- <span class="n">func</span><span class="p">(</span><span class="n">dvs</span><span class="p">,</span> <span class="n">ax</span><span class="p">)</span>
- <span class="k">else</span><span class="p">:</span>
- <span class="n">func</span><span class="p">(</span><span class="n">dvs</span><span class="p">,</span> <span class="n">ax</span><span class="p">,</span> <span class="n">arg</span><span class="p">)</span>
- <span class="n">plt</span><span class="o">.</span><span class="n">show</span><span class="p">()</span></div></div>
- </pre></div>
- </div>
- </div>
- </div>
- <div class="sphinxsidebar" role="navigation" aria-label="main navigation">
- <div class="sphinxsidebarwrapper">
- <div id="searchbox" style="display: none" role="search">
- <h3>Quick search</h3>
- <div class="searchformwrapper">
- <form class="search" action="../../../search.html" method="get">
- <input type="text" name="q" />
- <input type="submit" value="Go" />
- <input type="hidden" name="check_keywords" value="yes" />
- <input type="hidden" name="area" value="default" />
- </form>
- </div>
- </div>
- <script type="text/javascript">$('#searchbox').show(0);</script>
- </div>
- </div>
- <div class="clearer"></div>
- </div>
- <div class="related" role="navigation" aria-label="related navigation">
- <h3>Navigation</h3>
- <ul>
- <li class="right" style="margin-right: 10px">
- <a href="../../../genindex.html" title="General Index"
- >index</a></li>
- <li class="right" >
- <a href="../../../py-modindex.html" title="Python Module Index"
- >modules</a> |</li>
- <li class="nav-item nav-item-0"><a href="../../../index.html">Python(P)DEVS-BBL documentation</a> »</li>
- <li class="nav-item nav-item-1"><a href="../../index.html" >Module code</a> »</li>
- <li class="nav-item nav-item-2"><a href="../tracers.html" >pypdevsbbl.tracers</a> »</li>
- </ul>
- </div>
- <div class="footer" role="contentinfo">
- © Copyright 2020, Randy Paredis.
- Created using <a href="http://sphinx-doc.org/">Sphinx</a> 1.8.5.
- </div>
- </body>
- </html>
|