100 lines
2.9 KiB
C++
100 lines
2.9 KiB
C++
#include "weapon_inventory.h"
|
|
#include "macros.h"
|
|
#include "weapon_base.h"
|
|
|
|
void WeaponInventory::_bind_methods() {
|
|
BIND_HPROPERTY(Variant::OBJECT, fallback_weapon, PROPERTY_HINT_RESOURCE_TYPE, "PackedScene");
|
|
}
|
|
|
|
void WeaponInventory::on_switch_input() {
|
|
this->current = (this->current + 1) % 1;
|
|
this->select_weapon(this->weapons[this->current]);
|
|
}
|
|
|
|
void WeaponInventory::ready() {
|
|
// initialize the fallback weapon
|
|
this->weapon_parent = cast_to<Node3D>(get_node(NodePath("%PlayerCamera")));
|
|
if (this->fallback_weapon_scene.is_valid()) {
|
|
Node *fallback_as_node{ this->fallback_weapon_scene->instantiate() };
|
|
this->weapon_parent->add_child(fallback_as_node);
|
|
if ((this->fallback_weapon = cast_to<WeaponBase>(fallback_as_node))) {
|
|
this->select_weapon(this->fallback_weapon);
|
|
} else if(fallback_as_node != nullptr) {
|
|
fallback_as_node->queue_free();
|
|
}
|
|
}
|
|
if (this->fallback_weapon == nullptr) {
|
|
print_error("WeaponInventory::ready(): fallback weapon is invalid, expect unintended behaviour.");
|
|
}
|
|
}
|
|
|
|
void WeaponInventory::_notification(int what) {
|
|
if (Engine::get_singleton()->is_editor_hint()) {
|
|
return;
|
|
}
|
|
switch (what) {
|
|
default:
|
|
return;
|
|
case NOTIFICATION_READY:
|
|
ready();
|
|
return;
|
|
}
|
|
}
|
|
|
|
void WeaponInventory::set_fallback_weapon(Ref<PackedScene> scene) {
|
|
this->fallback_weapon_scene = scene;
|
|
}
|
|
|
|
Ref<PackedScene> WeaponInventory::get_fallback_weapon() const {
|
|
return this->fallback_weapon_scene;
|
|
}
|
|
|
|
void WeaponInventory::select_weapon(WeaponBase *new_weapon) {
|
|
if (new_weapon == nullptr) {
|
|
new_weapon = this->fallback_weapon;
|
|
}
|
|
if (this->current_weapon != nullptr) {
|
|
this->current_weapon->set_process_mode(PROCESS_MODE_DISABLED);
|
|
this->current_weapon->set_visible(false);
|
|
}
|
|
this->current_weapon = new_weapon;
|
|
if (new_weapon != nullptr) {
|
|
this->current_weapon->set_process_mode(PROCESS_MODE_PAUSABLE);
|
|
this->current_weapon->set_visible(true);
|
|
this->current_weapon->notify_selected();
|
|
if (new_weapon == this->weapons[0]) {
|
|
this->current = 0;
|
|
} else if (new_weapon == this->weapons[1]) {
|
|
this->current = 1;
|
|
} else {
|
|
print_error("WeaponInventory::select_weapon: Attempt to select weapon that is not in inventory");
|
|
}
|
|
}
|
|
}
|
|
|
|
WeaponBase *WeaponInventory::get_current_weapon() const {
|
|
return this->current_weapon;
|
|
}
|
|
|
|
void WeaponInventory::pickup_weapon(Ref<PackedScene> weapon_scene) {
|
|
Node *weapon_as_node{ weapon_scene->instantiate() };
|
|
if (WeaponBase *weapon{ cast_to<WeaponBase>(weapon_as_node) }) {
|
|
if (this->weapons[0] == nullptr) {
|
|
this->weapons[0] = weapon;
|
|
select_weapon(weapon);
|
|
return;
|
|
} else if (this->weapons[1] == nullptr) {
|
|
this->weapons[1] = weapon;
|
|
select_weapon(weapon);
|
|
return;
|
|
} else {
|
|
this->current_weapon->queue_free();
|
|
this->weapons[this->current] = weapon;
|
|
this->current_weapon = nullptr;
|
|
select_weapon(weapon);
|
|
}
|
|
} else if (weapon_as_node != nullptr) {
|
|
weapon_as_node->queue_free();
|
|
}
|
|
}
|