/*
    CuckooChess - A java chess program.
    Copyright (C) 2011  Peter Österlund, peterosterlund2@gmail.com

    This program is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program.  If not, see <http://www.gnu.org/licenses/>.
*/

package chess;

import org.junit.AfterClass;
import org.junit.BeforeClass;
import org.junit.Test;
import static org.junit.Assert.*;

/**
 *
 * @author petero
 */
public class EvaluateTest {

    public EvaluateTest() {
    }

    @BeforeClass
    public static void setUpClass() throws Exception {
    }

    @AfterClass
    public static void tearDownClass() throws Exception {
    }

    /**
     * Test of evalPos method, of class Evaluate.
     */
    @Test
    public void testEvalPos() throws ChessParseError {
        System.out.println("evalPos");
        Position pos = TextIO.readFEN(TextIO.startPosFEN);
        UndoInfo ui = new UndoInfo();
        pos.makeMove(TextIO.stringToMove(pos, "e4"), ui);
        pos.makeMove(TextIO.stringToMove(pos, "e5"), ui);
        pos.makeMove(TextIO.stringToMove(pos, "Nf3"), ui);
        pos.makeMove(TextIO.stringToMove(pos, "Nc6"), ui);
        pos.makeMove(TextIO.stringToMove(pos, "Bb5"), ui);
        pos.makeMove(TextIO.stringToMove(pos, "Nge7"), ui);
        assertTrue(moveScore(pos, "O-O") > 0);      // Castling is good
        assertTrue(moveScore(pos, "Ke2") < 0);      // Losing right to castle is bad
        assertTrue(moveScore(pos, "Kf1") < 0);
        assertTrue(moveScore(pos, "Rg1") < 0);
        assertTrue(moveScore(pos, "Rf1") < 0);

        pos = TextIO.readFEN("8/8/8/1r3k2/4pP2/4P3/8/4K2R w K - 0 1");
        assertEquals(true, pos.h1Castle());
        int cs1 = evalWhite(pos);
        pos.setCastleMask(pos.getCastleMask() & ~(1 << Position.H1_CASTLE));
        assertEquals(false, pos.h1Castle());
        int cs2 = evalWhite(pos);
        assertTrue(cs2 >= cs1);    // No bonus for useless castle right

        // Test rook open file bonus
        pos = TextIO.readFEN("r4rk1/1pp1qppp/3b1n2/4p3/2B1P1b1/1QN2N2/PP3PPP/R3R1K1 w - - 0 1");
        int ms1 = moveScore(pos, "Red1");
        int ms2 = moveScore(pos, "Rec1");
        int ms3 = moveScore(pos, "Rac1");
        int ms4 = moveScore(pos, "Rad1");
        assertTrue(ms1 > 0);        // Good to have rook on open file
        assertTrue(ms2 > 0);        // Good to have rook on half-open file
        assertTrue(ms1 > ms2);      // Open file better than half-open file
        assertTrue(ms3 > 0);
        assertTrue(ms4 > 0);
        assertTrue(ms4 > ms1);
        assertTrue(ms3 > ms2);
        
        pos = TextIO.readFEN("r3kb1r/p3pp1p/bpPq1np1/4N3/2pP4/2N1PQ2/P1PB1PPP/R3K2R b KQkq - 0 12");
        assertTrue(moveScore(pos, "O-O-O") > 0);    // Black long castle is bad for black
        pos.makeMove(TextIO.stringToMove(pos, "O-O-O"), ui);
        assertTrue(moveScore(pos, "O-O") > 0);      // White short castle is good for white
        
        pos = TextIO.readFEN("8/3k4/2p5/1pp5/1P1P4/3K4/8/8 w - - 0 1");
        int sc1 = moveScore(pos, "bxc5");
        int sc2 = moveScore(pos, "dxc5");
        assertTrue(sc1 < sc2);      // Don't give opponent a passed pawn.
        
        pos = TextIO.readFEN("8/pp1bk3/8/8/8/8/PPPBK3/8 w - - 0 1");
        sc1 = evalWhite(pos);
        pos.setPiece(Position.getSquare(3, 1), Piece.EMPTY);
        pos.setPiece(Position.getSquare(3, 0), Piece.WBISHOP);
        sc2 = evalWhite(pos);
        assertTrue(sc2 > sc1);      // Easier to win if bishops on same color
        
        // Test bishop mobility
        pos = TextIO.readFEN("r1bqkbnr/pppp1ppp/2n5/4p3/4P3/5N2/PPPP1PPP/RNBQKB1R w KQkq - 2 3");
        sc1 = moveScore(pos, "Bd3");
        sc2 = moveScore(pos, "Bc4");
        assertTrue(sc2 > sc1);
    }

