package com.dlsc.flexgantt.showcase.demo.resources;

import com.dlsc.flexgantt.model.gantt.DefaultGanttChartModel;
import com.dlsc.flexgantt.model.gantt.DefaultGanttChartNode;
import com.dlsc.flexgantt.model.gantt.ILayer.Feature;
import com.dlsc.flexgantt.model.gantt.ITimelineObject;
import com.dlsc.flexgantt.model.gantt.Layer;

public class ResourceModel
        extends
        DefaultGanttChartModel<DefaultGanttChartNode<Resource, ITimelineObject>, ITimelineObject> {
    
    public static final Layer CAP_LAYER = new Layer("Capacities");

    @SuppressWarnings("unchecked")
    public ResourceModel() {
        super(new DefaultGanttChartNode(true));

        addLayer(CAP_LAYER);
        CAP_LAYER.removeFeature(Feature.RELATIONSHIPS);

        DefaultGanttChartNode rootNode = getRoot();
        for (int i = 0; i < 20; i++) {
            rootNode.add(new ResourceNode(new Resource()));
        }
    }
}
