blob: dcd459593e6ddbd5a1ada2f2b9ee984f333d6142 [file] [log] [blame]
/*
* Copyright 2021 Google LLC
*
* Use of this source code is governed by a BSD-style license that can be
* found in the LICENSE file.
*/
#ifndef skgpu_KeyBuilder_DEFINED
#define skgpu_KeyBuilder_DEFINED
#include "include/core/SkString.h"
#include "include/private/base/SkAssert.h"
#include "include/private/base/SkTArray.h"
#include <cstdint>
#include <string_view>
namespace skgpu {
class KeyBuilder {
public:
KeyBuilder(skia_private::TArray<uint32_t, true>* data) : fData(data) {}
virtual ~KeyBuilder() {
// Ensure that flush was called before we went out of scope
SkASSERT(fBitsUsed == 0);
}
virtual void addBits(uint32_t numBits, uint32_t val, std::string_view label) {
SkASSERT(numBits > 0 && numBits <= 32);
SkASSERT(numBits == 32 || (val < (1u << numBits)));
fCurValue |= (val << fBitsUsed);
fBitsUsed += numBits;
if (fBitsUsed >= 32) {
// Overflow, start a new working value
fData->push_back(fCurValue);
uint32_t excess = fBitsUsed - 32;
fCurValue = excess ? (val >> (numBits - excess)) : 0;
fBitsUsed = excess;
}
SkASSERT(fCurValue < (1u << fBitsUsed));
}
void addBytes(uint32_t numBytes, const void* data, std::string_view label) {
const uint8_t* bytes = reinterpret_cast<const uint8_t*>(data);
for (; numBytes --> 0; bytes++) {
this->addBits(8, *bytes, label);
}
}
void addBool(bool b, std::string_view label) {
this->addBits(1, b, label);
}
void add32(uint32_t v, std::string_view label = "unknown") {
this->addBits(32, v, label);
}
virtual void appendComment(const char* comment) {}
// Introduces a word-boundary in the key. Must be called before using the key with any cache,
// but can also be called to create a break between generic data and backend-specific data.
void flush() {
if (fBitsUsed) {
fData->push_back(fCurValue);
fCurValue = 0;
fBitsUsed = 0;
}
}
private:
skia_private::TArray<uint32_t, true>* fData;
uint32_t fCurValue = 0;
uint32_t fBitsUsed = 0; // ... in current value
};
class StringKeyBuilder : public KeyBuilder {
public:
StringKeyBuilder(skia_private::TArray<uint32_t, true>* data) : KeyBuilder(data) {}
void addBits(uint32_t numBits, uint32_t val, std::string_view label) override {
KeyBuilder::addBits(numBits, val, label);
fDescription.appendf("%.*s: %u\n", (int)label.size(), label.data(), val);
}
void appendComment(const char* comment) override {
fDescription.appendf("%s\n", comment);
}
SkString description() const { return fDescription; }
private:
SkString fDescription;
};
} // namespace skgpu
#endif // skgpu_KeyBuilder_DEFINED