problem.py

import evaluation

class problem:
    def __init__(self,pieces,frame):
        self.pieces = pieces
        self.frame = frame
        self.merge_history = []
        self.depth_max = 0

    def search_match_pieces(self,frame_and_hist):
        """枠にはまるピースを探索し、評価値をつけ、リストにまとめて返します。

        frame:
            フレームと結合履歴のタプル
        戻り値:
            piece,評価値,フレームの頂点1,フレームの頂点2,ピースの頂点1,ピースの頂点2のタプルのリスト

        戻り値は以下のようにpieceオブジェクトと評価値が含まれるタプルのリストになります。
        [(pieceオブジェクト0,評価値0),(pieceオブジェクト1,評価値1),...,(pieceオブジェクトn,評価値n)]
        なおリストはソートされません。
        """
        enable_pieces=[] #枠と結合可能なピース,結合する枠の頂点,結合するピースの頂点のタプルをここに格納
        for piece in self.pieces: #全てのピースをみる
            for i,frame_vertex in enumerate(frame_and_hist[0].vertexes): #フレームの頂点全てをみる
                for j,piece_vertex in enumerate(piece.vertexes): #1ピースの頂点全てをみる

                    frame_vertex1=i
                    piece_vertex1=j
                    frame_vertex2=(i+1)%len(frame_and_hist[0].vertexes)
                    piece_vertex2=(j+1)%len(piece.vertexes)

                    #回転→is_overlapped()→merge()→is_on_grid()     結合判定
                    #     ↑←frip()←←←↓                          結合判定 piece内で関数作るべき？

                    enable_pieces.append((piece,frame_vertex1,frame_vertex2,piece_vertex1,piece_vertex2)) 
                    enable_pieces.append((piece,frame_vertex1,frame_vertex2,piece_vertex2,piece_vertex1))#逆
        return [(enable_piece[0],evaluation.evaluation.calc_eval_value(frame_and_hist[0].vertexes,enable_piece[0].vertexes),enable_piece[1],enable_piece[2],enable_piece[3],enable_piece[4])for enable_piece in enable_pieces]

    def sorting(self,pieces):
        """枠にはまるピースのリストを評価値順に降順にソートします。
        

        pieces: 
            piece,評価値,フレームの頂点1,フレームの頂点2,ピースの頂点1,ピースの頂点2のタプルのリスト
        戻り値:
            piece,評価値,フレームの頂点1,フレームの頂点2,ピースの頂点1,ピースの頂点2のタプルのリスト
        """
        return sorted(pieces, key = lambda t: t[1], reverse = True)

    def merge_pieces(self,frame,pieces):
        """枠にはまるピースのリストを枠と結合し、リストにまとめて返します(frame)。
        frame:
            piece,結合履歴のタプル
        pieces: 
            0     1      2               3               4           5
            piece,評価値,フレームの頂点1,フレームの頂点2,ピースの頂点1,ピースの頂点2のタプルのリスト
        戻り値:
            piece,評価値,結合履歴のタプルのリスト(frame)
        """

        #TODO:これだけややこしいならnamedtupleの方がいいかもね

        frames=[]
        for P in pieces:
            frames += [frame[0].merge(P[0],P[2],P[3],P[4],P[5]),P[1],frame[1]+[(frame[0],P[0],P[2],P[3],P[4],P[5])]]
        return frames

    def dfs_corner(self,frames,history,depth):
        """角の深さ優先探索を行います

        frames: piece
            フレームと結合履歴のタプルのリスト
        history:
            前のノードの結合履歴
        depth:
            深さ
        戻り値: bool
        """

        print("深さ",depth)
        print("frames",frames)
        for frame in frames:
            match_frames=self.merge_pieces(frame,self.sorting(self.search_match_pieces(frame)))
            #結合可能なピースを評価値順にソートし、枠と結合する
        if len(match_frames)==0: #結合可能なピースが無い、行き止まり
            if depth==len(self.pieces): #パズル完成
                self.merge_history=history
                return True
            if depth > self.depth_max:
                self.merge_history=history
                self.gui_api.draw_history(self)
            return False

        #再帰部
        same_value_frames=[] #評価値が等しいframeを格納するリスト
        for i,match_frame in enumerate(match_frames): #結合後のframeを全て見る
            if match_frames[i][1]!=match_frames[i-1][1]: #評価値が等しいframeを格納し終えたとき
                if self.dfs_corner(same_value_frames,same_value_frames[0][1],depth+1)==True: #それらを引数にして再帰
                    return True
            else:
                same_value_frames.append(match_frames[i]) #評価値が等しいframeを追加していく
        return False