EDIT

/**

* Copyright (C) 2015 EDIT

* European Distributed Institute of Taxonomy

* http://www.e-taxonomy.eu

*

* The contents of this file are subject to the Mozilla Public License Version 1.1

* See LICENSE.TXT at the top of this package for the full license terms.

*/

package eu.etaxonomy.cdm.ext.geo;

import java.util.ArrayList;

import java.util.Collection;

import java.util.Collections;

import java.util.HashMap;

import java.util.HashSet;

import java.util.List;

import java.util.Map;

import java.util.Set;

import java.util.UUID;

import org.apache.commons.lang.StringUtils;

import org.apache.log4j.Logger;

import eu.etaxonomy.cdm.api.service.dto.CondensedDistribution;

import eu.etaxonomy.cdm.common.UTF8;

import eu.etaxonomy.cdm.model.common.Language;

import eu.etaxonomy.cdm.model.description.Distribution;

import eu.etaxonomy.cdm.model.description.PresenceAbsenceTerm;

import eu.etaxonomy.cdm.model.location.NamedArea;

/**

 * @author a.kohlbecker

 * @since Jun 24, 2015

 *

 */

public class EuroPlusMedCondensedDistributionComposer extends CondensedDistributionComposerBase {

    @SuppressWarnings("unused")

    private static final Logger logger = Logger.getLogger(EuroPlusMedCondensedDistributionComposer.class);

    private final CondensedDistribution condensedDistribution;

    private static Set<UUID> foreignStatusUuids;

    // these status uuids are special for EuroPlusMed and might also be used

    private final static UUID REPORTED_IN_ERROR_UUID =  UUID.fromString("38604788-cf05-4607-b155-86db456f7680");

    static {

        // ==================================================

        // Mapping as defined in ticket https://dev.e-taxonomy.eu/redmine/issues/3907

        // ==================================================

        statusSymbols = new HashMap<> ();

        // â—� endemic (U+25CF BLACK CIRCLE)

        statusSymbols.put(PresenceAbsenceTerm.ENDEMIC_FOR_THE_RELEVANT_AREA().getUuid(), "\u25CF");

        // Lu native (incl. archaeophytes) TODO status archaeophytes?

        statusSymbols.put(PresenceAbsenceTerm.NATIVE().getUuid(), "");

        statusSymbols.put(PresenceAbsenceTerm.NATIVE_FORMERLY_NATIVE().getUuid(), "");  //wanted? differs from default "ne"

        // ?Lu doubtfully present (U+3F QUESTION MARK)

        statusSymbols.put(PresenceAbsenceTerm.INTRODUCED_PRESENCE_QUESTIONABLE().getUuid(), "?");

        statusSymbols.put(PresenceAbsenceTerm.NATIVE_PRESENCE_QUESTIONABLE().getUuid(), "?");

        statusSymbols.put(PresenceAbsenceTerm.PRESENT_DOUBTFULLY().getUuid(), "?");

        // dLu doubtfully native

        statusSymbols.put(PresenceAbsenceTerm.NATIVE_DOUBTFULLY_NATIVE().getUuid(), "d");

        // -Lu absent but reported in error (U+2D HYPHEN-MINUS)

        statusSymbols.put(PresenceAbsenceTerm.INTRODUCED_REPORTED_IN_ERROR().getUuid(), UTF8.EN_DASH.toString());

        statusSymbols.put(PresenceAbsenceTerm.NATIVE_REPORTED_IN_ERROR().getUuid(), UTF8.EN_DASH.toString());

        statusSymbols.put(REPORTED_IN_ERROR_UUID, "-");

        // †Lu (presumably) extinct (U+2020 DAGGER)

        // no such status in database!!!

        // [Lu] introduced (casual or naturalized) =  introduced, introduced: naturalized

        statusSymbols.put(PresenceAbsenceTerm.INTRODUCED().getUuid(), ""); //wanted? differs from default "i"

        // [aLu] casual alien = introduced: adventitious (casual)

        statusSymbols.put(PresenceAbsenceTerm.CASUAL().getUuid(), "a");

        // [cLu] cultivated

        statusSymbols.put(PresenceAbsenceTerm.CULTIVATED() .getUuid(), "c");

        statusSymbols.put(PresenceAbsenceTerm.INTRODUCED_CULTIVATED().getUuid(), "c");

        // [nLu] naturalized

        statusSymbols.put(PresenceAbsenceTerm.NATURALISED().getUuid(), "n");

        statusSymbols.put(PresenceAbsenceTerm.NATURALISED().getUuid(), "n");

        foreignStatusUuids = new HashSet<UUID>();

        foreignStatusUuids.add(PresenceAbsenceTerm.INTRODUCED().getUuid());

        foreignStatusUuids.add(PresenceAbsenceTerm.NATURALISED().getUuid());

        foreignStatusUuids.add(PresenceAbsenceTerm.CASUAL().getUuid());

        foreignStatusUuids.add(PresenceAbsenceTerm.INTRODUCED_CULTIVATED().getUuid());

        foreignStatusUuids.add(PresenceAbsenceTerm.NATURALISED().getUuid());

        foreignStatusUuids.add(PresenceAbsenceTerm.CULTIVATED().getUuid());

    }

