tools/llvm-diff/DifferenceEngine.h

   1 //===-- DifferenceEngine.h - Module comparator ------------------*- C++ -*-===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file is distributed under the University of Illinois Open Source
   6 // License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9 //
  10 // This header defines the interface to the LLVM difference engine,
  11 // which structurally compares functions within a module.
  12 //
  13 //===----------------------------------------------------------------------===//
  14
  15 #ifndef _LLVM_DIFFERENCE_ENGINE_H_
  16 #define _LLVM_DIFFERENCE_ENGINE_H_
  17
  18 #include "llvm/ADT/SmallVector.h"
  19 #include "llvm/ADT/StringRef.h"
  20 #include "DiffLog.h"
  21 #include "DiffConsumer.h"
  22
  23 #include <utility>
  24
  25 namespace llvm {
  26   class Function;
  27   class GlobalValue;
  28   class Instruction;
  29   class LLVMContext;
  30   class Module;
  31   class Twine;
  32   class Value;
  33
  34   /// A class for performing structural comparisons of LLVM assembly.
  35   class DifferenceEngine {
  36   public:
  37     /// A RAII object for recording the current context.
  38     struct Context {
  39       Context(DifferenceEngine &Engine, Value *L, Value *R) : Engine(Engine) {
  40         Engine.consumer.enterContext(L, R);
  41       }
  42
  43       ~Context() {
  44         Engine.consumer.exitContext();
  45       }
  46
  47     private:
  48       DifferenceEngine &Engine;
  49     };
  50
  51     /// An oracle for answering whether two values are equivalent as
  52     /// operands.
  53     struct Oracle {
  54       virtual bool operator()(Value *L, Value *R) = 0;
  55
  56     protected:
  57       virtual ~Oracle() {}
  58     };
  59
  60     DifferenceEngine(LLVMContext &context, Consumer &consumer)
  61       : context(context), consumer(consumer), globalValueOracle(0) {}
  62
  63     void diff(Module *L, Module *R);
  64     void diff(Function *L, Function *R);
  65     void log(StringRef text) {
  66       consumer.log(text);
  67     }
  68     LogBuilder logf(StringRef text) {
  69       return LogBuilder(consumer, text);
  70     }
  71     Consumer& getConsumer() const { return consumer; }
  72
  73     /// Installs an oracle to decide whether two global values are
  74     /// equivalent as operands.  Without an oracle, global values are
  75     /// considered equivalent as operands precisely when they have the
  76     /// same name.
  77     void setGlobalValueOracle(Oracle *oracle) {
  78       globalValueOracle = oracle;
  79     }
  80
  81     /// Determines whether two global values are equivalent.
  82     bool equivalentAsOperands(GlobalValue *L, GlobalValue *R);
  83
  84   private:
  85     LLVMContext &context;
  86     Consumer &consumer;
  87     Oracle *globalValueOracle;
  88   };
  89 }
  90
  91 #endif