162 lines
4.6 KiB
C++
162 lines
4.6 KiB
C++
/* -*- mode: C++; c-basic-offset: 2; indent-tabs-mode: nil -*- */
|
|
/*
|
|
* Main authors:
|
|
* Maxim Shishmarev <maxim.shishmarev@monash.edu>
|
|
*
|
|
* Contributing authors:
|
|
* Kevin Leo <kevin.leo@monash.edu>
|
|
* Christian Schulte <schulte@gecode.org>
|
|
*
|
|
* Copyright:
|
|
* Kevin Leo, 2017
|
|
* Christian Schulte, 2017
|
|
* Maxim Shishmarev, 2017
|
|
*
|
|
* This file is part of Gecode, the generic constraint
|
|
* development environment:
|
|
* http://www.gecode.org
|
|
*
|
|
* Permission is hereby granted, free of charge, to any person obtaining
|
|
* a copy of this software and associated documentation files (the
|
|
* "Software"), to deal in the Software without restriction, including
|
|
* without limitation the rights to use, copy, modify, merge, publish,
|
|
* distribute, sublicense, and/or sell copies of the Software, and to
|
|
* permit persons to whom the Software is furnished to do so, subject to
|
|
* the following conditions:
|
|
*
|
|
* The above copyright notice and this permission notice shall be
|
|
* included in all copies or substantial portions of the Software.
|
|
*
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
|
|
* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
|
|
* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
|
|
* NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
|
|
* LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
|
|
* OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
|
|
* WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
|
*
|
|
*/
|
|
|
|
#include <gecode/search.hh>
|
|
|
|
#ifdef GECODE_HAS_CPPROFILER
|
|
|
|
#include <gecode/search/cpprofiler/message.hpp>
|
|
#include <gecode/search/cpprofiler/connector.hpp>
|
|
|
|
namespace Gecode {
|
|
|
|
CPProfilerSearchTracer::GetInfo::GetInfo(void) {}
|
|
|
|
CPProfilerSearchTracer::GetInfo::~GetInfo(void) {}
|
|
|
|
|
|
CPProfilerSearchTracer::CPProfilerSearchTracer(int eid, std::string name0,
|
|
unsigned int port,
|
|
const GetInfo* pgetinfo) :
|
|
connector(new CPProfiler::Connector(port)), execution_id(eid), name(name0), restart(0),
|
|
pgi(pgetinfo) {
|
|
}
|
|
|
|
void
|
|
CPProfilerSearchTracer::init(void) {
|
|
// Try to find out whether engine is a restart engine
|
|
bool restarts = ((engines() == 2U) &&
|
|
(engine(0U).type() == EngineType::RBS));
|
|
connector->connect();
|
|
connector->start(name, execution_id, restarts);
|
|
}
|
|
|
|
void
|
|
CPProfilerSearchTracer::round(unsigned int) {
|
|
restart++;
|
|
connector->restart(restart);
|
|
}
|
|
|
|
void
|
|
CPProfilerSearchTracer::skip(const EdgeInfo& ei) {
|
|
CPProfiler::NodeUID dummy_node{-1, restart, -1};
|
|
CPProfiler::NodeUID parent{-1,restart,-1};
|
|
|
|
int alt = 0;
|
|
std::string label;
|
|
if (ei) {
|
|
parent.nid = static_cast<int>(ei.nid());
|
|
parent.tid = static_cast<int>(ei.wid());
|
|
dummy_node.tid = static_cast<int>(ei.wid());
|
|
alt = static_cast<int>(ei.alternative());
|
|
label = ei.string();
|
|
}
|
|
|
|
auto node = connector->createNode(dummy_node, parent,
|
|
alt, 0, CPProfiler::NodeStatus::SKIPPED)
|
|
.set_label(label)
|
|
.set_info("");
|
|
connector->sendNode(node);
|
|
}
|
|
|
|
void
|
|
CPProfilerSearchTracer::node(const EdgeInfo& ei, const NodeInfo& ni) {
|
|
CPProfiler::NodeUID this_node{static_cast<int>(ni.nid()),
|
|
restart,
|
|
static_cast<int>(ni.wid())};
|
|
CPProfiler::NodeUID parent {-1,restart,-1};
|
|
|
|
int alt = 0;
|
|
int alts = 0;
|
|
|
|
CPProfiler::NodeStatus ns = CPProfiler::NodeStatus::FAILED;
|
|
switch(ni.type()) {
|
|
case NodeType::SOLVED:
|
|
ns = CPProfiler::NodeStatus::SOLVED;
|
|
break;
|
|
case NodeType::BRANCH:
|
|
ns = CPProfiler::NodeStatus::BRANCH;
|
|
alts = static_cast<int>(ni.choice().alternatives());
|
|
break;
|
|
case NodeType::FAILED:
|
|
ns = CPProfiler::NodeStatus::FAILED;
|
|
break;
|
|
default:
|
|
GECODE_NEVER;
|
|
}
|
|
|
|
std::string label;
|
|
if(ei) {
|
|
parent = {static_cast<int>(ei.nid()),
|
|
restart,
|
|
static_cast<int>(ei.wid())};
|
|
|
|
alt = static_cast<int>(ei.alternative());
|
|
label = ei.string();
|
|
} else {
|
|
alt = restart;
|
|
}
|
|
|
|
std::string info;
|
|
if(pgi) {
|
|
info = pgi->getInfo(ni.space());
|
|
}
|
|
|
|
auto node = connector->createNode(this_node, parent, alt, alts, ns)
|
|
.set_label(label)
|
|
.set_info(info);
|
|
connector->sendNode(node);
|
|
}
|
|
|
|
void
|
|
CPProfilerSearchTracer::done(void) {
|
|
connector->disconnect();
|
|
}
|
|
|
|
CPProfilerSearchTracer::~CPProfilerSearchTracer(void) {
|
|
delete connector;
|
|
delete pgi;
|
|
}
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
// STATISTICS: search-trace
|