"""
    TestObsConversions.py: CCP4 GUI Project
    Copyright (C) 2015 Newcastle University
    
    This library is free software: you can redistribute it and/or
    modify it under the terms of the GNU Lesser General Public License
    version 3, modified in accordance with the provisions of the
    license to address the requirements of UK law.
    
    You should have received a copy of the modified GNU Lesser General
    Public License along with this library.  If not, copies may be
    downloaded from http://www.ccp4.ac.uk/ccp4license.php
    
    This program is distributed in the hope that it will be useful,S
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU Lesser General Public License for more details.
    """

from CCP4PluginScript import CPluginScript
import CCP4ErrorHandling
from CCP4ErrorHandling import *
import CCP4Modules
from lxml import etree

class TestObsConversions(CPluginScript):
    TASKTITLE = 'TestObsConversions'
    TASKNAME = 'TestObsConversions'
    TASKVERSION= 0.0
    RUNEXTERNALPROCESS= False
    PERFORMANCECLASS = 'CTestObsConversionsPerformance'
    ERROR_CODES = { 201 : {'description' : 'Failed to turn the provided data object into the requested input representation' },}
    ERROR_CODES = { 202 : {'description' : 'Failed to turn the intermediate data object into the requested output representation' },}

    def __init__(self,parent=None,name=None,workDirectory=None,dummy=False,taskName=None,**kw):
        super(TestObsConversions,self).__init__(parent, name, workDirectory, dummy, taskName, **kw)
        self.xmlroot = etree.Element('TestObsConversions')

    def startProcess(self, command):
        with open(self.makeFileName('PROGRAMXML'),'w') as programXML:
            programXML.write(etree.tostring(self.xmlroot, pretty_print=True))
        return CPluginScript.SUCCEEDED

    def processInputFiles(self):

        if int(self.container.controlParameters.INPUT_REPRESENTATION) == 1:
            inputType = 'F_SIGF_AS_IPAIR'
        elif int(self.container.controlParameters.INPUT_REPRESENTATION) == 2:
            inputType = 'F_SIGF_AS_FPAIR'
        elif int(self.container.controlParameters.INPUT_REPRESENTATION) == 3:
            inputType = 'F_SIGF_AS_IMEAN'
        
        fileNode = etree.SubElement(self.xmlroot,'File')
        typeNode = etree.SubElement(fileNode,'Role')
        typeNode.text = 'Input'
        contentFlagNode = etree.SubElement(fileNode,'ContentFlag')
        contentFlagNode.text = str(getattr(self.container.inputData,inputType).contentFlag)
        columnsNode = etree.SubElement(fileNode,'Columns')
        fileToInterrogate = getattr(self.container.inputData,inputType)
        columnsInFile = fileToInterrogate.getFileContent().getListOfColumns()
        columnsNode.text = str([column.columnLabel.__str__() for column in columnsInFile])
        pathNode = etree.SubElement(fileNode,'Path')
        pathNode.text = str(getattr(self.container.inputData,inputType))
        
        self.hklin,error = self.makeHklin([[inputType, int(self.container.controlParameters.INPUT_REPRESENTATION)]])
        if error.maxSeverity()>CCP4ErrorHandling.SEVERITY_WARNING:
            self.appendErrorReport(201, inputType + " conversion to type number " + str(self.container.controlParameters.INPUT_REPRESENTATION))
            return CPluginScript.FAILED
        import os
        intermediateFilePath = os.path.join(self.getWorkDirectory(),'intermediate.mtz')
        os.rename(self.hklin, intermediateFilePath)

        self.container.inputData.F_SIGF_INTERMEDIATE.setFullPath(intermediateFilePath)
        self.container.inputData.F_SIGF_INTERMEDIATE.setContentFlag(reset=True)
        
        inputType = 'F_SIGF_INTERMEDIATE'
        fileNode = etree.SubElement(self.xmlroot,'File')
        typeNode = etree.SubElement(fileNode,'Role')
        typeNode.text = 'Intermediate'
        contentFlagNode = etree.SubElement(fileNode,'ContentFlag')
        contentFlagNode.text = str(getattr(self.container.inputData,inputType).contentFlag)
        columnsNode = etree.SubElement(fileNode,'Columns')
        fileToInterrogate = getattr(self.container.inputData,inputType)
        columnsInFile = fileToInterrogate.getFileContent().getListOfColumns()
        columnsNode.text = str([column.columnLabel.__str__() for column in columnsInFile])
        pathNode = etree.SubElement(fileNode,'Path')
        pathNode.text = str(getattr(self.container.inputData,inputType))

        self.hklin,columns,error = self.makeHklin0([['F_SIGF_INTERMEDIATE', int(self.container.controlParameters.OUTPUT_REPRESENTATION)]])
        if error.maxSeverity()>CCP4ErrorHandling.SEVERITY_WARNING:
            self.appendErrorReport(202, 'F_SIGF_INTERMEDIATE' + " conversion to type number " + str(self.container.controlParameters.OUTPUT_REPRESENTATION))
            return CPluginScript.FAILED

        from CCP4XtalData import CObsDataFile
        outputFiles = ['F_SIGF_FINAL']
        outputColumns = [columns]
        error = self.splitHklout(outputFiles,outputColumns,infile=self.hklin)
        if error.maxSeverity()>CCP4ErrorHandling.SEVERITY_WARNING:
            return CPluginScript.FAILED
        self.container.outputData.F_SIGF_FINAL.setContentFlag(reset=True)
        
        inputType = 'F_SIGF_FINAL'
        fileNode = etree.SubElement(self.xmlroot,'File')
        typeNode = etree.SubElement(fileNode,'Role')
        typeNode.text = 'Final'
        contentFlagNode = etree.SubElement(fileNode,'ContentFlag')
        contentFlagNode.text = str(getattr(self.container.outputData,inputType).contentFlag)
        columnsNode = etree.SubElement(fileNode,'Columns')
        fileToInterrogate = getattr(self.container.outputData,inputType)
        columnsInFile = fileToInterrogate.getFileContent().getListOfColumns()
        columnsNode.text = str([column.columnLabel.__str__() for column in columnsInFile])
        self.container.outputData.PERFORMANCEINDICATOR.columnLabelsString = columnsNode.text
        pathNode = etree.SubElement(fileNode,'Path')
        pathNode.text = str(getattr(self.container.outputData,inputType))

        return CPluginScript.SUCCEEDED