/**************************************************************************** Copyright (c) 2018-2019 Xiamen Yaji Software Co., Ltd. Copyright (c) 2020 c4games.com 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. ****************************************************************************/ #include "UtilsMTL.h" #include "DeviceMTL.h" #include "TextureMTL.h" #include "base/CCConfiguration.h" #define COLOR_ATTAHCMENT_PIXEL_FORMAT MTLPixelFormatBGRA8Unorm CC_BACKEND_BEGIN id UtilsMTL::_defaultColorAttachmentTexture = nil; id UtilsMTL::_defaultDepthStencilAttachmentTexture = nil; namespace { #define byte(n) ((n) * 8) #define bit(n) (n) uint8_t getBitsPerElement(MTLPixelFormat pixleFormat) { switch (pixleFormat) { case MTLPixelFormatDepth32Float_Stencil8: return byte(8); case MTLPixelFormatBGRA8Unorm: case MTLPixelFormatRGBA8Unorm: case MTLPixelFormatDepth32Float: return byte(4); #if (CC_TARGET_PLATFORM == CC_PLATFORM_MAC) case MTLPixelFormatDepth24Unorm_Stencil8: return byte(4); #else case MTLPixelFormatABGR4Unorm: case MTLPixelFormatBGR5A1Unorm: case MTLPixelFormatB5G6R5Unorm: case MTLPixelFormatA1BGR5Unorm: return byte(2); #endif case MTLPixelFormatA8Unorm: case MTLPixelFormatR8Unorm: return byte(1); default: assert(false); break; } return 0; } MTLPixelFormat getSupportedDepthStencilFormat() { MTLPixelFormat pixelFormat = MTLPixelFormatDepth32Float_Stencil8; #if (CC_TARGET_PLATFORM == CC_PLATFORM_MAC) bool isDepth24Stencil8PixelFormatSupported = Configuration::getInstance()->supportsOESPackedDepthStencil(); if(isDepth24Stencil8PixelFormatSupported) pixelFormat = MTLPixelFormatDepth24Unorm_Stencil8; #endif return pixelFormat; } } MTLPixelFormat UtilsMTL::getDefaultDepthStencilAttachmentPixelFormat() { return getSupportedDepthStencilFormat(); } MTLPixelFormat UtilsMTL::getDefaultColorAttachmentPixelFormat() { return COLOR_ATTAHCMENT_PIXEL_FORMAT; } id UtilsMTL::getDefaultDepthStencilTexture() { if (! _defaultDepthStencilAttachmentTexture) _defaultDepthStencilAttachmentTexture = UtilsMTL::createDepthStencilAttachmentTexture(); return _defaultDepthStencilAttachmentTexture; } void UtilsMTL::updateDefaultColorAttachmentTexture(id texture) { UtilsMTL::_defaultColorAttachmentTexture = texture; } MTLPixelFormat UtilsMTL::toMTLPixelFormat(PixelFormat textureFormat) { switch (textureFormat) { #if (CC_TARGET_PLATFORM == CC_PLATFORM_IOS) case PixelFormat::MTL_ABGR4: return MTLPixelFormatABGR4Unorm; case PixelFormat::MTL_BGR5A1: return MTLPixelFormatBGR5A1Unorm; case PixelFormat::MTL_B5G6R5: return MTLPixelFormatB5G6R5Unorm; case PixelFormat::PVRTC4A: return MTLPixelFormatPVRTC_RGBA_4BPP; case PixelFormat::PVRTC4: return MTLPixelFormatPVRTC_RGB_4BPP; case PixelFormat::PVRTC2A: return MTLPixelFormatPVRTC_RGBA_2BPP; case PixelFormat::PVRTC2: return MTLPixelFormatPVRTC_RGB_2BPP; case PixelFormat::ETC1: case PixelFormat::ETC2_RGB: return MTLPixelFormatETC2_RGB8; case PixelFormat::ETC2_RGBA: return MTLPixelFormatEAC_RGBA8; case PixelFormat::ASTC4x4: return MTLPixelFormatASTC_4x4_LDR; case PixelFormat::ASTC6x6: return MTLPixelFormatASTC_6x6_LDR; case PixelFormat::ASTC8x8: return MTLPixelFormatASTC_8x8_LDR; #else case PixelFormat::S3TC_DXT1: return MTLPixelFormatBC1_RGBA; case PixelFormat::S3TC_DXT3: return MTLPixelFormatBC2_RGBA; case PixelFormat::S3TC_DXT5: return MTLPixelFormatBC3_RGBA; #endif case PixelFormat::RGBA8888: return MTLPixelFormatRGBA8Unorm; // Should transfer the data to match pixel format when updating data. case PixelFormat::RGB888: return MTLPixelFormatRGBA8Unorm; case PixelFormat::A8: return MTLPixelFormatA8Unorm; case PixelFormat::BGRA8888: return MTLPixelFormatBGRA8Unorm; //on mac, D24S8 means MTLPixelFormatDepth24Unorm_Stencil8, while on ios it means MTLPixelFormatDepth32Float_Stencil8 case PixelFormat::D24S8: return getSupportedDepthStencilFormat(); case PixelFormat::DEFAULT: return COLOR_ATTAHCMENT_PIXEL_FORMAT; case PixelFormat::NONE: default: return MTLPixelFormatInvalid; } } void UtilsMTL::resizeDefaultAttachmentTexture(std::size_t width, std::size_t height) { [backend::DeviceMTL::getCAMetalLayer() setDrawableSize:CGSizeMake(width, height)]; [_defaultDepthStencilAttachmentTexture release]; _defaultDepthStencilAttachmentTexture = UtilsMTL::createDepthStencilAttachmentTexture(); } id UtilsMTL::createDepthStencilAttachmentTexture() { auto CAMetalLayer = DeviceMTL::getCAMetalLayer(); MTLTextureDescriptor* textureDescriptor = [[MTLTextureDescriptor alloc] init]; textureDescriptor.width = CAMetalLayer.drawableSize.width; textureDescriptor.height = CAMetalLayer.drawableSize.height; textureDescriptor.pixelFormat = getSupportedDepthStencilFormat(); textureDescriptor.resourceOptions = MTLResourceStorageModePrivate; textureDescriptor.usage = MTLTextureUsageRenderTarget; auto ret = [CAMetalLayer.device newTextureWithDescriptor:textureDescriptor]; [textureDescriptor release]; return ret; } void UtilsMTL::generateMipmaps(id texture) { auto commandQueue = static_cast(DeviceMTL::getInstance())->getMTLCommandQueue(); auto commandBuffer = [commandQueue commandBuffer]; id commandEncoder = [commandBuffer blitCommandEncoder]; [commandEncoder generateMipmapsForTexture:texture]; [commandEncoder endEncoding]; [commandBuffer commit]; } void UtilsMTL::swizzleImage(unsigned char *image, std::size_t width, std::size_t height, MTLPixelFormat format) { if(!image) return; auto len = width * height; switch (format) { //convert to RGBA case MTLPixelFormatBGRA8Unorm: for(int i=0; igetTextureType()){ case TextureType::TEXTURE_2D: UtilsMTL::readPixels(static_cast(texture)->getMTLTexture(), origX, origY, rectWidth, rectHeight, pbd); break; case TextureType::TEXTURE_CUBE: UtilsMTL::readPixels(static_cast(texture)->getMTLTexture(), origX, origY, rectWidth, rectHeight, pbd); break; } } void UtilsMTL::readPixels(id texture, std::size_t origX, std::size_t origY, std::size_t rectWidth, std::size_t rectHeight, PixelBufferDescriptor& pbd) { NSUInteger texWidth = texture.width; NSUInteger texHeight = texture.height; MTLRegion region = MTLRegionMake2D(0, 0, texWidth, texHeight); MTLRegion imageRegion = MTLRegionMake2D(origX, origY, rectWidth, rectHeight); MTLTextureDescriptor* textureDescriptor = [MTLTextureDescriptor texture2DDescriptorWithPixelFormat:[texture pixelFormat] width:texWidth height:texHeight mipmapped:NO]; id device = static_cast(DeviceMTL::getInstance())->getMTLDevice(); id readPixelsTexture = [device newTextureWithDescriptor:textureDescriptor]; id commandQueue = static_cast(DeviceMTL::getInstance())->getMTLCommandQueue(); auto commandBuffer = [commandQueue commandBuffer]; // [commandBuffer enqueue]; id blitCommandEncoder = [commandBuffer blitCommandEncoder]; [blitCommandEncoder copyFromTexture:texture sourceSlice:0 sourceLevel:0 sourceOrigin:region.origin sourceSize:region.size toTexture:readPixelsTexture destinationSlice:0 destinationLevel:0 destinationOrigin:region.origin]; #if (CC_TARGET_PLATFORM == CC_PLATFORM_MAC) [blitCommandEncoder synchronizeResource:readPixelsTexture]; #endif [blitCommandEncoder endEncoding]; [commandBuffer addCompletedHandler:^(id commandBufferMTL) { auto bytePerRow = rectWidth * getBitsPerElement(texture.pixelFormat) / 8; auto texelData = pbd._data.resize(bytePerRow * rectHeight); if(texelData != nullptr) { [readPixelsTexture getBytes:texelData bytesPerRow:bytePerRow fromRegion:imageRegion mipmapLevel:0]; swizzleImage(texelData, rectWidth, rectHeight, readPixelsTexture.pixelFormat); pbd._width = rectWidth; pbd._height = rectHeight; } [readPixelsTexture release]; }]; [commandBuffer commit]; [commandBuffer waitUntilCompleted]; } CC_BACKEND_END