Parcourir la source

Add script to import SDE; add universe loading code.

master
Ben Kurtovic il y a 7 ans
Parent
révision
462785f0fa
3 fichiers modifiés avec 246 ajouts et 10 suppressions
  1. +40
    -6
      calefaction/eve/universe.py
  2. +206
    -0
      scripts/import_sde.py
  3. +0
    -4
      scripts/read_sde.py

+ 40
- 6
calefaction/eve/universe.py Voir le fichier

@@ -1,5 +1,10 @@
# -*- coding: utf-8 -*-

import gzip
from threading import Lock

import yaml

__all__ = ["Universe"]

class _SolarSystem:
@@ -115,27 +120,56 @@ class Universe:

def __init__(self, datadir):
self._dir = datadir
self._lock = Lock()
self._loaded = False
self._systems = {}
self._constellations = {}
self._regions = {}

def _load(self):
"""Load in universe data. This can be called multiple times safely."""
if self._loaded:
return

with self._lock:
if self._loaded:
return

filename = str(self._dir / "galaxy.yml.gz")
with gzip.open(filename, "rb") as fp:
data = yaml.load(fp, Loader=yaml.CLoader)

self._systems = data["systems"]
self._constellations = data["constellations"]
self._regions = data["regions"]
self._loaded = True

def system(self, sid):
"""Return a _SolarSystem with the given ID.

If the ID is invalid, return a dummy unknown object with ID -1.
"""
...
return _DummySolarSystem(self)
self._load()
if sid not in self._systems:
return _DummySolarSystem(self)
return _SolarSystem(self, sid, self._systems[sid])

def constellation(self, cid):
"""Return a _Constellation with the given ID.

If the ID is invalid, return a dummy unknown object with ID -1.
"""
...
return _DummyConstellation(self)
self._load()
if cid not in self._constellations:
return _DummyConstellation(self)
return _Constellation(self, cid, self._constellations[cid])

def region(self, rid):
"""Return a _Region with the given ID.

If the ID is invalid, return a dummy unknown object with ID -1.
"""
...
return _DummyRegion(self)
self._load()
if rid not in self._regions:
return _DummyRegion(self)
return _Region(self, rid, self._regions[rid])

+ 206
- 0
scripts/import_sde.py Voir le fichier

@@ -0,0 +1,206 @@
#!/usr/bin/env python3
# -*- coding: utf-8 -*-

import gzip
from pathlib import Path
import shutil
import sys

import yaml

_REGION = 3
_CONSTELLATION = 4
_SOLAR_SYSTEM = 5

def _load_yaml(filename):
with filename.open("rb") as fp:
return yaml.load(fp, Loader=yaml.CLoader)

def _verify_typeids(sde_dir):
print("Verifying typeIDs... ", end="", flush=True)

filename = sde_dir / "fsd" / "typeIDs.yaml"
with filename.open("rb") as fp:
data = yaml.load(fp.read(1024 * 16), Loader=yaml.CLoader)

assert data[_REGION]["groupID"] == _REGION
assert data[_REGION]["name"]["en"] == "Region"
assert data[_CONSTELLATION]["groupID"] == _CONSTELLATION
assert data[_CONSTELLATION]["name"]["en"] == "Constellation"
assert data[_SOLAR_SYSTEM]["groupID"] == _SOLAR_SYSTEM
assert data[_SOLAR_SYSTEM]["name"]["en"] == "Solar System"

print("done.")

def _load_ids(sde_dir):
print("Loading itemIDs... ", end="", flush=True)

data = _load_yaml(sde_dir / "bsd" / "invItems.yaml")

ids = {_REGION: [], _CONSTELLATION: [], _SOLAR_SYSTEM: []}
for entry in data:
if entry["typeID"] in ids:
ids[entry["typeID"]].append(entry["itemID"])

print("done.")
return ids

def _load_names(sde_dir):
print("Loading itemNames... ", end="", flush=True)

data = _load_yaml(sde_dir / "bsd" / "invNames.yaml")

names = {}
for entry in data:
name = entry["itemName"]
assert isinstance(name, str)
names[entry["itemID"]] = name

print("done.")
return names

def _build_galaxy_skeleton(ids, names):
print("Building galaxy skeleton... ", end="", flush=True)

galaxy = {"regions": {}, "constellations": {}, "systems": {}}

