src/main/java/de/ids_mannheim/korap/query/object/KoralGroup.java

package de.ids_mannheim.korap.query.object;

import java.util.ArrayList;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;

import de.ids_mannheim.korap.query.serialize.MapBuilder;
import de.ids_mannheim.korap.query.object.KoralObject;
import de.ids_mannheim.korap.query.object.KoralOperation;
import de.ids_mannheim.korap.query.object.KoralType;

/** Definition of koral:group in KoralQuery.
 * @author margaretha
 * 
 */
public class KoralGroup implements KoralObject {

    private static final KoralType type = KoralType.GROUP;

    private KoralOperation operation;

    private boolean inOrder = false;
    private List<KoralObject> operands;
    private List<KoralDistance> distances;
    private List<Frame> frames;
    private KoralBoundary boundary;
    
//    private int[] classIn;
//    private int classOut;
//    private int classRefCheck;
//    private int classRefOp;
//    private boolean exclude = false;
    
    public KoralGroup (KoralOperation operation) {
        this.operation = operation;
    }

    public boolean isInOrder() {
        return inOrder;
    }

    public void setInOrder(boolean inOrder) {
        this.inOrder = inOrder;
    }

    public List<KoralObject> getOperands() {
		return operands;
	}
    
    public void setOperands(List<KoralObject> operands) {
		this.operands = operands;
	}

    public KoralOperation getOperation() {
        return operation;
    }

    public void setOperation(KoralOperation operation) {
        this.operation = operation;
    }

    public List<KoralDistance> getDistances() {
        return distances;
    }

    public void setDistances(List<KoralDistance> distances) {
        this.distances = distances;
    }
    
    public List<Frame> getFrames() {
		return frames;
	}

	public void setFrames(List<Frame> frames) {
		this.frames = frames;
	}

    public KoralBoundary getBoundary() {
        return boundary;
    }

    public void setBoundary(KoralBoundary boundary) {
        this.boundary = boundary;
    }

    @Override
    public Map<String, Object> buildMap() {
        Map<String, Object> map = new LinkedHashMap<String, Object>();
        map.put("@type", type.toString());
        map.put("operation", operation.toString());

        if (getDistances() != null) {
            map.put("inOrder", isInOrder());
            List<Map<String, Object>> distanceList = new ArrayList<Map<String, Object>>();
            for (KoralDistance d : distances) {
                distanceList.add(d.buildMap());
            }
            map.put("distances", distanceList);
        }

        List<Map<String, Object>> operandList = new ArrayList<Map<String, Object>>();
        for (KoralObject o : operands) {
            operandList.add(MapBuilder.buildQueryMap(o));
        }
        map.put("operands", operandList);

        if (boundary != null) {
            map.put("boundary", boundary.buildMap());
        }
        return map;
    }

    public enum Frame{
		SUCCEDS("succeeds"), SUCCEDS_DIRECTLY("succeedsDirectly"), OVERLAPS_RIGHT("overlapsRight"), 
		ALIGNS_RIGHT("alignsRight"), IS_WITHIN("isWithin"), STARTS_WITH("startsWith"), 
		MATCHES("matches"), ALIGNS_LEFT("alignsLeft"), IS_AROUND("isAround"), ENDS_WITH("endsWith"),
		OVERLAPS_LEFT("overlapsLeft"), PRECEEDS_DIRECTLY("precedesDirectly"), PRECEDES("precedes");
		
		private String value;
		Frame(String value) {
			this.value = value;
		}
		
		@Override
		public String toString() {
			return "frame:"+value;
		}
	}
}