Extern/Beef
1
0
Fork 0
mirror of https://github.com/beefytech/Beef.git synced 2025-06-15 06:44:10 +02:00
Code Activity

Added const equality checking for string literals

Browse source
This commit is contained in:
Brian Fiete 2020-02-28 14:46:12 -08:00
parent 5bed292e87
commit a43d4e8bd9
3 changed files with 60 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

35
IDEHelper/Compiler/BfExprEvaluator.cpp
View file

@ -17660,6 +17660,41 @@ void BfExprEvaluator::PerformBinaryOperation(BfAstNode* leftExpression, BfAstNod
} }
} }
// Check for constant equality checks (mostly for strings)
if ((binaryOp == BfBinaryOp_Equality) || (binaryOp == BfBinaryOp_InEquality))
{
auto leftConstant = mModule->mBfIRBuilder->GetConstant(leftValue.mValue);
auto rightConstant = mModule->mBfIRBuilder->GetConstant(rightValue.mValue);
if ((leftConstant != NULL) && (rightConstant != NULL))
{
auto boolType = mModule->GetPrimitiveType(BfTypeCode_Boolean);
int leftStringPoolIdx = mModule->GetStringPoolIdx(leftValue.mValue, mModule->mBfIRBuilder);
if (leftStringPoolIdx != -1)
{
int rightStringPoolIdx = mModule->GetStringPoolIdx(rightValue.mValue, mModule->mBfIRBuilder);
if (rightStringPoolIdx != -1)
{
bool isEqual = leftStringPoolIdx == rightStringPoolIdx;
if (binaryOp == BfBinaryOp_InEquality)
isEqual = !isEqual;
mResult = BfTypedValue(mModule->GetConstValue(isEqual ? 1 : 0, boolType), boolType);
return;
}
}
int eqResult = mModule->mBfIRBuilder->CheckConstEquality(leftValue.mValue, rightValue.mValue);
if (eqResult != -1)
{
bool isEqual = eqResult == 1;
if (binaryOp == BfBinaryOp_InEquality)
isEqual = !isEqual;
mResult = BfTypedValue(mModule->GetConstValue(isEqual ? 1 : 0, boolType), boolType);
return;
}
}
}
if ((leftValue.mType->IsTypeInstance()) || (leftValue.mType->IsGenericParam()) || if ((leftValue.mType->IsTypeInstance()) || (leftValue.mType->IsGenericParam()) ||
(rightValue.mType->IsTypeInstance()) || (rightValue.mType->IsGenericParam())) (rightValue.mType->IsTypeInstance()) || (rightValue.mType->IsGenericParam()))
{ {

23
IDEHelper/Compiler/BfIRBuilder.cpp
View file

@ -344,6 +344,29 @@ bool BfIRConstHolder::TryGetBool(BfIRValue id, bool& boolVal)
return false; return false;
} }
int BfIRConstHolder::CheckConstEquality(BfIRValue lhs, BfIRValue rhs)
{
auto constLHS = GetConstant(lhs);
if (constLHS == NULL)
return -1;
auto constRHS = GetConstant(rhs);
if (constRHS == NULL)
return -1;
if (constLHS->mConstType == BfConstType_BitCast)
return CheckConstEquality(BfIRValue(BfIRValueFlags_Const, ((BfConstantBitCast*)constLHS)->mTarget), rhs);
if (constRHS->mConstType == BfConstType_BitCast)
return CheckConstEquality(lhs, BfIRValue(BfIRValueFlags_Const, ((BfConstantBitCast*)constRHS)->mTarget));
if ((constLHS->mConstType == BfConstType_GlobalVar) &&
(constRHS->mConstType == BfConstType_GlobalVar))
{
return (constLHS == constRHS) ? 1 : 0;
}
return -1;
}
BfIRValue BfIRConstHolder::CreateConst(BfTypeCode typeCode, uint64 val) BfIRValue BfIRConstHolder::CreateConst(BfTypeCode typeCode, uint64 val)
{ {
if (typeCode == BfTypeCode_IntUnknown) if (typeCode == BfTypeCode_IntUnknown)

1
IDEHelper/Compiler/BfIRBuilder.h
View file

@ -830,6 +830,7 @@ public:
BfConstant* GetConstantById(int id); BfConstant* GetConstantById(int id);
BfConstant* GetConstant(BfIRValue id); BfConstant* GetConstant(BfIRValue id);
bool TryGetBool(BfIRValue id, bool& boolVal); bool TryGetBool(BfIRValue id, bool& boolVal);
int CheckConstEquality(BfIRValue lhs, BfIRValue rhs); // -1 = fail, 0 = false, 1 = true
BfIRValue CreateConst(BfTypeCode typeCode, uint64 val); BfIRValue CreateConst(BfTypeCode typeCode, uint64 val);
BfIRValue CreateConst(BfTypeCode typeCode, int val); BfIRValue CreateConst(BfTypeCode typeCode, int val);