sirit/src/op.cpp
2018-10-03 00:32:45 -03:00

94 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
* Lesser General Public License version 2.1 or any later version.
*/
#include <cassert>
#include "common_types.h"
#include "lnumber.h"
#include "lstring.h"
#include "op.h"
#include "operand.h"
namespace Sirit {
Op::Op(spv::Op opcode, std::optional<u32> id, Ref result_type)
: opcode(opcode), id(id), result_type(result_type) {
operand_type = OperandType::Op;
}
Op::~Op() = default;
void Op::Fetch(Stream& stream) const {
assert(id.has_value());
stream.Write(id.value());
}
u16 Op::GetWordCount() const { return 1; }
bool Op::operator==(const Operand& other) const {
if (operand_type != other.GetType()) {
return false;
}
const Op& op = dynamic_cast<const Op&>(other);
if (op.opcode == opcode && result_type == op.result_type &&
operands.size() == op.operands.size()) {
for (std::size_t i{}; i < operands.size(); i++) {
if (*operands[i] != *op.operands[i]) {
return false;
}
}
return true;
}
return false;
}
void Op::Write(Stream& stream) const {
stream.Write(static_cast<u16>(opcode));
stream.Write(WordCount());
if (result_type) {
result_type->Fetch(stream);
}
if (id.has_value()) {
stream.Write(id.value());
}
for (const Operand* operand : operands) {
operand->Fetch(stream);
}
}
void Op::Add(Operand* operand) {
Add(static_cast<const Operand*>(operand));
operand_store.push_back(std::unique_ptr<Operand>(operand));
}
void Op::Add(const Operand* operand) { operands.push_back(operand); }
void Op::Add(u32 integer) { Add(LiteralNumber::Create<u32>(integer)); }
void Op::Add(const std::string& string) { Add(new LiteralString(string)); }
void Op::Add(const std::vector<Ref>& ids) {
for (Ref op : ids) {
Add(op);
}
}
u16 Op::WordCount() const {
u16 count{1};
if (result_type) {
count++;
}
if (id.has_value()) {
count++;
}
for (const Operand* operand : operands) {
count += operand->GetWordCount();
}
return count;
}
} // namespace Sirit