csvReader.h

Go to the documentation of this file.

/*
 * Copyright (c) 2019, NVIDIA CORPORATION. All rights reserved.
 *
 * Permission is hereby granted, free of charge, to any person obtaining a
 * copy of this software and associated documentation files (the "Software"),
 * to deal in the Software without restriction, including without limitation
 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
 * and/or sell copies of the Software, and to permit persons to whom the
 * Software is furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
 * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
 * DEALINGS IN THE SOFTWARE.
 */
 
#ifndef __CSV_READER_H_
#define __CSV_READER_H_
 
#include <stdio.h>
#include <stdlib.h>
#include <errno.h>
 
#include <string>
#include <vector>
#include <iostream>
 
 
class csvData
{
public:
        // constructors
        csvData( char* str )                                                    { string = str; }
        csvData( const char* str )                                              { string = str; }
        csvData( std::string& str )                                             { string = str; }
 
        // assignment
        template<typename T> csvData( T value )                         { set(value); }
        template<typename T> void set( T value )                        { string = std::to_string(value); }
 
        // cast to string
        inline operator std::string&()                                  { return string; }
        inline operator const std::string&() const                      { return string; }
        inline operator const char*() const                             { return string.c_str(); }
        
        // cast to number
        inline operator int() const                                             { return toInt(); }
        inline operator float() const                                   { return toFloat(); }
        inline operator double() const                                  { return toDouble(); }
 
        // convert to number (return true if valid)
        inline bool toInt( int* value ) const;
        inline bool toFloat( float* value ) const;
        inline bool toDouble( double* value ) const;
 
        // convert to number (valid->false on error)
        inline int toInt( bool* valid=NULL ) const;
        inline float toFloat( bool* valid=NULL ) const;
        inline double toDouble( bool* valid=NULL ) const;
 
        // stream insertion/extraction operators
        inline friend std::istream& operator >> (std::istream& in, csvData& obj); 
        inline friend std::ostream& operator << (std::ostream& out, const csvData& obj);
        
        // split string by delimiters into list of tokens
        inline static std::vector<csvData> Parse( const char* str, const char* delimiters=",;\t " );
 
        // fill list of tokens with string split by delimiters
        inline static bool Parse( std::vector<csvData>& data, const char* str, const char* delimiters=",;\t " );
 
        // data storage
        std::string string;
};
 
 
class csvReader
{
public:
        // constructor/destructor
        csvReader( const char* filename, const char* delimiters=",;\t " );
        ~csvReader();
 
        // open
        inline static csvReader* Open( const char* filename, const char* delimiters=",;\t " );
 
        // close
        inline void Close();
 
        // is open and not EOF
        inline bool IsOpen() const;
        inline bool IsClosed() const;
 
        // read line, return list of tokens
        inline std::vector<csvData> Read();
        inline std::vector<csvData> Read( const char* delimiters );
 
        // read line, fill list of tokens
        inline bool Read( std::vector<csvData>& data );
        inline bool Read( std::vector<csvData>& data, const char* delimiters );
 
        // set default delimiters
        inline void SetDelimiters( const char* delimiters );
 
        // retrieve default delimiters
        inline const char* GetDelimiters() const;
 
        // retrieve the filename
        inline const char* GetFilename() const;
 
        // maximum line length
        const size_t MaxLineLength=2048;
 
private:
        FILE* mFile;
 
        std::string mFilename;
        std::string mDelimiters;
};
 
 
// internal functions
#include "csvReader.hpp"
 
#endif