1 files changed, 93 insertions, 0 deletions
diff --git a/src/csvbuilder.h b/src/csvbuilder.h
new file mode 100644
index 00000000..f2e0e987
--- /dev/null
+++ b/src/csvbuilder.h
@@ -0,0 +1,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