Patrick Kelley 8fd444092b initial
2025-05-07 15:35:15 -04:00

74 lines
2.9 KiB
C++

// Copyright (c) 2020-now by the Zeek Project. See LICENSE for details.
#pragma once
#include <string>
#include <utility>
#include <hilti/ast/expression.h>
#include <hilti/ast/operator.h>
#include <hilti/ast/type.h>
namespace hilti::expression {
/** AST node for an expression representing an unresolved operator usage. */
class UnresolvedOperator : public Expression {
public:
auto kind() const { return _kind; }
// Checks if all operands are fully unified and hence ready for type
// comparison. Note that being unified is subtly different from being
// resolved: Being resolved is a dynamically computed property that can be
// checked anytime, whereas unification is computed regularly but may not
// always fully reflect the current state.
bool areOperandsUnified() const {
for ( auto e : children<Expression>(1, {}) ) {
if ( ! e->type()->type()->unification() )
return false;
}
return true;
}
// Accelerated accessors for the first three operands, returning raw pointers.
Expression* op0() const { return static_cast<Expression*>(children()[1]); }
Expression* op1() const { return static_cast<Expression*>(children()[2]); }
Expression* op2() const { return static_cast<Expression*>(children()[3]); }
/** Implements interface for use with `OverloadRegistry`. */
hilti::node::Range<Expression> operands() const { return children<Expression>(1, {}); }
// Dummy implementations as the node will be rejected in validation anyway.
QualifiedType* type() const final { return child<QualifiedType>(0); }
std::string printSignature() const { return operator_::detail::printSignature(kind(), operands(), Meta()); }
node::Properties properties() const final {
auto p = node::Properties{{"kind", to_string(_kind)}};
return Expression::properties() + std::move(p);
}
static auto create(ASTContext* ctx, operator_::Kind kind, Expressions operands, const Meta& meta = {}) {
return ctx->make<UnresolvedOperator>(ctx,
node::flatten(QualifiedType::createAuto(ctx, meta), std::move(operands)),
kind, meta);
}
static auto create(ASTContext* ctx, operator_::Kind kind, hilti::node::Range<Expression> operands,
const Meta& meta = {}) {
return ctx->make<UnresolvedOperator>(ctx, node::flatten(QualifiedType::createAuto(ctx, meta), operands), kind,
meta);
}
protected:
UnresolvedOperator(ASTContext* ctx, Nodes children, operator_::Kind kind, Meta meta)
: Expression(ctx, NodeTags, std::move(children), std::move(meta)), _kind(kind) {}
HILTI_NODE_1(expression::UnresolvedOperator, Expression, final);
private:
operator_::Kind _kind;
};
} // namespace hilti::expression