1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
|
#ifndef CSVBUILDER_H
#define CSVBUILDER_H
#include <vector>
#include <QString>
#include <QVariant>
#include <QTextStream>
class CSVException : public std::exception {
public:
CSVException(const QString &text)
: std::exception(), m_Message(text.toLocal8Bit()) {}
virtual const char* what() const throw()
{ return m_Message.constData(); }
private:
QByteArray m_Message;
};
class CSVBuilder
{
public:
enum EFieldType {
TYPE_INTEGER,
TYPE_STRING,
TYPE_FLOAT
};
enum EQuoteMode {
QUOTE_NEVER,
QUOTE_ONDEMAND,
QUOTE_ALWAYS
};
enum ELineBreak {
BREAK_LF,
BREAK_CRLF,
BREAK_CR
};
public:
CSVBuilder(QIODevice *target);
~CSVBuilder();
void setFieldSeparator(char sep);
void setLineBreak(ELineBreak lineBreak);
void setEscapeMode(EFieldType type, EQuoteMode mode);
void setFields(const std::vector<std::pair<QString, EFieldType> > &fields);
void setDefault(const QString &field, const QVariant &value);
void writeHeader();
void setRowField(const QString &field, const QVariant &value);
void writeRow();
void addRow(const std::map<QString, QVariant> &data);
private:
const char *lineBreak();
const char separator();
void fieldValid();
void checkFields(const std::vector<QString> &fields);
void checkValue(const QString &field, const QVariant &value);
void writeData(const std::map<QString, QVariant> &data, bool check);
void quoteInsert(QTextStream &stream, int value);
void quoteInsert(QTextStream &stream, float value);
void quoteInsert(QTextStream &stream, const QString &value);
private:
QTextStream m_Out;
char m_Separator;
ELineBreak m_LineBreak;
std::map<EFieldType, EQuoteMode> m_QuoteMode;
std::vector<QString> m_Fields;
std::map<QString, EFieldType> m_FieldTypes;
std::map<QString, QVariant> m_Defaults;
std::map<QString, QVariant> m_RowBuffer;
};
#endif // CSVBUILDER_H
|
