6502/kickc
1
0
Fork 0
mirror of https://gitlab.com/camelot/kickc.git synced 2025-08-17 11:27:50 +00:00
Code Issues Projects Releases Wiki Activity

Added a cache for improved performance.

Browse Source
This commit is contained in:
jespergravgaard
2020-06-03 01:08:16 +02:00
parent bff7cd7f9f
commit 5d1aff1630
1 changed files with 11 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
src/main/java/dk/camelot64/kickc/model/VariableReferenceInfos.java
View File

@@ -4,10 +4,7 @@ import dk.camelot64.kickc.model.statements.Statement;
import dk.camelot64.kickc.model.values.*; import dk.camelot64.kickc.model.values.*;
import dk.camelot64.kickc.passes.calcs.PassNCalcVariableReferenceInfos; import dk.camelot64.kickc.passes.calcs.PassNCalcVariableReferenceInfos;
import java.util.ArrayList; import java.util.*;
import java.util.Collection;
import java.util.LinkedHashSet;
import java.util.Map;
import java.util.stream.Collectors; import java.util.stream.Collectors;
/** /**
@@ -169,6 +166,9 @@ public class VariableReferenceInfos {
return PassNCalcVariableReferenceInfos.getReferencedVars(rValue); return PassNCalcVariableReferenceInfos.getReferencedVars(rValue);
} }
/** Caches results for getDefinedVars(). */
private Map<Integer, List<VariableRef>> cachedDefinedVars = new HashMap<>();
/** /**
* Get the variables defined by a statement * Get the variables defined by a statement
* *
@@ -176,17 +176,22 @@ public class VariableReferenceInfos {
* @return Variables defined by the statement * @return Variables defined by the statement
*/ */
public Collection<VariableRef> getDefinedVars(Statement stmt) { public Collection<VariableRef> getDefinedVars(Statement stmt) {
List<VariableRef> result = cachedDefinedVars.get(stmt.getIndex());
if(result!=null)
return result;
Collection<ReferenceToSymbolVar> referenceToSymbolVars = statementVarReferences.get(stmt.getIndex()); Collection<ReferenceToSymbolVar> referenceToSymbolVars = statementVarReferences.get(stmt.getIndex());
// TODO: This may cause problems since it is a sign that the maps are out of date!
if(referenceToSymbolVars == null) if(referenceToSymbolVars == null)
// TODO: This may cause problems since it is a sign that the maps are out of date!
return new ArrayList<>(); return new ArrayList<>();
return referenceToSymbolVars result = referenceToSymbolVars
.stream() .stream()
.filter(referenceToSymbolVar -> referenceToSymbolVar.getReferenced() instanceof VariableRef) .filter(referenceToSymbolVar -> referenceToSymbolVar.getReferenced() instanceof VariableRef)
.filter(referenceToSymbolVar -> ReferenceToSymbolVar.ReferenceType.DEFINE.equals(referenceToSymbolVar.getReferenceType())) .filter(referenceToSymbolVar -> ReferenceToSymbolVar.ReferenceType.DEFINE.equals(referenceToSymbolVar.getReferenceType()))
.map(ReferenceToSymbolVar::getReferenced) .map(ReferenceToSymbolVar::getReferenced)
.map(symbolVariableRef -> (VariableRef) symbolVariableRef) .map(symbolVariableRef -> (VariableRef) symbolVariableRef)
.collect(Collectors.toList()); .collect(Collectors.toList());
cachedDefinedVars.put(stmt.getIndex(), result);
return result;
} }
/** /**