001: /*
002: * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
003: *
004: * Copyright 1997-2007 Sun Microsystems, Inc. All rights reserved.
005: *
006: * The contents of this file are subject to the terms of either the GNU
007: * General Public License Version 2 only ("GPL") or the Common
008: * Development and Distribution License("CDDL") (collectively, the
009: * "License"). You may not use this file except in compliance with the
010: * License. You can obtain a copy of the License at
011: * http://www.netbeans.org/cddl-gplv2.html
012: * or nbbuild/licenses/CDDL-GPL-2-CP. See the License for the
013: * specific language governing permissions and limitations under the
014: * License. When distributing the software, include this License Header
015: * Notice in each file and include the License file at
016: * nbbuild/licenses/CDDL-GPL-2-CP. Sun designates this
017: * particular file as subject to the "Classpath" exception as provided
018: * by Sun in the GPL Version 2 section of the License file that
019: * accompanied this code. If applicable, add the following below the
020: * License Header, with the fields enclosed by brackets [] replaced by
021: * your own identifying information:
022: * "Portions Copyrighted [year] [name of copyright owner]"
023: *
024: * Contributor(s):
025: *
026: * The Original Software is NetBeans. The Initial Developer of the Original
027: * Software is Sun Microsystems, Inc. Portions Copyright 1997-2006 Sun
028: * Microsystems, Inc. All Rights Reserved.
029: *
030: * If you wish your version of this file to be governed by only the CDDL
031: * or only the GPL Version 2, indicate your decision by adding
032: * "[Contributor] elects to include this software in this distribution
033: * under the [CDDL or GPL Version 2] license." If you do not indicate a
034: * single choice of license, a recipient has the option to distribute
035: * your version of this file under either the CDDL, the GPL Version 2 or
036: * to extend the choice of license to its licensees as provided above.
037: * However, if you add GPL Version 2 code and therefore, elected the GPL
038: * Version 2 license, then the option applies only if the new code is
039: * made subject to such option by the copyright holder.
040: */
041:
042: package org.netbeans.insane.scanner;
043:
044: import java.util.*;
045:
046: /**
047: * A visitor implementation that counts occurence and total size of found
048: * objects, classified by their class.
049: *
050: * Usage: use it as a {@link Visitor} for an engine. After the engine finishes,
051: * you can query found classes and per class statistics using
052: * {@link #getClasses()}, {@link #getCountForClass(java.lang.Class)},
053: * {@link #getSizeForClass(java.lang.Class)}, and gobal statistics
054: * using {@link #getTotalCount()} and {@link #getTotalSize}.
055: *
056: * @author Nenik
057: */
058: public class CountingVisitor implements Visitor {
059:
060: private Map<Class, Info> infoMap = new HashMap<Class, Info>();
061: private int count;
062: private int size;
063:
064: /** Creates a new instance of CountingVisitor */
065: public CountingVisitor() {
066: }
067:
068: public void visitClass(Class cls) {
069: infoMap.put(cls, new Info());
070: }
071:
072: public void visitObject(ObjectMap map, Object obj) {
073: Info info = infoMap.get(obj.getClass());
074: assert info != null : "Engine shall announce the class before instance";
075:
076: info.count++;
077: count++;
078: int objSize = ScannerUtils.sizeOf(obj);
079: info.size += objSize;
080: size += objSize;
081: }
082:
083: public void visitStaticReference(ObjectMap map, Object to,
084: java.lang.reflect.Field ref) {
085: }
086:
087: public void visitObjectReference(ObjectMap map, Object from,
088: Object to, java.lang.reflect.Field ref) {
089: }
090:
091: public void visitArrayReference(ObjectMap map, Object from,
092: Object to, int index) {
093: }
094:
095: public Set<Class> getClasses() {
096: return Collections.unmodifiableSet(infoMap.keySet());
097: }
098:
099: public int getCountForClass(Class cls) {
100: Info info = infoMap.get(cls);
101: if (info == null)
102: throw new IllegalArgumentException("Unknown class");
103:
104: return info.count;
105: }
106:
107: public int getSizeForClass(Class cls) {
108: Info info = infoMap.get(cls);
109: if (info == null)
110: throw new IllegalArgumentException("Unknown class");
111:
112: return info.size;
113: }
114:
115: public int getTotalCount() {
116: return count;
117: }
118:
119: public int getTotalSize() {
120: return size;
121: }
122:
123: // A structure holding statistics for one class.
124: private static class Info {
125: int count;
126: int size;
127: }
128: }
|