SpeciesHandler.h

Go to the documentation of this file.

// This file is part of the MercuryDPM project (https://www.mercurydpm.org).
// Copyright (c), The MercuryDPM Developers Team. All rights reserved.
// License: BSD 3-Clause License; see the LICENSE file in the root directory.
 
#ifndef SPECIESHANDLER_H
#define SPECIESHANDLER_H
 
#include "BaseHandler.h"
#include "Species/ParticleSpecies.h"
 
class SpeciesHandler final : public BaseHandler<ParticleSpecies>
{
public:
    SpeciesHandler();
    
    SpeciesHandler(const SpeciesHandler& other);
    
    SpeciesHandler& operator=(const SpeciesHandler& rhs);
    
    ~SpeciesHandler() override;
    
    void addObject(ParticleSpecies* S) override;
    
    void clear() override
    {
        BaseHandler<ParticleSpecies>::clear();
        mixedObjects_.clear();
    }
    
    void removeObject(unsigned int index) override;
    
    void readAndAddObject(std::istream& is) override;
    
    ParticleSpecies* readOldObject(std::istream& is);
    
    unsigned int getMixedId(unsigned int id1, unsigned int id2) const;
    
    template<typename U>
    typename std::enable_if<!std::is_pointer<typename U::MixedSpeciesType>::value, typename U::MixedSpeciesType*>::type
    getMixedObject(const U* S, const U* T)
    {
        return static_cast<typename U::MixedSpeciesType*>(getMixedObject(S->getIndex(), T->getIndex()));
    }
    
    BaseSpecies* getMixedObject(unsigned int id1, unsigned int id2);
    
    const std::vector<BaseSpecies*>& getMixedObjects() const;
    
    void updateMixedObjects();
    
    virtual void write(std::ostream& os) const;
    
    std::string getName() const override;
    
    bool useAngularDOFs();
 
private:
    std::vector<BaseSpecies*> mixedObjects_;
};
 
#endif