# $Id$
#
#  Copyright (C) 2004-2006  Rational Discovery LLC
#
#   @@ All Rights Reserved  @@
#
import rdBase
import cDataStructs
__doc__=cDataStructs.__doc__
from cDataStructs import *


similarityFunctions=[
  ('Tanimoto',TanimotoSimilarity,''),
  ("Dice",DiceSimilarity,''),
  ("Cosine",CosineSimilarity,''),
  ("Sokal",SokalSimilarity,''),
  ("Russel",RusselSimilarity,''),
  ("Kulczynski",KulczynskiSimilarity,''),
  ("McConnaughey",McConnaugheySimilarity,''),
  ("Asymmetric",AsymmetricSimilarity,''),
  ("BraunBlanquet",BraunBlanquetSimilarity,''),
  ]

def FingerprintSimilarity(fp1,fp2,metric=TanimotoSimilarity):
  """ returns the calculated similarity between two fingerprints,
    handles any folding that may need to be done to ensure that they
    are compatible

  """
  sz1 = fp1.GetNumBits()
  sz2 = fp2.GetNumBits()
  if sz1<sz2:
    fp2 = FoldFingerprint(fp2,sz2/sz1)
  elif sz2<sz1:
    fp1 = FoldFingerprint(fp1,sz1/sz2)
  return metric(fp1,fp2)

ExplicitBitVect.ToBitString = BitVectToText
SparseBitVect.ToBitString = BitVectToText