source: sasview/_modules/sas/plottools/arrow3d.html @ a462c6a

gh-pages
Last change on this file since a462c6a was a462c6a, checked in by ajj, 9 years ago

Rebuild to fix index and modules docs

  • Property mode set to 100644
File size: 13.2 KB
Line 
1<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN"
2  "http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd">
3
4
5<html xmlns="http://www.w3.org/1999/xhtml">
6  <head>
7    <meta http-equiv="Content-Type" content="text/html; charset=utf-8" />
8   
9    <title>sas.plottools.arrow3d &mdash; SasView 3.0.0 documentation</title>
10   
11    <link rel="stylesheet" href="../../../_static/default.css" type="text/css" />
12    <link rel="stylesheet" href="../../../_static/pygments.css" type="text/css" />
13   
14    <script type="text/javascript">
15      var DOCUMENTATION_OPTIONS = {
16        URL_ROOT:    '../../../',
17        VERSION:     '3.0.0',
18        COLLAPSE_INDEX: false,
19        FILE_SUFFIX: '.html',
20        HAS_SOURCE:  true
21      };
22    </script>
23    <script type="text/javascript" src="../../../_static/jquery.js"></script>
24    <script type="text/javascript" src="../../../_static/underscore.js"></script>
25    <script type="text/javascript" src="../../../_static/doctools.js"></script>
26    <script type="text/javascript" src="http://cdn.mathjax.org/mathjax/latest/MathJax.js?config=TeX-AMS-MML_HTMLorMML"></script>
27    <link rel="top" title="SasView 3.0.0 documentation" href="../../../index.html" />
28    <link rel="up" title="Module code" href="../../index.html" /> 
29  </head>
30  <body>
31    <div class="related">
32      <h3>Navigation</h3>
33      <ul>
34        <li class="right" style="margin-right: 10px">
35          <a href="../../../genindex.html" title="General Index"
36             accesskey="I">index</a></li>
37        <li class="right" >
38          <a href="../../../py-modindex.html" title="Python Module Index"
39             >modules</a> |</li>
40        <li><a href="../../../index.html">SasView 3.0.0 documentation</a> &raquo;</li>
41          <li><a href="../../index.html" accesskey="U">Module code</a> &raquo;</li> 
42      </ul>
43    </div> 
44
45    <div class="document">
46      <div class="documentwrapper">
47        <div class="bodywrapper">
48          <div class="body">
49           
50  <h1>Source code for sas.plottools.arrow3d</h1><div class="highlight"><pre>
51<span class="sd">&quot;&quot;&quot;</span>
52<span class="sd">Module that draws multiple arrows in 3D coordinates</span>
53<span class="sd">&quot;&quot;&quot;</span>
54
55<span class="kn">from</span> <span class="nn">matplotlib.patches</span> <span class="kn">import</span> <span class="n">FancyArrowPatch</span>
56<span class="kn">from</span> <span class="nn">mpl_toolkits.mplot3d</span> <span class="kn">import</span> <span class="n">proj3d</span>
57<span class="kn">import</span> <span class="nn">time</span>
58<span class="c">#from matplotlib.artist import allow_rasterization</span>
59<div class="viewcode-block" id="Arrow3D"><a class="viewcode-back" href="../../../dev/api/sas.plottools.html#sas.plottools.arrow3d.Arrow3D">[docs]</a><span class="k">class</span> <span class="nc">Arrow3D</span><span class="p">(</span><span class="n">FancyArrowPatch</span><span class="p">):</span>
60    <span class="sd">&quot;&quot;&quot;</span>
61<span class="sd">    Draw 3D arrow</span>
62<span class="sd">    &quot;&quot;&quot;</span>
63   
64    <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">base</span><span class="p">,</span> <span class="n">xs</span><span class="p">,</span> <span class="n">ys</span><span class="p">,</span> <span class="n">zs</span><span class="p">,</span> <span class="n">colors</span><span class="p">,</span> <span class="o">*</span><span class="n">args</span><span class="p">,</span> <span class="o">**</span><span class="n">kwargs</span><span class="p">):</span>
65        <span class="sd">&quot;&quot;&quot;</span>
66<span class="sd">        Init</span>
67<span class="sd">            </span>
68<span class="sd">        :Params xs: [[x0, x0+dx0], [x1, x1+dx1], ...]</span>
69<span class="sd">        :Params ys: [[y0, y0+dy0], [y1, y1+dy1], ...]</span>
70<span class="sd">        :Params zs: [[z0, z0+dz0], [z1, z1+dz1], ...]</span>
71<span class="sd">        :Params colors: [[R0, G0, B0], [R1, G1, B1], ...] </span>
72<span class="sd">        where R, G, B ranges (0,1)</span>
73<span class="sd">        &quot;&quot;&quot;</span>
74        <span class="n">FancyArrowPatch</span><span class="o">.</span><span class="n">__init__</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="p">(</span><span class="mi">0</span><span class="p">,</span><span class="mi">0</span><span class="p">),</span> <span class="p">(</span><span class="mi">0</span><span class="p">,</span><span class="mi">0</span><span class="p">),</span> <span class="o">*</span><span class="n">args</span><span class="p">,</span> <span class="o">**</span><span class="n">kwargs</span><span class="p">)</span>
75        <span class="bp">self</span><span class="o">.</span><span class="n">leftdown</span> <span class="o">=</span> <span class="bp">False</span>
76        <span class="bp">self</span><span class="o">.</span><span class="n">t_click</span> <span class="o">=</span> <span class="mi">0</span>
77        <span class="bp">self</span><span class="o">.</span><span class="n">_verts3d</span> <span class="o">=</span> <span class="n">xs</span><span class="p">,</span> <span class="n">ys</span><span class="p">,</span> <span class="n">zs</span>
78        <span class="bp">self</span><span class="o">.</span><span class="n">colors</span> <span class="o">=</span> <span class="n">colors</span>
79        <span class="bp">self</span><span class="o">.</span><span class="n">base</span> <span class="o">=</span> <span class="n">base</span>
80       
81        <span class="k">if</span> <span class="n">base</span> <span class="o">!=</span> <span class="bp">None</span><span class="p">:</span>
82            <span class="c"># To turn the updating off during dragging</span>
83            <span class="n">base</span><span class="o">.</span><span class="n">canvas</span><span class="o">.</span><span class="n">mpl_connect</span><span class="p">(</span><span class="s">&#39;button_press_event&#39;</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">onLeftDown</span><span class="p">)</span>
84            <span class="n">base</span><span class="o">.</span><span class="n">canvas</span><span class="o">.</span><span class="n">mpl_connect</span><span class="p">(</span><span class="s">&#39;button_release_event&#39;</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">onLeftUp</span><span class="p">)</span>
85           
86<div class="viewcode-block" id="Arrow3D.onLeftDown"><a class="viewcode-back" href="../../../dev/api/sas.plottools.html#sas.plottools.arrow3d.Arrow3D.onLeftDown">[docs]</a>    <span class="k">def</span> <span class="nf">onLeftDown</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">event</span><span class="p">):</span>
87        <span class="sd">&quot;&quot;&quot;</span>
88<span class="sd">        &quot;&quot;&quot;</span>
89        <span class="bp">self</span><span class="o">.</span><span class="n">leftdown</span> <span class="o">=</span> <span class="bp">True</span>
90        <span class="bp">self</span><span class="o">.</span><span class="n">t_click</span> <span class="o">=</span> <span class="n">time</span><span class="o">.</span><span class="n">time</span><span class="p">()</span>
91        </div>
92<div class="viewcode-block" id="Arrow3D.onLeftUp"><a class="viewcode-back" href="../../../dev/api/sas.plottools.html#sas.plottools.arrow3d.Arrow3D.onLeftUp">[docs]</a>    <span class="k">def</span> <span class="nf">onLeftUp</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">event</span><span class="p">):</span>
93        <span class="sd">&quot;&quot;&quot;</span>
94<span class="sd">        &quot;&quot;&quot;</span>
95        <span class="n">t_up</span> <span class="o">=</span> <span class="n">time</span><span class="o">.</span><span class="n">time</span><span class="p">()</span> <span class="o">-</span> <span class="bp">self</span><span class="o">.</span><span class="n">t_click</span>
96        <span class="c"># Avoid just clicking</span>
97        <span class="k">if</span> <span class="n">t_up</span> <span class="o">&gt;</span> <span class="mf">0.1</span><span class="p">:</span>
98            <span class="bp">self</span><span class="o">.</span><span class="n">leftdown</span> <span class="o">=</span> <span class="bp">False</span>
99            <span class="bp">self</span><span class="o">.</span><span class="n">base</span><span class="o">.</span><span class="n">canvas</span><span class="o">.</span><span class="n">draw</span><span class="p">()</span>
100    </div>
101<div class="viewcode-block" id="Arrow3D.draw"><a class="viewcode-back" href="../../../dev/api/sas.plottools.html#sas.plottools.arrow3d.Arrow3D.draw">[docs]</a>    <span class="k">def</span> <span class="nf">draw</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">renderer</span><span class="p">,</span> <span class="n">rasterized</span><span class="o">=</span><span class="bp">True</span><span class="p">):</span>
102        <span class="sd">&quot;&quot;&quot;</span>
103<span class="sd">        Drawing actually happens here</span>
104<span class="sd">        &quot;&quot;&quot;</span>
105        <span class="c"># Draws only when the dragging finished</span>
106        <span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">leftdown</span><span class="p">:</span>
107            <span class="k">return</span>
108        <span class="n">xs3d</span><span class="p">,</span> <span class="n">ys3d</span><span class="p">,</span> <span class="n">zs3d</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">_verts3d</span>
109        <span class="k">for</span> <span class="n">i</span> <span class="ow">in</span> <span class="nb">xrange</span><span class="p">(</span><span class="nb">len</span><span class="p">(</span><span class="n">xs3d</span><span class="p">)):</span>
110            <span class="n">xs</span><span class="p">,</span> <span class="n">ys</span><span class="p">,</span> <span class="n">_</span> <span class="o">=</span> <span class="n">proj3d</span><span class="o">.</span><span class="n">proj_transform</span><span class="p">(</span><span class="n">xs3d</span><span class="p">[</span><span class="n">i</span><span class="p">],</span> <span class="n">ys3d</span><span class="p">[</span><span class="n">i</span><span class="p">],</span> <span class="n">zs3d</span><span class="p">[</span><span class="n">i</span><span class="p">],</span> 
111                                               <span class="n">renderer</span><span class="o">.</span><span class="n">M</span><span class="p">)</span>
112            <span class="bp">self</span><span class="o">.</span><span class="n">set_positions</span><span class="p">((</span><span class="n">xs</span><span class="p">[</span><span class="mi">0</span><span class="p">],</span><span class="n">ys</span><span class="p">[</span><span class="mi">0</span><span class="p">]),(</span><span class="n">xs</span><span class="p">[</span><span class="mi">1</span><span class="p">],</span><span class="n">ys</span><span class="p">[</span><span class="mi">1</span><span class="p">]))</span>
113            <span class="bp">self</span><span class="o">.</span><span class="n">set_color</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">colors</span><span class="p">[</span><span class="n">i</span><span class="p">])</span>
114            <span class="n">FancyArrowPatch</span><span class="o">.</span><span class="n">draw</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">renderer</span><span class="p">)</span>
115           
116        <span class="bp">self</span><span class="o">.</span><span class="n">leftdown</span> <span class="o">=</span> <span class="bp">False</span></div></div>
117</pre></div>
118
119          </div>
120        </div>
121      </div>
122      <div class="sphinxsidebar">
123        <div class="sphinxsidebarwrapper">
124<div id="searchbox" style="display: none">
125  <h3>Quick search</h3>
126    <form class="search" action="../../../search.html" method="get">
127      <input type="text" name="q" />
128      <input type="submit" value="Go" />
129      <input type="hidden" name="check_keywords" value="yes" />
130      <input type="hidden" name="area" value="default" />
131    </form>
132    <p class="searchtip" style="font-size: 90%">
133    Enter search terms or a module, class or function name.
134    </p>
135</div>
136<script type="text/javascript">$('#searchbox').show(0);</script>
137        </div>
138      </div>
139      <div class="clearer"></div>
140    </div>
141    <div class="related">
142      <h3>Navigation</h3>
143      <ul>
144        <li class="right" style="margin-right: 10px">
145          <a href="../../../genindex.html" title="General Index"
146             >index</a></li>
147        <li class="right" >
148          <a href="../../../py-modindex.html" title="Python Module Index"
149             >modules</a> |</li>
150        <li><a href="../../../index.html">SasView 3.0.0 documentation</a> &raquo;</li>
151          <li><a href="../../index.html" >Module code</a> &raquo;</li> 
152      </ul>
153    </div>
154    <div class="footer">
155        &copy; Copyright 2013, The SasView Project.
156      Created using <a href="http://sphinx-doc.org/">Sphinx</a> 1.2.3.
157    </div>
158  </body>
159</html>
Note: See TracBrowser for help on using the repository browser.