slip = None if(len(structure) > 2): slip = read_slip_file.read_slip_file(structure[2]) body_types.append(num_bodies_struct) body_names.append(structures_ID[ID]) # Create each body of type structure for i in range(num_bodies_struct): b = body.Body(struct_locations[i], struct_orientations[i], struct_ref_config, a) b.mobility_blobs = set_mobility_blobs(read.mobility_blobs_implementation) b.ID = structures_ID[ID] # Calculate body length for the RFD if i == 0: b.calc_body_length() else: b.body_length = bodies[-1].body_length multi_bodies_functions.set_slip_by_ID(b, slip) # Append bodies to total bodies list bodies.append(b) bodies = np.array(bodies) # Set some more variables num_of_body_types = len(body_types) num_bodies = bodies.size Nblobs = sum([x.Nblobs for x in bodies]) # Save bodies information with open(output_name + '.bodies_info', 'w') as f: f.write('num_of_body_types ' + str(num_of_body_types) + '\n') f.write('body_names ' + str(body_names) + '\n') f.write('body_types ' + str(body_types) + '\n') f.write('num_bodies ' + str(num_bodies) + '\n')
slip = None if(len(structure) > 2): slip = read_slip_file.read_slip_file(structure[2]) body_types.append(num_bodies_struct) body_names.append(structures_ID[ID]) # Create each body of type structure for i in range(num_bodies_struct): b = body.Body(struct_locations[i], struct_orientations[i], struct_ref_config, a) b.mobility_blobs = set_mobility_blobs(read.mobility_blobs_implementation) b.ID = structures_ID[ID] # Calculate body length for the RFD if i == 0: b.calc_body_length() else: b.body_length = bodies[-1].body_length multi_bodies_functions.set_slip_by_ID(b, slip) # Append bodies to total bodies list bodies.append(b) bodies = np.array(bodies) # Set some more variables num_of_body_types = len(body_types) num_bodies = bodies.size Nblobs = sum([x.Nblobs for x in bodies]) # Save bodies information with open(output_name + '.bodies_info', 'w') as f: f.write('num_of_body_types ' + str(num_of_body_types) + '\n') f.write('body_names ' + str(body_names) + '\n') f.write('body_types ' + str(body_types) + '\n') f.write('num_bodies ' + str(num_bodies) + '\n')