72 lines
1.3 KiB
C++
72 lines
1.3 KiB
C++
#pragma once
|
|
|
|
#include "buffer/buffer.h"
|
|
#include "pch.h"
|
|
|
|
struct Shard {
|
|
enum struct ShardKind {
|
|
Branch,
|
|
Petal
|
|
} kind;
|
|
|
|
uint32_t length;
|
|
uint32_t lines;
|
|
|
|
std::atomic_uint32_t refs;
|
|
|
|
Shard(ShardKind kind, uint32_t length, uint32_t lines)
|
|
: kind(kind), length(length), lines(lines), refs(0) {};
|
|
};
|
|
|
|
struct ShardPtr {
|
|
Shard *ptr;
|
|
|
|
ShardPtr(const ShardPtr &other) : ptr(other.ptr) {
|
|
if (ptr)
|
|
ptr->refs++;
|
|
}
|
|
|
|
ShardPtr &operator=(const ShardPtr &other) {
|
|
if (ptr == other.ptr)
|
|
return *this;
|
|
if (ptr && --ptr->refs == 0)
|
|
delete ptr;
|
|
ptr = other.ptr;
|
|
if (ptr)
|
|
ptr->refs++;
|
|
return *this;
|
|
}
|
|
|
|
ShardPtr(Shard *p = nullptr) : ptr(p) {
|
|
if (ptr)
|
|
ptr->refs++;
|
|
}
|
|
|
|
~ShardPtr() {
|
|
if (ptr && --ptr->refs == 0)
|
|
delete ptr;
|
|
}
|
|
};
|
|
|
|
struct Branch : Shard {
|
|
ShardPtr left;
|
|
ShardPtr right;
|
|
|
|
Branch(Shard *l, Shard *r)
|
|
: Shard(ShardKind::Branch, l->length + r->length, l->lines + r->lines),
|
|
left(l), right(r) {};
|
|
};
|
|
|
|
struct Petal : Shard {
|
|
Buffer *source;
|
|
|
|
uint32_t pos;
|
|
|
|
Petal(uint32_t length, uint32_t lines, Buffer *source, uint32_t pos)
|
|
: Shard(ShardKind::Petal, length, lines), source(source), pos(pos) {};
|
|
};
|
|
|
|
std::pair<ShardPtr, ShardPtr> split(Shard *n, uint32_t offset);
|
|
|
|
void print_shard(const Shard *shard, int depth = 0);
|