// This software is released into the Public Domain. See copying.txt for details.
package org.openstreetmap.osmosis.dataset.v0_6;
import java.io.File;
import java.util.Map;
import org.openstreetmap.osmosis.core.container.v0_6.EntityContainer;
import org.openstreetmap.osmosis.dataset.v0_6.impl.DatasetStore;
import org.openstreetmap.osmosis.dataset.v0_6.impl.DatasetStoreFileManager;
import org.openstreetmap.osmosis.dataset.v0_6.impl.PermanentFileDatasetStoreFileManager;
import org.openstreetmap.osmosis.core.task.v0_6.Sink;
/**
* Receives input data as a stream and builds a dataset containing all of the
* data.
*
* @author Brett Henderson
*/
public class WriteDataset implements Sink {
private DatasetStoreFileManager fileManager;
private DatasetStore store;
/**
* Creates a new instance.
*
* @param directory
* The directory to store all data files in.
* @param enableWayTileIndex
* If true a tile index is created for ways, otherwise a node-way
* index is used.
*/
public WriteDataset(File directory, boolean enableWayTileIndex) {
fileManager = new PermanentFileDatasetStoreFileManager(directory);
store = new DatasetStore(fileManager, enableWayTileIndex);
}
/**
* {@inheritDoc}
*/
public void initialize(Map<String, Object> metaData) {
store.initialize(metaData);
}
/**
* {@inheritDoc}
*/
public void process(EntityContainer entityContainer) {
store.process(entityContainer);
}
/**
* {@inheritDoc}
*/
public void complete() {
store.complete();
}
/**
* {@inheritDoc}
*/
public void release() {
// We must release the store last because downstream tasks must be able
// to release store readers first.
store.release();
// We must release the file manager after the store to ensure all open
// files are closed.
fileManager.release();
}
}