PlotTracer.html 18 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190
  1. <!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN"
  2. "http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd">
  3. <html xmlns="http://www.w3.org/1999/xhtml">
  4. <head>
  5. <meta http-equiv="X-UA-Compatible" content="IE=Edge" />
  6. <meta http-equiv="Content-Type" content="text/html; charset=utf-8" />
  7. <title>pypdevsbbl.tracers.PlotTracer &#8212; Python(P)DEVS-BBL documentation</title>
  8. <link rel="stylesheet" href="../../../_static/nature.css" type="text/css" />
  9. <link rel="stylesheet" href="../../../_static/pygments.css" type="text/css" />
  10. <link rel="stylesheet" type="text/css" href="../../../_static/custom.css" />
  11. <script type="text/javascript" id="documentation_options" data-url_root="../../../" src="../../../_static/documentation_options.js"></script>
  12. <script type="text/javascript" src="../../../_static/jquery.js"></script>
  13. <script type="text/javascript" src="../../../_static/underscore.js"></script>
  14. <script type="text/javascript" src="../../../_static/doctools.js"></script>
  15. <script type="text/javascript" src="../../../_static/language_data.js"></script>
  16. <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>
  17. <link rel="index" title="Index" href="../../../genindex.html" />
  18. <link rel="search" title="Search" href="../../../search.html" />
  19. </head><body>
  20. <div class="related" role="navigation" aria-label="related navigation">
  21. <h3>Navigation</h3>
  22. <ul>
  23. <li class="right" style="margin-right: 10px">
  24. <a href="../../../genindex.html" title="General Index"
  25. accesskey="I">index</a></li>
  26. <li class="right" >
  27. <a href="../../../py-modindex.html" title="Python Module Index"
  28. >modules</a> |</li>
  29. <li class="nav-item nav-item-0"><a href="../../../index.html">Python(P)DEVS-BBL documentation</a> &#187;</li>
  30. <li class="nav-item nav-item-1"><a href="../../index.html" >Module code</a> &#187;</li>
  31. <li class="nav-item nav-item-2"><a href="../tracers.html" accesskey="U">pypdevsbbl.tracers</a> &#187;</li>
  32. </ul>
  33. </div>
  34. <div class="document">
  35. <div class="documentwrapper">
  36. <div class="bodywrapper">
  37. <div class="body" role="main">
  38. <h1>Source code for pypdevsbbl.tracers.PlotTracer</h1><div class="highlight"><pre>
  39. <span></span><span class="c1"># Copyright 2020 Modelling, Simulation and Design Lab (MSDL) at</span>
  40. <span class="c1"># McGill University and the University of Antwerp (http://msdl.cs.mcgill.ca/)</span>
  41. <span class="c1">#</span>
  42. <span class="c1"># Licensed under the Apache License, Version 2.0 (the &quot;License&quot;);</span>
  43. <span class="c1"># you may not use this file except in compliance with the License.</span>
  44. <span class="c1"># You may obtain a copy of the License at</span>
  45. <span class="c1">#</span>
  46. <span class="c1"># http://www.apache.org/licenses/LICENSE-2.0</span>
  47. <span class="c1">#</span>
  48. <span class="c1"># Unless required by applicable law or agreed to in writing, software</span>
  49. <span class="c1"># distributed under the License is distributed on an &quot;AS IS&quot; BASIS,</span>
  50. <span class="c1"># WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.</span>
  51. <span class="c1"># See the License for the specific language governing permissions and</span>
  52. <span class="c1"># limitations under the License.</span>
  53. <span class="sd">&quot;&quot;&quot;This file contains the tracer that can provide plots of the :mod:`pypdevsbbl.generic.collectors`.&quot;&quot;&quot;</span>
  54. <span class="kn">from</span> <span class="nn">pypdevsbbl.generic.collectors</span> <span class="k">import</span> <span class="n">BaseCollector</span>
  55. <span class="kn">from</span> <span class="nn">pypdevsbbl.tracers</span> <span class="k">import</span> <span class="n">BaseTracer</span>
  56. <span class="kn">import</span> <span class="nn">matplotlib.pyplot</span> <span class="k">as</span> <span class="nn">plt</span>
  57. <span class="kn">import</span> <span class="nn">inspect</span>
  58. <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>
  59. <span class="sd">&quot;&quot;&quot;Plots data from the provided collectors.</span>
  60. <span class="sd"> Note:</span>
  61. <span class="sd"> Because of how :mod:`matplotlib` and :mod:`pypdevs` work, this</span>
  62. <span class="sd"> tracer only seems to be working within a `Jupyter` notebook.</span>
  63. <span class="sd"> See :file:`src/notebook/PlotTracer.ipynb` for an example on how</span>
  64. <span class="sd"> to do this.</span>
  65. <span class="sd"> Args:</span>
  66. <span class="sd"> uid: A unique identifier for distributed simulation.</span>
  67. <span class="sd"> Will be set automatically.</span>
  68. <span class="sd"> server: The server that runs the distributed simulation.</span>
  69. <span class="sd"> Will be set automatically.</span>
  70. <span class="sd"> *collectors: The collectors to plot. This is a pair in the</span>
  71. <span class="sd"> form of ``(BaseCollector, Axis)`` or a triple in</span>
  72. <span class="sd"> the form of ``(BaseDEVS, Axis, func)``, where</span>
  73. <span class="sd"> ``BaseCollector`` or ``BaseDEVS`` are instances</span>
  74. <span class="sd"> of those classes, ``Axis`` a :class:`matplotlib.Axes`</span>
  75. <span class="sd"> and ``func`` a callable that takes the first two</span>
  76. <span class="sd"> members and plots the data on the axis.</span>
  77. <span class="sd"> Additionally, the tuples/lists may be extended</span>
  78. <span class="sd"> with a thrird/fourth member, which is an additional</span>
  79. <span class="sd"> argument to pass to the plot function. What this is</span>
  80. <span class="sd"> for each collector is described in their docs.</span>
  81. <span class="sd"> &quot;&quot;&quot;</span>
  82. <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>
  83. <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>
  84. <span class="bp">self</span><span class="o">.</span><span class="n">blocks</span> <span class="o">=</span> <span class="p">{}</span>
  85. <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>
  86. <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>
  87. <span class="k">if</span> <span class="mi">2</span> <span class="o">&lt;=</span> <span class="nb">len</span><span class="p">(</span><span class="bp">cls</span><span class="p">)</span> <span class="o">&lt;=</span> <span class="mi">4</span><span class="p">:</span>
  88. <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>
  89. <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>
  90. <span class="n">func</span> <span class="o">=</span> <span class="n">PlotTracer</span><span class="o">.</span><span class="n">defPlot</span>
  91. <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>
  92. <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>
  93. <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>
  94. <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>
  95. <span class="k">else</span><span class="p">:</span>
  96. <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>
  97. <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>
  98. <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>
  99. <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>
  100. <span class="k">raise</span> <span class="ne">ValueError</span><span class="p">(</span><span class="s2">&quot;Third argument must be callable.&quot;</span><span class="p">)</span>
  101. <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>
  102. <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>
  103. <span class="k">else</span><span class="p">:</span>
  104. <span class="k">raise</span> <span class="ne">TypeError</span><span class="p">(</span><span class="s2">&quot;Cannot unpack collector info. Invalid number of arguments.&quot;</span><span class="p">)</span>
  105. <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>
  106. <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>
  107. <span class="sd">&quot;&quot;&quot;Default plotting function of the collectors.&quot;&quot;&quot;</span>
  108. <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>
  109. <span class="k">raise</span> <span class="ne">ValueError</span><span class="p">(</span><span class="s2">&quot;No collector to plot found. Please specify a plotting function.&quot;</span><span class="p">)</span>
  110. <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>
  111. <span class="n">dft</span> <span class="o">=</span> <span class="kc">None</span>
  112. <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>
  113. <span class="k">if</span> <span class="n">p</span> <span class="o">!=</span> <span class="s1">&#39;axis&#39;</span><span class="p">:</span>
  114. <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>
  115. <span class="k">break</span>
  116. <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>
  117. <span class="n">dft</span> <span class="o">=</span> <span class="n">extra</span>
  118. <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>
  119. <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>
  120. <span class="n">dvs</span> <span class="o">=</span> <span class="n">aDEVS</span><span class="o">.</span><span class="n">model</span>
  121. <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>
  122. <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>
  123. <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>
  124. <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>
  125. <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>
  126. <span class="k">else</span><span class="p">:</span>
  127. <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>
  128. <span class="n">plt</span><span class="o">.</span><span class="n">show</span><span class="p">()</span></div></div>
  129. </pre></div>
  130. </div>
  131. </div>
  132. </div>
  133. <div class="sphinxsidebar" role="navigation" aria-label="main navigation">
  134. <div class="sphinxsidebarwrapper">
  135. <div id="searchbox" style="display: none" role="search">
  136. <h3>Quick search</h3>
  137. <div class="searchformwrapper">
  138. <form class="search" action="../../../search.html" method="get">
  139. <input type="text" name="q" />
  140. <input type="submit" value="Go" />
  141. <input type="hidden" name="check_keywords" value="yes" />
  142. <input type="hidden" name="area" value="default" />
  143. </form>
  144. </div>
  145. </div>
  146. <script type="text/javascript">$('#searchbox').show(0);</script>
  147. </div>
  148. </div>
  149. <div class="clearer"></div>
  150. </div>
  151. <div class="related" role="navigation" aria-label="related navigation">
  152. <h3>Navigation</h3>
  153. <ul>
  154. <li class="right" style="margin-right: 10px">
  155. <a href="../../../genindex.html" title="General Index"
  156. >index</a></li>
  157. <li class="right" >
  158. <a href="../../../py-modindex.html" title="Python Module Index"
  159. >modules</a> |</li>
  160. <li class="nav-item nav-item-0"><a href="../../../index.html">Python(P)DEVS-BBL documentation</a> &#187;</li>
  161. <li class="nav-item nav-item-1"><a href="../../index.html" >Module code</a> &#187;</li>
  162. <li class="nav-item nav-item-2"><a href="../tracers.html" >pypdevsbbl.tracers</a> &#187;</li>
  163. </ul>
  164. </div>
  165. <div class="footer" role="contentinfo">
  166. &#169; Copyright 2020, Randy Paredis.
  167. Created using <a href="http://sphinx-doc.org/">Sphinx</a> 1.8.5.
  168. </div>
  169. </body>
  170. </html>