# -*- coding: ISO-8859-1 -*-

####################################################################
#                                                                  #
#   Programme permettant de tracer un profil d'aile dans Catia V5  #
#   à partir d'un fichier texte de coordonnées XY donné            #
#                                                                  #
#                                               PSX59              #
#                               Apprendre-la-CAO.com               #
####################################################################

import sys, os
from _NewPart_profil import Cree_NewPart_profil
from _Recup_profil import Recup_profil



# Recupère le répertoire de travail 

try:
 dir_path = os.path.dirname(os.path.abspath(__file__))
except NameError: # We are the main py2exe script, not a module
 import sys
 dir_path = os.path.dirname(os.path.abspath(sys.argv[0]))

print 'repertoire de travail: \n',dir_path

# Définition du répertoire de stockage des profils
Chemin_txt_profil = "NACA"

# Définition du nom du profil à tracer dans CV5
Nom_profil = "NACA4418"

#######################
# Programme principal #
#######################

#Cree une nouvelle part avec les sets geom prédéfinis
Cree_NewPart_profil(dir_path,"Profil_"+Nom_profil,"Profil_"+Nom_profil)
print ("Part creee")

#Recupération du profil  et creation dans catia
Recup_profil(Nom_profil,"Profil_"+Nom_profil,"Profil_"+Nom_profil,dir_path+"//"+Chemin_txt_profil)

print ("Profil "+Nom_profil+" cree et Catpart sauvegardee dans le repertoire de travail")

# -*- coding: ISO-8859-1 -*-


def Cree_NewPart_profil(Chemin,Nom_part,Nom_set):

    

    import win32com.client

    CATIA=win32com.client.Dispatch('catia.application') #->la communication est établie

    PartDoc = CATIA.Documents.Add("Part")

    partDocument1 = CATIA.ActiveDocument

    #recup nom de la part

    Nom_ProvisoirComplet_CATPart = partDocument1.Name

    Coupure_Nom_provisoir_CATPart = Nom_ProvisoirComplet_CATPart.split(".")

    Nom_prov_CatPart = Coupure_Nom_provisoir_CATPart[0]

    #change le nom de la part

    part1 = partDocument1.Part

    parameters1 = part1.Parameters

    Nom_parametre = str(Nom_prov_CatPart) + "\Référence"



    strParam1 = parameters1.Item(Nom_parametre)

    strParam1.Value = Nom_part

    #creation du set geom pour le profil

    hybridBodies1 = PartDoc.Part.HybridBodies
    Set_Import = hybridBodies1.Add()
    print Nom_set
    Set_Import.Name = Nom_set

    part1.Update()



    # Sauvegarde de la part
    #print Chemin + "\\" + Nom_part + ".CATPart"
    partDocument1.SaveAs(Chemin + "\\" + Nom_part + ".CATPart")
    

# -*- coding: cp1252 -*-

from _lect_profil import lect_profil
from _Courbe_TxtCv5 import Courbe_CV5

def Recup_profil(Nom_profil,Set_geom, Nom_part, Rep):
    Chemin_Fich = Rep + "\\" + Nom_profil + ".txt"
    print Chemin_Fich
    Coord_Nds_Txt = lect_profil(Chemin_Fich) #On lit le fichier texte
    print len(Coord_Nds_Txt)
    Nom_courbes = ['Extrados','Intrados']
    a = 0
    for Liste in Coord_Nds_Txt: #pour chacune des deux courbes...
        if Liste==[]:
            pass
        else:
            Courbe_CV5(Nom_courbes[a] , Liste,   Set_geom, Nom_part) #On crée les courbes dans Catia
            a = a + 1
    return 'profil ',Nom_profil,' charge'

# -*- coding: ISO-8859-1 -*-


