You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
83 lines
2.7 KiB
83 lines
2.7 KiB
#include "PityModel.hh"
|
|
#include "PityUnit.hh"
|
|
#include "PitySwarm.hh"
|
|
|
|
#include <vector>
|
|
#include <functional>
|
|
#include <memory>
|
|
|
|
namespace pEp {
|
|
namespace PityTest11 {
|
|
bool PitySwarm::debug_log_enabled = false;
|
|
|
|
void createPerspective(const PityModel& model, PityPerspective* psp, int node_nr)
|
|
{
|
|
psp->name = model.nodeNr(node_nr)->getName();
|
|
|
|
// Default partner is next node, its a circle
|
|
int partner_node_index = (node_nr + 1) % model.nodes().size();
|
|
psp->partner = model.nodes().at(partner_node_index)->getName();
|
|
|
|
// Create peers, everyone but me
|
|
auto nodes = model.nodes();
|
|
for (int i = 0; i < nodes.size(); i++) {
|
|
if (i != node_nr) {
|
|
psp->peers.push_back(nodes.at(i)->getName());
|
|
}
|
|
}
|
|
}
|
|
|
|
PitySwarm::PitySwarm(PityModel& model) : _model{ model }
|
|
{
|
|
pEpLogClass("called");
|
|
// Create perspective
|
|
for (auto n : _model.nodes()) {
|
|
auto tmp = std::make_shared<PityPerspective>();
|
|
createPerspective(_model, tmp.get(), n->getNr());
|
|
_perspectives.push_back(tmp);
|
|
}
|
|
|
|
// Construct Tree
|
|
_rootUnit = std::make_shared<PityUnit<PityModel, PityPerspective>>(
|
|
nullptr,
|
|
_model.getName(),
|
|
nullptr,
|
|
&_model);
|
|
|
|
for (auto n : _model.nodes()) {
|
|
_nodeUnits.push_back(std::make_shared<PityUnit<>>(
|
|
_rootUnit.get(),
|
|
n->getName(),
|
|
nullptr,
|
|
// std::bind(
|
|
// &PityNode::_init,
|
|
// this,
|
|
// std::placeholders::_1,
|
|
// std::placeholders::_2,
|
|
// std::placeholders::_3),
|
|
&_model,
|
|
_perspectives.at(n->getNr()).get(),
|
|
PityUnit<>::ExecutionMode::PROCESS_PARALLEL));
|
|
}
|
|
}
|
|
|
|
void PitySwarm::addTestUnit(
|
|
int nodeNr,
|
|
const std::string& name,
|
|
std::function<void(PityUnit<>&, PityModel*, PityPerspective*)> test_func)
|
|
{
|
|
auto tmp = std::make_shared <PityUnit<>>(_nodeUnits.at(nodeNr).get(), name, test_func);
|
|
_testUnits.push_back(tmp);
|
|
}
|
|
|
|
|
|
|
|
void PitySwarm::run()
|
|
{
|
|
|
|
_rootUnit->run();
|
|
}
|
|
|
|
|
|
} // namespace PityTest11
|
|
} // namespace pEp
|
|
|