mirror of
https://github.com/rdkit/rdkit.git
synced 2026-06-04 21:54:27 +08:00
67 lines
2.0 KiB
C++
67 lines
2.0 KiB
C++
//
|
|
// Copyright (C) 2020 Brian P. Kelley
|
|
//
|
|
// @@ 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 "SubstructLibrary.h"
|
|
#include "PatternFactory.h"
|
|
#include <RDGeneral/RDThreads.h>
|
|
|
|
#ifdef RDK_THREADSAFE_SSS
|
|
#include <thread>
|
|
#include <future>
|
|
#endif
|
|
|
|
namespace RDKit {
|
|
namespace {
|
|
void fillPatterns(const SubstructLibrary &slib,
|
|
const PatternHolder &fph,
|
|
std::vector<ExplicitBitVect*> &fps,
|
|
unsigned int start, unsigned int end, unsigned int numThreads) {
|
|
for (unsigned int idx = start;
|
|
idx < end;
|
|
idx += numThreads) {
|
|
fps[idx] = fph.makeFingerprint( *slib.getMol(idx).get() );
|
|
}
|
|
}
|
|
}
|
|
|
|
void addPatterns(SubstructLibrary &sslib, int numThreads) {
|
|
PRECONDITION(sslib.getFpHolder().get() == nullptr, "Substruct library already has fingerprints");
|
|
numThreads = (int)getNumThreadsToUse(numThreads);
|
|
|
|
boost::shared_ptr<PatternHolder> ptr(new PatternHolder);
|
|
std::vector<ExplicitBitVect *> & fps = ptr->getFingerprints();
|
|
unsigned int startIdx = 0;
|
|
unsigned int endIdx = sslib.getMolecules().size();
|
|
fps.resize(endIdx);
|
|
|
|
#ifdef RDK_THREADSAFE_SSS
|
|
std::vector<std::future<void>> thread_group;
|
|
for (int thread_group_idx = 0; thread_group_idx < numThreads;
|
|
++thread_group_idx) {
|
|
// need to use std::ref otherwise things are passed by value
|
|
thread_group.emplace_back(
|
|
std::async(std::launch::async, fillPatterns,
|
|
std::ref(sslib), std::ref(*ptr.get()), std::ref(fps),
|
|
startIdx + thread_group_idx, endIdx, numThreads));
|
|
}
|
|
for (auto &fut : thread_group) {
|
|
fut.get();
|
|
}
|
|
#else
|
|
fillPaterns(lib, fps, 0, sslib.size(), 1);
|
|
#endif
|
|
if (ptr->size() != sslib.size()) {
|
|
throw ValueErrorException("Number of fingerprints generated not equal to current number of molecules");
|
|
}
|
|
|
|
sslib.getFpHolder() = ptr;
|
|
sslib.resetHolders();
|
|
}
|
|
}
|