D-Net

package eu.dnetlib.utils.ontologies;

import java.util.ArrayList;

import java.util.HashMap;

import java.util.List;

import java.util.Map;

import com.google.common.collect.Maps;

import com.google.gson.GsonBuilder;

import eu.dnetlib.data.mapreduce.util.RelDescriptor;

import org.apache.commons.logging.Log;

import org.apache.commons.logging.LogFactory;

/**

 * Created by sandro on 12/13/16.

 */

public class Ontologies extends HashMap<String, Ontology> {

    private static final Log log = LogFactory.getLog(Ontologies.class);

    private Map<String, List<OntologyTerm>> inverse = Maps.newHashMap();

    public String inverseOf(final RelDescriptor rd) {

        if (!containsKey(rd.getRelType())) {

            log.warn(String.format("unable to find ontology '%s'", rd.getRelType()));

            return null;

        }

        return get(rd.getRelType()).inverseOf(rd.getRelClass());

    }

    public List<OntologyTerm> getTerms(final String termCode) {

        if (inverse.isEmpty()) {

            initInverse();

        }

        return inverse.get(termCode);

    }

    private void initInverse() {

        log.info("initialising inverse Ontology terms");

        values().forEach(o -> o.getTerms().values().forEach(t -> {

            if (!inverse.containsKey(t.getCode())) {

                inverse.put(t.getCode(), new ArrayList<>());

            }

            inverse.get(t.getCode()).add(t);

        }));

    }

    public String toJson() {

        return toJson(false);

    }

    public String toJson(boolean pretty) {

        final GsonBuilder gson = new GsonBuilder();

        if (pretty) {

            gson.setPrettyPrinting();

        }

        return gson.create().toJson(this);

    }

    @Override

    public String toString() {

        return toJson();

    }

}