root/06/libkombilo-branches/hash_center/test_hash_center2.py

#! /usr/bin/env python2.4

import os
import sys
import time
from libkombilo import *

def timed_search(gl, p):
    gl.reset()
    so = SearchOptions()
    so.algos = ALGO_FINALPOS | ALGO_MOVELIST
    start = time.time()
    gl.search(p, so)
    nohash_time = time.time() - start
    nh_result = []
    nh_result.append('\n'.join(gl.currentEntriesAsStrings()))
    nh_result.append(str(gl.size()) + ' games, ' + str(gl.numHits()) + ' hits.')
    nh_result.append(p.printPattern())
    for y in range(p.sizeY):
        for x in range(p.sizeX):
            nh_result.append(gl.lookupLabel(x,y))
        nh_result.append('\n')

    nh_result.append('\n')
    for y in range(p.sizeY):
        for x in range(p.sizeX):
            if gl.lookupLabel(x,y) != '.':
                cont = gl.lookupContinuation(x,y);
                nh_result.append("      %c      |   %3d (    %3d /    %3d ) |   %3d (   %3d /    %3d) |" % \
                      (gl.lookupLabel(x,y), cont.B, cont.wB, cont.lB, cont.W, cont.wW, cont.lW))

    gl.reset()
    so.algos = ALGO_FINALPOS | ALGO_MOVELIST | ALGO_HASH_FULL | ALGO_HASH_CORNER | ALGO_HASH_CENTER
    start = time.time()
    gl.search(p, so)
    hash_time = time.time() - start
    h_result = []
    h_result.append('\n'.join(gl.currentEntriesAsStrings()))
    h_result.append(str(gl.size()) + ' games, ' + str(gl.numHits()) + ' hits.')
    h_result.append(p.printPattern())
    for y in range(p.sizeY):
        for x in range(p.sizeX):
            h_result.append(gl.lookupLabel(x,y))
        h_result.append('\n')

    h_result.append('\n')
    for y in range(p.sizeY):
        for x in range(p.sizeX):
            if gl.lookupLabel(x,y) != '.':
                cont = gl.lookupContinuation(x,y);
                h_result.append("      %c      |   %3d (    %3d /    %3d ) |   %3d (   %3d /    %3d) |" % \
                      (gl.lookupLabel(x,y), cont.B, cont.wB, cont.lB, cont.W, cont.wW, cont.lW))

    if ''.join(nh_result) != ''.join(h_result):
        out1 = open('nh_result', 'ab')
        out1.write(''.join(nh_result))
        out1.close()
        out2 = open('h_result', 'ab')
        out2.write(''.join(h_result))
        out1.close()
    return (nohash_time, hash_time) 

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

pl = [ 
       Pattern(CORNER_NW_PATTERN, 19, 7, 7, 
      '''.......
         ....X..
         ...XO..
         ..XO...
         ..XO...
         .......
         .......'''.replace(' ','').replace('\n','')),
       Pattern(CORNER_NW_PATTERN, 19, 7, 7, 
      '''.......
         .......
         ..X..X.
         .O.X...
         .......
         ..O....
         .......'''.replace(' ','').replace('\n','')),
       Pattern(CORNER_NW_PATTERN, 19, 7, 7, 
      '''.......
         .......
         ...O...
         ...X...
         .......
         ..X....
         .......'''.replace(' ','').replace('\n','')),
       Pattern(CORNER_NW_PATTERN, 19, 7, 7, 
      '''.......
         .......
         .......
         ...X.X.
         .......
         ...O.O.
         .......'''.replace(' ','').replace('\n','')),
       Pattern(CORNER_SW_PATTERN, 19, 7, 7, '........................X........................'),
       Pattern(CENTER_PATTERN, 19, 3, 3, '.X.XXXXOX'),
       Pattern(CENTER_PATTERN, 19, 3, 5, '.X.' + '.OX' + '.OX' + '.OX' + 'OXO'),
       Pattern(CENTER_PATTERN, 19, 4, 5, '.X..' + '.OX.' + '.OX.' + '.OX.' + 'OXO.'),
       Pattern(CENTER_PATTERN, 19, 5, 4, '..XOO'+ '...XX'+ '.....'+ '..X..'),
       Pattern(CENTER_PATTERN, 19, 5, 4, "...O." + "....." + ".XOO." + "..OXO"),
     ]

start = time.time()
gl = GameList(sys.argv[1] + '.db')
end = time.time()
print gl.size(), 'games in the database'

for pattern in pl:
    print pattern.printPattern()
    print timed_search(gl, pattern)
    print 'num hits:', gl.num_hits
    print '..................................................................'
    print