'''
    SASSIE: Copyright (C) 2011 Joseph E. Curtis, Ph.D. 

    This program 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 3 of the License, or
    (at your option) any later version.

    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 General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program.  If not, see <http://www.gnu.org/licenses/>.
'''
import os,sys,string,locale,glob
import input_filter

def parse_grammar(basis, error):

        import re

        p=re.compile('\s*(Rg|RG|X2)\s*(>|<|>=|<=)\s*(\'[\d\s\.]+\'|"[\d\s\.]+"|[\d\.]+)\s*')
        if p.search(basis):
            error.append('expression "'+basis+'" not understood!')
            error.append('you may need to change "Rg|RG|X2" to lower cases')
            return error

        p=re.compile('\s*(rg|x2)\s*(>|<|>=|<=)\s*(\'[\d\s\.]+\'|"[\d\s\.]+"|[\d\.]+)\s*')
        list_indices = []
        for m in p.finditer(basis):
            list_indices.append([m.start(),m.end()])

        #print 'list_indices: ',list_indices

        words = []
        words.append(basis[0:list_indices[0][0]].strip())
        for i in range(1,len(list_indices)):
            words.append(basis[list_indices[i-1][1] : list_indices[i][0]].strip())
        words.append(basis[list_indices[-1][1]:].strip())

        #print 'words: ',words

        if not len(words):
            error.append('grammer wrong in expression "'+basis+'"')
            return error

        if len(words[0]) and words[0][0]!='(' or len(words[-1]) and words[-1][-1]!=')':
            error.append('grammer wrong in expression "'+basis+'"')
            return error
        for word in words[1:-1]:
            if word[0]=='(' or word[-1]==')':
                    #print word
                    error.append('grammer wrong in expression "'+basis+'"')
                    return error
                

        r = re.compile('(\(\s*\)|\(\s*(and|or)|(and|or)\s*(and|or)|(and|or)\s*\))')
        for word in words:
            if r.search(word):
                error.append('grammer wrong in expression "'+basis+'"')
                return error

        complete_words=''.join(words)
        if complete_words.replace('and','').replace('or','').replace('(','').replace(')','').strip():
            error.append('expression "'+basis+'" not understood!')
            return error
        if complete_words.count('(') != complete_words.count(')'):
            error.append('parenthesis in expression "'+basis+'" does not match!')
            return error

def check_chi_square_filter(variables,**kwargs):

    runname            = variables['runname'][0]           
    saspath            = variables['saspath'][0]           
    sasintfile        = variables['sasintfile'][0]        
    io            = variables['io'][0]                
    basis_string    = variables['basis_string'][0]                
    x2highcut        = variables['x2highcut'][0]         
    x2highweightfile    = variables['x2highweightfile'][0]  
    x2lowcut        = variables['x2lowcut'][0]          
    x2lowweightfile        = variables['x2lowweightfile'][0]   
    rghighcut        = variables['rghighcut'][0]         
    rghighweightfile    = variables['rghighweightfile'][0]  
    rglowcut        = variables['rglowcut'][0]          
    rglowweightfile        = variables['rglowweightfile'][0] 
    sastype            = variables['sastype'][0]           
    number_of_weight_files = variables['number_of_weight_files'][0]
    plotflag = variables['plotflag'][0]
    reduced_x2 = variables['reduced_x2'][0]
    weight_file_names = variables['weight_file_names'][0]
    number_of_weight_files = variables['number_of_weight_files'][0]
    plotflag = variables['plotflag'][0]
    reduced_x2 = variables['reduced_x2'][0]
    weight_file_names = variables['weight_file_names'][0]

    error=[]
    error = input_filter.check_name(runname)
    if(error!=[]):
        return error
    
    if "no_file_check" not in kwargs:

        ev,rv,wv=input_filter.check_permissions(saspath)

        if(not ev or not rv):
            error.append('permission error in input file path '+saspath+'  [code = '+str(ev)+str(rv)+str(wv)+']')
            if(ev==False):
                error.append('path does not exist')
            elif(rv==False):
                error.append('read permission not allowed')
            return error

    if(x2highcut < 0.0 or x2lowcut < 0.0):
        error.append('x2 cutoffs need to be > 0, x2highcut = '+str(x2highcut)+' x2lowcut = '+str(x2lowcut))
        return error
    elif(x2highcut < x2lowcut):
        error.append('x2highcut < x2lowcut, x2highcut = '+str(x2highcut)+' x2lowcut = '+str(x2lowcut))
        return error
    elif(rghighcut < 0.0 or rglowcut < 0.0):
        error.append('Rg cutoffs need to be > 0, Rghighcut = '+str(rghighcut)+' Rglowcut = '+str(rglowcut))
        return error
    elif(rghighcut < rglowcut):
        error.append('Rghighcut < Rglowcut, Rghighcut = '+str(rghighcut)+' Rglowcut = '+str(rglowcut))
        return error
    elif(sastype != 1 and sastype != 2 and sastype !=3):
        error.append('sastype needs to be equal to 1, 2, or 3 : sastype = '+str(sastype))
        return error

    error=input_filter.check_file_exists(sasintfile)
    if(len(error) != 0):
        error.append('check input SAS exp. data file path + filename : '+os.path.basename(sasintfile))
        return error