    /**
     * Test of pieceSquareEval method, of class Evaluate.
     */
    @Test
    public void testPieceSquareEval() throws ChessParseError {
        System.out.println("pieceSquareEval");
        Position pos = TextIO.readFEN(TextIO.startPosFEN);
        int score = evalWhite(pos);
        assertEquals(0, score);    // Should be zero, by symmetry
        UndoInfo ui = new UndoInfo();
        pos.makeMove(TextIO.stringToMove(pos, "e4"), ui);
        score = evalWhite(pos);
        assertTrue(score > 0);     // Centralizing a pawn is a good thing
        pos.makeMove(TextIO.stringToMove(pos, "e5"), ui);
        score = evalWhite(pos);
        assertEquals(0, score);    // Should be zero, by symmetry
        assertTrue(moveScore(pos, "Nf3") > 0);      // Developing knight is good        
        pos.makeMove(TextIO.stringToMove(pos, "Nf3"), ui);
        assertTrue(moveScore(pos, "Nc6") < 0);      // Developing knight is good        
        pos.makeMove(TextIO.stringToMove(pos, "Nc6"), ui);
        assertTrue(moveScore(pos, "Bb5") > 0);      // Developing bishop is good
        pos.makeMove(TextIO.stringToMove(pos, "Bb5"), ui);
        pos.makeMove(TextIO.stringToMove(pos, "Nge7"), ui);
        assertTrue(moveScore(pos, "Qe2") > 0);      // Queen away from edge is good
        score = evalWhite(pos);
        pos.makeMove(TextIO.stringToMove(pos, "Bxc6"), ui);
        pos.makeMove(TextIO.stringToMove(pos, "Nxc6"), ui);
        int score2 = evalWhite(pos);
        assertTrue(score2 < score);                 // Bishop worth more than knight in this case
        
        pos = TextIO.readFEN("5k2/4nppp/p1n5/1pp1p3/4P3/2P1BN2/PP3PPP/3R2K1 w - - 0 1");
        assertTrue(moveScore(pos, "Rd7") > 0);      // Rook on 7:th rank is good
        assertTrue(moveScore(pos, "Rd8") <= 0);     // Rook on 8:th rank not particularly good
        pos.setPiece(TextIO.getSquare("a1"), Piece.WROOK);
        assertTrue(moveScore(pos, "Rac1") > 0);     // Rook on c-f files considered good

        pos = TextIO.readFEN("r4rk1/pppRRppp/1q4b1/n7/8/2N3B1/PPP1QPPP/6K1 w - - 0 1");
        score = evalWhite(pos);
        assertTrue(score > 100); // Two rooks on 7:th rank is very good
    }

