搭建 Optix 环境

　　我参考了第0个示例 OptixHello 学习Optix的工程配置以及基本框架的配置过程，该文对于 Optix 的框架介绍的很好，但是按照该文配置遇到了一些问题，我花费了一番功夫自己摸索终于配置好了环境，实现了用Optix计算然后在OpenGL上展示结果的一个简单的Demo。

　　我的配置环境为：Win10，GTX1080，驱动版本431.6，Cuda9.0，Optix6.0，VS2015

　　Cuda我很早就安装了，很简单，这里就不做介绍了。

　　1、下载Optix6.0

　　　　地址：https://developer.nvidia.com/designworks/optix/downloads/legacy

　　　　GTX1080应该是只能支持到6.0版本。我刚开始下载了6.5版本，然后顺利使用Cmake打包、VS编译，将Optix引擎库引入VS工程后，运行 optix::Context::create(); 都会报错 “缺失Optix Library”！！！

　　　　查阅文档后发现可能是驱动的问题：

　　　　Optix6.5需要435以上的驱动，我更新了我的驱动后只能到431.6，所以只能下载Optix6.0，Optix6.0需要418以上的驱动，好极了！我的驱动合适 ^_^

　　　　至此，optix::Context::create()是可以运行了。

　　2、设置VS生成ptx

　　　　我们可以在.cu文件中写 Optix 的 shader 代码，这个代码运行在GPU端，需要使用nvcc编译（cuda），但是编译完不会保留中间汇编文件.ptx，而 Optix 需要使用这个 .ptx 文件生成“光线发生器模块”。

　　　　换句话说，Optix得使用一个.ptx格式的 shader 进行GPU端的计算。那么我们必须在VS中进行相关设置，保留.ptx中间文件。　　　　

　　　　两种方式：

　cuda命令行中设置，加入 --keep ，保留所有的中间结果

　cuda的common属性配置中设置

　　然后就可以获取ptx生成“光线发生器模块”了

std::string ptxPath(const char* filename) { return std::string("./x64/Release/") + std::string(filename) + ".cu.ptx"; }
optix::Program ray_gen_program = context->createProgramFromPTXFile(ptx, "draw_solid_color");

　　这里除了从ptx文件生成program，还可以从ptx字符串生成program

context->createProgramFromPTXString("……", "draw_solid_color");

　　3、设置.cu编译方式

　　我写了一个简单的程序，编译就报错，.cu文件编译出错

　　我想到可能是编译方式的问题。我最开始设置的混编，因为cuda程序需要混编。

　　我查了资料，有人说可以混编，可是我觉得可能是混编的问题，所以修改为了只编译GPU端代码，但是！！！我之前写过cuda混编的代码，工程中含有cuda混编的代码，不能直接修改编译方式，修改了直接报错。

　　如果剔除之前cuda的混编代码还有点麻烦，想了想既然有人说可以混编，那我就不改编译方式了，结果。。。。。。，一晚上查了各种资料找不到问题所在。。。。。。

　　已经晚上11多了，一个师兄还在实验室，我过去请教了一把，师兄看到我设置的混编，告诉我不能设置为混编，他从一开始就设置的Generate .ptx file(-ptx)，但是言语之间仿佛师兄也没有设置过混编，不知道行不行。我试着改为编译GPU端-ptx方式，然后剔除了cuda的混编代码，然后就好了！

　　惊不惊喜，意不意外！

　　附上文件配置路径：

附加依赖项

　　代码：

draw_color.cu文件

#include <optix.h>
#include <optixu/optixu_math_namespace.h>
using namespace optix;

rtDeclareVariable(uint2, launch_index, rtLaunchIndex, );
rtBuffer<float4, 2> result_buffer;

rtDeclareVariable(float3, draw_color, , );

