/* FichothequeLib_Tools - Copyright (c) 2013-2025 Vincent Calame - Exemole
 * Logiciel libre donné sous triple licence :
 * 1) selon les termes de la CeCILL V2
 * 2) selon les termes de l’EUPL V.1.1
 * 3) selon les termes de la GNU GPLv3
 * Voir le fichier licences.txt
 */


package net.fichotheque.tools.extraction.corpus;

import java.util.SortedMap;
import java.util.TreeMap;
import net.fichotheque.corpus.FicheMeta;
import net.fichotheque.extraction.def.FicheFilter;
import net.fichotheque.extraction.def.GroupClause;
import net.fichotheque.extraction.run.FicheGroup;
import static net.fichotheque.tools.extraction.corpus.GroupEngine.toGroupArray;
import net.fichotheque.junction.Junction;


/**
 *
 * @author Vincent Calame
 */
class WeightGroupEngine extends GroupEngine {

    private final SortedMap<Integer, GroupBuilder> treeMap = new TreeMap<Integer, GroupBuilder>();
    private final GroupClause groupClause;
    private final GroupEngineProvider subGroupEngineProvider;

    WeightGroupEngine(GroupClause groupClause, GroupEngineProvider subGroupEngineProvider) {
        this.groupClause = groupClause;
        this.subGroupEngineProvider = subGroupEngineProvider;
    }

    @Override
    public void add(FicheMeta ficheMeta, FicheFilter ficheFilter, Junction junction) {
        int weight = 0;
        if (junction != null) {
            weight = junction.getTieList().get(0).getWeight();
        }
        Integer weightInteger = weight;
        GroupBuilder groupBuilder = treeMap.get(weightInteger);
        if (groupBuilder == null) {
            groupBuilder = GroupBuilder.newInstance(weightInteger.toString(), groupClause.getTagNameInfo(), subGroupEngineProvider);
            treeMap.put(weightInteger, groupBuilder);
        }
        groupBuilder.add(ficheMeta, ficheFilter, junction);
    }

    @Override
    public FicheGroup[] toFicheGroupArray() {
        return toGroupArray(treeMap.values(), groupClause, null);
    }

}
