mingzailao
/
mitsuba
mirror of https://github.com/Quartenia/mitsuba.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

even nicer XML output

metadata
Wenzel Jakob 2010-11-22 09:18:15 +01:00
parent cc8761db46
commit 0015f5e53b
1 changed files with 38 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
src/converter/converter.cpp
View File

@ -27,7 +27,7 @@
#include <xercesc/parsers/XercesDOMParser.hpp> #include <xercesc/parsers/XercesDOMParser.hpp>
#include <xercesc/framework/MemBufInputSource.hpp> #include <xercesc/framework/MemBufInputSource.hpp>
#include <xercesc/framework/Wrapper4InputSource.hpp> #include <xercesc/framework/Wrapper4InputSource.hpp>
#include <xercesc/framework/LocalFileFormatTarget.hpp> #include <xercesc/framework/MemBufFormatTarget.hpp>
#include <xercesc/util/XMLUni.hpp> #include <xercesc/util/XMLUni.hpp>
#include <mitsuba/core/fresolver.h> #include <mitsuba/core/fresolver.h>
#include <mitsuba/core/fstream.h> #include <mitsuba/core/fstream.h>
@ -231,12 +231,46 @@ void GeometryConverter::convert(const fs::path &inputFile,
serConf->setParameter(XMLUni::fgDOMErrorHandler, &errorHandler); serConf->setParameter(XMLUni::fgDOMErrorHandler, &errorHandler);
if (serConf->canSetParameter(XMLUni::fgDOMWRTFormatPrettyPrint, true)) if (serConf->canSetParameter(XMLUni::fgDOMWRTFormatPrettyPrint, true))
serConf->setParameter(XMLUni::fgDOMWRTFormatPrettyPrint, true); serConf->setParameter(XMLUni::fgDOMWRTFormatPrettyPrint, true);
if (serConf->canSetParameter(XMLUni::fgDOMWRTXercesPrettyPrint, true))
serConf->setParameter(XMLUni::fgDOMWRTXercesPrettyPrint, true);
DOMLSOutput *output = impl->createLSOutput(); DOMLSOutput *output = impl->createLSOutput();
XMLFormatTarget *target = new LocalFileFormatTarget(outputFile.file_string().c_str()); MemBufFormatTarget *target = new MemBufFormatTarget();
output->setByteStream(target); output->setByteStream(target);
serializer->write(doc, output); serializer->write(doc, output);
const XMLByte *content = target->getRawBuffer();
std::ostringstream oss;
/* Turn leading spaces into tabs */
bool newline = true;
int numSpaces = 0;
for (size_t i=0; i<target->getLen(); ++i) {
char data = content[i];
switch (data) {
case ' ':
if (newline)
numSpaces++;
else
oss << data;
break;
case '\r':
case '\n':
oss << data;
newline = true;
numSpaces = 0;
break;
default:
if (newline) {
for (int i=0; i<numSpaces/2; ++i)
oss << '\t';
}
oss << data;
newline = false;
numSpaces = 0;
break;
}
}
fs::ofstream os(outputFile);
os << oss.str();
os.close();
delete output; delete output;
delete target; delete target;
delete wrapper; delete wrapper;