prog_reducer.h

// This file if part of the llir-opt project.
// Licensing information can be found in the LICENSE file.
// (C) 2018 Nandor Licker. All rights reserved.
 
#pragma once
 
#include <random>
#include <queue>
 
#include "core/inst_visitor.h"
#include "core/insts.h"
#include "timeout.h"
 
class Func;
 
template <typename T>
using Iterator = std::optional<std::pair<std::unique_ptr<Prog>, T *>>;
 
 
 
class ProgReducerBase : public InstVisitor<Iterator<Inst>> {
public:
  ProgReducerBase(unsigned threads) : threads_(threads) {}
 
  std::unique_ptr<Prog> Reduce(
      std::unique_ptr<Prog> &&prog,
      const Timeout &Timeout
  );
 
  virtual bool Verify(const Prog &prog) const = 0;
 
private:
  using It = Iterator<Inst>;
  using Bt = Iterator<Block>;
  using At = Iterator<Atom>;
  using Ft = Iterator<Func>;
 
  It ReduceInst(Inst *i) { return Dispatch(*i); }
  Bt ReduceBlock(Block *b);
  At ReduceAtom(Atom *a);
  Ft ReduceFunc(Func *f);
 
  It VisitArg(ArgInst &i);
  It VisitCall(CallInst &i);
  It VisitInvoke(InvokeInst &i);
  It VisitTailCall(TailCallInst &i);
  It VisitSyscall(SyscallInst &i);
  It VisitClone(CloneInst &i);
  It VisitRaise(RaiseInst &i);
  It VisitStore(StoreInst &i);
  It VisitMov(MovInst &i);
  It VisitSwitch(SwitchInst &i);
  It VisitJmp(JumpInst &i);
  It VisitJcc(JumpCondInst &i);
  It VisitReturn(ReturnInst &i);
  It VisitPhi(PhiInst &i);
  It VisitUndef(UndefInst &i);
  It VisitVAStart(VaStartInst &i);
  It VisitSet(SetInst &i);
  It VisitAlloca(AllocaInst &i) { return ReduceOperator(i); }
  It VisitFrame(FrameInst &i) { return ReduceOperator(i); }
  It VisitLoad(LoadInst &i) { return ReduceOperator(i); }
  It VisitUnary(UnaryInst &i) { return ReduceOperator(i); }
  It VisitBinary(BinaryInst &i) { return ReduceOperator(i); }
  It VisitSelect(SelectInst &i) { return ReduceOperator(i); }
  It VisitXchg(X86_XchgInst &i) { return ReduceOperator(i); }
  It VisitCmpXchg(X86_CmpXchgInst &i) { return ReduceOperator(i); }
  It VisitX86_Rdtsc(X86_RdTscInst &i) { return ReduceOperator(i); }
  It VisitX86_FPUControlInst(X86_FPUControlInst &i);
  It VisitInst(Inst &i);
 
  It ReduceOperator(Inst &i);
 
  using Candidate = std::pair<std::unique_ptr<Prog>, Inst *>;
  using CandidateList = std::queue<Candidate>;
 
  template <typename T>
  void RemoveArg(CandidateList &cand, T &i);
  void ReduceToOp(CandidateList &cand, Inst &i);
  void ReduceToRet(CandidateList &cand, Inst &i);
  void ReduceToTrap(CandidateList &cand, Inst &i);
  void ReduceToUndef(CandidateList &cand, Inst &i);
  void ReduceZero(CandidateList &cand, Inst &i);
  void ReduceErase(CandidateList &cand, Inst &i);
  void ReduceToArg(CandidateList &cand, Inst &i);
  void ReduceOperator(CandidateList &cand, Inst &i);
 
  It Evaluate(CandidateList &&cand);
 
  void RemoveEdge(Block *from, Block *to);
 
  Constant *GetZero(Type type);
 
private:
  unsigned threads_;
};