~bertrand-nouvel/pycvf-indexes/trunk

#!/usr/bin/env python

# -*- coding: utf-8 -*-

#########################################################################################################################################

# 2009 CNRS Postdoctorate JFLI

#

# (c) All rights reserved

# ###############################################

#

#########################################################################################################################################

# Import required objects

#########################################################################################################################################

 

import time, numpy

from pycvf.core.errors import *

from pycvf.core.generic_application import *

from pycvf.core.utilities import with_progressbar

 

class DbIdxTester(IndexUsingApplication):

  class ProgramMetadata(object):

      name="Index Implementation Tester"

      version="1.0"

      author="Bertrand Nouvel bertrand.nouvel@gmail.com"

      copyright="        COPYRIGHT Bertrand Nouvel - JFLI - CNRS 2010"

      license="GPLv3"

 

        

 

  key=CmdLineString(None,"key","modelpath","specified within the model what should be used as key when indexing","0")

  value=CmdLineString(None,"value","modelpath","specified within the model what should be used as value when indexing","@")

 

  @classmethod

  def process(cls,*args, **kwargs):

      keyval=cls.mdl.resolve_modelpath("0")

      valueval=cls.mdl.resolve_modelpath("@")

      #cls.vdb=

      for e in cls.vdb:

          keys=cls.mdl.process_path([e[0]],[e[1]],keyval,lambda x:x)

          values=cls.mdl.process_path([e[0]],[e[1]],valueval,lambda x:x)

          for c in zip(keys,values):

               cls.idx.add(c[0],c[1])

      try:

        cls.idx.recompute()

      except:

        pass

 

      idb=iter(cls.vdb)

      try:

        while True:

          nq=3

          nr=5

          e=[ idb.next() for i in range(nq) ]

          keys=cls.mdl.process_path(map(lambda f:f[0],e),map(lambda f:f[1],e),keyval,lambda x:x)

          keys=numpy.array(keys)

          l=cls.idx.getitems(keys,nr)

          if (type(l)==numpy.ndarray):

            l=l.tolist()

          print type(l),type(l[0]),type(l[0][0]), type(l[0][0][0]),type(l[0][0][1])

          print len(l),len(l[0])

          #print l

          assert (type(l)==list)      # queries

          assert (len(l)==nq)

          assert (type(l[0])==list)   # results

          assert (len(l[0])<=nr)

          assert (len(l[0][0])==2)

          #assert (type(l[0][0])==tuple)           

          xxx=map (lambda y:map(lambda x:(cls.vdb[x[0]],x[1]),y),l)

          #xxx=map (lambda y:map(lambda x:(cls.vdb[x[0][:]],x[1]),y),l)

          print xxx

      except StopIteration:

        print "OK"

  

if __name__=="__main__":       

  DbIdxTester.run(sys.argv[1:])