package eu.dnetlib.utils.ontologies;

import java.util.Map;

import com.google.gson.Gson;
import eu.dnetlib.data.mapreduce.util.RelDescriptor;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

/**
 * Created by claudio on 12/12/2016.
 */
public class Ontology {

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

    private String code;
    private String description;

    private Map<String, OntologyTerm> terms;

    public String getCode() {
        return code;
    }

    public Ontology setCode(final String code) {
        this.code = code;
        return this;
    }

    public String getDescription() {
        return description;
    }

    public Ontology setDescription(final String description) {
        this.description = description;
        return this;
    }

    public Map<String, OntologyTerm> getTerms() {
        return terms;
    }

    public Ontology setTerms(final Map<String, OntologyTerm> terms) {
        this.terms = terms;
        return this;
    }

    public String inverseOf(final String termCode) {
        if (getTerms() == null || !getTerms().containsKey(termCode)) {
            //log.warn(String.format("unable to find inverse for term '%s', terms: %s", termCode, getTerms()));
            return null;
        }
        return getTerms().get(termCode).getInverseCode();
    }

    public String inverseOf(final RelDescriptor rd) {
        if (getTerms() == null || !getTerms().containsKey(rd.getRelClass())) {
            return null;
        }
        return getTerms().get(rd.getRelClass()).getInverseCode();
    }

    @Override
    public String toString() {
        return new Gson().toJson(this);
    }

}