mirror of https://github.com/axmolengine/axmol.git
144 lines
4.7 KiB
Objective-C
144 lines
4.7 KiB
Objective-C
/****************************************************************************
|
|
Copyright (c) 2018-2019 Xiamen Yaji Software Co., Ltd.
|
|
|
|
http://www.cocos2d-x.org
|
|
|
|
Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
of this software and associated documentation files (the "Software"), to deal
|
|
in the Software without restriction, including without limitation the rights
|
|
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
copies of the Software, and to permit persons to whom the Software is
|
|
furnished to do so, subject to the following conditions:
|
|
|
|
The above copyright notice and this permission notice shall be included in
|
|
all copies or substantial portions of the Software.
|
|
|
|
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
|
THE SOFTWARE.
|
|
****************************************************************************/
|
|
|
|
#pragma once
|
|
|
|
#include "../ShaderModule.h"
|
|
#include "../Types.h"
|
|
|
|
#include <string>
|
|
#include <vector>
|
|
#include <memory>
|
|
#include <unordered_map>
|
|
#import <Metal/Metal.h>
|
|
|
|
struct glslopt_shader;
|
|
|
|
CC_BACKEND_BEGIN
|
|
/**
|
|
* @addtogroup _metal
|
|
* @{
|
|
*/
|
|
|
|
/**
|
|
* To Create a vertex or fragment shader.
|
|
*/
|
|
class ShaderModuleMTL : public ShaderModule
|
|
{
|
|
public:
|
|
/**
|
|
* @param mtlDevice The device for which MTLFunction object was created.
|
|
* @param stage Specify what kinds of shader to be created.
|
|
* @param source Specify the shader source.
|
|
*/
|
|
ShaderModuleMTL(id<MTLDevice> mtlDevice, ShaderStage stage, std::string_view source);
|
|
~ShaderModuleMTL();
|
|
|
|
/**
|
|
* Get MTLFunction object.
|
|
* @return A MTLFunction object.
|
|
*/
|
|
inline id<MTLFunction> getMTLFunction() const { return _mtlFunction; }
|
|
|
|
/**
|
|
* Get current shader uniform informatino.
|
|
* @return Uniform information. Key is each uniform name, Value is corresponding uniform info.
|
|
*/
|
|
inline const UniformInfo& getActiveUniform(int location) { return _activeUniformInfos[location]; }
|
|
|
|
/**
|
|
* Get all uniformInfos.
|
|
* @return The uniformInfos.
|
|
*/
|
|
inline const hlookup::string_map<UniformInfo>& getAllActiveUniformInfo() const { return _uniformInfos; }
|
|
|
|
/**
|
|
* Get maximum uniform location.
|
|
* @return Maximum uniform location.
|
|
*/
|
|
inline const int getMaxLocation() const { return _maxLocation; }
|
|
|
|
/**
|
|
* Get active attribute informations.
|
|
* @return Active attribute informations. key is attribute name and Value is corresponding attribute info.
|
|
*/
|
|
inline const hlookup::string_map<AttributeBindInfo> getAttributeInfo() const { return _attributeInfo; }
|
|
|
|
/**
|
|
* Get uniform location by engine built-in uniform enum name.
|
|
* @param name Specifies the engine built-in uniform enum name.
|
|
* @return The uniform location.
|
|
*/
|
|
int getUniformLocation(Uniform name) const;
|
|
|
|
/**
|
|
* Get uniform location by name.
|
|
* @param uniform Specifies the uniform name.
|
|
* @return The uniform location.
|
|
*/
|
|
int getUniformLocation(std::string_view name) const;
|
|
|
|
/**
|
|
* Get attribute location by engine built-in attribute enum name.
|
|
* @param name Specifies the engine built-in attribute enum name.
|
|
* @return The attribute location.
|
|
*/
|
|
int getAttributeLocation(Attribute name) const;
|
|
|
|
/**
|
|
* Get attribute location by attribute name.
|
|
* @param name Specifies the attribute name.
|
|
* @return The attribute location.
|
|
*/
|
|
int getAttributeLocation(std::string name);
|
|
|
|
/**
|
|
* Get uniform buffer size in bytes that holds all the uniforms.
|
|
* @return The uniform buffer size.
|
|
*/
|
|
inline std::size_t getUniformBufferSize() const { return _uniformBufferSize; }
|
|
|
|
private:
|
|
void parseAttibute(id<MTLDevice> mtlDevice, glslopt_shader* shader);
|
|
void parseUniform(id<MTLDevice> mtlDevice, glslopt_shader* shader);
|
|
void parseTexture(id<MTLDevice> mtlDevice, glslopt_shader* shader);
|
|
void setBuiltinUniformLocation();
|
|
void setBuiltinAttributeLocation();
|
|
|
|
id<MTLFunction> _mtlFunction = nil;
|
|
|
|
hlookup::string_map<UniformInfo> _uniformInfos;
|
|
std::unordered_map<int, UniformInfo> _activeUniformInfos;
|
|
hlookup::string_map<AttributeBindInfo> _attributeInfo;
|
|
|
|
int _maxLocation = -1;
|
|
int _uniformLocation[UNIFORM_MAX];
|
|
int _attributeLocation[ATTRIBUTE_MAX];
|
|
std::size_t _uniformBufferSize = 0;
|
|
};
|
|
|
|
// end of _metal group
|
|
/// @}
|
|
CC_BACKEND_END
|