84a12611d1503bf5e1ec788274ebe5b2648d195f
[des2015.git] / dsl / xtend / src / robots / missions / generator / TaskDSLGenerator.xtend
1 /*
2 * generated by Xtext
3 */
4 package robots.missions.generator
5
6 import org.eclipse.emf.common.util.EList
7 import org.eclipse.emf.ecore.resource.Resource
8 import org.eclipse.xtext.generator.IFileSystemAccess
9 import org.eclipse.xtext.generator.IGenerator
10 import robots.missions.taskDSL.Behaviour
11 import robots.missions.taskDSL.Mission
12 import robots.missions.taskDSL.OperatorE
13 import robots.missions.taskDSL.Robot
14 import robots.missions.taskDSL.StoppingExpression
15
16 /**
17 * Generates code from your model files on save.
18 *
19 * See https://www.eclipse.org/Xtext/documentation/303_runtime_concepts.html#code-generation
20 */
21 class TaskDSLGenerator implements IGenerator {
22
23 override void doGenerate(Resource resource, IFileSystemAccess fsa) {
24 var root = resource.allContents.head as Robot;
25 if(root != null){
26 fsa.generateFile("nl/ru/des/Constants.java", makeConstants(root));
27 for(Behaviour b : root.behaviour){
28 fsa.generateFile("nl/ru/des/" + b.name + "Behaviour.java", makeBehaviour(b));
29 }
30 //fsa.generateFile("nl/ru/des/Missions.java", makeMissions(root.mission));
31 }
32 }
33
34 def CharSequence makeMissions(EList<Mission> list)'''
35 package nl.ru.des;
36
37 import java.util.LinkedList;
38
39 import lejos.hardware.motor.EV3LargeRegulatedMotor;
40 import lejos.robotics.subsumption.Behavior;
41 import nl.ru.des.Behaviours;
42
43 public class Missions{
44 public static LinkedList<Mission> getMissions(SensorCollector sensors, EV3LargeRegulatedMotor rightMotor,
45 EV3LargeRegulatedMotor leftMotor, ColorMemory colors){
46 LinkedList<Mission> missions = new LinkedList<Mission>();
47 «FOR m : list»
48 missions.add(new Mission("«m.name»", new Behavior[]{
49 «FOR b : m.behaviours SEPARATOR ","»
50 new Behaviours.«b.name»Behaviour(sensors, rightMotor, leftMotor, colors)
51 «ENDFOR»,
52 new ShutdownBehaviour(sensors, rightMotor, leftMotor, colors){
53 @Override public boolean takeControl(){
54 return »;
55 }
56 }}
57 «ENDFOR»));
58 return missions;
59 }
60 }
61 '''
62
63 def CharSequence makeBehaviour(Behaviour b)'''
64 package nl.ru.des;
65
66 import lejos.robotics.RegulatedMotor;
67 import nl.ru.des.sensors.SensorCollector;
68
69 public class «b.name»Behaviour extends BasicBehaviour {
70 public «b.name»Behaviour(SensorCollector sensors, RegulatedMotor rightMotor,
71 RegulatedMotor leftMotor, RegulatedMotor measMotor){
72 super(sensors, rightMotor, leftMotor, measMotor);
73 }
74 «IF b.tc != null»
75 @Override public boolean takeControl(){
76 return «printExpression(b.tc)»;
77 }
78 «ENDIF»
79
80 @Override public void action(){
81 super.action();
82 «FOR a : b.actions»
83 «IF a.whichMotor != null»
84 «IF a.acc > 0»
85 «a.whichMotor.d.toString()»Motor.setAcceleration(«a.acc»);
86 «a.whichMotor.d.toString()»Motor.setSpeed(«a.spd»);
87 «ENDIF»
88 «a.whichMotor.d.toString()»Motor.«a.moveDir.d.toString()»();
89 «ELSEIF a.measureWhat != null»
90 «a.measureWhat.d.toString()»Measure();
91 «ELSE»
92 time = System.currentTimeMillis();
93 while(!suppressed«IF a.time.time > 0» && System.currentTimeMillis()-time>«a.time.time»«ENDIF»){
94 Thread.yield();
95 }
96 «ENDIF»
97 «ENDFOR»
98 reset();
99 }
100 }
101 '''
102
103 def CharSequence makeConstants(Robot robot)'''
104 package nl.ru.des;
105
106 public class Constants{
107 public final static int speed = «robot.spd»;
108 public final static int acceleration = «robot.acc»;
109 }'''
110
111 def CharSequence printExpression(StoppingExpression e)'''
112 «IF e.scond != null»
113 «IF !e.scond.colors.nullOrEmpty»
114 sensors.collected(new int[]{«FOR c : e.scond.colors SEPARATOR ","»«c.d.ordinal»«ENDFOR»})
115 «ELSEIF e.scond.touch != null»
116 sensors.«e.scond.touch.d.toString()»Touch()
117 «ELSEIF e.scond.dist != null»
118 sensors.distance() «e.scond.dist.d.toString()» «e.scond.dist»
119 «ELSEIF e.scond.color != null»
120 sensors.color() == «e.scond.color.d.ordinal»
121 «ENDIF»
122 «ELSE»
123 «IF e.op.d.equals(OperatorE.AND)»
124 «FOR ex : e.s BEFORE "(" SEPARATOR "&&" AFTER ")"»«printExpression(ex)»«ENDFOR»
125 «ELSE»
126 «FOR ex : e.s BEFORE "(" SEPARATOR "&&" AFTER ")"»«printExpression(ex)»«ENDFOR»
127 «ENDIF»
128 «ENDIF»
129 '''
130 }