from __future__ import absolute_import, division, print_function
# basic test reading an Raxis II image & producing a png file
# requires LABELIT and Python Image Library
# future: remove the LABELIT dependency

from six.moves import StringIO
import sys

if (__name__ == "__main__"):
  from labelit.command_line.overlay_distl import OverlayDriverClass

  filnm = sys.argv[1] # raxis-2 *.osc file
  out = sys.argv[2] # *.png file for graphical output

  ODC = OverlayDriverClass(filnm)
  ODC.pd={'xbeam':ODC.I.a.parameters['BEAM_CENTER_X'],
          'ybeam':ODC.I.a.parameters['BEAM_CENTER_Y'],
          'pixel_size':ODC.I.a.parameters['PIXEL_SIZE']}
  ODC.beam_overlay()
  ioout = StringIO()
  ODC.I.output(ioout)
  g = open(out,"wb")
  g.write(ioout.getvalue())
  g.close()