hierarchy.h

Go to the documentation of this file.

#pragma once
 
#include <string>
#include <vector>
#include <map>
#include <blam/api/api.h>
 
#ifdef SAPIEN_LIB_EXPORTS
#define SAPIEN_LIB_API __declspec(dllexport) 
#else
#define SAPIEN_LIB_API __declspec(dllimport) 
#endif
 
class BlamScenarioHierarchyNode_Object;
 
enum class BlamScenarioHierarchyNodeType
{
    Group,     
    Object,    
    Undefined  
};
 
class SAPIEN_LIB_API BlamScenarioHierarchyNode
{
private:
    BlamScenarioHierarchyNodeType node_type = BlamScenarioHierarchyNodeType::Undefined; 
 
protected:
    bool use_custom_name = false; 
 
    std::string display_name_override = "unspecified";
public:
    BlamScenarioHierarchyNode* parent_node = nullptr; 
 
    BlamScenarioHierarchyNode(BlamScenarioHierarchyNodeType _node_type);
 
    BlamScenarioHierarchyNode(BlamScenarioHierarchyNodeType _node_type, std::string _display_name_override);
 
    virtual ~BlamScenarioHierarchyNode() {};
 
    BlamScenarioHierarchyNodeType GetType();
 
    virtual std::string GetName();
};
 
class SAPIEN_LIB_API BlamScenarioHierarchyNode_Group : public BlamScenarioHierarchyNode
{
private:
    std::vector<BlamScenarioHierarchyNode*> child_nodes = std::vector<BlamScenarioHierarchyNode*>(); 
 
public:
    int folder_id = -1; 
 
    BlamScenarioHierarchyNode_Group(std::string group_name, int _folder_id);
 
    ~BlamScenarioHierarchyNode_Group();
 
    void ReleaseData();
 
    void AddChild(BlamScenarioHierarchyNode* node);
 
    BlamScenarioHierarchyNode* GetChildAt(int index);
 
    int GetChildCount();
 
    void RemoveNodeAt(int index, bool delete_object = true);
 
    void RemoveNode(BlamScenarioHierarchyNode* node, bool delete_object = true);
 
    bool HasChild(BlamScenarioHierarchyNode* node);
 
    std::vector<BlamScenarioHierarchyNode_Object*> GetAllObjectNodes();
};
 
class SAPIEN_LIB_API BlamScenarioHierarchyNode_Object : public BlamScenarioHierarchyNode
{
public:
    BlamWorldObject* world_object = nullptr; 
 
    BlamScenarioHierarchyNode_Object();
 
    BlamScenarioHierarchyNode_Object(std::string custom_name);
 
    ~BlamScenarioHierarchyNode_Object();
 
    std::string GetName();
 
    std::string GetTypeLabel();
 
    std::string GetClassLabel();
};
 
class SAPIEN_LIB_API BlamScenarioHierarchy
{
public:
 
    BlamScenarioHierarchyNode_Group* scenario_root_node = nullptr;
 
    std::map<int, BlamScenarioHierarchyNode_Group*> folder_ids = std::map<int, BlamScenarioHierarchyNode_Group*>();
 
    BlamScenarioHierarchy();
 
    ~BlamScenarioHierarchy();
 
    void Refresh();
 
    void ReleaseData();
 
    std::vector<BlamScenarioHierarchyNode_Object*> GetAllObjectNodes();
};
 
namespace Sapien::Hierarchy
{
    SAPIEN_LIB_API void PrepareHierarchy();
 
    SAPIEN_LIB_API void ReleaseHierarchy();
 
    SAPIEN_LIB_API BlamScenarioHierarchy* GetHierarchy();
 
    SAPIEN_LIB_API void SaveScenario();
 
    SAPIEN_LIB_API int ReserveFolderIndex();
}