B
    bU^                 @   s6   d dl mZ d dlZd dlmZ d dlZd	ddZdS )
    )RN皙?   2   varc          	      s  dkr8t ddd}t  fdd|D }d}n dkrjt ddd}fdd|D }d}nbdkrt d	d
d} fdd|D }d}n0dkrt ddd} fdd|D }d}t |}dddg}tt|d t|d  }t|d d|  t|d d|  }	fdd|D }
tj||d |
dd|d d xBt	ddD ]4}tj|||d  || ddd|  || d q`W t
|	 tt|t| tj|dd d S )Nr   g?   r   c                s   g | ]}t | d qS ))HRCRT_isot_iso)r   ).0r   )r	   r
   r    @/Users/vegardjervell/Documents/6_semester/matmod/matmod7/plot.py
<listcomp>   s    zplot.<locals>.<listcomp>zupper rightc                s   g | ]}t  |d qS ))r   r	   r
   r   )r   )r   r	   )r   r
   r   r   r   r      s    i  ix  d   c                s   g | ]}t  |d qS ))r   r	   r
   r   )r   )r   r
   )r	   r   r   r   r   r      s    z
upper left   c                s   g | ]}t  |d qS ))r   r	   r
   r   )r   )r   r   )r	   r   r
   r   r   r      s    zR after heatingzR after isothermzR after coolingr   c                s   g | ]} qS r   r   )r   i)baser   r   r   #   s    redg?)coloralphalabel   g333333?   )locZfontsize)nplinspacearray	transposeabsmaxminpltZfill_betweenrangeZylimZxlimlegend)r	   r
   r   r   Zvar_axisZR_matrZ
legend_loclabelsZdelta_Rtopbaseliner   r   )r	   r   r
   r   r   r   plot   s8    

4r(   )r   r   r   r   )	Zmatter   numpyr   matplotlib.pyplotpyplotr"   timer(   r   r   r   r   <module>   s   