csvformatter.h
Go to the documentation of this file.
1 /*
2  * Copyright (C) 2011 Tommi Maekitalo
3  *
4  * This library is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU Lesser General Public
6  * License as published by the Free Software Foundation; either
7  * version 2.1 of the License, or (at your option) any later version.
8  *
9  * As a special exception, you may use this file as part of a free
10  * software library without restriction. Specifically, if other files
11  * instantiate templates or use macros or inline functions from this
12  * file, or you compile this file and link it with other files to
13  * produce an executable, this file does not by itself cause the
14  * resulting executable to be covered by the GNU General Public
15  * License. This exception does not however invalidate any other
16  * reasons why the executable file might be covered by the GNU Library
17  * General Public License.
18  *
19  * This library is distributed in the hope that it will be useful,
20  * but WITHOUT ANY WARRANTY; without even the implied warranty of
21  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
22  * Lesser General Public License for more details.
23  *
24  * You should have received a copy of the GNU Lesser General Public
25  * License along with this library; if not, write to the Free Software
26  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
27  */
28 
29 #ifndef CXXTOOLS_CSVFORMATTER_H
30 #define CXXTOOLS_CSVFORMATTER_H
31 
32 #include <cxxtools/formatter.h>
33 #include <cxxtools/textstream.h>
34 #include <cxxtools/utf8codec.h>
35 #include <cxxtools/string.h>
36 #include <cxxtools/char.h>
37 #include <vector>
38 
39 namespace cxxtools
40 {
41  class CsvFormatter : public Formatter
42  {
43  public:
44  CsvFormatter(std::ostream& os, TextCodec<Char, char>* codec = new Utf8Codec());
45 
47 
48  ~CsvFormatter();
49 
50  void selectColumn(const std::string& title);
51 
52  void selectColumn(const std::string& memberName, const std::string& title);
53 
55  { _delimiter = delimiter; }
56 
58  { _quote = quote; }
59 
60  void lineEnding(const String& le)
61  { _lineEnding = le; }
62 
63  virtual void addValueString(const std::string& name, const std::string& type,
64  const String& value);
65 
66  virtual void beginArray(const std::string& name, const std::string& type);
67 
68  virtual void finishArray();
69 
70  virtual void beginObject(const std::string& name, const std::string& type);
71 
72  virtual void beginMember(const std::string& name);
73 
74  virtual void finishMember();
75 
76  virtual void finishObject();
77 
78  virtual void finish();
79 
80  private:
81  void toCsvData(String& ret, const std::string& type, const String& value);
82  void dataOut();
83 
84  bool _firstline;
85  bool _collectTitles;
86  unsigned _level;
87 
88  Char _delimiter;
89  Char _quote;
90  String _lineEnding;
91 
92  // titles and member names
93  struct Title
94  {
95  std::string _memberName;
96  std::string _title;
97  };
98 
99  std::vector<Title> _titles;
100 
101  std::vector<String> _data;
102  std::string _memberName;
103  TextOStream* _ts;
104  TextOStream& _os;
105  };
106 }
107 
108 #endif // CXXTOOLS_CSVFORMATTER_H