Merge pull request #206 from mbasmanova/estimate-memory-size-for-accelerators

Add accelerator size to OGCGeometry#estimateMemorySize

Author: stolstov

src/main/java/com/esri/core/geometry/GeometryAccelerators.java

@@ -23,8 +23,6 @@
  */
 package com.esri.core.geometry;
 
-import java.util.ArrayList;
-
 class GeometryAccelerators {
 
 	private RasterizedGeometry2D m_rasterizedGeometry;
@@ -84,4 +82,11 @@ static boolean canUseQuadTreeForPaths(Geometry geom) {
 
         return true;
     }
+
+	public long estimateMemorySize()
+	{
+		return (m_rasterizedGeometry != null ? m_rasterizedGeometry.estimateMemorySize() : 0) +
+			(m_quad_tree != null ? m_quad_tree.estimateMemorySize() : 0) +
+			(m_quad_tree_for_paths != null ? m_quad_tree_for_paths.estimateMemorySize() : 0);
+	}
 }

src/main/java/com/esri/core/geometry/MultiPathImpl.java

@@ -77,6 +77,11 @@ public long estimateMemorySize()
 				size += m_vertexAttributes[i].estimateMemorySize();
 			}
 		}
+
+		if (m_accelerators != null) {
+			size += m_accelerators.estimateMemorySize();
+		}
+
 		return size;
 	}
 

src/main/java/com/esri/core/geometry/QuadTreeImpl.java

@@ -29,6 +29,10 @@
 import java.io.Serializable;
 import java.util.ArrayList;
 
