1 files changed, 116 insertions, 0 deletions
diff --git a/src/gallium/auxiliary/gallivm/instructionssoa.cpp b/src/gallium/auxiliary/gallivm/instructionssoa.cpp
index a4d5046637..5739cf0cde 100644
--- a/src/gallium/auxiliary/gallivm/instructionssoa.cpp
+++ b/src/gallium/auxiliary/gallivm/instructionssoa.cpp
@@ -2,9 +2,28 @@
 
 #include "storagesoa.h"
 
+#include "pipe/p_shader_tokens.h"
+
+#include <llvm/CallingConv.h>
 #include <llvm/Constants.h>
+#include <llvm/Module.h>
+#include <llvm/Function.h>
+#include <llvm/Instructions.h>
+#include <llvm/Transforms/Utils/Cloning.h>
+#include <llvm/ParamAttrsList.h>
+
+#include <iostream>
+
 
+/* disable some warnings. this file is autogenerated */
+#if defined(__GNUC__)
+#pragma GCC diagnostic ignored "-Wunused-variable"
+#endif
 using namespace llvm;
+#include "gallivmsoabuiltins.cpp"
+#if defined(__GNUC__)
+#pragma GCC diagnostic warning "-Wunused-variable"
+#endif
 
 InstructionsSoa::InstructionsSoa(llvm::Module *mod, llvm::Function *func,
                                  llvm::BasicBlock *block, StorageSoa *storage)
@@ -12,6 +31,8 @@ InstructionsSoa::InstructionsSoa(llvm::Module *mod, llvm::Function *func,
      m_storage(storage),
      m_idx(0)
 {
+   createFunctionMap();
+   createBuiltins();
 }
 
 const char * InstructionsSoa::name(const char *prefix) const
@@ -119,3 +140,98 @@ std::vector<llvm::Value*> InstructionsSoa::extractVector(llvm::Value *vector)
 
    return res;
 }
+
+void InstructionsSoa::createFunctionMap()
+{
+   m_functionsMap[TGSI_OPCODE_DP3] = "dp3";
+}
+
+llvm::Function * InstructionsSoa::function(int op)
+{
+    if (m_functions.find(op) != m_functions.end())
+       return m_functions[op];
+
+    std::string name = m_functionsMap[op];
+
+    llvm::Function *originalFunc = m_builtins->getFunction(name);
+    llvm::Function *func = CloneFunction(originalFunc);
+    currentModule()->getFunctionList().push_back(func);
+    std::cout << "Func parent is "<<func->getParent()
+              <<", cur is "<<currentModule() <<std::endl;
+       //func->setParent(currentModule());
+    m_functions[op] = func;
+    return func;
+}
+
+llvm::Module * InstructionsSoa::currentModule() const
+{
+   BasicBlock *block = m_builder.GetInsertBlock();
+   if (!block || !block->getParent())
+      return 0;
+
+   return block->getParent()->getParent();
+}
+
+void InstructionsSoa::createBuiltins()
+{
+   m_builtins = createSoaBuiltins();
+}
+
+std::vector<llvm::Value*> InstructionsSoa::dp3(const std::vector<llvm::Value*> in1,
+                                               const std::vector<llvm::Value*> in2)
+{
+   llvm::Function *func = function(TGSI_OPCODE_DP3);
+   std::vector<Value*> params;
+
+   llvm::Value *tmp = allocaTemp();
+   params.push_back(tmp);
+
+   params.push_back(in1[0]);
+   params.push_back(in1[1]);
+   params.push_back(in1[2]);
+   params.push_back(in1[3]);
+   params.push_back(in2[0]);
+   params.push_back(in2[1]);
+   params.push_back(in2[2]);
+   params.push_back(in2[3]);
+   CallInst *call = m_builder.CreateCall(func, params.begin(), params.end(),
+                                         name("dp3"));
+   call->setCallingConv(CallingConv::C);
+   call->setTailCall(false);
+
+   std::vector<Value*> indices;
+   indices.push_back(m_storage->constantInt(0));
+   indices.push_back(m_storage->constantInt(0));
+
+   GetElementPtrInst *getElem = new GetElementPtrInst(tmp,
+                                                      indices.begin(),
+                                                      indices.end(),
+                                                      name("allocaPtr"),
+                                                      m_builder.GetInsertBlock());
+   indices = std::vector<Value*>();
+   indices.push_back(m_storage->constantInt(0));
+   GetElementPtrInst *xElemPtr =  new GetElementPtrInst(getElem,
+                                                        m_storage->constantInt(0),
+                                                        name("xPtr"),
+                                                        m_builder.GetInsertBlock());
+   GetElementPtrInst *yElemPtr =  new GetElementPtrInst(getElem,
+                                                        m_storage->constantInt(1),
+                                                        name("yPtr"),
+                                                        m_builder.GetInsertBlock());
+   GetElementPtrInst *zElemPtr =  new GetElementPtrInst(getElem,
+                                                        m_storage->constantInt(2),
+                                                        name("zPtr"),
+                                                        m_builder.GetInsertBlock());
+   GetElementPtrInst *wElemPtr =  new GetElementPtrInst(getElem,
+                                                        m_storage->constantInt(3),
+                                                        name("wPtr"),
+                                                        m_builder.GetInsertBlock());
+
+   std::vector<llvm::Value*> res(4);
+   res[0] = new LoadInst(xElemPtr);
+   res[1] = new LoadInst(yElemPtr);
+   res[2] = new LoadInst(zElemPtr);
+   res[3] = new LoadInst(wElemPtr);
+
+   return res;
+}