Created
May 14, 2018 22:48
-
-
Save nathan-sixnines/0c419fad542f6fde538eacfdb3c926b8 to your computer and use it in GitHub Desktop.
array padding function
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
<script src="https://cdn.jsdelivr.net/gh/nicolaspanel/[email protected]/dist/numjs.min.js"></script> | |
<script> | |
function modularPaddArray(prePadArray, border){ // buffer should be input tile radius | |
var ppa = prePadArray | |
var b = border | |
// 9 parts, original array in center, 4 sides, 4 corners | |
var botSlice = ppa.slice([ppa.shape[0]-b,ppa.shape[0]]) | |
var topSlice = ppa.slice([ 0, b]) | |
var rightSlice = ppa.slice(null, [ppa.shape[1]-b,ppa.shape[1]]) | |
var leftSlice = ppa.slice(null, [ 0, b]) | |
var topLeftCor = ppa.slice([ 0, b],[ 0, b]) | |
var topRightCor = ppa.slice([ 0, b],[ppa.shape[1]-b,ppa.shape[1]]) | |
var botLeftCor = ppa.slice([ppa.shape[0]-b,ppa.shape[0]],[ 0, b]) | |
var botRightCor = ppa.slice([ppa.shape[0]-b,ppa.shape[0]],[ppa.shape[1]-b,ppa.shape[1]]) | |
console.log(botRightCor.shape) | |
console.log(botSlice.shape) | |
var topPad = nj.concatenate( nj.concatenate( botRightCor, botSlice), botLeftCor) | |
var midPad = nj.concatenate( nj.concatenate( rightSlice, ppa), leftSlice) | |
var botPad = nj.concatenate( nj.concatenate( topRightCor, topSlice), topLeftCor) | |
var padded = nj.concatenate(nj.concatenate(topPad.T, midPad.T), botPad.T).T | |
return padded //console.log(padded) | |
} | |
var b = nj.arange(5 * 5).reshape(5,5); | |
padded = modularPaddArray(b, 2) | |
console.log(padded.shape) | |
console.log(padded) | |
</script> |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment