Subversion Repositories public iLand

#include "csvfile.h"
#include <QtCore>
#include "helper.h"

CSVFile::CSVFile()
{
    clear();
}

void CSVFile::clear()
{
    mColCount = mRowCount = -1;
    mHasCaptions = true;
    mCaptions.clear();
    mRows.clear();
}

bool CSVFile::loadFile(const QString &fileName)
{
    clear();
    QString content = Helper::loadTextFile(fileName);
    if (content.isEmpty()) {
        qDebug() << "CSVFile::loadFile" << fileName << "does not exist or is empty.";
        return false;
    }
    // split into rows
    mRows = content.split("\n", QString::SkipEmptyParts);
    if (mRows.count()==0)
        return false;

    for (int i=0;i<mRows.count();i++)
        mRows[i] = mRows[i].trimmed();

    // detect separator
    QString first = mRows.first();
    int c_tab = first.count('\t');
    int c_semi = first.count(';');
    int c_comma = first.count(',');
    if (c_tab+c_semi+c_comma == 0) {
        qDebug() << "CSVFile::loadFile: cannot recognize separator. first line:" << first;
        return false;
    }
    mSeparator="\t";
    if (c_tab > c_semi && c_tab>c_comma) mSeparator="\t";
    if (c_semi > c_tab && c_semi>c_comma) mSeparator=";";
    if (c_comma > c_tab && c_comma>c_semi) mSeparator=",";

    // captions
    if (mHasCaptions) {
        mCaptions = first.split(mSeparator);
        mRows.pop_front(); // drop the first line
    } else {
        // create pseudo captions
        mCaptions = first.split(mSeparator);
        for (int i=0;i<mCaptions.count();i++)
            mCaptions[i] = QString("c%1").arg(i);
    }
    mColCount = mCaptions.count();
    mRowCount = mRows.count();
    mStreamingMode = false;
    return true;
}

QVariant CSVFile::cell(const int row, const int col)
{
    if (mStreamingMode)
        return QVariant();

    if (row<0 || row>=mRowCount || col<0 || col>mColCount) {
        qDebug() << "CSVFile::cell: invalid index: row col:" << row << col << ". Size is:" << mRowCount << mColCount;
        return QVariant();
    }
    QStringList line = mRows[row].split(mSeparator);
    QVariant result;
    if (col<line.count()) {
        result = line[col];
    }
    return result;
}

bool CSVFile::openFile(const QString &fileName)
{
    mStreamingMode = true;
    return false;
}