CSVReader.cpp

Go to the documentation of this file.

// ==============================================================================================
// Microflow 3D, http://www.microflow.pwr.edu.pl/
// Created by Roman Szafran on 17.05.19.
// Copyright (c) 2019 Wroclaw University of Science and Technology.
// Distributed under the Apache License, Version 2.0. You may obtain a copy of the License at
// http://www.apache.org/licenses/LICENSE-2.0 or see accompanying file license.txt.
// Redistributions of source code must retain the above copyright and license notice.
// ==============================================================================================

#include "CSVReader.h"


void MF::RW::CSVReader::DataReadFromCSVFile(const std::string & DataFilePath, const std::vector <std::string> & VariableNameList, std::vector<double>* pDataArray) {

    std::string buffer = "";
    std::string substring = "";
    std::size_t pos1 = 0;
    std::size_t pos2 = 0;
    std::size_t pos3 = 0;
    unsigned int NodeNumber = 0;
    unsigned int ColumnNr = 0;
    unsigned int ColumnNumber[VariableNameList.size()];
    double DoubleDataTable[VariableNameList.size()];

    // Initialization
    for (unsigned long int i = 0; i < VariableNameList.size(); i++) {
        ColumnNumber[i] = -1;
        DoubleDataTable[i] = 0;
    }

    // File open
    std::ifstream FileInput(DataFilePath);
    if (!FileInput.good()) {
        std::cout << "Data file not found: " << DataFilePath << std::endl;
        exit(EXIT_FAILURE);
    }

    // Header processing
    // -------------------------------------------------------------------

    // Column numbers distinction
    std::getline(FileInput, buffer); //One line reading
    for (unsigned long int i = 0; i < VariableNameList.size(); i++) {
        pos1 = 0;
        pos2 = 0;
        pos3 = 0;
        substring = "";
        while (substring != VariableNameList.at(i)) {
            pos2 = buffer.find(",", pos1);
            if (pos2 != std::string::npos)
                substring = buffer.substr(pos1, pos2 - pos1);
            else
                substring = buffer.substr(pos1);
            pos3 = 0;
            while (pos3 != std::string::npos) // quotation marks removal
            {
                pos3 = substring.find("\"");
                if (pos3 != std::string::npos)
                    substring = substring.erase(pos3, 1);
            }
            if ((substring != VariableNameList.at(i)) & (pos2 == std::string::npos)) // If file end and data name not found
            {
                std::cout << "Column name " << VariableNameList.at(i) << " not found in the file: " << DataFilePath << std::endl;
                exit(EXIT_FAILURE);
            }

            pos1 = pos2 + 1;
            ColumnNumber[i]++;
        }
    }

    // Data processing
    //-----------------------------------------------------

    while (!FileInput.eof()) {
        std::getline(FileInput, buffer); //One line reading
        pos1 = buffer.find_first_of("-0123456789"); //Check if the line starts from number
        if (pos1 == 0) {
            pos1 = 0;
            pos2 = 0;
            ColumnNr = 0;
            while (pos2 != std::string::npos) {
                pos2 = buffer.find(",", pos1);
                if (pos2 != std::string::npos)
                    substring = buffer.substr(pos1, pos2 - pos1);
                else
                    substring = buffer.substr(pos1);
                for (unsigned long int i = 0; i < VariableNameList.size(); i++)
                    if (ColumnNumber[i] == ColumnNr)
                        DoubleDataTable[i] = boost::lexical_cast<double>(substring); // string to number
                ColumnNr++;
                pos1 = pos2 + 1;
            }
            for (unsigned long int i = 0; i < VariableNameList.size(); i++)
                pDataArray->push_back(DoubleDataTable[i]);
            NodeNumber++;
        }
    }

    std::cout << NodeNumber << " data values are read from the file: " << DataFilePath << std::endl;
    FileInput.close();
}