初步构建分析器，增加控制流分析，实现支配节点计算，支配树构建，支配边界计算，为后续Mem2reg做准备

2025-06-24 22:39:20 +08:00
parent d50f76a770
commit 3dbb394bc2
2 changed files with 558 additions and 0 deletions
--- a/src/include/SysYIRAnalyser.h
+++ b/src/include/SysYIRAnalyser.h
@@ -0,0 +1,299 @@
+#pragma once
+
+#include "IR.h"
+
+namespace sysy {
+
+// 前向声明
+
+class Loop;
+// 基本块分析信息类
+class BlockAnalysisInfo {
+
+public:
+  using block_list = std::vector<BasicBlock*>;
+  using block_set = std::unordered_set<BasicBlock*>;
+
+protected:
+  // 支配树相关
+  int domdepth = 0;          ///< 支配节点所在深度
+  BasicBlock* idom = nullptr;    ///< 直接支配结点
+  block_list sdoms;              ///< 支配树后继
+  block_set dominants;           ///< 必经结点集合
+  block_set dominant_frontiers;  ///< 支配边界
+  
+  // 后续添加循环分析相关
+  // Loop* loopbelong = nullptr;    ///< 所属循环
+  // int loopdepth = 0;             ///< 循环深度
+
+public:
+  // getterface
+  const int getDomDepth() const { return domdepth; }
+  const BasicBlock* getIdom() const { return idom; }
+  const block_list& getSdoms() const { return sdoms; }
+  const block_set& getDominants() const { return dominants; }
+  const block_set& getDomFrontiers() const { return dominant_frontiers; }
+  
+  // 支配树操作
+  void setDomDepth(int depth) { domdepth = depth; }
+  void setIdom(BasicBlock* block) { idom = block; }
+  void addSdoms(BasicBlock* block) { sdoms.push_back(block); }
+  void clearSdoms() { sdoms.clear(); }
+  void removeSdoms(BasicBlock* block) {
+    sdoms.erase(std::remove(sdoms.begin(), sdoms.end(), block), sdoms.end());
+  }
+  void addDominants(BasicBlock* block) { dominants.emplace(block); }
+  void addDominants(const block_set& blocks) { dominants.insert(blocks.begin(), blocks.end()); }
+  void setDominants(BasicBlock* block) { 
+    dominants.clear(); 
+    addDominants(block); 
+  }
+  void setDominants(const block_set& doms) { 
+    dominants = doms; 
+  }
+  void setDomFrontiers(const block_set& df) { 
+    dominant_frontiers = df; 
+  }
+  
+  
+  // TODO：循环分析操作方法
+  
+  // 清空所有分析信息
+  void clear() {
+    domdepth =  -1;
+    idom = nullptr;
+    sdoms.clear();
+    dominants.clear();
+    dominant_frontiers.clear();
+    // loopbelong = nullptr;
+    // loopdepth = 0;
+  }
+};
+
+// 函数分析信息类
+class FunctionAnalysisInfo {
+
+
+public:
+  // 函数属性
+  enum FunctionAttribute : uint64_t {
+      PlaceHolder = 0x0UL,
+      Pure = 0x1UL << 0,
+      SelfRecursive = 0x1UL << 1,
+      SideEffect = 0x1UL << 2,
+      NoPureCauseMemRead = 0x1UL << 3
+  };
+  
+  // 数据结构
+  using Loop_list = std::list<std::unique_ptr<Loop>>;
+  using block_loop_map = std::unordered_map<BasicBlock*, Loop*>;
+  using value_block_map = std::unordered_map<Value*, BasicBlock*>;
+  using value_block_count_map = std::unordered_map<Value*, std::unordered_map<BasicBlock*, int>>;
+  
+  // 分析数据
+  FunctionAttribute attribute = PlaceHolder;  ///< 函数属性
+  std::set<Function*> callees;               ///< 函数调用集合
+  Loop_list loops;                           ///< 所有循环
+  Loop_list topLoops;                        ///< 顶层循环
+  block_loop_map basicblock2Loop;            ///< 基本块到循环映射
+  std::list<std::unique_ptr<AllocaInst>> indirectAllocas; ///< 间接分配内存
+  
+  // 值定义/使用信息
+  value_block_map value2AllocBlocks;         ///< 值分配位置映射
+  value_block_count_map value2DefBlocks;     ///< 值定义位置映射
+  value_block_count_map value2UseBlocks;     ///< 值使用位置映射
+  
+  // 函数属性操作
+  FunctionAttribute getAttribute() const { return attribute; }
+  void setAttribute(FunctionAttribute attr) { attribute = static_cast<FunctionAttribute>(attribute | attr); }
+  void clearAttribute() { attribute = PlaceHolder; }
+  
+  // 调用关系操作
+  void addCallee(Function* callee) { callees.insert(callee); }
+  void removeCallee(Function* callee) { callees.erase(callee); }
+  void clearCallees() { callees.clear(); }
+  
+  // 循环分析操作
+  Loop* getLoopOfBasicBlock(BasicBlock* bb) { 
+    auto it = basicblock2Loop.find(bb);
+    return it != basicblock2Loop.end() ? it->second : nullptr; 
+  }
+  
+  void addBBToLoop(BasicBlock* bb, Loop* loop) { basicblock2Loop[bb] = loop; }
+  
+  unsigned getLoopDepthByBlock(BasicBlock* bb) {
+    Loop* loop = getLoopOfBasicBlock(bb);
+    return loop ? loop->getLoopDepth() : 0;
+  }
+  
+  // 值-块映射操作
+  void addValue2AllocBlocks(Value* value, BasicBlock* block) { value2AllocBlocks[value] = block; }
+  
+  BasicBlock* getAllocBlockByValue(Value* value) { 
+      auto it = value2AllocBlocks.find(value);
+      return it != value2AllocBlocks.end() ? it->second : nullptr;
+  }
+  
+  // 值定义/使用操作
+  void addValue2DefBlocks(Value* value, BasicBlock* block) { ++value2DefBlocks[value][block]; }
+  void addValue2UseBlocks(Value* value, BasicBlock* block) { ++value2UseBlocks[value][block]; }
+  
+  // 间接分配操作
+  void addIndirectAlloca(AllocaInst* alloca) { indirectAllocas.emplace_back(alloca); }
+  
+  // 清空所有分析信息
+  void clear() {
+    attribute = PlaceHolder;
+    callees.clear();
+    loops.clear();
+    topLoops.clear();
+    basicblock2Loop.clear();
+    indirectAllocas.clear();
+    value2AllocBlocks.clear();
+    value2DefBlocks.clear();
+    value2UseBlocks.clear();
+  }
+};
+// 循环类 - 未实现优化
+class Loop {
+public:
+  using block_list = std::vector<BasicBlock *>;
+  using block_set = std::unordered_set<BasicBlock *>;
+  using Loop_list = std::vector<Loop *>;
+
+protected:
+  Function *parent;                      // 所属函数
+  block_list blocksInLoop;               // 循环内的基本块
+  BasicBlock *preheaderBlock = nullptr;  // 前驱块
+  BasicBlock *headerBlock = nullptr;     // 循环头
+  block_list latchBlock;                 // 回边块
+  block_set exitingBlocks;               // 退出块
+  block_set exitBlocks;                  // 退出目标块
+  Loop *parentloop = nullptr;            // 父循环
+  Loop_list subLoops;                    // 子循环
+  size_t loopID;                         // 循环ID
+  unsigned loopDepth;                    // 循环深度
+
+  Instruction *indCondVar = nullptr;     // 循环条件变量
+  Instruction::Kind IcmpKind;            // 比较类型
+  Value *indEnd = nullptr;               // 循环结束值
+  AllocaInst *IndPhi = nullptr;          // 循环变量
+
+  ConstantValue *indBegin = nullptr;     // 循环起始值
+  ConstantValue *indStep = nullptr;      // 循环步长
+  
+  std::set<GlobalValue *> GlobalValuechange;  // 循环内改变的全局变量
+  
+  int StepType = 0;                      // 循环步长类型
+  bool parallelable = false;             // 是否可并行
+
+public:
+  explicit Loop(BasicBlock *header, const std::string &name = "") 
+      : headerBlock(header) {
+    blocksInLoop.push_back(header);
+  }
+  
+  void setloopID() {
+    static unsigned loopCount = 0;
+    loopCount = loopCount + 1;
+    loopID = loopCount;
+  }
+  ConstantValue* getindBegin() { return indBegin; }
+  ConstantValue* getindStep() { return indStep; }
+  void setindBegin(ConstantValue *indBegin2set) { indBegin = indBegin2set; }
+  void setindStep(ConstantValue *indStep2set) { indStep = indStep2set; }
+  void setStepType(int StepType2Set) { StepType = StepType2Set; }
+  int getStepType() { return StepType; }
+  size_t getLoopID() { return loopID; }
+
+  BasicBlock* getHeader() const { return headerBlock; }
+  BasicBlock* getPreheaderBlock() const { return preheaderBlock; }
+  block_list& getLatchBlocks() { return latchBlock; }
+  block_set& getExitingBlocks() { return exitingBlocks; }
+  block_set& getExitBlocks() { return exitBlocks; }
+  Loop* getParentLoop() const { return parentloop; }
+  void setParentLoop(Loop *parent) { parentloop = parent; }
+  void addBasicBlock(BasicBlock *bb) { blocksInLoop.push_back(bb); }
+  void addSubLoop(Loop *loop) { subLoops.push_back(loop); }
+  void setLoopDepth(unsigned depth) { loopDepth = depth; }
+  block_list& getBasicBlocks() { return blocksInLoop; }
+  Loop_list& getSubLoops() { return subLoops; }
+  unsigned getLoopDepth() const { return loopDepth; }
+  
+  bool isLoopContainsBasicBlock(BasicBlock *bb) const {
+    return std::find(blocksInLoop.begin(), blocksInLoop.end(), bb) != blocksInLoop.end();
+  }
+  
+  void addExitingBlock(BasicBlock *bb) { exitingBlocks.insert(bb); }
+  void addExitBlock(BasicBlock *bb) { exitBlocks.insert(bb); }
+  void addLatchBlock(BasicBlock *bb) { latchBlock.push_back(bb); }
+  void setPreheaderBlock(BasicBlock *bb) { preheaderBlock = bb; }
+  
+  void setIndexCondInstr(Instruction *instr) { indCondVar = instr; }
+  void setIcmpKind(Instruction::Kind kind) { IcmpKind = kind; }
+  Instruction::Kind getIcmpKind() const { return IcmpKind; }
+  
+  bool isSimpleLoopInvariant(Value *value) ; 
+  
+  void setIndEnd(Value *value) { indEnd = value; }
+  void setIndPhi(AllocaInst *phi) { IndPhi = phi; }
+  Value* getIndEnd() const { return indEnd; }
+  AllocaInst* getIndPhi() const { return IndPhi; }
+  Instruction* getIndCondVar() const { return indCondVar; }
+  
+  void addGlobalValuechange(GlobalValue *globalvaluechange2add) {
+    GlobalValuechange.insert(globalvaluechange2add);
+  }
+  std::set<GlobalValue *>& getGlobalValuechange() {
+    return GlobalValuechange;
+  }
+
+  void setParallelable(bool flag) { parallelable = flag; }
+  bool isParallelable() const { return parallelable; }
+};
+
+class ControlFlowAnalysis {
+private:
+  Module *pModule;  ///< 模块
+  std::unordered_map<BasicBlock*, BlockAnalysisInfo*> blockAnalysisInfo;  // 基本块分析信息
+  std::unordered_map<Function*, FunctionAnalysisInfo*> functionAnalysisInfo;  // 函数分析信息
+
+public:
+  explicit ControlFlowAnalysis(Module *pMoudle) : pModule(pMoudle) {} 
+
+  void init();  // 初始化分析器
+  void computeDomNode();  // 计算必经结点
+  void computeDomTree();  // 构造支配树
+  // std::unordered_set<BasicBlock *> computeDomFrontier(BasicBlock *block) ;  // 计算单个块的支配边界(弃用)
+  void computeDomFrontierAllBlk();  // 计算所有块的支配边界
+  void runControlFlowAnalysis();  // 运行控制流分析(主要是支配树和支配边界)
+  void clear(){
+    for (auto &pair : blockAnalysisInfo) {
+      delete pair.second;  // 清理基本块分析信息
+    }
+    blockAnalysisInfo.clear();
+    
+    for (auto &pair : functionAnalysisInfo) {
+      delete pair.second;  // 清理函数分析信息
+    }
+    functionAnalysisInfo.clear();
+  } // 清空分析结果
+  ~ControlFlowAnalysis() {
+    clear();  // 析构时清理所有分析信息
+  }
+
+private:
+  void intersectOP4Dom(std::unordered_set<BasicBlock *> &dom, const std::unordered_set<BasicBlock *> &other);  // 交集运算，
+  BasicBlock* findCommonDominator(BasicBlock *a, BasicBlock *b);  // 查找两个基本块的共同支配结点
+};
+
+
+// 分析管理器（整合版）
+class AnalysisManager {
+
+};
+
+
+
+
+}  // namespace sysy