org.mycore.datamodel.classifications2.mapper.MCRCategoryMapperBase Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of mycore-base Show documentation
There is a newer version: 2024.05
/*
 * This file is part of ***  M y C o R e  ***
 * See http://www.mycore.de/ for details.
 *
 * MyCoRe is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * MyCoRe is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with MyCoRe.  If not, see .
 */

package org.mycore.datamodel.classifications2.mapper;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collection;
import java.util.List;
import java.util.Set;
import java.util.SortedSet;
import java.util.TreeSet;
import java.util.stream.Collectors;

import org.mycore.datamodel.classifications2.MCRCategoryID;

/**
 * @author Frank Lützenkirchen
 */
public abstract class MCRCategoryMapperBase {

    public Set map(Collection input) {
        SortedSet output = new TreeSet<>();

        for (MCRCategoryID categoryID : input) {
            Set mapped = collectMappings(categoryID);
            output.addAll(mapped);
        }

        return output;
    }

    private Set collectMappings(MCRCategoryID categoryID) {
        Set mapped = new TreeSet<>();

        for (MCRCategoryID parent : resolveParentOrSelf(categoryID)) {
            for (MCRCategoryID mapping : getMappings(parent)) {
                if (!alreadyContainsCategoryOfSameClassification(mapped, mapping)) {
                    mapped.add(mapping);
                }
            }
        }

        return mapped;
    }

    private boolean alreadyContainsCategoryOfSameClassification(Collection collection,
        MCRCategoryID candidate) {
        String classificationID = candidate.getRootID();
        return collection.stream().map(MCRCategoryID::getRootID).anyMatch(classificationID::equals);
    }

    private List getMappings(MCRCategoryID categoryID) {
        String mappingRule = getMappingRule(categoryID);
        String[] mappings = mappingRule.split("\\s+");

        return Arrays.stream(mappings).map(this::buildMappedID).collect(Collectors.toList());
    }

    private MCRCategoryID buildMappedID(String mapping) {
        int pos = mapping.indexOf(":");
        String mappedClassificationID = mapping.substring(0, pos);
        String mappedCategoryID = mapping.substring(pos + 1);
        return new MCRCategoryID(mappedClassificationID, mappedCategoryID);
    }

    private List resolveParentOrSelf(MCRCategoryID childID) {
        List parentOrSelf = new ArrayList<>();
        parentOrSelf.add(childID);
        addParentsToList(childID, parentOrSelf);
        return parentOrSelf;
    }

    protected abstract void addParentsToList(MCRCategoryID childID, List list);

    protected abstract String getMappingRule(MCRCategoryID categoryID);
}