PredictionEntity.cpp

Go to the documentation of this file.

/*
 * This file is part of ArmarX.
 *
 * Copyright (C) 2012-2016, High Performance Humanoid Technologies (H2T), Karlsruhe Institute of Technology (KIT), all rights reserved.
 *
 * ArmarX is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2 as
 * published by the Free Software Foundation.
 *
 * ArmarX is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 *
 * @package    MemoryX::Core
 * @author     Manfred Kroehnert (Manfred dot Kroehnert at kit dot edu)
 * @date       2015
 * @copyright  http://www.gnu.org/licenses/gpl-2.0.txt
 *             GNU General Public License
 */
 
#include "PredictionEntity.h"
 
#include "../../../../core/entity/EntityRef.h"
 
namespace memoryx
{
    const std::string PredictionEntity::SNAPSHOT_ATTRIBUTE("memorySnapshot");
    const std::string PredictionEntity::SOURCE_SNAPSHOT_ATTRIBUTE("sourceMemorySnapshot");
    const std::string PredictionEntity::PREDICTION_ATTRIBUTE("predictedMemorySnapshot");
    const std::string PredictionEntity::PREDICTION_LIST_ATTRIBUTE("predictedMemorySnapshotList");
    const std::string PredictionEntity::PREDICTION_METHOD_NAME_ATTRIBUTE("predictionMethodName");
    const std::string PredictionEntity::TASK_COUNT_ATTRIBUTE("taskCount");
    const std::string PredictionEntity::PREDICTION_SCORE_ATTRIBUTE("predictionScore");
 
    PredictionEntity::PredictionEntity() : Entity()
    {
        initializeAttributes();
    }
 
    PredictionEntity::PredictionEntity(const EntityRefBasePtr& memorySnapshotRef,
                                       const EntityRefBasePtr& sourceMemorySnapshotRef,
                                       const EntityRefBasePtr& predictedMemorySnapshotRef,
                                       const EntityRefBaseList& predictedMemorySnapshotRefList,
                                       const std::string& predictionMethodName,
                                       Ice::Int taskCount,
                                       Ice::Float predictionScore) :
        Entity()
    {
        initializeAttributes();
        setProfilerMemorySnapshotRef(memorySnapshotRef);
        setSourceProfilerMemorySnapshotRef(sourceMemorySnapshotRef);
        setPredictedProfilerMemorySnapshotRef(predictedMemorySnapshotRef);
        setPredictedProfilerMemorySnapshotRefList(predictedMemorySnapshotRefList);
        setPredictionMethodName(predictionMethodName);
        setTaskCount(taskCount);
        setPredictionScore(predictionScore);
    }
 
    PredictionEntity::PredictionEntity(const PredictionEntity& source) :
        IceUtil::Shared(source),
        ::armarx::Serializable(source),
        EntityBase(), // dont copy
        PredictionEntityBase(source),
        Entity(source)
    {
    }
 
    PredictionEntity::~PredictionEntity()
    {
    }
 
    void
    PredictionEntity::initializeAttributes()
    {
        putAttribute(new EntityAttribute(PredictionEntity::SNAPSHOT_ATTRIBUTE));
        putAttribute(new EntityAttribute(PredictionEntity::SOURCE_SNAPSHOT_ATTRIBUTE));
        putAttribute(new EntityAttribute(PredictionEntity::PREDICTION_ATTRIBUTE));
        putAttribute(new EntityAttribute(PredictionEntity::PREDICTION_LIST_ATTRIBUTE));
        putAttribute(new EntityAttribute(PredictionEntity::PREDICTION_METHOD_NAME_ATTRIBUTE));
        putAttribute(new EntityAttribute(PredictionEntity::TASK_COUNT_ATTRIBUTE));
        putAttribute(new EntityAttribute(PredictionEntity::PREDICTION_SCORE_ATTRIBUTE));
    }
 
    void
    PredictionEntity::output(std::ostream& stream) const
    {
        Entity::output(stream);
    }
 
    Ice::ObjectPtr
    PredictionEntity::ice_clone() const
    {
        return this->clone();
    }
 
    PredictionEntityPtr
    PredictionEntity::clone(const Ice::Current& context) const
    {
        std::shared_lock entityLock(entityMutex);
        std::scoped_lock attributesLock(attributesMutex);
        std::scoped_lock wrappersLock(wrappersMutex);
        return new PredictionEntity(*this);
    }
 
