libsumo/MeanData.cpp

Go to the documentation of this file.

/****************************************************************************/
// Eclipse SUMO, Simulation of Urban MObility; see https://eclipse.dev/sumo
// Copyright (C) 2017-2026 German Aerospace Center (DLR) and others.
// This program and the accompanying materials are made available under the
// terms of the Eclipse Public License 2.0 which is available at
// https://www.eclipse.org/legal/epl-2.0/
// This Source Code may also be made available under the following Secondary
// Licenses when the conditions for such availability set forth in the Eclipse
// Public License 2.0 are satisfied: GNU General Public License, version 2
// or later which is available at
// https://www.gnu.org/licenses/old-licenses/gpl-2.0-standalone.html
// SPDX-License-Identifier: EPL-2.0 OR GPL-2.0-or-later
/****************************************************************************/
// C++ TraCI client API implementation
/****************************************************************************/
#include <config.h>
 
#include <microsim/MSNet.h>
#include <microsim/MSEdge.h>
#include <microsim/output/MSDetectorControl.h>
#include <microsim/output/MSMeanData.h>
#include <libsumo/TraCIConstants.h>
#include "StorageHelper.h"
#include "Helper.h"
#include "MeanData.h"
 
 
namespace libsumo {
// ===========================================================================
// static member initializations
// ===========================================================================
SubscriptionResults MeanData::mySubscriptionResults;
ContextSubscriptionResults MeanData::myContextSubscriptionResults;
 
 
// ===========================================================================
// static member definitions
// ===========================================================================
std::vector<std::string>
MeanData::getIDList() {
    std::vector<std::string> ids;
    for (const auto& item : MSNet::getInstance()->getDetectorControl().getMeanData()) {
        ids.push_back(item.first);
    }
    std::sort(ids.begin(), ids.end());
    return ids;
}
 
int
MeanData::getIDCount() {
    return (int)getIDList().size();
}
 
 
double
MeanData::getAttributeValue(const std::string& meanDataID, const std::string& laneID, const std::string& attr) {
    if (!SUMOXMLDefinitions::Attrs.hasString(attr)) {
        throw TraCIException("Unknown Meandata attribute '" + attr + "'.");
    }
    MSLane* const lane = MSLane::dictionary(laneID);
    if (lane == nullptr) {
        throw TraCIException("Lane '" + laneID + "' is not known");
    }
    MSMeanData* md = getMeanData(meanDataID);
    SumoXMLAttr a = (SumoXMLAttr)SUMOXMLDefinitions::Attrs.get(attr);
    return md->getAttributeValue(lane, a, INVALID_DOUBLE_VALUE);
}
 
 
std::vector<std::string>
MeanData::getIDs(const std::string&meanDataID) {
    MSMeanData* md = getMeanData(meanDataID);
    std::vector<std::string> result;
    if (md->isEdgeData()) {
        for (const MSEdge* e : md->getEdges()) {
            result.push_back(e->getID());
        }
    } else {
        for (const MSEdge* e : md->getEdges()) {
            for (const MSLane* lane : e->getLanes()) {
                result.push_back(lane->getID());
            }
        }
    }
    return result;
}
 
 
std::vector<double>
MeanData::getAttributeValues(const std::string& meanDataID, const std::string& attr) {
    if (!SUMOXMLDefinitions::Attrs.hasString(attr)) {
        throw TraCIException("Unknown Meandata attribute '" + attr + "'.");
    }
    MSMeanData* md = getMeanData(meanDataID);
    SumoXMLAttr a = (SumoXMLAttr)SUMOXMLDefinitions::Attrs.get(attr);
    std::vector<double> result;
    if (md->isEdgeData()) {
        for (const MSEdge* e : md->getEdges()) {
            result.push_back(md->getAttributeValue(e->getLanes().front(), a, INVALID_DOUBLE_VALUE));
        }
    } else {
        for (const MSEdge* e : md->getEdges()) {
            for (const MSLane* lane : e->getLanes()) {
                result.push_back(md->getAttributeValue(lane, a, INVALID_DOUBLE_VALUE));
            }
        }
    }
    return result;
}
 
 
std::string
MeanData::getParameter(const std::string& /* dataID */, const std::string& /* param */) {
    return "";
}
 
LIBSUMO_GET_PARAMETER_WITH_KEY_IMPLEMENTATION(MeanData)
 
void
MeanData::setParameter(const std::string& /* dataID */, const std::string& /* key */, const std::string& /* value */) {
    //MSMeanData* r = const_cast<MSMeanData*>(getMeanData(dataID));
    //r->setParameter(key, value);
}
 
 
LIBSUMO_SUBSCRIPTION_IMPLEMENTATION(MeanData, MEANDATA)
 
 
MSMeanData*
MeanData::getMeanData(const std::string& id) {
    auto mdMap = MSNet::getInstance()->getDetectorControl().getMeanData();
    auto it = mdMap.find(id);
    if (it == mdMap.end() || it->second.size() == 0) {
        throw TraCIException("MeanData '" + id + "' is not known");
    }
    if (it->second.size() > 1) {
        WRITE_WARNINGF(TL("Found % meanData definitions with id '%'."), toString(it->second.size()), id);
    }
    return it->second.front();
}
 
 
std::shared_ptr<VariableWrapper>
MeanData::makeWrapper() {
    return std::make_shared<Helper::SubscriptionWrapper>(handleVariable, mySubscriptionResults, myContextSubscriptionResults);
}
 
 
bool
MeanData::handleVariable(const std::string& objID, const int variable, VariableWrapper* wrapper, tcpip::Storage* paramData) {
    switch (variable) {
        case TRACI_ID_LIST:
            return wrapper->wrapStringList(objID, variable, getIDList());
        case ID_COUNT:
            return wrapper->wrapInt(objID, variable, getIDCount());
        case libsumo::VAR_PARAMETER:
            paramData->readUnsignedByte();
            return wrapper->wrapString(objID, variable, getParameter(objID, paramData->readString()));
        case libsumo::VAR_PARAMETER_WITH_KEY:
            paramData->readUnsignedByte();
            return wrapper->wrapStringPair(objID, variable, getParameterWithKey(objID, paramData->readString()));
        case libsumo::VAR_MEANDATA_LANE: {
            StoHelp::readCompound(*paramData);
            const std::string laneID = StoHelp::readTypedString(*paramData);
            const std::string attr = StoHelp::readTypedString(*paramData);
            return wrapper->wrapDouble(objID, variable, getAttributeValue(objID, laneID, attr));
        }
        case libsumo::VAR_MEANDATA_IDS: 
            return wrapper->wrapStringList(objID, variable, getIDs(objID));
        case libsumo::VAR_MEANDATA_VALUES: {
            return wrapper->wrapDoubleList(objID, variable, getAttributeValues(objID, StoHelp::readTypedString(*paramData)));
        }
        default:
            return false;
    }
}
 
}
 
 
/****************************************************************************/