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 support for "case when" (valueless) switch cases

Browse source
This commit is contained in:
Brian Fiete 2020-02-23 06:14:14 -08:00
parent 3fd81fc966
commit 857c1c384a
5 changed files with 25 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

6
IDEHelper/Backend/BeMCContext.cpp
View file

@ -7576,16 +7576,14 @@ void BeMCContext::DoInstCombinePass()
continue; continue;
auto vregInfoFrom = mVRegInfo[remapFrom]; auto vregInfoFrom = mVRegInfo[remapFrom];
auto vregInfoTo = mVRegInfo[remapTo]; auto vregInfoTo = mVRegInfo[remapTo];
int bestVRegIdx = (vregInfoFrom->mDefOrderIdx < vregInfoTo->mDefOrderIdx) ? remapFrom : remapTo; int bestVRegIdx = remapTo;
//auto itr = defMap.find(remapTo); //auto itr = defMap.find(remapTo);
//if (itr != defMap.end()) //if (itr != defMap.end())
int* prevBestVRegIdxPtr = NULL; int* prevBestVRegIdxPtr = NULL;
if (defMap.TryGetValue(remapTo, &prevBestVRegIdxPtr)) if (defMap.TryGetValue(remapTo, &prevBestVRegIdxPtr))
{ {
auto prevBestVRegIdx = *prevBestVRegIdxPtr;
if (mVRegInfo[bestVRegIdx]->mDefOrderIdx < mVRegInfo[prevBestVRegIdx]->mDefOrderIdx)
defMap[remapTo] = bestVRegIdx;
} }
else else
defMap[remapTo] = bestVRegIdx; defMap[remapTo] = bestVRegIdx;

2
IDEHelper/Backend/BeMCContext.h
View file

@ -749,7 +749,6 @@ public:
// Must be refreshed with RefreshRefCounts // Must be refreshed with RefreshRefCounts
int mRefCount; int mRefCount;
int mAssignCount; int mAssignCount;
int mDefOrderIdx; // Indicates instruction # in def assignment
BeMCBlock* mClosedBlock; BeMCBlock* mClosedBlock;
int mClosedInstIdx; int mClosedInstIdx;
@ -790,7 +789,6 @@ public:
mVolatileVRegSave = -1; mVolatileVRegSave = -1;
mMustExist = false; mMustExist = false;
mIsRetVal = false; mIsRetVal = false;
mDefOrderIdx = -1;
mClosedBlock = NULL; mClosedBlock = NULL;
mClosedInstIdx = -1; mClosedInstIdx = -1;

6
IDEHelper/Compiler/BfReducer.cpp
View file

@ -3264,7 +3264,11 @@ BfSwitchStatement* BfReducer::CreateSwitchStatement(BfTokenNode* tokenNode)
{ {
auto nextNode = mVisitorPos.GetNext(); auto nextNode = mVisitorPos.GetNext();
whenToken = BfNodeDynCast<BfTokenNode>(nextNode); whenToken = BfNodeDynCast<BfTokenNode>(nextNode);
if ((whenToken != NULL) && (whenToken->GetToken() != BfToken_When)) if ((whenToken != NULL) && (whenToken->GetToken() == BfToken_When))
{
mVisitorPos.MoveNext();
}
else
whenToken = NULL; whenToken = NULL;
} }
if (whenToken != NULL) if (whenToken != NULL)

14
IDEHelper/Compiler/BfStmtEvaluator.cpp
View file

@ -4397,6 +4397,20 @@ void BfModule::Visit(BfSwitchStatement* switchStmt)
lastDefaultBlock = notEqBB; lastDefaultBlock = notEqBB;
} }
if ((whenExpr != NULL) && (switchCase->mCaseExpressions.size() == 1))
{
// This was a "case when" expression, always matches
mayHaveMatch = true;
auto notEqBB = mBfIRBuilder->CreateBlock(StrFormat("switch.notEq_when.%d", blockIdx));
mBfIRBuilder->CreateBr(caseBlock);
mBfIRBuilder->AddBlock(notEqBB);
mBfIRBuilder->SetInsertPoint(notEqBB);
lastNotEqBlock = notEqBB;
}
if (lastDefaultBlock) if (lastDefaultBlock)
mBfIRBuilder->SetSwitchDefaultDest(switchStatement, lastDefaultBlock); mBfIRBuilder->SetSwitchDefaultDest(switchStatement, lastDefaultBlock);