    public EuroPlusMedCondensedDistributionComposer() {

        super();

        replaceCommonAreaLabelStart = true;

        condensedDistribution = new CondensedDistribution();

    }

    @Override

    public CondensedDistribution createCondensedDistribution(Collection<Distribution> filteredDistributions,

            List<Language> langs) {

        //1. group by PresenceAbsenceTerms

        Map<PresenceAbsenceTerm, Collection<NamedArea>> areasByStatus = new HashMap<>();

        for(Distribution d : filteredDistributions) {

            PresenceAbsenceTerm status = d.getStatus();

            if(status == null) {

                continue;

            }

            if(!areasByStatus.containsKey(status)) {

                areasByStatus.put(status, new HashSet<>());

            }

            areasByStatus.get(status).add(d.getArea());

        }

        //2. build the area hierarchy

        for(PresenceAbsenceTerm status : areasByStatus.keySet()) {

            Map<NamedArea, AreaNode> areaNodeMap = new HashMap<>();

            for(NamedArea area : areasByStatus.get(status)) {

                AreaNode node;

                if(!areaNodeMap.containsKey(area)) {

                    // putting area into hierarchy as node

                    node = new AreaNode(area);

                    areaNodeMap.put(area, node);

                } else {

                    //  is parent of another and thus already has a node

                    node = areaNodeMap.get(area);

                }

                NamedArea parent = findParentIn(area, areasByStatus.get(status));

                if(parent != null) {

                    AreaNode parentNode;

                    if(!areaNodeMap.containsKey(parent)) {

                        parentNode = new AreaNode(parent);

                        areaNodeMap.put(parent, parentNode);

                    } else {

                        parentNode = areaNodeMap.get(parent);

                    }

                    parentNode.addSubArea(node);

                }

            }

            //3. find root nodes

            Set<AreaNode>hierarchy = new HashSet<>();

            for(AreaNode node : areaNodeMap.values()) {

                if(!node.hasParent()) {

                    hierarchy.add(node);

                }

            }

            //4. replace the area by the abbreviated representation and add symbols

            for(AreaNode topLevelNode : hierarchy) {

                StringBuilder areaStatusString = new StringBuilder();

                String statusSymbol = statusSymbol(status);

                areaStatusString.append(statusSymbol);

                String areaLabel = topLevelNode.area.getPreferredRepresentation(langs).getAbbreviatedLabel();

                areaStatusString.append(areaLabel);

                if(!topLevelNode.subAreas.isEmpty()) {

                    areaStatusString.append('(');

                    subAreaLabels(langs, topLevelNode.subAreas, areaStatusString, statusSymbol, areaLabel);

                    areaStatusString.append(')');

                }

                if(isForeignStatus(status)) {

                    condensedDistribution.addForeignDistributionItem(status, areaStatusString.toString(), areaLabel);

                } else {

                    condensedDistribution.addIndigenousDistributionItem(status, areaStatusString.toString(), areaLabel);

                }

            }

        }

        //5. order the condensedDistributions alphabetically

        condensedDistribution.sortForeign();

        condensedDistribution.sortIndigenous();

        return condensedDistribution;

    }

    private boolean isForeignStatus(PresenceAbsenceTerm status) {

        return foreignStatusUuids.contains(status.getUuid());

    }

    private void subAreaLabels(List<Language> langs, Collection<AreaNode> nodes, StringBuilder areaString,

            String statusSymbol, String parentLabel) {

        List<String> subAreaLabels = new ArrayList<String>();

        for(AreaNode node : nodes) {

            StringBuilder subAreaString = new StringBuilder();

            subAreaString.append(statusSymbol);

            String areaLabel = node.area.getPreferredRepresentation(langs).getAbbreviatedLabel();

            if (replaceCommonAreaLabelStart){

                String cleanSubAreaLabel = StringUtils.replaceEach(areaLabel, new String[] {parentLabel, "(", ")"}, new String[] {"", "", ""});

                subAreaString.append(cleanSubAreaLabel);

            }else{

                subAreaString.append(areaLabel);

            }

            if(!node.subAreas.isEmpty()) {

                subAreaString.append('(');

                subAreaLabels(langs, node.subAreas, subAreaString, statusSymbol, areaLabel);

                subAreaString.append(')');

            }

            subAreaLabels.add(subAreaString.toString());

        }

        Collections.sort(subAreaLabels);

        areaString.append(StringUtils.join(subAreaLabels, " "));

    }

}