Partager via


Julia fractal using C++ AMP

In this blog post, I will share a C++ AMP implementation of a fractal generator, rendering 4 dimensional Quaternion Julia fractals. I’ll show you screenshots of the app, then we’ll dive into the code, and then I’ll share a pointer of where you can get the Visual Studio project.

Credits

The original sample was written using DirectCompute by Jan Vlietinck. The shader code in the DirectCompute version was a port of a Cg version written by Keenan Crane. I have ported the DirectCompute portions to use C++ AMP.

The app

The application presents an animation of Julia fractals by changing the generation parameters by a small amount at each time quantum. Here are some snapshots of the fractal animation.

  • Use the mouse wheel to zoom in and out
  • Hold down the left mouse button to select and rotate the fractal
  • Use ‘space’ to turn the animation ‘on’ or ‘off’

 

clip_image002clip_image004

clip_image006clip_image008

 

You can also change the parameters to the Julia fractal during the animation.

  • Use w/x to increase/decrease the x component of the quaternion constant (MuC.x)
  • Use q/z to increase/decrease the y component of the quaternion constant (MuC.y)
  • Use a/d to increase/decrease the z component of the quaternion constant (MuC.z)
  • Use e/c to increase/decrease the w component of the quaternion constant (MuC.w)
  • Use -/+ to increase/decrease the precision of intersection (epsilon)

Porting to C++ AMP

The porting of this sample was achieved using the guide C++ AMP for DirectCompute programmer. The main win32 application is implemented in QJulia4D.cpp.

We first start by removing most of the boilerplate code to create constant buffers, resource views and to compile and launch the shader since this is not needed when using C++ AMP. C++ AMP containers automatically take care of creating the buffers and resource views on the accelerator. Constants can be directly captured into the parallel_for_each. This takes care of creating and setting the required constant buffers. Finally the code to explicitly compile and launch the shader is not required when using C++ AMP. This is automatically taken care by the call to parallel_for_each.

The shader was ported to use C++ AMP in julia4DAMP.cpp. The compute shader is replaced with a parallel_for_each call as shown below.

Original Compute Shader:

  1: [numthreads(4, 64, 1)]
  2: //****************************************************************************
  3: void CS_QJulia4D( uint3 Gid : SV_GroupID, uint3 DTid : SV_DispatchThreadID, uint3 GTid : SV_GroupThreadID, uint GI : SV_GroupIndex )
  4: //****************************************************************************
  5: { 
  6:     float4 coord = float4((float)DTid.x, (float)DTid.y, 0.0f, 0.0f);
  7:  
  8:     float2 size     = float2((float)c_width, (float)c_height);
  9:     float scale     = min(size.x, size.y);
  10:     float2 half     = float2(0.5f, 0.5f);
  11:     
  12:     float2 position = (coord.xy - half * size) / scale *BOUNDING_RADIUS_2 *zoom;    
  13:  
  14:     float3 light = float3(1.5f, 0.5f, 4.0f);
  15:     float3 eye   = float3(0.0f, 0.0f, 4.0f);
  16:     float3 ray   = float3(position.x, position.y, 0.0f);
  17:     
  18:     // rotate fractal
  19:     light = mul(light, rotation);
  20:     eye   = mul(  eye, rotation);
  21:     ray   = mul(  ray, rotation);
  22:  
  23:     // ray start and ray direction
  24:     float3 rO =  eye;
  25:     float3 rD =  ray - rO;
  26:     
  27:     float4 color = QJulia(rO, rD, c_mu, c_epsilon, eye, light, c_selfShadow);
  28:     output[DTid.xy] = color;
  29: }

C++ AMP kernel:

  1: // Launch kernel using C++ AMP
  2: void ampCompute(const writeonly_texture_view<unorm4, 2>& tv, const QJulia4DConstants& mc)
  3: {
  4:     parallel_for_each(tv.accelerator_view, tv.extent.tile<64, 4>().pad(), 
  5:                      [&,tv,mc](tiled_index<64,4> ti) restrict(amp)          
  6:     {
  7:         float4 coord = float4((float)ti.global[1], (float)ti.global[0], 0.0f, 0.0f);
  8:  
  9:         float2 size     = float2((float)mc.c_width, (float)mc.c_height);                
  10:         float scale     = fminf(size.x, size.y);
  11:         float2 half     = float2(0.5f, 0.5f);
  12:         float2 position = (coord.xy - half * size) / scale *BOUNDING_RADIUS_2 *mc.zoom;
  13:  
  14:         float3 light = float3(1.5f, 0.5f, 4.0f);
  15:         float3 eye   = float3(0.0f, 0.0f, 4.0f);
  16:         float3 ray   = float3(position.x, position.y, 0.0f);
  17:  
  18:         // rotate fractal 
  19:         light = mul(light, mc.orientation);        
  20:         eye   = mul( eye, mc.orientation);
  21:         ray   = mul(  ray, mc.orientation);
  22:  
  23:         // ray start and ray direction
  24:         float3 rO =  eye;
  25:         float3 rD =  ray - rO;
  26:     
  27:         bool selfshadow = mc.selfShadow == 0  ? false : true;
  28:         float4 color = QJulia(rO, rD, mc.mu, mc.epsilon, light, selfshadow, mc);    
  29:         unorm_4 u_color(unorm(color.x), unorm(color.y), unorm(color.z), unorm(color.w));
  30:         tv.set(ti.global,u_color);
  31:     });
  32: }

 

Any exceptions from launch and execution of the shader, including TDR, are bubbled up to the user using C++ AMP runtime exceptions. The rest of the shader code remains unchanged in this port. Note that you can also use other C++ constructs when porting since C++ AMP allows us to express the kernel using C++ with some restrictions. You are welcome to try this out, I just did a vanilla port instead of writing the code in C++ the way I would if I was writing it from scratch.

The parallel_for_each kernel stores the computation results in a writeonly_texture_view created from the interop texture object of the back buffer associated with the window using the IDXGISwapChain.

  1: ...
  2: ID3D11Texture2D* pTexture;
  3: hr = g_pSwapChain->GetBuffer( 0, __uuidof( ID3D11Texture2D ), ( LPVOID* )&pTexture );
  4:  
  5: if(FAILED(hr))
  6: {
  7:     throw std::exception("Failed to get buffer");
  8: }
  9:  
  10: // create amp texture using interop
  11: accelerator_view av = concurrency::direct3d::create_accelerator_view(g_pd3dDevice);
  12: texture<unorm4, 2> tex = make_texture<unorm4, 2>(av, pTexture);
  13: g_pAmpTextureView.reset(new writeonly_texture_view<unorm4, 2>(tex));

 

And that’s it! We now have a sample that is written entirely in C++.

Download

You can download the zipped file containing the sample. Please note that the attached sample code in the “.zip” is released under BSD license. You will need Visual Studio 2012 to build the sample.

As always, please feel free to share your thoughts and ask questions below or in our MSDN concurrency forum.

Julia.zip

Comments

  • Anonymous
    November 09, 2012
    This is  not  ruanjian  

  • Anonymous
    December 24, 2012
    Unfortunately the sample app instantly crashed my video driver. GTX 460M with the WHQL drivers on Windows 8 Pro. Curious if anyone else has this issue.

  • Anonymous
    December 26, 2012
    Please try to build and run in "Release" configuration, as the default "Debug" might take much longer and hit TDR (timeout), which would appear as crash. Regards,