Application: gvSIG desktop » gvSIG scripting

# Version: v0.4

#

def addDependencyWithPlugin(pluginCode):

    pluginsManager = PluginsLocator.getManager()

    scriptingPlugin = pluginsManager.getPlugin("org.gvsig.scripting.app.extension")

    scriptingPlugin.addDependencyWithPlugin(pluginsManager.getPlugin(pluginCode))

addDependencyWithPlugin("org.gvsig.geoprocess.app.mainplugin")

              if DEV_INFO: print "extent add+"

                          from es.unex.sextante.parameters import RasterLayerAndBand

                          if DEV_INFO: print "*************************** RASTERLAYERBAND", rlb

              from java.util import ArrayList

          if isinstance(frame, str) or isinstance(frame, gvsig.View):

              if isinstance(frame, gvsig.View): view = frame

          dalManager = gvsig.DALLocator.getDataManager()

          mapContextManager = gvsig.MapContextLocator.getMapContextManager()

            print "|\t\t", value.getFileName()[0]

                value = loadRasterLayerNew(value.getFileName()[0], tocName=kwparams["TOCNAME"])

                value = loadRasterLayerNew(value.getFileName()[0])

      print "\n"

    return gvsig.Layer(layer)

            gvsig.copyToDynObject(parameters, store_parameters)

        except Throwable, ex:

            raise RuntimeException("Can't load layer, "+ str(ex))  

        return layer

    #checkFilesExist(["C:/gvsig/ran10.shp"])

    #geoprocessSearch(" ")

    #geoprocessHelp("closegapsnn")

    #geoprocessHelp("perturbatepointslayer")

    #r = runalg("perturbatepointslayer", LAYER = gvsig.currentLayer(),MEAN = 10, STDDEV = 10 )

    #r = runalg("perturbatepointslayer", EXTENT = "VIEW", LAYER = currentLayer(),MEAN = 10, STDDEV = 10 )

    #r = runalg("perturbatepointslayer", EXTENT = [0,0,500,500], LAYER = currentLayer(), MEAN = 10, STDDEV = 10 )

    #r = runalg("perturbatepointslayer", PATH = "C://gvsig//perturbatepoints028.shp", LAYER = gvsig.currentLayer(),MEAN = 5, STDDEV = 5 )

    #layer = gvsig.currentView().getLayer("data_test_lines.shp")

    #r = runalg("linestoequispacedpoints", LINES=layer,DISTANCE=2)

    #for i in range(10):

    #r = runalg("perturbatepointslayer", LAYER = r[0],PATH = "C://gvsig//perturbatepoints028_" + str(i) + ".shp",MEAN =0.5, STDDEV = 0.5 )

    #r = runalg("fixeddistancebuffer", LAYER = r[0], DISTANCE=1, TYPES="", RINGS=3, NOTROUNDED=False)

    #r = runalg("randomvector", COUNT=20, TYPE=1, EXTENT=gvsig.currentView())

    #RASTER

    #r1 = runalg("generaterandomnormal", EXTENT = [0,0,500,500], PATH = "C://gvsig//perturbatepoints030.tif", MEAN =0.5, STDDEV = 0.5)

    #layer = gvsig.currentView().getLayer("perturbatepoints030")

    #raster = currentRaster()

    #print raster

    #layer = gvsig_raster.loadRasterLayer('c:/gvsig/test_low.tif')

    #r1 = runalg("gradientlines",INPUT = layer, MIN=1, MAX=10, SKIP=1)

    #r = runalg("gridorientation",INPUT=layer,METHOD=0)

    #r = runalg("gridorientation",INPUT=layer,METHOD=0, PATH = "C://gvsig//perturbatepoints010.tif")

    #r = runalg("gridorientation", INPUT=raster, METHOD=0, PATH = "C://gvsig//perturbatepoints011.tif")

    #r = runalg("gradientlines", INPUT = layer, MIN=1, MAX=10, SKIP=1, PATH = "C://gvsig//perturbatepoints012.tif")

    #r = runalg("generaterandomnormal", EXTENT = [0,0,500,500], PATH = "C://gvsig//perturbatepoints013.tif", MEAN =0.5, STDDEV = 0.5)

    #geoprocessHelp("randomvector")

    #r = runalg("randomvector", COUNT=20, TYPE=2, EXTENT=gvsig.currentLayer())

    #r = runalg("randomvector", 200, TYPE_POINT, EXTENT=gvsig.currentLayer(), PATH="C://gvsig//test_puntos_sm01.shp")

    #r = runalg("randomvector", COUNT=20, TYPE=1, EXTENT=gvsig.currentView())

    #r = runalg("randomvector", COUNT=20, TYPE=1, EXTENT="VIEW")

    #r = runalg("randomvector", COUNT=20, TYPE=1, EXTENT=currentRaster())

    #r = runalg("gvSIG-convexhull", LAYER="Puntos_de_interes_01.shp", CHECK=True, PATH = "C://gvsig//gvsigconvexhull_001.shp")

    #r = runalg("generaterandomnormal", PATH = "C://gvsig//per.tif", EXTENT=gvsig.currentLayer(), CELLSIZE = 100, PATH = "C://gvsig//perturbatepoints014.tif", MEAN =5, STDDEV = 5)

    #geoprocessHelp("tablebasicstats")

    #r =runalg("tablebasicstats",TABLE=gvsig.currentTable(), FIELD=0)

    #Without parameters label

    #layers = gvsig.currentView().getLayers()

    ##r = runalg("gridorientation", layer, 0, PATH = "C://gvsig//perturbatepoints012.tif")

    #r = runalg("gridorientation", layer, 0)

    #r = runalg("gradientlines", layers[0], 1, 10, 1, PATH = "C://gvsig//perturbatepoints012.shp")

    #r = runalg("cva", "test_low", "test_low", "test_low", "test_low")

    #r = runalg("cva", currentRaster(), currentRaster(), currentRaster(), currentRaster(),PATH=["C:/gvsig/1.tif","C:/gvsig/2.tif"])

    #layerRaster = gvsig_raster.loadRasterLayer('c:/gvsig/test_low.tif')

    #r = runalg("gridorientation",layerRaster,0, PATH = "C://gvsig//Grid_orientation.tif",EXTENT=layerRaster, CELLSIZE=1, CELLSIZEZ=10)

    #r2 = runalg("cva", r, r, r, r, PATH=["C:/gvsig/1.tif","C:/gvsig/2.tif"])

    #print r2[0], r2[1]

    #print layerRaster

    #print getProjectLayer("Vista1", "test_low")

    #layer = getProjectLayer("Vista1", "as.shp")

    #extent = getProjectLayer("Vista1", "analisis_extent")

    #runalg("difference", "as.shp", "vista2_testeo.shp", PATH="C:/gvsig/recorte_extent.shp", EXTENT=[100, 100, 0, 540, 500, 0])

    #runalg("difference", "vista2_testeo.shp", layer, PATH="C:/gvsig/recorte_extent_2.shp", EXTENT=layer, OUTVIEW="Nueva")

    #r = runalg("tablebasicstats", "species", 0)

    #print r.encode("UTF-8")

    #algHelp("generaterandomnormal")

    """

    r = runalg("generaterandomnormal", 100,100, CELLSIZE=100, EXTENT=[250,250,0,500,500,0])

    r = runalg("generaterandomnormal", 10, 10, CELLSIZE=50, EXTENT=[500,500,0, 1000,1000,0])

    r = runalg("generaterandombernoulli", 50.0, CELLSIZE=25, EXTENT=[1000,1000,0, 1250,1250,0])

    r = runalg("gradientlines", r, 1, 100, 1)

    v1 = runalg("randomvector",10, TYPE_POLYGON, EXTENT=[0,0,0,500,500,0])

    v2 = runalg("randomvector", 5, TYPE_POLYGON, EXTENT=v1)

    v3 = runalg("difference", v1, v2, PATH="C:/gvsig/Diferencia.shp")

    v4 = runalg("randomvector", 5, 0, PATH="C:/gvsig/randomvector.shp", EXTENT=v3)

    v5 = runalg("randomvector", 100, 2, PATH="C:/gvsig/randompoints.shp", EXTENT="randomvector", OUTVIEW="Nueva")

    v6 = runalg("gvSIG-xyshift", "randompoints", "false", "-250.0", "-250.0", PATH=["C:/gvsig/ran1010.shp","C:/gvsig/ran2020.shp","C:/gvsig/ran3030.shp"])

    algHelp("tablebasicstats")

    v7 = runalg("gvSIG-buffer", "randompoints", False, 50.0, 0, False, True, 0, 0, PATH="C:/gvsig/buffer_gvsig0138.shp")

    v5 = runalg("randomvector", 100, 2, EXTENT=[0,0,0,500,500,0])

    #Test for model sextante

    v5 = runalg("randomvector", 10, 2, EXTENT=[0,0,0,500,500,0], OUTVIEW="Nueva")

    v5 = runalg("randomvector", "5", 2, EXTENT=[0,0,0,500,500,0], OUTVIEW="Nueva")

    r = runalg("generaterandomnormal", 100,100, CELLSIZE=1)

    vista= (gvsig.currentView().name).encode('UTF-8')

    """

    """

    vista= "Sin título"

    #vista = vista.encode("UTF-8")

    print gvsig.currentProject().getView(vista)

    endid = "_104.shp"

    v = runalg("randomvector", 20, 0)

    v2 = runalg("randomvector", 20, 0,PATH=["C://gvsig//i"+endid])

    #fitpoints = runalg("fitnpointsinpolygon",v, "2", "74", 1, 0)

    intersection = runalg("gvSIG-intersection", v, getProjectLayer(gvsig.currentView(),"i_102"), True, True,PATH=["C://gvsig//inter_pol"+endid,"C://gvsig//inter_line"+endid,"C://gvsig//inter_point"+endid])

    """

    #algSearch("nodelines")

    #ID:  extractendpointsoflines  || GROUP:  Topolog????a  || NAME:  Obtener extremos de l??neas

    #v7 = runalg("gvSIG-buffer", v, "False", 50.0, 0, "false", "true", 0, 0)

    #print "**Addfield: ", gvpy.addField(v, "ID5")

    #import gvpy

    #gvpy.addField(v, "ID89")

    #runalg("vectoraddfield", v, "Campo1", 0, 1, 0, 1, PATH="C:\\Users\\Oscar\\Desktop\\testi.shp")

    #layer = runalg("randomvector", 20, TYPE_LINE, EXTENT=[0,0,0,500,500,0], OUTVIEW="Po1")

    #2015-1-2

    #runalg("graticulebuilder", "10.0", "10.0", 1, PATH="/home/oscm/temp/reticula10.shp")

    #layer1 = gvsig.currentView().getLayer("1")

    #layer2 = gvsig.currentView().getLayer("2")

    #print layer1, layer2

    #

    #algHelp("cluster")

    #runalg("merge", layer1, [layer2])

    #layer1 = sRaster(0)

    #layer2 = sRaster(1)

    #print "Layer1: ", layer1.name, layer1, layer1.getCellSize()

    #print "Layer2: ", layer2.name, layer2, layer2.getCellSize()

    print "end"

