ExprIRGenerator.hpp

Go to the documentation of this file.

 
#ifndef LLVMEXPR_CODEGEN_LLVM_EXPRIRGenerator_HPP
#define LLVMEXPR_CODEGEN_LLVM_EXPRIRGenerator_HPP
 
#include "IRGeneratorBase.hpp"
 
class ExprIRGenerator : public IRGeneratorBase {
  public:
    ExprIRGenerator(
        const std::vector<Token>& tokens_in, const VSVideoInfo* out_vi,
        const std::vector<const VSVideoInfo*>& in_vi, int width_in,
        int height_in, bool mirror,
        const std::map<std::pair<int, std::string>, int>& p_map,
        const analysis::ExpressionAnalysisResults& analysis_results_in,
        llvm::LLVMContext& context_ref, llvm::Module& module_ref,
        llvm::IRBuilder<>& builder_ref, MathLibraryManager& math_mgr,
        std::string func_name_in, int approx_math_in, int tile_x_in,
        int tile_y_in);
 
  protected:
    void defineFunctionSignature() override;
    void generateLoops() override;
 
    bool processModeSpecificToken(const Token& token,
                                  std::vector<llvm::Value*>& rpn_stack,
                                  llvm::Value* x, llvm::Value* y,
                                  llvm::Value* x_fp, llvm::Value* y_fp,
                                  bool no_x_bounds_check) override;
 
    void finalizeAndStoreResult(llvm::Value* result_val, llvm::Value* x,
                                llvm::Value* y) override;
 
  private:
    void generate_x_loop_body(llvm::Value* x_var, llvm::Value* x_fp_var,
                              llvm::Value* y_var, llvm::Value* y_fp_var,
                              bool no_x_bounds_check);
 
    int tile_x;
    int tile_y;
 
    // Arrays
    std::map<std::string, llvm::Value*> named_arrays;
};
 
#endif // LLVMEXPR_CODEGEN_LLVM_EXPRIRGenerator_HPP