#  Copyright (C) 2014
#      Pierre de Buyl
#  Copyright (C) 2012,2013
#      Max Planck Institute for Polymer Research
#  Copyright (C) 2008,2009,2010,2011
#      Max-Planck-Institute for Polymer Research & Fraunhofer SCAI
#
#  This file is part of ESPResSo++.
#
#  ESPResSo++ is free software: you can redistribute it and/or modify
#  it under the terms of the GNU General Public License as published by
#  the Free Software Foundation, either version 3 of the License, or
#  (at your option) any later version.
#
#  ESPResSo++ is distributed in the hope that it will be useful,
#  but WITHOUT ANY WARRANTY; without even the implied warranty of
#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#  GNU General Public License for more details.
#
#  You should have received a copy of the GNU General Public License
#  along with this program.  If not, see <http://www.gnu.org/licenses/>.

"""
*************************************
**espresso.interaction.HarmonicTrap**
*************************************

"""
from espresso import pmi
from espresso.esutil import *

from espresso.interaction.SingleParticlePotential import *
from espresso.interaction.Interaction import *
from _espresso import interaction_HarmonicTrap, interaction_SingleParticleHarmonicTrap


class HarmonicTrapLocal(SingleParticlePotentialLocal, interaction_HarmonicTrap):
    'The (local) HarmonicTrap potential.'
    def __init__(self):
        """Initialize the local HarmonicTrap object."""
        if not (pmi._PMIComm and pmi._PMIComm.isActive()) or pmi._MPIcomm.rank in pmi._PMIComm.getMPIcpugroup():
            cxxinit(self, interaction_HarmonicTrap)


class SingleParticleHarmonicTrapLocal(InteractionLocal, interaction_SingleParticleHarmonicTrap):
    'The (local) HarmonicTrap interaction.'
    def __init__(self, system, potential):
        if not (pmi._PMIComm and pmi._PMIComm.isActive()) or pmi._MPIcomm.rank in pmi._PMIComm.getMPIcpugroup():
            cxxinit(self, interaction_SingleParticleHarmonicTrap, system, potential)

    def setPotential(self, potential):
        if not (pmi._PMIComm and pmi._PMIComm.isActive()) or pmi._MPIcomm.rank in pmi._PMIComm.getMPIcpugroup():
            self.cxxclass.setPotential(self, potential)

if pmi.isController:
    class HarmonicTrap(SingleParticlePotential):
        'The HarmonicTrap potential.'
        pmiproxydefs = dict(
            cls = 'espresso.interaction.HarmonicTrapLocal',
            pmiproperty = ['k', 'center']
            )

    class SingleParticleHarmonicTrap(Interaction):
        __metaclass__ = pmi.Proxy
        pmiproxydefs = dict(
            cls = 'espresso.interaction.SingleParticleHarmonicTrapLocal',
            pmicall = ['setPotential']
            )
