org.openstreetmap.atlas.geography.sharding.LocationToShardCommand Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of atlas Show documentation

"Library to load OSM data into an Atlas format"

There is a newer version: 7.0.8

package org.openstreetmap.atlas.geography.sharding;

import java.util.Set;

import org.openstreetmap.atlas.geography.Location;
import org.openstreetmap.atlas.utilities.collections.Iterables;
import org.openstreetmap.atlas.utilities.runtime.Command;
import org.openstreetmap.atlas.utilities.runtime.CommandMap;

/**
 * @author matthieun
 */
public class LocationToShardCommand extends Command
{
    private static final Switch SHARDING = new Switch<>("sharding", "The sharding tree",
            Sharding::forString, Optionality.REQUIRED);
    private static final Switch LOCATION = new Switch<>("location",
            "The location to check as \"latitude,longitude\"", Location::forString,
            Optionality.OPTIONAL);
    private static final Switch WKT_POINT = new Switch<>("wktPoint",
            "The location to check as a WKT point", Location::forWkt, Optionality.OPTIONAL);

    public static void main(final String[] args)
    {
        new LocationToShardCommand().run(args);
    }

    @Override
    protected int onRun(final CommandMap command)
    {
        final Sharding sharding = (Sharding) command.get(SHARDING);
        Location location = (Location) command.get(LOCATION);
        if (location == null)
        {
            location = (Location) command.get(WKT_POINT);
        }
        if (location == null)
        {
            System.err.println("No location found! Make sure to use either the -"
                    + LOCATION.getName() + " or -" + WKT_POINT.getName() + " switch.");
            return 0;
        }
        final Set shards = Iterables.asSet(sharding.shardsCovering(location));
        if (shards.size() > 0)
        {
            System.out.println("Shard(s) covering " + location.toCompactString() + ":");
            shards.forEach(
                    shard -> System.out.println(shard.getName() + "; " + shard.bounds().toWkt()));
        }
        else
        {
            System.err.println("No shard found!");
            return 1;
        }
        return 0;
    }

    @Override
    protected SwitchList switches()
    {
        return new SwitchList().with(SHARDING, LOCATION, WKT_POINT);
    }
}