### check interpolated sasintfile for three columns (q,I(q),error)
#    nsfit.data 
#    0.000000        0.019000        0.000824

    try:
        lines = open(sasintfile,'r').readlines()
    except:
        error.append("unable to open and read your SAS data file : "+os.path.basename(sasintfile))
    else:
        sum_I = 0.0
        Q_exp = []
        for line in lines:
            words = line.split()
            if len(words)==0 or words[0][0]=='#':
                continue
            elif len(words)>3 and words[3][0]!='#':
                error.append('the number of columes should be 3 in your SAS data file : '+os.path.basename(sasintfile)+'  \n'+line)
                return error
            else:
                try:
                    q = locale.atof(words[0])
                    I = locale.atof(words[1])
                    Error = locale.atof(words[2])
                    Q_exp.append(q)
                except:
                    error.append('the following line cannot be parsed in your SAS data file : '+os.path.basename(sasintfile)+'  \n'+line)
                    return error
                    #pass
                else:
                    if q<0.0:
                        error.append('negative Q value found in your SAS data file : '+os.path.basename(sasintfile)+'  \n'+line)
                        return error
                    elif I<0.0:
                        error.append('negative I value found in your SAS data file : '+os.path.basename(sasintfile)+'  \n'+line)
                        return error
                    elif Error<0.0:
                        error.append('negative error value found in your SAS data file : '+os.path.basename(sasintfile)+'  \n'+line)
                        return error
                    else:
                        sum_I += I
        if sum_I==0.0:
            error.append("no lines or all-zero intensity entries found in your SAS data file : "+os.path.basename(sasintfile))
            return error

    if(plotflag != 0 and plotflag != 1):
        error.append('plot flag needs to be 1 or 0 ... you entered: '+str(plotflag))
        return error

    # check basis_string
    list_basis = basis_string.split(',')
    if len(list_basis) !=  number_of_weight_files:
        error.append('number of weight files does not match number of basis string: there may be "," in the basis string')
        return error 
    for basis in list_basis:
        parse_grammar(basis, error)
        if len(error): return error

    # check sastype
    if sastype not in [1,2,3]:
        error.append("sastype %d not supported!"%sastype)
        return error
    ev,rv,wv=input_filter.check_permissions(saspath)
    if(not ev or not rv):
        error.append('permission error in input file path '+saspath+'  [code = '+str(ev)+str(rv)+str(wv)+']')
        if(ev==False):
            error.append('sas path "'+saspath+'" does not exist')
            return error
        elif(rv==False):
            error.append('read permission not allowed for sas path "'+saspath+'"')
            return error
    if(sastype==1):
        suffix=['*.iq','*.log']
        extra = ['*.inf','*.crd','*.ans','*.pr']
    elif(sastype==2):
        suffix=['*.int','*.log']
        extra = ['*.sav','*.flm','*.alm','*.ans']
    elif(sastype==3):
        suffix=['*.int','*.log']
        extra = ['*.sav','*.flm','*.alm','*.ans']
    spec_files = glob.glob(os.path.join(saspath,suffix[0]))
    log_files = glob.glob(os.path.join(saspath,suffix[1]))
    #spec_files.sort()
    #log_files.sort()
    number_spec_files = len(spec_files)
    dict_sastype = {1:'xtal2sas', 2:'cryson', 3:'crysol'}
    name_sastype = dict_sastype[sastype]
    if number_spec_files==0:
        error.append("there is no scattering files found for the selected sas-type: '"+name_sastype+"' in folder: '"+saspath+"'")
        return error
    # check q values
    Q_spec = []
    for line in open(spec_files[0]).readlines():
        words = line.split()
        if len(words)==0 or words[0][0]=='#':
            continue
        else:
            try:
                q = locale.atof(words[0])
                I = locale.atof(words[1])
            except:
                #error.append('the following line cannot be parsed in your SAS file : '+os.path.basename(spec_files[0])+'  \n'+line)
                #return error
                pass
            else:
                Q_spec.append(q)
    if Q_spec != Q_exp:
        output_Q_spec = str(Q_spec[:min(10,len(Q_spec))])
        if len(Q_spec)>10: output_Q_spec += '...'
        output_Q_exp = str(Q_exp[:min(10,len(Q_exp))])
        if len(Q_exp)>10: output_Q_exp += '...'
        error.append('Q value in your SAS file : '+os.path.basename(spec_files[0])+':\n'+output_Q_spec+'\n  does not match that in your experimental file:\n'+output_Q_exp)
        return error
    #check file size in the SAS folder
    file_size = os.path.getsize(spec_files[0]) 
    if len(spec_files)>1:
       for f in spec_files[1:]:
           if os.path.getsize(f) != file_size:
               error.append('File size of "'+f+'" does not match "'+os.path.basename(spec_files[0])+'".\nPlease verify the files in the SAS path are consistent!')
               return error


    # check io
    if io<=0.0:
        error.append("I0 should be greater than 0")
        return error
         
    # return
    return error