LocalExpansion.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 LOCALEXPANSION_H_
#define LOCALEXPANSION_H_
 
#include "Math/NumericalVector.h"
#include "Math/Vector.h"
#include "Multipole.h"
#include <complex>
#include <vector>
 
class LocalExpansion
{
public:
    LocalExpansion(int p, NumericalVector<>* squaredFactorials, Vec3D location);
    
    virtual ~LocalExpansion();
    
    //Set correct length of the localExpansions, based on p
    void initialiseLocalExpansion();
    
    //Functions used in the downwards pass
    NumericalVector<std::complex<Mdouble>> translateLocalExpansion(Vec3D location);
    
    //add local expansion coefficients to the current coefficients
    void addLocalExpansionCoefficients(NumericalVector<std::complex<Mdouble>> localExpansionCoefficients);
    
    NumericalVector<std::complex<Mdouble>> getExpansionCoefficients()
    {
        return localExpansionCoefficients_;
    }
 
private:
    int p_;
    NumericalVector<>* squaredFactorials_;
    Vec3D location_;
    NumericalVector<std::complex<Mdouble>> localExpansionCoefficients_;
};
 
#endif /* LOCALEXPANSION_H_ */