main_helper.cpp File Reference

#include <HardwareBridge.h>
#include "SimulationBridge.h"
#include <main_helper.h>
#include <cassert>
#include <iostream>
#include <RobotController.h>
#include "Utilities/SegfaultHandler.h"

Include dependency graph for main_helper.cpp:

Go to the source code of this file.

Functions
void	printUsage ()
int	main_helper (int argc, char **argv, RobotController *ctrl)

Variables
MasterConfig	gMasterConfig

Function Documentation

int main_helper	(	int	argc,
		char **	argv,
		RobotController *	ctrl
	)

Definition at line 31 of file main_helper.cpp.

References MasterConfig::_robot, CHEETAH_3, install_segfault_handler(), MINI_CHEETAH, printUsage(), SimulationBridge::run(), MiniCheetahHardwareBridge::run(), and MasterConfig::simulated.

                                                              {
  install_segfault_handler();
  if (argc != 3) {
    printUsage();
    return EXIT_FAILURE;
  }

  if (argv[1][0] == '3') {
    gMasterConfig._robot = RobotType::CHEETAH_3;
  } else if (argv[1][0] == 'm') {
    gMasterConfig._robot = RobotType::MINI_CHEETAH;
  } else {
    printUsage();
    return EXIT_FAILURE;
  }

  if (argv[2][0] == 's') {
    gMasterConfig.simulated = true;
  } else if (argv[2][0] == 'r') {
    gMasterConfig.simulated = false;
  } else {
    printUsage();
    return EXIT_FAILURE;
  }

  printf("[Quadruped] Cheetah Software\n");
  printf("        Quadruped:  %s\n",
         gMasterConfig._robot == RobotType::MINI_CHEETAH ? "Mini Cheetah"
                                                         : "Cheetah 3");
  printf("        Driver: %s\n", gMasterConfig.simulated
                                     ? "Development Simulation Driver"
                                     : "Quadruped Driver");

  // dispatch the appropriate driver
  if (gMasterConfig.simulated) {
    if (gMasterConfig._robot == RobotType::MINI_CHEETAH) {
      SimulationBridge simulationBridge(gMasterConfig._robot, ctrl);
      simulationBridge.run();
      printf("[Quadruped] SimDriver run() has finished!\n");
    } else if (gMasterConfig._robot == RobotType::CHEETAH_3) {
      SimulationBridge simulationBridge(gMasterConfig._robot, ctrl);
      simulationBridge.run();
    } else {
      printf("[ERROR] unknown robot\n");
      assert(false);
    }
  } else {
    if (gMasterConfig._robot == RobotType::MINI_CHEETAH) {
      MiniCheetahHardwareBridge hw(ctrl);
      hw.run();
      printf("[Quadruped] SimDriver run() has finished!\n");
    } else if (gMasterConfig._robot == RobotType::CHEETAH_3) {
      printf("[ERROR] can't do cheetah 3 hardware\n");
      assert(false);
    } else {
      printf("[ERROR] unknown robot\n");
      assert(false);
    }
  }

  return 0;
}

Here is the call graph for this function:

void printUsage

(

)

Print a message describing the command line flags for the robot program

Definition at line 24 of file main_helper.cpp.

                  {
  printf(
      "Usage: robot [robot-id] [sim-or-robot]\n\twhere robot-id:     3 for "
      "cheetah 3, m for mini-cheetah\n\t      sim-or-robot: s for sim, r for "
      "robot\n");
}

Here is the caller graph for this function:

Variable Documentation

MasterConfig gMasterConfig

Definition at line 19 of file main_helper.cpp.