mirror of
https://github.com/rdkit/rdkit.git
synced 2026-06-04 21:54:27 +08:00
83 lines
2.2 KiB
C++
Executable File
83 lines
2.2 KiB
C++
Executable File
// $Id$
|
|
//
|
|
// Copyright (c) 2003-2008 greg Landrum and Rational Discovery LLC
|
|
//
|
|
// @@ All Rights Reserved @@
|
|
//
|
|
#include "BitVect.h"
|
|
#include <sstream>
|
|
#include <limits>
|
|
#include <RDGeneral/StreamOps.h>
|
|
#include "base64.h"
|
|
#ifdef WIN32
|
|
#include <ios>
|
|
#endif
|
|
#include <boost/cstdint.hpp>
|
|
|
|
BitVect::~BitVect() {}; // must always implement virtual destructors
|
|
|
|
void BitVect::InitFromText(const char *data,const unsigned int dataLen,
|
|
bool isBase64,bool allowOldFormat){
|
|
std::stringstream ss(std::ios_base::binary|std::ios_base::in|std::ios_base::out);
|
|
if(isBase64){
|
|
unsigned int actualLen;
|
|
char *decoded;
|
|
decoded = Base64Decode((const char *)data,&actualLen);
|
|
ss.write(decoded,actualLen);
|
|
free(decoded);
|
|
} else {
|
|
ss.write(data,dataLen);
|
|
}
|
|
|
|
boost::int32_t format=0;
|
|
boost::uint32_t nOn=0;
|
|
boost::int32_t size;
|
|
boost::int32_t version=0;
|
|
|
|
// earlier versions of the code did not have the version number encoded, so
|
|
// we'll use that to distinguish version 0
|
|
ss.read((char *)&size,sizeof(size));
|
|
if(size<0){
|
|
version = -1*size;
|
|
if (version == 16) {
|
|
format=1;
|
|
}
|
|
else if (version == 32) {
|
|
format=2;
|
|
}
|
|
else {
|
|
throw ValueErrorException("bad version in BitVect pickle");
|
|
}
|
|
|
|
ss.read((char *)&size,sizeof(size));
|
|
} else if( !allowOldFormat ) {
|
|
throw ValueErrorException("invalid BitVect pickle");
|
|
}
|
|
ss.read((char *)&nOn,sizeof(nOn));
|
|
_InitForSize(static_cast<int>(size));
|
|
|
|
// if the either have older version or or version 16 with ints for on bits
|
|
if( (format==0) ||
|
|
( (format == 1) && (size >= std::numeric_limits<unsigned short>::max()) ) ) {
|
|
boost::uint32_t tmp;
|
|
for(unsigned int i=0; i<nOn; i++){
|
|
ss.read((char *)&tmp,sizeof(tmp));
|
|
SetBit(tmp);
|
|
}
|
|
} else if (format == 1) { // version 16 and on bits stored as short ints
|
|
boost::uint16_t tmp;
|
|
for(unsigned int i=0; i<nOn; i++){
|
|
ss.read((char *)&tmp,sizeof(tmp));
|
|
SetBit(tmp);
|
|
}
|
|
} else if (format == 2) { // run length encoded format
|
|
boost::uint32_t curr=0;
|
|
for (unsigned int i=0; i<nOn; i++) {
|
|
curr += RDKit::readPackedIntFromStream(ss);
|
|
SetBit(curr);
|
|
curr++;
|
|
}
|
|
}
|
|
|
|
}
|