    EntityRefBasePtr
    PredictionEntity::getProfilerMemorySnapshotRef(const Ice::Current&) const
    {
        return getAttributeValue(PredictionEntity::SNAPSHOT_ATTRIBUTE)->get<EntityRef>();
    }
 
    void
    PredictionEntity::setProfilerMemorySnapshotRef(const EntityRefBasePtr& memorySnapshotRef,
                                                   const Ice::Current&)
    {
        getAttribute(PredictionEntity::SNAPSHOT_ATTRIBUTE)
            ->setValue(new armarx::Variant(memorySnapshotRef));
    }
 
    EntityRefBasePtr
    PredictionEntity::getSourceProfilerMemorySnapshotRef(const Ice::Current&) const
    {
        return getAttributeValue(PredictionEntity::SOURCE_SNAPSHOT_ATTRIBUTE)->get<EntityRef>();
    }
 
    void
    PredictionEntity::setSourceProfilerMemorySnapshotRef(
        const EntityRefBasePtr& sourceMemorySnapshotRef,
        const Ice::Current&)
    {
        getAttribute(PredictionEntity::SOURCE_SNAPSHOT_ATTRIBUTE)
            ->setValue(new armarx::Variant(sourceMemorySnapshotRef));
    }
 
    EntityRefBasePtr
    PredictionEntity::getPredictedProfilerMemorySnapshotRef(const Ice::Current&) const
    {
        return getAttributeValue(PredictionEntity::PREDICTION_ATTRIBUTE)->get<EntityRef>();
    }
 
    void
    PredictionEntity::setPredictedProfilerMemorySnapshotRef(
        const EntityRefBasePtr& predictedMemorySnapshotRef,
        const Ice::Current&)
    {
        getAttribute(PredictionEntity::PREDICTION_ATTRIBUTE)
            ->setValue(new armarx::Variant(predictedMemorySnapshotRef));
    }
 
    EntityRefBaseList
    PredictionEntity::getPredictedProfilerMemorySnapshotRefList(const Ice::Current&) const
    {
        EntityAttributeBasePtr attribute =
            getAttribute(PredictionEntity::PREDICTION_LIST_ATTRIBUTE);
        EntityRefBaseList entityList;
        for (int i = 0; i < attribute->size(); i++)
        {
            armarx::VariantPtr attributeValue =
                armarx::VariantPtr::dynamicCast(attribute->getValueAt(i));
            entityList.push_back(attributeValue->get<EntityRef>());
        }
        return entityList;
    }
 
    void
    PredictionEntity::setPredictedProfilerMemorySnapshotRefList(
        const EntityRefBaseList& predictedMemorySnapshotRefList,
        const Ice::Current&)
    {
        EntityAttributeBasePtr attribute =
            getAttribute(PredictionEntity::PREDICTION_LIST_ATTRIBUTE);
        attribute->clear();
        for (const EntityRefBasePtr& entityRef : predictedMemorySnapshotRefList)
        {
            attribute->addValue(new armarx::Variant(entityRef));
        }
    }
 
    std::string
    PredictionEntity::getPredictionMethodName(const Ice::Current&) const
    {
        return getAttribute(PredictionEntity::PREDICTION_METHOD_NAME_ATTRIBUTE)
            ->getValue()
            ->getString();
    }
 
    void
    PredictionEntity::setPredictionMethodName(const std::string& predictionMethodName,
                                              const Ice::Current&)
    {
        getAttribute(PredictionEntity::PREDICTION_METHOD_NAME_ATTRIBUTE)
            ->setValue(new armarx::Variant(predictionMethodName));
    }
 
    Ice::Int
    PredictionEntity::getTaskCount(const Ice::Current&) const
    {
        return getAttributeValue(PredictionEntity::TASK_COUNT_ATTRIBUTE)->getInt();
    }
 
    void
    PredictionEntity::setTaskCount(Ice::Int taskCount, const Ice::Current&)
    {
        getAttribute(PredictionEntity::TASK_COUNT_ATTRIBUTE)
            ->setValue(new armarx::Variant(taskCount));
    }
 
    Ice::Float
    PredictionEntity::getPredictionScore(const Ice::Current&) const
    {
        return getAttributeValue(PredictionEntity::PREDICTION_SCORE_ATTRIBUTE)->getFloat();
    }
 
    void
    PredictionEntity::setPredictionScore(Ice::Float predictionScore, const Ice::Current&)
    {
        getAttribute(PredictionEntity::PREDICTION_SCORE_ATTRIBUTE)
            ->setValue(new armarx::Variant(predictionScore));
    }
} // namespace memoryx