def lect_profil(Chemin_Fich):

    #lect_profil("G:/Etudes_avions/Sites_web/Apprendre_la_CAO/Sujets/Profil_aile/NACA/NACA4418_airfoiltools.txt")
    Fichier_profil = open(Chemin_Fich, 'r')
    Liste_ligne = Fichier_profil.readlines()

    Nb_lignes = len (Liste_ligne)

    print ("nombre de lignes dans le fichier : " + str(Nb_lignes))

    Liste1 = [] #extrados
    Liste2 = [] #intrados

    
    Liste_coord = [Liste1, Liste2]

    X = 1
    
    a = 0 #Numéro de ligne
        

    while a <  Nb_lignes:
        print 'ligne N°',a
        Elem_ligne = Liste_ligne[a].split()
        print '--->', Elem_ligne
        try:
            premier_elem = round(float(Elem_ligne[0]),6)
            second_elem = round(float(Elem_ligne[1]),6)
        except:
            print 'ligne inutile'
            print 'ligne decoupée',Elem_ligne
            a = a + 1


        else:
            deltaX = X - premier_elem #X est l'ancienne valeur d'abscisse et premier_elem est l'abscisse actuelle
            X = premier_elem
            Y = second_elem

            if deltaX >= 0: #Abscisses décroissantes ?
                Liste1.append([X,Y])
                print 'Liste1 -->',[X,Y]
                a=a+1

            if X == 0.000: #lorsque l'on tombe sur le point origine on le duplique en l'ajoutant à la liste2
                Liste2.append([0.00,0.00])
                
            if deltaX < 0: #Abscisses croissantes ?
                Liste2.append([X,Y])
                print 'Liste2 -->',[X,Y]
                a=a+1
                
                
    return Liste_coord

# -*- coding: ISO-8859-1 -*-


import win32com.client


def Courbe_CV5(a , Liste, Set_geom, Nom_part):
        CATIA=win32com.client.Dispatch('catia.application') #->la communication est établie

        PartDoc = CATIA.ActiveDocument
        part1 = PartDoc.Part

        sel = PartDoc.Selection
        sel.Clear()
        #ouvre le SET geom du profil
        hybridBodies1 = PartDoc.Part.HybridBodies
        hybridBody_import = hybridBodies1.Item(Set_geom)
        hybridBodies2 = hybridBody_import.HybridBodies
        #cree un set pour la courbe
        hybridBody_Courbe = hybridBodies2.Add()
        hybridBody_Courbe.Name = "Courbe_" + a

        if Liste==[]:
            pass
        else:
            hybridShapes1 = hybridBody_Courbe.HybridShapes
            b = 1
            for Coord in Liste:
                X = Coord[0]*100 # modifiez ici si vous voulez modifier l'échelle du profil (ici 100 fois plus grand soit 100mm de long) Rappel: le profil à une longueur de 1 (sans unité). Si le fcteur vaut 1 alors cela fera 1mm dans catia
                Y = Coord[1]*100 # idem
                Point1 = part1.HybridShapeFactory.AddNewPointCoord(X, Y, 0) #cree le point de ccord XY
                hybridBody_Courbe.AppendHybridShape(Point1)
                sel.Add(Point1)
                Point1.Name = "Point_" + str(b) # renomme le point
                part1.UpdateObject(Point1)
                b = b + 1
        
        Nb_pt_dans_set = hybridShapes1.Count #compte les points
        print 'Nb points = ',Nb_pt_dans_set 

        hybridShapeSpline1 = part1.HybridShapeFactory.AddNewSpline() #cree une spline vide
        hybridShapeSpline1.SetSplineType(0)
        hybridShapeSpline1.SetClosing(0)
        hybridShapeSpline1.Name = "Spline_" + str(a) # Nomme la spline
        # Je n'ai pas ajouté de tangence verticale au point d'origine (normale au plan XY), il y aura donc une vraie arête au bord d'attaque à imposer manuellement dans catia sur chaque spline ou ajouter une contrainte dans ce script

        nb = 1
        while nb < Nb_pt_dans_set +1 :
                hybridShapePointOnPlane1 = hybridShapes1.Item("Point_" + str(nb)) # appel le point par son nom
                reference1 = part1.CreateReferenceFromObject(hybridShapePointOnPlane1)
                hybridShapeSpline1.AddPoint(reference1) #ajoute le point à la spline
                nb = nb + 1
                
        hybridBody_import.AppendHybridShape(hybridShapeSpline1) #ajoute la spline au set geom
        part1.UpdateObject(hybridShapeSpline1)

        #Publication des courbes

        PartDoc = CATIA.ActiveDocument

        product1 = PartDoc.GetItem(Nom_part)
        publications1 = product1.Publications

        Nom_publication = Nom_part+"_Courbe_" + a
        publication1 = publications1.Add(Nom_publication)

        Nom_courbe = Nom_publication+"/!Spline_" + a

        reference1 = product1.CreateReferenceFromName(Nom_courbe)
        
        publications1.SetDirect (Nom_publication, reference1)

        product1.update()    

        #Sauvegarde du profil

        PartDoc.Save()