| 1 | /* | 
|---|
| 2 | * Project: MoleCuilder | 
|---|
| 3 | * Description: creates and alters molecular systems | 
|---|
| 4 | * Copyright (C)  2014 Frederik Heber. All rights reserved. | 
|---|
| 5 | * | 
|---|
| 6 | * | 
|---|
| 7 | *   This file is part of MoleCuilder. | 
|---|
| 8 | * | 
|---|
| 9 | *    MoleCuilder is free software: you can redistribute it and/or modify | 
|---|
| 10 | *    it under the terms of the GNU General Public License as published by | 
|---|
| 11 | *    the Free Software Foundation, either version 2 of the License, or | 
|---|
| 12 | *    (at your option) any later version. | 
|---|
| 13 | * | 
|---|
| 14 | *    MoleCuilder is distributed in the hope that it will be useful, | 
|---|
| 15 | *    but WITHOUT ANY WARRANTY; without even the implied warranty of | 
|---|
| 16 | *    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the | 
|---|
| 17 | *    GNU General Public License for more details. | 
|---|
| 18 | * | 
|---|
| 19 | *    You should have received a copy of the GNU General Public License | 
|---|
| 20 | *    along with MoleCuilder.  If not, see <http://www.gnu.org/licenses/>. | 
|---|
| 21 | */ | 
|---|
| 22 |  | 
|---|
| 23 | /* | 
|---|
| 24 | * MPQCCommandFragmentController.cpp | 
|---|
| 25 | * | 
|---|
| 26 | *  Created on: Apr 14, 2014 | 
|---|
| 27 | *      Author: heber | 
|---|
| 28 | */ | 
|---|
| 29 |  | 
|---|
| 30 |  | 
|---|
| 31 | // include config.h | 
|---|
| 32 | #ifdef HAVE_CONFIG_H | 
|---|
| 33 | #include <config.h> | 
|---|
| 34 | #endif | 
|---|
| 35 |  | 
|---|
| 36 | #include <boost/archive/text_iarchive.hpp> | 
|---|
| 37 | // boost asio needs specific operator new | 
|---|
| 38 | //#include <boost/asio.hpp> | 
|---|
| 39 |  | 
|---|
| 40 | #include "CodePatterns/MemDebug.hpp" | 
|---|
| 41 |  | 
|---|
| 42 | #include "MPQCCommandFragmentController.hpp" | 
|---|
| 43 |  | 
|---|
| 44 | #include <boost/cast.hpp> | 
|---|
| 45 | #include <boost/filesystem/operations.hpp> | 
|---|
| 46 | #include <vector> | 
|---|
| 47 |  | 
|---|
| 48 | #include "CodePatterns/Assert.hpp" | 
|---|
| 49 | #include "CodePatterns/Log.hpp" | 
|---|
| 50 |  | 
|---|
| 51 | #include "Box.hpp" | 
|---|
| 52 | #include "Helpers/defs.hpp" | 
|---|
| 53 | #include "Fragmentation/Automation/FragmentJobQueue.hpp" | 
|---|
| 54 | #include "Jobs/MPQCCommandJob.hpp" | 
|---|
| 55 | #include "LinearAlgebra/RealSpaceMatrix.hpp" | 
|---|
| 56 | #include "World.hpp" | 
|---|
| 57 |  | 
|---|
| 58 | // static entities | 
|---|
| 59 | unsigned int MPQCCommandFragmentController::nextid = 0; | 
|---|
| 60 |  | 
|---|
| 61 | unsigned int MPQCCommandFragmentController::addJobsFromQueue( | 
|---|
| 62 | const MPQCData::DoLongrange_t _DoLongrange, | 
|---|
| 63 | const MPQCData::DoValenceOnly_t _DoValenceOnly, | 
|---|
| 64 | const std::string &_executable | 
|---|
| 65 | ) | 
|---|
| 66 | { | 
|---|
| 67 | // give them all valid ids | 
|---|
| 68 | std::vector<FragmentJob::ptr> newjobs = FragmentJobQueue::getInstance().getJobs(); | 
|---|
| 69 | for (std::vector<FragmentJob::ptr>::iterator jobiter = newjobs.begin(); | 
|---|
| 70 | jobiter != newjobs.end(); | 
|---|
| 71 | ++jobiter) { | 
|---|
| 72 | MPQCCommandJob *job = boost::polymorphic_downcast<MPQCCommandJob *>(jobiter->get()); | 
|---|
| 73 | //    job->DoLongrange = _DoLongrange; | 
|---|
| 74 | //    job->DoValenceOnly = _DoValenceOnly; | 
|---|
| 75 | #ifndef HAVE_JOBMARKET | 
|---|
| 76 | JobId_t id = getAvailableId(); | 
|---|
| 77 | if (id == (JobId_t)JobId::IllegalJob) { | 
|---|
| 78 | queue.clear(); | 
|---|
| 79 | break; | 
|---|
| 80 | } | 
|---|
| 81 | job->setId(id); | 
|---|
| 82 | if (!_executable.empty()) | 
|---|
| 83 | job->setCommand(_executable); | 
|---|
| 84 | job->setSuffix(".in"); | 
|---|
| 85 | #endif /* cause we don't inherit from FragmentController */ | 
|---|
| 86 | // add the jobs | 
|---|
| 87 | queue.push_back(job); | 
|---|
| 88 | } | 
|---|
| 89 |  | 
|---|
| 90 | return queue.size(); | 
|---|
| 91 | } | 
|---|
| 92 |  | 
|---|
| 93 | void MPQCCommandFragmentController::run() | 
|---|
| 94 | { | 
|---|
| 95 | const size_t NoExpectedResults = queue.size(); | 
|---|
| 96 | // take jobs from queue and add them | 
|---|
| 97 | std::vector<FragmentResult::ptr> resultvector; | 
|---|
| 98 | while(!queue.empty()) { | 
|---|
| 99 | // pop job from list | 
|---|
| 100 | MPQCCommandJob *job = queue.front(); | 
|---|
| 101 | queue.pop_front(); | 
|---|
| 102 | // run job and set exitflag if failed (>0) | 
|---|
| 103 | FragmentResult::ptr result = job->Work(); | 
|---|
| 104 | if (result->exitflag) { | 
|---|
| 105 | ELOG(2, "Job #" << job->getId() << " has failed execution."); | 
|---|
| 106 | } | 
|---|
| 107 | exitflag = result->exitflag == 0 ? | 
|---|
| 108 | exitflag : | 
|---|
| 109 | ( result->exitflag > exitflag ? | 
|---|
| 110 | result->exitflag : | 
|---|
| 111 | exitflag ); | 
|---|
| 112 | // store result | 
|---|
| 113 | resultvector.push_back(result); | 
|---|
| 114 | handler(resultvector.size(), NoExpectedResults); | 
|---|
| 115 | } | 
|---|
| 116 |  | 
|---|
| 117 | // convert all results to MPQCData | 
|---|
| 118 | std::vector<MPQCData> fragmentData; | 
|---|
| 119 | results.ConvertFragmentResultTo(resultvector, fragmentData); | 
|---|
| 120 | // insert into final map | 
|---|
| 121 | results.insertResults(resultvector, fragmentData); | 
|---|
| 122 | } | 
|---|