summaryrefslogtreecommitdiff
path: root/gi/posterior-regularisation/prjava/src/util/MemoryTracker.java
blob: 83a65611566a2ef0ccaacd7bf9048b6d2cd4bdfd (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
package util;


public class MemoryTracker {
	
	double initM,finalM;
	boolean start = false,finish = false;
	
	public MemoryTracker(){
		
	}
	
	public void start(){
		System.gc();
	    System.gc();
	    System.gc();
	    initM = (Runtime.getRuntime().totalMemory() - Runtime.getRuntime().freeMemory())/(1024*1024);  
	    start = true;
	}
	
	public void finish(){
		if(!start){
			throw new RuntimeException("Canot stop before starting");
		}
		System.gc();
	    System.gc();
	    System.gc();
	    finalM = (Runtime.getRuntime().totalMemory() - Runtime.getRuntime().freeMemory())/(1024*1024);  
	    finish = true;
	}
	
	public String print(){
		if(!finish){
			throw new RuntimeException("Canot print before stopping");
		}
		return "Used: " + (finalM - initM) + "MB";
	}
	
	public void clear(){
		initM = 0;
		finalM = 0;
		finish = false;
		start = false;
	}
	
	
}