1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96
|
//===----------------------------------------------------------------------===//
//
// Copyright (c) 2012 The University of Utah
// All rights reserved.
//
// This file is distributed under the University of Illinois Open Source
// License. See the file COPYING for details.
//
//===----------------------------------------------------------------------===//
#ifndef BINOP_SIMPLIFICATION_H
#define BINOP_SIMPLIFICATION_H
#include <string>
#include "llvm/ADT/SmallVector.h"
#include "Transformation.h"
namespace clang {
class DeclGroupRef;
class ASTContext;
class FunctionDecl;
class Stmt;
class BinaryOperator;
}
class BSCollectionVisitor;
class BSStatementVisitor;
class BinOpSimplification : public Transformation {
friend class BSCollectionVisitor;
friend class BSStatementVisitor;
public:
BinOpSimplification(const char *TransName, const char *Desc)
: Transformation(TransName, Desc),
BinOpCollectionVisitor(NULL),
StmtVisitor(NULL),
NameQueryWrap(NULL),
TheFuncDecl(NULL),
TheStmt(NULL),
TheBinOp(NULL),
TmpVarName(""),
NeedParen(false)
{ }
~BinOpSimplification(void);
private:
virtual void Initialize(clang::ASTContext &context);
virtual bool HandleTopLevelDecl(clang::DeclGroupRef D);
virtual void HandleTranslationUnit(clang::ASTContext &Ctx);
bool addNewTmpVariable(void);
bool addNewAssignStmt(void);
bool replaceBinOp(void);
void setTmpVarName(std::string &Name) {
TmpVarName = Name;
}
std::string getTmpVarName(void) {
return TmpVarName;
}
clang::SmallVector<clang::BinaryOperator *, 10> ValidBinOps;
BSCollectionVisitor *BinOpCollectionVisitor;
BSStatementVisitor *StmtVisitor;
TransNameQueryWrap *NameQueryWrap;
clang::FunctionDecl *TheFuncDecl;
clang::Stmt *TheStmt;
clang::BinaryOperator *TheBinOp;
std::string TmpVarName;
bool NeedParen;
// Unimplemented
BinOpSimplification(void);
BinOpSimplification(const BinOpSimplification &);
void operator=(const BinOpSimplification &);
};
#endif
|