source: sasview/_modules/sas/dataloader/readers/IgorReader.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: 42.5 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.dataloader.readers.IgorReader &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="sas.dataloader.readers" href="../readers.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" >Module code</a> &raquo;</li>
42          <li><a href="../readers.html" accesskey="U">sas.dataloader.readers</a> &raquo;</li> 
43      </ul>
44    </div> 
45
46    <div class="document">
47      <div class="documentwrapper">
48        <div class="bodywrapper">
49          <div class="body">
50           
51  <h1>Source code for sas.dataloader.readers.IgorReader</h1><div class="highlight"><pre>
52<span class="sd">&quot;&quot;&quot;</span>
53<span class="sd">    IGOR 2D reduced file reader</span>
54<span class="sd">&quot;&quot;&quot;</span>
55<span class="c">############################################################################</span>
56<span class="c">#This software was developed by the University of Tennessee as part of the</span>
57<span class="c">#Distributed Data Analysis of Neutron Scattering Experiments (DANSE)</span>
58<span class="c">#project funded by the US National Science Foundation. </span>
59<span class="c">#If you use DANSE applications to do scientific research that leads to </span>
60<span class="c">#publication, we ask that you acknowledge the use of the software with the </span>
61<span class="c">#following sentence:</span>
62<span class="c">#This work benefited from DANSE software developed under NSF award DMR-0520547. </span>
63<span class="c">#copyright 2008, University of Tennessee</span>
64<span class="c">#############################################################################</span>
65<span class="kn">import</span> <span class="nn">os</span>
66<span class="kn">import</span> <span class="nn">numpy</span>
67<span class="kn">import</span> <span class="nn">math</span>
68<span class="c">#import logging</span>
69<span class="kn">from</span> <span class="nn">sas.dataloader.data_info</span> <span class="kn">import</span> <span class="n">Data2D</span>
70<span class="kn">from</span> <span class="nn">sas.dataloader.data_info</span> <span class="kn">import</span> <span class="n">Detector</span>
71<span class="kn">from</span> <span class="nn">sas.dataloader.manipulations</span> <span class="kn">import</span> <span class="n">reader2D_converter</span>
72
73<span class="c"># Look for unit converter</span>
74<span class="n">has_converter</span> <span class="o">=</span> <span class="bp">True</span>
75<span class="k">try</span><span class="p">:</span>
76    <span class="kn">from</span> <span class="nn">sas.data_util.nxsunit</span> <span class="kn">import</span> <span class="n">Converter</span>
77<span class="k">except</span><span class="p">:</span>
78    <span class="n">has_converter</span> <span class="o">=</span> <span class="bp">False</span>
79
80
81<div class="viewcode-block" id="Reader"><a class="viewcode-back" href="../../../../dev/api/sas.dataloader.readers.html#sas.dataloader.readers.IgorReader.Reader">[docs]</a><span class="k">class</span> <span class="nc">Reader</span><span class="p">:</span>
82    <span class="sd">&quot;&quot;&quot; Simple data reader for Igor data files &quot;&quot;&quot;</span>
83    <span class="c">## File type</span>
84    <span class="n">type_name</span> <span class="o">=</span> <span class="s">&quot;IGOR 2D&quot;</span>
85    <span class="c">## Wildcards</span>
86    <span class="nb">type</span> <span class="o">=</span> <span class="p">[</span><span class="s">&quot;IGOR 2D files (*.ASC)|*.ASC&quot;</span><span class="p">]</span>
87    <span class="c">## Extension</span>
88    <span class="n">ext</span><span class="o">=</span><span class="p">[</span><span class="s">&#39;.ASC&#39;</span><span class="p">,</span> <span class="s">&#39;.asc&#39;</span><span class="p">]</span>
89
90<div class="viewcode-block" id="Reader.read"><a class="viewcode-back" href="../../../../dev/api/sas.dataloader.readers.html#sas.dataloader.readers.IgorReader.Reader.read">[docs]</a>    <span class="k">def</span> <span class="nf">read</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">filename</span><span class="o">=</span><span class="bp">None</span><span class="p">):</span>
91        <span class="sd">&quot;&quot;&quot; Read file &quot;&quot;&quot;</span>
92        <span class="k">if</span> <span class="ow">not</span> <span class="n">os</span><span class="o">.</span><span class="n">path</span><span class="o">.</span><span class="n">isfile</span><span class="p">(</span><span class="n">filename</span><span class="p">):</span>
93            <span class="k">raise</span> <span class="ne">ValueError</span><span class="p">,</span> \
94            <span class="s">&quot;Specified file </span><span class="si">%s</span><span class="s"> is not a regular file&quot;</span> <span class="o">%</span> <span class="n">filename</span>
95       
96        <span class="c"># Read file</span>
97        <span class="n">f</span> <span class="o">=</span> <span class="nb">open</span><span class="p">(</span><span class="n">filename</span><span class="p">,</span> <span class="s">&#39;r&#39;</span><span class="p">)</span>
98        <span class="n">buf</span> <span class="o">=</span> <span class="n">f</span><span class="o">.</span><span class="n">read</span><span class="p">()</span>
99       
100        <span class="c"># Instantiate data object</span>
101        <span class="n">output</span> <span class="o">=</span> <span class="n">Data2D</span><span class="p">()</span>
102        <span class="n">output</span><span class="o">.</span><span class="n">filename</span> <span class="o">=</span> <span class="n">os</span><span class="o">.</span><span class="n">path</span><span class="o">.</span><span class="n">basename</span><span class="p">(</span><span class="n">filename</span><span class="p">)</span>
103        <span class="n">detector</span> <span class="o">=</span> <span class="n">Detector</span><span class="p">()</span>
104        <span class="k">if</span> <span class="nb">len</span><span class="p">(</span><span class="n">output</span><span class="o">.</span><span class="n">detector</span><span class="p">)</span> <span class="o">&gt;</span> <span class="mi">0</span><span class="p">:</span>
105            <span class="k">print</span> <span class="nb">str</span><span class="p">(</span><span class="n">output</span><span class="o">.</span><span class="n">detector</span><span class="p">[</span><span class="mi">0</span><span class="p">])</span>
106        <span class="n">output</span><span class="o">.</span><span class="n">detector</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">detector</span><span class="p">)</span>
107               
108        <span class="c"># Get content</span>
109        <span class="n">dataStarted</span> <span class="o">=</span> <span class="bp">False</span>
110       
111        <span class="n">lines</span> <span class="o">=</span> <span class="n">buf</span><span class="o">.</span><span class="n">split</span><span class="p">(</span><span class="s">&#39;</span><span class="se">\n</span><span class="s">&#39;</span><span class="p">)</span>
112        <span class="n">itot</span> <span class="o">=</span> <span class="mi">0</span>
113        <span class="n">x</span> <span class="o">=</span> <span class="p">[]</span>
114        <span class="n">y</span> <span class="o">=</span> <span class="p">[]</span>
115       
116        <span class="n">ncounts</span> <span class="o">=</span> <span class="mi">0</span>
117       
118        <span class="n">xmin</span> <span class="o">=</span> <span class="bp">None</span>
119        <span class="n">xmax</span> <span class="o">=</span> <span class="bp">None</span>
120        <span class="n">ymin</span> <span class="o">=</span> <span class="bp">None</span>
121        <span class="n">ymax</span> <span class="o">=</span> <span class="bp">None</span>
122       
123        <span class="n">i_x</span> <span class="o">=</span> <span class="mi">0</span>
124        <span class="n">i_y</span> <span class="o">=</span> <span class="o">-</span><span class="mi">1</span>
125        <span class="n">i_tot_row</span> <span class="o">=</span> <span class="mi">0</span>
126       
127        <span class="n">isInfo</span> <span class="o">=</span> <span class="bp">False</span>
128        <span class="n">isCenter</span> <span class="o">=</span> <span class="bp">False</span>
129       
130        <span class="n">data_conv_q</span> <span class="o">=</span> <span class="bp">None</span>
131        <span class="n">data_conv_i</span> <span class="o">=</span> <span class="bp">None</span>
132       
133        <span class="k">if</span> <span class="n">has_converter</span> <span class="o">==</span> <span class="bp">True</span> <span class="ow">and</span> <span class="n">output</span><span class="o">.</span><span class="n">Q_unit</span> <span class="o">!=</span> <span class="s">&#39;1/A&#39;</span><span class="p">:</span>
134            <span class="n">data_conv_q</span> <span class="o">=</span> <span class="n">Converter</span><span class="p">(</span><span class="s">&#39;1/A&#39;</span><span class="p">)</span>
135            <span class="c"># Test it</span>
136            <span class="n">data_conv_q</span><span class="p">(</span><span class="mf">1.0</span><span class="p">,</span> <span class="n">output</span><span class="o">.</span><span class="n">Q_unit</span><span class="p">)</span>
137           
138        <span class="k">if</span> <span class="n">has_converter</span> <span class="o">==</span> <span class="bp">True</span> <span class="ow">and</span> <span class="n">output</span><span class="o">.</span><span class="n">I_unit</span> <span class="o">!=</span> <span class="s">&#39;1/cm&#39;</span><span class="p">:</span>
139            <span class="n">data_conv_i</span> <span class="o">=</span> <span class="n">Converter</span><span class="p">(</span><span class="s">&#39;1/cm&#39;</span><span class="p">)</span>
140            <span class="c"># Test it</span>
141            <span class="n">data_conv_i</span><span class="p">(</span><span class="mf">1.0</span><span class="p">,</span> <span class="n">output</span><span class="o">.</span><span class="n">I_unit</span><span class="p">)</span>
142         
143        <span class="k">for</span> <span class="n">line</span> <span class="ow">in</span> <span class="n">lines</span><span class="p">:</span>
144           
145            <span class="c"># Find setup info line</span>
146            <span class="k">if</span> <span class="n">isInfo</span><span class="p">:</span>
147                <span class="n">isInfo</span> <span class="o">=</span> <span class="bp">False</span>
148                <span class="n">line_toks</span> <span class="o">=</span> <span class="n">line</span><span class="o">.</span><span class="n">split</span><span class="p">()</span>
149                <span class="c"># Wavelength in Angstrom</span>
150                <span class="k">try</span><span class="p">:</span>
151                    <span class="n">wavelength</span> <span class="o">=</span> <span class="nb">float</span><span class="p">(</span><span class="n">line_toks</span><span class="p">[</span><span class="mi">1</span><span class="p">])</span>
152                <span class="k">except</span><span class="p">:</span>
153                    <span class="n">msg</span> <span class="o">=</span> <span class="s">&quot;IgorReader: can&#39;t read this file, missing wavelength&quot;</span>
154                    <span class="k">raise</span> <span class="ne">ValueError</span><span class="p">,</span> <span class="n">msg</span>
155               
156            <span class="c">#Find # of bins in a row assuming the detector is square.</span>
157            <span class="k">if</span> <span class="n">dataStarted</span> <span class="o">==</span> <span class="bp">True</span><span class="p">:</span>
158                <span class="k">try</span><span class="p">:</span>
159                    <span class="n">value</span> <span class="o">=</span> <span class="nb">float</span><span class="p">(</span><span class="n">line</span><span class="p">)</span>
160                <span class="k">except</span><span class="p">:</span>
161                    <span class="c"># Found a non-float entry, skip it</span>
162                    <span class="k">continue</span>
163               
164                <span class="c"># Get total bin number</span>
165               
166            <span class="n">i_tot_row</span> <span class="o">+=</span> <span class="mi">1</span>
167        <span class="n">i_tot_row</span> <span class="o">=</span> <span class="n">math</span><span class="o">.</span><span class="n">ceil</span><span class="p">(</span><span class="n">math</span><span class="o">.</span><span class="n">sqrt</span><span class="p">(</span><span class="n">i_tot_row</span><span class="p">))</span> <span class="o">-</span> <span class="mi">1</span>
168        <span class="c">#print &quot;i_tot&quot;, i_tot_row</span>
169        <span class="n">size_x</span> <span class="o">=</span> <span class="n">i_tot_row</span>  <span class="c"># 192#128</span>
170        <span class="n">size_y</span> <span class="o">=</span> <span class="n">i_tot_row</span>  <span class="c"># 192#128</span>
171        <span class="n">output</span><span class="o">.</span><span class="n">data</span> <span class="o">=</span> <span class="n">numpy</span><span class="o">.</span><span class="n">zeros</span><span class="p">([</span><span class="n">size_x</span><span class="p">,</span> <span class="n">size_y</span><span class="p">])</span>
172        <span class="n">output</span><span class="o">.</span><span class="n">err_data</span> <span class="o">=</span> <span class="n">numpy</span><span class="o">.</span><span class="n">zeros</span><span class="p">([</span><span class="n">size_x</span><span class="p">,</span> <span class="n">size_y</span><span class="p">])</span>
173     
174        <span class="c">#Read Header and 2D data</span>
175        <span class="k">for</span> <span class="n">line</span> <span class="ow">in</span> <span class="n">lines</span><span class="p">:</span>
176            <span class="c"># Find setup info line</span>
177            <span class="k">if</span> <span class="n">isInfo</span><span class="p">:</span>
178                <span class="n">isInfo</span> <span class="o">=</span> <span class="bp">False</span>
179                <span class="n">line_toks</span> <span class="o">=</span> <span class="n">line</span><span class="o">.</span><span class="n">split</span><span class="p">()</span>
180                <span class="c"># Wavelength in Angstrom</span>
181                <span class="k">try</span><span class="p">:</span>
182                    <span class="n">wavelength</span> <span class="o">=</span> <span class="nb">float</span><span class="p">(</span><span class="n">line_toks</span><span class="p">[</span><span class="mi">1</span><span class="p">])</span>
183                <span class="k">except</span><span class="p">:</span>
184                    <span class="n">msg</span> <span class="o">=</span> <span class="s">&quot;IgorReader: can&#39;t read this file, missing wavelength&quot;</span>
185                    <span class="k">raise</span> <span class="ne">ValueError</span><span class="p">,</span> <span class="n">msg</span>
186                <span class="c"># Distance in meters</span>
187                <span class="k">try</span><span class="p">:</span>
188                    <span class="n">distance</span> <span class="o">=</span> <span class="nb">float</span><span class="p">(</span><span class="n">line_toks</span><span class="p">[</span><span class="mi">3</span><span class="p">])</span>
189                <span class="k">except</span><span class="p">:</span>
190                    <span class="n">msg</span> <span class="o">=</span> <span class="s">&quot;IgorReader: can&#39;t read this file, missing distance&quot;</span>
191                    <span class="k">raise</span> <span class="ne">ValueError</span><span class="p">,</span> <span class="n">msg</span>
192               
193                <span class="c"># Distance in meters</span>
194                <span class="k">try</span><span class="p">:</span>
195                    <span class="n">transmission</span> <span class="o">=</span> <span class="nb">float</span><span class="p">(</span><span class="n">line_toks</span><span class="p">[</span><span class="mi">4</span><span class="p">])</span>
196                <span class="k">except</span><span class="p">:</span>
197                    <span class="n">msg</span> <span class="o">=</span> <span class="s">&quot;IgorReader: can&#39;t read this file, &quot;</span>
198                    <span class="n">msg</span> <span class="o">+=</span> <span class="s">&quot;missing transmission&quot;</span>
199                    <span class="k">raise</span> <span class="ne">ValueError</span><span class="p">,</span> <span class="n">msg</span>
200                                           
201            <span class="k">if</span> <span class="n">line</span><span class="o">.</span><span class="n">count</span><span class="p">(</span><span class="s">&quot;LAMBDA&quot;</span><span class="p">)</span> <span class="o">&gt;</span> <span class="mi">0</span><span class="p">:</span>
202                <span class="n">isInfo</span> <span class="o">=</span> <span class="bp">True</span>
203               
204            <span class="c"># Find center info line</span>
205            <span class="k">if</span> <span class="n">isCenter</span><span class="p">:</span>
206                <span class="n">isCenter</span> <span class="o">=</span> <span class="bp">False</span>
207                <span class="n">line_toks</span> <span class="o">=</span> <span class="n">line</span><span class="o">.</span><span class="n">split</span><span class="p">()</span>
208               
209                <span class="c"># Center in bin number: Must substrate 1 because</span>
210                <span class="c">#the index starts from 1</span>
211                <span class="n">center_x</span> <span class="o">=</span> <span class="nb">float</span><span class="p">(</span><span class="n">line_toks</span><span class="p">[</span><span class="mi">0</span><span class="p">])</span> <span class="o">-</span> <span class="mi">1</span>
212                <span class="n">center_y</span> <span class="o">=</span> <span class="nb">float</span><span class="p">(</span><span class="n">line_toks</span><span class="p">[</span><span class="mi">1</span><span class="p">])</span> <span class="o">-</span> <span class="mi">1</span>
213
214            <span class="k">if</span> <span class="n">line</span><span class="o">.</span><span class="n">count</span><span class="p">(</span><span class="s">&quot;BCENT&quot;</span><span class="p">)</span> <span class="o">&gt;</span> <span class="mi">0</span><span class="p">:</span>
215                <span class="n">isCenter</span> <span class="o">=</span> <span class="bp">True</span>
216               
217            <span class="c"># Find data start</span>
218            <span class="k">if</span> <span class="n">line</span><span class="o">.</span><span class="n">count</span><span class="p">(</span><span class="s">&quot;***&quot;</span><span class="p">)</span><span class="o">&gt;</span><span class="mi">0</span><span class="p">:</span>
219                <span class="n">dataStarted</span> <span class="o">=</span> <span class="bp">True</span>
220               
221                <span class="c"># Check that we have all the info</span>
222                <span class="k">if</span> <span class="n">wavelength</span> <span class="o">==</span> <span class="bp">None</span> \
223                    <span class="ow">or</span> <span class="n">distance</span> <span class="o">==</span> <span class="bp">None</span> \
224                    <span class="ow">or</span> <span class="n">center_x</span> <span class="o">==</span> <span class="bp">None</span> \
225                    <span class="ow">or</span> <span class="n">center_y</span> <span class="o">==</span> <span class="bp">None</span><span class="p">:</span>
226                    <span class="n">msg</span> <span class="o">=</span> <span class="s">&quot;IgorReader:Missing information in data file&quot;</span>
227                    <span class="k">raise</span> <span class="ne">ValueError</span><span class="p">,</span> <span class="n">msg</span>
228               
229            <span class="k">if</span> <span class="n">dataStarted</span> <span class="o">==</span> <span class="bp">True</span><span class="p">:</span>
230                <span class="k">try</span><span class="p">:</span>
231                    <span class="n">value</span> <span class="o">=</span> <span class="nb">float</span><span class="p">(</span><span class="n">line</span><span class="p">)</span>
232                <span class="k">except</span><span class="p">:</span>
233                    <span class="c"># Found a non-float entry, skip it</span>
234                    <span class="k">continue</span>
235               
236                <span class="c"># Get bin number</span>
237                <span class="k">if</span> <span class="n">math</span><span class="o">.</span><span class="n">fmod</span><span class="p">(</span><span class="n">itot</span><span class="p">,</span> <span class="n">i_tot_row</span><span class="p">)</span> <span class="o">==</span> <span class="mi">0</span><span class="p">:</span>
238                    <span class="n">i_x</span> <span class="o">=</span> <span class="mi">0</span>
239                    <span class="n">i_y</span> <span class="o">+=</span> <span class="mi">1</span>
240                <span class="k">else</span><span class="p">:</span>
241                    <span class="n">i_x</span> <span class="o">+=</span> <span class="mi">1</span>
242                   
243                <span class="n">output</span><span class="o">.</span><span class="n">data</span><span class="p">[</span><span class="n">i_y</span><span class="p">][</span><span class="n">i_x</span><span class="p">]</span> <span class="o">=</span> <span class="n">value</span>
244                <span class="n">ncounts</span> <span class="o">+=</span> <span class="mi">1</span>
245               
246                <span class="c"># Det 640 x 640 mm</span>
247                <span class="c"># Q = 4pi/lambda sin(theta/2)</span>
248                <span class="c"># Bin size is 0.5 cm </span>
249                <span class="c">#REmoved +1 from theta = (i_x-center_x+1)*0.5 / distance</span>
250                <span class="c"># / 100.0 and </span>
251                <span class="c">#REmoved +1 from theta = (i_y-center_y+1)*0.5 /</span>
252                <span class="c"># distance / 100.0</span>
253                <span class="c">#ToDo: Need  complete check if the following</span>
254                <span class="c"># covert process is consistent with fitting.py.</span>
255                <span class="n">theta</span> <span class="o">=</span> <span class="p">(</span><span class="n">i_x</span> <span class="o">-</span> <span class="n">center_x</span><span class="p">)</span> <span class="o">*</span> <span class="mf">0.5</span> <span class="o">/</span> <span class="n">distance</span> <span class="o">/</span> <span class="mf">100.0</span>
256                <span class="n">qx</span> <span class="o">=</span> <span class="mf">4.0</span> <span class="o">*</span> <span class="n">math</span><span class="o">.</span><span class="n">pi</span> <span class="o">/</span> <span class="n">wavelength</span> <span class="o">*</span> <span class="n">math</span><span class="o">.</span><span class="n">sin</span><span class="p">(</span><span class="n">theta</span><span class="o">/</span><span class="mf">2.0</span><span class="p">)</span>
257
258                <span class="k">if</span> <span class="n">has_converter</span> <span class="o">==</span> <span class="bp">True</span> <span class="ow">and</span> <span class="n">output</span><span class="o">.</span><span class="n">Q_unit</span> <span class="o">!=</span> <span class="s">&#39;1/A&#39;</span><span class="p">:</span>
259                    <span class="n">qx</span> <span class="o">=</span> <span class="n">data_conv_q</span><span class="p">(</span><span class="n">qx</span><span class="p">,</span> <span class="n">units</span><span class="o">=</span><span class="n">output</span><span class="o">.</span><span class="n">Q_unit</span><span class="p">)</span>
260
261                <span class="k">if</span> <span class="n">xmin</span> <span class="o">==</span> <span class="bp">None</span> <span class="ow">or</span> <span class="n">qx</span> <span class="o">&lt;</span> <span class="n">xmin</span><span class="p">:</span>
262                    <span class="n">xmin</span> <span class="o">=</span> <span class="n">qx</span>
263                <span class="k">if</span> <span class="n">xmax</span> <span class="o">==</span> <span class="bp">None</span> <span class="ow">or</span> <span class="n">qx</span> <span class="o">&gt;</span> <span class="n">xmax</span><span class="p">:</span>
264                    <span class="n">xmax</span> <span class="o">=</span> <span class="n">qx</span>
265               
266                <span class="n">theta</span> <span class="o">=</span> <span class="p">(</span><span class="n">i_y</span> <span class="o">-</span> <span class="n">center_y</span><span class="p">)</span> <span class="o">*</span> <span class="mf">0.5</span> <span class="o">/</span> <span class="n">distance</span> <span class="o">/</span> <span class="mf">100.0</span>
267                <span class="n">qy</span> <span class="o">=</span> <span class="mf">4.0</span> <span class="o">*</span> <span class="n">math</span><span class="o">.</span><span class="n">pi</span> <span class="o">/</span> <span class="n">wavelength</span> <span class="o">*</span> <span class="n">math</span><span class="o">.</span><span class="n">sin</span><span class="p">(</span><span class="n">theta</span> <span class="o">/</span> <span class="mf">2.0</span><span class="p">)</span>
268
269                <span class="k">if</span> <span class="n">has_converter</span> <span class="o">==</span> <span class="bp">True</span> <span class="ow">and</span> <span class="n">output</span><span class="o">.</span><span class="n">Q_unit</span> <span class="o">!=</span> <span class="s">&#39;1/A&#39;</span><span class="p">:</span>
270                    <span class="n">qy</span> <span class="o">=</span> <span class="n">data_conv_q</span><span class="p">(</span><span class="n">qy</span><span class="p">,</span> <span class="n">units</span><span class="o">=</span><span class="n">output</span><span class="o">.</span><span class="n">Q_unit</span><span class="p">)</span>
271               
272                <span class="k">if</span> <span class="n">ymin</span> <span class="o">==</span> <span class="bp">None</span> <span class="ow">or</span> <span class="n">qy</span> <span class="o">&lt;</span> <span class="n">ymin</span><span class="p">:</span>
273                    <span class="n">ymin</span> <span class="o">=</span> <span class="n">qy</span>
274                <span class="k">if</span> <span class="n">ymax</span> <span class="o">==</span> <span class="bp">None</span> <span class="ow">or</span> <span class="n">qy</span> <span class="o">&gt;</span> <span class="n">ymax</span><span class="p">:</span>
275                    <span class="n">ymax</span> <span class="o">=</span> <span class="n">qy</span>
276               
277                <span class="k">if</span> <span class="ow">not</span> <span class="n">qx</span> <span class="ow">in</span> <span class="n">x</span><span class="p">:</span>
278                    <span class="n">x</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">qx</span><span class="p">)</span>
279                <span class="k">if</span> <span class="ow">not</span> <span class="n">qy</span> <span class="ow">in</span> <span class="n">y</span><span class="p">:</span>
280                    <span class="n">y</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">qy</span><span class="p">)</span>
281               
282                <span class="n">itot</span> <span class="o">+=</span> <span class="mi">1</span>
283                 
284                 
285        <span class="n">theta</span> <span class="o">=</span> <span class="mf">0.25</span> <span class="o">/</span> <span class="n">distance</span> <span class="o">/</span> <span class="mf">100.0</span>
286        <span class="n">xstep</span> <span class="o">=</span> <span class="mf">4.0</span> <span class="o">*</span> <span class="n">math</span><span class="o">.</span><span class="n">pi</span> <span class="o">/</span> <span class="n">wavelength</span> <span class="o">*</span> <span class="n">math</span><span class="o">.</span><span class="n">sin</span><span class="p">(</span><span class="n">theta</span> <span class="o">/</span> <span class="mf">2.0</span><span class="p">)</span>
287       
288        <span class="n">theta</span> <span class="o">=</span> <span class="mf">0.25</span> <span class="o">/</span> <span class="n">distance</span> <span class="o">/</span> <span class="mf">100.0</span>
289        <span class="n">ystep</span> <span class="o">=</span> <span class="mf">4.0</span> <span class="o">*</span> <span class="n">math</span><span class="o">.</span><span class="n">pi</span><span class="o">/</span> <span class="n">wavelength</span> <span class="o">*</span> <span class="n">math</span><span class="o">.</span><span class="n">sin</span><span class="p">(</span><span class="n">theta</span> <span class="o">/</span> <span class="mf">2.0</span><span class="p">)</span>
290       
291        <span class="c"># Store all data ######################################</span>
292        <span class="c"># Store wavelength</span>
293        <span class="k">if</span> <span class="n">has_converter</span> <span class="o">==</span> <span class="bp">True</span> <span class="ow">and</span> <span class="n">output</span><span class="o">.</span><span class="n">source</span><span class="o">.</span><span class="n">wavelength_unit</span> <span class="o">!=</span> <span class="s">&#39;A&#39;</span><span class="p">:</span>
294            <span class="n">conv</span> <span class="o">=</span> <span class="n">Converter</span><span class="p">(</span><span class="s">&#39;A&#39;</span><span class="p">)</span>
295            <span class="n">wavelength</span> <span class="o">=</span> <span class="n">conv</span><span class="p">(</span><span class="n">wavelength</span><span class="p">,</span> <span class="n">units</span><span class="o">=</span><span class="n">output</span><span class="o">.</span><span class="n">source</span><span class="o">.</span><span class="n">wavelength_unit</span><span class="p">)</span>
296        <span class="n">output</span><span class="o">.</span><span class="n">source</span><span class="o">.</span><span class="n">wavelength</span> <span class="o">=</span> <span class="n">wavelength</span>
297
298        <span class="c"># Store distance</span>
299        <span class="k">if</span> <span class="n">has_converter</span> <span class="o">==</span> <span class="bp">True</span> <span class="ow">and</span> <span class="n">detector</span><span class="o">.</span><span class="n">distance_unit</span> <span class="o">!=</span> <span class="s">&#39;m&#39;</span><span class="p">:</span>
300            <span class="n">conv</span> <span class="o">=</span> <span class="n">Converter</span><span class="p">(</span><span class="s">&#39;m&#39;</span><span class="p">)</span>
301            <span class="n">distance</span> <span class="o">=</span> <span class="n">conv</span><span class="p">(</span><span class="n">distance</span><span class="p">,</span> <span class="n">units</span><span class="o">=</span><span class="n">detector</span><span class="o">.</span><span class="n">distance_unit</span><span class="p">)</span>
302        <span class="n">detector</span><span class="o">.</span><span class="n">distance</span> <span class="o">=</span> <span class="n">distance</span>
303 
304        <span class="c"># Store transmission</span>
305        <span class="n">output</span><span class="o">.</span><span class="n">sample</span><span class="o">.</span><span class="n">transmission</span> <span class="o">=</span> <span class="n">transmission</span>
306       
307        <span class="c"># Store pixel size</span>
308        <span class="n">pixel</span> <span class="o">=</span> <span class="mf">5.0</span>
309        <span class="k">if</span> <span class="n">has_converter</span> <span class="o">==</span> <span class="bp">True</span> <span class="ow">and</span> <span class="n">detector</span><span class="o">.</span><span class="n">pixel_size_unit</span> <span class="o">!=</span> <span class="s">&#39;mm&#39;</span><span class="p">:</span>
310            <span class="n">conv</span> <span class="o">=</span> <span class="n">Converter</span><span class="p">(</span><span class="s">&#39;mm&#39;</span><span class="p">)</span>
311            <span class="n">pixel</span> <span class="o">=</span> <span class="n">conv</span><span class="p">(</span><span class="n">pixel</span><span class="p">,</span> <span class="n">units</span><span class="o">=</span><span class="n">detector</span><span class="o">.</span><span class="n">pixel_size_unit</span><span class="p">)</span>
312        <span class="n">detector</span><span class="o">.</span><span class="n">pixel_size</span><span class="o">.</span><span class="n">x</span> <span class="o">=</span> <span class="n">pixel</span>
313        <span class="n">detector</span><span class="o">.</span><span class="n">pixel_size</span><span class="o">.</span><span class="n">y</span> <span class="o">=</span> <span class="n">pixel</span>
314 
315        <span class="c"># Store beam center in distance units</span>
316        <span class="n">detector</span><span class="o">.</span><span class="n">beam_center</span><span class="o">.</span><span class="n">x</span> <span class="o">=</span> <span class="n">center_x</span> <span class="o">*</span> <span class="n">pixel</span>
317        <span class="n">detector</span><span class="o">.</span><span class="n">beam_center</span><span class="o">.</span><span class="n">y</span> <span class="o">=</span> <span class="n">center_y</span> <span class="o">*</span> <span class="n">pixel</span>
318       
319        <span class="c"># Store limits of the image (2D array)</span>
320        <span class="n">xmin</span> <span class="o">=</span> <span class="n">xmin</span> <span class="o">-</span> <span class="n">xstep</span> <span class="o">/</span> <span class="mf">2.0</span>
321        <span class="n">xmax</span> <span class="o">=</span> <span class="n">xmax</span> <span class="o">+</span> <span class="n">xstep</span> <span class="o">/</span> <span class="mf">2.0</span>
322        <span class="n">ymin</span> <span class="o">=</span> <span class="n">ymin</span> <span class="o">-</span> <span class="n">ystep</span> <span class="o">/</span> <span class="mf">2.0</span>
323        <span class="n">ymax</span> <span class="o">=</span> <span class="n">ymax</span> <span class="o">+</span> <span class="n">ystep</span> <span class="o">/</span> <span class="mf">2.0</span>
324        <span class="k">if</span> <span class="n">has_converter</span> <span class="o">==</span> <span class="bp">True</span> <span class="ow">and</span> <span class="n">output</span><span class="o">.</span><span class="n">Q_unit</span> <span class="o">!=</span> <span class="s">&#39;1/A&#39;</span><span class="p">:</span>
325            <span class="n">xmin</span> <span class="o">=</span> <span class="n">data_conv_q</span><span class="p">(</span><span class="n">xmin</span><span class="p">,</span> <span class="n">units</span><span class="o">=</span><span class="n">output</span><span class="o">.</span><span class="n">Q_unit</span><span class="p">)</span>
326            <span class="n">xmax</span> <span class="o">=</span> <span class="n">data_conv_q</span><span class="p">(</span><span class="n">xmax</span><span class="p">,</span> <span class="n">units</span><span class="o">=</span><span class="n">output</span><span class="o">.</span><span class="n">Q_unit</span><span class="p">)</span>
327            <span class="n">ymin</span> <span class="o">=</span> <span class="n">data_conv_q</span><span class="p">(</span><span class="n">ymin</span><span class="p">,</span> <span class="n">units</span><span class="o">=</span><span class="n">output</span><span class="o">.</span><span class="n">Q_unit</span><span class="p">)</span>
328            <span class="n">ymax</span> <span class="o">=</span> <span class="n">data_conv_q</span><span class="p">(</span><span class="n">ymax</span><span class="p">,</span> <span class="n">units</span><span class="o">=</span><span class="n">output</span><span class="o">.</span><span class="n">Q_unit</span><span class="p">)</span>
329        <span class="n">output</span><span class="o">.</span><span class="n">xmin</span> <span class="o">=</span> <span class="n">xmin</span>
330        <span class="n">output</span><span class="o">.</span><span class="n">xmax</span> <span class="o">=</span> <span class="n">xmax</span>
331        <span class="n">output</span><span class="o">.</span><span class="n">ymin</span> <span class="o">=</span> <span class="n">ymin</span>
332        <span class="n">output</span><span class="o">.</span><span class="n">ymax</span> <span class="o">=</span> <span class="n">ymax</span>
333       
334        <span class="c"># Store x and y axis bin centers</span>
335        <span class="n">output</span><span class="o">.</span><span class="n">x_bins</span> <span class="o">=</span> <span class="n">x</span>
336        <span class="n">output</span><span class="o">.</span><span class="n">y_bins</span> <span class="o">=</span> <span class="n">y</span>
337       
338        <span class="c"># Units</span>
339        <span class="k">if</span> <span class="n">data_conv_q</span> <span class="ow">is</span> <span class="ow">not</span> <span class="bp">None</span><span class="p">:</span>
340            <span class="n">output</span><span class="o">.</span><span class="n">xaxis</span><span class="p">(</span><span class="s">&quot;</span><span class="se">\\</span><span class="s">rm{Q_{x}}&quot;</span><span class="p">,</span> <span class="n">output</span><span class="o">.</span><span class="n">Q_unit</span><span class="p">)</span>
341            <span class="n">output</span><span class="o">.</span><span class="n">yaxis</span><span class="p">(</span><span class="s">&quot;</span><span class="se">\\</span><span class="s">rm{Q_{y}}&quot;</span><span class="p">,</span> <span class="n">output</span><span class="o">.</span><span class="n">Q_unit</span><span class="p">)</span>
342        <span class="k">else</span><span class="p">:</span>
343            <span class="n">output</span><span class="o">.</span><span class="n">xaxis</span><span class="p">(</span><span class="s">&quot;</span><span class="se">\\</span><span class="s">rm{Q_{x}}&quot;</span><span class="p">,</span> <span class="s">&#39;A^{-1}&#39;</span><span class="p">)</span>
344            <span class="n">output</span><span class="o">.</span><span class="n">yaxis</span><span class="p">(</span><span class="s">&quot;</span><span class="se">\\</span><span class="s">rm{Q_{y}}&quot;</span><span class="p">,</span> <span class="s">&#39;A^{-1}&#39;</span><span class="p">)</span>
345           
346        <span class="k">if</span> <span class="n">data_conv_i</span> <span class="ow">is</span> <span class="ow">not</span> <span class="bp">None</span><span class="p">:</span>
347            <span class="n">output</span><span class="o">.</span><span class="n">zaxis</span><span class="p">(</span><span class="s">&quot;</span><span class="se">\\</span><span class="s">rm{Intensity}&quot;</span><span class="p">,</span> <span class="n">output</span><span class="o">.</span><span class="n">I_unit</span><span class="p">)</span>
348        <span class="k">else</span><span class="p">:</span>
349            <span class="n">output</span><span class="o">.</span><span class="n">zaxis</span><span class="p">(</span><span class="s">&quot;</span><span class="se">\\</span><span class="s">rm{Intensity}&quot;</span><span class="p">,</span> <span class="s">&quot;cm^{-1}&quot;</span><span class="p">)</span>
350   
351        <span class="c"># Store loading process information</span>
352        <span class="n">output</span><span class="o">.</span><span class="n">meta_data</span><span class="p">[</span><span class="s">&#39;loader&#39;</span><span class="p">]</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">type_name</span>
353        <span class="n">output</span> <span class="o">=</span> <span class="n">reader2D_converter</span><span class="p">(</span><span class="n">output</span><span class="p">)</span>
354
355        <span class="k">return</span> <span class="n">output</span></div></div>
356</pre></div>
357
358          </div>
359        </div>
360      </div>
361      <div class="sphinxsidebar">
362        <div class="sphinxsidebarwrapper">
363<div id="searchbox" style="display: none">
364  <h3>Quick search</h3>
365    <form class="search" action="../../../../search.html" method="get">
366      <input type="text" name="q" />
367      <input type="submit" value="Go" />
368      <input type="hidden" name="check_keywords" value="yes" />
369      <input type="hidden" name="area" value="default" />
370    </form>
371    <p class="searchtip" style="font-size: 90%">
372    Enter search terms or a module, class or function name.
373    </p>
374</div>
375<script type="text/javascript">$('#searchbox').show(0);</script>
376        </div>
377      </div>
378      <div class="clearer"></div>
379    </div>
380    <div class="related">
381      <h3>Navigation</h3>
382      <ul>
383        <li class="right" style="margin-right: 10px">
384          <a href="../../../../genindex.html" title="General Index"
385             >index</a></li>
386        <li class="right" >
387          <a href="../../../../py-modindex.html" title="Python Module Index"
388             >modules</a> |</li>
389        <li><a href="../../../../index.html">SasView 3.0.0 documentation</a> &raquo;</li>
390          <li><a href="../../../index.html" >Module code</a> &raquo;</li>
391          <li><a href="../readers.html" >sas.dataloader.readers</a> &raquo;</li> 
392      </ul>
393    </div>
394    <div class="footer">
395        &copy; Copyright 2013, The SasView Project.
396      Created using <a href="http://sphinx-doc.org/">Sphinx</a> 1.2.3.
397    </div>
398  </body>
399</html>
Note: See TracBrowser for help on using the repository browser.