(mesh, info_on_pix, config)
| 199 | return [*mesh.neighbors(node)] |
| 200 | |
| 201 | def generate_face(mesh, info_on_pix, config): |
| 202 | H, W = mesh.graph['H'], mesh.graph['W'] |
| 203 | str_faces = [] |
| 204 | num_node = len(mesh.nodes) |
| 205 | ply_flag = config.get('save_ply') or config.get('save_obj') |
| 206 | def out_fmt(input, cur_id_b, cur_id_self, cur_id_a, ply_flag): |
| 207 | if ply_flag is True: |
| 208 | input.append(' '.join(['3', cur_id_b, cur_id_self, cur_id_a]) + '\n') |
| 209 | else: |
| 210 | input.append([cur_id_b, cur_id_self, cur_id_a]) |
| 211 | mesh_nodes = mesh.nodes |
| 212 | for node in mesh_nodes: |
| 213 | cur_id_self = mesh_nodes[node]['cur_id'] |
| 214 | ne_nodes = get_neighbors(mesh, node) |
| 215 | four_dir_nes = {'up': [], 'left': [], |
| 216 | 'down': [], 'right': []} |
| 217 | for ne_node in ne_nodes: |
| 218 | store_tuple = [ne_node, mesh_nodes[ne_node]['cur_id']] |
| 219 | if ne_node[0] == node[0]: |
| 220 | if ne_node[1] == ne_node[1] - 1: |
| 221 | four_dir_nes['left'].append(store_tuple) |
| 222 | else: |
| 223 | four_dir_nes['right'].append(store_tuple) |
| 224 | else: |
| 225 | if ne_node[0] == ne_node[0] - 1: |
| 226 | four_dir_nes['up'].append(store_tuple) |
| 227 | else: |
| 228 | four_dir_nes['down'].append(store_tuple) |
| 229 | for node_a, cur_id_a in four_dir_nes['up']: |
| 230 | for node_b, cur_id_b in four_dir_nes['right']: |
| 231 | out_fmt(str_faces, cur_id_b, cur_id_self, cur_id_a, ply_flag) |
| 232 | for node_a, cur_id_a in four_dir_nes['right']: |
| 233 | for node_b, cur_id_b in four_dir_nes['down']: |
| 234 | out_fmt(str_faces, cur_id_b, cur_id_self, cur_id_a, ply_flag) |
| 235 | for node_a, cur_id_a in four_dir_nes['down']: |
| 236 | for node_b, cur_id_b in four_dir_nes['left']: |
| 237 | out_fmt(str_faces, cur_id_b, cur_id_self, cur_id_a, ply_flag) |
| 238 | for node_a, cur_id_a in four_dir_nes['left']: |
| 239 | for node_b, cur_id_b in four_dir_nes['up']: |
| 240 | out_fmt(str_faces, cur_id_b, cur_id_self, cur_id_a, ply_flag) |
| 241 | |
| 242 | return str_faces |
| 243 | |
| 244 | def reassign_floating_island(mesh, info_on_pix, image, depth): |
| 245 | H, W = mesh.graph['H'], mesh.graph['W'], |
no test coverage detected