basic_funcs.h

Go to the documentation of this file.

#ifndef BASIC_FUNCS_H
#define BASIC_FUNCS_H
 
#include <cmath>
#include <vector>
#include <ostream>
 
#include "source_base/vector3.h"
 
double maxval_abs_2d(const std::vector<ModuleBase::Vector3<double>>& array);
 
std::pair<int,int> maxloc_abs_2d(const std::vector<ModuleBase::Vector3<double>>& array);
 
template <typename T>
T sum_2d(const std::vector<ModuleBase::Vector3<T>>& array);
 
void scalar_multiply_2d(const std::vector<ModuleBase::Vector3<double>>& array,
                        double scalar,
                        std::vector<ModuleBase::Vector3<double>>& result);
 
void add_scalar_multiply_2d(const std::vector<ModuleBase::Vector3<double>>& array_1,
                            const std::vector<ModuleBase::Vector3<double>>& array_2,
                            double scalar,
                            std::vector<ModuleBase::Vector3<double>>& result);
 
void subtract_2d(const std::vector<ModuleBase::Vector3<double>>& array_1,
                 const std::vector<ModuleBase::Vector3<double>>& array_2,
                 std::vector<ModuleBase::Vector3<double>>& result);
 
void fill_scalar_2d(double scalar, std::vector<ModuleBase::Vector3<double>>& result);
 
void where_fill_scalar_2d(const std::vector<ModuleBase::Vector3<int>>& array_mask,
                          int mask,
                          double scalar,
                          std::vector<ModuleBase::Vector3<double>>& result);
 
void where_fill_scalar_else_2d(const std::vector<ModuleBase::Vector3<int>>& array_mask,
                               int mask,
                               double scalar,
                               const std::vector<ModuleBase::Vector3<double>>& rest,
                               std::vector<ModuleBase::Vector3<double>>& result);
 
void print_2d(const std::string info, const std::vector<ModuleBase::Vector3<double>> &array, const int nspin, const double unit_convert = 1.0, std::ostream& ofs = std::cout);
 
#endif // BASIC_FUNCS_H