pub_soft/rdkit
3
0
Fork 0
mirror of https://github.com/rdkit/rdkit.git synced 2026-06-04 21:54:27 +08:00
Code Issues Packages Projects Releases Wiki Activity

add bitmap version of AllProbeBitsMatch()

Browse Source
This commit is contained in:
Greg Landrum
2016-01-08 06:12:39 +01:00
parent 8aae258e3c
commit 7c0d4bdd89
3 changed files with 49 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
Code/DataStructs/BitOps.cpp
View File

@@ -914,3 +914,33 @@ double CalcBitmapTversky(const unsigned char* afp, const unsigned char* bfp,
}
return intersect_popcount / denom;
}
bool CalcBitmapAllProbeBitsMatch(const unsigned char* probe,
const unsigned char* ref,
unsigned int nBytes) {
PRECONDITION(probe, "no probe");
PRECONDITION(ref, "no ref");
#ifndef USE_BUILTIN_POPCOUNT
for (unsigned int i = 0; i < nBytes; i++) {
if (byte_popcounts[probe[i] & ref[i]] != byte_popcounts[probe[i]]) {
return false;
}
}
#else
unsigned int eidx = nBytes / sizeof(unsigned int);
for (unsigned int i = 0; i < eidx; ++i) {
if (__builtin_popcount(((unsigned int*)probe)[i] &
((unsigned int*)ref)[i]) !=
__builtin_popcount(((unsigned int*)probe)[i])) {
return false;
}
}
for (unsigned int i = eidx * sizeof(unsigned int); i < nBytes; ++i) {
if (byte_popcounts[probe[i] & ref[i]] != byte_popcounts[probe[i]]) {
return false;
}
}
#endif
return true;
}