mirror of
https://github.com/PixelGuys/Cubyz.git
synced 2025-08-03 11:17:05 -04:00
Refactor common*
variables from assets.zig
into struct (#1289)
## Description This pull request moves all of the `common*` variables into `Assets` struct. As a bonus change `Managed` was special cased in `refAllDeclsRecursiveExceptCImports` to avoid infinite recursion for `*HashMap*` aliases. ## Links Resolves: #1205 Resolves: #1204 Related to: #1181
This commit is contained in:
parent
cbe400c9a4
commit
a15f1a96bc
607
src/assets.zig
607
src/assets.zig
@ -10,154 +10,285 @@ const main = @import("main");
|
||||
const biomes_zig = main.server.terrain.biomes;
|
||||
const sbb = main.server.terrain.structure_building_blocks;
|
||||
const NeverFailingAllocator = main.heap.NeverFailingAllocator;
|
||||
const NeverFailingArenaAllocator = main.heap.NeverFailingArenaAllocator;
|
||||
const ListUnmanaged = main.ListUnmanaged;
|
||||
const files = main.files;
|
||||
|
||||
var arena: main.heap.NeverFailingArenaAllocator = undefined;
|
||||
var arenaAllocator: NeverFailingAllocator = undefined;
|
||||
var commonBlocks: std.StringHashMap(ZonElement) = undefined;
|
||||
var commonBlockMigrations: std.StringHashMap(ZonElement) = undefined;
|
||||
var commonItems: std.StringHashMap(ZonElement) = undefined;
|
||||
var commonTools: std.StringHashMap(ZonElement) = undefined;
|
||||
var commonBiomes: std.StringHashMap(ZonElement) = undefined;
|
||||
var commonBiomeMigrations: std.StringHashMap(ZonElement) = undefined;
|
||||
var commonRecipes: std.StringHashMap(ZonElement) = undefined;
|
||||
var commonModels: std.StringHashMap([]const u8) = undefined;
|
||||
var commonStructureBuildingBlocks: std.StringHashMap(ZonElement) = undefined;
|
||||
var commonBlueprints: std.StringHashMap([]u8) = undefined;
|
||||
var commonAssetArena: NeverFailingArenaAllocator = undefined;
|
||||
var commonAssetAllocator: NeverFailingAllocator = undefined;
|
||||
var common: Assets = undefined;
|
||||
|
||||
pub fn init() void {
|
||||
biomes_zig.init();
|
||||
blocks_zig.init();
|
||||
pub const Assets = struct {
|
||||
pub const ZonHashMap = std.StringHashMapUnmanaged(ZonElement);
|
||||
pub const BytesHashMap = std.StringHashMapUnmanaged([]const u8);
|
||||
pub const AddonNameToZonMap = std.StringHashMapUnmanaged(ZonElement);
|
||||
|
||||
arena = .init(main.globalAllocator);
|
||||
arenaAllocator = arena.allocator();
|
||||
commonBlocks = .init(arenaAllocator.allocator);
|
||||
commonBlockMigrations = .init(arenaAllocator.allocator);
|
||||
commonItems = .init(arenaAllocator.allocator);
|
||||
commonTools = .init(arenaAllocator.allocator);
|
||||
commonBiomes = .init(arenaAllocator.allocator);
|
||||
commonBiomeMigrations = .init(arenaAllocator.allocator);
|
||||
commonRecipes = .init(arenaAllocator.allocator);
|
||||
commonModels = .init(arenaAllocator.allocator);
|
||||
commonStructureBuildingBlocks = .init(arenaAllocator.allocator);
|
||||
commonBlueprints = .init(arenaAllocator.allocator);
|
||||
blocks: ZonHashMap,
|
||||
blockMigrations: AddonNameToZonMap,
|
||||
items: ZonHashMap,
|
||||
tools: ZonHashMap,
|
||||
biomes: ZonHashMap,
|
||||
biomeMigrations: AddonNameToZonMap,
|
||||
recipes: ZonHashMap,
|
||||
models: BytesHashMap,
|
||||
structureBuildingBlocks: ZonHashMap,
|
||||
blueprints: BytesHashMap,
|
||||
|
||||
readAssets(
|
||||
arenaAllocator,
|
||||
"assets/",
|
||||
&commonBlocks,
|
||||
&commonBlockMigrations,
|
||||
&commonItems,
|
||||
&commonTools,
|
||||
&commonBiomes,
|
||||
&commonBiomeMigrations,
|
||||
&commonRecipes,
|
||||
&commonModels,
|
||||
&commonStructureBuildingBlocks,
|
||||
&commonBlueprints,
|
||||
);
|
||||
|
||||
std.log.info(
|
||||
"Finished assets init with {} blocks ({} migrations), {} items, {} tools, {} biomes ({} migrations), {} recipes, {} structure building blocks and {} blueprints",
|
||||
.{commonBlocks.count(), commonBlockMigrations.count(), commonItems.count(), commonTools.count(), commonBiomes.count(), commonBiomeMigrations.count(), commonRecipes.count(), commonStructureBuildingBlocks.count(), commonBlueprints.count()},
|
||||
);
|
||||
}
|
||||
|
||||
fn readDefaultFile(allocator: NeverFailingAllocator, dir: std.fs.Dir) !ZonElement {
|
||||
if(main.files.Dir.init(dir).readToZon(allocator, "_defaults.zig.zon")) |zon| {
|
||||
return zon;
|
||||
} else |err| {
|
||||
if(err != error.FileNotFound) return err;
|
||||
}
|
||||
|
||||
if(main.files.Dir.init(dir).readToZon(allocator, "_defaults.zon")) |zon| {
|
||||
return zon;
|
||||
} else |err| {
|
||||
if(err != error.FileNotFound) return err;
|
||||
}
|
||||
|
||||
return .null;
|
||||
}
|
||||
|
||||
/// Reads all asset `.zig.zon` files recursively from all sub folders.
|
||||
///
|
||||
/// Files red are stored in output hashmap with asset ID as key.
|
||||
/// Asset ID are constructed as `{addonName}:{relativePathNoSuffix}`.
|
||||
/// relativePathNoSuffix is always unix style path with all extensions removed.
|
||||
pub fn readAllZonFilesInAddons(
|
||||
externalAllocator: NeverFailingAllocator,
|
||||
addons: main.List(std.fs.Dir),
|
||||
addonNames: main.List([]const u8),
|
||||
subPath: []const u8,
|
||||
defaults: bool,
|
||||
output: *std.StringHashMap(ZonElement),
|
||||
migrations: ?*std.StringHashMap(ZonElement),
|
||||
) void {
|
||||
for(addons.items, addonNames.items) |addon, addonName| {
|
||||
var dir = addon.openDir(subPath, .{.iterate = true}) catch |err| {
|
||||
if(err != error.FileNotFound) {
|
||||
std.log.err("Could not open addon directory {s}: {s}", .{subPath, @errorName(err)});
|
||||
}
|
||||
continue;
|
||||
fn init() Assets {
|
||||
return .{
|
||||
.blocks = .{},
|
||||
.blockMigrations = .{},
|
||||
.items = .{},
|
||||
.tools = .{},
|
||||
.biomes = .{},
|
||||
.biomeMigrations = .{},
|
||||
.recipes = .{},
|
||||
.models = .{},
|
||||
.structureBuildingBlocks = .{},
|
||||
.blueprints = .{},
|
||||
};
|
||||
defer dir.close();
|
||||
}
|
||||
fn deinit(self: *Assets, allocator: NeverFailingAllocator) void {
|
||||
self.blocks.deinit(allocator.allocator);
|
||||
self.blockMigrations.deinit(allocator.allocator);
|
||||
self.items.deinit(allocator.allocator);
|
||||
self.tools.deinit(allocator.allocator);
|
||||
self.biomes.deinit(allocator.allocator);
|
||||
self.biomeMigrations.deinit(allocator.allocator);
|
||||
self.recipes.deinit(allocator.allocator);
|
||||
self.models.deinit(allocator.allocator);
|
||||
self.structureBuildingBlocks.deinit(allocator.allocator);
|
||||
self.blueprints.deinit(allocator.allocator);
|
||||
}
|
||||
fn clone(self: Assets, allocator: NeverFailingAllocator) Assets {
|
||||
return .{
|
||||
.blocks = self.blocks.clone(allocator.allocator) catch unreachable,
|
||||
.blockMigrations = self.blockMigrations.clone(allocator.allocator) catch unreachable,
|
||||
.items = self.items.clone(allocator.allocator) catch unreachable,
|
||||
.tools = self.tools.clone(allocator.allocator) catch unreachable,
|
||||
.biomes = self.biomes.clone(allocator.allocator) catch unreachable,
|
||||
.biomeMigrations = self.biomeMigrations.clone(allocator.allocator) catch unreachable,
|
||||
.recipes = self.recipes.clone(allocator.allocator) catch unreachable,
|
||||
.models = self.models.clone(allocator.allocator) catch unreachable,
|
||||
.structureBuildingBlocks = self.structureBuildingBlocks.clone(allocator.allocator) catch unreachable,
|
||||
.blueprints = self.blueprints.clone(allocator.allocator) catch unreachable,
|
||||
};
|
||||
}
|
||||
fn read(self: *Assets, allocator: NeverFailingAllocator, assetPath: []const u8) void {
|
||||
const addons = Addon.discoverAll(main.stackAllocator, assetPath);
|
||||
defer addons.deinit(main.stackAllocator);
|
||||
defer for(addons.items) |*addon| addon.deinit(main.stackAllocator);
|
||||
|
||||
var defaultsArena: main.heap.NeverFailingArenaAllocator = .init(main.stackAllocator);
|
||||
defer defaultsArena.deinit();
|
||||
for(addons.items) |addon| {
|
||||
addon.readAllZon(allocator, "blocks", true, &self.blocks, &self.blockMigrations);
|
||||
addon.readAllZon(allocator, "items", true, &self.items, null);
|
||||
addon.readAllZon(allocator, "tools", true, &self.tools, null);
|
||||
addon.readAllZon(allocator, "biomes", true, &self.biomes, &self.biomeMigrations);
|
||||
addon.readAllZon(allocator, "recipes", false, &self.recipes, null);
|
||||
addon.readAllZon(allocator, "sbb", true, &self.structureBuildingBlocks, null);
|
||||
addon.readAllBlueprints(allocator, &self.blueprints);
|
||||
addon.readAllModels(allocator, &self.models);
|
||||
}
|
||||
}
|
||||
fn log(self: *Assets, typ: enum {common, world}) void {
|
||||
std.log.info(
|
||||
"Finished {s} assets reading with {} blocks ({} migrations), {} items, {} tools, {} biomes ({} migrations), {} recipes, {} structure building blocks and {} blueprints",
|
||||
.{@tagName(typ), self.blocks.count(), self.blockMigrations.count(), self.items.count(), self.tools.count(), self.biomes.count(), self.biomeMigrations.count(), self.recipes.count(), self.structureBuildingBlocks.count(), self.blueprints.count()},
|
||||
);
|
||||
}
|
||||
|
||||
const defaultsArenaAllocator = defaultsArena.allocator();
|
||||
const Addon = struct {
|
||||
name: []const u8,
|
||||
dir: std.fs.Dir,
|
||||
|
||||
var defaultMap = std.StringHashMap(ZonElement).init(defaultsArenaAllocator.allocator);
|
||||
fn discoverAll(allocator: NeverFailingAllocator, path: []const u8) main.ListUnmanaged(Addon) {
|
||||
var addons: main.ListUnmanaged(Addon) = .{};
|
||||
|
||||
var walker = dir.walk(main.stackAllocator.allocator) catch unreachable;
|
||||
defer walker.deinit();
|
||||
var dir = std.fs.cwd().openDir(path, .{.iterate = true}) catch |err| {
|
||||
std.log.err("Can't open asset path {s}: {s}", .{path, @errorName(err)});
|
||||
return addons;
|
||||
};
|
||||
defer dir.close();
|
||||
|
||||
while(walker.next() catch |err| blk: {
|
||||
std.log.err("Got error while iterating addon directory {s}: {s}", .{subPath, @errorName(err)});
|
||||
break :blk null;
|
||||
}) |entry| {
|
||||
if(entry.kind == .file and
|
||||
!std.ascii.startsWithIgnoreCase(entry.basename, "_defaults") and
|
||||
std.ascii.endsWithIgnoreCase(entry.basename, ".zon") and
|
||||
!std.ascii.startsWithIgnoreCase(entry.path, "textures") and
|
||||
!std.ascii.eqlIgnoreCase(entry.basename, "_migrations.zig.zon"))
|
||||
{
|
||||
const id = createAssetStringID(externalAllocator, addonName, entry.path);
|
||||
var iterator = dir.iterate();
|
||||
while(iterator.next() catch |err| blk: {
|
||||
std.log.err("Got error while iterating over asset path {s}: {s}", .{path, @errorName(err)});
|
||||
break :blk null;
|
||||
}) |addon| {
|
||||
if(addon.kind != .directory) continue;
|
||||
|
||||
const zon = main.files.Dir.init(dir).readToZon(externalAllocator, entry.path) catch |err| {
|
||||
const directory = dir.openDir(addon.name, .{}) catch |err| {
|
||||
std.log.err("Got error while reading addon {s} from {s}: {s}", .{addon.name, path, @errorName(err)});
|
||||
continue;
|
||||
};
|
||||
addons.append(allocator, .{.name = allocator.dupe(u8, addon.name), .dir = directory});
|
||||
}
|
||||
return addons;
|
||||
}
|
||||
|
||||
fn deinit(self: *Addon, allocator: NeverFailingAllocator) void {
|
||||
self.dir.close();
|
||||
allocator.free(self.name);
|
||||
}
|
||||
|
||||
const Defaults = struct {
|
||||
localArena: NeverFailingArenaAllocator = undefined,
|
||||
localAllocator: NeverFailingAllocator = undefined,
|
||||
defaults: std.StringHashMapUnmanaged(ZonElement) = .{},
|
||||
|
||||
fn init(self: *Defaults, allocator: NeverFailingAllocator) void {
|
||||
self.localArena = .init(allocator);
|
||||
self.localAllocator = self.localArena.allocator();
|
||||
}
|
||||
|
||||
fn deinit(self: *Defaults) void {
|
||||
self.localArena.deinit();
|
||||
}
|
||||
|
||||
fn get(self: *Defaults, dir: std.fs.Dir) ZonElement {
|
||||
const path = dir.realpathAlloc(main.stackAllocator.allocator, ".") catch unreachable;
|
||||
defer main.stackAllocator.free(path);
|
||||
|
||||
const result = self.defaults.getOrPut(self.localAllocator.allocator, path) catch unreachable;
|
||||
|
||||
if(!result.found_existing) {
|
||||
result.key_ptr.* = self.localAllocator.dupe(u8, path);
|
||||
const default: ZonElement = self.read(dir) catch |err| blk: {
|
||||
std.log.err("Failed to read default file: {s}", .{@errorName(err)});
|
||||
break :blk .null;
|
||||
};
|
||||
|
||||
result.value_ptr.* = default;
|
||||
}
|
||||
|
||||
return result.value_ptr.*;
|
||||
}
|
||||
|
||||
fn read(self: *Defaults, dir: std.fs.Dir) !ZonElement {
|
||||
if(main.files.Dir.init(dir).readToZon(self.localAllocator, "_defaults.zig.zon")) |zon| {
|
||||
return zon;
|
||||
} else |err| {
|
||||
if(err != error.FileNotFound) return err;
|
||||
}
|
||||
|
||||
if(main.files.Dir.init(dir).readToZon(self.localAllocator, "_defaults.zon")) |zon| {
|
||||
return zon;
|
||||
} else |err| {
|
||||
if(err != error.FileNotFound) return err;
|
||||
}
|
||||
|
||||
return .null;
|
||||
}
|
||||
};
|
||||
|
||||
pub fn readAllZon(addon: Addon, allocator: NeverFailingAllocator, assetType: []const u8, hasDefaults: bool, output: *ZonHashMap, migrations: ?*AddonNameToZonMap) void {
|
||||
var assetsDirectory = addon.dir.openDir(assetType, .{.iterate = true}) catch |err| {
|
||||
if(err != error.FileNotFound) {
|
||||
std.log.err("Could not open addon directory {s}: {s}", .{assetType, @errorName(err)});
|
||||
}
|
||||
return;
|
||||
};
|
||||
defer assetsDirectory.close();
|
||||
|
||||
var defaultsStorage: Defaults = .{};
|
||||
defaultsStorage.init(main.stackAllocator);
|
||||
defer defaultsStorage.deinit();
|
||||
|
||||
var walker = assetsDirectory.walk(main.stackAllocator.allocator) catch unreachable;
|
||||
defer walker.deinit();
|
||||
|
||||
while(walker.next() catch |err| blk: {
|
||||
std.log.err("Got error while iterating addon directory {s}: {s}", .{assetType, @errorName(err)});
|
||||
break :blk null;
|
||||
}) |entry| {
|
||||
if(entry.kind != .file) continue;
|
||||
if(std.ascii.startsWithIgnoreCase(entry.basename, "_defaults")) continue;
|
||||
if(!std.ascii.endsWithIgnoreCase(entry.basename, ".zon")) continue;
|
||||
if(std.ascii.startsWithIgnoreCase(entry.path, "textures")) continue;
|
||||
if(std.ascii.eqlIgnoreCase(entry.basename, "_migrations.zig.zon")) continue;
|
||||
|
||||
const id = createAssetStringID(allocator, addon.name, entry.path);
|
||||
|
||||
const zon = files.Dir.init(assetsDirectory).readToZon(allocator, entry.path) catch |err| {
|
||||
std.log.err("Could not open {s}/{s}: {s}", .{assetType, entry.path, @errorName(err)});
|
||||
continue;
|
||||
};
|
||||
if(hasDefaults) {
|
||||
zon.join(defaultsStorage.get(entry.dir));
|
||||
}
|
||||
output.put(allocator.allocator, id, zon) catch unreachable;
|
||||
}
|
||||
if(migrations != null) blk: {
|
||||
const zon = files.Dir.init(assetsDirectory).readToZon(allocator, "_migrations.zig.zon") catch |err| {
|
||||
if(err != error.FileNotFound) std.log.err("Cannot read {s} migration file for addon {s}", .{assetType, addon.name});
|
||||
break :blk;
|
||||
};
|
||||
migrations.?.put(allocator.allocator, allocator.dupe(u8, addon.name), zon) catch unreachable;
|
||||
}
|
||||
}
|
||||
|
||||
pub fn readAllBlueprints(addon: Addon, allocator: NeverFailingAllocator, output: *BytesHashMap) void {
|
||||
const subPath = "blueprints";
|
||||
var assetsDirectory = addon.dir.openDir(subPath, .{.iterate = true}) catch |err| {
|
||||
if(err != error.FileNotFound) {
|
||||
std.log.err("Could not open addon directory {s}: {s}", .{subPath, @errorName(err)});
|
||||
}
|
||||
return;
|
||||
};
|
||||
defer assetsDirectory.close();
|
||||
|
||||
var walker = assetsDirectory.walk(main.stackAllocator.allocator) catch unreachable;
|
||||
defer walker.deinit();
|
||||
|
||||
while(walker.next() catch |err| blk: {
|
||||
std.log.err("Got error while iterating addon directory {s}: {s}", .{subPath, @errorName(err)});
|
||||
break :blk null;
|
||||
}) |entry| {
|
||||
if(entry.kind != .file) continue;
|
||||
if(std.ascii.startsWithIgnoreCase(entry.basename, "_defaults")) continue;
|
||||
if(!std.ascii.endsWithIgnoreCase(entry.basename, ".blp")) continue;
|
||||
if(std.ascii.startsWithIgnoreCase(entry.basename, "_migrations")) continue;
|
||||
|
||||
const id = createAssetStringID(allocator, addon.name, entry.path);
|
||||
|
||||
const data = files.Dir.init(assetsDirectory).read(allocator, entry.path) catch |err| {
|
||||
std.log.err("Could not open {s}/{s}: {s}", .{subPath, entry.path, @errorName(err)});
|
||||
continue;
|
||||
};
|
||||
|
||||
if(defaults) {
|
||||
const path = entry.dir.realpathAlloc(main.stackAllocator.allocator, ".") catch unreachable;
|
||||
defer main.stackAllocator.free(path);
|
||||
|
||||
const result = defaultMap.getOrPut(path) catch unreachable;
|
||||
|
||||
if(!result.found_existing) {
|
||||
result.key_ptr.* = defaultsArenaAllocator.dupe(u8, path);
|
||||
const default: ZonElement = readDefaultFile(defaultsArenaAllocator, entry.dir) catch |err| blk: {
|
||||
std.log.err("Failed to read default file: {s}", .{@errorName(err)});
|
||||
break :blk .null;
|
||||
};
|
||||
|
||||
result.value_ptr.* = default;
|
||||
}
|
||||
|
||||
zon.join(result.value_ptr.*);
|
||||
}
|
||||
output.put(id, zon) catch unreachable;
|
||||
output.put(allocator.allocator, id, data) catch unreachable;
|
||||
}
|
||||
}
|
||||
if(migrations != null) blk: {
|
||||
const zon = main.files.Dir.init(dir).readToZon(externalAllocator, "_migrations.zig.zon") catch |err| {
|
||||
if(err != error.FileNotFound) std.log.err("Cannot read {s} migration file for addon {s}", .{subPath, addonName});
|
||||
break :blk;
|
||||
|
||||
pub fn readAllModels(addon: Addon, allocator: NeverFailingAllocator, output: *BytesHashMap) void {
|
||||
const subPath = "models";
|
||||
var assetsDirectory = addon.dir.openDir(subPath, .{.iterate = true}) catch |err| {
|
||||
if(err != error.FileNotFound) {
|
||||
std.log.err("Could not open addon directory {s}: {s}", .{subPath, @errorName(err)});
|
||||
}
|
||||
return;
|
||||
};
|
||||
migrations.?.put(externalAllocator.dupe(u8, addonName), zon) catch unreachable;
|
||||
defer assetsDirectory.close();
|
||||
var walker = assetsDirectory.walk(main.stackAllocator.allocator) catch unreachable;
|
||||
defer walker.deinit();
|
||||
|
||||
while(walker.next() catch |err| blk: {
|
||||
std.log.err("Got error while iterating addon directory {s}: {s}", .{subPath, @errorName(err)});
|
||||
break :blk null;
|
||||
}) |entry| {
|
||||
if(entry.kind != .file) continue;
|
||||
if(!std.ascii.endsWithIgnoreCase(entry.basename, ".obj")) continue;
|
||||
|
||||
const id = createAssetStringID(allocator, addon.name, entry.path);
|
||||
|
||||
const string = assetsDirectory.readFileAlloc(allocator.allocator, entry.path, std.math.maxInt(usize)) catch |err| {
|
||||
std.log.err("Could not open {s}/{s}: {s}", .{subPath, entry.path, @errorName(err)});
|
||||
continue;
|
||||
};
|
||||
output.put(allocator.allocator, id, string) catch unreachable;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
};
|
||||
};
|
||||
|
||||
fn createAssetStringID(
|
||||
externalAllocator: NeverFailingAllocator,
|
||||
@ -184,133 +315,16 @@ fn createAssetStringID(
|
||||
return assetId;
|
||||
}
|
||||
|
||||
pub fn readAllBlueprintFilesInAddons(
|
||||
externalAllocator: NeverFailingAllocator,
|
||||
addons: main.List(std.fs.Dir),
|
||||
addonNames: main.List([]const u8),
|
||||
subPath: []const u8,
|
||||
output: *std.StringHashMap([]u8),
|
||||
) void {
|
||||
for(addons.items, addonNames.items) |addon, addonName| {
|
||||
var dir = addon.openDir(subPath, .{.iterate = true}) catch |err| {
|
||||
if(err != error.FileNotFound) {
|
||||
std.log.err("Could not open addon directory {s}: {s}", .{subPath, @errorName(err)});
|
||||
}
|
||||
continue;
|
||||
};
|
||||
defer dir.close();
|
||||
pub fn init() void {
|
||||
biomes_zig.init();
|
||||
blocks_zig.init();
|
||||
|
||||
var walker = dir.walk(main.stackAllocator.allocator) catch unreachable;
|
||||
defer walker.deinit();
|
||||
commonAssetArena = .init(main.globalAllocator);
|
||||
commonAssetAllocator = commonAssetArena.allocator();
|
||||
|
||||
while(walker.next() catch |err| blk: {
|
||||
std.log.err("Got error while iterating addon directory {s}: {s}", .{subPath, @errorName(err)});
|
||||
break :blk null;
|
||||
}) |entry| {
|
||||
if(entry.kind != .file) continue;
|
||||
if(std.ascii.startsWithIgnoreCase(entry.basename, "_defaults")) continue;
|
||||
if(!std.ascii.endsWithIgnoreCase(entry.basename, ".blp")) continue;
|
||||
if(std.ascii.startsWithIgnoreCase(entry.basename, "_migrations")) continue;
|
||||
|
||||
const stringId: []u8 = createAssetStringID(externalAllocator, addonName, entry.path);
|
||||
const data = main.files.Dir.init(dir).read(externalAllocator, entry.path) catch |err| {
|
||||
std.log.err("Could not open {s}/{s}: {s}", .{subPath, entry.path, @errorName(err)});
|
||||
continue;
|
||||
};
|
||||
output.put(stringId, data) catch unreachable;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/// Reads obj files recursively from all subfolders.
|
||||
pub fn readAllObjFilesInAddonsHashmap(
|
||||
externalAllocator: NeverFailingAllocator,
|
||||
addons: main.List(std.fs.Dir),
|
||||
addonNames: main.List([]const u8),
|
||||
subPath: []const u8,
|
||||
output: *std.StringHashMap([]const u8),
|
||||
) void {
|
||||
for(addons.items, addonNames.items) |addon, addonName| {
|
||||
var dir = addon.openDir(subPath, .{.iterate = true}) catch |err| {
|
||||
if(err != error.FileNotFound) {
|
||||
std.log.err("Could not open addon directory {s}: {s}", .{subPath, @errorName(err)});
|
||||
}
|
||||
continue;
|
||||
};
|
||||
defer dir.close();
|
||||
|
||||
var walker = dir.walk(main.stackAllocator.allocator) catch unreachable;
|
||||
defer walker.deinit();
|
||||
|
||||
while(walker.next() catch |err| blk: {
|
||||
std.log.err("Got error while iterating addon directory {s}: {s}", .{subPath, @errorName(err)});
|
||||
break :blk null;
|
||||
}) |entry| {
|
||||
if(entry.kind == .file and std.ascii.endsWithIgnoreCase(entry.basename, ".obj")) {
|
||||
const id: []u8 = createAssetStringID(externalAllocator, addonName, entry.path);
|
||||
|
||||
const string = dir.readFileAlloc(externalAllocator.allocator, entry.path, std.math.maxInt(usize)) catch |err| {
|
||||
std.log.err("Could not open {s}/{s}: {s}", .{subPath, entry.path, @errorName(err)});
|
||||
continue;
|
||||
};
|
||||
output.put(id, string) catch unreachable;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
pub fn readAssets(
|
||||
externalAllocator: NeverFailingAllocator,
|
||||
assetPath: []const u8,
|
||||
blocks: *std.StringHashMap(ZonElement),
|
||||
blockMigrations: *std.StringHashMap(ZonElement),
|
||||
items: *std.StringHashMap(ZonElement),
|
||||
tools: *std.StringHashMap(ZonElement),
|
||||
biomes: *std.StringHashMap(ZonElement),
|
||||
biomeMigrations: *std.StringHashMap(ZonElement),
|
||||
recipes: *std.StringHashMap(ZonElement),
|
||||
models: *std.StringHashMap([]const u8),
|
||||
structureBuildingBlocks: *std.StringHashMap(ZonElement),
|
||||
blueprints: *std.StringHashMap([]u8),
|
||||
) void {
|
||||
var addons = main.List(std.fs.Dir).init(main.stackAllocator);
|
||||
defer addons.deinit();
|
||||
var addonNames = main.List([]const u8).init(main.stackAllocator);
|
||||
defer addonNames.deinit();
|
||||
|
||||
{ // Find all the sub-directories to the assets folder.
|
||||
var dir = std.fs.cwd().openDir(assetPath, .{.iterate = true}) catch |err| {
|
||||
std.log.err("Can't open asset path {s}: {s}", .{assetPath, @errorName(err)});
|
||||
return;
|
||||
};
|
||||
defer dir.close();
|
||||
var iterator = dir.iterate();
|
||||
while(iterator.next() catch |err| blk: {
|
||||
std.log.err("Got error while iterating over asset path {s}: {s}", .{assetPath, @errorName(err)});
|
||||
break :blk null;
|
||||
}) |addon| {
|
||||
if(addon.kind == .directory) {
|
||||
addons.append(dir.openDir(addon.name, .{}) catch |err| {
|
||||
std.log.err("Got error while reading addon {s} from {s}: {s}", .{addon.name, assetPath, @errorName(err)});
|
||||
continue;
|
||||
});
|
||||
addonNames.append(main.stackAllocator.dupe(u8, addon.name));
|
||||
}
|
||||
}
|
||||
}
|
||||
defer for(addons.items, addonNames.items) |*dir, addonName| {
|
||||
dir.close();
|
||||
main.stackAllocator.free(addonName);
|
||||
};
|
||||
|
||||
readAllZonFilesInAddons(externalAllocator, addons, addonNames, "blocks", true, blocks, blockMigrations);
|
||||
readAllZonFilesInAddons(externalAllocator, addons, addonNames, "items", true, items, null);
|
||||
readAllZonFilesInAddons(externalAllocator, addons, addonNames, "tools", true, tools, null);
|
||||
readAllZonFilesInAddons(externalAllocator, addons, addonNames, "biomes", true, biomes, biomeMigrations);
|
||||
readAllZonFilesInAddons(externalAllocator, addons, addonNames, "recipes", false, recipes, null);
|
||||
readAllObjFilesInAddonsHashmap(externalAllocator, addons, addonNames, "models", models);
|
||||
readAllZonFilesInAddons(externalAllocator, addons, addonNames, "sbb", true, structureBuildingBlocks, null);
|
||||
readAllBlueprintFilesInAddons(externalAllocator, addons, addonNames, "blueprints", blueprints);
|
||||
common = .init();
|
||||
common.read(commonAssetAllocator, "assets/");
|
||||
common.log(.common);
|
||||
}
|
||||
|
||||
fn registerItem(assetFolder: []const u8, id: []const u8, zon: ZonElement) !void {
|
||||
@ -463,51 +477,23 @@ pub fn loadWorldAssets(assetFolder: []const u8, blockPalette: *Palette, itemPale
|
||||
if(loadedAssets) return; // The assets already got loaded by the server.
|
||||
loadedAssets = true;
|
||||
|
||||
var blocks = commonBlocks.cloneWithAllocator(main.stackAllocator.allocator) catch unreachable;
|
||||
defer blocks.clearAndFree();
|
||||
var blockMigrations = commonBlockMigrations.cloneWithAllocator(main.stackAllocator.allocator) catch unreachable;
|
||||
defer blockMigrations.clearAndFree();
|
||||
var items = commonItems.cloneWithAllocator(main.stackAllocator.allocator) catch unreachable;
|
||||
defer items.clearAndFree();
|
||||
var tools = commonTools.cloneWithAllocator(main.stackAllocator.allocator) catch unreachable;
|
||||
defer tools.clearAndFree();
|
||||
var biomes = commonBiomes.cloneWithAllocator(main.stackAllocator.allocator) catch unreachable;
|
||||
defer biomes.clearAndFree();
|
||||
var biomeMigrations = commonBiomeMigrations.cloneWithAllocator(main.stackAllocator.allocator) catch unreachable;
|
||||
defer biomeMigrations.clearAndFree();
|
||||
var recipes = commonRecipes.cloneWithAllocator(main.stackAllocator.allocator) catch unreachable;
|
||||
defer recipes.clearAndFree();
|
||||
var models = commonModels.cloneWithAllocator(main.stackAllocator.allocator) catch unreachable;
|
||||
defer models.clearAndFree();
|
||||
var structureBuildingBlocks = commonStructureBuildingBlocks.cloneWithAllocator(main.stackAllocator.allocator) catch unreachable;
|
||||
defer structureBuildingBlocks.clearAndFree();
|
||||
var blueprints = commonBlueprints.cloneWithAllocator(main.stackAllocator.allocator) catch unreachable;
|
||||
defer blueprints.clearAndFree();
|
||||
var worldArena: NeverFailingArenaAllocator = .init(main.stackAllocator);
|
||||
defer worldArena.deinit();
|
||||
const worldAllocator = worldArena.allocator();
|
||||
|
||||
var worldAssets = common.clone(worldAllocator);
|
||||
worldAssets.read(worldAllocator, assetFolder);
|
||||
|
||||
readAssets(
|
||||
arenaAllocator,
|
||||
assetFolder,
|
||||
&blocks,
|
||||
&blockMigrations,
|
||||
&items,
|
||||
&tools,
|
||||
&biomes,
|
||||
&biomeMigrations,
|
||||
&recipes,
|
||||
&models,
|
||||
&structureBuildingBlocks,
|
||||
&blueprints,
|
||||
);
|
||||
errdefer unloadAssets();
|
||||
|
||||
migrations_zig.registerAll(.block, &blockMigrations);
|
||||
migrations_zig.registerAll(.block, &worldAssets.blockMigrations);
|
||||
migrations_zig.apply(.block, blockPalette);
|
||||
|
||||
migrations_zig.registerAll(.biome, &biomeMigrations);
|
||||
migrations_zig.registerAll(.biome, &worldAssets.biomeMigrations);
|
||||
migrations_zig.apply(.biome, biomePalette);
|
||||
|
||||
// models:
|
||||
var modelIterator = models.iterator();
|
||||
var modelIterator = worldAssets.models.iterator();
|
||||
while(modelIterator.next()) |entry| {
|
||||
_ = main.models.registerModel(entry.key_ptr.*, entry.value_ptr.*);
|
||||
}
|
||||
@ -517,11 +503,11 @@ pub fn loadWorldAssets(assetFolder: []const u8, blockPalette: *Palette, itemPale
|
||||
// Blocks:
|
||||
// First blocks from the palette to enforce ID values.
|
||||
for(blockPalette.palette.items) |stringId| {
|
||||
try registerBlock(assetFolder, stringId, blocks.get(stringId) orelse .null);
|
||||
try registerBlock(assetFolder, stringId, worldAssets.blocks.get(stringId) orelse .null);
|
||||
}
|
||||
|
||||
// Then all the blocks that were missing in palette but are present in the game.
|
||||
var iterator = blocks.iterator();
|
||||
var iterator = worldAssets.blocks.iterator();
|
||||
while(iterator.next()) |entry| {
|
||||
const stringId = entry.key_ptr.*;
|
||||
const zon = entry.value_ptr.*;
|
||||
@ -538,16 +524,16 @@ pub fn loadWorldAssets(assetFolder: []const u8, blockPalette: *Palette, itemPale
|
||||
std.debug.assert(!items_zig.hasRegistered(stringId));
|
||||
|
||||
// Some items are created automatically from blocks.
|
||||
if(blocks.get(stringId)) |zon| {
|
||||
if(worldAssets.blocks.get(stringId)) |zon| {
|
||||
if(!zon.get(bool, "hasItem", true)) continue;
|
||||
try registerItem(assetFolder, stringId, zon.getChild("item"));
|
||||
if(items.get(stringId) != null) {
|
||||
if(worldAssets.items.get(stringId) != null) {
|
||||
std.log.err("Item {s} appears as standalone item and as block item.", .{stringId});
|
||||
}
|
||||
continue;
|
||||
}
|
||||
// Items not related to blocks should appear in items hash map.
|
||||
if(items.get(stringId)) |zon| {
|
||||
if(worldAssets.items.get(stringId)) |zon| {
|
||||
try registerItem(assetFolder, stringId, zon);
|
||||
continue;
|
||||
}
|
||||
@ -557,7 +543,7 @@ pub fn loadWorldAssets(assetFolder: []const u8, blockPalette: *Palette, itemPale
|
||||
|
||||
// Then missing block-items to keep backwards compatibility of ID order.
|
||||
for(blockPalette.palette.items) |stringId| {
|
||||
const zon = blocks.get(stringId) orelse .null;
|
||||
const zon = worldAssets.blocks.get(stringId) orelse .null;
|
||||
|
||||
if(!zon.get(bool, "hasItem", true)) continue;
|
||||
if(items_zig.hasRegistered(stringId)) continue;
|
||||
@ -567,7 +553,7 @@ pub fn loadWorldAssets(assetFolder: []const u8, blockPalette: *Palette, itemPale
|
||||
}
|
||||
|
||||
// And finally normal items.
|
||||
iterator = items.iterator();
|
||||
iterator = worldAssets.items.iterator();
|
||||
while(iterator.next()) |entry| {
|
||||
const stringId = entry.key_ptr.*;
|
||||
const zon = entry.value_ptr.*;
|
||||
@ -581,7 +567,7 @@ pub fn loadWorldAssets(assetFolder: []const u8, blockPalette: *Palette, itemPale
|
||||
|
||||
// After we have registered all items and all blocks, we can assign block references to those that come from blocks.
|
||||
for(blockPalette.palette.items) |stringId| {
|
||||
const zon = blocks.get(stringId) orelse .null;
|
||||
const zon = worldAssets.blocks.get(stringId) orelse .null;
|
||||
|
||||
if(!zon.get(bool, "hasItem", true)) continue;
|
||||
std.debug.assert(items_zig.hasRegistered(stringId));
|
||||
@ -590,29 +576,29 @@ pub fn loadWorldAssets(assetFolder: []const u8, blockPalette: *Palette, itemPale
|
||||
}
|
||||
|
||||
// tools:
|
||||
iterator = tools.iterator();
|
||||
iterator = worldAssets.tools.iterator();
|
||||
while(iterator.next()) |entry| {
|
||||
registerTool(assetFolder, entry.key_ptr.*, entry.value_ptr.*);
|
||||
}
|
||||
|
||||
// block drops:
|
||||
blocks_zig.finishBlocks(blocks);
|
||||
blocks_zig.finishBlocks(worldAssets.blocks);
|
||||
|
||||
iterator = recipes.iterator();
|
||||
iterator = worldAssets.recipes.iterator();
|
||||
while(iterator.next()) |entry| {
|
||||
registerRecipesFromZon(entry.value_ptr.*);
|
||||
}
|
||||
|
||||
try sbb.registerBlueprints(&blueprints);
|
||||
try sbb.registerSBB(&structureBuildingBlocks);
|
||||
try sbb.registerBlueprints(&worldAssets.blueprints);
|
||||
try sbb.registerSBB(&worldAssets.structureBuildingBlocks);
|
||||
|
||||
// Biomes:
|
||||
var nextBiomeNumericId: u32 = 0;
|
||||
for(biomePalette.palette.items) |id| {
|
||||
registerBiome(nextBiomeNumericId, id, biomes.get(id) orelse .null);
|
||||
registerBiome(nextBiomeNumericId, id, worldAssets.biomes.get(id) orelse .null);
|
||||
nextBiomeNumericId += 1;
|
||||
}
|
||||
iterator = biomes.iterator();
|
||||
iterator = worldAssets.biomes.iterator();
|
||||
while(iterator.next()) |entry| {
|
||||
if(biomes_zig.hasRegistered(entry.key_ptr.*)) continue;
|
||||
registerBiome(nextBiomeNumericId, entry.key_ptr.*, entry.value_ptr.*);
|
||||
@ -640,10 +626,7 @@ pub fn loadWorldAssets(assetFolder: []const u8, blockPalette: *Palette, itemPale
|
||||
}
|
||||
}
|
||||
|
||||
std.log.info(
|
||||
"Finished registering assets with {} blocks ({} migrations), {} items {} tools, {} biomes ({} migrations), {} recipes and {} models",
|
||||
.{blocks.count(), blockMigrations.count(), items.count(), tools.count(), biomes.count(), biomeMigrations.count(), recipes.count(), models.count()},
|
||||
);
|
||||
worldAssets.log(.world);
|
||||
}
|
||||
|
||||
pub fn unloadAssets() void { // MARK: unloadAssets()
|
||||
@ -680,7 +663,7 @@ pub fn unloadAssets() void { // MARK: unloadAssets()
|
||||
}
|
||||
|
||||
pub fn deinit() void {
|
||||
arena.deinit();
|
||||
commonAssetArena.deinit();
|
||||
biomes_zig.deinit();
|
||||
blocks_zig.deinit();
|
||||
migrations_zig.deinit();
|
||||
|
@ -23,6 +23,7 @@ const block_entity = @import("block_entity.zig");
|
||||
const BlockEntityType = block_entity.BlockEntityType;
|
||||
const sbb = main.server.terrain.structure_building_blocks;
|
||||
const blueprint = main.blueprint;
|
||||
const Assets = main.assets.Assets;
|
||||
|
||||
var arena = main.heap.NeverFailingArenaAllocator.init(main.globalAllocator);
|
||||
const allocator = arena.allocator();
|
||||
@ -212,7 +213,7 @@ fn registerOpaqueVariant(typ: u16, zon: ZonElement) void {
|
||||
}
|
||||
}
|
||||
|
||||
pub fn finishBlocks(zonElements: std.StringHashMap(ZonElement)) void {
|
||||
pub fn finishBlocks(zonElements: Assets.ZonHashMap) void {
|
||||
var i: u16 = 0;
|
||||
while(i < size) : (i += 1) {
|
||||
registerBlockDrop(i, zonElements.get(_id[i]) orelse continue);
|
||||
|
@ -170,7 +170,7 @@ pub const Blueprint = struct {
|
||||
}
|
||||
}
|
||||
}
|
||||
pub fn load(allocator: NeverFailingAllocator, inputBuffer: []u8) !Blueprint {
|
||||
pub fn load(allocator: NeverFailingAllocator, inputBuffer: []const u8) !Blueprint {
|
||||
var compressedReader = BinaryReader.init(inputBuffer);
|
||||
const version = try compressedReader.readInt(u16);
|
||||
|
||||
|
@ -739,6 +739,8 @@ pub fn refAllDeclsRecursiveExceptCImports(comptime T: type) void {
|
||||
if(comptime std.mem.eql(u8, decl.name, "c")) continue;
|
||||
if(comptime std.mem.eql(u8, decl.name, "hbft")) break :blk;
|
||||
if(comptime std.mem.eql(u8, decl.name, "stb_image")) break :blk;
|
||||
// TODO: Remove this after Zig removes Managed hashmap PixelGuys/Cubyz#308
|
||||
if(comptime std.mem.eql(u8, decl.name, "Managed")) continue;
|
||||
if(@TypeOf(@field(T, decl.name)) == type) {
|
||||
switch(@typeInfo(@field(T, decl.name))) {
|
||||
.@"struct", .@"enum", .@"union", .@"opaque" => refAllDeclsRecursiveExceptCImports(@field(T, decl.name)),
|
||||
|
@ -3,6 +3,7 @@ const std = @import("std");
|
||||
const main = @import("main");
|
||||
const ZonElement = @import("zon.zig").ZonElement;
|
||||
const Palette = @import("assets.zig").Palette;
|
||||
const Assets = main.assets.Assets;
|
||||
|
||||
var arenaAllocator = main.heap.NeverFailingArenaAllocator.init(main.globalAllocator);
|
||||
const migrationAllocator = arenaAllocator.allocator();
|
||||
@ -15,7 +16,7 @@ const MigrationType = enum {
|
||||
biome,
|
||||
};
|
||||
|
||||
pub fn registerAll(comptime typ: MigrationType, migrations: *std.StringHashMap(ZonElement)) void {
|
||||
pub fn registerAll(comptime typ: MigrationType, migrations: *Assets.AddonNameToZonMap) void {
|
||||
std.log.info("Registering {} {s} migrations", .{migrations.count(), @tagName(typ)});
|
||||
const collection = switch(typ) {
|
||||
.block => &blockMigrations,
|
||||
|
@ -11,6 +11,7 @@ const Neighbor = main.chunk.Neighbor;
|
||||
const Block = main.blocks.Block;
|
||||
const Degrees = main.rotation.Degrees;
|
||||
const NeverFailingAllocator = main.heap.NeverFailingAllocator;
|
||||
const Assets = main.assets.Assets;
|
||||
|
||||
var arena = main.heap.NeverFailingArenaAllocator.init(main.globalAllocator);
|
||||
const arenaAllocator = arena.allocator();
|
||||
@ -175,7 +176,7 @@ const Child = struct {
|
||||
}
|
||||
};
|
||||
|
||||
pub fn registerSBB(structures: *std.StringHashMap(ZonElement)) !void {
|
||||
pub fn registerSBB(structures: *Assets.ZonHashMap) !void {
|
||||
std.debug.assert(structureCache.capacity() == 0);
|
||||
structureCache.ensureTotalCapacity(arenaAllocator.allocator, structures.count()) catch unreachable;
|
||||
childrenToResolve = .init(main.stackAllocator);
|
||||
@ -217,7 +218,7 @@ pub fn registerChildBlock(numericId: u16, stringId: []const u8) void {
|
||||
childBlockStringId.append(arenaAllocator, arenaAllocator.dupe(u8, colorName));
|
||||
}
|
||||
|
||||
pub fn registerBlueprints(blueprints: *std.StringHashMap([]u8)) !void {
|
||||
pub fn registerBlueprints(blueprints: *Assets.BytesHashMap) !void {
|
||||
std.debug.assert(blueprintCache.capacity() == 0);
|
||||
|
||||
originBlockNumericId = main.blocks.parseBlock(originBlockStringId).typ;
|
||||
|
Loading…
x
Reference in New Issue
Block a user