PuGa/tools/network.py

#!/usr/bin/env python3
"""Model a multi-base chain: several plans, each on its own planet, with materials transferred
between bases instead of bought/sold twice at market. See puga/network.py for the method.

Spec (YAML): {name, bases: {key: plan-yaml-path}, transfers: [{material, src, dst, trip_cost, cargo}]}
  puga network chains/nike_deimos_lst.yaml --basis vwap30
"""
import argparse, sys
from pathlib import Path
import yaml
sys.path.insert(0, str(Path(__file__).resolve().parent.parent))
from puga import ROOT, config, market, prunplanner as pp, shipping
from puga.network import BaseNode, Transfer, combine

sys.path.insert(0, str(Path(__file__).resolve().parent))
import plan_push


def main():
    ap = argparse.ArgumentParser()
    ap.add_argument("spec", help="chain spec YAML")
    ap.add_argument("--cx", default=config.DEFAULT_CX)
    ap.add_argument("--basis", default="uni30", choices=["real", "uni30", "vwap30", "vwap7", "ask", "bid", "mid"])
    a = ap.parse_args()

    net = yaml.safe_load(Path(a.spec).read_text())
    recipes, blds = pp.recipes(), pp.buildings()
    snap = market.snapshot()

    def price(t, side="both"):
        if a.basis == "real":
            q = snap.get((t, a.cx))
            return None if not q else (q.ask if side == "buy" else (q.vwap7 or q.vwap30 or q.bid))
        if a.basis == "uni30":
            return market.uni30(snap, t) or (snap.get((t, a.cx)) or market.Quote(t, a.cx)).ask
        q = snap.get((t, a.cx))
        if not q:
            return None
        return {"vwap30": q.vwap30 or q.vwap7 or q.ask, "vwap7": q.vwap7 or q.vwap30 or q.ask, "ask": q.ask,
                "bid": q.bid, "mid": (q.ask + q.bid) / 2 if q.ask and q.bid else None}[a.basis]

    st = yaml.safe_load(config.state_path().read_text())
    faction = st.get("company", {}).get("faction")
    perm = (st.get("permits", {}).get("used", 1), st.get("permits", {}).get("total", 2))

    bases = {}
    for key, entry in net["bases"].items():
        entry = entry if isinstance(entry, dict) else {"plan": entry}
        spec = yaml.safe_load((ROOT / entry["plan"]).read_text())
        plan = plan_push.build_payload(spec, recipes, {b["building_ticker"] for b in blds})
        if not st.get("hq"):
            plan["plan_corphq"] = False
        planet = pp._g(f"/data/planet/{plan['planet_natural_id']}/", 3600)
        built = next((dict(b.get("buildings", {})) for b in st.get("bases", []) if b.get("planet") == plan["planet_natural_id"]), {})
        if entry.get("cm_free"):        # a Core Module Kit (founding) covers the CM: don't price it
            built["CM"] = 1
        bases[key] = BaseNode(key, plan, recipes, blds, planet["resources"], planet["fertility"], price, faction, perm, built)

    sf_px = (snap.get(("SF", a.cx)) or market.Quote("SF", a.cx)).ask or 0
    ff_px = (snap.get(("FF", a.cx)) or market.Quote("FF", a.cx)).ask or 0
    transfers, cost_notes = [], []
    for raw in net.get("transfers", []):
        t = dict(raw)
        if "trip_cost" not in t and t["src"] in bases and t["dst"] in bases:
            real = shipping.round_trip_cost(bases[t["src"]].plan["planet_natural_id"], bases[t["dst"]].plan["planet_natural_id"],
                                             sf_px, ff_px)
            if real is not None:
                t["trip_cost"] = real
                cost_notes.append(f"  {t['material']} {t['src']}->{t['dst']}: real round-trip cost {real:,.0f} AIC (puga.shipping)")
            else:
                cost_notes.append(f"  {t['material']} {t['src']}->{t['dst']}: no direct-jump route (gateway?) - using placeholder {Transfer.__dataclass_fields__['trip_cost'].default:,.0f} AIC, check SFC in-game")
        transfers.append(Transfer(**t))
    r = combine(bases, transfers)

    print(f"{net.get('name', a.spec)}  ({a.basis} prices)\n")
    for key, res in r["results"].items():
        line = f"[{key}]  area {res['area']:.0f}  profit/day {res['profit']:,.0f}  new capex {res['new_capex']:,.0f}"
        if res["new_capex"] > 0:
            line += f"  (payback {res['new_capex']/res['profit']:.1f} d)"
        print(line)
        for b in res["buildings"]:
            print(f"    {b['amount']:2} x {b['building']:4} eff {b['efficiency']*100:6.1f}%")

    if cost_notes:
        print("\nFREIGHT COST")
        print("\n".join(cost_notes))

    print("\nTRANSFERS")
    for t in r["ledger"]:
        note = f"SHORTFALL {t['shortfall']:.1f}/d bought at market by {t['dst']}" if t["shortfall"] > 0.01 else \
               (f"surplus {t['surplus']:.1f}/d sold at market by {t['src']}" if t["surplus"] > 0.01 else "fully matched")
        print(f"  {t['material']:5} {t['src']} -> {t['dst']}: moved {t['moved']:7.1f}/d  "
              f"({t['weight']:.0f} t, {t['volume']:.0f} m3/d, {t['trips']:.2f} trips/d)  freight {t['freight']:,.0f}/d  [{note}]")

    total_new_capex = sum(res["new_capex"] for res in r["results"].values())
    print(f"\nCOMBINED profit/day {r['total_profit']:,.0f}   new capex {total_new_capex:,.0f}"
          + (f"   payback {total_new_capex/r['total_profit']:.1f} d" if total_new_capex > 0 else ""))
    print("CAUTION: a base's 'profit/day' here is its WHOLE plan (existing buildings included), not just the new "
          "addition, so its payback here understates the marginal payback where new_capex is small relative to an "
          "existing base. For the true marginal payback, also simulate the base's plan WITHOUT the new "
          "building/transfer and diff the profit; see docs/library.md.")
    print("NOTE: trip_cost is computed from the real fuel model (puga.shipping, max sliders) for direct-jump routes "
          "between two bases in this chain (see FREIGHT COST above); a gateway route, or a transfer with an "
          "explicit trip_cost in the YAML, uses that value instead (9250 AIC if neither applies). cargo defaults "
          "to 500 t/m3 (the account's ship hold size, confirmed via FIO) - override per-transfer if a different "
          "ship is used.")


if __name__ == "__main__":
    main()