1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
|
//
// Copyright 2015 Pixar
//
// Licensed under the Apache License, Version 2.0 (the "Apache License")
// with the following modification; you may not use this file except in
// compliance with the Apache License and the following modification to it:
// Section 6. Trademarks. is deleted and replaced with:
//
// 6. Trademarks. This License does not grant permission to use the trade
// names, trademarks, service marks, or product names of the Licensor
// and its affiliates, except as required to comply with Section 4(c) of
// the License and to reproduce the content of the NOTICE file.
//
// You may obtain a copy of the Apache License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the Apache License with the above modification is
// distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
// KIND, either express or implied. See the Apache License for the specific
// language governing permissions and limitations under the Apache License.
//
#ifndef OPENSUBDIV_EXAMPLES_MTL_UTILS_H
#define OPENSUBDIV_EXAMPLES_MTL_UTILS_H
#include <Metal/Metal.h>
#include <cstring>
#include <TargetConditionals.h>
#include <vector>
namespace OpenSubdiv { namespace OPENSUBDIV_VERSION { namespace Osd {
#if TARGET_OS_EMBEDDED
constexpr auto MTLDefaultStorageMode = MTLResourceStorageModeShared;
#else
constexpr auto MTLDefaultStorageMode = MTLResourceStorageModeManaged;
#endif
template<typename T>
id<MTLBuffer> MTLNewBufferFromVector(id<MTLDevice> device, const std::vector<T>& vec) {
const auto typeSize = sizeof(T);
const auto bufferSize = typeSize * vec.size();
return [device newBufferWithBytes:vec.data() length:bufferSize options:MTLDefaultStorageMode];
}
template<typename DataType, size_t RingLength>
class MTLRingBuffer {
private:
id<MTLBuffer> _buffers[RingLength];
uint _bufferIndex;
public:
template<typename SizedType = DataType>
MTLRingBuffer(id<MTLDevice> device, size_t count, NSString* label, MTLResourceOptions options = MTLDefaultStorageMode) {
alloc<SizedType>(device, count, options, label);
}
MTLRingBuffer() {
for(uint i = 0; i < RingLength; i++) {
_buffers[i] = nil;
}
_bufferIndex = 0;
}
void markModified() const {
#if !TARGET_OS_EMBEDDED
const auto bufferLength = [buffer() length];
[buffer() didModifyRange: NSMakeRange(0, bufferLength)];
#endif
}
template<typename SizedType = DataType>
void alloc(id<MTLDevice> device, size_t count, NSString* label, MTLResourceOptions options = MTLDefaultStorageMode) {
if(count) {
for(uint i = 0; i < RingLength; i++) {
_buffers[i] = [device newBufferWithLength: count * sizeof(SizedType) options:options];
_buffers[i].label = label;
_bufferIndex = i;
if((options & MTLResourceStorageModePrivate) == 0)
{
memset(data(), 0, count * sizeof(SizedType));
markModified();
}
}
} else {
for(uint i = 0; i < RingLength; i++) {
_buffers[i] = nil;
}
}
_bufferIndex = 0;
}
id<MTLBuffer> buffer() const {
return _buffers[_bufferIndex];
}
DataType* data() const {
return reinterpret_cast<DataType*>([buffer() contents]);
}
void next() {
_bufferIndex = (_bufferIndex + 1) % RingLength;
}
operator id<MTLBuffer>() const {
return buffer();
}
DataType* operator->() const {
return data();
}
DataType& operator[](int idx) const {
return (data())[idx];
}
};
#endif // OPENSUBDIV_EXAMPLES_MTL_UTILS_H
} } using namespace OPENSUBDIV_VERSION; }
|
