world.h

Go to the documentation of this file.

#pragma once
//------------------------------------------------------------------------------
//------------------------------------------------------------------------------
#include "core/refcounted.h"
#include "entity.h"
#include "entitypool.h"
#include "component.h"
#include "memdb/tablesignature.h"
#include "util/queue.h"
#include "category.h"
#include "processorid.h"
#include "processor.h"
#include "memory/arenaallocator.h"
#include "frameevent.h"
#include "util/blob.h"
 
namespace MemDb { class Database; }
 
namespace Game
{
 
class PackedLevel;
 
template <typename COMPONENT_TYPE>
ComponentId RegisterType(ComponentRegisterInfo<COMPONENT_TYPE> info = {});
 
//------------------------------------------------------------------------------
struct EntityCreateInfo
{
    TemplateId templateId = TemplateId::Invalid();
    bool immediate = false;
};
 
//------------------------------------------------------------------------------
class World
{
public:
    // Generally, only the game server should create worlds
    World(uint32_t hash);
    ~World();
 
    Entity CreateEntity(bool immediate = true);
    Entity CreateEntity(EntityCreateInfo const& info);
    void DeleteEntity(Entity entity);
    bool IsValid(Entity entity);
    bool HasInstance(Entity entity);
    EntityMapping GetEntityMapping(Entity entity);
    bool HasComponent(Entity entity, ComponentId component);
    template <typename TYPE>
    bool HasComponent(Entity entity);
    MemDb::RowId GetInstance(Entity entity);
    template <typename TYPE>
    void RemoveComponent(Entity entity);
    void RemoveComponent(Entity entity, ComponentId component);
    template <typename TYPE>
    void SetComponent(Entity entity, TYPE const& value);
    template <typename TYPE>
    TYPE GetComponent(Entity entity);
    template <typename TYPE>
    TYPE* AddComponent(Entity entity);
    void* AddComponent(Entity entity, ComponentId component);
    ComponentDecayBuffer const GetDecayBuffer(ComponentId component);
 
    void SetComponentValue(Entity entity, ComponentId component, void* value, uint64_t size);
    
    void ReinitializeComponent(Entity entity, ComponentId component, void* value, uint64_t size);
    
    Dataset Query(Filter filter);
    Dataset Query(Filter filter, Util::Array<MemDb::TableId>& tids);
 
    Ptr<MemDb::Database> GetDatabase();
    MemDb::TableId CreateEntityTable(EntityTableCreateInfo const& info);
 
    FramePipeline& GetFramePipeline();
 
    void MarkAsModified(Game::Entity entity);
 
    PackedLevel* PreloadLevel(Util::String const& path);
    void UnloadLevel(PackedLevel* level);
 
    void ExportLevel(Util::String const& path);
 
    // -- Internal methods -- Use with caution! --
 
    static void Override(World* src, World* dst);
    Entity AllocateEntity();
    void DeallocateEntity(Entity entity);
    MemDb::RowId AllocateInstance(Entity entity, MemDb::TableId table, Util::Blob const* const data = nullptr);
    MemDb::RowId AllocateInstance(Entity entity, BlueprintId blueprint);
    MemDb::RowId AllocateInstance(Entity entity, TemplateId templateId);
    void DeallocateInstance(MemDb::TableId table, MemDb::RowId instance);
    void DeallocateInstance(Entity entity);
    void Defragment(MemDb::TableId tableId);
    void* GetInstanceBuffer(MemDb::TableId const tableId, uint16_t partitionId, ComponentId const component);
    void* GetColumnData(MemDb::TableId const tableId, uint16_t partitionId, MemDb::ColumnIndex const column);
    void ExecuteAddComponentCommands();
    bool componentInitializationEnabled = true;
 
    void RenderDebug();
 
private:
    friend class GameServer;
    friend class BlueprintManager;
    friend class PackedLevel;
    
