Main.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.h2owrapper;

import edu.oswego.csc480_hci521_2013.server.RestHandler;
import edu.oswego.csc480_hci521_2013.server.ServerUrlEncoder;
import edu.oswego.csc480_hci521_2013.shared.h2o.json.ImportUrl;
import edu.oswego.csc480_hci521_2013.shared.h2o.json.Inspect;
import edu.oswego.csc480_hci521_2013.shared.h2o.json.Parse;
import edu.oswego.csc480_hci521_2013.shared.h2o.json.Progress;
import edu.oswego.csc480_hci521_2013.shared.h2o.json.RF;
import edu.oswego.csc480_hci521_2013.shared.h2o.json.RFTreeView;
import edu.oswego.csc480_hci521_2013.shared.h2o.json.RFView;
import edu.oswego.csc480_hci521_2013.shared.h2o.json.ResponseStatus;
import edu.oswego.csc480_hci521_2013.shared.h2o.json.StoreView;
import edu.oswego.csc480_hci521_2013.shared.h2o.urlbuilders.ImportUrlBuilder;
import edu.oswego.csc480_hci521_2013.shared.h2o.urlbuilders.InspectBuilder;
import edu.oswego.csc480_hci521_2013.shared.h2o.urlbuilders.ParseBuilder;
import edu.oswego.csc480_hci521_2013.shared.h2o.urlbuilders.RFBuilder;
import edu.oswego.csc480_hci521_2013.shared.h2o.urlbuilders.RFScoreBuilder;
import edu.oswego.csc480_hci521_2013.shared.h2o.urlbuilders.RFViewBuilder;
import edu.oswego.csc480_hci521_2013.shared.h2o.urlbuilders.RFTreeViewBuilder;
import edu.oswego.csc480_hci521_2013.shared.h2o.urlbuilders.RedirectRequestFactory;
import edu.oswego.csc480_hci521_2013.shared.h2o.urlbuilders.StoreViewBuilder;
import edu.oswego.csc480_hci521_2013.shared.h2o.urlbuilders.UrlEncoder;

/**
 * Examples of usage
 */
public class Main
{
    private static UrlEncoder encoder = new ServerUrlEncoder();

    /**
     * @param args the command line arguments
     */
    public static void main(String[] args) throws Exception
    {
        RestHandler rest = new RestHandler();

        importUrl(rest);
        ResponseStatus status = parse(rest);
        progress(rest, status);
        inspect(rest);
        rf(rest);
        rfView(rest);
        rfTreeView(rest);
        storeView(rest);
        rfScore(rest);
    }

    private static void importUrl(RestHandler rest) throws Exception
    {
        String url = new ImportUrlBuilder("https://raw.github.com/0xdata/h2o/master/smalldata/cars.csv").setKey("cars.csv").build(encoder);
        System.out.println(url);
        String json = rest.fetch(url);
        System.out.println(json);
        ImportUrl val = rest.parse(json, ImportUrl.class);
        System.out.println(val);
    }

    private static ResponseStatus parse(RestHandler rest) throws Exception
    {
        String url = new ParseBuilder("cars.csv").setHeader(true).setDestinationKey("cars.hex").build(encoder);
        System.out.println(url);
        String json = rest.fetch(url);
        System.out.println(json);
        Parse val = rest.parse(json, Parse.class);
        System.out.println(val);
        return val.getResponse();
    }

    private static void inspect(RestHandler rest) throws Exception
    {
        String url = new InspectBuilder("cars.hex").build(encoder);
        System.out.println(url);
        String json = rest.fetch(url);
        System.out.println(json);
        Inspect val = rest.parse(json, Inspect.class);
        System.out.println(val);
    }

    private static void rf(RestHandler rest) throws Exception
    {
        String url = new RFBuilder("cars.hex").setModelKey("cars.model").setOutOfBagErrorEstimate(false).build(encoder);
        System.out.println(url);
        String json = rest.fetch(url);
        System.out.println(json);
        RF val = rest.parse(json, RF.class);
        System.out.println(val);
    }

    private static void storeView(RestHandler rest) throws Exception
    {
        String url = new StoreViewBuilder().build(encoder);
        System.out.println(url);
        String json = rest.fetch(url);
        System.out.println(json);
        StoreView val = rest.parse(json, StoreView.class);
        System.out.println(val);
    }

    private static void rfView(RestHandler rest) throws Exception
    {
        String url = new RFViewBuilder("cars.hex", "cars.model").build(encoder);
        System.out.println(url);

        for (;;) {
            String json = rest.fetch(url);
            System.out.println(json);
            RFView val = rest.parse(json, RFView.class);
            System.out.println(val);
            if (val.getResponse().isPoll()) {
                System.out.println("Progress: " + val.getResponse().getProgress());
                Thread.sleep(500);
            }
            else {
                break;
            }
        }
    }

    private static void rfScore(RestHandler rest) throws Exception
    {
        String url = new RFScoreBuilder("cars.hex", "cars.model").build(encoder);
        System.out.println(url);

        String json = rest.fetch(url);
        System.out.println(json);
        RFView val = rest.parse(json, RFView.class);
        System.out.println(val);
    }

    private static void rfTreeView(RestHandler rest) throws Exception
    {
        String url = new RFTreeViewBuilder("cars.hex", "cars.model").setTreeNumber(15).build(encoder);
        System.out.println(url);
        String json = rest.fetch(url);
        System.out.println(json);
        RFTreeView val = rest.parse(json, RFTreeView.class);
        System.out.println("value: " + val);

        System.out.println(val.getTree().toJson());
    }

    private static void progress(RestHandler rest, ResponseStatus status) throws Exception
    {
        if (!status.isRedirect() || !status.getRedirectRequest().equals("Progress")) {
            return;
        }

        String url = RedirectRequestFactory.getRequest(status).build(encoder);
        System.out.println(url);
        Progress val;
        for (;;) {
            String json = rest.fetch(url);
            System.out.println(json);
            val = rest.parse(json, Progress.class);
            System.out.println(val);
            if (val.getResponse().isPoll()) {
                System.out.println("Progress: " + val.getResponse().getProgress());
                Thread.sleep(500);
            }
            else {
                break;
            }
        }
    }
}