Level.java

package org.activityinfo.core.shared.form.tree;

import com.google.common.collect.Lists;
import org.activityinfo.model.resource.ResourceId;
import org.activityinfo.core.shared.application.ApplicationProperties;
import org.activityinfo.core.shared.criteria.FormClassSet;
import org.activityinfo.model.form.FormClass;
import org.activityinfo.model.form.FormField;
import org.activityinfo.model.type.ReferenceType;

import java.util.List;

/**
* Represents a level within a hierarchy.
*/
public class Level {

    private FormClass formClass;

    ResourceId parentId;
    ResourceId parentFieldId;
    Level parent;
    List<Level> children = Lists.newArrayList();

    Level(FormClass formClass) {
        this.formClass = formClass;
        for(FormField field : formClass.getFields()) {
            if(field.isSubPropertyOf(ApplicationProperties.PARENT_PROPERTY) &&
                    field.getType() instanceof ReferenceType) {
                ReferenceType type = (ReferenceType) field.getType();
                assert type.getRange().size() == 1;
                parentId = type.getRange().iterator().next();
                parentFieldId = field.getId();
            }
        }
    }

    public ResourceId getClassId() {
        return formClass.getId();
    }

    public String getLabel() {
        return formClass.getLabel();
    }

    public FormClass getFormClass() {
        return formClass;
    }

    public Level getParent() {
        return parent;
    }

    public ResourceId getParentFieldId() {
        return parentFieldId;
    }

    public boolean isRoot() {
        return parentId == null;
    }

    @Override
    public String toString() {
        return getLabel();
    }

    public List<Level> getChildren() {
        return children;
    }

}