package eu.dnetlib.data.transform;

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.graph.utils.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.getOntologyCode())) {
        	log.debug(String.format("unable to find ontology '%s'", rd.getOntologyCode()));
	        return null;
        }
        return get(rd.getOntologyCode()).inverseOf(rd.getTermCode());
    }

    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();
	}

}