ApproxState.java

package mhtree;

import messif.operations.Approximate;

public class ApproxState {
    protected int limit;
    protected int objectsChecked;
    protected int bucketsVisited;

    protected ApproxState() {
    }

    protected ApproxState(int limit) {
        this.limit = limit;
    }

    public static ApproxState create(Approximate limits, MHTree mhTree) {
        switch (limits.getLocalSearchType()) {
            case PERCENTAGE:
                return new ApproxStateObjects(Math.round((float) mhTree.getObjectCount() * (float) limits.getLocalSearchParam() / 100f));
            case ABS_OBJ_COUNT:
                return new ApproxStateObjects(limits.getLocalSearchParam());
            case DATA_PARTITIONS:
                return new ApproxStateBuckets(limits.getLocalSearchParam());
            default:
                return new ApproxState();
        }
    }

    public void update(LeafNode node) {
        objectsChecked += node.getObjectCount();
        bucketsVisited++;
    }

    public boolean stop() {
        return false;
    }

    private static class ApproxStateBuckets extends ApproxState {
        private ApproxStateBuckets(int limit) {
            super(limit);
        }

        @Override
        public boolean stop() {
            return bucketsVisited >= limit;
        }
    }

    private static class ApproxStateObjects extends ApproxState {
        private ApproxStateObjects(int limit) {
            super(limit);
        }

        @Override
        public boolean stop() {
            return objectsChecked >= limit;
        }
    }
}