    struct AllocateInstanceCommand
    {
        Game::Entity entity;
        TemplateId tid;
    };
    struct DeallocInstanceCommand
    {
        Game::Entity entity;
    };
    struct AddStagedComponentCommand
    {
        Game::Entity entity;
        ComponentId componentId;
        SizeT dataSize;
        void* data;
    };
    struct RemoveComponentCommand
    {
        Entity entity;
        ComponentId componentId;
    };
 
    // These functions are called from game server
    void Start();
    void BeginFrame();
    void SimFrame();
    void EndFrame();
    void OnLoad();
    void OnSave();
    void ManageEntities();
    void Reset();
    void PrefilterProcessors();
    bool Prefiltered() const;
    void ClearDecayBuffers();
 
    void ExecuteRemoveComponentCommands();
 
    void AddStagedComponentsToEntity(Entity entity, AddStagedComponentCommand* cmds, SizeT numCmds);
    void RemoveComponentsFromEntity(Entity entity, RemoveComponentCommand* cmds, SizeT numCmds);
 
    SizeT GetNumInstances(MemDb::TableId tid);
    
    MemDb::RowId Migrate(Entity entity, MemDb::TableId newTable);
    void Migrate(
        Util::Array<Entity> const& entities,
        MemDb::TableId fromTable,
        MemDb::TableId newTable,
        Util::FixedArray<MemDb::RowId>& newInstances
    );
 
    void DecayComponent(ComponentId component, MemDb::TableId tableId, MemDb::ColumnIndex column, MemDb::RowId instance);
 
    void MoveInstance(MemDb::Table::Partition* partition, MemDb::RowId from, MemDb::RowId to);
 
    void InitializeAllComponents(Entity entity, MemDb::TableId tableId, MemDb::RowId row);
 
    EntityPool pool;
    SizeT numEntities;
    Util::Array<EntityMapping> entityMap;
    Ptr<MemDb::Database> db;
    uint32_t hash;
    Util::HashTable<BlueprintId, MemDb::TableId> blueprintToTableMap;
    Util::Queue<AllocateInstanceCommand> allocQueue;
    Util::Queue<DeallocInstanceCommand> deallocQueue;
    Util::Array<AddStagedComponentCommand> addStagedQueue;
    Util::Array<RemoveComponentCommand> removeComponentQueue;
 
    Memory::ArenaAllocator<4096_KB> componentStageAllocator;
    
    bool cacheValid = false;
 
    FramePipeline pipeline;
 
