import jdk.incubator.vector.*; final sclass BWIntegralImage_vectorized implements MakesBufferedImage, IBWIntegralImage { int w, h; int vecShift; // vector size used as dual log int[] data; // 1 entry per pixel ifdef BWIntegralImage_CountAccesses long accesses; endifdef *() {} *(File f) { this(loadImage2(f)); } *(MakesBufferedImage img) { this(toBufferedImage(img)); } //*(BufferedImage img) { this(BWImage(img)); } *(BufferedImage image) ctex { alloc(image.getWidth(), image.getHeight()); // Just grab directly into data array PixelGrabber pixelGrabber = new PixelGrabber(image, 0, 0, w, h, data, 0, w); if (!pixelGrabber.grabPixels()) fail("Could not grab pixels"); VectorSpecies<Int> species = IntVector.SPECIES_PREFERRED; vecShift = dualLog(species.length()); int upperBound = species.loopBound(w); // for brightness of pixels, // for now we cheat by just using one of the channels // first row int i = 0, sum = 0; for (; i < upperBound; i += species.length()) { IntVector v = IntVector.fromArray(species, data, i); v = v.and(255).add(sum); v.intoArray(data, i); sum += v.reduceLanes(VectorOperators.ADD); } for (; i < w; i++) data[i] = (data[i] & 255) + sum; // subsequent rows int i = w; for (int y = 1; y < h; y++) { sum = 0; int x = 0; for (; x < upperBound; x += species.length(), i += species.length()) { IntVector v = IntVector.fromArray(species, data, i); IntVector v_last = IntVector.fromArray(species, data, i-w); v = v.and(255).add(v_last).add(sum); v.intoArray(data, i); sum += v.reduceLanes(VectorOperators.ADD); } for (; x < w; x++, i++) data[i] = (data[i] & 255) + sum; } } private void alloc(int w, int h) { this.w = w; this.h = h; if (w*h > 8*1024*1024) fail("Image too large (more than 8 MP): " + w + "*" + h); data = new int[w*h]; } // pixels outside of image are considered black int get(int x, int y) { ifdef BWIntegralImage_CountAccesses ++accesses; endifdef ret x < 0 || y < 0 || x >= w || y >= h ? 0 : data[min(y, h-1)*w+min(x, w-1)]; } public double getPixelAverage(int x1, int y1, int x2, int y2) { int area = (x2-x1)*(y2-y1); ret doubleRatio(bwIntegralImage_sumRect(this, x1, y1, x2, y2), area); } int getPixel(int x, int y) { ret bwIntegralImage_sumRect(this, x, y, x+1, y+1); } int getPixel(Pt p) { ret getPixel(p.x, p.y); } public int getWidth() { ret w; } public int getHeight() { ret h; } // unoptimized public BufferedImage getBufferedImage() { ret scaleDownUsingIntegralImageBW(this, w).getBufferedImage(); } }
Began life as a copy of #1019595
download show line numbers debug dex old transpilations
Travelled to 4 computer(s): bhatertpkbcr, mqqgnosmbjvj, pyentgdyhuwx, vouqrxazstgt
No comments. add comment
Snippet ID: | #1030523 |
Snippet name: | BWIntegralImage_vectorized [dev.] |
Eternal ID of this version: | #1030523/3 |
Text MD5: | 2dd0cd77b3fba772391f60454b1d19fd |
Author: | stefan |
Category: | javax / imaging |
Type: | JavaX fragment (include) |
Public (visible to everyone): | Yes |
Archived (hidden from active list): | No |
Created/modified: | 2021-01-05 15:22:49 |
Source code size: | 2899 bytes / 97 lines |
Pitched / IR pitched: | No / No |
Views / Downloads: | 217 / 252 |
Version history: | 2 change(s) |
Referenced in: | #1030531 - BWIntegralImage_doubleVectorized [pre-calcs 8x8 blocks using Vector API, dev.] |