"""
     cphasematch.py: CCP4 GUI Project

     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,
     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

class cphasematch(CPluginScript):

    TASKMODULE = 'expt_data_utility'
    TASKTITLE = 'Match and analyse phases to reference set'
    TASKNAME = 'cphasematch'
    TASKCOMMAND = 'cphasematch'
    DESCRIPTION = '''Compare phases from different sources (with option change of origin/hand)'''
    TASKVERSION= 0.0

    def processInputFiles ( self ):
        import CCP4XtalData
        inp = self.container.inputData
        colgrps = [ ['F_SIGF', CCP4XtalData.CObsDataFile.CONTENT_FLAG_FMEAN],
                    'ABCD1', 'ABCD2' ] 
        self.hklin, columns, error = self.makeHklin0 ( colgrps )


    def makeCommandAndScript(self):
        inp = self.container.inputData
        out = self.container.outputData

        import os
        self.hklout = os.path.join(self.workDirectory,"hklout.mtz")

        self.appendCommandLine([ '-stdin' ])

        self.appendCommandScript( '-mtzin ' + self.hklin )
        self.appendCommandScript( '-mtzout ' + self.hklout )
        self.appendCommandScript ([ '-colin-fo F_SIGF_F,F_SIGF_SIGF' ])
        self.appendCommandScript ([ '-colin-hl-1 ABCD1_HLA,ABCD1_HLB,ABCD1_HLC,ABCD1_HLD' ])
        self.appendCommandScript ([ '-colin-hl-2 ABCD2_HLA,ABCD2_HLB,ABCD2_HLC,ABCD2_HLD' ])
        self.appendCommandScript( '-colout i2' )

        return CPluginScript.SUCCEEDED


    def processOutputFiles(self):
        from lxml import etree

        error = self.splitHklout( [ 'ABCDOUT' ], [ 'i2.ABCD.A,i2.ABCD.B,i2.ABCD.C,i2.ABCD.D' ] )
        self.container.outputData.ABCDOUT.annotation = 'shifted ABCD'

        lines = open(self.makeFileName('LOG')).readlines()
        for i in range(len(lines)):
          if "Overall" in lines[i]:
            dphi = lines[i+2].split()[3]
            wdphi1 = lines[i+2].split()[4]
            wdphi2 = lines[i+2].split()[5]
            fcorr = lines[i+2].split()[6]
            ecorr = lines[i+2].split()[7]
            self.container.outputData.PERFORMANCE.phaseError = float(dphi)
            self.container.outputData.PERFORMANCE.weightedPhaseError = float(wdphi1)
            self.container.outputData.PERFORMANCE.reflectionCorrelation = float(ecorr)

            self.xmlnode = etree.Element('cphasematch')
            etree.SubElement(self.xmlnode,'phaseError').text = dphi
            etree.SubElement(self.xmlnode,'weightedPhaseError1').text = wdphi1
            etree.SubElement(self.xmlnode,'weightedPhaseError2').text = wdphi2
            etree.SubElement(self.xmlnode,'reflectionFCorrelation').text = fcorr
            etree.SubElement(self.xmlnode,'reflectionECorrelation').text = ecorr

            smartieNode = etree.SubElement(self.xmlnode,'SmartieGraphs')
            self.scrapeSmartieGraphs(smartieNode)

            etree.ElementTree(self.xmlnode).write(self.makeFileName('PROGRAMXML'))

            return CPluginScript.SUCCEEDED

        return CPluginScript.FAILED


    def scrapeSmartieGraphs(self, smartieNode):
        import sys, os
        import CCP4Utils
        import MGQTmatplotlib        
        from lxml import etree
        smartiePath = os.path.join(CCP4Utils.getCCP4I2Dir(),'smartie')
        sys.path.append(smartiePath)
        import smartie

        logfile = smartie.parselog(self.makeFileName('LOG'))
        for smartieTable in logfile.tables():
            if smartieTable.ngraphs() > 0:
                tableetree = MGQTmatplotlib.CCP4LogToEtree(smartieTable.rawtable())
                smartieNode.append(tableetree)
        return