de.julielab.geneexpbase.genemodel.CoreferenceSet Maven / Gradle / Ivy

Go to download
package de.julielab.geneexpbase.genemodel;

import java.util.Collection;
import java.util.Comparator;
import java.util.TreeSet;

public class CoreferenceSet extends TreeSet {
    private String docId;
    private String id;

    public CoreferenceSet() {
        super(Comparator.comparingInt(CoreferenceExpression::getBegin).thenComparing(System::identityHashCode));
    }

    @Override
    public boolean add(CoreferenceExpression coreferenceExpression) {
        coreferenceExpression.setCoreferenceSet(this);
        return super.add(coreferenceExpression);
    }

    @Override
    public boolean addAll(Collection c) {
        c.forEach(e -> e.setCoreferenceSet(this));
        return super.addAll(c);
    }

    public CoreferenceSet(String docId, String setId) {
        this();
        this.docId = docId;
        this.id = setId;
    }

    public String getDocId() {
        return docId;
    }

    public void setDocId(String docId) {
        this.docId = docId;
    }

    public String getId() {
        return id;
    }

    public void setId(String id) {
        this.id = id;
    }
}