PixelShader.h

浏览该文件的文档.

//
// Created by Administrator on 2025/4/9.
//
#include "../MutilThreadBase/RefCount.h"
#include "Vertex.h"
#include "Context.h"
#include "../RHI/IPipelineResource.h"
#include "../RHI/ShaderBinary.h"
#ifndef FCT_PIXELSHADER_H
#define FCT_PIXELSHADER_H
namespace FCT {
    class PixelShader : public IPipelineResource {
    public:
        PipelineResourceType getType() const override
        {
            return PipelineResourceType::PixelShader;
        }
        PixelShader(Context* ctx): m_pixelShader(nullptr)
        {
            m_ctx = ctx;
        }
 
        ~PixelShader()
        {
            if (m_pixelShader)
            {
                m_pixelShader->release();
            }
        }
        void code(std::string source)
        {
            m_userSource = source;
        }
        void create()
        {
            m_pixelShader = m_ctx->createResource<RHI::PixelShader>();
            if (m_userSource.empty())
            {
                generateDefaultCode();
            }
            if (m_binaryCode.code().empty())
            {
                preprocess();
                compile();
            }
            m_pixelShader->code(m_binaryCode.code());
            m_pixelShader->create();
        }
        RHI::ShaderBinary binaryCode()
        {
            return m_binaryCode;
        }
        RHI::PixelShader* pixelShader()
        {
            return m_pixelShader;
        }
        void pixelLayout(const PixelLayout& layout)
        {
            m_pixelLayout = layout;
        }
        void addUniform(ConstLayout layout)
        {
            m_uniformLayouts.push_back(layout);
        }
        void resourceLayout(ResourceLayout layout)
        {
            m_resourceLayout = layout;
        }
    protected:
        void generateDefaultCode()
        {
            m_userSource = m_ctx->getGenerator()->generateDefaultPixelMain(m_pixelLayout);
        }
        void compile()
        {
            m_binaryCode.code(
                m_ctx->getCompiler()->compile(m_source,"FCTEntry",ShaderKind::FragmentShader)
                );
        }
        void preprocess()
        {
            m_source = m_ctx->getGenerator()->generatePixelShader(m_pixelLayout,m_uniformLayouts,m_binaryCode,m_resourceLayout, m_userSource);
        }
        RHI::ShaderBinary m_binaryCode;
        Context* m_ctx;
        PixelLayout m_pixelLayout;
        RHI::PixelShader* m_pixelShader;
        std::string m_userSource;
        std::string m_source;
        std::vector<ConstLayout> m_uniformLayouts;
        ResourceLayout m_resourceLayout;
    };
 
}
#endif //FCT_PIXELSHADER_H