r_blend_rgb.py 2.2 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152
  1. """
  2. ***************************************************************************
  3. r_blend_rgb.py
  4. --------------
  5. Date : February 2016
  6. Copyright : (C) 2016 by Médéric Ribreux
  7. Email : medspx at medspx dot fr
  8. ***************************************************************************
  9. * *
  10. * This program is free software; you can redistribute it and/or modify *
  11. * it under the terms of the GNU General Public License as published by *
  12. * the Free Software Foundation; either version 2 of the License, or *
  13. * (at your option) any later version. *
  14. * *
  15. ***************************************************************************
  16. """
  17. __author__ = 'Médéric Ribreux'
  18. __date__ = 'February 2016'
  19. __copyright__ = '(C) 2016, Médéric Ribreux'
  20. import os
  21. from grassprovider.Grass7Utils import Grass7Utils
  22. def processInputs(alg, parameters, context, feedback):
  23. if 'first' and 'second' in alg.exportedLayers:
  24. return
  25. # Use v.in.ogr
  26. for name in ['first', 'second']:
  27. alg.loadRasterLayerFromParameter(name, parameters, context, False, None)
  28. alg.postInputs(context)
  29. def processCommand(alg, parameters, context, feedback):
  30. # We need to remove all outputs
  31. alg.processCommand(parameters, context, feedback, True)
  32. def processOutputs(alg, parameters, context, feedback):
  33. createOpt = alg.parameterAsString(parameters, alg.GRASS_RASTER_FORMAT_OPT, context)
  34. metaOpt = alg.parameterAsString(parameters, alg.GRASS_RASTER_FORMAT_META, context)
  35. # Export each color raster
  36. colors = ['red', 'green', 'blue']
  37. for color in colors:
  38. fileName = os.path.normpath(
  39. alg.parameterAsOutputLayer(parameters, 'output_{}'.format(color), context))
  40. outFormat = Grass7Utils.getRasterFormatFromFilename(fileName)
  41. alg.exportRasterLayer('blended.{}'.format(color[0]),
  42. fileName, True, outFormat, createOpt, metaOpt)