mirror of
https://github.com/ml-explore/mlx.git
synced 2025-06-25 18:11:15 +08:00
4172 lines
832 KiB
HTML
4172 lines
832 KiB
HTML
<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN" "https://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd">
|
|
<html xmlns="http://www.w3.org/1999/xhtml" lang="en-US">
|
|
<head>
|
|
<meta http-equiv="Content-Type" content="text/xhtml;charset=UTF-8"/>
|
|
<meta http-equiv="X-UA-Compatible" content="IE=11"/>
|
|
<meta name="generator" content="Doxygen 1.10.0"/>
|
|
<meta name="viewport" content="width=device-width, initial-scale=1"/>
|
|
<title>MLX: mlx/3rdparty/pocketfft.h Source File</title>
|
|
<link href="tabs.css" rel="stylesheet" type="text/css"/>
|
|
<script type="text/javascript" src="jquery.js"></script>
|
|
<script type="text/javascript" src="dynsections.js"></script>
|
|
<script type="text/javascript" src="clipboard.js"></script>
|
|
<script type="text/javascript" src="cookie.js"></script>
|
|
<link href="search/search.css" rel="stylesheet" type="text/css"/>
|
|
<script type="text/javascript" src="search/searchdata.js"></script>
|
|
<script type="text/javascript" src="search/search.js"></script>
|
|
<link href="doxygen.css" rel="stylesheet" type="text/css" />
|
|
</head>
|
|
<body>
|
|
<div id="top"><!-- do not remove this div, it is closed by doxygen! -->
|
|
<div id="titlearea">
|
|
<table cellspacing="0" cellpadding="0">
|
|
<tbody>
|
|
<tr id="projectrow">
|
|
<td id="projectalign">
|
|
<div id="projectname">MLX
|
|
</div>
|
|
</td>
|
|
</tr>
|
|
</tbody>
|
|
</table>
|
|
</div>
|
|
<!-- end header part -->
|
|
<!-- Generated by Doxygen 1.10.0 -->
|
|
<script type="text/javascript">
|
|
/* @license magnet:?xt=urn:btih:d3d9a9a6595521f9666a5e94cc830dab83b65699&dn=expat.txt MIT */
|
|
var searchBox = new SearchBox("searchBox", "search/",'.html');
|
|
/* @license-end */
|
|
</script>
|
|
<script type="text/javascript" src="menudata.js"></script>
|
|
<script type="text/javascript" src="menu.js"></script>
|
|
<script type="text/javascript">
|
|
/* @license magnet:?xt=urn:btih:d3d9a9a6595521f9666a5e94cc830dab83b65699&dn=expat.txt MIT */
|
|
$(function() {
|
|
initMenu('',true,false,'search.php','Search');
|
|
$(function() { init_search(); });
|
|
});
|
|
/* @license-end */
|
|
</script>
|
|
<div id="main-nav"></div>
|
|
<script type="text/javascript">
|
|
/* @license magnet:?xt=urn:btih:d3d9a9a6595521f9666a5e94cc830dab83b65699&dn=expat.txt MIT */
|
|
$(function() { codefold.init(0); });
|
|
/* @license-end */
|
|
</script>
|
|
<!-- window showing the filter options -->
|
|
<div id="MSearchSelectWindow"
|
|
onmouseover="return searchBox.OnSearchSelectShow()"
|
|
onmouseout="return searchBox.OnSearchSelectHide()"
|
|
onkeydown="return searchBox.OnSearchSelectKey(event)">
|
|
</div>
|
|
|
|
<!-- iframe showing the search results (closed by default) -->
|
|
<div id="MSearchResultsWindow">
|
|
<div id="MSearchResults">
|
|
<div class="SRPage">
|
|
<div id="SRIndex">
|
|
<div id="SRResults"></div>
|
|
<div class="SRStatus" id="Loading">Loading...</div>
|
|
<div class="SRStatus" id="Searching">Searching...</div>
|
|
<div class="SRStatus" id="NoMatches">No Matches</div>
|
|
</div>
|
|
</div>
|
|
</div>
|
|
</div>
|
|
|
|
<div id="nav-path" class="navpath">
|
|
<ul>
|
|
<li class="navelem"><a class="el" href="dir_938ab0ecf10b8b860ff766c820f665fd.html">mlx</a></li><li class="navelem"><a class="el" href="dir_ad00dcd1517bfdbe01f68ec9b4eff877.html">3rdparty</a></li> </ul>
|
|
</div>
|
|
</div><!-- top -->
|
|
<div class="header">
|
|
<div class="headertitle"><div class="title">pocketfft.h</div></div>
|
|
</div><!--header-->
|
|
<div class="contents">
|
|
<a href="pocketfft_8h.html">Go to the documentation of this file.</a><div class="fragment"><div class="line"><a id="l00001" name="l00001"></a><span class="lineno"> 1</span><span class="comment">/*</span></div>
|
|
<div class="line"><a id="l00002" name="l00002"></a><span class="lineno"> 2</span><span class="comment">This file is part of pocketfft.</span></div>
|
|
<div class="line"><a id="l00003" name="l00003"></a><span class="lineno"> 3</span><span class="comment"></span> </div>
|
|
<div class="line"><a id="l00004" name="l00004"></a><span class="lineno"> 4</span><span class="comment">Copyright (C) 2010-2022 Max-Planck-Society</span></div>
|
|
<div class="line"><a id="l00005" name="l00005"></a><span class="lineno"> 5</span><span class="comment">Copyright (C) 2019-2020 Peter Bell</span></div>
|
|
<div class="line"><a id="l00006" name="l00006"></a><span class="lineno"> 6</span><span class="comment"></span> </div>
|
|
<div class="line"><a id="l00007" name="l00007"></a><span class="lineno"> 7</span><span class="comment">For the odd-sized DCT-IV transforms:</span></div>
|
|
<div class="line"><a id="l00008" name="l00008"></a><span class="lineno"> 8</span><span class="comment"> Copyright (C) 2003, 2007-14 Matteo Frigo</span></div>
|
|
<div class="line"><a id="l00009" name="l00009"></a><span class="lineno"> 9</span><span class="comment"> Copyright (C) 2003, 2007-14 Massachusetts Institute of Technology</span></div>
|
|
<div class="line"><a id="l00010" name="l00010"></a><span class="lineno"> 10</span><span class="comment"></span> </div>
|
|
<div class="line"><a id="l00011" name="l00011"></a><span class="lineno"> 11</span><span class="comment">Authors: Martin Reinecke, Peter Bell</span></div>
|
|
<div class="line"><a id="l00012" name="l00012"></a><span class="lineno"> 12</span><span class="comment"></span> </div>
|
|
<div class="line"><a id="l00013" name="l00013"></a><span class="lineno"> 13</span><span class="comment">All rights reserved.</span></div>
|
|
<div class="line"><a id="l00014" name="l00014"></a><span class="lineno"> 14</span><span class="comment"></span> </div>
|
|
<div class="line"><a id="l00015" name="l00015"></a><span class="lineno"> 15</span><span class="comment">Redistribution and use in source and binary forms, with or without modification,</span></div>
|
|
<div class="line"><a id="l00016" name="l00016"></a><span class="lineno"> 16</span><span class="comment">are permitted provided that the following conditions are met:</span></div>
|
|
<div class="line"><a id="l00017" name="l00017"></a><span class="lineno"> 17</span><span class="comment"></span> </div>
|
|
<div class="line"><a id="l00018" name="l00018"></a><span class="lineno"> 18</span><span class="comment">* Redistributions of source code must retain the above copyright notice, this</span></div>
|
|
<div class="line"><a id="l00019" name="l00019"></a><span class="lineno"> 19</span><span class="comment"> list of conditions and the following disclaimer.</span></div>
|
|
<div class="line"><a id="l00020" name="l00020"></a><span class="lineno"> 20</span><span class="comment">* Redistributions in binary form must reproduce the above copyright notice, this</span></div>
|
|
<div class="line"><a id="l00021" name="l00021"></a><span class="lineno"> 21</span><span class="comment"> list of conditions and the following disclaimer in the documentation and/or</span></div>
|
|
<div class="line"><a id="l00022" name="l00022"></a><span class="lineno"> 22</span><span class="comment"> other materials provided with the distribution.</span></div>
|
|
<div class="line"><a id="l00023" name="l00023"></a><span class="lineno"> 23</span><span class="comment">* Neither the name of the copyright holder nor the names of its contributors may</span></div>
|
|
<div class="line"><a id="l00024" name="l00024"></a><span class="lineno"> 24</span><span class="comment"> be used to endorse or promote products derived from this software without</span></div>
|
|
<div class="line"><a id="l00025" name="l00025"></a><span class="lineno"> 25</span><span class="comment"> specific prior written permission.</span></div>
|
|
<div class="line"><a id="l00026" name="l00026"></a><span class="lineno"> 26</span><span class="comment"></span> </div>
|
|
<div class="line"><a id="l00027" name="l00027"></a><span class="lineno"> 27</span><span class="comment">THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND</span></div>
|
|
<div class="line"><a id="l00028" name="l00028"></a><span class="lineno"> 28</span><span class="comment">ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED</span></div>
|
|
<div class="line"><a id="l00029" name="l00029"></a><span class="lineno"> 29</span><span class="comment">WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE</span></div>
|
|
<div class="line"><a id="l00030" name="l00030"></a><span class="lineno"> 30</span><span class="comment">DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR</span></div>
|
|
<div class="line"><a id="l00031" name="l00031"></a><span class="lineno"> 31</span><span class="comment">ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES</span></div>
|
|
<div class="line"><a id="l00032" name="l00032"></a><span class="lineno"> 32</span><span class="comment">(INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;</span></div>
|
|
<div class="line"><a id="l00033" name="l00033"></a><span class="lineno"> 33</span><span class="comment">LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON</span></div>
|
|
<div class="line"><a id="l00034" name="l00034"></a><span class="lineno"> 34</span><span class="comment">ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT</span></div>
|
|
<div class="line"><a id="l00035" name="l00035"></a><span class="lineno"> 35</span><span class="comment">(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS</span></div>
|
|
<div class="line"><a id="l00036" name="l00036"></a><span class="lineno"> 36</span><span class="comment">SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.</span></div>
|
|
<div class="line"><a id="l00037" name="l00037"></a><span class="lineno"> 37</span><span class="comment">*/</span></div>
|
|
<div class="line"><a id="l00038" name="l00038"></a><span class="lineno"> 38</span> </div>
|
|
<div class="line"><a id="l00039" name="l00039"></a><span class="lineno"> 39</span><span class="preprocessor">#ifndef POCKETFFT_HDRONLY_H</span></div>
|
|
<div class="line"><a id="l00040" name="l00040"></a><span class="lineno"> 40</span><span class="preprocessor">#define POCKETFFT_HDRONLY_H</span></div>
|
|
<div class="line"><a id="l00041" name="l00041"></a><span class="lineno"> 41</span> </div>
|
|
<div class="line"><a id="l00042" name="l00042"></a><span class="lineno"> 42</span><span class="preprocessor">#ifndef __cplusplus</span></div>
|
|
<div class="line"><a id="l00043" name="l00043"></a><span class="lineno"> 43</span><span class="preprocessor">#error This file is C++ and requires a C++ compiler.</span></div>
|
|
<div class="line"><a id="l00044" name="l00044"></a><span class="lineno"> 44</span><span class="preprocessor">#endif</span></div>
|
|
<div class="line"><a id="l00045" name="l00045"></a><span class="lineno"> 45</span> </div>
|
|
<div class="line"><a id="l00046" name="l00046"></a><span class="lineno"> 46</span><span class="preprocessor">#if !(__cplusplus >= 201103L || _MSVC_LANG+0L >= 201103L)</span></div>
|
|
<div class="line"><a id="l00047" name="l00047"></a><span class="lineno"> 47</span><span class="preprocessor">#error This file requires at least C++11 support.</span></div>
|
|
<div class="line"><a id="l00048" name="l00048"></a><span class="lineno"> 48</span><span class="preprocessor">#endif</span></div>
|
|
<div class="line"><a id="l00049" name="l00049"></a><span class="lineno"> 49</span> </div>
|
|
<div class="line"><a id="l00050" name="l00050"></a><span class="lineno"> 50</span><span class="preprocessor">#ifndef POCKETFFT_CACHE_SIZE</span></div>
|
|
<div class="line"><a id="l00051" name="l00051"></a><span class="lineno"><a class="line" href="pocketfft_8h.html#a9e604bcf20603d70b62b233d3f306714"> 51</a></span><span class="preprocessor">#define POCKETFFT_CACHE_SIZE 0</span></div>
|
|
<div class="line"><a id="l00052" name="l00052"></a><span class="lineno"> 52</span><span class="preprocessor">#endif</span></div>
|
|
<div class="line"><a id="l00053" name="l00053"></a><span class="lineno"> 53</span> </div>
|
|
<div class="line"><a id="l00054" name="l00054"></a><span class="lineno"> 54</span><span class="preprocessor">#include <cmath></span></div>
|
|
<div class="line"><a id="l00055" name="l00055"></a><span class="lineno"> 55</span><span class="preprocessor">#include <cstdlib></span></div>
|
|
<div class="line"><a id="l00056" name="l00056"></a><span class="lineno"> 56</span><span class="preprocessor">#include <stdexcept></span></div>
|
|
<div class="line"><a id="l00057" name="l00057"></a><span class="lineno"> 57</span><span class="preprocessor">#include <memory></span></div>
|
|
<div class="line"><a id="l00058" name="l00058"></a><span class="lineno"> 58</span><span class="preprocessor">#include <vector></span></div>
|
|
<div class="line"><a id="l00059" name="l00059"></a><span class="lineno"> 59</span><span class="preprocessor">#include <complex></span></div>
|
|
<div class="line"><a id="l00060" name="l00060"></a><span class="lineno"> 60</span><span class="preprocessor">#include <algorithm></span></div>
|
|
<div class="line"><a id="l00061" name="l00061"></a><span class="lineno"> 61</span><span class="preprocessor">#if POCKETFFT_CACHE_SIZE!=0</span></div>
|
|
<div class="line"><a id="l00062" name="l00062"></a><span class="lineno"> 62</span><span class="preprocessor">#include <array></span></div>
|
|
<div class="line"><a id="l00063" name="l00063"></a><span class="lineno"> 63</span><span class="preprocessor">#include <mutex></span></div>
|
|
<div class="line"><a id="l00064" name="l00064"></a><span class="lineno"> 64</span><span class="preprocessor">#endif</span></div>
|
|
<div class="line"><a id="l00065" name="l00065"></a><span class="lineno"> 65</span> </div>
|
|
<div class="line"><a id="l00066" name="l00066"></a><span class="lineno"> 66</span><span class="preprocessor">#ifndef POCKETFFT_NO_MULTITHREADING</span></div>
|
|
<div class="line"><a id="l00067" name="l00067"></a><span class="lineno"> 67</span><span class="preprocessor">#include <mutex></span></div>
|
|
<div class="line"><a id="l00068" name="l00068"></a><span class="lineno"> 68</span><span class="preprocessor">#include <condition_variable></span></div>
|
|
<div class="line"><a id="l00069" name="l00069"></a><span class="lineno"> 69</span><span class="preprocessor">#include <thread></span></div>
|
|
<div class="line"><a id="l00070" name="l00070"></a><span class="lineno"> 70</span><span class="preprocessor">#include <queue></span></div>
|
|
<div class="line"><a id="l00071" name="l00071"></a><span class="lineno"> 71</span><span class="preprocessor">#include <atomic></span></div>
|
|
<div class="line"><a id="l00072" name="l00072"></a><span class="lineno"> 72</span><span class="preprocessor">#include <functional></span></div>
|
|
<div class="line"><a id="l00073" name="l00073"></a><span class="lineno"> 73</span><span class="preprocessor">#include <new></span></div>
|
|
<div class="line"><a id="l00074" name="l00074"></a><span class="lineno"> 74</span> </div>
|
|
<div class="line"><a id="l00075" name="l00075"></a><span class="lineno"> 75</span><span class="preprocessor">#ifdef POCKETFFT_PTHREADS</span></div>
|
|
<div class="line"><a id="l00076" name="l00076"></a><span class="lineno"> 76</span><span class="preprocessor"># include <pthread.h></span></div>
|
|
<div class="line"><a id="l00077" name="l00077"></a><span class="lineno"> 77</span><span class="preprocessor">#endif</span></div>
|
|
<div class="line"><a id="l00078" name="l00078"></a><span class="lineno"> 78</span><span class="preprocessor">#endif</span></div>
|
|
<div class="line"><a id="l00079" name="l00079"></a><span class="lineno"> 79</span> </div>
|
|
<div class="line"><a id="l00080" name="l00080"></a><span class="lineno"> 80</span><span class="preprocessor">#if defined(__GNUC__)</span></div>
|
|
<div class="line"><a id="l00081" name="l00081"></a><span class="lineno"> 81</span><span class="preprocessor">#define POCKETFFT_NOINLINE __attribute__((noinline))</span></div>
|
|
<div class="line"><a id="l00082" name="l00082"></a><span class="lineno"> 82</span><span class="preprocessor">#define POCKETFFT_RESTRICT __restrict__</span></div>
|
|
<div class="line"><a id="l00083" name="l00083"></a><span class="lineno"> 83</span><span class="preprocessor">#elif defined(_MSC_VER)</span></div>
|
|
<div class="line"><a id="l00084" name="l00084"></a><span class="lineno"> 84</span><span class="preprocessor">#define POCKETFFT_NOINLINE __declspec(noinline)</span></div>
|
|
<div class="line"><a id="l00085" name="l00085"></a><span class="lineno"> 85</span><span class="preprocessor">#define POCKETFFT_RESTRICT __restrict</span></div>
|
|
<div class="line"><a id="l00086" name="l00086"></a><span class="lineno"> 86</span><span class="preprocessor">#else</span></div>
|
|
<div class="line"><a id="l00087" name="l00087"></a><span class="lineno"><a class="line" href="pocketfft_8h.html#a7020984e0ca1d6e565629ca6e7c1a7e0"> 87</a></span><span class="preprocessor">#define POCKETFFT_NOINLINE</span></div>
|
|
<div class="line"><a id="l00088" name="l00088"></a><span class="lineno"><a class="line" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82"> 88</a></span><span class="preprocessor">#define POCKETFFT_RESTRICT</span></div>
|
|
<div class="line"><a id="l00089" name="l00089"></a><span class="lineno"> 89</span><span class="preprocessor">#endif</span></div>
|
|
<div class="line"><a id="l00090" name="l00090"></a><span class="lineno"> 90</span> </div>
|
|
<div class="foldopen" id="foldopen00091" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00091" name="l00091"></a><span class="lineno"><a class="line" href="namespacepocketfft.html"> 91</a></span><span class="keyword">namespace </span><a class="code hl_namespace" href="namespacepocketfft.html">pocketfft</a> {</div>
|
|
<div class="line"><a id="l00092" name="l00092"></a><span class="lineno"> 92</span> </div>
|
|
<div class="foldopen" id="foldopen00093" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00093" name="l00093"></a><span class="lineno"><a class="line" href="namespacepocketfft_1_1detail.html"> 93</a></span><span class="keyword">namespace </span>detail {</div>
|
|
<div class="line"><a id="l00094" name="l00094"></a><span class="lineno"> 94</span><span class="keyword">using </span>std::size_t;</div>
|
|
<div class="line"><a id="l00095" name="l00095"></a><span class="lineno"> 95</span><span class="keyword">using </span>std::ptrdiff_t;</div>
|
|
<div class="line"><a id="l00096" name="l00096"></a><span class="lineno"> 96</span> </div>
|
|
<div class="line"><a id="l00097" name="l00097"></a><span class="lineno"> 97</span><span class="comment">// Always use std:: for <cmath> functions</span></div>
|
|
<div class="line"><a id="l00098" name="l00098"></a><span class="lineno"><a class="line" href="namespacepocketfft_1_1detail.html#a499c1e8b7d79a5272af024f46c63ff9d"> 98</a></span><span class="keyword">template</span> <<span class="keyword">typename</span> T> T <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a499c1e8b7d79a5272af024f46c63ff9d">cos</a>(T) = <span class="keyword">delete</span>;</div>
|
|
<div class="line"><a id="l00099" name="l00099"></a><span class="lineno"><a class="line" href="namespacepocketfft_1_1detail.html#a07745f4a069f811859308281b2982258"> 99</a></span><span class="keyword">template</span> <<span class="keyword">typename</span> T> T <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a07745f4a069f811859308281b2982258">sin</a>(T) = <span class="keyword">delete</span>;</div>
|
|
<div class="line"><a id="l00100" name="l00100"></a><span class="lineno"><a class="line" href="namespacepocketfft_1_1detail.html#a774f8b73f28259d4276bd188b540a3e3"> 100</a></span><span class="keyword">template</span> <<span class="keyword">typename</span> T> T <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a774f8b73f28259d4276bd188b540a3e3">sqrt</a>(T) = <span class="keyword">delete</span>;</div>
|
|
<div class="line"><a id="l00101" name="l00101"></a><span class="lineno"> 101</span> </div>
|
|
<div class="line"><a id="l00102" name="l00102"></a><span class="lineno"><a class="line" href="namespacepocketfft_1_1detail.html#a885ee37fcf564a268a5c8ca9ea8603e1"> 102</a></span><span class="keyword">using </span><a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#a885ee37fcf564a268a5c8ca9ea8603e1">shape_t</a> = std::vector<size_t>;</div>
|
|
<div class="line"><a id="l00103" name="l00103"></a><span class="lineno"><a class="line" href="namespacepocketfft_1_1detail.html#afb987c919e9424a996d0fc8b3c23cc84"> 103</a></span><span class="keyword">using </span><a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#afb987c919e9424a996d0fc8b3c23cc84">stride_t</a> = std::vector<ptrdiff_t>;</div>
|
|
<div class="line"><a id="l00104" name="l00104"></a><span class="lineno"> 104</span> </div>
|
|
<div class="line"><a id="l00105" name="l00105"></a><span class="lineno"><a class="line" href="namespacepocketfft_1_1detail.html#aecc5444a333360628be65a6f91ceb824"> 105</a></span><span class="keyword">constexpr</span> <span class="keywordtype">bool</span> FORWARD = <span class="keyword">true</span>,</div>
|
|
<div class="line"><a id="l00106" name="l00106"></a><span class="lineno"><a class="line" href="namespacepocketfft_1_1detail.html#a9d1eaa7469c018c39e745733eab9a9c3"> 106</a></span> BACKWARD = <span class="keyword">false</span>;</div>
|
|
<div class="line"><a id="l00107" name="l00107"></a><span class="lineno"> 107</span> </div>
|
|
<div class="line"><a id="l00108" name="l00108"></a><span class="lineno"> 108</span><span class="comment">// only enable vector support for gcc>=5.0 and clang>=5.0</span></div>
|
|
<div class="line"><a id="l00109" name="l00109"></a><span class="lineno"> 109</span><span class="preprocessor">#ifndef POCKETFFT_NO_VECTORS</span></div>
|
|
<div class="line"><a id="l00110" name="l00110"></a><span class="lineno"><a class="line" href="pocketfft_8h.html#aa9cdaed0819c48f97fcd19f05c289160"> 110</a></span><span class="preprocessor">#define POCKETFFT_NO_VECTORS</span></div>
|
|
<div class="line"><a id="l00111" name="l00111"></a><span class="lineno"> 111</span><span class="preprocessor">#if defined(__INTEL_COMPILER)</span></div>
|
|
<div class="line"><a id="l00112" name="l00112"></a><span class="lineno"> 112</span><span class="comment">// do nothing. This is necessary because this compiler also sets __GNUC__.</span></div>
|
|
<div class="line"><a id="l00113" name="l00113"></a><span class="lineno"> 113</span><span class="preprocessor">#elif defined(__clang__)</span></div>
|
|
<div class="line"><a id="l00114" name="l00114"></a><span class="lineno"> 114</span><span class="comment">// AppleClang has their own version numbering</span></div>
|
|
<div class="line"><a id="l00115" name="l00115"></a><span class="lineno"> 115</span><span class="preprocessor">#ifdef __apple_build_version__</span></div>
|
|
<div class="line"><a id="l00116" name="l00116"></a><span class="lineno"> 116</span><span class="preprocessor"># if (__clang_major__ > 9) || (__clang_major__ == 9 && __clang_minor__ >= 1)</span></div>
|
|
<div class="line"><a id="l00117" name="l00117"></a><span class="lineno"> 117</span><span class="preprocessor"># undef POCKETFFT_NO_VECTORS</span></div>
|
|
<div class="line"><a id="l00118" name="l00118"></a><span class="lineno"> 118</span><span class="preprocessor"># endif</span></div>
|
|
<div class="line"><a id="l00119" name="l00119"></a><span class="lineno"> 119</span><span class="preprocessor">#elif __clang_major__ >= 5</span></div>
|
|
<div class="line"><a id="l00120" name="l00120"></a><span class="lineno"> 120</span><span class="preprocessor"># undef POCKETFFT_NO_VECTORS</span></div>
|
|
<div class="line"><a id="l00121" name="l00121"></a><span class="lineno"> 121</span><span class="preprocessor">#endif</span></div>
|
|
<div class="line"><a id="l00122" name="l00122"></a><span class="lineno"> 122</span><span class="preprocessor">#elif defined(__GNUC__)</span></div>
|
|
<div class="line"><a id="l00123" name="l00123"></a><span class="lineno"> 123</span><span class="preprocessor">#if __GNUC__>=5</span></div>
|
|
<div class="line"><a id="l00124" name="l00124"></a><span class="lineno"> 124</span><span class="preprocessor">#undef POCKETFFT_NO_VECTORS</span></div>
|
|
<div class="line"><a id="l00125" name="l00125"></a><span class="lineno"> 125</span><span class="preprocessor">#endif</span></div>
|
|
<div class="line"><a id="l00126" name="l00126"></a><span class="lineno"> 126</span><span class="preprocessor">#endif</span></div>
|
|
<div class="line"><a id="l00127" name="l00127"></a><span class="lineno"> 127</span><span class="preprocessor">#endif</span></div>
|
|
<div class="line"><a id="l00128" name="l00128"></a><span class="lineno"> 128</span> </div>
|
|
<div class="line"><a id="l00129" name="l00129"></a><span class="lineno"><a class="line" href="structpocketfft_1_1detail_1_1_v_l_e_n.html#ab1fdc340dedde723e636746c828a4534"> 129</a></span><span class="keyword">template</span><<span class="keyword">typename</span> T> <span class="keyword">struct </span><a class="code hl_struct" href="structpocketfft_1_1detail_1_1_v_l_e_n.html">VLEN</a> { <span class="keyword">static</span> <span class="keyword">constexpr</span> <span class="keywordtype">size_t</span> <a class="code hl_variable" href="structpocketfft_1_1detail_1_1_v_l_e_n.html#ab1fdc340dedde723e636746c828a4534">val</a>=1; };</div>
|
|
<div class="line"><a id="l00130" name="l00130"></a><span class="lineno"> 130</span> </div>
|
|
<div class="line"><a id="l00131" name="l00131"></a><span class="lineno"> 131</span><span class="preprocessor">#ifndef POCKETFFT_NO_VECTORS</span></div>
|
|
<div class="line"><a id="l00132" name="l00132"></a><span class="lineno"> 132</span><span class="preprocessor">#if (defined(__AVX512F__))</span></div>
|
|
<div class="line"><a id="l00133" name="l00133"></a><span class="lineno"> 133</span><span class="keyword">template</span><> <span class="keyword">struct </span><a class="code hl_struct" href="structpocketfft_1_1detail_1_1_v_l_e_n.html">VLEN</a><float> { <span class="keyword">static</span> <span class="keyword">constexpr</span> <span class="keywordtype">size_t</span> <a class="code hl_variable" href="structpocketfft_1_1detail_1_1_v_l_e_n.html#ab1fdc340dedde723e636746c828a4534">val</a>=16; };</div>
|
|
<div class="line"><a id="l00134" name="l00134"></a><span class="lineno"> 134</span><span class="keyword">template</span><> <span class="keyword">struct </span>VLEN<double> { <span class="keyword">static</span> <span class="keyword">constexpr</span> <span class="keywordtype">size_t</span> <a class="code hl_variable" href="structpocketfft_1_1detail_1_1_v_l_e_n.html#ab1fdc340dedde723e636746c828a4534">val</a>=8; };</div>
|
|
<div class="line"><a id="l00135" name="l00135"></a><span class="lineno"> 135</span><span class="preprocessor">#elif (defined(__AVX__))</span></div>
|
|
<div class="line"><a id="l00136" name="l00136"></a><span class="lineno"> 136</span><span class="keyword">template</span><> <span class="keyword">struct </span>VLEN<float> { <span class="keyword">static</span> <span class="keyword">constexpr</span> <span class="keywordtype">size_t</span> <a class="code hl_variable" href="structpocketfft_1_1detail_1_1_v_l_e_n.html#ab1fdc340dedde723e636746c828a4534">val</a>=8; };</div>
|
|
<div class="line"><a id="l00137" name="l00137"></a><span class="lineno"> 137</span><span class="keyword">template</span><> <span class="keyword">struct </span>VLEN<double> { <span class="keyword">static</span> <span class="keyword">constexpr</span> <span class="keywordtype">size_t</span> <a class="code hl_variable" href="structpocketfft_1_1detail_1_1_v_l_e_n.html#ab1fdc340dedde723e636746c828a4534">val</a>=4; };</div>
|
|
<div class="line"><a id="l00138" name="l00138"></a><span class="lineno"> 138</span><span class="preprocessor">#elif (defined(__SSE2__))</span></div>
|
|
<div class="line"><a id="l00139" name="l00139"></a><span class="lineno"> 139</span><span class="keyword">template</span><> <span class="keyword">struct </span>VLEN<float> { <span class="keyword">static</span> <span class="keyword">constexpr</span> <span class="keywordtype">size_t</span> <a class="code hl_variable" href="structpocketfft_1_1detail_1_1_v_l_e_n.html#ab1fdc340dedde723e636746c828a4534">val</a>=4; };</div>
|
|
<div class="line"><a id="l00140" name="l00140"></a><span class="lineno"> 140</span><span class="keyword">template</span><> <span class="keyword">struct </span>VLEN<double> { <span class="keyword">static</span> <span class="keyword">constexpr</span> <span class="keywordtype">size_t</span> <a class="code hl_variable" href="structpocketfft_1_1detail_1_1_v_l_e_n.html#ab1fdc340dedde723e636746c828a4534">val</a>=2; };</div>
|
|
<div class="line"><a id="l00141" name="l00141"></a><span class="lineno"> 141</span><span class="preprocessor">#elif (defined(__VSX__))</span></div>
|
|
<div class="line"><a id="l00142" name="l00142"></a><span class="lineno"> 142</span><span class="keyword">template</span><> <span class="keyword">struct </span>VLEN<float> { <span class="keyword">static</span> <span class="keyword">constexpr</span> <span class="keywordtype">size_t</span> <a class="code hl_variable" href="structpocketfft_1_1detail_1_1_v_l_e_n.html#ab1fdc340dedde723e636746c828a4534">val</a>=4; };</div>
|
|
<div class="line"><a id="l00143" name="l00143"></a><span class="lineno"> 143</span><span class="keyword">template</span><> <span class="keyword">struct </span>VLEN<double> { <span class="keyword">static</span> <span class="keyword">constexpr</span> <span class="keywordtype">size_t</span> <a class="code hl_variable" href="structpocketfft_1_1detail_1_1_v_l_e_n.html#ab1fdc340dedde723e636746c828a4534">val</a>=2; };</div>
|
|
<div class="line"><a id="l00144" name="l00144"></a><span class="lineno"> 144</span><span class="preprocessor">#elif (defined(__ARM_NEON__) || defined(__ARM_NEON))</span></div>
|
|
<div class="line"><a id="l00145" name="l00145"></a><span class="lineno"> 145</span><span class="keyword">template</span><> <span class="keyword">struct </span>VLEN<float> { <span class="keyword">static</span> <span class="keyword">constexpr</span> <span class="keywordtype">size_t</span> <a class="code hl_variable" href="structpocketfft_1_1detail_1_1_v_l_e_n.html#ab1fdc340dedde723e636746c828a4534">val</a>=4; };</div>
|
|
<div class="line"><a id="l00146" name="l00146"></a><span class="lineno"> 146</span><span class="keyword">template</span><> <span class="keyword">struct </span>VLEN<double> { <span class="keyword">static</span> <span class="keyword">constexpr</span> <span class="keywordtype">size_t</span> <a class="code hl_variable" href="structpocketfft_1_1detail_1_1_v_l_e_n.html#ab1fdc340dedde723e636746c828a4534">val</a>=2; };</div>
|
|
<div class="line"><a id="l00147" name="l00147"></a><span class="lineno"> 147</span><span class="preprocessor">#else</span></div>
|
|
<div class="line"><a id="l00148" name="l00148"></a><span class="lineno"> 148</span><span class="preprocessor">#define POCKETFFT_NO_VECTORS</span></div>
|
|
<div class="line"><a id="l00149" name="l00149"></a><span class="lineno"> 149</span><span class="preprocessor">#endif</span></div>
|
|
<div class="line"><a id="l00150" name="l00150"></a><span class="lineno"> 150</span><span class="preprocessor">#endif</span></div>
|
|
<div class="line"><a id="l00151" name="l00151"></a><span class="lineno"> 151</span> </div>
|
|
<div class="line"><a id="l00152" name="l00152"></a><span class="lineno"> 152</span><span class="comment">// the __MINGW32__ part in the conditional below works around the problem that</span></div>
|
|
<div class="line"><a id="l00153" name="l00153"></a><span class="lineno"> 153</span><span class="comment">// the standard C++ library on Windows does not provide aligned_alloc() even</span></div>
|
|
<div class="line"><a id="l00154" name="l00154"></a><span class="lineno"> 154</span><span class="comment">// though the MinGW compiler and MSVC may advertise C++17 compliance.</span></div>
|
|
<div class="line"><a id="l00155" name="l00155"></a><span class="lineno"> 155</span><span class="preprocessor">#if (__cplusplus >= 201703L) && (!defined(__MINGW32__)) && (!defined(_MSC_VER))</span></div>
|
|
<div class="line"><a id="l00156" name="l00156"></a><span class="lineno"> 156</span><span class="keyword">inline</span> <span class="keywordtype">void</span> *<a class="code hl_function" href="namespacepocketfft_1_1detail.html#ae397445c61400f47a8fe3f8e1b6d0b76">aligned_alloc</a>(<span class="keywordtype">size_t</span> align, <span class="keywordtype">size_t</span> size)</div>
|
|
<div class="line"><a id="l00157" name="l00157"></a><span class="lineno"> 157</span> {</div>
|
|
<div class="line"><a id="l00158" name="l00158"></a><span class="lineno"> 158</span> <span class="comment">// aligned_alloc() requires that the requested size is a multiple of "align"</span></div>
|
|
<div class="line"><a id="l00159" name="l00159"></a><span class="lineno"> 159</span> <span class="keywordtype">void</span> *ptr = <a class="code hl_function" href="namespacepocketfft_1_1detail.html#ae397445c61400f47a8fe3f8e1b6d0b76">::aligned_alloc</a>(align,(size+align-1)&(~(align-1)));</div>
|
|
<div class="line"><a id="l00160" name="l00160"></a><span class="lineno"> 160</span> <span class="keywordflow">if</span> (!ptr) <span class="keywordflow">throw</span> std::bad_alloc();</div>
|
|
<div class="line"><a id="l00161" name="l00161"></a><span class="lineno"> 161</span> <span class="keywordflow">return</span> ptr;</div>
|
|
<div class="line"><a id="l00162" name="l00162"></a><span class="lineno"> 162</span> }</div>
|
|
<div class="line"><a id="l00163" name="l00163"></a><span class="lineno"> 163</span><span class="keyword">inline</span> <span class="keywordtype">void</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#aec7820e36a33e0a8bb83aa03b04b81e8">aligned_dealloc</a>(<span class="keywordtype">void</span> *ptr)</div>
|
|
<div class="line"><a id="l00164" name="l00164"></a><span class="lineno"> 164</span> { <a class="code hl_function" href="namespacemlx_1_1core_1_1allocator.html#a77f0a1215be242db6485612bcb273af5">free</a>(ptr); }</div>
|
|
<div class="line"><a id="l00165" name="l00165"></a><span class="lineno"> 165</span><span class="preprocessor">#else </span><span class="comment">// portable emulation</span></div>
|
|
<div class="foldopen" id="foldopen00166" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00166" name="l00166"></a><span class="lineno"><a class="line" href="namespacepocketfft_1_1detail.html#ae397445c61400f47a8fe3f8e1b6d0b76"> 166</a></span><span class="keyword">inline</span> <span class="keywordtype">void</span> *<a class="code hl_function" href="namespacepocketfft_1_1detail.html#ae397445c61400f47a8fe3f8e1b6d0b76">aligned_alloc</a>(<span class="keywordtype">size_t</span> align, <span class="keywordtype">size_t</span> size)</div>
|
|
<div class="line"><a id="l00167" name="l00167"></a><span class="lineno"> 167</span> {</div>
|
|
<div class="line"><a id="l00168" name="l00168"></a><span class="lineno"> 168</span> align = std::max(align, <span class="keyword">alignof</span>(max_align_t));</div>
|
|
<div class="line"><a id="l00169" name="l00169"></a><span class="lineno"> 169</span> <span class="keywordtype">void</span> *ptr = malloc(size+align);</div>
|
|
<div class="line"><a id="l00170" name="l00170"></a><span class="lineno"> 170</span> <span class="keywordflow">if</span> (!ptr) <span class="keywordflow">throw</span> std::bad_alloc();</div>
|
|
<div class="line"><a id="l00171" name="l00171"></a><span class="lineno"> 171</span> <span class="keywordtype">void</span> *res = <span class="keyword">reinterpret_cast<</span><span class="keywordtype">void</span> *<span class="keyword">></span></div>
|
|
<div class="line"><a id="l00172" name="l00172"></a><span class="lineno"> 172</span> ((<span class="keyword">reinterpret_cast<</span>uintptr_t<span class="keyword">></span>(ptr) & ~(uintptr_t(align-1))) + uintptr_t(align));</div>
|
|
<div class="line"><a id="l00173" name="l00173"></a><span class="lineno"> 173</span> (<span class="keyword">reinterpret_cast<</span><span class="keywordtype">void</span>**<span class="keyword">></span>(res))[-1] = ptr;</div>
|
|
<div class="line"><a id="l00174" name="l00174"></a><span class="lineno"> 174</span> <span class="keywordflow">return</span> res;</div>
|
|
<div class="line"><a id="l00175" name="l00175"></a><span class="lineno"> 175</span> }</div>
|
|
</div>
|
|
<div class="foldopen" id="foldopen00176" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00176" name="l00176"></a><span class="lineno"><a class="line" href="namespacepocketfft_1_1detail.html#aec7820e36a33e0a8bb83aa03b04b81e8"> 176</a></span><span class="keyword">inline</span> <span class="keywordtype">void</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#aec7820e36a33e0a8bb83aa03b04b81e8">aligned_dealloc</a>(<span class="keywordtype">void</span> *ptr)</div>
|
|
<div class="line"><a id="l00177" name="l00177"></a><span class="lineno"> 177</span> { <span class="keywordflow">if</span> (ptr) free((<span class="keyword">reinterpret_cast<</span><span class="keywordtype">void</span>**<span class="keyword">></span>(ptr))[-1]); }</div>
|
|
</div>
|
|
<div class="line"><a id="l00178" name="l00178"></a><span class="lineno"> 178</span><span class="preprocessor">#endif</span></div>
|
|
<div class="line"><a id="l00179" name="l00179"></a><span class="lineno"> 179</span> </div>
|
|
<div class="foldopen" id="foldopen00180" data-start="{" data-end="};">
|
|
<div class="line"><a id="l00180" name="l00180"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1arr.html"> 180</a></span><span class="keyword">template</span><<span class="keyword">typename</span> T> <span class="keyword">class </span><a class="code hl_class" href="classpocketfft_1_1detail_1_1arr.html">arr</a></div>
|
|
<div class="line"><a id="l00181" name="l00181"></a><span class="lineno"> 181</span> {</div>
|
|
<div class="line"><a id="l00182" name="l00182"></a><span class="lineno"> 182</span> <span class="keyword">private</span>:</div>
|
|
<div class="line"><a id="l00183" name="l00183"></a><span class="lineno"> 183</span> T *p;</div>
|
|
<div class="line"><a id="l00184" name="l00184"></a><span class="lineno"> 184</span> <span class="keywordtype">size_t</span> sz;</div>
|
|
<div class="line"><a id="l00185" name="l00185"></a><span class="lineno"> 185</span> </div>
|
|
<div class="line"><a id="l00186" name="l00186"></a><span class="lineno"> 186</span><span class="preprocessor">#if defined(POCKETFFT_NO_VECTORS)</span></div>
|
|
<div class="line"><a id="l00187" name="l00187"></a><span class="lineno"> 187</span> <span class="keyword">static</span> T *ralloc(<span class="keywordtype">size_t</span> num)</div>
|
|
<div class="line"><a id="l00188" name="l00188"></a><span class="lineno"> 188</span> {</div>
|
|
<div class="line"><a id="l00189" name="l00189"></a><span class="lineno"> 189</span> <span class="keywordflow">if</span> (num==0) <span class="keywordflow">return</span> <span class="keyword">nullptr</span>;</div>
|
|
<div class="line"><a id="l00190" name="l00190"></a><span class="lineno"> 190</span> <span class="keywordtype">void</span> *res = malloc(num*<span class="keyword">sizeof</span>(T));</div>
|
|
<div class="line"><a id="l00191" name="l00191"></a><span class="lineno"> 191</span> <span class="keywordflow">if</span> (!res) <span class="keywordflow">throw</span> std::bad_alloc();</div>
|
|
<div class="line"><a id="l00192" name="l00192"></a><span class="lineno"> 192</span> <span class="keywordflow">return</span> <span class="keyword">reinterpret_cast<</span>T *<span class="keyword">></span>(res);</div>
|
|
<div class="line"><a id="l00193" name="l00193"></a><span class="lineno"> 193</span> }</div>
|
|
<div class="line"><a id="l00194" name="l00194"></a><span class="lineno"> 194</span> <span class="keyword">static</span> <span class="keywordtype">void</span> dealloc(T *ptr)</div>
|
|
<div class="line"><a id="l00195" name="l00195"></a><span class="lineno"> 195</span> { free(ptr); }</div>
|
|
<div class="line"><a id="l00196" name="l00196"></a><span class="lineno"> 196</span><span class="preprocessor">#else</span></div>
|
|
<div class="line"><a id="l00197" name="l00197"></a><span class="lineno"> 197</span> <span class="keyword">static</span> T *ralloc(<span class="keywordtype">size_t</span> num)</div>
|
|
<div class="line"><a id="l00198" name="l00198"></a><span class="lineno"> 198</span> {</div>
|
|
<div class="line"><a id="l00199" name="l00199"></a><span class="lineno"> 199</span> <span class="keywordflow">if</span> (num==0) <span class="keywordflow">return</span> <span class="keyword">nullptr</span>;</div>
|
|
<div class="line"><a id="l00200" name="l00200"></a><span class="lineno"> 200</span> <span class="keywordtype">void</span> *ptr = <a class="code hl_function" href="namespacepocketfft_1_1detail.html#ae397445c61400f47a8fe3f8e1b6d0b76">aligned_alloc</a>(64, num*<span class="keyword">sizeof</span>(T));</div>
|
|
<div class="line"><a id="l00201" name="l00201"></a><span class="lineno"> 201</span> <span class="keywordflow">return</span> <span class="keyword">static_cast<</span>T*<span class="keyword">></span>(ptr);</div>
|
|
<div class="line"><a id="l00202" name="l00202"></a><span class="lineno"> 202</span> }</div>
|
|
<div class="line"><a id="l00203" name="l00203"></a><span class="lineno"> 203</span> <span class="keyword">static</span> <span class="keywordtype">void</span> dealloc(T *ptr)</div>
|
|
<div class="line"><a id="l00204" name="l00204"></a><span class="lineno"> 204</span> { <a class="code hl_function" href="namespacepocketfft_1_1detail.html#aec7820e36a33e0a8bb83aa03b04b81e8">aligned_dealloc</a>(ptr); }</div>
|
|
<div class="line"><a id="l00205" name="l00205"></a><span class="lineno"> 205</span><span class="preprocessor">#endif</span></div>
|
|
<div class="line"><a id="l00206" name="l00206"></a><span class="lineno"> 206</span> </div>
|
|
<div class="line"><a id="l00207" name="l00207"></a><span class="lineno"> 207</span> <span class="keyword">public</span>:</div>
|
|
<div class="line"><a id="l00208" name="l00208"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1arr.html#a961a24410638b35129cd6b81850d2a42"> 208</a></span> <a class="code hl_function" href="classpocketfft_1_1detail_1_1arr.html#a961a24410638b35129cd6b81850d2a42">arr</a>() : p(0), sz(0) {}</div>
|
|
<div class="line"><a id="l00209" name="l00209"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1arr.html#a04f832b780a4453fdf3b69bf75b182bd"> 209</a></span> <a class="code hl_function" href="classpocketfft_1_1detail_1_1arr.html#a04f832b780a4453fdf3b69bf75b182bd">arr</a>(<span class="keywordtype">size_t</span> n) : p(ralloc(n)), sz(n) {}</div>
|
|
<div class="foldopen" id="foldopen00210" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00210" name="l00210"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1arr.html#a0cd8fb4a588a74d428a7349d38b477d0"> 210</a></span> <a class="code hl_function" href="classpocketfft_1_1detail_1_1arr.html#a0cd8fb4a588a74d428a7349d38b477d0">arr</a>(<a class="code hl_class" href="classpocketfft_1_1detail_1_1arr.html">arr</a> &&other)</div>
|
|
<div class="line"><a id="l00211" name="l00211"></a><span class="lineno"> 211</span> : p(other.p), sz(other.sz)</div>
|
|
<div class="line"><a id="l00212" name="l00212"></a><span class="lineno"> 212</span> { other.p=<span class="keyword">nullptr</span>; other.sz=0; }</div>
|
|
</div>
|
|
<div class="line"><a id="l00213" name="l00213"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1arr.html#a8519487ee7fa95d3937a8352dd289b56"> 213</a></span> <a class="code hl_function" href="classpocketfft_1_1detail_1_1arr.html#a8519487ee7fa95d3937a8352dd289b56">~arr</a>() { dealloc(p); }</div>
|
|
<div class="line"><a id="l00214" name="l00214"></a><span class="lineno"> 214</span> </div>
|
|
<div class="foldopen" id="foldopen00215" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00215" name="l00215"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1arr.html#a8d73baaefa02dff8714e4398c83917e0"> 215</a></span> <span class="keywordtype">void</span> <a class="code hl_function" href="classpocketfft_1_1detail_1_1arr.html#a8d73baaefa02dff8714e4398c83917e0">resize</a>(<span class="keywordtype">size_t</span> n)</div>
|
|
<div class="line"><a id="l00216" name="l00216"></a><span class="lineno"> 216</span> {</div>
|
|
<div class="line"><a id="l00217" name="l00217"></a><span class="lineno"> 217</span> <span class="keywordflow">if</span> (n==sz) <span class="keywordflow">return</span>;</div>
|
|
<div class="line"><a id="l00218" name="l00218"></a><span class="lineno"> 218</span> dealloc(p);</div>
|
|
<div class="line"><a id="l00219" name="l00219"></a><span class="lineno"> 219</span> p = ralloc(n);</div>
|
|
<div class="line"><a id="l00220" name="l00220"></a><span class="lineno"> 220</span> sz = n;</div>
|
|
<div class="line"><a id="l00221" name="l00221"></a><span class="lineno"> 221</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l00222" name="l00222"></a><span class="lineno"> 222</span> </div>
|
|
<div class="line"><a id="l00223" name="l00223"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1arr.html#aea0bd899b19e03f54dfd6c188727061a"> 223</a></span> T &<a class="code hl_function" href="classpocketfft_1_1detail_1_1arr.html#aea0bd899b19e03f54dfd6c188727061a">operator[]</a>(<span class="keywordtype">size_t</span> idx) { <span class="keywordflow">return</span> p[idx]; }</div>
|
|
<div class="line"><a id="l00224" name="l00224"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1arr.html#a99c54f96bc79c7cdd8925c1663462842"> 224</a></span> <span class="keyword">const</span> T &<a class="code hl_function" href="classpocketfft_1_1detail_1_1arr.html#a99c54f96bc79c7cdd8925c1663462842">operator[]</a>(<span class="keywordtype">size_t</span> idx)<span class="keyword"> const </span>{ <span class="keywordflow">return</span> p[idx]; }</div>
|
|
<div class="line"><a id="l00225" name="l00225"></a><span class="lineno"> 225</span> </div>
|
|
<div class="line"><a id="l00226" name="l00226"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1arr.html#aec0f2191b4663b4187aab92454c34de8"> 226</a></span> T *<a class="code hl_function" href="classpocketfft_1_1detail_1_1arr.html#aec0f2191b4663b4187aab92454c34de8">data</a>() { <span class="keywordflow">return</span> p; }</div>
|
|
<div class="line"><a id="l00227" name="l00227"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1arr.html#ac82daa17e9f991072b012343f9d7c182"> 227</a></span> <span class="keyword">const</span> T *<a class="code hl_function" href="classpocketfft_1_1detail_1_1arr.html#ac82daa17e9f991072b012343f9d7c182">data</a>()<span class="keyword"> const </span>{ <span class="keywordflow">return</span> p; }</div>
|
|
<div class="line"><a id="l00228" name="l00228"></a><span class="lineno"> 228</span> </div>
|
|
<div class="line"><a id="l00229" name="l00229"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1arr.html#a95bca00060957f540ff25b69632c6952"> 229</a></span> <span class="keywordtype">size_t</span> <a class="code hl_function" href="classpocketfft_1_1detail_1_1arr.html#a95bca00060957f540ff25b69632c6952">size</a>()<span class="keyword"> const </span>{ <span class="keywordflow">return</span> sz; }</div>
|
|
<div class="line"><a id="l00230" name="l00230"></a><span class="lineno"> 230</span> };</div>
|
|
</div>
|
|
<div class="line"><a id="l00231" name="l00231"></a><span class="lineno"> 231</span> </div>
|
|
<div class="foldopen" id="foldopen00232" data-start="{" data-end="};">
|
|
<div class="line"><a id="l00232" name="l00232"></a><span class="lineno"><a class="line" href="structpocketfft_1_1detail_1_1cmplx.html"> 232</a></span><span class="keyword">template</span><<span class="keyword">typename</span> T> <span class="keyword">struct </span><a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cmplx</a> {</div>
|
|
<div class="line"><a id="l00233" name="l00233"></a><span class="lineno"><a class="line" href="structpocketfft_1_1detail_1_1cmplx.html#a35d2dce1b7de5f37d7029e639bc7f23d"> 233</a></span> T <a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a>, <a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#a35d2dce1b7de5f37d7029e639bc7f23d">i</a>;</div>
|
|
<div class="line"><a id="l00234" name="l00234"></a><span class="lineno"><a class="line" href="structpocketfft_1_1detail_1_1cmplx.html#a5b1ce506f1023f5254025ac81b831a2c"> 234</a></span> <a class="code hl_function" href="structpocketfft_1_1detail_1_1cmplx.html#a5b1ce506f1023f5254025ac81b831a2c">cmplx</a>() {}</div>
|
|
<div class="line"><a id="l00235" name="l00235"></a><span class="lineno"><a class="line" href="structpocketfft_1_1detail_1_1cmplx.html#a05491b4f1f22ca0bc49012f6a1c1710a"> 235</a></span> <a class="code hl_function" href="structpocketfft_1_1detail_1_1cmplx.html#a05491b4f1f22ca0bc49012f6a1c1710a">cmplx</a>(T r_, T i_) : <a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a>(r_), <a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#a35d2dce1b7de5f37d7029e639bc7f23d">i</a>(i_) {}</div>
|
|
<div class="line"><a id="l00236" name="l00236"></a><span class="lineno"><a class="line" href="structpocketfft_1_1detail_1_1cmplx.html#a647fece372b64b13c4a7e5877d09a807"> 236</a></span> <span class="keywordtype">void</span> <a class="code hl_function" href="structpocketfft_1_1detail_1_1cmplx.html#a647fece372b64b13c4a7e5877d09a807">Set</a>(T r_, T i_) { <a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a>=r_; <a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#a35d2dce1b7de5f37d7029e639bc7f23d">i</a>=i_; }</div>
|
|
<div class="line"><a id="l00237" name="l00237"></a><span class="lineno"><a class="line" href="structpocketfft_1_1detail_1_1cmplx.html#a447d26b2e07f6e45f29d865e906c0a98"> 237</a></span> <span class="keywordtype">void</span> <a class="code hl_function" href="structpocketfft_1_1detail_1_1cmplx.html#a447d26b2e07f6e45f29d865e906c0a98">Set</a>(T r_) { <a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a>=r_; <a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#a35d2dce1b7de5f37d7029e639bc7f23d">i</a>=T(0); }</div>
|
|
<div class="foldopen" id="foldopen00238" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00238" name="l00238"></a><span class="lineno"><a class="line" href="structpocketfft_1_1detail_1_1cmplx.html#ad4e69dcd89bdb7764c9c5807168f911e"> 238</a></span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cmplx</a> &<a class="code hl_function" href="structpocketfft_1_1detail_1_1cmplx.html#ad4e69dcd89bdb7764c9c5807168f911e">operator+= </a>(<span class="keyword">const</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cmplx</a> &other)</div>
|
|
<div class="line"><a id="l00239" name="l00239"></a><span class="lineno"> 239</span> { <a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a>+=other.<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a>; <a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#a35d2dce1b7de5f37d7029e639bc7f23d">i</a>+=other.<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#a35d2dce1b7de5f37d7029e639bc7f23d">i</a>; <span class="keywordflow">return</span> *<span class="keyword">this</span>; }</div>
|
|
</div>
|
|
<div class="foldopen" id="foldopen00240" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00240" name="l00240"></a><span class="lineno"><a class="line" href="structpocketfft_1_1detail_1_1cmplx.html#a683fd490182c9189fa2c05b1823edd93"> 240</a></span> <span class="keyword">template</span><<span class="keyword">typename</span> T2><a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cmplx</a> &<a class="code hl_function" href="structpocketfft_1_1detail_1_1cmplx.html#a683fd490182c9189fa2c05b1823edd93">operator*= </a>(T2 other)</div>
|
|
<div class="line"><a id="l00241" name="l00241"></a><span class="lineno"> 241</span> { <a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a>*=other; <a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#a35d2dce1b7de5f37d7029e639bc7f23d">i</a>*=other; <span class="keywordflow">return</span> *<span class="keyword">this</span>; }</div>
|
|
</div>
|
|
<div class="foldopen" id="foldopen00242" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00242" name="l00242"></a><span class="lineno"><a class="line" href="structpocketfft_1_1detail_1_1cmplx.html#a06f2c26c6fc4722e61b44da4c242ed87"> 242</a></span> <span class="keyword">template</span><<span class="keyword">typename</span> T2><a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cmplx</a> &<a class="code hl_function" href="structpocketfft_1_1detail_1_1cmplx.html#a683fd490182c9189fa2c05b1823edd93">operator*= </a>(<span class="keyword">const</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cmplx<T2></a> &other)</div>
|
|
<div class="line"><a id="l00243" name="l00243"></a><span class="lineno"> 243</span> {</div>
|
|
<div class="line"><a id="l00244" name="l00244"></a><span class="lineno"> 244</span> T tmp = <a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a>*other.<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a> - <a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#a35d2dce1b7de5f37d7029e639bc7f23d">i</a>*other.<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#a35d2dce1b7de5f37d7029e639bc7f23d">i</a>;</div>
|
|
<div class="line"><a id="l00245" name="l00245"></a><span class="lineno"> 245</span> <a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#a35d2dce1b7de5f37d7029e639bc7f23d">i</a> = <a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a>*other.<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#a35d2dce1b7de5f37d7029e639bc7f23d">i</a> + <a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#a35d2dce1b7de5f37d7029e639bc7f23d">i</a>*other.<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a>;</div>
|
|
<div class="line"><a id="l00246" name="l00246"></a><span class="lineno"> 246</span> <a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a> = tmp;</div>
|
|
<div class="line"><a id="l00247" name="l00247"></a><span class="lineno"> 247</span> <span class="keywordflow">return</span> *<span class="keyword">this</span>;</div>
|
|
<div class="line"><a id="l00248" name="l00248"></a><span class="lineno"> 248</span> }</div>
|
|
</div>
|
|
<div class="foldopen" id="foldopen00249" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00249" name="l00249"></a><span class="lineno"><a class="line" href="structpocketfft_1_1detail_1_1cmplx.html#affa618d8850a7c232793b7c61db6d184"> 249</a></span> <span class="keyword">template</span><<span class="keyword">typename</span> T2><a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cmplx</a> &<a class="code hl_function" href="structpocketfft_1_1detail_1_1cmplx.html#ad4e69dcd89bdb7764c9c5807168f911e">operator+= </a>(<span class="keyword">const</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cmplx<T2></a> &other)</div>
|
|
<div class="line"><a id="l00250" name="l00250"></a><span class="lineno"> 250</span> { <a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a>+=other.<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a>; <a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#a35d2dce1b7de5f37d7029e639bc7f23d">i</a>+=other.<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#a35d2dce1b7de5f37d7029e639bc7f23d">i</a>; <span class="keywordflow">return</span> *<span class="keyword">this</span>; }</div>
|
|
</div>
|
|
<div class="foldopen" id="foldopen00251" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00251" name="l00251"></a><span class="lineno"><a class="line" href="structpocketfft_1_1detail_1_1cmplx.html#a12441ff423274bd1b54245933d69ad7e"> 251</a></span> <span class="keyword">template</span><<span class="keyword">typename</span> T2><a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cmplx</a> &<a class="code hl_function" href="structpocketfft_1_1detail_1_1cmplx.html#a12441ff423274bd1b54245933d69ad7e">operator-= </a>(<span class="keyword">const</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cmplx<T2></a> &other)</div>
|
|
<div class="line"><a id="l00252" name="l00252"></a><span class="lineno"> 252</span> { <a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a>-=other.<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a>; <a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#a35d2dce1b7de5f37d7029e639bc7f23d">i</a>-=other.<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#a35d2dce1b7de5f37d7029e639bc7f23d">i</a>; <span class="keywordflow">return</span> *<span class="keyword">this</span>; }</div>
|
|
</div>
|
|
<div class="foldopen" id="foldopen00253" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00253" name="l00253"></a><span class="lineno"><a class="line" href="structpocketfft_1_1detail_1_1cmplx.html#a26bf3d709a58f06228e502af6db8e5ac"> 253</a></span> <span class="keyword">template</span><<span class="keyword">typename</span> T2> <span class="keyword">auto</span> <a class="code hl_function" href="structpocketfft_1_1detail_1_1cmplx.html#a26bf3d709a58f06228e502af6db8e5ac">operator* </a>(<span class="keyword">const</span> T2 &other) <span class="keyword">const</span></div>
|
|
<div class="line"><a id="l00254" name="l00254"></a><span class="lineno"> 254</span> -> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cmplx</a><<span class="keyword">decltype</span>(<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a>*other)></div>
|
|
<div class="line"><a id="l00255" name="l00255"></a><span class="lineno"> 255</span> { <span class="keywordflow">return</span> {<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a>*other, <a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#a35d2dce1b7de5f37d7029e639bc7f23d">i</a>*other}; }</div>
|
|
</div>
|
|
<div class="foldopen" id="foldopen00256" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00256" name="l00256"></a><span class="lineno"><a class="line" href="structpocketfft_1_1detail_1_1cmplx.html#a76447ef141c8732d57421749fc81b236"> 256</a></span> <span class="keyword">template</span><<span class="keyword">typename</span> T2> <span class="keyword">auto</span> <a class="code hl_function" href="structpocketfft_1_1detail_1_1cmplx.html#a76447ef141c8732d57421749fc81b236">operator+ </a>(<span class="keyword">const</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cmplx<T2></a> &other) <span class="keyword">const</span></div>
|
|
<div class="line"><a id="l00257" name="l00257"></a><span class="lineno"> 257</span> -> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cmplx</a><<span class="keyword">decltype</span>(<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a>+other.r)></div>
|
|
<div class="line"><a id="l00258" name="l00258"></a><span class="lineno"> 258</span> { <span class="keywordflow">return</span> {<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a>+other.<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a>, <a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#a35d2dce1b7de5f37d7029e639bc7f23d">i</a>+other.i}; }</div>
|
|
</div>
|
|
<div class="foldopen" id="foldopen00259" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00259" name="l00259"></a><span class="lineno"><a class="line" href="structpocketfft_1_1detail_1_1cmplx.html#a460da5db36d1c72fb1ed3496fd3abde4"> 259</a></span> <span class="keyword">template</span><<span class="keyword">typename</span> T2> <span class="keyword">auto</span> <a class="code hl_function" href="structpocketfft_1_1detail_1_1cmplx.html#a460da5db36d1c72fb1ed3496fd3abde4">operator- </a>(<span class="keyword">const</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cmplx<T2></a> &other) <span class="keyword">const</span></div>
|
|
<div class="line"><a id="l00260" name="l00260"></a><span class="lineno"> 260</span> -> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cmplx</a><<span class="keyword">decltype</span>(<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a>+other.r)></div>
|
|
<div class="line"><a id="l00261" name="l00261"></a><span class="lineno"> 261</span> { <span class="keywordflow">return</span> {<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a>-other.<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a>, <a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#a35d2dce1b7de5f37d7029e639bc7f23d">i</a>-other.i}; }</div>
|
|
</div>
|
|
<div class="foldopen" id="foldopen00262" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00262" name="l00262"></a><span class="lineno"><a class="line" href="structpocketfft_1_1detail_1_1cmplx.html#ad9c591ef8ae976293f207937d273e9a1"> 262</a></span> <span class="keyword">template</span><<span class="keyword">typename</span> T2> <span class="keyword">auto</span> <a class="code hl_function" href="structpocketfft_1_1detail_1_1cmplx.html#a26bf3d709a58f06228e502af6db8e5ac">operator* </a>(<span class="keyword">const</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cmplx<T2></a> &other) <span class="keyword">const</span></div>
|
|
<div class="line"><a id="l00263" name="l00263"></a><span class="lineno"> 263</span> -> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cmplx</a><<span class="keyword">decltype</span>(<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a>+other.r)></div>
|
|
<div class="line"><a id="l00264" name="l00264"></a><span class="lineno"> 264</span> { <span class="keywordflow">return</span> {<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a>*other.<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a>-<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#a35d2dce1b7de5f37d7029e639bc7f23d">i</a>*other.i, <a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a>*other.i + <a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#a35d2dce1b7de5f37d7029e639bc7f23d">i</a>*other.r}; }</div>
|
|
</div>
|
|
<div class="foldopen" id="foldopen00265" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00265" name="l00265"></a><span class="lineno"><a class="line" href="structpocketfft_1_1detail_1_1cmplx.html#a2e79f5c73c1d926361ad126cf57c8874"> 265</a></span> <span class="keyword">template</span><<span class="keywordtype">bool</span> fwd, <span class="keyword">typename</span> T2> <span class="keyword">auto</span> <a class="code hl_function" href="structpocketfft_1_1detail_1_1cmplx.html#a2e79f5c73c1d926361ad126cf57c8874">special_mul</a> (<span class="keyword">const</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cmplx<T2></a> &other) <span class="keyword">const</span></div>
|
|
<div class="line"><a id="l00266" name="l00266"></a><span class="lineno"> 266</span> -> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cmplx</a><<span class="keyword">decltype</span>(<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a>+other.r)></div>
|
|
<div class="line"><a id="l00267" name="l00267"></a><span class="lineno"> 267</span> {</div>
|
|
<div class="line"><a id="l00268" name="l00268"></a><span class="lineno"> 268</span> <span class="keyword">using </span>Tres = <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cmplx</a><<span class="keyword">decltype</span>(<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a>+other.r)>;</div>
|
|
<div class="line"><a id="l00269" name="l00269"></a><span class="lineno"> 269</span> <span class="keywordflow">return</span> fwd ? Tres(<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a>*other.r+<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#a35d2dce1b7de5f37d7029e639bc7f23d">i</a>*other.i, <a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#a35d2dce1b7de5f37d7029e639bc7f23d">i</a>*other.r-<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a>*other.i)</div>
|
|
<div class="line"><a id="l00270" name="l00270"></a><span class="lineno"> 270</span> : Tres(<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a>*other.r-<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#a35d2dce1b7de5f37d7029e639bc7f23d">i</a>*other.i, <a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a>*other.i+<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#a35d2dce1b7de5f37d7029e639bc7f23d">i</a>*other.r);</div>
|
|
<div class="line"><a id="l00271" name="l00271"></a><span class="lineno"> 271</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l00272" name="l00272"></a><span class="lineno"> 272</span>};</div>
|
|
</div>
|
|
<div class="foldopen" id="foldopen00273" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00273" name="l00273"></a><span class="lineno"><a class="line" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2"> 273</a></span><span class="keyword">template</span><<span class="keyword">typename</span> T> <span class="keyword">inline</span> <span class="keywordtype">void</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a>(T &a, T &b, T c, T d)</div>
|
|
<div class="line"><a id="l00274" name="l00274"></a><span class="lineno"> 274</span> { a=c+d; b=c-d; }</div>
|
|
</div>
|
|
<div class="foldopen" id="foldopen00275" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00275" name="l00275"></a><span class="lineno"><a class="line" href="namespacepocketfft_1_1detail.html#aa3477781ef54788c9fa7755cb8ea75fd"> 275</a></span><span class="keyword">template</span><<span class="keyword">typename</span> T> <span class="keyword">inline</span> <span class="keywordtype">void</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#aa3477781ef54788c9fa7755cb8ea75fd">PMINPLACE</a>(T &a, T &b)</div>
|
|
<div class="line"><a id="l00276" name="l00276"></a><span class="lineno"> 276</span> { T t = a; a+=b; b=t-b; }</div>
|
|
</div>
|
|
<div class="foldopen" id="foldopen00277" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00277" name="l00277"></a><span class="lineno"><a class="line" href="namespacepocketfft_1_1detail.html#af5eedf3cdfc83c0a30807092c39a9ce2"> 277</a></span><span class="keyword">template</span><<span class="keyword">typename</span> T> <span class="keyword">inline</span> <span class="keywordtype">void</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#af5eedf3cdfc83c0a30807092c39a9ce2">MPINPLACE</a>(T &a, T &b)</div>
|
|
<div class="line"><a id="l00278" name="l00278"></a><span class="lineno"> 278</span> { T t = a; a-=b; b=t+b; }</div>
|
|
</div>
|
|
<div class="foldopen" id="foldopen00279" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00279" name="l00279"></a><span class="lineno"><a class="line" href="namespacepocketfft_1_1detail.html#a66d79051d502046a9b9f103e744dbad3"> 279</a></span><span class="keyword">template</span><<span class="keyword">typename</span> T> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cmplx<T></a> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a66d79051d502046a9b9f103e744dbad3">conj</a>(<span class="keyword">const</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cmplx<T></a> &a)</div>
|
|
<div class="line"><a id="l00280" name="l00280"></a><span class="lineno"> 280</span> { <span class="keywordflow">return</span> {a.<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a>, -a.<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#a35d2dce1b7de5f37d7029e639bc7f23d">i</a>}; }</div>
|
|
</div>
|
|
<div class="foldopen" id="foldopen00281" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00281" name="l00281"></a><span class="lineno"><a class="line" href="namespacepocketfft_1_1detail.html#a8da1f3d4a0b712a0285529f24187fe76"> 281</a></span><span class="keyword">template</span><<span class="keywordtype">bool</span> fwd, <span class="keyword">typename</span> T, <span class="keyword">typename</span> T2> <span class="keywordtype">void</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8da1f3d4a0b712a0285529f24187fe76">special_mul</a> (<span class="keyword">const</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cmplx<T></a> &v1, <span class="keyword">const</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cmplx<T2></a> &v2, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cmplx<T></a> &res)</div>
|
|
<div class="line"><a id="l00282" name="l00282"></a><span class="lineno"> 282</span> {</div>
|
|
<div class="line"><a id="l00283" name="l00283"></a><span class="lineno"> 283</span> res = fwd ? <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cmplx<T></a>(v1.<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a>*v2.<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a>+v1.<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#a35d2dce1b7de5f37d7029e639bc7f23d">i</a>*v2.<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#a35d2dce1b7de5f37d7029e639bc7f23d">i</a>, v1.<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#a35d2dce1b7de5f37d7029e639bc7f23d">i</a>*v2.<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a>-v1.<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a>*v2.<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#a35d2dce1b7de5f37d7029e639bc7f23d">i</a>)</div>
|
|
<div class="line"><a id="l00284" name="l00284"></a><span class="lineno"> 284</span> : <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cmplx<T></a>(v1.<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a>*v2.<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a>-v1.<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#a35d2dce1b7de5f37d7029e639bc7f23d">i</a>*v2.<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#a35d2dce1b7de5f37d7029e639bc7f23d">i</a>, v1.<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a>*v2.<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#a35d2dce1b7de5f37d7029e639bc7f23d">i</a>+v1.<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#a35d2dce1b7de5f37d7029e639bc7f23d">i</a>*v2.<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a>);</div>
|
|
<div class="line"><a id="l00285" name="l00285"></a><span class="lineno"> 285</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l00286" name="l00286"></a><span class="lineno"> 286</span> </div>
|
|
<div class="foldopen" id="foldopen00287" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00287" name="l00287"></a><span class="lineno"><a class="line" href="namespacepocketfft_1_1detail.html#a928bad5278df636ee47402c0a75f64ef"> 287</a></span><span class="keyword">template</span><<span class="keyword">typename</span> T> <span class="keywordtype">void</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a928bad5278df636ee47402c0a75f64ef">ROT90</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cmplx<T></a> &<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">a</a>)</div>
|
|
<div class="line"><a id="l00288" name="l00288"></a><span class="lineno"> 288</span> { <span class="keyword">auto</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tmp_</a>=<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">a</a>.<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a>; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">a</a>.<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a>=-<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">a</a>.<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#a35d2dce1b7de5f37d7029e639bc7f23d">i</a>; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">a</a>.<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#a35d2dce1b7de5f37d7029e639bc7f23d">i</a>=<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tmp_</a>; }</div>
|
|
</div>
|
|
<div class="foldopen" id="foldopen00289" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00289" name="l00289"></a><span class="lineno"><a class="line" href="namespacepocketfft_1_1detail.html#ab6a43dc0cec4291e163e68a0875ac501"> 289</a></span><span class="keyword">template</span><<span class="keywordtype">bool</span> fwd, <span class="keyword">typename</span> T> <span class="keywordtype">void</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#ab6a43dc0cec4291e163e68a0875ac501">ROTX90</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cmplx<T></a> &<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">a</a>)</div>
|
|
<div class="line"><a id="l00290" name="l00290"></a><span class="lineno"> 290</span> { <span class="keyword">auto</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tmp_</a>= <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">fwd</a> ? -<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">a</a>.<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a> : <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">a</a>.<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a>; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">a</a>.<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a> = <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">fwd</a> ? <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">a</a>.<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#a35d2dce1b7de5f37d7029e639bc7f23d">i</a> : -<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">a</a>.<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#a35d2dce1b7de5f37d7029e639bc7f23d">i</a>; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">a</a>.<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#a35d2dce1b7de5f37d7029e639bc7f23d">i</a>=<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tmp_</a>; }</div>
|
|
</div>
|
|
<div class="line"><a id="l00291" name="l00291"></a><span class="lineno"> 291</span> </div>
|
|
<div class="line"><a id="l00292" name="l00292"></a><span class="lineno"> 292</span><span class="comment">//</span></div>
|
|
<div class="line"><a id="l00293" name="l00293"></a><span class="lineno"> 293</span><span class="comment">// twiddle factor section</span></div>
|
|
<div class="line"><a id="l00294" name="l00294"></a><span class="lineno"> 294</span><span class="comment">//</span></div>
|
|
<div class="foldopen" id="foldopen00295" data-start="{" data-end="};">
|
|
<div class="line"><a id="l00295" name="l00295"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1sincos__2pibyn.html"> 295</a></span><span class="keyword">template</span><<span class="keyword">typename</span> T> <span class="keyword">class </span><a class="code hl_class" href="classpocketfft_1_1detail_1_1sincos__2pibyn.html">sincos_2pibyn</a></div>
|
|
<div class="line"><a id="l00296" name="l00296"></a><span class="lineno"> 296</span> {</div>
|
|
<div class="line"><a id="l00297" name="l00297"></a><span class="lineno"> 297</span> <span class="keyword">private</span>:</div>
|
|
<div class="line"><a id="l00298" name="l00298"></a><span class="lineno"> 298</span> <span class="keyword">using </span>Thigh = <span class="keyword">typename</span> std::conditional<(<span class="keyword">sizeof</span>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a>)><span class="keyword">sizeof</span>(<span class="keywordtype">double</span>)), <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a>, <span class="keywordtype">double</span>>::type;</div>
|
|
<div class="line"><a id="l00299" name="l00299"></a><span class="lineno"> 299</span> <span class="keywordtype">size_t</span> N, mask, shift;</div>
|
|
<div class="line"><a id="l00300" name="l00300"></a><span class="lineno"> 300</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">arr<cmplx<Thigh></a>> v1, v2;</div>
|
|
<div class="line"><a id="l00301" name="l00301"></a><span class="lineno"> 301</span> </div>
|
|
<div class="line"><a id="l00302" name="l00302"></a><span class="lineno"> 302</span> <span class="keyword">static</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cmplx<Thigh></a> calc(<span class="keywordtype">size_t</span> x, <span class="keywordtype">size_t</span> n, Thigh <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ang</a>)</div>
|
|
<div class="line"><a id="l00303" name="l00303"></a><span class="lineno"> 303</span> {</div>
|
|
<div class="line"><a id="l00304" name="l00304"></a><span class="lineno"> 304</span> x<<=3;</div>
|
|
<div class="line"><a id="l00305" name="l00305"></a><span class="lineno"> 305</span> <span class="keywordflow">if</span> (x<4*n) <span class="comment">// first half</span></div>
|
|
<div class="line"><a id="l00306" name="l00306"></a><span class="lineno"> 306</span> {</div>
|
|
<div class="line"><a id="l00307" name="l00307"></a><span class="lineno"> 307</span> <span class="keywordflow">if</span> (x<2*n) <span class="comment">// first quadrant</span></div>
|
|
<div class="line"><a id="l00308" name="l00308"></a><span class="lineno"> 308</span> {</div>
|
|
<div class="line"><a id="l00309" name="l00309"></a><span class="lineno"> 309</span> <span class="keywordflow">if</span> (x<n) <span class="keywordflow">return</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cmplx<Thigh></a>(std::cos(Thigh(x)*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ang</a>), std::sin(Thigh(x)*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ang</a>));</div>
|
|
<div class="line"><a id="l00310" name="l00310"></a><span class="lineno"> 310</span> <span class="keywordflow">return</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cmplx<Thigh></a>(std::sin(Thigh(2*n-x)*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ang</a>), std::cos(Thigh(2*n-x)*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ang</a>));</div>
|
|
<div class="line"><a id="l00311" name="l00311"></a><span class="lineno"> 311</span> }</div>
|
|
<div class="line"><a id="l00312" name="l00312"></a><span class="lineno"> 312</span> <span class="keywordflow">else</span> <span class="comment">// second quadrant</span></div>
|
|
<div class="line"><a id="l00313" name="l00313"></a><span class="lineno"> 313</span> {</div>
|
|
<div class="line"><a id="l00314" name="l00314"></a><span class="lineno"> 314</span> x-=2*n;</div>
|
|
<div class="line"><a id="l00315" name="l00315"></a><span class="lineno"> 315</span> <span class="keywordflow">if</span> (x<n) <span class="keywordflow">return</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cmplx<Thigh></a>(-std::sin(Thigh(x)*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ang</a>), std::cos(Thigh(x)*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ang</a>));</div>
|
|
<div class="line"><a id="l00316" name="l00316"></a><span class="lineno"> 316</span> <span class="keywordflow">return</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cmplx<Thigh></a>(-std::cos(Thigh(2*n-x)*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ang</a>), std::sin(Thigh(2*n-x)*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ang</a>));</div>
|
|
<div class="line"><a id="l00317" name="l00317"></a><span class="lineno"> 317</span> }</div>
|
|
<div class="line"><a id="l00318" name="l00318"></a><span class="lineno"> 318</span> }</div>
|
|
<div class="line"><a id="l00319" name="l00319"></a><span class="lineno"> 319</span> <span class="keywordflow">else</span></div>
|
|
<div class="line"><a id="l00320" name="l00320"></a><span class="lineno"> 320</span> {</div>
|
|
<div class="line"><a id="l00321" name="l00321"></a><span class="lineno"> 321</span> x=8*n-x;</div>
|
|
<div class="line"><a id="l00322" name="l00322"></a><span class="lineno"> 322</span> <span class="keywordflow">if</span> (x<2*n) <span class="comment">// third quadrant</span></div>
|
|
<div class="line"><a id="l00323" name="l00323"></a><span class="lineno"> 323</span> {</div>
|
|
<div class="line"><a id="l00324" name="l00324"></a><span class="lineno"> 324</span> <span class="keywordflow">if</span> (x<n) <span class="keywordflow">return</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cmplx<Thigh></a>(std::cos(Thigh(x)*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ang</a>), -std::sin(Thigh(x)*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ang</a>));</div>
|
|
<div class="line"><a id="l00325" name="l00325"></a><span class="lineno"> 325</span> <span class="keywordflow">return</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cmplx<Thigh></a>(std::sin(Thigh(2*n-x)*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ang</a>), -std::cos(Thigh(2*n-x)*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ang</a>));</div>
|
|
<div class="line"><a id="l00326" name="l00326"></a><span class="lineno"> 326</span> }</div>
|
|
<div class="line"><a id="l00327" name="l00327"></a><span class="lineno"> 327</span> <span class="keywordflow">else</span> <span class="comment">// fourth quadrant</span></div>
|
|
<div class="line"><a id="l00328" name="l00328"></a><span class="lineno"> 328</span> {</div>
|
|
<div class="line"><a id="l00329" name="l00329"></a><span class="lineno"> 329</span> x-=2*n;</div>
|
|
<div class="line"><a id="l00330" name="l00330"></a><span class="lineno"> 330</span> <span class="keywordflow">if</span> (x<n) <span class="keywordflow">return</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cmplx<Thigh></a>(-std::sin(Thigh(x)*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ang</a>), -std::cos(Thigh(x)*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ang</a>));</div>
|
|
<div class="line"><a id="l00331" name="l00331"></a><span class="lineno"> 331</span> <span class="keywordflow">return</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cmplx<Thigh></a>(-std::cos(Thigh(2*n-x)*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ang</a>), -std::sin(Thigh(2*n-x)*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ang</a>));</div>
|
|
<div class="line"><a id="l00332" name="l00332"></a><span class="lineno"> 332</span> }</div>
|
|
<div class="line"><a id="l00333" name="l00333"></a><span class="lineno"> 333</span> }</div>
|
|
<div class="line"><a id="l00334" name="l00334"></a><span class="lineno"> 334</span> }</div>
|
|
<div class="line"><a id="l00335" name="l00335"></a><span class="lineno"> 335</span> </div>
|
|
<div class="line"><a id="l00336" name="l00336"></a><span class="lineno"> 336</span> <span class="keyword">public</span>:</div>
|
|
<div class="foldopen" id="foldopen00337" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00337" name="l00337"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1sincos__2pibyn.html#a88518f2182d854c557edacd4ab8cbc40"> 337</a></span> <a class="code hl_define" href="pocketfft_8h.html#a7020984e0ca1d6e565629ca6e7c1a7e0">POCKETFFT_NOINLINE</a> <a class="code hl_function" href="classpocketfft_1_1detail_1_1sincos__2pibyn.html#a88518f2182d854c557edacd4ab8cbc40">sincos_2pibyn</a>(<span class="keywordtype">size_t</span> n)</div>
|
|
<div class="line"><a id="l00338" name="l00338"></a><span class="lineno"> 338</span> : N(n)</div>
|
|
<div class="line"><a id="l00339" name="l00339"></a><span class="lineno"> 339</span> {</div>
|
|
<div class="line"><a id="l00340" name="l00340"></a><span class="lineno"> 340</span> <span class="keyword">constexpr</span> <span class="keyword">auto</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">pi</a> = 3.141592653589793238462643383279502884197L;</div>
|
|
<div class="line"><a id="l00341" name="l00341"></a><span class="lineno"> 341</span> Thigh <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ang</a> = Thigh(0.25L*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">pi</a>/n);</div>
|
|
<div class="line"><a id="l00342" name="l00342"></a><span class="lineno"> 342</span> <span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nval</a> = (n+2)/2;</div>
|
|
<div class="line"><a id="l00343" name="l00343"></a><span class="lineno"> 343</span> shift = 1;</div>
|
|
<div class="line"><a id="l00344" name="l00344"></a><span class="lineno"> 344</span> <span class="keywordflow">while</span>((<span class="keywordtype">size_t</span>(1)<<shift)*(size_t(1)<<shift) < <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nval</a>) ++shift;</div>
|
|
<div class="line"><a id="l00345" name="l00345"></a><span class="lineno"> 345</span> mask = (size_t(1)<<shift)-1;</div>
|
|
<div class="line"><a id="l00346" name="l00346"></a><span class="lineno"> 346</span> v1.resize(mask+1);</div>
|
|
<div class="line"><a id="l00347" name="l00347"></a><span class="lineno"> 347</span> v1[0].<a class="code hl_function" href="structpocketfft_1_1detail_1_1cmplx.html#a647fece372b64b13c4a7e5877d09a807">Set</a>(Thigh(1), Thigh(0));</div>
|
|
<div class="line"><a id="l00348" name="l00348"></a><span class="lineno"> 348</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i=1; i<v1.size(); ++i)</div>
|
|
<div class="line"><a id="l00349" name="l00349"></a><span class="lineno"> 349</span> v1[i]=calc(i,n,<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ang</a>);</div>
|
|
<div class="line"><a id="l00350" name="l00350"></a><span class="lineno"> 350</span> v2.resize((<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nval</a>+mask)/(mask+1));</div>
|
|
<div class="line"><a id="l00351" name="l00351"></a><span class="lineno"> 351</span> v2[0].<a class="code hl_function" href="structpocketfft_1_1detail_1_1cmplx.html#a647fece372b64b13c4a7e5877d09a807">Set</a>(Thigh(1), Thigh(0));</div>
|
|
<div class="line"><a id="l00352" name="l00352"></a><span class="lineno"> 352</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i=1; i<v2.size(); ++i)</div>
|
|
<div class="line"><a id="l00353" name="l00353"></a><span class="lineno"> 353</span> v2[i]=calc(i*(mask+1),n,<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ang</a>);</div>
|
|
<div class="line"><a id="l00354" name="l00354"></a><span class="lineno"> 354</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l00355" name="l00355"></a><span class="lineno"> 355</span> </div>
|
|
<div class="foldopen" id="foldopen00356" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00356" name="l00356"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1sincos__2pibyn.html#a71b02f67c47b24adb296eafd2c7a3598"> 356</a></span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cmplx<T></a> <a class="code hl_function" href="classpocketfft_1_1detail_1_1sincos__2pibyn.html#a71b02f67c47b24adb296eafd2c7a3598">operator[]</a>(<span class="keywordtype">size_t</span> idx)<span class="keyword"> const</span></div>
|
|
<div class="line"><a id="l00357" name="l00357"></a><span class="lineno"> 357</span><span class="keyword"> </span>{</div>
|
|
<div class="line"><a id="l00358" name="l00358"></a><span class="lineno"> 358</span> <span class="keywordflow">if</span> (2*idx<=N)</div>
|
|
<div class="line"><a id="l00359" name="l00359"></a><span class="lineno"> 359</span> {</div>
|
|
<div class="line"><a id="l00360" name="l00360"></a><span class="lineno"> 360</span> <span class="keyword">auto</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">x1</a>=v1[idx&mask], <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">x2</a>=v2[idx>>shift];</div>
|
|
<div class="line"><a id="l00361" name="l00361"></a><span class="lineno"> 361</span> <span class="keywordflow">return</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cmplx<T></a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">x1</a>.<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a>*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">x2</a>.<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a>-<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">x1</a>.<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#a35d2dce1b7de5f37d7029e639bc7f23d">i</a>*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">x2</a>.<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#a35d2dce1b7de5f37d7029e639bc7f23d">i</a>), <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">x1</a>.<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a>*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">x2</a>.<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#a35d2dce1b7de5f37d7029e639bc7f23d">i</a>+<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">x1</a>.<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#a35d2dce1b7de5f37d7029e639bc7f23d">i</a>*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">x2</a>.<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a>));</div>
|
|
<div class="line"><a id="l00362" name="l00362"></a><span class="lineno"> 362</span> }</div>
|
|
<div class="line"><a id="l00363" name="l00363"></a><span class="lineno"> 363</span> idx = N-idx;</div>
|
|
<div class="line"><a id="l00364" name="l00364"></a><span class="lineno"> 364</span> <span class="keyword">auto</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">x1</a>=v1[idx&mask], <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">x2</a>=v2[idx>>shift];</div>
|
|
<div class="line"><a id="l00365" name="l00365"></a><span class="lineno"> 365</span> <span class="keywordflow">return</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cmplx<T></a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">x1</a>.<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a>*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">x2</a>.<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a>-<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">x1</a>.<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#a35d2dce1b7de5f37d7029e639bc7f23d">i</a>*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">x2</a>.<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#a35d2dce1b7de5f37d7029e639bc7f23d">i</a>), -<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">x1</a>.<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a>*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">x2</a>.<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#a35d2dce1b7de5f37d7029e639bc7f23d">i</a>+<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">x1</a>.<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#a35d2dce1b7de5f37d7029e639bc7f23d">i</a>*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">x2</a>.<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a>));</div>
|
|
<div class="line"><a id="l00366" name="l00366"></a><span class="lineno"> 366</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l00367" name="l00367"></a><span class="lineno"> 367</span> };</div>
|
|
</div>
|
|
<div class="line"><a id="l00368" name="l00368"></a><span class="lineno"> 368</span> </div>
|
|
<div class="foldopen" id="foldopen00369" data-start="{" data-end="};">
|
|
<div class="line"><a id="l00369" name="l00369"></a><span class="lineno"><a class="line" href="structpocketfft_1_1detail_1_1util.html"> 369</a></span><span class="keyword">struct </span><a class="code hl_struct" href="structpocketfft_1_1detail_1_1util.html">util</a> <span class="comment">// hack to avoid duplicate symbols</span></div>
|
|
<div class="line"><a id="l00370" name="l00370"></a><span class="lineno"> 370</span> {</div>
|
|
<div class="foldopen" id="foldopen00371" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00371" name="l00371"></a><span class="lineno"><a class="line" href="structpocketfft_1_1detail_1_1util.html#a0e9b92280f25063d1ebef942dd0d233c"> 371</a></span> <span class="keyword">static</span> <a class="code hl_define" href="pocketfft_8h.html#a7020984e0ca1d6e565629ca6e7c1a7e0">POCKETFFT_NOINLINE</a> <span class="keywordtype">size_t</span> <a class="code hl_function" href="structpocketfft_1_1detail_1_1util.html#a0e9b92280f25063d1ebef942dd0d233c">largest_prime_factor</a> (<span class="keywordtype">size_t</span> n)</div>
|
|
<div class="line"><a id="l00372" name="l00372"></a><span class="lineno"> 372</span> {</div>
|
|
<div class="line"><a id="l00373" name="l00373"></a><span class="lineno"> 373</span> <span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">res</a>=1;</div>
|
|
<div class="line"><a id="l00374" name="l00374"></a><span class="lineno"> 374</span> <span class="keywordflow">while</span> ((n&1)==0)</div>
|
|
<div class="line"><a id="l00375" name="l00375"></a><span class="lineno"> 375</span> { <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">res</a>=2; n>>=1; }</div>
|
|
<div class="line"><a id="l00376" name="l00376"></a><span class="lineno"> 376</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> x=3; x*x<=n; x+=2)</div>
|
|
<div class="line"><a id="l00377" name="l00377"></a><span class="lineno"> 377</span> <span class="keywordflow">while</span> ((n%x)==0)</div>
|
|
<div class="line"><a id="l00378" name="l00378"></a><span class="lineno"> 378</span> { <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">res</a>=x; n/=x; }</div>
|
|
<div class="line"><a id="l00379" name="l00379"></a><span class="lineno"> 379</span> <span class="keywordflow">if</span> (n>1) <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">res</a>=n;</div>
|
|
<div class="line"><a id="l00380" name="l00380"></a><span class="lineno"> 380</span> <span class="keywordflow">return</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">res</a>;</div>
|
|
<div class="line"><a id="l00381" name="l00381"></a><span class="lineno"> 381</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l00382" name="l00382"></a><span class="lineno"> 382</span> </div>
|
|
<div class="foldopen" id="foldopen00383" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00383" name="l00383"></a><span class="lineno"><a class="line" href="structpocketfft_1_1detail_1_1util.html#ad3d874bc3fb0048df2270779a15d4bd0"> 383</a></span> <span class="keyword">static</span> <a class="code hl_define" href="pocketfft_8h.html#a7020984e0ca1d6e565629ca6e7c1a7e0">POCKETFFT_NOINLINE</a> <span class="keywordtype">double</span> <a class="code hl_function" href="structpocketfft_1_1detail_1_1util.html#ad3d874bc3fb0048df2270779a15d4bd0">cost_guess</a> (<span class="keywordtype">size_t</span> n)</div>
|
|
<div class="line"><a id="l00384" name="l00384"></a><span class="lineno"> 384</span> {</div>
|
|
<div class="line"><a id="l00385" name="l00385"></a><span class="lineno"> 385</span> <span class="keyword">constexpr</span> <span class="keywordtype">double</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">lfp</a>=1.1; <span class="comment">// penalty for non-hardcoded larger factors</span></div>
|
|
<div class="line"><a id="l00386" name="l00386"></a><span class="lineno"> 386</span> <span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ni</a>=n;</div>
|
|
<div class="line"><a id="l00387" name="l00387"></a><span class="lineno"> 387</span> <span class="keywordtype">double</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">result</a>=0.;</div>
|
|
<div class="line"><a id="l00388" name="l00388"></a><span class="lineno"> 388</span> <span class="keywordflow">while</span> ((n&1)==0)</div>
|
|
<div class="line"><a id="l00389" name="l00389"></a><span class="lineno"> 389</span> { <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">result</a>+=2; n>>=1; }</div>
|
|
<div class="line"><a id="l00390" name="l00390"></a><span class="lineno"> 390</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> x=3; x*x<=n; x+=2)</div>
|
|
<div class="line"><a id="l00391" name="l00391"></a><span class="lineno"> 391</span> <span class="keywordflow">while</span> ((n%x)==0)</div>
|
|
<div class="line"><a id="l00392" name="l00392"></a><span class="lineno"> 392</span> {</div>
|
|
<div class="line"><a id="l00393" name="l00393"></a><span class="lineno"> 393</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">result</a>+= (x<=5) ? <span class="keywordtype">double</span>(x) : <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">lfp</a>*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">double</a>(x); <span class="comment">// penalize larger prime factors</span></div>
|
|
<div class="line"><a id="l00394" name="l00394"></a><span class="lineno"> 394</span> n/=x;</div>
|
|
<div class="line"><a id="l00395" name="l00395"></a><span class="lineno"> 395</span> }</div>
|
|
<div class="line"><a id="l00396" name="l00396"></a><span class="lineno"> 396</span> <span class="keywordflow">if</span> (n>1) <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">result</a>+=(n<=5) ? <span class="keywordtype">double</span>(n) : <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">lfp</a>*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">double</a>(n);</div>
|
|
<div class="line"><a id="l00397" name="l00397"></a><span class="lineno"> 397</span> <span class="keywordflow">return</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">result</a>*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">double</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ni</a>);</div>
|
|
<div class="line"><a id="l00398" name="l00398"></a><span class="lineno"> 398</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l00399" name="l00399"></a><span class="lineno"> 399</span> </div>
|
|
<div class="line"><a id="l00400" name="l00400"></a><span class="lineno"> 400</span> <span class="comment">/* returns the smallest composite of 2, 3, 5, 7 and 11 which is >= n */</span></div>
|
|
<div class="foldopen" id="foldopen00401" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00401" name="l00401"></a><span class="lineno"><a class="line" href="structpocketfft_1_1detail_1_1util.html#a758e00d242a1b7eda8f9f0c21f35c624"> 401</a></span> <span class="keyword">static</span> <a class="code hl_define" href="pocketfft_8h.html#a7020984e0ca1d6e565629ca6e7c1a7e0">POCKETFFT_NOINLINE</a> <span class="keywordtype">size_t</span> <a class="code hl_function" href="structpocketfft_1_1detail_1_1util.html#a758e00d242a1b7eda8f9f0c21f35c624">good_size_cmplx</a>(<span class="keywordtype">size_t</span> n)</div>
|
|
<div class="line"><a id="l00402" name="l00402"></a><span class="lineno"> 402</span> {</div>
|
|
<div class="line"><a id="l00403" name="l00403"></a><span class="lineno"> 403</span> <span class="keywordflow">if</span> (n<=12) <span class="keywordflow">return</span> n;</div>
|
|
<div class="line"><a id="l00404" name="l00404"></a><span class="lineno"> 404</span> </div>
|
|
<div class="line"><a id="l00405" name="l00405"></a><span class="lineno"> 405</span> <span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">bestfac</a>=2*n;</div>
|
|
<div class="line"><a id="l00406" name="l00406"></a><span class="lineno"> 406</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">f11</a>=1; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">f11</a><<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">bestfac</a>; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">f11</a>*=11)</div>
|
|
<div class="line"><a id="l00407" name="l00407"></a><span class="lineno"> 407</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">f117</a>=<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">f11</a>; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">f117</a><<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">bestfac</a>; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">f117</a>*=7)</div>
|
|
<div class="line"><a id="l00408" name="l00408"></a><span class="lineno"> 408</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">f1175</a>=<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">f117</a>; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">f1175</a><<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">bestfac</a>; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">f1175</a>*=5)</div>
|
|
<div class="line"><a id="l00409" name="l00409"></a><span class="lineno"> 409</span> {</div>
|
|
<div class="line"><a id="l00410" name="l00410"></a><span class="lineno"> 410</span> <span class="keywordtype">size_t</span> x=<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">f1175</a>;</div>
|
|
<div class="line"><a id="l00411" name="l00411"></a><span class="lineno"> 411</span> <span class="keywordflow">while</span> (x<n) x*=2;</div>
|
|
<div class="line"><a id="l00412" name="l00412"></a><span class="lineno"> 412</span> <span class="keywordflow">for</span> (;;)</div>
|
|
<div class="line"><a id="l00413" name="l00413"></a><span class="lineno"> 413</span> {</div>
|
|
<div class="line"><a id="l00414" name="l00414"></a><span class="lineno"> 414</span> <span class="keywordflow">if</span> (x<n)</div>
|
|
<div class="line"><a id="l00415" name="l00415"></a><span class="lineno"> 415</span> x*=3;</div>
|
|
<div class="line"><a id="l00416" name="l00416"></a><span class="lineno"> 416</span> <span class="keywordflow">else</span> <span class="keywordflow">if</span> (x>n)</div>
|
|
<div class="line"><a id="l00417" name="l00417"></a><span class="lineno"> 417</span> {</div>
|
|
<div class="line"><a id="l00418" name="l00418"></a><span class="lineno"> 418</span> <span class="keywordflow">if</span> (x<<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">bestfac</a>) <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">bestfac</a>=x;</div>
|
|
<div class="line"><a id="l00419" name="l00419"></a><span class="lineno"> 419</span> <span class="keywordflow">if</span> (x&1) <span class="keywordflow">break</span>;</div>
|
|
<div class="line"><a id="l00420" name="l00420"></a><span class="lineno"> 420</span> x>>=1;</div>
|
|
<div class="line"><a id="l00421" name="l00421"></a><span class="lineno"> 421</span> }</div>
|
|
<div class="line"><a id="l00422" name="l00422"></a><span class="lineno"> 422</span> <span class="keywordflow">else</span></div>
|
|
<div class="line"><a id="l00423" name="l00423"></a><span class="lineno"> 423</span> <span class="keywordflow">return</span> n;</div>
|
|
<div class="line"><a id="l00424" name="l00424"></a><span class="lineno"> 424</span> }</div>
|
|
<div class="line"><a id="l00425" name="l00425"></a><span class="lineno"> 425</span> }</div>
|
|
<div class="line"><a id="l00426" name="l00426"></a><span class="lineno"> 426</span> <span class="keywordflow">return</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">bestfac</a>;</div>
|
|
<div class="line"><a id="l00427" name="l00427"></a><span class="lineno"> 427</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l00428" name="l00428"></a><span class="lineno"> 428</span> </div>
|
|
<div class="line"><a id="l00429" name="l00429"></a><span class="lineno"> 429</span> <span class="comment">/* returns the smallest composite of 2, 3, 5 which is >= n */</span></div>
|
|
<div class="foldopen" id="foldopen00430" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00430" name="l00430"></a><span class="lineno"><a class="line" href="structpocketfft_1_1detail_1_1util.html#a173da7d5929ded86fffcebcfdc5086aa"> 430</a></span> <span class="keyword">static</span> <a class="code hl_define" href="pocketfft_8h.html#a7020984e0ca1d6e565629ca6e7c1a7e0">POCKETFFT_NOINLINE</a> <span class="keywordtype">size_t</span> <a class="code hl_function" href="structpocketfft_1_1detail_1_1util.html#a173da7d5929ded86fffcebcfdc5086aa">good_size_real</a>(<span class="keywordtype">size_t</span> n)</div>
|
|
<div class="line"><a id="l00431" name="l00431"></a><span class="lineno"> 431</span> {</div>
|
|
<div class="line"><a id="l00432" name="l00432"></a><span class="lineno"> 432</span> <span class="keywordflow">if</span> (n<=6) <span class="keywordflow">return</span> n;</div>
|
|
<div class="line"><a id="l00433" name="l00433"></a><span class="lineno"> 433</span> </div>
|
|
<div class="line"><a id="l00434" name="l00434"></a><span class="lineno"> 434</span> <span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">bestfac</a>=2*n;</div>
|
|
<div class="line"><a id="l00435" name="l00435"></a><span class="lineno"> 435</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">f5</a>=1; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">f5</a><<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">bestfac</a>; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">f5</a>*=5)</div>
|
|
<div class="line"><a id="l00436" name="l00436"></a><span class="lineno"> 436</span> {</div>
|
|
<div class="line"><a id="l00437" name="l00437"></a><span class="lineno"> 437</span> <span class="keywordtype">size_t</span> x = <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">f5</a>;</div>
|
|
<div class="line"><a id="l00438" name="l00438"></a><span class="lineno"> 438</span> <span class="keywordflow">while</span> (x<n) x *= 2;</div>
|
|
<div class="line"><a id="l00439" name="l00439"></a><span class="lineno"> 439</span> <span class="keywordflow">for</span> (;;)</div>
|
|
<div class="line"><a id="l00440" name="l00440"></a><span class="lineno"> 440</span> {</div>
|
|
<div class="line"><a id="l00441" name="l00441"></a><span class="lineno"> 441</span> <span class="keywordflow">if</span> (x<n)</div>
|
|
<div class="line"><a id="l00442" name="l00442"></a><span class="lineno"> 442</span> x*=3;</div>
|
|
<div class="line"><a id="l00443" name="l00443"></a><span class="lineno"> 443</span> <span class="keywordflow">else</span> <span class="keywordflow">if</span> (x>n)</div>
|
|
<div class="line"><a id="l00444" name="l00444"></a><span class="lineno"> 444</span> {</div>
|
|
<div class="line"><a id="l00445" name="l00445"></a><span class="lineno"> 445</span> <span class="keywordflow">if</span> (x<<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">bestfac</a>) <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">bestfac</a>=x;</div>
|
|
<div class="line"><a id="l00446" name="l00446"></a><span class="lineno"> 446</span> <span class="keywordflow">if</span> (x&1) <span class="keywordflow">break</span>;</div>
|
|
<div class="line"><a id="l00447" name="l00447"></a><span class="lineno"> 447</span> x>>=1;</div>
|
|
<div class="line"><a id="l00448" name="l00448"></a><span class="lineno"> 448</span> }</div>
|
|
<div class="line"><a id="l00449" name="l00449"></a><span class="lineno"> 449</span> <span class="keywordflow">else</span></div>
|
|
<div class="line"><a id="l00450" name="l00450"></a><span class="lineno"> 450</span> <span class="keywordflow">return</span> n;</div>
|
|
<div class="line"><a id="l00451" name="l00451"></a><span class="lineno"> 451</span> }</div>
|
|
<div class="line"><a id="l00452" name="l00452"></a><span class="lineno"> 452</span> }</div>
|
|
<div class="line"><a id="l00453" name="l00453"></a><span class="lineno"> 453</span> <span class="keywordflow">return</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">bestfac</a>;</div>
|
|
<div class="line"><a id="l00454" name="l00454"></a><span class="lineno"> 454</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l00455" name="l00455"></a><span class="lineno"> 455</span> </div>
|
|
<div class="foldopen" id="foldopen00456" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00456" name="l00456"></a><span class="lineno"><a class="line" href="structpocketfft_1_1detail_1_1util.html#a33acae07b20b28fe4658bc338fce1b89"> 456</a></span> <span class="keyword">static</span> <span class="keywordtype">size_t</span> <a class="code hl_function" href="structpocketfft_1_1detail_1_1util.html#a33acae07b20b28fe4658bc338fce1b89">prod</a>(<span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#a885ee37fcf564a268a5c8ca9ea8603e1">shape_t</a> &shape)</div>
|
|
<div class="line"><a id="l00457" name="l00457"></a><span class="lineno"> 457</span> {</div>
|
|
<div class="line"><a id="l00458" name="l00458"></a><span class="lineno"> 458</span> <span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">res</a>=1;</div>
|
|
<div class="line"><a id="l00459" name="l00459"></a><span class="lineno"> 459</span> <span class="keywordflow">for</span> (<span class="keyword">auto</span> sz: shape)</div>
|
|
<div class="line"><a id="l00460" name="l00460"></a><span class="lineno"> 460</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">res</a>*=sz;</div>
|
|
<div class="line"><a id="l00461" name="l00461"></a><span class="lineno"> 461</span> <span class="keywordflow">return</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">res</a>;</div>
|
|
<div class="line"><a id="l00462" name="l00462"></a><span class="lineno"> 462</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l00463" name="l00463"></a><span class="lineno"> 463</span> </div>
|
|
<div class="foldopen" id="foldopen00464" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00464" name="l00464"></a><span class="lineno"><a class="line" href="structpocketfft_1_1detail_1_1util.html#a3d2f5f00d4fed5c09bd1509ffa9a44cd"> 464</a></span> <span class="keyword">static</span> <a class="code hl_define" href="pocketfft_8h.html#a7020984e0ca1d6e565629ca6e7c1a7e0">POCKETFFT_NOINLINE</a> <span class="keywordtype">void</span> <a class="code hl_function" href="structpocketfft_1_1detail_1_1util.html#a3d2f5f00d4fed5c09bd1509ffa9a44cd">sanity_check</a>(<span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#a885ee37fcf564a268a5c8ca9ea8603e1">shape_t</a> &shape,</div>
|
|
<div class="line"><a id="l00465" name="l00465"></a><span class="lineno"> 465</span> <span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#afb987c919e9424a996d0fc8b3c23cc84">stride_t</a> &stride_in, <span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#afb987c919e9424a996d0fc8b3c23cc84">stride_t</a> &stride_out, <span class="keywordtype">bool</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">inplace</a>)</div>
|
|
<div class="line"><a id="l00466" name="l00466"></a><span class="lineno"> 466</span> {</div>
|
|
<div class="line"><a id="l00467" name="l00467"></a><span class="lineno"> 467</span> <span class="keyword">auto</span> ndim = shape.size();</div>
|
|
<div class="line"><a id="l00468" name="l00468"></a><span class="lineno"> 468</span> <span class="keywordflow">if</span> (ndim<1) <span class="keywordflow">throw</span> std::runtime_error(<span class="stringliteral">"ndim must be >= 1"</span>);</div>
|
|
<div class="line"><a id="l00469" name="l00469"></a><span class="lineno"> 469</span> <span class="keywordflow">if</span> ((stride_in.size()!=ndim) || (stride_out.size()!=ndim))</div>
|
|
<div class="line"><a id="l00470" name="l00470"></a><span class="lineno"> 470</span> <span class="keywordflow">throw</span> std::runtime_error(<span class="stringliteral">"stride dimension mismatch"</span>);</div>
|
|
<div class="line"><a id="l00471" name="l00471"></a><span class="lineno"> 471</span> <span class="keywordflow">if</span> (<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">inplace</a> && (stride_in!=stride_out))</div>
|
|
<div class="line"><a id="l00472" name="l00472"></a><span class="lineno"> 472</span> <span class="keywordflow">throw</span> std::runtime_error(<span class="stringliteral">"stride mismatch"</span>);</div>
|
|
<div class="line"><a id="l00473" name="l00473"></a><span class="lineno"> 473</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l00474" name="l00474"></a><span class="lineno"> 474</span> </div>
|
|
<div class="foldopen" id="foldopen00475" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00475" name="l00475"></a><span class="lineno"><a class="line" href="structpocketfft_1_1detail_1_1util.html#a280945ba7af964102b1d0d1d83469130"> 475</a></span> <span class="keyword">static</span> <a class="code hl_define" href="pocketfft_8h.html#a7020984e0ca1d6e565629ca6e7c1a7e0">POCKETFFT_NOINLINE</a> <span class="keywordtype">void</span> <a class="code hl_function" href="structpocketfft_1_1detail_1_1util.html#a280945ba7af964102b1d0d1d83469130">sanity_check</a>(<span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#a885ee37fcf564a268a5c8ca9ea8603e1">shape_t</a> &shape,</div>
|
|
<div class="line"><a id="l00476" name="l00476"></a><span class="lineno"> 476</span> <span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#afb987c919e9424a996d0fc8b3c23cc84">stride_t</a> &stride_in, <span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#afb987c919e9424a996d0fc8b3c23cc84">stride_t</a> &stride_out, <span class="keywordtype">bool</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">inplace</a>,</div>
|
|
<div class="line"><a id="l00477" name="l00477"></a><span class="lineno"> 477</span> <span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#a885ee37fcf564a268a5c8ca9ea8603e1">shape_t</a> &<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axes</a>)</div>
|
|
<div class="line"><a id="l00478" name="l00478"></a><span class="lineno"> 478</span> {</div>
|
|
<div class="line"><a id="l00479" name="l00479"></a><span class="lineno"> 479</span> <a class="code hl_function" href="structpocketfft_1_1detail_1_1util.html#a3d2f5f00d4fed5c09bd1509ffa9a44cd">sanity_check</a>(shape, stride_in, stride_out, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">inplace</a>);</div>
|
|
<div class="line"><a id="l00480" name="l00480"></a><span class="lineno"> 480</span> <span class="keyword">auto</span> ndim = shape.size();</div>
|
|
<div class="line"><a id="l00481" name="l00481"></a><span class="lineno"> 481</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#a885ee37fcf564a268a5c8ca9ea8603e1">shape_t</a> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tmp</a>(ndim,0);</div>
|
|
<div class="line"><a id="l00482" name="l00482"></a><span class="lineno"> 482</span> <span class="keywordflow">for</span> (<span class="keyword">auto</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ax</a> : <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axes</a>)</div>
|
|
<div class="line"><a id="l00483" name="l00483"></a><span class="lineno"> 483</span> {</div>
|
|
<div class="line"><a id="l00484" name="l00484"></a><span class="lineno"> 484</span> <span class="keywordflow">if</span> (<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ax</a>>=ndim) <span class="keywordflow">throw</span> std::invalid_argument(<span class="stringliteral">"bad axis number"</span>);</div>
|
|
<div class="line"><a id="l00485" name="l00485"></a><span class="lineno"> 485</span> <span class="keywordflow">if</span> (++<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tmp</a>[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ax</a>]>1) <span class="keywordflow">throw</span> std::invalid_argument(<span class="stringliteral">"axis specified repeatedly"</span>);</div>
|
|
<div class="line"><a id="l00486" name="l00486"></a><span class="lineno"> 486</span> }</div>
|
|
<div class="line"><a id="l00487" name="l00487"></a><span class="lineno"> 487</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l00488" name="l00488"></a><span class="lineno"> 488</span> </div>
|
|
<div class="foldopen" id="foldopen00489" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00489" name="l00489"></a><span class="lineno"><a class="line" href="structpocketfft_1_1detail_1_1util.html#a670e044dadbbf680537e1fb11bc8b724"> 489</a></span> <span class="keyword">static</span> <a class="code hl_define" href="pocketfft_8h.html#a7020984e0ca1d6e565629ca6e7c1a7e0">POCKETFFT_NOINLINE</a> <span class="keywordtype">void</span> <a class="code hl_function" href="structpocketfft_1_1detail_1_1util.html#a670e044dadbbf680537e1fb11bc8b724">sanity_check</a>(<span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#a885ee37fcf564a268a5c8ca9ea8603e1">shape_t</a> &shape,</div>
|
|
<div class="line"><a id="l00490" name="l00490"></a><span class="lineno"> 490</span> <span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#afb987c919e9424a996d0fc8b3c23cc84">stride_t</a> &stride_in, <span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#afb987c919e9424a996d0fc8b3c23cc84">stride_t</a> &stride_out, <span class="keywordtype">bool</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">inplace</a>,</div>
|
|
<div class="line"><a id="l00491" name="l00491"></a><span class="lineno"> 491</span> <span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axis</a>)</div>
|
|
<div class="line"><a id="l00492" name="l00492"></a><span class="lineno"> 492</span> {</div>
|
|
<div class="line"><a id="l00493" name="l00493"></a><span class="lineno"> 493</span> <a class="code hl_function" href="structpocketfft_1_1detail_1_1util.html#a3d2f5f00d4fed5c09bd1509ffa9a44cd">sanity_check</a>(shape, stride_in, stride_out, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">inplace</a>);</div>
|
|
<div class="line"><a id="l00494" name="l00494"></a><span class="lineno"> 494</span> <span class="keywordflow">if</span> (<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axis</a>>=shape.size()) <span class="keywordflow">throw</span> std::invalid_argument(<span class="stringliteral">"bad axis number"</span>);</div>
|
|
<div class="line"><a id="l00495" name="l00495"></a><span class="lineno"> 495</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l00496" name="l00496"></a><span class="lineno"> 496</span> </div>
|
|
<div class="line"><a id="l00497" name="l00497"></a><span class="lineno"> 497</span><span class="preprocessor">#ifdef POCKETFFT_NO_MULTITHREADING</span></div>
|
|
<div class="line"><a id="l00498" name="l00498"></a><span class="lineno"> 498</span> <span class="keyword">static</span> <span class="keywordtype">size_t</span> <a class="code hl_function" href="structpocketfft_1_1detail_1_1util.html#a3b012d5a19215bcd32cf6e228556fa87">thread_count</a> (<span class="keywordtype">size_t</span> <span class="comment">/*nthreads*/</span>, <span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#a885ee37fcf564a268a5c8ca9ea8603e1">shape_t</a> &<span class="comment">/*shape*/</span>,</div>
|
|
<div class="line"><a id="l00499" name="l00499"></a><span class="lineno"> 499</span> <span class="keywordtype">size_t</span> <span class="comment">/*axis*/</span>, <span class="keywordtype">size_t</span> <span class="comment">/*vlen*/</span>)</div>
|
|
<div class="line"><a id="l00500" name="l00500"></a><span class="lineno"> 500</span> { <span class="keywordflow">return</span> 1; }</div>
|
|
<div class="line"><a id="l00501" name="l00501"></a><span class="lineno"> 501</span><span class="preprocessor">#else</span></div>
|
|
<div class="foldopen" id="foldopen00502" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00502" name="l00502"></a><span class="lineno"><a class="line" href="structpocketfft_1_1detail_1_1util.html#a3b012d5a19215bcd32cf6e228556fa87"> 502</a></span> <span class="keyword">static</span> <span class="keywordtype">size_t</span> <a class="code hl_function" href="structpocketfft_1_1detail_1_1util.html#a3b012d5a19215bcd32cf6e228556fa87">thread_count</a> (<span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nthreads</a>, <span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#a885ee37fcf564a268a5c8ca9ea8603e1">shape_t</a> &shape,</div>
|
|
<div class="line"><a id="l00503" name="l00503"></a><span class="lineno"> 503</span> <span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axis</a>, <span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">vlen</a>)</div>
|
|
<div class="line"><a id="l00504" name="l00504"></a><span class="lineno"> 504</span> {</div>
|
|
<div class="line"><a id="l00505" name="l00505"></a><span class="lineno"> 505</span> <span class="keywordflow">if</span> (<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nthreads</a>==1) <span class="keywordflow">return</span> 1;</div>
|
|
<div class="line"><a id="l00506" name="l00506"></a><span class="lineno"> 506</span> <span class="keywordtype">size_t</span> size = <a class="code hl_function" href="structpocketfft_1_1detail_1_1util.html#a33acae07b20b28fe4658bc338fce1b89">prod</a>(shape);</div>
|
|
<div class="line"><a id="l00507" name="l00507"></a><span class="lineno"> 507</span> <span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">parallel</a> = size / (shape[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axis</a>] * <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">vlen</a>);</div>
|
|
<div class="line"><a id="l00508" name="l00508"></a><span class="lineno"> 508</span> <span class="keywordflow">if</span> (shape[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axis</a>] < 1000)</div>
|
|
<div class="line"><a id="l00509" name="l00509"></a><span class="lineno"> 509</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">parallel</a> /= 4;</div>
|
|
<div class="line"><a id="l00510" name="l00510"></a><span class="lineno"> 510</span> <span class="keywordtype">size_t</span> max_threads = <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nthreads</a> == 0 ?</div>
|
|
<div class="line"><a id="l00511" name="l00511"></a><span class="lineno"> 511</span> std::thread::hardware_concurrency() : <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nthreads</a>;</div>
|
|
<div class="line"><a id="l00512" name="l00512"></a><span class="lineno"> 512</span> <span class="keywordflow">return</span> std::max(<span class="keywordtype">size_t</span>(1), std::min(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">parallel</a>, max_threads));</div>
|
|
<div class="line"><a id="l00513" name="l00513"></a><span class="lineno"> 513</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l00514" name="l00514"></a><span class="lineno"> 514</span><span class="preprocessor">#endif</span></div>
|
|
<div class="line"><a id="l00515" name="l00515"></a><span class="lineno"> 515</span> };</div>
|
|
</div>
|
|
<div class="line"><a id="l00516" name="l00516"></a><span class="lineno"> 516</span> </div>
|
|
<div class="foldopen" id="foldopen00517" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00517" name="l00517"></a><span class="lineno"><a class="line" href="namespacepocketfft_1_1detail_1_1threading.html"> 517</a></span><span class="keyword">namespace </span>threading {</div>
|
|
<div class="line"><a id="l00518" name="l00518"></a><span class="lineno"> 518</span> </div>
|
|
<div class="line"><a id="l00519" name="l00519"></a><span class="lineno"> 519</span><span class="preprocessor">#ifdef POCKETFFT_NO_MULTITHREADING</span></div>
|
|
<div class="line"><a id="l00520" name="l00520"></a><span class="lineno"> 520</span> </div>
|
|
<div class="line"><a id="l00521" name="l00521"></a><span class="lineno"> 521</span><span class="keyword">constexpr</span> <span class="keyword">inline</span> <span class="keywordtype">size_t</span> <a class="code hl_function" href="namespacepocketfft_1_1detail_1_1threading.html#aebe85d6273d92c7d3728e2c621ccc82b">thread_id</a>() { <span class="keywordflow">return</span> 0; }</div>
|
|
<div class="line"><a id="l00522" name="l00522"></a><span class="lineno"> 522</span><span class="keyword">constexpr</span> <span class="keyword">inline</span> <span class="keywordtype">size_t</span> <a class="code hl_function" href="namespacepocketfft_1_1detail_1_1threading.html#af5432c2e25aed679a73fe7b29534c833">num_threads</a>() { <span class="keywordflow">return</span> 1; }</div>
|
|
<div class="line"><a id="l00523" name="l00523"></a><span class="lineno"> 523</span> </div>
|
|
<div class="line"><a id="l00524" name="l00524"></a><span class="lineno"> 524</span><span class="keyword">template</span> <<span class="keyword">typename</span> Func></div>
|
|
<div class="line"><a id="l00525" name="l00525"></a><span class="lineno"> 525</span><span class="keywordtype">void</span> <a class="code hl_function" href="namespacepocketfft_1_1detail_1_1threading.html#a4fcf674db39f0e2c1c59d48491daed6e">thread_map</a>(<span class="keywordtype">size_t</span> <span class="comment">/* nthreads */</span>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">Func</a> <a class="code hl_variable" href="types_2bf16_8h.html#af900396d7b72ff2a7002e8befe8cf8f1">f</a>)</div>
|
|
<div class="line"><a id="l00526" name="l00526"></a><span class="lineno"> 526</span> { <a class="code hl_variable" href="types_2bf16_8h.html#af900396d7b72ff2a7002e8befe8cf8f1">f</a>(); }</div>
|
|
<div class="line"><a id="l00527" name="l00527"></a><span class="lineno"> 527</span> </div>
|
|
<div class="line"><a id="l00528" name="l00528"></a><span class="lineno"> 528</span><span class="preprocessor">#else</span></div>
|
|
<div class="line"><a id="l00529" name="l00529"></a><span class="lineno"> 529</span> </div>
|
|
<div class="foldopen" id="foldopen00530" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00530" name="l00530"></a><span class="lineno"><a class="line" href="namespacepocketfft_1_1detail_1_1threading.html#aebe85d6273d92c7d3728e2c621ccc82b"> 530</a></span><span class="keyword">inline</span> <span class="keywordtype">size_t</span> &<a class="code hl_function" href="namespacepocketfft_1_1detail_1_1threading.html#aebe85d6273d92c7d3728e2c621ccc82b">thread_id</a>()</div>
|
|
<div class="line"><a id="l00531" name="l00531"></a><span class="lineno"> 531</span> {</div>
|
|
<div class="line"><a id="l00532" name="l00532"></a><span class="lineno"> 532</span> <span class="keyword">static</span> <span class="keyword">thread_local</span> <span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">thread_id_</a>=0;</div>
|
|
<div class="line"><a id="l00533" name="l00533"></a><span class="lineno"> 533</span> <span class="keywordflow">return</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">thread_id_</a>;</div>
|
|
<div class="line"><a id="l00534" name="l00534"></a><span class="lineno"> 534</span> }</div>
|
|
</div>
|
|
<div class="foldopen" id="foldopen00535" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00535" name="l00535"></a><span class="lineno"><a class="line" href="namespacepocketfft_1_1detail_1_1threading.html#af5432c2e25aed679a73fe7b29534c833"> 535</a></span><span class="keyword">inline</span> <span class="keywordtype">size_t</span> &<a class="code hl_function" href="namespacepocketfft_1_1detail_1_1threading.html#af5432c2e25aed679a73fe7b29534c833">num_threads</a>()</div>
|
|
<div class="line"><a id="l00536" name="l00536"></a><span class="lineno"> 536</span> {</div>
|
|
<div class="line"><a id="l00537" name="l00537"></a><span class="lineno"> 537</span> <span class="keyword">static</span> <span class="keyword">thread_local</span> <span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">num_threads_</a>=1;</div>
|
|
<div class="line"><a id="l00538" name="l00538"></a><span class="lineno"> 538</span> <span class="keywordflow">return</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">num_threads_</a>;</div>
|
|
<div class="line"><a id="l00539" name="l00539"></a><span class="lineno"> 539</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l00540" name="l00540"></a><span class="lineno"><a class="line" href="namespacepocketfft_1_1detail_1_1threading.html#a2d5c0729f0b66cf061918baea4337d70"> 540</a></span><span class="keyword">static</span> <span class="keyword">const</span> <span class="keywordtype">size_t</span> <a class="code hl_variable" href="namespacepocketfft_1_1detail_1_1threading.html#a2d5c0729f0b66cf061918baea4337d70">max_threads</a> = std::max(1u, std::thread::hardware_concurrency());</div>
|
|
<div class="line"><a id="l00541" name="l00541"></a><span class="lineno"> 541</span> </div>
|
|
<div class="foldopen" id="foldopen00542" data-start="{" data-end="};">
|
|
<div class="line"><a id="l00542" name="l00542"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1threading_1_1latch.html"> 542</a></span><span class="keyword">class </span><a class="code hl_class" href="classpocketfft_1_1detail_1_1threading_1_1latch.html">latch</a></div>
|
|
<div class="line"><a id="l00543" name="l00543"></a><span class="lineno"> 543</span> {</div>
|
|
<div class="line"><a id="l00544" name="l00544"></a><span class="lineno"> 544</span> std::atomic<size_t> num_left_;</div>
|
|
<div class="line"><a id="l00545" name="l00545"></a><span class="lineno"> 545</span> std::mutex mut_;</div>
|
|
<div class="line"><a id="l00546" name="l00546"></a><span class="lineno"> 546</span> std::condition_variable completed_;</div>
|
|
<div class="line"><a id="l00547" name="l00547"></a><span class="lineno"> 547</span> <span class="keyword">using </span>lock_t = std::unique_lock<std::mutex>;</div>
|
|
<div class="line"><a id="l00548" name="l00548"></a><span class="lineno"> 548</span> </div>
|
|
<div class="line"><a id="l00549" name="l00549"></a><span class="lineno"> 549</span> <span class="keyword">public</span>:</div>
|
|
<div class="line"><a id="l00550" name="l00550"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1threading_1_1latch.html#a9260c871bb927525f7f53caa23a19c15"> 550</a></span> <a class="code hl_function" href="classpocketfft_1_1detail_1_1threading_1_1latch.html#a9260c871bb927525f7f53caa23a19c15">latch</a>(<span class="keywordtype">size_t</span> n): num_left_(n) {}</div>
|
|
<div class="line"><a id="l00551" name="l00551"></a><span class="lineno"> 551</span> </div>
|
|
<div class="foldopen" id="foldopen00552" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00552" name="l00552"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1threading_1_1latch.html#a81d6597189b40410e35f3cd653fd1342"> 552</a></span> <span class="keywordtype">void</span> <a class="code hl_function" href="classpocketfft_1_1detail_1_1threading_1_1latch.html#a81d6597189b40410e35f3cd653fd1342">count_down</a>()</div>
|
|
<div class="line"><a id="l00553" name="l00553"></a><span class="lineno"> 553</span> {</div>
|
|
<div class="line"><a id="l00554" name="l00554"></a><span class="lineno"> 554</span> lock_t <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">lock</a>(mut_);</div>
|
|
<div class="line"><a id="l00555" name="l00555"></a><span class="lineno"> 555</span> <span class="keywordflow">if</span> (--num_left_)</div>
|
|
<div class="line"><a id="l00556" name="l00556"></a><span class="lineno"> 556</span> <span class="keywordflow">return</span>;</div>
|
|
<div class="line"><a id="l00557" name="l00557"></a><span class="lineno"> 557</span> completed_.notify_all();</div>
|
|
<div class="line"><a id="l00558" name="l00558"></a><span class="lineno"> 558</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l00559" name="l00559"></a><span class="lineno"> 559</span> </div>
|
|
<div class="foldopen" id="foldopen00560" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00560" name="l00560"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1threading_1_1latch.html#af503189cc9247047fbdfc3ebf1daacc1"> 560</a></span> <span class="keywordtype">void</span> <a class="code hl_function" href="classpocketfft_1_1detail_1_1threading_1_1latch.html#af503189cc9247047fbdfc3ebf1daacc1">wait</a>()</div>
|
|
<div class="line"><a id="l00561" name="l00561"></a><span class="lineno"> 561</span> {</div>
|
|
<div class="line"><a id="l00562" name="l00562"></a><span class="lineno"> 562</span> lock_t <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">lock</a>(mut_);</div>
|
|
<div class="line"><a id="l00563" name="l00563"></a><span class="lineno"> 563</span> completed_.wait(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">lock</a>, [<span class="keyword">this</span>]{ <span class="keywordflow">return</span> <a class="code hl_function" href="classpocketfft_1_1detail_1_1threading_1_1latch.html#ab41ecc5adb6187aa2682ca190fd920f3">is_ready</a>(); });</div>
|
|
<div class="line"><a id="l00564" name="l00564"></a><span class="lineno"> 564</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l00565" name="l00565"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1threading_1_1latch.html#ab41ecc5adb6187aa2682ca190fd920f3"> 565</a></span> <span class="keywordtype">bool</span> <a class="code hl_function" href="classpocketfft_1_1detail_1_1threading_1_1latch.html#ab41ecc5adb6187aa2682ca190fd920f3">is_ready</a>() { <span class="keywordflow">return</span> num_left_ == 0; }</div>
|
|
<div class="line"><a id="l00566" name="l00566"></a><span class="lineno"> 566</span> };</div>
|
|
</div>
|
|
<div class="line"><a id="l00567" name="l00567"></a><span class="lineno"> 567</span> </div>
|
|
<div class="foldopen" id="foldopen00568" data-start="{" data-end="};">
|
|
<div class="line"><a id="l00568" name="l00568"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1threading_1_1concurrent__queue.html"> 568</a></span><span class="keyword">template</span> <<span class="keyword">typename</span> T> <span class="keyword">class </span><a class="code hl_class" href="classpocketfft_1_1detail_1_1threading_1_1concurrent__queue.html">concurrent_queue</a></div>
|
|
<div class="line"><a id="l00569" name="l00569"></a><span class="lineno"> 569</span> {</div>
|
|
<div class="line"><a id="l00570" name="l00570"></a><span class="lineno"> 570</span> std::queue<T> q_;</div>
|
|
<div class="line"><a id="l00571" name="l00571"></a><span class="lineno"> 571</span> std::mutex mut_;</div>
|
|
<div class="line"><a id="l00572" name="l00572"></a><span class="lineno"> 572</span> std::atomic<size_t> size_;</div>
|
|
<div class="line"><a id="l00573" name="l00573"></a><span class="lineno"> 573</span> <span class="keyword">using </span>lock_t = std::lock_guard<std::mutex>;</div>
|
|
<div class="line"><a id="l00574" name="l00574"></a><span class="lineno"> 574</span> </div>
|
|
<div class="line"><a id="l00575" name="l00575"></a><span class="lineno"> 575</span> <span class="keyword">public</span>:</div>
|
|
<div class="line"><a id="l00576" name="l00576"></a><span class="lineno"> 576</span> </div>
|
|
<div class="foldopen" id="foldopen00577" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00577" name="l00577"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1threading_1_1concurrent__queue.html#ae8ebd6bc1b4d18e75bd046005e7dde7b"> 577</a></span> <span class="keywordtype">void</span> <a class="code hl_function" href="classpocketfft_1_1detail_1_1threading_1_1concurrent__queue.html#ae8ebd6bc1b4d18e75bd046005e7dde7b">push</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a> val)</div>
|
|
<div class="line"><a id="l00578" name="l00578"></a><span class="lineno"> 578</span> {</div>
|
|
<div class="line"><a id="l00579" name="l00579"></a><span class="lineno"> 579</span> lock_t <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">lock</a>(mut_);</div>
|
|
<div class="line"><a id="l00580" name="l00580"></a><span class="lineno"> 580</span> ++size_;</div>
|
|
<div class="line"><a id="l00581" name="l00581"></a><span class="lineno"> 581</span> q_.push(std::move(val));</div>
|
|
<div class="line"><a id="l00582" name="l00582"></a><span class="lineno"> 582</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l00583" name="l00583"></a><span class="lineno"> 583</span> </div>
|
|
<div class="foldopen" id="foldopen00584" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00584" name="l00584"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1threading_1_1concurrent__queue.html#aa3807d46a126d229f9054c779105ea43"> 584</a></span> <span class="keywordtype">bool</span> <a class="code hl_function" href="classpocketfft_1_1detail_1_1threading_1_1concurrent__queue.html#aa3807d46a126d229f9054c779105ea43">try_pop</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a> &val)</div>
|
|
<div class="line"><a id="l00585" name="l00585"></a><span class="lineno"> 585</span> {</div>
|
|
<div class="line"><a id="l00586" name="l00586"></a><span class="lineno"> 586</span> <span class="keywordflow">if</span> (size_ == 0) <span class="keywordflow">return</span> <span class="keyword">false</span>;</div>
|
|
<div class="line"><a id="l00587" name="l00587"></a><span class="lineno"> 587</span> lock_t <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">lock</a>(mut_);</div>
|
|
<div class="line"><a id="l00588" name="l00588"></a><span class="lineno"> 588</span> <span class="comment">// Queue might have been emptied while we acquired the lock</span></div>
|
|
<div class="line"><a id="l00589" name="l00589"></a><span class="lineno"> 589</span> <span class="keywordflow">if</span> (q_.empty()) <span class="keywordflow">return</span> <span class="keyword">false</span>;</div>
|
|
<div class="line"><a id="l00590" name="l00590"></a><span class="lineno"> 590</span> </div>
|
|
<div class="line"><a id="l00591" name="l00591"></a><span class="lineno"> 591</span> val = std::move(q_.front());</div>
|
|
<div class="line"><a id="l00592" name="l00592"></a><span class="lineno"> 592</span> --size_;</div>
|
|
<div class="line"><a id="l00593" name="l00593"></a><span class="lineno"> 593</span> q_.pop();</div>
|
|
<div class="line"><a id="l00594" name="l00594"></a><span class="lineno"> 594</span> <span class="keywordflow">return</span> <span class="keyword">true</span>;</div>
|
|
<div class="line"><a id="l00595" name="l00595"></a><span class="lineno"> 595</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l00596" name="l00596"></a><span class="lineno"> 596</span> </div>
|
|
<div class="line"><a id="l00597" name="l00597"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1threading_1_1concurrent__queue.html#a1269e5da40c3f5145c895cee3641879a"> 597</a></span> <span class="keywordtype">bool</span> <a class="code hl_function" href="classpocketfft_1_1detail_1_1threading_1_1concurrent__queue.html#a1269e5da40c3f5145c895cee3641879a">empty</a>()<span class="keyword"> const </span>{ <span class="keywordflow">return</span> size_==0; }</div>
|
|
<div class="line"><a id="l00598" name="l00598"></a><span class="lineno"> 598</span> };</div>
|
|
</div>
|
|
<div class="line"><a id="l00599" name="l00599"></a><span class="lineno"> 599</span> </div>
|
|
<div class="line"><a id="l00600" name="l00600"></a><span class="lineno"> 600</span><span class="comment">// C++ allocator with support for over-aligned types</span></div>
|
|
<div class="foldopen" id="foldopen00601" data-start="{" data-end="};">
|
|
<div class="line"><a id="l00601" name="l00601"></a><span class="lineno"><a class="line" href="structpocketfft_1_1detail_1_1threading_1_1aligned__allocator.html"> 601</a></span><span class="keyword">template</span> <<span class="keyword">typename</span> T> <span class="keyword">struct </span><a class="code hl_struct" href="structpocketfft_1_1detail_1_1threading_1_1aligned__allocator.html">aligned_allocator</a></div>
|
|
<div class="line"><a id="l00602" name="l00602"></a><span class="lineno"> 602</span> {</div>
|
|
<div class="line"><a id="l00603" name="l00603"></a><span class="lineno"><a class="line" href="structpocketfft_1_1detail_1_1threading_1_1aligned__allocator.html#ab86a2740ed9ce3199135372ff1d88c76"> 603</a></span> <span class="keyword">using </span><a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">value_type</a> = <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a>;</div>
|
|
<div class="line"><a id="l00604" name="l00604"></a><span class="lineno"> 604</span> <span class="keyword">template</span> <<span class="keyword">class</span> U></div>
|
|
<div class="line"><a id="l00605" name="l00605"></a><span class="lineno"><a class="line" href="structpocketfft_1_1detail_1_1threading_1_1aligned__allocator.html#a57c07047ac09c6cf48a269429de2b0fb"> 605</a></span> <a class="code hl_function" href="structpocketfft_1_1detail_1_1threading_1_1aligned__allocator.html#a57c07047ac09c6cf48a269429de2b0fb">aligned_allocator</a>(<span class="keyword">const</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">aligned_allocator<U></a>&) {}</div>
|
|
<div class="line"><a id="l00606" name="l00606"></a><span class="lineno"><a class="line" href="structpocketfft_1_1detail_1_1threading_1_1aligned__allocator.html#a0c390851ec37c5cdc5c1e7c6232a0b94"> 606</a></span> <a class="code hl_function" href="structpocketfft_1_1detail_1_1threading_1_1aligned__allocator.html#a0c390851ec37c5cdc5c1e7c6232a0b94">aligned_allocator</a>() = <span class="keywordflow">default</span>;</div>
|
|
<div class="line"><a id="l00607" name="l00607"></a><span class="lineno"> 607</span> </div>
|
|
<div class="foldopen" id="foldopen00608" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00608" name="l00608"></a><span class="lineno"><a class="line" href="structpocketfft_1_1detail_1_1threading_1_1aligned__allocator.html#a4f785747b898980756af9e5515363826"> 608</a></span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a> *<a class="code hl_function" href="structpocketfft_1_1detail_1_1threading_1_1aligned__allocator.html#a4f785747b898980756af9e5515363826">allocate</a>(<span class="keywordtype">size_t</span> n)</div>
|
|
<div class="line"><a id="l00609" name="l00609"></a><span class="lineno"> 609</span> {</div>
|
|
<div class="line"><a id="l00610" name="l00610"></a><span class="lineno"> 610</span> <span class="keywordtype">void</span>* mem = <a class="code hl_function" href="namespacepocketfft_1_1detail.html#ae397445c61400f47a8fe3f8e1b6d0b76">aligned_alloc</a>(<span class="keyword">alignof</span>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a>), n*<span class="keyword">sizeof</span>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a>));</div>
|
|
<div class="line"><a id="l00611" name="l00611"></a><span class="lineno"> 611</span> <span class="keywordflow">return</span> <span class="keyword">static_cast<</span><a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a>*<span class="keyword">></span>(mem);</div>
|
|
<div class="line"><a id="l00612" name="l00612"></a><span class="lineno"> 612</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l00613" name="l00613"></a><span class="lineno"> 613</span> </div>
|
|
<div class="foldopen" id="foldopen00614" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00614" name="l00614"></a><span class="lineno"><a class="line" href="structpocketfft_1_1detail_1_1threading_1_1aligned__allocator.html#a2a99b8e296d26b255e9937ba5f30e76f"> 614</a></span> <span class="keywordtype">void</span> <a class="code hl_function" href="structpocketfft_1_1detail_1_1threading_1_1aligned__allocator.html#a2a99b8e296d26b255e9937ba5f30e76f">deallocate</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a> *p, <span class="keywordtype">size_t</span> <span class="comment">/*n*/</span>)</div>
|
|
<div class="line"><a id="l00615" name="l00615"></a><span class="lineno"> 615</span> { <a class="code hl_function" href="namespacepocketfft_1_1detail.html#aec7820e36a33e0a8bb83aa03b04b81e8">aligned_dealloc</a>(p); }</div>
|
|
</div>
|
|
<div class="line"><a id="l00616" name="l00616"></a><span class="lineno"> 616</span> };</div>
|
|
</div>
|
|
<div class="line"><a id="l00617" name="l00617"></a><span class="lineno"> 617</span> </div>
|
|
<div class="foldopen" id="foldopen00618" data-start="{" data-end="};">
|
|
<div class="line"><a id="l00618" name="l00618"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1threading_1_1thread__pool.html"> 618</a></span><span class="keyword">class </span><a class="code hl_class" href="classpocketfft_1_1detail_1_1threading_1_1thread__pool.html">thread_pool</a></div>
|
|
<div class="line"><a id="l00619" name="l00619"></a><span class="lineno"> 619</span> {</div>
|
|
<div class="line"><a id="l00620" name="l00620"></a><span class="lineno"> 620</span> <span class="comment">// A reasonable guess, probably close enough for most hardware</span></div>
|
|
<div class="line"><a id="l00621" name="l00621"></a><span class="lineno"> 621</span> <span class="keyword">static</span> <span class="keyword">constexpr</span> <span class="keywordtype">size_t</span> cache_line_size = 64;</div>
|
|
<div class="line"><a id="l00622" name="l00622"></a><span class="lineno"> 622</span> <span class="keyword">struct </span><a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">alignas(cache_line_size)</a> worker</div>
|
|
<div class="line"><a id="l00623" name="l00623"></a><span class="lineno"> 623</span> {</div>
|
|
<div class="line"><a id="l00624" name="l00624"></a><span class="lineno"> 624</span> std::thread thread;</div>
|
|
<div class="line"><a id="l00625" name="l00625"></a><span class="lineno"> 625</span> std::condition_variable work_ready;</div>
|
|
<div class="line"><a id="l00626" name="l00626"></a><span class="lineno"> 626</span> std::mutex mut;</div>
|
|
<div class="line"><a id="l00627" name="l00627"></a><span class="lineno"> 627</span> std::atomic_flag busy_flag = <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ATOMIC_FLAG_INIT</a>;</div>
|
|
<div class="line"><a id="l00628" name="l00628"></a><span class="lineno"> 628</span> std::function<<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">void</a>()> work;</div>
|
|
<div class="line"><a id="l00629" name="l00629"></a><span class="lineno"> 629</span> </div>
|
|
<div class="line"><a id="l00630" name="l00630"></a><span class="lineno"> 630</span> <span class="keywordtype">void</span> worker_main(</div>
|
|
<div class="line"><a id="l00631" name="l00631"></a><span class="lineno"> 631</span> std::atomic<bool> &<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">shutdown_flag</a>,</div>
|
|
<div class="line"><a id="l00632" name="l00632"></a><span class="lineno"> 632</span> std::atomic<size_t> &<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">unscheduled_tasks</a>,</div>
|
|
<div class="line"><a id="l00633" name="l00633"></a><span class="lineno"> 633</span> <a class="code hl_class" href="classpocketfft_1_1detail_1_1threading_1_1concurrent__queue.html">concurrent_queue</a><std::function<<span class="keywordtype">void</span>()>> &<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">overflow_work</a>)</div>
|
|
<div class="line"><a id="l00634" name="l00634"></a><span class="lineno"> 634</span> {</div>
|
|
<div class="line"><a id="l00635" name="l00635"></a><span class="lineno"> 635</span> <span class="keyword">using </span>lock_t = std::unique_lock<std::mutex>;</div>
|
|
<div class="line"><a id="l00636" name="l00636"></a><span class="lineno"> 636</span> <span class="keywordtype">bool</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">expect_work</a> = <span class="keyword">true</span>;</div>
|
|
<div class="line"><a id="l00637" name="l00637"></a><span class="lineno"> 637</span> <span class="keywordflow">while</span> (!<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">shutdown_flag</a> || <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">expect_work</a>)</div>
|
|
<div class="line"><a id="l00638" name="l00638"></a><span class="lineno"> 638</span> {</div>
|
|
<div class="line"><a id="l00639" name="l00639"></a><span class="lineno"> 639</span> std::function<<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">void</a>()> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">local_work</a>;</div>
|
|
<div class="line"><a id="l00640" name="l00640"></a><span class="lineno"> 640</span> <span class="keywordflow">if</span> (<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">expect_work</a> || <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">unscheduled_tasks</a> == 0)</div>
|
|
<div class="line"><a id="l00641" name="l00641"></a><span class="lineno"> 641</span> {</div>
|
|
<div class="line"><a id="l00642" name="l00642"></a><span class="lineno"> 642</span> lock_t <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">lock</a>(mut);</div>
|
|
<div class="line"><a id="l00643" name="l00643"></a><span class="lineno"> 643</span> <span class="comment">// Wait until there is work to be executed</span></div>
|
|
<div class="line"><a id="l00644" name="l00644"></a><span class="lineno"> 644</span> work_ready.wait(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">lock</a>, [&]{ <span class="keywordflow">return</span> (work || <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">shutdown_flag</a>); });</div>
|
|
<div class="line"><a id="l00645" name="l00645"></a><span class="lineno"> 645</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">local_work</a>.swap(work);</div>
|
|
<div class="line"><a id="l00646" name="l00646"></a><span class="lineno"> 646</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">expect_work</a> = <span class="keyword">false</span>;</div>
|
|
<div class="line"><a id="l00647" name="l00647"></a><span class="lineno"> 647</span> }</div>
|
|
<div class="line"><a id="l00648" name="l00648"></a><span class="lineno"> 648</span> </div>
|
|
<div class="line"><a id="l00649" name="l00649"></a><span class="lineno"> 649</span> <span class="keywordtype">bool</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">marked_busy</a> = <span class="keyword">false</span>;</div>
|
|
<div class="line"><a id="l00650" name="l00650"></a><span class="lineno"> 650</span> <span class="keywordflow">if</span> (<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">local_work</a>)</div>
|
|
<div class="line"><a id="l00651" name="l00651"></a><span class="lineno"> 651</span> {</div>
|
|
<div class="line"><a id="l00652" name="l00652"></a><span class="lineno"> 652</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">marked_busy</a> = <span class="keyword">true</span>;</div>
|
|
<div class="line"><a id="l00653" name="l00653"></a><span class="lineno"> 653</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">local_work</a>();</div>
|
|
<div class="line"><a id="l00654" name="l00654"></a><span class="lineno"> 654</span> }</div>
|
|
<div class="line"><a id="l00655" name="l00655"></a><span class="lineno"> 655</span> </div>
|
|
<div class="line"><a id="l00656" name="l00656"></a><span class="lineno"> 656</span> <span class="keywordflow">if</span> (!<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">overflow_work</a>.empty())</div>
|
|
<div class="line"><a id="l00657" name="l00657"></a><span class="lineno"> 657</span> {</div>
|
|
<div class="line"><a id="l00658" name="l00658"></a><span class="lineno"> 658</span> <span class="keywordflow">if</span> (!<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">marked_busy</a> && busy_flag.test_and_set())</div>
|
|
<div class="line"><a id="l00659" name="l00659"></a><span class="lineno"> 659</span> {</div>
|
|
<div class="line"><a id="l00660" name="l00660"></a><span class="lineno"> 660</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">expect_work</a> = <span class="keyword">true</span>;</div>
|
|
<div class="line"><a id="l00661" name="l00661"></a><span class="lineno"> 661</span> <span class="keywordflow">continue</span>;</div>
|
|
<div class="line"><a id="l00662" name="l00662"></a><span class="lineno"> 662</span> }</div>
|
|
<div class="line"><a id="l00663" name="l00663"></a><span class="lineno"> 663</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">marked_busy</a> = <span class="keyword">true</span>;</div>
|
|
<div class="line"><a id="l00664" name="l00664"></a><span class="lineno"> 664</span> </div>
|
|
<div class="line"><a id="l00665" name="l00665"></a><span class="lineno"> 665</span> <span class="keywordflow">while</span> (<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">overflow_work</a>.try_pop(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">local_work</a>))</div>
|
|
<div class="line"><a id="l00666" name="l00666"></a><span class="lineno"> 666</span> {</div>
|
|
<div class="line"><a id="l00667" name="l00667"></a><span class="lineno"> 667</span> --<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">unscheduled_tasks</a>;</div>
|
|
<div class="line"><a id="l00668" name="l00668"></a><span class="lineno"> 668</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">local_work</a>();</div>
|
|
<div class="line"><a id="l00669" name="l00669"></a><span class="lineno"> 669</span> }</div>
|
|
<div class="line"><a id="l00670" name="l00670"></a><span class="lineno"> 670</span> }</div>
|
|
<div class="line"><a id="l00671" name="l00671"></a><span class="lineno"> 671</span> </div>
|
|
<div class="line"><a id="l00672" name="l00672"></a><span class="lineno"> 672</span> <span class="keywordflow">if</span> (<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">marked_busy</a>) busy_flag.clear();</div>
|
|
<div class="line"><a id="l00673" name="l00673"></a><span class="lineno"> 673</span> }</div>
|
|
<div class="line"><a id="l00674" name="l00674"></a><span class="lineno"> 674</span> }</div>
|
|
<div class="line"><a id="l00675" name="l00675"></a><span class="lineno"> 675</span> };</div>
|
|
<div class="line"><a id="l00676" name="l00676"></a><span class="lineno"> 676</span> </div>
|
|
<div class="line"><a id="l00677" name="l00677"></a><span class="lineno"> 677</span> <a class="code hl_class" href="classpocketfft_1_1detail_1_1threading_1_1concurrent__queue.html">concurrent_queue</a><std::function<<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">void</a>()>> overflow_work_;</div>
|
|
<div class="line"><a id="l00678" name="l00678"></a><span class="lineno"> 678</span> std::mutex mut_;</div>
|
|
<div class="line"><a id="l00679" name="l00679"></a><span class="lineno"> 679</span> std::vector<worker, aligned_allocator<worker>> workers_;</div>
|
|
<div class="line"><a id="l00680" name="l00680"></a><span class="lineno"> 680</span> std::atomic<bool> shutdown_;</div>
|
|
<div class="line"><a id="l00681" name="l00681"></a><span class="lineno"> 681</span> std::atomic<size_t> unscheduled_tasks_;</div>
|
|
<div class="line"><a id="l00682" name="l00682"></a><span class="lineno"> 682</span> <span class="keyword">using </span>lock_t = std::lock_guard<std::mutex>;</div>
|
|
<div class="line"><a id="l00683" name="l00683"></a><span class="lineno"> 683</span> </div>
|
|
<div class="line"><a id="l00684" name="l00684"></a><span class="lineno"> 684</span> <span class="keywordtype">void</span> create_threads()</div>
|
|
<div class="line"><a id="l00685" name="l00685"></a><span class="lineno"> 685</span> {</div>
|
|
<div class="line"><a id="l00686" name="l00686"></a><span class="lineno"> 686</span> lock_t <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">lock</a>(mut_);</div>
|
|
<div class="line"><a id="l00687" name="l00687"></a><span class="lineno"> 687</span> <span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nthreads</a>=workers_.size();</div>
|
|
<div class="line"><a id="l00688" name="l00688"></a><span class="lineno"> 688</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i=0; i<<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nthreads</a>; ++i)</div>
|
|
<div class="line"><a id="l00689" name="l00689"></a><span class="lineno"> 689</span> {</div>
|
|
<div class="line"><a id="l00690" name="l00690"></a><span class="lineno"> 690</span> <span class="keywordflow">try</span></div>
|
|
<div class="line"><a id="l00691" name="l00691"></a><span class="lineno"> 691</span> {</div>
|
|
<div class="line"><a id="l00692" name="l00692"></a><span class="lineno"> 692</span> <span class="keyword">auto</span> *worker = &workers_[i];</div>
|
|
<div class="line"><a id="l00693" name="l00693"></a><span class="lineno"> 693</span> worker->busy_flag.clear();</div>
|
|
<div class="line"><a id="l00694" name="l00694"></a><span class="lineno"> 694</span> worker->work = <span class="keyword">nullptr</span>;</div>
|
|
<div class="line"><a id="l00695" name="l00695"></a><span class="lineno"> 695</span> worker->thread = std::thread([worker, <span class="keyword">this</span>]</div>
|
|
<div class="line"><a id="l00696" name="l00696"></a><span class="lineno"> 696</span> {</div>
|
|
<div class="line"><a id="l00697" name="l00697"></a><span class="lineno"> 697</span> worker->worker_main(shutdown_, unscheduled_tasks_, overflow_work_);</div>
|
|
<div class="line"><a id="l00698" name="l00698"></a><span class="lineno"> 698</span> });</div>
|
|
<div class="line"><a id="l00699" name="l00699"></a><span class="lineno"> 699</span> }</div>
|
|
<div class="line"><a id="l00700" name="l00700"></a><span class="lineno"> 700</span> <span class="keywordflow">catch</span> (...)</div>
|
|
<div class="line"><a id="l00701" name="l00701"></a><span class="lineno"> 701</span> {</div>
|
|
<div class="line"><a id="l00702" name="l00702"></a><span class="lineno"> 702</span> shutdown_locked();</div>
|
|
<div class="line"><a id="l00703" name="l00703"></a><span class="lineno"> 703</span> <span class="keywordflow">throw</span>;</div>
|
|
<div class="line"><a id="l00704" name="l00704"></a><span class="lineno"> 704</span> }</div>
|
|
<div class="line"><a id="l00705" name="l00705"></a><span class="lineno"> 705</span> }</div>
|
|
<div class="line"><a id="l00706" name="l00706"></a><span class="lineno"> 706</span> }</div>
|
|
<div class="line"><a id="l00707" name="l00707"></a><span class="lineno"> 707</span> </div>
|
|
<div class="line"><a id="l00708" name="l00708"></a><span class="lineno"> 708</span> <span class="keywordtype">void</span> shutdown_locked()</div>
|
|
<div class="line"><a id="l00709" name="l00709"></a><span class="lineno"> 709</span> {</div>
|
|
<div class="line"><a id="l00710" name="l00710"></a><span class="lineno"> 710</span> shutdown_ = <span class="keyword">true</span>;</div>
|
|
<div class="line"><a id="l00711" name="l00711"></a><span class="lineno"> 711</span> <span class="keywordflow">for</span> (<span class="keyword">auto</span> &worker : workers_)</div>
|
|
<div class="line"><a id="l00712" name="l00712"></a><span class="lineno"> 712</span> worker.work_ready.notify_all();</div>
|
|
<div class="line"><a id="l00713" name="l00713"></a><span class="lineno"> 713</span> </div>
|
|
<div class="line"><a id="l00714" name="l00714"></a><span class="lineno"> 714</span> <span class="keywordflow">for</span> (<span class="keyword">auto</span> &worker : workers_)</div>
|
|
<div class="line"><a id="l00715" name="l00715"></a><span class="lineno"> 715</span> <span class="keywordflow">if</span> (worker.thread.joinable())</div>
|
|
<div class="line"><a id="l00716" name="l00716"></a><span class="lineno"> 716</span> worker.thread.join();</div>
|
|
<div class="line"><a id="l00717" name="l00717"></a><span class="lineno"> 717</span> }</div>
|
|
<div class="line"><a id="l00718" name="l00718"></a><span class="lineno"> 718</span> </div>
|
|
<div class="line"><a id="l00719" name="l00719"></a><span class="lineno"> 719</span> <span class="keyword">public</span>:</div>
|
|
<div class="foldopen" id="foldopen00720" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00720" name="l00720"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#a37a8121a99dd06a9d44b3e80ba0ea560"> 720</a></span> <span class="keyword">explicit</span> <a class="code hl_function" href="classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#a37a8121a99dd06a9d44b3e80ba0ea560">thread_pool</a>(<span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nthreads</a>):</div>
|
|
<div class="line"><a id="l00721" name="l00721"></a><span class="lineno"> 721</span> workers_(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nthreads</a>)</div>
|
|
<div class="line"><a id="l00722" name="l00722"></a><span class="lineno"> 722</span> { create_threads(); }</div>
|
|
</div>
|
|
<div class="line"><a id="l00723" name="l00723"></a><span class="lineno"> 723</span> </div>
|
|
<div class="line"><a id="l00724" name="l00724"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#aefaadaa60c0183b862ad96338177a5e0"> 724</a></span> <a class="code hl_function" href="classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#aefaadaa60c0183b862ad96338177a5e0">thread_pool</a>(): <a class="code hl_class" href="classpocketfft_1_1detail_1_1threading_1_1thread__pool.html">thread_pool</a>(<a class="code hl_variable" href="namespacepocketfft_1_1detail_1_1threading.html#a2d5c0729f0b66cf061918baea4337d70">max_threads</a>) {}</div>
|
|
<div class="line"><a id="l00725" name="l00725"></a><span class="lineno"> 725</span> </div>
|
|
<div class="line"><a id="l00726" name="l00726"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#a68f71e79fa93379fcc5819e533b45472"> 726</a></span> <a class="code hl_function" href="classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#a68f71e79fa93379fcc5819e533b45472">~thread_pool</a>() { shutdown(); }</div>
|
|
<div class="line"><a id="l00727" name="l00727"></a><span class="lineno"> 727</span> </div>
|
|
<div class="foldopen" id="foldopen00728" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00728" name="l00728"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#a8698d49e8f406cdb88006aac6a91f9a4"> 728</a></span> <span class="keywordtype">void</span> <a class="code hl_function" href="classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#a8698d49e8f406cdb88006aac6a91f9a4">submit</a>(std::function<<span class="keywordtype">void</span>()> work)</div>
|
|
<div class="line"><a id="l00729" name="l00729"></a><span class="lineno"> 729</span> {</div>
|
|
<div class="line"><a id="l00730" name="l00730"></a><span class="lineno"> 730</span> lock_t <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">lock</a>(mut_);</div>
|
|
<div class="line"><a id="l00731" name="l00731"></a><span class="lineno"> 731</span> <span class="keywordflow">if</span> (shutdown_)</div>
|
|
<div class="line"><a id="l00732" name="l00732"></a><span class="lineno"> 732</span> <span class="keywordflow">throw</span> std::runtime_error(<span class="stringliteral">"Work item submitted after shutdown"</span>);</div>
|
|
<div class="line"><a id="l00733" name="l00733"></a><span class="lineno"> 733</span> </div>
|
|
<div class="line"><a id="l00734" name="l00734"></a><span class="lineno"> 734</span> ++unscheduled_tasks_;</div>
|
|
<div class="line"><a id="l00735" name="l00735"></a><span class="lineno"> 735</span> </div>
|
|
<div class="line"><a id="l00736" name="l00736"></a><span class="lineno"> 736</span> <span class="comment">// First check for any idle workers and wake those</span></div>
|
|
<div class="line"><a id="l00737" name="l00737"></a><span class="lineno"> 737</span> <span class="keywordflow">for</span> (<span class="keyword">auto</span> &worker : workers_)</div>
|
|
<div class="line"><a id="l00738" name="l00738"></a><span class="lineno"> 738</span> <span class="keywordflow">if</span> (!worker.busy_flag.test_and_set())</div>
|
|
<div class="line"><a id="l00739" name="l00739"></a><span class="lineno"> 739</span> {</div>
|
|
<div class="line"><a id="l00740" name="l00740"></a><span class="lineno"> 740</span> --unscheduled_tasks_;</div>
|
|
<div class="line"><a id="l00741" name="l00741"></a><span class="lineno"> 741</span> {</div>
|
|
<div class="line"><a id="l00742" name="l00742"></a><span class="lineno"> 742</span> lock_t <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">lock</a>(worker.mut);</div>
|
|
<div class="line"><a id="l00743" name="l00743"></a><span class="lineno"> 743</span> worker.work = std::move(work);</div>
|
|
<div class="line"><a id="l00744" name="l00744"></a><span class="lineno"> 744</span> }</div>
|
|
<div class="line"><a id="l00745" name="l00745"></a><span class="lineno"> 745</span> worker.work_ready.notify_one();</div>
|
|
<div class="line"><a id="l00746" name="l00746"></a><span class="lineno"> 746</span> <span class="keywordflow">return</span>;</div>
|
|
<div class="line"><a id="l00747" name="l00747"></a><span class="lineno"> 747</span> }</div>
|
|
<div class="line"><a id="l00748" name="l00748"></a><span class="lineno"> 748</span> </div>
|
|
<div class="line"><a id="l00749" name="l00749"></a><span class="lineno"> 749</span> <span class="comment">// If no workers were idle, push onto the overflow queue for later</span></div>
|
|
<div class="line"><a id="l00750" name="l00750"></a><span class="lineno"> 750</span> overflow_work_.push(std::move(work));</div>
|
|
<div class="line"><a id="l00751" name="l00751"></a><span class="lineno"> 751</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l00752" name="l00752"></a><span class="lineno"> 752</span> </div>
|
|
<div class="foldopen" id="foldopen00753" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00753" name="l00753"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#a0c7c29290fde806031c497f24c4ad411"> 753</a></span> <span class="keywordtype">void</span> <a class="code hl_function" href="classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#a0c7c29290fde806031c497f24c4ad411">shutdown</a>()</div>
|
|
<div class="line"><a id="l00754" name="l00754"></a><span class="lineno"> 754</span> {</div>
|
|
<div class="line"><a id="l00755" name="l00755"></a><span class="lineno"> 755</span> lock_t <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">lock</a>(mut_);</div>
|
|
<div class="line"><a id="l00756" name="l00756"></a><span class="lineno"> 756</span> shutdown_locked();</div>
|
|
<div class="line"><a id="l00757" name="l00757"></a><span class="lineno"> 757</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l00758" name="l00758"></a><span class="lineno"> 758</span> </div>
|
|
<div class="foldopen" id="foldopen00759" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00759" name="l00759"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#a51d252df8d0cd060f15be8ba2bfe3288"> 759</a></span> <span class="keywordtype">void</span> <a class="code hl_function" href="classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#a51d252df8d0cd060f15be8ba2bfe3288">restart</a>()</div>
|
|
<div class="line"><a id="l00760" name="l00760"></a><span class="lineno"> 760</span> {</div>
|
|
<div class="line"><a id="l00761" name="l00761"></a><span class="lineno"> 761</span> shutdown_ = <span class="keyword">false</span>;</div>
|
|
<div class="line"><a id="l00762" name="l00762"></a><span class="lineno"> 762</span> create_threads();</div>
|
|
<div class="line"><a id="l00763" name="l00763"></a><span class="lineno"> 763</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l00764" name="l00764"></a><span class="lineno"> 764</span> };</div>
|
|
</div>
|
|
<div class="line"><a id="l00765" name="l00765"></a><span class="lineno"> 765</span> </div>
|
|
<div class="foldopen" id="foldopen00766" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00766" name="l00766"></a><span class="lineno"><a class="line" href="namespacepocketfft_1_1detail_1_1threading.html#a7ec2b3f99232bd0f15f7b022c59d139a"> 766</a></span><span class="keyword">inline</span> <a class="code hl_class" href="classpocketfft_1_1detail_1_1threading_1_1thread__pool.html">thread_pool</a> & <a class="code hl_function" href="namespacepocketfft_1_1detail_1_1threading.html#a7ec2b3f99232bd0f15f7b022c59d139a">get_pool</a>()</div>
|
|
<div class="line"><a id="l00767" name="l00767"></a><span class="lineno"> 767</span> {</div>
|
|
<div class="line"><a id="l00768" name="l00768"></a><span class="lineno"> 768</span> <span class="keyword">static</span> <a class="code hl_class" href="classpocketfft_1_1detail_1_1threading_1_1thread__pool.html">thread_pool</a> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">pool</a>;</div>
|
|
<div class="line"><a id="l00769" name="l00769"></a><span class="lineno"> 769</span><span class="preprocessor">#ifdef POCKETFFT_PTHREADS</span></div>
|
|
<div class="line"><a id="l00770" name="l00770"></a><span class="lineno"> 770</span> <span class="keyword">static</span> std::once_flag <a class="code hl_variable" href="types_2bf16_8h.html#af900396d7b72ff2a7002e8befe8cf8f1">f</a>;</div>
|
|
<div class="line"><a id="l00771" name="l00771"></a><span class="lineno"> 771</span> std::call_once(<a class="code hl_variable" href="types_2bf16_8h.html#af900396d7b72ff2a7002e8befe8cf8f1">f</a>,</div>
|
|
<div class="line"><a id="l00772" name="l00772"></a><span class="lineno"> 772</span> []{</div>
|
|
<div class="line"><a id="l00773" name="l00773"></a><span class="lineno"> 773</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">pthread_atfork</a>(</div>
|
|
<div class="line"><a id="l00774" name="l00774"></a><span class="lineno"> 774</span> +[]{ <a class="code hl_function" href="namespacepocketfft_1_1detail_1_1threading.html#a7ec2b3f99232bd0f15f7b022c59d139a">get_pool</a>().<a class="code hl_function" href="classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#a0c7c29290fde806031c497f24c4ad411">shutdown</a>(); }, <span class="comment">// prepare</span></div>
|
|
<div class="line"><a id="l00775" name="l00775"></a><span class="lineno"> 775</span> +[]{ <a class="code hl_function" href="namespacepocketfft_1_1detail_1_1threading.html#a7ec2b3f99232bd0f15f7b022c59d139a">get_pool</a>().<a class="code hl_function" href="classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#a51d252df8d0cd060f15be8ba2bfe3288">restart</a>(); }, <span class="comment">// parent</span></div>
|
|
<div class="line"><a id="l00776" name="l00776"></a><span class="lineno"> 776</span> +[]{ <a class="code hl_function" href="namespacepocketfft_1_1detail_1_1threading.html#a7ec2b3f99232bd0f15f7b022c59d139a">get_pool</a>().<a class="code hl_function" href="classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#a51d252df8d0cd060f15be8ba2bfe3288">restart</a>(); } <span class="comment">// child</span></div>
|
|
<div class="line"><a id="l00777" name="l00777"></a><span class="lineno"> 777</span> );</div>
|
|
<div class="line"><a id="l00778" name="l00778"></a><span class="lineno"> 778</span> });</div>
|
|
<div class="line"><a id="l00779" name="l00779"></a><span class="lineno"> 779</span><span class="preprocessor">#endif</span></div>
|
|
<div class="line"><a id="l00780" name="l00780"></a><span class="lineno"> 780</span> </div>
|
|
<div class="line"><a id="l00781" name="l00781"></a><span class="lineno"> 781</span> <span class="keywordflow">return</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">pool</a>;</div>
|
|
<div class="line"><a id="l00782" name="l00782"></a><span class="lineno"> 782</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l00783" name="l00783"></a><span class="lineno"> 783</span> </div>
|
|
<div class="line"><a id="l00785" name="l00785"></a><span class="lineno"> 785</span><span class="keyword">template</span> <<span class="keyword">typename</span> Func></div>
|
|
<div class="foldopen" id="foldopen00786" data-start="{" data-end="}">
|
|
<div class="line"><a id="l00786" name="l00786"></a><span class="lineno"><a class="line" href="namespacepocketfft_1_1detail_1_1threading.html#a4fcf674db39f0e2c1c59d48491daed6e"> 786</a></span><span class="keywordtype">void</span> <a class="code hl_function" href="namespacepocketfft_1_1detail_1_1threading.html#a4fcf674db39f0e2c1c59d48491daed6e">thread_map</a>(<span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nthreads</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">Func</a> <a class="code hl_variable" href="types_2bf16_8h.html#af900396d7b72ff2a7002e8befe8cf8f1">f</a>)</div>
|
|
<div class="line"><a id="l00787" name="l00787"></a><span class="lineno"> 787</span> {</div>
|
|
<div class="line"><a id="l00788" name="l00788"></a><span class="lineno"> 788</span> <span class="keywordflow">if</span> (<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nthreads</a> == 0)</div>
|
|
<div class="line"><a id="l00789" name="l00789"></a><span class="lineno"> 789</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nthreads</a> = <a class="code hl_variable" href="namespacepocketfft_1_1detail_1_1threading.html#a2d5c0729f0b66cf061918baea4337d70">max_threads</a>;</div>
|
|
<div class="line"><a id="l00790" name="l00790"></a><span class="lineno"> 790</span> </div>
|
|
<div class="line"><a id="l00791" name="l00791"></a><span class="lineno"> 791</span> <span class="keywordflow">if</span> (<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nthreads</a> == 1)</div>
|
|
<div class="line"><a id="l00792" name="l00792"></a><span class="lineno"> 792</span> { <a class="code hl_variable" href="types_2bf16_8h.html#af900396d7b72ff2a7002e8befe8cf8f1">f</a>(); <span class="keywordflow">return</span>; }</div>
|
|
<div class="line"><a id="l00793" name="l00793"></a><span class="lineno"> 793</span> </div>
|
|
<div class="line"><a id="l00794" name="l00794"></a><span class="lineno"> 794</span> <span class="keyword">auto</span> & <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">pool</a> = <a class="code hl_function" href="namespacepocketfft_1_1detail_1_1threading.html#a7ec2b3f99232bd0f15f7b022c59d139a">get_pool</a>();</div>
|
|
<div class="line"><a id="l00795" name="l00795"></a><span class="lineno"> 795</span> <a class="code hl_class" href="classpocketfft_1_1detail_1_1threading_1_1latch.html">latch</a> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">counter</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nthreads</a>);</div>
|
|
<div class="line"><a id="l00796" name="l00796"></a><span class="lineno"> 796</span> std::exception_ptr <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ex</a>;</div>
|
|
<div class="line"><a id="l00797" name="l00797"></a><span class="lineno"> 797</span> std::mutex <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ex_mut</a>;</div>
|
|
<div class="line"><a id="l00798" name="l00798"></a><span class="lineno"> 798</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i=0; i<<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nthreads</a>; ++i)</div>
|
|
<div class="line"><a id="l00799" name="l00799"></a><span class="lineno"> 799</span> {</div>
|
|
<div class="line"><a id="l00800" name="l00800"></a><span class="lineno"> 800</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">pool</a>.submit(</div>
|
|
<div class="line"><a id="l00801" name="l00801"></a><span class="lineno"> 801</span> [&<a class="code hl_variable" href="types_2bf16_8h.html#af900396d7b72ff2a7002e8befe8cf8f1">f</a>, &<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">counter</a>, &<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ex</a>, &<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ex_mut</a>, i, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nthreads</a>] {</div>
|
|
<div class="line"><a id="l00802" name="l00802"></a><span class="lineno"> 802</span> <a class="code hl_function" href="namespacepocketfft_1_1detail_1_1threading.html#aebe85d6273d92c7d3728e2c621ccc82b">thread_id</a>() = i;</div>
|
|
<div class="line"><a id="l00803" name="l00803"></a><span class="lineno"> 803</span> <a class="code hl_function" href="namespacepocketfft_1_1detail_1_1threading.html#af5432c2e25aed679a73fe7b29534c833">num_threads</a>() = <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nthreads</a>;</div>
|
|
<div class="line"><a id="l00804" name="l00804"></a><span class="lineno"> 804</span> <span class="keywordflow">try</span> { <a class="code hl_variable" href="types_2bf16_8h.html#af900396d7b72ff2a7002e8befe8cf8f1">f</a>(); }</div>
|
|
<div class="line"><a id="l00805" name="l00805"></a><span class="lineno"> 805</span> <span class="keywordflow">catch</span> (...)</div>
|
|
<div class="line"><a id="l00806" name="l00806"></a><span class="lineno"> 806</span> {</div>
|
|
<div class="line"><a id="l00807" name="l00807"></a><span class="lineno"> 807</span> std::lock_guard<std::mutex> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">lock</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ex_mut</a>);</div>
|
|
<div class="line"><a id="l00808" name="l00808"></a><span class="lineno"> 808</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ex</a> = std::current_exception();</div>
|
|
<div class="line"><a id="l00809" name="l00809"></a><span class="lineno"> 809</span> }</div>
|
|
<div class="line"><a id="l00810" name="l00810"></a><span class="lineno"> 810</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">counter</a>.count_down();</div>
|
|
<div class="line"><a id="l00811" name="l00811"></a><span class="lineno"> 811</span> });</div>
|
|
<div class="line"><a id="l00812" name="l00812"></a><span class="lineno"> 812</span> }</div>
|
|
<div class="line"><a id="l00813" name="l00813"></a><span class="lineno"> 813</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">counter</a>.wait();</div>
|
|
<div class="line"><a id="l00814" name="l00814"></a><span class="lineno"> 814</span> <span class="keywordflow">if</span> (<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ex</a>)</div>
|
|
<div class="line"><a id="l00815" name="l00815"></a><span class="lineno"> 815</span> std::rethrow_exception(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ex</a>);</div>
|
|
<div class="line"><a id="l00816" name="l00816"></a><span class="lineno"> 816</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l00817" name="l00817"></a><span class="lineno"> 817</span> </div>
|
|
<div class="line"><a id="l00818" name="l00818"></a><span class="lineno"> 818</span><span class="preprocessor">#endif</span></div>
|
|
<div class="line"><a id="l00819" name="l00819"></a><span class="lineno"> 819</span> </div>
|
|
<div class="line"><a id="l00820" name="l00820"></a><span class="lineno"> 820</span>}</div>
|
|
</div>
|
|
<div class="line"><a id="l00821" name="l00821"></a><span class="lineno"> 821</span> </div>
|
|
<div class="line"><a id="l00822" name="l00822"></a><span class="lineno"> 822</span><span class="comment">//</span></div>
|
|
<div class="line"><a id="l00823" name="l00823"></a><span class="lineno"> 823</span><span class="comment">// complex FFTPACK transforms</span></div>
|
|
<div class="line"><a id="l00824" name="l00824"></a><span class="lineno"> 824</span><span class="comment">//</span></div>
|
|
<div class="line"><a id="l00825" name="l00825"></a><span class="lineno"> 825</span> </div>
|
|
<div class="foldopen" id="foldopen00826" data-start="{" data-end="};">
|
|
<div class="line"><a id="l00826" name="l00826"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1cfftp.html"> 826</a></span><span class="keyword">template</span><<span class="keyword">typename</span> T0> <span class="keyword">class </span><a class="code hl_class" href="classpocketfft_1_1detail_1_1cfftp.html">cfftp</a></div>
|
|
<div class="line"><a id="l00827" name="l00827"></a><span class="lineno"> 827</span> {</div>
|
|
<div class="line"><a id="l00828" name="l00828"></a><span class="lineno"> 828</span> <span class="keyword">private</span>:</div>
|
|
<div class="line"><a id="l00829" name="l00829"></a><span class="lineno"> 829</span> <span class="keyword">struct </span>fctdata</div>
|
|
<div class="line"><a id="l00830" name="l00830"></a><span class="lineno"> 830</span> {</div>
|
|
<div class="line"><a id="l00831" name="l00831"></a><span class="lineno"> 831</span> <span class="keywordtype">size_t</span> fct;</div>
|
|
<div class="line"><a id="l00832" name="l00832"></a><span class="lineno"> 832</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cmplx<T0></a> *tw, *tws;</div>
|
|
<div class="line"><a id="l00833" name="l00833"></a><span class="lineno"> 833</span> };</div>
|
|
<div class="line"><a id="l00834" name="l00834"></a><span class="lineno"> 834</span> </div>
|
|
<div class="line"><a id="l00835" name="l00835"></a><span class="lineno"> 835</span> <span class="keywordtype">size_t</span> length;</div>
|
|
<div class="line"><a id="l00836" name="l00836"></a><span class="lineno"> 836</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">arr<cmplx<T0></a>> mem;</div>
|
|
<div class="line"><a id="l00837" name="l00837"></a><span class="lineno"> 837</span> std::vector<fctdata> fact;</div>
|
|
<div class="line"><a id="l00838" name="l00838"></a><span class="lineno"> 838</span> </div>
|
|
<div class="line"><a id="l00839" name="l00839"></a><span class="lineno"> 839</span> <span class="keywordtype">void</span> add_factor(<span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">factor</a>)</div>
|
|
<div class="line"><a id="l00840" name="l00840"></a><span class="lineno"> 840</span> { fact.push_back({<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">factor</a>, <span class="keyword">nullptr</span>, <span class="keyword">nullptr</span>}); }</div>
|
|
<div class="line"><a id="l00841" name="l00841"></a><span class="lineno"> 841</span> </div>
|
|
<div class="line"><a id="l00842" name="l00842"></a><span class="lineno"> 842</span><span class="keyword">template</span><<span class="keywordtype">bool</span> fwd, <span class="keyword">typename</span> T> <span class="keywordtype">void</span> pass2 (<span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ido</a>, <span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">l1</a>,</div>
|
|
<div class="line"><a id="l00843" name="l00843"></a><span class="lineno"> 843</span> <span class="keyword">const</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a> * <a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cc</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a> * <a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ch</a>,</div>
|
|
<div class="line"><a id="l00844" name="l00844"></a><span class="lineno"> 844</span> <span class="keyword">const</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cmplx<T0></a> * <a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">wa</a>)<span class="keyword"> const</span></div>
|
|
<div class="line"><a id="l00845" name="l00845"></a><span class="lineno"> 845</span><span class="keyword"> </span>{</div>
|
|
<div class="line"><a id="l00846" name="l00846"></a><span class="lineno"> 846</span> <span class="keyword">auto</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">CH</a> = [<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ch</a>,<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ido</a>,<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">l1</a>](<span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">a</a>, <span class="keywordtype">size_t</span> b, <span class="keywordtype">size_t</span> c) -> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a>&</div>
|
|
<div class="line"><a id="l00847" name="l00847"></a><span class="lineno"> 847</span> { <span class="keywordflow">return</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ch</a>[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">a</a>+<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ido</a>*(b+<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">l1</a>*c)]; };</div>
|
|
<div class="line"><a id="l00848" name="l00848"></a><span class="lineno"> 848</span> <span class="keyword">auto</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">CC</a> = [<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cc</a>,<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ido</a>](<span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">a</a>, <span class="keywordtype">size_t</span> b, <span class="keywordtype">size_t</span> c) -> <span class="keyword">const</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a>&</div>
|
|
<div class="line"><a id="l00849" name="l00849"></a><span class="lineno"> 849</span> { <span class="keywordflow">return</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cc</a>[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">a</a>+<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ido</a>*(b+2*c)]; };</div>
|
|
<div class="line"><a id="l00850" name="l00850"></a><span class="lineno"> 850</span> <span class="keyword">auto</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">WA</a> = [<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">wa</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ido</a>](<span class="keywordtype">size_t</span> x, <span class="keywordtype">size_t</span> i)</div>
|
|
<div class="line"><a id="l00851" name="l00851"></a><span class="lineno"> 851</span> { <span class="keywordflow">return</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">wa</a>[i-1+x*(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ido</a>-1)]; };</div>
|
|
<div class="line"><a id="l00852" name="l00852"></a><span class="lineno"> 852</span> </div>
|
|
<div class="line"><a id="l00853" name="l00853"></a><span class="lineno"> 853</span> <span class="keywordflow">if</span> (<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ido</a>==1)</div>
|
|
<div class="line"><a id="l00854" name="l00854"></a><span class="lineno"> 854</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>=0; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a><<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">l1</a>; ++<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>)</div>
|
|
<div class="line"><a id="l00855" name="l00855"></a><span class="lineno"> 855</span> {</div>
|
|
<div class="line"><a id="l00856" name="l00856"></a><span class="lineno"> 856</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">CH</a>(0,<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>,0) = <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">CC</a>(0,0,<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>)+<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">CC</a>(0,1,<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>);</div>
|
|
<div class="line"><a id="l00857" name="l00857"></a><span class="lineno"> 857</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">CH</a>(0,<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>,1) = <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">CC</a>(0,0,<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>)-<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">CC</a>(0,1,<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>);</div>
|
|
<div class="line"><a id="l00858" name="l00858"></a><span class="lineno"> 858</span> }</div>
|
|
<div class="line"><a id="l00859" name="l00859"></a><span class="lineno"> 859</span> <span class="keywordflow">else</span></div>
|
|
<div class="line"><a id="l00860" name="l00860"></a><span class="lineno"> 860</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>=0; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a><<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">l1</a>; ++<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>)</div>
|
|
<div class="line"><a id="l00861" name="l00861"></a><span class="lineno"> 861</span> {</div>
|
|
<div class="line"><a id="l00862" name="l00862"></a><span class="lineno"> 862</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">CH</a>(0,<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>,0) = <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">CC</a>(0,0,<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>)+<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">CC</a>(0,1,<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>);</div>
|
|
<div class="line"><a id="l00863" name="l00863"></a><span class="lineno"> 863</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">CH</a>(0,<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>,1) = <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">CC</a>(0,0,<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>)-<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">CC</a>(0,1,<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>);</div>
|
|
<div class="line"><a id="l00864" name="l00864"></a><span class="lineno"> 864</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i=1; i<<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ido</a>; ++i)</div>
|
|
<div class="line"><a id="l00865" name="l00865"></a><span class="lineno"> 865</span> {</div>
|
|
<div class="line"><a id="l00866" name="l00866"></a><span class="lineno"> 866</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">CH</a>(i,<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>,0) = <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">CC</a>(i,0,<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>)+<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">CC</a>(i,1,<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>);</div>
|
|
<div class="line"><a id="l00867" name="l00867"></a><span class="lineno"> 867</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">special_mul<fwd></a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">CC</a>(i,0,<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>)-<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">CC</a>(i,1,<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>),<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">WA</a>(0,i),<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">CH</a>(i,<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>,1));</div>
|
|
<div class="line"><a id="l00868" name="l00868"></a><span class="lineno"> 868</span> }</div>
|
|
<div class="line"><a id="l00869" name="l00869"></a><span class="lineno"> 869</span> }</div>
|
|
<div class="line"><a id="l00870" name="l00870"></a><span class="lineno"> 870</span> }</div>
|
|
<div class="line"><a id="l00871" name="l00871"></a><span class="lineno"> 871</span> </div>
|
|
<div class="foldopen" id="foldopen00872" data-start="" data-end="">
|
|
<div class="line"><a id="l00872" name="l00872"></a><span class="lineno"><a class="line" href="pocketfft_8h.html#ae2fd9d433c417f0768fe1b58145b2e59"> 872</a></span><span class="preprocessor">#define POCKETFFT_PREP3(idx) \</span></div>
|
|
<div class="line"><a id="l00873" name="l00873"></a><span class="lineno"> 873</span><span class="preprocessor"> T t0 = CC(idx,0,k), t1, t2; \</span></div>
|
|
<div class="line"><a id="l00874" name="l00874"></a><span class="lineno"> 874</span><span class="preprocessor"> PM (t1,t2,CC(idx,1,k),CC(idx,2,k)); \</span></div>
|
|
<div class="line"><a id="l00875" name="l00875"></a><span class="lineno"> 875</span><span class="preprocessor"> CH(idx,k,0)=t0+t1;</span></div>
|
|
</div>
|
|
<div class="foldopen" id="foldopen00876" data-start="" data-end="">
|
|
<div class="line"><a id="l00876" name="l00876"></a><span class="lineno"><a class="line" href="pocketfft_8h.html#ac112b26e5130636ac1d91c2f0af45e0b"> 876</a></span><span class="preprocessor">#define POCKETFFT_PARTSTEP3a(u1,u2,twr,twi) \</span></div>
|
|
<div class="line"><a id="l00877" name="l00877"></a><span class="lineno"> 877</span><span class="preprocessor"> { \</span></div>
|
|
<div class="line"><a id="l00878" name="l00878"></a><span class="lineno"> 878</span><span class="preprocessor"> T ca=t0+t1*twr; \</span></div>
|
|
<div class="line"><a id="l00879" name="l00879"></a><span class="lineno"> 879</span><span class="preprocessor"> T cb{-t2.i*twi, t2.r*twi}; \</span></div>
|
|
<div class="line"><a id="l00880" name="l00880"></a><span class="lineno"> 880</span><span class="preprocessor"> PM(CH(0,k,u1),CH(0,k,u2),ca,cb) ;\</span></div>
|
|
<div class="line"><a id="l00881" name="l00881"></a><span class="lineno"> 881</span><span class="preprocessor"> }</span></div>
|
|
</div>
|
|
<div class="foldopen" id="foldopen00882" data-start="" data-end="">
|
|
<div class="line"><a id="l00882" name="l00882"></a><span class="lineno"><a class="line" href="pocketfft_8h.html#a41e646e5535a3a1c6e0d0e67122382f5"> 882</a></span><span class="preprocessor">#define POCKETFFT_PARTSTEP3b(u1,u2,twr,twi) \</span></div>
|
|
<div class="line"><a id="l00883" name="l00883"></a><span class="lineno"> 883</span><span class="preprocessor"> { \</span></div>
|
|
<div class="line"><a id="l00884" name="l00884"></a><span class="lineno"> 884</span><span class="preprocessor"> T ca=t0+t1*twr; \</span></div>
|
|
<div class="line"><a id="l00885" name="l00885"></a><span class="lineno"> 885</span><span class="preprocessor"> T cb{-t2.i*twi, t2.r*twi}; \</span></div>
|
|
<div class="line"><a id="l00886" name="l00886"></a><span class="lineno"> 886</span><span class="preprocessor"> special_mul<fwd>(ca+cb,WA(u1-1,i),CH(i,k,u1)); \</span></div>
|
|
<div class="line"><a id="l00887" name="l00887"></a><span class="lineno"> 887</span><span class="preprocessor"> special_mul<fwd>(ca-cb,WA(u2-1,i),CH(i,k,u2)); \</span></div>
|
|
<div class="line"><a id="l00888" name="l00888"></a><span class="lineno"> 888</span><span class="preprocessor"> }</span></div>
|
|
</div>
|
|
<div class="line"><a id="l00889" name="l00889"></a><span class="lineno"> 889</span><span class="keyword">template</span><<span class="keywordtype">bool</span> fwd, <span class="keyword">typename</span> T> <span class="keywordtype">void</span> pass3 (<span class="keywordtype">size_t</span> ido, <span class="keywordtype">size_t</span> l1,</div>
|
|
<div class="line"><a id="l00890" name="l00890"></a><span class="lineno"> 890</span> <span class="keyword">const</span> T * <a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> cc, T * <a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> ch,</div>
|
|
<div class="line"><a id="l00891" name="l00891"></a><span class="lineno"> 891</span> <span class="keyword">const</span> cmplx<T0> * <a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> wa)<span class="keyword"> const</span></div>
|
|
<div class="line"><a id="l00892" name="l00892"></a><span class="lineno"> 892</span><span class="keyword"> </span>{</div>
|
|
<div class="line"><a id="l00893" name="l00893"></a><span class="lineno"> 893</span> <span class="keyword">constexpr</span> T0 tw1r=-0.5,</div>
|
|
<div class="line"><a id="l00894" name="l00894"></a><span class="lineno"> 894</span> tw1i= (fwd ? -1: 1) * T0(0.8660254037844386467637231707529362L);</div>
|
|
<div class="line"><a id="l00895" name="l00895"></a><span class="lineno"> 895</span> </div>
|
|
<div class="line"><a id="l00896" name="l00896"></a><span class="lineno"> 896</span> <span class="keyword">auto</span> CH = [ch,ido,l1](<span class="keywordtype">size_t</span> a, <span class="keywordtype">size_t</span> b, <span class="keywordtype">size_t</span> c) -> T&</div>
|
|
<div class="line"><a id="l00897" name="l00897"></a><span class="lineno"> 897</span> { <span class="keywordflow">return</span> ch[a+ido*(b+l1*c)]; };</div>
|
|
<div class="line"><a id="l00898" name="l00898"></a><span class="lineno"> 898</span> <span class="keyword">auto</span> CC = [cc,ido](<span class="keywordtype">size_t</span> a, <span class="keywordtype">size_t</span> b, <span class="keywordtype">size_t</span> c) -> <span class="keyword">const</span> T&</div>
|
|
<div class="line"><a id="l00899" name="l00899"></a><span class="lineno"> 899</span> { <span class="keywordflow">return</span> cc[a+ido*(b+3*c)]; };</div>
|
|
<div class="line"><a id="l00900" name="l00900"></a><span class="lineno"> 900</span> <span class="keyword">auto</span> WA = [wa, ido](<span class="keywordtype">size_t</span> x, <span class="keywordtype">size_t</span> i)</div>
|
|
<div class="line"><a id="l00901" name="l00901"></a><span class="lineno"> 901</span> { <span class="keywordflow">return</span> wa[i-1+x*(ido-1)]; };</div>
|
|
<div class="line"><a id="l00902" name="l00902"></a><span class="lineno"> 902</span> </div>
|
|
<div class="line"><a id="l00903" name="l00903"></a><span class="lineno"> 903</span> <span class="keywordflow">if</span> (ido==1)</div>
|
|
<div class="line"><a id="l00904" name="l00904"></a><span class="lineno"> 904</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> k=0; k<l1; ++k)</div>
|
|
<div class="line"><a id="l00905" name="l00905"></a><span class="lineno"> 905</span> {</div>
|
|
<div class="line"><a id="l00906" name="l00906"></a><span class="lineno"> 906</span> <a class="code hl_define" href="pocketfft_8h.html#ae2fd9d433c417f0768fe1b58145b2e59">POCKETFFT_PREP3</a>(0)</div>
|
|
<div class="line"><a id="l00907" name="l00907"></a><span class="lineno"> 907</span> <a class="code hl_define" href="pocketfft_8h.html#ac112b26e5130636ac1d91c2f0af45e0b">POCKETFFT_PARTSTEP3a</a>(1,2,tw1r,tw1i)</div>
|
|
<div class="line"><a id="l00908" name="l00908"></a><span class="lineno"> 908</span> }</div>
|
|
<div class="line"><a id="l00909" name="l00909"></a><span class="lineno"> 909</span> <span class="keywordflow">else</span></div>
|
|
<div class="line"><a id="l00910" name="l00910"></a><span class="lineno"> 910</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> k=0; k<l1; ++k)</div>
|
|
<div class="line"><a id="l00911" name="l00911"></a><span class="lineno"> 911</span> {</div>
|
|
<div class="line"><a id="l00912" name="l00912"></a><span class="lineno"> 912</span> {</div>
|
|
<div class="line"><a id="l00913" name="l00913"></a><span class="lineno"> 913</span> <a class="code hl_define" href="pocketfft_8h.html#ae2fd9d433c417f0768fe1b58145b2e59">POCKETFFT_PREP3</a>(0)</div>
|
|
<div class="line"><a id="l00914" name="l00914"></a><span class="lineno"> 914</span> <a class="code hl_define" href="pocketfft_8h.html#ac112b26e5130636ac1d91c2f0af45e0b">POCKETFFT_PARTSTEP3a</a>(1,2,tw1r,tw1i)</div>
|
|
<div class="line"><a id="l00915" name="l00915"></a><span class="lineno"> 915</span> }</div>
|
|
<div class="line"><a id="l00916" name="l00916"></a><span class="lineno"> 916</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i=1; i<ido; ++i)</div>
|
|
<div class="line"><a id="l00917" name="l00917"></a><span class="lineno"> 917</span> {</div>
|
|
<div class="line"><a id="l00918" name="l00918"></a><span class="lineno"> 918</span> <a class="code hl_define" href="pocketfft_8h.html#ae2fd9d433c417f0768fe1b58145b2e59">POCKETFFT_PREP3</a>(i)</div>
|
|
<div class="line"><a id="l00919" name="l00919"></a><span class="lineno"> 919</span> <a class="code hl_define" href="pocketfft_8h.html#a41e646e5535a3a1c6e0d0e67122382f5">POCKETFFT_PARTSTEP3b</a>(1,2,tw1r,tw1i)</div>
|
|
<div class="line"><a id="l00920" name="l00920"></a><span class="lineno"> 920</span> }</div>
|
|
<div class="line"><a id="l00921" name="l00921"></a><span class="lineno"> 921</span> }</div>
|
|
<div class="line"><a id="l00922" name="l00922"></a><span class="lineno"> 922</span> }</div>
|
|
<div class="line"><a id="l00923" name="l00923"></a><span class="lineno"> 923</span> </div>
|
|
<div class="line"><a id="l00924" name="l00924"></a><span class="lineno"> 924</span><span class="preprocessor">#undef POCKETFFT_PARTSTEP3b</span></div>
|
|
<div class="line"><a id="l00925" name="l00925"></a><span class="lineno"> 925</span><span class="preprocessor">#undef POCKETFFT_PARTSTEP3a</span></div>
|
|
<div class="line"><a id="l00926" name="l00926"></a><span class="lineno"> 926</span><span class="preprocessor">#undef POCKETFFT_PREP3</span></div>
|
|
<div class="line"><a id="l00927" name="l00927"></a><span class="lineno"> 927</span> </div>
|
|
<div class="line"><a id="l00928" name="l00928"></a><span class="lineno"> 928</span><span class="keyword">template</span><<span class="keywordtype">bool</span> fwd, <span class="keyword">typename</span> T> <span class="keywordtype">void</span> pass4 (<span class="keywordtype">size_t</span> ido, <span class="keywordtype">size_t</span> l1,</div>
|
|
<div class="line"><a id="l00929" name="l00929"></a><span class="lineno"> 929</span> <span class="keyword">const</span> T * <a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> cc, T * <a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> ch,</div>
|
|
<div class="line"><a id="l00930" name="l00930"></a><span class="lineno"> 930</span> <span class="keyword">const</span> cmplx<T0> * <a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> wa)<span class="keyword"> const</span></div>
|
|
<div class="line"><a id="l00931" name="l00931"></a><span class="lineno"> 931</span><span class="keyword"> </span>{</div>
|
|
<div class="line"><a id="l00932" name="l00932"></a><span class="lineno"> 932</span> <span class="keyword">auto</span> CH = [ch,ido,l1](<span class="keywordtype">size_t</span> a, <span class="keywordtype">size_t</span> b, <span class="keywordtype">size_t</span> c) -> T&</div>
|
|
<div class="line"><a id="l00933" name="l00933"></a><span class="lineno"> 933</span> { <span class="keywordflow">return</span> ch[a+ido*(b+l1*c)]; };</div>
|
|
<div class="line"><a id="l00934" name="l00934"></a><span class="lineno"> 934</span> <span class="keyword">auto</span> CC = [cc,ido](<span class="keywordtype">size_t</span> a, <span class="keywordtype">size_t</span> b, <span class="keywordtype">size_t</span> c) -> <span class="keyword">const</span> T&</div>
|
|
<div class="line"><a id="l00935" name="l00935"></a><span class="lineno"> 935</span> { <span class="keywordflow">return</span> cc[a+ido*(b+4*c)]; };</div>
|
|
<div class="line"><a id="l00936" name="l00936"></a><span class="lineno"> 936</span> <span class="keyword">auto</span> WA = [wa, ido](<span class="keywordtype">size_t</span> x, <span class="keywordtype">size_t</span> i)</div>
|
|
<div class="line"><a id="l00937" name="l00937"></a><span class="lineno"> 937</span> { <span class="keywordflow">return</span> wa[i-1+x*(ido-1)]; };</div>
|
|
<div class="line"><a id="l00938" name="l00938"></a><span class="lineno"> 938</span> </div>
|
|
<div class="line"><a id="l00939" name="l00939"></a><span class="lineno"> 939</span> <span class="keywordflow">if</span> (ido==1)</div>
|
|
<div class="line"><a id="l00940" name="l00940"></a><span class="lineno"> 940</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> k=0; k<l1; ++k)</div>
|
|
<div class="line"><a id="l00941" name="l00941"></a><span class="lineno"> 941</span> {</div>
|
|
<div class="line"><a id="l00942" name="l00942"></a><span class="lineno"> 942</span> T t1, t2, t3, t4;</div>
|
|
<div class="line"><a id="l00943" name="l00943"></a><span class="lineno"> 943</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a>(t2,t1,CC(0,0,k),CC(0,2,k));</div>
|
|
<div class="line"><a id="l00944" name="l00944"></a><span class="lineno"> 944</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a>(t3,t4,CC(0,1,k),CC(0,3,k));</div>
|
|
<div class="line"><a id="l00945" name="l00945"></a><span class="lineno"> 945</span> ROTX90<fwd>(t4);</div>
|
|
<div class="line"><a id="l00946" name="l00946"></a><span class="lineno"> 946</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a>(CH(0,k,0),CH(0,k,2),t2,t3);</div>
|
|
<div class="line"><a id="l00947" name="l00947"></a><span class="lineno"> 947</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a>(CH(0,k,1),CH(0,k,3),t1,t4);</div>
|
|
<div class="line"><a id="l00948" name="l00948"></a><span class="lineno"> 948</span> }</div>
|
|
<div class="line"><a id="l00949" name="l00949"></a><span class="lineno"> 949</span> <span class="keywordflow">else</span></div>
|
|
<div class="line"><a id="l00950" name="l00950"></a><span class="lineno"> 950</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> k=0; k<l1; ++k)</div>
|
|
<div class="line"><a id="l00951" name="l00951"></a><span class="lineno"> 951</span> {</div>
|
|
<div class="line"><a id="l00952" name="l00952"></a><span class="lineno"> 952</span> {</div>
|
|
<div class="line"><a id="l00953" name="l00953"></a><span class="lineno"> 953</span> T t1, t2, t3, t4;</div>
|
|
<div class="line"><a id="l00954" name="l00954"></a><span class="lineno"> 954</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a>(t2,t1,CC(0,0,k),CC(0,2,k));</div>
|
|
<div class="line"><a id="l00955" name="l00955"></a><span class="lineno"> 955</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a>(t3,t4,CC(0,1,k),CC(0,3,k));</div>
|
|
<div class="line"><a id="l00956" name="l00956"></a><span class="lineno"> 956</span> ROTX90<fwd>(t4);</div>
|
|
<div class="line"><a id="l00957" name="l00957"></a><span class="lineno"> 957</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a>(CH(0,k,0),CH(0,k,2),t2,t3);</div>
|
|
<div class="line"><a id="l00958" name="l00958"></a><span class="lineno"> 958</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a>(CH(0,k,1),CH(0,k,3),t1,t4);</div>
|
|
<div class="line"><a id="l00959" name="l00959"></a><span class="lineno"> 959</span> }</div>
|
|
<div class="line"><a id="l00960" name="l00960"></a><span class="lineno"> 960</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i=1; i<ido; ++i)</div>
|
|
<div class="line"><a id="l00961" name="l00961"></a><span class="lineno"> 961</span> {</div>
|
|
<div class="line"><a id="l00962" name="l00962"></a><span class="lineno"> 962</span> T t1, t2, t3, t4;</div>
|
|
<div class="line"><a id="l00963" name="l00963"></a><span class="lineno"> 963</span> T cc0=CC(i,0,k), cc1=CC(i,1,k),cc2=CC(i,2,k),cc3=CC(i,3,k);</div>
|
|
<div class="line"><a id="l00964" name="l00964"></a><span class="lineno"> 964</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a>(t2,t1,cc0,cc2);</div>
|
|
<div class="line"><a id="l00965" name="l00965"></a><span class="lineno"> 965</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a>(t3,t4,cc1,cc3);</div>
|
|
<div class="line"><a id="l00966" name="l00966"></a><span class="lineno"> 966</span> ROTX90<fwd>(t4);</div>
|
|
<div class="line"><a id="l00967" name="l00967"></a><span class="lineno"> 967</span> CH(i,k,0) = t2+t3;</div>
|
|
<div class="line"><a id="l00968" name="l00968"></a><span class="lineno"> 968</span> special_mul<fwd>(t1+t4,WA(0,i),CH(i,k,1));</div>
|
|
<div class="line"><a id="l00969" name="l00969"></a><span class="lineno"> 969</span> special_mul<fwd>(t2-t3,WA(1,i),CH(i,k,2));</div>
|
|
<div class="line"><a id="l00970" name="l00970"></a><span class="lineno"> 970</span> special_mul<fwd>(t1-t4,WA(2,i),CH(i,k,3));</div>
|
|
<div class="line"><a id="l00971" name="l00971"></a><span class="lineno"> 971</span> }</div>
|
|
<div class="line"><a id="l00972" name="l00972"></a><span class="lineno"> 972</span> }</div>
|
|
<div class="line"><a id="l00973" name="l00973"></a><span class="lineno"> 973</span> }</div>
|
|
<div class="line"><a id="l00974" name="l00974"></a><span class="lineno"> 974</span> </div>
|
|
<div class="foldopen" id="foldopen00975" data-start="" data-end="">
|
|
<div class="line"><a id="l00975" name="l00975"></a><span class="lineno"><a class="line" href="pocketfft_8h.html#a73077c26d2a82754db2a9c48bc0e11a6"> 975</a></span><span class="preprocessor">#define POCKETFFT_PREP5(idx) \</span></div>
|
|
<div class="line"><a id="l00976" name="l00976"></a><span class="lineno"> 976</span><span class="preprocessor"> T t0 = CC(idx,0,k), t1, t2, t3, t4; \</span></div>
|
|
<div class="line"><a id="l00977" name="l00977"></a><span class="lineno"> 977</span><span class="preprocessor"> PM (t1,t4,CC(idx,1,k),CC(idx,4,k)); \</span></div>
|
|
<div class="line"><a id="l00978" name="l00978"></a><span class="lineno"> 978</span><span class="preprocessor"> PM (t2,t3,CC(idx,2,k),CC(idx,3,k)); \</span></div>
|
|
<div class="line"><a id="l00979" name="l00979"></a><span class="lineno"> 979</span><span class="preprocessor"> CH(idx,k,0).r=t0.r+t1.r+t2.r; \</span></div>
|
|
<div class="line"><a id="l00980" name="l00980"></a><span class="lineno"> 980</span><span class="preprocessor"> CH(idx,k,0).i=t0.i+t1.i+t2.i;</span></div>
|
|
</div>
|
|
<div class="line"><a id="l00981" name="l00981"></a><span class="lineno"> 981</span> </div>
|
|
<div class="foldopen" id="foldopen00982" data-start="" data-end="">
|
|
<div class="line"><a id="l00982" name="l00982"></a><span class="lineno"><a class="line" href="pocketfft_8h.html#a078bc2bd38ab0ffb15b981878c9de03c"> 982</a></span><span class="preprocessor">#define POCKETFFT_PARTSTEP5a(u1,u2,twar,twbr,twai,twbi) \</span></div>
|
|
<div class="line"><a id="l00983" name="l00983"></a><span class="lineno"> 983</span><span class="preprocessor"> { \</span></div>
|
|
<div class="line"><a id="l00984" name="l00984"></a><span class="lineno"> 984</span><span class="preprocessor"> T ca,cb; \</span></div>
|
|
<div class="line"><a id="l00985" name="l00985"></a><span class="lineno"> 985</span><span class="preprocessor"> ca.r=t0.r+twar*t1.r+twbr*t2.r; \</span></div>
|
|
<div class="line"><a id="l00986" name="l00986"></a><span class="lineno"> 986</span><span class="preprocessor"> ca.i=t0.i+twar*t1.i+twbr*t2.i; \</span></div>
|
|
<div class="line"><a id="l00987" name="l00987"></a><span class="lineno"> 987</span><span class="preprocessor"> cb.i=twai*t4.r twbi*t3.r; \</span></div>
|
|
<div class="line"><a id="l00988" name="l00988"></a><span class="lineno"> 988</span><span class="preprocessor"> cb.r=-(twai*t4.i twbi*t3.i); \</span></div>
|
|
<div class="line"><a id="l00989" name="l00989"></a><span class="lineno"> 989</span><span class="preprocessor"> PM(CH(0,k,u1),CH(0,k,u2),ca,cb); \</span></div>
|
|
<div class="line"><a id="l00990" name="l00990"></a><span class="lineno"> 990</span><span class="preprocessor"> }</span></div>
|
|
</div>
|
|
<div class="line"><a id="l00991" name="l00991"></a><span class="lineno"> 991</span> </div>
|
|
<div class="foldopen" id="foldopen00992" data-start="" data-end="">
|
|
<div class="line"><a id="l00992" name="l00992"></a><span class="lineno"><a class="line" href="pocketfft_8h.html#ab8a5da142555e059c5e9c618f75b46fa"> 992</a></span><span class="preprocessor">#define POCKETFFT_PARTSTEP5b(u1,u2,twar,twbr,twai,twbi) \</span></div>
|
|
<div class="line"><a id="l00993" name="l00993"></a><span class="lineno"> 993</span><span class="preprocessor"> { \</span></div>
|
|
<div class="line"><a id="l00994" name="l00994"></a><span class="lineno"> 994</span><span class="preprocessor"> T ca,cb,da,db; \</span></div>
|
|
<div class="line"><a id="l00995" name="l00995"></a><span class="lineno"> 995</span><span class="preprocessor"> ca.r=t0.r+twar*t1.r+twbr*t2.r; \</span></div>
|
|
<div class="line"><a id="l00996" name="l00996"></a><span class="lineno"> 996</span><span class="preprocessor"> ca.i=t0.i+twar*t1.i+twbr*t2.i; \</span></div>
|
|
<div class="line"><a id="l00997" name="l00997"></a><span class="lineno"> 997</span><span class="preprocessor"> cb.i=twai*t4.r twbi*t3.r; \</span></div>
|
|
<div class="line"><a id="l00998" name="l00998"></a><span class="lineno"> 998</span><span class="preprocessor"> cb.r=-(twai*t4.i twbi*t3.i); \</span></div>
|
|
<div class="line"><a id="l00999" name="l00999"></a><span class="lineno"> 999</span><span class="preprocessor"> special_mul<fwd>(ca+cb,WA(u1-1,i),CH(i,k,u1)); \</span></div>
|
|
<div class="line"><a id="l01000" name="l01000"></a><span class="lineno"> 1000</span><span class="preprocessor"> special_mul<fwd>(ca-cb,WA(u2-1,i),CH(i,k,u2)); \</span></div>
|
|
<div class="line"><a id="l01001" name="l01001"></a><span class="lineno"> 1001</span><span class="preprocessor"> }</span></div>
|
|
</div>
|
|
<div class="line"><a id="l01002" name="l01002"></a><span class="lineno"> 1002</span><span class="keyword">template</span><<span class="keywordtype">bool</span> fwd, <span class="keyword">typename</span> T> <span class="keywordtype">void</span> pass5 (<span class="keywordtype">size_t</span> ido, <span class="keywordtype">size_t</span> l1,</div>
|
|
<div class="line"><a id="l01003" name="l01003"></a><span class="lineno"> 1003</span> <span class="keyword">const</span> T * <a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> cc, T * <a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> ch,</div>
|
|
<div class="line"><a id="l01004" name="l01004"></a><span class="lineno"> 1004</span> <span class="keyword">const</span> cmplx<T0> * <a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> wa)<span class="keyword"> const</span></div>
|
|
<div class="line"><a id="l01005" name="l01005"></a><span class="lineno"> 1005</span><span class="keyword"> </span>{</div>
|
|
<div class="line"><a id="l01006" name="l01006"></a><span class="lineno"> 1006</span> <span class="keyword">constexpr</span> T0 tw1r= T0(0.3090169943749474241022934171828191L),</div>
|
|
<div class="line"><a id="l01007" name="l01007"></a><span class="lineno"> 1007</span> tw1i= (fwd ? -1: 1) * T0(0.9510565162951535721164393333793821L),</div>
|
|
<div class="line"><a id="l01008" name="l01008"></a><span class="lineno"> 1008</span> tw2r= T0(-0.8090169943749474241022934171828191L),</div>
|
|
<div class="line"><a id="l01009" name="l01009"></a><span class="lineno"> 1009</span> tw2i= (fwd ? -1: 1) * T0(0.5877852522924731291687059546390728L);</div>
|
|
<div class="line"><a id="l01010" name="l01010"></a><span class="lineno"> 1010</span> </div>
|
|
<div class="line"><a id="l01011" name="l01011"></a><span class="lineno"> 1011</span> <span class="keyword">auto</span> CH = [ch,ido,l1](<span class="keywordtype">size_t</span> a, <span class="keywordtype">size_t</span> b, <span class="keywordtype">size_t</span> c) -> T&</div>
|
|
<div class="line"><a id="l01012" name="l01012"></a><span class="lineno"> 1012</span> { <span class="keywordflow">return</span> ch[a+ido*(b+l1*c)]; };</div>
|
|
<div class="line"><a id="l01013" name="l01013"></a><span class="lineno"> 1013</span> <span class="keyword">auto</span> CC = [cc,ido](<span class="keywordtype">size_t</span> a, <span class="keywordtype">size_t</span> b, <span class="keywordtype">size_t</span> c) -> <span class="keyword">const</span> T&</div>
|
|
<div class="line"><a id="l01014" name="l01014"></a><span class="lineno"> 1014</span> { <span class="keywordflow">return</span> cc[a+ido*(b+5*c)]; };</div>
|
|
<div class="line"><a id="l01015" name="l01015"></a><span class="lineno"> 1015</span> <span class="keyword">auto</span> WA = [wa, ido](<span class="keywordtype">size_t</span> x, <span class="keywordtype">size_t</span> i)</div>
|
|
<div class="line"><a id="l01016" name="l01016"></a><span class="lineno"> 1016</span> { <span class="keywordflow">return</span> wa[i-1+x*(ido-1)]; };</div>
|
|
<div class="line"><a id="l01017" name="l01017"></a><span class="lineno"> 1017</span> </div>
|
|
<div class="line"><a id="l01018" name="l01018"></a><span class="lineno"> 1018</span> <span class="keywordflow">if</span> (ido==1)</div>
|
|
<div class="line"><a id="l01019" name="l01019"></a><span class="lineno"> 1019</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> k=0; k<l1; ++k)</div>
|
|
<div class="line"><a id="l01020" name="l01020"></a><span class="lineno"> 1020</span> {</div>
|
|
<div class="line"><a id="l01021" name="l01021"></a><span class="lineno"> 1021</span> <a class="code hl_define" href="pocketfft_8h.html#a73077c26d2a82754db2a9c48bc0e11a6">POCKETFFT_PREP5</a>(0)</div>
|
|
<div class="line"><a id="l01022" name="l01022"></a><span class="lineno"> 1022</span> <a class="code hl_define" href="pocketfft_8h.html#a078bc2bd38ab0ffb15b981878c9de03c">POCKETFFT_PARTSTEP5a</a>(1,4,tw1r,tw2r,+tw1i,+tw2i)</div>
|
|
<div class="line"><a id="l01023" name="l01023"></a><span class="lineno"> 1023</span> <a class="code hl_define" href="pocketfft_8h.html#a078bc2bd38ab0ffb15b981878c9de03c">POCKETFFT_PARTSTEP5a</a>(2,3,tw2r,tw1r,+tw2i,-tw1i)</div>
|
|
<div class="line"><a id="l01024" name="l01024"></a><span class="lineno"> 1024</span> }</div>
|
|
<div class="line"><a id="l01025" name="l01025"></a><span class="lineno"> 1025</span> <span class="keywordflow">else</span></div>
|
|
<div class="line"><a id="l01026" name="l01026"></a><span class="lineno"> 1026</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> k=0; k<l1; ++k)</div>
|
|
<div class="line"><a id="l01027" name="l01027"></a><span class="lineno"> 1027</span> {</div>
|
|
<div class="line"><a id="l01028" name="l01028"></a><span class="lineno"> 1028</span> {</div>
|
|
<div class="line"><a id="l01029" name="l01029"></a><span class="lineno"> 1029</span> <a class="code hl_define" href="pocketfft_8h.html#a73077c26d2a82754db2a9c48bc0e11a6">POCKETFFT_PREP5</a>(0)</div>
|
|
<div class="line"><a id="l01030" name="l01030"></a><span class="lineno"> 1030</span> <a class="code hl_define" href="pocketfft_8h.html#a078bc2bd38ab0ffb15b981878c9de03c">POCKETFFT_PARTSTEP5a</a>(1,4,tw1r,tw2r,+tw1i,+tw2i)</div>
|
|
<div class="line"><a id="l01031" name="l01031"></a><span class="lineno"> 1031</span> <a class="code hl_define" href="pocketfft_8h.html#a078bc2bd38ab0ffb15b981878c9de03c">POCKETFFT_PARTSTEP5a</a>(2,3,tw2r,tw1r,+tw2i,-tw1i)</div>
|
|
<div class="line"><a id="l01032" name="l01032"></a><span class="lineno"> 1032</span> }</div>
|
|
<div class="line"><a id="l01033" name="l01033"></a><span class="lineno"> 1033</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i=1; i<ido; ++i)</div>
|
|
<div class="line"><a id="l01034" name="l01034"></a><span class="lineno"> 1034</span> {</div>
|
|
<div class="line"><a id="l01035" name="l01035"></a><span class="lineno"> 1035</span> <a class="code hl_define" href="pocketfft_8h.html#a73077c26d2a82754db2a9c48bc0e11a6">POCKETFFT_PREP5</a>(i)</div>
|
|
<div class="line"><a id="l01036" name="l01036"></a><span class="lineno"> 1036</span> <a class="code hl_define" href="pocketfft_8h.html#ab8a5da142555e059c5e9c618f75b46fa">POCKETFFT_PARTSTEP5b</a>(1,4,tw1r,tw2r,+tw1i,+tw2i)</div>
|
|
<div class="line"><a id="l01037" name="l01037"></a><span class="lineno"> 1037</span> <a class="code hl_define" href="pocketfft_8h.html#ab8a5da142555e059c5e9c618f75b46fa">POCKETFFT_PARTSTEP5b</a>(2,3,tw2r,tw1r,+tw2i,-tw1i)</div>
|
|
<div class="line"><a id="l01038" name="l01038"></a><span class="lineno"> 1038</span> }</div>
|
|
<div class="line"><a id="l01039" name="l01039"></a><span class="lineno"> 1039</span> }</div>
|
|
<div class="line"><a id="l01040" name="l01040"></a><span class="lineno"> 1040</span> }</div>
|
|
<div class="line"><a id="l01041" name="l01041"></a><span class="lineno"> 1041</span> </div>
|
|
<div class="line"><a id="l01042" name="l01042"></a><span class="lineno"> 1042</span><span class="preprocessor">#undef POCKETFFT_PARTSTEP5b</span></div>
|
|
<div class="line"><a id="l01043" name="l01043"></a><span class="lineno"> 1043</span><span class="preprocessor">#undef POCKETFFT_PARTSTEP5a</span></div>
|
|
<div class="line"><a id="l01044" name="l01044"></a><span class="lineno"> 1044</span><span class="preprocessor">#undef POCKETFFT_PREP5</span></div>
|
|
<div class="line"><a id="l01045" name="l01045"></a><span class="lineno"> 1045</span> </div>
|
|
<div class="foldopen" id="foldopen01046" data-start="" data-end="">
|
|
<div class="line"><a id="l01046" name="l01046"></a><span class="lineno"><a class="line" href="pocketfft_8h.html#ae7c4d0cda5b3824f84eac54addabd6ec"> 1046</a></span><span class="preprocessor">#define POCKETFFT_PREP7(idx) \</span></div>
|
|
<div class="line"><a id="l01047" name="l01047"></a><span class="lineno"> 1047</span><span class="preprocessor"> T t1 = CC(idx,0,k), t2, t3, t4, t5, t6, t7; \</span></div>
|
|
<div class="line"><a id="l01048" name="l01048"></a><span class="lineno"> 1048</span><span class="preprocessor"> PM (t2,t7,CC(idx,1,k),CC(idx,6,k)); \</span></div>
|
|
<div class="line"><a id="l01049" name="l01049"></a><span class="lineno"> 1049</span><span class="preprocessor"> PM (t3,t6,CC(idx,2,k),CC(idx,5,k)); \</span></div>
|
|
<div class="line"><a id="l01050" name="l01050"></a><span class="lineno"> 1050</span><span class="preprocessor"> PM (t4,t5,CC(idx,3,k),CC(idx,4,k)); \</span></div>
|
|
<div class="line"><a id="l01051" name="l01051"></a><span class="lineno"> 1051</span><span class="preprocessor"> CH(idx,k,0).r=t1.r+t2.r+t3.r+t4.r; \</span></div>
|
|
<div class="line"><a id="l01052" name="l01052"></a><span class="lineno"> 1052</span><span class="preprocessor"> CH(idx,k,0).i=t1.i+t2.i+t3.i+t4.i;</span></div>
|
|
</div>
|
|
<div class="line"><a id="l01053" name="l01053"></a><span class="lineno"> 1053</span> </div>
|
|
<div class="foldopen" id="foldopen01054" data-start="" data-end="">
|
|
<div class="line"><a id="l01054" name="l01054"></a><span class="lineno"><a class="line" href="pocketfft_8h.html#a9c2fc2de74a031c38e9d8a21249ae1cd"> 1054</a></span><span class="preprocessor">#define POCKETFFT_PARTSTEP7a0(u1,u2,x1,x2,x3,y1,y2,y3,out1,out2) \</span></div>
|
|
<div class="line"><a id="l01055" name="l01055"></a><span class="lineno"> 1055</span><span class="preprocessor"> { \</span></div>
|
|
<div class="line"><a id="l01056" name="l01056"></a><span class="lineno"> 1056</span><span class="preprocessor"> T ca,cb; \</span></div>
|
|
<div class="line"><a id="l01057" name="l01057"></a><span class="lineno"> 1057</span><span class="preprocessor"> ca.r=t1.r+x1*t2.r+x2*t3.r+x3*t4.r; \</span></div>
|
|
<div class="line"><a id="l01058" name="l01058"></a><span class="lineno"> 1058</span><span class="preprocessor"> ca.i=t1.i+x1*t2.i+x2*t3.i+x3*t4.i; \</span></div>
|
|
<div class="line"><a id="l01059" name="l01059"></a><span class="lineno"> 1059</span><span class="preprocessor"> cb.i=y1*t7.r y2*t6.r y3*t5.r; \</span></div>
|
|
<div class="line"><a id="l01060" name="l01060"></a><span class="lineno"> 1060</span><span class="preprocessor"> cb.r=-(y1*t7.i y2*t6.i y3*t5.i); \</span></div>
|
|
<div class="line"><a id="l01061" name="l01061"></a><span class="lineno"> 1061</span><span class="preprocessor"> PM(out1,out2,ca,cb); \</span></div>
|
|
<div class="line"><a id="l01062" name="l01062"></a><span class="lineno"> 1062</span><span class="preprocessor"> }</span></div>
|
|
</div>
|
|
<div class="foldopen" id="foldopen01063" data-start="" data-end="">
|
|
<div class="line"><a id="l01063" name="l01063"></a><span class="lineno"><a class="line" href="pocketfft_8h.html#a2b27f6e1f0ee131765186870517255c6"> 1063</a></span><span class="preprocessor">#define POCKETFFT_PARTSTEP7a(u1,u2,x1,x2,x3,y1,y2,y3) \</span></div>
|
|
<div class="line"><a id="l01064" name="l01064"></a><span class="lineno"> 1064</span><span class="preprocessor"> POCKETFFT_PARTSTEP7a0(u1,u2,x1,x2,x3,y1,y2,y3,CH(0,k,u1),CH(0,k,u2))</span></div>
|
|
</div>
|
|
<div class="foldopen" id="foldopen01065" data-start="" data-end="">
|
|
<div class="line"><a id="l01065" name="l01065"></a><span class="lineno"><a class="line" href="pocketfft_8h.html#af7de1f82911a973d8446cf3f40ff3044"> 1065</a></span><span class="preprocessor">#define POCKETFFT_PARTSTEP7(u1,u2,x1,x2,x3,y1,y2,y3) \</span></div>
|
|
<div class="line"><a id="l01066" name="l01066"></a><span class="lineno"> 1066</span><span class="preprocessor"> { \</span></div>
|
|
<div class="line"><a id="l01067" name="l01067"></a><span class="lineno"> 1067</span><span class="preprocessor"> T da,db; \</span></div>
|
|
<div class="line"><a id="l01068" name="l01068"></a><span class="lineno"> 1068</span><span class="preprocessor"> POCKETFFT_PARTSTEP7a0(u1,u2,x1,x2,x3,y1,y2,y3,da,db) \</span></div>
|
|
<div class="line"><a id="l01069" name="l01069"></a><span class="lineno"> 1069</span><span class="preprocessor"> special_mul<fwd>(da,WA(u1-1,i),CH(i,k,u1)); \</span></div>
|
|
<div class="line"><a id="l01070" name="l01070"></a><span class="lineno"> 1070</span><span class="preprocessor"> special_mul<fwd>(db,WA(u2-1,i),CH(i,k,u2)); \</span></div>
|
|
<div class="line"><a id="l01071" name="l01071"></a><span class="lineno"> 1071</span><span class="preprocessor"> }</span></div>
|
|
</div>
|
|
<div class="line"><a id="l01072" name="l01072"></a><span class="lineno"> 1072</span> </div>
|
|
<div class="line"><a id="l01073" name="l01073"></a><span class="lineno"> 1073</span><span class="keyword">template</span><<span class="keywordtype">bool</span> fwd, <span class="keyword">typename</span> T> <span class="keywordtype">void</span> pass7(<span class="keywordtype">size_t</span> ido, <span class="keywordtype">size_t</span> l1,</div>
|
|
<div class="line"><a id="l01074" name="l01074"></a><span class="lineno"> 1074</span> <span class="keyword">const</span> T * <a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> cc, T * <a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> ch,</div>
|
|
<div class="line"><a id="l01075" name="l01075"></a><span class="lineno"> 1075</span> <span class="keyword">const</span> cmplx<T0> * <a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> wa)<span class="keyword"> const</span></div>
|
|
<div class="line"><a id="l01076" name="l01076"></a><span class="lineno"> 1076</span><span class="keyword"> </span>{</div>
|
|
<div class="line"><a id="l01077" name="l01077"></a><span class="lineno"> 1077</span> <span class="keyword">constexpr</span> T0 tw1r= T0(0.6234898018587335305250048840042398L),</div>
|
|
<div class="line"><a id="l01078" name="l01078"></a><span class="lineno"> 1078</span> tw1i= (fwd ? -1 : 1) * T0(0.7818314824680298087084445266740578L),</div>
|
|
<div class="line"><a id="l01079" name="l01079"></a><span class="lineno"> 1079</span> tw2r= T0(-0.2225209339563144042889025644967948L),</div>
|
|
<div class="line"><a id="l01080" name="l01080"></a><span class="lineno"> 1080</span> tw2i= (fwd ? -1 : 1) * T0(0.9749279121818236070181316829939312L),</div>
|
|
<div class="line"><a id="l01081" name="l01081"></a><span class="lineno"> 1081</span> tw3r= T0(-0.9009688679024191262361023195074451L),</div>
|
|
<div class="line"><a id="l01082" name="l01082"></a><span class="lineno"> 1082</span> tw3i= (fwd ? -1 : 1) * T0(0.433883739117558120475768332848359L);</div>
|
|
<div class="line"><a id="l01083" name="l01083"></a><span class="lineno"> 1083</span> </div>
|
|
<div class="line"><a id="l01084" name="l01084"></a><span class="lineno"> 1084</span> <span class="keyword">auto</span> CH = [ch,ido,l1](<span class="keywordtype">size_t</span> a, <span class="keywordtype">size_t</span> b, <span class="keywordtype">size_t</span> c) -> T&</div>
|
|
<div class="line"><a id="l01085" name="l01085"></a><span class="lineno"> 1085</span> { <span class="keywordflow">return</span> ch[a+ido*(b+l1*c)]; };</div>
|
|
<div class="line"><a id="l01086" name="l01086"></a><span class="lineno"> 1086</span> <span class="keyword">auto</span> CC = [cc,ido](<span class="keywordtype">size_t</span> a, <span class="keywordtype">size_t</span> b, <span class="keywordtype">size_t</span> c) -> <span class="keyword">const</span> T&</div>
|
|
<div class="line"><a id="l01087" name="l01087"></a><span class="lineno"> 1087</span> { <span class="keywordflow">return</span> cc[a+ido*(b+7*c)]; };</div>
|
|
<div class="line"><a id="l01088" name="l01088"></a><span class="lineno"> 1088</span> <span class="keyword">auto</span> WA = [wa, ido](<span class="keywordtype">size_t</span> x, <span class="keywordtype">size_t</span> i)</div>
|
|
<div class="line"><a id="l01089" name="l01089"></a><span class="lineno"> 1089</span> { <span class="keywordflow">return</span> wa[i-1+x*(ido-1)]; };</div>
|
|
<div class="line"><a id="l01090" name="l01090"></a><span class="lineno"> 1090</span> </div>
|
|
<div class="line"><a id="l01091" name="l01091"></a><span class="lineno"> 1091</span> <span class="keywordflow">if</span> (ido==1)</div>
|
|
<div class="line"><a id="l01092" name="l01092"></a><span class="lineno"> 1092</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> k=0; k<l1; ++k)</div>
|
|
<div class="line"><a id="l01093" name="l01093"></a><span class="lineno"> 1093</span> {</div>
|
|
<div class="line"><a id="l01094" name="l01094"></a><span class="lineno"> 1094</span> <a class="code hl_define" href="pocketfft_8h.html#ae7c4d0cda5b3824f84eac54addabd6ec">POCKETFFT_PREP7</a>(0)</div>
|
|
<div class="line"><a id="l01095" name="l01095"></a><span class="lineno"> 1095</span> <a class="code hl_define" href="pocketfft_8h.html#a2b27f6e1f0ee131765186870517255c6">POCKETFFT_PARTSTEP7a</a>(1,6,tw1r,tw2r,tw3r,+tw1i,+tw2i,+tw3i)</div>
|
|
<div class="line"><a id="l01096" name="l01096"></a><span class="lineno"> 1096</span> <a class="code hl_define" href="pocketfft_8h.html#a2b27f6e1f0ee131765186870517255c6">POCKETFFT_PARTSTEP7a</a>(2,5,tw2r,tw3r,tw1r,+tw2i,-tw3i,-tw1i)</div>
|
|
<div class="line"><a id="l01097" name="l01097"></a><span class="lineno"> 1097</span> <a class="code hl_define" href="pocketfft_8h.html#a2b27f6e1f0ee131765186870517255c6">POCKETFFT_PARTSTEP7a</a>(3,4,tw3r,tw1r,tw2r,+tw3i,-tw1i,+tw2i)</div>
|
|
<div class="line"><a id="l01098" name="l01098"></a><span class="lineno"> 1098</span> }</div>
|
|
<div class="line"><a id="l01099" name="l01099"></a><span class="lineno"> 1099</span> <span class="keywordflow">else</span></div>
|
|
<div class="line"><a id="l01100" name="l01100"></a><span class="lineno"> 1100</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> k=0; k<l1; ++k)</div>
|
|
<div class="line"><a id="l01101" name="l01101"></a><span class="lineno"> 1101</span> {</div>
|
|
<div class="line"><a id="l01102" name="l01102"></a><span class="lineno"> 1102</span> {</div>
|
|
<div class="line"><a id="l01103" name="l01103"></a><span class="lineno"> 1103</span> <a class="code hl_define" href="pocketfft_8h.html#ae7c4d0cda5b3824f84eac54addabd6ec">POCKETFFT_PREP7</a>(0)</div>
|
|
<div class="line"><a id="l01104" name="l01104"></a><span class="lineno"> 1104</span> <a class="code hl_define" href="pocketfft_8h.html#a2b27f6e1f0ee131765186870517255c6">POCKETFFT_PARTSTEP7a</a>(1,6,tw1r,tw2r,tw3r,+tw1i,+tw2i,+tw3i)</div>
|
|
<div class="line"><a id="l01105" name="l01105"></a><span class="lineno"> 1105</span> <a class="code hl_define" href="pocketfft_8h.html#a2b27f6e1f0ee131765186870517255c6">POCKETFFT_PARTSTEP7a</a>(2,5,tw2r,tw3r,tw1r,+tw2i,-tw3i,-tw1i)</div>
|
|
<div class="line"><a id="l01106" name="l01106"></a><span class="lineno"> 1106</span> <a class="code hl_define" href="pocketfft_8h.html#a2b27f6e1f0ee131765186870517255c6">POCKETFFT_PARTSTEP7a</a>(3,4,tw3r,tw1r,tw2r,+tw3i,-tw1i,+tw2i)</div>
|
|
<div class="line"><a id="l01107" name="l01107"></a><span class="lineno"> 1107</span> }</div>
|
|
<div class="line"><a id="l01108" name="l01108"></a><span class="lineno"> 1108</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i=1; i<ido; ++i)</div>
|
|
<div class="line"><a id="l01109" name="l01109"></a><span class="lineno"> 1109</span> {</div>
|
|
<div class="line"><a id="l01110" name="l01110"></a><span class="lineno"> 1110</span> <a class="code hl_define" href="pocketfft_8h.html#ae7c4d0cda5b3824f84eac54addabd6ec">POCKETFFT_PREP7</a>(i)</div>
|
|
<div class="line"><a id="l01111" name="l01111"></a><span class="lineno"> 1111</span> <a class="code hl_define" href="pocketfft_8h.html#af7de1f82911a973d8446cf3f40ff3044">POCKETFFT_PARTSTEP7</a>(1,6,tw1r,tw2r,tw3r,+tw1i,+tw2i,+tw3i)</div>
|
|
<div class="line"><a id="l01112" name="l01112"></a><span class="lineno"> 1112</span> <a class="code hl_define" href="pocketfft_8h.html#af7de1f82911a973d8446cf3f40ff3044">POCKETFFT_PARTSTEP7</a>(2,5,tw2r,tw3r,tw1r,+tw2i,-tw3i,-tw1i)</div>
|
|
<div class="line"><a id="l01113" name="l01113"></a><span class="lineno"> 1113</span> <a class="code hl_define" href="pocketfft_8h.html#af7de1f82911a973d8446cf3f40ff3044">POCKETFFT_PARTSTEP7</a>(3,4,tw3r,tw1r,tw2r,+tw3i,-tw1i,+tw2i)</div>
|
|
<div class="line"><a id="l01114" name="l01114"></a><span class="lineno"> 1114</span> }</div>
|
|
<div class="line"><a id="l01115" name="l01115"></a><span class="lineno"> 1115</span> }</div>
|
|
<div class="line"><a id="l01116" name="l01116"></a><span class="lineno"> 1116</span> }</div>
|
|
<div class="line"><a id="l01117" name="l01117"></a><span class="lineno"> 1117</span> </div>
|
|
<div class="line"><a id="l01118" name="l01118"></a><span class="lineno"> 1118</span><span class="preprocessor">#undef POCKETFFT_PARTSTEP7</span></div>
|
|
<div class="line"><a id="l01119" name="l01119"></a><span class="lineno"> 1119</span><span class="preprocessor">#undef POCKETFFT_PARTSTEP7a0</span></div>
|
|
<div class="line"><a id="l01120" name="l01120"></a><span class="lineno"> 1120</span><span class="preprocessor">#undef POCKETFFT_PARTSTEP7a</span></div>
|
|
<div class="line"><a id="l01121" name="l01121"></a><span class="lineno"> 1121</span><span class="preprocessor">#undef POCKETFFT_PREP7</span></div>
|
|
<div class="line"><a id="l01122" name="l01122"></a><span class="lineno"> 1122</span> </div>
|
|
<div class="line"><a id="l01123" name="l01123"></a><span class="lineno"> 1123</span><span class="keyword">template</span> <<span class="keywordtype">bool</span> fwd, <span class="keyword">typename</span> T> <span class="keywordtype">void</span> ROTX45(T &a)<span class="keyword"> const</span></div>
|
|
<div class="line"><a id="l01124" name="l01124"></a><span class="lineno"> 1124</span><span class="keyword"> </span>{</div>
|
|
<div class="line"><a id="l01125" name="l01125"></a><span class="lineno"> 1125</span> <span class="keyword">constexpr</span> T0 hsqt2=T0(0.707106781186547524400844362104849L);</div>
|
|
<div class="line"><a id="l01126" name="l01126"></a><span class="lineno"> 1126</span> <span class="keywordflow">if</span> (fwd)</div>
|
|
<div class="line"><a id="l01127" name="l01127"></a><span class="lineno"> 1127</span> { <span class="keyword">auto</span> tmp_=a.r; a.r=hsqt2*(a.r+a.i); a.i=hsqt2*(a.i-tmp_); }</div>
|
|
<div class="line"><a id="l01128" name="l01128"></a><span class="lineno"> 1128</span> <span class="keywordflow">else</span></div>
|
|
<div class="line"><a id="l01129" name="l01129"></a><span class="lineno"> 1129</span> { <span class="keyword">auto</span> tmp_=a.r; a.r=hsqt2*(a.r-a.i); a.i=hsqt2*(a.i+tmp_); }</div>
|
|
<div class="line"><a id="l01130" name="l01130"></a><span class="lineno"> 1130</span> }</div>
|
|
<div class="line"><a id="l01131" name="l01131"></a><span class="lineno"> 1131</span><span class="keyword">template</span> <<span class="keywordtype">bool</span> fwd, <span class="keyword">typename</span> T> <span class="keywordtype">void</span> ROTX135(T &a)<span class="keyword"> const</span></div>
|
|
<div class="line"><a id="l01132" name="l01132"></a><span class="lineno"> 1132</span><span class="keyword"> </span>{</div>
|
|
<div class="line"><a id="l01133" name="l01133"></a><span class="lineno"> 1133</span> <span class="keyword">constexpr</span> T0 hsqt2=T0(0.707106781186547524400844362104849L);</div>
|
|
<div class="line"><a id="l01134" name="l01134"></a><span class="lineno"> 1134</span> <span class="keywordflow">if</span> (fwd)</div>
|
|
<div class="line"><a id="l01135" name="l01135"></a><span class="lineno"> 1135</span> { <span class="keyword">auto</span> tmp_=a.r; a.r=hsqt2*(a.i-a.r); a.i=hsqt2*(-tmp_-a.i); }</div>
|
|
<div class="line"><a id="l01136" name="l01136"></a><span class="lineno"> 1136</span> <span class="keywordflow">else</span></div>
|
|
<div class="line"><a id="l01137" name="l01137"></a><span class="lineno"> 1137</span> { <span class="keyword">auto</span> tmp_=a.r; a.r=hsqt2*(-a.r-a.i); a.i=hsqt2*(tmp_-a.i); }</div>
|
|
<div class="line"><a id="l01138" name="l01138"></a><span class="lineno"> 1138</span> }</div>
|
|
<div class="line"><a id="l01139" name="l01139"></a><span class="lineno"> 1139</span> </div>
|
|
<div class="line"><a id="l01140" name="l01140"></a><span class="lineno"> 1140</span><span class="keyword">template</span><<span class="keywordtype">bool</span> fwd, <span class="keyword">typename</span> T> <span class="keywordtype">void</span> pass8 (<span class="keywordtype">size_t</span> ido, <span class="keywordtype">size_t</span> l1,</div>
|
|
<div class="line"><a id="l01141" name="l01141"></a><span class="lineno"> 1141</span> <span class="keyword">const</span> T * <a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> cc, T * <a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> ch,</div>
|
|
<div class="line"><a id="l01142" name="l01142"></a><span class="lineno"> 1142</span> <span class="keyword">const</span> cmplx<T0> * <a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> wa)<span class="keyword"> const</span></div>
|
|
<div class="line"><a id="l01143" name="l01143"></a><span class="lineno"> 1143</span><span class="keyword"> </span>{</div>
|
|
<div class="line"><a id="l01144" name="l01144"></a><span class="lineno"> 1144</span> <span class="keyword">auto</span> CH = [ch,ido,l1](<span class="keywordtype">size_t</span> a, <span class="keywordtype">size_t</span> b, <span class="keywordtype">size_t</span> c) -> T&</div>
|
|
<div class="line"><a id="l01145" name="l01145"></a><span class="lineno"> 1145</span> { <span class="keywordflow">return</span> ch[a+ido*(b+l1*c)]; };</div>
|
|
<div class="line"><a id="l01146" name="l01146"></a><span class="lineno"> 1146</span> <span class="keyword">auto</span> CC = [cc,ido](<span class="keywordtype">size_t</span> a, <span class="keywordtype">size_t</span> b, <span class="keywordtype">size_t</span> c) -> <span class="keyword">const</span> T&</div>
|
|
<div class="line"><a id="l01147" name="l01147"></a><span class="lineno"> 1147</span> { <span class="keywordflow">return</span> cc[a+ido*(b+8*c)]; };</div>
|
|
<div class="line"><a id="l01148" name="l01148"></a><span class="lineno"> 1148</span> <span class="keyword">auto</span> WA = [wa, ido](<span class="keywordtype">size_t</span> x, <span class="keywordtype">size_t</span> i)</div>
|
|
<div class="line"><a id="l01149" name="l01149"></a><span class="lineno"> 1149</span> { <span class="keywordflow">return</span> wa[i-1+x*(ido-1)]; };</div>
|
|
<div class="line"><a id="l01150" name="l01150"></a><span class="lineno"> 1150</span> </div>
|
|
<div class="line"><a id="l01151" name="l01151"></a><span class="lineno"> 1151</span> <span class="keywordflow">if</span> (ido==1)</div>
|
|
<div class="line"><a id="l01152" name="l01152"></a><span class="lineno"> 1152</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> k=0; k<l1; ++k)</div>
|
|
<div class="line"><a id="l01153" name="l01153"></a><span class="lineno"> 1153</span> {</div>
|
|
<div class="line"><a id="l01154" name="l01154"></a><span class="lineno"> 1154</span> T a0, a1, a2, a3, a4, a5, a6, a7;</div>
|
|
<div class="line"><a id="l01155" name="l01155"></a><span class="lineno"> 1155</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a>(a1,a5,CC(0,1,k),CC(0,5,k));</div>
|
|
<div class="line"><a id="l01156" name="l01156"></a><span class="lineno"> 1156</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a>(a3,a7,CC(0,3,k),CC(0,7,k));</div>
|
|
<div class="line"><a id="l01157" name="l01157"></a><span class="lineno"> 1157</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#aa3477781ef54788c9fa7755cb8ea75fd">PMINPLACE</a>(a1,a3);</div>
|
|
<div class="line"><a id="l01158" name="l01158"></a><span class="lineno"> 1158</span> ROTX90<fwd>(a3);</div>
|
|
<div class="line"><a id="l01159" name="l01159"></a><span class="lineno"> 1159</span> </div>
|
|
<div class="line"><a id="l01160" name="l01160"></a><span class="lineno"> 1160</span> ROTX90<fwd>(a7);</div>
|
|
<div class="line"><a id="l01161" name="l01161"></a><span class="lineno"> 1161</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#aa3477781ef54788c9fa7755cb8ea75fd">PMINPLACE</a>(a5,a7);</div>
|
|
<div class="line"><a id="l01162" name="l01162"></a><span class="lineno"> 1162</span> ROTX45<fwd>(a5);</div>
|
|
<div class="line"><a id="l01163" name="l01163"></a><span class="lineno"> 1163</span> ROTX135<fwd>(a7);</div>
|
|
<div class="line"><a id="l01164" name="l01164"></a><span class="lineno"> 1164</span> </div>
|
|
<div class="line"><a id="l01165" name="l01165"></a><span class="lineno"> 1165</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a>(a0,a4,CC(0,0,k),CC(0,4,k));</div>
|
|
<div class="line"><a id="l01166" name="l01166"></a><span class="lineno"> 1166</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a>(a2,a6,CC(0,2,k),CC(0,6,k));</div>
|
|
<div class="line"><a id="l01167" name="l01167"></a><span class="lineno"> 1167</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a>(CH(0,k,0),CH(0,k,4),a0+a2,a1);</div>
|
|
<div class="line"><a id="l01168" name="l01168"></a><span class="lineno"> 1168</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a>(CH(0,k,2),CH(0,k,6),a0-a2,a3);</div>
|
|
<div class="line"><a id="l01169" name="l01169"></a><span class="lineno"> 1169</span> ROTX90<fwd>(a6);</div>
|
|
<div class="line"><a id="l01170" name="l01170"></a><span class="lineno"> 1170</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a>(CH(0,k,1),CH(0,k,5),a4+a6,a5);</div>
|
|
<div class="line"><a id="l01171" name="l01171"></a><span class="lineno"> 1171</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a>(CH(0,k,3),CH(0,k,7),a4-a6,a7);</div>
|
|
<div class="line"><a id="l01172" name="l01172"></a><span class="lineno"> 1172</span> }</div>
|
|
<div class="line"><a id="l01173" name="l01173"></a><span class="lineno"> 1173</span> <span class="keywordflow">else</span></div>
|
|
<div class="line"><a id="l01174" name="l01174"></a><span class="lineno"> 1174</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> k=0; k<l1; ++k)</div>
|
|
<div class="line"><a id="l01175" name="l01175"></a><span class="lineno"> 1175</span> {</div>
|
|
<div class="line"><a id="l01176" name="l01176"></a><span class="lineno"> 1176</span> {</div>
|
|
<div class="line"><a id="l01177" name="l01177"></a><span class="lineno"> 1177</span> T a0, a1, a2, a3, a4, a5, a6, a7;</div>
|
|
<div class="line"><a id="l01178" name="l01178"></a><span class="lineno"> 1178</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a>(a1,a5,CC(0,1,k),CC(0,5,k));</div>
|
|
<div class="line"><a id="l01179" name="l01179"></a><span class="lineno"> 1179</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a>(a3,a7,CC(0,3,k),CC(0,7,k));</div>
|
|
<div class="line"><a id="l01180" name="l01180"></a><span class="lineno"> 1180</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#aa3477781ef54788c9fa7755cb8ea75fd">PMINPLACE</a>(a1,a3);</div>
|
|
<div class="line"><a id="l01181" name="l01181"></a><span class="lineno"> 1181</span> ROTX90<fwd>(a3);</div>
|
|
<div class="line"><a id="l01182" name="l01182"></a><span class="lineno"> 1182</span> </div>
|
|
<div class="line"><a id="l01183" name="l01183"></a><span class="lineno"> 1183</span> ROTX90<fwd>(a7);</div>
|
|
<div class="line"><a id="l01184" name="l01184"></a><span class="lineno"> 1184</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#aa3477781ef54788c9fa7755cb8ea75fd">PMINPLACE</a>(a5,a7);</div>
|
|
<div class="line"><a id="l01185" name="l01185"></a><span class="lineno"> 1185</span> ROTX45<fwd>(a5);</div>
|
|
<div class="line"><a id="l01186" name="l01186"></a><span class="lineno"> 1186</span> ROTX135<fwd>(a7);</div>
|
|
<div class="line"><a id="l01187" name="l01187"></a><span class="lineno"> 1187</span> </div>
|
|
<div class="line"><a id="l01188" name="l01188"></a><span class="lineno"> 1188</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a>(a0,a4,CC(0,0,k),CC(0,4,k));</div>
|
|
<div class="line"><a id="l01189" name="l01189"></a><span class="lineno"> 1189</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a>(a2,a6,CC(0,2,k),CC(0,6,k));</div>
|
|
<div class="line"><a id="l01190" name="l01190"></a><span class="lineno"> 1190</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a>(CH(0,k,0),CH(0,k,4),a0+a2,a1);</div>
|
|
<div class="line"><a id="l01191" name="l01191"></a><span class="lineno"> 1191</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a>(CH(0,k,2),CH(0,k,6),a0-a2,a3);</div>
|
|
<div class="line"><a id="l01192" name="l01192"></a><span class="lineno"> 1192</span> ROTX90<fwd>(a6);</div>
|
|
<div class="line"><a id="l01193" name="l01193"></a><span class="lineno"> 1193</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a>(CH(0,k,1),CH(0,k,5),a4+a6,a5);</div>
|
|
<div class="line"><a id="l01194" name="l01194"></a><span class="lineno"> 1194</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a>(CH(0,k,3),CH(0,k,7),a4-a6,a7);</div>
|
|
<div class="line"><a id="l01195" name="l01195"></a><span class="lineno"> 1195</span> }</div>
|
|
<div class="line"><a id="l01196" name="l01196"></a><span class="lineno"> 1196</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i=1; i<ido; ++i)</div>
|
|
<div class="line"><a id="l01197" name="l01197"></a><span class="lineno"> 1197</span> {</div>
|
|
<div class="line"><a id="l01198" name="l01198"></a><span class="lineno"> 1198</span> T a0, a1, a2, a3, a4, a5, a6, a7;</div>
|
|
<div class="line"><a id="l01199" name="l01199"></a><span class="lineno"> 1199</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a>(a1,a5,CC(i,1,k),CC(i,5,k));</div>
|
|
<div class="line"><a id="l01200" name="l01200"></a><span class="lineno"> 1200</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a>(a3,a7,CC(i,3,k),CC(i,7,k));</div>
|
|
<div class="line"><a id="l01201" name="l01201"></a><span class="lineno"> 1201</span> ROTX90<fwd>(a7);</div>
|
|
<div class="line"><a id="l01202" name="l01202"></a><span class="lineno"> 1202</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#aa3477781ef54788c9fa7755cb8ea75fd">PMINPLACE</a>(a1,a3);</div>
|
|
<div class="line"><a id="l01203" name="l01203"></a><span class="lineno"> 1203</span> ROTX90<fwd>(a3);</div>
|
|
<div class="line"><a id="l01204" name="l01204"></a><span class="lineno"> 1204</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#aa3477781ef54788c9fa7755cb8ea75fd">PMINPLACE</a>(a5,a7);</div>
|
|
<div class="line"><a id="l01205" name="l01205"></a><span class="lineno"> 1205</span> ROTX45<fwd>(a5);</div>
|
|
<div class="line"><a id="l01206" name="l01206"></a><span class="lineno"> 1206</span> ROTX135<fwd>(a7);</div>
|
|
<div class="line"><a id="l01207" name="l01207"></a><span class="lineno"> 1207</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a>(a0,a4,CC(i,0,k),CC(i,4,k));</div>
|
|
<div class="line"><a id="l01208" name="l01208"></a><span class="lineno"> 1208</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a>(a2,a6,CC(i,2,k),CC(i,6,k));</div>
|
|
<div class="line"><a id="l01209" name="l01209"></a><span class="lineno"> 1209</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#aa3477781ef54788c9fa7755cb8ea75fd">PMINPLACE</a>(a0,a2);</div>
|
|
<div class="line"><a id="l01210" name="l01210"></a><span class="lineno"> 1210</span> CH(i,k,0) = a0+a1;</div>
|
|
<div class="line"><a id="l01211" name="l01211"></a><span class="lineno"> 1211</span> special_mul<fwd>(a0-a1,WA(3,i),CH(i,k,4));</div>
|
|
<div class="line"><a id="l01212" name="l01212"></a><span class="lineno"> 1212</span> special_mul<fwd>(a2+a3,WA(1,i),CH(i,k,2));</div>
|
|
<div class="line"><a id="l01213" name="l01213"></a><span class="lineno"> 1213</span> special_mul<fwd>(a2-a3,WA(5,i),CH(i,k,6));</div>
|
|
<div class="line"><a id="l01214" name="l01214"></a><span class="lineno"> 1214</span> ROTX90<fwd>(a6);</div>
|
|
<div class="line"><a id="l01215" name="l01215"></a><span class="lineno"> 1215</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#aa3477781ef54788c9fa7755cb8ea75fd">PMINPLACE</a>(a4,a6);</div>
|
|
<div class="line"><a id="l01216" name="l01216"></a><span class="lineno"> 1216</span> special_mul<fwd>(a4+a5,WA(0,i),CH(i,k,1));</div>
|
|
<div class="line"><a id="l01217" name="l01217"></a><span class="lineno"> 1217</span> special_mul<fwd>(a4-a5,WA(4,i),CH(i,k,5));</div>
|
|
<div class="line"><a id="l01218" name="l01218"></a><span class="lineno"> 1218</span> special_mul<fwd>(a6+a7,WA(2,i),CH(i,k,3));</div>
|
|
<div class="line"><a id="l01219" name="l01219"></a><span class="lineno"> 1219</span> special_mul<fwd>(a6-a7,WA(6,i),CH(i,k,7));</div>
|
|
<div class="line"><a id="l01220" name="l01220"></a><span class="lineno"> 1220</span> }</div>
|
|
<div class="line"><a id="l01221" name="l01221"></a><span class="lineno"> 1221</span> }</div>
|
|
<div class="line"><a id="l01222" name="l01222"></a><span class="lineno"> 1222</span> }</div>
|
|
<div class="line"><a id="l01223" name="l01223"></a><span class="lineno"> 1223</span> </div>
|
|
<div class="line"><a id="l01224" name="l01224"></a><span class="lineno"> 1224</span> </div>
|
|
<div class="foldopen" id="foldopen01225" data-start="" data-end="">
|
|
<div class="line"><a id="l01225" name="l01225"></a><span class="lineno"><a class="line" href="pocketfft_8h.html#a536d2ea61479d4b074bf52ce09fdbc3a"> 1225</a></span><span class="preprocessor">#define POCKETFFT_PREP11(idx) \</span></div>
|
|
<div class="line"><a id="l01226" name="l01226"></a><span class="lineno"> 1226</span><span class="preprocessor"> T t1 = CC(idx,0,k), t2, t3, t4, t5, t6, t7, t8, t9, t10, t11; \</span></div>
|
|
<div class="line"><a id="l01227" name="l01227"></a><span class="lineno"> 1227</span><span class="preprocessor"> PM (t2,t11,CC(idx,1,k),CC(idx,10,k)); \</span></div>
|
|
<div class="line"><a id="l01228" name="l01228"></a><span class="lineno"> 1228</span><span class="preprocessor"> PM (t3,t10,CC(idx,2,k),CC(idx, 9,k)); \</span></div>
|
|
<div class="line"><a id="l01229" name="l01229"></a><span class="lineno"> 1229</span><span class="preprocessor"> PM (t4,t9 ,CC(idx,3,k),CC(idx, 8,k)); \</span></div>
|
|
<div class="line"><a id="l01230" name="l01230"></a><span class="lineno"> 1230</span><span class="preprocessor"> PM (t5,t8 ,CC(idx,4,k),CC(idx, 7,k)); \</span></div>
|
|
<div class="line"><a id="l01231" name="l01231"></a><span class="lineno"> 1231</span><span class="preprocessor"> PM (t6,t7 ,CC(idx,5,k),CC(idx, 6,k)); \</span></div>
|
|
<div class="line"><a id="l01232" name="l01232"></a><span class="lineno"> 1232</span><span class="preprocessor"> CH(idx,k,0).r=t1.r+t2.r+t3.r+t4.r+t5.r+t6.r; \</span></div>
|
|
<div class="line"><a id="l01233" name="l01233"></a><span class="lineno"> 1233</span><span class="preprocessor"> CH(idx,k,0).i=t1.i+t2.i+t3.i+t4.i+t5.i+t6.i;</span></div>
|
|
</div>
|
|
<div class="line"><a id="l01234" name="l01234"></a><span class="lineno"> 1234</span> </div>
|
|
<div class="foldopen" id="foldopen01235" data-start="" data-end="">
|
|
<div class="line"><a id="l01235" name="l01235"></a><span class="lineno"><a class="line" href="pocketfft_8h.html#ab2df44457945ab625fb38a777a46af1b"> 1235</a></span><span class="preprocessor">#define POCKETFFT_PARTSTEP11a0(u1,u2,x1,x2,x3,x4,x5,y1,y2,y3,y4,y5,out1,out2) \</span></div>
|
|
<div class="line"><a id="l01236" name="l01236"></a><span class="lineno"> 1236</span><span class="preprocessor"> { \</span></div>
|
|
<div class="line"><a id="l01237" name="l01237"></a><span class="lineno"> 1237</span><span class="preprocessor"> T ca = t1 + t2*x1 + t3*x2 + t4*x3 + t5*x4 +t6*x5, \</span></div>
|
|
<div class="line"><a id="l01238" name="l01238"></a><span class="lineno"> 1238</span><span class="preprocessor"> cb; \</span></div>
|
|
<div class="line"><a id="l01239" name="l01239"></a><span class="lineno"> 1239</span><span class="preprocessor"> cb.i=y1*t11.r y2*t10.r y3*t9.r y4*t8.r y5*t7.r; \</span></div>
|
|
<div class="line"><a id="l01240" name="l01240"></a><span class="lineno"> 1240</span><span class="preprocessor"> cb.r=-(y1*t11.i y2*t10.i y3*t9.i y4*t8.i y5*t7.i ); \</span></div>
|
|
<div class="line"><a id="l01241" name="l01241"></a><span class="lineno"> 1241</span><span class="preprocessor"> PM(out1,out2,ca,cb); \</span></div>
|
|
<div class="line"><a id="l01242" name="l01242"></a><span class="lineno"> 1242</span><span class="preprocessor"> }</span></div>
|
|
</div>
|
|
<div class="foldopen" id="foldopen01243" data-start="" data-end="">
|
|
<div class="line"><a id="l01243" name="l01243"></a><span class="lineno"><a class="line" href="pocketfft_8h.html#ac35e1aa5ae84d655256b7a0afd9051c2"> 1243</a></span><span class="preprocessor">#define POCKETFFT_PARTSTEP11a(u1,u2,x1,x2,x3,x4,x5,y1,y2,y3,y4,y5) \</span></div>
|
|
<div class="line"><a id="l01244" name="l01244"></a><span class="lineno"> 1244</span><span class="preprocessor"> POCKETFFT_PARTSTEP11a0(u1,u2,x1,x2,x3,x4,x5,y1,y2,y3,y4,y5,CH(0,k,u1),CH(0,k,u2))</span></div>
|
|
</div>
|
|
<div class="foldopen" id="foldopen01245" data-start="" data-end="">
|
|
<div class="line"><a id="l01245" name="l01245"></a><span class="lineno"><a class="line" href="pocketfft_8h.html#a1793d0d00f2e13101eb5ad0719c40817"> 1245</a></span><span class="preprocessor">#define POCKETFFT_PARTSTEP11(u1,u2,x1,x2,x3,x4,x5,y1,y2,y3,y4,y5) \</span></div>
|
|
<div class="line"><a id="l01246" name="l01246"></a><span class="lineno"> 1246</span><span class="preprocessor"> { \</span></div>
|
|
<div class="line"><a id="l01247" name="l01247"></a><span class="lineno"> 1247</span><span class="preprocessor"> T da,db; \</span></div>
|
|
<div class="line"><a id="l01248" name="l01248"></a><span class="lineno"> 1248</span><span class="preprocessor"> POCKETFFT_PARTSTEP11a0(u1,u2,x1,x2,x3,x4,x5,y1,y2,y3,y4,y5,da,db) \</span></div>
|
|
<div class="line"><a id="l01249" name="l01249"></a><span class="lineno"> 1249</span><span class="preprocessor"> special_mul<fwd>(da,WA(u1-1,i),CH(i,k,u1)); \</span></div>
|
|
<div class="line"><a id="l01250" name="l01250"></a><span class="lineno"> 1250</span><span class="preprocessor"> special_mul<fwd>(db,WA(u2-1,i),CH(i,k,u2)); \</span></div>
|
|
<div class="line"><a id="l01251" name="l01251"></a><span class="lineno"> 1251</span><span class="preprocessor"> }</span></div>
|
|
</div>
|
|
<div class="line"><a id="l01252" name="l01252"></a><span class="lineno"> 1252</span> </div>
|
|
<div class="line"><a id="l01253" name="l01253"></a><span class="lineno"> 1253</span><span class="keyword">template</span><<span class="keywordtype">bool</span> fwd, <span class="keyword">typename</span> T> <span class="keywordtype">void</span> pass11 (<span class="keywordtype">size_t</span> ido, <span class="keywordtype">size_t</span> l1,</div>
|
|
<div class="line"><a id="l01254" name="l01254"></a><span class="lineno"> 1254</span> <span class="keyword">const</span> T * <a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> cc, T * <a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> ch,</div>
|
|
<div class="line"><a id="l01255" name="l01255"></a><span class="lineno"> 1255</span> <span class="keyword">const</span> cmplx<T0> * <a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> wa)<span class="keyword"> const</span></div>
|
|
<div class="line"><a id="l01256" name="l01256"></a><span class="lineno"> 1256</span><span class="keyword"> </span>{</div>
|
|
<div class="line"><a id="l01257" name="l01257"></a><span class="lineno"> 1257</span> <span class="keyword">constexpr</span> T0 tw1r= T0(0.8412535328311811688618116489193677L),</div>
|
|
<div class="line"><a id="l01258" name="l01258"></a><span class="lineno"> 1258</span> tw1i= (fwd ? -1 : 1) * T0(0.5406408174555975821076359543186917L),</div>
|
|
<div class="line"><a id="l01259" name="l01259"></a><span class="lineno"> 1259</span> tw2r= T0(0.4154150130018864255292741492296232L),</div>
|
|
<div class="line"><a id="l01260" name="l01260"></a><span class="lineno"> 1260</span> tw2i= (fwd ? -1 : 1) * T0(0.9096319953545183714117153830790285L),</div>
|
|
<div class="line"><a id="l01261" name="l01261"></a><span class="lineno"> 1261</span> tw3r= T0(-0.1423148382732851404437926686163697L),</div>
|
|
<div class="line"><a id="l01262" name="l01262"></a><span class="lineno"> 1262</span> tw3i= (fwd ? -1 : 1) * T0(0.9898214418809327323760920377767188L),</div>
|
|
<div class="line"><a id="l01263" name="l01263"></a><span class="lineno"> 1263</span> tw4r= T0(-0.6548607339452850640569250724662936L),</div>
|
|
<div class="line"><a id="l01264" name="l01264"></a><span class="lineno"> 1264</span> tw4i= (fwd ? -1 : 1) * T0(0.7557495743542582837740358439723444L),</div>
|
|
<div class="line"><a id="l01265" name="l01265"></a><span class="lineno"> 1265</span> tw5r= T0(-0.9594929736144973898903680570663277L),</div>
|
|
<div class="line"><a id="l01266" name="l01266"></a><span class="lineno"> 1266</span> tw5i= (fwd ? -1 : 1) * T0(0.2817325568414296977114179153466169L);</div>
|
|
<div class="line"><a id="l01267" name="l01267"></a><span class="lineno"> 1267</span> </div>
|
|
<div class="line"><a id="l01268" name="l01268"></a><span class="lineno"> 1268</span> <span class="keyword">auto</span> CH = [ch,ido,l1](<span class="keywordtype">size_t</span> a, <span class="keywordtype">size_t</span> b, <span class="keywordtype">size_t</span> c) -> T&</div>
|
|
<div class="line"><a id="l01269" name="l01269"></a><span class="lineno"> 1269</span> { <span class="keywordflow">return</span> ch[a+ido*(b+l1*c)]; };</div>
|
|
<div class="line"><a id="l01270" name="l01270"></a><span class="lineno"> 1270</span> <span class="keyword">auto</span> CC = [cc,ido](<span class="keywordtype">size_t</span> a, <span class="keywordtype">size_t</span> b, <span class="keywordtype">size_t</span> c) -> <span class="keyword">const</span> T&</div>
|
|
<div class="line"><a id="l01271" name="l01271"></a><span class="lineno"> 1271</span> { <span class="keywordflow">return</span> cc[a+ido*(b+11*c)]; };</div>
|
|
<div class="line"><a id="l01272" name="l01272"></a><span class="lineno"> 1272</span> <span class="keyword">auto</span> WA = [wa, ido](<span class="keywordtype">size_t</span> x, <span class="keywordtype">size_t</span> i)</div>
|
|
<div class="line"><a id="l01273" name="l01273"></a><span class="lineno"> 1273</span> { <span class="keywordflow">return</span> wa[i-1+x*(ido-1)]; };</div>
|
|
<div class="line"><a id="l01274" name="l01274"></a><span class="lineno"> 1274</span> </div>
|
|
<div class="line"><a id="l01275" name="l01275"></a><span class="lineno"> 1275</span> <span class="keywordflow">if</span> (ido==1)</div>
|
|
<div class="line"><a id="l01276" name="l01276"></a><span class="lineno"> 1276</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> k=0; k<l1; ++k)</div>
|
|
<div class="line"><a id="l01277" name="l01277"></a><span class="lineno"> 1277</span> {</div>
|
|
<div class="line"><a id="l01278" name="l01278"></a><span class="lineno"> 1278</span> <a class="code hl_define" href="pocketfft_8h.html#a536d2ea61479d4b074bf52ce09fdbc3a">POCKETFFT_PREP11</a>(0)</div>
|
|
<div class="line"><a id="l01279" name="l01279"></a><span class="lineno"> 1279</span> <a class="code hl_define" href="pocketfft_8h.html#ac35e1aa5ae84d655256b7a0afd9051c2">POCKETFFT_PARTSTEP11a</a>(1,10,tw1r,tw2r,tw3r,tw4r,tw5r,+tw1i,+tw2i,+tw3i,+tw4i,+tw5i)</div>
|
|
<div class="line"><a id="l01280" name="l01280"></a><span class="lineno"> 1280</span> <a class="code hl_define" href="pocketfft_8h.html#ac35e1aa5ae84d655256b7a0afd9051c2">POCKETFFT_PARTSTEP11a</a>(2, 9,tw2r,tw4r,tw5r,tw3r,tw1r,+tw2i,+tw4i,-tw5i,-tw3i,-tw1i)</div>
|
|
<div class="line"><a id="l01281" name="l01281"></a><span class="lineno"> 1281</span> <a class="code hl_define" href="pocketfft_8h.html#ac35e1aa5ae84d655256b7a0afd9051c2">POCKETFFT_PARTSTEP11a</a>(3, 8,tw3r,tw5r,tw2r,tw1r,tw4r,+tw3i,-tw5i,-tw2i,+tw1i,+tw4i)</div>
|
|
<div class="line"><a id="l01282" name="l01282"></a><span class="lineno"> 1282</span> <a class="code hl_define" href="pocketfft_8h.html#ac35e1aa5ae84d655256b7a0afd9051c2">POCKETFFT_PARTSTEP11a</a>(4, 7,tw4r,tw3r,tw1r,tw5r,tw2r,+tw4i,-tw3i,+tw1i,+tw5i,-tw2i)</div>
|
|
<div class="line"><a id="l01283" name="l01283"></a><span class="lineno"> 1283</span> <a class="code hl_define" href="pocketfft_8h.html#ac35e1aa5ae84d655256b7a0afd9051c2">POCKETFFT_PARTSTEP11a</a>(5, 6,tw5r,tw1r,tw4r,tw2r,tw3r,+tw5i,-tw1i,+tw4i,-tw2i,+tw3i)</div>
|
|
<div class="line"><a id="l01284" name="l01284"></a><span class="lineno"> 1284</span> }</div>
|
|
<div class="line"><a id="l01285" name="l01285"></a><span class="lineno"> 1285</span> <span class="keywordflow">else</span></div>
|
|
<div class="line"><a id="l01286" name="l01286"></a><span class="lineno"> 1286</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> k=0; k<l1; ++k)</div>
|
|
<div class="line"><a id="l01287" name="l01287"></a><span class="lineno"> 1287</span> {</div>
|
|
<div class="line"><a id="l01288" name="l01288"></a><span class="lineno"> 1288</span> {</div>
|
|
<div class="line"><a id="l01289" name="l01289"></a><span class="lineno"> 1289</span> <a class="code hl_define" href="pocketfft_8h.html#a536d2ea61479d4b074bf52ce09fdbc3a">POCKETFFT_PREP11</a>(0)</div>
|
|
<div class="line"><a id="l01290" name="l01290"></a><span class="lineno"> 1290</span> <a class="code hl_define" href="pocketfft_8h.html#ac35e1aa5ae84d655256b7a0afd9051c2">POCKETFFT_PARTSTEP11a</a>(1,10,tw1r,tw2r,tw3r,tw4r,tw5r,+tw1i,+tw2i,+tw3i,+tw4i,+tw5i)</div>
|
|
<div class="line"><a id="l01291" name="l01291"></a><span class="lineno"> 1291</span> <a class="code hl_define" href="pocketfft_8h.html#ac35e1aa5ae84d655256b7a0afd9051c2">POCKETFFT_PARTSTEP11a</a>(2, 9,tw2r,tw4r,tw5r,tw3r,tw1r,+tw2i,+tw4i,-tw5i,-tw3i,-tw1i)</div>
|
|
<div class="line"><a id="l01292" name="l01292"></a><span class="lineno"> 1292</span> <a class="code hl_define" href="pocketfft_8h.html#ac35e1aa5ae84d655256b7a0afd9051c2">POCKETFFT_PARTSTEP11a</a>(3, 8,tw3r,tw5r,tw2r,tw1r,tw4r,+tw3i,-tw5i,-tw2i,+tw1i,+tw4i)</div>
|
|
<div class="line"><a id="l01293" name="l01293"></a><span class="lineno"> 1293</span> <a class="code hl_define" href="pocketfft_8h.html#ac35e1aa5ae84d655256b7a0afd9051c2">POCKETFFT_PARTSTEP11a</a>(4, 7,tw4r,tw3r,tw1r,tw5r,tw2r,+tw4i,-tw3i,+tw1i,+tw5i,-tw2i)</div>
|
|
<div class="line"><a id="l01294" name="l01294"></a><span class="lineno"> 1294</span> <a class="code hl_define" href="pocketfft_8h.html#ac35e1aa5ae84d655256b7a0afd9051c2">POCKETFFT_PARTSTEP11a</a>(5, 6,tw5r,tw1r,tw4r,tw2r,tw3r,+tw5i,-tw1i,+tw4i,-tw2i,+tw3i)</div>
|
|
<div class="line"><a id="l01295" name="l01295"></a><span class="lineno"> 1295</span> }</div>
|
|
<div class="line"><a id="l01296" name="l01296"></a><span class="lineno"> 1296</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i=1; i<ido; ++i)</div>
|
|
<div class="line"><a id="l01297" name="l01297"></a><span class="lineno"> 1297</span> {</div>
|
|
<div class="line"><a id="l01298" name="l01298"></a><span class="lineno"> 1298</span> <a class="code hl_define" href="pocketfft_8h.html#a536d2ea61479d4b074bf52ce09fdbc3a">POCKETFFT_PREP11</a>(i)</div>
|
|
<div class="line"><a id="l01299" name="l01299"></a><span class="lineno"> 1299</span> <a class="code hl_define" href="pocketfft_8h.html#a1793d0d00f2e13101eb5ad0719c40817">POCKETFFT_PARTSTEP11</a>(1,10,tw1r,tw2r,tw3r,tw4r,tw5r,+tw1i,+tw2i,+tw3i,+tw4i,+tw5i)</div>
|
|
<div class="line"><a id="l01300" name="l01300"></a><span class="lineno"> 1300</span> <a class="code hl_define" href="pocketfft_8h.html#a1793d0d00f2e13101eb5ad0719c40817">POCKETFFT_PARTSTEP11</a>(2, 9,tw2r,tw4r,tw5r,tw3r,tw1r,+tw2i,+tw4i,-tw5i,-tw3i,-tw1i)</div>
|
|
<div class="line"><a id="l01301" name="l01301"></a><span class="lineno"> 1301</span> <a class="code hl_define" href="pocketfft_8h.html#a1793d0d00f2e13101eb5ad0719c40817">POCKETFFT_PARTSTEP11</a>(3, 8,tw3r,tw5r,tw2r,tw1r,tw4r,+tw3i,-tw5i,-tw2i,+tw1i,+tw4i)</div>
|
|
<div class="line"><a id="l01302" name="l01302"></a><span class="lineno"> 1302</span> <a class="code hl_define" href="pocketfft_8h.html#a1793d0d00f2e13101eb5ad0719c40817">POCKETFFT_PARTSTEP11</a>(4, 7,tw4r,tw3r,tw1r,tw5r,tw2r,+tw4i,-tw3i,+tw1i,+tw5i,-tw2i)</div>
|
|
<div class="line"><a id="l01303" name="l01303"></a><span class="lineno"> 1303</span> <a class="code hl_define" href="pocketfft_8h.html#a1793d0d00f2e13101eb5ad0719c40817">POCKETFFT_PARTSTEP11</a>(5, 6,tw5r,tw1r,tw4r,tw2r,tw3r,+tw5i,-tw1i,+tw4i,-tw2i,+tw3i)</div>
|
|
<div class="line"><a id="l01304" name="l01304"></a><span class="lineno"> 1304</span> }</div>
|
|
<div class="line"><a id="l01305" name="l01305"></a><span class="lineno"> 1305</span> }</div>
|
|
<div class="line"><a id="l01306" name="l01306"></a><span class="lineno"> 1306</span> }</div>
|
|
<div class="line"><a id="l01307" name="l01307"></a><span class="lineno"> 1307</span> </div>
|
|
<div class="line"><a id="l01308" name="l01308"></a><span class="lineno"> 1308</span><span class="preprocessor">#undef POCKETFFT_PARTSTEP11</span></div>
|
|
<div class="line"><a id="l01309" name="l01309"></a><span class="lineno"> 1309</span><span class="preprocessor">#undef POCKETFFT_PARTSTEP11a0</span></div>
|
|
<div class="line"><a id="l01310" name="l01310"></a><span class="lineno"> 1310</span><span class="preprocessor">#undef POCKETFFT_PARTSTEP11a</span></div>
|
|
<div class="line"><a id="l01311" name="l01311"></a><span class="lineno"> 1311</span><span class="preprocessor">#undef POCKETFFT_PREP11</span></div>
|
|
<div class="line"><a id="l01312" name="l01312"></a><span class="lineno"> 1312</span> </div>
|
|
<div class="line"><a id="l01313" name="l01313"></a><span class="lineno"> 1313</span><span class="keyword">template</span><<span class="keywordtype">bool</span> fwd, <span class="keyword">typename</span> T> <span class="keywordtype">void</span> passg (<span class="keywordtype">size_t</span> ido, <span class="keywordtype">size_t</span> ip,</div>
|
|
<div class="line"><a id="l01314" name="l01314"></a><span class="lineno"> 1314</span> <span class="keywordtype">size_t</span> l1, T * <a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> cc, T * <a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> ch,</div>
|
|
<div class="line"><a id="l01315" name="l01315"></a><span class="lineno"> 1315</span> <span class="keyword">const</span> cmplx<T0> * <a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> wa,</div>
|
|
<div class="line"><a id="l01316" name="l01316"></a><span class="lineno"> 1316</span> <span class="keyword">const</span> cmplx<T0> * <a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> csarr)<span class="keyword"> const</span></div>
|
|
<div class="line"><a id="l01317" name="l01317"></a><span class="lineno"> 1317</span><span class="keyword"> </span>{</div>
|
|
<div class="line"><a id="l01318" name="l01318"></a><span class="lineno"> 1318</span> <span class="keyword">const</span> <span class="keywordtype">size_t</span> cdim=ip;</div>
|
|
<div class="line"><a id="l01319" name="l01319"></a><span class="lineno"> 1319</span> <span class="keywordtype">size_t</span> ipph = (ip+1)/2;</div>
|
|
<div class="line"><a id="l01320" name="l01320"></a><span class="lineno"> 1320</span> <span class="keywordtype">size_t</span> idl1 = ido*l1;</div>
|
|
<div class="line"><a id="l01321" name="l01321"></a><span class="lineno"> 1321</span> </div>
|
|
<div class="line"><a id="l01322" name="l01322"></a><span class="lineno"> 1322</span> <span class="keyword">auto</span> CH = [ch,ido,l1](<span class="keywordtype">size_t</span> a, <span class="keywordtype">size_t</span> b, <span class="keywordtype">size_t</span> c) -> T&</div>
|
|
<div class="line"><a id="l01323" name="l01323"></a><span class="lineno"> 1323</span> { <span class="keywordflow">return</span> ch[a+ido*(b+l1*c)]; };</div>
|
|
<div class="line"><a id="l01324" name="l01324"></a><span class="lineno"> 1324</span> <span class="keyword">auto</span> CC = [cc,ido,cdim](<span class="keywordtype">size_t</span> a, <span class="keywordtype">size_t</span> b, <span class="keywordtype">size_t</span> c) -> <span class="keyword">const</span> T&</div>
|
|
<div class="line"><a id="l01325" name="l01325"></a><span class="lineno"> 1325</span> { <span class="keywordflow">return</span> cc[a+ido*(b+cdim*c)]; };</div>
|
|
<div class="line"><a id="l01326" name="l01326"></a><span class="lineno"> 1326</span> <span class="keyword">auto</span> CX = [cc, ido, l1](<span class="keywordtype">size_t</span> a, <span class="keywordtype">size_t</span> b, <span class="keywordtype">size_t</span> c) -> T&</div>
|
|
<div class="line"><a id="l01327" name="l01327"></a><span class="lineno"> 1327</span> { <span class="keywordflow">return</span> cc[a+ido*(b+l1*c)]; };</div>
|
|
<div class="line"><a id="l01328" name="l01328"></a><span class="lineno"> 1328</span> <span class="keyword">auto</span> CX2 = [cc, idl1](<span class="keywordtype">size_t</span> a, <span class="keywordtype">size_t</span> b) -> T&</div>
|
|
<div class="line"><a id="l01329" name="l01329"></a><span class="lineno"> 1329</span> { <span class="keywordflow">return</span> cc[a+idl1*b]; };</div>
|
|
<div class="line"><a id="l01330" name="l01330"></a><span class="lineno"> 1330</span> <span class="keyword">auto</span> CH2 = [ch, idl1](<span class="keywordtype">size_t</span> a, <span class="keywordtype">size_t</span> b) -> <span class="keyword">const</span> T&</div>
|
|
<div class="line"><a id="l01331" name="l01331"></a><span class="lineno"> 1331</span> { <span class="keywordflow">return</span> ch[a+idl1*b]; };</div>
|
|
<div class="line"><a id="l01332" name="l01332"></a><span class="lineno"> 1332</span> </div>
|
|
<div class="line"><a id="l01333" name="l01333"></a><span class="lineno"> 1333</span> arr<cmplx<T0>> wal(ip);</div>
|
|
<div class="line"><a id="l01334" name="l01334"></a><span class="lineno"> 1334</span> wal[0] = cmplx<T0>(1., 0.);</div>
|
|
<div class="line"><a id="l01335" name="l01335"></a><span class="lineno"> 1335</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i=1; i<ip; ++i)</div>
|
|
<div class="line"><a id="l01336" name="l01336"></a><span class="lineno"> 1336</span> wal[i]=cmplx<T0>(csarr[i].r,fwd ? -csarr[i].i : csarr[i].i);</div>
|
|
<div class="line"><a id="l01337" name="l01337"></a><span class="lineno"> 1337</span> </div>
|
|
<div class="line"><a id="l01338" name="l01338"></a><span class="lineno"> 1338</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> k=0; k<l1; ++k)</div>
|
|
<div class="line"><a id="l01339" name="l01339"></a><span class="lineno"> 1339</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i=0; i<ido; ++i)</div>
|
|
<div class="line"><a id="l01340" name="l01340"></a><span class="lineno"> 1340</span> CH(i,k,0) = CC(i,0,k);</div>
|
|
<div class="line"><a id="l01341" name="l01341"></a><span class="lineno"> 1341</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> j=1, jc=ip-1; j<ipph; ++j, --jc)</div>
|
|
<div class="line"><a id="l01342" name="l01342"></a><span class="lineno"> 1342</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> k=0; k<l1; ++k)</div>
|
|
<div class="line"><a id="l01343" name="l01343"></a><span class="lineno"> 1343</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i=0; i<ido; ++i)</div>
|
|
<div class="line"><a id="l01344" name="l01344"></a><span class="lineno"> 1344</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a>(CH(i,k,j),CH(i,k,jc),CC(i,j,k),CC(i,jc,k));</div>
|
|
<div class="line"><a id="l01345" name="l01345"></a><span class="lineno"> 1345</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> k=0; k<l1; ++k)</div>
|
|
<div class="line"><a id="l01346" name="l01346"></a><span class="lineno"> 1346</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i=0; i<ido; ++i)</div>
|
|
<div class="line"><a id="l01347" name="l01347"></a><span class="lineno"> 1347</span> {</div>
|
|
<div class="line"><a id="l01348" name="l01348"></a><span class="lineno"> 1348</span> T tmp = CH(i,k,0);</div>
|
|
<div class="line"><a id="l01349" name="l01349"></a><span class="lineno"> 1349</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> j=1; j<ipph; ++j)</div>
|
|
<div class="line"><a id="l01350" name="l01350"></a><span class="lineno"> 1350</span> tmp+=CH(i,k,j);</div>
|
|
<div class="line"><a id="l01351" name="l01351"></a><span class="lineno"> 1351</span> CX(i,k,0) = tmp;</div>
|
|
<div class="line"><a id="l01352" name="l01352"></a><span class="lineno"> 1352</span> }</div>
|
|
<div class="line"><a id="l01353" name="l01353"></a><span class="lineno"> 1353</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> l=1, lc=ip-1; l<ipph; ++l, --lc)</div>
|
|
<div class="line"><a id="l01354" name="l01354"></a><span class="lineno"> 1354</span> {</div>
|
|
<div class="line"><a id="l01355" name="l01355"></a><span class="lineno"> 1355</span> <span class="comment">// j=0</span></div>
|
|
<div class="line"><a id="l01356" name="l01356"></a><span class="lineno"> 1356</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> ik=0; ik<idl1; ++ik)</div>
|
|
<div class="line"><a id="l01357" name="l01357"></a><span class="lineno"> 1357</span> {</div>
|
|
<div class="line"><a id="l01358" name="l01358"></a><span class="lineno"> 1358</span> CX2(ik,l).r = CH2(ik,0).r+wal[l].r*CH2(ik,1).r+wal[2*l].r*CH2(ik,2).r;</div>
|
|
<div class="line"><a id="l01359" name="l01359"></a><span class="lineno"> 1359</span> CX2(ik,l).i = CH2(ik,0).i+wal[l].r*CH2(ik,1).i+wal[2*l].r*CH2(ik,2).i;</div>
|
|
<div class="line"><a id="l01360" name="l01360"></a><span class="lineno"> 1360</span> CX2(ik,lc).r=-wal[l].i*CH2(ik,ip-1).i-wal[2*l].i*CH2(ik,ip-2).i;</div>
|
|
<div class="line"><a id="l01361" name="l01361"></a><span class="lineno"> 1361</span> CX2(ik,lc).i=wal[l].i*CH2(ik,ip-1).r+wal[2*l].i*CH2(ik,ip-2).r;</div>
|
|
<div class="line"><a id="l01362" name="l01362"></a><span class="lineno"> 1362</span> }</div>
|
|
<div class="line"><a id="l01363" name="l01363"></a><span class="lineno"> 1363</span> </div>
|
|
<div class="line"><a id="l01364" name="l01364"></a><span class="lineno"> 1364</span> <span class="keywordtype">size_t</span> iwal=2*l;</div>
|
|
<div class="line"><a id="l01365" name="l01365"></a><span class="lineno"> 1365</span> <span class="keywordtype">size_t</span> j=3, jc=ip-3;</div>
|
|
<div class="line"><a id="l01366" name="l01366"></a><span class="lineno"> 1366</span> <span class="keywordflow">for</span> (; j<ipph-1; j+=2, jc-=2)</div>
|
|
<div class="line"><a id="l01367" name="l01367"></a><span class="lineno"> 1367</span> {</div>
|
|
<div class="line"><a id="l01368" name="l01368"></a><span class="lineno"> 1368</span> iwal+=l; <span class="keywordflow">if</span> (iwal>ip) iwal-=ip;</div>
|
|
<div class="line"><a id="l01369" name="l01369"></a><span class="lineno"> 1369</span> cmplx<T0> xwal=wal[iwal];</div>
|
|
<div class="line"><a id="l01370" name="l01370"></a><span class="lineno"> 1370</span> iwal+=l; <span class="keywordflow">if</span> (iwal>ip) iwal-=ip;</div>
|
|
<div class="line"><a id="l01371" name="l01371"></a><span class="lineno"> 1371</span> cmplx<T0> xwal2=wal[iwal];</div>
|
|
<div class="line"><a id="l01372" name="l01372"></a><span class="lineno"> 1372</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> ik=0; ik<idl1; ++ik)</div>
|
|
<div class="line"><a id="l01373" name="l01373"></a><span class="lineno"> 1373</span> {</div>
|
|
<div class="line"><a id="l01374" name="l01374"></a><span class="lineno"> 1374</span> CX2(ik,l).r += CH2(ik,j).r*xwal.r+CH2(ik,j+1).r*xwal2.r;</div>
|
|
<div class="line"><a id="l01375" name="l01375"></a><span class="lineno"> 1375</span> CX2(ik,l).i += CH2(ik,j).i*xwal.r+CH2(ik,j+1).i*xwal2.r;</div>
|
|
<div class="line"><a id="l01376" name="l01376"></a><span class="lineno"> 1376</span> CX2(ik,lc).r -= CH2(ik,jc).i*xwal.i+CH2(ik,jc-1).i*xwal2.i;</div>
|
|
<div class="line"><a id="l01377" name="l01377"></a><span class="lineno"> 1377</span> CX2(ik,lc).i += CH2(ik,jc).r*xwal.i+CH2(ik,jc-1).r*xwal2.i;</div>
|
|
<div class="line"><a id="l01378" name="l01378"></a><span class="lineno"> 1378</span> }</div>
|
|
<div class="line"><a id="l01379" name="l01379"></a><span class="lineno"> 1379</span> }</div>
|
|
<div class="line"><a id="l01380" name="l01380"></a><span class="lineno"> 1380</span> <span class="keywordflow">for</span> (; j<ipph; ++j, --jc)</div>
|
|
<div class="line"><a id="l01381" name="l01381"></a><span class="lineno"> 1381</span> {</div>
|
|
<div class="line"><a id="l01382" name="l01382"></a><span class="lineno"> 1382</span> iwal+=l; <span class="keywordflow">if</span> (iwal>ip) iwal-=ip;</div>
|
|
<div class="line"><a id="l01383" name="l01383"></a><span class="lineno"> 1383</span> cmplx<T0> xwal=wal[iwal];</div>
|
|
<div class="line"><a id="l01384" name="l01384"></a><span class="lineno"> 1384</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> ik=0; ik<idl1; ++ik)</div>
|
|
<div class="line"><a id="l01385" name="l01385"></a><span class="lineno"> 1385</span> {</div>
|
|
<div class="line"><a id="l01386" name="l01386"></a><span class="lineno"> 1386</span> CX2(ik,l).r += CH2(ik,j).r*xwal.r;</div>
|
|
<div class="line"><a id="l01387" name="l01387"></a><span class="lineno"> 1387</span> CX2(ik,l).i += CH2(ik,j).i*xwal.r;</div>
|
|
<div class="line"><a id="l01388" name="l01388"></a><span class="lineno"> 1388</span> CX2(ik,lc).r -= CH2(ik,jc).i*xwal.i;</div>
|
|
<div class="line"><a id="l01389" name="l01389"></a><span class="lineno"> 1389</span> CX2(ik,lc).i += CH2(ik,jc).r*xwal.i;</div>
|
|
<div class="line"><a id="l01390" name="l01390"></a><span class="lineno"> 1390</span> }</div>
|
|
<div class="line"><a id="l01391" name="l01391"></a><span class="lineno"> 1391</span> }</div>
|
|
<div class="line"><a id="l01392" name="l01392"></a><span class="lineno"> 1392</span> }</div>
|
|
<div class="line"><a id="l01393" name="l01393"></a><span class="lineno"> 1393</span> </div>
|
|
<div class="line"><a id="l01394" name="l01394"></a><span class="lineno"> 1394</span> <span class="comment">// shuffling and twiddling</span></div>
|
|
<div class="line"><a id="l01395" name="l01395"></a><span class="lineno"> 1395</span> <span class="keywordflow">if</span> (ido==1)</div>
|
|
<div class="line"><a id="l01396" name="l01396"></a><span class="lineno"> 1396</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> j=1, jc=ip-1; j<ipph; ++j, --jc)</div>
|
|
<div class="line"><a id="l01397" name="l01397"></a><span class="lineno"> 1397</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> ik=0; ik<idl1; ++ik)</div>
|
|
<div class="line"><a id="l01398" name="l01398"></a><span class="lineno"> 1398</span> {</div>
|
|
<div class="line"><a id="l01399" name="l01399"></a><span class="lineno"> 1399</span> T t1=CX2(ik,j), t2=CX2(ik,jc);</div>
|
|
<div class="line"><a id="l01400" name="l01400"></a><span class="lineno"> 1400</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a>(CX2(ik,j),CX2(ik,jc),t1,t2);</div>
|
|
<div class="line"><a id="l01401" name="l01401"></a><span class="lineno"> 1401</span> }</div>
|
|
<div class="line"><a id="l01402" name="l01402"></a><span class="lineno"> 1402</span> <span class="keywordflow">else</span></div>
|
|
<div class="line"><a id="l01403" name="l01403"></a><span class="lineno"> 1403</span> {</div>
|
|
<div class="line"><a id="l01404" name="l01404"></a><span class="lineno"> 1404</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> j=1, jc=ip-1; j<ipph; ++j,--jc)</div>
|
|
<div class="line"><a id="l01405" name="l01405"></a><span class="lineno"> 1405</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> k=0; k<l1; ++k)</div>
|
|
<div class="line"><a id="l01406" name="l01406"></a><span class="lineno"> 1406</span> {</div>
|
|
<div class="line"><a id="l01407" name="l01407"></a><span class="lineno"> 1407</span> T t1=CX(0,k,j), t2=CX(0,k,jc);</div>
|
|
<div class="line"><a id="l01408" name="l01408"></a><span class="lineno"> 1408</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a>(CX(0,k,j),CX(0,k,jc),t1,t2);</div>
|
|
<div class="line"><a id="l01409" name="l01409"></a><span class="lineno"> 1409</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i=1; i<ido; ++i)</div>
|
|
<div class="line"><a id="l01410" name="l01410"></a><span class="lineno"> 1410</span> {</div>
|
|
<div class="line"><a id="l01411" name="l01411"></a><span class="lineno"> 1411</span> T x1, x2;</div>
|
|
<div class="line"><a id="l01412" name="l01412"></a><span class="lineno"> 1412</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a>(x1,x2,CX(i,k,j),CX(i,k,jc));</div>
|
|
<div class="line"><a id="l01413" name="l01413"></a><span class="lineno"> 1413</span> <span class="keywordtype">size_t</span> idij=(j-1)*(ido-1)+i-1;</div>
|
|
<div class="line"><a id="l01414" name="l01414"></a><span class="lineno"> 1414</span> special_mul<fwd>(x1,wa[idij],CX(i,k,j));</div>
|
|
<div class="line"><a id="l01415" name="l01415"></a><span class="lineno"> 1415</span> idij=(jc-1)*(ido-1)+i-1;</div>
|
|
<div class="line"><a id="l01416" name="l01416"></a><span class="lineno"> 1416</span> special_mul<fwd>(x2,wa[idij],CX(i,k,jc));</div>
|
|
<div class="line"><a id="l01417" name="l01417"></a><span class="lineno"> 1417</span> }</div>
|
|
<div class="line"><a id="l01418" name="l01418"></a><span class="lineno"> 1418</span> }</div>
|
|
<div class="line"><a id="l01419" name="l01419"></a><span class="lineno"> 1419</span> }</div>
|
|
<div class="line"><a id="l01420" name="l01420"></a><span class="lineno"> 1420</span> }</div>
|
|
<div class="line"><a id="l01421" name="l01421"></a><span class="lineno"> 1421</span> </div>
|
|
<div class="line"><a id="l01422" name="l01422"></a><span class="lineno"> 1422</span><span class="keyword">template</span><<span class="keywordtype">bool</span> fwd, <span class="keyword">typename</span> T> <span class="keywordtype">void</span> pass_all(T c[], T0 fct)<span class="keyword"> const</span></div>
|
|
<div class="line"><a id="l01423" name="l01423"></a><span class="lineno"> 1423</span><span class="keyword"> </span>{</div>
|
|
<div class="line"><a id="l01424" name="l01424"></a><span class="lineno"> 1424</span> <span class="keywordflow">if</span> (length==1) { c[0]*=fct; <span class="keywordflow">return</span>; }</div>
|
|
<div class="line"><a id="l01425" name="l01425"></a><span class="lineno"> 1425</span> <span class="keywordtype">size_t</span> l1=1;</div>
|
|
<div class="line"><a id="l01426" name="l01426"></a><span class="lineno"> 1426</span> arr<T> ch(length);</div>
|
|
<div class="line"><a id="l01427" name="l01427"></a><span class="lineno"> 1427</span> T *p1=c, *p2=ch.data();</div>
|
|
<div class="line"><a id="l01428" name="l01428"></a><span class="lineno"> 1428</span> </div>
|
|
<div class="line"><a id="l01429" name="l01429"></a><span class="lineno"> 1429</span> <span class="keywordflow">for</span>(<span class="keywordtype">size_t</span> k1=0; k1<fact.size(); k1++)</div>
|
|
<div class="line"><a id="l01430" name="l01430"></a><span class="lineno"> 1430</span> {</div>
|
|
<div class="line"><a id="l01431" name="l01431"></a><span class="lineno"> 1431</span> <span class="keywordtype">size_t</span> ip=fact[k1].fct;</div>
|
|
<div class="line"><a id="l01432" name="l01432"></a><span class="lineno"> 1432</span> <span class="keywordtype">size_t</span> l2=ip*l1;</div>
|
|
<div class="line"><a id="l01433" name="l01433"></a><span class="lineno"> 1433</span> <span class="keywordtype">size_t</span> ido = length/l2;</div>
|
|
<div class="line"><a id="l01434" name="l01434"></a><span class="lineno"> 1434</span> <span class="keywordflow">if</span> (ip==4)</div>
|
|
<div class="line"><a id="l01435" name="l01435"></a><span class="lineno"> 1435</span> pass4<fwd> (ido, l1, p1, p2, fact[k1].tw);</div>
|
|
<div class="line"><a id="l01436" name="l01436"></a><span class="lineno"> 1436</span> <span class="keywordflow">else</span> <span class="keywordflow">if</span>(ip==8)</div>
|
|
<div class="line"><a id="l01437" name="l01437"></a><span class="lineno"> 1437</span> pass8<fwd>(ido, l1, p1, p2, fact[k1].tw);</div>
|
|
<div class="line"><a id="l01438" name="l01438"></a><span class="lineno"> 1438</span> <span class="keywordflow">else</span> <span class="keywordflow">if</span>(ip==2)</div>
|
|
<div class="line"><a id="l01439" name="l01439"></a><span class="lineno"> 1439</span> pass2<fwd>(ido, l1, p1, p2, fact[k1].tw);</div>
|
|
<div class="line"><a id="l01440" name="l01440"></a><span class="lineno"> 1440</span> <span class="keywordflow">else</span> <span class="keywordflow">if</span>(ip==3)</div>
|
|
<div class="line"><a id="l01441" name="l01441"></a><span class="lineno"> 1441</span> pass3<fwd> (ido, l1, p1, p2, fact[k1].tw);</div>
|
|
<div class="line"><a id="l01442" name="l01442"></a><span class="lineno"> 1442</span> <span class="keywordflow">else</span> <span class="keywordflow">if</span>(ip==5)</div>
|
|
<div class="line"><a id="l01443" name="l01443"></a><span class="lineno"> 1443</span> pass5<fwd> (ido, l1, p1, p2, fact[k1].tw);</div>
|
|
<div class="line"><a id="l01444" name="l01444"></a><span class="lineno"> 1444</span> <span class="keywordflow">else</span> <span class="keywordflow">if</span>(ip==7)</div>
|
|
<div class="line"><a id="l01445" name="l01445"></a><span class="lineno"> 1445</span> pass7<fwd> (ido, l1, p1, p2, fact[k1].tw);</div>
|
|
<div class="line"><a id="l01446" name="l01446"></a><span class="lineno"> 1446</span> <span class="keywordflow">else</span> <span class="keywordflow">if</span>(ip==11)</div>
|
|
<div class="line"><a id="l01447" name="l01447"></a><span class="lineno"> 1447</span> pass11<fwd> (ido, l1, p1, p2, fact[k1].tw);</div>
|
|
<div class="line"><a id="l01448" name="l01448"></a><span class="lineno"> 1448</span> <span class="keywordflow">else</span></div>
|
|
<div class="line"><a id="l01449" name="l01449"></a><span class="lineno"> 1449</span> {</div>
|
|
<div class="line"><a id="l01450" name="l01450"></a><span class="lineno"> 1450</span> passg<fwd>(ido, ip, l1, p1, p2, fact[k1].tw, fact[k1].tws);</div>
|
|
<div class="line"><a id="l01451" name="l01451"></a><span class="lineno"> 1451</span> std::swap(p1,p2);</div>
|
|
<div class="line"><a id="l01452" name="l01452"></a><span class="lineno"> 1452</span> }</div>
|
|
<div class="line"><a id="l01453" name="l01453"></a><span class="lineno"> 1453</span> std::swap(p1,p2);</div>
|
|
<div class="line"><a id="l01454" name="l01454"></a><span class="lineno"> 1454</span> l1=l2;</div>
|
|
<div class="line"><a id="l01455" name="l01455"></a><span class="lineno"> 1455</span> }</div>
|
|
<div class="line"><a id="l01456" name="l01456"></a><span class="lineno"> 1456</span> <span class="keywordflow">if</span> (p1!=c)</div>
|
|
<div class="line"><a id="l01457" name="l01457"></a><span class="lineno"> 1457</span> {</div>
|
|
<div class="line"><a id="l01458" name="l01458"></a><span class="lineno"> 1458</span> <span class="keywordflow">if</span> (fct!=1.)</div>
|
|
<div class="line"><a id="l01459" name="l01459"></a><span class="lineno"> 1459</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i=0; i<length; ++i)</div>
|
|
<div class="line"><a id="l01460" name="l01460"></a><span class="lineno"> 1460</span> c[i] = ch[i]*fct;</div>
|
|
<div class="line"><a id="l01461" name="l01461"></a><span class="lineno"> 1461</span> <span class="keywordflow">else</span></div>
|
|
<div class="line"><a id="l01462" name="l01462"></a><span class="lineno"> 1462</span> std::copy_n (p1, length, c);</div>
|
|
<div class="line"><a id="l01463" name="l01463"></a><span class="lineno"> 1463</span> }</div>
|
|
<div class="line"><a id="l01464" name="l01464"></a><span class="lineno"> 1464</span> <span class="keywordflow">else</span></div>
|
|
<div class="line"><a id="l01465" name="l01465"></a><span class="lineno"> 1465</span> <span class="keywordflow">if</span> (fct!=1.)</div>
|
|
<div class="line"><a id="l01466" name="l01466"></a><span class="lineno"> 1466</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i=0; i<length; ++i)</div>
|
|
<div class="line"><a id="l01467" name="l01467"></a><span class="lineno"> 1467</span> c[i] *= fct;</div>
|
|
<div class="line"><a id="l01468" name="l01468"></a><span class="lineno"> 1468</span> }</div>
|
|
<div class="line"><a id="l01469" name="l01469"></a><span class="lineno"> 1469</span> </div>
|
|
<div class="line"><a id="l01470" name="l01470"></a><span class="lineno"> 1470</span> <span class="keyword">public</span>:</div>
|
|
<div class="foldopen" id="foldopen01471" data-start="{" data-end="}">
|
|
<div class="line"><a id="l01471" name="l01471"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1cfftp.html#a95211024bf007d27e700835db556fbd2"> 1471</a></span> <span class="keyword">template</span><<span class="keyword">typename</span> T> <span class="keywordtype">void</span> <a class="code hl_function" href="classpocketfft_1_1detail_1_1cfftp.html#a95211024bf007d27e700835db556fbd2">exec</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a> c[], <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T0</a> fct, <span class="keywordtype">bool</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">fwd</a>)<span class="keyword"> const</span></div>
|
|
<div class="line"><a id="l01472" name="l01472"></a><span class="lineno"> 1472</span><span class="keyword"> </span>{ <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">fwd</a> ? <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">pass_all<true></a>(c, fct) : <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">pass_all<false></a>(c, fct); }</div>
|
|
</div>
|
|
<div class="line"><a id="l01473" name="l01473"></a><span class="lineno"> 1473</span> </div>
|
|
<div class="line"><a id="l01474" name="l01474"></a><span class="lineno"> 1474</span> <span class="keyword">private</span>:</div>
|
|
<div class="line"><a id="l01475" name="l01475"></a><span class="lineno"> 1475</span> <a class="code hl_define" href="pocketfft_8h.html#a7020984e0ca1d6e565629ca6e7c1a7e0">POCKETFFT_NOINLINE</a> <span class="keywordtype">void</span> factorize()</div>
|
|
<div class="line"><a id="l01476" name="l01476"></a><span class="lineno"> 1476</span> {</div>
|
|
<div class="line"><a id="l01477" name="l01477"></a><span class="lineno"> 1477</span> <span class="keywordtype">size_t</span> len=length;</div>
|
|
<div class="line"><a id="l01478" name="l01478"></a><span class="lineno"> 1478</span> <span class="keywordflow">while</span> ((len&7)==0)</div>
|
|
<div class="line"><a id="l01479" name="l01479"></a><span class="lineno"> 1479</span> { add_factor(8); len>>=3; }</div>
|
|
<div class="line"><a id="l01480" name="l01480"></a><span class="lineno"> 1480</span> <span class="keywordflow">while</span> ((len&3)==0)</div>
|
|
<div class="line"><a id="l01481" name="l01481"></a><span class="lineno"> 1481</span> { add_factor(4); len>>=2; }</div>
|
|
<div class="line"><a id="l01482" name="l01482"></a><span class="lineno"> 1482</span> <span class="keywordflow">if</span> ((len&1)==0)</div>
|
|
<div class="line"><a id="l01483" name="l01483"></a><span class="lineno"> 1483</span> {</div>
|
|
<div class="line"><a id="l01484" name="l01484"></a><span class="lineno"> 1484</span> len>>=1;</div>
|
|
<div class="line"><a id="l01485" name="l01485"></a><span class="lineno"> 1485</span> <span class="comment">// factor 2 should be at the front of the factor list</span></div>
|
|
<div class="line"><a id="l01486" name="l01486"></a><span class="lineno"> 1486</span> add_factor(2);</div>
|
|
<div class="line"><a id="l01487" name="l01487"></a><span class="lineno"> 1487</span> std::swap(fact[0].fct, fact.back().fct);</div>
|
|
<div class="line"><a id="l01488" name="l01488"></a><span class="lineno"> 1488</span> }</div>
|
|
<div class="line"><a id="l01489" name="l01489"></a><span class="lineno"> 1489</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> divisor=3; divisor*divisor<=len; divisor+=2)</div>
|
|
<div class="line"><a id="l01490" name="l01490"></a><span class="lineno"> 1490</span> <span class="keywordflow">while</span> ((len%divisor)==0)</div>
|
|
<div class="line"><a id="l01491" name="l01491"></a><span class="lineno"> 1491</span> {</div>
|
|
<div class="line"><a id="l01492" name="l01492"></a><span class="lineno"> 1492</span> add_factor(divisor);</div>
|
|
<div class="line"><a id="l01493" name="l01493"></a><span class="lineno"> 1493</span> len/=divisor;</div>
|
|
<div class="line"><a id="l01494" name="l01494"></a><span class="lineno"> 1494</span> }</div>
|
|
<div class="line"><a id="l01495" name="l01495"></a><span class="lineno"> 1495</span> <span class="keywordflow">if</span> (len>1) add_factor(len);</div>
|
|
<div class="line"><a id="l01496" name="l01496"></a><span class="lineno"> 1496</span> }</div>
|
|
<div class="line"><a id="l01497" name="l01497"></a><span class="lineno"> 1497</span> </div>
|
|
<div class="line"><a id="l01498" name="l01498"></a><span class="lineno"> 1498</span> <span class="keywordtype">size_t</span> twsize()<span class="keyword"> const</span></div>
|
|
<div class="line"><a id="l01499" name="l01499"></a><span class="lineno"> 1499</span><span class="keyword"> </span>{</div>
|
|
<div class="line"><a id="l01500" name="l01500"></a><span class="lineno"> 1500</span> <span class="keywordtype">size_t</span> twsize=0, l1=1;</div>
|
|
<div class="line"><a id="l01501" name="l01501"></a><span class="lineno"> 1501</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> k=0; k<fact.size(); ++k)</div>
|
|
<div class="line"><a id="l01502" name="l01502"></a><span class="lineno"> 1502</span> {</div>
|
|
<div class="line"><a id="l01503" name="l01503"></a><span class="lineno"> 1503</span> <span class="keywordtype">size_t</span> ip=fact[k].fct, ido= length/(l1*ip);</div>
|
|
<div class="line"><a id="l01504" name="l01504"></a><span class="lineno"> 1504</span> twsize+=(ip-1)*(ido-1);</div>
|
|
<div class="line"><a id="l01505" name="l01505"></a><span class="lineno"> 1505</span> <span class="keywordflow">if</span> (ip>11)</div>
|
|
<div class="line"><a id="l01506" name="l01506"></a><span class="lineno"> 1506</span> twsize+=ip;</div>
|
|
<div class="line"><a id="l01507" name="l01507"></a><span class="lineno"> 1507</span> l1*=ip;</div>
|
|
<div class="line"><a id="l01508" name="l01508"></a><span class="lineno"> 1508</span> }</div>
|
|
<div class="line"><a id="l01509" name="l01509"></a><span class="lineno"> 1509</span> <span class="keywordflow">return</span> twsize;</div>
|
|
<div class="line"><a id="l01510" name="l01510"></a><span class="lineno"> 1510</span> }</div>
|
|
<div class="line"><a id="l01511" name="l01511"></a><span class="lineno"> 1511</span> </div>
|
|
<div class="line"><a id="l01512" name="l01512"></a><span class="lineno"> 1512</span> <span class="keywordtype">void</span> comp_twiddle()</div>
|
|
<div class="line"><a id="l01513" name="l01513"></a><span class="lineno"> 1513</span> {</div>
|
|
<div class="line"><a id="l01514" name="l01514"></a><span class="lineno"> 1514</span> sincos_2pibyn<T0> twiddle(length);</div>
|
|
<div class="line"><a id="l01515" name="l01515"></a><span class="lineno"> 1515</span> <span class="keywordtype">size_t</span> l1=1;</div>
|
|
<div class="line"><a id="l01516" name="l01516"></a><span class="lineno"> 1516</span> <span class="keywordtype">size_t</span> memofs=0;</div>
|
|
<div class="line"><a id="l01517" name="l01517"></a><span class="lineno"> 1517</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> k=0; k<fact.size(); ++k)</div>
|
|
<div class="line"><a id="l01518" name="l01518"></a><span class="lineno"> 1518</span> {</div>
|
|
<div class="line"><a id="l01519" name="l01519"></a><span class="lineno"> 1519</span> <span class="keywordtype">size_t</span> ip=fact[k].fct, ido=length/(l1*ip);</div>
|
|
<div class="line"><a id="l01520" name="l01520"></a><span class="lineno"> 1520</span> fact[k].tw=mem.data()+memofs;</div>
|
|
<div class="line"><a id="l01521" name="l01521"></a><span class="lineno"> 1521</span> memofs+=(ip-1)*(ido-1);</div>
|
|
<div class="line"><a id="l01522" name="l01522"></a><span class="lineno"> 1522</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> j=1; j<ip; ++j)</div>
|
|
<div class="line"><a id="l01523" name="l01523"></a><span class="lineno"> 1523</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i=1; i<ido; ++i)</div>
|
|
<div class="line"><a id="l01524" name="l01524"></a><span class="lineno"> 1524</span> fact[k].tw[(j-1)*(ido-1)+i-1] = twiddle[j*l1*i];</div>
|
|
<div class="line"><a id="l01525" name="l01525"></a><span class="lineno"> 1525</span> <span class="keywordflow">if</span> (ip>11)</div>
|
|
<div class="line"><a id="l01526" name="l01526"></a><span class="lineno"> 1526</span> {</div>
|
|
<div class="line"><a id="l01527" name="l01527"></a><span class="lineno"> 1527</span> fact[k].tws=mem.data()+memofs;</div>
|
|
<div class="line"><a id="l01528" name="l01528"></a><span class="lineno"> 1528</span> memofs+=ip;</div>
|
|
<div class="line"><a id="l01529" name="l01529"></a><span class="lineno"> 1529</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> j=0; j<ip; ++j)</div>
|
|
<div class="line"><a id="l01530" name="l01530"></a><span class="lineno"> 1530</span> fact[k].tws[j] = twiddle[j*l1*ido];</div>
|
|
<div class="line"><a id="l01531" name="l01531"></a><span class="lineno"> 1531</span> }</div>
|
|
<div class="line"><a id="l01532" name="l01532"></a><span class="lineno"> 1532</span> l1*=ip;</div>
|
|
<div class="line"><a id="l01533" name="l01533"></a><span class="lineno"> 1533</span> }</div>
|
|
<div class="line"><a id="l01534" name="l01534"></a><span class="lineno"> 1534</span> }</div>
|
|
<div class="line"><a id="l01535" name="l01535"></a><span class="lineno"> 1535</span> </div>
|
|
<div class="line"><a id="l01536" name="l01536"></a><span class="lineno"> 1536</span> <span class="keyword">public</span>:</div>
|
|
<div class="foldopen" id="foldopen01537" data-start="{" data-end="}">
|
|
<div class="line"><a id="l01537" name="l01537"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1cfftp.html#a121dcd8d4db180061d09fc1c5eb7da27"> 1537</a></span> <a class="code hl_define" href="pocketfft_8h.html#a7020984e0ca1d6e565629ca6e7c1a7e0">POCKETFFT_NOINLINE</a> <a class="code hl_function" href="classpocketfft_1_1detail_1_1cfftp.html#a121dcd8d4db180061d09fc1c5eb7da27">cfftp</a>(<span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">length_</a>)</div>
|
|
<div class="line"><a id="l01538" name="l01538"></a><span class="lineno"> 1538</span> : length(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">length_</a>)</div>
|
|
<div class="line"><a id="l01539" name="l01539"></a><span class="lineno"> 1539</span> {</div>
|
|
<div class="line"><a id="l01540" name="l01540"></a><span class="lineno"> 1540</span> <span class="keywordflow">if</span> (length==0) <span class="keywordflow">throw</span> std::runtime_error(<span class="stringliteral">"zero-length FFT requested"</span>);</div>
|
|
<div class="line"><a id="l01541" name="l01541"></a><span class="lineno"> 1541</span> <span class="keywordflow">if</span> (length==1) <span class="keywordflow">return</span>;</div>
|
|
<div class="line"><a id="l01542" name="l01542"></a><span class="lineno"> 1542</span> factorize();</div>
|
|
<div class="line"><a id="l01543" name="l01543"></a><span class="lineno"> 1543</span> mem.resize(twsize());</div>
|
|
<div class="line"><a id="l01544" name="l01544"></a><span class="lineno"> 1544</span> comp_twiddle();</div>
|
|
<div class="line"><a id="l01545" name="l01545"></a><span class="lineno"> 1545</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l01546" name="l01546"></a><span class="lineno"> 1546</span> };</div>
|
|
</div>
|
|
<div class="line"><a id="l01547" name="l01547"></a><span class="lineno"> 1547</span> </div>
|
|
<div class="line"><a id="l01548" name="l01548"></a><span class="lineno"> 1548</span><span class="comment">//</span></div>
|
|
<div class="line"><a id="l01549" name="l01549"></a><span class="lineno"> 1549</span><span class="comment">// real-valued FFTPACK transforms</span></div>
|
|
<div class="line"><a id="l01550" name="l01550"></a><span class="lineno"> 1550</span><span class="comment">//</span></div>
|
|
<div class="line"><a id="l01551" name="l01551"></a><span class="lineno"> 1551</span> </div>
|
|
<div class="foldopen" id="foldopen01552" data-start="{" data-end="};">
|
|
<div class="line"><a id="l01552" name="l01552"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1rfftp.html"> 1552</a></span><span class="keyword">template</span><<span class="keyword">typename</span> T0> <span class="keyword">class </span><a class="code hl_class" href="classpocketfft_1_1detail_1_1rfftp.html">rfftp</a></div>
|
|
<div class="line"><a id="l01553" name="l01553"></a><span class="lineno"> 1553</span> {</div>
|
|
<div class="line"><a id="l01554" name="l01554"></a><span class="lineno"> 1554</span> <span class="keyword">private</span>:</div>
|
|
<div class="line"><a id="l01555" name="l01555"></a><span class="lineno"> 1555</span> <span class="keyword">struct </span>fctdata</div>
|
|
<div class="line"><a id="l01556" name="l01556"></a><span class="lineno"> 1556</span> {</div>
|
|
<div class="line"><a id="l01557" name="l01557"></a><span class="lineno"> 1557</span> <span class="keywordtype">size_t</span> fct;</div>
|
|
<div class="line"><a id="l01558" name="l01558"></a><span class="lineno"> 1558</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T0</a> *tw, *tws;</div>
|
|
<div class="line"><a id="l01559" name="l01559"></a><span class="lineno"> 1559</span> };</div>
|
|
<div class="line"><a id="l01560" name="l01560"></a><span class="lineno"> 1560</span> </div>
|
|
<div class="line"><a id="l01561" name="l01561"></a><span class="lineno"> 1561</span> <span class="keywordtype">size_t</span> length;</div>
|
|
<div class="line"><a id="l01562" name="l01562"></a><span class="lineno"> 1562</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">arr<T0></a> mem;</div>
|
|
<div class="line"><a id="l01563" name="l01563"></a><span class="lineno"> 1563</span> std::vector<fctdata> fact;</div>
|
|
<div class="line"><a id="l01564" name="l01564"></a><span class="lineno"> 1564</span> </div>
|
|
<div class="line"><a id="l01565" name="l01565"></a><span class="lineno"> 1565</span> <span class="keywordtype">void</span> add_factor(<span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">factor</a>)</div>
|
|
<div class="line"><a id="l01566" name="l01566"></a><span class="lineno"> 1566</span> { fact.push_back({<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">factor</a>, <span class="keyword">nullptr</span>, <span class="keyword">nullptr</span>}); }</div>
|
|
<div class="line"><a id="l01567" name="l01567"></a><span class="lineno"> 1567</span> </div>
|
|
<div class="line"><a id="l01568" name="l01568"></a><span class="lineno"> 1568</span><span class="comment">/* (a+ib) = conj(c+id) * (e+if) */</span></div>
|
|
<div class="line"><a id="l01569" name="l01569"></a><span class="lineno"> 1569</span><span class="keyword">template</span><<span class="keyword">typename</span> T1, <span class="keyword">typename</span> T2, <span class="keyword">typename</span> T3> <span class="keyword">inline</span> <span class="keywordtype">void</span> MULPM</div>
|
|
<div class="line"><a id="l01570" name="l01570"></a><span class="lineno"> 1570</span> (<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T1</a> &<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">a</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T1</a> &b, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T2</a> c, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T2</a> d, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T3</a> e, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T3</a> <a class="code hl_variable" href="types_2bf16_8h.html#af900396d7b72ff2a7002e8befe8cf8f1">f</a>)<span class="keyword"> const</span></div>
|
|
<div class="line"><a id="l01571" name="l01571"></a><span class="lineno"> 1571</span><span class="keyword"> </span>{ <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">a</a>=c*e+d*<a class="code hl_variable" href="types_2bf16_8h.html#af900396d7b72ff2a7002e8befe8cf8f1">f</a>; b=c*<a class="code hl_variable" href="types_2bf16_8h.html#af900396d7b72ff2a7002e8befe8cf8f1">f</a>-d*e; }</div>
|
|
<div class="line"><a id="l01572" name="l01572"></a><span class="lineno"> 1572</span> </div>
|
|
<div class="line"><a id="l01573" name="l01573"></a><span class="lineno"> 1573</span><span class="keyword">template</span><<span class="keyword">typename</span> T> <span class="keywordtype">void</span> radf2 (<span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ido</a>, <span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">l1</a>,</div>
|
|
<div class="line"><a id="l01574" name="l01574"></a><span class="lineno"> 1574</span> <span class="keyword">const</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a> * <a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cc</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a> * <a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ch</a>,</div>
|
|
<div class="line"><a id="l01575" name="l01575"></a><span class="lineno"> 1575</span> <span class="keyword">const</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T0</a> * <a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">wa</a>)<span class="keyword"> const</span></div>
|
|
<div class="line"><a id="l01576" name="l01576"></a><span class="lineno"> 1576</span><span class="keyword"> </span>{</div>
|
|
<div class="line"><a id="l01577" name="l01577"></a><span class="lineno"> 1577</span> <span class="keyword">auto</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">WA</a> = [<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">wa</a>,<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ido</a>](<span class="keywordtype">size_t</span> x, <span class="keywordtype">size_t</span> i) { <span class="keywordflow">return</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">wa</a>[i+x*(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ido</a>-1)]; };</div>
|
|
<div class="line"><a id="l01578" name="l01578"></a><span class="lineno"> 1578</span> <span class="keyword">auto</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">CC</a> = [<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cc</a>,<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ido</a>,<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">l1</a>](<span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">a</a>, <span class="keywordtype">size_t</span> b, <span class="keywordtype">size_t</span> c) -> <span class="keyword">const</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a>&</div>
|
|
<div class="line"><a id="l01579" name="l01579"></a><span class="lineno"> 1579</span> { <span class="keywordflow">return</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cc</a>[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">a</a>+<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ido</a>*(b+<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">l1</a>*c)]; };</div>
|
|
<div class="line"><a id="l01580" name="l01580"></a><span class="lineno"> 1580</span> <span class="keyword">auto</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">CH</a> = [<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ch</a>,<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ido</a>](<span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">a</a>, <span class="keywordtype">size_t</span> b, <span class="keywordtype">size_t</span> c) -> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a>&</div>
|
|
<div class="line"><a id="l01581" name="l01581"></a><span class="lineno"> 1581</span> { <span class="keywordflow">return</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ch</a>[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">a</a>+<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ido</a>*(b+2*c)]; };</div>
|
|
<div class="line"><a id="l01582" name="l01582"></a><span class="lineno"> 1582</span> </div>
|
|
<div class="line"><a id="l01583" name="l01583"></a><span class="lineno"> 1583</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>=0; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a><<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">l1</a>; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>++)</div>
|
|
<div class="line"><a id="l01584" name="l01584"></a><span class="lineno"> 1584</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a> (<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">CH</a>(0,0,<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>),<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">CH</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ido</a>-1,1,<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>),<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">CC</a>(0,<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>,0),<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">CC</a>(0,<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>,1));</div>
|
|
<div class="line"><a id="l01585" name="l01585"></a><span class="lineno"> 1585</span> <span class="keywordflow">if</span> ((<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ido</a>&1)==0)</div>
|
|
<div class="line"><a id="l01586" name="l01586"></a><span class="lineno"> 1586</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>=0; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a><<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">l1</a>; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>++)</div>
|
|
<div class="line"><a id="l01587" name="l01587"></a><span class="lineno"> 1587</span> {</div>
|
|
<div class="line"><a id="l01588" name="l01588"></a><span class="lineno"> 1588</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">CH</a>( 0,1,<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>) = -<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">CC</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ido</a>-1,<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>,1);</div>
|
|
<div class="line"><a id="l01589" name="l01589"></a><span class="lineno"> 1589</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">CH</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ido</a>-1,0,<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>) = <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">CC</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ido</a>-1,<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>,0);</div>
|
|
<div class="line"><a id="l01590" name="l01590"></a><span class="lineno"> 1590</span> }</div>
|
|
<div class="line"><a id="l01591" name="l01591"></a><span class="lineno"> 1591</span> <span class="keywordflow">if</span> (<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ido</a><=2) <span class="keywordflow">return</span>;</div>
|
|
<div class="line"><a id="l01592" name="l01592"></a><span class="lineno"> 1592</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>=0; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a><<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">l1</a>; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>++)</div>
|
|
<div class="line"><a id="l01593" name="l01593"></a><span class="lineno"> 1593</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i=2; i<<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ido</a>; i+=2)</div>
|
|
<div class="line"><a id="l01594" name="l01594"></a><span class="lineno"> 1594</span> {</div>
|
|
<div class="line"><a id="l01595" name="l01595"></a><span class="lineno"> 1595</span> <span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ic</a>=<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ido</a>-i;</div>
|
|
<div class="line"><a id="l01596" name="l01596"></a><span class="lineno"> 1596</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tr2</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ti2</a>;</div>
|
|
<div class="line"><a id="l01597" name="l01597"></a><span class="lineno"> 1597</span> MULPM (<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tr2</a>,<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ti2</a>,<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">WA</a>(0,i-2),<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">WA</a>(0,i-1),<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">CC</a>(i-1,<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>,1),<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">CC</a>(i,<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>,1));</div>
|
|
<div class="line"><a id="l01598" name="l01598"></a><span class="lineno"> 1598</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a> (<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">CH</a>(i-1,0,<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>),<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">CH</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ic</a>-1,1,<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>),<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">CC</a>(i-1,<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>,0),<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tr2</a>);</div>
|
|
<div class="line"><a id="l01599" name="l01599"></a><span class="lineno"> 1599</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a> (<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">CH</a>(i ,0,<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>),<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">CH</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ic</a> ,1,<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>),<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ti2</a>,<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">CC</a>(i ,<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>,0));</div>
|
|
<div class="line"><a id="l01600" name="l01600"></a><span class="lineno"> 1600</span> }</div>
|
|
<div class="line"><a id="l01601" name="l01601"></a><span class="lineno"> 1601</span> }</div>
|
|
<div class="line"><a id="l01602" name="l01602"></a><span class="lineno"> 1602</span> </div>
|
|
<div class="line"><a id="l01603" name="l01603"></a><span class="lineno"> 1603</span><span class="comment">// a2=a+b; b2=i*(b-a);</span></div>
|
|
<div class="foldopen" id="foldopen01604" data-start="" data-end="">
|
|
<div class="line"><a id="l01604" name="l01604"></a><span class="lineno"><a class="line" href="pocketfft_8h.html#acffdf2e1ab84f36a7a097e1b8b87a9f9"> 1604</a></span><span class="preprocessor">#define POCKETFFT_REARRANGE(rx, ix, ry, iy) \</span></div>
|
|
<div class="line"><a id="l01605" name="l01605"></a><span class="lineno"> 1605</span><span class="preprocessor"> {\</span></div>
|
|
<div class="line"><a id="l01606" name="l01606"></a><span class="lineno"> 1606</span><span class="preprocessor"> auto t1=rx+ry, t2=ry-rx, t3=ix+iy, t4=ix-iy; \</span></div>
|
|
<div class="line"><a id="l01607" name="l01607"></a><span class="lineno"> 1607</span><span class="preprocessor"> rx=t1; ix=t3; ry=t4; iy=t2; \</span></div>
|
|
<div class="line"><a id="l01608" name="l01608"></a><span class="lineno"> 1608</span><span class="preprocessor"> }</span></div>
|
|
</div>
|
|
<div class="line"><a id="l01609" name="l01609"></a><span class="lineno"> 1609</span> </div>
|
|
<div class="line"><a id="l01610" name="l01610"></a><span class="lineno"> 1610</span><span class="keyword">template</span><<span class="keyword">typename</span> T> <span class="keywordtype">void</span> radf3(<span class="keywordtype">size_t</span> ido, <span class="keywordtype">size_t</span> l1,</div>
|
|
<div class="line"><a id="l01611" name="l01611"></a><span class="lineno"> 1611</span> <span class="keyword">const</span> T * <a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> cc, T * <a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> ch,</div>
|
|
<div class="line"><a id="l01612" name="l01612"></a><span class="lineno"> 1612</span> <span class="keyword">const</span> T0 * <a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> wa)<span class="keyword"> const</span></div>
|
|
<div class="line"><a id="l01613" name="l01613"></a><span class="lineno"> 1613</span><span class="keyword"> </span>{</div>
|
|
<div class="line"><a id="l01614" name="l01614"></a><span class="lineno"> 1614</span> <span class="keyword">constexpr</span> T0 taur=-0.5, taui=T0(0.8660254037844386467637231707529362L);</div>
|
|
<div class="line"><a id="l01615" name="l01615"></a><span class="lineno"> 1615</span> </div>
|
|
<div class="line"><a id="l01616" name="l01616"></a><span class="lineno"> 1616</span> <span class="keyword">auto</span> WA = [wa,ido](<span class="keywordtype">size_t</span> x, <span class="keywordtype">size_t</span> i) { <span class="keywordflow">return</span> wa[i+x*(ido-1)]; };</div>
|
|
<div class="line"><a id="l01617" name="l01617"></a><span class="lineno"> 1617</span> <span class="keyword">auto</span> CC = [cc,ido,l1](<span class="keywordtype">size_t</span> a, <span class="keywordtype">size_t</span> b, <span class="keywordtype">size_t</span> c) -> <span class="keyword">const</span> T&</div>
|
|
<div class="line"><a id="l01618" name="l01618"></a><span class="lineno"> 1618</span> { <span class="keywordflow">return</span> cc[a+ido*(b+l1*c)]; };</div>
|
|
<div class="line"><a id="l01619" name="l01619"></a><span class="lineno"> 1619</span> <span class="keyword">auto</span> CH = [ch,ido](<span class="keywordtype">size_t</span> a, <span class="keywordtype">size_t</span> b, <span class="keywordtype">size_t</span> c) -> T&</div>
|
|
<div class="line"><a id="l01620" name="l01620"></a><span class="lineno"> 1620</span> { <span class="keywordflow">return</span> ch[a+ido*(b+3*c)]; };</div>
|
|
<div class="line"><a id="l01621" name="l01621"></a><span class="lineno"> 1621</span> </div>
|
|
<div class="line"><a id="l01622" name="l01622"></a><span class="lineno"> 1622</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> k=0; k<l1; k++)</div>
|
|
<div class="line"><a id="l01623" name="l01623"></a><span class="lineno"> 1623</span> {</div>
|
|
<div class="line"><a id="l01624" name="l01624"></a><span class="lineno"> 1624</span> T cr2=CC(0,k,1)+CC(0,k,2);</div>
|
|
<div class="line"><a id="l01625" name="l01625"></a><span class="lineno"> 1625</span> CH(0,0,k) = CC(0,k,0)+cr2;</div>
|
|
<div class="line"><a id="l01626" name="l01626"></a><span class="lineno"> 1626</span> CH(0,2,k) = taui*(CC(0,k,2)-CC(0,k,1));</div>
|
|
<div class="line"><a id="l01627" name="l01627"></a><span class="lineno"> 1627</span> CH(ido-1,1,k) = CC(0,k,0)+taur*cr2;</div>
|
|
<div class="line"><a id="l01628" name="l01628"></a><span class="lineno"> 1628</span> }</div>
|
|
<div class="line"><a id="l01629" name="l01629"></a><span class="lineno"> 1629</span> <span class="keywordflow">if</span> (ido==1) <span class="keywordflow">return</span>;</div>
|
|
<div class="line"><a id="l01630" name="l01630"></a><span class="lineno"> 1630</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> k=0; k<l1; k++)</div>
|
|
<div class="line"><a id="l01631" name="l01631"></a><span class="lineno"> 1631</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i=2; i<ido; i+=2)</div>
|
|
<div class="line"><a id="l01632" name="l01632"></a><span class="lineno"> 1632</span> {</div>
|
|
<div class="line"><a id="l01633" name="l01633"></a><span class="lineno"> 1633</span> <span class="keywordtype">size_t</span> ic=ido-i;</div>
|
|
<div class="line"><a id="l01634" name="l01634"></a><span class="lineno"> 1634</span> T di2, di3, dr2, dr3;</div>
|
|
<div class="line"><a id="l01635" name="l01635"></a><span class="lineno"> 1635</span> MULPM (dr2,di2,WA(0,i-2),WA(0,i-1),CC(i-1,k,1),CC(i,k,1)); <span class="comment">// d2=conj(WA0)*CC1</span></div>
|
|
<div class="line"><a id="l01636" name="l01636"></a><span class="lineno"> 1636</span> MULPM (dr3,di3,WA(1,i-2),WA(1,i-1),CC(i-1,k,2),CC(i,k,2)); <span class="comment">// d3=conj(WA1)*CC2</span></div>
|
|
<div class="line"><a id="l01637" name="l01637"></a><span class="lineno"> 1637</span> <a class="code hl_define" href="pocketfft_8h.html#acffdf2e1ab84f36a7a097e1b8b87a9f9">POCKETFFT_REARRANGE</a>(dr2, di2, dr3, di3);</div>
|
|
<div class="line"><a id="l01638" name="l01638"></a><span class="lineno"> 1638</span> CH(i-1,0,k) = CC(i-1,k,0)+dr2; <span class="comment">// c add</span></div>
|
|
<div class="line"><a id="l01639" name="l01639"></a><span class="lineno"> 1639</span> CH(i ,0,k) = CC(i ,k,0)+di2;</div>
|
|
<div class="line"><a id="l01640" name="l01640"></a><span class="lineno"> 1640</span> T tr2 = CC(i-1,k,0)+taur*dr2; <span class="comment">// c add</span></div>
|
|
<div class="line"><a id="l01641" name="l01641"></a><span class="lineno"> 1641</span> T ti2 = CC(i ,k,0)+taur*di2;</div>
|
|
<div class="line"><a id="l01642" name="l01642"></a><span class="lineno"> 1642</span> T tr3 = taui*dr3; <span class="comment">// t3 = taui*i*(d3-d2)?</span></div>
|
|
<div class="line"><a id="l01643" name="l01643"></a><span class="lineno"> 1643</span> T ti3 = taui*di3;</div>
|
|
<div class="line"><a id="l01644" name="l01644"></a><span class="lineno"> 1644</span> PM(CH(i-1,2,k),CH(ic-1,1,k),tr2,tr3); <span class="comment">// PM(i) = t2+t3</span></div>
|
|
<div class="line"><a id="l01645" name="l01645"></a><span class="lineno"> 1645</span> PM(CH(i ,2,k),CH(ic ,1,k),ti3,ti2); <span class="comment">// PM(ic) = conj(t2-t3)</span></div>
|
|
<div class="line"><a id="l01646" name="l01646"></a><span class="lineno"> 1646</span> }</div>
|
|
<div class="line"><a id="l01647" name="l01647"></a><span class="lineno"> 1647</span> }</div>
|
|
<div class="line"><a id="l01648" name="l01648"></a><span class="lineno"> 1648</span> </div>
|
|
<div class="line"><a id="l01649" name="l01649"></a><span class="lineno"> 1649</span><span class="keyword">template</span><<span class="keyword">typename</span> T> <span class="keywordtype">void</span> radf4(<span class="keywordtype">size_t</span> ido, <span class="keywordtype">size_t</span> l1,</div>
|
|
<div class="line"><a id="l01650" name="l01650"></a><span class="lineno"> 1650</span> <span class="keyword">const</span> T * <a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> cc, T * <a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> ch,</div>
|
|
<div class="line"><a id="l01651" name="l01651"></a><span class="lineno"> 1651</span> <span class="keyword">const</span> T0 * <a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> wa)<span class="keyword"> const</span></div>
|
|
<div class="line"><a id="l01652" name="l01652"></a><span class="lineno"> 1652</span><span class="keyword"> </span>{</div>
|
|
<div class="line"><a id="l01653" name="l01653"></a><span class="lineno"> 1653</span> <span class="keyword">constexpr</span> T0 hsqt2=T0(0.707106781186547524400844362104849L);</div>
|
|
<div class="line"><a id="l01654" name="l01654"></a><span class="lineno"> 1654</span> </div>
|
|
<div class="line"><a id="l01655" name="l01655"></a><span class="lineno"> 1655</span> <span class="keyword">auto</span> WA = [wa,ido](<span class="keywordtype">size_t</span> x, <span class="keywordtype">size_t</span> i) { <span class="keywordflow">return</span> wa[i+x*(ido-1)]; };</div>
|
|
<div class="line"><a id="l01656" name="l01656"></a><span class="lineno"> 1656</span> <span class="keyword">auto</span> CC = [cc,ido,l1](<span class="keywordtype">size_t</span> a, <span class="keywordtype">size_t</span> b, <span class="keywordtype">size_t</span> c) -> <span class="keyword">const</span> T&</div>
|
|
<div class="line"><a id="l01657" name="l01657"></a><span class="lineno"> 1657</span> { <span class="keywordflow">return</span> cc[a+ido*(b+l1*c)]; };</div>
|
|
<div class="line"><a id="l01658" name="l01658"></a><span class="lineno"> 1658</span> <span class="keyword">auto</span> CH = [ch,ido](<span class="keywordtype">size_t</span> a, <span class="keywordtype">size_t</span> b, <span class="keywordtype">size_t</span> c) -> T&</div>
|
|
<div class="line"><a id="l01659" name="l01659"></a><span class="lineno"> 1659</span> { <span class="keywordflow">return</span> ch[a+ido*(b+4*c)]; };</div>
|
|
<div class="line"><a id="l01660" name="l01660"></a><span class="lineno"> 1660</span> </div>
|
|
<div class="line"><a id="l01661" name="l01661"></a><span class="lineno"> 1661</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> k=0; k<l1; k++)</div>
|
|
<div class="line"><a id="l01662" name="l01662"></a><span class="lineno"> 1662</span> {</div>
|
|
<div class="line"><a id="l01663" name="l01663"></a><span class="lineno"> 1663</span> T tr1,tr2;</div>
|
|
<div class="line"><a id="l01664" name="l01664"></a><span class="lineno"> 1664</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a> (tr1,CH(0,2,k),CC(0,k,3),CC(0,k,1));</div>
|
|
<div class="line"><a id="l01665" name="l01665"></a><span class="lineno"> 1665</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a> (tr2,CH(ido-1,1,k),CC(0,k,0),CC(0,k,2));</div>
|
|
<div class="line"><a id="l01666" name="l01666"></a><span class="lineno"> 1666</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a> (CH(0,0,k),CH(ido-1,3,k),tr2,tr1);</div>
|
|
<div class="line"><a id="l01667" name="l01667"></a><span class="lineno"> 1667</span> }</div>
|
|
<div class="line"><a id="l01668" name="l01668"></a><span class="lineno"> 1668</span> <span class="keywordflow">if</span> ((ido&1)==0)</div>
|
|
<div class="line"><a id="l01669" name="l01669"></a><span class="lineno"> 1669</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> k=0; k<l1; k++)</div>
|
|
<div class="line"><a id="l01670" name="l01670"></a><span class="lineno"> 1670</span> {</div>
|
|
<div class="line"><a id="l01671" name="l01671"></a><span class="lineno"> 1671</span> T ti1=-hsqt2*(CC(ido-1,k,1)+CC(ido-1,k,3));</div>
|
|
<div class="line"><a id="l01672" name="l01672"></a><span class="lineno"> 1672</span> T tr1= hsqt2*(CC(ido-1,k,1)-CC(ido-1,k,3));</div>
|
|
<div class="line"><a id="l01673" name="l01673"></a><span class="lineno"> 1673</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a> (CH(ido-1,0,k),CH(ido-1,2,k),CC(ido-1,k,0),tr1);</div>
|
|
<div class="line"><a id="l01674" name="l01674"></a><span class="lineno"> 1674</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a> (CH( 0,3,k),CH( 0,1,k),ti1,CC(ido-1,k,2));</div>
|
|
<div class="line"><a id="l01675" name="l01675"></a><span class="lineno"> 1675</span> }</div>
|
|
<div class="line"><a id="l01676" name="l01676"></a><span class="lineno"> 1676</span> <span class="keywordflow">if</span> (ido<=2) <span class="keywordflow">return</span>;</div>
|
|
<div class="line"><a id="l01677" name="l01677"></a><span class="lineno"> 1677</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> k=0; k<l1; k++)</div>
|
|
<div class="line"><a id="l01678" name="l01678"></a><span class="lineno"> 1678</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i=2; i<ido; i+=2)</div>
|
|
<div class="line"><a id="l01679" name="l01679"></a><span class="lineno"> 1679</span> {</div>
|
|
<div class="line"><a id="l01680" name="l01680"></a><span class="lineno"> 1680</span> <span class="keywordtype">size_t</span> ic=ido-i;</div>
|
|
<div class="line"><a id="l01681" name="l01681"></a><span class="lineno"> 1681</span> T ci2, ci3, ci4, cr2, cr3, cr4, ti1, ti2, ti3, ti4, tr1, tr2, tr3, tr4;</div>
|
|
<div class="line"><a id="l01682" name="l01682"></a><span class="lineno"> 1682</span> MULPM(cr2,ci2,WA(0,i-2),WA(0,i-1),CC(i-1,k,1),CC(i,k,1));</div>
|
|
<div class="line"><a id="l01683" name="l01683"></a><span class="lineno"> 1683</span> MULPM(cr3,ci3,WA(1,i-2),WA(1,i-1),CC(i-1,k,2),CC(i,k,2));</div>
|
|
<div class="line"><a id="l01684" name="l01684"></a><span class="lineno"> 1684</span> MULPM(cr4,ci4,WA(2,i-2),WA(2,i-1),CC(i-1,k,3),CC(i,k,3));</div>
|
|
<div class="line"><a id="l01685" name="l01685"></a><span class="lineno"> 1685</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a>(tr1,tr4,cr4,cr2);</div>
|
|
<div class="line"><a id="l01686" name="l01686"></a><span class="lineno"> 1686</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a>(ti1,ti4,ci2,ci4);</div>
|
|
<div class="line"><a id="l01687" name="l01687"></a><span class="lineno"> 1687</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a>(tr2,tr3,CC(i-1,k,0),cr3);</div>
|
|
<div class="line"><a id="l01688" name="l01688"></a><span class="lineno"> 1688</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a>(ti2,ti3,CC(i ,k,0),ci3);</div>
|
|
<div class="line"><a id="l01689" name="l01689"></a><span class="lineno"> 1689</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a>(CH(i-1,0,k),CH(ic-1,3,k),tr2,tr1);</div>
|
|
<div class="line"><a id="l01690" name="l01690"></a><span class="lineno"> 1690</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a>(CH(i ,0,k),CH(ic ,3,k),ti1,ti2);</div>
|
|
<div class="line"><a id="l01691" name="l01691"></a><span class="lineno"> 1691</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a>(CH(i-1,2,k),CH(ic-1,1,k),tr3,ti4);</div>
|
|
<div class="line"><a id="l01692" name="l01692"></a><span class="lineno"> 1692</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a>(CH(i ,2,k),CH(ic ,1,k),tr4,ti3);</div>
|
|
<div class="line"><a id="l01693" name="l01693"></a><span class="lineno"> 1693</span> }</div>
|
|
<div class="line"><a id="l01694" name="l01694"></a><span class="lineno"> 1694</span> }</div>
|
|
<div class="line"><a id="l01695" name="l01695"></a><span class="lineno"> 1695</span> </div>
|
|
<div class="line"><a id="l01696" name="l01696"></a><span class="lineno"> 1696</span><span class="keyword">template</span><<span class="keyword">typename</span> T> <span class="keywordtype">void</span> radf5(<span class="keywordtype">size_t</span> ido, <span class="keywordtype">size_t</span> l1,</div>
|
|
<div class="line"><a id="l01697" name="l01697"></a><span class="lineno"> 1697</span> <span class="keyword">const</span> T * <a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> cc, T * <a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> ch,</div>
|
|
<div class="line"><a id="l01698" name="l01698"></a><span class="lineno"> 1698</span> <span class="keyword">const</span> T0 * <a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> wa)<span class="keyword"> const</span></div>
|
|
<div class="line"><a id="l01699" name="l01699"></a><span class="lineno"> 1699</span><span class="keyword"> </span>{</div>
|
|
<div class="line"><a id="l01700" name="l01700"></a><span class="lineno"> 1700</span> <span class="keyword">constexpr</span> T0 tr11= T0(0.3090169943749474241022934171828191L),</div>
|
|
<div class="line"><a id="l01701" name="l01701"></a><span class="lineno"> 1701</span> ti11= T0(0.9510565162951535721164393333793821L),</div>
|
|
<div class="line"><a id="l01702" name="l01702"></a><span class="lineno"> 1702</span> tr12= T0(-0.8090169943749474241022934171828191L),</div>
|
|
<div class="line"><a id="l01703" name="l01703"></a><span class="lineno"> 1703</span> ti12= T0(0.5877852522924731291687059546390728L);</div>
|
|
<div class="line"><a id="l01704" name="l01704"></a><span class="lineno"> 1704</span> </div>
|
|
<div class="line"><a id="l01705" name="l01705"></a><span class="lineno"> 1705</span> <span class="keyword">auto</span> WA = [wa,ido](<span class="keywordtype">size_t</span> x, <span class="keywordtype">size_t</span> i) { <span class="keywordflow">return</span> wa[i+x*(ido-1)]; };</div>
|
|
<div class="line"><a id="l01706" name="l01706"></a><span class="lineno"> 1706</span> <span class="keyword">auto</span> CC = [cc,ido,l1](<span class="keywordtype">size_t</span> a, <span class="keywordtype">size_t</span> b, <span class="keywordtype">size_t</span> c) -> <span class="keyword">const</span> T&</div>
|
|
<div class="line"><a id="l01707" name="l01707"></a><span class="lineno"> 1707</span> { <span class="keywordflow">return</span> cc[a+ido*(b+l1*c)]; };</div>
|
|
<div class="line"><a id="l01708" name="l01708"></a><span class="lineno"> 1708</span> <span class="keyword">auto</span> CH = [ch,ido](<span class="keywordtype">size_t</span> a, <span class="keywordtype">size_t</span> b, <span class="keywordtype">size_t</span> c) -> T&</div>
|
|
<div class="line"><a id="l01709" name="l01709"></a><span class="lineno"> 1709</span> { <span class="keywordflow">return</span> ch[a+ido*(b+5*c)]; };</div>
|
|
<div class="line"><a id="l01710" name="l01710"></a><span class="lineno"> 1710</span> </div>
|
|
<div class="line"><a id="l01711" name="l01711"></a><span class="lineno"> 1711</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> k=0; k<l1; k++)</div>
|
|
<div class="line"><a id="l01712" name="l01712"></a><span class="lineno"> 1712</span> {</div>
|
|
<div class="line"><a id="l01713" name="l01713"></a><span class="lineno"> 1713</span> T cr2, cr3, ci4, ci5;</div>
|
|
<div class="line"><a id="l01714" name="l01714"></a><span class="lineno"> 1714</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a> (cr2,ci5,CC(0,k,4),CC(0,k,1));</div>
|
|
<div class="line"><a id="l01715" name="l01715"></a><span class="lineno"> 1715</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a> (cr3,ci4,CC(0,k,3),CC(0,k,2));</div>
|
|
<div class="line"><a id="l01716" name="l01716"></a><span class="lineno"> 1716</span> CH(0,0,k)=CC(0,k,0)+cr2+cr3;</div>
|
|
<div class="line"><a id="l01717" name="l01717"></a><span class="lineno"> 1717</span> CH(ido-1,1,k)=CC(0,k,0)+tr11*cr2+tr12*cr3;</div>
|
|
<div class="line"><a id="l01718" name="l01718"></a><span class="lineno"> 1718</span> CH(0,2,k)=ti11*ci5+ti12*ci4;</div>
|
|
<div class="line"><a id="l01719" name="l01719"></a><span class="lineno"> 1719</span> CH(ido-1,3,k)=CC(0,k,0)+tr12*cr2+tr11*cr3;</div>
|
|
<div class="line"><a id="l01720" name="l01720"></a><span class="lineno"> 1720</span> CH(0,4,k)=ti12*ci5-ti11*ci4;</div>
|
|
<div class="line"><a id="l01721" name="l01721"></a><span class="lineno"> 1721</span> }</div>
|
|
<div class="line"><a id="l01722" name="l01722"></a><span class="lineno"> 1722</span> <span class="keywordflow">if</span> (ido==1) <span class="keywordflow">return</span>;</div>
|
|
<div class="line"><a id="l01723" name="l01723"></a><span class="lineno"> 1723</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> k=0; k<l1;++k)</div>
|
|
<div class="line"><a id="l01724" name="l01724"></a><span class="lineno"> 1724</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i=2, ic=ido-2; i<ido; i+=2, ic-=2)</div>
|
|
<div class="line"><a id="l01725" name="l01725"></a><span class="lineno"> 1725</span> {</div>
|
|
<div class="line"><a id="l01726" name="l01726"></a><span class="lineno"> 1726</span> T di2, di3, di4, di5, dr2, dr3, dr4, dr5;</div>
|
|
<div class="line"><a id="l01727" name="l01727"></a><span class="lineno"> 1727</span> MULPM (dr2,di2,WA(0,i-2),WA(0,i-1),CC(i-1,k,1),CC(i,k,1));</div>
|
|
<div class="line"><a id="l01728" name="l01728"></a><span class="lineno"> 1728</span> MULPM (dr3,di3,WA(1,i-2),WA(1,i-1),CC(i-1,k,2),CC(i,k,2));</div>
|
|
<div class="line"><a id="l01729" name="l01729"></a><span class="lineno"> 1729</span> MULPM (dr4,di4,WA(2,i-2),WA(2,i-1),CC(i-1,k,3),CC(i,k,3));</div>
|
|
<div class="line"><a id="l01730" name="l01730"></a><span class="lineno"> 1730</span> MULPM (dr5,di5,WA(3,i-2),WA(3,i-1),CC(i-1,k,4),CC(i,k,4));</div>
|
|
<div class="line"><a id="l01731" name="l01731"></a><span class="lineno"> 1731</span> <a class="code hl_define" href="pocketfft_8h.html#acffdf2e1ab84f36a7a097e1b8b87a9f9">POCKETFFT_REARRANGE</a>(dr2, di2, dr5, di5);</div>
|
|
<div class="line"><a id="l01732" name="l01732"></a><span class="lineno"> 1732</span> <a class="code hl_define" href="pocketfft_8h.html#acffdf2e1ab84f36a7a097e1b8b87a9f9">POCKETFFT_REARRANGE</a>(dr3, di3, dr4, di4);</div>
|
|
<div class="line"><a id="l01733" name="l01733"></a><span class="lineno"> 1733</span> CH(i-1,0,k)=CC(i-1,k,0)+dr2+dr3;</div>
|
|
<div class="line"><a id="l01734" name="l01734"></a><span class="lineno"> 1734</span> CH(i ,0,k)=CC(i ,k,0)+di2+di3;</div>
|
|
<div class="line"><a id="l01735" name="l01735"></a><span class="lineno"> 1735</span> T tr2=CC(i-1,k,0)+tr11*dr2+tr12*dr3;</div>
|
|
<div class="line"><a id="l01736" name="l01736"></a><span class="lineno"> 1736</span> T ti2=CC(i ,k,0)+tr11*di2+tr12*di3;</div>
|
|
<div class="line"><a id="l01737" name="l01737"></a><span class="lineno"> 1737</span> T tr3=CC(i-1,k,0)+tr12*dr2+tr11*dr3;</div>
|
|
<div class="line"><a id="l01738" name="l01738"></a><span class="lineno"> 1738</span> T ti3=CC(i ,k,0)+tr12*di2+tr11*di3;</div>
|
|
<div class="line"><a id="l01739" name="l01739"></a><span class="lineno"> 1739</span> T tr5 = ti11*dr5 + ti12*dr4;</div>
|
|
<div class="line"><a id="l01740" name="l01740"></a><span class="lineno"> 1740</span> T ti5 = ti11*di5 + ti12*di4;</div>
|
|
<div class="line"><a id="l01741" name="l01741"></a><span class="lineno"> 1741</span> T tr4 = ti12*dr5 - ti11*dr4;</div>
|
|
<div class="line"><a id="l01742" name="l01742"></a><span class="lineno"> 1742</span> T ti4 = ti12*di5 - ti11*di4;</div>
|
|
<div class="line"><a id="l01743" name="l01743"></a><span class="lineno"> 1743</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a>(CH(i-1,2,k),CH(ic-1,1,k),tr2,tr5);</div>
|
|
<div class="line"><a id="l01744" name="l01744"></a><span class="lineno"> 1744</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a>(CH(i ,2,k),CH(ic ,1,k),ti5,ti2);</div>
|
|
<div class="line"><a id="l01745" name="l01745"></a><span class="lineno"> 1745</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a>(CH(i-1,4,k),CH(ic-1,3,k),tr3,tr4);</div>
|
|
<div class="line"><a id="l01746" name="l01746"></a><span class="lineno"> 1746</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a>(CH(i ,4,k),CH(ic ,3,k),ti4,ti3);</div>
|
|
<div class="line"><a id="l01747" name="l01747"></a><span class="lineno"> 1747</span> }</div>
|
|
<div class="line"><a id="l01748" name="l01748"></a><span class="lineno"> 1748</span> }</div>
|
|
<div class="line"><a id="l01749" name="l01749"></a><span class="lineno"> 1749</span> </div>
|
|
<div class="line"><a id="l01750" name="l01750"></a><span class="lineno"> 1750</span><span class="preprocessor">#undef POCKETFFT_REARRANGE</span></div>
|
|
<div class="line"><a id="l01751" name="l01751"></a><span class="lineno"> 1751</span> </div>
|
|
<div class="line"><a id="l01752" name="l01752"></a><span class="lineno"> 1752</span><span class="keyword">template</span><<span class="keyword">typename</span> T> <span class="keywordtype">void</span> radfg(<span class="keywordtype">size_t</span> ido, <span class="keywordtype">size_t</span> ip, <span class="keywordtype">size_t</span> l1,</div>
|
|
<div class="line"><a id="l01753" name="l01753"></a><span class="lineno"> 1753</span> T * <a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> cc, T * <a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> ch,</div>
|
|
<div class="line"><a id="l01754" name="l01754"></a><span class="lineno"> 1754</span> <span class="keyword">const</span> T0 * <a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> wa, <span class="keyword">const</span> T0 * <a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> csarr)<span class="keyword"> const</span></div>
|
|
<div class="line"><a id="l01755" name="l01755"></a><span class="lineno"> 1755</span><span class="keyword"> </span>{</div>
|
|
<div class="line"><a id="l01756" name="l01756"></a><span class="lineno"> 1756</span> <span class="keyword">const</span> <span class="keywordtype">size_t</span> cdim=ip;</div>
|
|
<div class="line"><a id="l01757" name="l01757"></a><span class="lineno"> 1757</span> <span class="keywordtype">size_t</span> ipph=(ip+1)/2;</div>
|
|
<div class="line"><a id="l01758" name="l01758"></a><span class="lineno"> 1758</span> <span class="keywordtype">size_t</span> idl1 = ido*l1;</div>
|
|
<div class="line"><a id="l01759" name="l01759"></a><span class="lineno"> 1759</span> </div>
|
|
<div class="line"><a id="l01760" name="l01760"></a><span class="lineno"> 1760</span> <span class="keyword">auto</span> CC = [cc,ido,cdim](<span class="keywordtype">size_t</span> a, <span class="keywordtype">size_t</span> b, <span class="keywordtype">size_t</span> c) -> T&</div>
|
|
<div class="line"><a id="l01761" name="l01761"></a><span class="lineno"> 1761</span> { <span class="keywordflow">return</span> cc[a+ido*(b+cdim*c)]; };</div>
|
|
<div class="line"><a id="l01762" name="l01762"></a><span class="lineno"> 1762</span> <span class="keyword">auto</span> CH = [ch,ido,l1](<span class="keywordtype">size_t</span> a, <span class="keywordtype">size_t</span> b, <span class="keywordtype">size_t</span> c) -> <span class="keyword">const</span> T&</div>
|
|
<div class="line"><a id="l01763" name="l01763"></a><span class="lineno"> 1763</span> { <span class="keywordflow">return</span> ch[a+ido*(b+l1*c)]; };</div>
|
|
<div class="line"><a id="l01764" name="l01764"></a><span class="lineno"> 1764</span> <span class="keyword">auto</span> C1 = [cc,ido,l1] (<span class="keywordtype">size_t</span> a, <span class="keywordtype">size_t</span> b, <span class="keywordtype">size_t</span> c) -> T&</div>
|
|
<div class="line"><a id="l01765" name="l01765"></a><span class="lineno"> 1765</span> { <span class="keywordflow">return</span> cc[a+ido*(b+l1*c)]; };</div>
|
|
<div class="line"><a id="l01766" name="l01766"></a><span class="lineno"> 1766</span> <span class="keyword">auto</span> C2 = [cc,idl1] (<span class="keywordtype">size_t</span> a, <span class="keywordtype">size_t</span> b) -> T&</div>
|
|
<div class="line"><a id="l01767" name="l01767"></a><span class="lineno"> 1767</span> { <span class="keywordflow">return</span> cc[a+idl1*b]; };</div>
|
|
<div class="line"><a id="l01768" name="l01768"></a><span class="lineno"> 1768</span> <span class="keyword">auto</span> CH2 = [ch,idl1] (<span class="keywordtype">size_t</span> a, <span class="keywordtype">size_t</span> b) -> T&</div>
|
|
<div class="line"><a id="l01769" name="l01769"></a><span class="lineno"> 1769</span> { <span class="keywordflow">return</span> ch[a+idl1*b]; };</div>
|
|
<div class="line"><a id="l01770" name="l01770"></a><span class="lineno"> 1770</span> </div>
|
|
<div class="line"><a id="l01771" name="l01771"></a><span class="lineno"> 1771</span> <span class="keywordflow">if</span> (ido>1)</div>
|
|
<div class="line"><a id="l01772" name="l01772"></a><span class="lineno"> 1772</span> {</div>
|
|
<div class="line"><a id="l01773" name="l01773"></a><span class="lineno"> 1773</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> j=1, jc=ip-1; j<ipph; ++j,--jc) <span class="comment">// 114</span></div>
|
|
<div class="line"><a id="l01774" name="l01774"></a><span class="lineno"> 1774</span> {</div>
|
|
<div class="line"><a id="l01775" name="l01775"></a><span class="lineno"> 1775</span> <span class="keywordtype">size_t</span> is=(j-1)*(ido-1),</div>
|
|
<div class="line"><a id="l01776" name="l01776"></a><span class="lineno"> 1776</span> is2=(jc-1)*(ido-1);</div>
|
|
<div class="line"><a id="l01777" name="l01777"></a><span class="lineno"> 1777</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> k=0; k<l1; ++k) <span class="comment">// 113</span></div>
|
|
<div class="line"><a id="l01778" name="l01778"></a><span class="lineno"> 1778</span> {</div>
|
|
<div class="line"><a id="l01779" name="l01779"></a><span class="lineno"> 1779</span> <span class="keywordtype">size_t</span> idij=is;</div>
|
|
<div class="line"><a id="l01780" name="l01780"></a><span class="lineno"> 1780</span> <span class="keywordtype">size_t</span> idij2=is2;</div>
|
|
<div class="line"><a id="l01781" name="l01781"></a><span class="lineno"> 1781</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i=1; i<=ido-2; i+=2) <span class="comment">// 112</span></div>
|
|
<div class="line"><a id="l01782" name="l01782"></a><span class="lineno"> 1782</span> {</div>
|
|
<div class="line"><a id="l01783" name="l01783"></a><span class="lineno"> 1783</span> T t1=C1(i,k,j ), t2=C1(i+1,k,j ),</div>
|
|
<div class="line"><a id="l01784" name="l01784"></a><span class="lineno"> 1784</span> t3=C1(i,k,jc), t4=C1(i+1,k,jc);</div>
|
|
<div class="line"><a id="l01785" name="l01785"></a><span class="lineno"> 1785</span> T x1=wa[idij]*t1 + wa[idij+1]*t2,</div>
|
|
<div class="line"><a id="l01786" name="l01786"></a><span class="lineno"> 1786</span> x2=wa[idij]*t2 - wa[idij+1]*t1,</div>
|
|
<div class="line"><a id="l01787" name="l01787"></a><span class="lineno"> 1787</span> x3=wa[idij2]*t3 + wa[idij2+1]*t4,</div>
|
|
<div class="line"><a id="l01788" name="l01788"></a><span class="lineno"> 1788</span> x4=wa[idij2]*t4 - wa[idij2+1]*t3;</div>
|
|
<div class="line"><a id="l01789" name="l01789"></a><span class="lineno"> 1789</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a>(C1(i,k,j),C1(i+1,k,jc),x3,x1);</div>
|
|
<div class="line"><a id="l01790" name="l01790"></a><span class="lineno"> 1790</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a>(C1(i+1,k,j),C1(i,k,jc),x2,x4);</div>
|
|
<div class="line"><a id="l01791" name="l01791"></a><span class="lineno"> 1791</span> idij+=2;</div>
|
|
<div class="line"><a id="l01792" name="l01792"></a><span class="lineno"> 1792</span> idij2+=2;</div>
|
|
<div class="line"><a id="l01793" name="l01793"></a><span class="lineno"> 1793</span> }</div>
|
|
<div class="line"><a id="l01794" name="l01794"></a><span class="lineno"> 1794</span> }</div>
|
|
<div class="line"><a id="l01795" name="l01795"></a><span class="lineno"> 1795</span> }</div>
|
|
<div class="line"><a id="l01796" name="l01796"></a><span class="lineno"> 1796</span> }</div>
|
|
<div class="line"><a id="l01797" name="l01797"></a><span class="lineno"> 1797</span> </div>
|
|
<div class="line"><a id="l01798" name="l01798"></a><span class="lineno"> 1798</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> j=1, jc=ip-1; j<ipph; ++j,--jc) <span class="comment">// 123</span></div>
|
|
<div class="line"><a id="l01799" name="l01799"></a><span class="lineno"> 1799</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> k=0; k<l1; ++k) <span class="comment">// 122</span></div>
|
|
<div class="line"><a id="l01800" name="l01800"></a><span class="lineno"> 1800</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#af5eedf3cdfc83c0a30807092c39a9ce2">MPINPLACE</a>(C1(0,k,jc), C1(0,k,j));</div>
|
|
<div class="line"><a id="l01801" name="l01801"></a><span class="lineno"> 1801</span> </div>
|
|
<div class="line"><a id="l01802" name="l01802"></a><span class="lineno"> 1802</span><span class="comment">//everything in C</span></div>
|
|
<div class="line"><a id="l01803" name="l01803"></a><span class="lineno"> 1803</span><span class="comment">//memset(ch,0,ip*l1*ido*sizeof(double));</span></div>
|
|
<div class="line"><a id="l01804" name="l01804"></a><span class="lineno"> 1804</span> </div>
|
|
<div class="line"><a id="l01805" name="l01805"></a><span class="lineno"> 1805</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> l=1,lc=ip-1; l<ipph; ++l,--lc) <span class="comment">// 127</span></div>
|
|
<div class="line"><a id="l01806" name="l01806"></a><span class="lineno"> 1806</span> {</div>
|
|
<div class="line"><a id="l01807" name="l01807"></a><span class="lineno"> 1807</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> ik=0; ik<idl1; ++ik) <span class="comment">// 124</span></div>
|
|
<div class="line"><a id="l01808" name="l01808"></a><span class="lineno"> 1808</span> {</div>
|
|
<div class="line"><a id="l01809" name="l01809"></a><span class="lineno"> 1809</span> CH2(ik,l ) = C2(ik,0)+csarr[2*l]*C2(ik,1)+csarr[4*l]*C2(ik,2);</div>
|
|
<div class="line"><a id="l01810" name="l01810"></a><span class="lineno"> 1810</span> CH2(ik,lc) = csarr[2*l+1]*C2(ik,ip-1)+csarr[4*l+1]*C2(ik,ip-2);</div>
|
|
<div class="line"><a id="l01811" name="l01811"></a><span class="lineno"> 1811</span> }</div>
|
|
<div class="line"><a id="l01812" name="l01812"></a><span class="lineno"> 1812</span> <span class="keywordtype">size_t</span> iang = 2*l;</div>
|
|
<div class="line"><a id="l01813" name="l01813"></a><span class="lineno"> 1813</span> <span class="keywordtype">size_t</span> j=3, jc=ip-3;</div>
|
|
<div class="line"><a id="l01814" name="l01814"></a><span class="lineno"> 1814</span> <span class="keywordflow">for</span> (; j<ipph-3; j+=4,jc-=4) <span class="comment">// 126</span></div>
|
|
<div class="line"><a id="l01815" name="l01815"></a><span class="lineno"> 1815</span> {</div>
|
|
<div class="line"><a id="l01816" name="l01816"></a><span class="lineno"> 1816</span> iang+=l; <span class="keywordflow">if</span> (iang>=ip) iang-=ip;</div>
|
|
<div class="line"><a id="l01817" name="l01817"></a><span class="lineno"> 1817</span> T0 ar1=csarr[2*iang], ai1=csarr[2*iang+1];</div>
|
|
<div class="line"><a id="l01818" name="l01818"></a><span class="lineno"> 1818</span> iang+=l; <span class="keywordflow">if</span> (iang>=ip) iang-=ip;</div>
|
|
<div class="line"><a id="l01819" name="l01819"></a><span class="lineno"> 1819</span> T0 ar2=csarr[2*iang], ai2=csarr[2*iang+1];</div>
|
|
<div class="line"><a id="l01820" name="l01820"></a><span class="lineno"> 1820</span> iang+=l; <span class="keywordflow">if</span> (iang>=ip) iang-=ip;</div>
|
|
<div class="line"><a id="l01821" name="l01821"></a><span class="lineno"> 1821</span> T0 ar3=csarr[2*iang], ai3=csarr[2*iang+1];</div>
|
|
<div class="line"><a id="l01822" name="l01822"></a><span class="lineno"> 1822</span> iang+=l; <span class="keywordflow">if</span> (iang>=ip) iang-=ip;</div>
|
|
<div class="line"><a id="l01823" name="l01823"></a><span class="lineno"> 1823</span> T0 ar4=csarr[2*iang], ai4=csarr[2*iang+1];</div>
|
|
<div class="line"><a id="l01824" name="l01824"></a><span class="lineno"> 1824</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> ik=0; ik<idl1; ++ik) <span class="comment">// 125</span></div>
|
|
<div class="line"><a id="l01825" name="l01825"></a><span class="lineno"> 1825</span> {</div>
|
|
<div class="line"><a id="l01826" name="l01826"></a><span class="lineno"> 1826</span> CH2(ik,l ) += ar1*C2(ik,j )+ar2*C2(ik,j +1)</div>
|
|
<div class="line"><a id="l01827" name="l01827"></a><span class="lineno"> 1827</span> +ar3*C2(ik,j +2)+ar4*C2(ik,j +3);</div>
|
|
<div class="line"><a id="l01828" name="l01828"></a><span class="lineno"> 1828</span> CH2(ik,lc) += ai1*C2(ik,jc)+ai2*C2(ik,jc-1)</div>
|
|
<div class="line"><a id="l01829" name="l01829"></a><span class="lineno"> 1829</span> +ai3*C2(ik,jc-2)+ai4*C2(ik,jc-3);</div>
|
|
<div class="line"><a id="l01830" name="l01830"></a><span class="lineno"> 1830</span> }</div>
|
|
<div class="line"><a id="l01831" name="l01831"></a><span class="lineno"> 1831</span> }</div>
|
|
<div class="line"><a id="l01832" name="l01832"></a><span class="lineno"> 1832</span> <span class="keywordflow">for</span> (; j<ipph-1; j+=2,jc-=2) <span class="comment">// 126</span></div>
|
|
<div class="line"><a id="l01833" name="l01833"></a><span class="lineno"> 1833</span> {</div>
|
|
<div class="line"><a id="l01834" name="l01834"></a><span class="lineno"> 1834</span> iang+=l; <span class="keywordflow">if</span> (iang>=ip) iang-=ip;</div>
|
|
<div class="line"><a id="l01835" name="l01835"></a><span class="lineno"> 1835</span> T0 ar1=csarr[2*iang], ai1=csarr[2*iang+1];</div>
|
|
<div class="line"><a id="l01836" name="l01836"></a><span class="lineno"> 1836</span> iang+=l; <span class="keywordflow">if</span> (iang>=ip) iang-=ip;</div>
|
|
<div class="line"><a id="l01837" name="l01837"></a><span class="lineno"> 1837</span> T0 ar2=csarr[2*iang], ai2=csarr[2*iang+1];</div>
|
|
<div class="line"><a id="l01838" name="l01838"></a><span class="lineno"> 1838</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> ik=0; ik<idl1; ++ik) <span class="comment">// 125</span></div>
|
|
<div class="line"><a id="l01839" name="l01839"></a><span class="lineno"> 1839</span> {</div>
|
|
<div class="line"><a id="l01840" name="l01840"></a><span class="lineno"> 1840</span> CH2(ik,l ) += ar1*C2(ik,j )+ar2*C2(ik,j +1);</div>
|
|
<div class="line"><a id="l01841" name="l01841"></a><span class="lineno"> 1841</span> CH2(ik,lc) += ai1*C2(ik,jc)+ai2*C2(ik,jc-1);</div>
|
|
<div class="line"><a id="l01842" name="l01842"></a><span class="lineno"> 1842</span> }</div>
|
|
<div class="line"><a id="l01843" name="l01843"></a><span class="lineno"> 1843</span> }</div>
|
|
<div class="line"><a id="l01844" name="l01844"></a><span class="lineno"> 1844</span> <span class="keywordflow">for</span> (; j<ipph; ++j,--jc) <span class="comment">// 126</span></div>
|
|
<div class="line"><a id="l01845" name="l01845"></a><span class="lineno"> 1845</span> {</div>
|
|
<div class="line"><a id="l01846" name="l01846"></a><span class="lineno"> 1846</span> iang+=l; <span class="keywordflow">if</span> (iang>=ip) iang-=ip;</div>
|
|
<div class="line"><a id="l01847" name="l01847"></a><span class="lineno"> 1847</span> T0 ar=csarr[2*iang], ai=csarr[2*iang+1];</div>
|
|
<div class="line"><a id="l01848" name="l01848"></a><span class="lineno"> 1848</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> ik=0; ik<idl1; ++ik) <span class="comment">// 125</span></div>
|
|
<div class="line"><a id="l01849" name="l01849"></a><span class="lineno"> 1849</span> {</div>
|
|
<div class="line"><a id="l01850" name="l01850"></a><span class="lineno"> 1850</span> CH2(ik,l ) += ar*C2(ik,j );</div>
|
|
<div class="line"><a id="l01851" name="l01851"></a><span class="lineno"> 1851</span> CH2(ik,lc) += ai*C2(ik,jc);</div>
|
|
<div class="line"><a id="l01852" name="l01852"></a><span class="lineno"> 1852</span> }</div>
|
|
<div class="line"><a id="l01853" name="l01853"></a><span class="lineno"> 1853</span> }</div>
|
|
<div class="line"><a id="l01854" name="l01854"></a><span class="lineno"> 1854</span> }</div>
|
|
<div class="line"><a id="l01855" name="l01855"></a><span class="lineno"> 1855</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> ik=0; ik<idl1; ++ik) <span class="comment">// 101</span></div>
|
|
<div class="line"><a id="l01856" name="l01856"></a><span class="lineno"> 1856</span> CH2(ik,0) = C2(ik,0);</div>
|
|
<div class="line"><a id="l01857" name="l01857"></a><span class="lineno"> 1857</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> j=1; j<ipph; ++j) <span class="comment">// 129</span></div>
|
|
<div class="line"><a id="l01858" name="l01858"></a><span class="lineno"> 1858</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> ik=0; ik<idl1; ++ik) <span class="comment">// 128</span></div>
|
|
<div class="line"><a id="l01859" name="l01859"></a><span class="lineno"> 1859</span> CH2(ik,0) += C2(ik,j);</div>
|
|
<div class="line"><a id="l01860" name="l01860"></a><span class="lineno"> 1860</span> </div>
|
|
<div class="line"><a id="l01861" name="l01861"></a><span class="lineno"> 1861</span><span class="comment">// everything in CH at this point!</span></div>
|
|
<div class="line"><a id="l01862" name="l01862"></a><span class="lineno"> 1862</span><span class="comment">//memset(cc,0,ip*l1*ido*sizeof(double));</span></div>
|
|
<div class="line"><a id="l01863" name="l01863"></a><span class="lineno"> 1863</span> </div>
|
|
<div class="line"><a id="l01864" name="l01864"></a><span class="lineno"> 1864</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> k=0; k<l1; ++k) <span class="comment">// 131</span></div>
|
|
<div class="line"><a id="l01865" name="l01865"></a><span class="lineno"> 1865</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i=0; i<ido; ++i) <span class="comment">// 130</span></div>
|
|
<div class="line"><a id="l01866" name="l01866"></a><span class="lineno"> 1866</span> CC(i,0,k) = CH(i,k,0);</div>
|
|
<div class="line"><a id="l01867" name="l01867"></a><span class="lineno"> 1867</span> </div>
|
|
<div class="line"><a id="l01868" name="l01868"></a><span class="lineno"> 1868</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> j=1, jc=ip-1; j<ipph; ++j,--jc) <span class="comment">// 137</span></div>
|
|
<div class="line"><a id="l01869" name="l01869"></a><span class="lineno"> 1869</span> {</div>
|
|
<div class="line"><a id="l01870" name="l01870"></a><span class="lineno"> 1870</span> <span class="keywordtype">size_t</span> j2=2*j-1;</div>
|
|
<div class="line"><a id="l01871" name="l01871"></a><span class="lineno"> 1871</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> k=0; k<l1; ++k) <span class="comment">// 136</span></div>
|
|
<div class="line"><a id="l01872" name="l01872"></a><span class="lineno"> 1872</span> {</div>
|
|
<div class="line"><a id="l01873" name="l01873"></a><span class="lineno"> 1873</span> CC(ido-1,j2,k) = CH(0,k,j);</div>
|
|
<div class="line"><a id="l01874" name="l01874"></a><span class="lineno"> 1874</span> CC(0,j2+1,k) = CH(0,k,jc);</div>
|
|
<div class="line"><a id="l01875" name="l01875"></a><span class="lineno"> 1875</span> }</div>
|
|
<div class="line"><a id="l01876" name="l01876"></a><span class="lineno"> 1876</span> }</div>
|
|
<div class="line"><a id="l01877" name="l01877"></a><span class="lineno"> 1877</span> </div>
|
|
<div class="line"><a id="l01878" name="l01878"></a><span class="lineno"> 1878</span> <span class="keywordflow">if</span> (ido==1) <span class="keywordflow">return</span>;</div>
|
|
<div class="line"><a id="l01879" name="l01879"></a><span class="lineno"> 1879</span> </div>
|
|
<div class="line"><a id="l01880" name="l01880"></a><span class="lineno"> 1880</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> j=1, jc=ip-1; j<ipph; ++j,--jc) <span class="comment">// 140</span></div>
|
|
<div class="line"><a id="l01881" name="l01881"></a><span class="lineno"> 1881</span> {</div>
|
|
<div class="line"><a id="l01882" name="l01882"></a><span class="lineno"> 1882</span> <span class="keywordtype">size_t</span> j2=2*j-1;</div>
|
|
<div class="line"><a id="l01883" name="l01883"></a><span class="lineno"> 1883</span> <span class="keywordflow">for</span>(<span class="keywordtype">size_t</span> k=0; k<l1; ++k) <span class="comment">// 139</span></div>
|
|
<div class="line"><a id="l01884" name="l01884"></a><span class="lineno"> 1884</span> <span class="keywordflow">for</span>(<span class="keywordtype">size_t</span> i=1, ic=ido-i-2; i<=ido-2; i+=2, ic-=2) <span class="comment">// 138</span></div>
|
|
<div class="line"><a id="l01885" name="l01885"></a><span class="lineno"> 1885</span> {</div>
|
|
<div class="line"><a id="l01886" name="l01886"></a><span class="lineno"> 1886</span> CC(i ,j2+1,k) = CH(i ,k,j )+CH(i ,k,jc);</div>
|
|
<div class="line"><a id="l01887" name="l01887"></a><span class="lineno"> 1887</span> CC(ic ,j2 ,k) = CH(i ,k,j )-CH(i ,k,jc);</div>
|
|
<div class="line"><a id="l01888" name="l01888"></a><span class="lineno"> 1888</span> CC(i+1 ,j2+1,k) = CH(i+1,k,j )+CH(i+1,k,jc);</div>
|
|
<div class="line"><a id="l01889" name="l01889"></a><span class="lineno"> 1889</span> CC(ic+1,j2 ,k) = CH(i+1,k,jc)-CH(i+1,k,j );</div>
|
|
<div class="line"><a id="l01890" name="l01890"></a><span class="lineno"> 1890</span> }</div>
|
|
<div class="line"><a id="l01891" name="l01891"></a><span class="lineno"> 1891</span> }</div>
|
|
<div class="line"><a id="l01892" name="l01892"></a><span class="lineno"> 1892</span> }</div>
|
|
<div class="line"><a id="l01893" name="l01893"></a><span class="lineno"> 1893</span> </div>
|
|
<div class="line"><a id="l01894" name="l01894"></a><span class="lineno"> 1894</span><span class="keyword">template</span><<span class="keyword">typename</span> T> <span class="keywordtype">void</span> radb2(<span class="keywordtype">size_t</span> ido, <span class="keywordtype">size_t</span> l1,</div>
|
|
<div class="line"><a id="l01895" name="l01895"></a><span class="lineno"> 1895</span> <span class="keyword">const</span> T * <a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> cc, T * <a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> ch,</div>
|
|
<div class="line"><a id="l01896" name="l01896"></a><span class="lineno"> 1896</span> <span class="keyword">const</span> T0 * <a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> wa)<span class="keyword"> const</span></div>
|
|
<div class="line"><a id="l01897" name="l01897"></a><span class="lineno"> 1897</span><span class="keyword"> </span>{</div>
|
|
<div class="line"><a id="l01898" name="l01898"></a><span class="lineno"> 1898</span> <span class="keyword">auto</span> WA = [wa,ido](<span class="keywordtype">size_t</span> x, <span class="keywordtype">size_t</span> i) { <span class="keywordflow">return</span> wa[i+x*(ido-1)]; };</div>
|
|
<div class="line"><a id="l01899" name="l01899"></a><span class="lineno"> 1899</span> <span class="keyword">auto</span> CC = [cc,ido](<span class="keywordtype">size_t</span> a, <span class="keywordtype">size_t</span> b, <span class="keywordtype">size_t</span> c) -> <span class="keyword">const</span> T&</div>
|
|
<div class="line"><a id="l01900" name="l01900"></a><span class="lineno"> 1900</span> { <span class="keywordflow">return</span> cc[a+ido*(b+2*c)]; };</div>
|
|
<div class="line"><a id="l01901" name="l01901"></a><span class="lineno"> 1901</span> <span class="keyword">auto</span> CH = [ch,ido,l1](<span class="keywordtype">size_t</span> a, <span class="keywordtype">size_t</span> b, <span class="keywordtype">size_t</span> c) -> T&</div>
|
|
<div class="line"><a id="l01902" name="l01902"></a><span class="lineno"> 1902</span> { <span class="keywordflow">return</span> ch[a+ido*(b+l1*c)]; };</div>
|
|
<div class="line"><a id="l01903" name="l01903"></a><span class="lineno"> 1903</span> </div>
|
|
<div class="line"><a id="l01904" name="l01904"></a><span class="lineno"> 1904</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> k=0; k<l1; k++)</div>
|
|
<div class="line"><a id="l01905" name="l01905"></a><span class="lineno"> 1905</span> PM (CH(0,k,0),CH(0,k,1),CC(0,0,k),CC(ido-1,1,k));</div>
|
|
<div class="line"><a id="l01906" name="l01906"></a><span class="lineno"> 1906</span> <span class="keywordflow">if</span> ((ido&1)==0)</div>
|
|
<div class="line"><a id="l01907" name="l01907"></a><span class="lineno"> 1907</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> k=0; k<l1; k++)</div>
|
|
<div class="line"><a id="l01908" name="l01908"></a><span class="lineno"> 1908</span> {</div>
|
|
<div class="line"><a id="l01909" name="l01909"></a><span class="lineno"> 1909</span> CH(ido-1,k,0) = 2*CC(ido-1,0,k);</div>
|
|
<div class="line"><a id="l01910" name="l01910"></a><span class="lineno"> 1910</span> CH(ido-1,k,1) =-2*CC(0 ,1,k);</div>
|
|
<div class="line"><a id="l01911" name="l01911"></a><span class="lineno"> 1911</span> }</div>
|
|
<div class="line"><a id="l01912" name="l01912"></a><span class="lineno"> 1912</span> <span class="keywordflow">if</span> (ido<=2) <span class="keywordflow">return</span>;</div>
|
|
<div class="line"><a id="l01913" name="l01913"></a><span class="lineno"> 1913</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> k=0; k<l1;++k)</div>
|
|
<div class="line"><a id="l01914" name="l01914"></a><span class="lineno"> 1914</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i=2; i<ido; i+=2)</div>
|
|
<div class="line"><a id="l01915" name="l01915"></a><span class="lineno"> 1915</span> {</div>
|
|
<div class="line"><a id="l01916" name="l01916"></a><span class="lineno"> 1916</span> <span class="keywordtype">size_t</span> ic=ido-i;</div>
|
|
<div class="line"><a id="l01917" name="l01917"></a><span class="lineno"> 1917</span> T ti2, tr2;</div>
|
|
<div class="line"><a id="l01918" name="l01918"></a><span class="lineno"> 1918</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a> (CH(i-1,k,0),tr2,CC(i-1,0,k),CC(ic-1,1,k));</div>
|
|
<div class="line"><a id="l01919" name="l01919"></a><span class="lineno"> 1919</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a> (ti2,CH(i ,k,0),CC(i ,0,k),CC(ic ,1,k));</div>
|
|
<div class="line"><a id="l01920" name="l01920"></a><span class="lineno"> 1920</span> MULPM (CH(i,k,1),CH(i-1,k,1),WA(0,i-2),WA(0,i-1),ti2,tr2);</div>
|
|
<div class="line"><a id="l01921" name="l01921"></a><span class="lineno"> 1921</span> }</div>
|
|
<div class="line"><a id="l01922" name="l01922"></a><span class="lineno"> 1922</span> }</div>
|
|
<div class="line"><a id="l01923" name="l01923"></a><span class="lineno"> 1923</span> </div>
|
|
<div class="line"><a id="l01924" name="l01924"></a><span class="lineno"> 1924</span><span class="keyword">template</span><<span class="keyword">typename</span> T> <span class="keywordtype">void</span> radb3(<span class="keywordtype">size_t</span> ido, <span class="keywordtype">size_t</span> l1,</div>
|
|
<div class="line"><a id="l01925" name="l01925"></a><span class="lineno"> 1925</span> <span class="keyword">const</span> T * <a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> cc, T * <a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> ch,</div>
|
|
<div class="line"><a id="l01926" name="l01926"></a><span class="lineno"> 1926</span> <span class="keyword">const</span> T0 * <a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> wa)<span class="keyword"> const</span></div>
|
|
<div class="line"><a id="l01927" name="l01927"></a><span class="lineno"> 1927</span><span class="keyword"> </span>{</div>
|
|
<div class="line"><a id="l01928" name="l01928"></a><span class="lineno"> 1928</span> <span class="keyword">constexpr</span> T0 taur=-0.5, taui=T0(0.8660254037844386467637231707529362L);</div>
|
|
<div class="line"><a id="l01929" name="l01929"></a><span class="lineno"> 1929</span> </div>
|
|
<div class="line"><a id="l01930" name="l01930"></a><span class="lineno"> 1930</span> <span class="keyword">auto</span> WA = [wa,ido](<span class="keywordtype">size_t</span> x, <span class="keywordtype">size_t</span> i) { <span class="keywordflow">return</span> wa[i+x*(ido-1)]; };</div>
|
|
<div class="line"><a id="l01931" name="l01931"></a><span class="lineno"> 1931</span> <span class="keyword">auto</span> CC = [cc,ido](<span class="keywordtype">size_t</span> a, <span class="keywordtype">size_t</span> b, <span class="keywordtype">size_t</span> c) -> <span class="keyword">const</span> T&</div>
|
|
<div class="line"><a id="l01932" name="l01932"></a><span class="lineno"> 1932</span> { <span class="keywordflow">return</span> cc[a+ido*(b+3*c)]; };</div>
|
|
<div class="line"><a id="l01933" name="l01933"></a><span class="lineno"> 1933</span> <span class="keyword">auto</span> CH = [ch,ido,l1](<span class="keywordtype">size_t</span> a, <span class="keywordtype">size_t</span> b, <span class="keywordtype">size_t</span> c) -> T&</div>
|
|
<div class="line"><a id="l01934" name="l01934"></a><span class="lineno"> 1934</span> { <span class="keywordflow">return</span> ch[a+ido*(b+l1*c)]; };</div>
|
|
<div class="line"><a id="l01935" name="l01935"></a><span class="lineno"> 1935</span> </div>
|
|
<div class="line"><a id="l01936" name="l01936"></a><span class="lineno"> 1936</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> k=0; k<l1; k++)</div>
|
|
<div class="line"><a id="l01937" name="l01937"></a><span class="lineno"> 1937</span> {</div>
|
|
<div class="line"><a id="l01938" name="l01938"></a><span class="lineno"> 1938</span> T tr2=2*CC(ido-1,1,k);</div>
|
|
<div class="line"><a id="l01939" name="l01939"></a><span class="lineno"> 1939</span> T cr2=CC(0,0,k)+taur*tr2;</div>
|
|
<div class="line"><a id="l01940" name="l01940"></a><span class="lineno"> 1940</span> CH(0,k,0)=CC(0,0,k)+tr2;</div>
|
|
<div class="line"><a id="l01941" name="l01941"></a><span class="lineno"> 1941</span> T ci3=2*taui*CC(0,2,k);</div>
|
|
<div class="line"><a id="l01942" name="l01942"></a><span class="lineno"> 1942</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a> (CH(0,k,2),CH(0,k,1),cr2,ci3);</div>
|
|
<div class="line"><a id="l01943" name="l01943"></a><span class="lineno"> 1943</span> }</div>
|
|
<div class="line"><a id="l01944" name="l01944"></a><span class="lineno"> 1944</span> <span class="keywordflow">if</span> (ido==1) <span class="keywordflow">return</span>;</div>
|
|
<div class="line"><a id="l01945" name="l01945"></a><span class="lineno"> 1945</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> k=0; k<l1; k++)</div>
|
|
<div class="line"><a id="l01946" name="l01946"></a><span class="lineno"> 1946</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i=2, ic=ido-2; i<ido; i+=2, ic-=2)</div>
|
|
<div class="line"><a id="l01947" name="l01947"></a><span class="lineno"> 1947</span> {</div>
|
|
<div class="line"><a id="l01948" name="l01948"></a><span class="lineno"> 1948</span> T tr2=CC(i-1,2,k)+CC(ic-1,1,k); <span class="comment">// t2=CC(I) + conj(CC(ic))</span></div>
|
|
<div class="line"><a id="l01949" name="l01949"></a><span class="lineno"> 1949</span> T ti2=CC(i ,2,k)-CC(ic ,1,k);</div>
|
|
<div class="line"><a id="l01950" name="l01950"></a><span class="lineno"> 1950</span> T cr2=CC(i-1,0,k)+taur*tr2; <span class="comment">// c2=CC +taur*t2</span></div>
|
|
<div class="line"><a id="l01951" name="l01951"></a><span class="lineno"> 1951</span> T ci2=CC(i ,0,k)+taur*ti2;</div>
|
|
<div class="line"><a id="l01952" name="l01952"></a><span class="lineno"> 1952</span> CH(i-1,k,0)=CC(i-1,0,k)+tr2; <span class="comment">// CH=CC+t2</span></div>
|
|
<div class="line"><a id="l01953" name="l01953"></a><span class="lineno"> 1953</span> CH(i ,k,0)=CC(i ,0,k)+ti2;</div>
|
|
<div class="line"><a id="l01954" name="l01954"></a><span class="lineno"> 1954</span> T cr3=taui*(CC(i-1,2,k)-CC(ic-1,1,k));<span class="comment">// c3=taui*(CC(i)-conj(CC(ic)))</span></div>
|
|
<div class="line"><a id="l01955" name="l01955"></a><span class="lineno"> 1955</span> T ci3=taui*(CC(i ,2,k)+CC(ic ,1,k));</div>
|
|
<div class="line"><a id="l01956" name="l01956"></a><span class="lineno"> 1956</span> T di2, di3, dr2, dr3;</div>
|
|
<div class="line"><a id="l01957" name="l01957"></a><span class="lineno"> 1957</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a>(dr3,dr2,cr2,ci3); <span class="comment">// d2= (cr2-ci3, ci2+cr3) = c2+i*c3</span></div>
|
|
<div class="line"><a id="l01958" name="l01958"></a><span class="lineno"> 1958</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a>(di2,di3,ci2,cr3); <span class="comment">// d3= (cr2+ci3, ci2-cr3) = c2-i*c3</span></div>
|
|
<div class="line"><a id="l01959" name="l01959"></a><span class="lineno"> 1959</span> MULPM(CH(i,k,1),CH(i-1,k,1),WA(0,i-2),WA(0,i-1),di2,dr2); <span class="comment">// ch = WA*d2</span></div>
|
|
<div class="line"><a id="l01960" name="l01960"></a><span class="lineno"> 1960</span> MULPM(CH(i,k,2),CH(i-1,k,2),WA(1,i-2),WA(1,i-1),di3,dr3);</div>
|
|
<div class="line"><a id="l01961" name="l01961"></a><span class="lineno"> 1961</span> }</div>
|
|
<div class="line"><a id="l01962" name="l01962"></a><span class="lineno"> 1962</span> }</div>
|
|
<div class="line"><a id="l01963" name="l01963"></a><span class="lineno"> 1963</span> </div>
|
|
<div class="line"><a id="l01964" name="l01964"></a><span class="lineno"> 1964</span><span class="keyword">template</span><<span class="keyword">typename</span> T> <span class="keywordtype">void</span> radb4(<span class="keywordtype">size_t</span> ido, <span class="keywordtype">size_t</span> l1,</div>
|
|
<div class="line"><a id="l01965" name="l01965"></a><span class="lineno"> 1965</span> <span class="keyword">const</span> T * <a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> cc, T * <a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> ch,</div>
|
|
<div class="line"><a id="l01966" name="l01966"></a><span class="lineno"> 1966</span> <span class="keyword">const</span> T0 * <a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> wa)<span class="keyword"> const</span></div>
|
|
<div class="line"><a id="l01967" name="l01967"></a><span class="lineno"> 1967</span><span class="keyword"> </span>{</div>
|
|
<div class="line"><a id="l01968" name="l01968"></a><span class="lineno"> 1968</span> <span class="keyword">constexpr</span> T0 sqrt2=T0(1.414213562373095048801688724209698L);</div>
|
|
<div class="line"><a id="l01969" name="l01969"></a><span class="lineno"> 1969</span> </div>
|
|
<div class="line"><a id="l01970" name="l01970"></a><span class="lineno"> 1970</span> <span class="keyword">auto</span> WA = [wa,ido](<span class="keywordtype">size_t</span> x, <span class="keywordtype">size_t</span> i) { <span class="keywordflow">return</span> wa[i+x*(ido-1)]; };</div>
|
|
<div class="line"><a id="l01971" name="l01971"></a><span class="lineno"> 1971</span> <span class="keyword">auto</span> CC = [cc,ido](<span class="keywordtype">size_t</span> a, <span class="keywordtype">size_t</span> b, <span class="keywordtype">size_t</span> c) -> <span class="keyword">const</span> T&</div>
|
|
<div class="line"><a id="l01972" name="l01972"></a><span class="lineno"> 1972</span> { <span class="keywordflow">return</span> cc[a+ido*(b+4*c)]; };</div>
|
|
<div class="line"><a id="l01973" name="l01973"></a><span class="lineno"> 1973</span> <span class="keyword">auto</span> CH = [ch,ido,l1](<span class="keywordtype">size_t</span> a, <span class="keywordtype">size_t</span> b, <span class="keywordtype">size_t</span> c) -> T&</div>
|
|
<div class="line"><a id="l01974" name="l01974"></a><span class="lineno"> 1974</span> { <span class="keywordflow">return</span> ch[a+ido*(b+l1*c)]; };</div>
|
|
<div class="line"><a id="l01975" name="l01975"></a><span class="lineno"> 1975</span> </div>
|
|
<div class="line"><a id="l01976" name="l01976"></a><span class="lineno"> 1976</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> k=0; k<l1; k++)</div>
|
|
<div class="line"><a id="l01977" name="l01977"></a><span class="lineno"> 1977</span> {</div>
|
|
<div class="line"><a id="l01978" name="l01978"></a><span class="lineno"> 1978</span> T tr1, tr2;</div>
|
|
<div class="line"><a id="l01979" name="l01979"></a><span class="lineno"> 1979</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a> (tr2,tr1,CC(0,0,k),CC(ido-1,3,k));</div>
|
|
<div class="line"><a id="l01980" name="l01980"></a><span class="lineno"> 1980</span> T tr3=2*CC(ido-1,1,k);</div>
|
|
<div class="line"><a id="l01981" name="l01981"></a><span class="lineno"> 1981</span> T tr4=2*CC(0,2,k);</div>
|
|
<div class="line"><a id="l01982" name="l01982"></a><span class="lineno"> 1982</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a> (CH(0,k,0),CH(0,k,2),tr2,tr3);</div>
|
|
<div class="line"><a id="l01983" name="l01983"></a><span class="lineno"> 1983</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a> (CH(0,k,3),CH(0,k,1),tr1,tr4);</div>
|
|
<div class="line"><a id="l01984" name="l01984"></a><span class="lineno"> 1984</span> }</div>
|
|
<div class="line"><a id="l01985" name="l01985"></a><span class="lineno"> 1985</span> <span class="keywordflow">if</span> ((ido&1)==0)</div>
|
|
<div class="line"><a id="l01986" name="l01986"></a><span class="lineno"> 1986</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> k=0; k<l1; k++)</div>
|
|
<div class="line"><a id="l01987" name="l01987"></a><span class="lineno"> 1987</span> {</div>
|
|
<div class="line"><a id="l01988" name="l01988"></a><span class="lineno"> 1988</span> T tr1,tr2,ti1,ti2;</div>
|
|
<div class="line"><a id="l01989" name="l01989"></a><span class="lineno"> 1989</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a> (ti1,ti2,CC(0 ,3,k),CC(0 ,1,k));</div>
|
|
<div class="line"><a id="l01990" name="l01990"></a><span class="lineno"> 1990</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a> (tr2,tr1,CC(ido-1,0,k),CC(ido-1,2,k));</div>
|
|
<div class="line"><a id="l01991" name="l01991"></a><span class="lineno"> 1991</span> CH(ido-1,k,0)=tr2+tr2;</div>
|
|
<div class="line"><a id="l01992" name="l01992"></a><span class="lineno"> 1992</span> CH(ido-1,k,1)=sqrt2*(tr1-ti1);</div>
|
|
<div class="line"><a id="l01993" name="l01993"></a><span class="lineno"> 1993</span> CH(ido-1,k,2)=ti2+ti2;</div>
|
|
<div class="line"><a id="l01994" name="l01994"></a><span class="lineno"> 1994</span> CH(ido-1,k,3)=-sqrt2*(tr1+ti1);</div>
|
|
<div class="line"><a id="l01995" name="l01995"></a><span class="lineno"> 1995</span> }</div>
|
|
<div class="line"><a id="l01996" name="l01996"></a><span class="lineno"> 1996</span> <span class="keywordflow">if</span> (ido<=2) <span class="keywordflow">return</span>;</div>
|
|
<div class="line"><a id="l01997" name="l01997"></a><span class="lineno"> 1997</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> k=0; k<l1;++k)</div>
|
|
<div class="line"><a id="l01998" name="l01998"></a><span class="lineno"> 1998</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i=2; i<ido; i+=2)</div>
|
|
<div class="line"><a id="l01999" name="l01999"></a><span class="lineno"> 1999</span> {</div>
|
|
<div class="line"><a id="l02000" name="l02000"></a><span class="lineno"> 2000</span> T ci2, ci3, ci4, cr2, cr3, cr4, ti1, ti2, ti3, ti4, tr1, tr2, tr3, tr4;</div>
|
|
<div class="line"><a id="l02001" name="l02001"></a><span class="lineno"> 2001</span> <span class="keywordtype">size_t</span> ic=ido-i;</div>
|
|
<div class="line"><a id="l02002" name="l02002"></a><span class="lineno"> 2002</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a> (tr2,tr1,CC(i-1,0,k),CC(ic-1,3,k));</div>
|
|
<div class="line"><a id="l02003" name="l02003"></a><span class="lineno"> 2003</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a> (ti1,ti2,CC(i ,0,k),CC(ic ,3,k));</div>
|
|
<div class="line"><a id="l02004" name="l02004"></a><span class="lineno"> 2004</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a> (tr4,ti3,CC(i ,2,k),CC(ic ,1,k));</div>
|
|
<div class="line"><a id="l02005" name="l02005"></a><span class="lineno"> 2005</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a> (tr3,ti4,CC(i-1,2,k),CC(ic-1,1,k));</div>
|
|
<div class="line"><a id="l02006" name="l02006"></a><span class="lineno"> 2006</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a> (CH(i-1,k,0),cr3,tr2,tr3);</div>
|
|
<div class="line"><a id="l02007" name="l02007"></a><span class="lineno"> 2007</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a> (CH(i ,k,0),ci3,ti2,ti3);</div>
|
|
<div class="line"><a id="l02008" name="l02008"></a><span class="lineno"> 2008</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a> (cr4,cr2,tr1,tr4);</div>
|
|
<div class="line"><a id="l02009" name="l02009"></a><span class="lineno"> 2009</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a> (ci2,ci4,ti1,ti4);</div>
|
|
<div class="line"><a id="l02010" name="l02010"></a><span class="lineno"> 2010</span> MULPM (CH(i,k,1),CH(i-1,k,1),WA(0,i-2),WA(0,i-1),ci2,cr2);</div>
|
|
<div class="line"><a id="l02011" name="l02011"></a><span class="lineno"> 2011</span> MULPM (CH(i,k,2),CH(i-1,k,2),WA(1,i-2),WA(1,i-1),ci3,cr3);</div>
|
|
<div class="line"><a id="l02012" name="l02012"></a><span class="lineno"> 2012</span> MULPM (CH(i,k,3),CH(i-1,k,3),WA(2,i-2),WA(2,i-1),ci4,cr4);</div>
|
|
<div class="line"><a id="l02013" name="l02013"></a><span class="lineno"> 2013</span> }</div>
|
|
<div class="line"><a id="l02014" name="l02014"></a><span class="lineno"> 2014</span> }</div>
|
|
<div class="line"><a id="l02015" name="l02015"></a><span class="lineno"> 2015</span> </div>
|
|
<div class="line"><a id="l02016" name="l02016"></a><span class="lineno"> 2016</span><span class="keyword">template</span><<span class="keyword">typename</span> T> <span class="keywordtype">void</span> radb5(<span class="keywordtype">size_t</span> ido, <span class="keywordtype">size_t</span> l1,</div>
|
|
<div class="line"><a id="l02017" name="l02017"></a><span class="lineno"> 2017</span> <span class="keyword">const</span> T * <a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> cc, T * <a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> ch,</div>
|
|
<div class="line"><a id="l02018" name="l02018"></a><span class="lineno"> 2018</span> <span class="keyword">const</span> T0 * <a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> wa)<span class="keyword"> const</span></div>
|
|
<div class="line"><a id="l02019" name="l02019"></a><span class="lineno"> 2019</span><span class="keyword"> </span>{</div>
|
|
<div class="line"><a id="l02020" name="l02020"></a><span class="lineno"> 2020</span> <span class="keyword">constexpr</span> T0 tr11= T0(0.3090169943749474241022934171828191L),</div>
|
|
<div class="line"><a id="l02021" name="l02021"></a><span class="lineno"> 2021</span> ti11= T0(0.9510565162951535721164393333793821L),</div>
|
|
<div class="line"><a id="l02022" name="l02022"></a><span class="lineno"> 2022</span> tr12= T0(-0.8090169943749474241022934171828191L),</div>
|
|
<div class="line"><a id="l02023" name="l02023"></a><span class="lineno"> 2023</span> ti12= T0(0.5877852522924731291687059546390728L);</div>
|
|
<div class="line"><a id="l02024" name="l02024"></a><span class="lineno"> 2024</span> </div>
|
|
<div class="line"><a id="l02025" name="l02025"></a><span class="lineno"> 2025</span> <span class="keyword">auto</span> WA = [wa,ido](<span class="keywordtype">size_t</span> x, <span class="keywordtype">size_t</span> i) { <span class="keywordflow">return</span> wa[i+x*(ido-1)]; };</div>
|
|
<div class="line"><a id="l02026" name="l02026"></a><span class="lineno"> 2026</span> <span class="keyword">auto</span> CC = [cc,ido](<span class="keywordtype">size_t</span> a, <span class="keywordtype">size_t</span> b, <span class="keywordtype">size_t</span> c) -> <span class="keyword">const</span> T&</div>
|
|
<div class="line"><a id="l02027" name="l02027"></a><span class="lineno"> 2027</span> { <span class="keywordflow">return</span> cc[a+ido*(b+5*c)]; };</div>
|
|
<div class="line"><a id="l02028" name="l02028"></a><span class="lineno"> 2028</span> <span class="keyword">auto</span> CH = [ch,ido,l1](<span class="keywordtype">size_t</span> a, <span class="keywordtype">size_t</span> b, <span class="keywordtype">size_t</span> c) -> T&</div>
|
|
<div class="line"><a id="l02029" name="l02029"></a><span class="lineno"> 2029</span> { <span class="keywordflow">return</span> ch[a+ido*(b+l1*c)]; };</div>
|
|
<div class="line"><a id="l02030" name="l02030"></a><span class="lineno"> 2030</span> </div>
|
|
<div class="line"><a id="l02031" name="l02031"></a><span class="lineno"> 2031</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> k=0; k<l1; k++)</div>
|
|
<div class="line"><a id="l02032" name="l02032"></a><span class="lineno"> 2032</span> {</div>
|
|
<div class="line"><a id="l02033" name="l02033"></a><span class="lineno"> 2033</span> T ti5=CC(0,2,k)+CC(0,2,k);</div>
|
|
<div class="line"><a id="l02034" name="l02034"></a><span class="lineno"> 2034</span> T ti4=CC(0,4,k)+CC(0,4,k);</div>
|
|
<div class="line"><a id="l02035" name="l02035"></a><span class="lineno"> 2035</span> T tr2=CC(ido-1,1,k)+CC(ido-1,1,k);</div>
|
|
<div class="line"><a id="l02036" name="l02036"></a><span class="lineno"> 2036</span> T tr3=CC(ido-1,3,k)+CC(ido-1,3,k);</div>
|
|
<div class="line"><a id="l02037" name="l02037"></a><span class="lineno"> 2037</span> CH(0,k,0)=CC(0,0,k)+tr2+tr3;</div>
|
|
<div class="line"><a id="l02038" name="l02038"></a><span class="lineno"> 2038</span> T cr2=CC(0,0,k)+tr11*tr2+tr12*tr3;</div>
|
|
<div class="line"><a id="l02039" name="l02039"></a><span class="lineno"> 2039</span> T cr3=CC(0,0,k)+tr12*tr2+tr11*tr3;</div>
|
|
<div class="line"><a id="l02040" name="l02040"></a><span class="lineno"> 2040</span> T ci4, ci5;</div>
|
|
<div class="line"><a id="l02041" name="l02041"></a><span class="lineno"> 2041</span> MULPM(ci5,ci4,ti5,ti4,ti11,ti12);</div>
|
|
<div class="line"><a id="l02042" name="l02042"></a><span class="lineno"> 2042</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a>(CH(0,k,4),CH(0,k,1),cr2,ci5);</div>
|
|
<div class="line"><a id="l02043" name="l02043"></a><span class="lineno"> 2043</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a>(CH(0,k,3),CH(0,k,2),cr3,ci4);</div>
|
|
<div class="line"><a id="l02044" name="l02044"></a><span class="lineno"> 2044</span> }</div>
|
|
<div class="line"><a id="l02045" name="l02045"></a><span class="lineno"> 2045</span> <span class="keywordflow">if</span> (ido==1) <span class="keywordflow">return</span>;</div>
|
|
<div class="line"><a id="l02046" name="l02046"></a><span class="lineno"> 2046</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> k=0; k<l1;++k)</div>
|
|
<div class="line"><a id="l02047" name="l02047"></a><span class="lineno"> 2047</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i=2, ic=ido-2; i<ido; i+=2, ic-=2)</div>
|
|
<div class="line"><a id="l02048" name="l02048"></a><span class="lineno"> 2048</span> {</div>
|
|
<div class="line"><a id="l02049" name="l02049"></a><span class="lineno"> 2049</span> T tr2, tr3, tr4, tr5, ti2, ti3, ti4, ti5;</div>
|
|
<div class="line"><a id="l02050" name="l02050"></a><span class="lineno"> 2050</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a>(tr2,tr5,CC(i-1,2,k),CC(ic-1,1,k));</div>
|
|
<div class="line"><a id="l02051" name="l02051"></a><span class="lineno"> 2051</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a>(ti5,ti2,CC(i ,2,k),CC(ic ,1,k));</div>
|
|
<div class="line"><a id="l02052" name="l02052"></a><span class="lineno"> 2052</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a>(tr3,tr4,CC(i-1,4,k),CC(ic-1,3,k));</div>
|
|
<div class="line"><a id="l02053" name="l02053"></a><span class="lineno"> 2053</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a>(ti4,ti3,CC(i ,4,k),CC(ic ,3,k));</div>
|
|
<div class="line"><a id="l02054" name="l02054"></a><span class="lineno"> 2054</span> CH(i-1,k,0)=CC(i-1,0,k)+tr2+tr3;</div>
|
|
<div class="line"><a id="l02055" name="l02055"></a><span class="lineno"> 2055</span> CH(i ,k,0)=CC(i ,0,k)+ti2+ti3;</div>
|
|
<div class="line"><a id="l02056" name="l02056"></a><span class="lineno"> 2056</span> T cr2=CC(i-1,0,k)+tr11*tr2+tr12*tr3;</div>
|
|
<div class="line"><a id="l02057" name="l02057"></a><span class="lineno"> 2057</span> T ci2=CC(i ,0,k)+tr11*ti2+tr12*ti3;</div>
|
|
<div class="line"><a id="l02058" name="l02058"></a><span class="lineno"> 2058</span> T cr3=CC(i-1,0,k)+tr12*tr2+tr11*tr3;</div>
|
|
<div class="line"><a id="l02059" name="l02059"></a><span class="lineno"> 2059</span> T ci3=CC(i ,0,k)+tr12*ti2+tr11*ti3;</div>
|
|
<div class="line"><a id="l02060" name="l02060"></a><span class="lineno"> 2060</span> T ci4, ci5, cr5, cr4;</div>
|
|
<div class="line"><a id="l02061" name="l02061"></a><span class="lineno"> 2061</span> MULPM(cr5,cr4,tr5,tr4,ti11,ti12);</div>
|
|
<div class="line"><a id="l02062" name="l02062"></a><span class="lineno"> 2062</span> MULPM(ci5,ci4,ti5,ti4,ti11,ti12);</div>
|
|
<div class="line"><a id="l02063" name="l02063"></a><span class="lineno"> 2063</span> T dr2, dr3, dr4, dr5, di2, di3, di4, di5;</div>
|
|
<div class="line"><a id="l02064" name="l02064"></a><span class="lineno"> 2064</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a>(dr4,dr3,cr3,ci4);</div>
|
|
<div class="line"><a id="l02065" name="l02065"></a><span class="lineno"> 2065</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a>(di3,di4,ci3,cr4);</div>
|
|
<div class="line"><a id="l02066" name="l02066"></a><span class="lineno"> 2066</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a>(dr5,dr2,cr2,ci5);</div>
|
|
<div class="line"><a id="l02067" name="l02067"></a><span class="lineno"> 2067</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a>(di2,di5,ci2,cr5);</div>
|
|
<div class="line"><a id="l02068" name="l02068"></a><span class="lineno"> 2068</span> MULPM(CH(i,k,1),CH(i-1,k,1),WA(0,i-2),WA(0,i-1),di2,dr2);</div>
|
|
<div class="line"><a id="l02069" name="l02069"></a><span class="lineno"> 2069</span> MULPM(CH(i,k,2),CH(i-1,k,2),WA(1,i-2),WA(1,i-1),di3,dr3);</div>
|
|
<div class="line"><a id="l02070" name="l02070"></a><span class="lineno"> 2070</span> MULPM(CH(i,k,3),CH(i-1,k,3),WA(2,i-2),WA(2,i-1),di4,dr4);</div>
|
|
<div class="line"><a id="l02071" name="l02071"></a><span class="lineno"> 2071</span> MULPM(CH(i,k,4),CH(i-1,k,4),WA(3,i-2),WA(3,i-1),di5,dr5);</div>
|
|
<div class="line"><a id="l02072" name="l02072"></a><span class="lineno"> 2072</span> }</div>
|
|
<div class="line"><a id="l02073" name="l02073"></a><span class="lineno"> 2073</span> }</div>
|
|
<div class="line"><a id="l02074" name="l02074"></a><span class="lineno"> 2074</span> </div>
|
|
<div class="line"><a id="l02075" name="l02075"></a><span class="lineno"> 2075</span><span class="keyword">template</span><<span class="keyword">typename</span> T> <span class="keywordtype">void</span> radbg(<span class="keywordtype">size_t</span> ido, <span class="keywordtype">size_t</span> ip, <span class="keywordtype">size_t</span> l1,</div>
|
|
<div class="line"><a id="l02076" name="l02076"></a><span class="lineno"> 2076</span> T * <a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> cc, T * <a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> ch,</div>
|
|
<div class="line"><a id="l02077" name="l02077"></a><span class="lineno"> 2077</span> <span class="keyword">const</span> T0 * <a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> wa, <span class="keyword">const</span> T0 * <a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> csarr)<span class="keyword"> const</span></div>
|
|
<div class="line"><a id="l02078" name="l02078"></a><span class="lineno"> 2078</span><span class="keyword"> </span>{</div>
|
|
<div class="line"><a id="l02079" name="l02079"></a><span class="lineno"> 2079</span> <span class="keyword">const</span> <span class="keywordtype">size_t</span> cdim=ip;</div>
|
|
<div class="line"><a id="l02080" name="l02080"></a><span class="lineno"> 2080</span> <span class="keywordtype">size_t</span> ipph=(ip+1)/ 2;</div>
|
|
<div class="line"><a id="l02081" name="l02081"></a><span class="lineno"> 2081</span> <span class="keywordtype">size_t</span> idl1 = ido*l1;</div>
|
|
<div class="line"><a id="l02082" name="l02082"></a><span class="lineno"> 2082</span> </div>
|
|
<div class="line"><a id="l02083" name="l02083"></a><span class="lineno"> 2083</span> <span class="keyword">auto</span> CC = [cc,ido,cdim](<span class="keywordtype">size_t</span> a, <span class="keywordtype">size_t</span> b, <span class="keywordtype">size_t</span> c) -> <span class="keyword">const</span> T&</div>
|
|
<div class="line"><a id="l02084" name="l02084"></a><span class="lineno"> 2084</span> { <span class="keywordflow">return</span> cc[a+ido*(b+cdim*c)]; };</div>
|
|
<div class="line"><a id="l02085" name="l02085"></a><span class="lineno"> 2085</span> <span class="keyword">auto</span> CH = [ch,ido,l1](<span class="keywordtype">size_t</span> a, <span class="keywordtype">size_t</span> b, <span class="keywordtype">size_t</span> c) -> T&</div>
|
|
<div class="line"><a id="l02086" name="l02086"></a><span class="lineno"> 2086</span> { <span class="keywordflow">return</span> ch[a+ido*(b+l1*c)]; };</div>
|
|
<div class="line"><a id="l02087" name="l02087"></a><span class="lineno"> 2087</span> <span class="keyword">auto</span> C1 = [cc,ido,l1](<span class="keywordtype">size_t</span> a, <span class="keywordtype">size_t</span> b, <span class="keywordtype">size_t</span> c) -> <span class="keyword">const</span> T&</div>
|
|
<div class="line"><a id="l02088" name="l02088"></a><span class="lineno"> 2088</span> { <span class="keywordflow">return</span> cc[a+ido*(b+l1*c)]; };</div>
|
|
<div class="line"><a id="l02089" name="l02089"></a><span class="lineno"> 2089</span> <span class="keyword">auto</span> C2 = [cc,idl1](<span class="keywordtype">size_t</span> a, <span class="keywordtype">size_t</span> b) -> T&</div>
|
|
<div class="line"><a id="l02090" name="l02090"></a><span class="lineno"> 2090</span> { <span class="keywordflow">return</span> cc[a+idl1*b]; };</div>
|
|
<div class="line"><a id="l02091" name="l02091"></a><span class="lineno"> 2091</span> <span class="keyword">auto</span> CH2 = [ch,idl1](<span class="keywordtype">size_t</span> a, <span class="keywordtype">size_t</span> b) -> T&</div>
|
|
<div class="line"><a id="l02092" name="l02092"></a><span class="lineno"> 2092</span> { <span class="keywordflow">return</span> ch[a+idl1*b]; };</div>
|
|
<div class="line"><a id="l02093" name="l02093"></a><span class="lineno"> 2093</span> </div>
|
|
<div class="line"><a id="l02094" name="l02094"></a><span class="lineno"> 2094</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> k=0; k<l1; ++k) <span class="comment">// 102</span></div>
|
|
<div class="line"><a id="l02095" name="l02095"></a><span class="lineno"> 2095</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i=0; i<ido; ++i) <span class="comment">// 101</span></div>
|
|
<div class="line"><a id="l02096" name="l02096"></a><span class="lineno"> 2096</span> CH(i,k,0) = CC(i,0,k);</div>
|
|
<div class="line"><a id="l02097" name="l02097"></a><span class="lineno"> 2097</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> j=1, jc=ip-1; j<ipph; ++j, --jc) <span class="comment">// 108</span></div>
|
|
<div class="line"><a id="l02098" name="l02098"></a><span class="lineno"> 2098</span> {</div>
|
|
<div class="line"><a id="l02099" name="l02099"></a><span class="lineno"> 2099</span> <span class="keywordtype">size_t</span> j2=2*j-1;</div>
|
|
<div class="line"><a id="l02100" name="l02100"></a><span class="lineno"> 2100</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> k=0; k<l1; ++k)</div>
|
|
<div class="line"><a id="l02101" name="l02101"></a><span class="lineno"> 2101</span> {</div>
|
|
<div class="line"><a id="l02102" name="l02102"></a><span class="lineno"> 2102</span> CH(0,k,j ) = 2*CC(ido-1,j2,k);</div>
|
|
<div class="line"><a id="l02103" name="l02103"></a><span class="lineno"> 2103</span> CH(0,k,jc) = 2*CC(0,j2+1,k);</div>
|
|
<div class="line"><a id="l02104" name="l02104"></a><span class="lineno"> 2104</span> }</div>
|
|
<div class="line"><a id="l02105" name="l02105"></a><span class="lineno"> 2105</span> }</div>
|
|
<div class="line"><a id="l02106" name="l02106"></a><span class="lineno"> 2106</span> </div>
|
|
<div class="line"><a id="l02107" name="l02107"></a><span class="lineno"> 2107</span> <span class="keywordflow">if</span> (ido!=1)</div>
|
|
<div class="line"><a id="l02108" name="l02108"></a><span class="lineno"> 2108</span> {</div>
|
|
<div class="line"><a id="l02109" name="l02109"></a><span class="lineno"> 2109</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> j=1, jc=ip-1; j<ipph; ++j,--jc) <span class="comment">// 111</span></div>
|
|
<div class="line"><a id="l02110" name="l02110"></a><span class="lineno"> 2110</span> {</div>
|
|
<div class="line"><a id="l02111" name="l02111"></a><span class="lineno"> 2111</span> <span class="keywordtype">size_t</span> j2=2*j-1;</div>
|
|
<div class="line"><a id="l02112" name="l02112"></a><span class="lineno"> 2112</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> k=0; k<l1; ++k)</div>
|
|
<div class="line"><a id="l02113" name="l02113"></a><span class="lineno"> 2113</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i=1, ic=ido-i-2; i<=ido-2; i+=2, ic-=2) <span class="comment">// 109</span></div>
|
|
<div class="line"><a id="l02114" name="l02114"></a><span class="lineno"> 2114</span> {</div>
|
|
<div class="line"><a id="l02115" name="l02115"></a><span class="lineno"> 2115</span> CH(i ,k,j ) = CC(i ,j2+1,k)+CC(ic ,j2,k);</div>
|
|
<div class="line"><a id="l02116" name="l02116"></a><span class="lineno"> 2116</span> CH(i ,k,jc) = CC(i ,j2+1,k)-CC(ic ,j2,k);</div>
|
|
<div class="line"><a id="l02117" name="l02117"></a><span class="lineno"> 2117</span> CH(i+1,k,j ) = CC(i+1,j2+1,k)-CC(ic+1,j2,k);</div>
|
|
<div class="line"><a id="l02118" name="l02118"></a><span class="lineno"> 2118</span> CH(i+1,k,jc) = CC(i+1,j2+1,k)+CC(ic+1,j2,k);</div>
|
|
<div class="line"><a id="l02119" name="l02119"></a><span class="lineno"> 2119</span> }</div>
|
|
<div class="line"><a id="l02120" name="l02120"></a><span class="lineno"> 2120</span> }</div>
|
|
<div class="line"><a id="l02121" name="l02121"></a><span class="lineno"> 2121</span> }</div>
|
|
<div class="line"><a id="l02122" name="l02122"></a><span class="lineno"> 2122</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> l=1,lc=ip-1; l<ipph; ++l,--lc)</div>
|
|
<div class="line"><a id="l02123" name="l02123"></a><span class="lineno"> 2123</span> {</div>
|
|
<div class="line"><a id="l02124" name="l02124"></a><span class="lineno"> 2124</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> ik=0; ik<idl1; ++ik)</div>
|
|
<div class="line"><a id="l02125" name="l02125"></a><span class="lineno"> 2125</span> {</div>
|
|
<div class="line"><a id="l02126" name="l02126"></a><span class="lineno"> 2126</span> C2(ik,l ) = CH2(ik,0)+csarr[2*l]*CH2(ik,1)+csarr[4*l]*CH2(ik,2);</div>
|
|
<div class="line"><a id="l02127" name="l02127"></a><span class="lineno"> 2127</span> C2(ik,lc) = csarr[2*l+1]*CH2(ik,ip-1)+csarr[4*l+1]*CH2(ik,ip-2);</div>
|
|
<div class="line"><a id="l02128" name="l02128"></a><span class="lineno"> 2128</span> }</div>
|
|
<div class="line"><a id="l02129" name="l02129"></a><span class="lineno"> 2129</span> <span class="keywordtype">size_t</span> iang=2*l;</div>
|
|
<div class="line"><a id="l02130" name="l02130"></a><span class="lineno"> 2130</span> <span class="keywordtype">size_t</span> j=3,jc=ip-3;</div>
|
|
<div class="line"><a id="l02131" name="l02131"></a><span class="lineno"> 2131</span> <span class="keywordflow">for</span>(; j<ipph-3; j+=4,jc-=4)</div>
|
|
<div class="line"><a id="l02132" name="l02132"></a><span class="lineno"> 2132</span> {</div>
|
|
<div class="line"><a id="l02133" name="l02133"></a><span class="lineno"> 2133</span> iang+=l; <span class="keywordflow">if</span>(iang>ip) iang-=ip;</div>
|
|
<div class="line"><a id="l02134" name="l02134"></a><span class="lineno"> 2134</span> T0 ar1=csarr[2*iang], ai1=csarr[2*iang+1];</div>
|
|
<div class="line"><a id="l02135" name="l02135"></a><span class="lineno"> 2135</span> iang+=l; <span class="keywordflow">if</span>(iang>ip) iang-=ip;</div>
|
|
<div class="line"><a id="l02136" name="l02136"></a><span class="lineno"> 2136</span> T0 ar2=csarr[2*iang], ai2=csarr[2*iang+1];</div>
|
|
<div class="line"><a id="l02137" name="l02137"></a><span class="lineno"> 2137</span> iang+=l; <span class="keywordflow">if</span>(iang>ip) iang-=ip;</div>
|
|
<div class="line"><a id="l02138" name="l02138"></a><span class="lineno"> 2138</span> T0 ar3=csarr[2*iang], ai3=csarr[2*iang+1];</div>
|
|
<div class="line"><a id="l02139" name="l02139"></a><span class="lineno"> 2139</span> iang+=l; <span class="keywordflow">if</span>(iang>ip) iang-=ip;</div>
|
|
<div class="line"><a id="l02140" name="l02140"></a><span class="lineno"> 2140</span> T0 ar4=csarr[2*iang], ai4=csarr[2*iang+1];</div>
|
|
<div class="line"><a id="l02141" name="l02141"></a><span class="lineno"> 2141</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> ik=0; ik<idl1; ++ik)</div>
|
|
<div class="line"><a id="l02142" name="l02142"></a><span class="lineno"> 2142</span> {</div>
|
|
<div class="line"><a id="l02143" name="l02143"></a><span class="lineno"> 2143</span> C2(ik,l ) += ar1*CH2(ik,j )+ar2*CH2(ik,j +1)</div>
|
|
<div class="line"><a id="l02144" name="l02144"></a><span class="lineno"> 2144</span> +ar3*CH2(ik,j +2)+ar4*CH2(ik,j +3);</div>
|
|
<div class="line"><a id="l02145" name="l02145"></a><span class="lineno"> 2145</span> C2(ik,lc) += ai1*CH2(ik,jc)+ai2*CH2(ik,jc-1)</div>
|
|
<div class="line"><a id="l02146" name="l02146"></a><span class="lineno"> 2146</span> +ai3*CH2(ik,jc-2)+ai4*CH2(ik,jc-3);</div>
|
|
<div class="line"><a id="l02147" name="l02147"></a><span class="lineno"> 2147</span> }</div>
|
|
<div class="line"><a id="l02148" name="l02148"></a><span class="lineno"> 2148</span> }</div>
|
|
<div class="line"><a id="l02149" name="l02149"></a><span class="lineno"> 2149</span> <span class="keywordflow">for</span>(; j<ipph-1; j+=2,jc-=2)</div>
|
|
<div class="line"><a id="l02150" name="l02150"></a><span class="lineno"> 2150</span> {</div>
|
|
<div class="line"><a id="l02151" name="l02151"></a><span class="lineno"> 2151</span> iang+=l; <span class="keywordflow">if</span>(iang>ip) iang-=ip;</div>
|
|
<div class="line"><a id="l02152" name="l02152"></a><span class="lineno"> 2152</span> T0 ar1=csarr[2*iang], ai1=csarr[2*iang+1];</div>
|
|
<div class="line"><a id="l02153" name="l02153"></a><span class="lineno"> 2153</span> iang+=l; <span class="keywordflow">if</span>(iang>ip) iang-=ip;</div>
|
|
<div class="line"><a id="l02154" name="l02154"></a><span class="lineno"> 2154</span> T0 ar2=csarr[2*iang], ai2=csarr[2*iang+1];</div>
|
|
<div class="line"><a id="l02155" name="l02155"></a><span class="lineno"> 2155</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> ik=0; ik<idl1; ++ik)</div>
|
|
<div class="line"><a id="l02156" name="l02156"></a><span class="lineno"> 2156</span> {</div>
|
|
<div class="line"><a id="l02157" name="l02157"></a><span class="lineno"> 2157</span> C2(ik,l ) += ar1*CH2(ik,j )+ar2*CH2(ik,j +1);</div>
|
|
<div class="line"><a id="l02158" name="l02158"></a><span class="lineno"> 2158</span> C2(ik,lc) += ai1*CH2(ik,jc)+ai2*CH2(ik,jc-1);</div>
|
|
<div class="line"><a id="l02159" name="l02159"></a><span class="lineno"> 2159</span> }</div>
|
|
<div class="line"><a id="l02160" name="l02160"></a><span class="lineno"> 2160</span> }</div>
|
|
<div class="line"><a id="l02161" name="l02161"></a><span class="lineno"> 2161</span> <span class="keywordflow">for</span>(; j<ipph; ++j,--jc)</div>
|
|
<div class="line"><a id="l02162" name="l02162"></a><span class="lineno"> 2162</span> {</div>
|
|
<div class="line"><a id="l02163" name="l02163"></a><span class="lineno"> 2163</span> iang+=l; <span class="keywordflow">if</span>(iang>ip) iang-=ip;</div>
|
|
<div class="line"><a id="l02164" name="l02164"></a><span class="lineno"> 2164</span> T0 war=csarr[2*iang], wai=csarr[2*iang+1];</div>
|
|
<div class="line"><a id="l02165" name="l02165"></a><span class="lineno"> 2165</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> ik=0; ik<idl1; ++ik)</div>
|
|
<div class="line"><a id="l02166" name="l02166"></a><span class="lineno"> 2166</span> {</div>
|
|
<div class="line"><a id="l02167" name="l02167"></a><span class="lineno"> 2167</span> C2(ik,l ) += war*CH2(ik,j );</div>
|
|
<div class="line"><a id="l02168" name="l02168"></a><span class="lineno"> 2168</span> C2(ik,lc) += wai*CH2(ik,jc);</div>
|
|
<div class="line"><a id="l02169" name="l02169"></a><span class="lineno"> 2169</span> }</div>
|
|
<div class="line"><a id="l02170" name="l02170"></a><span class="lineno"> 2170</span> }</div>
|
|
<div class="line"><a id="l02171" name="l02171"></a><span class="lineno"> 2171</span> }</div>
|
|
<div class="line"><a id="l02172" name="l02172"></a><span class="lineno"> 2172</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> j=1; j<ipph; ++j)</div>
|
|
<div class="line"><a id="l02173" name="l02173"></a><span class="lineno"> 2173</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> ik=0; ik<idl1; ++ik)</div>
|
|
<div class="line"><a id="l02174" name="l02174"></a><span class="lineno"> 2174</span> CH2(ik,0) += CH2(ik,j);</div>
|
|
<div class="line"><a id="l02175" name="l02175"></a><span class="lineno"> 2175</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> j=1, jc=ip-1; j<ipph; ++j,--jc) <span class="comment">// 124</span></div>
|
|
<div class="line"><a id="l02176" name="l02176"></a><span class="lineno"> 2176</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> k=0; k<l1; ++k)</div>
|
|
<div class="line"><a id="l02177" name="l02177"></a><span class="lineno"> 2177</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">PM</a>(CH(0,k,jc),CH(0,k,j),C1(0,k,j),C1(0,k,jc));</div>
|
|
<div class="line"><a id="l02178" name="l02178"></a><span class="lineno"> 2178</span> </div>
|
|
<div class="line"><a id="l02179" name="l02179"></a><span class="lineno"> 2179</span> <span class="keywordflow">if</span> (ido==1) <span class="keywordflow">return</span>;</div>
|
|
<div class="line"><a id="l02180" name="l02180"></a><span class="lineno"> 2180</span> </div>
|
|
<div class="line"><a id="l02181" name="l02181"></a><span class="lineno"> 2181</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> j=1, jc=ip-1; j<ipph; ++j, --jc) <span class="comment">// 127</span></div>
|
|
<div class="line"><a id="l02182" name="l02182"></a><span class="lineno"> 2182</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> k=0; k<l1; ++k)</div>
|
|
<div class="line"><a id="l02183" name="l02183"></a><span class="lineno"> 2183</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i=1; i<=ido-2; i+=2)</div>
|
|
<div class="line"><a id="l02184" name="l02184"></a><span class="lineno"> 2184</span> {</div>
|
|
<div class="line"><a id="l02185" name="l02185"></a><span class="lineno"> 2185</span> CH(i ,k,j ) = C1(i ,k,j)-C1(i+1,k,jc);</div>
|
|
<div class="line"><a id="l02186" name="l02186"></a><span class="lineno"> 2186</span> CH(i ,k,jc) = C1(i ,k,j)+C1(i+1,k,jc);</div>
|
|
<div class="line"><a id="l02187" name="l02187"></a><span class="lineno"> 2187</span> CH(i+1,k,j ) = C1(i+1,k,j)+C1(i ,k,jc);</div>
|
|
<div class="line"><a id="l02188" name="l02188"></a><span class="lineno"> 2188</span> CH(i+1,k,jc) = C1(i+1,k,j)-C1(i ,k,jc);</div>
|
|
<div class="line"><a id="l02189" name="l02189"></a><span class="lineno"> 2189</span> }</div>
|
|
<div class="line"><a id="l02190" name="l02190"></a><span class="lineno"> 2190</span> </div>
|
|
<div class="line"><a id="l02191" name="l02191"></a><span class="lineno"> 2191</span><span class="comment">// All in CH</span></div>
|
|
<div class="line"><a id="l02192" name="l02192"></a><span class="lineno"> 2192</span> </div>
|
|
<div class="line"><a id="l02193" name="l02193"></a><span class="lineno"> 2193</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> j=1; j<ip; ++j)</div>
|
|
<div class="line"><a id="l02194" name="l02194"></a><span class="lineno"> 2194</span> {</div>
|
|
<div class="line"><a id="l02195" name="l02195"></a><span class="lineno"> 2195</span> <span class="keywordtype">size_t</span> is = (j-1)*(ido-1);</div>
|
|
<div class="line"><a id="l02196" name="l02196"></a><span class="lineno"> 2196</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> k=0; k<l1; ++k)</div>
|
|
<div class="line"><a id="l02197" name="l02197"></a><span class="lineno"> 2197</span> {</div>
|
|
<div class="line"><a id="l02198" name="l02198"></a><span class="lineno"> 2198</span> <span class="keywordtype">size_t</span> idij = is;</div>
|
|
<div class="line"><a id="l02199" name="l02199"></a><span class="lineno"> 2199</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i=1; i<=ido-2; i+=2)</div>
|
|
<div class="line"><a id="l02200" name="l02200"></a><span class="lineno"> 2200</span> {</div>
|
|
<div class="line"><a id="l02201" name="l02201"></a><span class="lineno"> 2201</span> T t1=CH(i,k,j), t2=CH(i+1,k,j);</div>
|
|
<div class="line"><a id="l02202" name="l02202"></a><span class="lineno"> 2202</span> CH(i ,k,j) = wa[idij]*t1-wa[idij+1]*t2;</div>
|
|
<div class="line"><a id="l02203" name="l02203"></a><span class="lineno"> 2203</span> CH(i+1,k,j) = wa[idij]*t2+wa[idij+1]*t1;</div>
|
|
<div class="line"><a id="l02204" name="l02204"></a><span class="lineno"> 2204</span> idij+=2;</div>
|
|
<div class="line"><a id="l02205" name="l02205"></a><span class="lineno"> 2205</span> }</div>
|
|
<div class="line"><a id="l02206" name="l02206"></a><span class="lineno"> 2206</span> }</div>
|
|
<div class="line"><a id="l02207" name="l02207"></a><span class="lineno"> 2207</span> }</div>
|
|
<div class="line"><a id="l02208" name="l02208"></a><span class="lineno"> 2208</span> }</div>
|
|
<div class="line"><a id="l02209" name="l02209"></a><span class="lineno"> 2209</span> </div>
|
|
<div class="line"><a id="l02210" name="l02210"></a><span class="lineno"> 2210</span> <span class="keyword">template</span><<span class="keyword">typename</span> T> <span class="keywordtype">void</span> copy_and_norm(T *c, T *p1, T0 fct)<span class="keyword"> const</span></div>
|
|
<div class="line"><a id="l02211" name="l02211"></a><span class="lineno"> 2211</span><span class="keyword"> </span>{</div>
|
|
<div class="line"><a id="l02212" name="l02212"></a><span class="lineno"> 2212</span> <span class="keywordflow">if</span> (p1!=c)</div>
|
|
<div class="line"><a id="l02213" name="l02213"></a><span class="lineno"> 2213</span> {</div>
|
|
<div class="line"><a id="l02214" name="l02214"></a><span class="lineno"> 2214</span> <span class="keywordflow">if</span> (fct!=1.)</div>
|
|
<div class="line"><a id="l02215" name="l02215"></a><span class="lineno"> 2215</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i=0; i<length; ++i)</div>
|
|
<div class="line"><a id="l02216" name="l02216"></a><span class="lineno"> 2216</span> c[i] = fct*p1[i];</div>
|
|
<div class="line"><a id="l02217" name="l02217"></a><span class="lineno"> 2217</span> <span class="keywordflow">else</span></div>
|
|
<div class="line"><a id="l02218" name="l02218"></a><span class="lineno"> 2218</span> std::copy_n (p1, length, c);</div>
|
|
<div class="line"><a id="l02219" name="l02219"></a><span class="lineno"> 2219</span> }</div>
|
|
<div class="line"><a id="l02220" name="l02220"></a><span class="lineno"> 2220</span> <span class="keywordflow">else</span></div>
|
|
<div class="line"><a id="l02221" name="l02221"></a><span class="lineno"> 2221</span> <span class="keywordflow">if</span> (fct!=1.)</div>
|
|
<div class="line"><a id="l02222" name="l02222"></a><span class="lineno"> 2222</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i=0; i<length; ++i)</div>
|
|
<div class="line"><a id="l02223" name="l02223"></a><span class="lineno"> 2223</span> c[i] *= fct;</div>
|
|
<div class="line"><a id="l02224" name="l02224"></a><span class="lineno"> 2224</span> }</div>
|
|
<div class="line"><a id="l02225" name="l02225"></a><span class="lineno"> 2225</span> </div>
|
|
<div class="line"><a id="l02226" name="l02226"></a><span class="lineno"> 2226</span> <span class="keyword">public</span>:</div>
|
|
<div class="foldopen" id="foldopen02227" data-start="{" data-end="}">
|
|
<div class="line"><a id="l02227" name="l02227"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1rfftp.html#a073972f42bdd3617693be7be2cb5e0ac"> 2227</a></span> <span class="keyword">template</span><<span class="keyword">typename</span> T> <span class="keywordtype">void</span> <a class="code hl_function" href="classpocketfft_1_1detail_1_1rfftp.html#a073972f42bdd3617693be7be2cb5e0ac">exec</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a> c[], <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T0</a> fct, <span class="keywordtype">bool</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">r2hc</a>)<span class="keyword"> const</span></div>
|
|
<div class="line"><a id="l02228" name="l02228"></a><span class="lineno"> 2228</span><span class="keyword"> </span>{</div>
|
|
<div class="line"><a id="l02229" name="l02229"></a><span class="lineno"> 2229</span> <span class="keywordflow">if</span> (length==1) { c[0]*=fct; <span class="keywordflow">return</span>; }</div>
|
|
<div class="line"><a id="l02230" name="l02230"></a><span class="lineno"> 2230</span> <span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nf</a>=fact.size();</div>
|
|
<div class="line"><a id="l02231" name="l02231"></a><span class="lineno"> 2231</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">arr<T></a> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ch</a>(length);</div>
|
|
<div class="line"><a id="l02232" name="l02232"></a><span class="lineno"> 2232</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a> *<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">p1</a>=c, *<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">p2</a>=<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ch</a>.data();</div>
|
|
<div class="line"><a id="l02233" name="l02233"></a><span class="lineno"> 2233</span> </div>
|
|
<div class="line"><a id="l02234" name="l02234"></a><span class="lineno"> 2234</span> <span class="keywordflow">if</span> (<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">r2hc</a>)</div>
|
|
<div class="line"><a id="l02235" name="l02235"></a><span class="lineno"> 2235</span> <span class="keywordflow">for</span>(<span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k1</a>=0, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">l1</a>=length; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k1</a><<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nf</a>;++<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k1</a>)</div>
|
|
<div class="line"><a id="l02236" name="l02236"></a><span class="lineno"> 2236</span> {</div>
|
|
<div class="line"><a id="l02237" name="l02237"></a><span class="lineno"> 2237</span> <span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>=<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nf</a>-<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k1</a>-1;</div>
|
|
<div class="line"><a id="l02238" name="l02238"></a><span class="lineno"> 2238</span> <span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ip</a>=fact[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>].fct;</div>
|
|
<div class="line"><a id="l02239" name="l02239"></a><span class="lineno"> 2239</span> <span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ido</a>=length / <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">l1</a>;</div>
|
|
<div class="line"><a id="l02240" name="l02240"></a><span class="lineno"> 2240</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">l1</a> /= <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ip</a>;</div>
|
|
<div class="line"><a id="l02241" name="l02241"></a><span class="lineno"> 2241</span> <span class="keywordflow">if</span>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ip</a>==4)</div>
|
|
<div class="line"><a id="l02242" name="l02242"></a><span class="lineno"> 2242</span> radf4(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ido</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">l1</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">p1</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">p2</a>, fact[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>].tw);</div>
|
|
<div class="line"><a id="l02243" name="l02243"></a><span class="lineno"> 2243</span> <span class="keywordflow">else</span> <span class="keywordflow">if</span>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ip</a>==2)</div>
|
|
<div class="line"><a id="l02244" name="l02244"></a><span class="lineno"> 2244</span> radf2(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ido</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">l1</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">p1</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">p2</a>, fact[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>].tw);</div>
|
|
<div class="line"><a id="l02245" name="l02245"></a><span class="lineno"> 2245</span> <span class="keywordflow">else</span> <span class="keywordflow">if</span>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ip</a>==3)</div>
|
|
<div class="line"><a id="l02246" name="l02246"></a><span class="lineno"> 2246</span> radf3(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ido</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">l1</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">p1</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">p2</a>, fact[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>].tw);</div>
|
|
<div class="line"><a id="l02247" name="l02247"></a><span class="lineno"> 2247</span> <span class="keywordflow">else</span> <span class="keywordflow">if</span>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ip</a>==5)</div>
|
|
<div class="line"><a id="l02248" name="l02248"></a><span class="lineno"> 2248</span> radf5(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ido</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">l1</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">p1</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">p2</a>, fact[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>].tw);</div>
|
|
<div class="line"><a id="l02249" name="l02249"></a><span class="lineno"> 2249</span> <span class="keywordflow">else</span></div>
|
|
<div class="line"><a id="l02250" name="l02250"></a><span class="lineno"> 2250</span> { radfg(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ido</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ip</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">l1</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">p1</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">p2</a>, fact[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>].tw, fact[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>].tws); std::swap (<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">p1</a>,<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">p2</a>); }</div>
|
|
<div class="line"><a id="l02251" name="l02251"></a><span class="lineno"> 2251</span> std::swap (<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">p1</a>,<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">p2</a>);</div>
|
|
<div class="line"><a id="l02252" name="l02252"></a><span class="lineno"> 2252</span> }</div>
|
|
<div class="line"><a id="l02253" name="l02253"></a><span class="lineno"> 2253</span> <span class="keywordflow">else</span></div>
|
|
<div class="line"><a id="l02254" name="l02254"></a><span class="lineno"> 2254</span> <span class="keywordflow">for</span>(<span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>=0, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">l1</a>=1; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a><<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nf</a>; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>++)</div>
|
|
<div class="line"><a id="l02255" name="l02255"></a><span class="lineno"> 2255</span> {</div>
|
|
<div class="line"><a id="l02256" name="l02256"></a><span class="lineno"> 2256</span> <span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ip</a> = fact[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>].fct,</div>
|
|
<div class="line"><a id="l02257" name="l02257"></a><span class="lineno"> 2257</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ido</a>= length/(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ip</a>*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">l1</a>);</div>
|
|
<div class="line"><a id="l02258" name="l02258"></a><span class="lineno"> 2258</span> <span class="keywordflow">if</span>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ip</a>==4)</div>
|
|
<div class="line"><a id="l02259" name="l02259"></a><span class="lineno"> 2259</span> radb4(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ido</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">l1</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">p1</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">p2</a>, fact[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>].tw);</div>
|
|
<div class="line"><a id="l02260" name="l02260"></a><span class="lineno"> 2260</span> <span class="keywordflow">else</span> <span class="keywordflow">if</span>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ip</a>==2)</div>
|
|
<div class="line"><a id="l02261" name="l02261"></a><span class="lineno"> 2261</span> radb2(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ido</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">l1</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">p1</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">p2</a>, fact[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>].tw);</div>
|
|
<div class="line"><a id="l02262" name="l02262"></a><span class="lineno"> 2262</span> <span class="keywordflow">else</span> <span class="keywordflow">if</span>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ip</a>==3)</div>
|
|
<div class="line"><a id="l02263" name="l02263"></a><span class="lineno"> 2263</span> radb3(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ido</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">l1</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">p1</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">p2</a>, fact[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>].tw);</div>
|
|
<div class="line"><a id="l02264" name="l02264"></a><span class="lineno"> 2264</span> <span class="keywordflow">else</span> <span class="keywordflow">if</span>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ip</a>==5)</div>
|
|
<div class="line"><a id="l02265" name="l02265"></a><span class="lineno"> 2265</span> radb5(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ido</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">l1</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">p1</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">p2</a>, fact[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>].tw);</div>
|
|
<div class="line"><a id="l02266" name="l02266"></a><span class="lineno"> 2266</span> <span class="keywordflow">else</span></div>
|
|
<div class="line"><a id="l02267" name="l02267"></a><span class="lineno"> 2267</span> radbg(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ido</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ip</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">l1</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">p1</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">p2</a>, fact[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>].tw, fact[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>].tws);</div>
|
|
<div class="line"><a id="l02268" name="l02268"></a><span class="lineno"> 2268</span> std::swap (<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">p1</a>,<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">p2</a>);</div>
|
|
<div class="line"><a id="l02269" name="l02269"></a><span class="lineno"> 2269</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">l1</a>*=<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ip</a>;</div>
|
|
<div class="line"><a id="l02270" name="l02270"></a><span class="lineno"> 2270</span> }</div>
|
|
<div class="line"><a id="l02271" name="l02271"></a><span class="lineno"> 2271</span> </div>
|
|
<div class="line"><a id="l02272" name="l02272"></a><span class="lineno"> 2272</span> copy_and_norm(c,<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">p1</a>,fct);</div>
|
|
<div class="line"><a id="l02273" name="l02273"></a><span class="lineno"> 2273</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l02274" name="l02274"></a><span class="lineno"> 2274</span> </div>
|
|
<div class="line"><a id="l02275" name="l02275"></a><span class="lineno"> 2275</span> <span class="keyword">private</span>:</div>
|
|
<div class="line"><a id="l02276" name="l02276"></a><span class="lineno"> 2276</span> <span class="keywordtype">void</span> factorize()</div>
|
|
<div class="line"><a id="l02277" name="l02277"></a><span class="lineno"> 2277</span> {</div>
|
|
<div class="line"><a id="l02278" name="l02278"></a><span class="lineno"> 2278</span> <span class="keywordtype">size_t</span> len=length;</div>
|
|
<div class="line"><a id="l02279" name="l02279"></a><span class="lineno"> 2279</span> <span class="keywordflow">while</span> ((len%4)==0)</div>
|
|
<div class="line"><a id="l02280" name="l02280"></a><span class="lineno"> 2280</span> { add_factor(4); len>>=2; }</div>
|
|
<div class="line"><a id="l02281" name="l02281"></a><span class="lineno"> 2281</span> <span class="keywordflow">if</span> ((len%2)==0)</div>
|
|
<div class="line"><a id="l02282" name="l02282"></a><span class="lineno"> 2282</span> {</div>
|
|
<div class="line"><a id="l02283" name="l02283"></a><span class="lineno"> 2283</span> len>>=1;</div>
|
|
<div class="line"><a id="l02284" name="l02284"></a><span class="lineno"> 2284</span> <span class="comment">// factor 2 should be at the front of the factor list</span></div>
|
|
<div class="line"><a id="l02285" name="l02285"></a><span class="lineno"> 2285</span> add_factor(2);</div>
|
|
<div class="line"><a id="l02286" name="l02286"></a><span class="lineno"> 2286</span> std::swap(fact[0].fct, fact.back().fct);</div>
|
|
<div class="line"><a id="l02287" name="l02287"></a><span class="lineno"> 2287</span> }</div>
|
|
<div class="line"><a id="l02288" name="l02288"></a><span class="lineno"> 2288</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> divisor=3; divisor*divisor<=len; divisor+=2)</div>
|
|
<div class="line"><a id="l02289" name="l02289"></a><span class="lineno"> 2289</span> <span class="keywordflow">while</span> ((len%divisor)==0)</div>
|
|
<div class="line"><a id="l02290" name="l02290"></a><span class="lineno"> 2290</span> {</div>
|
|
<div class="line"><a id="l02291" name="l02291"></a><span class="lineno"> 2291</span> add_factor(divisor);</div>
|
|
<div class="line"><a id="l02292" name="l02292"></a><span class="lineno"> 2292</span> len/=divisor;</div>
|
|
<div class="line"><a id="l02293" name="l02293"></a><span class="lineno"> 2293</span> }</div>
|
|
<div class="line"><a id="l02294" name="l02294"></a><span class="lineno"> 2294</span> <span class="keywordflow">if</span> (len>1) add_factor(len);</div>
|
|
<div class="line"><a id="l02295" name="l02295"></a><span class="lineno"> 2295</span> }</div>
|
|
<div class="line"><a id="l02296" name="l02296"></a><span class="lineno"> 2296</span> </div>
|
|
<div class="line"><a id="l02297" name="l02297"></a><span class="lineno"> 2297</span> <span class="keywordtype">size_t</span> twsize()<span class="keyword"> const</span></div>
|
|
<div class="line"><a id="l02298" name="l02298"></a><span class="lineno"> 2298</span><span class="keyword"> </span>{</div>
|
|
<div class="line"><a id="l02299" name="l02299"></a><span class="lineno"> 2299</span> <span class="keywordtype">size_t</span> twsz=0, l1=1;</div>
|
|
<div class="line"><a id="l02300" name="l02300"></a><span class="lineno"> 2300</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> k=0; k<fact.size(); ++k)</div>
|
|
<div class="line"><a id="l02301" name="l02301"></a><span class="lineno"> 2301</span> {</div>
|
|
<div class="line"><a id="l02302" name="l02302"></a><span class="lineno"> 2302</span> <span class="keywordtype">size_t</span> ip=fact[k].fct, ido=length/(l1*ip);</div>
|
|
<div class="line"><a id="l02303" name="l02303"></a><span class="lineno"> 2303</span> twsz+=(ip-1)*(ido-1);</div>
|
|
<div class="line"><a id="l02304" name="l02304"></a><span class="lineno"> 2304</span> <span class="keywordflow">if</span> (ip>5) twsz+=2*ip;</div>
|
|
<div class="line"><a id="l02305" name="l02305"></a><span class="lineno"> 2305</span> l1*=ip;</div>
|
|
<div class="line"><a id="l02306" name="l02306"></a><span class="lineno"> 2306</span> }</div>
|
|
<div class="line"><a id="l02307" name="l02307"></a><span class="lineno"> 2307</span> <span class="keywordflow">return</span> twsz;</div>
|
|
<div class="line"><a id="l02308" name="l02308"></a><span class="lineno"> 2308</span> }</div>
|
|
<div class="line"><a id="l02309" name="l02309"></a><span class="lineno"> 2309</span> </div>
|
|
<div class="line"><a id="l02310" name="l02310"></a><span class="lineno"> 2310</span> <span class="keywordtype">void</span> comp_twiddle()</div>
|
|
<div class="line"><a id="l02311" name="l02311"></a><span class="lineno"> 2311</span> {</div>
|
|
<div class="line"><a id="l02312" name="l02312"></a><span class="lineno"> 2312</span> sincos_2pibyn<T0> twid(length);</div>
|
|
<div class="line"><a id="l02313" name="l02313"></a><span class="lineno"> 2313</span> <span class="keywordtype">size_t</span> l1=1;</div>
|
|
<div class="line"><a id="l02314" name="l02314"></a><span class="lineno"> 2314</span> T0 *ptr=mem.data();</div>
|
|
<div class="line"><a id="l02315" name="l02315"></a><span class="lineno"> 2315</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> k=0; k<fact.size(); ++k)</div>
|
|
<div class="line"><a id="l02316" name="l02316"></a><span class="lineno"> 2316</span> {</div>
|
|
<div class="line"><a id="l02317" name="l02317"></a><span class="lineno"> 2317</span> <span class="keywordtype">size_t</span> ip=fact[k].fct, ido=length/(l1*ip);</div>
|
|
<div class="line"><a id="l02318" name="l02318"></a><span class="lineno"> 2318</span> <span class="keywordflow">if</span> (k<fact.size()-1) <span class="comment">// last factor doesn't need twiddles</span></div>
|
|
<div class="line"><a id="l02319" name="l02319"></a><span class="lineno"> 2319</span> {</div>
|
|
<div class="line"><a id="l02320" name="l02320"></a><span class="lineno"> 2320</span> fact[k].tw=ptr; ptr+=(ip-1)*(ido-1);</div>
|
|
<div class="line"><a id="l02321" name="l02321"></a><span class="lineno"> 2321</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> j=1; j<ip; ++j)</div>
|
|
<div class="line"><a id="l02322" name="l02322"></a><span class="lineno"> 2322</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i=1; i<=(ido-1)/2; ++i)</div>
|
|
<div class="line"><a id="l02323" name="l02323"></a><span class="lineno"> 2323</span> {</div>
|
|
<div class="line"><a id="l02324" name="l02324"></a><span class="lineno"> 2324</span> fact[k].tw[(j-1)*(ido-1)+2*i-2] = twid[j*l1*i].r;</div>
|
|
<div class="line"><a id="l02325" name="l02325"></a><span class="lineno"> 2325</span> fact[k].tw[(j-1)*(ido-1)+2*i-1] = twid[j*l1*i].i;</div>
|
|
<div class="line"><a id="l02326" name="l02326"></a><span class="lineno"> 2326</span> }</div>
|
|
<div class="line"><a id="l02327" name="l02327"></a><span class="lineno"> 2327</span> }</div>
|
|
<div class="line"><a id="l02328" name="l02328"></a><span class="lineno"> 2328</span> <span class="keywordflow">if</span> (ip>5) <span class="comment">// special factors required by *g functions</span></div>
|
|
<div class="line"><a id="l02329" name="l02329"></a><span class="lineno"> 2329</span> {</div>
|
|
<div class="line"><a id="l02330" name="l02330"></a><span class="lineno"> 2330</span> fact[k].tws=ptr; ptr+=2*ip;</div>
|
|
<div class="line"><a id="l02331" name="l02331"></a><span class="lineno"> 2331</span> fact[k].tws[0] = 1.;</div>
|
|
<div class="line"><a id="l02332" name="l02332"></a><span class="lineno"> 2332</span> fact[k].tws[1] = 0.;</div>
|
|
<div class="line"><a id="l02333" name="l02333"></a><span class="lineno"> 2333</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i=2, ic=2*ip-2; i<=ic; i+=2, ic-=2)</div>
|
|
<div class="line"><a id="l02334" name="l02334"></a><span class="lineno"> 2334</span> {</div>
|
|
<div class="line"><a id="l02335" name="l02335"></a><span class="lineno"> 2335</span> fact[k].tws[i ] = twid[i/2*(length/ip)].r;</div>
|
|
<div class="line"><a id="l02336" name="l02336"></a><span class="lineno"> 2336</span> fact[k].tws[i+1] = twid[i/2*(length/ip)].i;</div>
|
|
<div class="line"><a id="l02337" name="l02337"></a><span class="lineno"> 2337</span> fact[k].tws[ic] = twid[i/2*(length/ip)].r;</div>
|
|
<div class="line"><a id="l02338" name="l02338"></a><span class="lineno"> 2338</span> fact[k].tws[ic+1] = -twid[i/2*(length/ip)].i;</div>
|
|
<div class="line"><a id="l02339" name="l02339"></a><span class="lineno"> 2339</span> }</div>
|
|
<div class="line"><a id="l02340" name="l02340"></a><span class="lineno"> 2340</span> }</div>
|
|
<div class="line"><a id="l02341" name="l02341"></a><span class="lineno"> 2341</span> l1*=ip;</div>
|
|
<div class="line"><a id="l02342" name="l02342"></a><span class="lineno"> 2342</span> }</div>
|
|
<div class="line"><a id="l02343" name="l02343"></a><span class="lineno"> 2343</span> }</div>
|
|
<div class="line"><a id="l02344" name="l02344"></a><span class="lineno"> 2344</span> </div>
|
|
<div class="line"><a id="l02345" name="l02345"></a><span class="lineno"> 2345</span> <span class="keyword">public</span>:</div>
|
|
<div class="foldopen" id="foldopen02346" data-start="{" data-end="}">
|
|
<div class="line"><a id="l02346" name="l02346"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1rfftp.html#a0c590f917b8e8afa3ff53ccff52e68c5"> 2346</a></span> <a class="code hl_define" href="pocketfft_8h.html#a7020984e0ca1d6e565629ca6e7c1a7e0">POCKETFFT_NOINLINE</a> <a class="code hl_function" href="classpocketfft_1_1detail_1_1rfftp.html#a0c590f917b8e8afa3ff53ccff52e68c5">rfftp</a>(<span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">length_</a>)</div>
|
|
<div class="line"><a id="l02347" name="l02347"></a><span class="lineno"> 2347</span> : length(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">length_</a>)</div>
|
|
<div class="line"><a id="l02348" name="l02348"></a><span class="lineno"> 2348</span> {</div>
|
|
<div class="line"><a id="l02349" name="l02349"></a><span class="lineno"> 2349</span> <span class="keywordflow">if</span> (length==0) <span class="keywordflow">throw</span> std::runtime_error(<span class="stringliteral">"zero-length FFT requested"</span>);</div>
|
|
<div class="line"><a id="l02350" name="l02350"></a><span class="lineno"> 2350</span> <span class="keywordflow">if</span> (length==1) <span class="keywordflow">return</span>;</div>
|
|
<div class="line"><a id="l02351" name="l02351"></a><span class="lineno"> 2351</span> factorize();</div>
|
|
<div class="line"><a id="l02352" name="l02352"></a><span class="lineno"> 2352</span> mem.resize(twsize());</div>
|
|
<div class="line"><a id="l02353" name="l02353"></a><span class="lineno"> 2353</span> comp_twiddle();</div>
|
|
<div class="line"><a id="l02354" name="l02354"></a><span class="lineno"> 2354</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l02355" name="l02355"></a><span class="lineno"> 2355</span>};</div>
|
|
</div>
|
|
<div class="line"><a id="l02356" name="l02356"></a><span class="lineno"> 2356</span> </div>
|
|
<div class="line"><a id="l02357" name="l02357"></a><span class="lineno"> 2357</span><span class="comment">//</span></div>
|
|
<div class="line"><a id="l02358" name="l02358"></a><span class="lineno"> 2358</span><span class="comment">// complex Bluestein transforms</span></div>
|
|
<div class="line"><a id="l02359" name="l02359"></a><span class="lineno"> 2359</span><span class="comment">//</span></div>
|
|
<div class="line"><a id="l02360" name="l02360"></a><span class="lineno"> 2360</span> </div>
|
|
<div class="foldopen" id="foldopen02361" data-start="{" data-end="};">
|
|
<div class="line"><a id="l02361" name="l02361"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1fftblue.html"> 2361</a></span><span class="keyword">template</span><<span class="keyword">typename</span> T0> <span class="keyword">class </span><a class="code hl_class" href="classpocketfft_1_1detail_1_1fftblue.html">fftblue</a></div>
|
|
<div class="line"><a id="l02362" name="l02362"></a><span class="lineno"> 2362</span> {</div>
|
|
<div class="line"><a id="l02363" name="l02363"></a><span class="lineno"> 2363</span> <span class="keyword">private</span>:</div>
|
|
<div class="line"><a id="l02364" name="l02364"></a><span class="lineno"> 2364</span> <span class="keywordtype">size_t</span> n, n2;</div>
|
|
<div class="line"><a id="l02365" name="l02365"></a><span class="lineno"> 2365</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cfftp<T0></a> plan;</div>
|
|
<div class="line"><a id="l02366" name="l02366"></a><span class="lineno"> 2366</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">arr<cmplx<T0></a>> mem;</div>
|
|
<div class="line"><a id="l02367" name="l02367"></a><span class="lineno"> 2367</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cmplx<T0></a> *bk, *bkf;</div>
|
|
<div class="line"><a id="l02368" name="l02368"></a><span class="lineno"> 2368</span> </div>
|
|
<div class="line"><a id="l02369" name="l02369"></a><span class="lineno"> 2369</span> <span class="keyword">template</span><<span class="keywordtype">bool</span> fwd, <span class="keyword">typename</span> T> <span class="keywordtype">void</span> <a class="code hl_function" href="backend_2metal_2kernels_2fft_8h.html#a4010b0e151e5f01e610e9c32234458c7">fft</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cmplx<T></a> c[], <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T0</a> fct)<span class="keyword"> const</span></div>
|
|
<div class="line"><a id="l02370" name="l02370"></a><span class="lineno"> 2370</span><span class="keyword"> </span>{</div>
|
|
<div class="line"><a id="l02371" name="l02371"></a><span class="lineno"> 2371</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">arr<cmplx<T></a>> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">akf</a>(n2);</div>
|
|
<div class="line"><a id="l02372" name="l02372"></a><span class="lineno"> 2372</span> </div>
|
|
<div class="line"><a id="l02373" name="l02373"></a><span class="lineno"> 2373</span> <span class="comment">/* initialize a_k and FFT it */</span></div>
|
|
<div class="line"><a id="l02374" name="l02374"></a><span class="lineno"> 2374</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a>=0; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a><n; ++<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a>)</div>
|
|
<div class="line"><a id="l02375" name="l02375"></a><span class="lineno"> 2375</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">special_mul<fwd></a>(c[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a>],bk[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a>],<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">akf</a>[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a>]);</div>
|
|
<div class="line"><a id="l02376" name="l02376"></a><span class="lineno"> 2376</span> <span class="keyword">auto</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">zero</a> = <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">akf</a>[0]*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T0</a>(0);</div>
|
|
<div class="line"><a id="l02377" name="l02377"></a><span class="lineno"> 2377</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a>=n; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a><n2; ++<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a>)</div>
|
|
<div class="line"><a id="l02378" name="l02378"></a><span class="lineno"> 2378</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">akf</a>[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a>]=<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">zero</a>;</div>
|
|
<div class="line"><a id="l02379" name="l02379"></a><span class="lineno"> 2379</span> </div>
|
|
<div class="line"><a id="l02380" name="l02380"></a><span class="lineno"> 2380</span> plan.exec (<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">akf</a>.data(),1.,<span class="keyword">true</span>);</div>
|
|
<div class="line"><a id="l02381" name="l02381"></a><span class="lineno"> 2381</span> </div>
|
|
<div class="line"><a id="l02382" name="l02382"></a><span class="lineno"> 2382</span> <span class="comment">/* do the convolution */</span></div>
|
|
<div class="line"><a id="l02383" name="l02383"></a><span class="lineno"> 2383</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">akf</a>[0] = <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">akf</a>[0].template <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">special_mul<!fwd></a>(bkf[0]);</div>
|
|
<div class="line"><a id="l02384" name="l02384"></a><span class="lineno"> 2384</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a>=1; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a><(n2+1)/2; ++<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a>)</div>
|
|
<div class="line"><a id="l02385" name="l02385"></a><span class="lineno"> 2385</span> {</div>
|
|
<div class="line"><a id="l02386" name="l02386"></a><span class="lineno"> 2386</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">akf</a>[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a>] = <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">akf</a>[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a>].template <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">special_mul<!fwd></a>(bkf[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a>]);</div>
|
|
<div class="line"><a id="l02387" name="l02387"></a><span class="lineno"> 2387</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">akf</a>[n2-<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a>] = <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">akf</a>[n2-<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a>].template <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">special_mul<!fwd></a>(bkf[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a>]);</div>
|
|
<div class="line"><a id="l02388" name="l02388"></a><span class="lineno"> 2388</span> }</div>
|
|
<div class="line"><a id="l02389" name="l02389"></a><span class="lineno"> 2389</span> <span class="keywordflow">if</span> ((n2&1)==0)</div>
|
|
<div class="line"><a id="l02390" name="l02390"></a><span class="lineno"> 2390</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">akf</a>[n2/2] = <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">akf</a>[n2/2].template <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">special_mul<!fwd></a>(bkf[n2/2]);</div>
|
|
<div class="line"><a id="l02391" name="l02391"></a><span class="lineno"> 2391</span> </div>
|
|
<div class="line"><a id="l02392" name="l02392"></a><span class="lineno"> 2392</span> <span class="comment">/* inverse FFT */</span></div>
|
|
<div class="line"><a id="l02393" name="l02393"></a><span class="lineno"> 2393</span> plan.exec (<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">akf</a>.data(),1.,<span class="keyword">false</span>);</div>
|
|
<div class="line"><a id="l02394" name="l02394"></a><span class="lineno"> 2394</span> </div>
|
|
<div class="line"><a id="l02395" name="l02395"></a><span class="lineno"> 2395</span> <span class="comment">/* multiply by b_k */</span></div>
|
|
<div class="line"><a id="l02396" name="l02396"></a><span class="lineno"> 2396</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a>=0; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a><n; ++<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a>)</div>
|
|
<div class="line"><a id="l02397" name="l02397"></a><span class="lineno"> 2397</span> c[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a>] = <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">akf</a>[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a>].<span class="keyword">template</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">special_mul<fwd></a>(bk[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a>])*fct;</div>
|
|
<div class="line"><a id="l02398" name="l02398"></a><span class="lineno"> 2398</span> }</div>
|
|
<div class="line"><a id="l02399" name="l02399"></a><span class="lineno"> 2399</span> </div>
|
|
<div class="line"><a id="l02400" name="l02400"></a><span class="lineno"> 2400</span> <span class="keyword">public</span>:</div>
|
|
<div class="foldopen" id="foldopen02401" data-start="{" data-end="}">
|
|
<div class="line"><a id="l02401" name="l02401"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1fftblue.html#a8613b2131e7843bbca9e1cd4cc15bc01"> 2401</a></span> <a class="code hl_define" href="pocketfft_8h.html#a7020984e0ca1d6e565629ca6e7c1a7e0">POCKETFFT_NOINLINE</a> <a class="code hl_function" href="classpocketfft_1_1detail_1_1fftblue.html#a8613b2131e7843bbca9e1cd4cc15bc01">fftblue</a>(<span class="keywordtype">size_t</span> length)</div>
|
|
<div class="line"><a id="l02402" name="l02402"></a><span class="lineno"> 2402</span> : n(length), n2(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1util.html">util</a>::good_size_cmplx(n*2-1)), plan(n2), mem(n+n2/2+1),</div>
|
|
<div class="line"><a id="l02403" name="l02403"></a><span class="lineno"> 2403</span> bk(mem.data()), bkf(mem.data()+n)</div>
|
|
<div class="line"><a id="l02404" name="l02404"></a><span class="lineno"> 2404</span> {</div>
|
|
<div class="line"><a id="l02405" name="l02405"></a><span class="lineno"> 2405</span> <span class="comment">/* initialize b_k */</span></div>
|
|
<div class="line"><a id="l02406" name="l02406"></a><span class="lineno"> 2406</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">sincos_2pibyn<T0></a> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tmp</a>(2*n);</div>
|
|
<div class="line"><a id="l02407" name="l02407"></a><span class="lineno"> 2407</span> bk[0].<a class="code hl_function" href="structpocketfft_1_1detail_1_1cmplx.html#a647fece372b64b13c4a7e5877d09a807">Set</a>(1, 0);</div>
|
|
<div class="line"><a id="l02408" name="l02408"></a><span class="lineno"> 2408</span> </div>
|
|
<div class="line"><a id="l02409" name="l02409"></a><span class="lineno"> 2409</span> <span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">coeff</a>=0;</div>
|
|
<div class="line"><a id="l02410" name="l02410"></a><span class="lineno"> 2410</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a>=1; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a><n; ++<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a>)</div>
|
|
<div class="line"><a id="l02411" name="l02411"></a><span class="lineno"> 2411</span> {</div>
|
|
<div class="line"><a id="l02412" name="l02412"></a><span class="lineno"> 2412</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">coeff</a>+=2*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a>-1;</div>
|
|
<div class="line"><a id="l02413" name="l02413"></a><span class="lineno"> 2413</span> <span class="keywordflow">if</span> (<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">coeff</a>>=2*n) <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">coeff</a>-=2*n;</div>
|
|
<div class="line"><a id="l02414" name="l02414"></a><span class="lineno"> 2414</span> bk[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a>] = <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tmp</a>[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">coeff</a>];</div>
|
|
<div class="line"><a id="l02415" name="l02415"></a><span class="lineno"> 2415</span> }</div>
|
|
<div class="line"><a id="l02416" name="l02416"></a><span class="lineno"> 2416</span> </div>
|
|
<div class="line"><a id="l02417" name="l02417"></a><span class="lineno"> 2417</span> <span class="comment">/* initialize the zero-padded, Fourier transformed b_k. Add normalisation. */</span></div>
|
|
<div class="line"><a id="l02418" name="l02418"></a><span class="lineno"> 2418</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">arr<cmplx<T0></a>> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tbkf</a>(n2);</div>
|
|
<div class="line"><a id="l02419" name="l02419"></a><span class="lineno"> 2419</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T0</a> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">xn2</a> = <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T0</a>(1)/<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T0</a>(n2);</div>
|
|
<div class="line"><a id="l02420" name="l02420"></a><span class="lineno"> 2420</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tbkf</a>[0] = bk[0]*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">xn2</a>;</div>
|
|
<div class="line"><a id="l02421" name="l02421"></a><span class="lineno"> 2421</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a>=1; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a><n; ++<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a>)</div>
|
|
<div class="line"><a id="l02422" name="l02422"></a><span class="lineno"> 2422</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tbkf</a>[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a>] = <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tbkf</a>[n2-<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a>] = bk[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a>]*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">xn2</a>;</div>
|
|
<div class="line"><a id="l02423" name="l02423"></a><span class="lineno"> 2423</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a>=n;<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a><=(n2-n);++<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a>)</div>
|
|
<div class="line"><a id="l02424" name="l02424"></a><span class="lineno"> 2424</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tbkf</a>[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a>].Set(0.,0.);</div>
|
|
<div class="line"><a id="l02425" name="l02425"></a><span class="lineno"> 2425</span> plan.exec(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tbkf</a>.data(),1.,<span class="keyword">true</span>);</div>
|
|
<div class="line"><a id="l02426" name="l02426"></a><span class="lineno"> 2426</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i=0; i<n2/2+1; ++i)</div>
|
|
<div class="line"><a id="l02427" name="l02427"></a><span class="lineno"> 2427</span> bkf[i] = <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tbkf</a>[i];</div>
|
|
<div class="line"><a id="l02428" name="l02428"></a><span class="lineno"> 2428</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l02429" name="l02429"></a><span class="lineno"> 2429</span> </div>
|
|
<div class="foldopen" id="foldopen02430" data-start="{" data-end="}">
|
|
<div class="line"><a id="l02430" name="l02430"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1fftblue.html#a5fb03413a3d1a653842875adcf87ae8c"> 2430</a></span> <span class="keyword">template</span><<span class="keyword">typename</span> T> <span class="keywordtype">void</span> <a class="code hl_function" href="classpocketfft_1_1detail_1_1fftblue.html#a5fb03413a3d1a653842875adcf87ae8c">exec</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cmplx<T></a> c[], <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T0</a> fct, <span class="keywordtype">bool</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">fwd</a>)<span class="keyword"> const</span></div>
|
|
<div class="line"><a id="l02431" name="l02431"></a><span class="lineno"> 2431</span><span class="keyword"> </span>{ <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">fwd</a> ? <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">fft<true></a>(c,fct) : <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">fft<false></a>(c,fct); }</div>
|
|
</div>
|
|
<div class="line"><a id="l02432" name="l02432"></a><span class="lineno"> 2432</span> </div>
|
|
<div class="foldopen" id="foldopen02433" data-start="{" data-end="}">
|
|
<div class="line"><a id="l02433" name="l02433"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1fftblue.html#a642b4aff0485c7d9c8794161a1464f00"> 2433</a></span> <span class="keyword">template</span><<span class="keyword">typename</span> T> <span class="keywordtype">void</span> <a class="code hl_function" href="classpocketfft_1_1detail_1_1fftblue.html#a642b4aff0485c7d9c8794161a1464f00">exec_r</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a> c[], <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T0</a> fct, <span class="keywordtype">bool</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">fwd</a>)</div>
|
|
<div class="line"><a id="l02434" name="l02434"></a><span class="lineno"> 2434</span> {</div>
|
|
<div class="line"><a id="l02435" name="l02435"></a><span class="lineno"> 2435</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">arr<cmplx<T></a>> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tmp</a>(n);</div>
|
|
<div class="line"><a id="l02436" name="l02436"></a><span class="lineno"> 2436</span> <span class="keywordflow">if</span> (<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">fwd</a>)</div>
|
|
<div class="line"><a id="l02437" name="l02437"></a><span class="lineno"> 2437</span> {</div>
|
|
<div class="line"><a id="l02438" name="l02438"></a><span class="lineno"> 2438</span> <span class="keyword">auto</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">zero</a> = <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T0</a>(0)*c[0];</div>
|
|
<div class="line"><a id="l02439" name="l02439"></a><span class="lineno"> 2439</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a>=0; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a><n; ++<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a>)</div>
|
|
<div class="line"><a id="l02440" name="l02440"></a><span class="lineno"> 2440</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tmp</a>[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a>].Set(c[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a>], <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">zero</a>);</div>
|
|
<div class="line"><a id="l02441" name="l02441"></a><span class="lineno"> 2441</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">fft<true></a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tmp</a>.data(),fct);</div>
|
|
<div class="line"><a id="l02442" name="l02442"></a><span class="lineno"> 2442</span> c[0] = <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tmp</a>[0].<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a>;</div>
|
|
<div class="line"><a id="l02443" name="l02443"></a><span class="lineno"> 2443</span> std::copy_n (&<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tmp</a>[1].r, n-1, &c[1]);</div>
|
|
<div class="line"><a id="l02444" name="l02444"></a><span class="lineno"> 2444</span> }</div>
|
|
<div class="line"><a id="l02445" name="l02445"></a><span class="lineno"> 2445</span> <span class="keywordflow">else</span></div>
|
|
<div class="line"><a id="l02446" name="l02446"></a><span class="lineno"> 2446</span> {</div>
|
|
<div class="line"><a id="l02447" name="l02447"></a><span class="lineno"> 2447</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tmp</a>[0].<a class="code hl_function" href="structpocketfft_1_1detail_1_1cmplx.html#a647fece372b64b13c4a7e5877d09a807">Set</a>(c[0],c[0]*0);</div>
|
|
<div class="line"><a id="l02448" name="l02448"></a><span class="lineno"> 2448</span> std::copy_n (c+1, n-1, &<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tmp</a>[1].r);</div>
|
|
<div class="line"><a id="l02449" name="l02449"></a><span class="lineno"> 2449</span> <span class="keywordflow">if</span> ((n&1)==0) <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tmp</a>[n/2].<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#a35d2dce1b7de5f37d7029e639bc7f23d">i</a>=<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T0</a>(0)*c[0];</div>
|
|
<div class="line"><a id="l02450" name="l02450"></a><span class="lineno"> 2450</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a>=1; 2*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a><n; ++<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a>)</div>
|
|
<div class="line"><a id="l02451" name="l02451"></a><span class="lineno"> 2451</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tmp</a>[n-<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a>].Set(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tmp</a>[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a>].r, -<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tmp</a>[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a>].i);</div>
|
|
<div class="line"><a id="l02452" name="l02452"></a><span class="lineno"> 2452</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">fft<false></a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tmp</a>.data(),fct);</div>
|
|
<div class="line"><a id="l02453" name="l02453"></a><span class="lineno"> 2453</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a>=0; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a><n; ++<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a>)</div>
|
|
<div class="line"><a id="l02454" name="l02454"></a><span class="lineno"> 2454</span> c[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a>] = <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tmp</a>[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a>].r;</div>
|
|
<div class="line"><a id="l02455" name="l02455"></a><span class="lineno"> 2455</span> }</div>
|
|
<div class="line"><a id="l02456" name="l02456"></a><span class="lineno"> 2456</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l02457" name="l02457"></a><span class="lineno"> 2457</span> };</div>
|
|
</div>
|
|
<div class="line"><a id="l02458" name="l02458"></a><span class="lineno"> 2458</span> </div>
|
|
<div class="line"><a id="l02459" name="l02459"></a><span class="lineno"> 2459</span><span class="comment">//</span></div>
|
|
<div class="line"><a id="l02460" name="l02460"></a><span class="lineno"> 2460</span><span class="comment">// flexible (FFTPACK/Bluestein) complex 1D transform</span></div>
|
|
<div class="line"><a id="l02461" name="l02461"></a><span class="lineno"> 2461</span><span class="comment">//</span></div>
|
|
<div class="line"><a id="l02462" name="l02462"></a><span class="lineno"> 2462</span> </div>
|
|
<div class="foldopen" id="foldopen02463" data-start="{" data-end="};">
|
|
<div class="line"><a id="l02463" name="l02463"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1pocketfft__c.html"> 2463</a></span><span class="keyword">template</span><<span class="keyword">typename</span> T0> <span class="keyword">class </span><a class="code hl_class" href="classpocketfft_1_1detail_1_1pocketfft__c.html">pocketfft_c</a></div>
|
|
<div class="line"><a id="l02464" name="l02464"></a><span class="lineno"> 2464</span> {</div>
|
|
<div class="line"><a id="l02465" name="l02465"></a><span class="lineno"> 2465</span> <span class="keyword">private</span>:</div>
|
|
<div class="line"><a id="l02466" name="l02466"></a><span class="lineno"> 2466</span> std::unique_ptr<cfftp<T0>> packplan;</div>
|
|
<div class="line"><a id="l02467" name="l02467"></a><span class="lineno"> 2467</span> std::unique_ptr<fftblue<T0>> blueplan;</div>
|
|
<div class="line"><a id="l02468" name="l02468"></a><span class="lineno"> 2468</span> <span class="keywordtype">size_t</span> len;</div>
|
|
<div class="line"><a id="l02469" name="l02469"></a><span class="lineno"> 2469</span> </div>
|
|
<div class="line"><a id="l02470" name="l02470"></a><span class="lineno"> 2470</span> <span class="keyword">public</span>:</div>
|
|
<div class="foldopen" id="foldopen02471" data-start="{" data-end="}">
|
|
<div class="line"><a id="l02471" name="l02471"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1pocketfft__c.html#a44f46cf06f572b42ad8b53c7c6f38e4d"> 2471</a></span> <a class="code hl_define" href="pocketfft_8h.html#a7020984e0ca1d6e565629ca6e7c1a7e0">POCKETFFT_NOINLINE</a> <a class="code hl_function" href="classpocketfft_1_1detail_1_1pocketfft__c.html#a44f46cf06f572b42ad8b53c7c6f38e4d">pocketfft_c</a>(<span class="keywordtype">size_t</span> length)</div>
|
|
<div class="line"><a id="l02472" name="l02472"></a><span class="lineno"> 2472</span> : len(length)</div>
|
|
<div class="line"><a id="l02473" name="l02473"></a><span class="lineno"> 2473</span> {</div>
|
|
<div class="line"><a id="l02474" name="l02474"></a><span class="lineno"> 2474</span> <span class="keywordflow">if</span> (length==0) <span class="keywordflow">throw</span> std::runtime_error(<span class="stringliteral">"zero-length FFT requested"</span>);</div>
|
|
<div class="line"><a id="l02475" name="l02475"></a><span class="lineno"> 2475</span> <span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tmp</a> = (length<50) ? 0 : util::largest_prime_factor(length);</div>
|
|
<div class="line"><a id="l02476" name="l02476"></a><span class="lineno"> 2476</span> <span class="keywordflow">if</span> (<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tmp</a>*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tmp</a> <= length)</div>
|
|
<div class="line"><a id="l02477" name="l02477"></a><span class="lineno"> 2477</span> {</div>
|
|
<div class="line"><a id="l02478" name="l02478"></a><span class="lineno"> 2478</span> packplan=std::unique_ptr<cfftp<T0>>(<span class="keyword">new</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cfftp<T0></a>(length));</div>
|
|
<div class="line"><a id="l02479" name="l02479"></a><span class="lineno"> 2479</span> <span class="keywordflow">return</span>;</div>
|
|
<div class="line"><a id="l02480" name="l02480"></a><span class="lineno"> 2480</span> }</div>
|
|
<div class="line"><a id="l02481" name="l02481"></a><span class="lineno"> 2481</span> <span class="keywordtype">double</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">comp1</a> = util::cost_guess(length);</div>
|
|
<div class="line"><a id="l02482" name="l02482"></a><span class="lineno"> 2482</span> <span class="keywordtype">double</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">comp2</a> = 2*util::cost_guess(util::good_size_cmplx(2*length-1));</div>
|
|
<div class="line"><a id="l02483" name="l02483"></a><span class="lineno"> 2483</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">comp2</a>*=1.5; <span class="comment">/* fudge factor that appears to give good overall performance */</span></div>
|
|
<div class="line"><a id="l02484" name="l02484"></a><span class="lineno"> 2484</span> <span class="keywordflow">if</span> (<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">comp2</a><<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">comp1</a>) <span class="comment">// use Bluestein</span></div>
|
|
<div class="line"><a id="l02485" name="l02485"></a><span class="lineno"> 2485</span> blueplan=std::unique_ptr<fftblue<T0>>(<span class="keyword">new</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">fftblue<T0></a>(length));</div>
|
|
<div class="line"><a id="l02486" name="l02486"></a><span class="lineno"> 2486</span> <span class="keywordflow">else</span></div>
|
|
<div class="line"><a id="l02487" name="l02487"></a><span class="lineno"> 2487</span> packplan=std::unique_ptr<cfftp<T0>>(<span class="keyword">new</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cfftp<T0></a>(length));</div>
|
|
<div class="line"><a id="l02488" name="l02488"></a><span class="lineno"> 2488</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l02489" name="l02489"></a><span class="lineno"> 2489</span> </div>
|
|
<div class="foldopen" id="foldopen02490" data-start="{" data-end="}">
|
|
<div class="line"><a id="l02490" name="l02490"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1pocketfft__c.html#a436afd63e8e130f97aff103ae964a45d"> 2490</a></span> <span class="keyword">template</span><<span class="keyword">typename</span> T> <a class="code hl_define" href="pocketfft_8h.html#a7020984e0ca1d6e565629ca6e7c1a7e0">POCKETFFT_NOINLINE</a> <span class="keywordtype">void</span> <a class="code hl_function" href="classpocketfft_1_1detail_1_1pocketfft__c.html#a436afd63e8e130f97aff103ae964a45d">exec</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cmplx<T></a> c[], <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T0</a> fct, <span class="keywordtype">bool</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">fwd</a>)<span class="keyword"> const</span></div>
|
|
<div class="line"><a id="l02491" name="l02491"></a><span class="lineno"> 2491</span><span class="keyword"> </span>{ packplan ? packplan->exec(c,fct,<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">fwd</a>) : blueplan->exec(c,fct,<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">fwd</a>); }</div>
|
|
</div>
|
|
<div class="line"><a id="l02492" name="l02492"></a><span class="lineno"> 2492</span> </div>
|
|
<div class="line"><a id="l02493" name="l02493"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1pocketfft__c.html#a1fd1a2f9b3ae5ee9f00b9ca6946eb16d"> 2493</a></span> <span class="keywordtype">size_t</span> <a class="code hl_function" href="classpocketfft_1_1detail_1_1pocketfft__c.html#a1fd1a2f9b3ae5ee9f00b9ca6946eb16d">length</a>()<span class="keyword"> const </span>{ <span class="keywordflow">return</span> len; }</div>
|
|
<div class="line"><a id="l02494" name="l02494"></a><span class="lineno"> 2494</span> };</div>
|
|
</div>
|
|
<div class="line"><a id="l02495" name="l02495"></a><span class="lineno"> 2495</span> </div>
|
|
<div class="line"><a id="l02496" name="l02496"></a><span class="lineno"> 2496</span><span class="comment">//</span></div>
|
|
<div class="line"><a id="l02497" name="l02497"></a><span class="lineno"> 2497</span><span class="comment">// flexible (FFTPACK/Bluestein) real-valued 1D transform</span></div>
|
|
<div class="line"><a id="l02498" name="l02498"></a><span class="lineno"> 2498</span><span class="comment">//</span></div>
|
|
<div class="line"><a id="l02499" name="l02499"></a><span class="lineno"> 2499</span> </div>
|
|
<div class="foldopen" id="foldopen02500" data-start="{" data-end="};">
|
|
<div class="line"><a id="l02500" name="l02500"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1pocketfft__r.html"> 2500</a></span><span class="keyword">template</span><<span class="keyword">typename</span> T0> <span class="keyword">class </span><a class="code hl_class" href="classpocketfft_1_1detail_1_1pocketfft__r.html">pocketfft_r</a></div>
|
|
<div class="line"><a id="l02501" name="l02501"></a><span class="lineno"> 2501</span> {</div>
|
|
<div class="line"><a id="l02502" name="l02502"></a><span class="lineno"> 2502</span> <span class="keyword">private</span>:</div>
|
|
<div class="line"><a id="l02503" name="l02503"></a><span class="lineno"> 2503</span> std::unique_ptr<rfftp<T0>> packplan;</div>
|
|
<div class="line"><a id="l02504" name="l02504"></a><span class="lineno"> 2504</span> std::unique_ptr<fftblue<T0>> blueplan;</div>
|
|
<div class="line"><a id="l02505" name="l02505"></a><span class="lineno"> 2505</span> <span class="keywordtype">size_t</span> len;</div>
|
|
<div class="line"><a id="l02506" name="l02506"></a><span class="lineno"> 2506</span> </div>
|
|
<div class="line"><a id="l02507" name="l02507"></a><span class="lineno"> 2507</span> <span class="keyword">public</span>:</div>
|
|
<div class="foldopen" id="foldopen02508" data-start="{" data-end="}">
|
|
<div class="line"><a id="l02508" name="l02508"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1pocketfft__r.html#a60e9b3d1f9b20ec5b86d46b0398f6f7d"> 2508</a></span> <a class="code hl_define" href="pocketfft_8h.html#a7020984e0ca1d6e565629ca6e7c1a7e0">POCKETFFT_NOINLINE</a> <a class="code hl_function" href="classpocketfft_1_1detail_1_1pocketfft__r.html#a60e9b3d1f9b20ec5b86d46b0398f6f7d">pocketfft_r</a>(<span class="keywordtype">size_t</span> length)</div>
|
|
<div class="line"><a id="l02509" name="l02509"></a><span class="lineno"> 2509</span> : len(length)</div>
|
|
<div class="line"><a id="l02510" name="l02510"></a><span class="lineno"> 2510</span> {</div>
|
|
<div class="line"><a id="l02511" name="l02511"></a><span class="lineno"> 2511</span> <span class="keywordflow">if</span> (length==0) <span class="keywordflow">throw</span> std::runtime_error(<span class="stringliteral">"zero-length FFT requested"</span>);</div>
|
|
<div class="line"><a id="l02512" name="l02512"></a><span class="lineno"> 2512</span> <span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tmp</a> = (length<50) ? 0 : util::largest_prime_factor(length);</div>
|
|
<div class="line"><a id="l02513" name="l02513"></a><span class="lineno"> 2513</span> <span class="keywordflow">if</span> (<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tmp</a>*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tmp</a> <= length)</div>
|
|
<div class="line"><a id="l02514" name="l02514"></a><span class="lineno"> 2514</span> {</div>
|
|
<div class="line"><a id="l02515" name="l02515"></a><span class="lineno"> 2515</span> packplan=std::unique_ptr<rfftp<T0>>(<span class="keyword">new</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">rfftp<T0></a>(length));</div>
|
|
<div class="line"><a id="l02516" name="l02516"></a><span class="lineno"> 2516</span> <span class="keywordflow">return</span>;</div>
|
|
<div class="line"><a id="l02517" name="l02517"></a><span class="lineno"> 2517</span> }</div>
|
|
<div class="line"><a id="l02518" name="l02518"></a><span class="lineno"> 2518</span> <span class="keywordtype">double</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">comp1</a> = 0.5*util::cost_guess(length);</div>
|
|
<div class="line"><a id="l02519" name="l02519"></a><span class="lineno"> 2519</span> <span class="keywordtype">double</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">comp2</a> = 2*util::cost_guess(util::good_size_cmplx(2*length-1));</div>
|
|
<div class="line"><a id="l02520" name="l02520"></a><span class="lineno"> 2520</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">comp2</a>*=1.5; <span class="comment">/* fudge factor that appears to give good overall performance */</span></div>
|
|
<div class="line"><a id="l02521" name="l02521"></a><span class="lineno"> 2521</span> <span class="keywordflow">if</span> (<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">comp2</a><<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">comp1</a>) <span class="comment">// use Bluestein</span></div>
|
|
<div class="line"><a id="l02522" name="l02522"></a><span class="lineno"> 2522</span> blueplan=std::unique_ptr<fftblue<T0>>(<span class="keyword">new</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">fftblue<T0></a>(length));</div>
|
|
<div class="line"><a id="l02523" name="l02523"></a><span class="lineno"> 2523</span> <span class="keywordflow">else</span></div>
|
|
<div class="line"><a id="l02524" name="l02524"></a><span class="lineno"> 2524</span> packplan=std::unique_ptr<rfftp<T0>>(<span class="keyword">new</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">rfftp<T0></a>(length));</div>
|
|
<div class="line"><a id="l02525" name="l02525"></a><span class="lineno"> 2525</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l02526" name="l02526"></a><span class="lineno"> 2526</span> </div>
|
|
<div class="foldopen" id="foldopen02527" data-start="{" data-end="}">
|
|
<div class="line"><a id="l02527" name="l02527"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1pocketfft__r.html#a2815bc8aa04fa986834b02e502f98b33"> 2527</a></span> <span class="keyword">template</span><<span class="keyword">typename</span> T> <a class="code hl_define" href="pocketfft_8h.html#a7020984e0ca1d6e565629ca6e7c1a7e0">POCKETFFT_NOINLINE</a> <span class="keywordtype">void</span> <a class="code hl_function" href="classpocketfft_1_1detail_1_1pocketfft__r.html#a2815bc8aa04fa986834b02e502f98b33">exec</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a> c[], <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T0</a> fct, <span class="keywordtype">bool</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">fwd</a>)<span class="keyword"> const</span></div>
|
|
<div class="line"><a id="l02528" name="l02528"></a><span class="lineno"> 2528</span><span class="keyword"> </span>{ packplan ? packplan->exec(c,fct,<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">fwd</a>) : blueplan->exec_r(c,fct,<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">fwd</a>); }</div>
|
|
</div>
|
|
<div class="line"><a id="l02529" name="l02529"></a><span class="lineno"> 2529</span> </div>
|
|
<div class="line"><a id="l02530" name="l02530"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1pocketfft__r.html#a83222fdbf81a7c6d560e0841cdfca8c6"> 2530</a></span> <span class="keywordtype">size_t</span> <a class="code hl_function" href="classpocketfft_1_1detail_1_1pocketfft__r.html#a83222fdbf81a7c6d560e0841cdfca8c6">length</a>()<span class="keyword"> const </span>{ <span class="keywordflow">return</span> len; }</div>
|
|
<div class="line"><a id="l02531" name="l02531"></a><span class="lineno"> 2531</span> };</div>
|
|
</div>
|
|
<div class="line"><a id="l02532" name="l02532"></a><span class="lineno"> 2532</span> </div>
|
|
<div class="line"><a id="l02533" name="l02533"></a><span class="lineno"> 2533</span> </div>
|
|
<div class="line"><a id="l02534" name="l02534"></a><span class="lineno"> 2534</span><span class="comment">//</span></div>
|
|
<div class="line"><a id="l02535" name="l02535"></a><span class="lineno"> 2535</span><span class="comment">// sine/cosine transforms</span></div>
|
|
<div class="line"><a id="l02536" name="l02536"></a><span class="lineno"> 2536</span><span class="comment">//</span></div>
|
|
<div class="line"><a id="l02537" name="l02537"></a><span class="lineno"> 2537</span> </div>
|
|
<div class="foldopen" id="foldopen02538" data-start="{" data-end="};">
|
|
<div class="line"><a id="l02538" name="l02538"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1_t__dct1.html"> 2538</a></span><span class="keyword">template</span><<span class="keyword">typename</span> T0> <span class="keyword">class </span><a class="code hl_class" href="classpocketfft_1_1detail_1_1_t__dct1.html">T_dct1</a></div>
|
|
<div class="line"><a id="l02539" name="l02539"></a><span class="lineno"> 2539</span> {</div>
|
|
<div class="line"><a id="l02540" name="l02540"></a><span class="lineno"> 2540</span> <span class="keyword">private</span>:</div>
|
|
<div class="line"><a id="l02541" name="l02541"></a><span class="lineno"> 2541</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">pocketfft_r<T0></a> fftplan;</div>
|
|
<div class="line"><a id="l02542" name="l02542"></a><span class="lineno"> 2542</span> </div>
|
|
<div class="line"><a id="l02543" name="l02543"></a><span class="lineno"> 2543</span> <span class="keyword">public</span>:</div>
|
|
<div class="foldopen" id="foldopen02544" data-start="{" data-end="}">
|
|
<div class="line"><a id="l02544" name="l02544"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1_t__dct1.html#ac153e82c8bf0943638a88486e35c16d6"> 2544</a></span> <a class="code hl_define" href="pocketfft_8h.html#a7020984e0ca1d6e565629ca6e7c1a7e0">POCKETFFT_NOINLINE</a> <a class="code hl_function" href="classpocketfft_1_1detail_1_1_t__dct1.html#ac153e82c8bf0943638a88486e35c16d6">T_dct1</a>(<span class="keywordtype">size_t</span> length)</div>
|
|
<div class="line"><a id="l02545" name="l02545"></a><span class="lineno"> 2545</span> : fftplan(2*(length-1)) {}</div>
|
|
</div>
|
|
<div class="line"><a id="l02546" name="l02546"></a><span class="lineno"> 2546</span> </div>
|
|
<div class="foldopen" id="foldopen02547" data-start="{" data-end="}">
|
|
<div class="line"><a id="l02547" name="l02547"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1_t__dct1.html#a7736111ff9d220f983e41a6fecd5f058"> 2547</a></span> <span class="keyword">template</span><<span class="keyword">typename</span> T> <a class="code hl_define" href="pocketfft_8h.html#a7020984e0ca1d6e565629ca6e7c1a7e0">POCKETFFT_NOINLINE</a> <span class="keywordtype">void</span> <a class="code hl_function" href="classpocketfft_1_1detail_1_1_t__dct1.html#a7736111ff9d220f983e41a6fecd5f058">exec</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a> c[], <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T0</a> fct, <span class="keywordtype">bool</span> ortho,</div>
|
|
<div class="line"><a id="l02548" name="l02548"></a><span class="lineno"> 2548</span> <span class="keywordtype">int</span> <span class="comment">/*type*/</span>, <span class="keywordtype">bool</span> <span class="comment">/*cosine*/</span>)<span class="keyword"> const</span></div>
|
|
<div class="line"><a id="l02549" name="l02549"></a><span class="lineno"> 2549</span><span class="keyword"> </span>{</div>
|
|
<div class="line"><a id="l02550" name="l02550"></a><span class="lineno"> 2550</span> <span class="keyword">constexpr</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T0</a> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">sqrt2</a>=<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T0</a>(1.414213562373095048801688724209698L);</div>
|
|
<div class="line"><a id="l02551" name="l02551"></a><span class="lineno"> 2551</span> <span class="keywordtype">size_t</span> N=fftplan.length(), n=N/2+1;</div>
|
|
<div class="line"><a id="l02552" name="l02552"></a><span class="lineno"> 2552</span> <span class="keywordflow">if</span> (ortho)</div>
|
|
<div class="line"><a id="l02553" name="l02553"></a><span class="lineno"> 2553</span> { c[0]*=<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">sqrt2</a>; c[n-1]*=<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">sqrt2</a>; }</div>
|
|
<div class="line"><a id="l02554" name="l02554"></a><span class="lineno"> 2554</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">arr<T></a> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tmp</a>(N);</div>
|
|
<div class="line"><a id="l02555" name="l02555"></a><span class="lineno"> 2555</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tmp</a>[0] = c[0];</div>
|
|
<div class="line"><a id="l02556" name="l02556"></a><span class="lineno"> 2556</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i=1; i<n; ++i)</div>
|
|
<div class="line"><a id="l02557" name="l02557"></a><span class="lineno"> 2557</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tmp</a>[i] = <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tmp</a>[N-i] = c[i];</div>
|
|
<div class="line"><a id="l02558" name="l02558"></a><span class="lineno"> 2558</span> fftplan.exec(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tmp</a>.data(), fct, <span class="keyword">true</span>);</div>
|
|
<div class="line"><a id="l02559" name="l02559"></a><span class="lineno"> 2559</span> c[0] = <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tmp</a>[0];</div>
|
|
<div class="line"><a id="l02560" name="l02560"></a><span class="lineno"> 2560</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i=1; i<n; ++i)</div>
|
|
<div class="line"><a id="l02561" name="l02561"></a><span class="lineno"> 2561</span> c[i] = <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tmp</a>[2*i-1];</div>
|
|
<div class="line"><a id="l02562" name="l02562"></a><span class="lineno"> 2562</span> <span class="keywordflow">if</span> (ortho)</div>
|
|
<div class="line"><a id="l02563" name="l02563"></a><span class="lineno"> 2563</span> { c[0]*=<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">sqrt2</a>*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T0</a>(0.5); c[n-1]*=<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">sqrt2</a>*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T0</a>(0.5); }</div>
|
|
<div class="line"><a id="l02564" name="l02564"></a><span class="lineno"> 2564</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l02565" name="l02565"></a><span class="lineno"> 2565</span> </div>
|
|
<div class="line"><a id="l02566" name="l02566"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1_t__dct1.html#ac7a04c91d507bd8f173d2266bb5bb168"> 2566</a></span> <span class="keywordtype">size_t</span> <a class="code hl_function" href="classpocketfft_1_1detail_1_1_t__dct1.html#ac7a04c91d507bd8f173d2266bb5bb168">length</a>()<span class="keyword"> const </span>{ <span class="keywordflow">return</span> fftplan.length()/2+1; }</div>
|
|
<div class="line"><a id="l02567" name="l02567"></a><span class="lineno"> 2567</span> };</div>
|
|
</div>
|
|
<div class="line"><a id="l02568" name="l02568"></a><span class="lineno"> 2568</span> </div>
|
|
<div class="foldopen" id="foldopen02569" data-start="{" data-end="};">
|
|
<div class="line"><a id="l02569" name="l02569"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1_t__dst1.html"> 2569</a></span><span class="keyword">template</span><<span class="keyword">typename</span> T0> <span class="keyword">class </span><a class="code hl_class" href="classpocketfft_1_1detail_1_1_t__dst1.html">T_dst1</a></div>
|
|
<div class="line"><a id="l02570" name="l02570"></a><span class="lineno"> 2570</span> {</div>
|
|
<div class="line"><a id="l02571" name="l02571"></a><span class="lineno"> 2571</span> <span class="keyword">private</span>:</div>
|
|
<div class="line"><a id="l02572" name="l02572"></a><span class="lineno"> 2572</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">pocketfft_r<T0></a> fftplan;</div>
|
|
<div class="line"><a id="l02573" name="l02573"></a><span class="lineno"> 2573</span> </div>
|
|
<div class="line"><a id="l02574" name="l02574"></a><span class="lineno"> 2574</span> <span class="keyword">public</span>:</div>
|
|
<div class="foldopen" id="foldopen02575" data-start="{" data-end="}">
|
|
<div class="line"><a id="l02575" name="l02575"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1_t__dst1.html#ad0b6647266ef0e87d34730774b92da84"> 2575</a></span> <a class="code hl_define" href="pocketfft_8h.html#a7020984e0ca1d6e565629ca6e7c1a7e0">POCKETFFT_NOINLINE</a> <a class="code hl_function" href="classpocketfft_1_1detail_1_1_t__dst1.html#ad0b6647266ef0e87d34730774b92da84">T_dst1</a>(<span class="keywordtype">size_t</span> length)</div>
|
|
<div class="line"><a id="l02576" name="l02576"></a><span class="lineno"> 2576</span> : fftplan(2*(length+1)) {}</div>
|
|
</div>
|
|
<div class="line"><a id="l02577" name="l02577"></a><span class="lineno"> 2577</span> </div>
|
|
<div class="foldopen" id="foldopen02578" data-start="{" data-end="}">
|
|
<div class="line"><a id="l02578" name="l02578"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1_t__dst1.html#a598a9511004263eb3610053d7efc9e26"> 2578</a></span> <span class="keyword">template</span><<span class="keyword">typename</span> T> <a class="code hl_define" href="pocketfft_8h.html#a7020984e0ca1d6e565629ca6e7c1a7e0">POCKETFFT_NOINLINE</a> <span class="keywordtype">void</span> <a class="code hl_function" href="classpocketfft_1_1detail_1_1_t__dst1.html#a598a9511004263eb3610053d7efc9e26">exec</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a> c[], <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T0</a> fct,</div>
|
|
<div class="line"><a id="l02579" name="l02579"></a><span class="lineno"> 2579</span> <span class="keywordtype">bool</span> <span class="comment">/*ortho*/</span>, <span class="keywordtype">int</span> <span class="comment">/*type*/</span>, <span class="keywordtype">bool</span> <span class="comment">/*cosine*/</span>)<span class="keyword"> const</span></div>
|
|
<div class="line"><a id="l02580" name="l02580"></a><span class="lineno"> 2580</span><span class="keyword"> </span>{</div>
|
|
<div class="line"><a id="l02581" name="l02581"></a><span class="lineno"> 2581</span> <span class="keywordtype">size_t</span> N=fftplan.length(), n=N/2-1;</div>
|
|
<div class="line"><a id="l02582" name="l02582"></a><span class="lineno"> 2582</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">arr<T></a> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tmp</a>(N);</div>
|
|
<div class="line"><a id="l02583" name="l02583"></a><span class="lineno"> 2583</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tmp</a>[0] = <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tmp</a>[n+1] = c[0]*0;</div>
|
|
<div class="line"><a id="l02584" name="l02584"></a><span class="lineno"> 2584</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i=0; i<n; ++i)</div>
|
|
<div class="line"><a id="l02585" name="l02585"></a><span class="lineno"> 2585</span> { <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tmp</a>[i+1]=c[i]; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tmp</a>[N-1-i]=-c[i]; }</div>
|
|
<div class="line"><a id="l02586" name="l02586"></a><span class="lineno"> 2586</span> fftplan.exec(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tmp</a>.data(), fct, <span class="keyword">true</span>);</div>
|
|
<div class="line"><a id="l02587" name="l02587"></a><span class="lineno"> 2587</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i=0; i<n; ++i)</div>
|
|
<div class="line"><a id="l02588" name="l02588"></a><span class="lineno"> 2588</span> c[i] = -<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tmp</a>[2*i+2];</div>
|
|
<div class="line"><a id="l02589" name="l02589"></a><span class="lineno"> 2589</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l02590" name="l02590"></a><span class="lineno"> 2590</span> </div>
|
|
<div class="line"><a id="l02591" name="l02591"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1_t__dst1.html#ab205d901650e38b592ff860b7978fa3e"> 2591</a></span> <span class="keywordtype">size_t</span> <a class="code hl_function" href="classpocketfft_1_1detail_1_1_t__dst1.html#ab205d901650e38b592ff860b7978fa3e">length</a>()<span class="keyword"> const </span>{ <span class="keywordflow">return</span> fftplan.length()/2-1; }</div>
|
|
<div class="line"><a id="l02592" name="l02592"></a><span class="lineno"> 2592</span> };</div>
|
|
</div>
|
|
<div class="line"><a id="l02593" name="l02593"></a><span class="lineno"> 2593</span> </div>
|
|
<div class="foldopen" id="foldopen02594" data-start="{" data-end="};">
|
|
<div class="line"><a id="l02594" name="l02594"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1_t__dcst23.html"> 2594</a></span><span class="keyword">template</span><<span class="keyword">typename</span> T0> <span class="keyword">class </span><a class="code hl_class" href="classpocketfft_1_1detail_1_1_t__dcst23.html">T_dcst23</a></div>
|
|
<div class="line"><a id="l02595" name="l02595"></a><span class="lineno"> 2595</span> {</div>
|
|
<div class="line"><a id="l02596" name="l02596"></a><span class="lineno"> 2596</span> <span class="keyword">private</span>:</div>
|
|
<div class="line"><a id="l02597" name="l02597"></a><span class="lineno"> 2597</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">pocketfft_r<T0></a> fftplan;</div>
|
|
<div class="line"><a id="l02598" name="l02598"></a><span class="lineno"> 2598</span> std::vector<T0> twiddle;</div>
|
|
<div class="line"><a id="l02599" name="l02599"></a><span class="lineno"> 2599</span> </div>
|
|
<div class="line"><a id="l02600" name="l02600"></a><span class="lineno"> 2600</span> <span class="keyword">public</span>:</div>
|
|
<div class="foldopen" id="foldopen02601" data-start="{" data-end="}">
|
|
<div class="line"><a id="l02601" name="l02601"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1_t__dcst23.html#a7cbe3fd1b64383d8b466bcdc2f594900"> 2601</a></span> <a class="code hl_define" href="pocketfft_8h.html#a7020984e0ca1d6e565629ca6e7c1a7e0">POCKETFFT_NOINLINE</a> <a class="code hl_function" href="classpocketfft_1_1detail_1_1_t__dcst23.html#a7cbe3fd1b64383d8b466bcdc2f594900">T_dcst23</a>(<span class="keywordtype">size_t</span> length)</div>
|
|
<div class="line"><a id="l02602" name="l02602"></a><span class="lineno"> 2602</span> : fftplan(length), twiddle(length)</div>
|
|
<div class="line"><a id="l02603" name="l02603"></a><span class="lineno"> 2603</span> {</div>
|
|
<div class="line"><a id="l02604" name="l02604"></a><span class="lineno"> 2604</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">sincos_2pibyn<T0></a> tw(4*length);</div>
|
|
<div class="line"><a id="l02605" name="l02605"></a><span class="lineno"> 2605</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i=0; i<length; ++i)</div>
|
|
<div class="line"><a id="l02606" name="l02606"></a><span class="lineno"> 2606</span> twiddle[i] = tw[i+1].r;</div>
|
|
<div class="line"><a id="l02607" name="l02607"></a><span class="lineno"> 2607</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l02608" name="l02608"></a><span class="lineno"> 2608</span> </div>
|
|
<div class="foldopen" id="foldopen02609" data-start="{" data-end="}">
|
|
<div class="line"><a id="l02609" name="l02609"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1_t__dcst23.html#a2a45b7b4612904c2be69c01f6d5029ac"> 2609</a></span> <span class="keyword">template</span><<span class="keyword">typename</span> T> <a class="code hl_define" href="pocketfft_8h.html#a7020984e0ca1d6e565629ca6e7c1a7e0">POCKETFFT_NOINLINE</a> <span class="keywordtype">void</span> <a class="code hl_function" href="classpocketfft_1_1detail_1_1_t__dcst23.html#a2a45b7b4612904c2be69c01f6d5029ac">exec</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a> c[], <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T0</a> fct, <span class="keywordtype">bool</span> ortho,</div>
|
|
<div class="line"><a id="l02610" name="l02610"></a><span class="lineno"> 2610</span> <span class="keywordtype">int</span> type, <span class="keywordtype">bool</span> cosine)<span class="keyword"> const</span></div>
|
|
<div class="line"><a id="l02611" name="l02611"></a><span class="lineno"> 2611</span><span class="keyword"> </span>{</div>
|
|
<div class="line"><a id="l02612" name="l02612"></a><span class="lineno"> 2612</span> <span class="keyword">constexpr</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T0</a> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">sqrt2</a>=<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T0</a>(1.414213562373095048801688724209698L);</div>
|
|
<div class="line"><a id="l02613" name="l02613"></a><span class="lineno"> 2613</span> <span class="keywordtype">size_t</span> N=length();</div>
|
|
<div class="line"><a id="l02614" name="l02614"></a><span class="lineno"> 2614</span> <span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">NS2</a> = (N+1)/2;</div>
|
|
<div class="line"><a id="l02615" name="l02615"></a><span class="lineno"> 2615</span> <span class="keywordflow">if</span> (type==2)</div>
|
|
<div class="line"><a id="l02616" name="l02616"></a><span class="lineno"> 2616</span> {</div>
|
|
<div class="line"><a id="l02617" name="l02617"></a><span class="lineno"> 2617</span> <span class="keywordflow">if</span> (!cosine)</div>
|
|
<div class="line"><a id="l02618" name="l02618"></a><span class="lineno"> 2618</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>=1; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a><N; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>+=2)</div>
|
|
<div class="line"><a id="l02619" name="l02619"></a><span class="lineno"> 2619</span> c[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>] = -c[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>];</div>
|
|
<div class="line"><a id="l02620" name="l02620"></a><span class="lineno"> 2620</span> c[0] *= 2;</div>
|
|
<div class="line"><a id="l02621" name="l02621"></a><span class="lineno"> 2621</span> <span class="keywordflow">if</span> ((N&1)==0) c[N-1]*=2;</div>
|
|
<div class="line"><a id="l02622" name="l02622"></a><span class="lineno"> 2622</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>=1; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a><N-1; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>+=2)</div>
|
|
<div class="line"><a id="l02623" name="l02623"></a><span class="lineno"> 2623</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#af5eedf3cdfc83c0a30807092c39a9ce2">MPINPLACE</a>(c[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>+1], c[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>]);</div>
|
|
<div class="line"><a id="l02624" name="l02624"></a><span class="lineno"> 2624</span> fftplan.exec(c, fct, <span class="keyword">false</span>);</div>
|
|
<div class="line"><a id="l02625" name="l02625"></a><span class="lineno"> 2625</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>=1, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">kc</a>=N-1; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a><<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">NS2</a>; ++<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>, --<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">kc</a>)</div>
|
|
<div class="line"><a id="l02626" name="l02626"></a><span class="lineno"> 2626</span> {</div>
|
|
<div class="line"><a id="l02627" name="l02627"></a><span class="lineno"> 2627</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">t1</a> = twiddle[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>-1]*c[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">kc</a>]+twiddle[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">kc</a>-1]*c[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>];</div>
|
|
<div class="line"><a id="l02628" name="l02628"></a><span class="lineno"> 2628</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">t2</a> = twiddle[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>-1]*c[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>]-twiddle[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">kc</a>-1]*c[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">kc</a>];</div>
|
|
<div class="line"><a id="l02629" name="l02629"></a><span class="lineno"> 2629</span> c[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>] = <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T0</a>(0.5)*(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">t1</a>+<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">t2</a>); c[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">kc</a>]=<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T0</a>(0.5)*(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">t1</a>-<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">t2</a>);</div>
|
|
<div class="line"><a id="l02630" name="l02630"></a><span class="lineno"> 2630</span> }</div>
|
|
<div class="line"><a id="l02631" name="l02631"></a><span class="lineno"> 2631</span> <span class="keywordflow">if</span> ((N&1)==0)</div>
|
|
<div class="line"><a id="l02632" name="l02632"></a><span class="lineno"> 2632</span> c[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">NS2</a>] *= twiddle[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">NS2</a>-1];</div>
|
|
<div class="line"><a id="l02633" name="l02633"></a><span class="lineno"> 2633</span> <span class="keywordflow">if</span> (!cosine)</div>
|
|
<div class="line"><a id="l02634" name="l02634"></a><span class="lineno"> 2634</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>=0, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">kc</a>=N-1; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a><<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">kc</a>; ++<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>, --<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">kc</a>)</div>
|
|
<div class="line"><a id="l02635" name="l02635"></a><span class="lineno"> 2635</span> std::swap(c[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>], c[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">kc</a>]);</div>
|
|
<div class="line"><a id="l02636" name="l02636"></a><span class="lineno"> 2636</span> <span class="keywordflow">if</span> (ortho) c[0]*=<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">sqrt2</a>*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T0</a>(0.5);</div>
|
|
<div class="line"><a id="l02637" name="l02637"></a><span class="lineno"> 2637</span> }</div>
|
|
<div class="line"><a id="l02638" name="l02638"></a><span class="lineno"> 2638</span> <span class="keywordflow">else</span></div>
|
|
<div class="line"><a id="l02639" name="l02639"></a><span class="lineno"> 2639</span> {</div>
|
|
<div class="line"><a id="l02640" name="l02640"></a><span class="lineno"> 2640</span> <span class="keywordflow">if</span> (ortho) c[0]*=<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">sqrt2</a>;</div>
|
|
<div class="line"><a id="l02641" name="l02641"></a><span class="lineno"> 2641</span> <span class="keywordflow">if</span> (!cosine)</div>
|
|
<div class="line"><a id="l02642" name="l02642"></a><span class="lineno"> 2642</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>=0, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">kc</a>=N-1; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a><<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">NS2</a>; ++<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>, --<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">kc</a>)</div>
|
|
<div class="line"><a id="l02643" name="l02643"></a><span class="lineno"> 2643</span> std::swap(c[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>], c[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">kc</a>]);</div>
|
|
<div class="line"><a id="l02644" name="l02644"></a><span class="lineno"> 2644</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>=1, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">kc</a>=N-1; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a><<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">NS2</a>; ++<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>, --<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">kc</a>)</div>
|
|
<div class="line"><a id="l02645" name="l02645"></a><span class="lineno"> 2645</span> {</div>
|
|
<div class="line"><a id="l02646" name="l02646"></a><span class="lineno"> 2646</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">t1</a>=c[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>]+c[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">kc</a>], <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">t2</a>=c[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>]-c[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">kc</a>];</div>
|
|
<div class="line"><a id="l02647" name="l02647"></a><span class="lineno"> 2647</span> c[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>] = twiddle[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>-1]*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">t2</a>+twiddle[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">kc</a>-1]*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">t1</a>;</div>
|
|
<div class="line"><a id="l02648" name="l02648"></a><span class="lineno"> 2648</span> c[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">kc</a>]= twiddle[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>-1]*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">t1</a>-twiddle[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">kc</a>-1]*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">t2</a>;</div>
|
|
<div class="line"><a id="l02649" name="l02649"></a><span class="lineno"> 2649</span> }</div>
|
|
<div class="line"><a id="l02650" name="l02650"></a><span class="lineno"> 2650</span> <span class="keywordflow">if</span> ((N&1)==0)</div>
|
|
<div class="line"><a id="l02651" name="l02651"></a><span class="lineno"> 2651</span> c[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">NS2</a>] *= 2*twiddle[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">NS2</a>-1];</div>
|
|
<div class="line"><a id="l02652" name="l02652"></a><span class="lineno"> 2652</span> fftplan.exec(c, fct, <span class="keyword">true</span>);</div>
|
|
<div class="line"><a id="l02653" name="l02653"></a><span class="lineno"> 2653</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>=1; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a><N-1; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>+=2)</div>
|
|
<div class="line"><a id="l02654" name="l02654"></a><span class="lineno"> 2654</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#af5eedf3cdfc83c0a30807092c39a9ce2">MPINPLACE</a>(c[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>], c[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>+1]);</div>
|
|
<div class="line"><a id="l02655" name="l02655"></a><span class="lineno"> 2655</span> <span class="keywordflow">if</span> (!cosine)</div>
|
|
<div class="line"><a id="l02656" name="l02656"></a><span class="lineno"> 2656</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>=1; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a><N; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>+=2)</div>
|
|
<div class="line"><a id="l02657" name="l02657"></a><span class="lineno"> 2657</span> c[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>] = -c[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>];</div>
|
|
<div class="line"><a id="l02658" name="l02658"></a><span class="lineno"> 2658</span> }</div>
|
|
<div class="line"><a id="l02659" name="l02659"></a><span class="lineno"> 2659</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l02660" name="l02660"></a><span class="lineno"> 2660</span> </div>
|
|
<div class="line"><a id="l02661" name="l02661"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1_t__dcst23.html#a6dab012b487ff98d11b8a9418653a478"> 2661</a></span> <span class="keywordtype">size_t</span> <a class="code hl_function" href="classpocketfft_1_1detail_1_1_t__dcst23.html#a6dab012b487ff98d11b8a9418653a478">length</a>()<span class="keyword"> const </span>{ <span class="keywordflow">return</span> fftplan.length(); }</div>
|
|
<div class="line"><a id="l02662" name="l02662"></a><span class="lineno"> 2662</span> };</div>
|
|
</div>
|
|
<div class="line"><a id="l02663" name="l02663"></a><span class="lineno"> 2663</span> </div>
|
|
<div class="foldopen" id="foldopen02664" data-start="{" data-end="};">
|
|
<div class="line"><a id="l02664" name="l02664"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1_t__dcst4.html"> 2664</a></span><span class="keyword">template</span><<span class="keyword">typename</span> T0> <span class="keyword">class </span><a class="code hl_class" href="classpocketfft_1_1detail_1_1_t__dcst4.html">T_dcst4</a></div>
|
|
<div class="line"><a id="l02665" name="l02665"></a><span class="lineno"> 2665</span> {</div>
|
|
<div class="line"><a id="l02666" name="l02666"></a><span class="lineno"> 2666</span> <span class="keyword">private</span>:</div>
|
|
<div class="line"><a id="l02667" name="l02667"></a><span class="lineno"> 2667</span> <span class="keywordtype">size_t</span> N;</div>
|
|
<div class="line"><a id="l02668" name="l02668"></a><span class="lineno"> 2668</span> std::unique_ptr<pocketfft_c<T0>> <a class="code hl_function" href="backend_2metal_2kernels_2fft_8h.html#a4010b0e151e5f01e610e9c32234458c7">fft</a>;</div>
|
|
<div class="line"><a id="l02669" name="l02669"></a><span class="lineno"> 2669</span> std::unique_ptr<pocketfft_r<T0>> rfft;</div>
|
|
<div class="line"><a id="l02670" name="l02670"></a><span class="lineno"> 2670</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">arr<cmplx<T0></a>> C2;</div>
|
|
<div class="line"><a id="l02671" name="l02671"></a><span class="lineno"> 2671</span> </div>
|
|
<div class="line"><a id="l02672" name="l02672"></a><span class="lineno"> 2672</span> <span class="keyword">public</span>:</div>
|
|
<div class="foldopen" id="foldopen02673" data-start="{" data-end="}">
|
|
<div class="line"><a id="l02673" name="l02673"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1_t__dcst4.html#a4a623ebc5dfe8c3cad153c642663099b"> 2673</a></span> <a class="code hl_define" href="pocketfft_8h.html#a7020984e0ca1d6e565629ca6e7c1a7e0">POCKETFFT_NOINLINE</a> <a class="code hl_function" href="classpocketfft_1_1detail_1_1_t__dcst4.html#a4a623ebc5dfe8c3cad153c642663099b">T_dcst4</a>(<span class="keywordtype">size_t</span> length)</div>
|
|
<div class="line"><a id="l02674" name="l02674"></a><span class="lineno"> 2674</span> : N(length),</div>
|
|
<div class="line"><a id="l02675" name="l02675"></a><span class="lineno"> 2675</span> <a class="code hl_function" href="backend_2metal_2kernels_2fft_8h.html#a4010b0e151e5f01e610e9c32234458c7">fft</a>((N&1) ? <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nullptr</a> : <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">new</a> <a class="code hl_class" href="classpocketfft_1_1detail_1_1pocketfft__c.html">pocketfft_c</a><<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T0</a>>(N/2)),</div>
|
|
<div class="line"><a id="l02676" name="l02676"></a><span class="lineno"> 2676</span> rfft((N&1)? <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">new</a> <a class="code hl_class" href="classpocketfft_1_1detail_1_1pocketfft__r.html">pocketfft_r</a><<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T0</a>>(N) : <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nullptr</a>),</div>
|
|
<div class="line"><a id="l02677" name="l02677"></a><span class="lineno"> 2677</span> C2((N&1) ? 0 : N/2)</div>
|
|
<div class="line"><a id="l02678" name="l02678"></a><span class="lineno"> 2678</span> {</div>
|
|
<div class="line"><a id="l02679" name="l02679"></a><span class="lineno"> 2679</span> <span class="keywordflow">if</span> ((N&1)==0)</div>
|
|
<div class="line"><a id="l02680" name="l02680"></a><span class="lineno"> 2680</span> {</div>
|
|
<div class="line"><a id="l02681" name="l02681"></a><span class="lineno"> 2681</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">sincos_2pibyn<T0></a> tw(16*N);</div>
|
|
<div class="line"><a id="l02682" name="l02682"></a><span class="lineno"> 2682</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i=0; i<N/2; ++i)</div>
|
|
<div class="line"><a id="l02683" name="l02683"></a><span class="lineno"> 2683</span> C2[i] = <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a66d79051d502046a9b9f103e744dbad3">conj</a>(tw[8*i+1]);</div>
|
|
<div class="line"><a id="l02684" name="l02684"></a><span class="lineno"> 2684</span> }</div>
|
|
<div class="line"><a id="l02685" name="l02685"></a><span class="lineno"> 2685</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l02686" name="l02686"></a><span class="lineno"> 2686</span> </div>
|
|
<div class="foldopen" id="foldopen02687" data-start="{" data-end="}">
|
|
<div class="line"><a id="l02687" name="l02687"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1_t__dcst4.html#af794ebf21009d5f918681188081df708"> 2687</a></span> <span class="keyword">template</span><<span class="keyword">typename</span> T> <a class="code hl_define" href="pocketfft_8h.html#a7020984e0ca1d6e565629ca6e7c1a7e0">POCKETFFT_NOINLINE</a> <span class="keywordtype">void</span> <a class="code hl_function" href="classpocketfft_1_1detail_1_1_t__dcst4.html#af794ebf21009d5f918681188081df708">exec</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a> c[], <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T0</a> fct,</div>
|
|
<div class="line"><a id="l02688" name="l02688"></a><span class="lineno"> 2688</span> <span class="keywordtype">bool</span> <span class="comment">/*ortho*/</span>, <span class="keywordtype">int</span> <span class="comment">/*type*/</span>, <span class="keywordtype">bool</span> cosine)<span class="keyword"> const</span></div>
|
|
<div class="line"><a id="l02689" name="l02689"></a><span class="lineno"> 2689</span><span class="keyword"> </span>{</div>
|
|
<div class="line"><a id="l02690" name="l02690"></a><span class="lineno"> 2690</span> <span class="keywordtype">size_t</span> n2 = N/2;</div>
|
|
<div class="line"><a id="l02691" name="l02691"></a><span class="lineno"> 2691</span> <span class="keywordflow">if</span> (!cosine)</div>
|
|
<div class="line"><a id="l02692" name="l02692"></a><span class="lineno"> 2692</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>=0, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">kc</a>=N-1; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a><n2; ++<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>, --<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">kc</a>)</div>
|
|
<div class="line"><a id="l02693" name="l02693"></a><span class="lineno"> 2693</span> std::swap(c[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>], c[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">kc</a>]);</div>
|
|
<div class="line"><a id="l02694" name="l02694"></a><span class="lineno"> 2694</span> <span class="keywordflow">if</span> (N&1)</div>
|
|
<div class="line"><a id="l02695" name="l02695"></a><span class="lineno"> 2695</span> {</div>
|
|
<div class="line"><a id="l02696" name="l02696"></a><span class="lineno"> 2696</span> <span class="comment">// The following code is derived from the FFTW3 function apply_re11()</span></div>
|
|
<div class="line"><a id="l02697" name="l02697"></a><span class="lineno"> 2697</span> <span class="comment">// and is released under the 3-clause BSD license with friendly</span></div>
|
|
<div class="line"><a id="l02698" name="l02698"></a><span class="lineno"> 2698</span> <span class="comment">// permission of Matteo Frigo and Steven G. Johnson.</span></div>
|
|
<div class="line"><a id="l02699" name="l02699"></a><span class="lineno"> 2699</span> </div>
|
|
<div class="line"><a id="l02700" name="l02700"></a><span class="lineno"> 2700</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">arr<T></a> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">y</a>(N);</div>
|
|
<div class="line"><a id="l02701" name="l02701"></a><span class="lineno"> 2701</span> {</div>
|
|
<div class="line"><a id="l02702" name="l02702"></a><span class="lineno"> 2702</span> <span class="keywordtype">size_t</span> i=0, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a>=n2;</div>
|
|
<div class="line"><a id="l02703" name="l02703"></a><span class="lineno"> 2703</span> <span class="keywordflow">for</span> (; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a><N; ++i, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a>+=4)</div>
|
|
<div class="line"><a id="l02704" name="l02704"></a><span class="lineno"> 2704</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">y</a>[i] = c[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a>];</div>
|
|
<div class="line"><a id="l02705" name="l02705"></a><span class="lineno"> 2705</span> <span class="keywordflow">for</span> (; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a><2*N; ++i, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a>+=4)</div>
|
|
<div class="line"><a id="l02706" name="l02706"></a><span class="lineno"> 2706</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">y</a>[i] = -c[2*N-<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a>-1];</div>
|
|
<div class="line"><a id="l02707" name="l02707"></a><span class="lineno"> 2707</span> <span class="keywordflow">for</span> (; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a><3*N; ++i, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a>+=4)</div>
|
|
<div class="line"><a id="l02708" name="l02708"></a><span class="lineno"> 2708</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">y</a>[i] = -c[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a>-2*N];</div>
|
|
<div class="line"><a id="l02709" name="l02709"></a><span class="lineno"> 2709</span> <span class="keywordflow">for</span> (; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a><4*N; ++i, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a>+=4)</div>
|
|
<div class="line"><a id="l02710" name="l02710"></a><span class="lineno"> 2710</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">y</a>[i] = c[4*N-<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a>-1];</div>
|
|
<div class="line"><a id="l02711" name="l02711"></a><span class="lineno"> 2711</span> <span class="keywordflow">for</span> (; i<N; ++i, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a>+=4)</div>
|
|
<div class="line"><a id="l02712" name="l02712"></a><span class="lineno"> 2712</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">y</a>[i] = c[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">m</a>-4*N];</div>
|
|
<div class="line"><a id="l02713" name="l02713"></a><span class="lineno"> 2713</span> }</div>
|
|
<div class="line"><a id="l02714" name="l02714"></a><span class="lineno"> 2714</span> rfft->exec(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">y</a>.data(), fct, <span class="keyword">true</span>);</div>
|
|
<div class="line"><a id="l02715" name="l02715"></a><span class="lineno"> 2715</span> {</div>
|
|
<div class="line"><a id="l02716" name="l02716"></a><span class="lineno"> 2716</span> <span class="keyword">auto</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">SGN</a> = [](<span class="keywordtype">size_t</span> i)</div>
|
|
<div class="line"><a id="l02717" name="l02717"></a><span class="lineno"> 2717</span> {</div>
|
|
<div class="line"><a id="l02718" name="l02718"></a><span class="lineno"> 2718</span> <span class="keyword">constexpr</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T0</a> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">sqrt2</a>=<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T0</a>(1.414213562373095048801688724209698L);</div>
|
|
<div class="line"><a id="l02719" name="l02719"></a><span class="lineno"> 2719</span> <span class="keywordflow">return</span> (i&2) ? -<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">sqrt2</a> : <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">sqrt2</a>;</div>
|
|
<div class="line"><a id="l02720" name="l02720"></a><span class="lineno"> 2720</span> };</div>
|
|
<div class="line"><a id="l02721" name="l02721"></a><span class="lineno"> 2721</span> c[n2] = <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">y</a>[0]*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">SGN</a>(n2+1);</div>
|
|
<div class="line"><a id="l02722" name="l02722"></a><span class="lineno"> 2722</span> <span class="keywordtype">size_t</span> i=0, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">i1</a>=1, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>=1;</div>
|
|
<div class="line"><a id="l02723" name="l02723"></a><span class="lineno"> 2723</span> <span class="keywordflow">for</span> (; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a><n2; ++i, ++<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">i1</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>+=2)</div>
|
|
<div class="line"><a id="l02724" name="l02724"></a><span class="lineno"> 2724</span> {</div>
|
|
<div class="line"><a id="l02725" name="l02725"></a><span class="lineno"> 2725</span> c[i ] = <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">y</a>[2*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>-1]*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">SGN</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">i1</a>) + <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">y</a>[2*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a> ]*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">SGN</a>(i);</div>
|
|
<div class="line"><a id="l02726" name="l02726"></a><span class="lineno"> 2726</span> c[N -<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">i1</a>] = <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">y</a>[2*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>-1]*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">SGN</a>(N -i) - <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">y</a>[2*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a> ]*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">SGN</a>(N -<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">i1</a>);</div>
|
|
<div class="line"><a id="l02727" name="l02727"></a><span class="lineno"> 2727</span> c[n2-<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">i1</a>] = <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">y</a>[2*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>+1]*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">SGN</a>(n2-i) - <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">y</a>[2*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>+2]*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">SGN</a>(n2-<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">i1</a>);</div>
|
|
<div class="line"><a id="l02728" name="l02728"></a><span class="lineno"> 2728</span> c[n2+<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">i1</a>] = <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">y</a>[2*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>+1]*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">SGN</a>(n2+i+2) + <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">y</a>[2*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>+2]*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">SGN</a>(n2+<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">i1</a>);</div>
|
|
<div class="line"><a id="l02729" name="l02729"></a><span class="lineno"> 2729</span> }</div>
|
|
<div class="line"><a id="l02730" name="l02730"></a><span class="lineno"> 2730</span> <span class="keywordflow">if</span> (<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a> == n2)</div>
|
|
<div class="line"><a id="l02731" name="l02731"></a><span class="lineno"> 2731</span> {</div>
|
|
<div class="line"><a id="l02732" name="l02732"></a><span class="lineno"> 2732</span> c[i ] = <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">y</a>[2*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>-1]*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">SGN</a>(i+1) + <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">y</a>[2*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>]*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">SGN</a>(i);</div>
|
|
<div class="line"><a id="l02733" name="l02733"></a><span class="lineno"> 2733</span> c[N-<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">i1</a>] = <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">y</a>[2*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>-1]*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">SGN</a>(i+2) + <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">y</a>[2*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>]*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">SGN</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">i1</a>);</div>
|
|
<div class="line"><a id="l02734" name="l02734"></a><span class="lineno"> 2734</span> }</div>
|
|
<div class="line"><a id="l02735" name="l02735"></a><span class="lineno"> 2735</span> }</div>
|
|
<div class="line"><a id="l02736" name="l02736"></a><span class="lineno"> 2736</span> </div>
|
|
<div class="line"><a id="l02737" name="l02737"></a><span class="lineno"> 2737</span> <span class="comment">// FFTW-derived code ends here</span></div>
|
|
<div class="line"><a id="l02738" name="l02738"></a><span class="lineno"> 2738</span> }</div>
|
|
<div class="line"><a id="l02739" name="l02739"></a><span class="lineno"> 2739</span> <span class="keywordflow">else</span></div>
|
|
<div class="line"><a id="l02740" name="l02740"></a><span class="lineno"> 2740</span> {</div>
|
|
<div class="line"><a id="l02741" name="l02741"></a><span class="lineno"> 2741</span> <span class="comment">// even length algorithm from</span></div>
|
|
<div class="line"><a id="l02742" name="l02742"></a><span class="lineno"> 2742</span> <span class="comment">// https://www.appletonaudio.com/blog/2013/derivation-of-fast-dct-4-algorithm-based-on-dft/</span></div>
|
|
<div class="line"><a id="l02743" name="l02743"></a><span class="lineno"> 2743</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">arr<cmplx<T></a>> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">y</a>(n2);</div>
|
|
<div class="line"><a id="l02744" name="l02744"></a><span class="lineno"> 2744</span> <span class="keywordflow">for</span>(<span class="keywordtype">size_t</span> i=0; i<n2; ++i)</div>
|
|
<div class="line"><a id="l02745" name="l02745"></a><span class="lineno"> 2745</span> {</div>
|
|
<div class="line"><a id="l02746" name="l02746"></a><span class="lineno"> 2746</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">y</a>[i].<a class="code hl_function" href="structpocketfft_1_1detail_1_1cmplx.html#a647fece372b64b13c4a7e5877d09a807">Set</a>(c[2*i],c[N-1-2*i]);</div>
|
|
<div class="line"><a id="l02747" name="l02747"></a><span class="lineno"> 2747</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">y</a>[i] *= C2[i];</div>
|
|
<div class="line"><a id="l02748" name="l02748"></a><span class="lineno"> 2748</span> }</div>
|
|
<div class="line"><a id="l02749" name="l02749"></a><span class="lineno"> 2749</span> <a class="code hl_function" href="backend_2metal_2kernels_2fft_8h.html#a4010b0e151e5f01e610e9c32234458c7">fft</a>->exec(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">y</a>.data(), fct, <span class="keyword">true</span>);</div>
|
|
<div class="line"><a id="l02750" name="l02750"></a><span class="lineno"> 2750</span> <span class="keywordflow">for</span>(<span class="keywordtype">size_t</span> i=0, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ic</a>=n2-1; i<n2; ++i, --<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ic</a>)</div>
|
|
<div class="line"><a id="l02751" name="l02751"></a><span class="lineno"> 2751</span> {</div>
|
|
<div class="line"><a id="l02752" name="l02752"></a><span class="lineno"> 2752</span> c[2*i ] = 2*(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">y</a>[i ].<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a>*C2[i ].<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a>-<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">y</a>[i ].<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#a35d2dce1b7de5f37d7029e639bc7f23d">i</a>*C2[i ].<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#a35d2dce1b7de5f37d7029e639bc7f23d">i</a>);</div>
|
|
<div class="line"><a id="l02753" name="l02753"></a><span class="lineno"> 2753</span> c[2*i+1] = -2*(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">y</a>[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ic</a>].<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#a35d2dce1b7de5f37d7029e639bc7f23d">i</a>*C2[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ic</a>].<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a>+<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">y</a>[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ic</a>].<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a>*C2[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ic</a>].<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#a35d2dce1b7de5f37d7029e639bc7f23d">i</a>);</div>
|
|
<div class="line"><a id="l02754" name="l02754"></a><span class="lineno"> 2754</span> }</div>
|
|
<div class="line"><a id="l02755" name="l02755"></a><span class="lineno"> 2755</span> }</div>
|
|
<div class="line"><a id="l02756" name="l02756"></a><span class="lineno"> 2756</span> <span class="keywordflow">if</span> (!cosine)</div>
|
|
<div class="line"><a id="l02757" name="l02757"></a><span class="lineno"> 2757</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>=1; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a><N; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>+=2)</div>
|
|
<div class="line"><a id="l02758" name="l02758"></a><span class="lineno"> 2758</span> c[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>] = -c[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">k</a>];</div>
|
|
<div class="line"><a id="l02759" name="l02759"></a><span class="lineno"> 2759</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l02760" name="l02760"></a><span class="lineno"> 2760</span> </div>
|
|
<div class="line"><a id="l02761" name="l02761"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1_t__dcst4.html#af25bf28a7ccd4690ca9934e3aa79c12f"> 2761</a></span> <span class="keywordtype">size_t</span> <a class="code hl_function" href="classpocketfft_1_1detail_1_1_t__dcst4.html#af25bf28a7ccd4690ca9934e3aa79c12f">length</a>()<span class="keyword"> const </span>{ <span class="keywordflow">return</span> N; }</div>
|
|
<div class="line"><a id="l02762" name="l02762"></a><span class="lineno"> 2762</span> };</div>
|
|
</div>
|
|
<div class="line"><a id="l02763" name="l02763"></a><span class="lineno"> 2763</span> </div>
|
|
<div class="line"><a id="l02764" name="l02764"></a><span class="lineno"> 2764</span> </div>
|
|
<div class="line"><a id="l02765" name="l02765"></a><span class="lineno"> 2765</span><span class="comment">//</span></div>
|
|
<div class="line"><a id="l02766" name="l02766"></a><span class="lineno"> 2766</span><span class="comment">// multi-D infrastructure</span></div>
|
|
<div class="line"><a id="l02767" name="l02767"></a><span class="lineno"> 2767</span><span class="comment">//</span></div>
|
|
<div class="line"><a id="l02768" name="l02768"></a><span class="lineno"> 2768</span> </div>
|
|
<div class="foldopen" id="foldopen02769" data-start="{" data-end="}">
|
|
<div class="line"><a id="l02769" name="l02769"></a><span class="lineno"><a class="line" href="namespacepocketfft_1_1detail.html#ab24cdb6118901f4d3c8df06ef0f8390b"> 2769</a></span><span class="keyword">template</span><<span class="keyword">typename</span> T> std::shared_ptr<T> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#ab24cdb6118901f4d3c8df06ef0f8390b">get_plan</a>(<span class="keywordtype">size_t</span> length)</div>
|
|
<div class="line"><a id="l02770" name="l02770"></a><span class="lineno"> 2770</span> {</div>
|
|
<div class="line"><a id="l02771" name="l02771"></a><span class="lineno"> 2771</span><span class="preprocessor">#if POCKETFFT_CACHE_SIZE==0</span></div>
|
|
<div class="line"><a id="l02772" name="l02772"></a><span class="lineno"> 2772</span> <span class="keywordflow">return</span> std::make_shared<T>(length);</div>
|
|
<div class="line"><a id="l02773" name="l02773"></a><span class="lineno"> 2773</span><span class="preprocessor">#else</span></div>
|
|
<div class="line"><a id="l02774" name="l02774"></a><span class="lineno"> 2774</span> <span class="keyword">constexpr</span> <span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nmax</a>=<a class="code hl_define" href="pocketfft_8h.html#a9e604bcf20603d70b62b233d3f306714">POCKETFFT_CACHE_SIZE</a>;</div>
|
|
<div class="line"><a id="l02775" name="l02775"></a><span class="lineno"> 2775</span> <span class="keyword">static</span> std::array<std::shared_ptr<T>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nmax</a>> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cache</a>;</div>
|
|
<div class="line"><a id="l02776" name="l02776"></a><span class="lineno"> 2776</span> <span class="keyword">static</span> std::array<size_t, nmax> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">last_access</a>{{0}};</div>
|
|
<div class="line"><a id="l02777" name="l02777"></a><span class="lineno"> 2777</span> <span class="keyword">static</span> <span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">access_counter</a> = 0;</div>
|
|
<div class="line"><a id="l02778" name="l02778"></a><span class="lineno"> 2778</span> <span class="keyword">static</span> std::mutex mut;</div>
|
|
<div class="line"><a id="l02779" name="l02779"></a><span class="lineno"> 2779</span> </div>
|
|
<div class="line"><a id="l02780" name="l02780"></a><span class="lineno"> 2780</span> <span class="keyword">auto</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">find_in_cache</a> = [&]() -> std::shared_ptr<T></div>
|
|
<div class="line"><a id="l02781" name="l02781"></a><span class="lineno"> 2781</span> {</div>
|
|
<div class="line"><a id="l02782" name="l02782"></a><span class="lineno"> 2782</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i=0; i<<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nmax</a>; ++i)</div>
|
|
<div class="line"><a id="l02783" name="l02783"></a><span class="lineno"> 2783</span> <span class="keywordflow">if</span> (<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cache</a>[i] && (<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cache</a>[i]->length()==length))</div>
|
|
<div class="line"><a id="l02784" name="l02784"></a><span class="lineno"> 2784</span> {</div>
|
|
<div class="line"><a id="l02785" name="l02785"></a><span class="lineno"> 2785</span> <span class="comment">// no need to update if this is already the most recent entry</span></div>
|
|
<div class="line"><a id="l02786" name="l02786"></a><span class="lineno"> 2786</span> <span class="keywordflow">if</span> (<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">last_access</a>[i]!=<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">access_counter</a>)</div>
|
|
<div class="line"><a id="l02787" name="l02787"></a><span class="lineno"> 2787</span> {</div>
|
|
<div class="line"><a id="l02788" name="l02788"></a><span class="lineno"> 2788</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">last_access</a>[i] = ++<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">access_counter</a>;</div>
|
|
<div class="line"><a id="l02789" name="l02789"></a><span class="lineno"> 2789</span> <span class="comment">// Guard against overflow</span></div>
|
|
<div class="line"><a id="l02790" name="l02790"></a><span class="lineno"> 2790</span> <span class="keywordflow">if</span> (<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">access_counter</a> == 0)</div>
|
|
<div class="line"><a id="l02791" name="l02791"></a><span class="lineno"> 2791</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">last_access</a>.fill(0);</div>
|
|
<div class="line"><a id="l02792" name="l02792"></a><span class="lineno"> 2792</span> }</div>
|
|
<div class="line"><a id="l02793" name="l02793"></a><span class="lineno"> 2793</span> <span class="keywordflow">return</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cache</a>[i];</div>
|
|
<div class="line"><a id="l02794" name="l02794"></a><span class="lineno"> 2794</span> }</div>
|
|
<div class="line"><a id="l02795" name="l02795"></a><span class="lineno"> 2795</span> </div>
|
|
<div class="line"><a id="l02796" name="l02796"></a><span class="lineno"> 2796</span> <span class="keywordflow">return</span> <span class="keyword">nullptr</span>;</div>
|
|
<div class="line"><a id="l02797" name="l02797"></a><span class="lineno"> 2797</span> };</div>
|
|
<div class="line"><a id="l02798" name="l02798"></a><span class="lineno"> 2798</span> </div>
|
|
<div class="line"><a id="l02799" name="l02799"></a><span class="lineno"> 2799</span> {</div>
|
|
<div class="line"><a id="l02800" name="l02800"></a><span class="lineno"> 2800</span> std::lock_guard<std::mutex> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">lock</a>(mut);</div>
|
|
<div class="line"><a id="l02801" name="l02801"></a><span class="lineno"> 2801</span> <span class="keyword">auto</span> p = <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">find_in_cache</a>();</div>
|
|
<div class="line"><a id="l02802" name="l02802"></a><span class="lineno"> 2802</span> <span class="keywordflow">if</span> (p) <span class="keywordflow">return</span> p;</div>
|
|
<div class="line"><a id="l02803" name="l02803"></a><span class="lineno"> 2803</span> }</div>
|
|
<div class="line"><a id="l02804" name="l02804"></a><span class="lineno"> 2804</span> <span class="keyword">auto</span> plan = std::make_shared<T>(length);</div>
|
|
<div class="line"><a id="l02805" name="l02805"></a><span class="lineno"> 2805</span> {</div>
|
|
<div class="line"><a id="l02806" name="l02806"></a><span class="lineno"> 2806</span> std::lock_guard<std::mutex> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">lock</a>(mut);</div>
|
|
<div class="line"><a id="l02807" name="l02807"></a><span class="lineno"> 2807</span> <span class="keyword">auto</span> p = <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">find_in_cache</a>();</div>
|
|
<div class="line"><a id="l02808" name="l02808"></a><span class="lineno"> 2808</span> <span class="keywordflow">if</span> (p) <span class="keywordflow">return</span> p;</div>
|
|
<div class="line"><a id="l02809" name="l02809"></a><span class="lineno"> 2809</span> </div>
|
|
<div class="line"><a id="l02810" name="l02810"></a><span class="lineno"> 2810</span> <span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">lru</a> = 0;</div>
|
|
<div class="line"><a id="l02811" name="l02811"></a><span class="lineno"> 2811</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i=1; i<<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nmax</a>; ++i)</div>
|
|
<div class="line"><a id="l02812" name="l02812"></a><span class="lineno"> 2812</span> <span class="keywordflow">if</span> (<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">last_access</a>[i] < <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">last_access</a>[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">lru</a>])</div>
|
|
<div class="line"><a id="l02813" name="l02813"></a><span class="lineno"> 2813</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">lru</a> = i;</div>
|
|
<div class="line"><a id="l02814" name="l02814"></a><span class="lineno"> 2814</span> </div>
|
|
<div class="line"><a id="l02815" name="l02815"></a><span class="lineno"> 2815</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cache</a>[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">lru</a>] = plan;</div>
|
|
<div class="line"><a id="l02816" name="l02816"></a><span class="lineno"> 2816</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">last_access</a>[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">lru</a>] = ++<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">access_counter</a>;</div>
|
|
<div class="line"><a id="l02817" name="l02817"></a><span class="lineno"> 2817</span> }</div>
|
|
<div class="line"><a id="l02818" name="l02818"></a><span class="lineno"> 2818</span> <span class="keywordflow">return</span> plan;</div>
|
|
<div class="line"><a id="l02819" name="l02819"></a><span class="lineno"> 2819</span><span class="preprocessor">#endif</span></div>
|
|
<div class="line"><a id="l02820" name="l02820"></a><span class="lineno"> 2820</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l02821" name="l02821"></a><span class="lineno"> 2821</span> </div>
|
|
<div class="foldopen" id="foldopen02822" data-start="{" data-end="};">
|
|
<div class="line"><a id="l02822" name="l02822"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1arr__info.html"> 2822</a></span><span class="keyword">class </span><a class="code hl_class" href="classpocketfft_1_1detail_1_1arr__info.html">arr_info</a></div>
|
|
<div class="line"><a id="l02823" name="l02823"></a><span class="lineno"> 2823</span> {</div>
|
|
<div class="line"><a id="l02824" name="l02824"></a><span class="lineno"> 2824</span> <span class="keyword">protected</span>:</div>
|
|
<div class="line"><a id="l02825" name="l02825"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1arr__info.html#a2467e9e01de1ba4d7cd28c1af783da8d"> 2825</a></span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#a885ee37fcf564a268a5c8ca9ea8603e1">shape_t</a> <a class="code hl_variable" href="classpocketfft_1_1detail_1_1arr__info.html#a2467e9e01de1ba4d7cd28c1af783da8d">shp</a>;</div>
|
|
<div class="line"><a id="l02826" name="l02826"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1arr__info.html#abe1f7b92501b4e0e5a38fd26294ac5a4"> 2826</a></span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#afb987c919e9424a996d0fc8b3c23cc84">stride_t</a> <a class="code hl_variable" href="classpocketfft_1_1detail_1_1arr__info.html#abe1f7b92501b4e0e5a38fd26294ac5a4">str</a>;</div>
|
|
<div class="line"><a id="l02827" name="l02827"></a><span class="lineno"> 2827</span> </div>
|
|
<div class="line"><a id="l02828" name="l02828"></a><span class="lineno"> 2828</span> <span class="keyword">public</span>:</div>
|
|
<div class="foldopen" id="foldopen02829" data-start="{" data-end="}">
|
|
<div class="line"><a id="l02829" name="l02829"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1arr__info.html#a0dbddb7d86ca306159fc9ef9a453b21e"> 2829</a></span> <a class="code hl_function" href="classpocketfft_1_1detail_1_1arr__info.html#a0dbddb7d86ca306159fc9ef9a453b21e">arr_info</a>(<span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#a885ee37fcf564a268a5c8ca9ea8603e1">shape_t</a> &shape_, <span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#afb987c919e9424a996d0fc8b3c23cc84">stride_t</a> &<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">stride_</a>)</div>
|
|
<div class="line"><a id="l02830" name="l02830"></a><span class="lineno"> 2830</span> : shp(shape_), str(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">stride_</a>) {}</div>
|
|
</div>
|
|
<div class="line"><a id="l02831" name="l02831"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1arr__info.html#ac608c8af2a59a28a0012e308be7ee414"> 2831</a></span> <span class="keywordtype">size_t</span> <a class="code hl_function" href="classpocketfft_1_1detail_1_1arr__info.html#ac608c8af2a59a28a0012e308be7ee414">ndim</a>()<span class="keyword"> const </span>{ <span class="keywordflow">return</span> shp.size(); }</div>
|
|
<div class="line"><a id="l02832" name="l02832"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1arr__info.html#a003a7106f7fa59a3c55ac1f0116313a5"> 2832</a></span> <span class="keywordtype">size_t</span> <a class="code hl_function" href="classpocketfft_1_1detail_1_1arr__info.html#a003a7106f7fa59a3c55ac1f0116313a5">size</a>()<span class="keyword"> const </span>{ <span class="keywordflow">return</span> util::prod(shp); }</div>
|
|
<div class="line"><a id="l02833" name="l02833"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1arr__info.html#accada8146cb8d3ab7facb4c1e3413ec0"> 2833</a></span> <span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#a885ee37fcf564a268a5c8ca9ea8603e1">shape_t</a> &<a class="code hl_function" href="classpocketfft_1_1detail_1_1arr__info.html#accada8146cb8d3ab7facb4c1e3413ec0">shape</a>()<span class="keyword"> const </span>{ <span class="keywordflow">return</span> shp; }</div>
|
|
<div class="line"><a id="l02834" name="l02834"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1arr__info.html#ac601c660c64a4c252aa8be4ae7dfa7a8"> 2834</a></span> <span class="keywordtype">size_t</span> <a class="code hl_function" href="classpocketfft_1_1detail_1_1arr__info.html#ac601c660c64a4c252aa8be4ae7dfa7a8">shape</a>(<span class="keywordtype">size_t</span> i)<span class="keyword"> const </span>{ <span class="keywordflow">return</span> shp[i]; }</div>
|
|
<div class="line"><a id="l02835" name="l02835"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1arr__info.html#a9d10aa83a1117e75d36f7396b8c2a093"> 2835</a></span> <span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#afb987c919e9424a996d0fc8b3c23cc84">stride_t</a> &<a class="code hl_function" href="classpocketfft_1_1detail_1_1arr__info.html#a9d10aa83a1117e75d36f7396b8c2a093">stride</a>()<span class="keyword"> const </span>{ <span class="keywordflow">return</span> str; }</div>
|
|
<div class="line"><a id="l02836" name="l02836"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1arr__info.html#ac1f6a9bd6703eceef6003f5f6315d39b"> 2836</a></span> <span class="keyword">const</span> ptrdiff_t &<a class="code hl_function" href="classpocketfft_1_1detail_1_1arr__info.html#ac1f6a9bd6703eceef6003f5f6315d39b">stride</a>(<span class="keywordtype">size_t</span> i)<span class="keyword"> const </span>{ <span class="keywordflow">return</span> str[i]; }</div>
|
|
<div class="line"><a id="l02837" name="l02837"></a><span class="lineno"> 2837</span> };</div>
|
|
</div>
|
|
<div class="line"><a id="l02838" name="l02838"></a><span class="lineno"> 2838</span> </div>
|
|
<div class="foldopen" id="foldopen02839" data-start="{" data-end="};">
|
|
<div class="line"><a id="l02839" name="l02839"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1cndarr.html"> 2839</a></span><span class="keyword">template</span><<span class="keyword">typename</span> T> <span class="keyword">class </span><a class="code hl_class" href="classpocketfft_1_1detail_1_1cndarr.html">cndarr</a>: <span class="keyword">public</span> <a class="code hl_class" href="classpocketfft_1_1detail_1_1arr__info.html">arr_info</a></div>
|
|
<div class="line"><a id="l02840" name="l02840"></a><span class="lineno"> 2840</span> {</div>
|
|
<div class="line"><a id="l02841" name="l02841"></a><span class="lineno"> 2841</span> <span class="keyword">protected</span>:</div>
|
|
<div class="line"><a id="l02842" name="l02842"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1cndarr.html#ac29c769aebb03f81fbcf16ba6e766af2"> 2842</a></span> <span class="keyword">const</span> <span class="keywordtype">char</span> *<a class="code hl_variable" href="classpocketfft_1_1detail_1_1cndarr.html#ac29c769aebb03f81fbcf16ba6e766af2">d</a>;</div>
|
|
<div class="line"><a id="l02843" name="l02843"></a><span class="lineno"> 2843</span> </div>
|
|
<div class="line"><a id="l02844" name="l02844"></a><span class="lineno"> 2844</span> <span class="keyword">public</span>:</div>
|
|
<div class="foldopen" id="foldopen02845" data-start="{" data-end="}">
|
|
<div class="line"><a id="l02845" name="l02845"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1cndarr.html#abf73f1b4ddcfb27d7f85cfa441607129"> 2845</a></span> <a class="code hl_function" href="classpocketfft_1_1detail_1_1cndarr.html#abf73f1b4ddcfb27d7f85cfa441607129">cndarr</a>(<span class="keyword">const</span> <span class="keywordtype">void</span> *<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">data_</a>, <span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#a885ee37fcf564a268a5c8ca9ea8603e1">shape_t</a> &shape_, <span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#afb987c919e9424a996d0fc8b3c23cc84">stride_t</a> &<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">stride_</a>)</div>
|
|
<div class="line"><a id="l02846" name="l02846"></a><span class="lineno"> 2846</span> : <a class="code hl_class" href="classpocketfft_1_1detail_1_1arr__info.html">arr_info</a>(shape_, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">stride_</a>),</div>
|
|
<div class="line"><a id="l02847" name="l02847"></a><span class="lineno"> 2847</span> d(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">reinterpret_cast</a><<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">const</a> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">char</a> *>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">data_</a>)) {}</div>
|
|
</div>
|
|
<div class="foldopen" id="foldopen02848" data-start="{" data-end="}">
|
|
<div class="line"><a id="l02848" name="l02848"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1cndarr.html#ae4852d1fe936a5d61832b507816c7054"> 2848</a></span> <span class="keyword">const</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a> &<a class="code hl_function" href="classpocketfft_1_1detail_1_1cndarr.html#ae4852d1fe936a5d61832b507816c7054">operator[]</a>(ptrdiff_t ofs)<span class="keyword"> const</span></div>
|
|
<div class="line"><a id="l02849" name="l02849"></a><span class="lineno"> 2849</span><span class="keyword"> </span>{ <span class="keywordflow">return</span> *<span class="keyword">reinterpret_cast<</span><span class="keyword">const </span><a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a> *<span class="keyword">></span>(d+ofs); }</div>
|
|
</div>
|
|
<div class="line"><a id="l02850" name="l02850"></a><span class="lineno"> 2850</span> };</div>
|
|
</div>
|
|
<div class="line"><a id="l02851" name="l02851"></a><span class="lineno"> 2851</span> </div>
|
|
<div class="foldopen" id="foldopen02852" data-start="{" data-end="};">
|
|
<div class="line"><a id="l02852" name="l02852"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1ndarr.html"> 2852</a></span><span class="keyword">template</span><<span class="keyword">typename</span> T> <span class="keyword">class </span><a class="code hl_class" href="classpocketfft_1_1detail_1_1ndarr.html">ndarr</a>: <span class="keyword">public</span> <a class="code hl_class" href="classpocketfft_1_1detail_1_1cndarr.html">cndarr</a><T></div>
|
|
<div class="line"><a id="l02853" name="l02853"></a><span class="lineno"> 2853</span> {</div>
|
|
<div class="line"><a id="l02854" name="l02854"></a><span class="lineno"> 2854</span> <span class="keyword">public</span>:</div>
|
|
<div class="foldopen" id="foldopen02855" data-start="{" data-end="}">
|
|
<div class="line"><a id="l02855" name="l02855"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1ndarr.html#a8f0037a172d96cb1ad915a5069175fa2"> 2855</a></span> <a class="code hl_function" href="classpocketfft_1_1detail_1_1ndarr.html#a8f0037a172d96cb1ad915a5069175fa2">ndarr</a>(<span class="keywordtype">void</span> *<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">data_</a>, <span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#a885ee37fcf564a268a5c8ca9ea8603e1">shape_t</a> &shape_, <span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#afb987c919e9424a996d0fc8b3c23cc84">stride_t</a> &<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">stride_</a>)</div>
|
|
<div class="line"><a id="l02856" name="l02856"></a><span class="lineno"> 2856</span> : <a class="code hl_class" href="classpocketfft_1_1detail_1_1cndarr.html">cndarr</a><<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a>>::<a class="code hl_class" href="classpocketfft_1_1detail_1_1cndarr.html">cndarr</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">const_cast</a><<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">const</a> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">void</a> *>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">data_</a>), shape_, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">stride_</a>)</div>
|
|
<div class="line"><a id="l02857" name="l02857"></a><span class="lineno"> 2857</span> {}</div>
|
|
</div>
|
|
<div class="foldopen" id="foldopen02858" data-start="{" data-end="}">
|
|
<div class="line"><a id="l02858" name="l02858"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1ndarr.html#a2b2c4e205e8b5c32c9fe55dfd7b8c8d8"> 2858</a></span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a> &<a class="code hl_function" href="classpocketfft_1_1detail_1_1ndarr.html#a2b2c4e205e8b5c32c9fe55dfd7b8c8d8">operator[]</a>(ptrdiff_t ofs)</div>
|
|
<div class="line"><a id="l02859" name="l02859"></a><span class="lineno"> 2859</span> { <span class="keywordflow">return</span> *<span class="keyword">reinterpret_cast<</span><a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a> *<span class="keyword">></span>(<span class="keyword">const_cast<</span><span class="keywordtype">char</span> *<span class="keyword">></span>(<a class="code hl_class" href="classpocketfft_1_1detail_1_1cndarr.html">cndarr<T>::d</a>+ofs)); }</div>
|
|
</div>
|
|
<div class="line"><a id="l02860" name="l02860"></a><span class="lineno"> 2860</span> };</div>
|
|
</div>
|
|
<div class="line"><a id="l02861" name="l02861"></a><span class="lineno"> 2861</span> </div>
|
|
<div class="foldopen" id="foldopen02862" data-start="{" data-end="};">
|
|
<div class="line"><a id="l02862" name="l02862"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1multi__iter.html"> 2862</a></span><span class="keyword">template</span><<span class="keywordtype">size_t</span> N> <span class="keyword">class </span><a class="code hl_class" href="classpocketfft_1_1detail_1_1multi__iter.html">multi_iter</a></div>
|
|
<div class="line"><a id="l02863" name="l02863"></a><span class="lineno"> 2863</span> {</div>
|
|
<div class="line"><a id="l02864" name="l02864"></a><span class="lineno"> 2864</span> <span class="keyword">private</span>:</div>
|
|
<div class="line"><a id="l02865" name="l02865"></a><span class="lineno"> 2865</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#a885ee37fcf564a268a5c8ca9ea8603e1">shape_t</a> pos;</div>
|
|
<div class="line"><a id="l02866" name="l02866"></a><span class="lineno"> 2866</span> <span class="keyword">const</span> <a class="code hl_class" href="classpocketfft_1_1detail_1_1arr__info.html">arr_info</a> &iarr, &oarr;</div>
|
|
<div class="line"><a id="l02867" name="l02867"></a><span class="lineno"> 2867</span> ptrdiff_t p_ii, p_i[N], str_i, p_oi, p_o[N], str_o;</div>
|
|
<div class="line"><a id="l02868" name="l02868"></a><span class="lineno"> 2868</span> <span class="keywordtype">size_t</span> idim, rem;</div>
|
|
<div class="line"><a id="l02869" name="l02869"></a><span class="lineno"> 2869</span> </div>
|
|
<div class="line"><a id="l02870" name="l02870"></a><span class="lineno"> 2870</span> <span class="keywordtype">void</span> advance_i()</div>
|
|
<div class="line"><a id="l02871" name="l02871"></a><span class="lineno"> 2871</span> {</div>
|
|
<div class="line"><a id="l02872" name="l02872"></a><span class="lineno"> 2872</span> <span class="keywordflow">for</span> (<span class="keywordtype">int</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">i_</a>=<span class="keywordtype">int</span>(pos.size())-1; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">i_</a>>=0; --<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">i_</a>)</div>
|
|
<div class="line"><a id="l02873" name="l02873"></a><span class="lineno"> 2873</span> {</div>
|
|
<div class="line"><a id="l02874" name="l02874"></a><span class="lineno"> 2874</span> <span class="keyword">auto</span> i = size_t(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">i_</a>);</div>
|
|
<div class="line"><a id="l02875" name="l02875"></a><span class="lineno"> 2875</span> <span class="keywordflow">if</span> (i==idim) <span class="keywordflow">continue</span>;</div>
|
|
<div class="line"><a id="l02876" name="l02876"></a><span class="lineno"> 2876</span> p_ii += iarr.<a class="code hl_function" href="classpocketfft_1_1detail_1_1arr__info.html#a9d10aa83a1117e75d36f7396b8c2a093">stride</a>(i);</div>
|
|
<div class="line"><a id="l02877" name="l02877"></a><span class="lineno"> 2877</span> p_oi += oarr.<a class="code hl_function" href="classpocketfft_1_1detail_1_1arr__info.html#a9d10aa83a1117e75d36f7396b8c2a093">stride</a>(i);</div>
|
|
<div class="line"><a id="l02878" name="l02878"></a><span class="lineno"> 2878</span> <span class="keywordflow">if</span> (++pos[i] < iarr.<a class="code hl_function" href="classpocketfft_1_1detail_1_1arr__info.html#accada8146cb8d3ab7facb4c1e3413ec0">shape</a>(i))</div>
|
|
<div class="line"><a id="l02879" name="l02879"></a><span class="lineno"> 2879</span> <span class="keywordflow">return</span>;</div>
|
|
<div class="line"><a id="l02880" name="l02880"></a><span class="lineno"> 2880</span> pos[i] = 0;</div>
|
|
<div class="line"><a id="l02881" name="l02881"></a><span class="lineno"> 2881</span> p_ii -= ptrdiff_t(iarr.<a class="code hl_function" href="classpocketfft_1_1detail_1_1arr__info.html#accada8146cb8d3ab7facb4c1e3413ec0">shape</a>(i))*iarr.<a class="code hl_function" href="classpocketfft_1_1detail_1_1arr__info.html#a9d10aa83a1117e75d36f7396b8c2a093">stride</a>(i);</div>
|
|
<div class="line"><a id="l02882" name="l02882"></a><span class="lineno"> 2882</span> p_oi -= ptrdiff_t(oarr.<a class="code hl_function" href="classpocketfft_1_1detail_1_1arr__info.html#accada8146cb8d3ab7facb4c1e3413ec0">shape</a>(i))*oarr.<a class="code hl_function" href="classpocketfft_1_1detail_1_1arr__info.html#a9d10aa83a1117e75d36f7396b8c2a093">stride</a>(i);</div>
|
|
<div class="line"><a id="l02883" name="l02883"></a><span class="lineno"> 2883</span> }</div>
|
|
<div class="line"><a id="l02884" name="l02884"></a><span class="lineno"> 2884</span> }</div>
|
|
<div class="line"><a id="l02885" name="l02885"></a><span class="lineno"> 2885</span> </div>
|
|
<div class="line"><a id="l02886" name="l02886"></a><span class="lineno"> 2886</span> <span class="keyword">public</span>:</div>
|
|
<div class="foldopen" id="foldopen02887" data-start="{" data-end="}">
|
|
<div class="line"><a id="l02887" name="l02887"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1multi__iter.html#a9be43bb18840202da6d17988fccc64b9"> 2887</a></span> <a class="code hl_function" href="classpocketfft_1_1detail_1_1multi__iter.html#a9be43bb18840202da6d17988fccc64b9">multi_iter</a>(<span class="keyword">const</span> <a class="code hl_class" href="classpocketfft_1_1detail_1_1arr__info.html">arr_info</a> &<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">iarr_</a>, <span class="keyword">const</span> <a class="code hl_class" href="classpocketfft_1_1detail_1_1arr__info.html">arr_info</a> &<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">oarr_</a>, <span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">idim_</a>)</div>
|
|
<div class="line"><a id="l02888" name="l02888"></a><span class="lineno"> 2888</span> : pos(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">iarr_</a>.ndim(), 0), iarr(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">iarr_</a>), oarr(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">oarr_</a>), p_ii(0),</div>
|
|
<div class="line"><a id="l02889" name="l02889"></a><span class="lineno"> 2889</span> str_i(iarr.stride(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">idim_</a>)), p_oi(0), str_o(oarr.stride(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">idim_</a>)),</div>
|
|
<div class="line"><a id="l02890" name="l02890"></a><span class="lineno"> 2890</span> idim(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">idim_</a>), rem(iarr.size()/iarr.shape(idim))</div>
|
|
<div class="line"><a id="l02891" name="l02891"></a><span class="lineno"> 2891</span> {</div>
|
|
<div class="line"><a id="l02892" name="l02892"></a><span class="lineno"> 2892</span> <span class="keyword">auto</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nshares</a> = threading::num_threads();</div>
|
|
<div class="line"><a id="l02893" name="l02893"></a><span class="lineno"> 2893</span> <span class="keywordflow">if</span> (<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nshares</a>==1) <span class="keywordflow">return</span>;</div>
|
|
<div class="line"><a id="l02894" name="l02894"></a><span class="lineno"> 2894</span> <span class="keywordflow">if</span> (<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nshares</a>==0) <span class="keywordflow">throw</span> std::runtime_error(<span class="stringliteral">"can't run with zero threads"</span>);</div>
|
|
<div class="line"><a id="l02895" name="l02895"></a><span class="lineno"> 2895</span> <span class="keyword">auto</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">myshare</a> = threading::thread_id();</div>
|
|
<div class="line"><a id="l02896" name="l02896"></a><span class="lineno"> 2896</span> <span class="keywordflow">if</span> (<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">myshare</a>>=<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nshares</a>) <span class="keywordflow">throw</span> std::runtime_error(<span class="stringliteral">"impossible share requested"</span>);</div>
|
|
<div class="line"><a id="l02897" name="l02897"></a><span class="lineno"> 2897</span> <span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nbase</a> = rem/<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nshares</a>;</div>
|
|
<div class="line"><a id="l02898" name="l02898"></a><span class="lineno"> 2898</span> <span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">additional</a> = rem%<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nshares</a>;</div>
|
|
<div class="line"><a id="l02899" name="l02899"></a><span class="lineno"> 2899</span> <span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">lo</a> = <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">myshare</a>*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nbase</a> + ((<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">myshare</a><<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">additional</a>) ? <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">myshare</a> : <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">additional</a>);</div>
|
|
<div class="line"><a id="l02900" name="l02900"></a><span class="lineno"> 2900</span> <span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">hi</a> = <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">lo</a>+<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nbase</a>+(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">myshare</a><<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">additional</a>);</div>
|
|
<div class="line"><a id="l02901" name="l02901"></a><span class="lineno"> 2901</span> <span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">todo</a> = <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">hi</a>-<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">lo</a>;</div>
|
|
<div class="line"><a id="l02902" name="l02902"></a><span class="lineno"> 2902</span> </div>
|
|
<div class="line"><a id="l02903" name="l02903"></a><span class="lineno"> 2903</span> <span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">chunk</a> = rem;</div>
|
|
<div class="line"><a id="l02904" name="l02904"></a><span class="lineno"> 2904</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i=0; i<pos.size(); ++i)</div>
|
|
<div class="line"><a id="l02905" name="l02905"></a><span class="lineno"> 2905</span> {</div>
|
|
<div class="line"><a id="l02906" name="l02906"></a><span class="lineno"> 2906</span> <span class="keywordflow">if</span> (i==idim) <span class="keywordflow">continue</span>;</div>
|
|
<div class="line"><a id="l02907" name="l02907"></a><span class="lineno"> 2907</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">chunk</a> /= iarr.<a class="code hl_function" href="classpocketfft_1_1detail_1_1arr__info.html#accada8146cb8d3ab7facb4c1e3413ec0">shape</a>(i);</div>
|
|
<div class="line"><a id="l02908" name="l02908"></a><span class="lineno"> 2908</span> <span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">n_advance</a> = <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">lo</a>/<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">chunk</a>;</div>
|
|
<div class="line"><a id="l02909" name="l02909"></a><span class="lineno"> 2909</span> pos[i] += <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">n_advance</a>;</div>
|
|
<div class="line"><a id="l02910" name="l02910"></a><span class="lineno"> 2910</span> p_ii += ptrdiff_t(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">n_advance</a>)*iarr.<a class="code hl_function" href="classpocketfft_1_1detail_1_1arr__info.html#a9d10aa83a1117e75d36f7396b8c2a093">stride</a>(i);</div>
|
|
<div class="line"><a id="l02911" name="l02911"></a><span class="lineno"> 2911</span> p_oi += ptrdiff_t(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">n_advance</a>)*oarr.<a class="code hl_function" href="classpocketfft_1_1detail_1_1arr__info.html#a9d10aa83a1117e75d36f7396b8c2a093">stride</a>(i);</div>
|
|
<div class="line"><a id="l02912" name="l02912"></a><span class="lineno"> 2912</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">lo</a> -= <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">n_advance</a>*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">chunk</a>;</div>
|
|
<div class="line"><a id="l02913" name="l02913"></a><span class="lineno"> 2913</span> }</div>
|
|
<div class="line"><a id="l02914" name="l02914"></a><span class="lineno"> 2914</span> rem = <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">todo</a>;</div>
|
|
<div class="line"><a id="l02915" name="l02915"></a><span class="lineno"> 2915</span> }</div>
|
|
</div>
|
|
<div class="foldopen" id="foldopen02916" data-start="{" data-end="}">
|
|
<div class="line"><a id="l02916" name="l02916"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1multi__iter.html#a5ddcc0666125b3cb6c0d62b30befdd2c"> 2916</a></span> <span class="keywordtype">void</span> <a class="code hl_function" href="classpocketfft_1_1detail_1_1multi__iter.html#a5ddcc0666125b3cb6c0d62b30befdd2c">advance</a>(<span class="keywordtype">size_t</span> n)</div>
|
|
<div class="line"><a id="l02917" name="l02917"></a><span class="lineno"> 2917</span> {</div>
|
|
<div class="line"><a id="l02918" name="l02918"></a><span class="lineno"> 2918</span> <span class="keywordflow">if</span> (rem<n) <span class="keywordflow">throw</span> std::runtime_error(<span class="stringliteral">"underrun"</span>);</div>
|
|
<div class="line"><a id="l02919" name="l02919"></a><span class="lineno"> 2919</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i=0; i<n; ++i)</div>
|
|
<div class="line"><a id="l02920" name="l02920"></a><span class="lineno"> 2920</span> {</div>
|
|
<div class="line"><a id="l02921" name="l02921"></a><span class="lineno"> 2921</span> p_i[i] = p_ii;</div>
|
|
<div class="line"><a id="l02922" name="l02922"></a><span class="lineno"> 2922</span> p_o[i] = p_oi;</div>
|
|
<div class="line"><a id="l02923" name="l02923"></a><span class="lineno"> 2923</span> advance_i();</div>
|
|
<div class="line"><a id="l02924" name="l02924"></a><span class="lineno"> 2924</span> }</div>
|
|
<div class="line"><a id="l02925" name="l02925"></a><span class="lineno"> 2925</span> rem -= n;</div>
|
|
<div class="line"><a id="l02926" name="l02926"></a><span class="lineno"> 2926</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l02927" name="l02927"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1multi__iter.html#ad33360d4a8ab8e6d72efadc6f9cb5bfa"> 2927</a></span> ptrdiff_t <a class="code hl_function" href="classpocketfft_1_1detail_1_1multi__iter.html#ad33360d4a8ab8e6d72efadc6f9cb5bfa">iofs</a>(<span class="keywordtype">size_t</span> i)<span class="keyword"> const </span>{ <span class="keywordflow">return</span> p_i[0] + ptrdiff_t(i)*str_i; }</div>
|
|
<div class="line"><a id="l02928" name="l02928"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1multi__iter.html#a97462d97bdca6419d8d2f37c2031fe83"> 2928</a></span> ptrdiff_t <a class="code hl_function" href="classpocketfft_1_1detail_1_1multi__iter.html#a97462d97bdca6419d8d2f37c2031fe83">iofs</a>(<span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">j</a>, <span class="keywordtype">size_t</span> i)<span class="keyword"> const </span>{ <span class="keywordflow">return</span> p_i[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">j</a>] + ptrdiff_t(i)*str_i; }</div>
|
|
<div class="line"><a id="l02929" name="l02929"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1multi__iter.html#aae63e67caac095d474ddd32daa5ffa34"> 2929</a></span> ptrdiff_t <a class="code hl_function" href="classpocketfft_1_1detail_1_1multi__iter.html#aae63e67caac095d474ddd32daa5ffa34">oofs</a>(<span class="keywordtype">size_t</span> i)<span class="keyword"> const </span>{ <span class="keywordflow">return</span> p_o[0] + ptrdiff_t(i)*str_o; }</div>
|
|
<div class="line"><a id="l02930" name="l02930"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1multi__iter.html#a9236047e7419e5d21379cbf95eb3a78e"> 2930</a></span> ptrdiff_t <a class="code hl_function" href="classpocketfft_1_1detail_1_1multi__iter.html#a9236047e7419e5d21379cbf95eb3a78e">oofs</a>(<span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">j</a>, <span class="keywordtype">size_t</span> i)<span class="keyword"> const </span>{ <span class="keywordflow">return</span> p_o[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">j</a>] + ptrdiff_t(i)*str_o; }</div>
|
|
<div class="line"><a id="l02931" name="l02931"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1multi__iter.html#a5318b79d934cddf109dff7bf96a330c8"> 2931</a></span> <span class="keywordtype">size_t</span> <a class="code hl_function" href="classpocketfft_1_1detail_1_1multi__iter.html#a5318b79d934cddf109dff7bf96a330c8">length_in</a>()<span class="keyword"> const </span>{ <span class="keywordflow">return</span> iarr.<a class="code hl_function" href="classpocketfft_1_1detail_1_1arr__info.html#accada8146cb8d3ab7facb4c1e3413ec0">shape</a>(idim); }</div>
|
|
<div class="line"><a id="l02932" name="l02932"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1multi__iter.html#a93cd515d07cd479138a35da9df66bd41"> 2932</a></span> <span class="keywordtype">size_t</span> <a class="code hl_function" href="classpocketfft_1_1detail_1_1multi__iter.html#a93cd515d07cd479138a35da9df66bd41">length_out</a>()<span class="keyword"> const </span>{ <span class="keywordflow">return</span> oarr.<a class="code hl_function" href="classpocketfft_1_1detail_1_1arr__info.html#accada8146cb8d3ab7facb4c1e3413ec0">shape</a>(idim); }</div>
|
|
<div class="line"><a id="l02933" name="l02933"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1multi__iter.html#ac947f03b1cfcb63436a7e61ff020a88c"> 2933</a></span> ptrdiff_t <a class="code hl_function" href="classpocketfft_1_1detail_1_1multi__iter.html#ac947f03b1cfcb63436a7e61ff020a88c">stride_in</a>()<span class="keyword"> const </span>{ <span class="keywordflow">return</span> str_i; }</div>
|
|
<div class="line"><a id="l02934" name="l02934"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1multi__iter.html#a81d71a13bf0b85e556fbb9834167ecc7"> 2934</a></span> ptrdiff_t <a class="code hl_function" href="classpocketfft_1_1detail_1_1multi__iter.html#a81d71a13bf0b85e556fbb9834167ecc7">stride_out</a>()<span class="keyword"> const </span>{ <span class="keywordflow">return</span> str_o; }</div>
|
|
<div class="line"><a id="l02935" name="l02935"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1multi__iter.html#a034d12f842df90e6471dffd3fa6ba4bd"> 2935</a></span> <span class="keywordtype">size_t</span> <a class="code hl_function" href="classpocketfft_1_1detail_1_1multi__iter.html#a034d12f842df90e6471dffd3fa6ba4bd">remaining</a>()<span class="keyword"> const </span>{ <span class="keywordflow">return</span> rem; }</div>
|
|
<div class="line"><a id="l02936" name="l02936"></a><span class="lineno"> 2936</span> };</div>
|
|
</div>
|
|
<div class="line"><a id="l02937" name="l02937"></a><span class="lineno"> 2937</span> </div>
|
|
<div class="foldopen" id="foldopen02938" data-start="{" data-end="};">
|
|
<div class="line"><a id="l02938" name="l02938"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1simple__iter.html"> 2938</a></span><span class="keyword">class </span><a class="code hl_class" href="classpocketfft_1_1detail_1_1simple__iter.html">simple_iter</a></div>
|
|
<div class="line"><a id="l02939" name="l02939"></a><span class="lineno"> 2939</span> {</div>
|
|
<div class="line"><a id="l02940" name="l02940"></a><span class="lineno"> 2940</span> <span class="keyword">private</span>:</div>
|
|
<div class="line"><a id="l02941" name="l02941"></a><span class="lineno"> 2941</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#a885ee37fcf564a268a5c8ca9ea8603e1">shape_t</a> pos;</div>
|
|
<div class="line"><a id="l02942" name="l02942"></a><span class="lineno"> 2942</span> <span class="keyword">const</span> <a class="code hl_class" href="classpocketfft_1_1detail_1_1arr__info.html">arr_info</a> &<a class="code hl_class" href="classpocketfft_1_1detail_1_1arr.html">arr</a>;</div>
|
|
<div class="line"><a id="l02943" name="l02943"></a><span class="lineno"> 2943</span> ptrdiff_t p;</div>
|
|
<div class="line"><a id="l02944" name="l02944"></a><span class="lineno"> 2944</span> <span class="keywordtype">size_t</span> rem;</div>
|
|
<div class="line"><a id="l02945" name="l02945"></a><span class="lineno"> 2945</span> </div>
|
|
<div class="line"><a id="l02946" name="l02946"></a><span class="lineno"> 2946</span> <span class="keyword">public</span>:</div>
|
|
<div class="foldopen" id="foldopen02947" data-start="{" data-end="}">
|
|
<div class="line"><a id="l02947" name="l02947"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1simple__iter.html#a1e455c615825bebd5f1f62665027b398"> 2947</a></span> <a class="code hl_function" href="classpocketfft_1_1detail_1_1simple__iter.html#a1e455c615825bebd5f1f62665027b398">simple_iter</a>(<span class="keyword">const</span> <a class="code hl_class" href="classpocketfft_1_1detail_1_1arr__info.html">arr_info</a> &<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">arr_</a>)</div>
|
|
<div class="line"><a id="l02948" name="l02948"></a><span class="lineno"> 2948</span> : pos(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">arr_</a>.ndim(), 0), <a class="code hl_class" href="classpocketfft_1_1detail_1_1arr.html">arr</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">arr_</a>), p(0), rem(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">arr_</a>.size()) {}</div>
|
|
</div>
|
|
<div class="foldopen" id="foldopen02949" data-start="{" data-end="}">
|
|
<div class="line"><a id="l02949" name="l02949"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1simple__iter.html#a73a9ecd3008d2bd35aaa00bf9fac074f"> 2949</a></span> <span class="keywordtype">void</span> <a class="code hl_function" href="classpocketfft_1_1detail_1_1simple__iter.html#a73a9ecd3008d2bd35aaa00bf9fac074f">advance</a>()</div>
|
|
<div class="line"><a id="l02950" name="l02950"></a><span class="lineno"> 2950</span> {</div>
|
|
<div class="line"><a id="l02951" name="l02951"></a><span class="lineno"> 2951</span> --rem;</div>
|
|
<div class="line"><a id="l02952" name="l02952"></a><span class="lineno"> 2952</span> <span class="keywordflow">for</span> (<span class="keywordtype">int</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">i_</a>=<span class="keywordtype">int</span>(pos.size())-1; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">i_</a>>=0; --<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">i_</a>)</div>
|
|
<div class="line"><a id="l02953" name="l02953"></a><span class="lineno"> 2953</span> {</div>
|
|
<div class="line"><a id="l02954" name="l02954"></a><span class="lineno"> 2954</span> <span class="keyword">auto</span> i = size_t(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">i_</a>);</div>
|
|
<div class="line"><a id="l02955" name="l02955"></a><span class="lineno"> 2955</span> p += <a class="code hl_class" href="classpocketfft_1_1detail_1_1arr.html">arr</a>.stride(i);</div>
|
|
<div class="line"><a id="l02956" name="l02956"></a><span class="lineno"> 2956</span> <span class="keywordflow">if</span> (++pos[i] < <a class="code hl_class" href="classpocketfft_1_1detail_1_1arr.html">arr</a>.shape(i))</div>
|
|
<div class="line"><a id="l02957" name="l02957"></a><span class="lineno"> 2957</span> <span class="keywordflow">return</span>;</div>
|
|
<div class="line"><a id="l02958" name="l02958"></a><span class="lineno"> 2958</span> pos[i] = 0;</div>
|
|
<div class="line"><a id="l02959" name="l02959"></a><span class="lineno"> 2959</span> p -= ptrdiff_t(<a class="code hl_class" href="classpocketfft_1_1detail_1_1arr.html">arr</a>.shape(i))*<a class="code hl_class" href="classpocketfft_1_1detail_1_1arr.html">arr</a>.stride(i);</div>
|
|
<div class="line"><a id="l02960" name="l02960"></a><span class="lineno"> 2960</span> }</div>
|
|
<div class="line"><a id="l02961" name="l02961"></a><span class="lineno"> 2961</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l02962" name="l02962"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1simple__iter.html#ab59481ad9c8f04addb907c3ebb89f8fa"> 2962</a></span> ptrdiff_t <a class="code hl_function" href="classpocketfft_1_1detail_1_1simple__iter.html#ab59481ad9c8f04addb907c3ebb89f8fa">ofs</a>()<span class="keyword"> const </span>{ <span class="keywordflow">return</span> p; }</div>
|
|
<div class="line"><a id="l02963" name="l02963"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1simple__iter.html#a9267d37f51a9a5aecc69293c7ed1b1f6"> 2963</a></span> <span class="keywordtype">size_t</span> <a class="code hl_function" href="classpocketfft_1_1detail_1_1simple__iter.html#a9267d37f51a9a5aecc69293c7ed1b1f6">remaining</a>()<span class="keyword"> const </span>{ <span class="keywordflow">return</span> rem; }</div>
|
|
<div class="line"><a id="l02964" name="l02964"></a><span class="lineno"> 2964</span> };</div>
|
|
</div>
|
|
<div class="line"><a id="l02965" name="l02965"></a><span class="lineno"> 2965</span> </div>
|
|
<div class="foldopen" id="foldopen02966" data-start="{" data-end="};">
|
|
<div class="line"><a id="l02966" name="l02966"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1rev__iter.html"> 2966</a></span><span class="keyword">class </span><a class="code hl_class" href="classpocketfft_1_1detail_1_1rev__iter.html">rev_iter</a></div>
|
|
<div class="line"><a id="l02967" name="l02967"></a><span class="lineno"> 2967</span> {</div>
|
|
<div class="line"><a id="l02968" name="l02968"></a><span class="lineno"> 2968</span> <span class="keyword">private</span>:</div>
|
|
<div class="line"><a id="l02969" name="l02969"></a><span class="lineno"> 2969</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#a885ee37fcf564a268a5c8ca9ea8603e1">shape_t</a> pos;</div>
|
|
<div class="line"><a id="l02970" name="l02970"></a><span class="lineno"> 2970</span> <span class="keyword">const</span> <a class="code hl_class" href="classpocketfft_1_1detail_1_1arr__info.html">arr_info</a> &<a class="code hl_class" href="classpocketfft_1_1detail_1_1arr.html">arr</a>;</div>
|
|
<div class="line"><a id="l02971" name="l02971"></a><span class="lineno"> 2971</span> std::vector<char> rev_axis;</div>
|
|
<div class="line"><a id="l02972" name="l02972"></a><span class="lineno"> 2972</span> std::vector<char> rev_jump;</div>
|
|
<div class="line"><a id="l02973" name="l02973"></a><span class="lineno"> 2973</span> <span class="keywordtype">size_t</span> last_axis, last_size;</div>
|
|
<div class="line"><a id="l02974" name="l02974"></a><span class="lineno"> 2974</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#a885ee37fcf564a268a5c8ca9ea8603e1">shape_t</a> shp;</div>
|
|
<div class="line"><a id="l02975" name="l02975"></a><span class="lineno"> 2975</span> ptrdiff_t p, rp;</div>
|
|
<div class="line"><a id="l02976" name="l02976"></a><span class="lineno"> 2976</span> <span class="keywordtype">size_t</span> rem;</div>
|
|
<div class="line"><a id="l02977" name="l02977"></a><span class="lineno"> 2977</span> </div>
|
|
<div class="line"><a id="l02978" name="l02978"></a><span class="lineno"> 2978</span> <span class="keyword">public</span>:</div>
|
|
<div class="foldopen" id="foldopen02979" data-start="{" data-end="}">
|
|
<div class="line"><a id="l02979" name="l02979"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1rev__iter.html#af7b8c2f1534d3038ba2a3c6b9919e134"> 2979</a></span> <a class="code hl_function" href="classpocketfft_1_1detail_1_1rev__iter.html#af7b8c2f1534d3038ba2a3c6b9919e134">rev_iter</a>(<span class="keyword">const</span> <a class="code hl_class" href="classpocketfft_1_1detail_1_1arr__info.html">arr_info</a> &<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">arr_</a>, <span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#a885ee37fcf564a268a5c8ca9ea8603e1">shape_t</a> &<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axes</a>)</div>
|
|
<div class="line"><a id="l02980" name="l02980"></a><span class="lineno"> 2980</span> : pos(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">arr_</a>.ndim(), 0), <a class="code hl_class" href="classpocketfft_1_1detail_1_1arr.html">arr</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">arr_</a>), rev_axis(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">arr_</a>.ndim(), 0),</div>
|
|
<div class="line"><a id="l02981" name="l02981"></a><span class="lineno"> 2981</span> rev_jump(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">arr_</a>.ndim(), 1), p(0), rp(0)</div>
|
|
<div class="line"><a id="l02982" name="l02982"></a><span class="lineno"> 2982</span> {</div>
|
|
<div class="line"><a id="l02983" name="l02983"></a><span class="lineno"> 2983</span> <span class="keywordflow">for</span> (<span class="keyword">auto</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ax</a>: <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axes</a>)</div>
|
|
<div class="line"><a id="l02984" name="l02984"></a><span class="lineno"> 2984</span> rev_axis[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ax</a>]=1;</div>
|
|
<div class="line"><a id="l02985" name="l02985"></a><span class="lineno"> 2985</span> last_axis = <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axes</a>.back();</div>
|
|
<div class="line"><a id="l02986" name="l02986"></a><span class="lineno"> 2986</span> last_size = <a class="code hl_class" href="classpocketfft_1_1detail_1_1arr.html">arr</a>.shape(last_axis)/2 + 1;</div>
|
|
<div class="line"><a id="l02987" name="l02987"></a><span class="lineno"> 2987</span> shp = <a class="code hl_class" href="classpocketfft_1_1detail_1_1arr.html">arr</a>.shape();</div>
|
|
<div class="line"><a id="l02988" name="l02988"></a><span class="lineno"> 2988</span> shp[last_axis] = last_size;</div>
|
|
<div class="line"><a id="l02989" name="l02989"></a><span class="lineno"> 2989</span> rem=1;</div>
|
|
<div class="line"><a id="l02990" name="l02990"></a><span class="lineno"> 2990</span> <span class="keywordflow">for</span> (<span class="keyword">auto</span> i: shp)</div>
|
|
<div class="line"><a id="l02991" name="l02991"></a><span class="lineno"> 2991</span> rem *= i;</div>
|
|
<div class="line"><a id="l02992" name="l02992"></a><span class="lineno"> 2992</span> }</div>
|
|
</div>
|
|
<div class="foldopen" id="foldopen02993" data-start="{" data-end="}">
|
|
<div class="line"><a id="l02993" name="l02993"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1rev__iter.html#ad1918c84ae963188afc7599629b29686"> 2993</a></span> <span class="keywordtype">void</span> <a class="code hl_function" href="classpocketfft_1_1detail_1_1rev__iter.html#ad1918c84ae963188afc7599629b29686">advance</a>()</div>
|
|
<div class="line"><a id="l02994" name="l02994"></a><span class="lineno"> 2994</span> {</div>
|
|
<div class="line"><a id="l02995" name="l02995"></a><span class="lineno"> 2995</span> --rem;</div>
|
|
<div class="line"><a id="l02996" name="l02996"></a><span class="lineno"> 2996</span> <span class="keywordflow">for</span> (<span class="keywordtype">int</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">i_</a>=<span class="keywordtype">int</span>(pos.size())-1; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">i_</a>>=0; --<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">i_</a>)</div>
|
|
<div class="line"><a id="l02997" name="l02997"></a><span class="lineno"> 2997</span> {</div>
|
|
<div class="line"><a id="l02998" name="l02998"></a><span class="lineno"> 2998</span> <span class="keyword">auto</span> i = size_t(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">i_</a>);</div>
|
|
<div class="line"><a id="l02999" name="l02999"></a><span class="lineno"> 2999</span> p += <a class="code hl_class" href="classpocketfft_1_1detail_1_1arr.html">arr</a>.stride(i);</div>
|
|
<div class="line"><a id="l03000" name="l03000"></a><span class="lineno"> 3000</span> <span class="keywordflow">if</span> (!rev_axis[i])</div>
|
|
<div class="line"><a id="l03001" name="l03001"></a><span class="lineno"> 3001</span> rp += <a class="code hl_class" href="classpocketfft_1_1detail_1_1arr.html">arr</a>.stride(i);</div>
|
|
<div class="line"><a id="l03002" name="l03002"></a><span class="lineno"> 3002</span> <span class="keywordflow">else</span></div>
|
|
<div class="line"><a id="l03003" name="l03003"></a><span class="lineno"> 3003</span> {</div>
|
|
<div class="line"><a id="l03004" name="l03004"></a><span class="lineno"> 3004</span> rp -= <a class="code hl_class" href="classpocketfft_1_1detail_1_1arr.html">arr</a>.stride(i);</div>
|
|
<div class="line"><a id="l03005" name="l03005"></a><span class="lineno"> 3005</span> <span class="keywordflow">if</span> (rev_jump[i])</div>
|
|
<div class="line"><a id="l03006" name="l03006"></a><span class="lineno"> 3006</span> {</div>
|
|
<div class="line"><a id="l03007" name="l03007"></a><span class="lineno"> 3007</span> rp += ptrdiff_t(<a class="code hl_class" href="classpocketfft_1_1detail_1_1arr.html">arr</a>.shape(i))*<a class="code hl_class" href="classpocketfft_1_1detail_1_1arr.html">arr</a>.stride(i);</div>
|
|
<div class="line"><a id="l03008" name="l03008"></a><span class="lineno"> 3008</span> rev_jump[i] = 0;</div>
|
|
<div class="line"><a id="l03009" name="l03009"></a><span class="lineno"> 3009</span> }</div>
|
|
<div class="line"><a id="l03010" name="l03010"></a><span class="lineno"> 3010</span> }</div>
|
|
<div class="line"><a id="l03011" name="l03011"></a><span class="lineno"> 3011</span> <span class="keywordflow">if</span> (++pos[i] < shp[i])</div>
|
|
<div class="line"><a id="l03012" name="l03012"></a><span class="lineno"> 3012</span> <span class="keywordflow">return</span>;</div>
|
|
<div class="line"><a id="l03013" name="l03013"></a><span class="lineno"> 3013</span> pos[i] = 0;</div>
|
|
<div class="line"><a id="l03014" name="l03014"></a><span class="lineno"> 3014</span> p -= ptrdiff_t(shp[i])*<a class="code hl_class" href="classpocketfft_1_1detail_1_1arr.html">arr</a>.stride(i);</div>
|
|
<div class="line"><a id="l03015" name="l03015"></a><span class="lineno"> 3015</span> <span class="keywordflow">if</span> (rev_axis[i])</div>
|
|
<div class="line"><a id="l03016" name="l03016"></a><span class="lineno"> 3016</span> {</div>
|
|
<div class="line"><a id="l03017" name="l03017"></a><span class="lineno"> 3017</span> rp -= ptrdiff_t(<a class="code hl_class" href="classpocketfft_1_1detail_1_1arr.html">arr</a>.shape(i)-shp[i])*<a class="code hl_class" href="classpocketfft_1_1detail_1_1arr.html">arr</a>.stride(i);</div>
|
|
<div class="line"><a id="l03018" name="l03018"></a><span class="lineno"> 3018</span> rev_jump[i] = 1;</div>
|
|
<div class="line"><a id="l03019" name="l03019"></a><span class="lineno"> 3019</span> }</div>
|
|
<div class="line"><a id="l03020" name="l03020"></a><span class="lineno"> 3020</span> <span class="keywordflow">else</span></div>
|
|
<div class="line"><a id="l03021" name="l03021"></a><span class="lineno"> 3021</span> rp -= ptrdiff_t(shp[i])*<a class="code hl_class" href="classpocketfft_1_1detail_1_1arr.html">arr</a>.stride(i);</div>
|
|
<div class="line"><a id="l03022" name="l03022"></a><span class="lineno"> 3022</span> }</div>
|
|
<div class="line"><a id="l03023" name="l03023"></a><span class="lineno"> 3023</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l03024" name="l03024"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1rev__iter.html#a78c3b4ad19edf9d20cab40ad109e9dd1"> 3024</a></span> ptrdiff_t <a class="code hl_function" href="classpocketfft_1_1detail_1_1rev__iter.html#a78c3b4ad19edf9d20cab40ad109e9dd1">ofs</a>()<span class="keyword"> const </span>{ <span class="keywordflow">return</span> p; }</div>
|
|
<div class="line"><a id="l03025" name="l03025"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1rev__iter.html#a7f112afa76cb7a4c29cff217a6f5f5a9"> 3025</a></span> ptrdiff_t <a class="code hl_function" href="classpocketfft_1_1detail_1_1rev__iter.html#a7f112afa76cb7a4c29cff217a6f5f5a9">rev_ofs</a>()<span class="keyword"> const </span>{ <span class="keywordflow">return</span> rp; }</div>
|
|
<div class="line"><a id="l03026" name="l03026"></a><span class="lineno"><a class="line" href="classpocketfft_1_1detail_1_1rev__iter.html#a143637135c441a4b9a2959c2370d8c63"> 3026</a></span> <span class="keywordtype">size_t</span> <a class="code hl_function" href="classpocketfft_1_1detail_1_1rev__iter.html#a143637135c441a4b9a2959c2370d8c63">remaining</a>()<span class="keyword"> const </span>{ <span class="keywordflow">return</span> rem; }</div>
|
|
<div class="line"><a id="l03027" name="l03027"></a><span class="lineno"> 3027</span> };</div>
|
|
</div>
|
|
<div class="line"><a id="l03028" name="l03028"></a><span class="lineno"> 3028</span> </div>
|
|
<div class="line"><a id="l03029" name="l03029"></a><span class="lineno"><a class="line" href="structpocketfft_1_1detail_1_1_v_t_y_p_e.html"> 3029</a></span><span class="keyword">template</span><<span class="keyword">typename</span> T> <span class="keyword">struct </span><a class="code hl_struct" href="structpocketfft_1_1detail_1_1_v_t_y_p_e.html">VTYPE</a> {};</div>
|
|
<div class="line"><a id="l03030" name="l03030"></a><span class="lineno"><a class="line" href="namespacepocketfft_1_1detail.html#a3edfb93aeed2f8258183d463ea291d62"> 3030</a></span><span class="keyword">template</span> <<span class="keyword">typename</span> T> <span class="keyword">using </span><a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#a3edfb93aeed2f8258183d463ea291d62">vtype_t</a> = <span class="keyword">typename</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1_v_t_y_p_e.html">VTYPE<T>::type</a>;</div>
|
|
<div class="line"><a id="l03031" name="l03031"></a><span class="lineno"> 3031</span> </div>
|
|
<div class="line"><a id="l03032" name="l03032"></a><span class="lineno"> 3032</span><span class="preprocessor">#ifndef POCKETFFT_NO_VECTORS</span></div>
|
|
<div class="line"><a id="l03033" name="l03033"></a><span class="lineno"> 3033</span><span class="keyword">template</span><> <span class="keyword">struct </span><a class="code hl_struct" href="structpocketfft_1_1detail_1_1_v_t_y_p_e.html">VTYPE</a><<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">float</a>></div>
|
|
<div class="line"><a id="l03034" name="l03034"></a><span class="lineno"> 3034</span> {</div>
|
|
<div class="line"><a id="l03035" name="l03035"></a><span class="lineno"> 3035</span> <span class="keyword">using </span>type = <span class="keywordtype">float</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">__attribute__</a> ((<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">vector_size</a> (<a class="code hl_struct" href="structpocketfft_1_1detail_1_1_v_l_e_n.html">VLEN<float>::val</a>*<span class="keyword">sizeof</span>(<span class="keywordtype">float</span>))));</div>
|
|
<div class="line"><a id="l03036" name="l03036"></a><span class="lineno"> 3036</span> };</div>
|
|
<div class="line"><a id="l03037" name="l03037"></a><span class="lineno"> 3037</span><span class="keyword">template</span><> <span class="keyword">struct </span>VTYPE<double></div>
|
|
<div class="line"><a id="l03038" name="l03038"></a><span class="lineno"> 3038</span> {</div>
|
|
<div class="line"><a id="l03039" name="l03039"></a><span class="lineno"> 3039</span> <span class="keyword">using </span>type = <span class="keywordtype">double</span> __attribute__ ((vector_size (VLEN<double>::val*<span class="keyword">sizeof</span>(<span class="keywordtype">double</span>))));</div>
|
|
<div class="line"><a id="l03040" name="l03040"></a><span class="lineno"> 3040</span> };</div>
|
|
<div class="line"><a id="l03041" name="l03041"></a><span class="lineno"> 3041</span><span class="keyword">template</span><> <span class="keyword">struct </span>VTYPE<long double></div>
|
|
<div class="line"><a id="l03042" name="l03042"></a><span class="lineno"> 3042</span> {</div>
|
|
<div class="line"><a id="l03043" name="l03043"></a><span class="lineno"> 3043</span> <span class="keyword">using </span>type = <span class="keywordtype">long</span> <span class="keywordtype">double</span> __attribute__ ((vector_size (VLEN<long double>::val*<span class="keyword">sizeof</span>(<span class="keywordtype">long</span> <span class="keywordtype">double</span>))));</div>
|
|
<div class="line"><a id="l03044" name="l03044"></a><span class="lineno"> 3044</span> };</div>
|
|
<div class="line"><a id="l03045" name="l03045"></a><span class="lineno"> 3045</span><span class="preprocessor">#endif</span></div>
|
|
<div class="line"><a id="l03046" name="l03046"></a><span class="lineno"> 3046</span> </div>
|
|
<div class="foldopen" id="foldopen03047" data-start="{" data-end="}">
|
|
<div class="line"><a id="l03047" name="l03047"></a><span class="lineno"><a class="line" href="namespacepocketfft_1_1detail.html#a4db03cbcd9d43d9e0b0b9067713c80e9"> 3047</a></span><span class="keyword">template</span><<span class="keyword">typename</span> T> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">arr<char></a> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a4db03cbcd9d43d9e0b0b9067713c80e9">alloc_tmp</a>(<span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#a885ee37fcf564a268a5c8ca9ea8603e1">shape_t</a> &shape,</div>
|
|
<div class="line"><a id="l03048" name="l03048"></a><span class="lineno"> 3048</span> <span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axsize</a>, <span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">elemsize</a>)</div>
|
|
<div class="line"><a id="l03049" name="l03049"></a><span class="lineno"> 3049</span> {</div>
|
|
<div class="line"><a id="l03050" name="l03050"></a><span class="lineno"> 3050</span> <span class="keyword">auto</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">othersize</a> = <a class="code hl_function" href="structpocketfft_1_1detail_1_1util.html#a33acae07b20b28fe4658bc338fce1b89">util::prod</a>(shape)/<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axsize</a>;</div>
|
|
<div class="line"><a id="l03051" name="l03051"></a><span class="lineno"> 3051</span> <span class="keyword">auto</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tmpsize</a> = <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axsize</a>*((<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">othersize</a>>=<a class="code hl_struct" href="structpocketfft_1_1detail_1_1_v_l_e_n.html">VLEN<T>::val</a>) ? <a class="code hl_struct" href="structpocketfft_1_1detail_1_1_v_l_e_n.html">VLEN<T>::val</a> : 1);</div>
|
|
<div class="line"><a id="l03052" name="l03052"></a><span class="lineno"> 3052</span> <span class="keywordflow">return</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">arr<char></a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tmpsize</a>*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">elemsize</a>);</div>
|
|
<div class="line"><a id="l03053" name="l03053"></a><span class="lineno"> 3053</span> }</div>
|
|
</div>
|
|
<div class="foldopen" id="foldopen03054" data-start="{" data-end="}">
|
|
<div class="line"><a id="l03054" name="l03054"></a><span class="lineno"><a class="line" href="namespacepocketfft_1_1detail.html#a13832735696303b9559c4663631d5475"> 3054</a></span><span class="keyword">template</span><<span class="keyword">typename</span> T> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">arr<char></a> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a4db03cbcd9d43d9e0b0b9067713c80e9">alloc_tmp</a>(<span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#a885ee37fcf564a268a5c8ca9ea8603e1">shape_t</a> &shape,</div>
|
|
<div class="line"><a id="l03055" name="l03055"></a><span class="lineno"> 3055</span> <span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#a885ee37fcf564a268a5c8ca9ea8603e1">shape_t</a> &<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axes</a>, <span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">elemsize</a>)</div>
|
|
<div class="line"><a id="l03056" name="l03056"></a><span class="lineno"> 3056</span> {</div>
|
|
<div class="line"><a id="l03057" name="l03057"></a><span class="lineno"> 3057</span> <span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">fullsize</a>=<a class="code hl_function" href="structpocketfft_1_1detail_1_1util.html#a33acae07b20b28fe4658bc338fce1b89">util::prod</a>(shape);</div>
|
|
<div class="line"><a id="l03058" name="l03058"></a><span class="lineno"> 3058</span> <span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tmpsize</a>=0;</div>
|
|
<div class="line"><a id="l03059" name="l03059"></a><span class="lineno"> 3059</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i=0; i<<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axes</a>.size(); ++i)</div>
|
|
<div class="line"><a id="l03060" name="l03060"></a><span class="lineno"> 3060</span> {</div>
|
|
<div class="line"><a id="l03061" name="l03061"></a><span class="lineno"> 3061</span> <span class="keyword">auto</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axsize</a> = shape[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axes</a>[i]];</div>
|
|
<div class="line"><a id="l03062" name="l03062"></a><span class="lineno"> 3062</span> <span class="keyword">auto</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">othersize</a> = <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">fullsize</a>/<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axsize</a>;</div>
|
|
<div class="line"><a id="l03063" name="l03063"></a><span class="lineno"> 3063</span> <span class="keyword">auto</span> sz = <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axsize</a>*((<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">othersize</a>>=<a class="code hl_struct" href="structpocketfft_1_1detail_1_1_v_l_e_n.html">VLEN<T>::val</a>) ? <a class="code hl_struct" href="structpocketfft_1_1detail_1_1_v_l_e_n.html">VLEN<T>::val</a> : 1);</div>
|
|
<div class="line"><a id="l03064" name="l03064"></a><span class="lineno"> 3064</span> <span class="keywordflow">if</span> (sz><a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tmpsize</a>) <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tmpsize</a>=sz;</div>
|
|
<div class="line"><a id="l03065" name="l03065"></a><span class="lineno"> 3065</span> }</div>
|
|
<div class="line"><a id="l03066" name="l03066"></a><span class="lineno"> 3066</span> <span class="keywordflow">return</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">arr<char></a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tmpsize</a>*<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">elemsize</a>);</div>
|
|
<div class="line"><a id="l03067" name="l03067"></a><span class="lineno"> 3067</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l03068" name="l03068"></a><span class="lineno"> 3068</span> </div>
|
|
<div class="foldopen" id="foldopen03069" data-start="{" data-end="}">
|
|
<div class="line"><a id="l03069" name="l03069"></a><span class="lineno"><a class="line" href="namespacepocketfft_1_1detail.html#aff05be3064743c1143b19318ab12ad4a"> 3069</a></span><span class="keyword">template</span> <<span class="keyword">typename</span> T, <span class="keywordtype">size_t</span> vlen> <span class="keywordtype">void</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#aff05be3064743c1143b19318ab12ad4a">copy_input</a>(<span class="keyword">const</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">multi_iter<vlen></a> &<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">it</a>,</div>
|
|
<div class="line"><a id="l03070" name="l03070"></a><span class="lineno"> 3070</span> <span class="keyword">const</span> <a class="code hl_class" href="classpocketfft_1_1detail_1_1cndarr.html">cndarr</a><<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cmplx<T></a>> &src, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cmplx</a><<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">vtype_t<T></a>> *<a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> dst)</div>
|
|
<div class="line"><a id="l03071" name="l03071"></a><span class="lineno"> 3071</span> {</div>
|
|
<div class="line"><a id="l03072" name="l03072"></a><span class="lineno"> 3072</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i=0; i<<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">it</a>.length_in(); ++i)</div>
|
|
<div class="line"><a id="l03073" name="l03073"></a><span class="lineno"> 3073</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">j</a>=0; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">j</a><<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">vlen</a>; ++<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">j</a>)</div>
|
|
<div class="line"><a id="l03074" name="l03074"></a><span class="lineno"> 3074</span> {</div>
|
|
<div class="line"><a id="l03075" name="l03075"></a><span class="lineno"> 3075</span> dst[i].<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a>[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">j</a>] = src[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">it</a>.iofs(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">j</a>,i)].<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a>;</div>
|
|
<div class="line"><a id="l03076" name="l03076"></a><span class="lineno"> 3076</span> dst[i].<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#a35d2dce1b7de5f37d7029e639bc7f23d">i</a>[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">j</a>] = src[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">it</a>.iofs(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">j</a>,i)].<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#a35d2dce1b7de5f37d7029e639bc7f23d">i</a>;</div>
|
|
<div class="line"><a id="l03077" name="l03077"></a><span class="lineno"> 3077</span> }</div>
|
|
<div class="line"><a id="l03078" name="l03078"></a><span class="lineno"> 3078</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l03079" name="l03079"></a><span class="lineno"> 3079</span> </div>
|
|
<div class="foldopen" id="foldopen03080" data-start="{" data-end="}">
|
|
<div class="line"><a id="l03080" name="l03080"></a><span class="lineno"><a class="line" href="namespacepocketfft_1_1detail.html#a30fc708f9d8f9cfa74194925c7863c0a"> 3080</a></span><span class="keyword">template</span> <<span class="keyword">typename</span> T, <span class="keywordtype">size_t</span> vlen> <span class="keywordtype">void</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#aff05be3064743c1143b19318ab12ad4a">copy_input</a>(<span class="keyword">const</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">multi_iter<vlen></a> &<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">it</a>,</div>
|
|
<div class="line"><a id="l03081" name="l03081"></a><span class="lineno"> 3081</span> <span class="keyword">const</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cndarr<T></a> &src, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">vtype_t<T></a> *<a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> dst)</div>
|
|
<div class="line"><a id="l03082" name="l03082"></a><span class="lineno"> 3082</span> {</div>
|
|
<div class="line"><a id="l03083" name="l03083"></a><span class="lineno"> 3083</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i=0; i<<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">it</a>.length_in(); ++i)</div>
|
|
<div class="line"><a id="l03084" name="l03084"></a><span class="lineno"> 3084</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">j</a>=0; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">j</a><<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">vlen</a>; ++<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">j</a>)</div>
|
|
<div class="line"><a id="l03085" name="l03085"></a><span class="lineno"> 3085</span> dst[i][<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">j</a>] = src[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">it</a>.iofs(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">j</a>,i)];</div>
|
|
<div class="line"><a id="l03086" name="l03086"></a><span class="lineno"> 3086</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l03087" name="l03087"></a><span class="lineno"> 3087</span> </div>
|
|
<div class="foldopen" id="foldopen03088" data-start="{" data-end="}">
|
|
<div class="line"><a id="l03088" name="l03088"></a><span class="lineno"><a class="line" href="namespacepocketfft_1_1detail.html#a3387bd35f237870e42b8461769e6aec4"> 3088</a></span><span class="keyword">template</span> <<span class="keyword">typename</span> T, <span class="keywordtype">size_t</span> vlen> <span class="keywordtype">void</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#aff05be3064743c1143b19318ab12ad4a">copy_input</a>(<span class="keyword">const</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">multi_iter<vlen></a> &<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">it</a>,</div>
|
|
<div class="line"><a id="l03089" name="l03089"></a><span class="lineno"> 3089</span> <span class="keyword">const</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cndarr<T></a> &src, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a> *<a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> dst)</div>
|
|
<div class="line"><a id="l03090" name="l03090"></a><span class="lineno"> 3090</span> {</div>
|
|
<div class="line"><a id="l03091" name="l03091"></a><span class="lineno"> 3091</span> <span class="keywordflow">if</span> (dst == &src[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">it</a>.iofs(0)]) <span class="keywordflow">return</span>; <span class="comment">// in-place</span></div>
|
|
<div class="line"><a id="l03092" name="l03092"></a><span class="lineno"> 3092</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i=0; i<<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">it</a>.length_in(); ++i)</div>
|
|
<div class="line"><a id="l03093" name="l03093"></a><span class="lineno"> 3093</span> dst[i] = src[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">it</a>.iofs(i)];</div>
|
|
<div class="line"><a id="l03094" name="l03094"></a><span class="lineno"> 3094</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l03095" name="l03095"></a><span class="lineno"> 3095</span> </div>
|
|
<div class="foldopen" id="foldopen03096" data-start="{" data-end="}">
|
|
<div class="line"><a id="l03096" name="l03096"></a><span class="lineno"><a class="line" href="namespacepocketfft_1_1detail.html#a1523a037300a8da05db210b802d9cb0e"> 3096</a></span><span class="keyword">template</span><<span class="keyword">typename</span> T, <span class="keywordtype">size_t</span> vlen> <span class="keywordtype">void</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a1523a037300a8da05db210b802d9cb0e">copy_output</a>(<span class="keyword">const</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">multi_iter<vlen></a> &<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">it</a>,</div>
|
|
<div class="line"><a id="l03097" name="l03097"></a><span class="lineno"> 3097</span> <span class="keyword">const</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cmplx</a><<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">vtype_t<T></a>> *<a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> src, <a class="code hl_class" href="classpocketfft_1_1detail_1_1ndarr.html">ndarr</a><<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cmplx<T></a>> &dst)</div>
|
|
<div class="line"><a id="l03098" name="l03098"></a><span class="lineno"> 3098</span> {</div>
|
|
<div class="line"><a id="l03099" name="l03099"></a><span class="lineno"> 3099</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i=0; i<<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">it</a>.length_out(); ++i)</div>
|
|
<div class="line"><a id="l03100" name="l03100"></a><span class="lineno"> 3100</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">j</a>=0; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">j</a><<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">vlen</a>; ++<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">j</a>)</div>
|
|
<div class="line"><a id="l03101" name="l03101"></a><span class="lineno"> 3101</span> dst[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">it</a>.oofs(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">j</a>,i)].<a class="code hl_function" href="structpocketfft_1_1detail_1_1cmplx.html#a647fece372b64b13c4a7e5877d09a807">Set</a>(src[i].r[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">j</a>],src[i].i[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">j</a>]);</div>
|
|
<div class="line"><a id="l03102" name="l03102"></a><span class="lineno"> 3102</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l03103" name="l03103"></a><span class="lineno"> 3103</span> </div>
|
|
<div class="foldopen" id="foldopen03104" data-start="{" data-end="}">
|
|
<div class="line"><a id="l03104" name="l03104"></a><span class="lineno"><a class="line" href="namespacepocketfft_1_1detail.html#a21980853aca4d92ed06e3dcffe7ef660"> 3104</a></span><span class="keyword">template</span><<span class="keyword">typename</span> T, <span class="keywordtype">size_t</span> vlen> <span class="keywordtype">void</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a1523a037300a8da05db210b802d9cb0e">copy_output</a>(<span class="keyword">const</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">multi_iter<vlen></a> &<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">it</a>,</div>
|
|
<div class="line"><a id="l03105" name="l03105"></a><span class="lineno"> 3105</span> <span class="keyword">const</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">vtype_t<T></a> *<a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> src, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ndarr<T></a> &dst)</div>
|
|
<div class="line"><a id="l03106" name="l03106"></a><span class="lineno"> 3106</span> {</div>
|
|
<div class="line"><a id="l03107" name="l03107"></a><span class="lineno"> 3107</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i=0; i<<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">it</a>.length_out(); ++i)</div>
|
|
<div class="line"><a id="l03108" name="l03108"></a><span class="lineno"> 3108</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">j</a>=0; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">j</a><<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">vlen</a>; ++<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">j</a>)</div>
|
|
<div class="line"><a id="l03109" name="l03109"></a><span class="lineno"> 3109</span> dst[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">it</a>.oofs(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">j</a>,i)] = src[i][<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">j</a>];</div>
|
|
<div class="line"><a id="l03110" name="l03110"></a><span class="lineno"> 3110</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l03111" name="l03111"></a><span class="lineno"> 3111</span> </div>
|
|
<div class="foldopen" id="foldopen03112" data-start="{" data-end="}">
|
|
<div class="line"><a id="l03112" name="l03112"></a><span class="lineno"><a class="line" href="namespacepocketfft_1_1detail.html#a310481c334e46674710ba794ad7403c0"> 3112</a></span><span class="keyword">template</span><<span class="keyword">typename</span> T, <span class="keywordtype">size_t</span> vlen> <span class="keywordtype">void</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a1523a037300a8da05db210b802d9cb0e">copy_output</a>(<span class="keyword">const</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">multi_iter<vlen></a> &<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">it</a>,</div>
|
|
<div class="line"><a id="l03113" name="l03113"></a><span class="lineno"> 3113</span> <span class="keyword">const</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a> *<a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> src, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ndarr<T></a> &dst)</div>
|
|
<div class="line"><a id="l03114" name="l03114"></a><span class="lineno"> 3114</span> {</div>
|
|
<div class="line"><a id="l03115" name="l03115"></a><span class="lineno"> 3115</span> <span class="keywordflow">if</span> (src == &dst[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">it</a>.oofs(0)]) <span class="keywordflow">return</span>; <span class="comment">// in-place</span></div>
|
|
<div class="line"><a id="l03116" name="l03116"></a><span class="lineno"> 3116</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i=0; i<<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">it</a>.length_out(); ++i)</div>
|
|
<div class="line"><a id="l03117" name="l03117"></a><span class="lineno"> 3117</span> dst[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">it</a>.oofs(i)] = src[i];</div>
|
|
<div class="line"><a id="l03118" name="l03118"></a><span class="lineno"> 3118</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l03119" name="l03119"></a><span class="lineno"> 3119</span> </div>
|
|
<div class="line"><a id="l03120" name="l03120"></a><span class="lineno"><a class="line" href="structpocketfft_1_1detail_1_1add__vec.html#a7568dc83136c1b41eb71dcb78527227e"> 3120</a></span><span class="keyword">template</span> <<span class="keyword">typename</span> T> <span class="keyword">struct </span><a class="code hl_struct" href="structpocketfft_1_1detail_1_1add__vec.html">add_vec</a> { <span class="keyword">using </span><a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">type</a> = <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">vtype_t<T></a>; };</div>
|
|
<div class="foldopen" id="foldopen03121" data-start="{" data-end="};">
|
|
<div class="line"><a id="l03121" name="l03121"></a><span class="lineno"><a class="line" href="structpocketfft_1_1detail_1_1add__vec_3_01cmplx_3_01_t_01_4_01_4.html"> 3121</a></span><span class="keyword">template</span> <<span class="keyword">typename</span> T> <span class="keyword">struct </span><a class="code hl_struct" href="structpocketfft_1_1detail_1_1add__vec.html">add_vec</a><<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cmplx</a><<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a>>></div>
|
|
<div class="line"><a id="l03122" name="l03122"></a><span class="lineno"><a class="line" href="structpocketfft_1_1detail_1_1add__vec_3_01cmplx_3_01_t_01_4_01_4.html#a257b1c81fb9f559c48ee90497013494e"> 3122</a></span> { <span class="keyword">using </span><a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">type</a> = <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cmplx<vtype_t<T></a>>; };</div>
|
|
</div>
|
|
<div class="line"><a id="l03123" name="l03123"></a><span class="lineno"><a class="line" href="namespacepocketfft_1_1detail.html#a421aa74fbee775a96463246f72b144d6"> 3123</a></span><span class="keyword">template</span> <<span class="keyword">typename</span> T> <span class="keyword">using </span><a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#a421aa74fbee775a96463246f72b144d6">add_vec_t</a> = <span class="keyword">typename</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">add_vec<T>::type</a>;</div>
|
|
<div class="line"><a id="l03124" name="l03124"></a><span class="lineno"> 3124</span> </div>
|
|
<div class="line"><a id="l03125" name="l03125"></a><span class="lineno"> 3125</span><span class="keyword">template</span><<span class="keyword">typename</span> Tplan, <span class="keyword">typename</span> T, <span class="keyword">typename</span> T0, <span class="keyword">typename</span> Exec></div>
|
|
<div class="foldopen" id="foldopen03126" data-start="{" data-end="}">
|
|
<div class="line"><a id="l03126" name="l03126"></a><span class="lineno"><a class="line" href="namespacepocketfft_1_1detail.html#ab47f52551920af5eb9f57fbbded0f4f0"> 3126</a></span><a class="code hl_define" href="pocketfft_8h.html#a7020984e0ca1d6e565629ca6e7c1a7e0">POCKETFFT_NOINLINE</a> <span class="keywordtype">void</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#ab47f52551920af5eb9f57fbbded0f4f0">general_nd</a>(<span class="keyword">const</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cndarr<T></a> &in, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ndarr<T></a> &out,</div>
|
|
<div class="line"><a id="l03127" name="l03127"></a><span class="lineno"> 3127</span> <span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#a885ee37fcf564a268a5c8ca9ea8603e1">shape_t</a> &<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axes</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T0</a> fct, <span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nthreads</a>, <span class="keyword">const</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">Exec</a> & exec,</div>
|
|
<div class="line"><a id="l03128" name="l03128"></a><span class="lineno"> 3128</span> <span class="keyword">const</span> <span class="keywordtype">bool</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">allow_inplace</a>=<span class="keyword">true</span>)</div>
|
|
<div class="line"><a id="l03129" name="l03129"></a><span class="lineno"> 3129</span> {</div>
|
|
<div class="line"><a id="l03130" name="l03130"></a><span class="lineno"> 3130</span> std::shared_ptr<Tplan> plan;</div>
|
|
<div class="line"><a id="l03131" name="l03131"></a><span class="lineno"> 3131</span> </div>
|
|
<div class="line"><a id="l03132" name="l03132"></a><span class="lineno"> 3132</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">iax</a>=0; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">iax</a><<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axes</a>.size(); ++<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">iax</a>)</div>
|
|
<div class="line"><a id="l03133" name="l03133"></a><span class="lineno"> 3133</span> {</div>
|
|
<div class="line"><a id="l03134" name="l03134"></a><span class="lineno"> 3134</span> <span class="keywordtype">size_t</span> len=in.shape(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axes</a>[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">iax</a>]);</div>
|
|
<div class="line"><a id="l03135" name="l03135"></a><span class="lineno"> 3135</span> <span class="keywordflow">if</span> ((!plan) || (len!=plan->length()))</div>
|
|
<div class="line"><a id="l03136" name="l03136"></a><span class="lineno"> 3136</span> plan = <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">get_plan<Tplan></a>(len);</div>
|
|
<div class="line"><a id="l03137" name="l03137"></a><span class="lineno"> 3137</span> </div>
|
|
<div class="line"><a id="l03138" name="l03138"></a><span class="lineno"> 3138</span> <a class="code hl_function" href="namespacepocketfft_1_1detail_1_1threading.html#a4fcf674db39f0e2c1c59d48491daed6e">threading::thread_map</a>(</div>
|
|
<div class="line"><a id="l03139" name="l03139"></a><span class="lineno"> 3139</span> <a class="code hl_function" href="structpocketfft_1_1detail_1_1util.html#a3b012d5a19215bcd32cf6e228556fa87">util::thread_count</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nthreads</a>, in.shape(), <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axes</a>[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">iax</a>], <a class="code hl_struct" href="structpocketfft_1_1detail_1_1_v_l_e_n.html">VLEN<T>::val</a>),</div>
|
|
<div class="line"><a id="l03140" name="l03140"></a><span class="lineno"> 3140</span> [&] {</div>
|
|
<div class="line"><a id="l03141" name="l03141"></a><span class="lineno"> 3141</span> constexpr auto vlen = VLEN<T0>::val;</div>
|
|
<div class="line"><a id="l03142" name="l03142"></a><span class="lineno"> 3142</span> auto storage = alloc_tmp<T0>(in.shape(), len, sizeof(T));</div>
|
|
<div class="line"><a id="l03143" name="l03143"></a><span class="lineno"> 3143</span> const auto &tin(iax==0? in : out);</div>
|
|
<div class="line"><a id="l03144" name="l03144"></a><span class="lineno"> 3144</span> multi_iter<vlen> it(tin, out, axes[iax]);</div>
|
|
<div class="line"><a id="l03145" name="l03145"></a><span class="lineno"> 3145</span>#ifndef POCKETFFT_NO_VECTORS</div>
|
|
<div class="line"><a id="l03146" name="l03146"></a><span class="lineno"> 3146</span> if (vlen>1)</div>
|
|
<div class="line"><a id="l03147" name="l03147"></a><span class="lineno"> 3147</span> while (it.remaining()>=vlen)</div>
|
|
<div class="line"><a id="l03148" name="l03148"></a><span class="lineno"> 3148</span> {</div>
|
|
<div class="line"><a id="l03149" name="l03149"></a><span class="lineno"> 3149</span> it.advance(vlen);</div>
|
|
<div class="line"><a id="l03150" name="l03150"></a><span class="lineno"> 3150</span> auto tdatav = reinterpret_cast<add_vec_t<T> *>(storage.data());</div>
|
|
<div class="line"><a id="l03151" name="l03151"></a><span class="lineno"> 3151</span> exec(it, tin, out, tdatav, *plan, fct);</div>
|
|
<div class="line"><a id="l03152" name="l03152"></a><span class="lineno"> 3152</span> }</div>
|
|
<div class="line"><a id="l03153" name="l03153"></a><span class="lineno"> 3153</span>#<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">endif</a></div>
|
|
<div class="line"><a id="l03154" name="l03154"></a><span class="lineno"> 3154</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">while</a> (<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">it</a>.remaining()>0)</div>
|
|
<div class="line"><a id="l03155" name="l03155"></a><span class="lineno"> 3155</span> {</div>
|
|
<div class="line"><a id="l03156" name="l03156"></a><span class="lineno"> 3156</span> it.advance(1);</div>
|
|
<div class="line"><a id="l03157" name="l03157"></a><span class="lineno"> 3157</span> auto buf = allow_inplace && it.stride_out() == sizeof(T) ?</div>
|
|
<div class="line"><a id="l03158" name="l03158"></a><span class="lineno"> 3158</span> &out[it.oofs(0)] : reinterpret_cast<T *>(storage.data());</div>
|
|
<div class="line"><a id="l03159" name="l03159"></a><span class="lineno"> 3159</span> exec(it, tin, out, buf, *plan, fct);</div>
|
|
<div class="line"><a id="l03160" name="l03160"></a><span class="lineno"> 3160</span> }</div>
|
|
<div class="line"><a id="l03161" name="l03161"></a><span class="lineno"> 3161</span> }); <span class="comment">// end of parallel region</span></div>
|
|
<div class="line"><a id="l03162" name="l03162"></a><span class="lineno"> 3162</span> fct = <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T0</a>(1); <span class="comment">// factor has been applied, use 1 for remaining axes</span></div>
|
|
<div class="line"><a id="l03163" name="l03163"></a><span class="lineno"> 3163</span> }</div>
|
|
<div class="line"><a id="l03164" name="l03164"></a><span class="lineno"> 3164</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l03165" name="l03165"></a><span class="lineno"> 3165</span> </div>
|
|
<div class="foldopen" id="foldopen03166" data-start="{" data-end="};">
|
|
<div class="line"><a id="l03166" name="l03166"></a><span class="lineno"><a class="line" href="structpocketfft_1_1detail_1_1_exec_c2_c.html"> 3166</a></span><span class="keyword">struct </span><a class="code hl_struct" href="structpocketfft_1_1detail_1_1_exec_c2_c.html">ExecC2C</a></div>
|
|
<div class="line"><a id="l03167" name="l03167"></a><span class="lineno"> 3167</span> {</div>
|
|
<div class="line"><a id="l03168" name="l03168"></a><span class="lineno"><a class="line" href="structpocketfft_1_1detail_1_1_exec_c2_c.html#a63e27292b327597674deede9debe1c43"> 3168</a></span> <span class="keywordtype">bool</span> <a class="code hl_variable" href="structpocketfft_1_1detail_1_1_exec_c2_c.html#a63e27292b327597674deede9debe1c43">forward</a>;</div>
|
|
<div class="line"><a id="l03169" name="l03169"></a><span class="lineno"> 3169</span> </div>
|
|
<div class="foldopen" id="foldopen03170" data-start="{" data-end="}">
|
|
<div class="line"><a id="l03170" name="l03170"></a><span class="lineno"><a class="line" href="structpocketfft_1_1detail_1_1_exec_c2_c.html#a4fd637f1a6d335826789af28ac089ecb"> 3170</a></span> <span class="keyword">template</span> <<span class="keyword">typename</span> T0, <span class="keyword">typename</span> T, <span class="keywordtype">size_t</span> vlen> <span class="keywordtype">void</span> <a class="code hl_function" href="structpocketfft_1_1detail_1_1_exec_c2_c.html#a4fd637f1a6d335826789af28ac089ecb">operator () </a>(</div>
|
|
<div class="line"><a id="l03171" name="l03171"></a><span class="lineno"> 3171</span> <span class="keyword">const</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">multi_iter<vlen></a> &<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">it</a>, <span class="keyword">const</span> <a class="code hl_class" href="classpocketfft_1_1detail_1_1cndarr.html">cndarr</a><<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cmplx<T0></a>> &in,</div>
|
|
<div class="line"><a id="l03172" name="l03172"></a><span class="lineno"> 3172</span> <a class="code hl_class" href="classpocketfft_1_1detail_1_1ndarr.html">ndarr</a><<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cmplx<T0></a>> &out, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a> * <a class="code hl_variable" href="backend_2metal_2allocator_8h.html#a15aa5cc1baf29be08d55cca88509e697">buf</a>, <span class="keyword">const</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">pocketfft_c<T0></a> &plan, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T0</a> fct)<span class="keyword"> const</span></div>
|
|
<div class="line"><a id="l03173" name="l03173"></a><span class="lineno"> 3173</span><span class="keyword"> </span>{</div>
|
|
<div class="line"><a id="l03174" name="l03174"></a><span class="lineno"> 3174</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#aff05be3064743c1143b19318ab12ad4a">copy_input</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">it</a>, in, <a class="code hl_variable" href="backend_2metal_2allocator_8h.html#a15aa5cc1baf29be08d55cca88509e697">buf</a>);</div>
|
|
<div class="line"><a id="l03175" name="l03175"></a><span class="lineno"> 3175</span> plan.exec(<a class="code hl_variable" href="backend_2metal_2allocator_8h.html#a15aa5cc1baf29be08d55cca88509e697">buf</a>, fct, <a class="code hl_variable" href="structpocketfft_1_1detail_1_1_exec_c2_c.html#a63e27292b327597674deede9debe1c43">forward</a>);</div>
|
|
<div class="line"><a id="l03176" name="l03176"></a><span class="lineno"> 3176</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a1523a037300a8da05db210b802d9cb0e">copy_output</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">it</a>, <a class="code hl_variable" href="backend_2metal_2allocator_8h.html#a15aa5cc1baf29be08d55cca88509e697">buf</a>, out);</div>
|
|
<div class="line"><a id="l03177" name="l03177"></a><span class="lineno"> 3177</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l03178" name="l03178"></a><span class="lineno"> 3178</span> };</div>
|
|
</div>
|
|
<div class="line"><a id="l03179" name="l03179"></a><span class="lineno"> 3179</span> </div>
|
|
<div class="foldopen" id="foldopen03180" data-start="{" data-end="}">
|
|
<div class="line"><a id="l03180" name="l03180"></a><span class="lineno"><a class="line" href="namespacepocketfft_1_1detail.html#abac3fcc8ce83800d228774f64c28d4c3"> 3180</a></span><span class="keyword">template</span> <<span class="keyword">typename</span> T, <span class="keywordtype">size_t</span> vlen> <span class="keywordtype">void</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#abac3fcc8ce83800d228774f64c28d4c3">copy_hartley</a>(<span class="keyword">const</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">multi_iter<vlen></a> &<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">it</a>,</div>
|
|
<div class="line"><a id="l03181" name="l03181"></a><span class="lineno"> 3181</span> <span class="keyword">const</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">vtype_t<T></a> *<a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> src, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ndarr<T></a> &dst)</div>
|
|
<div class="line"><a id="l03182" name="l03182"></a><span class="lineno"> 3182</span> {</div>
|
|
<div class="line"><a id="l03183" name="l03183"></a><span class="lineno"> 3183</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">j</a>=0; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">j</a><<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">vlen</a>; ++<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">j</a>)</div>
|
|
<div class="line"><a id="l03184" name="l03184"></a><span class="lineno"> 3184</span> dst[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">it</a>.oofs(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">j</a>,0)] = src[0][<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">j</a>];</div>
|
|
<div class="line"><a id="l03185" name="l03185"></a><span class="lineno"> 3185</span> <span class="keywordtype">size_t</span> i=1, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">i1</a>=1, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">i2</a>=<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">it</a>.length_out()-1;</div>
|
|
<div class="line"><a id="l03186" name="l03186"></a><span class="lineno"> 3186</span> <span class="keywordflow">for</span> (i=1; i<<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">it</a>.length_out()-1; i+=2, ++<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">i1</a>, --<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">i2</a>)</div>
|
|
<div class="line"><a id="l03187" name="l03187"></a><span class="lineno"> 3187</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">j</a>=0; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">j</a><<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">vlen</a>; ++<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">j</a>)</div>
|
|
<div class="line"><a id="l03188" name="l03188"></a><span class="lineno"> 3188</span> {</div>
|
|
<div class="line"><a id="l03189" name="l03189"></a><span class="lineno"> 3189</span> dst[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">it</a>.oofs(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">j</a>,<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">i1</a>)] = src[i][<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">j</a>]+src[i+1][<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">j</a>];</div>
|
|
<div class="line"><a id="l03190" name="l03190"></a><span class="lineno"> 3190</span> dst[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">it</a>.oofs(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">j</a>,<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">i2</a>)] = src[i][<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">j</a>]-src[i+1][<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">j</a>];</div>
|
|
<div class="line"><a id="l03191" name="l03191"></a><span class="lineno"> 3191</span> }</div>
|
|
<div class="line"><a id="l03192" name="l03192"></a><span class="lineno"> 3192</span> <span class="keywordflow">if</span> (i<<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">it</a>.length_out())</div>
|
|
<div class="line"><a id="l03193" name="l03193"></a><span class="lineno"> 3193</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">j</a>=0; <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">j</a><<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">vlen</a>; ++<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">j</a>)</div>
|
|
<div class="line"><a id="l03194" name="l03194"></a><span class="lineno"> 3194</span> dst[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">it</a>.oofs(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">j</a>,<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">i1</a>)] = src[i][<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">j</a>];</div>
|
|
<div class="line"><a id="l03195" name="l03195"></a><span class="lineno"> 3195</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l03196" name="l03196"></a><span class="lineno"> 3196</span> </div>
|
|
<div class="foldopen" id="foldopen03197" data-start="{" data-end="}">
|
|
<div class="line"><a id="l03197" name="l03197"></a><span class="lineno"><a class="line" href="namespacepocketfft_1_1detail.html#ae7b44d2773d9d06a9787aff01d66b3ed"> 3197</a></span><span class="keyword">template</span> <<span class="keyword">typename</span> T, <span class="keywordtype">size_t</span> vlen> <span class="keywordtype">void</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#abac3fcc8ce83800d228774f64c28d4c3">copy_hartley</a>(<span class="keyword">const</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">multi_iter<vlen></a> &<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">it</a>,</div>
|
|
<div class="line"><a id="l03198" name="l03198"></a><span class="lineno"> 3198</span> <span class="keyword">const</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a> *<a class="code hl_define" href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a> src, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ndarr<T></a> &dst)</div>
|
|
<div class="line"><a id="l03199" name="l03199"></a><span class="lineno"> 3199</span> {</div>
|
|
<div class="line"><a id="l03200" name="l03200"></a><span class="lineno"> 3200</span> dst[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">it</a>.oofs(0)] = src[0];</div>
|
|
<div class="line"><a id="l03201" name="l03201"></a><span class="lineno"> 3201</span> <span class="keywordtype">size_t</span> i=1, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">i1</a>=1, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">i2</a>=<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">it</a>.length_out()-1;</div>
|
|
<div class="line"><a id="l03202" name="l03202"></a><span class="lineno"> 3202</span> <span class="keywordflow">for</span> (i=1; i<<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">it</a>.length_out()-1; i+=2, ++<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">i1</a>, --<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">i2</a>)</div>
|
|
<div class="line"><a id="l03203" name="l03203"></a><span class="lineno"> 3203</span> {</div>
|
|
<div class="line"><a id="l03204" name="l03204"></a><span class="lineno"> 3204</span> dst[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">it</a>.oofs(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">i1</a>)] = src[i]+src[i+1];</div>
|
|
<div class="line"><a id="l03205" name="l03205"></a><span class="lineno"> 3205</span> dst[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">it</a>.oofs(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">i2</a>)] = src[i]-src[i+1];</div>
|
|
<div class="line"><a id="l03206" name="l03206"></a><span class="lineno"> 3206</span> }</div>
|
|
<div class="line"><a id="l03207" name="l03207"></a><span class="lineno"> 3207</span> <span class="keywordflow">if</span> (i<<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">it</a>.length_out())</div>
|
|
<div class="line"><a id="l03208" name="l03208"></a><span class="lineno"> 3208</span> dst[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">it</a>.oofs(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">i1</a>)] = src[i];</div>
|
|
<div class="line"><a id="l03209" name="l03209"></a><span class="lineno"> 3209</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l03210" name="l03210"></a><span class="lineno"> 3210</span> </div>
|
|
<div class="foldopen" id="foldopen03211" data-start="{" data-end="};">
|
|
<div class="line"><a id="l03211" name="l03211"></a><span class="lineno"><a class="line" href="structpocketfft_1_1detail_1_1_exec_hartley.html"> 3211</a></span><span class="keyword">struct </span><a class="code hl_struct" href="structpocketfft_1_1detail_1_1_exec_hartley.html">ExecHartley</a></div>
|
|
<div class="line"><a id="l03212" name="l03212"></a><span class="lineno"> 3212</span> {</div>
|
|
<div class="foldopen" id="foldopen03213" data-start="{" data-end="}">
|
|
<div class="line"><a id="l03213" name="l03213"></a><span class="lineno"><a class="line" href="structpocketfft_1_1detail_1_1_exec_hartley.html#a67c98b38d12440781053552b9a33bba1"> 3213</a></span> <span class="keyword">template</span> <<span class="keyword">typename</span> T0, <span class="keyword">typename</span> T, <span class="keywordtype">size_t</span> vlen> <span class="keywordtype">void</span> <a class="code hl_function" href="structpocketfft_1_1detail_1_1_exec_hartley.html#a67c98b38d12440781053552b9a33bba1">operator () </a>(</div>
|
|
<div class="line"><a id="l03214" name="l03214"></a><span class="lineno"> 3214</span> <span class="keyword">const</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">multi_iter<vlen></a> &<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">it</a>, <span class="keyword">const</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cndarr<T0></a> &in, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ndarr<T0></a> &out,</div>
|
|
<div class="line"><a id="l03215" name="l03215"></a><span class="lineno"> 3215</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a> * <a class="code hl_variable" href="backend_2metal_2allocator_8h.html#a15aa5cc1baf29be08d55cca88509e697">buf</a>, <span class="keyword">const</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">pocketfft_r<T0></a> &plan, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T0</a> fct)<span class="keyword"> const</span></div>
|
|
<div class="line"><a id="l03216" name="l03216"></a><span class="lineno"> 3216</span><span class="keyword"> </span>{</div>
|
|
<div class="line"><a id="l03217" name="l03217"></a><span class="lineno"> 3217</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#aff05be3064743c1143b19318ab12ad4a">copy_input</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">it</a>, in, <a class="code hl_variable" href="backend_2metal_2allocator_8h.html#a15aa5cc1baf29be08d55cca88509e697">buf</a>);</div>
|
|
<div class="line"><a id="l03218" name="l03218"></a><span class="lineno"> 3218</span> plan.exec(<a class="code hl_variable" href="backend_2metal_2allocator_8h.html#a15aa5cc1baf29be08d55cca88509e697">buf</a>, fct, <span class="keyword">true</span>);</div>
|
|
<div class="line"><a id="l03219" name="l03219"></a><span class="lineno"> 3219</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#abac3fcc8ce83800d228774f64c28d4c3">copy_hartley</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">it</a>, <a class="code hl_variable" href="backend_2metal_2allocator_8h.html#a15aa5cc1baf29be08d55cca88509e697">buf</a>, out);</div>
|
|
<div class="line"><a id="l03220" name="l03220"></a><span class="lineno"> 3220</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l03221" name="l03221"></a><span class="lineno"> 3221</span> };</div>
|
|
</div>
|
|
<div class="line"><a id="l03222" name="l03222"></a><span class="lineno"> 3222</span> </div>
|
|
<div class="foldopen" id="foldopen03223" data-start="{" data-end="};">
|
|
<div class="line"><a id="l03223" name="l03223"></a><span class="lineno"><a class="line" href="structpocketfft_1_1detail_1_1_exec_dcst.html"> 3223</a></span><span class="keyword">struct </span><a class="code hl_struct" href="structpocketfft_1_1detail_1_1_exec_dcst.html">ExecDcst</a></div>
|
|
<div class="line"><a id="l03224" name="l03224"></a><span class="lineno"> 3224</span> {</div>
|
|
<div class="line"><a id="l03225" name="l03225"></a><span class="lineno"><a class="line" href="structpocketfft_1_1detail_1_1_exec_dcst.html#aea17551a49acaca5e7808dc181d38b7f"> 3225</a></span> <span class="keywordtype">bool</span> <a class="code hl_variable" href="structpocketfft_1_1detail_1_1_exec_dcst.html#aea17551a49acaca5e7808dc181d38b7f">ortho</a>;</div>
|
|
<div class="line"><a id="l03226" name="l03226"></a><span class="lineno"><a class="line" href="structpocketfft_1_1detail_1_1_exec_dcst.html#a9b170cbd74a9c6f45ac014ce349219ea"> 3226</a></span> <span class="keywordtype">int</span> <a class="code hl_variable" href="structpocketfft_1_1detail_1_1_exec_dcst.html#a9b170cbd74a9c6f45ac014ce349219ea">type</a>;</div>
|
|
<div class="line"><a id="l03227" name="l03227"></a><span class="lineno"><a class="line" href="structpocketfft_1_1detail_1_1_exec_dcst.html#a185023fc1e386cc8f233b79c49c1fd8a"> 3227</a></span> <span class="keywordtype">bool</span> <a class="code hl_variable" href="structpocketfft_1_1detail_1_1_exec_dcst.html#a185023fc1e386cc8f233b79c49c1fd8a">cosine</a>;</div>
|
|
<div class="line"><a id="l03228" name="l03228"></a><span class="lineno"> 3228</span> </div>
|
|
<div class="line"><a id="l03229" name="l03229"></a><span class="lineno"> 3229</span> <span class="keyword">template</span> <<span class="keyword">typename</span> T0, <span class="keyword">typename</span> T, <span class="keyword">typename</span> Tplan, <span class="keywordtype">size_t</span> vlen></div>
|
|
<div class="foldopen" id="foldopen03230" data-start="{" data-end="}">
|
|
<div class="line"><a id="l03230" name="l03230"></a><span class="lineno"><a class="line" href="structpocketfft_1_1detail_1_1_exec_dcst.html#a67f4f56e3574c491695f8cb8a1e983d8"> 3230</a></span> <span class="keywordtype">void</span> <a class="code hl_function" href="structpocketfft_1_1detail_1_1_exec_dcst.html#a67f4f56e3574c491695f8cb8a1e983d8">operator () </a>(<span class="keyword">const</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">multi_iter<vlen></a> &<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">it</a>, <span class="keyword">const</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cndarr<T0></a> &in,</div>
|
|
<div class="line"><a id="l03231" name="l03231"></a><span class="lineno"> 3231</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ndarr<T0></a> &out, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a> * <a class="code hl_variable" href="backend_2metal_2allocator_8h.html#a15aa5cc1baf29be08d55cca88509e697">buf</a>, <span class="keyword">const</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">Tplan</a> &plan, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T0</a> fct)<span class="keyword"> const</span></div>
|
|
<div class="line"><a id="l03232" name="l03232"></a><span class="lineno"> 3232</span><span class="keyword"> </span>{</div>
|
|
<div class="line"><a id="l03233" name="l03233"></a><span class="lineno"> 3233</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#aff05be3064743c1143b19318ab12ad4a">copy_input</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">it</a>, in, <a class="code hl_variable" href="backend_2metal_2allocator_8h.html#a15aa5cc1baf29be08d55cca88509e697">buf</a>);</div>
|
|
<div class="line"><a id="l03234" name="l03234"></a><span class="lineno"> 3234</span> plan.exec(<a class="code hl_variable" href="backend_2metal_2allocator_8h.html#a15aa5cc1baf29be08d55cca88509e697">buf</a>, fct, <a class="code hl_variable" href="structpocketfft_1_1detail_1_1_exec_dcst.html#aea17551a49acaca5e7808dc181d38b7f">ortho</a>, <a class="code hl_variable" href="structpocketfft_1_1detail_1_1_exec_dcst.html#a9b170cbd74a9c6f45ac014ce349219ea">type</a>, <a class="code hl_variable" href="structpocketfft_1_1detail_1_1_exec_dcst.html#a185023fc1e386cc8f233b79c49c1fd8a">cosine</a>);</div>
|
|
<div class="line"><a id="l03235" name="l03235"></a><span class="lineno"> 3235</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a1523a037300a8da05db210b802d9cb0e">copy_output</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">it</a>, <a class="code hl_variable" href="backend_2metal_2allocator_8h.html#a15aa5cc1baf29be08d55cca88509e697">buf</a>, out);</div>
|
|
<div class="line"><a id="l03236" name="l03236"></a><span class="lineno"> 3236</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l03237" name="l03237"></a><span class="lineno"> 3237</span> };</div>
|
|
</div>
|
|
<div class="line"><a id="l03238" name="l03238"></a><span class="lineno"> 3238</span> </div>
|
|
<div class="foldopen" id="foldopen03239" data-start="{" data-end="}">
|
|
<div class="line"><a id="l03239" name="l03239"></a><span class="lineno"><a class="line" href="namespacepocketfft_1_1detail.html#a055a39b0a337ca12217717196eb92fed"> 3239</a></span><span class="keyword">template</span><<span class="keyword">typename</span> T> <a class="code hl_define" href="pocketfft_8h.html#a7020984e0ca1d6e565629ca6e7c1a7e0">POCKETFFT_NOINLINE</a> <span class="keywordtype">void</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a055a39b0a337ca12217717196eb92fed">general_r2c</a>(</div>
|
|
<div class="line"><a id="l03240" name="l03240"></a><span class="lineno"> 3240</span> <span class="keyword">const</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cndarr<T></a> &in, <a class="code hl_class" href="classpocketfft_1_1detail_1_1ndarr.html">ndarr</a><<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cmplx<T></a>> &out, <span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axis</a>, <span class="keywordtype">bool</span> forward, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a> fct,</div>
|
|
<div class="line"><a id="l03241" name="l03241"></a><span class="lineno"> 3241</span> <span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nthreads</a>)</div>
|
|
<div class="line"><a id="l03242" name="l03242"></a><span class="lineno"> 3242</span> {</div>
|
|
<div class="line"><a id="l03243" name="l03243"></a><span class="lineno"> 3243</span> <span class="keyword">auto</span> plan = <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">get_plan<pocketfft_r<T></a>>(in.shape(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axis</a>));</div>
|
|
<div class="line"><a id="l03244" name="l03244"></a><span class="lineno"> 3244</span> <span class="keywordtype">size_t</span> len=in.shape(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axis</a>);</div>
|
|
<div class="line"><a id="l03245" name="l03245"></a><span class="lineno"> 3245</span> <a class="code hl_function" href="namespacepocketfft_1_1detail_1_1threading.html#a4fcf674db39f0e2c1c59d48491daed6e">threading::thread_map</a>(</div>
|
|
<div class="line"><a id="l03246" name="l03246"></a><span class="lineno"> 3246</span> <a class="code hl_function" href="structpocketfft_1_1detail_1_1util.html#a3b012d5a19215bcd32cf6e228556fa87">util::thread_count</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nthreads</a>, in.shape(), <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axis</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1_v_l_e_n.html">VLEN<T>::val</a>),</div>
|
|
<div class="line"><a id="l03247" name="l03247"></a><span class="lineno"> 3247</span> [&] {</div>
|
|
<div class="line"><a id="l03248" name="l03248"></a><span class="lineno"> 3248</span> constexpr auto vlen = VLEN<T>::val;</div>
|
|
<div class="line"><a id="l03249" name="l03249"></a><span class="lineno"> 3249</span> auto storage = alloc_tmp<T>(in.shape(), len, sizeof(T));</div>
|
|
<div class="line"><a id="l03250" name="l03250"></a><span class="lineno"> 3250</span> multi_iter<vlen> it(in, out, axis);</div>
|
|
<div class="line"><a id="l03251" name="l03251"></a><span class="lineno"> 3251</span>#ifndef POCKETFFT_NO_VECTORS</div>
|
|
<div class="line"><a id="l03252" name="l03252"></a><span class="lineno"> 3252</span> if (vlen>1)</div>
|
|
<div class="line"><a id="l03253" name="l03253"></a><span class="lineno"> 3253</span> while (it.remaining()>=vlen)</div>
|
|
<div class="line"><a id="l03254" name="l03254"></a><span class="lineno"> 3254</span> {</div>
|
|
<div class="line"><a id="l03255" name="l03255"></a><span class="lineno"> 3255</span> it.advance(vlen);</div>
|
|
<div class="line"><a id="l03256" name="l03256"></a><span class="lineno"> 3256</span> auto tdatav = reinterpret_cast<vtype_t<T> *>(storage.data());</div>
|
|
<div class="line"><a id="l03257" name="l03257"></a><span class="lineno"> 3257</span> copy_input(it, in, tdatav);</div>
|
|
<div class="line"><a id="l03258" name="l03258"></a><span class="lineno"> 3258</span> plan->exec(tdatav, fct, true);</div>
|
|
<div class="line"><a id="l03259" name="l03259"></a><span class="lineno"> 3259</span> for (size_t j=0; j<vlen; ++j)</div>
|
|
<div class="line"><a id="l03260" name="l03260"></a><span class="lineno"> 3260</span> out[it.oofs(j,0)].Set(tdatav[0][j]);</div>
|
|
<div class="line"><a id="l03261" name="l03261"></a><span class="lineno"> 3261</span> size_t i=1, ii=1;</div>
|
|
<div class="line"><a id="l03262" name="l03262"></a><span class="lineno"> 3262</span> if (forward)</div>
|
|
<div class="line"><a id="l03263" name="l03263"></a><span class="lineno"> 3263</span> for (; i<len-1; i+=2, ++ii)</div>
|
|
<div class="line"><a id="l03264" name="l03264"></a><span class="lineno"> 3264</span> for (size_t j=0; j<vlen; ++j)</div>
|
|
<div class="line"><a id="l03265" name="l03265"></a><span class="lineno"> 3265</span> out[it.oofs(j,ii)].Set(tdatav[i][j], tdatav[i+1][j]);</div>
|
|
<div class="line"><a id="l03266" name="l03266"></a><span class="lineno"> 3266</span> else</div>
|
|
<div class="line"><a id="l03267" name="l03267"></a><span class="lineno"> 3267</span> for (; i<len-1; i+=2, ++ii)</div>
|
|
<div class="line"><a id="l03268" name="l03268"></a><span class="lineno"> 3268</span> for (size_t j=0; j<vlen; ++j)</div>
|
|
<div class="line"><a id="l03269" name="l03269"></a><span class="lineno"> 3269</span> out[it.oofs(j,ii)].Set(tdatav[i][j], -tdatav[i+1][j]);</div>
|
|
<div class="line"><a id="l03270" name="l03270"></a><span class="lineno"> 3270</span> if (i<len)</div>
|
|
<div class="line"><a id="l03271" name="l03271"></a><span class="lineno"> 3271</span> for (size_t j=0; j<vlen; ++j)</div>
|
|
<div class="line"><a id="l03272" name="l03272"></a><span class="lineno"> 3272</span> out[it.oofs(j,ii)].Set(tdatav[i][j]);</div>
|
|
<div class="line"><a id="l03273" name="l03273"></a><span class="lineno"> 3273</span> }</div>
|
|
<div class="line"><a id="l03274" name="l03274"></a><span class="lineno"> 3274</span>#<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">endif</a></div>
|
|
<div class="line"><a id="l03275" name="l03275"></a><span class="lineno"> 3275</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">while</a> (<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">it</a>.remaining()>0)</div>
|
|
<div class="line"><a id="l03276" name="l03276"></a><span class="lineno"> 3276</span> {</div>
|
|
<div class="line"><a id="l03277" name="l03277"></a><span class="lineno"> 3277</span> it.advance(1);</div>
|
|
<div class="line"><a id="l03278" name="l03278"></a><span class="lineno"> 3278</span> auto tdata = reinterpret_cast<T *>(storage.data());</div>
|
|
<div class="line"><a id="l03279" name="l03279"></a><span class="lineno"> 3279</span> copy_input(it, in, tdata);</div>
|
|
<div class="line"><a id="l03280" name="l03280"></a><span class="lineno"> 3280</span> plan->exec(tdata, fct, true);</div>
|
|
<div class="line"><a id="l03281" name="l03281"></a><span class="lineno"> 3281</span> out[it.oofs(0)].Set(tdata[0]);</div>
|
|
<div class="line"><a id="l03282" name="l03282"></a><span class="lineno"> 3282</span> size_t i=1, ii=1;</div>
|
|
<div class="line"><a id="l03283" name="l03283"></a><span class="lineno"> 3283</span> if (forward)</div>
|
|
<div class="line"><a id="l03284" name="l03284"></a><span class="lineno"> 3284</span> for (; i<len-1; i+=2, ++ii)</div>
|
|
<div class="line"><a id="l03285" name="l03285"></a><span class="lineno"> 3285</span> out[it.oofs(ii)].Set(tdata[i], tdata[i+1]);</div>
|
|
<div class="line"><a id="l03286" name="l03286"></a><span class="lineno"> 3286</span> else</div>
|
|
<div class="line"><a id="l03287" name="l03287"></a><span class="lineno"> 3287</span> for (; i<len-1; i+=2, ++ii)</div>
|
|
<div class="line"><a id="l03288" name="l03288"></a><span class="lineno"> 3288</span> out[it.oofs(ii)].Set(tdata[i], -tdata[i+1]);</div>
|
|
<div class="line"><a id="l03289" name="l03289"></a><span class="lineno"> 3289</span> if (i<len)</div>
|
|
<div class="line"><a id="l03290" name="l03290"></a><span class="lineno"> 3290</span> out[it.oofs(ii)].Set(tdata[i]);</div>
|
|
<div class="line"><a id="l03291" name="l03291"></a><span class="lineno"> 3291</span> }</div>
|
|
<div class="line"><a id="l03292" name="l03292"></a><span class="lineno"> 3292</span> }); <span class="comment">// end of parallel region</span></div>
|
|
<div class="line"><a id="l03293" name="l03293"></a><span class="lineno"> 3293</span> }</div>
|
|
</div>
|
|
<div class="foldopen" id="foldopen03294" data-start="{" data-end="}">
|
|
<div class="line"><a id="l03294" name="l03294"></a><span class="lineno"><a class="line" href="namespacepocketfft_1_1detail.html#ac8ee38e8d8bcda875c99eeaf567550fc"> 3294</a></span><span class="keyword">template</span><<span class="keyword">typename</span> T> <a class="code hl_define" href="pocketfft_8h.html#a7020984e0ca1d6e565629ca6e7c1a7e0">POCKETFFT_NOINLINE</a> <span class="keywordtype">void</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#ac8ee38e8d8bcda875c99eeaf567550fc">general_c2r</a>(</div>
|
|
<div class="line"><a id="l03295" name="l03295"></a><span class="lineno"> 3295</span> <span class="keyword">const</span> <a class="code hl_class" href="classpocketfft_1_1detail_1_1cndarr.html">cndarr</a><<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cmplx<T></a>> &in, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ndarr<T></a> &out, <span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axis</a>, <span class="keywordtype">bool</span> forward, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a> fct,</div>
|
|
<div class="line"><a id="l03296" name="l03296"></a><span class="lineno"> 3296</span> <span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nthreads</a>)</div>
|
|
<div class="line"><a id="l03297" name="l03297"></a><span class="lineno"> 3297</span> {</div>
|
|
<div class="line"><a id="l03298" name="l03298"></a><span class="lineno"> 3298</span> <span class="keyword">auto</span> plan = <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">get_plan<pocketfft_r<T></a>>(out.shape(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axis</a>));</div>
|
|
<div class="line"><a id="l03299" name="l03299"></a><span class="lineno"> 3299</span> <span class="keywordtype">size_t</span> len=out.shape(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axis</a>);</div>
|
|
<div class="line"><a id="l03300" name="l03300"></a><span class="lineno"> 3300</span> threading::thread_map(</div>
|
|
<div class="line"><a id="l03301" name="l03301"></a><span class="lineno"> 3301</span> util::thread_count(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nthreads</a>, in.shape(), <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axis</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1_v_l_e_n.html">VLEN<T>::val</a>),</div>
|
|
<div class="line"><a id="l03302" name="l03302"></a><span class="lineno"> 3302</span> [&] {</div>
|
|
<div class="line"><a id="l03303" name="l03303"></a><span class="lineno"> 3303</span> constexpr auto vlen = VLEN<T>::val;</div>
|
|
<div class="line"><a id="l03304" name="l03304"></a><span class="lineno"> 3304</span> auto storage = alloc_tmp<T>(out.shape(), len, sizeof(T));</div>
|
|
<div class="line"><a id="l03305" name="l03305"></a><span class="lineno"> 3305</span> multi_iter<vlen> it(in, out, axis);</div>
|
|
<div class="line"><a id="l03306" name="l03306"></a><span class="lineno"> 3306</span>#ifndef POCKETFFT_NO_VECTORS</div>
|
|
<div class="line"><a id="l03307" name="l03307"></a><span class="lineno"> 3307</span> if (vlen>1)</div>
|
|
<div class="line"><a id="l03308" name="l03308"></a><span class="lineno"> 3308</span> while (it.remaining()>=vlen)</div>
|
|
<div class="line"><a id="l03309" name="l03309"></a><span class="lineno"> 3309</span> {</div>
|
|
<div class="line"><a id="l03310" name="l03310"></a><span class="lineno"> 3310</span> it.advance(vlen);</div>
|
|
<div class="line"><a id="l03311" name="l03311"></a><span class="lineno"> 3311</span> auto tdatav = reinterpret_cast<vtype_t<T> *>(storage.data());</div>
|
|
<div class="line"><a id="l03312" name="l03312"></a><span class="lineno"> 3312</span> for (size_t j=0; j<vlen; ++j)</div>
|
|
<div class="line"><a id="l03313" name="l03313"></a><span class="lineno"> 3313</span> tdatav[0][j]=in[it.iofs(j,0)].r;</div>
|
|
<div class="line"><a id="l03314" name="l03314"></a><span class="lineno"> 3314</span> {</div>
|
|
<div class="line"><a id="l03315" name="l03315"></a><span class="lineno"> 3315</span> size_t i=1, ii=1;</div>
|
|
<div class="line"><a id="l03316" name="l03316"></a><span class="lineno"> 3316</span> if (forward)</div>
|
|
<div class="line"><a id="l03317" name="l03317"></a><span class="lineno"> 3317</span> for (; i<len-1; i+=2, ++ii)</div>
|
|
<div class="line"><a id="l03318" name="l03318"></a><span class="lineno"> 3318</span> for (size_t j=0; j<vlen; ++j)</div>
|
|
<div class="line"><a id="l03319" name="l03319"></a><span class="lineno"> 3319</span> {</div>
|
|
<div class="line"><a id="l03320" name="l03320"></a><span class="lineno"> 3320</span> tdatav[i ][j] = in[it.iofs(j,ii)].r;</div>
|
|
<div class="line"><a id="l03321" name="l03321"></a><span class="lineno"> 3321</span> tdatav[i+1][j] = -in[it.iofs(j,ii)].i;</div>
|
|
<div class="line"><a id="l03322" name="l03322"></a><span class="lineno"> 3322</span> }</div>
|
|
<div class="line"><a id="l03323" name="l03323"></a><span class="lineno"> 3323</span> else</div>
|
|
<div class="line"><a id="l03324" name="l03324"></a><span class="lineno"> 3324</span> for (; i<len-1; i+=2, ++ii)</div>
|
|
<div class="line"><a id="l03325" name="l03325"></a><span class="lineno"> 3325</span> for (size_t j=0; j<vlen; ++j)</div>
|
|
<div class="line"><a id="l03326" name="l03326"></a><span class="lineno"> 3326</span> {</div>
|
|
<div class="line"><a id="l03327" name="l03327"></a><span class="lineno"> 3327</span> tdatav[i ][j] = in[it.iofs(j,ii)].r;</div>
|
|
<div class="line"><a id="l03328" name="l03328"></a><span class="lineno"> 3328</span> tdatav[i+1][j] = in[it.iofs(j,ii)].i;</div>
|
|
<div class="line"><a id="l03329" name="l03329"></a><span class="lineno"> 3329</span> }</div>
|
|
<div class="line"><a id="l03330" name="l03330"></a><span class="lineno"> 3330</span> if (i<len)</div>
|
|
<div class="line"><a id="l03331" name="l03331"></a><span class="lineno"> 3331</span> for (size_t j=0; j<vlen; ++j)</div>
|
|
<div class="line"><a id="l03332" name="l03332"></a><span class="lineno"> 3332</span> tdatav[i][j] = in[it.iofs(j,ii)].r;</div>
|
|
<div class="line"><a id="l03333" name="l03333"></a><span class="lineno"> 3333</span> }</div>
|
|
<div class="line"><a id="l03334" name="l03334"></a><span class="lineno"> 3334</span> plan->exec(tdatav, fct, false);</div>
|
|
<div class="line"><a id="l03335" name="l03335"></a><span class="lineno"> 3335</span> copy_output(it, tdatav, out);</div>
|
|
<div class="line"><a id="l03336" name="l03336"></a><span class="lineno"> 3336</span> }</div>
|
|
<div class="line"><a id="l03337" name="l03337"></a><span class="lineno"> 3337</span>#<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">endif</a></div>
|
|
<div class="line"><a id="l03338" name="l03338"></a><span class="lineno"> 3338</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">while</a> (<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">it</a>.remaining()>0)</div>
|
|
<div class="line"><a id="l03339" name="l03339"></a><span class="lineno"> 3339</span> {</div>
|
|
<div class="line"><a id="l03340" name="l03340"></a><span class="lineno"> 3340</span> it.advance(1);</div>
|
|
<div class="line"><a id="l03341" name="l03341"></a><span class="lineno"> 3341</span> auto tdata = reinterpret_cast<T *>(storage.data());</div>
|
|
<div class="line"><a id="l03342" name="l03342"></a><span class="lineno"> 3342</span> tdata[0]=in[it.iofs(0)].r;</div>
|
|
<div class="line"><a id="l03343" name="l03343"></a><span class="lineno"> 3343</span> {</div>
|
|
<div class="line"><a id="l03344" name="l03344"></a><span class="lineno"> 3344</span> size_t i=1, ii=1;</div>
|
|
<div class="line"><a id="l03345" name="l03345"></a><span class="lineno"> 3345</span> if (forward)</div>
|
|
<div class="line"><a id="l03346" name="l03346"></a><span class="lineno"> 3346</span> for (; i<len-1; i+=2, ++ii)</div>
|
|
<div class="line"><a id="l03347" name="l03347"></a><span class="lineno"> 3347</span> {</div>
|
|
<div class="line"><a id="l03348" name="l03348"></a><span class="lineno"> 3348</span> tdata[i ] = in[it.iofs(ii)].r;</div>
|
|
<div class="line"><a id="l03349" name="l03349"></a><span class="lineno"> 3349</span> tdata[i+1] = -in[it.iofs(ii)].i;</div>
|
|
<div class="line"><a id="l03350" name="l03350"></a><span class="lineno"> 3350</span> }</div>
|
|
<div class="line"><a id="l03351" name="l03351"></a><span class="lineno"> 3351</span> else</div>
|
|
<div class="line"><a id="l03352" name="l03352"></a><span class="lineno"> 3352</span> for (; i<len-1; i+=2, ++ii)</div>
|
|
<div class="line"><a id="l03353" name="l03353"></a><span class="lineno"> 3353</span> {</div>
|
|
<div class="line"><a id="l03354" name="l03354"></a><span class="lineno"> 3354</span> tdata[i ] = in[it.iofs(ii)].r;</div>
|
|
<div class="line"><a id="l03355" name="l03355"></a><span class="lineno"> 3355</span> tdata[i+1] = in[it.iofs(ii)].i;</div>
|
|
<div class="line"><a id="l03356" name="l03356"></a><span class="lineno"> 3356</span> }</div>
|
|
<div class="line"><a id="l03357" name="l03357"></a><span class="lineno"> 3357</span> if (i<len)</div>
|
|
<div class="line"><a id="l03358" name="l03358"></a><span class="lineno"> 3358</span> tdata[i] = in[it.iofs(ii)].r;</div>
|
|
<div class="line"><a id="l03359" name="l03359"></a><span class="lineno"> 3359</span> }</div>
|
|
<div class="line"><a id="l03360" name="l03360"></a><span class="lineno"> 3360</span> plan->exec(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tdata</a>, fct, <span class="keyword">false</span>);</div>
|
|
<div class="line"><a id="l03361" name="l03361"></a><span class="lineno"> 3361</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a1523a037300a8da05db210b802d9cb0e">copy_output</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">it</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tdata</a>, out);</div>
|
|
<div class="line"><a id="l03362" name="l03362"></a><span class="lineno"> 3362</span> }</div>
|
|
<div class="line"><a id="l03363" name="l03363"></a><span class="lineno"> 3363</span> }); <span class="comment">// end of parallel region</span></div>
|
|
<div class="line"><a id="l03364" name="l03364"></a><span class="lineno"> 3364</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l03365" name="l03365"></a><span class="lineno"> 3365</span> </div>
|
|
<div class="foldopen" id="foldopen03366" data-start="{" data-end="};">
|
|
<div class="line"><a id="l03366" name="l03366"></a><span class="lineno"><a class="line" href="structpocketfft_1_1detail_1_1_exec_r2_r.html"> 3366</a></span><span class="keyword">struct </span><a class="code hl_struct" href="structpocketfft_1_1detail_1_1_exec_r2_r.html">ExecR2R</a></div>
|
|
<div class="line"><a id="l03367" name="l03367"></a><span class="lineno"> 3367</span> {</div>
|
|
<div class="line"><a id="l03368" name="l03368"></a><span class="lineno"><a class="line" href="structpocketfft_1_1detail_1_1_exec_r2_r.html#a5ec66ebb2ccd079f62b068ddd1fc7bdf"> 3368</a></span> <span class="keywordtype">bool</span> r2h, <a class="code hl_variable" href="structpocketfft_1_1detail_1_1_exec_r2_r.html#a5ec66ebb2ccd079f62b068ddd1fc7bdf">forward</a>;</div>
|
|
<div class="line"><a id="l03369" name="l03369"></a><span class="lineno"> 3369</span> </div>
|
|
<div class="foldopen" id="foldopen03370" data-start="{" data-end="}">
|
|
<div class="line"><a id="l03370" name="l03370"></a><span class="lineno"><a class="line" href="structpocketfft_1_1detail_1_1_exec_r2_r.html#acdba1650962714e6afff51e9ca456970"> 3370</a></span> <span class="keyword">template</span> <<span class="keyword">typename</span> T0, <span class="keyword">typename</span> T, <span class="keywordtype">size_t</span> vlen> <span class="keywordtype">void</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">operator () </a>(</div>
|
|
<div class="line"><a id="l03371" name="l03371"></a><span class="lineno"> 3371</span> <span class="keyword">const</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">multi_iter<vlen></a> &<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">it</a>, <span class="keyword">const</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cndarr<T0></a> &in, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ndarr<T0></a> &out, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a> * <a class="code hl_variable" href="backend_2metal_2allocator_8h.html#a15aa5cc1baf29be08d55cca88509e697">buf</a>,</div>
|
|
<div class="line"><a id="l03372" name="l03372"></a><span class="lineno"> 3372</span> <span class="keyword">const</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">pocketfft_r<T0></a> &plan, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T0</a> fct)<span class="keyword"> const</span></div>
|
|
<div class="line"><a id="l03373" name="l03373"></a><span class="lineno"> 3373</span><span class="keyword"> </span>{</div>
|
|
<div class="line"><a id="l03374" name="l03374"></a><span class="lineno"> 3374</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#aff05be3064743c1143b19318ab12ad4a">copy_input</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">it</a>, in, <a class="code hl_variable" href="backend_2metal_2allocator_8h.html#a15aa5cc1baf29be08d55cca88509e697">buf</a>);</div>
|
|
<div class="line"><a id="l03375" name="l03375"></a><span class="lineno"> 3375</span> <span class="keywordflow">if</span> ((!r2h) && forward)</div>
|
|
<div class="line"><a id="l03376" name="l03376"></a><span class="lineno"> 3376</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i=2; i<<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">it</a>.length_out(); i+=2)</div>
|
|
<div class="line"><a id="l03377" name="l03377"></a><span class="lineno"> 3377</span> <a class="code hl_variable" href="backend_2metal_2allocator_8h.html#a15aa5cc1baf29be08d55cca88509e697">buf</a>[i] = -<a class="code hl_variable" href="backend_2metal_2allocator_8h.html#a15aa5cc1baf29be08d55cca88509e697">buf</a>[i];</div>
|
|
<div class="line"><a id="l03378" name="l03378"></a><span class="lineno"> 3378</span> plan.exec(<a class="code hl_variable" href="backend_2metal_2allocator_8h.html#a15aa5cc1baf29be08d55cca88509e697">buf</a>, fct, r2h);</div>
|
|
<div class="line"><a id="l03379" name="l03379"></a><span class="lineno"> 3379</span> <span class="keywordflow">if</span> (r2h && (!forward))</div>
|
|
<div class="line"><a id="l03380" name="l03380"></a><span class="lineno"> 3380</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i=2; i<<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">it</a>.length_out(); i+=2)</div>
|
|
<div class="line"><a id="l03381" name="l03381"></a><span class="lineno"> 3381</span> <a class="code hl_variable" href="backend_2metal_2allocator_8h.html#a15aa5cc1baf29be08d55cca88509e697">buf</a>[i] = -<a class="code hl_variable" href="backend_2metal_2allocator_8h.html#a15aa5cc1baf29be08d55cca88509e697">buf</a>[i];</div>
|
|
<div class="line"><a id="l03382" name="l03382"></a><span class="lineno"> 3382</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a1523a037300a8da05db210b802d9cb0e">copy_output</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">it</a>, <a class="code hl_variable" href="backend_2metal_2allocator_8h.html#a15aa5cc1baf29be08d55cca88509e697">buf</a>, out);</div>
|
|
<div class="line"><a id="l03383" name="l03383"></a><span class="lineno"> 3383</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l03384" name="l03384"></a><span class="lineno"> 3384</span> };</div>
|
|
</div>
|
|
<div class="line"><a id="l03385" name="l03385"></a><span class="lineno"> 3385</span> </div>
|
|
<div class="foldopen" id="foldopen03386" data-start="{" data-end="}">
|
|
<div class="line"><a id="l03386" name="l03386"></a><span class="lineno"><a class="line" href="namespacepocketfft_1_1detail.html#ab585ac594ae1253d4659e7b9e1623c8a"> 3386</a></span><span class="keyword">template</span><<span class="keyword">typename</span> T> <span class="keywordtype">void</span> c2c(<span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#a885ee37fcf564a268a5c8ca9ea8603e1">shape_t</a> &shape, <span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#afb987c919e9424a996d0fc8b3c23cc84">stride_t</a> &stride_in,</div>
|
|
<div class="line"><a id="l03387" name="l03387"></a><span class="lineno"> 3387</span> <span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#afb987c919e9424a996d0fc8b3c23cc84">stride_t</a> &stride_out, <span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#a885ee37fcf564a268a5c8ca9ea8603e1">shape_t</a> &<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axes</a>, <span class="keywordtype">bool</span> forward,</div>
|
|
<div class="line"><a id="l03388" name="l03388"></a><span class="lineno"> 3388</span> <span class="keyword">const</span> std::complex<T> *<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">data_in</a>, std::complex<T> *<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">data_out</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a> fct,</div>
|
|
<div class="line"><a id="l03389" name="l03389"></a><span class="lineno"> 3389</span> <span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nthreads</a>=1)</div>
|
|
<div class="line"><a id="l03390" name="l03390"></a><span class="lineno"> 3390</span> {</div>
|
|
<div class="line"><a id="l03391" name="l03391"></a><span class="lineno"> 3391</span> <span class="keywordflow">if</span> (util::prod(shape)==0) <span class="keywordflow">return</span>;</div>
|
|
<div class="line"><a id="l03392" name="l03392"></a><span class="lineno"> 3392</span> util::sanity_check(shape, stride_in, stride_out, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">data_in</a>==<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">data_out</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axes</a>);</div>
|
|
<div class="line"><a id="l03393" name="l03393"></a><span class="lineno"> 3393</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cndarr<cmplx<T></a>> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ain</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">data_in</a>, shape, stride_in);</div>
|
|
<div class="line"><a id="l03394" name="l03394"></a><span class="lineno"> 3394</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ndarr<cmplx<T></a>> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">aout</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">data_out</a>, shape, stride_out);</div>
|
|
<div class="line"><a id="l03395" name="l03395"></a><span class="lineno"> 3395</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">general_nd<pocketfft_c<T></a>>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ain</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">aout</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axes</a>, fct, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nthreads</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1_exec_c2_c.html">ExecC2C</a>{forward});</div>
|
|
<div class="line"><a id="l03396" name="l03396"></a><span class="lineno"> 3396</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l03397" name="l03397"></a><span class="lineno"> 3397</span> </div>
|
|
<div class="foldopen" id="foldopen03398" data-start="{" data-end="}">
|
|
<div class="line"><a id="l03398" name="l03398"></a><span class="lineno"><a class="line" href="namespacepocketfft_1_1detail.html#a60615f5b685314c658346c309d5ef2ba"> 3398</a></span><span class="keyword">template</span><<span class="keyword">typename</span> T> <span class="keywordtype">void</span> dct(<span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#a885ee37fcf564a268a5c8ca9ea8603e1">shape_t</a> &shape,</div>
|
|
<div class="line"><a id="l03399" name="l03399"></a><span class="lineno"> 3399</span> <span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#afb987c919e9424a996d0fc8b3c23cc84">stride_t</a> &stride_in, <span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#afb987c919e9424a996d0fc8b3c23cc84">stride_t</a> &stride_out, <span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#a885ee37fcf564a268a5c8ca9ea8603e1">shape_t</a> &<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axes</a>,</div>
|
|
<div class="line"><a id="l03400" name="l03400"></a><span class="lineno"> 3400</span> <span class="keywordtype">int</span> type, <span class="keyword">const</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a> *<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">data_in</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a> *<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">data_out</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a> fct, <span class="keywordtype">bool</span> ortho, <span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nthreads</a>=1)</div>
|
|
<div class="line"><a id="l03401" name="l03401"></a><span class="lineno"> 3401</span> {</div>
|
|
<div class="line"><a id="l03402" name="l03402"></a><span class="lineno"> 3402</span> <span class="keywordflow">if</span> ((type<1) || (type>4)) <span class="keywordflow">throw</span> std::invalid_argument(<span class="stringliteral">"invalid DCT type"</span>);</div>
|
|
<div class="line"><a id="l03403" name="l03403"></a><span class="lineno"> 3403</span> <span class="keywordflow">if</span> (util::prod(shape)==0) <span class="keywordflow">return</span>;</div>
|
|
<div class="line"><a id="l03404" name="l03404"></a><span class="lineno"> 3404</span> util::sanity_check(shape, stride_in, stride_out, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">data_in</a>==<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">data_out</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axes</a>);</div>
|
|
<div class="line"><a id="l03405" name="l03405"></a><span class="lineno"> 3405</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cndarr<T></a> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ain</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">data_in</a>, shape, stride_in);</div>
|
|
<div class="line"><a id="l03406" name="l03406"></a><span class="lineno"> 3406</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ndarr<T></a> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">aout</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">data_out</a>, shape, stride_out);</div>
|
|
<div class="line"><a id="l03407" name="l03407"></a><span class="lineno"> 3407</span> <span class="keyword">const</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1_exec_dcst.html">ExecDcst</a> exec{ortho, type, <span class="keyword">true</span>};</div>
|
|
<div class="line"><a id="l03408" name="l03408"></a><span class="lineno"> 3408</span> <span class="keywordflow">if</span> (type==1)</div>
|
|
<div class="line"><a id="l03409" name="l03409"></a><span class="lineno"> 3409</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">general_nd<T_dct1<T></a>>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ain</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">aout</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axes</a>, fct, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nthreads</a>, exec);</div>
|
|
<div class="line"><a id="l03410" name="l03410"></a><span class="lineno"> 3410</span> <span class="keywordflow">else</span> <span class="keywordflow">if</span> (type==4)</div>
|
|
<div class="line"><a id="l03411" name="l03411"></a><span class="lineno"> 3411</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">general_nd<T_dcst4<T></a>>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ain</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">aout</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axes</a>, fct, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nthreads</a>, exec);</div>
|
|
<div class="line"><a id="l03412" name="l03412"></a><span class="lineno"> 3412</span> <span class="keywordflow">else</span></div>
|
|
<div class="line"><a id="l03413" name="l03413"></a><span class="lineno"> 3413</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">general_nd<T_dcst23<T></a>>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ain</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">aout</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axes</a>, fct, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nthreads</a>, exec);</div>
|
|
<div class="line"><a id="l03414" name="l03414"></a><span class="lineno"> 3414</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l03415" name="l03415"></a><span class="lineno"> 3415</span> </div>
|
|
<div class="foldopen" id="foldopen03416" data-start="{" data-end="}">
|
|
<div class="line"><a id="l03416" name="l03416"></a><span class="lineno"><a class="line" href="namespacepocketfft_1_1detail.html#add0f231fc8a1ce01b90a90faeebcb4eb"> 3416</a></span><span class="keyword">template</span><<span class="keyword">typename</span> T> <span class="keywordtype">void</span> dst(<span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#a885ee37fcf564a268a5c8ca9ea8603e1">shape_t</a> &shape,</div>
|
|
<div class="line"><a id="l03417" name="l03417"></a><span class="lineno"> 3417</span> <span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#afb987c919e9424a996d0fc8b3c23cc84">stride_t</a> &stride_in, <span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#afb987c919e9424a996d0fc8b3c23cc84">stride_t</a> &stride_out, <span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#a885ee37fcf564a268a5c8ca9ea8603e1">shape_t</a> &<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axes</a>,</div>
|
|
<div class="line"><a id="l03418" name="l03418"></a><span class="lineno"> 3418</span> <span class="keywordtype">int</span> type, <span class="keyword">const</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a> *<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">data_in</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a> *<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">data_out</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a> fct, <span class="keywordtype">bool</span> ortho, <span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nthreads</a>=1)</div>
|
|
<div class="line"><a id="l03419" name="l03419"></a><span class="lineno"> 3419</span> {</div>
|
|
<div class="line"><a id="l03420" name="l03420"></a><span class="lineno"> 3420</span> <span class="keywordflow">if</span> ((type<1) || (type>4)) <span class="keywordflow">throw</span> std::invalid_argument(<span class="stringliteral">"invalid DST type"</span>);</div>
|
|
<div class="line"><a id="l03421" name="l03421"></a><span class="lineno"> 3421</span> <span class="keywordflow">if</span> (util::prod(shape)==0) <span class="keywordflow">return</span>;</div>
|
|
<div class="line"><a id="l03422" name="l03422"></a><span class="lineno"> 3422</span> util::sanity_check(shape, stride_in, stride_out, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">data_in</a>==<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">data_out</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axes</a>);</div>
|
|
<div class="line"><a id="l03423" name="l03423"></a><span class="lineno"> 3423</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cndarr<T></a> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ain</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">data_in</a>, shape, stride_in);</div>
|
|
<div class="line"><a id="l03424" name="l03424"></a><span class="lineno"> 3424</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ndarr<T></a> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">aout</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">data_out</a>, shape, stride_out);</div>
|
|
<div class="line"><a id="l03425" name="l03425"></a><span class="lineno"> 3425</span> <span class="keyword">const</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1_exec_dcst.html">ExecDcst</a> exec{ortho, type, <span class="keyword">false</span>};</div>
|
|
<div class="line"><a id="l03426" name="l03426"></a><span class="lineno"> 3426</span> <span class="keywordflow">if</span> (type==1)</div>
|
|
<div class="line"><a id="l03427" name="l03427"></a><span class="lineno"> 3427</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">general_nd<T_dst1<T></a>>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ain</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">aout</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axes</a>, fct, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nthreads</a>, exec);</div>
|
|
<div class="line"><a id="l03428" name="l03428"></a><span class="lineno"> 3428</span> <span class="keywordflow">else</span> <span class="keywordflow">if</span> (type==4)</div>
|
|
<div class="line"><a id="l03429" name="l03429"></a><span class="lineno"> 3429</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">general_nd<T_dcst4<T></a>>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ain</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">aout</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axes</a>, fct, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nthreads</a>, exec);</div>
|
|
<div class="line"><a id="l03430" name="l03430"></a><span class="lineno"> 3430</span> <span class="keywordflow">else</span></div>
|
|
<div class="line"><a id="l03431" name="l03431"></a><span class="lineno"> 3431</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">general_nd<T_dcst23<T></a>>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ain</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">aout</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axes</a>, fct, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nthreads</a>, exec);</div>
|
|
<div class="line"><a id="l03432" name="l03432"></a><span class="lineno"> 3432</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l03433" name="l03433"></a><span class="lineno"> 3433</span> </div>
|
|
<div class="foldopen" id="foldopen03434" data-start="{" data-end="}">
|
|
<div class="line"><a id="l03434" name="l03434"></a><span class="lineno"><a class="line" href="namespacepocketfft_1_1detail.html#a4e46762466d399e35b79c324cfe21616"> 3434</a></span><span class="keyword">template</span><<span class="keyword">typename</span> T> <span class="keywordtype">void</span> r2c(<span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#a885ee37fcf564a268a5c8ca9ea8603e1">shape_t</a> &<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">shape_in</a>,</div>
|
|
<div class="line"><a id="l03435" name="l03435"></a><span class="lineno"> 3435</span> <span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#afb987c919e9424a996d0fc8b3c23cc84">stride_t</a> &stride_in, <span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#afb987c919e9424a996d0fc8b3c23cc84">stride_t</a> &stride_out, <span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axis</a>,</div>
|
|
<div class="line"><a id="l03436" name="l03436"></a><span class="lineno"> 3436</span> <span class="keywordtype">bool</span> forward, <span class="keyword">const</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a> *<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">data_in</a>, std::complex<T> *<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">data_out</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a> fct,</div>
|
|
<div class="line"><a id="l03437" name="l03437"></a><span class="lineno"> 3437</span> <span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nthreads</a>=1)</div>
|
|
<div class="line"><a id="l03438" name="l03438"></a><span class="lineno"> 3438</span> {</div>
|
|
<div class="line"><a id="l03439" name="l03439"></a><span class="lineno"> 3439</span> <span class="keywordflow">if</span> (util::prod(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">shape_in</a>)==0) <span class="keywordflow">return</span>;</div>
|
|
<div class="line"><a id="l03440" name="l03440"></a><span class="lineno"> 3440</span> util::sanity_check(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">shape_in</a>, stride_in, stride_out, <span class="keyword">false</span>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axis</a>);</div>
|
|
<div class="line"><a id="l03441" name="l03441"></a><span class="lineno"> 3441</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cndarr<T></a> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ain</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">data_in</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">shape_in</a>, stride_in);</div>
|
|
<div class="line"><a id="l03442" name="l03442"></a><span class="lineno"> 3442</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#a885ee37fcf564a268a5c8ca9ea8603e1">shape_t</a> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">shape_out</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">shape_in</a>);</div>
|
|
<div class="line"><a id="l03443" name="l03443"></a><span class="lineno"> 3443</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">shape_out</a>[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axis</a>] = <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">shape_in</a>[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axis</a>]/2 + 1;</div>
|
|
<div class="line"><a id="l03444" name="l03444"></a><span class="lineno"> 3444</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ndarr<cmplx<T></a>> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">aout</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">data_out</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">shape_out</a>, stride_out);</div>
|
|
<div class="line"><a id="l03445" name="l03445"></a><span class="lineno"> 3445</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#a055a39b0a337ca12217717196eb92fed">general_r2c</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ain</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">aout</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axis</a>, forward, fct, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nthreads</a>);</div>
|
|
<div class="line"><a id="l03446" name="l03446"></a><span class="lineno"> 3446</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l03447" name="l03447"></a><span class="lineno"> 3447</span> </div>
|
|
<div class="foldopen" id="foldopen03448" data-start="{" data-end="}">
|
|
<div class="line"><a id="l03448" name="l03448"></a><span class="lineno"><a class="line" href="namespacepocketfft_1_1detail.html#a454179497c44714d4b7425f116468c17"> 3448</a></span><span class="keyword">template</span><<span class="keyword">typename</span> T> <span class="keywordtype">void</span> r2c(<span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#a885ee37fcf564a268a5c8ca9ea8603e1">shape_t</a> &<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">shape_in</a>,</div>
|
|
<div class="line"><a id="l03449" name="l03449"></a><span class="lineno"> 3449</span> <span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#afb987c919e9424a996d0fc8b3c23cc84">stride_t</a> &stride_in, <span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#afb987c919e9424a996d0fc8b3c23cc84">stride_t</a> &stride_out, <span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#a885ee37fcf564a268a5c8ca9ea8603e1">shape_t</a> &<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axes</a>,</div>
|
|
<div class="line"><a id="l03450" name="l03450"></a><span class="lineno"> 3450</span> <span class="keywordtype">bool</span> forward, <span class="keyword">const</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a> *<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">data_in</a>, std::complex<T> *<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">data_out</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a> fct,</div>
|
|
<div class="line"><a id="l03451" name="l03451"></a><span class="lineno"> 3451</span> <span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nthreads</a>=1)</div>
|
|
<div class="line"><a id="l03452" name="l03452"></a><span class="lineno"> 3452</span> {</div>
|
|
<div class="line"><a id="l03453" name="l03453"></a><span class="lineno"> 3453</span> <span class="keywordflow">if</span> (util::prod(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">shape_in</a>)==0) <span class="keywordflow">return</span>;</div>
|
|
<div class="line"><a id="l03454" name="l03454"></a><span class="lineno"> 3454</span> util::sanity_check(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">shape_in</a>, stride_in, stride_out, <span class="keyword">false</span>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axes</a>);</div>
|
|
<div class="line"><a id="l03455" name="l03455"></a><span class="lineno"> 3455</span> r2c(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">shape_in</a>, stride_in, stride_out, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axes</a>.back(), forward, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">data_in</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">data_out</a>,</div>
|
|
<div class="line"><a id="l03456" name="l03456"></a><span class="lineno"> 3456</span> fct, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nthreads</a>);</div>
|
|
<div class="line"><a id="l03457" name="l03457"></a><span class="lineno"> 3457</span> <span class="keywordflow">if</span> (<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axes</a>.size()==1) <span class="keywordflow">return</span>;</div>
|
|
<div class="line"><a id="l03458" name="l03458"></a><span class="lineno"> 3458</span> </div>
|
|
<div class="line"><a id="l03459" name="l03459"></a><span class="lineno"> 3459</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#a885ee37fcf564a268a5c8ca9ea8603e1">shape_t</a> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">shape_out</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">shape_in</a>);</div>
|
|
<div class="line"><a id="l03460" name="l03460"></a><span class="lineno"> 3460</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">shape_out</a>[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axes</a>.back()] = <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">shape_in</a>[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axes</a>.back()]/2 + 1;</div>
|
|
<div class="line"><a id="l03461" name="l03461"></a><span class="lineno"> 3461</span> <span class="keyword">auto</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">newaxes</a> = <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#a885ee37fcf564a268a5c8ca9ea8603e1">shape_t</a>{<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axes</a>.begin(), --<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axes</a>.end()};</div>
|
|
<div class="line"><a id="l03462" name="l03462"></a><span class="lineno"> 3462</span> c2c(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">shape_out</a>, stride_out, stride_out, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">newaxes</a>, forward, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">data_out</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">data_out</a>,</div>
|
|
<div class="line"><a id="l03463" name="l03463"></a><span class="lineno"> 3463</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a>(1), <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nthreads</a>);</div>
|
|
<div class="line"><a id="l03464" name="l03464"></a><span class="lineno"> 3464</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l03465" name="l03465"></a><span class="lineno"> 3465</span> </div>
|
|
<div class="foldopen" id="foldopen03466" data-start="{" data-end="}">
|
|
<div class="line"><a id="l03466" name="l03466"></a><span class="lineno"><a class="line" href="namespacepocketfft_1_1detail.html#ab26cbfed16f487b987f50bf63bfc1ab9"> 3466</a></span><span class="keyword">template</span><<span class="keyword">typename</span> T> <span class="keywordtype">void</span> c2r(<span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#a885ee37fcf564a268a5c8ca9ea8603e1">shape_t</a> &<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">shape_out</a>,</div>
|
|
<div class="line"><a id="l03467" name="l03467"></a><span class="lineno"> 3467</span> <span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#afb987c919e9424a996d0fc8b3c23cc84">stride_t</a> &stride_in, <span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#afb987c919e9424a996d0fc8b3c23cc84">stride_t</a> &stride_out, <span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axis</a>,</div>
|
|
<div class="line"><a id="l03468" name="l03468"></a><span class="lineno"> 3468</span> <span class="keywordtype">bool</span> forward, <span class="keyword">const</span> std::complex<T> *<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">data_in</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a> *<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">data_out</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a> fct,</div>
|
|
<div class="line"><a id="l03469" name="l03469"></a><span class="lineno"> 3469</span> <span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nthreads</a>=1)</div>
|
|
<div class="line"><a id="l03470" name="l03470"></a><span class="lineno"> 3470</span> {</div>
|
|
<div class="line"><a id="l03471" name="l03471"></a><span class="lineno"> 3471</span> <span class="keywordflow">if</span> (util::prod(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">shape_out</a>)==0) <span class="keywordflow">return</span>;</div>
|
|
<div class="line"><a id="l03472" name="l03472"></a><span class="lineno"> 3472</span> util::sanity_check(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">shape_out</a>, stride_in, stride_out, <span class="keyword">false</span>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axis</a>);</div>
|
|
<div class="line"><a id="l03473" name="l03473"></a><span class="lineno"> 3473</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#a885ee37fcf564a268a5c8ca9ea8603e1">shape_t</a> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">shape_in</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">shape_out</a>);</div>
|
|
<div class="line"><a id="l03474" name="l03474"></a><span class="lineno"> 3474</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">shape_in</a>[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axis</a>] = <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">shape_out</a>[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axis</a>]/2 + 1;</div>
|
|
<div class="line"><a id="l03475" name="l03475"></a><span class="lineno"> 3475</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cndarr<cmplx<T></a>> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ain</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">data_in</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">shape_in</a>, stride_in);</div>
|
|
<div class="line"><a id="l03476" name="l03476"></a><span class="lineno"> 3476</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ndarr<T></a> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">aout</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">data_out</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">shape_out</a>, stride_out);</div>
|
|
<div class="line"><a id="l03477" name="l03477"></a><span class="lineno"> 3477</span> <a class="code hl_function" href="namespacepocketfft_1_1detail.html#ac8ee38e8d8bcda875c99eeaf567550fc">general_c2r</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ain</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">aout</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axis</a>, forward, fct, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nthreads</a>);</div>
|
|
<div class="line"><a id="l03478" name="l03478"></a><span class="lineno"> 3478</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l03479" name="l03479"></a><span class="lineno"> 3479</span> </div>
|
|
<div class="foldopen" id="foldopen03480" data-start="{" data-end="}">
|
|
<div class="line"><a id="l03480" name="l03480"></a><span class="lineno"><a class="line" href="namespacepocketfft_1_1detail.html#a788506fff59f8e13056247076cac51c1"> 3480</a></span><span class="keyword">template</span><<span class="keyword">typename</span> T> <span class="keywordtype">void</span> c2r(<span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#a885ee37fcf564a268a5c8ca9ea8603e1">shape_t</a> &<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">shape_out</a>,</div>
|
|
<div class="line"><a id="l03481" name="l03481"></a><span class="lineno"> 3481</span> <span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#afb987c919e9424a996d0fc8b3c23cc84">stride_t</a> &stride_in, <span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#afb987c919e9424a996d0fc8b3c23cc84">stride_t</a> &stride_out, <span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#a885ee37fcf564a268a5c8ca9ea8603e1">shape_t</a> &<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axes</a>,</div>
|
|
<div class="line"><a id="l03482" name="l03482"></a><span class="lineno"> 3482</span> <span class="keywordtype">bool</span> forward, <span class="keyword">const</span> std::complex<T> *<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">data_in</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a> *<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">data_out</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a> fct,</div>
|
|
<div class="line"><a id="l03483" name="l03483"></a><span class="lineno"> 3483</span> <span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nthreads</a>=1)</div>
|
|
<div class="line"><a id="l03484" name="l03484"></a><span class="lineno"> 3484</span> {</div>
|
|
<div class="line"><a id="l03485" name="l03485"></a><span class="lineno"> 3485</span> <span class="keywordflow">if</span> (util::prod(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">shape_out</a>)==0) <span class="keywordflow">return</span>;</div>
|
|
<div class="line"><a id="l03486" name="l03486"></a><span class="lineno"> 3486</span> <span class="keywordflow">if</span> (<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axes</a>.size()==1)</div>
|
|
<div class="line"><a id="l03487" name="l03487"></a><span class="lineno"> 3487</span> <span class="keywordflow">return</span> c2r(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">shape_out</a>, stride_in, stride_out, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axes</a>[0], forward,</div>
|
|
<div class="line"><a id="l03488" name="l03488"></a><span class="lineno"> 3488</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">data_in</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">data_out</a>, fct, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nthreads</a>);</div>
|
|
<div class="line"><a id="l03489" name="l03489"></a><span class="lineno"> 3489</span> util::sanity_check(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">shape_out</a>, stride_in, stride_out, <span class="keyword">false</span>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axes</a>);</div>
|
|
<div class="line"><a id="l03490" name="l03490"></a><span class="lineno"> 3490</span> <span class="keyword">auto</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">shape_in</a> = <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">shape_out</a>;</div>
|
|
<div class="line"><a id="l03491" name="l03491"></a><span class="lineno"> 3491</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">shape_in</a>[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axes</a>.back()] = <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">shape_out</a>[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axes</a>.back()]/2 + 1;</div>
|
|
<div class="line"><a id="l03492" name="l03492"></a><span class="lineno"> 3492</span> <span class="keyword">auto</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nval</a> = util::prod(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">shape_in</a>);</div>
|
|
<div class="line"><a id="l03493" name="l03493"></a><span class="lineno"> 3493</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#afb987c919e9424a996d0fc8b3c23cc84">stride_t</a> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">stride_inter</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">shape_in</a>.size());</div>
|
|
<div class="line"><a id="l03494" name="l03494"></a><span class="lineno"> 3494</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">stride_inter</a>.back() = <span class="keyword">sizeof</span>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cmplx<T></a>);</div>
|
|
<div class="line"><a id="l03495" name="l03495"></a><span class="lineno"> 3495</span> <span class="keywordflow">for</span> (<span class="keywordtype">int</span> i=<span class="keywordtype">int</span>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">shape_in</a>.size())-2; i>=0; --i)</div>
|
|
<div class="line"><a id="l03496" name="l03496"></a><span class="lineno"> 3496</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">stride_inter</a>[<span class="keywordtype">size_t</span>(i)] =</div>
|
|
<div class="line"><a id="l03497" name="l03497"></a><span class="lineno"> 3497</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">stride_inter</a>[size_t(i+1)]*ptrdiff_t(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">shape_in</a>[<span class="keywordtype">size_t</span>(i+1)]);</div>
|
|
<div class="line"><a id="l03498" name="l03498"></a><span class="lineno"> 3498</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">arr<std::complex<T></a>> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tmp</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nval</a>);</div>
|
|
<div class="line"><a id="l03499" name="l03499"></a><span class="lineno"> 3499</span> <span class="keyword">auto</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">newaxes</a> = <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#a885ee37fcf564a268a5c8ca9ea8603e1">shape_t</a>{<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axes</a>.begin(), --<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axes</a>.end()};</div>
|
|
<div class="line"><a id="l03500" name="l03500"></a><span class="lineno"> 3500</span> c2c(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">shape_in</a>, stride_in, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">stride_inter</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">newaxes</a>, forward, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">data_in</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tmp</a>.data(),</div>
|
|
<div class="line"><a id="l03501" name="l03501"></a><span class="lineno"> 3501</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a>(1), <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nthreads</a>);</div>
|
|
<div class="line"><a id="l03502" name="l03502"></a><span class="lineno"> 3502</span> c2r(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">shape_out</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">stride_inter</a>, stride_out, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axes</a>.back(), forward,</div>
|
|
<div class="line"><a id="l03503" name="l03503"></a><span class="lineno"> 3503</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tmp</a>.data(), <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">data_out</a>, fct, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nthreads</a>);</div>
|
|
<div class="line"><a id="l03504" name="l03504"></a><span class="lineno"> 3504</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l03505" name="l03505"></a><span class="lineno"> 3505</span> </div>
|
|
<div class="foldopen" id="foldopen03506" data-start="{" data-end="}">
|
|
<div class="line"><a id="l03506" name="l03506"></a><span class="lineno"><a class="line" href="namespacepocketfft_1_1detail.html#a1ccca4cbbc6150d65620e2f9cdff62ac"> 3506</a></span><span class="keyword">template</span><<span class="keyword">typename</span> T> <span class="keywordtype">void</span> r2r_fftpack(<span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#a885ee37fcf564a268a5c8ca9ea8603e1">shape_t</a> &shape,</div>
|
|
<div class="line"><a id="l03507" name="l03507"></a><span class="lineno"> 3507</span> <span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#afb987c919e9424a996d0fc8b3c23cc84">stride_t</a> &stride_in, <span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#afb987c919e9424a996d0fc8b3c23cc84">stride_t</a> &stride_out, <span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#a885ee37fcf564a268a5c8ca9ea8603e1">shape_t</a> &<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axes</a>,</div>
|
|
<div class="line"><a id="l03508" name="l03508"></a><span class="lineno"> 3508</span> <span class="keywordtype">bool</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">real2hermitian</a>, <span class="keywordtype">bool</span> forward, <span class="keyword">const</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a> *<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">data_in</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a> *<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">data_out</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a> fct,</div>
|
|
<div class="line"><a id="l03509" name="l03509"></a><span class="lineno"> 3509</span> <span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nthreads</a>=1)</div>
|
|
<div class="line"><a id="l03510" name="l03510"></a><span class="lineno"> 3510</span> {</div>
|
|
<div class="line"><a id="l03511" name="l03511"></a><span class="lineno"> 3511</span> <span class="keywordflow">if</span> (util::prod(shape)==0) <span class="keywordflow">return</span>;</div>
|
|
<div class="line"><a id="l03512" name="l03512"></a><span class="lineno"> 3512</span> util::sanity_check(shape, stride_in, stride_out, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">data_in</a>==<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">data_out</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axes</a>);</div>
|
|
<div class="line"><a id="l03513" name="l03513"></a><span class="lineno"> 3513</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cndarr<T></a> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ain</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">data_in</a>, shape, stride_in);</div>
|
|
<div class="line"><a id="l03514" name="l03514"></a><span class="lineno"> 3514</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ndarr<T></a> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">aout</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">data_out</a>, shape, stride_out);</div>
|
|
<div class="line"><a id="l03515" name="l03515"></a><span class="lineno"> 3515</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">general_nd<pocketfft_r<T></a>>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ain</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">aout</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axes</a>, fct, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nthreads</a>,</div>
|
|
<div class="line"><a id="l03516" name="l03516"></a><span class="lineno"> 3516</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1_exec_r2_r.html">ExecR2R</a>{<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">real2hermitian</a>, forward});</div>
|
|
<div class="line"><a id="l03517" name="l03517"></a><span class="lineno"> 3517</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l03518" name="l03518"></a><span class="lineno"> 3518</span> </div>
|
|
<div class="foldopen" id="foldopen03519" data-start="{" data-end="}">
|
|
<div class="line"><a id="l03519" name="l03519"></a><span class="lineno"><a class="line" href="namespacepocketfft_1_1detail.html#a40b56ae882eea17e6960618415f8eb42"> 3519</a></span><span class="keyword">template</span><<span class="keyword">typename</span> T> <span class="keywordtype">void</span> r2r_separable_hartley(<span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#a885ee37fcf564a268a5c8ca9ea8603e1">shape_t</a> &shape,</div>
|
|
<div class="line"><a id="l03520" name="l03520"></a><span class="lineno"> 3520</span> <span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#afb987c919e9424a996d0fc8b3c23cc84">stride_t</a> &stride_in, <span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#afb987c919e9424a996d0fc8b3c23cc84">stride_t</a> &stride_out, <span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#a885ee37fcf564a268a5c8ca9ea8603e1">shape_t</a> &<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axes</a>,</div>
|
|
<div class="line"><a id="l03521" name="l03521"></a><span class="lineno"> 3521</span> <span class="keyword">const</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a> *<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">data_in</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a> *<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">data_out</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a> fct, <span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nthreads</a>=1)</div>
|
|
<div class="line"><a id="l03522" name="l03522"></a><span class="lineno"> 3522</span> {</div>
|
|
<div class="line"><a id="l03523" name="l03523"></a><span class="lineno"> 3523</span> <span class="keywordflow">if</span> (util::prod(shape)==0) <span class="keywordflow">return</span>;</div>
|
|
<div class="line"><a id="l03524" name="l03524"></a><span class="lineno"> 3524</span> util::sanity_check(shape, stride_in, stride_out, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">data_in</a>==<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">data_out</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axes</a>);</div>
|
|
<div class="line"><a id="l03525" name="l03525"></a><span class="lineno"> 3525</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cndarr<T></a> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ain</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">data_in</a>, shape, stride_in);</div>
|
|
<div class="line"><a id="l03526" name="l03526"></a><span class="lineno"> 3526</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ndarr<T></a> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">aout</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">data_out</a>, shape, stride_out);</div>
|
|
<div class="line"><a id="l03527" name="l03527"></a><span class="lineno"> 3527</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">general_nd<pocketfft_r<T></a>>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ain</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">aout</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axes</a>, fct, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nthreads</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1_exec_hartley.html">ExecHartley</a>{},</div>
|
|
<div class="line"><a id="l03528" name="l03528"></a><span class="lineno"> 3528</span> <span class="keyword">false</span>);</div>
|
|
<div class="line"><a id="l03529" name="l03529"></a><span class="lineno"> 3529</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l03530" name="l03530"></a><span class="lineno"> 3530</span> </div>
|
|
<div class="foldopen" id="foldopen03531" data-start="{" data-end="}">
|
|
<div class="line"><a id="l03531" name="l03531"></a><span class="lineno"><a class="line" href="namespacepocketfft_1_1detail.html#a072a67f2c4b3b3ebd030604f3383e1ed"> 3531</a></span><span class="keyword">template</span><<span class="keyword">typename</span> T> <span class="keywordtype">void</span> r2r_genuine_hartley(<span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#a885ee37fcf564a268a5c8ca9ea8603e1">shape_t</a> &shape,</div>
|
|
<div class="line"><a id="l03532" name="l03532"></a><span class="lineno"> 3532</span> <span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#afb987c919e9424a996d0fc8b3c23cc84">stride_t</a> &stride_in, <span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#afb987c919e9424a996d0fc8b3c23cc84">stride_t</a> &stride_out, <span class="keyword">const</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#a885ee37fcf564a268a5c8ca9ea8603e1">shape_t</a> &<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axes</a>,</div>
|
|
<div class="line"><a id="l03533" name="l03533"></a><span class="lineno"> 3533</span> <span class="keyword">const</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a> *<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">data_in</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a> *<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">data_out</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">T</a> fct, <span class="keywordtype">size_t</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nthreads</a>=1)</div>
|
|
<div class="line"><a id="l03534" name="l03534"></a><span class="lineno"> 3534</span> {</div>
|
|
<div class="line"><a id="l03535" name="l03535"></a><span class="lineno"> 3535</span> <span class="keywordflow">if</span> (util::prod(shape)==0) <span class="keywordflow">return</span>;</div>
|
|
<div class="line"><a id="l03536" name="l03536"></a><span class="lineno"> 3536</span> <span class="keywordflow">if</span> (<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axes</a>.size()==1)</div>
|
|
<div class="line"><a id="l03537" name="l03537"></a><span class="lineno"> 3537</span> <span class="keywordflow">return</span> r2r_separable_hartley(shape, stride_in, stride_out, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axes</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">data_in</a>,</div>
|
|
<div class="line"><a id="l03538" name="l03538"></a><span class="lineno"> 3538</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">data_out</a>, fct, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nthreads</a>);</div>
|
|
<div class="line"><a id="l03539" name="l03539"></a><span class="lineno"> 3539</span> util::sanity_check(shape, stride_in, stride_out, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">data_in</a>==<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">data_out</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axes</a>);</div>
|
|
<div class="line"><a id="l03540" name="l03540"></a><span class="lineno"> 3540</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#a885ee37fcf564a268a5c8ca9ea8603e1">shape_t</a> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tshp</a>(shape);</div>
|
|
<div class="line"><a id="l03541" name="l03541"></a><span class="lineno"> 3541</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tshp</a>[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axes</a>.back()] = <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tshp</a>[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axes</a>.back()]/2+1;</div>
|
|
<div class="line"><a id="l03542" name="l03542"></a><span class="lineno"> 3542</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">arr<std::complex<T></a>> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tdata</a>(util::prod(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tshp</a>));</div>
|
|
<div class="line"><a id="l03543" name="l03543"></a><span class="lineno"> 3543</span> <a class="code hl_typedef" href="namespacepocketfft_1_1detail.html#afb987c919e9424a996d0fc8b3c23cc84">stride_t</a> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tstride</a>(shape.size());</div>
|
|
<div class="line"><a id="l03544" name="l03544"></a><span class="lineno"> 3544</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tstride</a>.back()=<span class="keyword">sizeof</span>(std::complex<T>);</div>
|
|
<div class="line"><a id="l03545" name="l03545"></a><span class="lineno"> 3545</span> <span class="keywordflow">for</span> (<span class="keywordtype">size_t</span> i=<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tstride</a>.size()-1; i>0; --i)</div>
|
|
<div class="line"><a id="l03546" name="l03546"></a><span class="lineno"> 3546</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tstride</a>[i-1]=<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tstride</a>[i]*ptrdiff_t(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tshp</a>[i]);</div>
|
|
<div class="line"><a id="l03547" name="l03547"></a><span class="lineno"> 3547</span> r2c(shape, stride_in, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tstride</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axes</a>, <span class="keyword">true</span>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">data_in</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tdata</a>.data(), fct, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">nthreads</a>);</div>
|
|
<div class="line"><a id="l03548" name="l03548"></a><span class="lineno"> 3548</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">cndarr<cmplx<T></a>> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">atmp</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tdata</a>.data(), <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tshp</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">tstride</a>);</div>
|
|
<div class="line"><a id="l03549" name="l03549"></a><span class="lineno"> 3549</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">ndarr<T></a> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">aout</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">data_out</a>, shape, stride_out);</div>
|
|
<div class="line"><a id="l03550" name="l03550"></a><span class="lineno"> 3550</span> <a class="code hl_class" href="classpocketfft_1_1detail_1_1simple__iter.html">simple_iter</a> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">iin</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">atmp</a>);</div>
|
|
<div class="line"><a id="l03551" name="l03551"></a><span class="lineno"> 3551</span> <a class="code hl_class" href="classpocketfft_1_1detail_1_1rev__iter.html">rev_iter</a> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">iout</a>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">aout</a>, <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">axes</a>);</div>
|
|
<div class="line"><a id="l03552" name="l03552"></a><span class="lineno"> 3552</span> <span class="keywordflow">while</span>(<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">iin</a>.remaining()>0)</div>
|
|
<div class="line"><a id="l03553" name="l03553"></a><span class="lineno"> 3553</span> {</div>
|
|
<div class="line"><a id="l03554" name="l03554"></a><span class="lineno"> 3554</span> <span class="keyword">auto</span> v = <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">atmp</a>[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">iin</a>.ofs()];</div>
|
|
<div class="line"><a id="l03555" name="l03555"></a><span class="lineno"> 3555</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">aout</a>[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">iout</a>.ofs()] = v.<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a>+v.<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#a35d2dce1b7de5f37d7029e639bc7f23d">i</a>;</div>
|
|
<div class="line"><a id="l03556" name="l03556"></a><span class="lineno"> 3556</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">aout</a>[<a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">iout</a>.rev_ofs()] = v.<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">r</a>-v.<a class="code hl_variable" href="structpocketfft_1_1detail_1_1cmplx.html#a35d2dce1b7de5f37d7029e639bc7f23d">i</a>;</div>
|
|
<div class="line"><a id="l03557" name="l03557"></a><span class="lineno"> 3557</span> <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">iin</a>.advance(); <a class="code hl_struct" href="structpocketfft_1_1detail_1_1cmplx.html">iout</a>.advance();</div>
|
|
<div class="line"><a id="l03558" name="l03558"></a><span class="lineno"> 3558</span> }</div>
|
|
<div class="line"><a id="l03559" name="l03559"></a><span class="lineno"> 3559</span> }</div>
|
|
</div>
|
|
<div class="line"><a id="l03560" name="l03560"></a><span class="lineno"> 3560</span> </div>
|
|
<div class="line"><a id="l03561" name="l03561"></a><span class="lineno"> 3561</span>} <span class="comment">// namespace detail</span></div>
|
|
</div>
|
|
<div class="line"><a id="l03562" name="l03562"></a><span class="lineno"> 3562</span> </div>
|
|
<div class="line"><a id="l03563" name="l03563"></a><span class="lineno"> 3563</span><span class="keyword">using </span>detail::FORWARD;</div>
|
|
<div class="line"><a id="l03564" name="l03564"></a><span class="lineno"> 3564</span><span class="keyword">using </span>detail::BACKWARD;</div>
|
|
<div class="line"><a id="l03565" name="l03565"></a><span class="lineno"> 3565</span><span class="keyword">using </span>detail::shape_t;</div>
|
|
<div class="line"><a id="l03566" name="l03566"></a><span class="lineno"> 3566</span><span class="keyword">using </span>detail::stride_t;</div>
|
|
<div class="line"><a id="l03567" name="l03567"></a><span class="lineno"> 3567</span><span class="keyword">using </span>detail::c2c;</div>
|
|
<div class="line"><a id="l03568" name="l03568"></a><span class="lineno"> 3568</span><span class="keyword">using </span>detail::c2r;</div>
|
|
<div class="line"><a id="l03569" name="l03569"></a><span class="lineno"> 3569</span><span class="keyword">using </span>detail::r2c;</div>
|
|
<div class="line"><a id="l03570" name="l03570"></a><span class="lineno"> 3570</span><span class="keyword">using </span>detail::r2r_fftpack;</div>
|
|
<div class="line"><a id="l03571" name="l03571"></a><span class="lineno"> 3571</span><span class="keyword">using </span>detail::r2r_separable_hartley;</div>
|
|
<div class="line"><a id="l03572" name="l03572"></a><span class="lineno"> 3572</span><span class="keyword">using </span>detail::r2r_genuine_hartley;</div>
|
|
<div class="line"><a id="l03573" name="l03573"></a><span class="lineno"> 3573</span><span class="keyword">using </span>detail::dct;</div>
|
|
<div class="line"><a id="l03574" name="l03574"></a><span class="lineno"> 3574</span><span class="keyword">using </span>detail::dst;</div>
|
|
<div class="line"><a id="l03575" name="l03575"></a><span class="lineno"> 3575</span> </div>
|
|
<div class="line"><a id="l03576" name="l03576"></a><span class="lineno"> 3576</span>} <span class="comment">// namespace pocketfft</span></div>
|
|
</div>
|
|
<div class="line"><a id="l03577" name="l03577"></a><span class="lineno"> 3577</span> </div>
|
|
<div class="line"><a id="l03578" name="l03578"></a><span class="lineno"> 3578</span><span class="preprocessor">#undef POCKETFFT_NOINLINE</span></div>
|
|
<div class="line"><a id="l03579" name="l03579"></a><span class="lineno"> 3579</span><span class="preprocessor">#undef POCKETFFT_RESTRICT</span></div>
|
|
<div class="line"><a id="l03580" name="l03580"></a><span class="lineno"> 3580</span> </div>
|
|
<div class="line"><a id="l03581" name="l03581"></a><span class="lineno"> 3581</span><span class="preprocessor">#endif </span><span class="comment">// POCKETFFT_HDRONLY_H</span></div>
|
|
<div class="ttc" id="abackend_2metal_2allocator_8h_html_a15aa5cc1baf29be08d55cca88509e697"><div class="ttname"><a href="backend_2metal_2allocator_8h.html#a15aa5cc1baf29be08d55cca88509e697">buf</a></div><div class="ttdeci">MTL::Buffer * buf</div><div class="ttdef"><b>Definition</b> allocator.h:38</div></div>
|
|
<div class="ttc" id="abackend_2metal_2kernels_2fft_8h_html_a4010b0e151e5f01e610e9c32234458c7"><div class="ttname"><a href="backend_2metal_2kernels_2fft_8h.html#a4010b0e151e5f01e610e9c32234458c7">fft</a></div><div class="ttdeci">void fft(const device in_T *in, device out_T *out, constant const int &n, constant const int &batch_size, uint3 elem, uint3 grid)</div><div class="ttdef"><b>Definition</b> fft.h:180</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1_t__dcst23_html"><div class="ttname"><a href="classpocketfft_1_1detail_1_1_t__dcst23.html">pocketfft::detail::T_dcst23</a></div><div class="ttdef"><b>Definition</b> pocketfft.h:2595</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1_t__dcst23_html_a2a45b7b4612904c2be69c01f6d5029ac"><div class="ttname"><a href="classpocketfft_1_1detail_1_1_t__dcst23.html#a2a45b7b4612904c2be69c01f6d5029ac">pocketfft::detail::T_dcst23::exec</a></div><div class="ttdeci">void exec(T c[], T0 fct, bool ortho, int type, bool cosine) const</div><div class="ttdef"><b>Definition</b> pocketfft.h:2609</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1_t__dcst23_html_a6dab012b487ff98d11b8a9418653a478"><div class="ttname"><a href="classpocketfft_1_1detail_1_1_t__dcst23.html#a6dab012b487ff98d11b8a9418653a478">pocketfft::detail::T_dcst23::length</a></div><div class="ttdeci">size_t length() const</div><div class="ttdef"><b>Definition</b> pocketfft.h:2661</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1_t__dcst23_html_a7cbe3fd1b64383d8b466bcdc2f594900"><div class="ttname"><a href="classpocketfft_1_1detail_1_1_t__dcst23.html#a7cbe3fd1b64383d8b466bcdc2f594900">pocketfft::detail::T_dcst23::T_dcst23</a></div><div class="ttdeci">T_dcst23(size_t length)</div><div class="ttdef"><b>Definition</b> pocketfft.h:2601</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1_t__dcst4_html"><div class="ttname"><a href="classpocketfft_1_1detail_1_1_t__dcst4.html">pocketfft::detail::T_dcst4</a></div><div class="ttdef"><b>Definition</b> pocketfft.h:2665</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1_t__dcst4_html_a4a623ebc5dfe8c3cad153c642663099b"><div class="ttname"><a href="classpocketfft_1_1detail_1_1_t__dcst4.html#a4a623ebc5dfe8c3cad153c642663099b">pocketfft::detail::T_dcst4::T_dcst4</a></div><div class="ttdeci">T_dcst4(size_t length)</div><div class="ttdef"><b>Definition</b> pocketfft.h:2673</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1_t__dcst4_html_af25bf28a7ccd4690ca9934e3aa79c12f"><div class="ttname"><a href="classpocketfft_1_1detail_1_1_t__dcst4.html#af25bf28a7ccd4690ca9934e3aa79c12f">pocketfft::detail::T_dcst4::length</a></div><div class="ttdeci">size_t length() const</div><div class="ttdef"><b>Definition</b> pocketfft.h:2761</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1_t__dcst4_html_af794ebf21009d5f918681188081df708"><div class="ttname"><a href="classpocketfft_1_1detail_1_1_t__dcst4.html#af794ebf21009d5f918681188081df708">pocketfft::detail::T_dcst4::exec</a></div><div class="ttdeci">void exec(T c[], T0 fct, bool, int, bool cosine) const</div><div class="ttdef"><b>Definition</b> pocketfft.h:2687</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1_t__dct1_html"><div class="ttname"><a href="classpocketfft_1_1detail_1_1_t__dct1.html">pocketfft::detail::T_dct1</a></div><div class="ttdef"><b>Definition</b> pocketfft.h:2539</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1_t__dct1_html_a7736111ff9d220f983e41a6fecd5f058"><div class="ttname"><a href="classpocketfft_1_1detail_1_1_t__dct1.html#a7736111ff9d220f983e41a6fecd5f058">pocketfft::detail::T_dct1::exec</a></div><div class="ttdeci">void exec(T c[], T0 fct, bool ortho, int, bool) const</div><div class="ttdef"><b>Definition</b> pocketfft.h:2547</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1_t__dct1_html_ac153e82c8bf0943638a88486e35c16d6"><div class="ttname"><a href="classpocketfft_1_1detail_1_1_t__dct1.html#ac153e82c8bf0943638a88486e35c16d6">pocketfft::detail::T_dct1::T_dct1</a></div><div class="ttdeci">T_dct1(size_t length)</div><div class="ttdef"><b>Definition</b> pocketfft.h:2544</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1_t__dct1_html_ac7a04c91d507bd8f173d2266bb5bb168"><div class="ttname"><a href="classpocketfft_1_1detail_1_1_t__dct1.html#ac7a04c91d507bd8f173d2266bb5bb168">pocketfft::detail::T_dct1::length</a></div><div class="ttdeci">size_t length() const</div><div class="ttdef"><b>Definition</b> pocketfft.h:2566</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1_t__dst1_html"><div class="ttname"><a href="classpocketfft_1_1detail_1_1_t__dst1.html">pocketfft::detail::T_dst1</a></div><div class="ttdef"><b>Definition</b> pocketfft.h:2570</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1_t__dst1_html_a598a9511004263eb3610053d7efc9e26"><div class="ttname"><a href="classpocketfft_1_1detail_1_1_t__dst1.html#a598a9511004263eb3610053d7efc9e26">pocketfft::detail::T_dst1::exec</a></div><div class="ttdeci">void exec(T c[], T0 fct, bool, int, bool) const</div><div class="ttdef"><b>Definition</b> pocketfft.h:2578</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1_t__dst1_html_ab205d901650e38b592ff860b7978fa3e"><div class="ttname"><a href="classpocketfft_1_1detail_1_1_t__dst1.html#ab205d901650e38b592ff860b7978fa3e">pocketfft::detail::T_dst1::length</a></div><div class="ttdeci">size_t length() const</div><div class="ttdef"><b>Definition</b> pocketfft.h:2591</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1_t__dst1_html_ad0b6647266ef0e87d34730774b92da84"><div class="ttname"><a href="classpocketfft_1_1detail_1_1_t__dst1.html#ad0b6647266ef0e87d34730774b92da84">pocketfft::detail::T_dst1::T_dst1</a></div><div class="ttdeci">T_dst1(size_t length)</div><div class="ttdef"><b>Definition</b> pocketfft.h:2575</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1arr__info_html"><div class="ttname"><a href="classpocketfft_1_1detail_1_1arr__info.html">pocketfft::detail::arr_info</a></div><div class="ttdef"><b>Definition</b> pocketfft.h:2823</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1arr__info_html_a003a7106f7fa59a3c55ac1f0116313a5"><div class="ttname"><a href="classpocketfft_1_1detail_1_1arr__info.html#a003a7106f7fa59a3c55ac1f0116313a5">pocketfft::detail::arr_info::size</a></div><div class="ttdeci">size_t size() const</div><div class="ttdef"><b>Definition</b> pocketfft.h:2832</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1arr__info_html_a0dbddb7d86ca306159fc9ef9a453b21e"><div class="ttname"><a href="classpocketfft_1_1detail_1_1arr__info.html#a0dbddb7d86ca306159fc9ef9a453b21e">pocketfft::detail::arr_info::arr_info</a></div><div class="ttdeci">arr_info(const shape_t &shape_, const stride_t &stride_)</div><div class="ttdef"><b>Definition</b> pocketfft.h:2829</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1arr__info_html_a2467e9e01de1ba4d7cd28c1af783da8d"><div class="ttname"><a href="classpocketfft_1_1detail_1_1arr__info.html#a2467e9e01de1ba4d7cd28c1af783da8d">pocketfft::detail::arr_info::shp</a></div><div class="ttdeci">shape_t shp</div><div class="ttdef"><b>Definition</b> pocketfft.h:2825</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1arr__info_html_a9d10aa83a1117e75d36f7396b8c2a093"><div class="ttname"><a href="classpocketfft_1_1detail_1_1arr__info.html#a9d10aa83a1117e75d36f7396b8c2a093">pocketfft::detail::arr_info::stride</a></div><div class="ttdeci">const stride_t & stride() const</div><div class="ttdef"><b>Definition</b> pocketfft.h:2835</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1arr__info_html_abe1f7b92501b4e0e5a38fd26294ac5a4"><div class="ttname"><a href="classpocketfft_1_1detail_1_1arr__info.html#abe1f7b92501b4e0e5a38fd26294ac5a4">pocketfft::detail::arr_info::str</a></div><div class="ttdeci">stride_t str</div><div class="ttdef"><b>Definition</b> pocketfft.h:2826</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1arr__info_html_ac1f6a9bd6703eceef6003f5f6315d39b"><div class="ttname"><a href="classpocketfft_1_1detail_1_1arr__info.html#ac1f6a9bd6703eceef6003f5f6315d39b">pocketfft::detail::arr_info::stride</a></div><div class="ttdeci">const ptrdiff_t & stride(size_t i) const</div><div class="ttdef"><b>Definition</b> pocketfft.h:2836</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1arr__info_html_ac601c660c64a4c252aa8be4ae7dfa7a8"><div class="ttname"><a href="classpocketfft_1_1detail_1_1arr__info.html#ac601c660c64a4c252aa8be4ae7dfa7a8">pocketfft::detail::arr_info::shape</a></div><div class="ttdeci">size_t shape(size_t i) const</div><div class="ttdef"><b>Definition</b> pocketfft.h:2834</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1arr__info_html_ac608c8af2a59a28a0012e308be7ee414"><div class="ttname"><a href="classpocketfft_1_1detail_1_1arr__info.html#ac608c8af2a59a28a0012e308be7ee414">pocketfft::detail::arr_info::ndim</a></div><div class="ttdeci">size_t ndim() const</div><div class="ttdef"><b>Definition</b> pocketfft.h:2831</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1arr__info_html_accada8146cb8d3ab7facb4c1e3413ec0"><div class="ttname"><a href="classpocketfft_1_1detail_1_1arr__info.html#accada8146cb8d3ab7facb4c1e3413ec0">pocketfft::detail::arr_info::shape</a></div><div class="ttdeci">const shape_t & shape() const</div><div class="ttdef"><b>Definition</b> pocketfft.h:2833</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1arr_html"><div class="ttname"><a href="classpocketfft_1_1detail_1_1arr.html">pocketfft::detail::arr</a></div><div class="ttdef"><b>Definition</b> pocketfft.h:181</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1arr_html_a04f832b780a4453fdf3b69bf75b182bd"><div class="ttname"><a href="classpocketfft_1_1detail_1_1arr.html#a04f832b780a4453fdf3b69bf75b182bd">pocketfft::detail::arr::arr</a></div><div class="ttdeci">arr(size_t n)</div><div class="ttdef"><b>Definition</b> pocketfft.h:209</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1arr_html_a0cd8fb4a588a74d428a7349d38b477d0"><div class="ttname"><a href="classpocketfft_1_1detail_1_1arr.html#a0cd8fb4a588a74d428a7349d38b477d0">pocketfft::detail::arr::arr</a></div><div class="ttdeci">arr(arr &&other)</div><div class="ttdef"><b>Definition</b> pocketfft.h:210</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1arr_html_a8519487ee7fa95d3937a8352dd289b56"><div class="ttname"><a href="classpocketfft_1_1detail_1_1arr.html#a8519487ee7fa95d3937a8352dd289b56">pocketfft::detail::arr::~arr</a></div><div class="ttdeci">~arr()</div><div class="ttdef"><b>Definition</b> pocketfft.h:213</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1arr_html_a8d73baaefa02dff8714e4398c83917e0"><div class="ttname"><a href="classpocketfft_1_1detail_1_1arr.html#a8d73baaefa02dff8714e4398c83917e0">pocketfft::detail::arr::resize</a></div><div class="ttdeci">void resize(size_t n)</div><div class="ttdef"><b>Definition</b> pocketfft.h:215</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1arr_html_a95bca00060957f540ff25b69632c6952"><div class="ttname"><a href="classpocketfft_1_1detail_1_1arr.html#a95bca00060957f540ff25b69632c6952">pocketfft::detail::arr::size</a></div><div class="ttdeci">size_t size() const</div><div class="ttdef"><b>Definition</b> pocketfft.h:229</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1arr_html_a961a24410638b35129cd6b81850d2a42"><div class="ttname"><a href="classpocketfft_1_1detail_1_1arr.html#a961a24410638b35129cd6b81850d2a42">pocketfft::detail::arr::arr</a></div><div class="ttdeci">arr()</div><div class="ttdef"><b>Definition</b> pocketfft.h:208</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1arr_html_a99c54f96bc79c7cdd8925c1663462842"><div class="ttname"><a href="classpocketfft_1_1detail_1_1arr.html#a99c54f96bc79c7cdd8925c1663462842">pocketfft::detail::arr::operator[]</a></div><div class="ttdeci">const T & operator[](size_t idx) const</div><div class="ttdef"><b>Definition</b> pocketfft.h:224</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1arr_html_ac82daa17e9f991072b012343f9d7c182"><div class="ttname"><a href="classpocketfft_1_1detail_1_1arr.html#ac82daa17e9f991072b012343f9d7c182">pocketfft::detail::arr::data</a></div><div class="ttdeci">const T * data() const</div><div class="ttdef"><b>Definition</b> pocketfft.h:227</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1arr_html_aea0bd899b19e03f54dfd6c188727061a"><div class="ttname"><a href="classpocketfft_1_1detail_1_1arr.html#aea0bd899b19e03f54dfd6c188727061a">pocketfft::detail::arr::operator[]</a></div><div class="ttdeci">T & operator[](size_t idx)</div><div class="ttdef"><b>Definition</b> pocketfft.h:223</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1arr_html_aec0f2191b4663b4187aab92454c34de8"><div class="ttname"><a href="classpocketfft_1_1detail_1_1arr.html#aec0f2191b4663b4187aab92454c34de8">pocketfft::detail::arr::data</a></div><div class="ttdeci">T * data()</div><div class="ttdef"><b>Definition</b> pocketfft.h:226</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1cfftp_html"><div class="ttname"><a href="classpocketfft_1_1detail_1_1cfftp.html">pocketfft::detail::cfftp</a></div><div class="ttdef"><b>Definition</b> pocketfft.h:827</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1cfftp_html_a121dcd8d4db180061d09fc1c5eb7da27"><div class="ttname"><a href="classpocketfft_1_1detail_1_1cfftp.html#a121dcd8d4db180061d09fc1c5eb7da27">pocketfft::detail::cfftp::cfftp</a></div><div class="ttdeci">cfftp(size_t length_)</div><div class="ttdef"><b>Definition</b> pocketfft.h:1537</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1cfftp_html_a95211024bf007d27e700835db556fbd2"><div class="ttname"><a href="classpocketfft_1_1detail_1_1cfftp.html#a95211024bf007d27e700835db556fbd2">pocketfft::detail::cfftp::exec</a></div><div class="ttdeci">void exec(T c[], T0 fct, bool fwd) const</div><div class="ttdef"><b>Definition</b> pocketfft.h:1471</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1cndarr_html"><div class="ttname"><a href="classpocketfft_1_1detail_1_1cndarr.html">pocketfft::detail::cndarr</a></div><div class="ttdef"><b>Definition</b> pocketfft.h:2840</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1cndarr_html_abf73f1b4ddcfb27d7f85cfa441607129"><div class="ttname"><a href="classpocketfft_1_1detail_1_1cndarr.html#abf73f1b4ddcfb27d7f85cfa441607129">pocketfft::detail::cndarr::cndarr</a></div><div class="ttdeci">cndarr(const void *data_, const shape_t &shape_, const stride_t &stride_)</div><div class="ttdef"><b>Definition</b> pocketfft.h:2845</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1cndarr_html_ac29c769aebb03f81fbcf16ba6e766af2"><div class="ttname"><a href="classpocketfft_1_1detail_1_1cndarr.html#ac29c769aebb03f81fbcf16ba6e766af2">pocketfft::detail::cndarr::d</a></div><div class="ttdeci">const char * d</div><div class="ttdef"><b>Definition</b> pocketfft.h:2842</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1cndarr_html_ae4852d1fe936a5d61832b507816c7054"><div class="ttname"><a href="classpocketfft_1_1detail_1_1cndarr.html#ae4852d1fe936a5d61832b507816c7054">pocketfft::detail::cndarr::operator[]</a></div><div class="ttdeci">const T & operator[](ptrdiff_t ofs) const</div><div class="ttdef"><b>Definition</b> pocketfft.h:2848</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1fftblue_html"><div class="ttname"><a href="classpocketfft_1_1detail_1_1fftblue.html">pocketfft::detail::fftblue</a></div><div class="ttdef"><b>Definition</b> pocketfft.h:2362</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1fftblue_html_a5fb03413a3d1a653842875adcf87ae8c"><div class="ttname"><a href="classpocketfft_1_1detail_1_1fftblue.html#a5fb03413a3d1a653842875adcf87ae8c">pocketfft::detail::fftblue::exec</a></div><div class="ttdeci">void exec(cmplx< T > c[], T0 fct, bool fwd) const</div><div class="ttdef"><b>Definition</b> pocketfft.h:2430</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1fftblue_html_a642b4aff0485c7d9c8794161a1464f00"><div class="ttname"><a href="classpocketfft_1_1detail_1_1fftblue.html#a642b4aff0485c7d9c8794161a1464f00">pocketfft::detail::fftblue::exec_r</a></div><div class="ttdeci">void exec_r(T c[], T0 fct, bool fwd)</div><div class="ttdef"><b>Definition</b> pocketfft.h:2433</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1fftblue_html_a8613b2131e7843bbca9e1cd4cc15bc01"><div class="ttname"><a href="classpocketfft_1_1detail_1_1fftblue.html#a8613b2131e7843bbca9e1cd4cc15bc01">pocketfft::detail::fftblue::fftblue</a></div><div class="ttdeci">fftblue(size_t length)</div><div class="ttdef"><b>Definition</b> pocketfft.h:2401</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1multi__iter_html"><div class="ttname"><a href="classpocketfft_1_1detail_1_1multi__iter.html">pocketfft::detail::multi_iter</a></div><div class="ttdef"><b>Definition</b> pocketfft.h:2863</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1multi__iter_html_a034d12f842df90e6471dffd3fa6ba4bd"><div class="ttname"><a href="classpocketfft_1_1detail_1_1multi__iter.html#a034d12f842df90e6471dffd3fa6ba4bd">pocketfft::detail::multi_iter::remaining</a></div><div class="ttdeci">size_t remaining() const</div><div class="ttdef"><b>Definition</b> pocketfft.h:2935</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1multi__iter_html_a5318b79d934cddf109dff7bf96a330c8"><div class="ttname"><a href="classpocketfft_1_1detail_1_1multi__iter.html#a5318b79d934cddf109dff7bf96a330c8">pocketfft::detail::multi_iter::length_in</a></div><div class="ttdeci">size_t length_in() const</div><div class="ttdef"><b>Definition</b> pocketfft.h:2931</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1multi__iter_html_a5ddcc0666125b3cb6c0d62b30befdd2c"><div class="ttname"><a href="classpocketfft_1_1detail_1_1multi__iter.html#a5ddcc0666125b3cb6c0d62b30befdd2c">pocketfft::detail::multi_iter::advance</a></div><div class="ttdeci">void advance(size_t n)</div><div class="ttdef"><b>Definition</b> pocketfft.h:2916</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1multi__iter_html_a81d71a13bf0b85e556fbb9834167ecc7"><div class="ttname"><a href="classpocketfft_1_1detail_1_1multi__iter.html#a81d71a13bf0b85e556fbb9834167ecc7">pocketfft::detail::multi_iter::stride_out</a></div><div class="ttdeci">ptrdiff_t stride_out() const</div><div class="ttdef"><b>Definition</b> pocketfft.h:2934</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1multi__iter_html_a9236047e7419e5d21379cbf95eb3a78e"><div class="ttname"><a href="classpocketfft_1_1detail_1_1multi__iter.html#a9236047e7419e5d21379cbf95eb3a78e">pocketfft::detail::multi_iter::oofs</a></div><div class="ttdeci">ptrdiff_t oofs(size_t j, size_t i) const</div><div class="ttdef"><b>Definition</b> pocketfft.h:2930</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1multi__iter_html_a93cd515d07cd479138a35da9df66bd41"><div class="ttname"><a href="classpocketfft_1_1detail_1_1multi__iter.html#a93cd515d07cd479138a35da9df66bd41">pocketfft::detail::multi_iter::length_out</a></div><div class="ttdeci">size_t length_out() const</div><div class="ttdef"><b>Definition</b> pocketfft.h:2932</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1multi__iter_html_a97462d97bdca6419d8d2f37c2031fe83"><div class="ttname"><a href="classpocketfft_1_1detail_1_1multi__iter.html#a97462d97bdca6419d8d2f37c2031fe83">pocketfft::detail::multi_iter::iofs</a></div><div class="ttdeci">ptrdiff_t iofs(size_t j, size_t i) const</div><div class="ttdef"><b>Definition</b> pocketfft.h:2928</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1multi__iter_html_a9be43bb18840202da6d17988fccc64b9"><div class="ttname"><a href="classpocketfft_1_1detail_1_1multi__iter.html#a9be43bb18840202da6d17988fccc64b9">pocketfft::detail::multi_iter::multi_iter</a></div><div class="ttdeci">multi_iter(const arr_info &iarr_, const arr_info &oarr_, size_t idim_)</div><div class="ttdef"><b>Definition</b> pocketfft.h:2887</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1multi__iter_html_aae63e67caac095d474ddd32daa5ffa34"><div class="ttname"><a href="classpocketfft_1_1detail_1_1multi__iter.html#aae63e67caac095d474ddd32daa5ffa34">pocketfft::detail::multi_iter::oofs</a></div><div class="ttdeci">ptrdiff_t oofs(size_t i) const</div><div class="ttdef"><b>Definition</b> pocketfft.h:2929</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1multi__iter_html_ac947f03b1cfcb63436a7e61ff020a88c"><div class="ttname"><a href="classpocketfft_1_1detail_1_1multi__iter.html#ac947f03b1cfcb63436a7e61ff020a88c">pocketfft::detail::multi_iter::stride_in</a></div><div class="ttdeci">ptrdiff_t stride_in() const</div><div class="ttdef"><b>Definition</b> pocketfft.h:2933</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1multi__iter_html_ad33360d4a8ab8e6d72efadc6f9cb5bfa"><div class="ttname"><a href="classpocketfft_1_1detail_1_1multi__iter.html#ad33360d4a8ab8e6d72efadc6f9cb5bfa">pocketfft::detail::multi_iter::iofs</a></div><div class="ttdeci">ptrdiff_t iofs(size_t i) const</div><div class="ttdef"><b>Definition</b> pocketfft.h:2927</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1ndarr_html"><div class="ttname"><a href="classpocketfft_1_1detail_1_1ndarr.html">pocketfft::detail::ndarr</a></div><div class="ttdef"><b>Definition</b> pocketfft.h:2853</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1ndarr_html_a2b2c4e205e8b5c32c9fe55dfd7b8c8d8"><div class="ttname"><a href="classpocketfft_1_1detail_1_1ndarr.html#a2b2c4e205e8b5c32c9fe55dfd7b8c8d8">pocketfft::detail::ndarr::operator[]</a></div><div class="ttdeci">T & operator[](ptrdiff_t ofs)</div><div class="ttdef"><b>Definition</b> pocketfft.h:2858</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1ndarr_html_a8f0037a172d96cb1ad915a5069175fa2"><div class="ttname"><a href="classpocketfft_1_1detail_1_1ndarr.html#a8f0037a172d96cb1ad915a5069175fa2">pocketfft::detail::ndarr::ndarr</a></div><div class="ttdeci">ndarr(void *data_, const shape_t &shape_, const stride_t &stride_)</div><div class="ttdef"><b>Definition</b> pocketfft.h:2855</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1pocketfft__c_html"><div class="ttname"><a href="classpocketfft_1_1detail_1_1pocketfft__c.html">pocketfft::detail::pocketfft_c</a></div><div class="ttdef"><b>Definition</b> pocketfft.h:2464</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1pocketfft__c_html_a1fd1a2f9b3ae5ee9f00b9ca6946eb16d"><div class="ttname"><a href="classpocketfft_1_1detail_1_1pocketfft__c.html#a1fd1a2f9b3ae5ee9f00b9ca6946eb16d">pocketfft::detail::pocketfft_c::length</a></div><div class="ttdeci">size_t length() const</div><div class="ttdef"><b>Definition</b> pocketfft.h:2493</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1pocketfft__c_html_a436afd63e8e130f97aff103ae964a45d"><div class="ttname"><a href="classpocketfft_1_1detail_1_1pocketfft__c.html#a436afd63e8e130f97aff103ae964a45d">pocketfft::detail::pocketfft_c::exec</a></div><div class="ttdeci">void exec(cmplx< T > c[], T0 fct, bool fwd) const</div><div class="ttdef"><b>Definition</b> pocketfft.h:2490</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1pocketfft__c_html_a44f46cf06f572b42ad8b53c7c6f38e4d"><div class="ttname"><a href="classpocketfft_1_1detail_1_1pocketfft__c.html#a44f46cf06f572b42ad8b53c7c6f38e4d">pocketfft::detail::pocketfft_c::pocketfft_c</a></div><div class="ttdeci">pocketfft_c(size_t length)</div><div class="ttdef"><b>Definition</b> pocketfft.h:2471</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1pocketfft__r_html"><div class="ttname"><a href="classpocketfft_1_1detail_1_1pocketfft__r.html">pocketfft::detail::pocketfft_r</a></div><div class="ttdef"><b>Definition</b> pocketfft.h:2501</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1pocketfft__r_html_a2815bc8aa04fa986834b02e502f98b33"><div class="ttname"><a href="classpocketfft_1_1detail_1_1pocketfft__r.html#a2815bc8aa04fa986834b02e502f98b33">pocketfft::detail::pocketfft_r::exec</a></div><div class="ttdeci">void exec(T c[], T0 fct, bool fwd) const</div><div class="ttdef"><b>Definition</b> pocketfft.h:2527</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1pocketfft__r_html_a60e9b3d1f9b20ec5b86d46b0398f6f7d"><div class="ttname"><a href="classpocketfft_1_1detail_1_1pocketfft__r.html#a60e9b3d1f9b20ec5b86d46b0398f6f7d">pocketfft::detail::pocketfft_r::pocketfft_r</a></div><div class="ttdeci">pocketfft_r(size_t length)</div><div class="ttdef"><b>Definition</b> pocketfft.h:2508</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1pocketfft__r_html_a83222fdbf81a7c6d560e0841cdfca8c6"><div class="ttname"><a href="classpocketfft_1_1detail_1_1pocketfft__r.html#a83222fdbf81a7c6d560e0841cdfca8c6">pocketfft::detail::pocketfft_r::length</a></div><div class="ttdeci">size_t length() const</div><div class="ttdef"><b>Definition</b> pocketfft.h:2530</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1rev__iter_html"><div class="ttname"><a href="classpocketfft_1_1detail_1_1rev__iter.html">pocketfft::detail::rev_iter</a></div><div class="ttdef"><b>Definition</b> pocketfft.h:2967</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1rev__iter_html_a143637135c441a4b9a2959c2370d8c63"><div class="ttname"><a href="classpocketfft_1_1detail_1_1rev__iter.html#a143637135c441a4b9a2959c2370d8c63">pocketfft::detail::rev_iter::remaining</a></div><div class="ttdeci">size_t remaining() const</div><div class="ttdef"><b>Definition</b> pocketfft.h:3026</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1rev__iter_html_a78c3b4ad19edf9d20cab40ad109e9dd1"><div class="ttname"><a href="classpocketfft_1_1detail_1_1rev__iter.html#a78c3b4ad19edf9d20cab40ad109e9dd1">pocketfft::detail::rev_iter::ofs</a></div><div class="ttdeci">ptrdiff_t ofs() const</div><div class="ttdef"><b>Definition</b> pocketfft.h:3024</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1rev__iter_html_a7f112afa76cb7a4c29cff217a6f5f5a9"><div class="ttname"><a href="classpocketfft_1_1detail_1_1rev__iter.html#a7f112afa76cb7a4c29cff217a6f5f5a9">pocketfft::detail::rev_iter::rev_ofs</a></div><div class="ttdeci">ptrdiff_t rev_ofs() const</div><div class="ttdef"><b>Definition</b> pocketfft.h:3025</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1rev__iter_html_ad1918c84ae963188afc7599629b29686"><div class="ttname"><a href="classpocketfft_1_1detail_1_1rev__iter.html#ad1918c84ae963188afc7599629b29686">pocketfft::detail::rev_iter::advance</a></div><div class="ttdeci">void advance()</div><div class="ttdef"><b>Definition</b> pocketfft.h:2993</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1rev__iter_html_af7b8c2f1534d3038ba2a3c6b9919e134"><div class="ttname"><a href="classpocketfft_1_1detail_1_1rev__iter.html#af7b8c2f1534d3038ba2a3c6b9919e134">pocketfft::detail::rev_iter::rev_iter</a></div><div class="ttdeci">rev_iter(const arr_info &arr_, const shape_t &axes)</div><div class="ttdef"><b>Definition</b> pocketfft.h:2979</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1rfftp_html"><div class="ttname"><a href="classpocketfft_1_1detail_1_1rfftp.html">pocketfft::detail::rfftp</a></div><div class="ttdef"><b>Definition</b> pocketfft.h:1553</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1rfftp_html_a073972f42bdd3617693be7be2cb5e0ac"><div class="ttname"><a href="classpocketfft_1_1detail_1_1rfftp.html#a073972f42bdd3617693be7be2cb5e0ac">pocketfft::detail::rfftp::exec</a></div><div class="ttdeci">void exec(T c[], T0 fct, bool r2hc) const</div><div class="ttdef"><b>Definition</b> pocketfft.h:2227</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1rfftp_html_a0c590f917b8e8afa3ff53ccff52e68c5"><div class="ttname"><a href="classpocketfft_1_1detail_1_1rfftp.html#a0c590f917b8e8afa3ff53ccff52e68c5">pocketfft::detail::rfftp::rfftp</a></div><div class="ttdeci">rfftp(size_t length_)</div><div class="ttdef"><b>Definition</b> pocketfft.h:2346</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1simple__iter_html"><div class="ttname"><a href="classpocketfft_1_1detail_1_1simple__iter.html">pocketfft::detail::simple_iter</a></div><div class="ttdef"><b>Definition</b> pocketfft.h:2939</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1simple__iter_html_a1e455c615825bebd5f1f62665027b398"><div class="ttname"><a href="classpocketfft_1_1detail_1_1simple__iter.html#a1e455c615825bebd5f1f62665027b398">pocketfft::detail::simple_iter::simple_iter</a></div><div class="ttdeci">simple_iter(const arr_info &arr_)</div><div class="ttdef"><b>Definition</b> pocketfft.h:2947</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1simple__iter_html_a73a9ecd3008d2bd35aaa00bf9fac074f"><div class="ttname"><a href="classpocketfft_1_1detail_1_1simple__iter.html#a73a9ecd3008d2bd35aaa00bf9fac074f">pocketfft::detail::simple_iter::advance</a></div><div class="ttdeci">void advance()</div><div class="ttdef"><b>Definition</b> pocketfft.h:2949</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1simple__iter_html_a9267d37f51a9a5aecc69293c7ed1b1f6"><div class="ttname"><a href="classpocketfft_1_1detail_1_1simple__iter.html#a9267d37f51a9a5aecc69293c7ed1b1f6">pocketfft::detail::simple_iter::remaining</a></div><div class="ttdeci">size_t remaining() const</div><div class="ttdef"><b>Definition</b> pocketfft.h:2963</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1simple__iter_html_ab59481ad9c8f04addb907c3ebb89f8fa"><div class="ttname"><a href="classpocketfft_1_1detail_1_1simple__iter.html#ab59481ad9c8f04addb907c3ebb89f8fa">pocketfft::detail::simple_iter::ofs</a></div><div class="ttdeci">ptrdiff_t ofs() const</div><div class="ttdef"><b>Definition</b> pocketfft.h:2962</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1sincos__2pibyn_html"><div class="ttname"><a href="classpocketfft_1_1detail_1_1sincos__2pibyn.html">pocketfft::detail::sincos_2pibyn</a></div><div class="ttdef"><b>Definition</b> pocketfft.h:296</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1sincos__2pibyn_html_a71b02f67c47b24adb296eafd2c7a3598"><div class="ttname"><a href="classpocketfft_1_1detail_1_1sincos__2pibyn.html#a71b02f67c47b24adb296eafd2c7a3598">pocketfft::detail::sincos_2pibyn::operator[]</a></div><div class="ttdeci">cmplx< T > operator[](size_t idx) const</div><div class="ttdef"><b>Definition</b> pocketfft.h:356</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1sincos__2pibyn_html_a88518f2182d854c557edacd4ab8cbc40"><div class="ttname"><a href="classpocketfft_1_1detail_1_1sincos__2pibyn.html#a88518f2182d854c557edacd4ab8cbc40">pocketfft::detail::sincos_2pibyn::sincos_2pibyn</a></div><div class="ttdeci">sincos_2pibyn(size_t n)</div><div class="ttdef"><b>Definition</b> pocketfft.h:337</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1threading_1_1concurrent__queue_html"><div class="ttname"><a href="classpocketfft_1_1detail_1_1threading_1_1concurrent__queue.html">pocketfft::detail::threading::concurrent_queue</a></div><div class="ttdef"><b>Definition</b> pocketfft.h:569</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1threading_1_1concurrent__queue_html_a1269e5da40c3f5145c895cee3641879a"><div class="ttname"><a href="classpocketfft_1_1detail_1_1threading_1_1concurrent__queue.html#a1269e5da40c3f5145c895cee3641879a">pocketfft::detail::threading::concurrent_queue::empty</a></div><div class="ttdeci">bool empty() const</div><div class="ttdef"><b>Definition</b> pocketfft.h:597</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1threading_1_1concurrent__queue_html_aa3807d46a126d229f9054c779105ea43"><div class="ttname"><a href="classpocketfft_1_1detail_1_1threading_1_1concurrent__queue.html#aa3807d46a126d229f9054c779105ea43">pocketfft::detail::threading::concurrent_queue::try_pop</a></div><div class="ttdeci">bool try_pop(T &val)</div><div class="ttdef"><b>Definition</b> pocketfft.h:584</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1threading_1_1concurrent__queue_html_ae8ebd6bc1b4d18e75bd046005e7dde7b"><div class="ttname"><a href="classpocketfft_1_1detail_1_1threading_1_1concurrent__queue.html#ae8ebd6bc1b4d18e75bd046005e7dde7b">pocketfft::detail::threading::concurrent_queue::push</a></div><div class="ttdeci">void push(T val)</div><div class="ttdef"><b>Definition</b> pocketfft.h:577</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1threading_1_1latch_html"><div class="ttname"><a href="classpocketfft_1_1detail_1_1threading_1_1latch.html">pocketfft::detail::threading::latch</a></div><div class="ttdef"><b>Definition</b> pocketfft.h:543</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1threading_1_1latch_html_a81d6597189b40410e35f3cd653fd1342"><div class="ttname"><a href="classpocketfft_1_1detail_1_1threading_1_1latch.html#a81d6597189b40410e35f3cd653fd1342">pocketfft::detail::threading::latch::count_down</a></div><div class="ttdeci">void count_down()</div><div class="ttdef"><b>Definition</b> pocketfft.h:552</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1threading_1_1latch_html_a9260c871bb927525f7f53caa23a19c15"><div class="ttname"><a href="classpocketfft_1_1detail_1_1threading_1_1latch.html#a9260c871bb927525f7f53caa23a19c15">pocketfft::detail::threading::latch::latch</a></div><div class="ttdeci">latch(size_t n)</div><div class="ttdef"><b>Definition</b> pocketfft.h:550</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1threading_1_1latch_html_ab41ecc5adb6187aa2682ca190fd920f3"><div class="ttname"><a href="classpocketfft_1_1detail_1_1threading_1_1latch.html#ab41ecc5adb6187aa2682ca190fd920f3">pocketfft::detail::threading::latch::is_ready</a></div><div class="ttdeci">bool is_ready()</div><div class="ttdef"><b>Definition</b> pocketfft.h:565</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1threading_1_1latch_html_af503189cc9247047fbdfc3ebf1daacc1"><div class="ttname"><a href="classpocketfft_1_1detail_1_1threading_1_1latch.html#af503189cc9247047fbdfc3ebf1daacc1">pocketfft::detail::threading::latch::wait</a></div><div class="ttdeci">void wait()</div><div class="ttdef"><b>Definition</b> pocketfft.h:560</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1threading_1_1thread__pool_html"><div class="ttname"><a href="classpocketfft_1_1detail_1_1threading_1_1thread__pool.html">pocketfft::detail::threading::thread_pool</a></div><div class="ttdef"><b>Definition</b> pocketfft.h:619</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1threading_1_1thread__pool_html_a0c7c29290fde806031c497f24c4ad411"><div class="ttname"><a href="classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#a0c7c29290fde806031c497f24c4ad411">pocketfft::detail::threading::thread_pool::shutdown</a></div><div class="ttdeci">void shutdown()</div><div class="ttdef"><b>Definition</b> pocketfft.h:753</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1threading_1_1thread__pool_html_a37a8121a99dd06a9d44b3e80ba0ea560"><div class="ttname"><a href="classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#a37a8121a99dd06a9d44b3e80ba0ea560">pocketfft::detail::threading::thread_pool::thread_pool</a></div><div class="ttdeci">thread_pool(size_t nthreads)</div><div class="ttdef"><b>Definition</b> pocketfft.h:720</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1threading_1_1thread__pool_html_a51d252df8d0cd060f15be8ba2bfe3288"><div class="ttname"><a href="classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#a51d252df8d0cd060f15be8ba2bfe3288">pocketfft::detail::threading::thread_pool::restart</a></div><div class="ttdeci">void restart()</div><div class="ttdef"><b>Definition</b> pocketfft.h:759</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1threading_1_1thread__pool_html_a68f71e79fa93379fcc5819e533b45472"><div class="ttname"><a href="classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#a68f71e79fa93379fcc5819e533b45472">pocketfft::detail::threading::thread_pool::~thread_pool</a></div><div class="ttdeci">~thread_pool()</div><div class="ttdef"><b>Definition</b> pocketfft.h:726</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1threading_1_1thread__pool_html_a8698d49e8f406cdb88006aac6a91f9a4"><div class="ttname"><a href="classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#a8698d49e8f406cdb88006aac6a91f9a4">pocketfft::detail::threading::thread_pool::submit</a></div><div class="ttdeci">void submit(std::function< void()> work)</div><div class="ttdef"><b>Definition</b> pocketfft.h:728</div></div>
|
|
<div class="ttc" id="aclasspocketfft_1_1detail_1_1threading_1_1thread__pool_html_aefaadaa60c0183b862ad96338177a5e0"><div class="ttname"><a href="classpocketfft_1_1detail_1_1threading_1_1thread__pool.html#aefaadaa60c0183b862ad96338177a5e0">pocketfft::detail::threading::thread_pool::thread_pool</a></div><div class="ttdeci">thread_pool()</div><div class="ttdef"><b>Definition</b> pocketfft.h:724</div></div>
|
|
<div class="ttc" id="anamespacemlx_1_1core_1_1allocator_html_a77f0a1215be242db6485612bcb273af5"><div class="ttname"><a href="namespacemlx_1_1core_1_1allocator.html#a77f0a1215be242db6485612bcb273af5">mlx::core::allocator::free</a></div><div class="ttdeci">void free(Buffer buffer)</div></div>
|
|
<div class="ttc" id="anamespacepocketfft_1_1detail_1_1threading_html_a2d5c0729f0b66cf061918baea4337d70"><div class="ttname"><a href="namespacepocketfft_1_1detail_1_1threading.html#a2d5c0729f0b66cf061918baea4337d70">pocketfft::detail::threading::max_threads</a></div><div class="ttdeci">static const size_t max_threads</div><div class="ttdef"><b>Definition</b> pocketfft.h:540</div></div>
|
|
<div class="ttc" id="anamespacepocketfft_1_1detail_1_1threading_html_a4fcf674db39f0e2c1c59d48491daed6e"><div class="ttname"><a href="namespacepocketfft_1_1detail_1_1threading.html#a4fcf674db39f0e2c1c59d48491daed6e">pocketfft::detail::threading::thread_map</a></div><div class="ttdeci">void thread_map(size_t nthreads, Func f)</div><div class="ttdoc">Map a function f over nthreads.</div><div class="ttdef"><b>Definition</b> pocketfft.h:786</div></div>
|
|
<div class="ttc" id="anamespacepocketfft_1_1detail_1_1threading_html_a7ec2b3f99232bd0f15f7b022c59d139a"><div class="ttname"><a href="namespacepocketfft_1_1detail_1_1threading.html#a7ec2b3f99232bd0f15f7b022c59d139a">pocketfft::detail::threading::get_pool</a></div><div class="ttdeci">thread_pool & get_pool()</div><div class="ttdef"><b>Definition</b> pocketfft.h:766</div></div>
|
|
<div class="ttc" id="anamespacepocketfft_1_1detail_1_1threading_html_aebe85d6273d92c7d3728e2c621ccc82b"><div class="ttname"><a href="namespacepocketfft_1_1detail_1_1threading.html#aebe85d6273d92c7d3728e2c621ccc82b">pocketfft::detail::threading::thread_id</a></div><div class="ttdeci">size_t & thread_id()</div><div class="ttdef"><b>Definition</b> pocketfft.h:530</div></div>
|
|
<div class="ttc" id="anamespacepocketfft_1_1detail_1_1threading_html_af5432c2e25aed679a73fe7b29534c833"><div class="ttname"><a href="namespacepocketfft_1_1detail_1_1threading.html#af5432c2e25aed679a73fe7b29534c833">pocketfft::detail::threading::num_threads</a></div><div class="ttdeci">size_t & num_threads()</div><div class="ttdef"><b>Definition</b> pocketfft.h:535</div></div>
|
|
<div class="ttc" id="anamespacepocketfft_1_1detail_html_a055a39b0a337ca12217717196eb92fed"><div class="ttname"><a href="namespacepocketfft_1_1detail.html#a055a39b0a337ca12217717196eb92fed">pocketfft::detail::general_r2c</a></div><div class="ttdeci">void general_r2c(const cndarr< T > &in, ndarr< cmplx< T > > &out, size_t axis, bool forward, T fct, size_t nthreads)</div><div class="ttdef"><b>Definition</b> pocketfft.h:3239</div></div>
|
|
<div class="ttc" id="anamespacepocketfft_1_1detail_html_a07745f4a069f811859308281b2982258"><div class="ttname"><a href="namespacepocketfft_1_1detail.html#a07745f4a069f811859308281b2982258">pocketfft::detail::sin</a></div><div class="ttdeci">T sin(T)=delete</div></div>
|
|
<div class="ttc" id="anamespacepocketfft_1_1detail_html_a1523a037300a8da05db210b802d9cb0e"><div class="ttname"><a href="namespacepocketfft_1_1detail.html#a1523a037300a8da05db210b802d9cb0e">pocketfft::detail::copy_output</a></div><div class="ttdeci">void copy_output(const multi_iter< vlen > &it, const cmplx< vtype_t< T > > *src, ndarr< cmplx< T > > &dst)</div><div class="ttdef"><b>Definition</b> pocketfft.h:3096</div></div>
|
|
<div class="ttc" id="anamespacepocketfft_1_1detail_html_a3edfb93aeed2f8258183d463ea291d62"><div class="ttname"><a href="namespacepocketfft_1_1detail.html#a3edfb93aeed2f8258183d463ea291d62">pocketfft::detail::vtype_t</a></div><div class="ttdeci">typename VTYPE< T >::type vtype_t</div><div class="ttdef"><b>Definition</b> pocketfft.h:3030</div></div>
|
|
<div class="ttc" id="anamespacepocketfft_1_1detail_html_a421aa74fbee775a96463246f72b144d6"><div class="ttname"><a href="namespacepocketfft_1_1detail.html#a421aa74fbee775a96463246f72b144d6">pocketfft::detail::add_vec_t</a></div><div class="ttdeci">typename add_vec< T >::type add_vec_t</div><div class="ttdef"><b>Definition</b> pocketfft.h:3123</div></div>
|
|
<div class="ttc" id="anamespacepocketfft_1_1detail_html_a499c1e8b7d79a5272af024f46c63ff9d"><div class="ttname"><a href="namespacepocketfft_1_1detail.html#a499c1e8b7d79a5272af024f46c63ff9d">pocketfft::detail::cos</a></div><div class="ttdeci">T cos(T)=delete</div></div>
|
|
<div class="ttc" id="anamespacepocketfft_1_1detail_html_a4db03cbcd9d43d9e0b0b9067713c80e9"><div class="ttname"><a href="namespacepocketfft_1_1detail.html#a4db03cbcd9d43d9e0b0b9067713c80e9">pocketfft::detail::alloc_tmp</a></div><div class="ttdeci">arr< char > alloc_tmp(const shape_t &shape, size_t axsize, size_t elemsize)</div><div class="ttdef"><b>Definition</b> pocketfft.h:3047</div></div>
|
|
<div class="ttc" id="anamespacepocketfft_1_1detail_html_a66d79051d502046a9b9f103e744dbad3"><div class="ttname"><a href="namespacepocketfft_1_1detail.html#a66d79051d502046a9b9f103e744dbad3">pocketfft::detail::conj</a></div><div class="ttdeci">cmplx< T > conj(const cmplx< T > &a)</div><div class="ttdef"><b>Definition</b> pocketfft.h:279</div></div>
|
|
<div class="ttc" id="anamespacepocketfft_1_1detail_html_a774f8b73f28259d4276bd188b540a3e3"><div class="ttname"><a href="namespacepocketfft_1_1detail.html#a774f8b73f28259d4276bd188b540a3e3">pocketfft::detail::sqrt</a></div><div class="ttdeci">T sqrt(T)=delete</div></div>
|
|
<div class="ttc" id="anamespacepocketfft_1_1detail_html_a885ee37fcf564a268a5c8ca9ea8603e1"><div class="ttname"><a href="namespacepocketfft_1_1detail.html#a885ee37fcf564a268a5c8ca9ea8603e1">pocketfft::detail::shape_t</a></div><div class="ttdeci">std::vector< size_t > shape_t</div><div class="ttdef"><b>Definition</b> pocketfft.h:102</div></div>
|
|
<div class="ttc" id="anamespacepocketfft_1_1detail_html_a8c6ec35091d5136d991577623133fcf2"><div class="ttname"><a href="namespacepocketfft_1_1detail.html#a8c6ec35091d5136d991577623133fcf2">pocketfft::detail::PM</a></div><div class="ttdeci">void PM(T &a, T &b, T c, T d)</div><div class="ttdef"><b>Definition</b> pocketfft.h:273</div></div>
|
|
<div class="ttc" id="anamespacepocketfft_1_1detail_html_a8da1f3d4a0b712a0285529f24187fe76"><div class="ttname"><a href="namespacepocketfft_1_1detail.html#a8da1f3d4a0b712a0285529f24187fe76">pocketfft::detail::special_mul</a></div><div class="ttdeci">void special_mul(const cmplx< T > &v1, const cmplx< T2 > &v2, cmplx< T > &res)</div><div class="ttdef"><b>Definition</b> pocketfft.h:281</div></div>
|
|
<div class="ttc" id="anamespacepocketfft_1_1detail_html_a928bad5278df636ee47402c0a75f64ef"><div class="ttname"><a href="namespacepocketfft_1_1detail.html#a928bad5278df636ee47402c0a75f64ef">pocketfft::detail::ROT90</a></div><div class="ttdeci">void ROT90(cmplx< T > &a)</div><div class="ttdef"><b>Definition</b> pocketfft.h:287</div></div>
|
|
<div class="ttc" id="anamespacepocketfft_1_1detail_html_aa3477781ef54788c9fa7755cb8ea75fd"><div class="ttname"><a href="namespacepocketfft_1_1detail.html#aa3477781ef54788c9fa7755cb8ea75fd">pocketfft::detail::PMINPLACE</a></div><div class="ttdeci">void PMINPLACE(T &a, T &b)</div><div class="ttdef"><b>Definition</b> pocketfft.h:275</div></div>
|
|
<div class="ttc" id="anamespacepocketfft_1_1detail_html_ab24cdb6118901f4d3c8df06ef0f8390b"><div class="ttname"><a href="namespacepocketfft_1_1detail.html#ab24cdb6118901f4d3c8df06ef0f8390b">pocketfft::detail::get_plan</a></div><div class="ttdeci">std::shared_ptr< T > get_plan(size_t length)</div><div class="ttdef"><b>Definition</b> pocketfft.h:2769</div></div>
|
|
<div class="ttc" id="anamespacepocketfft_1_1detail_html_ab47f52551920af5eb9f57fbbded0f4f0"><div class="ttname"><a href="namespacepocketfft_1_1detail.html#ab47f52551920af5eb9f57fbbded0f4f0">pocketfft::detail::general_nd</a></div><div class="ttdeci">void general_nd(const cndarr< T > &in, ndarr< T > &out, const shape_t &axes, T0 fct, size_t nthreads, const Exec &exec, const bool allow_inplace=true)</div><div class="ttdef"><b>Definition</b> pocketfft.h:3126</div></div>
|
|
<div class="ttc" id="anamespacepocketfft_1_1detail_html_ab6a43dc0cec4291e163e68a0875ac501"><div class="ttname"><a href="namespacepocketfft_1_1detail.html#ab6a43dc0cec4291e163e68a0875ac501">pocketfft::detail::ROTX90</a></div><div class="ttdeci">void ROTX90(cmplx< T > &a)</div><div class="ttdef"><b>Definition</b> pocketfft.h:289</div></div>
|
|
<div class="ttc" id="anamespacepocketfft_1_1detail_html_abac3fcc8ce83800d228774f64c28d4c3"><div class="ttname"><a href="namespacepocketfft_1_1detail.html#abac3fcc8ce83800d228774f64c28d4c3">pocketfft::detail::copy_hartley</a></div><div class="ttdeci">void copy_hartley(const multi_iter< vlen > &it, const vtype_t< T > *src, ndarr< T > &dst)</div><div class="ttdef"><b>Definition</b> pocketfft.h:3180</div></div>
|
|
<div class="ttc" id="anamespacepocketfft_1_1detail_html_ac8ee38e8d8bcda875c99eeaf567550fc"><div class="ttname"><a href="namespacepocketfft_1_1detail.html#ac8ee38e8d8bcda875c99eeaf567550fc">pocketfft::detail::general_c2r</a></div><div class="ttdeci">void general_c2r(const cndarr< cmplx< T > > &in, ndarr< T > &out, size_t axis, bool forward, T fct, size_t nthreads)</div><div class="ttdef"><b>Definition</b> pocketfft.h:3294</div></div>
|
|
<div class="ttc" id="anamespacepocketfft_1_1detail_html_ae397445c61400f47a8fe3f8e1b6d0b76"><div class="ttname"><a href="namespacepocketfft_1_1detail.html#ae397445c61400f47a8fe3f8e1b6d0b76">pocketfft::detail::aligned_alloc</a></div><div class="ttdeci">void * aligned_alloc(size_t align, size_t size)</div><div class="ttdef"><b>Definition</b> pocketfft.h:166</div></div>
|
|
<div class="ttc" id="anamespacepocketfft_1_1detail_html_aec7820e36a33e0a8bb83aa03b04b81e8"><div class="ttname"><a href="namespacepocketfft_1_1detail.html#aec7820e36a33e0a8bb83aa03b04b81e8">pocketfft::detail::aligned_dealloc</a></div><div class="ttdeci">void aligned_dealloc(void *ptr)</div><div class="ttdef"><b>Definition</b> pocketfft.h:176</div></div>
|
|
<div class="ttc" id="anamespacepocketfft_1_1detail_html_af5eedf3cdfc83c0a30807092c39a9ce2"><div class="ttname"><a href="namespacepocketfft_1_1detail.html#af5eedf3cdfc83c0a30807092c39a9ce2">pocketfft::detail::MPINPLACE</a></div><div class="ttdeci">void MPINPLACE(T &a, T &b)</div><div class="ttdef"><b>Definition</b> pocketfft.h:277</div></div>
|
|
<div class="ttc" id="anamespacepocketfft_1_1detail_html_afb987c919e9424a996d0fc8b3c23cc84"><div class="ttname"><a href="namespacepocketfft_1_1detail.html#afb987c919e9424a996d0fc8b3c23cc84">pocketfft::detail::stride_t</a></div><div class="ttdeci">std::vector< ptrdiff_t > stride_t</div><div class="ttdef"><b>Definition</b> pocketfft.h:103</div></div>
|
|
<div class="ttc" id="anamespacepocketfft_1_1detail_html_aff05be3064743c1143b19318ab12ad4a"><div class="ttname"><a href="namespacepocketfft_1_1detail.html#aff05be3064743c1143b19318ab12ad4a">pocketfft::detail::copy_input</a></div><div class="ttdeci">void copy_input(const multi_iter< vlen > &it, const cndarr< cmplx< T > > &src, cmplx< vtype_t< T > > *dst)</div><div class="ttdef"><b>Definition</b> pocketfft.h:3069</div></div>
|
|
<div class="ttc" id="anamespacepocketfft_html"><div class="ttname"><a href="namespacepocketfft.html">pocketfft</a></div><div class="ttdef"><b>Definition</b> pocketfft.h:91</div></div>
|
|
<div class="ttc" id="apocketfft_8h_html_a078bc2bd38ab0ffb15b981878c9de03c"><div class="ttname"><a href="pocketfft_8h.html#a078bc2bd38ab0ffb15b981878c9de03c">POCKETFFT_PARTSTEP5a</a></div><div class="ttdeci">#define POCKETFFT_PARTSTEP5a(u1, u2, twar, twbr, twai, twbi)</div><div class="ttdef"><b>Definition</b> pocketfft.h:982</div></div>
|
|
<div class="ttc" id="apocketfft_8h_html_a1793d0d00f2e13101eb5ad0719c40817"><div class="ttname"><a href="pocketfft_8h.html#a1793d0d00f2e13101eb5ad0719c40817">POCKETFFT_PARTSTEP11</a></div><div class="ttdeci">#define POCKETFFT_PARTSTEP11(u1, u2, x1, x2, x3, x4, x5, y1, y2, y3, y4, y5)</div><div class="ttdef"><b>Definition</b> pocketfft.h:1245</div></div>
|
|
<div class="ttc" id="apocketfft_8h_html_a2b27f6e1f0ee131765186870517255c6"><div class="ttname"><a href="pocketfft_8h.html#a2b27f6e1f0ee131765186870517255c6">POCKETFFT_PARTSTEP7a</a></div><div class="ttdeci">#define POCKETFFT_PARTSTEP7a(u1, u2, x1, x2, x3, y1, y2, y3)</div><div class="ttdef"><b>Definition</b> pocketfft.h:1063</div></div>
|
|
<div class="ttc" id="apocketfft_8h_html_a41e646e5535a3a1c6e0d0e67122382f5"><div class="ttname"><a href="pocketfft_8h.html#a41e646e5535a3a1c6e0d0e67122382f5">POCKETFFT_PARTSTEP3b</a></div><div class="ttdeci">#define POCKETFFT_PARTSTEP3b(u1, u2, twr, twi)</div><div class="ttdef"><b>Definition</b> pocketfft.h:882</div></div>
|
|
<div class="ttc" id="apocketfft_8h_html_a536d2ea61479d4b074bf52ce09fdbc3a"><div class="ttname"><a href="pocketfft_8h.html#a536d2ea61479d4b074bf52ce09fdbc3a">POCKETFFT_PREP11</a></div><div class="ttdeci">#define POCKETFFT_PREP11(idx)</div><div class="ttdef"><b>Definition</b> pocketfft.h:1225</div></div>
|
|
<div class="ttc" id="apocketfft_8h_html_a7020984e0ca1d6e565629ca6e7c1a7e0"><div class="ttname"><a href="pocketfft_8h.html#a7020984e0ca1d6e565629ca6e7c1a7e0">POCKETFFT_NOINLINE</a></div><div class="ttdeci">#define POCKETFFT_NOINLINE</div><div class="ttdef"><b>Definition</b> pocketfft.h:87</div></div>
|
|
<div class="ttc" id="apocketfft_8h_html_a73077c26d2a82754db2a9c48bc0e11a6"><div class="ttname"><a href="pocketfft_8h.html#a73077c26d2a82754db2a9c48bc0e11a6">POCKETFFT_PREP5</a></div><div class="ttdeci">#define POCKETFFT_PREP5(idx)</div><div class="ttdef"><b>Definition</b> pocketfft.h:975</div></div>
|
|
<div class="ttc" id="apocketfft_8h_html_a9e604bcf20603d70b62b233d3f306714"><div class="ttname"><a href="pocketfft_8h.html#a9e604bcf20603d70b62b233d3f306714">POCKETFFT_CACHE_SIZE</a></div><div class="ttdeci">#define POCKETFFT_CACHE_SIZE</div><div class="ttdef"><b>Definition</b> pocketfft.h:51</div></div>
|
|
<div class="ttc" id="apocketfft_8h_html_ab8a5da142555e059c5e9c618f75b46fa"><div class="ttname"><a href="pocketfft_8h.html#ab8a5da142555e059c5e9c618f75b46fa">POCKETFFT_PARTSTEP5b</a></div><div class="ttdeci">#define POCKETFFT_PARTSTEP5b(u1, u2, twar, twbr, twai, twbi)</div><div class="ttdef"><b>Definition</b> pocketfft.h:992</div></div>
|
|
<div class="ttc" id="apocketfft_8h_html_abbe177c4872821b32d76d5ce08d6ce82"><div class="ttname"><a href="pocketfft_8h.html#abbe177c4872821b32d76d5ce08d6ce82">POCKETFFT_RESTRICT</a></div><div class="ttdeci">#define POCKETFFT_RESTRICT</div><div class="ttdef"><b>Definition</b> pocketfft.h:88</div></div>
|
|
<div class="ttc" id="apocketfft_8h_html_ac112b26e5130636ac1d91c2f0af45e0b"><div class="ttname"><a href="pocketfft_8h.html#ac112b26e5130636ac1d91c2f0af45e0b">POCKETFFT_PARTSTEP3a</a></div><div class="ttdeci">#define POCKETFFT_PARTSTEP3a(u1, u2, twr, twi)</div><div class="ttdef"><b>Definition</b> pocketfft.h:876</div></div>
|
|
<div class="ttc" id="apocketfft_8h_html_ac35e1aa5ae84d655256b7a0afd9051c2"><div class="ttname"><a href="pocketfft_8h.html#ac35e1aa5ae84d655256b7a0afd9051c2">POCKETFFT_PARTSTEP11a</a></div><div class="ttdeci">#define POCKETFFT_PARTSTEP11a(u1, u2, x1, x2, x3, x4, x5, y1, y2, y3, y4, y5)</div><div class="ttdef"><b>Definition</b> pocketfft.h:1243</div></div>
|
|
<div class="ttc" id="apocketfft_8h_html_acffdf2e1ab84f36a7a097e1b8b87a9f9"><div class="ttname"><a href="pocketfft_8h.html#acffdf2e1ab84f36a7a097e1b8b87a9f9">POCKETFFT_REARRANGE</a></div><div class="ttdeci">#define POCKETFFT_REARRANGE(rx, ix, ry, iy)</div><div class="ttdef"><b>Definition</b> pocketfft.h:1604</div></div>
|
|
<div class="ttc" id="apocketfft_8h_html_ae2fd9d433c417f0768fe1b58145b2e59"><div class="ttname"><a href="pocketfft_8h.html#ae2fd9d433c417f0768fe1b58145b2e59">POCKETFFT_PREP3</a></div><div class="ttdeci">#define POCKETFFT_PREP3(idx)</div><div class="ttdef"><b>Definition</b> pocketfft.h:872</div></div>
|
|
<div class="ttc" id="apocketfft_8h_html_ae7c4d0cda5b3824f84eac54addabd6ec"><div class="ttname"><a href="pocketfft_8h.html#ae7c4d0cda5b3824f84eac54addabd6ec">POCKETFFT_PREP7</a></div><div class="ttdeci">#define POCKETFFT_PREP7(idx)</div><div class="ttdef"><b>Definition</b> pocketfft.h:1046</div></div>
|
|
<div class="ttc" id="apocketfft_8h_html_af7de1f82911a973d8446cf3f40ff3044"><div class="ttname"><a href="pocketfft_8h.html#af7de1f82911a973d8446cf3f40ff3044">POCKETFFT_PARTSTEP7</a></div><div class="ttdeci">#define POCKETFFT_PARTSTEP7(u1, u2, x1, x2, x3, y1, y2, y3)</div><div class="ttdef"><b>Definition</b> pocketfft.h:1065</div></div>
|
|
<div class="ttc" id="astructpocketfft_1_1detail_1_1_exec_c2_c_html"><div class="ttname"><a href="structpocketfft_1_1detail_1_1_exec_c2_c.html">pocketfft::detail::ExecC2C</a></div><div class="ttdef"><b>Definition</b> pocketfft.h:3167</div></div>
|
|
<div class="ttc" id="astructpocketfft_1_1detail_1_1_exec_c2_c_html_a4fd637f1a6d335826789af28ac089ecb"><div class="ttname"><a href="structpocketfft_1_1detail_1_1_exec_c2_c.html#a4fd637f1a6d335826789af28ac089ecb">pocketfft::detail::ExecC2C::operator()</a></div><div class="ttdeci">void operator()(const multi_iter< vlen > &it, const cndarr< cmplx< T0 > > &in, ndarr< cmplx< T0 > > &out, T *buf, const pocketfft_c< T0 > &plan, T0 fct) const</div><div class="ttdef"><b>Definition</b> pocketfft.h:3170</div></div>
|
|
<div class="ttc" id="astructpocketfft_1_1detail_1_1_exec_c2_c_html_a63e27292b327597674deede9debe1c43"><div class="ttname"><a href="structpocketfft_1_1detail_1_1_exec_c2_c.html#a63e27292b327597674deede9debe1c43">pocketfft::detail::ExecC2C::forward</a></div><div class="ttdeci">bool forward</div><div class="ttdef"><b>Definition</b> pocketfft.h:3168</div></div>
|
|
<div class="ttc" id="astructpocketfft_1_1detail_1_1_exec_dcst_html"><div class="ttname"><a href="structpocketfft_1_1detail_1_1_exec_dcst.html">pocketfft::detail::ExecDcst</a></div><div class="ttdef"><b>Definition</b> pocketfft.h:3224</div></div>
|
|
<div class="ttc" id="astructpocketfft_1_1detail_1_1_exec_dcst_html_a185023fc1e386cc8f233b79c49c1fd8a"><div class="ttname"><a href="structpocketfft_1_1detail_1_1_exec_dcst.html#a185023fc1e386cc8f233b79c49c1fd8a">pocketfft::detail::ExecDcst::cosine</a></div><div class="ttdeci">bool cosine</div><div class="ttdef"><b>Definition</b> pocketfft.h:3227</div></div>
|
|
<div class="ttc" id="astructpocketfft_1_1detail_1_1_exec_dcst_html_a67f4f56e3574c491695f8cb8a1e983d8"><div class="ttname"><a href="structpocketfft_1_1detail_1_1_exec_dcst.html#a67f4f56e3574c491695f8cb8a1e983d8">pocketfft::detail::ExecDcst::operator()</a></div><div class="ttdeci">void operator()(const multi_iter< vlen > &it, const cndarr< T0 > &in, ndarr< T0 > &out, T *buf, const Tplan &plan, T0 fct) const</div><div class="ttdef"><b>Definition</b> pocketfft.h:3230</div></div>
|
|
<div class="ttc" id="astructpocketfft_1_1detail_1_1_exec_dcst_html_a9b170cbd74a9c6f45ac014ce349219ea"><div class="ttname"><a href="structpocketfft_1_1detail_1_1_exec_dcst.html#a9b170cbd74a9c6f45ac014ce349219ea">pocketfft::detail::ExecDcst::type</a></div><div class="ttdeci">int type</div><div class="ttdef"><b>Definition</b> pocketfft.h:3226</div></div>
|
|
<div class="ttc" id="astructpocketfft_1_1detail_1_1_exec_dcst_html_aea17551a49acaca5e7808dc181d38b7f"><div class="ttname"><a href="structpocketfft_1_1detail_1_1_exec_dcst.html#aea17551a49acaca5e7808dc181d38b7f">pocketfft::detail::ExecDcst::ortho</a></div><div class="ttdeci">bool ortho</div><div class="ttdef"><b>Definition</b> pocketfft.h:3225</div></div>
|
|
<div class="ttc" id="astructpocketfft_1_1detail_1_1_exec_hartley_html"><div class="ttname"><a href="structpocketfft_1_1detail_1_1_exec_hartley.html">pocketfft::detail::ExecHartley</a></div><div class="ttdef"><b>Definition</b> pocketfft.h:3212</div></div>
|
|
<div class="ttc" id="astructpocketfft_1_1detail_1_1_exec_hartley_html_a67c98b38d12440781053552b9a33bba1"><div class="ttname"><a href="structpocketfft_1_1detail_1_1_exec_hartley.html#a67c98b38d12440781053552b9a33bba1">pocketfft::detail::ExecHartley::operator()</a></div><div class="ttdeci">void operator()(const multi_iter< vlen > &it, const cndarr< T0 > &in, ndarr< T0 > &out, T *buf, const pocketfft_r< T0 > &plan, T0 fct) const</div><div class="ttdef"><b>Definition</b> pocketfft.h:3213</div></div>
|
|
<div class="ttc" id="astructpocketfft_1_1detail_1_1_exec_r2_r_html"><div class="ttname"><a href="structpocketfft_1_1detail_1_1_exec_r2_r.html">pocketfft::detail::ExecR2R</a></div><div class="ttdef"><b>Definition</b> pocketfft.h:3367</div></div>
|
|
<div class="ttc" id="astructpocketfft_1_1detail_1_1_exec_r2_r_html_a5ec66ebb2ccd079f62b068ddd1fc7bdf"><div class="ttname"><a href="structpocketfft_1_1detail_1_1_exec_r2_r.html#a5ec66ebb2ccd079f62b068ddd1fc7bdf">pocketfft::detail::ExecR2R::forward</a></div><div class="ttdeci">bool forward</div><div class="ttdef"><b>Definition</b> pocketfft.h:3368</div></div>
|
|
<div class="ttc" id="astructpocketfft_1_1detail_1_1_v_l_e_n_html"><div class="ttname"><a href="structpocketfft_1_1detail_1_1_v_l_e_n.html">pocketfft::detail::VLEN</a></div><div class="ttdef"><b>Definition</b> pocketfft.h:129</div></div>
|
|
<div class="ttc" id="astructpocketfft_1_1detail_1_1_v_l_e_n_html_ab1fdc340dedde723e636746c828a4534"><div class="ttname"><a href="structpocketfft_1_1detail_1_1_v_l_e_n.html#ab1fdc340dedde723e636746c828a4534">pocketfft::detail::VLEN::val</a></div><div class="ttdeci">static constexpr size_t val</div><div class="ttdef"><b>Definition</b> pocketfft.h:129</div></div>
|
|
<div class="ttc" id="astructpocketfft_1_1detail_1_1_v_t_y_p_e_html"><div class="ttname"><a href="structpocketfft_1_1detail_1_1_v_t_y_p_e.html">pocketfft::detail::VTYPE</a></div><div class="ttdef"><b>Definition</b> pocketfft.h:3029</div></div>
|
|
<div class="ttc" id="astructpocketfft_1_1detail_1_1add__vec_html"><div class="ttname"><a href="structpocketfft_1_1detail_1_1add__vec.html">pocketfft::detail::add_vec</a></div><div class="ttdef"><b>Definition</b> pocketfft.h:3120</div></div>
|
|
<div class="ttc" id="astructpocketfft_1_1detail_1_1cmplx_html"><div class="ttname"><a href="structpocketfft_1_1detail_1_1cmplx.html">pocketfft::detail::cmplx</a></div><div class="ttdef"><b>Definition</b> pocketfft.h:232</div></div>
|
|
<div class="ttc" id="astructpocketfft_1_1detail_1_1cmplx_html_a05491b4f1f22ca0bc49012f6a1c1710a"><div class="ttname"><a href="structpocketfft_1_1detail_1_1cmplx.html#a05491b4f1f22ca0bc49012f6a1c1710a">pocketfft::detail::cmplx::cmplx</a></div><div class="ttdeci">cmplx(T r_, T i_)</div><div class="ttdef"><b>Definition</b> pocketfft.h:235</div></div>
|
|
<div class="ttc" id="astructpocketfft_1_1detail_1_1cmplx_html_a12441ff423274bd1b54245933d69ad7e"><div class="ttname"><a href="structpocketfft_1_1detail_1_1cmplx.html#a12441ff423274bd1b54245933d69ad7e">pocketfft::detail::cmplx::operator-=</a></div><div class="ttdeci">cmplx & operator-=(const cmplx< T2 > &other)</div><div class="ttdef"><b>Definition</b> pocketfft.h:251</div></div>
|
|
<div class="ttc" id="astructpocketfft_1_1detail_1_1cmplx_html_a26bf3d709a58f06228e502af6db8e5ac"><div class="ttname"><a href="structpocketfft_1_1detail_1_1cmplx.html#a26bf3d709a58f06228e502af6db8e5ac">pocketfft::detail::cmplx::operator*</a></div><div class="ttdeci">auto operator*(const T2 &other) const -> cmplx< decltype(r *other)></div><div class="ttdef"><b>Definition</b> pocketfft.h:253</div></div>
|
|
<div class="ttc" id="astructpocketfft_1_1detail_1_1cmplx_html_a2e79f5c73c1d926361ad126cf57c8874"><div class="ttname"><a href="structpocketfft_1_1detail_1_1cmplx.html#a2e79f5c73c1d926361ad126cf57c8874">pocketfft::detail::cmplx::special_mul</a></div><div class="ttdeci">auto special_mul(const cmplx< T2 > &other) const -> cmplx< decltype(r+other.r)></div><div class="ttdef"><b>Definition</b> pocketfft.h:265</div></div>
|
|
<div class="ttc" id="astructpocketfft_1_1detail_1_1cmplx_html_a35d2dce1b7de5f37d7029e639bc7f23d"><div class="ttname"><a href="structpocketfft_1_1detail_1_1cmplx.html#a35d2dce1b7de5f37d7029e639bc7f23d">pocketfft::detail::cmplx::i</a></div><div class="ttdeci">T i</div><div class="ttdef"><b>Definition</b> pocketfft.h:233</div></div>
|
|
<div class="ttc" id="astructpocketfft_1_1detail_1_1cmplx_html_a447d26b2e07f6e45f29d865e906c0a98"><div class="ttname"><a href="structpocketfft_1_1detail_1_1cmplx.html#a447d26b2e07f6e45f29d865e906c0a98">pocketfft::detail::cmplx::Set</a></div><div class="ttdeci">void Set(T r_)</div><div class="ttdef"><b>Definition</b> pocketfft.h:237</div></div>
|
|
<div class="ttc" id="astructpocketfft_1_1detail_1_1cmplx_html_a460da5db36d1c72fb1ed3496fd3abde4"><div class="ttname"><a href="structpocketfft_1_1detail_1_1cmplx.html#a460da5db36d1c72fb1ed3496fd3abde4">pocketfft::detail::cmplx::operator-</a></div><div class="ttdeci">auto operator-(const cmplx< T2 > &other) const -> cmplx< decltype(r+other.r)></div><div class="ttdef"><b>Definition</b> pocketfft.h:259</div></div>
|
|
<div class="ttc" id="astructpocketfft_1_1detail_1_1cmplx_html_a5b1ce506f1023f5254025ac81b831a2c"><div class="ttname"><a href="structpocketfft_1_1detail_1_1cmplx.html#a5b1ce506f1023f5254025ac81b831a2c">pocketfft::detail::cmplx::cmplx</a></div><div class="ttdeci">cmplx()</div><div class="ttdef"><b>Definition</b> pocketfft.h:234</div></div>
|
|
<div class="ttc" id="astructpocketfft_1_1detail_1_1cmplx_html_a647fece372b64b13c4a7e5877d09a807"><div class="ttname"><a href="structpocketfft_1_1detail_1_1cmplx.html#a647fece372b64b13c4a7e5877d09a807">pocketfft::detail::cmplx::Set</a></div><div class="ttdeci">void Set(T r_, T i_)</div><div class="ttdef"><b>Definition</b> pocketfft.h:236</div></div>
|
|
<div class="ttc" id="astructpocketfft_1_1detail_1_1cmplx_html_a683fd490182c9189fa2c05b1823edd93"><div class="ttname"><a href="structpocketfft_1_1detail_1_1cmplx.html#a683fd490182c9189fa2c05b1823edd93">pocketfft::detail::cmplx::operator*=</a></div><div class="ttdeci">cmplx & operator*=(T2 other)</div><div class="ttdef"><b>Definition</b> pocketfft.h:240</div></div>
|
|
<div class="ttc" id="astructpocketfft_1_1detail_1_1cmplx_html_a76447ef141c8732d57421749fc81b236"><div class="ttname"><a href="structpocketfft_1_1detail_1_1cmplx.html#a76447ef141c8732d57421749fc81b236">pocketfft::detail::cmplx::operator+</a></div><div class="ttdeci">auto operator+(const cmplx< T2 > &other) const -> cmplx< decltype(r+other.r)></div><div class="ttdef"><b>Definition</b> pocketfft.h:256</div></div>
|
|
<div class="ttc" id="astructpocketfft_1_1detail_1_1cmplx_html_ad4e69dcd89bdb7764c9c5807168f911e"><div class="ttname"><a href="structpocketfft_1_1detail_1_1cmplx.html#ad4e69dcd89bdb7764c9c5807168f911e">pocketfft::detail::cmplx::operator+=</a></div><div class="ttdeci">cmplx & operator+=(const cmplx &other)</div><div class="ttdef"><b>Definition</b> pocketfft.h:238</div></div>
|
|
<div class="ttc" id="astructpocketfft_1_1detail_1_1cmplx_html_afc51cdf222d77690953a8cb8ce3ee692"><div class="ttname"><a href="structpocketfft_1_1detail_1_1cmplx.html#afc51cdf222d77690953a8cb8ce3ee692">pocketfft::detail::cmplx::r</a></div><div class="ttdeci">T r</div><div class="ttdef"><b>Definition</b> pocketfft.h:233</div></div>
|
|
<div class="ttc" id="astructpocketfft_1_1detail_1_1threading_1_1aligned__allocator_html"><div class="ttname"><a href="structpocketfft_1_1detail_1_1threading_1_1aligned__allocator.html">pocketfft::detail::threading::aligned_allocator</a></div><div class="ttdef"><b>Definition</b> pocketfft.h:602</div></div>
|
|
<div class="ttc" id="astructpocketfft_1_1detail_1_1threading_1_1aligned__allocator_html_a0c390851ec37c5cdc5c1e7c6232a0b94"><div class="ttname"><a href="structpocketfft_1_1detail_1_1threading_1_1aligned__allocator.html#a0c390851ec37c5cdc5c1e7c6232a0b94">pocketfft::detail::threading::aligned_allocator::aligned_allocator</a></div><div class="ttdeci">aligned_allocator()=default</div></div>
|
|
<div class="ttc" id="astructpocketfft_1_1detail_1_1threading_1_1aligned__allocator_html_a2a99b8e296d26b255e9937ba5f30e76f"><div class="ttname"><a href="structpocketfft_1_1detail_1_1threading_1_1aligned__allocator.html#a2a99b8e296d26b255e9937ba5f30e76f">pocketfft::detail::threading::aligned_allocator::deallocate</a></div><div class="ttdeci">void deallocate(T *p, size_t)</div><div class="ttdef"><b>Definition</b> pocketfft.h:614</div></div>
|
|
<div class="ttc" id="astructpocketfft_1_1detail_1_1threading_1_1aligned__allocator_html_a4f785747b898980756af9e5515363826"><div class="ttname"><a href="structpocketfft_1_1detail_1_1threading_1_1aligned__allocator.html#a4f785747b898980756af9e5515363826">pocketfft::detail::threading::aligned_allocator::allocate</a></div><div class="ttdeci">T * allocate(size_t n)</div><div class="ttdef"><b>Definition</b> pocketfft.h:608</div></div>
|
|
<div class="ttc" id="astructpocketfft_1_1detail_1_1threading_1_1aligned__allocator_html_a57c07047ac09c6cf48a269429de2b0fb"><div class="ttname"><a href="structpocketfft_1_1detail_1_1threading_1_1aligned__allocator.html#a57c07047ac09c6cf48a269429de2b0fb">pocketfft::detail::threading::aligned_allocator::aligned_allocator</a></div><div class="ttdeci">aligned_allocator(const aligned_allocator< U > &)</div><div class="ttdef"><b>Definition</b> pocketfft.h:605</div></div>
|
|
<div class="ttc" id="astructpocketfft_1_1detail_1_1util_html"><div class="ttname"><a href="structpocketfft_1_1detail_1_1util.html">pocketfft::detail::util</a></div><div class="ttdef"><b>Definition</b> pocketfft.h:370</div></div>
|
|
<div class="ttc" id="astructpocketfft_1_1detail_1_1util_html_a0e9b92280f25063d1ebef942dd0d233c"><div class="ttname"><a href="structpocketfft_1_1detail_1_1util.html#a0e9b92280f25063d1ebef942dd0d233c">pocketfft::detail::util::largest_prime_factor</a></div><div class="ttdeci">static size_t largest_prime_factor(size_t n)</div><div class="ttdef"><b>Definition</b> pocketfft.h:371</div></div>
|
|
<div class="ttc" id="astructpocketfft_1_1detail_1_1util_html_a173da7d5929ded86fffcebcfdc5086aa"><div class="ttname"><a href="structpocketfft_1_1detail_1_1util.html#a173da7d5929ded86fffcebcfdc5086aa">pocketfft::detail::util::good_size_real</a></div><div class="ttdeci">static size_t good_size_real(size_t n)</div><div class="ttdef"><b>Definition</b> pocketfft.h:430</div></div>
|
|
<div class="ttc" id="astructpocketfft_1_1detail_1_1util_html_a280945ba7af964102b1d0d1d83469130"><div class="ttname"><a href="structpocketfft_1_1detail_1_1util.html#a280945ba7af964102b1d0d1d83469130">pocketfft::detail::util::sanity_check</a></div><div class="ttdeci">static void sanity_check(const shape_t &shape, const stride_t &stride_in, const stride_t &stride_out, bool inplace, const shape_t &axes)</div><div class="ttdef"><b>Definition</b> pocketfft.h:475</div></div>
|
|
<div class="ttc" id="astructpocketfft_1_1detail_1_1util_html_a33acae07b20b28fe4658bc338fce1b89"><div class="ttname"><a href="structpocketfft_1_1detail_1_1util.html#a33acae07b20b28fe4658bc338fce1b89">pocketfft::detail::util::prod</a></div><div class="ttdeci">static size_t prod(const shape_t &shape)</div><div class="ttdef"><b>Definition</b> pocketfft.h:456</div></div>
|
|
<div class="ttc" id="astructpocketfft_1_1detail_1_1util_html_a3b012d5a19215bcd32cf6e228556fa87"><div class="ttname"><a href="structpocketfft_1_1detail_1_1util.html#a3b012d5a19215bcd32cf6e228556fa87">pocketfft::detail::util::thread_count</a></div><div class="ttdeci">static size_t thread_count(size_t nthreads, const shape_t &shape, size_t axis, size_t vlen)</div><div class="ttdef"><b>Definition</b> pocketfft.h:502</div></div>
|
|
<div class="ttc" id="astructpocketfft_1_1detail_1_1util_html_a3d2f5f00d4fed5c09bd1509ffa9a44cd"><div class="ttname"><a href="structpocketfft_1_1detail_1_1util.html#a3d2f5f00d4fed5c09bd1509ffa9a44cd">pocketfft::detail::util::sanity_check</a></div><div class="ttdeci">static void sanity_check(const shape_t &shape, const stride_t &stride_in, const stride_t &stride_out, bool inplace)</div><div class="ttdef"><b>Definition</b> pocketfft.h:464</div></div>
|
|
<div class="ttc" id="astructpocketfft_1_1detail_1_1util_html_a670e044dadbbf680537e1fb11bc8b724"><div class="ttname"><a href="structpocketfft_1_1detail_1_1util.html#a670e044dadbbf680537e1fb11bc8b724">pocketfft::detail::util::sanity_check</a></div><div class="ttdeci">static void sanity_check(const shape_t &shape, const stride_t &stride_in, const stride_t &stride_out, bool inplace, size_t axis)</div><div class="ttdef"><b>Definition</b> pocketfft.h:489</div></div>
|
|
<div class="ttc" id="astructpocketfft_1_1detail_1_1util_html_a758e00d242a1b7eda8f9f0c21f35c624"><div class="ttname"><a href="structpocketfft_1_1detail_1_1util.html#a758e00d242a1b7eda8f9f0c21f35c624">pocketfft::detail::util::good_size_cmplx</a></div><div class="ttdeci">static size_t good_size_cmplx(size_t n)</div><div class="ttdef"><b>Definition</b> pocketfft.h:401</div></div>
|
|
<div class="ttc" id="astructpocketfft_1_1detail_1_1util_html_ad3d874bc3fb0048df2270779a15d4bd0"><div class="ttname"><a href="structpocketfft_1_1detail_1_1util.html#ad3d874bc3fb0048df2270779a15d4bd0">pocketfft::detail::util::cost_guess</a></div><div class="ttdeci">static double cost_guess(size_t n)</div><div class="ttdef"><b>Definition</b> pocketfft.h:383</div></div>
|
|
<div class="ttc" id="atypes_2bf16_8h_html_af900396d7b72ff2a7002e8befe8cf8f1"><div class="ttname"><a href="types_2bf16_8h.html#af900396d7b72ff2a7002e8befe8cf8f1">f</a></div><div class="ttdeci">float f</div><div class="ttdef"><b>Definition</b> bf16.h:16</div></div>
|
|
</div><!-- fragment --></div><!-- contents -->
|
|
<!-- start footer part -->
|
|
<hr class="footer"/><address class="footer"><small>
|
|
Generated by <a href="https://www.doxygen.org/index.html"><img class="footer" src="doxygen.svg" width="104" height="31" alt="doxygen"/></a> 1.10.0
|
|
</small></address>
|
|
</body>
|
|
</html>
|