    /**
     * Test of tradeBonus method, of class Evaluate.
     */
    @Test
    public void testTradeBonus() throws ChessParseError {
        System.out.println("tradeBonus");
        String fen = "8/5k2/6r1/2p1p3/3p4/2P2N2/3PPP2/4K1R1 w - - 0 1";
        Position pos = TextIO.readFEN(fen);
        int score1 = evalWhite(pos);
        UndoInfo ui = new UndoInfo();
        pos.makeMove(TextIO.stringToMove(pos, "Rxg6"), ui);
        pos.makeMove(TextIO.stringToMove(pos, "Kxg6"), ui);
        int score2 = evalWhite(pos);
        assertTrue(score2 > score1);    // White ahead, trading pieces is good
        
        pos = TextIO.readFEN(fen);
        pos.makeMove(TextIO.stringToMove(pos, "cxd4"), ui);
        pos.makeMove(TextIO.stringToMove(pos, "cxd4"), ui);
        score2 = evalWhite(pos);
        assertTrue(score2 < score1);    // White ahead, trading pawns is bad

        pos = TextIO.readFEN("8/8/1b2b3/4kp2/5N2/4NKP1/6B1/8 w - - 0 62");
        score1 = evalWhite(pos);
        pos.makeMove(TextIO.stringToMove(pos, "Nxe6"), ui);
        pos.makeMove(TextIO.stringToMove(pos, "Kxe6"), ui);
        score2 = evalWhite(pos);
        assertTrue(score2 > score1); // White ahead, trading pieces is good
    }

    /**
     * Test of material method, of class Evaluate.
     */
    @Test
    public void testMaterial() throws ChessParseError {
        System.out.println("material");
        Position pos = TextIO.readFEN(TextIO.startPosFEN);
        assertEquals(0, Evaluate.material(pos));
        
        final int pV = Evaluate.pV;
        final int qV = Evaluate.qV;
        assertTrue(pV != 0);
        assertTrue(qV != 0);
        assertTrue(qV > pV);
        
        UndoInfo ui = new UndoInfo();
        pos.makeMove(TextIO.stringToMove(pos, "e4"), ui);
        assertEquals(0, Evaluate.material(pos));
        pos.makeMove(TextIO.stringToMove(pos, "d5"), ui);
        assertEquals(0, Evaluate.material(pos));
        pos.makeMove(TextIO.stringToMove(pos, "exd5"), ui);
        assertEquals(pV, Evaluate.material(pos));
        pos.makeMove(TextIO.stringToMove(pos, "Qxd5"), ui);
        assertEquals(0, Evaluate.material(pos));
        pos.makeMove(TextIO.stringToMove(pos, "Nc3"), ui);
        assertEquals(0, Evaluate.material(pos));
        pos.makeMove(TextIO.stringToMove(pos, "Qxd2"), ui);
        assertEquals(-pV, Evaluate.material(pos));
        pos.makeMove(TextIO.stringToMove(pos, "Qxd2"), ui);
        assertEquals(-pV+qV, Evaluate.material(pos));
    }

    /**
     * Test of kingSafety method, of class Evaluate.
     */
    @Test
    public void testKingSafety() throws ChessParseError {
        System.out.println("kingSafety");
        Position pos = TextIO.readFEN("r3kb1r/p1p1pppp/b2q1n2/4N3/3P4/2N1PQ2/P2B1PPP/R3R1K1 w kq - 0 1");
        int s1 = evalWhite(pos);
        pos.setPiece(TextIO.getSquare("g7"), Piece.EMPTY);
        pos.setPiece(TextIO.getSquare("b7"), Piece.BPAWN);
        int s2 = evalWhite(pos);
        assertTrue(s2 < s1);    // Half-open g-file is bad for white

        // Trapping rook with own king is bad
        pos = TextIO.readFEN("rnbqk1nr/pppp1ppp/8/8/1bBpP3/8/PPP2PPP/RNBQK1NR w KQkq - 2 4");
        s1 = evalWhite(pos);
        pos.setPiece(TextIO.getSquare("e1"), Piece.EMPTY);
        pos.setPiece(TextIO.getSquare("f1"), Piece.WKING);
        s2 = evalWhite(pos);
        assertTrue(s2 < s1);

        pos = TextIO.readFEN("rnbqk1nr/pppp1ppp/8/8/1bBpPB2/8/PPP1QPPP/RN1K2NR w kq - 0 1");
        s1 = evalWhite(pos);
        pos.setPiece(TextIO.getSquare("d1"), Piece.EMPTY);
        pos.setPiece(TextIO.getSquare("c1"), Piece.WKING);
        s2 = evalWhite(pos);
        assertTrue(s2 < s1);
    }

