朱丽叶—Cuda+OSG

#include <cuda_runtime.h>
#include <osg/Image>
const int DIM = ;
 
typedef struct cuComplex
{
    float r;
    float i;
    __device__ cuComplex(float a, float b)
        :r(a)
        ,i(b)
    {
    }
 
    __device__ float magnitude2(void)
    {
        return r * r + i * i;
    }
 
    __device__ cuComplex operator * (const cuComplex & a)
    {
        return cuComplex(r * a.r - i * a.i, i * a.r + r * a.i);
    }
 
    __device__ cuComplex operator + (const cuComplex & a)
    {
        return cuComplex(r + a.r, i + a.i);
    }
};
 
__device__ int julia(int x, int y)
{
    const float scale = 1.5;
    float jx = scale * (float)(DIM/ - x)/(DIM/);
    float jy = scale * (float)(DIM/ -y)/(DIM/);
    cuComplex c(-0.8, 0.156);
    cuComplex a(jx, jy);
 
    for (int i = ; i < ; ++i)
    {
        a = a * a + c;
        if (a.magnitude2() > )
            return ;
    }
    return ;
}
 
__global__ void kernel(unsigned char * ptr)
{
    int x = blockIdx.x;
    int y = blockIdx.y;
    int offset = x + y * gridDim.x;
    int juliaValue = julia(x, y);
    ptr[offset *  + ] =  * juliaValue;
    ptr[offset *  + ] = ;
    ptr[offset *  + ] = ;
    ptr[offset *  + ] = ;
 
}
 
extern "C" void SetUp(osg::Image * image)
{
    unsigned char * dev_bitmap;
    cudaMalloc((void **)&dev_bitmap, DIM * DIM * );
    dim3 grid(DIM, DIM);
    kernel<<<grid, >>>(dev_bitmap);
    cudaMemcpy(image->data(), dev_bitmap, DIM * DIM * , cudaMemcpyDeviceToHost);
    cudaFree(dev_bitmap);
}

#include <osgViewer/Viewer>
#include <osg/Texture2D>
#include <osg/Image>
#include <osgDB/WriteFile>
#include <osgViewer/ViewerEventHandlers>
#pragma comment(lib, "osgViewerd.lib")
#pragma comment(lib, "osgDBd.lib")
#pragma comment(lib, "osgd.lib")
#pragma comment(lib, "osgGAd.lib")
const int DIM = ; 
 
osg::ref_ptr<osg::Geode> CreateQuad()
{
    osg::ref_ptr<osg::Geode> geode = new osg::Geode;
    osg::ref_ptr<osg::Geometry> geometry = new osg::Geometry;
    osg::ref_ptr<osg::Vec3Array> vArray = new osg::Vec3Array;
    osg::ref_ptr<osg::Vec2Array> tArray = new osg::Vec2Array;
    osg::ref_ptr<osg::Vec3Array> nArray = new osg::Vec3Array;
    vArray->push_back(osg::Vec3(-1.0, 0.0, -1.0));
    vArray->push_back(osg::Vec3(1.0, 0.0, -1.0));
    vArray->push_back(osg::Vec3(1.0, 0.0, 1.0));
    vArray->push_back(osg::Vec3(-1.0, 0.0, 1.0));
 
    tArray->push_back(osg::Vec2(0.0, 0.0));
    tArray->push_back(osg::Vec2(1.0, 0.0));
    tArray->push_back(osg::Vec2(1.0, 1.0));
    tArray->push_back(osg::Vec2(0.0, 1.0));
 
    nArray->push_back(osg::Vec3(0.0, 1.0, 0.0));
    geometry->setVertexArray(vArray.get());
    geometry->setTexCoordArray(, tArray.get());
    geometry->setNormalArray(nArray.get());
    geometry->setNormalBinding(osg::Geometry::BIND_OVERALL);
    geometry->addPrimitiveSet(new osg::DrawArrays(osg::PrimitiveSet::QUADS, , vArray->size()));
    geode->addDrawable(geometry.get());
    return geode.get();
}
 
osg::ref_ptr<osg::StateSet> CreateStateSet(osg::Image * image)
{
    osg::ref_ptr<osg::StateSet> stateSet = new osg::StateSet;
 
    osg::ref_ptr<osg::Texture2D> texture = new osg::Texture2D();
    texture->setImage(image);
    stateSet->setTextureAttributeAndModes(, texture.get(), osg::StateAttribute::ON);
    return stateSet.get();
}
extern "C" void SetUp(osg::Image * image);
int main()
{
    osg::ref_ptr<osgViewer::Viewer> viewer = new osgViewer::Viewer;
 
    osg::ref_ptr<osg::Image> image = new osg::Image;
    image->allocateImage(DIM, DIM, , GL_RGBA, GL_UNSIGNED_BYTE);
    SetUp(image.get());
 
    osg::ref_ptr<osg::StateSet> stateSet = CreateStateSet(image.get());
    osg::ref_ptr<osg::Geode> quad = CreateQuad();
    quad->setStateSet(stateSet.get());
    viewer->setSceneData(quad.get());
    viewer->getCamera()->getOrCreateStateSet()->setMode(GL_LIGHTING, osg::StateAttribute::OFF);
    viewer->addEventHandler(new osgViewer::StatsHandler);
    viewer->setUpViewInWindow(, , , );
    viewer->run();
    return ;
}