KeyBuilder.h

Go to the documentation of this file.

/*
 * 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