コード例 #1
0
ファイル: cencen.py プロジェクト: cejgrove/FastHodFitting
x = np.append(x, x_t)
y = np.append(y, y_t)
z = np.append(z, z_t)
Mvir = np.append(Mvir, Mvir_t)

import multiprocessing

num_threads = multiprocessing.cpu_count()
print("CPUS total ", num_threads)
print("total particles ", len(x))

start_time = time.time()
samples_test = fasthod.mass_mask(x, y, z, Mvir, mass_bin_edges)

samples_test = fasthod.subsample(samples_test, subsample_array)

end_time_1 = time.time()
print('starting pair counting')
npairs_test = fasthod.create_npairs_corrfunc(samples_test, samples_test,
                                             r_bin_edges, boxsize, num_threads)
npairs_mass_r_bins_test = fasthod.npairs_conversion(samples_test, samples_test,
                                                    npairs_test, r_bin_edges)
print('pair counting done')
end_time_2 = time.time()
np.save(run_label + "_cencen.npy", npairs_mass_r_bins_test)

print(run_label)
print(path)
print("time to split into mass bins: ", end_time_1 - start_time)
print("total time to run code including paircounting: ",
コード例 #2
0
x_sat = x_sat[::num_sat_parts]
y_sat = y_sat[::num_sat_parts]
z_sat = z_sat[::num_sat_parts]
Mvir_sat = Mvir_sat[::num_sat_parts]

import multiprocessing

num_threads = multiprocessing.cpu_count()
print("CPUS total ", num_threads)
print("total particles ", len(x))

start_time = time.time()

samples_sat = fasthod.mass_mask(x_sat, y_sat, z_sat, Mvir_sat, mass_bin_edges)

samples_sat = fasthod.subsample(samples_sat, subsample_array)

end_time_1 = time.time()
print('starting pair counting')
npairs_test = fasthod.create_npairs_corrfunc(samples_sat, samples_sat,
                                             r_bin_edges, boxsize, num_threads)
npairs_mass_r_bins_test = fasthod.npairs_conversion(samples_sat, samples_sat,
                                                    npairs_test, r_bin_edges)
print('pair counting done')
end_time_2 = time.time()
np.save(run_label + "_satsat.npy", npairs_mass_r_bins_test)

print(run_label)
print(path)
print("time to split into mass bins: ", end_time_1 - start_time)
print("total time to run code including paircounting: ",