read = read_input.ReadInput(input_file) # Set some variables for the simulation n_steps = read.n_steps n_save = read.n_save n_relaxation = read.n_relaxation dt = read.dt eta = read.eta g = read.g a = read.blob_radius scheme = read.scheme output_name = read.output_name structures = read.structures structures_ID = read.structures_ID mobility_vector_prod = set_mobility_vector_prod(read.mobility_vector_prod_implementation) multi_bodies_functions.calc_blob_blob_forces = multi_bodies_functions.set_blob_blob_forces(read.blob_blob_force_implementation) multi_bodies_functions.calc_body_body_forces_torques = multi_bodies_functions.set_body_body_forces_torques(read.body_body_force_torque_implementation) # Copy input file to output subprocess.call(["cp", input_file, output_name + '.inputfile']) # Set random generator state if read.random_state is not None: with open(read.random_state, 'rb') as f: np.random.set_state(cPickle.load(f)) elif read.seed is not None: np.random.seed(int(read.seed)) # Save random generator state with open(output_name + '.random_state', 'wb') as f: cPickle.dump(np.random.get_state(), f)
# Set some variables for the simulation n_steps = read.n_steps n_save = read.n_save n_relaxation = read.n_relaxation dt = read.dt eta = read.eta g = read.g a = read.blob_radius scheme = read.scheme output_name = read.output_name structures = read.structures structures_ID = read.structures_ID mobility_vector_prod = set_mobility_vector_prod(read.mobility_vector_prod_implementation) multi_bodies_functions.calc_blob_blob_forces = multi_bodies_functions.set_blob_blob_forces(read.blob_blob_force_implementation) multi_bodies_functions.calc_body_body_forces_torques = multi_bodies_functions.set_body_body_forces_torques(read.body_body_force_torque_implementation) # Copy input file to output subprocess.call(["cp", input_file, output_name + '.inputfile']) # Set random generator state if read.random_state is not None: with open(read.random_state, 'rb') as f: np.random.set_state(cpickle.load(f)) elif read.seed is not None: np.random.seed(int(read.seed)) # Save random generator state with open(output_name + '.random_state', 'wb') as f: cpickle.dump(np.random.get_state(), f)