    /**
     * Test of endGameEval method, of class Evaluate.
     */
    @Test
    public void testEndGameEval() throws ChessParseError {
        System.out.println("endGameEval");
        Position pos = new Position();
        pos.setPiece(Position.getSquare(4, 1), Piece.WKING);
        pos.setPiece(Position.getSquare(4, 6), Piece.BKING);
        int score = evalWhite(pos);
        assertEquals(0, score);

        pos.setPiece(Position.getSquare(3, 1), Piece.WBISHOP);
        score = evalWhite(pos);
        assertTrue(Math.abs(score) < 50);   // Insufficient material to mate

        pos.setPiece(Position.getSquare(3, 1), Piece.WKNIGHT);
        score = evalWhite(pos);
        assertTrue(Math.abs(score) < 50);   // Insufficient material to mate

        pos.setPiece(Position.getSquare(3, 1), Piece.WROOK);
        score = evalWhite(pos);
        final int rV = Evaluate.rV;
        assertTrue(Math.abs(score) > rV + 100);   // Enough material to force mate
        
        pos.setPiece(Position.getSquare(3, 6), Piece.BBISHOP);
        score = evalWhite(pos);
        final int bV = Evaluate.bV;
        assertTrue(score >= 0);
        assertTrue(score < rV - bV);   // Insufficient excess material to mate
        
        pos.setPiece(Position.getSquare(5, 6), Piece.BROOK);
        score = evalWhite(pos);
        assertTrue(score <= 0);
        assertTrue(-score < bV);
        
        pos.setPiece(Position.getSquare(2, 6), Piece.BBISHOP);
        score = evalWhite(pos);
        assertTrue(-score > bV * 2 + 100);
        
        // KrpKn is win for white
        pos = TextIO.readFEN("8/3bk3/8/8/8/3P4/3RK3/8 w - - 0 1");
        score = evalWhite(pos);
        final int pV = Evaluate.pV;
        assertTrue(score > rV + pV - bV - 100);
        
        // KNNK is a draw
        pos = TextIO.readFEN("8/8/4k3/8/8/3NK3/3N4/8 w - - 0 1");
        score = evalWhite(pos);
        assertTrue(Math.abs(score) < 50);
        
        pos = TextIO.readFEN("8/8/3k4/8/8/3NK3/2B5/8 b - - 0 1");
        score = evalWhite(pos);
        final int nV = Evaluate.nV;
        assertTrue(score > bV + nV + 150);  // KBNK is won, should have a bonus
        score = moveScore(pos, "Kc6");
        assertTrue(score > 0);      // Black king going into wrong corner, good for white
        score = moveScore(pos, "Ke6");
        assertTrue(score < 0);      // Black king going away from wrong corner, good for black
        
        // KRN vs KR is generally drawn
        pos = TextIO.readFEN("rk/p/8/8/8/8/NKR/8 w - - 0 1");
        score = evalWhite(pos);
        assertTrue(score < nV - 2 * pV);
    }

    /**
     * Test of endGameEval method, of class Evaluate.
     */
    @Test
    public void testPassedPawns() throws ChessParseError {
        System.out.println("passedPawns");
        Position pos = TextIO.readFEN("8/8/8/P3k/8/8/p/K w");
        int score = evalWhite(pos);
        assertTrue(score > 300); // Unstoppable passed pawn
        pos.whiteMove = false;
        score = evalWhite(pos);
        assertTrue(score <= 0); // Not unstoppable
        
        pos = TextIO.readFEN("4R3/8/8/3K4/8/4pk2/8/8 w - - 0 1");
        score = evalWhite(pos);
        pos.setPiece(TextIO.getSquare("d5"), Piece.EMPTY);
        pos.setPiece(TextIO.getSquare("d4"), Piece.WKING);
        int score2 = evalWhite(pos);
        assertTrue(score2 > score); // King closer to passed pawn promotion square

        // Connected passed pawn test. Disabled because it didn't help in tests
//        pos = TextIO.readFEN("4k3/8/8/4P3/3P1K2/8/8/8 w - - 0 1");
//        score = evalWhite(pos);
//        pos.setPiece(TextIO.getSquare("d4"), Piece.EMPTY);
//        pos.setPiece(TextIO.getSquare("d5"), Piece.WPAWN);
//        score2 = evalWhite(pos);
//        assertTrue(score2 > score); // Advancing passed pawn is good
    }

