ó
­žéPc           @   s    d  d l  Td  d l Td  d l Z d  d l m Z d  d l m Z d  d l Td  d l j	 Z
 d e j d <d e j d <d a d	 a d
 e d a d d d „ Z d S(   iÿÿÿÿ(   t   *N(   t   norm(   t   fftconvolvet   outs   xtick.directions   ytick.directiongÃõ(\Âµ?gš™™™™™Ù?i   gH¯¼šò×z>c   $      C   s  d } t  t d t t d t | ƒ } t  t d d t | ƒ } t j | | ƒ \ }	 }
 t  d d t d ƒ } t | ƒ } t | ƒ } t j	 d } t
 d d d d d	 d
 ƒ } | d d d | d d d d ƒ } t d d. ƒ } | j d ƒ } d/ } x2 | D]* } | j t |	 |
 | d | d ƒ ƒ qþ Wt t | t t | d ƒ t g  g  d ƒ \ } t t | d | d f ƒ d t t t t t t g ƒ} t |	 |	 |	 d d ƒ} t | d d d d ƒ} | j | d t |  ƒ ƒ xt t |  ƒ ƒ D]û} t j ƒ  } | j d d |  | ƒ t |	 |
 | | d f | | d f ƒ } x+ t | | ƒ D] \ } } | | | 7} q\Wt | d0 d | d1 d ƒ } t |	 |
 | | d f | | d f | | d! d" t ƒ\ } }  t | | d# d$ ƒ| d | d | d | d }! | j | |  | | d f | |  | | d f ƒ |! d 9}! | j |! | d% | d% d& … | d% | d% … f ƒ | j d' t ƒ | j  ƒ  x! | j! D] }" | j! j" |" ƒ q¨Wx | j# D] }# |# j" ƒ  qÌWt |	 | d% | d% … | d% | d% … f |
 | d% | d% … | d% | d% … f |! | d% | d% … | d% | d% … f d d ƒ} | j d d d d ƒ } | j$ ƒ  | j% ƒ  | t |  ƒ d k rø| j& d( d) d' d d2 d+ d, ƒ| j& d- d) d' d d3 d+ d, ƒqøqøWWd  QXd  S(4   Nid   i   i    t   ffmpegt   titles
   Movie Testt   artistt
   Matplotlibt   comments   Movie support!t   fpsi
   t   metadatat   codect   mpeg4t   bitrateiÐ  t   figsizei   io   i   s   r--s   g--t   extentt   colorst   kt   inlinet   fontsizes   video/animation.wmvs#   Pol. Mag. field [mT]; time %1.2f msg     @@.gš™™™™™É?t   centeredt   modet   samei   iÿÿÿÿt   tights   icon.pngt   bbox_inchesi   t   dpii(   s   large_icon.png(   i   i   (    (   .i   (   .i    (   i   i   (   i   i   ('   t   linspacet   at   R_0t   npt   meshgridt   pit   cost   sint
   manimationt   writerst   dictt   figuret   add_subplott   appendt   Bloop_analytict   plott   imshowt   zerost   contourt   clabelt   savingt   lent   ranget   timet	   set_titlet   zipt   sqrtt   CurrentProfilet   TrueR   t   set_datat   autoscale_viewt	   autoscalet   collectionst   removet
   labelTextst   relimt
   grab_framet   savefig($   t   tvect   I_pt   pos_pt   namet   I_coilst	   pos_coilst   nt   rt   zt   Rt   Zt   phit   ct   st   FFMpegWriterR
   t   writert   figt   axt   B_coilt   coordin_coilt   plasma_edge_plott   mag_imgt   CSt   clt   it   tt   Bt   B_extt   I_extR   t   jt   aot   normconvt   collt   text(    (    sO   ../includes/diagnostics/Magnetic/0612PlasmaPosition_TO.ON/MagFieldProfileGen.pyt   PlotProfile   s^    #	!(<) ?54
8


(    (    (   t   numpyt   matplotlib.pyplotR1   t   numpy.linalgR   t   scipy.signalR   t   MagFieldCalct   matplotlib.animationt	   animationR"   t
   matplotlibt   rcParamsR   R   R   t   mu_0Rb   (    (    (    sO   ../includes/diagnostics/Magnetic/0612PlasmaPosition_TO.ON/MagFieldProfileGen.pyt   <module>   s   


