# $python --pylab
# %load cobr.py 


def gama(b):
    # interval 0 .. 1
    b1 = (b - 1000)/3000

    # orezani 
    b2 = b1
    numpy.clip(b1,0.0,1.0,b2)

    # sRGB transformace (gamma funkce)
    b3 = 1.055*b2**(1/2.4) - 0.055 

    # final clip
    b4 = b3
    numpy.clip(b3,0.0,1.0,b4)
    return b4


import pyfits

pfR = pyfits.open('pfR.fits')
r = pfR[0].data
pfR.close()

pfV = pyfits.open('pfV.fits')
v = pfV[0].data
pfV.close()

pfB = pyfits.open('pfB.fits')
b = pfB[0].data
pfB.close()

import matplotlib.pyplot as plt
import numpy

plt.imshow(b)
plt.set_cmap('gray')

pf = numpy.zeros((1020,1530,3))
pf[:,:,0] = r
pf[:,:,1] = v
pf[:,:,2] = b

pf4 = gama(pf)
plt.imshow(pf4)