from __future__ import absolute_import, division, print_function
import sys

class ImageWorker(object):
  def __init__(self,relpath):
    from iotbx.detectors import mar
    m = mar.MARImage(relpath)
    m.read()
    self.fi = m.get_flex_image()
    self.fi.setWindow(1.0) #fraction of original image dimension written to graphics display
    self.fi.adjust()

  def output(self,outputfile):
    import PIL.Image # dependency on Python Image Library
    im = PIL.Image.new("RGB",(self.fi.ex_size1(), self.fi.ex_size2())) # 'L' is grayscale
    r,g,b = im.split()
    r.putdata(self.fi.channel(0))
    g.putdata(self.fi.channel(1))
    b.putdata(self.fi.channel(2))
    imageout = PIL.Image.merge("RGB",(r,g,b))
    imageout.save(outputfile,"PNG")

if __name__=='__main__':
  infile = sys.argv[1]
  outfile = "/net/racer/scratch1/ttleese/test2.png"
  I = ImageWorker(infile)
  print("Finished read")
  I.output(outfile)
  print("Finished write")