o ~*d)@s<ddlZddlmZmZmZddlZdZ GdddZdS)N)UNKNOWNMINE get_neighborsFc@sDeZdZddZddZddZddZd d Zd d Zd dZ dS)PlayercCs||_||_||_||_t|||_t||fd|_t||fd|_ t |jD]}t |jD]}t |j||f|j ||f<q0q)t |_ dS)Nr)rowscolumnsmine_prbgamer neighborsnumpyfullminesunknownrangelensetinvalid)selfrrr r ijreC:\Users\mikol\Documents\MFF-PRG\LS2023\UdAI\working\05-probability_minesweeper\minesweeper_player.py__init__s  zPlayer.__init__cCs"|}|s |}|||SN) preprocessingprobability_playerinvalidate_with_neighborsrposrrrturn1s  z Player.turncCs|}d}t|jD];}t|jD]3}|j||ftkrE|||fdkr3t|j||f<|||f||||fkrE|||f}||f}qq |S)NgH.?)get_each_mine_probabilityrrrr rrr)rprbZmin_prbrrZ best_positionrrrr9s zPlayer.probability_playercCs,|j||j|D]}|j|q dSr)raddr )rrneighrrrrHs z Player.invalidate_with_neighborscs6tfddtjD_}|dksJdS)Nc3s(|]}tjD]}||fVq qdSr)rr).0rrrrr Ps&z(Player.preprocess_all..)rrrrrrrr'rpreprocess_allNszPlayer.preprocess_allc sjrj}tfddj|Dj|<j|dkrj|tfddj|Dj|<dj|krGj|j|ksIJj|dkrj|j|krtj|D]}j|tkrrtj|< |q_n"j|dkrj |j|D] }j|tkr|SqJjst sdSt j D]L}t jD]D}j||ftfddj||fDksJj||fdkrj||fj||ftfddj||fDksJqqdS) a3 Update precomputed information of cells in the set self.invalid. Using a simple counting, check cells which have to contain a mine. If this simple counting finds a cell which cannot contain a mine, then returns its position. Otherwise, returns None. c3&|]}j|tkr dndVqdSr!rNr rr&r%r'rrr(_$z'Player.preprocessing..rc3r*r+r rr-r'rrr(cr.FNc3r*r+r,r-r'rrr(}r.c3r*r+r/r-r'rrr(r.)rpopsumr rr rrrrr$ RUN_TESTSrrr)rrr%rrrr'rrTs@ ",&   2@zPlayer.preprocessingcCsft|j|jf}||jt}t}t |jD]}t |jD]}|j ||ft kr4d|||f<q"|j ||fdkrd|||f<g}d}|j ||fD]:}|d} |d} | |j| } |j | | ft kr{| |jvru|| ddg|| qN|j | | ft kr|d7}qN|j ||f|} t|dkr|t| ||D] } | |vr|| qq"q|}|D]+}d|d<|D]"} || dkr|d|j9<|| dkr|dd|j9<qq|D]I}||j} ||j} d}d}|D]}||dkr ||d7}q||d7}q|}|}||dkr"d|| | f<qd||}|||| | f<q|S)Nr!rr#)r zerosrrfillr rctZProblemrr rr rZ _variablesZ addVariableappenditemrZ addConstraintZExactSumConstraintr$Z getSolutions)rZ probabilityZall_unknown_positionsZmines_position_problemrrZunknown_cells_aroundZnum_of_revealed_mines_aroundZ neighbourxyvariableZ required_sumrZ solutionsZsolutionZunknown_positionZ true_prbsZ false_prbsZtrue_prbZ false_prbalpharrrr"sr           z Player.get_each_mine_probabilityN) __name__ __module__ __qualname__rr rrr)rr"rrrrrs -r) r minesweeper_commonrrrZ constraintr5r2rrrrrs