    /**
     * Test of endGameEval method, of class Evaluate.
     */
    @Test
    public void testBishAndRookPawns() throws ChessParseError {
        System.out.println("bishAndRookPawns");
        final int pV = Evaluate.pV;
        final int bV = Evaluate.bV;
        final int winScore = pV + bV;
        final int drawish = (pV + bV) / 20;
        Position pos = TextIO.readFEN("k7/8/8/8/2B5/2K5/P7/8 w - - 0 1");
        assertTrue(evalWhite(pos) > winScore);

        pos = TextIO.readFEN("k7/8/8/8/3B4/2K5/P7/8 w - - 0 1");
        assertTrue(evalWhite(pos) < drawish);

        pos = TextIO.readFEN("8/2k5/8/8/3B4/2K5/P7/8 w - - 0 1");
        assertTrue(evalWhite(pos) > winScore);

        pos = TextIO.readFEN("8/2k5/8/8/3B4/2K4P/8/8 w - - 0 1");
        assertTrue(evalWhite(pos) > winScore);

        pos = TextIO.readFEN("8/2k5/8/8/4B3/2K4P/8/8 w - - 0 1");
        assertTrue(evalWhite(pos) > winScore);

        pos = TextIO.readFEN("8/6k1/8/8/4B3/2K4P/8/8 w - - 0 1");
        assertTrue(evalWhite(pos) < drawish);

        pos = TextIO.readFEN("8/6k1/8/8/4B3/2K4P/7P/8 w - - 0 1");
        assertTrue(evalWhite(pos) < drawish);

        pos = TextIO.readFEN("8/6k1/8/8/2B1B3/2K4P/7P/8 w - - 0 1");
        assertTrue(evalWhite(pos) < drawish);

        pos = TextIO.readFEN("8/6k1/8/2B5/4B3/2K4P/7P/8 w - - 0 1");
        assertTrue(evalWhite(pos) > winScore);

        pos = TextIO.readFEN("8/6k1/8/8/4B3/2K4P/P7/8 w - - 0 1");
        assertTrue(evalWhite(pos) > winScore);
        
        pos = TextIO.readFEN("8/6k1/8/8/4B3/2K3PP/8/8 w - - 0 1");
        assertTrue(evalWhite(pos) > winScore);
    }
    
    @Test
    public void testTrappedBishop() throws ChessParseError {
        Position pos = TextIO.readFEN("r2q1rk1/ppp2ppp/3p1n2/8/3P4/1P1Q1NP1/b1P2PBP/2KR3R w - - 0 1");
        assertTrue(evalWhite(pos) > 0); // Black has trapped bishop

        pos = TextIO.readFEN("r2q2k1/pp1b1p1p/2p2np1/3p4/3P4/1BNQ2P1/PPPB1P1b/2KR4 w - - 0 1");
        assertTrue(evalWhite(pos) > 0); // Black has trapped bishop
    }
    
