multivariate_noise_generator.h
Go to the documentation of this file.
1 /*
2  * Copyright (c) 2019, the mcl_3dl authors
3  * All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions are met:
7  *
8  * * Redistributions of source code must retain the above copyright
9  * notice, this list of conditions and the following disclaimer.
10  * * Redistributions in binary form must reproduce the above copyright
11  * notice, this list of conditions and the following disclaimer in the
12  * documentation and/or other materials provided with the distribution.
13  * * Neither the name of the copyright holder nor the names of its
14  * contributors may be used to endorse or promote products derived from
15  * this software without specific prior written permission.
16  *
17  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
18  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
19  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20  * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
21  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
22  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
23  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
24  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
25  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
26  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
27  * POSSIBILITY OF SUCH DAMAGE.
28  */
29 
30 #ifndef MCL_3DL_NOISE_GENERATORS_MULTIVARIATE_NOISE_GENERATOR_H
31 #define MCL_3DL_NOISE_GENERATORS_MULTIVARIATE_NOISE_GENERATOR_H
32 
33 #include <random>
34 #include <vector>
35 
36 #include <Eigen/Core>
37 #include <Eigen/Eigenvalues>
38 
40 
41 namespace mcl_3dl
42 {
43 template <typename FLT_TYPE>
45 {
46 public:
48  using Matrix = Eigen::Matrix<FLT_TYPE, Eigen::Dynamic, Eigen::Dynamic>;
49  using Vector = Eigen::Matrix<FLT_TYPE, Eigen::Dynamic, 1>;
50 
51  template <typename MEAN_TYPE, typename COV_TYPE>
52  explicit MultivariateNoiseGenerator(const MEAN_TYPE& mean, const COV_TYPE& covariance)
53  {
54  Parent::setMean(mean);
55  mean_vec_.resize(Parent::mean_.size());
56  for (int i = 0; i < mean_vec_.size(); ++i)
57  {
58  mean_vec_[i] = Parent::mean_[i];
59  }
60  setCovariance(covariance);
61  }
62 
63  template <typename COV_TYPE>
64  void setCovariance(const COV_TYPE& covariance)
65  {
66  const size_t dim = Parent::getDimension();
67  Matrix cov_matrix(dim, dim);
68  for (size_t i = 0; i < dim; ++i)
69  {
70  for (size_t j = 0; j < dim; ++j)
71  {
72  cov_matrix(i, j) = covariance[j + i * dim];
73  }
74  }
75  const Eigen::SelfAdjointEigenSolver<Matrix> eigen_solver(cov_matrix);
76  norm_transform_ = eigen_solver.eigenvectors() * eigen_solver.eigenvalues().cwiseSqrt().asDiagonal();
77  }
78 
79  template <typename RANDOM_ENGINE>
80  std::vector<FLT_TYPE> operator()(RANDOM_ENGINE& engine) const
81  {
82  const size_t dim = Parent::getDimension();
83  Vector random(dim);
84  std::normal_distribution<FLT_TYPE> nd(0.0, 1.0);
85  for (size_t j = 0; j < dim; ++j)
86  {
87  random[j] = nd(engine);
88  }
89  const Vector sample_noise = mean_vec_ + norm_transform_ * random;
90  return std::vector<FLT_TYPE>(sample_noise.data(), sample_noise.data() + dim);
91  }
92 
93 protected:
96 };
97 
98 } // namespace mcl_3dl
99 
100 #endif // MCL_3DL_NOISE_GENERATORS_MULTIVARIATE_NOISE_GENERATOR_H
std::vector< FLT_TYPE > operator()(RANDOM_ENGINE &engine) const
std::vector< FLT_TYPE > mean_
Eigen::Matrix< FLT_TYPE, Eigen::Dynamic, Eigen::Dynamic > Matrix
Eigen::Matrix< FLT_TYPE, Eigen::Dynamic, 1 > Vector
MultivariateNoiseGenerator(const MEAN_TYPE &mean, const COV_TYPE &covariance)
void setCovariance(const COV_TYPE &covariance)
IMETHOD void random(Vector &a)


mcl_3dl
Author(s): Atsushi Watanabe
autogenerated on Wed May 12 2021 02:16:29