def init_orbital_parameter_sorted_sets(self): """ Initialize particle sets which sort the particles by their bounds at each timestep """ # Update current mask self.mask = self.tmp_mask for i,b in enumerate(self.bodies.history): two_stars = b[0:2] planetesimals = b[2:] if (self.mask[i] and (not isinstance(self.unbound[i], iterable)) ): print "Iteration ", i # iterable is defined in the header # only sort if mask says to sort and array does not exist yet unbound, central, passing, both = ps.orbital_parameter_sorted_sets(planetesimals, two_stars) (self.unbound)[i] = unbound (self.central)[i] = central (self.passing)[i] = passing (self.both)[i] = both print " **** Saving Sorted Particle Sets **** " self.save() # Also, partition 'both' self.partition_both() self.save()
def split_sort(self, start, end): for i,b in enumerate(self.bodies.history): if (start <= i <= end): two_stars = b[0:2] planetesimals = b[2:] if (self.mask[i] and (not isinstance(self.unbound[i], iterable)) ): print "Iteration ", i # iterable is defined in the header # only sort if mask says to sort and array does not exist yet unbound, central, passing, both = ps.orbital_parameter_sorted_sets(planetesimals, two_stars) (self.unbound)[i] = unbound (self.central)[i] = central (self.passing)[i] = passing (self.both)[i] = both
def split_sort(self, start, end): for i, b in enumerate(self.bodies.history): if (start <= i <= end): two_stars = b[0:2] planetesimals = b[2:] if (self.mask[i] and (not isinstance(self.unbound[i], iterable))): print "Iteration ", i # iterable is defined in the header # only sort if mask says to sort and array does not exist yet unbound, central, passing, both = ps.orbital_parameter_sorted_sets( planetesimals, two_stars) (self.unbound)[i] = unbound (self.central)[i] = central (self.passing)[i] = passing (self.both)[i] = both