rairlab/Spectra
1
0
Fork 0
mirror of https://github.com/RAIRLab/Spectra.git synced 2025-11-02 23:01:20 +00:00
Code Issues Projects Releases Packages Wiki Activity

clean up tests

Browse source
This commit is contained in:
Naveen Sundar Govindarajulu 2021-01-04 15:24:49 -08:00
parent eb31602b35
commit 344824aae8
9 changed files with 252 additions and 37 deletions
Download patch file Download diff file Expand all files Collapse all files

6
src/main/java/com/naveensundarg/planner/DepthFirstPlanner.java
View file

@ -19,8 +19,8 @@ import java.util.stream.Collectors;
public class DepthFirstPlanner implements Planner {
private static int MAX_DEPTH = 6;
private static boolean EXHAUSTIVE_TILL_MAX_DEPTH = false;
private static int MAX_DEPTH = 7;
private static boolean EXHAUSTIVE_TILL_MAX_DEPTH = true;
private boolean USE_METHODS, WORK_FROM_SCRATCH;
@ -203,7 +203,7 @@ public class DepthFirstPlanner implements Planner {
current.removeAll(current.stream().filter(u-> deletions.stream().anyMatch(d-> Operations.equivalent(background, d, u))).collect(Collectors.toSet()));
current.removeAll(deletions);
current.addAll(additions.stream().map(Simplifier::simplify).collect(Collectors.toSet()));
current.addAll(additions.stream().collect(Collectors.toSet()));
expectedStates.add(State.initializeWith(current));

34
src/main/java/com/naveensundarg/planner/utils/Sandbox.java
View file

@ -1,12 +1,13 @@
package com.naveensundarg.planner.utils;
import com.naveensundarg.planner.DepthFirstPlanner;
import com.naveensundarg.planner.PlanMethod;
import com.naveensundarg.planner.Planner;
import com.naveensundarg.planner.Simplifier;
import com.diogonunes.jcdp.color.ColoredPrinter;
import com.diogonunes.jcdp.color.api.Ansi;
import com.naveensundarg.planner.*;
import com.naveensundarg.shadow.prover.representations.formula.Predicate;
import java.util.Comparator;
import java.util.List;
import java.util.stream.Collectors;
import static com.naveensundarg.shadow.prover.utils.Reader.readFormulaFromString;
@ -44,21 +45,38 @@ public class Sandbox {
}
static ColoredPrinter cp = new ColoredPrinter.Builder(1, false).build();
public static void main(String[] args) throws com.naveensundarg.shadow.prover.utils.Reader.ParsingException {
List<PlanningProblem> planningProblemList = (PlanningProblem.readFromFile(Sandbox.class.getResourceAsStream("../problems/tora/attend.clj")));
List<PlanningProblem> planningProblemList = (PlanningProblem.readFromFile(Sandbox.class.getResourceAsStream("../problems/ai2thor/FloorPlan28.clj")));
Planner depthFirstPlanner = new DepthFirstPlanner();
PlanningProblem planningProblem = planningProblemList.stream().filter(problem -> problem.getName().equals("soda can challenge 2")).findFirst().get();
PlanningProblem planningProblem = planningProblemList.stream().filter(problem -> problem.getName().equals("FloorPlan28")).findFirst().get();
depthFirstPlanner.plan(planningProblem.getBackground(), planningProblem.getActions(), planningProblem.getStart(), planningProblem.getGoal()).ifPresent(plans-> {
plans.stream().forEach(System.out::println);
});
// System.out.println(plans);
List<Plan> plansList = plans.stream().sorted(Comparator.comparing(plan -> plan.getActions().size())).collect(Collectors.toList());
if(!plansList.isEmpty()) {
System.out.println("***************************");
cp.setAttribute(Ansi.Attribute.BOLD);
cp.println("PLAN FOUND");
cp.clear();
System.out.println("------------------------------");
cp.setForegroundColor(Ansi.FColor.BLACK);
cp.setBackgroundColor(Ansi.BColor.GREEN); //setting format
cp.println(plansList.get(0));
}
});