csvwriter.hpp

Go to the documentation of this file.

#ifndef CSVWRITER_HPP
#define CSVWRITER_HPP
#include <QObject>
#include <QString>
#define DEFAULT_ESCAPE_CHARACTER = '"' 
#define DEFAULT_SEPARATOR = ','         
#define DEFAULT_QUOTE_CHARACTER = '"'   
#define NO_QUOTE_CHARACTER = '\u0000';  
#define NO_ESCAPE_CHARACTER = '\u0000'; 
#define DEFAULT_LINE_END = "\n";        
public class CSVWriter
{

public:
    CSVWriter(Writer writer)
    {
        this(writer, DEFAULT_SEPARATOR, DEFAULT_QUOTE_CHARACTER,
            DEFAULT_ESCAPE_CHARACTER, DEFAULT_LINE_END);
    }

    CSVWriter(Writer writer, char separator, char quotechar, char escapechar, String lineEnd)
    {
        this.pw = new PrintWriter(writer);
        this.separator = separator;
        this.quotechar = quotechar;
        this.escapechar = escapechar;
        this.lineEnd = lineEnd;
    }


    void writeNext(QString nextLine)
    {
        if ( nextLine == NULL ) return;

        QString sb;
        for ( int i = 0; i < nextLine.length(); ++i )
        {
            if ( i != 0 ) sb.append(separator);

            QString nextElement = nextLine[i];
            if ( nextElement == NULL ) continue;
            if ( quotechar !=  NO_QUOTE_CHARACTER ) sb.append(quotechar);

            for ( int j = 0; j < nextElement.length(); ++j )
            {
                char nextChar = nextElement.mid(j,1);
                if ( (escapechar != NO_ESCAPE_CHARACTER) &&
                     (nextChar   == quotechar)
                   )
                {
                    sb.append(escapechar).append(nextChar);
                }
                else if ( ( escapechar != NO_ESCAPE_CHARACTER ) &&
                          ( nextChar == escapechar )
                        )
                {
                    sb.append(escapechar).append(nextChar);
                }
                else
                {
                    sb.append(nextChar);
                }
            }
            if ( quotechar != NO_QUOTE_CHARACTER ) sb.append(quotechar);
        }

        sb.append(lineEnd);
        pw.write(sb.toString());

    }

    void flush() throws IOException
    {
        pw.flush();
    }

    void close() throws IOException
    {
        pw.flush();
        pw.close();
    }


private:
    PrintWriter pw;
    char separator;
    char quotechar;
    char escapechar;
    QString lineEnd;
}
#endif // CSVWRITER_HPP