import os
from PyQt4 import QtGui
from sextante.parameters.ParameterString import ParameterString
from sextante.lastools.LasToolsUtils import LasToolsUtils
from sextante.parameters.ParameterBoolean import ParameterBoolean
from sextante.lastools.LasToolsAlgorithm import LasToolsAlgorithm
from sextante.parameters.ParameterNumber import ParameterNumber
from sextante.outputs.OutputVector import OutputVector
from sextante.parameters.ParameterFile import ParameterFile

class las2iso(LasToolsAlgorithm):

    INPUT = "INPUT"
    OUTPUT = "OUTPUT"
    CLEAN = "CLEAN"
    SIMPLIFY = "SIMPLIFY"
    INTERVAL = "INTERVAL"

    def getIcon(self):
        filepath = os.path.dirname(__file__) + "/../images/tool.png"
        return QtGui.QIcon(filepath)

    def defineCharacteristics(self):
        self.name = "las2iso"
        self.group = "Tools"
        self.addParameter(ParameterFile(las2iso.INPUT, "Input las layer", ""))
        self.addParameter(ParameterNumber(las2iso.INTERVAL, "Interval between isolines", 0, None, 10.0))
        self.addParameter(ParameterNumber(las2iso.CLEAN, "Clean isolines shorter than (0 = do not clean)", None, None, 0.0))
        self.addParameter(ParameterNumber(las2iso.SIMPLIFY, "simplify segments shorter than (0 = do not simplify)", None, None, 0.0))
        self.addOutput(OutputVector(las2iso.OUTPUT, "Output isolines"))
        self.addCommonParameters()

    def processAlgorithm(self, progress):
        commands = [os.path.join(LasToolsUtils.LasToolsPath(), "bin", "las2iso.exe")]
        commands.append("-i")
        commands.append(self.getParameterValue(las2iso.INPUT))
        commands.append("-o")
        commands.append(self.getOutputValue(las2iso.OUTPUT))
        commands.append("-iso_every")
        commands.append(str(self.getParameterValue(las2iso.INTERVAL)))
        simplify = self.getParameterValue(las2iso.SIMPLIFY)
        if simplify != 0:
            commands.append("-simplify")
            commands.append(str(simplify))
        clean = self.getParameterValue(las2iso.CLEAN)
        if clean != 0:
            commands.append("-clean")
            commands.append(str(clean))
        self.addCommonParameterValuesToCommand(commands)

        LasToolsUtils.runLasTools(commands, progress)
