source: src/Fragmentation/Automation/poolworker.cpp@ d9373b

Action_Thermostats Add_AtomRandomPerturbation Add_FitFragmentPartialChargesAction Add_RotateAroundBondAction Add_SelectAtomByNameAction Added_ParseSaveFragmentResults AddingActions_SaveParseParticleParameters Adding_Graph_to_ChangeBondActions Adding_MD_integration_tests Adding_ParticleName_to_Atom Adding_StructOpt_integration_tests AtomFragments Automaking_mpqc_open AutomationFragmentation_failures Candidate_v1.5.4 Candidate_v1.6.0 Candidate_v1.6.1 ChangeBugEmailaddress ChangingTestPorts ChemicalSpaceEvaluator CombiningParticlePotentialParsing Combining_Subpackages Debian_Package_split Debian_package_split_molecuildergui_only Disabling_MemDebug Docu_Python_wait EmpiricalPotential_contain_HomologyGraph EmpiricalPotential_contain_HomologyGraph_documentation Enable_parallel_make_install Enhance_userguide Enhanced_StructuralOptimization Enhanced_StructuralOptimization_continued Example_ManyWaysToTranslateAtom Exclude_Hydrogens_annealWithBondGraph FitPartialCharges_GlobalError Fix_BoundInBox_CenterInBox_MoleculeActions Fix_ChargeSampling_PBC Fix_ChronosMutex Fix_FitPartialCharges Fix_FitPotential_needs_atomicnumbers Fix_ForceAnnealing Fix_IndependentFragmentGrids Fix_ParseParticles Fix_ParseParticles_split_forward_backward_Actions Fix_PopActions Fix_QtFragmentList_sorted_selection Fix_Restrictedkeyset_FragmentMolecule Fix_StatusMsg Fix_StepWorldTime_single_argument Fix_Verbose_Codepatterns Fix_fitting_potentials Fixes ForceAnnealing_goodresults ForceAnnealing_oldresults ForceAnnealing_tocheck ForceAnnealing_with_BondGraph ForceAnnealing_with_BondGraph_continued ForceAnnealing_with_BondGraph_continued_betteresults ForceAnnealing_with_BondGraph_contraction-expansion FragmentAction_writes_AtomFragments FragmentMolecule_checks_bonddegrees GeometryObjects Gui_Fixes Gui_displays_atomic_force_velocity ImplicitCharges IndependentFragmentGrids IndependentFragmentGrids_IndividualZeroInstances IndependentFragmentGrids_IntegrationTest IndependentFragmentGrids_Sole_NN_Calculation JobMarket_RobustOnKillsSegFaults JobMarket_StableWorkerPool JobMarket_unresolvable_hostname_fix MoreRobust_FragmentAutomation ODR_violation_mpqc_open PartialCharges_OrthogonalSummation PdbParser_setsAtomName PythonUI_with_named_parameters QtGui_reactivate_TimeChanged_changes Recreated_GuiChecks Rewrite_FitPartialCharges RotateToPrincipalAxisSystem_UndoRedo SaturateAtoms_findBestMatching SaturateAtoms_singleDegree StoppableMakroAction Subpackage_CodePatterns Subpackage_JobMarket Subpackage_LinearAlgebra Subpackage_levmar Subpackage_mpqc_open Subpackage_vmg Switchable_LogView ThirdParty_MPQC_rebuilt_buildsystem TrajectoryDependenant_MaxOrder TremoloParser_IncreasedPrecision TremoloParser_MultipleTimesteps TremoloParser_setsAtomName Ubuntu_1604_changes stable
Last change on this file since d9373b was 5ac92e, checked in by Frederik Heber <heber@…>, 13 years ago

PoolWorker now parses command line options via boost::program_options.

  • all of poolworkers command line arguments are now parsed via boost::program_options.
  • As a sideeffect multiple signals may now be specified to catch on the command line.
  • TESTFIX: Usage of poolworker has changed, hence adapted regression tests.
  • Property mode set to 100644
