mirror of
https://github.com/rdkit/rdkit.git
synced 2026-06-04 21:54:27 +08:00
50 lines
912 B
Python
Executable File
50 lines
912 B
Python
Executable File
# $Id: BitUtils.py 5009 2006-02-22 15:19:49Z glandrum $
|
|
#
|
|
# Copyright (C) 2005-2006 greg Landrum and Rational Discovery LLC
|
|
#
|
|
# @@ All Rights Reserved @@
|
|
#
|
|
def ConstructEnsembleBV(bv,bitsToKeep):
|
|
"""
|
|
|
|
>>> import DataStructs
|
|
>>> bv = DataStructs.ExplicitBitVect(128)
|
|
>>> bv.SetBitsFromList((1,5,47,99,120))
|
|
>>> r = ConstructEnsembleBV(bv,(0,1,2,3,45,46,47,48,49))
|
|
>>> r.GetNumBits()
|
|
9
|
|
>>> r.GetBit(0)
|
|
0
|
|
>>> r.GetBit(1)
|
|
1
|
|
>>> r.GetBit(5)
|
|
0
|
|
>>> r.GetBit(6) # old bit 47
|
|
1
|
|
|
|
|
|
|
|
"""
|
|
finalSize=len(bitsToKeep)
|
|
res = bv.__class__(finalSize)
|
|
|
|
|
|
for i,bit in enumerate(bitsToKeep):
|
|
if bv.GetBit(bit):
|
|
res.SetBit(i)
|
|
return res
|
|
|
|
|
|
#------------------------------------
|
|
#
|
|
# doctest boilerplate
|
|
#
|
|
def _test():
|
|
import doctest,sys
|
|
return doctest.testmod(sys.modules["__main__"])
|
|
|
|
if __name__ == '__main__':
|
|
import sys
|
|
failed,tried = _test()
|
|
sys.exit(failed)
|