d = galaxy["regions"]
for rid in ids[_REGION]:
assert isinstance(rid, int)
d[rid] = {
"name": names[rid]
}

d = galaxy["constellations"]
for cid in ids[_CONSTELLATION]:
assert isinstance(cid, int)
d[cid] = {
"name": names[cid],
"region": -1
}

d = galaxy["systems"]
for sid in ids[_SOLAR_SYSTEM]:
assert isinstance(sid, int)
d[sid] = {
"name": names[sid],
"constellation": -1, "region": -1, "security": 0.0
}

print("done.")
return galaxy

def _load_galaxy_associations(sde_dir, galaxy):
print("Loading galaxy staticdata... ", end="", flush=True)

univdir = sde_dir / "fsd" / "universe"
for base in univdir.iterdir():
if not base.is_dir():
continue

for region in base.iterdir():
if not region.is_dir():
continue
rdata = _load_yaml(region / "region.staticdata")
rid = rdata["regionID"]

assert isinstance(rid, int)
assert rid >= 0

for constellation in region.iterdir():
if not constellation.is_dir():
continue
cdata = _load_yaml(constellation / "constellation.staticdata")
cid = cdata["constellationID"]

assert isinstance(cid, int)
assert cid >= 0

galaxy["constellations"][cid]["region"] = rid

for system in constellation.iterdir():
if not system.is_dir():
continue
sdata = _load_yaml(system / "solarsystem.staticdata")
sid = sdata["solarSystemID"]
sec = sdata["security"]

assert isinstance(sid, int)
assert isinstance(sec, float)
assert sid >= 0
assert sec >= -1.0 and sec <= 1.0

galaxy["systems"][sid]["constellation"] = cid
galaxy["systems"][sid]["region"] = rid
galaxy["systems"][sid]["security"] = sec

print("done.")

for cid, constellation in galaxy["constellations"].items():
if constellation["region"] < 0:
print("[WARNING] Orphaned constellation: %d=%s" % (
cid, constellation["name"]))

for sid, system in galaxy["systems"].items():
if system["region"] < 0 or system["constellation"] < 0:
print("[WARNING] Orphaned system: %d=%s" % (sid, system["name"]))

def _dump_galaxy(out_dir, galaxy):
print("Dumping galaxy... ", end="", flush=True)

filename = out_dir / "galaxy.yml"
with filename.open("w") as fp:
fp.write(yaml.dump(galaxy, Dumper=yaml.CDumper))

print("done.")

def _compress_galaxy(out_dir):
print("Compressing galaxy... ", end="", flush=True)

fn_src = out_dir / "galaxy.yml"
fn_dst = out_dir / "galaxy.yml.gz"

with fn_src.open("rb") as f_in:
with gzip.open(str(fn_dst), "wb") as f_out:
shutil.copyfileobj(f_in, f_out)

print("done.")

def _cleanup(out_dir):
print("Cleaning up... ", end="", flush=True)

(out_dir / "galaxy.yml").unlink()

print("done.")

def import_sde(sde_dir, out_dir):
"""Import the SDE unzipped at sde_dir to out_dir."""
print("EVE Online static data import")
print("- from: %s" % sde_dir)
print("- to: %s" % out_dir)

_verify_typeids(sde_dir)
ids = _load_ids(sde_dir)
print("Counts: regions=%d, constellations=%d, systems=%d" % (
len(ids[_REGION]), len(ids[_CONSTELLATION]), len(ids[_SOLAR_SYSTEM])))
names = _load_names(sde_dir)
galaxy = _build_galaxy_skeleton(ids, names)
del ids
del names
_load_galaxy_associations(sde_dir, galaxy)
_dump_galaxy(out_dir, galaxy)
del galaxy
_compress_galaxy(out_dir)
_cleanup(out_dir)

def main():
if len(sys.argv) < 2:
print("usage: %s <sde_directory>" % sys.argv[0])
exit(1)

sde_dir = Path(sys.argv[1]).resolve()
out_dir = Path(__file__).resolve().parent.parent / "data" / "universe"
import_sde(sde_dir, out_dir)

if __name__ == "__main__":
main()

+ 0
- 4
scripts/read_sde.py Voir le fichier

@@ -1,4 +0,0 @@
#!/usr/bin/env python3
# -*- coding: utf-8 -*-

...

Chargement…
Annuler
Enregistrer