LearnModels.java

Go to the documentation of this file.

package edu.wpi.rail.jinteractiveworld.ros.srvs.interactiveworldmsgs;

import javax.json.Json;
import javax.json.JsonObject;

import edu.wpi.rail.jinteractiveworld.ros.msgs.interactiveworldmsgs.TaskModels;
import edu.wpi.rail.jinteractiveworld.ros.msgs.interactiveworldmsgs.TaskTrainingData;
import edu.wpi.rail.jrosbridge.services.ServiceRequest;
import edu.wpi.rail.jrosbridge.services.ServiceResponse;

import java.io.StringReader;
import java.util.Arrays;

public class LearnModels {

    public static final String TYPE = "interactive_world_msgs/LearnModels";

    public static class Request extends ServiceRequest {

        public static final String FIELD_HOST = "host";

        public static final String FIELD_PORT = "port";

        public static final String FIELD_USER = "user";

        public static final String FIELD_PASSWORD = "password";

        public static final String FIELD_DATABASE = "database";

        public static final String FIELD_CONDITION_ID = "condition_id";

        public static final String FIELD_DATA = "data";

        private final TaskTrainingData data;
        private final int port, conditionId;
        private final String host, user, password, database;

        public Request() {
            this("", 0, "", "", "", 0, new TaskTrainingData());
        }

        public Request(String host, int port, String user, String password, String database, int conditionId, TaskTrainingData data) {
            super(Json.createObjectBuilder().add(Request.FIELD_DATA, data.toJsonObject())
                    .add(Request.FIELD_CONDITION_ID, conditionId).build(), LearnModels.TYPE);
            this.host = host;
            this.port = port;
            this.user = user;
            this.password = password;
            this.database = database;
            this.conditionId = conditionId;
            this.data = data;
        }

        public int getPort() {
            return port;
        }

        public String getHost() {
            return host;
        }

        public String getUser() {
            return user;
        }

        public String getPassword() {
            return password;
        }

        public String getDatabase() {
            return database;
        }

        public int getConditionId() {
            return this.conditionId;
        }

        public TaskTrainingData getData() {
            return this.data;
        }

        @Override
        public Request clone() {
            return new LearnModels.Request(this.host, this.port, this.user, this.password, this.database, this.conditionId, this.data);
        }

        public static Request fromJsonString(String jsonString) {
            // convert to a ServiceRequest
            return LearnModels.Request.fromServiceRequest(new ServiceRequest(
                    jsonString));
        }

        public static Request fromServiceRequest(ServiceRequest req) {
            // get it from the JSON object
            return LearnModels.Request.fromJsonObject(req.toJsonObject());
        }

        public static Request fromJsonObject(JsonObject jsonObject) {
            String host = jsonObject.containsKey(Request.FIELD_HOST) ?
                    jsonObject.getString(Request.FIELD_HOST) : "";
            int port = jsonObject.containsKey(Request.FIELD_PORT) ? jsonObject
                    .getJsonNumber(Request.FIELD_PORT).intValue() : 0;
            String user = jsonObject.containsKey(Request.FIELD_USER) ?
                    jsonObject.getString(Request.FIELD_USER) : "";
            String password = jsonObject.containsKey(Request.FIELD_PASSWORD) ?
                    jsonObject.getString(Request.FIELD_PASSWORD) : "";
            String database = jsonObject.containsKey(Request.FIELD_DATABASE) ?
                    jsonObject.getString(Request.FIELD_DATABASE) : "";
            int conditionId = jsonObject.containsKey(Request.FIELD_CONDITION_ID) ? jsonObject
                    .getJsonNumber(Request.FIELD_CONDITION_ID).intValue() : 0;
            TaskTrainingData data = jsonObject.containsKey(Request.FIELD_DATA) ? TaskTrainingData
                    .fromJsonObject(jsonObject.getJsonObject(Request.FIELD_DATA))
                    : new TaskTrainingData();
            return new LearnModels.Request(host, port, user, password, database, conditionId, data);
        }
    }

    public static class Response extends ServiceResponse {

        public Response() {
            this(true);
        }

        public Response(boolean result) {
            super(ServiceResponse.EMPTY_MESSAGE, LearnModels.TYPE, result);
        }

        @Override
        public Response clone() {
            return new LearnModels.Response(this.getResult());
        }

        public static Response fromJsonString(String jsonString) {
            // convert to a ServiceResponse
            return LearnModels.Response.fromServiceResponse(new ServiceResponse(
                    jsonString, true));
        }

        public static Response fromServiceResponse(ServiceResponse resp) {
            // get it from the JSON object
            return LearnModels.Response.fromJsonObject(resp.toJsonObject());
        }

        public static Response fromJsonObject(JsonObject jsonObject) {
            return new LearnModels.Response(true);
        }
    }
}