source: sasview/_modules/sas/pr/num_term.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: 33.3 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.pr.num_term &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.pr.num_term</h1><div class="highlight"><pre>
51<span class="kn">import</span> <span class="nn">math</span>
52<span class="kn">import</span> <span class="nn">numpy</span>
53<span class="kn">import</span> <span class="nn">copy</span>
54<span class="kn">from</span> <span class="nn">sas.pr.invertor</span> <span class="kn">import</span> <span class="n">Invertor</span>
55
56
57<div class="viewcode-block" id="Num_terms"><a class="viewcode-back" href="../../../dev/api/sas.pr.html#sas.pr.num_term.Num_terms">[docs]</a><span class="k">class</span> <span class="nc">Num_terms</span><span class="p">():</span>
58    <span class="sd">&quot;&quot;&quot;</span>
59<span class="sd">    &quot;&quot;&quot;</span>
60    <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">invertor</span><span class="p">):</span>
61        <span class="sd">&quot;&quot;&quot;</span>
62<span class="sd">        &quot;&quot;&quot;</span>
63        <span class="bp">self</span><span class="o">.</span><span class="n">invertor</span> <span class="o">=</span> <span class="n">invertor</span>
64        <span class="bp">self</span><span class="o">.</span><span class="n">nterm_min</span> <span class="o">=</span> <span class="mi">10</span>
65        <span class="bp">self</span><span class="o">.</span><span class="n">nterm_max</span> <span class="o">=</span> <span class="nb">len</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">invertor</span><span class="o">.</span><span class="n">x</span><span class="p">)</span>
66        <span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">nterm_max</span> <span class="o">&gt;</span> <span class="mi">50</span><span class="p">:</span>
67            <span class="bp">self</span><span class="o">.</span><span class="n">nterm_max</span> <span class="o">=</span> <span class="mi">50</span>
68        <span class="bp">self</span><span class="o">.</span><span class="n">isquit_func</span> <span class="o">=</span> <span class="bp">None</span>
69         
70        <span class="bp">self</span><span class="o">.</span><span class="n">osc_list</span> <span class="o">=</span> <span class="p">[]</span>
71        <span class="bp">self</span><span class="o">.</span><span class="n">err_list</span> <span class="o">=</span> <span class="p">[]</span>
72        <span class="bp">self</span><span class="o">.</span><span class="n">alpha_list</span> <span class="o">=</span> <span class="p">[]</span>
73        <span class="bp">self</span><span class="o">.</span><span class="n">mess_list</span> <span class="o">=</span> <span class="p">[]</span>
74         
75        <span class="bp">self</span><span class="o">.</span><span class="n">dataset</span> <span class="o">=</span> <span class="p">[]</span>
76     
77<div class="viewcode-block" id="Num_terms.is_odd"><a class="viewcode-back" href="../../../dev/api/sas.pr.html#sas.pr.num_term.Num_terms.is_odd">[docs]</a>    <span class="k">def</span> <span class="nf">is_odd</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">n</span><span class="p">):</span>
78        <span class="sd">&quot;&quot;&quot;</span>
79<span class="sd">        &quot;&quot;&quot;</span>
80        <span class="k">return</span> <span class="nb">bool</span><span class="p">(</span><span class="n">n</span> <span class="o">%</span> <span class="mi">2</span><span class="p">)</span>
81</div>
82<div class="viewcode-block" id="Num_terms.sort_osc"><a class="viewcode-back" href="../../../dev/api/sas.pr.html#sas.pr.num_term.Num_terms.sort_osc">[docs]</a>    <span class="k">def</span> <span class="nf">sort_osc</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
83        <span class="sd">&quot;&quot;&quot;</span>
84<span class="sd">        &quot;&quot;&quot;</span>
85        <span class="c">#import copy</span>
86        <span class="n">osc</span> <span class="o">=</span> <span class="n">copy</span><span class="o">.</span><span class="n">deepcopy</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">dataset</span><span class="p">)</span>
87        <span class="n">lis</span> <span class="o">=</span> <span class="p">[]</span>
88        <span class="k">for</span> <span class="n">i</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">osc</span><span class="p">)):</span>
89            <span class="n">osc</span><span class="o">.</span><span class="n">sort</span><span class="p">()</span>
90            <span class="n">re</span> <span class="o">=</span> <span class="n">osc</span><span class="o">.</span><span class="n">pop</span><span class="p">(</span><span class="mi">0</span><span class="p">)</span>
91            <span class="n">lis</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">re</span><span class="p">)</span>
92        <span class="k">return</span> <span class="n">lis</span>
93           </div>
94<div class="viewcode-block" id="Num_terms.median_osc"><a class="viewcode-back" href="../../../dev/api/sas.pr.html#sas.pr.num_term.Num_terms.median_osc">[docs]</a>    <span class="k">def</span> <span class="nf">median_osc</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
95        <span class="sd">&quot;&quot;&quot;</span>
96<span class="sd">        &quot;&quot;&quot;</span>
97        <span class="n">osc</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">sort_osc</span><span class="p">()</span>
98        <span class="n">dv</span> <span class="o">=</span> <span class="nb">len</span><span class="p">(</span><span class="n">osc</span><span class="p">)</span>
99        <span class="n">med</span> <span class="o">=</span> <span class="nb">float</span><span class="p">(</span><span class="n">dv</span><span class="p">)</span> <span class="o">/</span> <span class="mf">2.0</span>
100        <span class="n">odd</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">is_odd</span><span class="p">(</span><span class="n">dv</span><span class="p">)</span>
101        <span class="n">medi</span> <span class="o">=</span> <span class="mi">0</span>
102        <span class="k">for</span> <span class="n">i</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="n">dv</span><span class="p">):</span>
103            <span class="k">if</span> <span class="n">odd</span> <span class="o">==</span> <span class="bp">True</span><span class="p">:</span>
104                <span class="n">medi</span> <span class="o">=</span> <span class="n">osc</span><span class="p">[</span><span class="nb">int</span><span class="p">(</span><span class="n">med</span><span class="p">)]</span>
105            <span class="k">else</span><span class="p">:</span>
106                <span class="n">medi</span> <span class="o">=</span> <span class="n">osc</span><span class="p">[</span><span class="nb">int</span><span class="p">(</span><span class="n">med</span><span class="p">)</span> <span class="o">-</span> <span class="mi">1</span><span class="p">]</span>
107        <span class="k">return</span> <span class="n">medi</span>
108</div>
109<div class="viewcode-block" id="Num_terms.get0_out"><a class="viewcode-back" href="../../../dev/api/sas.pr.html#sas.pr.num_term.Num_terms.get0_out">[docs]</a>    <span class="k">def</span> <span class="nf">get0_out</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
110        <span class="sd">&quot;&quot;&quot;</span>
111<span class="sd">        &quot;&quot;&quot;</span>
112        <span class="n">inver</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">invertor</span>
113        <span class="bp">self</span><span class="o">.</span><span class="n">osc_list</span> <span class="o">=</span> <span class="p">[]</span>
114        <span class="bp">self</span><span class="o">.</span><span class="n">err_list</span> <span class="o">=</span> <span class="p">[]</span>
115        <span class="bp">self</span><span class="o">.</span><span class="n">alpha_list</span> <span class="o">=</span> <span class="p">[]</span>
116        <span class="k">for</span> <span class="n">k</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">nterm_min</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">nterm_max</span><span class="p">,</span> <span class="mi">1</span><span class="p">):</span>
117            <span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">isquit_func</span> <span class="o">!=</span> <span class="bp">None</span><span class="p">:</span>
118                <span class="bp">self</span><span class="o">.</span><span class="n">isquit_func</span><span class="p">()</span>
119            <span class="n">best_alpha</span><span class="p">,</span> <span class="n">message</span><span class="p">,</span> <span class="n">_</span> <span class="o">=</span> <span class="n">inver</span><span class="o">.</span><span class="n">estimate_alpha</span><span class="p">(</span><span class="n">k</span><span class="p">)</span>
120            <span class="n">inver</span><span class="o">.</span><span class="n">alpha</span> <span class="o">=</span> <span class="n">best_alpha</span>
121            <span class="n">inver</span><span class="o">.</span><span class="n">out</span><span class="p">,</span> <span class="n">inver</span><span class="o">.</span><span class="n">cov</span> <span class="o">=</span> <span class="n">inver</span><span class="o">.</span><span class="n">lstsq</span><span class="p">(</span><span class="n">k</span><span class="p">)</span>
122            <span class="n">osc</span> <span class="o">=</span> <span class="n">inver</span><span class="o">.</span><span class="n">oscillations</span><span class="p">(</span><span class="n">inver</span><span class="o">.</span><span class="n">out</span><span class="p">)</span>
123            <span class="n">err</span> <span class="o">=</span> <span class="n">inver</span><span class="o">.</span><span class="n">get_pos_err</span><span class="p">(</span><span class="n">inver</span><span class="o">.</span><span class="n">out</span><span class="p">,</span> <span class="n">inver</span><span class="o">.</span><span class="n">cov</span><span class="p">)</span>
124            <span class="k">if</span> <span class="n">osc</span> <span class="o">&gt;</span> <span class="mf">10.0</span><span class="p">:</span>
125                <span class="k">break</span>
126            <span class="bp">self</span><span class="o">.</span><span class="n">osc_list</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">osc</span><span class="p">)</span>
127            <span class="bp">self</span><span class="o">.</span><span class="n">err_list</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">err</span><span class="p">)</span>
128            <span class="bp">self</span><span class="o">.</span><span class="n">alpha_list</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">inver</span><span class="o">.</span><span class="n">alpha</span><span class="p">)</span>
129            <span class="bp">self</span><span class="o">.</span><span class="n">mess_list</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">message</span><span class="p">)</span>
130         
131        <span class="n">new_osc1</span> <span class="o">=</span> <span class="p">[]</span>
132        <span class="n">new_osc2</span> <span class="o">=</span> <span class="p">[]</span>
133        <span class="n">new_osc3</span> <span class="o">=</span> <span class="p">[]</span>
134        <span class="n">flag9</span> <span class="o">=</span> <span class="bp">False</span>
135        <span class="n">flag8</span> <span class="o">=</span> <span class="bp">False</span>
136        <span class="n">flag7</span> <span class="o">=</span> <span class="bp">False</span>
137        <span class="k">for</span> <span class="n">i</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="bp">self</span><span class="o">.</span><span class="n">err_list</span><span class="p">)):</span>
138            <span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">err_list</span><span class="p">[</span><span class="n">i</span><span class="p">]</span> <span class="o">&lt;=</span> <span class="mf">1.0</span> <span class="ow">and</span> <span class="bp">self</span><span class="o">.</span><span class="n">err_list</span><span class="p">[</span><span class="n">i</span><span class="p">]</span> <span class="o">&gt;=</span> <span class="mf">0.9</span><span class="p">:</span>
139                <span class="n">new_osc1</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">osc_list</span><span class="p">[</span><span class="n">i</span><span class="p">])</span>
140                <span class="n">flag9</span> <span class="o">=</span> <span class="bp">True</span>
141            <span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">err_list</span><span class="p">[</span><span class="n">i</span><span class="p">]</span> <span class="o">&lt;</span> <span class="mf">0.9</span> <span class="ow">and</span> <span class="bp">self</span><span class="o">.</span><span class="n">err_list</span><span class="p">[</span><span class="n">i</span><span class="p">]</span> <span class="o">&gt;=</span> <span class="mf">0.8</span><span class="p">:</span>
142                <span class="n">new_osc2</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">osc_list</span><span class="p">[</span><span class="n">i</span><span class="p">])</span>
143                <span class="n">flag8</span> <span class="o">=</span> <span class="bp">True</span>
144            <span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">err_list</span><span class="p">[</span><span class="n">i</span><span class="p">]</span> <span class="o">&lt;</span> <span class="mf">0.8</span> <span class="ow">and</span> <span class="bp">self</span><span class="o">.</span><span class="n">err_list</span><span class="p">[</span><span class="n">i</span><span class="p">]</span> <span class="o">&gt;=</span> <span class="mf">0.7</span><span class="p">:</span>
145                <span class="n">new_osc3</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">osc_list</span><span class="p">[</span><span class="n">i</span><span class="p">])</span>
146                <span class="n">flag7</span> <span class="o">=</span> <span class="bp">True</span>
147                 
148        <span class="k">if</span> <span class="n">flag9</span> <span class="o">==</span> <span class="bp">True</span><span class="p">:</span>
149            <span class="bp">self</span><span class="o">.</span><span class="n">dataset</span> <span class="o">=</span> <span class="n">new_osc1</span>
150        <span class="k">elif</span> <span class="n">flag8</span> <span class="o">==</span> <span class="bp">True</span><span class="p">:</span>
151            <span class="bp">self</span><span class="o">.</span><span class="n">dataset</span> <span class="o">=</span> <span class="n">new_osc2</span>
152        <span class="k">else</span><span class="p">:</span>
153            <span class="bp">self</span><span class="o">.</span><span class="n">dataset</span> <span class="o">=</span> <span class="n">new_osc3</span>
154         
155        <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">dataset</span>
156        </div>
157<div class="viewcode-block" id="Num_terms.ls_osc"><a class="viewcode-back" href="../../../dev/api/sas.pr.html#sas.pr.num_term.Num_terms.ls_osc">[docs]</a>    <span class="k">def</span> <span class="nf">ls_osc</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
158        <span class="sd">&quot;&quot;&quot;</span>
159<span class="sd">        &quot;&quot;&quot;</span>
160        <span class="c"># Generate data</span>
161        <span class="n">ls_osc</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">get0_out</span><span class="p">()</span>
162        <span class="n">med</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">median_osc</span><span class="p">()</span>
163       
164        <span class="c">#TODO: check 1</span>
165        <span class="n">ls_osc</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">dataset</span>
166        <span class="n">ls</span> <span class="o">=</span> <span class="p">[]</span>
167        <span class="k">for</span> <span class="n">i</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">ls_osc</span><span class="p">)):</span>
168            <span class="k">if</span> <span class="nb">int</span><span class="p">(</span><span class="n">med</span><span class="p">)</span> <span class="o">==</span> <span class="nb">int</span><span class="p">(</span><span class="n">ls_osc</span><span class="p">[</span><span class="n">i</span><span class="p">]):</span>
169                <span class="n">ls</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">ls_osc</span><span class="p">[</span><span class="n">i</span><span class="p">])</span>
170        <span class="k">return</span> <span class="n">ls</span>
171</div>
172<div class="viewcode-block" id="Num_terms.compare_err"><a class="viewcode-back" href="../../../dev/api/sas.pr.html#sas.pr.num_term.Num_terms.compare_err">[docs]</a>    <span class="k">def</span> <span class="nf">compare_err</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
173        <span class="sd">&quot;&quot;&quot;</span>
174<span class="sd">        &quot;&quot;&quot;</span>
175        <span class="n">ls</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">ls_osc</span><span class="p">()</span>
176        <span class="c">#print &quot;ls&quot;, ls</span>
177        <span class="n">nt_ls</span> <span class="o">=</span> <span class="p">[]</span>
178        <span class="k">for</span> <span class="n">i</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">ls</span><span class="p">)):</span>
179            <span class="n">r</span> <span class="o">=</span> <span class="n">ls</span><span class="p">[</span><span class="n">i</span><span class="p">]</span>
180            <span class="n">n</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">osc_list</span><span class="o">.</span><span class="n">index</span><span class="p">(</span><span class="n">r</span><span class="p">)</span> <span class="o">+</span> <span class="mi">10</span>
181            <span class="c">#er = self.err_list[n]</span>
182            <span class="c">#nt = self.osc_list.index(r) + 10</span>
183            <span class="n">nt_ls</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">n</span><span class="p">)</span>
184        <span class="c">#print &quot;nt list&quot;, nt_ls</span>
185        <span class="k">return</span> <span class="n">nt_ls</span>
186</div>
187<div class="viewcode-block" id="Num_terms.num_terms"><a class="viewcode-back" href="../../../dev/api/sas.pr.html#sas.pr.num_term.Num_terms.num_terms">[docs]</a>    <span class="k">def</span> <span class="nf">num_terms</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">isquit_func</span><span class="o">=</span><span class="bp">None</span><span class="p">):</span>
188        <span class="sd">&quot;&quot;&quot;</span>
189<span class="sd">        &quot;&quot;&quot;</span>
190        <span class="k">try</span><span class="p">:</span>
191            <span class="bp">self</span><span class="o">.</span><span class="n">isquit_func</span> <span class="o">=</span> <span class="n">isquit_func</span>
192            <span class="n">nts</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">compare_err</span><span class="p">()</span>
193            <span class="n">div</span> <span class="o">=</span> <span class="nb">len</span><span class="p">(</span><span class="n">nts</span><span class="p">)</span>
194            <span class="n">tem</span> <span class="o">=</span> <span class="nb">float</span><span class="p">(</span><span class="n">div</span><span class="p">)</span><span class="o">/</span><span class="mf">2.0</span>
195            <span class="n">odd</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">is_odd</span><span class="p">(</span><span class="n">div</span><span class="p">)</span>
196            <span class="k">if</span> <span class="n">odd</span> <span class="o">==</span> <span class="bp">True</span><span class="p">:</span>
197                <span class="n">nt</span> <span class="o">=</span> <span class="n">nts</span><span class="p">[</span><span class="nb">int</span><span class="p">(</span><span class="n">tem</span><span class="p">)]</span>
198            <span class="k">else</span><span class="p">:</span>
199                <span class="n">nt</span> <span class="o">=</span> <span class="n">nts</span><span class="p">[</span><span class="nb">int</span><span class="p">(</span><span class="n">tem</span><span class="p">)</span> <span class="o">-</span> <span class="mi">1</span><span class="p">]</span>
200            <span class="k">return</span> <span class="n">nt</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">alpha_list</span><span class="p">[</span><span class="n">nt</span> <span class="o">-</span> <span class="mi">10</span><span class="p">],</span> <span class="bp">self</span><span class="o">.</span><span class="n">mess_list</span><span class="p">[</span><span class="n">nt</span><span class="o">-</span><span class="mi">10</span><span class="p">]</span>
201        <span class="k">except</span><span class="p">:</span>
202            <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">nterm_min</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">alpha_list</span><span class="p">[</span><span class="mi">10</span><span class="p">],</span> <span class="bp">self</span><span class="o">.</span><span class="n">mess_list</span><span class="p">[</span><span class="mi">10</span><span class="p">]</span>
203
204
205<span class="c">#For testing</span></div></div>
206<div class="viewcode-block" id="load"><a class="viewcode-back" href="../../../dev/api/sas.pr.html#sas.pr.num_term.load">[docs]</a><span class="k">def</span> <span class="nf">load</span><span class="p">(</span><span class="n">path</span><span class="p">):</span>
207    <span class="c"># Read the data from the data file</span>
208    <span class="n">data_x</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="mi">0</span><span class="p">)</span>
209    <span class="n">data_y</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="mi">0</span><span class="p">)</span>
210    <span class="n">data_err</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="mi">0</span><span class="p">)</span>
211    <span class="n">scale</span>    <span class="o">=</span> <span class="bp">None</span>
212    <span class="n">min_err</span>  <span class="o">=</span> <span class="mf">0.0</span>
213    <span class="k">if</span> <span class="ow">not</span> <span class="n">path</span> <span class="o">==</span> <span class="bp">None</span><span class="p">:</span>
214        <span class="n">input_f</span> <span class="o">=</span> <span class="nb">open</span><span class="p">(</span><span class="n">path</span><span class="p">,</span><span class="s">&#39;r&#39;</span><span class="p">)</span>
215        <span class="n">buff</span>    <span class="o">=</span> <span class="n">input_f</span><span class="o">.</span><span class="n">read</span><span class="p">()</span>
216        <span class="n">lines</span>   <span class="o">=</span> <span class="n">buff</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>
217        <span class="k">for</span> <span class="n">line</span> <span class="ow">in</span> <span class="n">lines</span><span class="p">:</span>
218            <span class="k">try</span><span class="p">:</span>
219                <span class="n">toks</span> <span class="o">=</span> <span class="n">line</span><span class="o">.</span><span class="n">split</span><span class="p">()</span>
220                <span class="n">test_x</span> <span class="o">=</span> <span class="nb">float</span><span class="p">(</span><span class="n">toks</span><span class="p">[</span><span class="mi">0</span><span class="p">])</span>
221                <span class="n">test_y</span> <span class="o">=</span> <span class="nb">float</span><span class="p">(</span><span class="n">toks</span><span class="p">[</span><span class="mi">1</span><span class="p">])</span>
222                <span class="k">if</span> <span class="nb">len</span><span class="p">(</span><span class="n">toks</span><span class="p">)</span> <span class="o">&gt;</span> <span class="mi">2</span><span class="p">:</span>
223                    <span class="n">err</span> <span class="o">=</span> <span class="nb">float</span><span class="p">(</span><span class="n">toks</span><span class="p">[</span><span class="mi">2</span><span class="p">])</span>
224                <span class="k">else</span><span class="p">:</span>
225                    <span class="k">if</span> <span class="n">scale</span> <span class="o">==</span> <span class="bp">None</span><span class="p">:</span>
226                        <span class="n">scale</span> <span class="o">=</span> <span class="mf">0.05</span> <span class="o">*</span> <span class="n">math</span><span class="o">.</span><span class="n">sqrt</span><span class="p">(</span><span class="n">test_y</span><span class="p">)</span>
227                        <span class="c">#scale = 0.05/math.sqrt(y)</span>
228                        <span class="n">min_err</span> <span class="o">=</span> <span class="mf">0.01</span> <span class="o">*</span> <span class="n">y</span>
229                    <span class="n">err</span> <span class="o">=</span> <span class="n">scale</span> <span class="o">*</span> <span class="n">math</span><span class="o">.</span><span class="n">sqrt</span><span class="p">(</span><span class="n">test_y</span><span class="p">)</span> <span class="o">+</span> <span class="n">min_err</span>
230                    <span class="c">#err = 0</span>
231                   
232                <span class="n">data_x</span> <span class="o">=</span> <span class="n">numpy</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">data_x</span><span class="p">,</span> <span class="n">test_x</span><span class="p">)</span>
233                <span class="n">data_y</span> <span class="o">=</span> <span class="n">numpy</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">data_y</span><span class="p">,</span> <span class="n">test_y</span><span class="p">)</span>
234                <span class="n">data_err</span> <span class="o">=</span> <span class="n">numpy</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">data_err</span><span class="p">,</span> <span class="n">err</span><span class="p">)</span>
235            <span class="k">except</span><span class="p">:</span>
236                <span class="k">pass</span>
237               
238    <span class="k">return</span> <span class="n">data_x</span><span class="p">,</span> <span class="n">data_y</span><span class="p">,</span> <span class="n">data_err</span>
239
240</div>
241<span class="k">if</span> <span class="n">__name__</span> <span class="o">==</span> <span class="s">&quot;__main__&quot;</span><span class="p">:</span>
242    <span class="n">i</span> <span class="o">=</span> <span class="n">Invertor</span><span class="p">()</span>
243    <span class="n">x</span><span class="p">,</span> <span class="n">y</span><span class="p">,</span> <span class="n">erro</span> <span class="o">=</span> <span class="n">load</span><span class="p">(</span><span class="s">&quot;test/Cyl_A_D102.txt&quot;</span><span class="p">)</span>
244    <span class="n">i</span><span class="o">.</span><span class="n">d_max</span> <span class="o">=</span> <span class="mf">102.0</span>
245    <span class="n">i</span><span class="o">.</span><span class="n">nfunc</span> <span class="o">=</span> <span class="mi">10</span>
246    <span class="c">#i.q_max = 0.4</span>
247    <span class="c">#i.q_min = 0.07</span>
248    <span class="n">i</span><span class="o">.</span><span class="n">x</span> <span class="o">=</span> <span class="n">x</span>
249    <span class="n">i</span><span class="o">.</span><span class="n">y</span> <span class="o">=</span> <span class="n">y</span>
250    <span class="n">i</span><span class="o">.</span><span class="n">err</span> <span class="o">=</span> <span class="n">erro</span>
251    <span class="c">#i.out, i.cov = i.lstsq(10)</span>
252    <span class="c"># Testing estimator</span>
253    <span class="n">est</span> <span class="o">=</span> <span class="n">Num_terms</span><span class="p">(</span><span class="n">i</span><span class="p">)</span>
254    <span class="k">print</span> <span class="n">est</span><span class="o">.</span><span class="n">num_terms</span><span class="p">()</span>
255</pre></div>
256
257          </div>
258        </div>
259      </div>
260      <div class="sphinxsidebar">
261        <div class="sphinxsidebarwrapper">
262<div id="searchbox" style="display: none">
263  <h3>Quick search</h3>
264    <form class="search" action="../../../search.html" method="get">
265      <input type="text" name="q" />
266      <input type="submit" value="Go" />
267      <input type="hidden" name="check_keywords" value="yes" />
268      <input type="hidden" name="area" value="default" />
269    </form>
270    <p class="searchtip" style="font-size: 90%">
271    Enter search terms or a module, class or function name.
272    </p>
273</div>
274<script type="text/javascript">$('#searchbox').show(0);</script>
275        </div>
276      </div>
277      <div class="clearer"></div>
278    </div>
279    <div class="related">
280      <h3>Navigation</h3>
281      <ul>
282        <li class="right" style="margin-right: 10px">
283          <a href="../../../genindex.html" title="General Index"
284             >index</a></li>
285        <li class="right" >
286          <a href="../../../py-modindex.html" title="Python Module Index"
287             >modules</a> |</li>
288        <li><a href="../../../index.html">SasView 3.0.0 documentation</a> &raquo;</li>
289          <li><a href="../../index.html" >Module code</a> &raquo;</li> 
290      </ul>
291    </div>
292    <div class="footer">
293        &copy; Copyright 2013, The SasView Project.
294      Created using <a href="http://sphinx-doc.org/">Sphinx</a> 1.2.3.
295    </div>
296  </body>
297</html>
Note: See TracBrowser for help on using the repository browser.