def mainLibrary():

    #showFields(gvsig.currentLayer())

    #removeField(layer, "campo3")

    #removeField(layer, "campo5")

    #addField(layer,"campo1")

    #addField(layer,"campo2")

    #renameField(layer, "campo5", "campo50")

    #print list2geompoly([[1,2],[3,10],[5,30]])

    #addFeature(layer, "test01", "test2", [15,200])

    #addFeature(layer, "test11", "test3", geom.createPoint(300, 301))

    #addFeature(layer, campo1 = "kwparam", campo2 = "kwparam2", GEOMETRY = geom.createPoint(300,300))

    #addFeature(layer, "linea", "01", [[1,2],[3,10],[5,30]])

    #addFeature(layer, "pol", "02", [[50,80],[150,50],[100,10],[0,10],[50,80]])

    """

    layer = newLayer()

    addFeature(layer, "pol", "01", [[50,80],[150,50],[100,10],[0,10],[50,80]])

    addFeature(layer, "pol", "02", [[0,0],[10,5],[10,10],[0,10],[5,5]])

    addFeature(layer, "pol", "03", [[-50, -34],[0,0], [-14,30]])

    addField(layer,"campo3")

    modifyFeatures(layer, "campo3", "nuevo poligono")

    """

    """

    #Create shapes

    layer1 = newLayer(layer,"C:/gvsig/point_shape.shp", 1)

    layer2 = newLayer(layer,"C:/gvsig/line_shape.shp", 2)

    layer3 = newLayer(layer,"C:/gvsig/polygon_shape", 3)

    #Add features

    addFeature(layer1, "point", "01", [50,80])

    addFeature(layer1, "point", "02",[150,50])

    addFeature(layer1, "point", "03",[100,10])

    addFeature(layer1, "point", "04",[0,10])

    addFeature(layer3, "polygon", "01", [[50,80],[150,50],[100,10],[0,10],[50,80]])

    addFeature(layer2, "line", "01", [[50,80],[150,50],[100,10],[0,10],[50,80]])

    #Modify all values in one column

    modifyFeatures(layer1, "campo1", "Points_gsoc")

    modifyFeatures(layer2, "campo1", "Lines_gsoc")

    #Modify schema

    addField(layer1,"Name")

    removeField(layer1,"Surname")

    ##Execute SEXTANTE

    #r = geoprocess("perturbatepointslayer", LAYER = currentLayer(),MEAN = 5, STDDEV = 5 )

    """

    #layer = gvsig.currentView().getLayer("line_04.shp")

    #newLayer(layer, "C:/gvsig/gvpy_test006.shp")

    #layer2 = copyLayer(layer, "C:/gvsig/gvpy_copylayer_012.shp")

    #v = copyLayer(layer, "C:/gvpy_copylayer_new_06.shp")

    #layer = gvsig.currentView().getLayer("gvpy_copylayer_new_06")

    #addFeature(v, "Camino", [[50,00],[50,50],[10,10],[0,1],[50,18]])

    #Basics field

    """

    addField(v, "Direccion")

    modifyFeatures(v, "Direccion", "Av")

    removeField(v, "Direccion")

    """

    """

    #EJEMPLO 1

    #New shapes

    layer = gvpy.runalg("randomvector", 20, gvpy.TYPE_LINE, EXTENT=[0,0,0,500,500,0], OUTVIEW="Po1")

    #layer = gvsig.currentLayer()

    #Advanced field

    removeField(layer, "ID")

    removeField(layer, "Distance")

    addField(layer, "ID") #Add fields

    addField(layer, "Distance", "STRING")

    addField(layer, "Long", "LONG")

    removeField(layer, "Long") #Remove field

    modifyFeatures(layer, "ID", "90") #Modify all features

    modifyField(layer, "ID", "LONG") #Modify type of field

    modifyField(layer, "Distance", "FLOAT")

    addFeature(layer, 1, 0, [[50,0],[1000,0]]) #Add new feature with geometry line

    for feature in layer.features():

        perimeter = feature.geometry().perimeter()

        modifyFeature(layer, feature, "Distance", perimeter) #Modify each feature

    pass

    #FIN EJEMPLO 1

    """

    #modifyFeatures(gvsig.currentLayer(), "ID", 100, "Distance == 90")

    layer = gvsig.currentLayer()

    for feature in gvsig.currentLayer().features():

        value = feature.Distance

        #modifyFeature(layer, feature, "ID", distance)

        feature.edit()

        feature.set("ID", value)

        layer.update(feature)

    #model2script("C://gsoc//test02.model", "C://gsoc//nuevoScript.py")

    print "END TEST"

