/*
* Copyright (c) 2009, Steve Ratcliffe
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License version 3 as
* published by the Free Software Foundation.
*
* This program is distributed in the hope that it will be useful, but
* WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* General Public License for more details.
*/
package uk.me.parabola.splitter.writer;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.io.OutputStreamWriter;
import java.io.Writer;
import java.nio.charset.StandardCharsets;
import java.text.DecimalFormat;
import java.text.DecimalFormatSymbols;
import java.util.Iterator;
import java.util.List;
import java.util.Locale;
import java.util.zip.GZIPOutputStream;
import it.unimi.dsi.fastutil.longs.LongArrayList;
import uk.me.parabola.splitter.Element;
import uk.me.parabola.splitter.Node;
import uk.me.parabola.splitter.OsmBounds;
import uk.me.parabola.splitter.Relation;
import uk.me.parabola.splitter.Way;
public class OSMXMLWriter
extends AbstractOSMWriter
{
private final DecimalFormat numberFormat =
new DecimalFormat(
"0.#######;-0.#######",
new DecimalFormatSymbols(Locale.
US)
);
private Writer writer
;
public OSMXMLWriter
(File oFile
) {
super(oFile
);
}
public OSMXMLWriter
(String baseName
) {
super(new File(baseName+
".osm.gz"));
}
@
Override
public void initForWrite
() {
try {
FileOutputStream fos =
new FileOutputStream(outputFile
);
OutputStream zos =
new GZIPOutputStream(fos
);
writer =
new OutputStreamWriter(zos, StandardCharsets.
UTF_8);
writeHeader
();
} catch (IOException e
) {
System.
out.
println("Could not open or write file header. Reason: " + e.
getMessage());
throw new RuntimeException(e
);
}
}
private void writeHeader
() throws IOException {
writeString
("<?xml version='1.0' encoding='UTF-8'?>\n");
String apiVersion =
(versionMethod == REMOVE_VERSION
) ? "version='0.5'" :
"version='0.6'";
writeString
("<osm " + apiVersion +
" generator='splitter' upload='false'>\n");
}
@
Override
public void write
(OsmBounds bounds
) throws IOException {
writeString
("<bounds minlat='");
writeLongDouble
(bounds.
getMinLat());
writeString
("' minlon='");
writeLongDouble
(bounds.
getMinLong());
writeString
("' maxlat='");
writeLongDouble
(bounds.
getMaxLat());
writeString
("' maxlon='");
writeLongDouble
(bounds.
getMaxLong());
writeString
("'/>\n");
}
@
Override
public void finishWrite
() {
try {
writeString
("</osm>\n");
flush
();
writer.
close();
writer =
null;
} catch (IOException e
) {
System.
out.
println("Could not write end of file: " + e
);
}
}
@
Override
public void write
(Node node
) throws IOException {
writeString
("<node id='");
writeLong
(node.
getId());
writeString
("' lat='");
writeDouble
(node.
getLat());
writeString
("' lon='");
writeDouble
(node.
getLon());
if (versionMethod
!= REMOVE_VERSION
)
writeString
("' version='" + getWriteVersion
(node
));
if (node.
hasTags()) {
writeString
("'>\n");
writeTags
(node
);
writeString
("</node>\n");
} else {
writeString
("'/>\n");
}
}
@
Override
public void write
(Way way
) throws IOException {
writeString
("<way id='");
writeLong
(way.
getId());
if (versionMethod
!= REMOVE_VERSION
)
writeString
("' version='" + getWriteVersion
(way
));
writeString
("'>\n");
LongArrayList refs = way.
getRefs();
for (int i =
0; i
< refs.
size(); i++
) {
writeString
("<nd ref='");
writeLong
(refs.
get(i
));
writeString
("'/>\n");
}
if (way.
hasTags())
writeTags
(way
);
writeString
("</way>\n");
}
@
Override
public void write
(Relation rel
) throws IOException {
writeString
("<relation id='");
writeLong
(rel.
getId());
if (versionMethod
!= REMOVE_VERSION
)
writeString
("' version='" + getWriteVersion
(rel
));
writeString
("'>\n");
List<Relation.
Member> memlist = rel.
getMembers();
for (Relation.
Member m : memlist
) {
if (m.
getType() ==
null || m.
getRef() ==
0) {
System.
err.
println("Invalid relation member found in relation " + rel.
getId() +
": member type="
+ m.
getType() +
", ref=" + m.
getRef() +
", role=" + m.
getRole() +
". Ignoring this member");
continue;
}
writeString
("<member type='");
writeAttribute
(m.
getType());
writeString
("' ref='");
writeLong
(m.
getRef());
writeString
("' role='");
if (m.
getRole() !=
null) {
writeAttribute
(m.
getRole());
}
writeString
("'/>\n");
}
if (rel.
hasTags())
writeTags
(rel
);
writeString
("</relation>\n");
}
private void writeTags
(Element element
) throws IOException {
Iterator<Element.
Tag> it = element.
tagsIterator();
while (it.
hasNext()) {
Element.
Tag entry = it.
next();
writeString
("<tag k='");
writeAttribute
(entry.
getKey());
writeString
("' v='");
writeAttribute
(entry.
getValue());
writeString
("'/>\n");
}
}
private void writeAttribute
(String value
) throws IOException {
for (int i =
0; i
< value.
length(); i++
) {
char c = value.
charAt(i
);
switch (c
) {
case '\'':
writeString
("'");
break;
case '&':
writeString
("&");
break;
case '<':
writeString
("<");
break;
case '\n':
writeString
("
");
break;
case '\r':
writeString
("
");
break;
case '\t':
writeString
("	");
break;
default:
writeChar
(c
);
}
}
}
private int index
;
private final char[] charBuf =
new char[4096];
private void checkFlush
(int i
) throws IOException {
if (charBuf.
length - index
< i
) {
flush
();
}
}
private void flush
() throws IOException {
writer.
write(charBuf,
0, index
);
index =
0;
}
private void writeString
(String value
) throws IOException {
int start =
0;
int end = value.
length();
int len
;
while ((len = charBuf.
length - index
) < end - start
) {
value.
getChars(start, start + len, charBuf, index
);
start += len
;
index = charBuf.
length;
flush
();
}
value.
getChars(start, end, charBuf, index
);
index += end - start
;
}
/** Write a double to full precision */
private void writeLongDouble
(double value
) throws IOException {
checkFlush
(22);
writeString
(Double.
toString(value
));
}
/**
* Write a double truncated to OSM's 7 digits of precision
*/
private void writeDouble
(double value
) throws IOException {
checkFlush
(22);
// Punt on some annoying specialcases
if (value
< -
200 || value
> 200 ||
(value
> -
1 && value
< 1))
writeString
(numberFormat.
format(value
));
else {
if (value
< 0) {
charBuf
[index++
] =
'-'; // Write directly.
value = -value
;
}
int val =
(int) Math.
round(value
* 10000000);
StringBuilder s =
new StringBuilder(Integer.
toString(val
));
s.
insert(s.
length() -
7,
'.');
writeString
(s.
toString());
}
}
private void writeLong
(long value
) throws IOException {
checkFlush
(20);
writeString
(Long.
toString(value
));
}
private void writeChar
(char value
) throws IOException {
checkFlush
(1);
charBuf
[index++
] = value
;
}
}