79 lines
2.5 KiB
C++
79 lines
2.5 KiB
C++
/* 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 "ICmpAIManager.h"
|
|
|
|
#include "simulation2/system/InterfaceScripted.h"
|
|
|
|
#include "lib/file/vfs/vfs_util.h"
|
|
#include "ps/Filesystem.h"
|
|
|
|
BEGIN_INTERFACE_WRAPPER(AIManager)
|
|
DEFINE_INTERFACE_METHOD_3("AddPlayer", void, ICmpAIManager, AddPlayer, std::wstring, player_id_t, uint8_t)
|
|
DEFINE_INTERFACE_METHOD_0("TryLoadSharedComponent", void, ICmpAIManager, TryLoadSharedComponent)
|
|
DEFINE_INTERFACE_METHOD_0("RunGamestateInit", void, ICmpAIManager, RunGamestateInit)
|
|
END_INTERFACE_WRAPPER(AIManager)
|
|
|
|
// Implement the static method that finds all AI scripts
|
|
// that can be loaded via AddPlayer:
|
|
|
|
struct GetAIsHelper
|
|
{
|
|
NONCOPYABLE(GetAIsHelper);
|
|
public:
|
|
GetAIsHelper(ScriptInterface& scriptInterface) :
|
|
m_ScriptInterface(scriptInterface)
|
|
{
|
|
}
|
|
|
|
void Run()
|
|
{
|
|
vfs::ForEachFile(g_VFS, L"simulation/ai/", Callback, (uintptr_t)this, L"*.json", vfs::DIR_RECURSIVE);
|
|
}
|
|
|
|
static Status Callback(const VfsPath& pathname, const CFileInfo& UNUSED(fileInfo), const uintptr_t cbData)
|
|
{
|
|
GetAIsHelper* self = (GetAIsHelper*)cbData;
|
|
|
|
// Extract the 3rd component of the path (i.e. the directory after simulation/ai/)
|
|
fs::wpath components = pathname.string();
|
|
fs::wpath::iterator it = components.begin();
|
|
std::advance(it, 2);
|
|
std::wstring dirname = GetWstringFromWpath(*it);
|
|
|
|
CScriptValRooted ai;
|
|
self->m_ScriptInterface.Eval("({})", ai);
|
|
self->m_ScriptInterface.SetProperty(ai.get(), "id", dirname, true);
|
|
self->m_ScriptInterface.SetProperty(ai.get(), "data", self->m_ScriptInterface.ReadJSONFile(pathname), true);
|
|
self->m_AIs.push_back(ai);
|
|
|
|
return INFO::OK;
|
|
}
|
|
|
|
std::vector<CScriptValRooted> m_AIs;
|
|
ScriptInterface& m_ScriptInterface;
|
|
};
|
|
|
|
std::vector<CScriptValRooted> ICmpAIManager::GetAIs(ScriptInterface& scriptInterface)
|
|
{
|
|
GetAIsHelper helper(scriptInterface);
|
|
helper.Run();
|
|
return helper.m_AIs;
|
|
}
|