def copyLayerFeatures2Layer(layer1, layer2):

    for i in layer1.features():

        layer2.append(i.getValues())

    layer2.commit()

def copyLayer(layer, path):

    output = newLayer(layer, path)

    copyLayerFeatures2Layer(layer, output)

    #addLayerView(output)

    return output

def newLayer(layer, path, geometryType=None):

    CRS = layer.getProjectionCode()

    schema = gvsig.createSchema(layer.getSchema())

    if geometryType==None: geometryType = layer.getTypeVectorLayer().getType()

    output = gvsig.createShape( schema, path, CRS=CRS, geometryType=geometryType )

    gvsig.currentView().addLayer(output)

    return output

def addLayerView(layer):

    gvsig.currentView().addLayer(layer)

def addFeature(layer, *params, **kwparams):

    #IN: layer, feature params + geometry

    typeLayer = layer.getTypeVectorLayer().name

    #if kwparams != {}:

    #    layer.append(kwparams)

    #    layer.commit()

    #    return

    if "COMMIT" in kwparams:

        pass

    else:

        COMMIT=1

    if params != ():

        schValues = layer.getSchema().getAttrNames()

        values = {}

        itera = iter((list(params)))

        value = itera.next()

        for sch in schValues:

            #Si el campo a modificar es una geometria

            #print "Comprobación:", sch, isinstance(value, list)

            #re comprobacion si es campo geometry

            #bug: Comprobar si es lista o objeto geom  en primer if

            #... sch == "Geometry" and ES UNA LISTA

            #... sino copia el valor directamente: caso de pasar geometrias

            if sch == "GEOMETRY":

                print typeLayer

                if typeLayer == "Point2D":

                    if isinstance(value, list):

                        values[sch] = geom.createPoint(value[0],value[1])

                elif typeLayer == "MultiCurve2D":

                    if isinstance(value, list):

                        values[sch] = list2geomcurve(value)

                elif typeLayer == "MultiSurface2D":

                    if isinstance(value, list):

                        values[sch] = list2geompoly(value)

                else: #Si son geometrias

                    values[sch] = value

            else:

                values[sch] = value

            try:

                value = itera.next()

            except:

                break

        layer.append(values)

        if COMMIT==1: layer.commit()

    print "Add feature ", params, " to ", layer

