
Z	                 @   sO   d  d l  j Z d  d l Z g  d d  Z d d   Z d d g  d d  Z d S)	    Nc       
      C   s  t  j   d } t | |  d  r} | j   } | j d  \ } } | j   j   } g  g  } } t |  | s x| r | j t | j d  d j	 d d    | j t | j d  d j	 d d    | j   j   } qy Wn x | rt |  x | D] }	 | j |	  } q
W| j t | j d  d j	 d d    | j t | j d  d j	 d d    | j   j   } q WWd  QRX| | | | f S)N/r;r   ,.   )
osgetcwdopenreadlinesplitstripprintappendfloatreplace)
filenamecharscwdfilelinex_namey_namexychar r   H/Users/vegardjervell/Documents/2semester/Uorganisk/oppg4/plot_csv_txt.pyget_values_csv   s(    
	,,	
,,r   c       
      C   s  t  j   d } t | |  d  M} | j   } | j   \ } } | j   j   } g  g  } } | j   j   } | s x | r | j t | d j d d    | j t | d j d d    | j   j   } q~ Wn x | rkt |  x | D] }	 | j	 |	  } q W| j t | d j d d    | j t | d j d d    | j   } q WWd  QRX| | } } | | | | f S)Nr   r   r   r   r   r   )
r   r	   r
   r   r   r   r   r   r   r   )
r   r   r   r   r   r   r   r   r   r   r   r   r   get_values_txt   s*    	##	
##r   Zmy_plotc             C   s  |  d  k r d Sd |  k r= t  |  d | \ } } } } n1 d |  k rj t |  d | \ } } } } n d St j   t |  d t |  d } } t |  d t |  d }	 }
 t j | |  t j | |  t j |	 |
  t j	 |  t j
 |  t j |  d  S)NzXplot(data = name of data file, name = name of plot, chars = list of characters to strip)z.txtr   z.csvzfile must be .csv or .txtr   )r   r   pltZfigureminmaxplotZxlimZylimZxlabelZylabelZsavefig)datanamer   r   r   r   r   Zmin_xZmax_xZmin_yZmax_yr   r   r   r#   6   s     !!
!!r#   )Zmatplotlib.pyplotZpyplotr    r   r   r   r#   r   r   r   r   <module>   s   