100 lines
2.2 KiB
C++
100 lines
2.2 KiB
C++
|
/* This file is part of the sirit project.
|
||
|
|
* Copyright (c) 2018 ReinUsesLisp
|
||
|
|
* This software may be used and distributed according to the terms of the GNU
|
||
|
|
* General Public License version 2 or any later version.
|
||
|
|
*/
|
||
|
|
|
||
|
|
#include <cassert>
|
||
|
|
#include "common_types.h"
|
||
|
|
#include "operand.h"
|
||
|
|
#include "ref.h"
|
||
|
|
|
||
|
|
namespace Sirit {
|
||
|
|
|
||
|
|
Ref::Ref(spv::Op opcode_, u32 id_, const Ref* result_type_)
|
||
|
|
: opcode(opcode_), id(id_), result_type(result_type_) {
|
||
|
|
operand_type = OperandType::Ref;
|
||
|
|
}
|
||
|
|
|
||
|
|
Ref::~Ref() = default;
|
||
|
|
|
||
|
|
void Ref::Fetch(Stream& stream) const {
|
||
|
|
assert(id != UINT32_MAX);
|
||
|
|
stream.Write(id);
|
||
|
|
}
|
||
|
|
|
||
|
|
u16 Ref::GetWordCount() const {
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
bool Ref::operator==(const Operand& other) const {
|
||
|
|
if (operand_type != other.GetType()) {
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
const Ref& ref = dynamic_cast<const Ref&>(other);
|
||
|
|
if (ref.opcode == opcode && result_type == ref.result_type &&
|
||
|
|
operands.size() == ref.operands.size()) {
|
||
|
|
for (std::size_t i{}; i < operands.size(); i++) {
|
||
|
|
if (*operands[i] != *ref.operands[i]) {
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
void Ref::Write(Stream& stream) const {
|
||
|
|
stream.Write(static_cast<u16>(opcode));
|
||
|
|
stream.Write(WordCount());
|
||
|
|
|
||
|
|
if (result_type) {
|
||
|
|
result_type->Fetch(stream);
|
||
|
|
}
|
||
|
|
if (id != UINT32_MAX) {
|
||
|
|
stream.Write(id);
|
||
|
|
}
|
||
|
|
for (const Operand* operand : operands) {
|
||
|
|
operand->Fetch(stream);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
void Ref::Add(Operand* operand) {
|
||
|
|
Add(static_cast<const Operand*>(operand));
|
||
|
|
operand_store.push_back(std::unique_ptr<Operand>(operand));
|
||
|
|
}
|
||
|
|
|
||
|
|
void Ref::Add(const Operand* operand) {
|
||
|
|
operands.push_back(operand);
|
||
|
|
}
|
||
|
|
|
||
|
|
void Ref::Add(u32 integer) {
|
||
|
|
Add(new LiteralInteger(integer));
|
||
|
|
}
|
||
|
|
|
||
|
|
void Ref::Add(const std::string& string) {
|
||
|
|
Add(new LiteralString(string));
|
||
|
|
}
|
||
|
|
|
||
|
|
void Ref::Add(const std::vector<const Ref*>& ids) {
|
||
|
|
for (const Ref* ref : ids) {
|
||
|
|
Add(ref);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
u16 Ref::WordCount() const {
|
||
|
|
u16 count{1};
|
||
|
|
if (result_type) {
|
||
|
|
count++;
|
||
|
|
}
|
||
|
|
if (id != UINT32_MAX) {
|
||
|
|
count++;
|
||
|
|
}
|
||
|
|
for (const Operand* operand : operands) {
|
||
|
|
count += operand->GetWordCount();
|
||
|
|
}
|
||
|
|
return count;
|
||
|
|
}
|
||
|
|
|
||
|
|
} // namespace Sirit
|