Upload MjaiToPaiju.py

MjaiToPaiju.py

@@ -0,0 +1,258 @@
+import json
+
+tiles_tenhou = {
+    '1m': 0, '2m': 1, '3m': 2, '4m': 3, '5m': 4, '5mr': 4.5, '6m': 5, '7m': 6, '8m': 7, '9m': 8,
+    '1p': 9, '2p': 10, '3p': 11, '4p': 12, '5p': 13, '5pr': 13.5, '6p': 14, '7p': 15, '8p': 16, '9p': 17,
+    '1s': 18, '2s': 19, '3s': 20, '4s': 21, '5s': 22, '5sr': 22.5, '6s': 23, '7s': 24, '8s': 25, '9s': 26,
+    'E': 27, 'S': 28, 'W': 29, 'N': 30, 'P': 31, 'F': 32, 'C': 33, '?': 34
+}
+
+def process_mjai_events(events, self_id):
+    state = {
+        'meta': {},
+        'self': self_id,
+        'status': 'processing',
+        'last_actor': 0,
+        'players': [{'reach': False, 'discards': [], 'hand': [], 'melds': []} for _ in range(4)]
+    }
+    
+    # 方向映射
+    direction_map = {
+        1: 'next', 2: 'front', 3: 'prev'
+    }
+    
+    current_player = 0
+    reach_declared = [False] * 4
+    reach_discard_pending = [False] * 4
+    
+    def is_same_tile(t1, t2):
+        """判断两张牌是否相同（考虑红宝牌）"""
+        if t1 == t2:
+            return True
+        if t1.endswith('r') and t1[:-1] == t2:
+            return True
+        if t2.endswith('r') and t2[:-1] == t1:
+            return True
+        return False
+    
+    def kakan_to_pon(player_idx, pai):
+        """找到可以加杠的碰副露"""
+        for i, meld in enumerate(state['players'][player_idx]['melds']):
+            if meld['type'] == 'pon':
+                if is_same_tile(pai, meld['tiles'][0]):
+                    state['players'][player_idx]['melds'][i]['type'] = 'kakan'
+                    state['players'][player_idx]['melds'][i]['key2'] = pai
+                    break
+        return None
+    self_last_tsumo = None
+    last_event_type = ''
+    for event in events:
+        event_type = event.get('type')
+            
+        if event_type == 'start_kyoku':
+            state['meta'] = {
+                'bakaze': event['bakaze'],
+                'kyoku': event['kyoku'],
+                'honba': event['honba'],
+                'kyotaku': event['kyotaku'],
+                'scores': event['scores'],
+                'oya': event['oya'],
+                'round_info': f"{event['bakaze']}{event['kyoku']}-{event['honba']}",
+                'dora_markers': [],
+                'score_delta': [0, 0, 0, 0],
+                'ura_dora_markers': []
+            }
+            state['meta']['dora_markers'].append(event['dora_marker'])
+            for i in range(4):
+                state['players'][i]['hand'] = event['tehais'][i][:]
+                state['players'][i]['melds'] = []
+                state['players'][i]['discards'] = []
+                state['players'][i]['reach'] = False
+                state['players'][i]['hand'] = sorted(state['players'][i]['hand'], key=lambda x: tiles_tenhou[x])
+        elif event_type == 'hora':
+            actor = event['actor']
+            state['players'][actor]['hand'] = event['tiles']
+            state['status'] = 'hora'
+            state['meta',]['score_delta'] = event['delta']
+            state['meta']['ura_dora_markers'] = event['ura_dora']
+            print(state)
+
+        elif event_type == 'ryukyoku':
+            state['status'] = 'ryukyoku'
+            state['meta']['score_delta'] = event['delta']
+            
+        elif event_type == 'tsumo':
+            actor = event['actor']
+            pai = event['pai']
+            state['players'][actor]['hand'].append(pai)
+            if actor == self_id:
+                self_last_tsumo = pai
+            
+        elif event_type == 'dahai':
+            actor = event['actor']
+            pai = event['pai']
+            tsumogiri = event['tsumogiri']
+            if last_event_type == 'tsumo' and actor == self_id:
+                tsumogiri = (pai == self_last_tsumo)
+            
+            discard_info = {
+                'pai': pai,
+                'tsumogiri': tsumogiri,
+                'reach_decl': False
+            }
+            
+            if reach_discard_pending[actor]:
+                discard_info['reach_decl'] = True
+                reach_discard_pending[actor] = False
+                reach_declared[actor] = True
+                state['players'][actor]['reach'] = True
+            
+            state['players'][actor]['discards'].append(discard_info)
+            
+            if pai in state['players'][actor]['hand']:
+                state['players'][actor]['hand'].remove(pai)
+            elif '?' in state['players'][actor]['hand']:
+                state['players'][actor]['hand'].remove('?')
+        
+        elif event_type == 'reach':
+            actor = event['actor']
+            reach_discard_pending[actor] = True
+            
+        elif event_type == 'reach_accepted':
+            if 'scores' in event:
+                state['meta']['scores'] = event['scores']
+        
+        elif event_type == 'pon':
+            actor = event['actor']
+            target = event['target']
+            consumed = event['consumed']
+            pai = event['pai']
+            
+            rel_pos = direction_map[(target - actor) % 4]
+            
+            meld_info = {
+                'type': 'pon',
+                'tiles': consumed,
+                'key': pai,
+                'callee': rel_pos
+            }
+
+            for tile in consumed:
+                if tile in state['players'][actor]['hand']:
+                    state['players'][actor]['hand'].remove(tile)
+                elif '?' in state['players'][actor]['hand']:
+                    state['players'][actor]['hand'].remove('?')
+            
+            state['players'][actor]['melds'].append(meld_info)
+
+            target_river = state['players'][target]['discards']
+            
+            if target_river[-1]['pai'] == pai:
+                if target_river[-1]['reach_decl']:
+                    reach_discard_pending[target] = True
+                state['players'][target]['discards'] = state['players'][target]['discards'][:-1]
+        
+        elif event_type == 'chi':
+            actor = event['actor']
+            target = event['target']
+            consumed = event['consumed']
+            pai = event['pai']
+            
+            rel_pos = direction_map[(target - actor) % 4]
+            
+            meld_info = {
+                'type': 'chi',
+                'tiles': consumed,
+                'key': pai,
+                'callee': rel_pos
+            }
+
+            for tile in consumed:
+                if tile in state['players'][actor]['hand']:
+                    state['players'][actor]['hand'].remove(tile)
+                elif '?' in state['players'][actor]['hand']:
+                    state['players'][actor]['hand'].remove('?')
+            
+            state['players'][actor]['melds'].append(meld_info)
+            
+            target_river = state['players'][target]['discards']
+            
+            if target_river[-1]['pai'] == pai:
+                if target_river[-1]['reach_decl']:
+                    reach_discard_pending[target] = True
+                state['players'][target]['discards'] = state['players'][target]['discards'][:-1]
+        
+        elif event_type == 'daiminkan':
+            actor = event['actor']
+            target = event['target']
+            consumed = event['consumed']
+            pai = event['pai']
+            
+            rel_pos = direction_map[(target - actor) % 4]
+            
+            meld_info = {
+                'type': 'daiminkan',
+                'tiles': consumed,
+                'key': pai,
+                'callee': rel_pos
+            }
+
+            for tile in consumed:
+                if tile in state['players'][actor]['hand']:
+                    state['players'][actor]['hand'].remove(tile)
+                elif '?' in state['players'][actor]['hand']:
+                    state['players'][actor]['hand'].remove('?')
+            
+            state['players'][actor]['melds'].append(meld_info)
+            
+            target_river = state['players'][target]['discards']
+            
+            if target_river[-1]['pai'] == pai:
+                if target_river[-1]['reach_decl']:
+                    reach_discard_pending[target] = True
+                state['players'][target]['discards'] = state['players'][target]['discards'][:-1]
+        
+        elif event_type == 'kakan':
+            actor = event['actor']
+            pai = event['pai']
+            
+            # 找到对应的碰副露并升级为加杠
+            kakan_to_pon(actor, pai)
+            
+            if pai in state['players'][actor]['hand']:
+                state['players'][actor]['hand'].remove(pai)
+            elif '?' in state['players'][actor]['hand']:
+                state['players'][actor]['hand'].remove('?')
+        
+        elif event_type == 'ankan':
+            actor = event['actor']
+            consumed = event['consumed']
+            
+            meld_info = {
+                'type': 'ankan',
+                'tiles': consumed,
+                'callee': 'self'
+            }
+            
+            state['players'][actor]['melds'].append(meld_info)
+            
+            for tile in consumed:
+                if tile in state['players'][actor]['hand']:
+                    state['players'][actor]['hand'].remove(tile)
+                elif '?' in state['players'][actor]['hand']:
+                    state['players'][actor]['hand'].remove('?')
+
+        elif event_type == 'dora':
+            state['meta']['dora_markers'].append(event['dora_marker'])
+
+        last_event_type = event_type
+        if 'actor' in event and event['type'] not in ('tsumo', 'hora'):
+            state['players'][actor]['hand'] = sorted(state['players'][actor]['hand'], key=lambda x: tiles_tenhou[x])
+        if 'actor' in event:
+            state['last_actor'] = event['actor']
+    return state
+
+# ���用示例
+if __name__ == "__main__":
+    events = json.load(open('parsedlogs.txt', 'r'))
+    print(process_mjai_events(events, 2))