+import static com.esri.core.geometry.SizeOf.SIZE_OF_DATA;
+import static com.esri.core.geometry.SizeOf.SIZE_OF_QUAD_TREE_IMPL;
+import static com.esri.core.geometry.SizeOf.sizeOfObjectArray;
+
 class QuadTreeImpl implements Serializable {
 	private static final long serialVersionUID = 1L;
 
@@ -777,6 +781,22 @@ QuadTreeSortedIteratorImpl getSortedIterator() {
 		return new QuadTreeSortedIteratorImpl(getIterator());
 	}
 
+	public long estimateMemorySize()
+	{
+		long size = SIZE_OF_QUAD_TREE_IMPL +
+				(m_extent != null ? m_extent.estimateMemorySize() : 0) +
+				(m_data_extent != null ? m_data_extent.estimateMemorySize() : 0) +
+				(m_quad_tree_nodes != null ? m_quad_tree_nodes.estimateMemorySize() : 0) +
+				(m_element_nodes != null ? m_element_nodes.estimateMemorySize() : 0) +
+				(m_free_data != null ? m_free_data.estimateMemorySize() : 0);
+
+		if (m_data != null) {
+			size += sizeOfObjectArray(m_data.size()) + m_data.size() * SIZE_OF_DATA;
+		}
+
+		return size;
+	}
+
 	private void reset_(Envelope2D extent, int height) {
 		if (height < 0 || height > 127)
 			throw new IllegalArgumentException("invalid height");
@@ -1268,9 +1288,6 @@ static final class Data {
 		int element;
 		Envelope2D box;
 
-		Data() {
-		}
-		
 		Data(int element_, double x1, double y1, double x2, double y2) {
 			element = element_;
 			box = new Envelope2D();

src/main/java/com/esri/core/geometry/RasterizedGeometry2D.java

@@ -136,4 +136,10 @@ static boolean canUseAccelerator(Geometry geom) {
 	 */
 	public abstract boolean dbgSaveToBitmap(String fileName);
 
+	/**
+	 * Returns an estimate of this object size in bytes.
+	 *
+	 * @return Returns an estimate of this object size in bytes.
+	 */
+	public abstract long estimateMemorySize();
 }

src/main/java/com/esri/core/geometry/RasterizedGeometry2DImpl.java

@@ -24,18 +24,14 @@
 
 package com.esri.core.geometry;
 
-import java.io.*;
+import java.io.FileOutputStream;
+import java.io.IOException;
 import java.nio.ByteBuffer;
 import java.nio.ByteOrder;
 
-import com.esri.core.geometry.Envelope2D;
-import com.esri.core.geometry.Geometry;
-import com.esri.core.geometry.GeometryException;
-import com.esri.core.geometry.NumberUtils;
-import com.esri.core.geometry.Point2D;
-import com.esri.core.geometry.Segment;
-import com.esri.core.geometry.SegmentIteratorImpl;
-import com.esri.core.geometry.SimpleRasterizer;
+import static com.esri.core.geometry.SizeOf.SIZE_OF_RASTERIZED_GEOMETRY_2D_IMPL;
+import static com.esri.core.geometry.SizeOf.SIZE_OF_SCAN_CALLBACK_IMPL;
+import static com.esri.core.geometry.SizeOf.sizeOfIntArray;
 
 final class RasterizedGeometry2DImpl extends RasterizedGeometry2D {
 	int[] m_bitmap;
@@ -89,6 +85,13 @@ public void drawScan(int[] scans, int scanCount3) {
 			}
 			}
 		}
+
+		@Override
+		public long estimateMemorySize()
+		{
+			return SIZE_OF_SCAN_CALLBACK_IMPL +
+					(m_bitmap != null ? sizeOfIntArray(m_bitmap.length) : 0);
+		}
 	}
 
 	void fillMultiPath(SimpleRasterizer rasterizer, Transformation2D trans, MultiPathImpl polygon, boolean isWinding) {
@@ -559,4 +562,14 @@ public boolean dbgSaveToBitmap(String fileName) {
 
 		}
 	}
+
+	@Override
+	public long estimateMemorySize()
+	{
+		return SIZE_OF_RASTERIZED_GEOMETRY_2D_IMPL +
+				(m_geomEnv != null ? m_geomEnv.estimateMemorySize() : 0) +
+				(m_transform != null ? m_transform.estimateMemorySize(): 0) +
+				(m_rasterizer != null ? m_rasterizer.estimateMemorySize(): 0) +
+				(m_callback != null ? m_callback.estimateMemorySize(): 0);
+	}
 }

src/main/java/com/esri/core/geometry/SimpleRasterizer.java

@@ -24,11 +24,14 @@
 
 package com.esri.core.geometry;
 
-import java.util.ArrayList;
 import java.util.Arrays;
-import java.util.Collections;
 import java.util.Comparator;
 
+import static com.esri.core.geometry.SizeOf.SIZE_OF_EDGE;
+import static com.esri.core.geometry.SizeOf.SIZE_OF_SIMPLE_RASTERIZER;
+import static com.esri.core.geometry.SizeOf.sizeOfIntArray;
+import static com.esri.core.geometry.SizeOf.sizeOfObjectArray;
+
 /**
  * Simple scanline rasterizer. Caller provides a callback to draw pixels to actual surface.
  *
@@ -44,15 +47,22 @@ public class SimpleRasterizer {
 	 * Winding fill rule
 	 */
 	public final static int WINDING = 1;
-	
-	public static interface ScanCallback {
+
+	public interface ScanCallback {
 		/**
 		 * Rasterizer calls this method for each scan it produced
 		 * @param scans array of scans. Scans are triplets of numbers. The start X coordinate for the scan (inclusive),
 		 * the end X coordinate of the scan (exclusive), the Y coordinate for the scan.
 		 * @param scanCount3 The number of initialized elements in the scans array. The scan count is scanCount3 / 3. 
 		 */
-		public abstract void drawScan(int[] scans, int scanCount3);
+		void drawScan(int[] scans, int scanCount3);
+
+		/**
+		 * Returns an estimate of this object size in bytes.
+		 *
+		 * @return Returns an estimate of this object size in bytes.
+		 */
+		long estimateMemorySize();
 	}
 
 	public SimpleRasterizer() {
@@ -340,17 +350,50 @@ final boolean addSegmentStroke(double x1, double y1, double x2, double y2, doubl
     }
 
 	public final ScanCallback getScanCallback() { return callback_; }
-	
-	
+
+	public long estimateMemorySize()
+	{
+		// callback_ is only a pointer, the actual size is accounted for in the caller of setup()
+		long size = SIZE_OF_SIMPLE_RASTERIZER +
+				(activeEdgesTable_ != null ? activeEdgesTable_.estimateMemorySize() : 0) +
+				(scanBuffer_ != null ? sizeOfIntArray(scanBuffer_.length) : 0);
+
+		if (ySortedEdges_ != null) {
+			size += sizeOfObjectArray(ySortedEdges_.length);
+			for (int i = 0; i < ySortedEdges_.length; i++) {
+				if (ySortedEdges_[i] != null) {
+					size += ySortedEdges_[i].estimateMemorySize();
+				}
+			}
+		}
+
+		if (sortBuffer_ != null) {
+			size += sizeOfObjectArray(sortBuffer_.length);
+			for (int i = 0; i < sortBuffer_.length; i++) {
+				if (sortBuffer_[i] != null) {
+					size += sortBuffer_[i].estimateMemorySize();
+				}
+			}
+		}
+
+		return size;
+	}
+
 	//PRIVATE
 
-	private static class Edge {
+	static class Edge {
 		long x;
 		long dxdy;
 		int y;
 		int ymax;
 		int dir;
 		Edge next;
+
+		long estimateMemorySize()
+		{
+			// next is only a pointer, the actual size is accounted for in SimpleRasterizer#estimateMemorySize
+			return SIZE_OF_EDGE;
+		}
 	}
 
 	private final void advanceAET_() {

src/main/java/com/esri/core/geometry/SizeOf.java

@@ -36,6 +36,8 @@
 import static sun.misc.Unsafe.ARRAY_INT_INDEX_SCALE;
 import static sun.misc.Unsafe.ARRAY_LONG_BASE_OFFSET;
 import static sun.misc.Unsafe.ARRAY_LONG_INDEX_SCALE;
+import static sun.misc.Unsafe.ARRAY_OBJECT_BASE_OFFSET;
+import static sun.misc.Unsafe.ARRAY_OBJECT_INDEX_SCALE;
 import static sun.misc.Unsafe.ARRAY_SHORT_BASE_OFFSET;
 import static sun.misc.Unsafe.ARRAY_SHORT_INDEX_SCALE;
 
@@ -88,6 +90,22 @@ public final class SizeOf {
 
 	public static final int SIZE_OF_MAPGEOMETRY = 24;
 
+	public static final int SIZE_OF_RASTERIZED_GEOMETRY_2D_IMPL = 112;
+
+	public static final int SIZE_OF_SCAN_CALLBACK_IMPL = 32;
+
+	public static final int SIZE_OF_TRANSFORMATION_2D = 64;
+
+	public static final int SIZE_OF_SIMPLE_RASTERIZER = 64;
+
+	public static final int SIZE_OF_EDGE = 48;
+
+	public static final int SIZE_OF_QUAD_TREE_IMPL = 48;
+
+	public static final int SIZE_OF_DATA = 24;
+
+	public static final int SIZE_OF_STRIDED_INDEX_TYPE_COLLECTION = 48;
+
 	public static long sizeOfByteArray(int length) {
 		return ARRAY_BYTE_BASE_OFFSET + (((long) ARRAY_BYTE_INDEX_SCALE) * length);
 	}
@@ -116,6 +134,11 @@ public static long sizeOfDoubleArray(int length) {
 		return ARRAY_DOUBLE_BASE_OFFSET + (((long) ARRAY_DOUBLE_INDEX_SCALE) * length);
 	}
 
+	public static long sizeOfObjectArray(int length)
+	{
+		return ARRAY_OBJECT_BASE_OFFSET + (((long) ARRAY_OBJECT_INDEX_SCALE) * length);
+	}
+
 	private SizeOf() {
 	}
 }

src/main/java/com/esri/core/geometry/StridedIndexTypeCollection.java

@@ -25,6 +25,10 @@
 
 import java.io.Serializable;
 
+import static com.esri.core.geometry.SizeOf.SIZE_OF_STRIDED_INDEX_TYPE_COLLECTION;
+import static com.esri.core.geometry.SizeOf.sizeOfIntArray;
+import static com.esri.core.geometry.SizeOf.sizeOfObjectArray;
+
 /**
  * A collection of strides of Index_type elements. To be used when one needs a
  * collection of homogeneous elements that contain only integer fields (i.e.
@@ -277,4 +281,18 @@ private void grow_(long newsize) {
 			}
 		}
 	}
+
+	public long estimateMemorySize()
+	{
+		long size = SIZE_OF_STRIDED_INDEX_TYPE_COLLECTION;
+		if (m_buffer != null) {
+			size += sizeOfObjectArray(m_buffer.length);
+			for (int i = 0; i< m_buffer.length; i++) {
+				if (m_buffer[i] != null) {
+					size += sizeOfIntArray(m_buffer[i].length);
+				}
+			}
+		}
+		return size;
+	}
 }

src/main/java/com/esri/core/geometry/Transformation2D.java

@@ -24,6 +24,8 @@
 
 package com.esri.core.geometry;
 
+import static com.esri.core.geometry.SizeOf.SIZE_OF_TRANSFORMATION_2D;
+
 /**
  * The affine transformation class for 2D.
  * 
@@ -921,4 +923,8 @@ public void extractScaleTransform(Transformation2D scale,
 		rotateNshearNshift.multiply(this);
 	}
 
+	public long estimateMemorySize()
+	{
+		return SIZE_OF_TRANSFORMATION_2D;
+	}
 }