#/*##########################################################################
# Copyright (C) 2004-2014 European Synchrotron Radiation Facility
#
# This file is part of the PyMca X-ray Fluorescence Toolkit developed at
# the ESRF by the Software group.
#
# This toolkit is free software; you can redistribute it and/or modify it
# under the terms of the GNU General Public License as published by the Free
# Software Foundation; either version 2 of the License, or (at your option)
# any later version.
#
# PyMca is distributed in the hope that it will be useful, but WITHOUT ANY
# WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
# FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
# details.
#
# You should have received a copy of the GNU General Public License along with
# PyMca; if not, write to the Free Software Foundation, Inc.,
# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
#
# PyMca follows the dual licensing model of Riverbank's PyQt and cannot be
# used as a free plugin for a non-free program.
#
# Please contact the ESRF industrial unit (industry@esrf.fr) if this license
# is a problem for you.
#############################################################################*/
import os
import sys
from PyMca.PyMcaIO import MEDFile
from PyMca.PyMcaIO import SpecFileAbstractClass

class APSMEDFileParser(object):
    def __init__(self, filename, sum_all=False):
        if not os.path.exists(filename):
            raise IOError("File %s does not exists"  % filename)

        self._medFileObject = MEDFile.MEDFile(filename)
        #read the data
        #potentially each detector could have a different number of channels
        #should I support that?
        header = []
        header.append('#S 1  %s Unknown command' % os.path.basename(filename))
        header.append('#D %s' % self._medFileObject.mcas[0].start_time)

        if sum_all:
            realtime = 0
            livetime = 0
            cal_offset = 0
            cal_slope = 0
            cal_quad = 0
            n = float(len(self._medFileObject.mcas))
            for mca in self._medFileObject.mcas:
                realtime += (mca.realtime / n)
                livetime += (mca.livetime / n)
                cal_offset += (cal_offset / n)
                cal_slope += (cal_slope / n)
                cal_quad += (cal_quad / n)
            data = [self._medFileObject.get_data(sum_all=sum_all)]
            # this makes no sense in any case
            header.append('#@CTIME %f %f %f' % (realtime,
                                                realtime,
                                                livetime))
            header.append('#@CALIB %f %f %f' % (cal_offset,
                                                cal_slope,
                                                cal_quad))
        else:
            data = []
            for mca in self._medFileObject.mcas:
                header.append('#@CTIME %f %f %f' % (mca.realtime,
                                                    mca.realtime,
                                                    mca.livetime))
                header.append('#@CALIB %f %f %f' % (mca.cal_offset,
                                                    mca.cal_slope,
                                                    mca.cal_quad))
                data.append(mca.data)

        self.motorNames = []
        motorValues = []
        for item in self._medFileObject.env:
            name, value = item.split("=")
            self.motorNames.append(name)
            motorValues.append(value)
            header.append('#' + item)

        #create an abstract scan object
        self._scan = [APSMEDScan(data, scanheader=header,
                                 motor_values=motorValues)]

        #the methods below are called by PyMca on any SPEC file

    def __getitem__(self, item):
        return self._scan[item]

    def scanno(self):
        """
        Gives back the number of scans in the file
        """
        return len(self._scan)

    def list(self):
        return "1:1"

    def select(self, key):
        """
        key is of the from s.o
        scan number, scan order
        """
        n = key.split(".")
        return self.__getitem__(int(n[0]) - 1)

    def allmotors(self):
        return self.motorNames


class APSMEDScan(SpecFileAbstractClass.SpecFileAbstractScan):
    def __init__(self, data, scantype='MCA',
                 identification="1.1", scanheader=None, labels=None,
                 motor_values=None):
        SpecFileAbstractClass.SpecFileAbstractScan.__init__(self,
                    data, scantype=scantype, identification=identification,
                    scanheader=scanheader, labels=labels)
        if motor_values is None:
            motor_values = []
        self.motorValues = motor_values

    def allmotorpos(self):
        return self.motorValues


def isAPSMEDFile(filename):
    #Obviously I should put a better test than this one
    if not filename.upper().endswith(".XRF"):
        return False
    return True


def test(filename):
    if isAPSMEDFile(filename):
        sf = APSMEDFileParser(filename)
    else:
        print("Not an APS Multi-element detector file")
    print(sf[0].header('S'))
    print(sf[0].header('D'))
    print(sf[0].header('ID13ds'))
    print(sf[0].alllabels())
    print(dir(sf[0]))
    print("number of mcas = %s " % sf[0].nbmca())
    try:
        import pylab
        for i in range(sf[0].nbmca()):
            pylab.plot(sf[0].mca(i + 1))
        pylab.show()
    except ImportError:
        pass


if __name__ == "__main__":
    test(sys.argv[1])