B
    ?e_                 @   s   d dl Zd dlZd dlmZ d dlmZ dddZdddgd d d gd	fd
dZdd Zedkrd dl	m
Z
 d dlZdZeed dS )    N)butterworth_filter)clean_negbgrQ?
   c             C   s*   || | |  }t j||d || }|S )N)out)nparcsinh)xscalenonlinearitybaseflx r    /gal/db1/Data/HiPS/lupton_rgb.pylup_func   s    r      Z
rgb_ukidssc             C   s  dd l }| | }tj|jd |jd ftjd}tj||dd d||dk < t|dddd	}d
|d  | d |d   | | }t|d|jd  d}|d8 }d||dk < |d7 }d||dk< d
|d  | d |d   | | }	t|	d|jd  d}	|	d8 }	d|	|	dk < |	d7 }	d|	|	dk< d
|d  | d |d   | | }
t|
d|jd  d}
|
d8 }
d|
|
dk < |
d7 }
d|
|
dk< t| jtj	}|tj	|d< |	tj	|d< |
tj	|d< t
j|d tj|dddd d S )Nr   r   )dtype)r   axisg|=gMb?g?)r
   r   r   g     o@).r   g
ףp=
?)Zomega_cg      .@   ).r      ).r   z.jpg)r   b   )quality)osr   zerosshapefloatsumr   r   astypeuint8imageioimwriteflip)Z
rgb_imagesscalesbase0out_namer   cILrgbZrgb_outr   r   r   make_rgb_lup   s:     $$$r,   c             C   sp  ddl m} dd l}||j| d}||j| d}||j| d}|d jj}t	|d |d df}t
|d j|d jd |d jd }	t
|d j|d jd |d jd }
t
|d j|d jd |d jd }|d j|	 |d< |d j|
 |d	< |d j| |d
< |d jj|d dddd t|dddgdddg|d |  |  |  d S )Nr   )fitszK_COADD.fitszH_COADD.fitszY_COADD.fitsr      ).r   ).r   ).r   z.hhh
FT)seppadding	overwriteg(\?g)\(?gffffff?g        )r#   r$   r%   )
astropy.ior-   r   openpathjoindatar   r   r   r   headertofiler,   close)	swarp_dirr%   r-   r   hrZhgZhbsZcube_rgbZbgr_rZbgr_gZbgr_br   r   r   YHKfits2RGBjpeg,   s$    &&&r>   __main__)r-   z)images/UKIDSS_J110731.23+134712.88/swarp/zUKIDSS_J110731.23+134712.88)r   r   r   )numpyr   r    r   r   r   r,   r>   __name__r3   r-   r   r;   r   r   r   r   <module>   s   
 
