B
    MW^^é  ã               @   s@   d dl Zd dlmZ d dlZdd„ Zdd„ Zdd„ Z	d	d
„ Z
dS )é    N)Úspecialc             C   s   t jt t j t j|   ¡ S )N)ÚcZD0ÚnpÚexpÚQÚR)ÚT© r	   úA/Users/vegardjervell/Documents/6_semester/matmod8_2/analytical.pyÚ<lambda>   s    r   c             C   s   t  || |¡}|S )N)r   Úarange)Zt_fÚdtZt_listr	   r	   r
   Út_list_funk   s    r   c       
      C   sè   d}xÞt | |ƒD ]Î}| dtj |  dt || ¡  }| d| tj tj  dt || ¡  }| d| tj  dt || ¡  }| d| d tj  dt || ¡  }	|dt |¡t |¡ t |¡ t |	¡  7 }qW |S )Nr   é   é   g      à?)Úranger   ÚLr   ÚsqrtÚssÚerf)
Úx_distÚDÚtÚkÚSZk_iZz_0Zz_1Zz_2Zz_3r	   r	   r
   ÚS_func
   s    $*$(4r   c             C   s0   t |ƒ}tjd |  }tjt|||tjƒ }|S )Nr   )ÚD_funcr   r   ÚC0r   Úanalytical_k)Úxr   r   r   r   ÚCr	   r	   r
   Úcons   s    r!   )Únumpyr   Zscipyr   r   Ú	constantsr   r   r   r   r!   r	   r	   r	   r
   Ú<module>   s   