B
    å9‡^ü  ã               @   sd   d dl Zd dlmZ d dlmZ d dlZdd„ Z	dd„ Z
dd„ Zejdfd	d
„Zejdfdd„ZdS )é    Nc          	   C   sr   t jt t jt j t jd | t j|  d   t jd t jt jt j d     t jt j d|  dt j    ¡ S )Né   é   )	ÚcÚtrÚnpÚexpZA0ÚRÚTeqZTrZQd)ÚT© r   úG/Users/vegardjervell/Documents/6_semester/matmod/matmod11V4/c_curves.pyÚt_star_func   s    
"
r   c             C   s8   t |ƒ}|t t d|  ¡t dtj ¡ dtj ¡ S )Nr   )r   r   ÚpowerÚlogr   ÚXcÚn)ÚXr
   Út_starr   r   r   Út_func_isotherm   s    r   c             C   sD   t |ƒ}t t d|  ¡t dtj ¡ dtj ¡}tj||ddS )Nr   r   )Úaxes)r   r   r   r   r   r   r   Ú	tensordot)r   r
   r   Úa1r   r   r   Út_func_iso_vectorized"   s    ,r   iÖ  c          
   C   sH  t jdddd}t  | |d¡d }tjddgd t d	¡}d
}||d ƒ}xt|D ]l}|d
krœ||d ƒ}tjt||ƒ|d |dt	t
|dƒƒ dd d}qRtjt||ƒ|d |d |d7 }qRW |dkr| dk rtjdtt|d |ƒƒgtjd tjd gdddd t d¡ t ¡  t d¡ t d¡ t d¡ t ¡  d S )Ngš™™™™™©?r   )Ústepéd   i  é   é   )ÚfigsizeÚplasma_ré   r   g333333Ã?zX = r   ú:)ÚcolorÚlabelÚ	linestyle)r!   i¸  i¤  éÿÿÿÿÚgreenz--z$T_{eq}$)r!   r#   r"   r   zTime [s]u   Temperature [â„ƒ]z[T-t curves for different fractions transformed, distance between lines is $\Delta X = 0.05$)r   ÚarangeÚlinspaceÚpltÚfigureÚcmÚget_cmapÚplotr   ÚstrÚroundÚmaxr   r	   ÚxscaleÚlegendÚxlabelÚylabelÚtitleÚshow)ÚT_minÚT_maxÚX_valsÚT_lineÚcmapÚlabel_counterr!   r   r   r   r   Úplot_C_curves)   s.    

*



r<   c       
   
   C   sP  t jdddd}t  | |d¡d }tjddgd t||ƒ}t d	¡}d
}||d ƒ}xrt||ƒD ]d\}}	|d
krª||	d ƒ}tj	||d |dt
t|	dƒƒ dd d}qbtj	||d |d |d7 }qbW |dkr| dk rtj	dtt|d |ƒƒgtjd tjd gdddd t d¡ t ¡  t d¡ t d¡ t d¡ t ¡  d S )Ngš™™™™™©?r   )r   r   i  r   r   )r   r   r   r   g333333Ã?zX = r   r    )r!   r"   r#   )r!   i¸  i¤  r$   r%   z--z$T_{eq}$)r!   r#   r"   r   zTime [s]u   Temperature [â„ƒ]z[T-t curves for different fractions transformed, distance between lines is $\Delta X = 0.05$)r   r&   r'   r(   r)   r   r*   r+   Úzipr,   r-   r.   r/   r   r   r	   r0   r1   r2   r3   r4   r5   )
r6   r7   r8   r9   Úcurvesr:   r;   r!   Zcurver   r   r   r   Úplot_C_curves_V2G   s0    

*



r?   )Únumpyr   Úmatplotlib.pyplotÚpyplotr(   Zmatplotlib.cmr*   Ú	constantsr   r   r   r   ZT_min_c_curver<   r?   r   r   r   r   Ú<module>   s   