refraction_shader_class.h

#ifndef _REFRACTIONSHADERCLASS_H_
#define _REFRACTIONSHADERCLASS_H_
 
 
// INCLUDES //
#include <d3d11.h>
#include <d3dcompiler.h>
#include <directxmath.h>
#include <fstream>
using namespace DirectX;
using namespace std;
 
 
// Class name: refraction_shader_class
class refraction_shader_class
{
private:
    struct matrix_buffer_type
    {
        XMMATRIX world;
        XMMATRIX view;
        XMMATRIX projection;
    };
 
    struct light_buffer_type
    {
        XMFLOAT4 ambient_color;
        XMFLOAT4 diffuse_color;
        XMFLOAT4 light_position;
        XMFLOAT3 light_direction;
        float padding;
    };
 
    struct clip_plane_buffer_type
    {
        XMFLOAT4 clip_plane;
    };
 
public:
    refraction_shader_class();
    refraction_shader_class(const refraction_shader_class&);
    ~refraction_shader_class();
 
    bool initialize(ID3D11Device*, HWND);
    void shutdown();
    bool render(ID3D11DeviceContext*, int, XMMATRIX, XMMATRIX, XMMATRIX, ID3D11ShaderResourceView*,
        XMFLOAT3, XMFLOAT4[], XMFLOAT4[], XMFLOAT4[], XMFLOAT4);
 
private:
    bool initialize_shader(ID3D11Device*, HWND, WCHAR*, WCHAR*);
    void shutdown_shader();
    void output_shader_error_message(ID3D10Blob*, HWND, WCHAR*);
 
    bool set_shader_parameters(ID3D11DeviceContext*, XMMATRIX, XMMATRIX, XMMATRIX, ID3D11ShaderResourceView*,
        XMFLOAT3, XMFLOAT4[], XMFLOAT4[], XMFLOAT4[], XMFLOAT4);
    void render_shader(ID3D11DeviceContext*, int);
 
private:
    ID3D11VertexShader* vertex_shader_;
    ID3D11PixelShader* pixel_shader_;
    ID3D11InputLayout* layout_;
    ID3D11SamplerState* sample_state_;
    ID3D11Buffer* matrix_buffer_;
    ID3D11Buffer* light_buffer_;
    ID3D11Buffer* clip_plane_buffer_;
};
 
#endif