#include"Std3DMaterial.h"
#include<hgl/shadergen/MaterialCreateInfo.h>

STD_MTL_NAMESPACE_BEGIN
namespace
{
    // Gizmo3D材质其实就是纯色的blinnphong材质，但不需要外部传入太阳光方向、高光系数等数据。
    // 其全部在Shader中直接包含，它是专门为Gizmo 3D控件所准备的一种材质。
    
    constexpr const char mi_codes[]="vec4 Color;";                      //材质实例代码
    constexpr const uint32_t mi_bytes=sizeof(Vector4f);                 //材质实例数据大小

    constexpr const char vs_main[]=R"(
void main()
{
    HandoverMI();

    Output.Normal   =GetNormal();
    Output.Position =GetPosition3D();

    gl_Position     =Output.Position;
})";

    //一个shader中输出的所有数据，会被定义在一个名为Output的结构中。所以编写时要用Output.XXXX来使用。
    //而同时，这个结构在下一个Shader中以Input名称出现，使用时以Input.XXX的形式使用。

    constexpr const char fs_main[]=R"(

const vec3 SUN_DIRECTION=vec3(0.655386,0.491539,0.573462);      //normalized(8,6,7)
const vec3 SUN_COLOR=vec3(1.0,1.0,1.0);

void main()
{
    MaterialInstance mi=GetMI();

    //点乘法线和光照
    float intensity=0.5*max(dot(Input.Normal,SUN_DIRECTION),0.0)+0.5;

    //直接光颜色
    vec3 direct_color=intensity*SUN_COLOR*mi.Color.rgb;

    vec3 spec_color=vec3(0);

    if(intensity>0.0)
    {
        vec3 half_vector=normalize(SUN_DIRECTION+normalize(Input.Position.xyz+camera.pos));

        float specular=max(dot(half_vector,Input.Normal),0.0);

        spec_color=specular*pow(specular,64)*SUN_COLOR;
    }
        
    FragColor=vec4(direct_color+spec_color,1.0);
})";

    class MaterialGizmo3D:public Std3DMaterial
    {
    public:

        using Std3DMaterial::Std3DMaterial;
        ~MaterialGizmo3D()=default;

        bool CustomVertexShader(ShaderCreateInfoVertex *vsc) override
        {
            vsc->AddInput(VAT_VEC3,VAN::Normal);

            vsc->AddOutput(SVT_VEC4,"Position");
            vsc->AddOutput(SVT_VEC3,"Normal");

            if(!Std3DMaterial::CustomVertexShader(vsc))     //会根据是否有GetNormal函数来决定是否添加Normal计算代码，所以需要放在AddInput后面
                return(false);

            vsc->SetMain(vs_main);
            return(true);
        }

        bool CustomFragmentShader(ShaderCreateInfoFragment *fsc) override
        {
            fsc->AddOutput(VAT_VEC4,"FragColor");       //Fragment shader的输出等于最终的RT了，所以这个名称其实随便起。

            fsc->SetMain(fs_main);
            return(true);
        }

        bool EndCustomShader() override
        {
            mci->SetMaterialInstance(mi_codes,                       //材质实例glsl代码
                                     mi_bytes,                       //材质实例数据大小
                                     VK_SHADER_STAGE_FRAGMENT_BIT);  //只在Vertex Shader中使用材质实例最终数据

            return(true);
        }
    };//class MaterialGizmo3D:public Std3DMaterial
}//namespace

MaterialCreateInfo *CreateMaterialGizmo3D(const Material3DCreateConfig *cfg)
{
    MaterialGizmo3D mg3d(cfg);

    return mg3d.Create();
}
STD_MTL_NAMESPACE_END