00001 #ifndef OPENMM_DENSITYFORCE_H_ 00002 #define OPENMM_DENSITYFORCE_H_ 00003 00004 /* -------------------------------------------------------------------------- * 00005 * OpenMM * 00006 * -------------------------------------------------------------------------- * 00007 * This is part of the OpenMM molecular simulation toolkit originating from * 00008 * Simbios, the NIH National Center for Physics-Based Simulation of * 00009 * Biological Structures at Stanford, funded under the NIH Roadmap for * 00010 * Medical Research, grant U54 GM072970. See https://simtk.org. * 00011 * * 00012 * Portions copyright (c) 2014 Stanford University and the Authors. * 00013 * Authors: Peter Eastman * 00014 * Contributors: * 00015 * * 00016 * Permission is hereby granted, free of charge, to any person obtaining a * 00017 * copy of this software and associated documentation files (the "Software"), * 00018 * to deal in the Software without restriction, including without limitation * 00019 * the rights to use, copy, modify, merge, publish, distribute, sublicense, * 00020 * and/or sell copies of the Software, and to permit persons to whom the * 00021 * Software is furnished to do so, subject to the following conditions: * 00022 * * 00023 * The above copyright notice and this permission notice shall be included in * 00024 * all copies or substantial portions of the Software. * 00025 * * 00026 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR * 00027 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, * 00028 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL * 00029 * THE AUTHORS, CONTRIBUTORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, * 00030 * DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR * 00031 * OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE * 00032 * USE OR OTHER DEALINGS IN THE SOFTWARE. * 00033 * -------------------------------------------------------------------------- */ 00034 00035 #include <openmm/Context.h> 00036 #include <openmm/Force.h> 00037 #include <ost/img/image.hh> 00038 00039 namespace ost { namespace mol{ namespace mm{ 00040 00041 00042 class DensityForce : public OpenMM::Force { 00043 public: 00049 //since the density is scaled in Angstrom, the resolution is also in Angstrom 00050 DensityForce(const ost::img::ImageHandle& d, Real r, Real s): density_(d), 00051 resolution_(r), 00052 scaling_(s) { } 00053 00054 void addParticle(Real mass); 00055 00056 int getNumParticles() const { return masses_.size(); } 00057 00058 int getNumBodies() const { return bodies_.size(); } 00059 00060 Real getParticleMass(int idx) const; 00061 00062 bool isInBody(int idx) const; 00063 00064 void defineBody(const std::vector<int>& indices); 00065 00066 const std::vector<int>& getBody(int idx) const; 00067 00068 void setScaling(Real scaling) { scaling_ = scaling; } 00069 00070 Real getScaling() const { return scaling_; } 00071 00072 ost::img::ImageHandle getDensity() const { return density_; } 00073 00074 Real getResolution() const { return resolution_; } 00075 00076 void updateParametersInContext(OpenMM::Context& context); 00077 00078 bool usesPeriodicBoundaryConditions() const { return false; } 00079 00080 protected: 00081 OpenMM::ForceImpl* createImpl() const; 00082 private: 00083 00084 ost::img::ImageHandle density_; 00085 std::vector<Real> masses_; 00086 std::vector<bool> in_body_; 00087 std::vector<std::vector<int> > bodies_; 00088 Real resolution_; 00089 Real scaling_; 00090 }; 00091 00092 }}} // ns 00093 00094 #endif /*OPENMM_DENSITYFORCE_H_*/