<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN"
"http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd">
<html xmlns="http://www.w3.org/1999/xhtml">
<head>
<meta http-equiv="Content-Type" content="text/html; charset=utf-8" />
<title>Golem No: 10104 - the only fully opensource tokamak </title>
<link rel="shortcut icon" href="/favicon.ico" type="image/x-icon" />
<link rel="stylesheet" href="/_static/default.css" type="text/css" />
<link rel="stylesheet" href="/_static/pygments.css" type="text/css" />
<link rel="stylesheet" href="/_static/lightbox.css" type="text/css" media="screen" />
<script type="text/javascript">
var DOCUMENTATION_OPTIONS = {
URL_ROOT: '',
VERSION: '0.2',
COLLAPSE_INDEX: false,
FILE_SUFFIX: '.html',
HAS_SOURCE: true
};
</script>
<script type="text/javascript" src="/_static/jquery.js"></script>
<script type="text/javascript" src="/_static/underscore.js"></script>
<script type="text/javascript" src="/_static/doctools.js"></script>
<script type="text/javascript" src="/_static/sidebar.js"></script>
<script src="/_static/lightbox_plus.js"></script>
<script type="text/javascript">
function getURL(val){
location = '/shots/' + val + '/' + "DAS/1011NIbasic.ON/";
return false;
}
</script>
</head>
<body>
<div class="related">
<h3>Navigation</h3>
<ul>
<li class="right" ><a href= /shots/0/DAS/1011NIbasic.ON/ title="current shot" >current</a> </li>
<li class="right" ><a href= /shots/10105/DAS/1011NIbasic.ON/ title="next shot" >next</a> |</li>
<li class="right" ><a href= /shots/10103/DAS/1011NIbasic.ON/ title="previous shot" >previous</a> | </li>
</ul>
<ul>
<li><a href="/" > GOLEM </a> »
<a href= /shots/10104 >Shot #10104</a> » </li>
<li><a href=/shots/10104/DAS/1011NIbasic.ON/ > main </a> »
</ul>
</div>
<div class="document">
<div class="documentwrapper">
<div class="bodywrapper">
<div class="body">
<div class="section" >
<h1>Source code :: main</a></h1>
<pre><font face="Lucida,Courier New"><a class="viewcode-back" href="javascript:history.go(-1)">[Return]</a><br/><a class="viewcode-back" href="main.py">[Download]</a><font color="#008000">#!/usr/bin/python2</font>
<font color="#008000"># -*- coding: utf-8 -*-</font>
<font color="#C00000">import</font> <font color="#000000">matplotlib</font>
<font color="#000000">matplotlib</font><font color="#0000C0">.</font><font color="#000000">rcParams</font><font color="#0000C0">[</font><font color="#004080">'backend'</font><font color="#0000C0">]</font> <font color="#0000C0">=</font> <font color="#004080">'Agg'</font>
<font color="#000000">matplotlib</font><font color="#0000C0">.</font><font color="#000000">rc</font><font color="#0000C0">(</font><font color="#004080">'font'</font><font color="#0000C0">,</font> <font color="#000000">size</font><font color="#0000C0">=</font><font color="#004080">'10'</font><font color="#0000C0">)</font>
<font color="#000000">matplotlib</font><font color="#0000C0">.</font><font color="#000000">rc</font><font color="#0000C0">(</font><font color="#004080">'text'</font><font color="#0000C0">,</font> <font color="#000000">usetex</font><font color="#0000C0">=</font><font color="#000000">True</font><font color="#0000C0">)</font> <font color="#008000"># FIXME !! nicer but slower !!! </font>
<font color="#C00000">from</font> <font color="#000000">numpy</font> <font color="#C00000">import</font> <font color="#0000C0">*</font>
<font color="#C00000">from</font> <font color="#000000">pygolem_lite</font><font color="#0000C0">.</font><font color="#000000">modules</font> <font color="#C00000">import</font> <font color="#0000C0">*</font>
<font color="#C00000">import</font> <font color="#000000">time</font>
<font color="#C00000">import</font> <font color="#000000">os</font><font color="#0000C0">,</font> <font color="#000000">sys</font>
<font color="#C00000">def</font> <font color="#000000">prepare_data</font><font color="#0000C0">(</font><font color="#0000C0">)</font><font color="#0000C0">:</font>
<font color="#000000">data</font> <font color="#0000C0">=</font> <font color="#000000">loadtxt</font><font color="#0000C0">(</font><font color="#004080">'Nidatap.lvm'</font><font color="#0000C0">)</font>
<font color="#000000">tvec</font> <font color="#0000C0">=</font> <font color="#000000">arange</font><font color="#0000C0">(</font><font color="#000000">len</font><font color="#0000C0">(</font><font color="#000000">data</font><font color="#0000C0">)</font><font color="#0000C0">)</font> <font color="#0000C0">*</font> <font color="#0080C0">5e-6</font> <font color="#008000"># 200kHz</font>
<font color="#000000">save_adv</font><font color="#0000C0">(</font><font color="#004080">'Nidatap'</font><font color="#0000C0">,</font> <font color="#000000">tvec</font><font color="#0000C0">,</font> <font color="#000000">data</font><font color="#0000C0">)</font>
<font color="#C00000">print</font> <font color="#004080">"prepared data"</font>
<font color="#C00000">def</font> <font color="#000000">plot_data</font><font color="#0000C0">(</font><font color="#000000">file_type</font><font color="#0000C0">)</font><font color="#0000C0">:</font>
<font color="#000000">name</font> <font color="#0000C0">=</font> <font color="#004080">"NI Basic"</font>
<font color="#008000"># graph all</font>
<font color="#000000">out</font> <font color="#0000C0">=</font> <font color="#000000">get_data</font><font color="#0000C0">(</font><font color="#004080">'Nidatap'</font><font color="#0000C0">,</font> <font color="#004080">'ch'</font><font color="#0000C0">,</font> <font color="#004080">""</font> <font color="#0000C0">,</font><font color="#000000">xlim</font><font color="#0000C0">=</font><font color="#0000C0">[</font><font color="#0080C0">0</font><font color="#0000C0">,</font><font color="#0080C0">40</font><font color="#0000C0">]</font><font color="#0000C0">,</font><font color="#000000">integrated</font><font color="#0000C0">=</font><font color="#000000">False</font> <font color="#0000C0">)</font>
<font color="#000000">multiplot</font><font color="#0000C0">(</font><font color="#000000">out</font><font color="#0000C0">,</font> <font color="#000000">name</font> <font color="#0000C0">,</font> <font color="#004080">'graph1'</font><font color="#0000C0">,</font> <font color="#0000C0">(</font><font color="#0080C0">6</font><font color="#0000C0">,</font><font color="#000000">len</font><font color="#0000C0">(</font><font color="#000000">out</font><font color="#0000C0">)</font><font color="#0000C0">*</font><font color="#0080C0">2</font><font color="#0000C0">)</font><font color="#0000C0">,</font> <font color="#0080C0">100</font><font color="#0000C0">,</font> <font color="#004080">'vertical'</font><font color="#0000C0">,</font> <font color="#000000">file_type</font><font color="#0000C0">)</font>
<font color="#000000">os</font><font color="#0000C0">.</font><font color="#000000">system</font><font color="#0000C0">(</font><font color="#004080">'convert -resize 150x120\! graph1.png icon.png'</font><font color="#0000C0">)</font>
<font color="#008000">## integrated </font>
<font color="#000000">out</font> <font color="#0000C0">=</font> <font color="#000000">get_data</font><font color="#0000C0">(</font><font color="#004080">'Nidatap'</font><font color="#0000C0">,</font> <font color="#004080">'ch'</font><font color="#0000C0">,</font> <font color="#004080">""</font><font color="#0000C0">,</font><font color="#000000">xlim</font><font color="#0000C0">=</font><font color="#0000C0">[</font><font color="#0080C0">0</font><font color="#0000C0">,</font><font color="#0080C0">40</font><font color="#0000C0">]</font> <font color="#0000C0">,</font> <font color="#000000">integrated</font><font color="#0000C0">=</font><font color="#000000">True</font> <font color="#0000C0">)</font>
<font color="#000000">multiplot</font><font color="#0000C0">(</font><font color="#000000">out</font><font color="#0000C0">,</font> <font color="#000000">name</font> <font color="#0000C0">+</font> <font color="#004080">' Integrated'</font> <font color="#0000C0">,</font> <font color="#004080">'graph2'</font><font color="#0000C0">,</font> <font color="#0000C0">(</font><font color="#0080C0">6</font><font color="#0000C0">,</font><font color="#000000">len</font><font color="#0000C0">(</font><font color="#000000">out</font><font color="#0000C0">)</font><font color="#0000C0">*</font><font color="#0080C0">2</font><font color="#0000C0">)</font><font color="#0000C0">,</font> <font color="#0080C0">100</font><font color="#0000C0">,</font> <font color="#004080">'vertical'</font><font color="#0000C0">,</font> <font color="#000000">file_type</font><font color="#0000C0">)</font>
<font color="#C00000">def</font> <font color="#000000">main</font><font color="#0000C0">(</font><font color="#0000C0">)</font><font color="#0000C0">:</font>
<font color="#C00000">if</font> <font color="#000000">sys</font><font color="#0000C0">.</font><font color="#000000">argv</font><font color="#0000C0">[</font><font color="#0080C0">1</font><font color="#0000C0">]</font> <font color="#0000C0">==</font> <font color="#004080">"acquisition"</font><font color="#0000C0">:</font>
<font color="#000000">prepare_data</font><font color="#0000C0">(</font><font color="#0000C0">)</font>
<font color="#C00000">if</font> <font color="#000000">sys</font><font color="#0000C0">.</font><font color="#000000">argv</font><font color="#0000C0">[</font><font color="#0080C0">1</font><font color="#0000C0">]</font> <font color="#0000C0">==</font> <font color="#004080">"plots"</font><font color="#0000C0">:</font>
<font color="#000000">plot_data</font><font color="#0000C0">(</font><font color="#004080">'png'</font><font color="#0000C0">)</font>
<font color="#008000">#plot_data('svg')</font>
<font color="#000000">saveconst</font><font color="#0000C0">(</font><font color="#004080">'status'</font><font color="#0000C0">,</font> <font color="#0080C0">0</font><font color="#0000C0">)</font>
<font color="#C00000">if</font> <font color="#000000">__name__</font> <font color="#0000C0">==</font> <font color="#004080">"__main__"</font><font color="#0000C0">:</font>
<font color="#000000">main</font><font color="#0000C0">(</font><font color="#0000C0">)</font><font color="#000000"></font><a class="viewcode-back" href="javascript:history.go(-1)">[Return]</a></font></pre>
</div>
</div>
</div>
</div>
<div class="sphinxsidebar">
<div class="sphinxsidebarwrapper">
<a href= /shots/10104 ><img src="/_static/logos/golem.svg" alt="" width=80 /></a>
<h4>Navigation</h4>
<ul class="this-page-menu">
<li>
<a class="reference internal" href="javascript:history.go(-1)"><b>Return</b></a>
</li>
</ul>
<div id="searchbox" style="display: none">
<h3>Go to shot</h3>
<form style='display:inline;' method="post" action="" onsubmit="return getURL(this.shot.value)">
<input size="5" type="text" name="shot" value="10104" />
<input size="5" type="submit" name="Go" value="Go" />
</form>
</div>
<script type="text/javascript">$('#searchbox').show(0);</script>
</div>
</div>
<div class="clearer"></div>
</div>
<div class="related">
<span class="base_template"><a href="/shots/10104/base.php.template" title="main template">[Main template]</a> Disk usage: 288KB </span>
<h3>Navigation</h3>
<ul>
<li class="right" ><a href= /shots/0/DAS/1011NIbasic.ON/ title="current shot" >current</a> </li>
<li class="right" ><a href= /shots/10105/DAS/1011NIbasic.ON/ title="next shot" >next</a> |</li>
<li class="right" ><a href= /shots/10103/DAS/1011NIbasic.ON/ title="previous shot" >previous</a> | </li>
</ul>
</div>
<div class="footer">
© Copyright 2012, <a href="mailto:svoboda@fjfi.cvut.cz">GOLEM Team</a>
</div>
</body>
</html>
