source: src/controller_main.cpp@ a10cc0

/*
 * Project: MoleCuilder
 * Description: creates and alters molecular systems
 * Copyright (C)  2011 University of Bonn. All rights reserved.
 * Please see the LICENSE file or "Copyright notice" in builder.cpp for details.
 */

/*
 * \file controller_main.cpp
 *
 * This file strongly follows the Serialization example from the boost::asio
 * library (see client.cpp)
 *
 *  Created on: Nov 27, 2011
 *      Author: heber
 */


// include config.h
#ifdef HAVE_CONFIG_H
#include <config.h>
#endif

// boost asio needs specific operator new
#include <boost/asio.hpp>

#include "CodePatterns/MemDebug.hpp"

#include <boost/assign.hpp>
#include <boost/program_options.hpp>
#include <deque>
#include <fstream>
#include <iostream>
#include <map>
#include <sstream>
#include <streambuf>
#include <vector>

#include "CodePatterns/Assert.hpp"
#include "CodePatterns/Info.hpp"
#include "CodePatterns/Log.hpp"
#include "atexit.hpp"
#include "FragmentController.hpp"

#include "ControllerOptions_SystemCommandJob.hpp"
#include "ControllerCommandRegistry.hpp"

#include "controller_AddOn.hpp"

/** Print the status of scheduled and done jobs.
 *
 * @param status pair of number of schedule and done jobs
 */
void printJobStatus(const std::pair<size_t, size_t> &JobStatus)
{
  LOG(1, "INFO: #" << JobStatus.first << " are waiting in the queue and #"
      << JobStatus.second << " jobs are calculated so far.");
}

inline std::vector<std::string> getListOfCommands(const ControllerCommandRegistry &ControllerCommands)
{
  std::vector<std::string> Commands;
  for (ControllerCommandRegistry::const_iterator iter = ControllerCommands.getBeginIter();
      iter != ControllerCommands.getEndIter(); ++iter)
    Commands.push_back(iter->first);

  return Commands;
}

int controller_main(int argc, char* argv[])
{
  // from this moment on, we need to be sure to deeinitialize in the correct order
  // this is handled by the cleanup function
  atexit(cleanUp);

  size_t Exitflag = 0;

  controller_AddOn *AddOn = getAddOn();
  ASSERT(AddOn != NULL,
      "main() - returned AddOn is NULL.");

  ControllerOptions *ControllerInfo = AddOn->allocateControllerInfo();
  boost::asio::io_service io_service;
  FragmentController controller(io_service);
  boost::program_options::variables_map vm;

  // prepare ControllerCommand
  // note: we need "< ControllerCommand::commands_t >" because parseExecutable(),... return int
  // in contrast to other functions that return void
  ControllerCommandRegistry ControllerCommands;
  boost::function<void (ControllerCommand *)> registrator =
      (boost::bind(&Registry<ControllerCommand>::registerInstance, boost::ref(ControllerCommands), _1));
  registrator(new ControllerCommand("checkresults",
      boost::assign::list_of< ControllerCommand::commands_t >
        (boost::bind(&FragmentController::checkResults,
            boost::ref(controller), boost::cref(ControllerInfo->server), boost::cref(ControllerInfo->serverport)))
        (boost::bind(&printJobStatus,
            boost::bind(&FragmentController::getJobStatus, boost::ref(controller))))
      ));
  registrator(new ControllerCommand("removeall",
      boost::assign::list_of< ControllerCommand::commands_t >
        (boost::bind(&FragmentController::removeall,
            boost::ref(controller), boost::cref(ControllerInfo->server), boost::cref(ControllerInfo->serverport)))
      ));
  registrator(new ControllerCommand("shutdown",
      boost::assign::list_of< ControllerCommand::commands_t >
        (boost::bind(&FragmentController::shutdown,
            boost::ref(controller), boost::cref(ControllerInfo->server), boost::cref(ControllerInfo->serverport)))
      ));
  AddOn->addSpecificCommands(registrator, controller, *ControllerInfo);

  // Declare the supported options.
  boost::program_options::options_description desc("Allowed options");
  desc.add_options()
      ("help,h", "produce help message")
      ("verbosity,v", boost::program_options::value<size_t>(), "set verbosity level")
      ("server", boost::program_options::value< std::string >(), "connect to server at this address (host:port)")
      ("command", boost::program_options::value< std::string >(), (std::string("command to send to server: ")+toString(getListOfCommands(ControllerCommands))).c_str())
  ;
  AddOn->addSpecificOptions(desc.add_options());

  // parse command line
  boost::program_options::store(boost::program_options::parse_command_line(argc, argv, desc), vm);
  boost::program_options::notify(vm);

  // set controller information
  int status = 0;
  status = ControllerInfo->parseHelp(vm, desc);
  if (status) return status;
  status = ControllerInfo->parseVerbosity(vm);
  if (status) return status;
  status = ControllerInfo->parseServerPort(vm);
  if (status) return status;
  status = ControllerInfo->parseCommand(vm, getListOfCommands(ControllerCommands));
  if (status) return status;

  // all later parse functions depend on parsed command
  status = AddOn->addOtherParsings(*ControllerInfo, vm);
  if (status) return status;

  // parse given ControllerCommand
  if(!ControllerCommands.isPresentByName(ControllerInfo->command)) {
    ELOG(1, "Unrecognized command '"+toString(ControllerInfo->command)+"'.");
    return 255;
  }
  const ControllerCommand *commands = ControllerCommands.getByName(ControllerInfo->command);
  try
  {
    // execute each command in the queue synchronously
    size_t phase = 1;
    for (ControllerCommand::const_iterator iter = commands->begin();
       iter != commands->end(); ++iter) {
      (*iter)();
      {
        io_service.reset();
        Info info((std::string("io_service: ")+toString(phase)).c_str());
        io_service.run();
      }
    }
    Exitflag = controller.getExitflag();
  }
  catch (std::exception& e)
  {
    std::cerr << e.what() << std::endl;
  }
  delete AddOn;

  return Exitflag;
}