SEGMapBase.h

//
// Created by sixiang on 9/30/24.
//
 
#ifndef CLEARBLUE_SEGMAPBASE_H
#define CLEARBLUE_SEGMAPBASE_H
 
#include <llvm/IR/Function.h>
#include <llvm/IR/Module.h>
#include <llvm/Support/raw_ostream.h>
 
#include <unordered_map>
 
#include "Analysis/CallGraph/CBCallGraph.h"
#include "IR/SEG/SymbolicExprGraph.h"
#include "Language/Spec/Memory.h"
 
using namespace llvm;
 
class CBCallGraphWrapper;
 
class SEGMapBase {
protected:
  std::unordered_map<const Function *, SymbolicExprGraph *> SEGMap;
 
public:
  SEGMapBase();
  ~SEGMapBase();
 
public:
  SymbolicExprGraph *getSymbolicExprGraph(const Function *F) const;
 
  bool hasSEG(const Function *F) const;
 
  bool isFuncBuildable(const Function *F) const;
 
  value_iterator<
      std::unordered_map<const Function *, SymbolicExprGraph *>::iterator>
  begin() {
    return {SEGMap.begin()};
  }
 
  value_iterator<
      std::unordered_map<const Function *, SymbolicExprGraph *>::iterator>
  end() {
    return {SEGMap.end()};
  }
 
protected:
  ExternalMemorySpec *MEM_SPEC = nullptr;
  CBCallGraphWrapper *CGWrapper = nullptr;
};
 
#endif // CLEARBLUE_SEGMAPBASE_H