/*
* Copyright (c) 2016, Gerd Petermann
*
* 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;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStream;
import java.io.RandomAccessFile;
import java.io.Reader;
import java.nio.channels.FileChannel;
import java.util.HashMap;
import java.util.List;
import java.util.concurrent.ArrayBlockingQueue;
import java.util.concurrent.BlockingQueue;
import org.xmlpull.v1.XmlPullParserException;
import crosby.binary.file.BlockInputStream;
import it.unimi.dsi.fastutil.shorts.ShortArrayList;
import uk.me.parabola.splitter.parser.BinaryMapParser;
import uk.me.parabola.splitter.parser.O5mMapParser;
import uk.me.parabola.splitter.parser.OSMXMLParser;
/**
* A class which stores parameters needed to process input (OSM) files
*
* @author Gerd Petermann
*
*/
public class OSMFileHandler
{
/** list of OSM input files to process */
private List<String> filenames
;
// for faster access on blocks in pbf files
private final HashMap<String, ShortArrayList
> blockTypeMap =
new HashMap<>();
// for faster access on blocks in o5m files
private final HashMap<String,
long[]> skipArrayMap =
new HashMap<>();
// Whether or not the source OSM file(s) contain strictly nodes first, then
// ways, then rels,
// or they're all mixed up. Running with mixed enabled takes longer.
private boolean mixed
;
private int maxThreads =
1;
public void setFileNames
(List<String> filenames
) {
this.
filenames = filenames
;
}
public void setMixed
(boolean f
) {
mixed = f
;
}
public void setMaxThreads
(int maxThreads
) {
this.
maxThreads = maxThreads
;
}
public boolean process
(MapProcessor processor
) {
// create appropriate parser for each input file
for (String filename : filenames
) {
System.
out.
println("Processing " + filename
);
processor.
startFile();
try {
if (filename.
endsWith(".o5m")) {
File file =
new File(filename
);
try (RandomAccessFile raf =
new RandomAccessFile(file,
"r");
FileChannel fileChannel = raf.
getChannel()) {
long[] skipArray = skipArrayMap.
get(filename
);
O5mMapParser o5mParser =
new O5mMapParser
(processor, fileChannel, skipArray
);
o5mParser.
parse();
if (skipArray ==
null) {
skipArray = o5mParser.
getNextSkipArray();
skipArrayMap.
put(filename, skipArray
);
}
}
} else if (filename.
endsWith(".pbf")) {
// Is it a binary file?
File file =
new File(filename
);
ShortArrayList blockTypes = blockTypeMap.
get(filename
);
BinaryMapParser binParser =
new BinaryMapParser
(processor, blockTypes,
1);
try (InputStream stream =
new FileInputStream(file
)) {
BlockInputStream blockinput =
(new BlockInputStream
(stream, binParser
));
blockinput.
process();
if (blockTypes ==
null) {
// remember this file
blockTypes = binParser.
getBlockList();
blockTypeMap.
put(filename, blockTypes
);
}
}
} else {
// No, try XML.
try (Reader reader = Utils.
openFile(filename, maxThreads
> 1)) {
OSMXMLParser parser =
new OSMXMLParser
(processor, mixed
);
parser.
setReader(reader
);
parser.
parse();
}
}
} catch (FileNotFoundException e
) {
System.
out.
println(e
);
throw new SplitFailedException
("ERROR: file " + filename +
" was not found");
} catch (XmlPullParserException e
) {
e.
printStackTrace();
throw new SplitFailedException
("ERROR: file " + filename +
" is not a valid OSM XML file");
} catch (IllegalArgumentException e
) {
e.
printStackTrace();
throw new SplitFailedException
("ERROR: file " + filename +
" contains unexpected data");
} catch (IOException e
) {
e.
printStackTrace();
throw new SplitFailedException
("ERROR: file " + filename +
" caused I/O exception");
} catch (RuntimeException e
) {
e.
printStackTrace();
throw new SplitFailedException
("ERROR: file " + filename +
" caused exception");
}
}
return processor.
endMap();
}
RuntimeException exception =
null;
public boolean execute
(MapProcessor processor
) {
if (maxThreads ==
1)
return process
(processor
);
// use two threads
BlockingQueue<OSMMessage
> queue =
new ArrayBlockingQueue<>(10);
QueueProcessor queueProcessor =
new QueueProcessor
(queue, processor
);
// start producer thread
new Thread("producer for " + processor.
getClass().
getSimpleName()) {
@
Override
public void run
() {
try {
process
(queueProcessor
);
} catch (SplitFailedException e
) {
try {
queue.
put(new OSMMessage
(OSMMessage.
Type.
EXIT));
exception = e
;
} catch (InterruptedException e1
) {
e1.
printStackTrace();
}
}
}
}.
start();
boolean done = processor.
consume(queue
);
if (exception
!=
null)
throw exception
;
return done
;
}
}