from transfer.spreadmetric import SpreadMetricCalculator
from transfer.frontends.swiftvr import SWIFTSnapshotData
from transfer.holder import SimulationData
import numpy
initial_snap = SWIFTSnapshotData('/cosma6/data/dp004/dc-borr1/swift-test-data/eagle_0000.hdf5', halo_filename = None)
final_snap = SWIFTSnapshotData('/cosma6/data/dp004/dc-borr1/swift-test-data/eagle_0037.hdf5',
halo_filename = '/cosma6/data/dp004/dc-borr1/swift-test-data/halo_0037/halo_0037.properties')
data = SimulationData(initial_snap, final_snap)
cal = SpreadMetricCalculator(simulation = data)
cal.find_neighbour_distances()
dm_spread = cal.dark_matter_spread
gas_spread = cal.gas_spread
star_spread = cal.star_spread
numpy.savetxt('/cosma5/data/durham/dc-murr1/dm_spread.txt', dm_spread)
numpy.savetxt('/cosma5/data/durham/dc-murr1/gas_spread.txt', gas_spread)
numpy.savetxt('/cosma5/data/durham/dc-murr1/star_spread.txt', star_spread)
final_halo_filename = (
"/cosma6/data/dp004/dc-borr1/simba-test-data/snap151Rpep..z0.000.AHF_halos"
)
initial = SIMBASnapshotData(initial_filename, initial_halo_filename)
truncate_ids = {
0: initial.gas.particle_ids.max() + 1,
1: None,
4: initial.gas.particle_ids.max() + 1,
}
final = SIMBASnapshotData(final_filename,
final_halo_filename,
truncate_ids=truncate_ids)
cross = SimulationData(initial_snapshot=initial, final_snapshot=final)
for particle_type in ["dark_matter", "stars", "gas"]:
x = {
n: getattr(getattr(cross, f"{particle_type}_transfer"), n)
for n in [
"in_halo",
"in_halo_from_own_lr",
"in_halo_from_other_lr",
"in_halo_from_outside_lr",
"in_lr",
"in_other_halo_from_lr",
"outside_haloes",
]
}
with open(f"{particle_type}.pickle", "wb") as handle: