|
| 1 | +<!doctype html> |
| 2 | +<html lang="en"> |
| 3 | +<head> |
| 4 | +<meta charset="utf-8"> |
| 5 | +<meta name="viewport" content="width=device-width, initial-scale=1, minimum-scale=1"> |
| 6 | +<meta name="generator" content="pdoc3 0.11.6"> |
| 7 | +<title>underworld3.ckdtree API documentation</title> |
| 8 | +<meta name="description" content=""> |
| 9 | +<link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/10up-sanitize.css/13.0.0/sanitize.min.css" integrity="sha512-y1dtMcuvtTMJc1yPgEqF0ZjQbhnc/bFhyvIyVNb9Zk5mIGtqVaAB1Ttl28su8AvFMOY0EwRbAe+HCLqj6W7/KA==" crossorigin> |
| 10 | +<link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/10up-sanitize.css/13.0.0/typography.min.css" integrity="sha512-Y1DYSb995BAfxobCkKepB1BqJJTPrOp3zPL74AWFugHHmmdcvO+C48WLrUOlhGMc0QG7AE3f7gmvvcrmX2fDoA==" crossorigin> |
| 11 | +<link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/highlight.js/11.9.0/styles/default.min.css" crossorigin> |
| 12 | +<style>:root{--highlight-color:#fe9}.flex{display:flex !important}body{line-height:1.5em}#content{padding:20px}#sidebar{padding:1.5em;overflow:hidden}#sidebar > *:last-child{margin-bottom:2cm}.http-server-breadcrumbs{font-size:130%;margin:0 0 15px 0}#footer{font-size:.75em;padding:5px 30px;border-top:1px solid #ddd;text-align:right}#footer p{margin:0 0 0 1em;display:inline-block}#footer p:last-child{margin-right:30px}h1,h2,h3,h4,h5{font-weight:300}h1{font-size:2.5em;line-height:1.1em}h2{font-size:1.75em;margin:2em 0 .50em 0}h3{font-size:1.4em;margin:1.6em 0 .7em 0}h4{margin:0;font-size:105%}h1:target,h2:target,h3:target,h4:target,h5:target,h6:target{background:var(--highlight-color);padding:.2em 0}a{color:#058;text-decoration:none;transition:color .2s ease-in-out}a:visited{color:#503}a:hover{color:#b62}.title code{font-weight:bold}h2[id^="header-"]{margin-top:2em}.ident{color:#900;font-weight:bold}pre code{font-size:.8em;line-height:1.4em;padding:1em;display:block}code{background:#f3f3f3;font-family:"DejaVu Sans Mono",monospace;padding:1px 4px;overflow-wrap:break-word}h1 code{background:transparent}pre{border-top:1px solid #ccc;border-bottom:1px solid #ccc;margin:1em 0}#http-server-module-list{display:flex;flex-flow:column}#http-server-module-list div{display:flex}#http-server-module-list dt{min-width:10%}#http-server-module-list p{margin-top:0}.toc ul,#index{list-style-type:none;margin:0;padding:0}#index code{background:transparent}#index h3{border-bottom:1px solid #ddd}#index ul{padding:0}#index h4{margin-top:.6em;font-weight:bold}@media (min-width:200ex){#index .two-column{column-count:2}}@media (min-width:300ex){#index .two-column{column-count:3}}dl{margin-bottom:2em}dl dl:last-child{margin-bottom:4em}dd{margin:0 0 1em 3em}#header-classes + dl > dd{margin-bottom:3em}dd dd{margin-left:2em}dd p{margin:10px 0}.name{background:#eee;font-size:.85em;padding:5px 10px;display:inline-block;min-width:40%}.name:hover{background:#e0e0e0}dt:target .name{background:var(--highlight-color)}.name > span:first-child{white-space:nowrap}.name.class > span:nth-child(2){margin-left:.4em}.inherited{color:#999;border-left:5px solid #eee;padding-left:1em}.inheritance em{font-style:normal;font-weight:bold}.desc h2{font-weight:400;font-size:1.25em}.desc h3{font-size:1em}.desc dt code{background:inherit}.source > summary,.git-link-div{color:#666;text-align:right;font-weight:400;font-size:.8em;text-transform:uppercase}.source summary > *{white-space:nowrap;cursor:pointer}.git-link{color:inherit;margin-left:1em}.source pre{max-height:500px;overflow:auto;margin:0}.source pre code{font-size:12px;overflow:visible;min-width:max-content}.hlist{list-style:none}.hlist li{display:inline}.hlist li:after{content:',\2002'}.hlist li:last-child:after{content:none}.hlist .hlist{display:inline;padding-left:1em}img{max-width:100%}td{padding:0 .5em}.admonition{padding:.1em 1em;margin:1em 0}.admonition-title{font-weight:bold}.admonition.note,.admonition.info,.admonition.important{background:#aef}.admonition.todo,.admonition.versionadded,.admonition.tip,.admonition.hint{background:#dfd}.admonition.warning,.admonition.versionchanged,.admonition.deprecated{background:#fd4}.admonition.error,.admonition.danger,.admonition.caution{background:lightpink}</style> |
| 13 | +<style media="screen and (min-width: 700px)">@media screen and (min-width:700px){#sidebar{width:30%;height:100vh;overflow:auto;position:sticky;top:0}#content{width:70%;max-width:100ch;padding:3em 4em;border-left:1px solid #ddd}pre code{font-size:1em}.name{font-size:1em}main{display:flex;flex-direction:row-reverse;justify-content:flex-end}.toc ul ul,#index ul ul{padding-left:1em}.toc > ul > li{margin-top:.5em}}</style> |
| 14 | +<style media="print">@media print{#sidebar h1{page-break-before:always}.source{display:none}}@media print{*{background:transparent !important;color:#000 !important;box-shadow:none !important;text-shadow:none !important}a[href]:after{content:" (" attr(href) ")";font-size:90%}a[href][title]:after{content:none}abbr[title]:after{content:" (" attr(title) ")"}.ir a:after,a[href^="javascript:"]:after,a[href^="#"]:after{content:""}pre,blockquote{border:1px solid #999;page-break-inside:avoid}thead{display:table-header-group}tr,img{page-break-inside:avoid}img{max-width:100% !important}@page{margin:0.5cm}p,h2,h3{orphans:3;widows:3}h1,h2,h3,h4,h5,h6{page-break-after:avoid}}</style> |
| 15 | +<script type="text/x-mathjax-config">MathJax.Hub.Config({ tex2jax: { inlineMath: [ ['$','$'], ["\\(","\\)"] ], processEscapes: true } });</script> |
| 16 | +<script async src="https://cdnjs.cloudflare.com/ajax/libs/mathjax/2.7.7/latest.js?config=TeX-AMS_CHTML" integrity="sha256-kZafAc6mZvK3W3v1pHOcUix30OHQN6pU/NO2oFkqZVw=" crossorigin></script> |
| 17 | +<script defer src="https://cdnjs.cloudflare.com/ajax/libs/highlight.js/11.9.0/highlight.min.js" integrity="sha512-D9gUyxqja7hBtkWpPWGt9wfbfaMGVt9gnyCvYa+jojwwPHLCzUm5i8rpk7vD7wNee9bA35eYIjobYPaQuKS1MQ==" crossorigin></script> |
| 18 | +<script>window.addEventListener('DOMContentLoaded', () => { |
| 19 | +hljs.configure({languages: ['bash', 'css', 'diff', 'graphql', 'ini', 'javascript', 'json', 'plaintext', 'python', 'python-repl', 'rust', 'shell', 'sql', 'typescript', 'xml', 'yaml']}); |
| 20 | +hljs.highlightAll(); |
| 21 | +/* Collapse source docstrings */ |
| 22 | +setTimeout(() => { |
| 23 | +[...document.querySelectorAll('.hljs.language-python > .hljs-string')] |
| 24 | +.filter(el => el.innerHTML.length > 200 && ['"""', "'''"].includes(el.innerHTML.substring(0, 3))) |
| 25 | +.forEach(el => { |
| 26 | +let d = document.createElement('details'); |
| 27 | +d.classList.add('hljs-string'); |
| 28 | +d.innerHTML = '<summary>"""</summary>' + el.innerHTML.substring(3); |
| 29 | +el.replaceWith(d); |
| 30 | +}); |
| 31 | +}, 100); |
| 32 | +})</script> |
| 33 | +</head> |
| 34 | +<body> |
| 35 | +<main> |
| 36 | +<article id="content"> |
| 37 | +<header> |
| 38 | +<h1 class="title">Module <code>underworld3.ckdtree</code></h1> |
| 39 | +</header> |
| 40 | +<section id="section-intro"> |
| 41 | +</section> |
| 42 | +<section> |
| 43 | +</section> |
| 44 | +<section> |
| 45 | +</section> |
| 46 | +<section> |
| 47 | +</section> |
| 48 | +<section> |
| 49 | +<h2 class="section-title" id="header-classes">Classes</h2> |
| 50 | +<dl> |
| 51 | +<dt id="underworld3.ckdtree.cKDTree"><code class="flex name class"> |
| 52 | +<span>class <span class="ident">cKDTree</span></span> |
| 53 | +<span>(</span><span>...)</span> |
| 54 | +</code></dt> |
| 55 | +<dd> |
| 56 | +<div class="desc"><p>KD-Tree indexes are data structures and algorithms for the efficient |
| 57 | +determination of nearest neighbours.</p> |
| 58 | +<p>This class generates a kd-tree index for the provided points, and provides |
| 59 | +the necessary methods for finding which points are closest to a given query |
| 60 | +location.</p> |
| 61 | +<p>This class utilises <code>nanoflann</code> for kd-tree functionality.</p> |
| 62 | +<h2 id="parameters">Parameters</h2> |
| 63 | +<p>points: |
| 64 | +The points for which the kd-tree index will be build. This |
| 65 | +should be a 2-dimensional array of size (n_points,dim).</p> |
| 66 | +<h2 id="example">Example</h2> |
| 67 | +<pre><code class="language-python-repl">>>> import numpy as np |
| 68 | +>>> import underworld3 as uw |
| 69 | +</code></pre> |
| 70 | +<p>Generate a random set of points</p> |
| 71 | +<pre><code class="language-python-repl">>>> pts = np.random.random( size=(100,2) ) |
| 72 | +</code></pre> |
| 73 | +<p>Build the index on the points</p> |
| 74 | +<pre><code class="language-python-repl">>>> index = uw.algorithms.KDTree(pts) |
| 75 | +>>> index.build_index() |
| 76 | +</code></pre> |
| 77 | +<p>Search the index for a coordinate</p> |
| 78 | +<pre><code class="language-python-repl">>>> coord = np.zeros((1,2)) |
| 79 | +>>> coord[0] = (0.5,0.5) |
| 80 | +>>> indices, dist_sqr, found = index.find_closest_point(coord) |
| 81 | +</code></pre> |
| 82 | +<p>Confirm that a point has been found</p> |
| 83 | +<pre><code class="language-python-repl">>>> found[0] |
| 84 | +True |
| 85 | +</code></pre></div> |
| 86 | +<h3>Methods</h3> |
| 87 | +<dl> |
| 88 | +<dt id="underworld3.ckdtree.cKDTree.build_index"><code class="name flex"> |
| 89 | +<span>def <span class="ident">build_index</span></span>(<span>self)</span> |
| 90 | +</code></dt> |
| 91 | +<dd> |
| 92 | +<div class="desc"><p>Build the kd-tree index.</p></div> |
| 93 | +</dd> |
| 94 | +<dt id="underworld3.ckdtree.cKDTree.find_closest_n_points"><code class="name flex"> |
| 95 | +<span>def <span class="ident">find_closest_n_points</span></span>(<span>self, nCount: numpy.int64, coords: numpy.ndarray)</span> |
| 96 | +</code></dt> |
| 97 | +<dd> |
| 98 | +<div class="desc"><p>Find the n points closest to the provided coordinates.</p> |
| 99 | +<h2 id="parameters">Parameters</h2> |
| 100 | +<p>nCount: |
| 101 | +The number of nearest neighbour points to find for each <code>coords</code>.</p> |
| 102 | +<p>coords: |
| 103 | +Coordinates of the points for which the kd-tree index will be searched for nearest |
| 104 | +neighbours. This should be a 2-dimensional array of size (n_coords,dim).</p> |
| 105 | +<h2 id="returns">Returns</h2> |
| 106 | +<dl> |
| 107 | +<dt><code>indices:</code></dt> |
| 108 | +<dd>An integer array of indices into the <code>points</code> array (passed into the constructor) corresponding to |
| 109 | +the nearest neighbour for the search coordinates. It will be of size (n_coords).</dd> |
| 110 | +<dt><code>dist_sqr:</code></dt> |
| 111 | +<dd>A float array of squred distances between the provided coords and the nearest neighbouring |
| 112 | +points. It will be of size (n_coords).</dd> |
| 113 | +</dl></div> |
| 114 | +</dd> |
| 115 | +<dt id="underworld3.ckdtree.cKDTree.find_closest_point"><code class="name flex"> |
| 116 | +<span>def <span class="ident">find_closest_point</span></span>(<span>self, coords: numpy.ndarray)</span> |
| 117 | +</code></dt> |
| 118 | +<dd> |
| 119 | +<div class="desc"><p>Find the points closest to the provided set of coordinates.</p> |
| 120 | +<h2 id="parameters">Parameters</h2> |
| 121 | +<p>coords: |
| 122 | +An array of coordinates for which the kd-tree index will be searched for nearest |
| 123 | +neighbours. This should be a 2-dimensional array of size (n_coords,dim).</p> |
| 124 | +<h2 id="returns">Returns</h2> |
| 125 | +<dl> |
| 126 | +<dt><code>indices:</code></dt> |
| 127 | +<dd>An integer array of indices into the <code>points</code> array (passed into the constructor) corresponding to |
| 128 | +the nearest neighbour for the search coordinates. It will be of size (n_coords).</dd> |
| 129 | +<dt><code>dist_sqr:</code></dt> |
| 130 | +<dd>A float array of squared distances between the provided coords and the nearest neighbouring |
| 131 | +points. It will be of size (n_coords).</dd> |
| 132 | +<dt><code>found:</code></dt> |
| 133 | +<dd>A bool array of flags which signals whether a nearest neighbour has been found for a given |
| 134 | +coordinate. It will be of size (n_coords).</dd> |
| 135 | +</dl></div> |
| 136 | +</dd> |
| 137 | +<dt id="underworld3.ckdtree.cKDTree.kdtree_points"><code class="name flex"> |
| 138 | +<span>def <span class="ident">kdtree_points</span></span>(<span>self)</span> |
| 139 | +</code></dt> |
| 140 | +<dd> |
| 141 | +<div class="desc"><p>Returns a view of the points used to define the kd-tree</p></div> |
| 142 | +</dd> |
| 143 | +<dt id="underworld3.ckdtree.cKDTree.query"><code class="name flex"> |
| 144 | +<span>def <span class="ident">query</span></span>(<span>self, coords: numpy.ndarray, k: numpy.int64, sqr_dists=True)</span> |
| 145 | +</code></dt> |
| 146 | +<dd> |
| 147 | +<div class="desc"><p>Find the n points closest to the provided coordinates.</p></div> |
| 148 | +</dd> |
| 149 | +<dt id="underworld3.ckdtree.cKDTree.rbf_interpolator_local"><code class="name flex"> |
| 150 | +<span>def <span class="ident">rbf_interpolator_local</span></span>(<span>self, coords, data, nnn=4, verbose=False)</span> |
| 151 | +</code></dt> |
| 152 | +<dd> |
| 153 | +<div class="desc"></div> |
| 154 | +</dd> |
| 155 | +<dt id="underworld3.ckdtree.cKDTree.rbf_interpolator_local_from_kdtree"><code class="name flex"> |
| 156 | +<span>def <span class="ident">rbf_interpolator_local_from_kdtree</span></span>(<span>self, coords, data, nnn=4, verbose=False)</span> |
| 157 | +</code></dt> |
| 158 | +<dd> |
| 159 | +<div class="desc"><p>An inverse (squared) distance weighted mapping of a numpy array from the |
| 160 | +set of coordinates defined by the kd-tree to the set of input points specified. |
| 161 | +This assumes all points are local to the same processor. |
| 162 | +If that is not the case, it is best to use a particle swarm |
| 163 | +to manage the distributed data.</p></div> |
| 164 | +</dd> |
| 165 | +<dt id="underworld3.ckdtree.cKDTree.rbf_interpolator_local_to_kdtree"><code class="name flex"> |
| 166 | +<span>def <span class="ident">rbf_interpolator_local_to_kdtree</span></span>(<span>self, coords, data, nnn=4, verbose=False, weights=None)</span> |
| 167 | +</code></dt> |
| 168 | +<dd> |
| 169 | +<div class="desc"><p>An inverse (squared) distance weighted mapping of a numpy array to the |
| 170 | +set of coordinates defined by the kd-tree from the set of input points specified. |
| 171 | +This assumes all points are local to the same processor. |
| 172 | +If that is not the case, it is sensible to use a particle swarm |
| 173 | +to manage the distributed data.</p></div> |
| 174 | +</dd> |
| 175 | +</dl> |
| 176 | +</dd> |
| 177 | +</dl> |
| 178 | +</section> |
| 179 | +</article> |
| 180 | +<nav id="sidebar"> |
| 181 | +<div class="toc"> |
| 182 | +<ul></ul> |
| 183 | +</div> |
| 184 | +<ul id="index"> |
| 185 | +<li><h3>Super-module</h3> |
| 186 | +<ul> |
| 187 | +<li><code><a title="underworld3" href="index.html">underworld3</a></code></li> |
| 188 | +</ul> |
| 189 | +</li> |
| 190 | +<li><h3><a href="#header-classes">Classes</a></h3> |
| 191 | +<ul> |
| 192 | +<li> |
| 193 | +<h4><code><a title="underworld3.ckdtree.cKDTree" href="#underworld3.ckdtree.cKDTree">cKDTree</a></code></h4> |
| 194 | +<ul class=""> |
| 195 | +<li><code><a title="underworld3.ckdtree.cKDTree.build_index" href="#underworld3.ckdtree.cKDTree.build_index">build_index</a></code></li> |
| 196 | +<li><code><a title="underworld3.ckdtree.cKDTree.find_closest_n_points" href="#underworld3.ckdtree.cKDTree.find_closest_n_points">find_closest_n_points</a></code></li> |
| 197 | +<li><code><a title="underworld3.ckdtree.cKDTree.find_closest_point" href="#underworld3.ckdtree.cKDTree.find_closest_point">find_closest_point</a></code></li> |
| 198 | +<li><code><a title="underworld3.ckdtree.cKDTree.kdtree_points" href="#underworld3.ckdtree.cKDTree.kdtree_points">kdtree_points</a></code></li> |
| 199 | +<li><code><a title="underworld3.ckdtree.cKDTree.query" href="#underworld3.ckdtree.cKDTree.query">query</a></code></li> |
| 200 | +<li><code><a title="underworld3.ckdtree.cKDTree.rbf_interpolator_local" href="#underworld3.ckdtree.cKDTree.rbf_interpolator_local">rbf_interpolator_local</a></code></li> |
| 201 | +<li><code><a title="underworld3.ckdtree.cKDTree.rbf_interpolator_local_from_kdtree" href="#underworld3.ckdtree.cKDTree.rbf_interpolator_local_from_kdtree">rbf_interpolator_local_from_kdtree</a></code></li> |
| 202 | +<li><code><a title="underworld3.ckdtree.cKDTree.rbf_interpolator_local_to_kdtree" href="#underworld3.ckdtree.cKDTree.rbf_interpolator_local_to_kdtree">rbf_interpolator_local_to_kdtree</a></code></li> |
| 203 | +</ul> |
| 204 | +</li> |
| 205 | +</ul> |
| 206 | +</li> |
| 207 | +</ul> |
| 208 | +</nav> |
| 209 | +</main> |
| 210 | +<footer id="footer"> |
| 211 | +<p>Generated by <a href="https://pdoc3.github.io/pdoc" title="pdoc: Python API documentation generator"><cite>pdoc</cite> 0.11.6</a>.</p> |
| 212 | +</footer> |
| 213 | +</body> |
| 214 | +</html> |
0 commit comments