encode_ctx.hpp

#pragma once

#include <stdint.h>
#include <unordered_set>
#include <vector>

#include <fmt/compile.h>
#include <fmt/core.h>

#include "common.hpp"

#define defaultBufferSize 4096

class EncodeContext {
public:
    std::vector<char> buffer;
    size_t stack_depth;
    std::unordered_set<uintptr_t> seen;

    EncodeContext() {
        debug_print("new EncodeContext");
        buffer = std::vector<char>();
        stack_depth = 0;
        buffer.reserve(defaultBufferSize);
    }

    ~EncodeContext() {
        debug_print("delete context");
        seen.clear();
        buffer.clear();
    }

    void reset() {
        stack_depth = 0;
        buffer.clear();
        seen.clear();
    }

    void write(std::string_view val) { buffer.insert(buffer.end(), val.begin(), val.end()); }

    void write(const char *data, Py_ssize_t size) {
        buffer.insert(buffer.end(), data, data + size);
    }

    void writeSize_t(size_t val) { fmt::format_to(std::back_inserter(buffer), "{}", val); }

    void writePySize_t(Py_ssize_t val) { fmt::format_to(std::back_inserter(buffer), "{}", val); }

    void writeLongLong(int64_t val) { fmt::format_to(std::back_inserter(buffer), "{}", val); }

    void writeChar(const char c) { buffer.push_back(c); }
};