1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
|
//
// Copyright (C) 2003-2008 greg Landrum and Rational Discovery LLC
//
// @@ All Rights Reserved @@
// This file is part of the RDKit.
// The contents are covered by the terms of the BSD license
// which is included in the file license.txt, found at the root
// of the RDKit source tree.
//
#include <RDGeneral/export.h>
#ifndef __RD_WRAPHELPERS_H__
#define __RD_WRAPHELPERS_H__
#include <DataStructs/BitVects.h>
#include <DataStructs/BitOps.h>
#include <RDBoost/Wrap.h>
#include <DataStructs/base64.h>
namespace python = boost::python;
template <typename T>
void InitFromBase64(T& self, const std::string& inD) {
self.initFromText(inD.c_str(), inD.length(), true);
};
template <typename T>
std::string ToBase64(T& self) {
std::string tmp;
tmp = self.toString();
const char* txt = Base64Encode(tmp.c_str(), tmp.length());
std::string res(txt);
delete[] txt;
return res;
};
template <typename T>
void SetBitsFromList(T* bv, python::object onBitList) {
PySequenceHolder<int> bitL(onBitList);
for (unsigned int i = 0; i < bitL.size(); i++) {
bv->setBit(bitL[i]);
}
}
template <typename T>
void UnSetBitsFromList(T* bv, python::object offBitList) {
PySequenceHolder<int> bitL(offBitList);
for (unsigned int i = 0; i < bitL.size(); i++) {
bv->unsetBit(bitL[i]);
}
}
// used to support __getitem__
template <typename T>
int get_VectItem(const T& self, int which) {
if (which < 0) {
if (which + static_cast<int>(self.getNumBits()) < 0) {
throw IndexErrorException(which);
} else {
which += self.getNumBits();
}
}
return self.getBit(static_cast<unsigned int>(which));
}
// used to support __setitem__
template <typename T>
int set_VectItem(T& self, int which, const int val) {
if (which < 0) {
if (which + static_cast<int>(self.getNumBits()) < 0) {
throw IndexErrorException(which);
} else {
which += self.getNumBits();
}
}
if (val) {
return self.setBit(static_cast<unsigned int>(which));
} else {
return self.unsetBit(static_cast<unsigned int>(which));
}
}
// used to support getOnBits()
template <typename T>
IntVect GetOnBits(const T& self) {
IntVect res;
self.getOnBits(res);
return res;
}
template <typename T>
python::object BVToBinary(const T& bv) {
std::string res = bv.toString();
python::object retval = python::object(
python::handle<>(PyBytes_FromStringAndSize(res.c_str(), res.length())));
return retval;
}
#endif
|
