Main Lemma Repository
您最多选择25个主题 主题必须以字母或数字开头,可以包含连字符 (-),并且长度不得超过35个字符

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322
  1. /* This file is part of Lemma, a geophysical modelling and inversion API.
  2. * More information is available at http://lemmasoftware.org
  3. */
  4. /* This Source Code Form is subject to the terms of the Mozilla Public
  5. * License, v. 2.0. If a copy of the MPL was not distributed with this
  6. * file, You can obtain one at http://mozilla.org/MPL/2.0/.
  7. */
  8. /**
  9. * @file
  10. * @date 05/02/2018 09:46:38 PM
  11. * @author Trevor Irons (ti)
  12. * @email Trevor.Irons@utah.edu
  13. * @copyright Copyright (c) 2018, University of Utah
  14. * @copyright Copyright (c) 2018, Trevor Irons & Lemma Software, LLC
  15. */
  16. #ifndef FHT_INC
  17. #define FHT_INC
  18. #pragma once
  19. #include "HankelTransform.h"
  20. #include "CubicSplineInterpolator.h"
  21. namespace Lemma {
  22. /**
  23. \ingroup FDEM1D
  24. \brief Impliments lagged and related fast Hankel transform through
  25. digital filtering.
  26. \details A general Fast Hankel Transform routine which uses the digital
  27. filter apporach. Both lagged and related kernels are supported in
  28. order to minimize kernel function calls.
  29. This approach performs a complete sweep of the
  30. coefficients , for a variant that uses a longer filter which may
  31. be truncated, see FHTAnderson801.
  32. @see FHTAnderson801
  33. @see GQChave
  34. @see QWEKey
  35. */
  36. template < HANKELTRANSFORMTYPE Type >
  37. class FHT : public HankelTransform {
  38. friend std::ostream &operator<<(std::ostream &stream, const FHT<Type> &ob) {
  39. stream << ob.Serialize(); // << "\n";
  40. return stream;
  41. }
  42. public:
  43. // ==================== LIFECYCLE =======================
  44. /**
  45. * Default protected constructor, use NewSP methods to construct
  46. * @see FHT::NewSP
  47. */
  48. explicit FHT (const ctor_key& key ) : HankelTransform( key ) {
  49. }
  50. /**
  51. * Protected DeDerializing constructor, use factory DeSerialize method.
  52. * @see FHT::DeSerialize
  53. */
  54. FHT (const YAML::Node& node, const ctor_key& key) : HankelTransform(node, key) {
  55. }
  56. /** Default protected destructor, use smart pointers (std::shared_ptr) */
  57. ~FHT () {
  58. }
  59. /**
  60. * Factory method for generating concrete class.
  61. * @return a std::shared_ptr of type FHT
  62. */
  63. static std::shared_ptr< FHT > NewSP() {
  64. return std::make_shared< FHT >( ctor_key() );
  65. }
  66. /**
  67. * Uses YAML to serialize this object.
  68. * @return a YAML::Node
  69. * @see FHT::DeSerialize
  70. */
  71. YAML::Node Serialize() const {
  72. YAML::Node node = HankelTransform::Serialize();
  73. node.SetTag( this->GetName() ); // + enum2String(Type) );
  74. //node.SetTag( enum2String(Type) );
  75. //node["var"] = 0;
  76. return node;
  77. }
  78. /**
  79. * Constructs an FHT object from a YAML::Node.
  80. * @see FHT::Serialize
  81. */
  82. static std::shared_ptr<FHT> DeSerialize(const YAML::Node& node);
  83. // ==================== OPERATORS =======================
  84. // ==================== OPERATIONS =======================
  85. Complex Zgauss(const int&, const Lemma::EMMODE&, const int&, const Real&,
  86. const Real&, Lemma::KernelEM1DBase* Kernel);
  87. /// Computes related kernels, if applicable, otherwise this is
  88. /// just a dummy function.
  89. void ComputeRelated(const Real& rho, std::shared_ptr<KernelEM1DBase> Kernel) {
  90. }
  91. void ComputeRelated(const Real& rho, std::vector< std::shared_ptr<KernelEM1DBase> > KernelVec) {
  92. }
  93. void ComputeRelated(const Real& rho, std::shared_ptr<KernelEM1DManager> KernelManager);
  94. void ComputeLaggedRelated(const Real& rho, const int& nlag, std::shared_ptr<KernelEM1DManager> KernelManager);
  95. // ==================== ACCESS =======================
  96. /**
  97. * @param[in] rho is the argument for lagged convolution evaluation from the
  98. * spline after calculation.
  99. */
  100. void SetLaggedArg(const Real& rho) {
  101. for (int i=0; i<Zans.cols(); ++ i) {
  102. Zans(0, i) = Complex( splineVecReal[i]->Interpolate(rho),
  103. splineVecImag[i]->Interpolate(rho) );
  104. }
  105. return ;
  106. }
  107. // ==================== INQUIRY =======================
  108. /**
  109. * @return filter asbscissa spacing
  110. */
  111. inline Real GetABSER();
  112. //{
  113. // return 0; //this->WT(0,0)/this->WT(1,0);
  114. //}
  115. /** Returns the name of the underlying class, similiar to Python's type */
  116. inline std::string GetName() const {
  117. return enum2String(Type); //this->CName;
  118. }
  119. protected:
  120. // ==================== LIFECYCLE =======================
  121. // ==================== DATA MEMBERS =========================
  122. private:
  123. // Filter Weights, these are specialized for each template type
  124. static const Eigen::Matrix<Real, Eigen::Dynamic, 3> WT;
  125. /// Spines for lagged convolutions (real part)
  126. std::vector <std::shared_ptr<CubicSplineInterpolator> > splineVecReal;
  127. /// Spines for lagged convolutions (imaginary part)
  128. std::vector < std::shared_ptr<CubicSplineInterpolator> > splineVecImag;
  129. /// Holds answer, dimensions are NumConv, and NumberRelated.
  130. Eigen::Matrix<Complex, Eigen::Dynamic, Eigen::Dynamic> Zans;
  131. /** ASCII string representation of the class name */
  132. //static constexpr auto CName = "FHT";
  133. }; // ----- end of class FHT ----
  134. // Forward declarations
  135. template<>
  136. const Eigen::Matrix<Real, Eigen::Dynamic, 3> FHT<FHTKEY201>::WT;
  137. template<>
  138. const Eigen::Matrix<Real, Eigen::Dynamic, 3> FHT<FHTKEY101>::WT;
  139. template<>
  140. const Eigen::Matrix<Real, Eigen::Dynamic, 3> FHT<FHTKEY51>::WT;
  141. template<>
  142. const Eigen::Matrix<Real, Eigen::Dynamic, 3> FHT<FHTKONG61>::WT;
  143. template<>
  144. const Eigen::Matrix<Real, Eigen::Dynamic, 3> FHT<FHTKONG121>::WT;
  145. template<>
  146. const Eigen::Matrix<Real, Eigen::Dynamic, 3> FHT<FHTKONG241>::WT;
  147. template<>
  148. const Eigen::Matrix<Real, Eigen::Dynamic, 3> FHT<IRONS>::WT;
  149. template < HANKELTRANSFORMTYPE Type >
  150. Complex FHT<Type>::Zgauss(const int& ii, const Lemma::EMMODE& mode, const int& jj, const Real& val,
  151. const Real& val2, Lemma::KernelEM1DBase* Kernel){
  152. // TODO, in 101 or 51 we never reach here!!
  153. //std::cout << "Zgauss " << std::endl;
  154. return this->Zans(0, Kernel->GetManagerIndex());
  155. }
  156. // Specialisations
  157. // Note that ANDERSON801, CHAVE, QWEKEY will throw errors as they are not consistent
  158. // part of this class
  159. template < HANKELTRANSFORMTYPE Type >
  160. Real FHT< Type >::GetABSER() {
  161. return WT(0,0)/WT(1,0);
  162. }
  163. /* specializations could provide slighly better performance by reducing divides */
  164. // template < >
  165. // Real FHT< FHTKEY201 >::GetABSER() {
  166. // return WT(0,0)/WT(1,0);
  167. // }
  168. //
  169. // template < >
  170. // Real FHT< FHTKEY101 >::GetABSER() {
  171. // return WT(0,0)/WT(1,0);
  172. // }
  173. //
  174. // template < >
  175. // Real FHT< FHTKEY51 >::GetABSER() {
  176. // return WT(0,0)/WT(1,0);
  177. // }
  178. //--------------------------------------------------------------------------------------
  179. // Class: FHT
  180. // Method: ComputeRelated
  181. //--------------------------------------------------------------------------------------
  182. template < HANKELTRANSFORMTYPE Type >
  183. void FHT<Type>::ComputeRelated ( const Real& rho, std::shared_ptr<KernelEM1DManager> KernelManager ) {
  184. int nrel = (int)(KernelManager->GetSTLVector().size());
  185. Eigen::Matrix<Complex, Eigen::Dynamic, Eigen::Dynamic > Zwork;
  186. Zans= Eigen::Matrix<Complex, Eigen::Dynamic, Eigen::Dynamic>::Zero(1, nrel);
  187. Zwork.resize(WT.rows(), nrel);
  188. VectorXr lambda = WT.col(0).array()/rho;
  189. int NumFun = 0;
  190. int idx = 0;
  191. // Get Kernel values
  192. for (int ir=0; ir<lambda.size(); ++ir) {
  193. // irelated loop
  194. ++NumFun;
  195. KernelManager->ComputeReflectionCoeffs(lambda(ir), idx, rho);
  196. for (int ir2=0; ir2<nrel; ++ir2) {
  197. // Zwork* needed due to sign convention of filter weights
  198. Zwork(ir, ir2) = std::conj(KernelManager->GetSTLVector()[ir2]->RelBesselArg(lambda(ir)));
  199. }
  200. }
  201. for (int ir2=0; ir2<nrel; ++ir2) {
  202. Zans(0, ir2) = Zwork.col(ir2).dot(WT.col(KernelManager->GetSTLVector()[ir2]->GetBesselOrder() + 1))/rho;
  203. }
  204. return ;
  205. } // ----- end of method FHT::ComputeRelated -----
  206. //--------------------------------------------------------------------------------------
  207. // Class: FHT
  208. // Method: ComputeLaggedRelated
  209. //--------------------------------------------------------------------------------------
  210. template < HANKELTRANSFORMTYPE Type >
  211. void FHT<Type>::ComputeLaggedRelated ( const Real& rho, const int& nlag, std::shared_ptr<KernelEM1DManager> KernelManager ) {
  212. int nrel = (int)(KernelManager->GetSTLVector().size());
  213. Eigen::Matrix< Complex, Eigen::Dynamic, Eigen::Dynamic > Zwork;
  214. Zans= Eigen::Matrix<Complex, Eigen::Dynamic, Eigen::Dynamic>::Zero(nlag, nrel);
  215. Zwork.resize(WT.rows()+nlag, nrel); // Zwork needs to be expanded to filter length + nlag
  216. // lambda needs to be expanded to include lagged results
  217. VectorXr lambda = (VectorXr(WT.rows()+nlag) << WT.col(0).array()/rho, VectorXr::Zero(nlag)).finished();
  218. for (int ilam =WT.rows(); ilam< nlag+WT.rows(); ++ilam) {
  219. lambda(ilam) = lambda(ilam-1)/GetABSER();
  220. }
  221. int NumFun = 0;
  222. int idx = 0;
  223. VectorXr Arg(nlag);
  224. Arg(nlag-1) = rho;
  225. for (int ilag=nlag-2; ilag>=0; --ilag) {
  226. Arg(ilag) = Arg(ilag+1) * GetABSER();
  227. }
  228. // Get Kernel values
  229. for (int ir=0; ir<lambda.size(); ++ir) {
  230. // irelated loop
  231. ++NumFun;
  232. KernelManager->ComputeReflectionCoeffs(lambda(ir), idx, rho);
  233. for (int ir2=0; ir2<nrel; ++ir2) {
  234. Zwork(ir, ir2) = std::conj(KernelManager->GetSTLVector()[ir2]->RelBesselArg(lambda(ir)));
  235. }
  236. }
  237. // Inner product and scale
  238. int ilagr = nlag-1; // Zwork is in opposite order from Arg
  239. for (int ilag=0; ilag<nlag; ++ilag) {
  240. for (int ir2=0; ir2<nrel; ++ir2) {
  241. Zans(ilagr, ir2) = Zwork.col(ir2).segment(ilag,WT.rows()).dot( WT.col(KernelManager->GetSTLVector()[ir2]->GetBesselOrder()+1) ) / Arg(ilagr);
  242. }
  243. ilagr -= 1;
  244. }
  245. // make sure vectors are empty
  246. splineVecReal.clear();
  247. splineVecImag.clear();
  248. // Now do cubic spline
  249. for (int ii=0; ii<Zans.cols(); ++ii) {
  250. auto SplineR = CubicSplineInterpolator::NewSP();
  251. SplineR->SetKnots( Arg, Zans.col(ii).real() );
  252. splineVecReal.push_back(SplineR);
  253. auto SplineI = CubicSplineInterpolator::NewSP();
  254. SplineI->SetKnots( Arg, Zans.col(ii).imag() );
  255. splineVecImag.push_back(SplineI);
  256. }
  257. return ;
  258. } // ----- end of method FHT::ComputeLaggedRelated -----
  259. } // ----- end of namespace Lemma ----
  260. #endif // ----- #ifndef FHT_INC -----
  261. /* vim: set tabstop=4 expandtab: */
  262. /* vim: set filetype=cpp: */