import os, numpy as np
from pymicro.xray.detectors import Mar165
from pymicro.file.file_utils import HST_read
from matplotlib import pyplot as plt, cm

if __name__ == '__main__':
    mar = Mar165()  # create a mar object
    mar.ucen = 981  # position of the direct beam on the x axis
    mar.vcen = 1060  # position of the direct beam on the y axis
    mar.calib = 495. / 15.  # identified on (111) ring of CeO2 powder
    mar.correction = 'bg'
    mar.compute_TwoTh_Psi_arrays()

    # load background for correction
    mar.bg = HST_read('../data/c1_exptime_bgair_5.raw', data_type=np.uint16, dims=(2048, 2048, 1))[:, :, 0].astype(
        np.float32)
    # load image
    mar.load_image('../data/c1_exptime_air_5.raw')

    image_name = os.path.splitext(__file__)[0] + '.png'
    print 'writting %s' % image_name
    plt.imsave(image_name, mar.corr_data, vmin=0, vmax=2000)

    from matplotlib import image

    image.thumbnail(image_name, 'thumb_' + image_name, scale=200. / 2048)
