Le portail francophone de la géomatique

# coding: latin-1

# ---------------------------------------------------------------------------
# Extraction_Envi_2009_V3
# nouvelle version à partir de Extraction_Envi_V2
# # Dernière modification
# 20/01/2009 à 14h45
# Révisé le 15/12/2009
# ---------------------------------------------------------------------------

# Import system modules
import sys, string, os, arcgisscripting, shutil, datetime, time

heure_debut_calcul = time.time()
#print heure_debut_calcul

# Create the Geoprocessor object
gp = arcgisscripting.create()
gp.setproduct("ArcInfo")
gp.overwriteoutput = 1

# Charge la boite à outils  A_CETIOM
gp.AddToolbox("D:/SIG/OUTILS/BOITES_A_OUTILS/A_CETIOM.tbx")


# Définir Espace de Travail
#gp.workspace = sys.argv[1]
gp.workspace = r"D:\SIG\PROJETS\2009-2010\FARMSTAR\Donnees_SIG_Infoterra\ENVI\Test"


# Fichier Données CETIOM
#Fichiers_Donnees = sys.argv[2]
Fichiers_Donnees = r"D:\SIG\PROJETS\2009-2010\FARMSTAR\Donnees_SIG_CETIOM\Shapes_CETIOM\Test_extraction.shp"

# Definition de la période travaillée et/ou version (type EH1)
#Periode = sys.argv[3]
Periode = "SHH"

# Calcul_long Oui/Non
#Calcul_long = sys.argv[4]
Calcul_long = "Non"

print "Démarrage"

# Répertoire intermédiaire
rep_fic_inter = r"D:\SIG\PROJETS\2009-2010\FARMSTAR\Donnees_SIG_Infoterra\ENVI\Fichiers_Intermediaires"

# Variables de travail temporaire
#R_NoData = rep_fic_inter + r"\R_Nodata"
#R_Extrait = rep_fic_inter + r"\R_Extrait"
R_Entier = rep_fic_inter + r"\R_Entier"
#R_Moy = rep_fic_inter + r"\R_moy"
#R_Med = rep_fic_inter + r"\R_Med"
#R_Min = rep_fic_inter + r"\R_Min"
#R_Max = rep_fic_inter + r"\R_Max"
#Masque_Extraction = rep_fic_inter + r"\Masque_Extraction.shp"
#Fichier_Temp = rep_fic_inter + r"\Fichier_Temp.shp"
Layer_Attente = rep_fic_inter + r"\Layer_Attente"

# Création Répertoire intermédiaire
if os.path.isdir(rep_fic_inter) == 1:
    shutil.rmtree(rep_fic_inter)
    os.mkdir(rep_fic_inter)
    
else :
    os.mkdir(rep_fic_inter)
try:

    
   

    if gp.CheckExtension("spatial") == "Available":     # Verifie si l'extention est disponible

      
        gp.CheckOutExtension("spatial")                 # Si oui, prend l'extension

           
        ## Définir la liste des rasters
        Liste_Rasters = gp.listrasters("*","IMA")

        ## Remise à zéro compteur
        Liste_Rasters.reset()

        ## Affecter à la variable le nom du premier fichier
        Raster = Liste_Rasters.next()
        

        print Raster

        ## Boucle
        while Raster:

            # Calcul Durée Calcul
            Traitement_Debut = datetime.datetime.now()
     
            # Récupération du code parcelle pour comparer avec fichier RCA
            # 2009 --> LC02_B_048952_2_COLZA01.07_flt_200902271728.ima
            # 2010 --> GC1_LC01_A_081675_2_COLZA07.07_flt_200912041329.ima   Nom organisme en plus
            Num_Saisi = str(Raster[9:19])
            Num_Saisi = Num_Saisi.replace("_","-")
            print "Num_Saisi --> " + Num_Saisi

            # Chemin du Raster à traiter
            Chemin_Raster = gp.workspace + "\\" + Raster
            print "Chemin_Raster --> " + Chemin_Raster

            ## Expression pour transfomer en g puis en entier
            Expression = "int(" + Chemin_Raster + " * 1000)"
            print "Expression --> " + Expression

            # Process: Algèbre spatial - sortie unique...
            gp.SingleOutputMapAlgebra_sa(Expression, R_Entier)
            print "Raster Entier créé"
                       
            # Supprime les 0 par NoData
            #Valeur_NoData = "0"
            # Process: Copier un raster...
            #gp.CopyRaster_management(R_Entier, R_Nodata, "", "", Valeur_NoData, "NONE", "NONE", "")

            print "Fichiers_Donnees --> " + Fichiers_Donnees

            # Expression pour définir la parcelle à traiter
            #Expression_Nom = "\"NUM_SAISI\" = Num_Saisi"
            #print "Expression Num Saisi"

            # Génère une couche .lyr utilisable pour SelectLayer
            #gp.MakeFeatureLayer_management(Fichiers_Donnees, Layer_Attente, "", "", "")
            #print "Fichier Lyr --> " + Layer_Attente
            
            # Process: Sélectionner une couche par attributs...
            #gp.SelectLayerByAttribute_management(Layer_Attente, "NEW_SELECTION", Expression_Nom)
            #print "Selection effectuée"
            
            # Pixel Process: Extraction de la valeur de cellule d'une grille (8)...
            gp.GpExtractFunction(Layer_Attente, "Pix_" + Periode, R_Entier)   # mettre le chemin et nom du fichier

            #gp.ExtractValue("", "Valeur", "")  via ModelBulder
            #gp.ExtractValue(Layer_Attente, "Pix_" + Periode, R_Entier)
            

            print "Fin Extraction"
                    
            print "Raster " + Raster + " traité"
            gp.addwarning("Raster " + Raster + " traité")

              
            # Raster suivant
            Raster = Liste_Rasters.next()

        # Redonne l'extension spatial
        gp.CheckInExtension("spatial")

        # Suppression du repertoire intermédiaires
        shutil.rmtree(r"D:\SIG\PROJETS\2009-2010\FARMSTAR\Donnees_SIG_Infoterra\ENVI\Fichiers_Intermediaires")

        # Calcul du temps de calcul
        temps_calcul = time.time()-heure_debut_calcul
        print "Temps de calculs : " + str(temps_calcul)[:5] + " secondes.\n"
            
        #print "Traitement des rasters terminé"
        gp.addwarning("Traitement des rasters terminé - Temps de calculs : " + str(temps_calcul)[:5] + " secondes.\n")

    else:
        print "La license Spatial Analyst n'est pas disponible."
        gp.addwarning("La license Spatial Analyst n'est pas disponible.")

except:
    print gp.GetMessages(2)