    /**
     * Test of endGameEval method, of class Evaluate.
     */
    @Test
    public void testKQKP() throws ChessParseError {
        System.out.println("KQKP");
        final int pV = Evaluate.pV;
        final int qV = Evaluate.qV;
        final int winScore = qV - pV - 200;
        final int drawish = (pV + qV) / 20;

        // Pawn on a2
        Position pos = TextIO.readFEN("8/8/1K6/8/8/Q7/p7/1k6 w - - 0 1");
        assertTrue(evalWhite(pos) < drawish);
        pos = TextIO.readFEN("8/8/8/1K6/8/Q7/p7/1k6 w - - 0 1");
        assertTrue(evalWhite(pos) > winScore);
        pos = TextIO.readFEN("3Q4/8/8/8/K7/8/1kp5/8 w - - 0 1");
        assertTrue(evalWhite(pos) > winScore);

        // Pawn on c2
        pos = TextIO.readFEN("3Q4/8/8/8/3K4/8/1kp5/8 w - - 0 1");
        assertTrue(evalWhite(pos) < drawish);
        pos = TextIO.readFEN("3Q4/8/8/8/8/4K3/1kp5/8 w - - 0 1");
        assertTrue(evalWhite(pos) > winScore);
    }
    
    @Test
    public void testKRKP() throws ChessParseError {
        System.out.println("KRKP");
        final int pV = Evaluate.pV;
        final int rV = Evaluate.rV;
        final int winScore = rV - pV;
        final int drawish = (pV + rV) / 20;
        Position pos = TextIO.readFEN("6R1/8/8/8/5K2/2kp4/8/8 w - - 0 1");
        assertTrue(evalWhite(pos) > winScore);
        pos.whiteMove = !pos.whiteMove;
        assertTrue(evalWhite(pos) < drawish);
    }

    @Test
    public void testCantWin() throws ChessParseError {
        Position pos = TextIO.readFEN("8/8/8/3k4/3p4/3K4/4N3/8 w - - 0 1");
        int score1 = evalWhite(pos);
        assertTrue(score1 < 0);
        UndoInfo ui = new UndoInfo();
        pos.makeMove(TextIO.stringToMove(pos, "Nxd4"), ui);
        int score2 = evalWhite(pos);
        assertTrue(score2 <= 0);
        assertTrue(score2 > score1);
    }

    /** Return static evaluation score for white, regardless of whose turn it is to move. */
    final static int evalWhite(Position pos) {
        Evaluate eval = new Evaluate();
        int ret = eval.evalPos(pos);
        Position symPos = swapColors(pos);
        int symScore = eval.evalPos(symPos);
        assertEquals(ret, symScore);
        if (!pos.whiteMove) {
            ret = -ret;
        }
        return ret;
    }

    final static Position swapColors(Position pos) {
        Position sym = new Position();
        sym.whiteMove = !pos.whiteMove;
        for (int x = 0; x < 8; x++) {
            for (int y = 0; y < 8; y++) {
                int p = pos.getPiece(Position.getSquare(x, y));
                p = Piece.isWhite(p) ? Piece.makeBlack(p) : Piece.makeWhite(p);
                sym.setPiece(Position.getSquare(x, 7-y), p);
            }
        }

        int castleMask = 0;
        if (pos.a1Castle()) castleMask |= 1 << Position.A8_CASTLE;
        if (pos.h1Castle()) castleMask |= 1 << Position.H8_CASTLE;
        if (pos.a8Castle()) castleMask |= 1 << Position.A1_CASTLE;
        if (pos.h8Castle()) castleMask |= 1 << Position.H1_CASTLE;
        sym.setCastleMask(castleMask);

        if (pos.getEpSquare() >= 0) {
            int x = Position.getX(pos.getEpSquare());
            int y = Position.getY(pos.getEpSquare());
            sym.setEpSquare(Position.getSquare(x, 7-y));
        }

        sym.halfMoveClock = pos.halfMoveClock;
        sym.fullMoveCounter = pos.fullMoveCounter;

        return sym;
    }

    /** Compute change in eval score for white after making "moveStr" in position "pos". */
    private final int moveScore(Position pos, String moveStr) {
        int score1 = evalWhite(pos);
        Position tmpPos = new Position(pos);
        UndoInfo ui = new UndoInfo();
        tmpPos.makeMove(TextIO.stringToMove(tmpPos, moveStr), ui);
        int score2 = evalWhite(tmpPos);
//        System.out.printf("move:%s s1:%d s2:%d\n", moveStr, score1, score2);
        return score2 - score1;
    }
}