root/06/devel-old/ktest.py

#! /usr/bin/env python

import time
import os
import sys
import cPickle
from copy import copy, deepcopy
from string import split, find, join, strip, replace, digits, maketrans, translate, lower, upper
import glob
import re
from array import *
import webbrowser

from abstractBoard import *
from board import *
from searchPY import *
from aglPY import *
from algosPY import *
try:
    from pattern import *
except:
    print "Using patternPY"
    from patternPY import *

# process sgf files in ./testdata

# test "process" directly

# progBar = dummy_progbar()
# db = SGFDatabase(19, progBar)
# ALGOS = 255
# db.process('./testdata', ('./testdata', ''), ALGOS, '*.sgf', )

# ----------------------------------------------------------

# test abstractGameList
 
agl = abstractGameList()
agl.addDB('./testdata', ('./testdata', ''), stdout_message())

# ----------------------------------------------------------

# set up a search pattern

p = Pattern(0, 13, 0, 14,
            5, 4,
            '..XO.'+
            '..OX.'+
            '.....'+
            '.....',
            '', 0, 'X') 

p.printPattern()

# do pattern search 

noMatches, Bwins, Wwins, noSwitched, continuations = \
      agl.search(p, {}, 0, {}, ['A', 'B', 'C', 'D', 'E', 'F', 'G'])

print
print 'Search results'
print 'Number of matches: ', noMatches
print 'Black wins: ', Bwins
print 'White wins: ', Wwins
print 'Number of color reversals: ', noSwitched
print 'Continuations:'
for k in continuations.keys(): print k, ': ', continuations[k]

# print results

# [do game info search]
# [...]

# clean up

# os.system('rm ./testdata/*.db')