source: datared/gen_0.py @ 9

#!/bin/env python
# coding: utf-8

VERSION='1.0'

import pyfits
from optparse import OptionParser

import numpy


parser = OptionParser()

(options, args) = parser.parse_args ()

if (len(args) !=2) :
    print "Usage : pre.py input.fits output.fits"

# Ouverture de l'image fits en argument
hdulist = pyfits.open(args[0])
scidata = hdulist[0].data


scidata = numpy.zeros([2048,2048])

R2 = 910**2 # pixels

# Tracé du disque
for i in range(2048):
    for j in range(2048):
        if ( ((1024-i)**2 + (1024-j)**2) < R2):
            scidata[i,j] = 7000

scidata_flou = numpy.zeros([2048,2048])

# Floutage du bord sur ~ 5 pixels
#fen=2
#for i in range(fen,2048-fen):
#    print i
#    for j in range(fen,2048-fen):
#        scidata_flou = sum(sum(scidata[i-fen:i+fen,j-fen:j+fen]))/\
#                       100



prihdr = hdulist[0].header
prihdr.add_history('Generation de la mire')
hdulist[0].data = scidata
hdulist.writeto(args[1],clobber=True)




hdulist.close()