| from pathlib import Path | |
| import sys,json,numpy as np | |
| ROOT=Path(__file__).resolve().parents[1];sys.path.insert(0,str(ROOT)) | |
| from model.samudrace import * | |
| c=load_config(ROOT);aa=[];oo=[];at=[];ot=[] | |
| for i,p in enumerate(c["data"]["tile_origins"]):a,o=synthetic_state(p,8,i);ta,to=synthetic_state(p,8,i,1);aa.append(a);oo.append(o);at.append(ta);ot.append(to) | |
| path=ROOT/c["data"]["path"];path.parent.mkdir(parents=True,exist_ok=True);np.savez_compressed(path,atmosphere=np.stack(aa),ocean=np.stack(oo),atmosphere_target=np.stack(at),ocean_target=np.stack(ot),origins=np.array(c["data"]["tile_origins"]),logical_shape=np.array([126,180,360]),is_complete_global=False);print(path) | |