#!/usr/bin/env python

import reflex 
import sys
import pyfits

if __name__ == '__main__':

  parser = reflex.ReflexIOParser()

  #Define inputs
  parser.add_option("-i", "--in_sof", dest="in_sof")
  #Define outputs
  parser.add_output("-o", "--out_sof", dest="out_sof")

  inputs  = parser.get_inputs()
  outputs = parser.get_outputs()

  #Retrieve input
  in_sof = inputs.in_sof

  #Get the input files
  files = in_sof.files

  #Do the stuff
  for file in files:
    if file.category == 'MAPPED_SCI_MOS' :
      print "I am going to change file ",file.name
      raw_input("press a key:")
      hdulist = pyfits.open(file.name,mode='update')
      newdata = hdulist[0].data / 2
      hdulist[0].data = newdata
      hdulist.flush()
      print "Mapped science mos ",file.name, "has been divided by 2"

  #Set the output 
  outputs.out_sof = inputs.in_sof

  parser.write_outputs()

  raw_input("press a key:")

  sys.exit()