Subversion Repositories public iLand

/********************************************************************************************
**    iLand - an individual based forest landscape and disturbance model
**    http://iland.boku.ac.at
**    Copyright (C) 2009-  Werner Rammer, Rupert Seidl
**
**    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 3 of the License, or
**    (at your option) any later version.
**
**    This program 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/>.
********************************************************************************************/

#include "global.h"
#include "climateconverter.h"
#include "csvfile.h"

#include <QJSEngine>
#include <QJSValue>
#include <QtSql>

/** @class ClimateConverter
  Converts text-file-based data into the iLand climate data format.
  For the iLand format see the wiki (ClimateFormat). For each column (i.e. year,month, day,
  temp, prec, rad, vpd), an expression providing access to the columns of the input file calculates
  the respective output value. Propertes tableName and fileName define the input file and the
  name of the output table (located in the "climate"-database of iLand) respectively.
*/
/** @group script
  @class ClimateConverter
  This is the Scripting related documentation for the ClimateConverter tool.
*/
//Q_SCRIPT_DECLARE_QMETAOBJECT(ClimateConverter, QObject*)
void ClimateConverter::addToScriptEngine(QJSEngine &engine)
{
    // about this kind of scripting magic see: http://qt.nokia.com/developer/faqs/faq.2007-06-25.9557303148
    //QJSValue cc_class = engine.scriptValueFromQMetaObject<ClimateConverter>();
    // the script name for the object is "ClimateConverter".
    QObject *cc = new ClimateConverter();
    QJSValue cc_class = engine.newQObject(cc);
    engine.globalObject().setProperty("ClimateConverter", cc_class);
}

ClimateConverter::ClimateConverter(QObject *)
{
    mCaptions = true;
    bindExpression(mExpYear, 0);
    bindExpression(mExpMonth, 1);
    bindExpression(mExpDay, 2);

    bindExpression(mExpTemp, 3);
    bindExpression(mExpMinTemp, 4);

    bindExpression(mExpPrec, 5);
    bindExpression(mExpRad, 6);
    bindExpression(mExpVpd, 7);
}



void ClimateConverter::bindExpression(Expression &expr, int index)
{
    expr.setExpression(QString("c%1").arg(index) ); // "cX" is the default expression
    for (int i=0;i<10;i++)
        mVars[index*10 + i] = expr.addVar( QString("c%1").arg(i) );
}

void ClimateConverter::run()
{
    mExpYear.setExpression(mYear);
    mExpMonth.setExpression(mMonth);
    mExpDay.setExpression(mDay);

    mExpTemp.setExpression(mTemp);
    mExpMinTemp.setExpression(mMinTemp);
    mExpPrec.setExpression(mPrec);
    mExpRad.setExpression(mRad);
    mExpVpd.setExpression(mVpd);
    // prepare output database
    if (mDatabase.isEmpty()) {
        qDebug() << "ClimateConverter: database is empty!";
        return;
    }
    QSqlDatabase db = QSqlDatabase::addDatabase("QSQLITE","cctemp");
    db.setDatabaseName(mDatabase);
    if (!db.open()) {
        qDebug() << "ClimateConverter: invalid database: " << db.lastError().text();
        return;
    }

    QSqlQuery creator(db);
    if (mTableName.isEmpty()) {
        qDebug() << "ClimateConverter::run: invalid climate database or table name.";
    }
    QString sql=QString("CREATE TABLE %1 ( " \
                "year INTEGER, month INTEGER, day INTEGER, " \
                "temp REAL, min_temp REAL, prec REAL, rad REAL, vpd REAL)").arg(mTableName);

    QString drop=QString("drop table if exists %1").arg(mTableName);

    creator.exec(drop); // drop table (if exists)
    if (creator.lastError().isValid()) {
        qDebug() << "ClimateConverter: Sql-Error (drop):" << creator.lastError().text();
        return;
    }
    creator.exec(sql); // (re-)create table
    if (creator.lastError().isValid()) {
        qDebug() << "ClimateConverter: Sql-Error (create table):" << creator.lastError().text();
        return;
    }

    // prepare insert statement
    sql = QString("insert into %1 (year, month, day, temp, min_temp, prec, rad, vpd) values (?,?,?, ?,?,?,?,?)").arg(mTableName);
    creator.prepare(sql);
    if (creator.lastError().isValid()) {
        qDebug() << "ClimateConverter: Sql-Error (prepare):" << creator.lastError().text();
        return;
    }
    // load file
    if (mFileName.isEmpty()) {
        qDebug() << "ClimateConverter::run: empty filename.";
        return;
    }
    CSVFile file;
    file.setHasCaptions(mCaptions);
    file.loadFile(mFileName);
    if (!file.rowCount()) {
        qDebug() << "ClimateConverter::run: cannot load file:" << mFileName;
        return;
    }

    // do this for each row
    double value;
    int year, month, day;
    double temp, min_temp, prec, rad, vpd;
    int rows=0;
    db.transaction();
    for (int row=0;row<file.rowCount(); row++) {
        // fetch values from input file
        for (int col=0;col<file.colCount(); col++) {
            value = file.value(row, col).toDouble();
            // store value in each of the expression variables
            for (int j=0;j<8;j++)
                *(mVars[j*10 + col]) = value; // store in the locataion mVars[x] points to.
        }
        // calculate new values....
        year = (int)mExpYear.execute();
        month = (int)mExpMonth.execute();
        day = (int)mExpDay.execute();
        temp = mExpTemp.execute();
        min_temp = mExpMinTemp.execute();
        prec = mExpPrec.execute();
        rad = mExpRad.execute();
        vpd = mExpVpd.execute();
        //qDebug() << year << month << day << temp << prec << rad << vpd;
        // bind values
        creator.bindValue(0,year);
        creator.bindValue(1,month);
        creator.bindValue(2,day);
        creator.bindValue(3,temp);
        creator.bindValue(4,min_temp);
        creator.bindValue(5,prec);
        creator.bindValue(6,rad);
        creator.bindValue(7,vpd);
        creator.exec();
        rows++;
        if (creator.lastError().isValid()) {
            qDebug() << "ClimateConverter: Sql-Error:" << creator.lastError().text();
            return;
        }
    }
    db.commit();
    creator.clear();
    db.close();
    QSqlDatabase::removeDatabase("cctemp");
    qDebug() << "ClimateConverter::run: processing complete." << rows << "rows inserted.";

}