mirror/monofonIC
1
0
Fork 0
mirror of https://github.com/cosmo-sims/monofonIC.git synced 2024-09-19 17:03:45 +02:00
Code Releases Activity

new particle IDs encoding Lagrangian coordinates

Browse source
This commit is contained in:
Oliver Hahn 2019-10-09 17:28:00 +02:00
parent 234dd539dc
commit b8acbf842d
1 changed files with 14 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
src/ic_generator.cc
View file

@ -436,20 +436,24 @@ int Run( ConfigFile& the_config )
if( the_output_plugin->write_species_as( this_species ) == output_type::particles )
{
// if particles occupy a bcc lattice, then there are 2 x N^3 of them...
if( initial_bcc_lattice )
particles.allocate( 2*num_p_in_load );
else
particles.allocate( num_p_in_load );
// generate particle IDs
auto ipcount0 = (initial_bcc_lattice? 2:1) * particles.get_local_offset();
if( !initial_bcc_lattice ){
particles.allocate( num_p_in_load );
for( size_t i=0,ipcount=0; i<tmp.size(0); ++i ){
for( size_t j=0; j<tmp.size(1); ++j){
for( size_t k=0; k<tmp.size(2); ++k){
particles.set_id( ipcount++, tmp.get_cell_idx_1d(i,j,k) );
}
}
}
}else{
// if particles occupy a bcc lattice, then there are 2 x N^3 of them...
particles.allocate( 2*num_p_in_load );
for( size_t i=0,ipcount=0; i<tmp.size(0); ++i ){
for( size_t j=0; j<tmp.size(1); ++j){
for( size_t k=0; k<tmp.size(2); ++k){
particles.set_id( ipcount, ipcount+ipcount0 );
++ipcount;
if( initial_bcc_lattice ){
particles.set_id( ipcount, ipcount+ipcount0 );
particles.set_id( ipcount, 2*tmp.get_cell_idx_1d(i,j,k) );
particles.set_id( ipcount+num_p_in_load, 2*tmp.get_cell_idx_1d(i,j,k)+1 );
++ipcount;
}
}