1 package org.simantics.scl.compiler.module.coverage;
\r
3 import org.simantics.scl.compiler.module.Module;
\r
4 import org.simantics.scl.runtime.profiling.BranchPoint;
\r
6 import gnu.trove.map.hash.THashMap;
\r
7 import gnu.trove.procedure.TObjectObjectProcedure;
\r
9 public class CoverageBuilder {
\r
10 THashMap<String, THashMap<String, BranchPoint[]>> combined =
\r
11 new THashMap<String, THashMap<String, BranchPoint[]>>();
\r
13 public void addCoverage(Module module, boolean persistOverBranchpointReset) {
\r
14 THashMap<String, BranchPoint[]> branchPointMap = module.getBranchPoints();
\r
15 if(branchPointMap == null)
\r
17 THashMap<String, BranchPoint[]> oldBranchPointMap = combined.get(module.getName());
\r
18 if(oldBranchPointMap == null) {
\r
19 oldBranchPointMap = new THashMap<String, BranchPoint[]>();
\r
20 combined.put(module.getName(), oldBranchPointMap);
\r
22 THashMap<String, BranchPoint[]> oldBranchPointMap_ = oldBranchPointMap;
\r
23 branchPointMap.forEachEntry(new TObjectObjectProcedure<String, BranchPoint[]>() {
\r
25 public boolean execute(String name, BranchPoint[] branchPoints) {
\r
26 BranchPoint[] oldBranchPoints = oldBranchPointMap_.get(name);
\r
27 if(oldBranchPoints == null) {
\r
28 if (persistOverBranchpointReset) {
\r
29 // Clone the branchPoints array so that if will last over a reset
\r
30 BranchPoint[] clonedBranchPoints = cloneBranchPoints(branchPoints);
\r
31 oldBranchPointMap_.put(name, clonedBranchPoints);
\r
33 // No need to copy so resetting branchpoints will reset this as well
\r
34 oldBranchPointMap_.put(name, branchPoints);
\r
37 combineCounters(oldBranchPoints, branchPoints);
\r
44 private static BranchPoint[] cloneBranchPoints(BranchPoint[] oldBranchPoints) {
\r
45 BranchPoint[] newBranchPoints = new BranchPoint[oldBranchPoints.length];
\r
46 for (int i = 0; i < oldBranchPoints.length; i++) {
\r
47 BranchPoint bp = oldBranchPoints[i];
\r
48 BranchPoint[] children = cloneBranchPoints(bp.getChildren());
\r
49 newBranchPoints[i] = new BranchPoint(bp.getLocation(), bp.getCodeSize(), children);
\r
51 return newBranchPoints;
\r
54 private static void combineCounters(BranchPoint[] oldBranchPoints, BranchPoint[] branchPoints) {
\r
55 if(oldBranchPoints.length != branchPoints.length)
\r
56 throw new IllegalArgumentException("Incompatible branch points.");
\r
57 for(int i=0;i<branchPoints.length;++i) {
\r
58 BranchPoint oldBP = oldBranchPoints[i];
\r
59 BranchPoint newBP = branchPoints[i];
\r
60 if(oldBP.getLocation() != newBP.getLocation())
\r
61 throw new IllegalArgumentException("Incompatible branch points.");
\r
62 oldBP.incrementVisitCounter(newBP.getVisitCounter());
\r
63 combineCounters(oldBP.getChildren(), newBP.getChildren());
\r
67 public CombinedCoverage getCoverage() {
\r
68 THashMap<String,ModuleCoverage> moduleCoverages =
\r
69 new THashMap<String,ModuleCoverage>();
\r
70 combined.forEachEntry(new TObjectObjectProcedure<String, THashMap<String, BranchPoint[]>>() {
\r
72 public boolean execute(String name, THashMap<String, BranchPoint[]> branchPoints) {
\r
73 moduleCoverages.put(name, CoverageUtils.getCoverage(name, branchPoints));
\r
77 return CoverageUtils.combineCoverages(moduleCoverages);
\r