angry1980.neo4j.louvain.LouvainResult Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of audio-neo4j Show documentation

There is a newer version: 0.0.10

package angry1980.neo4j.louvain;

import it.unimi.dsi.fastutil.ints.Int2ObjectMap;
import it.unimi.dsi.fastutil.ints.Int2ObjectOpenHashMap;
import it.unimi.dsi.fastutil.ints.IntArrayList;

import java.util.Collections;
import java.util.List;

public class LouvainResult {

    private Int2ObjectMap layerMap;

    public LouvainResult() {
        this.layerMap = new Int2ObjectOpenHashMap<>();
    }

    public LouvainLayer layer(int level) {
        if (!layerMap.containsKey(level)) {
            this.layerMap.put(level, new LouvainLayer(level));
        }
        return this.layerMap.get(level);
    }

    public List layers() {
        List keys = new IntArrayList(layerMap.keySet());
        Collections.sort(keys);
        return keys;
    }

}