
{Qc           @   s   d  Z  d GHd d l Z d e j d <e j d d d e j d	 d
 e d d l Td d l Td d l Td d l m	 Z	 d d l
 Z
 d d l m Z d d l Z d GHd   Z d   Z d   Z e d k r e   n  d S(   s.    CREATED: 7/2012
    AUTHOR: MICHAL ODSTRČIL
s   importing modules iNt   Aggt   backendt   fontt   sizet   10t   textt   usetex(   t   *(   t   Shot(   t
   mquantiless   importing modules done c         C   s  t    } | j } t d d^ d |   } | d } | rN | d } | d } n d } d	 } t j   } d } d
 }	 | j d  r | d \ }
 } | |	 9} t | |
 | k |
 | k  @d  } n  d } xa d d d g D]P } y@ | | \ } } t | t | | | k | | k  @d   } Wq q Xq Wd t | d | d  g } d t |  } t d d d d d d  g t d d d d d d  g d t
 | rt d d d d d d t
 n t d  d! d" d d d t
 t d d# d$ d | t d d% d$ d | g t d d& d$ d | g | j d  r\| d' d( k r\t d d) d$ d | d |	 d t
 n d  g | r| j d*  r| d+ d, k r| d- d. k  rt d* d/ d0 d d1 n d  g g } d2 GHt | | d3 |  d4 GHd5 d6 l m } m } | d7 |    GHt d d d  t d d d d t
 t d  d8 d" d d d t
 t d d# d$  t d d9 d$  g g } t | | d: |  t d d d d; d d  g t d d d d; d d  g d t
 t d d d< d; d d  g d d d t
 t d= d! d< d; d d  g d d d t
 t d  d8 d< d; d d  g d d d t
 g t d d# d$ d; d d  g t d d9 d$ d; d d  g g g } t | | d> d? |  t d d d d; d d  g t d d d d; d d  g d t
 | rt d  d8 d@ d; d d  g d d d t
 t d d d d; d d  g d d d t
 g n' t d  d8 d" d; d d  g d d d t
 t dA d# dB d; d d  g t dC d9 dB d; d d  g g g } t | | dD dE |  t dF d dB d; d d  g t dG dH dI d; d d  g t dJ dK dL d; d d  g t dM dN dO d; d d  g g t dA d# dB d; d d  g t dC d9 dB d; d d  g g g } t | | dP dQ |  t d dR dR dS dR | rt d  dR dR dS dR g t d dR dR dS dR g n t d  dR dR dS dR g g } t | dR dT d_ dW  t d dR d d; d d  g d d d  g dS dR | rt d  dR d@ d; d d  g d d d  g dS dR d d d t
 t d dR d d; d d  g d d d  g dS dR d d d t
 g n9 t d  dR d" d; d d  g d d d  g dS dR d d d t
 g } t | dR dX d` dZ d[  d\ t j   | GHt d] d  d  S(a   Nt   figsizei	   i   t	   file_typet   plasmat   plasma_startt
   plasma_endi    g{Gz?g{Gzt?t
   hxr_smoothgGz?g333333?t
   photodiodet   photodiode_alphat   photodiode_otherg333333?s   Golem shot No:t   loop_voltages   Loop voltages   U [V]t   ylimt   toroidal_fields   Toroidal mag. fields	   B$_t$ [T]t	   reductiont   plasma_currents   Plasma currents   I$_{p}$ [kA]t   data_rescalegMbP?t   rogowski_currents   Chamber currents   I$_{ch}$ [kA]t   Visibles   Intensity [a.u.]s
   H$_\alpha$t   Othert   hxr_meang{Gz?t   HXRt   electron_densityt   electron_density_meang 7yACs   electron_density:reliabilityg      ?s   Electron density ($n_e$)s    n$_{e}$ [10$^{19}\cdot$m$^{-3}$]gҶOɃ;s   start plottingt	   graphpress   end plottingi(   t   gmtimet   strftimes   %Y-%m-%d %H:%M:%Ss   Total currents	   H$_\alphat
   graphprintt   xlims   I [kA]t   chamber_currents    - final data t   graphpresfulls   I$_p$+I$_{ch}$ [kA]t   photos   [DAS V]t   haphotos    - integrated datat
   graphpresit   uloopt   btors   Derivative  of mag. fields   dB$_t$/dt [DAS V]t   irogs   Raw Rogowski signals   dI$_{p+ch}$/dt [DAS V]t   PlasmaDetects   Derivative of I$_{p}$s   dI$_{p}$/dt [a.u.]s    - raw datat
   graphpresbt    t   xlabelt   iconi   i   i(   t   graphicg      ?id   t
   horizontals	    Time: %gt   status(   i	   i   (   i   i   (   i	   g      ?(   R   t   shot_numt   dictt   timet   existR	   t   maxt   strt   get_datat   Nonet   Truet	   multiplotR!   R"   t   paralel_multiplott	   saveconst(   R   t   St   shott   plot_paramsR   t   startt   endt   t0t   max_HXRt	   HXR_constt   tvec_HXRR   t	   max_photot   sigt   tvecR   t   ph_ranget
   shot_titlet   dataR!   R"   (    (    s8   ../includes/analysis/Basics/0411ShotHomepage.ONN/main.pyt	   plot_data   s    		


&0!$FJY!''*$!Z'$$:-c    3   	   B   s  d d l  Te j   }  e d  } e   e   e   \ } } } } e   \ } } } }	 e |	  \ }
 } } } e | |
 |  \ } } e	 | |  \ } } } } | r e | |
 | | |  \ } } n  x e
 d d d g d d d	 g  D]e \ } } y, e | | | |  } e | | | |  Wq e k
 rR} d
 | d Ge |  GHq Xq We | | |  } e | | |  } e | | |  } e | | |  } e | |  } e | | | | | | | |  } | r;e | |  } e | |  } e | | | | |  \ }  }! }" }# e |  }$ e | | |$ |  }% e | | | |  \ }& }' }( e |  }) e | | | |  } e d d d d d d g  d }* e |*  }+ e |+  }, xT e  |+  D]F }- e! | | | | |* |-  \ }. |, |- <d |* |- |. |, |- f GHqWe" |, e# |,  <e! | | | | |* e$ |,   n  e% |  }/ e& | | |  \ } }0 e' |  }1 e( | | |  }2 e) | |
 | | |  y e*   Wn" e k
 r} d Ge |  GHn Xd Ge j   |  GHd  S(   Ni(   R   t   Aktual_PfeifferMerkaVakuat   photovrbR(   R'   t
   PhotodVrbat   PhotodHalphat   Photods   Photodiode s    failed g?g?g333333?g?g333333?g?gMbP?s   win %g rate %g err %gs   BreakdownProba faileds   time of basic diagn generation(+   t   basic_diagnR7   t	   loadconstt   getDatet   save_configt   getBtoroidalt   getUloopt   getIrogowskit
   getIplasmaR-   t   zipt	   getPhotodt   getMeanPhotodt	   ExceptionR:   t	   getMeanBtt   getTotalCharget   getMeanUloopt   getMeanCurrentt   getOhmicHeatingPowert   Failurest   getQedget   getMeanElectronTemperaturet   getBreakDownVoltaget   getStateEqElectronDensityt%   getElectronConfinementTimeFirstApproxt   getElectronTemperaturet   getGreenwaldDensityt   getQedgeTimet   arrayt   lent   zerost   ranget   getBreakDownRatet   inft   isnant   argmint   getChamberResistancet   getOhmicHeatingPowerTimet   getMagneticFluxt   getTransformatorSaturationt   EnergyBalancet   BreakdownProba(3   t   tRQ   t   Btort   dBtort   BtMaxt   BtMeant   Uloopt   UloopMaxt	   UloopMeant
   ReversedCDt   Irogt   dIdt_rogMaxt   IrogMaxt   I_startt   Iplat   Icht   Plasmat   PlasmaStartt	   PlasmaEndt   PlasmaTimeLengtht   namet   fnameR   t   et   MeanBtt   TotalCharget	   MeanUloopt   MeanIplat   OhmicHeatingPowert   PlasmaStatust   Qedget   ElectronTemperaturet   Umaxt   Btimet   Bbreakt   Biplat   StateEqElectronDensityt"   ElectronConfinementTimeFirstApproxt   ElectronTempTimet   medElectronTempt   maxTt   GreenwaldDensityt   trangest   ntt   errt   it   ratet   ChamberResistancet   OhmicHeatingChambert   MagneticFluxt
   Saturation(    (    s8   ../includes/analysis/Basics/0411ShotHomepage.ONN/main.pyt   prepare_data   sf    
!.!$"&#c           C   s   t  j d d k r t   nd t  j d d k rJ t d  t d d  n7 t  j d d k r t d  t d  t d d  n  d  S(	   Ni   t   acquisitiont   plotst   pngR4   i    t   postanalysist   svgz(   t   syst   argvR   RP   R@   (    (    (    s8   ../includes/analysis/Basics/0411ShotHomepage.ONN/main.pyt   main   s    



t   __main__(   t   __doc__t
   matplotlibt   rcParamst   rcR=   t   numpyt   pygolem_lite.configt   pygolem_lite.modulest   pygolem_liteR   R   t   scipy.stats.mstatsR	   R7   RP   R   R   t   __name__(    (    (    s8   ../includes/analysis/Basics/0411ShotHomepage.ONN/main.pyt   <module>   s$   


		K	