src/main/java/de/ids_mannheim/korap/response/match/MatchIdentifier.java

package de.ids_mannheim.korap.response.match;

import java.util.*;
import java.util.regex.*;

public class MatchIdentifier extends DocIdentifier {
    private int startPos, endPos = -1;

    private ArrayList<int[]> pos = new ArrayList<>(8);

    // TODO: "contains" is necessary for a compatibility bug in Kustvakt
    Pattern idRegex = Pattern.compile("^(?:match-|contains-)"
            + "(?:([^!]+?)[!\\.])?" + "([^!]+)-p([0-9]+)-([0-9]+)"
            + "((?:\\(-?[0-9]+\\)-?[0-9]+--?[0-9]+)*)" + "(?:c.+?)?$");
    Pattern posRegex = Pattern.compile("\\(([0-9]+)\\)([0-9]+)-([0-9]+)");


    public MatchIdentifier () {};


    public MatchIdentifier (String id) {

        // Replace for legacy reasons with incompatible versions of Kustvakt
        id = id.replaceAll("^(contains-|match-)([^-!_\\.]+?)!\\2_", "$1$2_");

        Matcher matcher = idRegex.matcher(id);
        if (matcher.matches()) {

            // <legacy>
            // and test compatibility
            if (id.contains("!") || !id.contains("_")) {
                this.setCorpusID(matcher.group(1));
                this.setDocID(matcher.group(2));
            }
            // </legacy>
            else {
                // this.getCorpusID() + "." + this.getDocID()
                this.setTextSigle(matcher.group(1) + '.' + matcher.group(2));
            };

            this.setStartPos(Integer.parseInt(matcher.group(3)));
            this.setEndPos(Integer.parseInt(matcher.group(4)));

            if (matcher.group(5) != null) {
                matcher = posRegex.matcher(matcher.group(5));
                while (matcher.find()) {
                    this.addPos(Integer.parseInt(matcher.group(2)),
                            Integer.parseInt(matcher.group(3)),
                            Integer.parseInt(matcher.group(1)));
                };
            };
        };
    };


    public int getStartPos () {
        return this.startPos;
    };


    public void setStartPos (int pos) {
        if (pos >= 0)
            this.startPos = pos;
    };


    public int getEndPos () {
        return this.endPos;
    };


    public void setEndPos (int pos) {
        if (pos >= 0)
            this.endPos = pos;
    };


    public void addPos (int start, int end, int number) {
        if (start >= 0 && end >= 0 && number >= 0)
            this.pos.add(new int[] { start, end, number });
    };


    public ArrayList<int[]> getPos () {
        return this.pos;
    };


    public String toString () {
        StringBuilder sb = new StringBuilder("match-");

        if (this.docID == null) {
            if (this.textSigle == null)
                return null;

            sb.append(this.textSigle);
        }

        // Get prefix string corpus/doc
        // LEGACY
        else if (this.corpusID != null) {
            sb.append(this.corpusID).append('!').append(this.docID);
        }
        else {
            sb.append(this.docID);
        };

        sb.append('-').append(this.getPositionString());
        return sb.toString();
    };


    public String getPositionString () {
        StringBuilder sb = new StringBuilder();
        sb.append('p').append(this.startPos).append('-').append(this.endPos);

        // Get Position information
        for (int[] i : this.pos) {
            sb.append('(').append(i[2]).append(')');
            sb.append(i[0]).append('-').append(i[1]);
        };

        return sb.toString();
    };
};