コード例 #1
0
    def init_nodes(self):
        ends = []
        forks_before_lights = []
        forward_right_lights = []
        left_backward_lights = []
        forks_after_lights = []

        for i in range(4):
            dest = Node(i * 2, NodeType.basic)
            ends.append(dest)
            src = Node(i * 2 + 1, NodeType.basic)
            ends.append(src)

        for i in range(4):
            src_fork_before_lights = Node(i + 8, NodeType.fork)
            ends[i*2+1].add_edge(src_fork_before_lights, 1000)
            forks_before_lights.append(src_fork_before_lights)

            forward_right_light = Node(i + 12, NodeType.traffic_light)
            src_fork_before_lights.add_edge(forward_right_light, 50)

            left_backward_light = Node(i + 16, NodeType.traffic_light)
            src_fork_before_lights.add_edge(left_backward_light, 50)

            left_backward_lights.append(left_backward_light)
            forward_right_lights.append(forward_right_light)

            forward_right_fork = Node(i + 20, NodeType.fork)
            forward_right_light.add_edge(forward_right_fork, 50)
            forward_right_fork.add_edge(ends[(i*2 + 4) % len(ends)], 1000)
            forward_right_fork.add_edge(ends[(i*2 + 2) % len(ends)], 1000)

            left_backward_fork = Node(i + 24, NodeType.fork)
            left_backward_light.add_edge(left_backward_fork, 50)
            left_backward_fork.add_edge(ends[(i*2+6) % len(ends)], 1000)
            left_backward_fork.add_edge(ends[(i*2) % len(ends)], 1000)

            forks_after_lights.append(left_backward_fork)
            forks_after_lights.append(forward_right_fork)






        # start_node = Node(1, NodeType.basic)
        #
        # fork_node = Node(2, NodeType.fork)
        # start_node.add_edge(fork_node, 50)
        #
        # left_node = Node(3, NodeType.basic)
        # fork_node.add_edge(left_node, 50)
        # right_node = Node(4, NodeType.basic)
        # fork_node.add_edge(right_node, 50)

        return ends