RFTreeViewBuilder.java

/*
 * Copyright 2013 State University of New York at Oswego
 *
 * Licensed under the Apache License, Version 2.0 (the "License"); 
 * you may not use this file except in compliance with the License. 
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software 
 * distributed under the License is distributed on an "AS IS" BASIS, 
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 
 * See the License for the specific language governing permissions and 
 * limitations under the License.
 */
package edu.oswego.csc480_hci521_2013.shared.h2o.urlbuilders;

import java.util.HashMap;

/**
 *
 */
public class RFTreeViewBuilder extends AbstractBuilder {

    static final String NAME = "RFTreeView";

    RFTreeViewBuilder() {
    }

    RFTreeViewBuilder(HashMap<String, String> args) {
        super(NAME);
        setArgs(args);
    }

    public RFTreeViewBuilder(String dataKey, String modelKey) {
        super(NAME);
        addArg("data_key", dataKey);
        addArg("model_key", modelKey);
    }

    public RFTreeViewBuilder setTreeNumber(int value) {
        if (value < 0) {
            throw new IllegalArgumentException("tree number must be positive");
        }
        addArg("tree_number", Integer.valueOf(value).toString());
        return this;
    }

    /**
     * Column name.
     *
     * @param value
     * @return
     */
    public RFTreeViewBuilder setResponseVariable(int value) {
        addArg("response_variable", Integer.valueOf(value).toString());
        return this;
    }
}