AppMixedLearning.h

/*
 * Copyright © 2012, United States Government, as represented by the
 * Administrator of the National Aeronautics and Space Administration.
 * All rights reserved.
 * 
 * The NASA Tensegrity Robotics Toolkit (NTRT) v1 platform is 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.
*/

#ifndef APP_MIXED_LEARNING
#define APP_MIXED_LEARNING

//robot
//#include "dev/btietz/TCContact/FlemonsSpineModelContact.h"
//#include "FlemonsSpineModelMixed.h"
#include "RibModelMixedContact.h"

// controller 
#include "dev/dhustigschultz/TC_MixedLearning/JSONMixedLearningControl.h"

// obstacles
#include "models/obstacles/tgBlockField.h"

// This library
#include "core/tgModel.h"
#include "core/tgSimViewGraphics.h"
#include "core/tgSimulation.h"
#include "core/tgWorld.h"
#include "core/terrain/tgBoxGround.h"
#include "core/terrain/tgHillyGround.h"

// Boost
#include <boost/program_options.hpp>

// The C++ Standard Library
#include <iostream>
#include <string>

namespace po = boost::program_options;

class AppMixedLearning
{
public:
    AppMixedLearning(int argc, char** argv);

    bool setup();
    bool run();

private:
    void handleOptions(int argc, char** argv);

    const tgHillyGround::Config getHillyConfig();
    
    const tgBoxGround::Config getBoxConfig();
    
    tgModel* getBlocks();
    
    tgWorld *createWorld();

    tgSimViewGraphics *createGraphicsView(tgWorld *world);

    tgSimView *createView(tgWorld *world);

    void simulate(tgSimulation *simulation);
    
    
    // Keep these around for cleanup
    tgWorld* world;
    tgSimView* view;
    tgSimulation* simulation;

    bool use_graphics;
    bool add_controller;
    bool add_blocks;
    bool add_hills;
    bool all_terrain;
    double timestep_physics; //Seconds
    double timestep_graphics; // Seconds, AKA render rate. Leave at 1/60 for real-time viewing
    int nEpisodes; // Number of episodes ("trial runs")
    int nSteps; // Number of steps in each episode, 60k is 100 seconds (timestep_physics*nSteps)
    int nSegments; // Number of segments in the tensegrity spine
    int nTypes; // Number of types of terrain to be used. Currently 3

    double startX;
    double startY;
    double startZ;
    double startAngle;
    double goalAngle;
    
    std::string lowerPath; 
    std::string suffix;
    
    bool bSetup;
};

#endif  // APP_MIXED_LEARNING