#include <config.h>
#include <string>
#include <utils/options/OptionsCont.h>
#include <utils/common/MsgHandler.h>
#include <netbuild/NBLoadedTLDef.h>
#include <netbuild/NBTrafficLightLogicCont.h>
#include <netbuild/NBEdgeCont.h>
#include "NIVisumTL.h"
NIVisumTL::NIVisumTL(const std::string& name, SUMOTime cycleTime, SUMOTime offset,
SUMOTime intermediateTime, bool phaseDefined)
: myName(name), myCycleTime(cycleTime), myOffset(offset),
myIntermediateTime(intermediateTime), myPhaseDefined(phaseDefined) {
}
NIVisumTL::~NIVisumTL() {
for (std::map<std::string, Phase*>::iterator i = myPhases.begin(); i != myPhases.end(); ++i) {
delete i->second;
}
for (std::map<std::string, SignalGroup*>::iterator i = mySignalGroups.begin(); i != mySignalGroups.end(); ++i) {
delete i->second;
}
}
void
NIVisumTL::addSignalGroup(const std::string& name, SUMOTime startTime, SUMOTime endTime, SUMOTime yellowTime) {
mySignalGroups[name] = new NIVisumTL::SignalGroup(name, startTime, endTime, yellowTime);
}
void
NIVisumTL::addPhase(const std::string& name, SUMOTime startTime, SUMOTime endTime, SUMOTime yellowTime) {
myPhases[name] = new NIVisumTL::Phase(startTime, endTime, yellowTime);
}
NIVisumTL::SignalGroup&
NIVisumTL::getSignalGroup(const std::string& name) {
return *mySignalGroups.find(name)->second;
}
void
NIVisumTL::build(NBEdgeCont& ec, NBTrafficLightLogicCont& tlc) {
for (std::vector<NBNode*>::iterator ni = myNodes.begin(); ni != myNodes.end(); ni++) {
NBNode* node = (*ni);
if (node == nullptr) {
WRITE_WARNINGF(TL("invalid node for traffic light '%'"), myName);
continue;
}
TrafficLightType type = SUMOXMLDefinitions::TrafficLightTypes.get(OptionsCont::getOptions().getString("tls.default-type"));
NBLoadedTLDef* def = new NBLoadedTLDef(ec, node->getID(), node, myOffset, type);
tlc.insert(def);
def->setCycleDuration(myCycleTime);
for (std::map<std::string, SignalGroup*>::iterator gi = mySignalGroups.begin(); gi != mySignalGroups.end(); gi++) {
std::string groupName = (*gi).first;
NIVisumTL::SignalGroup& SG = *(*gi).second;
def->addSignalGroup(groupName);
def->addToSignalGroup(groupName, SG.connections());
SUMOTime yellowTime = -1;
if (myPhaseDefined) {
for (std::map<std::string, Phase*>::iterator pi = SG.phases().begin(); pi != SG.phases().end(); pi++) {
NIVisumTL::Phase& PH = *(*pi).second;
def->addSignalGroupPhaseBegin(groupName, PH.getStartTime(), NBTrafficLightDefinition::TLCOLOR_GREEN);
def->addSignalGroupPhaseBegin(groupName, PH.getEndTime(), NBTrafficLightDefinition::TLCOLOR_RED);
yellowTime = MAX2(PH.getYellowTime(), yellowTime);
}
} else {
def->addSignalGroupPhaseBegin(groupName, SG.getStartTime(), NBTrafficLightDefinition::TLCOLOR_GREEN);
def->addSignalGroupPhaseBegin(groupName, SG.getEndTime(), NBTrafficLightDefinition::TLCOLOR_RED);
yellowTime = MAX2(SG.getYellowTime(), yellowTime);
}
def->setSignalYellowTimes(groupName, myIntermediateTime, yellowTime);
}
}
}