OpenVDB  8.1.0
ComputeGenerator.h
Go to the documentation of this file.
1 // Copyright Contributors to the OpenVDB Project
2 // SPDX-License-Identifier: MPL-2.0
3 
10 
11 #ifndef OPENVDB_AX_COMPUTE_GENERATOR_HAS_BEEN_INCLUDED
12 #define OPENVDB_AX_COMPUTE_GENERATOR_HAS_BEEN_INCLUDED
13 
14 #include "FunctionRegistry.h"
15 #include "FunctionTypes.h"
16 #include "SymbolTable.h"
17 
18 #include "../ast/AST.h"
19 #include "../ast/Visitor.h"
20 #include "../compiler/CompilerOptions.h"
21 #include "../compiler/Logger.h"
22 
23 #include <openvdb/version.h>
24 
25 #include <llvm/Analysis/TargetLibraryInfo.h>
26 #include <llvm/IR/BasicBlock.h>
27 #include <llvm/IR/Function.h>
28 #include <llvm/IR/IRBuilder.h>
29 #include <llvm/IR/LLVMContext.h>
30 #include <llvm/IR/Module.h>
31 
32 #include <stack>
33 
34 namespace openvdb {
36 namespace OPENVDB_VERSION_NAME {
37 
38 namespace ax {
39 namespace codegen {
40 
49 {
51  static const std::string Name;
52 
54  using Signature = void(const void* const);
56  static const size_t N_ARGS = FunctionTraitsT::N_ARGS;
57 
59  static const std::array<std::string, N_ARGS>& getArgumentKeys();
60  static std::string getDefaultName();
61 };
62 
63 
66 
67 namespace codegen_internal {
68 
86 struct ComputeGenerator : public ast::Visitor<ComputeGenerator>
87 {
88  ComputeGenerator(llvm::Module& module,
89  const FunctionOptions& options,
90  FunctionRegistry& functionRegistry,
91  Logger& logger);
92 
93  virtual ~ComputeGenerator() = default;
94 
95  bool generate(const ast::Tree&);
96 
97  inline SymbolTable& globals() { return mSymbolTables.globals(); }
98  inline const SymbolTable& globals() const { return mSymbolTables.globals(); }
99 
100  // Visitor pattern
101 
104 
106  inline bool postOrderNodes() const { return true; }
107 
111  bool traverse(const ast::Block* block)
112  {
113  if (!block) return true;
114  if (!this->visit(block)) return false;
115  return true;
116  }
117 
122  bool traverse(const ast::CommaOperator* comma)
123  {
124  if (!comma) return true;
125  if (!this->visit(comma)) return false;
126  return true;
127  }
128 
129 
134  {
135  if (!cond) return true;
136  if (!this->visit(cond)) return false;
137  return true;
138  }
139 
143  bool traverse(const ast::BinaryOperator* bin)
144  {
145  if (!bin) return true;
146  if (!this->visit(bin)) return false;
147  return true;
148  }
149 
153  bool traverse(const ast::TernaryOperator* tern)
154  {
155  if (!tern) return true;
156  if (!this->visit(tern)) return false;
157  return true;
158  }
159 
164  bool traverse(const ast::Loop* loop)
165  {
166  if (!loop) return true;
167  if (!this->visit(loop)) return false;
168  return true;
169  }
170 
175  bool traverse(const ast::DeclareLocal* decl)
176  {
177  if (!decl) return true;
178  if (!this->visit(decl)) return false;
179  return true;
180  }
181 
184  virtual bool visit(const ast::CommaOperator*);
185  virtual bool visit(const ast::AssignExpression*);
186  virtual bool visit(const ast::Crement*);
187  virtual bool visit(const ast::FunctionCall*);
188  virtual bool visit(const ast::Attribute*);
189  virtual bool visit(const ast::Tree*);
190  virtual bool visit(const ast::Block*);
191  virtual bool visit(const ast::ConditionalStatement*);
192  virtual bool visit(const ast::Loop*);
193  virtual bool visit(const ast::Keyword*);
194  virtual bool visit(const ast::UnaryOperator*);
195  virtual bool visit(const ast::BinaryOperator*);
196  virtual bool visit(const ast::TernaryOperator*);
197  virtual bool visit(const ast::Cast*);
198  virtual bool visit(const ast::DeclareLocal*);
199  virtual bool visit(const ast::Local*);
200  virtual bool visit(const ast::ExternalVariable*);
201  virtual bool visit(const ast::ArrayUnpack*);
202  virtual bool visit(const ast::ArrayPack*);
203  virtual bool visit(const ast::Value<bool>*);
204  virtual bool visit(const ast::Value<int16_t>*);
205  virtual bool visit(const ast::Value<int32_t>*);
206  virtual bool visit(const ast::Value<int64_t>*);
207  virtual bool visit(const ast::Value<float>*);
208  virtual bool visit(const ast::Value<double>*);
209  virtual bool visit(const ast::Value<std::string>*);
210 
211  template <typename ValueType>
212  typename std::enable_if<std::is_integral<ValueType>::value, bool>::type
213  visit(const ast::Value<ValueType>* node);
214  template <typename ValueType>
215 
216  typename std::enable_if<std::is_floating_point<ValueType>::value, bool>::type
217  visit(const ast::Value<ValueType>* node);
219 
220 protected:
221 
222  const FunctionGroup* getFunction(const std::string& identifier,
223  const bool allowInternal = false);
224 
225  bool binaryExpression(llvm::Value*& result, llvm::Value* lhs, llvm::Value* rhs,
226  const ast::tokens::OperatorToken op, const ast::Node* node);
227  bool assignExpression(llvm::Value* lhs, llvm::Value*& rhs, const ast::Node* node);
228 
234  void createFreeSymbolStrings(llvm::IRBuilder<>&);
235 
236  llvm::Module& mModule;
237  llvm::LLVMContext& mContext;
238  llvm::IRBuilder<> mBuilder;
239 
240  // The stack of accessed values
241  std::stack<llvm::Value*> mValues;
242 
243  // The stack of blocks for keyword branching
244  std::stack<std::pair<llvm::BasicBlock*, llvm::BasicBlock*>> mBreakContinueStack;
245 
246  // The current scope number used to track scoped declarations
247  size_t mScopeIndex;
248 
249  // The map of scope number to local variable names to values
251 
252  // The function used as the base code block
253  llvm::Function* mFunction;
254 
256 
258 
259 private:
260  FunctionRegistry& mFunctionRegistry;
261 };
262 
263 } // codegen_internal
264 
265 } // namespace codegen
266 } // namespace ax
267 } // namespace OPENVDB_VERSION_NAME
268 } // namespace openvdb
269 
270 #endif // OPENVDB_AX_COMPUTE_GENERATOR_HAS_BEEN_INCLUDED
271 
openvdb::v8_1::ax::ast::Local
Local AST nodes represent a single accesses to a local variable. The only store the name of the varia...
Definition: AST.h:2111
openvdb::v8_1::ax::codegen::codegen_internal::ComputeGenerator::mScopeIndex
size_t mScopeIndex
Definition: ComputeGenerator.h:247
FunctionRegistry.h
Contains the global function registration definition which described all available user front end fun...
openvdb::v8_1::ax::ast::AssignExpression
AssignExpressions represents a similar object construction to a BinaryOperator. AssignExpressions can...
Definition: AST.h:1197
openvdb::v8_1::ax::FunctionOptions
Options that control how functions behave.
Definition: CompilerOptions.h:24
openvdb::v8_1::ax::ast::DeclareLocal
DeclareLocal AST nodes symbolize a single type declaration of a local variable. These store the local...
Definition: AST.h:2138
openvdb::v8_1::ax::codegen::codegen_internal::ComputeGenerator::traverse
bool traverse(const ast::TernaryOperator *tern)
Custom traversal of ternary operators.
Definition: ComputeGenerator.h:153
openvdb::v8_1::ax::ast::Block
A Block node represents a scoped list of statements. It may comprise of 0 or more statements,...
Definition: AST.h:475
OPENVDB_VERSION_NAME
#define OPENVDB_VERSION_NAME
The version namespace name for this library version.
Definition: version.h.in:116
openvdb::v8_1::ax::codegen::codegen_internal::ComputeGenerator::traverse
bool traverse(const ast::Block *block)
Custom traversal of scoped blocks.
Definition: ComputeGenerator.h:111
OPENVDB_USE_VERSION_NAMESPACE
#define OPENVDB_USE_VERSION_NAMESPACE
Definition: version.h.in:178
openvdb::v8_1::ax::ast::ConditionalStatement
ConditionalStatements represents all combinations of 'if', 'else' and 'else if' syntax and semantics....
Definition: AST.h:863
openvdb::v8_1::ax::codegen::codegen_internal::ComputeGenerator::mBuilder
llvm::IRBuilder mBuilder
Definition: ComputeGenerator.h:238
openvdb::v8_1::ax::codegen::codegen_internal::ComputeGenerator::globals
SymbolTable & globals()
Definition: ComputeGenerator.h:97
openvdb::v8_1::ax::codegen::ComputeKernel::Name
static const std::string Name
The name of the generated function.
Definition: ComputeGenerator.h:51
openvdb::v8_1::ax::ast::Visitor
The Visitor class uses the Curiously Recursive Template Pattern (CRTP) to provide a customizable inte...
Definition: Visitor.h:95
openvdb::v8_1::ax::codegen::ComputeKernel::Signature
void(const void *const) Signature
The signature of the generated function.
Definition: ComputeGenerator.h:54
openvdb::v8_1::ax::codegen::codegen_internal::ComputeGenerator::mValues
std::stack< llvm::Value * > mValues
Definition: ComputeGenerator.h:241
openvdb::v8_1::ax::ast::Tree
A Tree is the highest concrete (non-abstract) node in the entire AX AST hierarchy....
Definition: AST.h:561
openvdb::v8_1::ax::codegen::FunctionTraits
Templated function traits which provides compile-time index access to the types of the function signa...
Definition: ax/openvdb_ax/codegen/Types.h:259
openvdb::v8_1::ax::ast::Loop
Loops represent for, while and do-while loop constructs. These all consist of a condition - evaluated...
Definition: AST.h:707
openvdb::v8_1::ax::ast::Value< std::string >
Specialization of Values for strings.
Definition: AST.h:2334
openvdb::v8_1::ax::ast::FunctionCall
FunctionCalls represent a single call to a function and any provided arguments. The argument list can...
Definition: AST.h:1540
openvdb::v8_1::ax::codegen::SymbolTableBlocks
A map of unique ids to symbol tables which can be used to represent local variables within a program....
Definition: SymbolTable.h:112
openvdb::v8_1::ax::codegen::codegen_internal::ComputeGenerator::mSymbolTables
SymbolTableBlocks mSymbolTables
Definition: ComputeGenerator.h:250
openvdb::v8_1::ax::ast::UnaryOperator
A UnaryOperator represents a single unary operation on an expression. The operation type is stored as...
Definition: AST.h:1388
openvdb::v8_1::ax::ast::CommaOperator
Definition: AST.h:606
openvdb::v8_1::ax::codegen::codegen_internal::ComputeGenerator::globals
const SymbolTable & globals() const
Definition: ComputeGenerator.h:98
openvdb::v8_1::ax::codegen::FunctionRegistry
The function registry which is used for function code generation. Each time a function is visited wit...
Definition: FunctionRegistry.h:36
openvdb::v8_1::ax::ast::Node
The base abstract node which determines the interface and required methods for all derived concrete n...
Definition: AST.h:101
openvdb::v8_1::ax::ast::Attribute
Attributes represent any access to a primitive value, typically associated with the '@' symbol syntax...
Definition: AST.h:1873
openvdb::v8_1::ax::ast::ExternalVariable
ExternalVariable represent any access to external (custom) data, typically associated with the '$' sy...
Definition: AST.h:2001
openvdb::v8_1::ax::codegen::codegen_internal::ComputeGenerator::traverse
bool traverse(const ast::BinaryOperator *bin)
Custom traversal of binary operators.
Definition: ComputeGenerator.h:143
openvdb::v8_1::ax::codegen::ComputeKernel
The function definition and signature which is built by the ComputeGenerator.
Definition: ComputeGenerator.h:48
openvdb::v8_1::ax::codegen::FunctionGroup
todo
Definition: FunctionTypes.h:790
openvdb::v8_1::ax::codegen::codegen_internal::ComputeGenerator
Visitor object which will generate llvm IR for a syntax tree. This provides the majority of the code ...
Definition: ComputeGenerator.h:86
openvdb::v8_1::ax::ast::Value
A Value (literal) AST node holds either literal text or absolute value information on all numerical,...
Definition: AST.h:2252
openvdb::v8_1::ax::codegen::SymbolTable
A symbol table which can be used to represent a single scoped set of a programs variables....
Definition: SymbolTable.h:35
openvdb::v8_1::ax::Logger
Logger for collecting errors and warnings that occur during AX compilation.
Definition: Logger.h:54
openvdb::v8_1::ax::ast::ArrayUnpack
ArrayUnpack represent indexing operations into AX container types, primarily vectors and matrices ind...
Definition: AST.h:1685
SymbolTable.h
Contains the symbol table which holds mappings of variables names to llvm::Values.
openvdb::v8_1::ax::ast::ArrayPack
ArrayPacks represent temporary container creations of arbitrary sizes, typically generated through th...
Definition: AST.h:1784
openvdb::v8_1::ax::codegen::codegen_internal::ComputeGenerator::mOptions
const FunctionOptions mOptions
Definition: ComputeGenerator.h:255
openvdb::v8_1::ax::codegen::codegen_internal::ComputeGenerator::mFunction
llvm::Function * mFunction
Definition: ComputeGenerator.h:253
openvdb::v8_1::ax::ast::BinaryOperator
A BinaryOperator represents a single binary operation between a left hand side (LHS) and right hand s...
Definition: AST.h:987
FunctionTypes.h
Contains frameworks for creating custom AX functions which can be registered within the FunctionRegis...
openvdb::v8_1::ax::codegen::codegen_internal::ComputeGenerator::mContext
llvm::LLVMContext & mContext
Definition: ComputeGenerator.h:237
openvdb::v8_1::ax::ast::tokens::OperatorToken
OperatorToken
Definition: Tokens.h:150
openvdb::v8_1::ax::ast::Crement
A Crement node represents a single increment '++' and decrement '–' operation. As well as it's crem...
Definition: AST.h:1293
openvdb::v8_1::ax::ast::TernaryOperator
A TernaryOperator represents a ternary (conditional) expression 'a ? b : c' which evaluates to 'b' if...
Definition: AST.h:1091
openvdb::v8_1::ax::codegen::codegen_internal::ComputeGenerator::mModule
llvm::Module & mModule
Definition: ComputeGenerator.h:236
openvdb::v8_1::ax::ast::Cast
Cast nodes represent the conversion of an underlying expression to a target type. Cast nodes are typi...
Definition: AST.h:1463
openvdb::v8_1::ax::codegen::codegen_internal::ComputeGenerator::mBreakContinueStack
std::stack< std::pair< llvm::BasicBlock *, llvm::BasicBlock * > > mBreakContinueStack
Definition: ComputeGenerator.h:244
openvdb
Definition: openvdb/Exceptions.h:13
openvdb::v8_1::ax::codegen::codegen_internal::ComputeGenerator::mLog
Logger & mLog
Definition: ComputeGenerator.h:257
openvdb::v8_1::ax::codegen::codegen_internal::ComputeGenerator::postOrderNodes
bool postOrderNodes() const
Code generation always runs post order.
Definition: ComputeGenerator.h:106
openvdb::v8_1::ax::ast::Keyword
Keywords represent keyword statements defining changes in execution. These include those that define ...
Definition: AST.h:1640
openvdb::v8_1::ax::codegen::codegen_internal::ComputeGenerator::traverse
bool traverse(const ast::CommaOperator *comma)
Custom traversal of comma expression.
Definition: ComputeGenerator.h:122
openvdb::v8_1::ax::codegen::codegen_internal::ComputeGenerator::traverse
bool traverse(const ast::Loop *loop)
Custom traversal of loops.
Definition: ComputeGenerator.h:164
openvdb::v8_1::ax::codegen::codegen_internal::ComputeGenerator::traverse
bool traverse(const ast::DeclareLocal *decl)
Custom traversal of declarations.
Definition: ComputeGenerator.h:175
openvdb::v8_1::ax::codegen::codegen_internal::ComputeGenerator::traverse
bool traverse(const ast::ConditionalStatement *cond)
Custom traversal of conditional statements.
Definition: ComputeGenerator.h:133