instruction-counter.h

Go to the documentation of this file.

// Copyright 2012 Cloudera Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
#ifndef IMPALA_CODEGEN_INSTRUCTION_COUNTER_H_
#define IMPALA_CODEGEN_INSTRUCTION_COUNTER_H_

#include <map>
#include <string>
#include <sstream>
#include <iomanip>
#include <algorithm>

#include "common/logging.h"
#include "base/logging.h"
#include "llvm/IR/Instruction.h"
#include "llvm/IR/Module.h"
#include "llvm/IR/Function.h"

namespace impala {

class InstructionCounter {
 public:
  static const char* TOTAL_INSTS;
  static const char* TOTAL_BLOCKS;
  static const char* TOTAL_FUNCTIONS;
  static const char* TERMINATOR_INSTS;
  static const char* BINARY_INSTS;
  static const char* MEMORY_INSTS;
  static const char* CAST_INSTS;
  static const char* OTHER_INSTS;

  InstructionCounter();

  void visit(const llvm::Module& M);

  void visit(const llvm::Function &F);

  void visit(const llvm::BasicBlock &BB);

  void visit(const llvm::Instruction &I);

  void PrintCounter(const char* name, int count, int max_count_len,
      std::stringstream* stream) const;

  std::string PrintCounters() const;

  int GetCount(const char* name);

  void ResetCount();

 private:
  typedef std::map<std::string, int> CounterMap;

  template<class Iterator>
  void visit(Iterator start, Iterator end) {
    while (start != end) {
      visit(*start++);
    }
  }

  void IncrementCount(const char* name);

  CounterMap counters_;
};

}  // namespace impala

#endif  // IMPALA_CODEGEN_INSTRUCTION_COUNTER_H_