#!/usr/bin/python
# -*- coding: utf-8 -*-
#
# Pyromaths
# Un programme en Python qui permet de créer des fiches d'exercices types de
# mathématiques niveau collège ainsi que leur corrigé en LaTeX.
# Copyright (C) 2006 -- Jérôme Ortais (jerome.ortais@pyromaths.org)
#
# 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 2 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, write to the Free Software
# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
#

import re
from .decimaux import decimaux, decimaux_quiz
from ..classes import Racine, Fractions

        
def tex_coef(coef, var, bplus=0, bpn=0, bpc=0, cero=0):
    """
    coef est le coefficient à écrire devant la variable var
    bplus est un booleen : s'il est vrai, il faut ecrire le signe +
    bpn est un booleen : s'il est vrai, il faut mettre des parentheses autour de
        l'écriture si coef est negatif.
    bpc est un booleen : s'il est vrai, il faut mettre des parentheses autour de
        l'écriture si coef =! 0 ou 1 et var est non vide
    """
    if coef != 0 and abs(coef) != 1:
        if var == '':
                a = TeX(coef)
        else:
                a = '%s\\,%s' % (TeX(coef), var)
        if bplus and coef > 0:
            a = '+' + a
    elif coef == 1:
        if var == '':
            a = '1'
        else:
            a = '%s' % var
        if bplus:
            a = '+' + a
    elif coef == 0:
        if cero:
            a = '0'
        else:
            a = ''
    elif coef == -1:
        if var == '':
            a = '-1'
        else:
            a = '-%s' % var
    if bpn and coef < 0 or bpc and coef != 0 and coef != 1 and var != '':
        a = '\\left( ' + a + '\\right)'
    return a


def TeXz(nombre):
    '''n'affiche pas b si b=0'''
    if nombre==0:
        return ""
    else:
        return TeX(nombre)
def tTeX(nombre):
    '''raccourci pour TeX(nombre,terme=True)'''
    return TeX(nombre,terme=True)
def pTeX(nombre):
    '''raccourci pour TeX(nombre,parenthese=True)'''
    return TeX(nombre,parenthese=True)
def fTeX(nombre):
    '''raccourci pour TeX(nombre,fractex="\\frac")'''
    return TeX(nombre,fractex="\\frac")
    
def TeX(nombre,parenthese=False,terme=False,fractex="\\dfrac"):
    '''renvoie une chaine de caractere pour TeX'''
    strTeX=finTeX=""
    
    #Affichage simplifié des racines ou fractions
    if isinstance(nombre,Racine.RacineDegre2) and nombre.radicande==0:
        #Affiche la RacineDegre2 comme une Fractions
        nombre=Fractions.Fractions(nombre.numerateur,nombre.denominateur)
    if isinstance(nombre,Fractions.Fractions) and nombre.denominateur==1:
        #Affiche la Fractions comme un entier
        nombre=nombre.numerateur
    #parentheses des fractions
    if parenthese and (
        isinstance(nombre,Racine.RacineDegre2)
                       and nombre.denominateur==1 and (nombre.numerateur or nombre.coeff<0 )
        #RacineDegre2 avec radicande nécessairement grâce au tri
        or isinstance(nombre,Fractions.Fractions) and nombre.numerateur<0
        or isinstance(nombre,int) and nombre<0
        or isinstance(nombre,float) and nombre<0):
        strTeX="\\left("
        finTeX="\\right)" 
    elif terme and (isinstance(nombre,Racine.RacineDegre2) and
                        (nombre.denominateur!=1 or (nombre.numerateur >0 or nombre.numerateur==0 and nombre.coeff>=0))
                    or nombre>=0) :
        strTeX="+"
        finTeX=""

    ##Affichage
    if nombre==float("inf"):
        return "+\\infty "
    elif nombre==float("-inf"):
        return "-\\infty "
    elif isinstance(nombre,int) or isinstance(nombre,float):
        return strTeX+decimaux(nombre)+finTeX
    elif isinstance(nombre,Fractions.Fractions):
        if nombre.numerateur < 0:
            strTeX += "-"+fractex+"{"+decimaux(-nombre.numerateur)+"}{"+decimaux(nombre.denominateur)+"} "
        else:
            strTeX += fractex+"{"+decimaux(nombre.numerateur)+"}{"+decimaux(nombre.denominateur)+"} "
        strTeX+=finTeX
        return strTeX
    elif isinstance(nombre,Racine.RacineDegre2):
        return strTeX+str(nombre)+finTeX
    else:
        return strTeX+str(nombre)+finTeX

def TeX_quiz(nombre,mathenv = 0,fractex="\\frac"):
    '''renvoie une chaine de caractere pour TeX'''
    strTeX=finTeX=""

    if isinstance(nombre,Racine.RacineDegre2) and nombre.radicande==0:
        #Affiche la RacineDegre2 comme une Fractions
        nombre=Fractions.Fractions(nombre.numerateur,nombre.denominateur).TeX_quiz(mathenv)
        return nombre
    if isinstance(nombre,Fractions.Fractions) and nombre.denominateur==1:
        #Affiche la Fractions comme un entier
        nombre=nombre.numerateur

    ##Affichage
    if nombre==float("inf"):
        return "+\\infty "
    elif nombre==float("-inf"):
        return "-\\infty "
    elif isinstance(nombre,int) or isinstance(nombre,float):
        if mathenv:
            return strTeX+decimaux_quiz(nombre,1)+finTeX
        else:
            return strTeX+decimaux_quiz(nombre)+finTeX
    elif isinstance(nombre,Fractions.Fractions):
        if nombre.numerateur < 0 and mathenv:
            strTeX += "-"+decimaux_quiz(-nombre.numerateur)+"/"+decimaux_quiz(nombre.denominateur)
        elif nombre.numerateur > 0 and mathenv:
            strTeX += decimaux_quiz(nombre.numerateur)+"/"+decimaux_quiz(nombre.denominateur)
        elif nombre.numerateur < 0:
            strTeX += "-"+fractex+"{"+decimaux_quiz(-nombre.numerateur)+"}{"+decimaux_quiz(nombre.denominateur)+"}"
        else:
            strTeX += fractex+"{"+decimaux_quiz(nombre.numerateur)+"}{"+decimaux_quiz(nombre.denominateur)+"}"
        strTeX+=finTeX
        return strTeX
    elif isinstance(nombre,Racine.RacineDegre2):
        if mathenv:
            #if nombre.denominateur == 1:
            #    return [nombre.numerateur, nombre.radicande]
            #else:
            #    return [nombre.numerateur, nombre.radicande, nombre.denominateur]
            return nombre
        else:
            return strTeX+str(nombre).replace("\\dfrac",fractex)+finTeX
    elif isinstance(nombre,str):
        nombre = nombre.replace(",",".")
        if mathenv:
            return strTeX+decimaux_quiz(float(nombre),1)+finTeX
        else:
            return strTeX+decimaux_quiz(float(nombre))+finTeX
    else:
        return strTeX+str(nombre)+finTeX

def radicalTeX(n):
    return "\\sqrt{%s}"%(decimaux(n))