Skip to content
Permalink
Browse files

Add i.rectify algorithm

  • Loading branch information
Médéric Ribreux Médéric RIBREUX
Médéric Ribreux authored and Médéric RIBREUX committed May 4, 2016
1 parent 8ff07cf commit fe5c6f51574f8c0d79a3e354c1492f1163c0e2c9
@@ -0,0 +1,14 @@
i.rectify
Rectifies an image by computing a coordinate transformation for each pixel in the image based on the control points.
Imagery (i.*)
ParameterMultipleInput|rasters|Name of raster maps to rectify|3|False
ParameterFile|gcp|Ground Control Points file|False|False
ParameterSelection|order|Rectification polynomial order|1;2;3|0
ParameterString|resolution|Target resolution|None|False|True
ParameterNumber|memory|Amount of memory to use in MB|1|None|300|True
ParameterSelection|method|Interpolation method to use|nearest;linear;cubic;lanczos;linear_f;cubic_f;lanczos_f|0
ParameterCrs|crs|Destination CRS|None|False
Hardcoded|extension=rectified
*ParameterBoolean|-t|Use thin plate spline|False
OutputDirectory|output|Output Directory

@@ -0,0 +1,102 @@
# -*- coding: utf-8 -*-

"""
***************************************************************************
i_rectify.py
------------
Date : April 2016
Copyright : (C) 2016 by Médéric Ribreux
Email : medspx at medspx dot fr
***************************************************************************
* *
* This program is free software; you can redistribute it and/or modify *
* it under the terms of the GNU General Public License as published by *
* the Free Software Foundation; either version 2 of the License, or *
* (at your option) any later version. *
* *
***************************************************************************
"""

__author__ = 'Médéric Ribreux'
__date__ = 'April 2016'
__copyright__ = '(C) 2016, Médéric Ribreux'

# This will get replaced with a git SHA1 when you do a git archive

__revision__ = '$Format:%H$'

from i import regroupRasters, copyFile, multipleOutputDir
from qgis.core import QgsMessageLog
from qgis.core import QgsCoordinateReferenceSystem
from ..Grass7Utils import Grass7Utils
from processing.core.parameters import getParameterFromString
from os import path


def processCommand(alg):
# Creates a new location with the CRS
crsParam = alg.getParameterFromName('crs')
crsId = int(crsParam.value[5:])
#QgsMessageLog.logMessage('crs = {}'.format(crs), 'DEBUG', QgsMessageLog.INFO)
crs = QgsCoordinateReferenceSystem()
crs.createFromId(crsId, QgsCoordinateReferenceSystem.EpsgCrsId)
command = "g.proj proj4=\"{}\" location=TARGET".format(crs.toProj4())
alg.commands.append(command)
alg.parameters.remove(crsParam)

# Regroup rasters
rasters = alg.getParameterFromName('rasters')
rastersList = rasters.value.split(';')
alg.parameters.remove(rasters)

# Insert a i.group command
group = getParameterFromString("ParameterString|group|group of rasters|None|False|False")
group.value = alg.getTempFilename()
alg.addParameter(group)

command = 'i.group group={} input={}'.format(
group.value,
','.join([alg.exportedLayers[f] for f in rastersList])
)
alg.commands.append(command)

# Handle POINT File
gcp = alg.getParameterFromName('gcp')
extFileName = gcp.value
destPath = path.join(Grass7Utils.grassMapsetFolder(),
'PERMANENT',
'group', group.value,
'POINTS')
copyFile(alg, extFileName, destPath)
alg.parameters.remove(gcp)

# Add a target destination for our group
command = "i.target group={} location=TARGET mapset=PERMANENT".format(group.value)
alg.commands.append(command)

# remove output
output = alg.getOutputFromName('output')
alg.removeOutputFromName('output')

# Add an extension
#extension = getParameterFromString("ParameterString|extension|Output raster map(s) suffix|None|False|False")
#extension.value = "rectified"
#alg.addParameter(extension)

# modify parameters values
alg.processCommand()

# Re-add input rasters
alg.addParameter(rasters)
alg.addParameter(gcp)
alg.addParameter(crs)

# Re-add output
alg.addOutput(output)


def processOutputs(alg):
# We need to export from the TARGET location
command = "g.mapset location=TARGET mapset=PERMANENT"
alg.commands.append(command)
multipleOutputDir(alg, 'output')

0 comments on commit fe5c6f5

Please sign in to comment.
You can’t perform that action at this time.