blob: d38d1b5a3308907b1bd4cc8a6f51a5aa0aee8506 [file] [log] [blame]
// Copyright 2013 The Flutter Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#pragma once
#include <Foundation/Foundation.h>
#include <Metal/Metal.h>
#include <memory>
#include <string>
#include <unordered_map>
#include "flutter/fml/macros.h"
#include "impeller/base/comparable.h"
#include "impeller/base/thread.h"
#include "impeller/renderer/shader_key.h"
#include "impeller/renderer/shader_library.h"
namespace impeller {
class ShaderLibraryMTL final : public ShaderLibrary {
public:
ShaderLibraryMTL();
// |ShaderLibrary|
~ShaderLibraryMTL() override;
// |ShaderLibrary|
bool IsValid() const override;
private:
friend class ContextMTL;
UniqueID library_id_;
mutable RWMutex libraries_mutex_;
NSMutableArray<id<MTLLibrary>>* libraries_ IPLR_GUARDED_BY(libraries_mutex_) =
nullptr;
ShaderFunctionMap functions_;
bool is_valid_ = false;
ShaderLibraryMTL(NSArray<id<MTLLibrary>>* libraries);
// |ShaderLibrary|
std::shared_ptr<const ShaderFunction> GetFunction(std::string_view name,
ShaderStage stage) override;
// |ShaderLibrary|
void RegisterFunction(std::string name,
ShaderStage stage,
std::shared_ptr<fml::Mapping> code,
RegistrationCallback callback) override;
// |ShaderLibrary|
void UnregisterFunction(std::string name, ShaderStage stage) override;
id<MTLDevice> GetDevice() const;
void RegisterLibrary(id<MTLLibrary> library);
FML_DISALLOW_COPY_AND_ASSIGN(ShaderLibraryMTL);
};
} // namespace impeller