MDWF_Calibrate_d.py

"""
***************************************************************************
   MDWF_Calibrate_d.py
-------------------------------------
    Copyright (C) 2014 TIGER-NET (www.tiger-net.org)

***************************************************************************
* This plugin is part of the Water Observation Information System (WOIS)  *
* developed under the TIGER-NET project funded by the European Space      *
* Agency as part of the long-term TIGER initiative aiming at promoting    *
* the use of Earth Observation (EO) for improved Integrated Water         *
* Resources Management (IWRM) in Africa.                                  *
*                                                                         *
* WOIS is a free software i.e. 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 3 of the License,       *
* or (at your option) any later version.                                  *
*                                                                         *
* WOIS is distributed in the hope that it will be useful, but WITHOUT ANY * 
* WARRANTY; without even the implied warranty of MERCHANTABILITY or       *
* FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License   *
* for more details.                                                       *
*                                                                         *
* You should have received a copy of the GNU General Public License along *
* with this program.  If not, see <http://www.gnu.org/licenses/>.         *
***************************************************************************
"""

import os
from datetime import date, timedelta, datetime
import numpy
import subprocess
from PyQt4 import QtGui
from processing.core.GeoAlgorithmExecutionException import GeoAlgorithmExecutionException
from processing.core.parameters import *
from SWATAlgorithm import SWATAlgorithm
from ModelFile import ModelFile
from SWAT_PEST_specs import SWAT_PEST_specs

class MDWF_Calibrate_d(SWATAlgorithm):

    PEST_mode = "PEST_mode"
    SRC_FOLDER = "SRC_FOLDER"
    CONTROL_FILE = "CONTROL_FILE"

    def __init__(self):
        super(MDWF_Calibrate_d, self).__init__(__file__)

    def defineCharacteristics(self):
        self.name = "5.8 - Sensitivity analysis and calibration of SWAT model with PEST (MDWF) - run PEST"
        self.group = "Model development workflow (MDWF)"
        self.addParameter(ParameterSelection(MDWF_Calibrate_d.PEST_mode, "Calibration algorithm", ['PEST','SCEUA_P','CMAES_P'], False))
        self.addParameter(ParameterFile(MDWF_Calibrate_d.SRC_FOLDER, "Select model source folder", True))
        self.addParameter(ParameterFile(MDWF_Calibrate_d.CONTROL_FILE, "Select PEST control file", False, False))


    def processAlgorithm(self, progress):
        PEST_mode = self.getParameterValue(MDWF_Calibrate_d.PEST_mode)
        SRC_FOLDER = self.getParameterValue(MDWF_Calibrate_d.SRC_FOLDER)
        CONTROL_FILE = self.getParameterValue(MDWF_Calibrate_d.CONTROL_FILE)
        CONTROL_FILE = os.path.split(CONTROL_FILE)[1]
        pestspecs = SWAT_PEST_specs()

        # Writing PEST batch file
        batname = SRC_FOLDER + os.sep + 'WOIS_PEST_run.bat'
        batfile = open(batname,'w')
        batfile.writelines('@ECHO OFF\r\n')
        batfile.writelines('cd ' + SRC_FOLDER + '\r\n')
        if PEST_mode == 0:
            batfile.writelines('"'+pestspecs.PESTexeFolder + os.sep + 'pest.exe" ' + CONTROL_FILE + '\r\n')
        elif PEST_mode ==1:
            batfile.writelines('"'+pestspecs.PESTexeFolder + os.sep + 'sceua_p.exe" ' + CONTROL_FILE + '\r\n')
        else:
            batfile.writelines('"'+pestspecs.PESTexeFolder + os.sep + 'cmaes_p.exe" ' + CONTROL_FILE + '\r\n')
        batfile.writelines('@PAUSE\r\n')
        batfile.close()

        # Running PEST
        currpath = os.getcwd()
        os.chdir(SRC_FOLDER)
        runres = subprocess.call(batname)
        os.chdir(currpath)