1 package Analysis.Scheduling;
4 import java.io.FileOutputStream;
5 import java.io.PrintWriter;
6 import java.util.Collection;
7 import java.util.Enumeration;
8 import java.util.HashSet;
9 import java.util.Hashtable;
10 import java.util.Iterator;
12 import java.util.Vector;
13 import java.util.Map.Entry;
15 import Analysis.Scheduling.ScheduleSimulator.Action;
16 import Analysis.Scheduling.ScheduleSimulator.CheckPoint;
17 import Analysis.TaskStateAnalysis.Allocations;
18 import Analysis.TaskStateAnalysis.FEdge;
19 import Analysis.TaskStateAnalysis.FlagState;
20 import Analysis.TaskStateAnalysis.FEdge.NewObjInfo;
21 import IR.ClassDescriptor;
23 import IR.Tree.FlagExpressionNode;
24 import IR.Tree.FlagNode;
25 import IR.Tree.FlagOpNode;
27 import Util.GraphNode;
30 public class SchedulingUtil {
32 /*public static int maxDivisor(int l, int r) {
44 if(((a&1)==0) && ((b&1)==0)) {
45 // a and b are both even
49 } else if(((a&1)==0) && ((b&1)!=0)) {
50 // a is even, b is odd
52 } else if (((a&1)!=0) && ((b&1)==0)) {
53 // a is odd, b is even
55 } else if (((a&1)!=0) && ((b&1)!=0)) {
56 // a and b are both odd
58 a = a>b ? (a-b):(b-a);
64 public static boolean isTaskTrigger_flag(FlagExpressionNode fen,FlagState fs) {
67 else if (fen instanceof FlagNode)
68 return fs.get(((FlagNode)fen).getFlag());
70 switch (((FlagOpNode)fen).getOp().getOp()) {
71 case Operation.LOGIC_AND:
72 return ((isTaskTrigger_flag(((FlagOpNode)fen).getLeft(),fs)) && (isTaskTrigger_flag(((FlagOpNode)fen).getRight(),fs)));
73 case Operation.LOGIC_OR:
74 return ((isTaskTrigger_flag(((FlagOpNode)fen).getLeft(),fs)) || (isTaskTrigger_flag(((FlagOpNode)fen).getRight(),fs)));
75 case Operation.LOGIC_NOT:
76 return !(isTaskTrigger_flag(((FlagOpNode)fen).getLeft(),fs));
82 public static void printScheduleGraph(String path, Vector<ScheduleNode> sNodes) {
84 File file=new File(path);
85 FileOutputStream dotstream=new FileOutputStream(file,false);
86 PrintWriter output = new java.io.PrintWriter(dotstream, true);
87 output.println("digraph G {");
88 output.println("\tcompound=true;\n");
89 traverseSNodes(output, sNodes);
90 output.println("}\n");
92 } catch (Exception e) {
98 private static void traverseSNodes(PrintWriter output, Vector<ScheduleNode> sNodes){
99 //Draw clusters representing ScheduleNodes
100 Iterator it = sNodes.iterator();
101 while (it.hasNext()) {
102 ScheduleNode gn = (ScheduleNode) it.next();
103 Iterator edges = gn.edges();
104 output.println("\tsubgraph " + gn.getLabel() + "{");
105 output.println("\t\tlabel=\"" + gn.getTextLabel() + "\";");
106 Iterator it_cnodes = gn.getClassNodesIterator();
107 traverseCNodes(output, it_cnodes);
108 //Draw the internal 'new' edges
109 Iterator it_edges =gn.getScheduleEdgesIterator();
110 while(it_edges.hasNext()) {
111 ScheduleEdge se = (ScheduleEdge)it_edges.next();
113 if(se.getSourceCNode().isclone()) {
114 output.print(se.getSourceCNode().getLabel());
116 if(se.getSourceFState() == null) {
117 output.print(se.getSourceCNode().getClusterLabel());
119 output.print(se.getSourceFState().getLabel());
123 output.print(" -> ");
125 if(se.getTargetCNode().isclone()) {
126 output.print(se.getTargetCNode().getLabel());
128 output.print(se.getTargetCNode().getClusterLabel());
130 output.println(" [label=\"" + se.getLabel() + "\", color=red];");
132 output.print(se.getTargetFState().getLabel() + " [label=\"" + se.getLabel() + "\", color=red, ltail=");
133 if(se.getSourceCNode().isclone()) {
134 output.println(se.getSourceCNode().getLabel() + "];");
136 output.println(se.getSourceCNode().getClusterLabel() + "];");
140 output.println("\t}\n");
141 //Draw 'new' edges of this ScheduleNode
142 while(edges.hasNext()) {
143 ScheduleEdge se = (ScheduleEdge)edges.next();
145 if(se.getSourceCNode().isclone()) {
146 output.print(se.getSourceCNode().getLabel());
148 if(se.getSourceFState() == null) {
149 output.print(se.getSourceCNode().getClusterLabel());
151 output.print(se.getSourceFState().getLabel());
155 output.print(" -> ");
157 if(se.getTargetCNode().isclone()) {
158 output.print(se.getTargetCNode().getLabel());
160 output.print(se.getTargetCNode().getClusterLabel());
162 output.println(" [label=\"" + se.getLabel() + "\", color=red, style=dashed];");
164 output.println(se.getTargetFState().getLabel() + " [label=\"" + se.getLabel() + "\", color=red, style=dashed];");
170 private static void traverseCNodes(PrintWriter output, Iterator it){
171 //Draw clusters representing ClassNodes
172 while (it.hasNext()) {
173 ClassNode gn = (ClassNode) it.next();
175 output.println("\t\t" + gn.getLabel() + " [style=dashed, label=\"" + gn.getTextLabel() + "\", shape=box];");
177 output.println("\tsubgraph " + gn.getClusterLabel() + "{");
178 output.println("\t\tstyle=dashed;");
179 output.println("\t\tlabel=\"" + gn.getTextLabel() + "\";");
180 traverseFlagStates(output, gn.getFlagStates());
181 output.println("\t}\n");
186 private static void traverseFlagStates(PrintWriter output, Collection nodes) {
187 Set cycleset=GraphNode.findcycles(nodes);
188 Vector namers=new Vector();
189 namers.add(new Namer());
190 namers.add(new Allocations());
192 Iterator it = nodes.iterator();
193 while (it.hasNext()) {
194 GraphNode gn = (GraphNode) it.next();
195 Iterator edges = gn.edges();
197 String dotnodeparams="";
199 for(int i=0;i<namers.size();i++) {
200 Namer name=(Namer) namers.get(i);
201 String newlabel=name.nodeLabel(gn);
202 String newparams=name.nodeOption(gn);
204 if (!newlabel.equals("") && !label.equals("")) {
207 if (!newparams.equals("")) {
208 dotnodeparams+=", " + name.nodeOption(gn);
210 label+=name.nodeLabel(gn);
212 label += ":[" + ((FlagState)gn).getExeTime() + "]";
215 output.println("\t" + gn.getLabel() + " [label=\"" + label + "\"" + dotnodeparams + "];");
218 while (edges.hasNext()) {
219 Edge edge = (Edge) edges.next();
220 GraphNode node = edge.getTarget();
221 if (nodes.contains(node)) {
222 for(Iterator nodeit=nonmerge(node, nodes).iterator();nodeit.hasNext();) {
223 GraphNode node2=(GraphNode)nodeit.next();
224 String edgelabel = "";
225 String edgedotnodeparams="";
227 for(int i=0;i<namers.size();i++) {
228 Namer name=(Namer) namers.get(i);
229 String newlabel=name.edgeLabel(edge);
230 String newoption=name.edgeOption(edge);
231 if (!newlabel.equals("")&& ! edgelabel.equals(""))
234 if (!newoption.equals(""))
235 edgedotnodeparams+=", "+newoption;
237 edgelabel+=":[" + ((FEdge)edge).getExeTime() + "]";
238 edgelabel+=":(" + ((FEdge)edge).getProbability() + "%)";
239 Hashtable<ClassDescriptor, NewObjInfo> hashtable = ((FEdge)edge).getNewObjInfoHashtable();
240 if(hashtable != null) {
241 Set<ClassDescriptor> keys = hashtable.keySet();
242 Iterator it_keys = keys.iterator();
243 while(it_keys.hasNext()) {
244 ClassDescriptor cd = (ClassDescriptor)it_keys.next();
245 NewObjInfo noi = hashtable.get(cd);
246 edgelabel += ":{ class " + cd.getSymbol() + " | " + noi.getNewRate() + " | (" + noi.getProbability() + "%) }";
249 output.println("\t" + gn.getLabel() + " -> " + node2.getLabel() + " [" + "label=\"" + edgelabel + "\"" + edgedotnodeparams + "];");
256 private static Set nonmerge(GraphNode gn, Collection nodes) {
257 HashSet newset=new HashSet();
258 HashSet toprocess=new HashSet();
260 while(!toprocess.isEmpty()) {
261 GraphNode gn2=(GraphNode)toprocess.iterator().next();
262 toprocess.remove(gn2);
266 Iterator edges = gn2.edges();
267 while (edges.hasNext()) {
268 Edge edge = (Edge) edges.next();
269 GraphNode node = edge.getTarget();
270 if (!newset.contains(node)&&nodes.contains(node))
278 public static void printSimulationResult(String path, int time, int coreNum, Vector<CheckPoint> checkpoints) {
280 File file=new File(path);
281 FileOutputStream dotstream=new FileOutputStream(file,false);
282 PrintWriter output = new java.io.PrintWriter(dotstream, true);
283 output.println("digraph simulation{");
285 output.println("node [shape=plaintext];");
287 output.println("edge [dir=none];");
289 output.println("ranksep=.05;");
295 output.print("{rank=source; \"Time\"; ");
296 for(j = 0; j < coreNum; j++) {
297 output.print("\"core " + j + "\"; ");
300 // time coordinate nodes
301 Vector<String> timeNodes = new Vector<String>();
302 String[] lastTaskNodes = new String[coreNum];
303 String[] lastTasks = new String[coreNum];
304 boolean[] isTaskFinish = new boolean[coreNum];
305 for(j = 0; j < coreNum; j++) {
306 lastTaskNodes[j] = "first";
307 isTaskFinish[j] = true;
311 for(j = 0; j < checkpoints.size(); j++) {
312 CheckPoint tcp = checkpoints.elementAt(j);
313 Hashtable<Integer, String> tmplastTasks = new Hashtable<Integer, String>();
314 Vector<Integer> tmpisTaskFinish = new Vector<Integer>();
315 Vector<Integer> tmpisset = new Vector<Integer>();
316 String tnode = String.valueOf(tcp.getTimepoint());
317 if(!timeNodes.contains(tnode)) {
318 timeNodes.add(tnode);
320 Vector<Action> actions = tcp.getActions();
321 Hashtable<String, StringBuffer> tmpTaskNodes = new Hashtable<String, StringBuffer>();
322 for(int i = 0; i < actions.size(); i++) {
323 Action taction = actions.elementAt(i);
324 int cNum = taction.getCoreNum();
325 if(!tmplastTasks.contains(cNum)) {
326 tmplastTasks.put(cNum, lastTasks[cNum]);
328 if(!(tmpisset.contains(cNum)) && (isTaskFinish[cNum]) && !(tmpisTaskFinish.contains(cNum))) {
329 tmpisTaskFinish.add(cNum); // records those with task finished the first time visit it
331 String tmpTaskNode = "\"" + tnode + "core" + cNum + "\"";
332 StringBuffer tmpLabel = null;
333 boolean isfirst = false;
334 if(!tmpTaskNodes.containsKey(tmpTaskNode)) {
335 tmpTaskNodes.put(tmpTaskNode, new StringBuffer(tnode + ":"));
338 tmpLabel = tmpTaskNodes.get(tmpTaskNode);
339 switch(taction.getType()){
340 case Action.ADDOBJ: {
342 tmpLabel.append("\\n");
344 tmpLabel.append("(" + taction.getTransObj().getSymbol() + ")arrives;");
345 if(!(lastTaskNodes[cNum].equals(tmpTaskNode))) {
347 if(lastTaskNodes[cNum].equals("first")) {
348 output.print("\"core " + cNum + "\"->" + tmpTaskNode);
350 output.print(lastTaskNodes[cNum] + "->" + tmpTaskNode);
352 if(tmpisTaskFinish.contains(cNum)) {
353 output.print(" [style=invis]");
356 lastTaskNodes[cNum] = tmpTaskNode;
360 case Action.TASKFINISH: {
362 tmpLabel.append("\\n");
364 tmpLabel.append("<" + taction.getTd().getSymbol() + ">finishes;");
365 if(!(lastTaskNodes[cNum].equals("first"))) {
366 if(!(lastTaskNodes[cNum].equals(tmpTaskNode))) {
368 output.println(lastTaskNodes[cNum] + "->" + tmpTaskNode + ";");
369 lastTaskNodes[cNum] = tmpTaskNode;
371 if(tmpisset.contains(cNum)) {
372 isTaskFinish[cNum] &= true;
374 isTaskFinish[cNum] = true;
377 lastTasks[cNum] = "";
379 throw new Exception("Error: unexpected task finish");
383 case Action.TFWITHOBJ: {
385 tmpLabel.append("\\n");
387 tmpLabel.append("<" + taction.getTd().getSymbol() + ">finishes; ");
388 Iterator<Entry<ClassDescriptor, Integer>> it_entry = (Iterator<Entry<ClassDescriptor, Integer>>)taction.getNObjs().entrySet().iterator();
389 while(it_entry.hasNext()) {
390 Entry<ClassDescriptor, Integer> entry = it_entry.next();
391 tmpLabel.append(entry.getValue() + "(" + entry.getKey().getSymbol() + ")");
392 if(it_entry.hasNext()) {
393 tmpLabel.append(",");
395 tmpLabel.append(";");
398 if(!(lastTaskNodes[cNum].equals("first"))) {
399 if (!(lastTaskNodes[cNum].equals(tmpTaskNode))) {
401 output.println(lastTaskNodes[cNum] + "->" + tmpTaskNode + ";");
402 lastTaskNodes[cNum] = tmpTaskNode;
404 if(tmpisset.contains(cNum)) {
405 isTaskFinish[cNum] &= true;
407 isTaskFinish[cNum] = true;
410 lastTasks[cNum] = "";
412 throw new Exception("Error: unexpected task finish");
416 case Action.TASKSTART: {
418 tmpLabel.append("\\n");
420 tmpLabel.append("<" + taction.getTd().getSymbol() + ">starts;");
421 lastTasks[cNum] = taction.getTd().getSymbol();
423 if (!(lastTaskNodes[cNum].equals(tmpTaskNode))) {
425 if(lastTaskNodes[cNum].equals("first")) {
426 output.print("\"core " + cNum + "\"->" + tmpTaskNode);
428 output.print(lastTaskNodes[cNum] + "->" + tmpTaskNode);
430 if(tmpisTaskFinish.contains(cNum)) {
431 output.print(" [style=invis]");
434 lastTaskNodes[cNum] = tmpTaskNode;
436 isTaskFinish[cNum] &= false;
439 case Action.TASKABORT: {
441 tmpLabel.append("\\n");
443 tmpLabel.append("<" + taction.getTd().getSymbol() + ">aborts;");
444 if(!(lastTaskNodes[cNum].equals("first")) &&
445 (tmplastTasks.get(cNum).equals(taction.getTd().getSymbol()))) {
446 if(!(lastTaskNodes[cNum].equals(tmpTaskNode))) {
448 output.println(lastTaskNodes[cNum] + "->" + tmpTaskNode + ";");
449 lastTaskNodes[cNum] = tmpTaskNode;
451 if(tmpisset.contains(cNum)) {
452 isTaskFinish[cNum] &= true;
454 isTaskFinish[cNum] = true;
457 lastTasks[cNum] = "";
459 throw new Exception("Error: unexpected task aborts");
463 case Action.TASKREMOVE: {
465 tmpLabel.append("\\n");
467 tmpLabel.append("<" + taction.getTd().getSymbol() + ">removes;");
468 if(!(lastTaskNodes[cNum].equals("first")) &&
469 (tmplastTasks.get(cNum).equals(taction.getTd().getSymbol()))) {
470 if(!(lastTaskNodes[cNum].equals(tmpTaskNode))) {
472 output.println(lastTaskNodes[cNum] + "->" + tmpTaskNode + ";");
473 lastTaskNodes[cNum] = tmpTaskNode;
475 if(tmpisset.contains(cNum)) {
476 isTaskFinish[cNum] &= true;
478 isTaskFinish[cNum] = true;
481 lastTasks[cNum] = "";
483 throw new Exception("Error: unexpected task remove");
489 Enumeration<String> keys = tmpTaskNodes.keys();
490 while(keys.hasMoreElements()) {
491 String tmpTaskNode = keys.nextElement();
493 output.println(tmpTaskNode + "[label=\"" + tmpTaskNodes.get(tmpTaskNode).toString() + "\"]");
496 output.print("{rank=same; rankdir=LR; " + tnode + "; ");
497 keys = tmpTaskNodes.keys();
498 while(keys.hasMoreElements()) {
499 String tmpTaskNode = keys.nextElement();
500 output.print(tmpTaskNode);
508 output.println("\"Time\"->" + timeNodes.elementAt(0) + "[style=invis];");
509 for(j = 0; j < time; j++) {
510 output.print(j + "->");
512 output.println(timeNodes.lastElement() + ";");
515 } catch (Exception e) {