csvfile.h

Go to the documentation of this file.

/********************************************************************************************
**    iLand - an individual based forest landscape and disturbance model
**    http://iland-model.org
**    Copyright (C) 2009-  Werner Rammer
**
**    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/>.
********************************************************************************************/
 
#ifndef CSVFILE_H
#define CSVFILE_H
 
#include <QObject>
#include <QStringList>
#include <QJSEngine>
class CSVFile : public QObject
{
    Q_OBJECT
    Q_PROPERTY(bool captions WRITE setHasCaptions READ hasCaptions) 
    Q_PROPERTY(bool flat WRITE setFlat READ flat) 
    Q_PROPERTY(int colCount READ colCount)
    Q_PROPERTY(int rowCount READ rowCount)
public:
    CSVFile(QObject *parent=0);
    CSVFile(const QString &fileName) {     mHasCaptions = true; mFlat = false; mFixedWidth=false; loadFile(fileName);} 
    // actions
    bool openFile(const QString &fileName); 
    QVariant colValue(const int col); 
    bool next(); 
    // properties
    bool streamingMode() const { return mStreamingMode; } 
    bool hasCaptions() const { return mHasCaptions; } 
    bool flat() const { return mFlat; } 
    int rowCount() const { return mRowCount; } 
    int colCount() const { return mColCount; } 
    bool isEmpty() const { return mIsEmpty; } 
    QStringList captions() const { return mCaptions; } 
    QStringList column(const int col) const; 
    QVariantList values(const int row) const; 
    // setters
    void setHasCaptions(const bool hasCaps) { mHasCaptions = hasCaps; }
    void setFixedWidth(const bool hasFixedWidth) { mFixedWidth = hasFixedWidth; }
    void setFlat(const bool isflat) { mFlat = isflat; }
    static void addToScriptEngine(QJSEngine &engine); // called during setup of ScriptEngine
public slots:
    bool loadFile(const QString &fileName); 
    bool loadFromString(const QString &content); 
    QString columnName(const int col) { if (col<mColCount) return mCaptions[col]; return QString(); } 
    int columnIndex(const QString &columnName) const { return mCaptions.indexOf(columnName); } 
    // value function with a column name
    QVariant value(const int row, const QString column_name) const { return value(row, columnIndex(column_name)); }
 
    QVariant value(const int row, const int col) const;
    QVariant row(const int row); 
    QJSValue jsValue(const int row, const int col) const;
    QJSValue jsValue(const int row, const QString column_name) const { return jsValue(row, columnIndex(column_name)); }
 
    void setValue(const int row, const int col, QVariant value); 
    void saveFile(const QString &fileName); 
 
private:
    void clear();
    bool mIsEmpty;
    bool mHasCaptions;
    bool mFixedWidth;
    bool mFlat;
    bool mStreamingMode;
    QStringList mCaptions;
    QStringList mRows;
    QString mSeparator;
    int mRowCount;
    int mColCount;
};
 
#endif // CSVFILE_H