🏡 index : github.com/captn3m0/okiya.git

aca9660e3530e7f9a65136b5268e01f39737a70f [patch]

6f170815c7daa20ba8af018234a4e79cce13bfba

aca9660e3530e7f9a65136b5268e01f39737a70f.tar.gz

Initial commit

Diff

 __init__.py      |  0 
 main.py          | 81 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 requirements.txt |  1 +
 3 files changed, 82 insertions(+)

diff --git a/__init__.py b/__init__.py
new file mode 100644
index 0000000..e69de29 100644
--- /dev/null
+++ a/__init__.py
diff --git a/main.py b/main.py
new file mode 100644
index 0000000..7ccd118 100644
--- /dev/null
+++ a/main.py
@@ -1,0 +1,81 @@
from mcts import mcts
from enum import Enum
from random import shuffle

class Attributes(Enum):
	sun = 1
	flag = 2
	bird = 3
	rain = 4

	star = 5
	tree = 6
	leaf = 7
	tulip = 8

class Card():
	def __init__(self, attr1, attr2):
		# print(attr1, attr2)
		self.attr1 = attr1
		self.attr2 = attr2
	
	def __str__(self):
		return str(self.attr1) +","+ str(self.attr2)

class State:
	GAME_START = 0

	def __init__(self):
		cards = []
		for attr1 in range(1,5):
			for attr2 in range (5,9):
				cards.append(Card(attr1, attr2))

		shuffle(cards)

		self.grid = [ [cards[i*4 + j] for j in range(0,4)] for i in range(0,4)]
		self.currentTurn = self.GAME_START
		self.currentPlayer = 0
		self.display()

	def display(self):
		for ii in range(0,4):
			for jj in range(0,4):
				print(self.grid[ii][jj], end=' ', flush=True)
			print()

	def getPossibleActions(self):
		if(self.currentTurn == self.GAME_START):
			actions = [[(i,j) for j in range(0,4)] for i in range(0,4)]
		else:
			actions = []
			for ii in range(0,4):
				for jj in range(0,4):
					gridcard=grid[ii][jj]
					if (gridcard.__class__ == Card and gridcard.attr1 == self.lastCard.attr1 or gridself.lastCard.attr2 == self.lastCard.attr2 or gridself.lastCard.attr1 == self.lastCard.attr2 or gridself.lastCard.attr2 == self.lastCard.attr1):
						actions.append((ii,jj))

		return actions

	def takeAction(action):
		i,j = action
		
		self.lastCard = grid[i][j]
		self.currentTurn = action
		self.currentPlayer += 1
		self.currentPlayer = self.currentPlayer%2
		
		grid[i][j] = self.currentPlayer

		# TODO: Detect a win condition
	def isTerminal():
		return len(self.getPossibleActions) == 0

		# TODO: Return a reward in case of a win condition
	def getReward():
		pass

initialState = State()

mcts = mcts(timeLimit=1000)
# bestAction = mcts.search(initialState=initialState)
diff --git a/requirements.txt b/requirements.txt
new file mode 100644
index 0000000..bc415c5 100644
--- /dev/null
+++ a/requirements.txt
@@ -1,0 +1,1 @@
mcts==1.0.4