CCmpVision.cpp

Go to the documentation of this file.

/* Copyright (C) 2012 Wildfire Games.
 * This file is part of 0 A.D.
 *
 * 0 A.D. is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 2 of the License, or
 * (at your option) any later version.
 *
 * 0 A.D. 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 0 A.D.  If not, see <http://www.gnu.org/licenses/>.
 */

#include "precompiled.h"

#include "simulation2/system/Component.h"
#include "ICmpVision.h"

#include "simulation2/MessageTypes.h"
#include "simulation2/components/ICmpOwnership.h"
#include "simulation2/components/ICmpPlayerManager.h"
#include "simulation2/components/ICmpTechnologyManager.h"

class CCmpVision : public ICmpVision
{
public:
    static void ClassInit(CComponentManager& componentManager)
    {
        componentManager.SubscribeGloballyToMessageType(MT_TechnologyModification);
    }

    DEFAULT_COMPONENT_ALLOCATOR(Vision)

    // Template state:

    entity_pos_t m_Range, m_BaseRange;
    bool m_RetainInFog;
    bool m_AlwaysVisible;

    static std::string GetSchema()
    {
        return
            "<element name='Range'>"
                "<data type='nonNegativeInteger'/>"
            "</element>"
            "<element name='RetainInFog'>"
                "<data type='boolean'/>"
            "</element>"
            "<element name='AlwaysVisible'>"
                "<data type='boolean'/>"
            "</element>";
    }

    virtual void Init(const CParamNode& paramNode)
    {
        m_BaseRange = m_Range = paramNode.GetChild("Range").ToFixed();
        m_RetainInFog = paramNode.GetChild("RetainInFog").ToBool();
        m_AlwaysVisible = paramNode.GetChild("AlwaysVisible").ToBool();
    }

    virtual void Deinit()
    {
    }

    virtual void Serialize(ISerializer& UNUSED(serialize))
    {
        // No dynamic state to serialize
    }

    virtual void Deserialize(const CParamNode& paramNode, IDeserializer& UNUSED(deserialize))
    {
        Init(paramNode);
    }

    virtual void HandleMessage(const CMessage& msg, bool UNUSED(global))
    {
        switch (msg.GetType())
        {
        case MT_TechnologyModification:
        {
            const CMessageTechnologyModification& msgData = static_cast<const CMessageTechnologyModification&> (msg);
            if (msgData.component == L"Vision")
            {
                CmpPtr<ICmpOwnership> cmpOwnership(GetEntityHandle());
                if (cmpOwnership)
                {
                    player_id_t owner = cmpOwnership->GetOwner();
                    if (owner != INVALID_PLAYER && owner == msgData.player)
                    {
                        CmpPtr<ICmpPlayerManager> cmpPlayerManager(GetSystemEntity());
                        entity_id_t playerEnt = cmpPlayerManager->GetPlayerByID(owner);
                        CmpPtr<ICmpTechnologyManager> cmpTechnologyManager(GetSimContext(), playerEnt);
                        if (playerEnt != INVALID_ENTITY && cmpTechnologyManager)
                        {
                            entity_pos_t newRange = cmpTechnologyManager->ApplyModifications(L"Vision/Range", m_BaseRange, GetEntityId());
                            if (newRange != m_Range)
                            {
                                // Update our vision range and broadcast message
                                entity_pos_t oldRange = m_Range;
                                m_Range = newRange;
                                CMessageVisionRangeChanged msg(GetEntityId(), oldRange, newRange);
                                GetSimContext().GetComponentManager().BroadcastMessage(msg);
                            }
                        }
                    }
                }
            }
            break;
        }
        }
    }

    virtual entity_pos_t GetRange()
    {
        return m_Range;
    }

    virtual bool GetRetainInFog()
    {
        return m_RetainInFog;
    }

    virtual bool GetAlwaysVisible()
    {
        return m_AlwaysVisible;
    }
};

REGISTER_COMPONENT_TYPE(Vision)