    MemDb::TableId defaultTableId;
};
 
//------------------------------------------------------------------------------
template <typename COMPONENT_TYPE>
ComponentId 
RegisterType(ComponentRegisterInfo<COMPONENT_TYPE> info)
{
    uint32_t componentFlags = 0;
    componentFlags |= (uint32_t)COMPONENTFLAG_DECAY * (uint32_t)info.decay;
 
    ComponentInterface* cInterface = new ComponentInterface(
        COMPONENT_TYPE::Traits::name,
        COMPONENT_TYPE(),
        componentFlags
    );
    cInterface->Init = reinterpret_cast<ComponentInterface::ComponentInitFunc>(info.OnInit);
    Game::ComponentId const cid = MemDb::AttributeRegistry::Register<COMPONENT_TYPE>(cInterface);
    Game::ComponentSerialization::Register<COMPONENT_TYPE>(cid);
    Game::ComponentInspection::Register(cid, &Game::ComponentDrawFuncT<COMPONENT_TYPE>);
    return cid;
}
 
//------------------------------------------------------------------------------
template <typename TYPE>
inline void
World::SetComponent(Entity entity, TYPE const& value)
{
#if NEBULA_DEBUG
    n_assert2(
        !this->pipeline.IsRunningAsync(),
        "Settings components in arbitrary entities while executing an async processor is currently not supported!"
    );
#endif
 
#if NEBULA_DEBUG
    n_assert2(
        sizeof(TYPE) == MemDb::AttributeRegistry::TypeSize(Game::GetComponentId<TYPE>()),
        "SetComponent: Provided value's type is not the correct size for the given ComponentId."
    );
#endif
    EntityMapping mapping = this->GetEntityMapping(entity);
    TYPE* ptr = (TYPE*)this->GetInstanceBuffer(mapping.table, mapping.instance.partition, GetComponentId<TYPE>());
    *(ptr + mapping.instance.index) = value;
}
 
//------------------------------------------------------------------------------
template <typename TYPE>
inline TYPE
World::GetComponent(Entity entity)
{
#if NEBULA_DEBUG
    n_assert2(
        !this->pipeline.IsRunningAsync(), "Getting components from entities while executing an async processor is currently not supported!"
    );
#endif
 
#if NEBULA_DEBUG
    n_assert2(
        sizeof(TYPE) == MemDb::AttributeRegistry::TypeSize(Game::GetComponentId<TYPE>()),
        "GetComponent: Provided value's type is not the correct size for the given ComponentId."
    );
#endif
    EntityMapping mapping = this->GetEntityMapping(entity);
    TYPE* ptr = (TYPE*)this->GetInstanceBuffer(mapping.table, mapping.instance.partition, GetComponentId<TYPE>());
    return *(ptr + mapping.instance.index);
}
 
//------------------------------------------------------------------------------
template <typename TYPE>
inline void
World::RemoveComponent(Entity entity)
{
#if NEBULA_DEBUG
    n_assert2(
        !this->pipeline.IsRunningAsync(),
        "Removing components from entities while executing an async processor is currently not supported!"
    );
#endif
    RemoveComponentCommand cmd = {
        .entity = entity,
        .componentId = Game::GetComponentId<TYPE>(),
    };
    this->removeComponentQueue.Append(cmd);
}
 
//------------------------------------------------------------------------------
template <typename TYPE>
inline bool
World::HasComponent(Game::Entity entity)
{
    return this->HasComponent(entity, Game::GetComponentId<TYPE>());
}
 
//------------------------------------------------------------------------------
template <typename TYPE>
inline TYPE*
World::AddComponent(Entity entity)
{
    /*
        We could possibly support this by either having thread local allocators and
        staged queues and gathering after the processors have finished (must clear
        allocators by issuing a separate job for this at the end of the frame), or by
        just introducing a simple mutex.
    */
#if NEBULA_DEBUG
    n_assert2(!this->pipeline.IsRunningAsync(), "Adding component to entities while in an async processor is currently not supported!");
#endif
    Game::ComponentId id = Game::GetComponentId<TYPE>();
#if NEBULA_DEBUG
    n_assert(MemDb::AttributeRegistry::TypeSize(id) == sizeof(TYPE));
    //n_assert(!this->HasComponent<TYPE>(entity));
#endif
    TYPE* data = this->componentStageAllocator.Alloc<TYPE>();
    const void* defaultValue = MemDb::AttributeRegistry::DefaultValue(id);
    Memory::Copy(defaultValue, data, sizeof(TYPE));
 
    AddStagedComponentCommand cmd = {
        .entity = entity,
        .componentId = id,
        .dataSize = sizeof(TYPE),
        .data = data,
    };
    this->addStagedQueue.Append(cmd);
 
    MemDb::Attribute* attr = MemDb::AttributeRegistry::GetAttribute(id);
    ComponentInterface* cInterface;
    cInterface = static_cast<ComponentInterface*>(attr);
 
    if (cInterface->Init != nullptr)
    {
        // run initialization function if it exists.
        cInterface->Init(this, entity, data);
    }
 
    return data;
}
 
template<> Game::Position World::GetComponent(Entity entity);
template<> Game::Orientation World::GetComponent(Entity entity);
template<> Game::Scale World::GetComponent(Entity entity);
 
template<> void World::SetComponent(Entity entity, Game::Position const&);
template<> void World::SetComponent(Entity entity, Game::Orientation const&);
template<> void World::SetComponent(Entity entity, Game::Scale const&);
 
 
} // namespace Game