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
|
/*
* Copyright (C) 2016 Sereina Riniker, Paolo Tosco
*
* @@ 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 <GraphMol/Trajectory/Snapshot.h>
#include <GraphMol/Trajectory/Trajectory.h>
%}
%include <GraphMol/Trajectory/Snapshot.h>
%include <GraphMol/Trajectory/Trajectory.h>
%extend RDKit::Snapshot {
Snapshot(std::vector<double> &posVect, double energy = 0.0) {
double *posArray = new double[posVect.size()];
for (unsigned int i = 0; i < posVect.size(); ++i)
posArray[i] = posVect[i];
boost::shared_array<double> pos(posArray);
RDKit::Snapshot *s = new RDKit::Snapshot(pos, energy);
return s;
}
Snapshot(const RDKit::Snapshot &other) {
return new RDKit::Snapshot(other);
}
%newobject SnapshotVect;
static std::vector<RDKit::Snapshot> *SnapshotVect() {
return new std::vector<RDKit::Snapshot>();
}
}
|