搭建 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上的显示代码就不做展示了。

  

  效果图:

  

  

  

  

posted @ 2019-10-25 14:44  茶飘香~  阅读(538)  评论(2编辑  收藏