Application: gvSIG desktop » gvSIG scripting

from gvsig import *

from commonsdialog import *

from libs.toolbox import *

class XYShift(ToolboxProcess):

  def defineCharacteristics(self):

    """

En esta operacion debemos definir los parametros de netrada y salida que va a precisar nuetro proceso.

    """

    # Fijamos el nombre con el que se va a mostrar nuestro proceso

    self.setName("Prueba desplazamiento en X e Y")

    # Indicamos el grupo en el que aparecera

    self.setGroup("Vectorial")

    params = self.getParameters()

    # Indicamos que precisamos un parametro LAYER, del tipo punto y que es obligatorio

    params.addInputVectorLayer("LAYER","Caoa de entrada", SHAPE_TYPE_POINT,True)

    # Indicamos que precisamos un par de valores numericos, X e Y 

    params.addNumericalValue("X", "X_traslation",0, NUMERICAL_VALUE_DOUBLE)

    params.addNumericalValue("Y", "Y_traslation", 0, NUMERICAL_VALUE_DOUBLE)

    # Y por ultimo indicamos que precisaremos una capa de salida de puntos.

    self.addOutputVectorLayer("RESULT_POINT", "XYShift_point", SHAPE_TYPE_POINT)

  def processAlgorithm(self):

    """

Esta operacion es la encargada de realizar nuetro proceso.

    """

    features=None

    try:

      """

      Recogemos los parametros y creamos el conjunto de entidades asociadas a la capa

      de entrada.

      """

      params = self.getParameters()

      layer = params.getParameterValueAsVectorLayer("LAYER")

      x = params.getParameterValueAsDouble("X")

      y = params.getParameterValueAsDouble("Y")

      input_store = layer.getFeatureStore()

      features = input_store.getFeatureSet()

      """

      Generamos la capa de salida con la misma estructura que la capa de entrada

      """

      output_store = self.buildOutPutStore(

        features.getDefaultFeatureType(), 

        SHAPE_TYPE_POINT,

        "XYShift_points",

        "RESULT_POINT"

      )

      """

      Nos recorremos todas las entidades de entrada, y creamos las de salida desplazando la geometria

      en los valores indicados por la X e Y de los parametros.

      """

      self.setRangeOfValues(0,features.getSize())

      for feature in features.iterator():

        if self.isCanceled():

          # Si el usuario indico que quiere cancelar el proceso abortamos.

          print "Proceso cancelado"

          break

        # Incrementamos el progreso de nuestro proceso.

        self.next()

        # Creamos una nueva entidad para nuestro almacen de salida.

        newfeature = self.createNewFeature(output_store,feature)

        # Desplazamos la geometria de la nueva entidad

        geom = newfeature.getDefaultGeometry()

        geom.move(x,y)

        # Guardamos la nueva entidad

        output_store.insert(newfeature)

      # Cuando hemos terminado de recorrernos las entidades terminamos la edicion.

      output_store.finishEditing()   

    finally:

      DisposeUtils.disposeQuietly(features)

      print "Proceso terminado %s" % self.getCommandLineName() 

def main(*args):

    # Creamos nuesto geoproceso

    process = XYShift()

    # Lo registramos entre los procesos disponibles en el grupo de "Scripting"

    process.selfregister("Scripting")

    # Actualizamos el interface de usuario de la Toolbox

    process.updateToolbox()

    msgbox("Incorporado el script '%s/%s/%s' a la paleta de geoprocesos." % (

        "Scripting",

        process.getGroup(),

        process.getName()

      )

    )