RT_PROGRAM void draw_solid_color()
{
    //result_buffer[launch_index] = make_float4(draw_color, 1.0f);
    float2 degree = make_float2(launch_index)*3.14 / 180.0f;
    result_buffer[launch_index] = make_float4(sin(degree.x), cos(degree.y), 0.0, 0.f);
}

OptixFrame.h

#pragma once
#include <iostream>
#include <sstream>
#include <fstream>
#include <optix.h>
#include <nvrtc.h>
#include <optixu/optixpp_namespace.h>
#include <sutil.h>
#include <GL/glew.h>
#include "../Model/Texture.h"

class OptixFrame
{
public:
    OptixFrame();
    ~OptixFrame();
    GLuint TransBuffer2Texture(GLfloat*);
    GLuint getSrcTexId() {
        return this->srcTexID;
    }

private:
    int width = 800, height = 800;
    optix::Context context;
    optix::Buffer outputBuffer;
    GLuint srcTexID;

private:
    void setup();
    void setColorSpace();
    std::string ptxPath(const char* filename) { return std::string("./x64/Release/") + std::string(filename) + ".cu.ptx"; }
};

OptixFrame.cpp

#include "./OptixFrame.h"


OptixFrame::OptixFrame()
{
    try
    {
        this->setup();
    }
    catch (const std::exception& e)
    {
        std::cerr << "Optix Error: " << e.what() << "\n";
    }
}

OptixFrame::~OptixFrame()
{
}

void OptixFrame::setColorSpace() {
    GLboolean use_SRGB = GL_FALSE;
    glGetBooleanv(GL_FRAMEBUFFER_SRGB_CAPABLE_EXT, &use_SRGB);
    if (use_SRGB) {
        glEnable(GL_FRAMEBUFFER_SRGB_EXT);
    }
}

GLuint OptixFrame::TransBuffer2Texture(GLfloat* data) {
    glGenTextures(1, &this->srcTexID);
    glBindTexture(GL_TEXTURE_2D, this->srcTexID);
    glTexImage2D(GL_TEXTURE_2D, 0, GL_RGBA, this->width,this->height, 0, GL_RGBA, GL_FLOAT, data);
    glGenerateMipmap(GL_TEXTURE_2D);
    //Parameters
    glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_S, GL_REPEAT);
    glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_T, GL_REPEAT);
    glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_LINEAR_MIPMAP_LINEAR);
    glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_LINEAR);
    glBindTexture(GL_TEXTURE_2D, 0);
    return this->srcTexID;
}

void OptixFrame::setup() {
    context = optix::Context::create();
    context->setRayTypeCount(1); //1种光线(有时可以2种，渲染+阴影)
    context->setEntryPointCount(1);
    //申请Buffer
    this->outputBuffer = context->createBuffer(RT_BUFFER_OUTPUT, RT_FORMAT_FLOAT4, width, height);
    context["result_buffer"]->set(outputBuffer);
    //编译Shader
    //const char* ptx = sutil::getPtxString("Optix","draw_color.cu");
    std::string ptx = this->ptxPath("draw_color");
    //从ptx字串中标识 "draw_solid_color"入口函数，并创建并返回光线发生器模块
    optix::Program ray_gen_program = context->createProgramFromPTXFile(ptx, "draw_solid_color");
    context->setRayGenerationProgram(0, ray_gen_program);
    context["draw_color"]->setFloat(0.8f, 0.7f, 0.0f);

    context->setPrintEnabled(true); //可以配合tfPrintf()进行shader端打印
    context->setPrintBufferSize(4096);

    //内部检查有效性
    context->validate();
    //开启第0个入口
    context->launch(0, width, height);

    this->setColorSpace();
    GLfloat* imageData = (GLfloat*)outputBuffer->map(0, RT_BUFFER_MAP_READ);
    this->TransBuffer2Texture(imageData);
}

　　这些代码生成了OpenGL的texture2D纹理，然后就可以在OpenGL上显示了，OpenGL上的显示代码就不做展示了。

　　效果图：

猜你喜欢