Python/logic.py at main · Jem1D/Python · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
import random

def start():
    mat=[]
    for i in range(4):
        mat.append([0]*4)
    print("Commands are: ")
    print("W : Move up")
    print("S : Move Down")
    print("A : Move left")
    print("D : Move Right")
    add_2(mat)
    return mat

def add_2(mat):
    r=random.randint(0,3)
    c=random.randint(0,3)
    while(mat[r][c]!=0):
        r=random.randint(0,3)
        c=random.randint(0,3)
    mat[r][c]=2

def get_current_state(mat):
    for i in range(4):
        for j in range(4):
            if(mat[i][j]==2048):
                return "Won"
    for i in range(4):
        for j in range(4):
            if(mat[i][j]==0):
                return "Game continues"

#To check if two consecutive cells have same value

    for i in range(3):
        for j in range(3):
            if(mat[i][j]== mat[i + 1][j] or mat[i][j]== mat[i][j + 1]):
                return "Game continues"

    for j in range(3):
        if(mat[3][j]== mat[3][j + 1]):
            return "Game continues"

    for i in range(3):
        if(mat[i][3]== mat[i + 1][3]):
            return "Game continues"

    return 'Game lost'
# To bring every number to one side
def compress(mat):
    changed = False
    new_mat = []

    for i in range(4):
        new_mat.append([0] * 4)

    for i in range(4):
        pos = 0

        for j in range(4):
            if(mat[i][j] != 0):

                new_mat[i][pos] = mat[i][j]

                if(j != pos):
                    changed = True
                pos += 1

    return new_mat, changed
#merge same numbers
def merge(mat):
    changed = False

    for i in range(4):
        for j in range(3):

            if(mat[i][j] == mat[i][j + 1] and mat[i][j] != 0):

                mat[i][j] = mat[i][j] * 2
                mat[i][j + 1] = 0

                changed = True
    return mat,changed

def reverse(mat):
    new_mat =[]
    for i in range(4):
        new_mat.append([])
        for j in range(4):
            new_mat[i].append(mat[i][3 - j])
    return new_mat
def transpose(mat):
    new_mat = []
    for i in range(4):
        new_mat.append([])
        for j in range(4):
            new_mat[i].append(mat[j][i])
    return new_mat

def move_left(grid):

    new_grid, changed1 = compress(grid)
    new_grid, changed2 = merge(new_grid)
    changed = changed1 or changed2
    new_grid, temp = compress(new_grid)
    return new_grid, changed

def move_right(grid):
    new_grid = reverse(grid)
    new_grid, changed = move_left(new_grid)
    new_grid = reverse(new_grid)
    return new_grid, changed

def move_up(grid):
    new_grid = transpose(grid)
    new_grid, changed = move_left(new_grid)
    new_grid = transpose(new_grid)
    return new_grid, changed

def move_down(grid):
    new_grid = transpose(grid)
    new_grid, changed = move_right(new_grid)
    new_grid = transpose(new_grid)
    return new_grid, changed