geode/loader/src/hooks/GeodeNodeMetadata.cpp

180 lines
5.1 KiB
C++
Raw Normal View History

2022-10-30 15:07:49 -04:00
#include <Geode/modify/Field.hpp>
2022-10-13 05:56:23 -04:00
#include <Geode/utils/cocos.hpp>
2022-10-13 04:31:23 -04:00
#include <Geode/modify/Field.hpp>
2022-10-17 09:08:12 -04:00
#include <Geode/modify/CCNode.hpp>
2022-10-30 14:56:36 -04:00
#include <cocos2d.h>
USE_GEODE_NAMESPACE();
using namespace geode::modifier;
#pragma warning(push)
2022-10-30 14:56:36 -04:00
#pragma warning(disable : 4273)
constexpr auto METADATA_TAG = 0xB324ABC;
struct ProxyCCNode;
class GeodeNodeMetadata final : public cocos2d::CCObject {
private:
FieldContainer* m_fieldContainer;
Ref<cocos2d::CCObject> m_userObject;
std::string m_id = "";
2022-10-10 13:58:47 -04:00
std::unique_ptr<Layout> m_layout = nullptr;
2023-02-06 14:36:08 -05:00
std::unique_ptr<LayoutOptions> m_layoutOptions = nullptr;
2022-10-10 13:58:47 -04:00
PositionHint m_positionHint = PositionHint::Default;
2022-10-25 15:57:40 -04:00
std::unordered_map<std::string, std::any> m_attributes;
friend class ProxyCCNode;
friend class cocos2d::CCNode;
GeodeNodeMetadata() : m_fieldContainer(new FieldContainer()) {}
2022-10-30 14:56:36 -04:00
virtual ~GeodeNodeMetadata() {
delete m_fieldContainer;
}
public:
static GeodeNodeMetadata* set(CCNode* target) {
if (!target) return nullptr;
2022-10-30 14:56:36 -04:00
auto old = target->m_pUserObject;
2022-10-30 14:56:36 -04:00
// faster than dynamic_cast, technically can
// but extremely unlikely to fail
if (old && old->getTag() == METADATA_TAG) {
return static_cast<GeodeNodeMetadata*>(old);
}
auto meta = new GeodeNodeMetadata();
meta->autorelease();
meta->setTag(METADATA_TAG);
// set user object
target->m_pUserObject = meta;
meta->retain();
if (old) {
meta->m_userObject = old;
// the old user object has been retained by CCNode
old->release();
}
return meta;
}
FieldContainer* getFieldContainer() {
return m_fieldContainer;
}
};
// proxy forwards
2022-10-30 14:56:36 -04:00
#include <Geode/modify/CCNode.hpp>
struct ProxyCCNode : Modify<ProxyCCNode, CCNode> {
virtual CCObject* getUserObject() {
if (static_cast<CCObject*>(this) == static_cast<CCObject*>(CCDirector::get())) {
// apparently this function is the same as
// CCDirector::getNextScene so yeah
2022-11-17 15:29:08 -05:00
return m_pUserObject;
}
return GeodeNodeMetadata::set(this)->m_userObject;
}
virtual void setUserObject(CCObject* obj) {
GeodeNodeMetadata::set(this)->m_userObject = obj;
}
};
2022-12-03 08:51:46 -05:00
static inline std::unordered_map<size_t, size_t> s_nextIndex;
2022-12-03 08:59:55 -05:00
size_t modifier::getFieldIndexForClass(size_t hash) {
2022-12-03 08:51:46 -05:00
return s_nextIndex[hash]++;
}
2022-10-30 14:56:36 -04:00
// not const because might modify contents
FieldContainer* CCNode::getFieldContainer() {
return GeodeNodeMetadata::set(this)->getFieldContainer();
}
std::string CCNode::getID() {
return GeodeNodeMetadata::set(this)->m_id;
}
void CCNode::setID(std::string const& id) {
GeodeNodeMetadata::set(this)->m_id = id;
}
CCNode* CCNode::getChildByID(std::string const& id) {
2022-10-13 05:56:23 -04:00
for (auto child : CCArrayExt<CCNode>(m_pChildren)) {
if (child->getID() == id) {
return child;
}
}
return nullptr;
}
CCNode* CCNode::getChildByIDRecursive(std::string const& id) {
if (auto child = this->getChildByID(id)) {
return child;
}
2022-10-13 05:56:23 -04:00
for (auto child : CCArrayExt<CCNode>(m_pChildren)) {
if ((child = child->getChildByIDRecursive(id))) {
return child;
}
}
return nullptr;
}
2023-02-02 10:08:13 -05:00
void CCNode::setLayout(Layout* layout, bool apply, bool respectAnchor) {
if (respectAnchor && this->isIgnoreAnchorPointForPosition()) {
for (auto child : CCArrayExt<CCNode>(m_pChildren)) {
child->setPosition(child->getPosition() + this->getScaledContentSize());
}
this->ignoreAnchorPointForPosition(false);
}
2022-10-10 13:58:47 -04:00
GeodeNodeMetadata::set(this)->m_layout.reset(layout);
if (apply) {
this->updateLayout();
}
}
Layout* CCNode::getLayout() {
return GeodeNodeMetadata::set(this)->m_layout.get();
}
2023-02-06 14:36:08 -05:00
void CCNode::setLayoutOptions(LayoutOptions* options, bool apply) {
GeodeNodeMetadata::set(this)->m_layoutOptions.reset(options);
if (apply && m_pParent) {
m_pParent->updateLayout();
}
}
LayoutOptions* CCNode::getLayoutOptions() {
return GeodeNodeMetadata::set(this)->m_layoutOptions.get();
}
void CCNode::updateLayout(bool updateChildOrder) {
if (updateChildOrder) {
this->sortAllChildren();
}
2022-10-10 13:58:47 -04:00
if (auto layout = GeodeNodeMetadata::set(this)->m_layout.get()) {
2023-02-02 10:08:13 -05:00
layout->apply(this);
2022-10-10 13:58:47 -04:00
}
}
void CCNode::setPositionHint(PositionHint hint) {
GeodeNodeMetadata::set(this)->m_positionHint = hint;
}
PositionHint CCNode::getPositionHint() {
return GeodeNodeMetadata::set(this)->m_positionHint;
}
2022-10-25 15:57:40 -04:00
void CCNode::setAttribute(std::string const& attr, std::any value) {
GeodeNodeMetadata::set(this)->m_attributes[attr] = value;
}
std::optional<std::any> CCNode::getAttributeInternal(std::string const& attr) {
auto meta = GeodeNodeMetadata::set(this);
if (meta->m_attributes.count(attr)) {
return meta->m_attributes.at(attr);
}
return std::nullopt;
}
#pragma warning(pop)