File size: 4.8 KB
Line 
1/*
2 * Project: MoleCuilder
3 * Description: creates and alters molecular systems
4 * Copyright (C) 2012 University of Bonn. All rights reserved.
5 * Please see the LICENSE file or "Copyright notice" in builder.cpp for details.
6 */
7
8/*
9 * \file poolworker.cpp
10 *
11 * This file strongly follows the Serialization example from the boost::asio
12 * library (see client.cpp)
13 *
14 * Created on: Feb 28, 2012
15 * Author: heber
16 */
17
18
19// include config.h
20#ifdef HAVE_CONFIG_H
21#include <config.h>
22#endif
23
24// boost asio needs specific operator new
25#include <boost/asio.hpp>
26
27#include "CodePatterns/MemDebug.hpp"
28
29#include <iostream>
30#include <boost/program_options.hpp>
31#include <boost/lexical_cast.hpp>
32#include <signal.h>
33#include <vector>
34
35#include "atexit.hpp"
36#include "CodePatterns/Info.hpp"
37#include "CodePatterns/Log.hpp"
38#include "Pool/PoolWorker.hpp"
39
40//!> global shutdown function
41boost::function<void (int)> shutdownfunction;
42//!> global signal vector for later releasing
43std::vector<size_t> signals;
44
45void signalhandler(int sig)
46{
47 // shutdown if we have handler
48 if (shutdownfunction)
49 shutdownfunction(sig);
50
51 // release signal hook again
52 for (std::vector<size_t>::const_iterator iter = signals.begin();
53 iter != signals.end(); ++iter)
54 signal(*iter, NULL);
55}
56
57int main(int argc, char* argv[])
58{
59 // from this moment on, we need to be sure to deeinitialize in the correct order
60 // this is handled by the cleanup function
61 atexit(cleanUp);
62
63 // Declare the supported options.
64 boost::program_options::options_description desc("Allowed options");
65 desc.add_options()
66 ("help,h", "produce help message")
67 ("verbosity,v", boost::program_options::value<size_t>(), "set verbosity level")
68 ("signal", boost::program_options::value< std::vector<size_t> >(), "set signal to catch (can be given multiple times)")
69 ("server", boost::program_options::value< std::string>(), "connect to server at this address (host:port)")
70 ("listen", boost::program_options::value< std::string >(), "listen on this port")
71 ("hostname", boost::program_options::value< std::string>(), "name of host on which this codes runs and which server can resolve")
72 ;
73
74 boost::program_options::variables_map vm;
75 boost::program_options::store(boost::program_options::parse_command_line(argc, argv, desc), vm);
76 boost::program_options::notify(vm);
77
78 if (vm.count("help")) {
79 std::cout << desc << "\n";
80 return 1;
81 }
82
83 if (vm.count("verbosity")) {
84 LOG(0, "STATUS: Verbosity level was set to " << vm["verbosity"].as<size_t>() << ".");
85 setVerbosity(vm["verbosity"].as<size_t>());
86 } else {
87 LOG(0, "STATUS: Verbosity level was not set, defaulting to 5.");
88 setVerbosity(5);
89 }
90
91 std::string server;
92 std::string serverport;
93 if (vm.count("server")) {
94 server = vm["server"].as< std::string >();
95 serverport = server.substr(server.find_last_of(':')+1, std::string::npos);
96 server = server.substr(0, server.find_last_of(':'));
97 try {
98 boost::lexical_cast<size_t>(serverport);
99 } catch (boost::bad_lexical_cast) {
100 ELOG(1, "Could not interpret " << serverport << " as server:port.");
101 return 255;
102 }
103 LOG(1, "INFO: Using " << server << ":" << serverport << " as server's address.");
104 } else {
105 ELOG(1, "Requiring server's address (host:port) to connect to.");
106 return 255;
107 }
108
109 std::string hostname("localhost");
110 if (vm.count("hostname")) {
111 hostname = vm["hostname"].as< std::string >();
112 LOG(1, "INFO: Using " << hostname << " as host's name.");
113 } else {
114 ELOG(1, "Requiring port to listen on .");
115 return 255;
116 }
117
118 std::string listenport;
119 if (vm.count("listen")) {
120 try {
121 listenport = vm["listen"].as< std::string >();
122 } catch (boost::bad_lexical_cast) {
123 ELOG(1, "Could not read " << vm["listen"].as< std::string >() << " as digits.");
124 return 255;
125 }
126 LOG(1, "INFO: Using port " << listenport << " to listen to server connects.");
127 } else {
128 ELOG(1, "Requiring port to listen on .");
129 return 255;
130 }
131
132 size_t exitflag = 0;
133 try
134 {
135 boost::asio::io_service io_service;
136 PoolWorker client(io_service, server, serverport, hostname, listenport);
137
138 // catch ctrl-c and shutdown worker properly
139 shutdownfunction = boost::bind(&PoolWorker::shutdown, boost::ref(client), _1);
140 if (vm.count("signal")) {
141 signals = vm["signal"].as< std::vector<size_t> >();
142 for (std::vector<size_t>::const_iterator iter = signals.begin();
143 iter != signals.end(); ++iter) {
144 LOG(0, "STATUS: Catching signal " << *iter << " via signal handler.");
145 signal(*iter, &signalhandler);
146 }
147 } else {
148 LOG(0, "STATUS: No signals are caught.");
149 }
150
151 // process io requests
152 {
153 Info info("io_service");
154 io_service.run();
155 }
156 exitflag = client.getFlag();
157 }
158 catch (std::exception& e)
159 {
160 std::cerr << e.what() << std::endl;
161 }
162
163
164 return exitflag;
165}
Note: See TracBrowser for help on using the repository browser.