def list2geompoly(listPoints):

    #IN: list[[x,y],...]

    #OUT: geometry polygon

    geometry = geom.createGeometry(3)

    for point in listPoints:

        geometry.addVertex(geom.createPoint(point[0],point[1]))

    if listPoints[0] != listPoints[len(listPoints)-1]:

        geometry.addVertex(geom.createPoint(listPoints[0][0], listPoints[0][1]))

    return geometry

def list2geomcurve(listPoints):

    #IN: list [[x,y],...]

    #OUT: geometry line

    geometry = geom.createGeometry(2)

    for point in listPoints:

        geometry.addVertex(geom.createPoint(point[0],point[1]))

    return geometry

def modifyFeatures(layer, field, value, COMMIT=1,FILTER=None):

    #IN: layer, field, new value

    if FILTER == None:

        features = layer.features()

    else:

        features = layer.features(FILTER)

    for feature in features:

        feature.edit()

        feature.set(field, value)

        layer.update(feature)

    if COMMIT==1: layer.commit()

    print "Modify feature ", layer.name, field, value

def showFields(layer):

    #IN: layer

    #OUT: str(atributos)

    print layer.getSchema().getAttrNames()

def addField(layer,field, sType = "STRING",iSize=20):

    #IN: layer, field, *sType, *iSize)

    #OUT: layer

    #addField(layer, "nombre")

    schema = gvsig.createSchema(layer.getSchema())

    schema.modify()

    if isinstance(field,str): schema.append(field,sType,iSize)

    layer.edit()

    layer.updateSchema(schema)

    layer.commit()

    print "Add field ", field, " to ", layer.name

    return layer

def modifyField(layer, field, iType="STRING", iSize=20):

    temp = []

    for i in layer.features():

        temp.append(i.get(field))

    removeField(layer, field)

    addField(layer, field, iType, iSize)

    n = 0

    for i in layer.features():

        modifyFeature(layer, i, field, temp[n],COMMIT=0)

        n += 1

    layer.commit()

    print "Modify field type to: ", field, " in ", layer.name

def modifyFeature(layer, feature, field, value, COMMIT=1):

    feature.edit()

    feature.set(field, value)

    layer.update(feature)

    print "Modify Feature field: ", field , " to ", value

    if COMMIT==1: layer.commit()

def removeField(layer, field):

    #IN: layer, field

    #OUT: layer

    #removeField(layer, "apellido")

    print "Campo %s eliminado" % (str(field))

    schema = layer.getSchema()

    schema.modify()

    if isinstance(field,str): schema.remove(field)

    else: return

    layer.edit()

    layer.updateSchema(schema)

    layer.commit()

    return layer