source: trunk/source/visualization/HepRep/src/XMLWriter.cc@ 1183

Last change on this file since 1183 was 834, checked in by garnier, 17 years ago

import all except CVS
File size: 8.1 KB
RevLine 
[834]1// Copyright FreeHEP, 2005.
2
3#include "cheprep/config.h"
4
5#include <cstdio>
6
7#include "cheprep/DefaultHepRepAttValue.h"
8#include "cheprep/XMLWriter.h"
9
10using namespace std;
11
12/**
13 * @author Mark Donszelmann
14 * @version $Id: XMLWriter.cc,v 1.12 2005/06/02 21:28:45 duns Exp $
15 */
16namespace cheprep {
17
18XMLWriter::XMLWriter(ostream* out, string indentString, string defaultNameSpace)
19 : AbstractXMLWriter(defaultNameSpace) {
20 writer = new IndentPrintWriter(out);
21 writer->setIndentString(indentString);
22 closed = false;
23 dtdName = "";
24}
25
26XMLWriter::~XMLWriter() {
27 writer->close();
28 delete writer;
29}
30
31void XMLWriter::close() {
32 closeDoc();
33 writer->close();
34}
35
36void XMLWriter::openDoc(string version, string encoding, bool standalone) {
37 string indentString = writer->getIndentString();
38 writer->setIndentString(indentString);
39
40// if (!XMLCharacterProperties.validVersionNum(version)) throw new RuntimeException("Invalid version number: "+version);
41 *writer << "<?xml version=\"" << version.c_str() << "\" ";
42 if (encoding.compare("") != 0) {
43// if (!XMLCharacterProperties.validEncName(encoding)) throw new RuntimeException("Invalid encoding name: "+encoding);
44 *writer << "encoding=\"" << encoding.c_str() << "\" ";
45 }
46 if (standalone) {
47 *writer << "standalone=\"yes\" ";
48 }
49 *writer << "?>";
50 *writer << endl;
51 writer->setIndentString(indentString);
52}
53
54void XMLWriter::referToDTD(string name, string pid, string ref) {
55 if (dtdName != "") {
56 cerr << "XMLWriter::ReferToDTD cannot be called twice" << endl;
57 }
58 dtdName = name;
59 *writer << "<!DOCTYPE " << name.c_str() << " PUBLIC \"" << pid.c_str() << "\" \"" << ref.c_str() << "\">" << endl;
60}
61
62void XMLWriter::referToDTD(string name, string system) {
63 if (dtdName != "") {
64 cerr << "XMLWriter::ReferToDTD cannot be called twice";
65 }
66 dtdName = name;
67 *writer << "<!DOCTYPE " << name.c_str() << " SYSTEM \"" << system.c_str() << "\">" << endl;
68}
69
70void XMLWriter::closeDoc(bool force) {
71 if (!closed) {
72 if (!openTags.empty()) {
73 if (!force) cerr << "Not all tags were closed before closing XML document:" << endl;
74 while (!openTags.empty()) {
75 if (force) {
76 closeTag();
77 } else {
78 cerr << " </" << openTags.top().c_str() << ">" << endl;
79 openTags.pop();
80 }
81 }
82 }
83 closed = true;
84 }
85}
86
87void XMLWriter::printComment(string comment) {
88 if (comment.find("--") != string::npos) {
89 cerr << "XMLWriter::printComment '--' sequence not allowed in comment" << endl;
90 }
91 *writer << "<!--" << normalizeText(comment).c_str() << "-->" << endl;
92}
93
94void XMLWriter::printPlain(string text) {
95 *writer << text.c_str();
96}
97
98void XMLWriter::print(string text) {
99 *writer << normalizeText(text).c_str();
100}
101
102void XMLWriter::println(string text) {
103 print(text);
104 *writer << endl;
105}
106
107void XMLWriter::openTag(string name) {
108 checkNameValid(name);
109 if (openTags.empty() && dtdName.compare("") && dtdName.compare(name)) {
110 cerr << "XMLWriter::openTag(), First tag: '" << name << "' not equal to DTD id: '" << dtdName << "'" << endl;
111 }
112 *writer << "<" << name.c_str();
113 printAttributes(name.length());
114 *writer << ">" << endl;
115 writer->indent();
116 openTags.push(name);
117}
118
119void XMLWriter::closeTag() {
120 if (openTags.empty()) {
121 writer->close();
122 cerr << "XMLWriter::closeTag(), No open tags" << endl;
123 }
124 string name = openTags.top();
125 openTags.pop();
126 writer->outdent();
127 *writer << "</" << name.c_str() << ">" << endl;
128}
129
130void XMLWriter::printTag(string name) {
131 checkNameValid(name);
132 *writer << "<" << name.c_str();
133 printAttributes(name.length());
134 *writer << "/>" << endl;
135}
136
137void XMLWriter::setAttribute(string name, char* value) {
138 setAttribute(name, (string)value);
139}
140
141void XMLWriter::setAttribute(string name, string value) {
142 attributes[name] = value;
143 // NOTE: never set type here
144}
145
146void XMLWriter::setAttribute(std::string name, std::vector<double> value) {
147 if (name == "value") setAttribute("type", (std::string)"Color");
148 setAttribute(name, DefaultHepRepAttValue::getAsString(value));
149}
150
151void XMLWriter::setAttribute(std::string name, int64 value) {
152 if (name == "value") setAttribute("type", (std::string)"long");
153 setAttribute(name, DefaultHepRepAttValue::getAsString(value));
154}
155
156void XMLWriter::setAttribute(std::string name, int value) {
157 if (name == "showlabel") {
158 string label = DefaultHepRepAttValue::toShowLabel(value);
159 setAttribute("showlabel", label);
160 } else {
161 if (name == "value") setAttribute("type", (std::string)"int");
162 setAttribute(name, DefaultHepRepAttValue::getAsString(value));
163 }
164}
165
166void XMLWriter::setAttribute(std::string name, bool value) {
167 if (name == "value") setAttribute("type", (std::string)"boolean");
168 setAttribute(name, DefaultHepRepAttValue::getAsString(value));
169}
170
171void XMLWriter::setAttribute(string name, double value) {
172 if (name == "value") setAttribute("type", (std::string)"double");
173 setAttribute(name, DefaultHepRepAttValue::getAsString(value));
174}
175
176void XMLWriter::printAttributes(int tagLength) {
177 int width = tagLength + 1;
178 bool extraIndent = false;
179 for (map<string,string>::iterator i = attributes.begin(); i != attributes.end(); i++) {
180 string key = i->first;
181 checkNameValid(key);
182 string value = normalize(i->second);
183 int length = key.length() + value.length() + 3;
184 if (width > 0 && width + length + 2*writer->getIndent() > 60) {
185 width = 0;
186 *writer << endl;
187 if (!extraIndent) {
188 writer->indent();
189 extraIndent = true;
190 }
191 } else {
192 width += length;
193 *writer << " ";
194 }
195 *writer << key.c_str() << "=\"" << value.c_str() << "\"";
196 }
197 attributes.clear();
198 if (extraIndent) writer->outdent();
199}
200
201string XMLWriter::normalize(string s) {
202 string str = "";
203 char buffer[20];
204
205 int len = s.length();
206 for (int i = 0; i < len; i++) {
207 char ch = s[i];
208 switch (ch) {
209 case '<': {
210 str.append("&lt;");
211 break;
212 }
213 case '>': {
214 str.append("&gt;");
215 break;
216 }
217 case '&': {
218 str.append("&amp;");
219 break;
220 }
221 case '"': {
222 str.append("&quot;");
223 break;
224 }
225 case '\r':
226 case '\n': {
227 sprintf(buffer, "&#%ud", ch);
228 str.append(buffer);
229 str.append(";");
230 break;
231 }
232 default: {
233// if (ch > 0x00FF) {
234// sprintf(buffer, "&#x%4.4x", ch);
235// str.append(buffer);
236// str.append(";");
237// } else {
238 str.append(&ch, 1);
239// }
240 }
241 }
242 }
243
244 return str;
245}
246
247string XMLWriter::normalizeText(string s) {
248 string str = "";
249
250 int len = s.length();
251 for (int i = 0; i < len; i++) {
252 char ch = s[i];
253 switch (ch) {
254 case '<': {
255 str.append("&lt;");
256 break;
257 }
258 case '>': {
259 str.append("&gt;");
260 break;
261 }
262 case '&': {
263 str.append("&amp;");
264 break;
265 }
266 default: {
267// if (ch > 0x00FF) {
268// sprintf(buffer, "&#x%4.4x", ch);
269// str.append(buffer);
270// str.append(";");
271// } else {
272 str.append(&ch, 1);
273// }
274 }
275 }
276 }
277 return str;
278}
279
280void XMLWriter::checkNameValid(string) {
281// Could be added.
282// if (!XMLCharacterProperties.validName(s)) throw new RuntimeException("Invalid name: "+s);
283}
284
285
286} // cheprep
Note: See TracBrowser for help on using the repository browser.