| // Owns one validated model.bin image. Slot pointers refer into bytes_ and stay | |
| // valid until the store is loaded again or destroyed. A failed Load resets the | |
| // store; callers must not use metadata or slots from an earlier model. | |
| class ModelStore { | |
| public: | |
| // Copies the caller's bytes, then accepts only the fixed format, supported | |
| // dimensions, complete ordered slot directory, aligned non-overlapping | |
| // payloads, and no trailing bytes. | |
| bool Load(const void* data, size_t size); | |
| // Returns a stable payload pointer only when both slot identity and expected | |
| // byte extent match the fixed graph's request. | |
| const void* Slot(uint32_t slot, size_t expected_size) const; | |
| uint32_t resolution() const { return resolution_; } | |
| uint32_t class_count() const { return class_count_; } | |
| uint32_t classification_width() const { return yolo26::ClassificationWidth(class_count_); } | |
| private: | |
| struct SlotRecord { | |
| uint64_t offset = 0; | |
| uint64_t size = 0; | |
| }; | |
| void Reset(); | |
| std::vector<uint8_t> bytes_; | |
| std::array<SlotRecord, yolo26::kRuntimeWeightSlotCount> slots_{}; | |
| uint32_t resolution_ = 0; | |
| uint32_t class_count_ = 0; | |
| }; | |