br.com.edu.dbpediaspotlight.db.java Source code

Introduction

Here is the source code for br.com.edu.dbpediaspotlight.db.java

Source

/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */
package br.com.edu.dbpediaspotlight;

import org.apache.commons.httpclient.Header;
import org.apache.commons.httpclient.methods.GetMethod;
import org.dbpedia.spotlight.exceptions.AnnotationException;
import org.dbpedia.spotlight.model.DBpediaResource;
import org.dbpedia.spotlight.model.Text;
import org.json.JSONArray;
import org.json.JSONException;
import org.json.JSONObject;

import java.io.UnsupportedEncodingException;
import java.net.URLEncoder;
import java.util.LinkedList;
import java.util.List;

/**
 *
 * @author angelo
 */

//English - http://spotlight.sztaki.hu:2222/rest/annotate
//German - http://spotlight.sztaki.hu:2226/rest/annotate
//Dutch - http://spotlight.sztaki.hu:2232/rest/annotate
//French - http://spotlight.sztaki.hu:2225/rest/annotate
//Italian - http://spotlight.sztaki.hu:2230/rest/annotate
//Russian - http://spotlight.sztaki.hu:2227/rest/annotate
//Spanish - http://spotlight.sztaki.hu:2231/rest/annotate
//Portuguese - http://spotlight.sztaki.hu:2228/rest/annotate
//Hungarian - http://spotlight.sztaki.hu:2229/rest/annotate
//and Turkish - http://spotlight.sztaki.hu:2235/rest/annotate
public class db extends AnnotationClient {

    //private static String  API_URL    = "http://spotlight.dbpedia.org:80/";
    private static String API_URL = "http://spotlight.sztaki.hu:2222/";
    private static double CONFIDENCE = 0.0;
    private static int SUPPORT = 0;
    private static String powered_by = "non";
    //private static String spotter ="Default";//"LingPipeSpotter"=Annotate all spots 
    //AtLeastOneNounSelector"=No verbs and adjs.    
    //"CoOccurrenceBasedSelector" =No 'common words'
    //"NESpotter"=Only Per.,Org.,Loc.
    private static String disambiguator = "Default";//Default ;Occurrences=Occurrence-centric;Document=Document-centric
    private static String showScores = "yes";

    @SuppressWarnings("static-access")
    public void configiration(double CONFIDENCE, int SUPPORT, String powered_by, String spotter,
            String disambiguator, String showScores) {
        this.CONFIDENCE = CONFIDENCE;
        this.SUPPORT = SUPPORT;
        this.powered_by = powered_by;
        //this.spotter=spotter;
        this.disambiguator = disambiguator;
        this.showScores = showScores;
    }

    public List<DBpediaResource> extract(Text text) throws AnnotationException {
        LOG.info("Querying API.");
        String spotlightResponse;
        try {
            String Query = API_URL + "rest/annotate/?" + "confidence=" + CONFIDENCE + "&support=" + SUPPORT
                    + "&spotter=Default" + "&disambiguator=" + disambiguator + "&showScores=" + showScores
                    + "&powered_by=" + powered_by + "&text=" + URLEncoder.encode(text.text(), "utf-8");
            LOG.info(Query);

            GetMethod getMethod = new GetMethod(Query);
            getMethod.addRequestHeader(new Header("Accept", "application/json"));
            spotlightResponse = request(getMethod);

        } catch (UnsupportedEncodingException e) {
            throw new AnnotationException("Could not encode text.", e);
        }

        assert spotlightResponse != null;
        JSONObject resultJSON = null;
        JSONArray entities = null;

        try {
            resultJSON = new JSONObject(spotlightResponse);
            entities = resultJSON.getJSONArray("Resources");
        } catch (JSONException e) {
            throw new AnnotationException("Received invalid response from DBpedia Spotlight API.");
        }

        LinkedList<DBpediaResource> resources = new LinkedList<DBpediaResource>();
        if (entities != null)
            for (int i = 0; i < entities.length(); i++) {
                try {
                    JSONObject entity = entities.getJSONObject(i);
                    resources.add(new DBpediaResource(entity.getString("@URI"),
                            Integer.parseInt(entity.getString("@support"))));
                } catch (JSONException e) {
                    LOG.error("JSON exception " + e);
                }
            }
        return resources;
    }
}