from __future__ import division
def print_sweep(list_of_images):

  from dxtbx.imageset import ImageSetFactory
  sweeps = ImageSetFactory.new(list_of_images)

  for s in sweeps:
    print s.get_detector()
    print s.get_beam()
    print s.get_goniometer()
    print s.get_scan()

    # compute the beam centre... in mm... w.r.t. fast, slow axis

    print 'Derived quantities:'

    from scitbx import matrix

    d = s.get_detector()[0]
    b = s.get_beam()

    o = matrix.col(d.get_origin())
    f = matrix.col(d.get_fast_axis())
    s = matrix.col(d.get_slow_axis())
    s0 = matrix.col(b.get_direction())

    n = f.cross(s)

    beam_offset = o - o.dot(s0) * s0
    print '    beam centre (mm, fast, slow): %.2f %.2f' % (- beam_offset.dot(f),
                                                            - beam_offset.dot(s))

if __name__ == '__main__':
  import sys

  if len(sys.argv) == 2:
    print_sweep(sys.argv[1])
  else:
    print_sweep(sys.argv[1:])