org.wikibrain.spatial.cookbook.CalculateGeographicDistanceBetweenPages Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of wikibrain-spatial Show documentation
There is a newer version: 0.9.1
package org.wikibrain.spatial.cookbook;

import com.google.common.collect.Lists;
import com.vividsolutions.jts.geom.Geometry;
import com.vividsolutions.jts.geom.Point;
import org.geotools.referencing.GeodeticCalculator;


import org.wikibrain.conf.Configurator;
import org.wikibrain.core.cmd.Env;
import org.wikibrain.core.cmd.EnvBuilder;
import org.wikibrain.core.dao.LocalPageDao;
import org.wikibrain.core.lang.LanguageSet;
import org.wikibrain.core.model.LocalPage;
import org.wikibrain.core.model.NameSpace;
import org.wikibrain.core.model.Title;
import org.wikibrain.spatial.dao.SpatialDataDao;
import org.wikibrain.wikidata.WikidataDao;

import java.util.List;

/**
 * Created by bjhecht on 4/7/14.
 */
public class CalculateGeographicDistanceBetweenPages {

    public static void main(String[] args){

        try {

            Env env = EnvBuilder.envFromArgs(args);
            Configurator c = env.getConfigurator();

            SpatialDataDao sdDao = c.get(SpatialDataDao.class);
            WikidataDao wdDao = c.get(WikidataDao.class);
            LocalPageDao lpDao = c.get(LocalPageDao.class);

            LanguageSet loadedLangs = lpDao.getLoadedLanguages();


            String[] pageNames = new String[]{"Minneapolis", "Chicago", "Beijing"};

            List itemIds = Lists.newArrayList();

            // Find the Wikidata item IDs for the local pages corresponding to the page names
            for (String pageName : pageNames){
                LocalPage localPage = lpDao.getByTitle(new Title(pageName, loadedLangs.getBestAvailableEnglishLang(true)), NameSpace.ARTICLE);
                Integer itemId = wdDao.getItemId(localPage);

                itemIds.add(itemId);
            }

            GeodeticCalculator calc = new GeodeticCalculator();

            // get the geometries for each local page/item id and calculate the distance between them
            for (int i = 0; i < itemIds.size(); i++){
                Geometry g1 = sdDao.getGeometry(itemIds.get(i), "wikidata", "earth");
                Point centroid = g1.getCentroid();
                calc.setStartingGeographicPoint(centroid.getX(), centroid.getY());
                for (int j = i; j < itemIds.size(); j++){
                    Geometry g2 = sdDao.getGeometry(itemIds.get(j), "wikidata", "earth");
                    centroid = g2.getCentroid();
                    calc.setDestinationGeographicPoint(centroid.getX(), centroid.getY());

                    String out = String.format("%s to %s is %fkm", pageNames[i], pageNames[j], calc.getOrthodromicDistance()/1000);
                    System.out.println(out);

                }
            }

        }catch(Exception e){
            e.printStackTrace();;
        }


    }

}