CmpPtr.h

Go to the documentation of this file.

/* Copyright (C) 2015 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/>.
 */

#ifndef INCLUDED_CMPPTR
#define INCLUDED_CMPPTR

#include "simulation2/system/Entity.h"

class CSimContext;
class CSimulation2;
class IComponent;

// Helper functions for CmpPtr
IComponent* QueryInterface(const CSimContext& context, entity_id_t ent, int iid);
IComponent* QueryInterface(const CSimulation2& simulation, entity_id_t ent, int iid);

/**
 * A simplified syntax for accessing entity components.
 * E.g. to get the @c Position component, write:
 *
 * @code
 * CmpPtr<ICmpPosition> cmpPosition(context, ent);
 * if (!cmpPosition)
 *     // do something (maybe just silently abort; you should never crash if the
 *     // component is missing, even if you're sure it should never be missing)
 * @endcode
 *
 * where @c context is (if you're writing component code) a CSimContext object, or
 * (if you're writing external engine code that makes use of the simulation system)
 * a CSimulation2 object; and @c ent is the entity ID.
 *
 * @c ent can be CComponentManager::SYSTEM_ENTITY (if you're writing a component), or
 * CSimulation2::SYSTEM_ENTITY (for external code), if you want to access the global
 * singleton system components.
 *
 * You should never hold onto a component pointer outside of the method in which you acquire
 * it, because it might get deleted and invalidate your pointer. (Components will never be
 * deleted while inside a simulation method.)
 */
template<typename T>
class CmpPtr
{
private:
    T* m;

public:
    CmpPtr(const CSimContext& context, entity_id_t ent)
    {
        m = static_cast<T*>(QueryInterface(context, ent, T::GetInterfaceId()));
    }

    CmpPtr(const CSimulation2& simulation, entity_id_t ent)
    {
        m = static_cast<T*>(QueryInterface(simulation, ent, T::GetInterfaceId()));
    }

    CmpPtr(CEntityHandle ent)
    {
        SEntityComponentCache* cache = ent.GetComponentCache();
        if (cache != NULL && T::GetInterfaceId() < (int)cache->numInterfaces)
            m = static_cast<T*>(cache->interfaces[T::GetInterfaceId()]);
        else
            m = NULL;
    }

    T* operator->() { return m; }

    explicit operator bool() const
    {
        return m != NULL;
    }
};

#endif // INCLUDED_CMPPTR