Examples with TaskInfo abs.backend.java.scheduling.SimpleTaskScheduler.TaskInfo used on opensource projects

Example 1 with TaskInfo

use of abs.backend.java.scheduling.SimpleTaskScheduler.TaskInfo in project abstools by abstools.

the class SchedulingTest method testNStepScheduler.

@Test
public void testNStepScheduler() {
    final SchedulingStrategy ss = mock(SchedulingStrategy.class);
    doCallRealMethod().when(ss).setBaseScheduler(any(TotalScheduler.class));
    doCallRealMethod().when(ss).setCurrentScheduler(any(TotalScheduler.class));
    doCallRealMethod().when(ss).doSingleStep();
    TotalScheduler ts = mock(TotalScheduler.class);
    ScheduleAction scheduleAction = mock(ScheduleAction.class, Mockito.RETURNS_DEEP_STUBS);
    when(ts.choose(any(ScheduleOptions.class))).thenReturn(scheduleAction);
    TaskInfo taskInfo = mock(TaskInfo.class);
    when(ts.schedule(any(TaskScheduler.class), any(List.class))).thenReturn(taskInfo);
    ss.setBaseScheduler(ts);
    final int numberSteps = new Random().nextInt(100);
    final NStepScheduler nss = new NStepScheduler(ss, numberSteps);
    ss.setCurrentScheduler(nss);
    Thread schedulingThread = new Thread(new Runnable() {

        @Override
        public void run() {
            ScheduleOptions scheduleOptions = mock(ScheduleOptions.class, Mockito.RETURNS_DEEP_STUBS);
            for (int i = 0; i < numberSteps; i++) {
                ss.curScheduler.choose(scheduleOptions);
            }
            assertEquals(nss, ss.curScheduler);
            ss.curScheduler.choose(scheduleOptions);
        }
    });
    try {
        schedulingThread.start();
        schedulingThread.join();
        verify(ss).awaitGUIAction(any(ScheduleOptions.class));
    } catch (InterruptedException e) {
        fail();
    }
}

Example 2 with TaskInfo

use of abs.backend.java.scheduling.SimpleTaskScheduler.TaskInfo in project abstools by abstools.

the class SchedulingTest method testRunToLineScheduler.

@Test
public void testRunToLineScheduler() {
    final SchedulingStrategy ss = mock(SchedulingStrategy.class);
    doCallRealMethod().when(ss).setBaseScheduler(any(TotalScheduler.class));
    doCallRealMethod().when(ss).setCurrentScheduler(any(TotalScheduler.class));
    doCallRealMethod().when(ss).doSingleStep();
    TotalScheduler ts = mock(TotalScheduler.class);
    ScheduleAction scheduleAction = mock(ScheduleAction.class, Mockito.RETURNS_DEEP_STUBS);
    when(ts.choose(any(ScheduleOptions.class))).thenReturn(scheduleAction);
    TaskInfo taskInfo = mock(TaskInfo.class);
    when(ts.schedule(any(TaskScheduler.class), any(List.class))).thenReturn(taskInfo);
    ss.setBaseScheduler(ts);
    final int line = new Random().nextInt(1000);
    final String fileName = "filename";
    abs.backend.java.debugging.TaskInfo ti = mock(abs.backend.java.debugging.TaskInfo.class);
    when(ti.getCurrentFile()).thenReturn("filename");
    when(ti.getCurrentLine()).thenReturn(line - 5).thenReturn(line - 4).thenReturn(line - 3).thenReturn(line - 2).thenReturn(line - 1).thenReturn(line);
    ss.steppedTask = ti;
    RunToLineScheduler rtls = new RunToLineScheduler(ss, fileName, line);
    ss.setCurrentScheduler(rtls);
    ScheduleOptions scheduleOptions = mock(ScheduleOptions.class, Mockito.RETURNS_DEEP_STUBS);
    for (int i = 0; i < 6; i++) {
        ss.curScheduler.choose(scheduleOptions);
    }
    verify(ss).awaitGUIAction(any(ScheduleOptions.class));
}

Example 3 with TaskInfo

use of abs.backend.java.scheduling.SimpleTaskScheduler.TaskInfo in project abstools by abstools.

the class Cog method setupAPI.

private static void setupAPI() {
    thisClass.setName("Cog");
    thisClass.addMethod(/*List<Process>*/
    "getQueue", new ABSClosure() {

        @Override
        public ABSValue exec(ABSDynamicObject t, ABSValue... params) {
            // TODO
            return ABSUnit.UNIT;
        }
    });
    thisClass.addMethod(/*ABSUnit*/
    "info", new ABSClosure() {

        @Override
        public ABSValue exec(ABSDynamicObject t, ABSValue... params) {
            COG cog = (COG) t.getFieldValue_Internal("cog");
            System.out.println("Cog scheduler " + cog.getScheduler().toString());
            if (cog.getScheduler() instanceof SimpleTaskScheduler) {
                SimpleTaskScheduler sch = (SimpleTaskScheduler) cog.getScheduler();
                TaskSchedulingStrategy strat = sch.getSchedulingStrategy();
                System.out.println("Strategy " + strat);
            }
            return ABSUnit.UNIT;
        }
    });
    thisClass.addMethod(/*ABSUnit*/
    "setScheduler", new ABSClosure() {

        @Override
        public ABSValue exec(ABSDynamicObject t, ABSValue... params) {
            COG cog = (COG) t.getFieldValue_Internal("cog");
            if (!(cog.getScheduler() instanceof SimpleTaskScheduler)) {
                throw new DynamicException("For user-defined scheduling to work, the Scheduler must be an instance of " + SimpleTaskScheduler.class.getName() + " (use \"-taskscheduler=simple\")");
            }
            // The user-defined scheduler
            final ABSDynamicObject userScheduler = (ABSDynamicObject) params[0];
            // Create a new scheduling strategy that invokes the user-defined scheduler's "schedule" method
            // - convert TaskSchedulingStrategy's List<TaskInfo> to ProcessScheduler's List<Process>
            // - convert returned Process to TaskInfo.
            // custom TaskSchedulingStrategy
            TaskSchedulingStrategy strategy = new TaskSchedulingStrategy() {

                @Override
                public synchronized TaskInfo schedule(final TaskScheduler scheduler, final List<TaskInfo> schedulableTasks) {
                    System.out.println("Scheduling (" + schedulableTasks.size() + " processes in queue)...");
                    // Remember TaskInfos based on their Pids to speed things up a little
                    HashMap<Long, TaskInfo> taskMap = new HashMap<>();
                    // Convert List<TaskInfo> to ArrayList<ABSProcess>
                    ArrayList<ABSProcess> processes = new ArrayList<>();
                    for (TaskInfo task : schedulableTasks) {
                        taskMap.put(task.id, task);
                        // TODO set: pid, method, arrival, cost, deadline, start, finish, critical, value
                        ABSProcess proc = new ABSProcess(task.task.getID(), task.task.getCall().methodName(), 0, 0, 0, 0, 0, false, 0);
                        processes.add(proc);
                        System.out.println("\t" + proc.toString());
                    }
                    // FIXME
                    return schedulableTasks.get(0);
                }
            };
            // Connect the strategy to the cog's scheduler
            System.out.println(cog.toString() + ": Setting TaskSchedulingStrategy to: " + userScheduler.getClassName());
            // move the userScheduler object to the Cog
            // this WILL cause problems if the user associates the scheduler with multiple cogs!!
            // userScheduler.setCOG(cog);
            SimpleTaskScheduler scheduler = (SimpleTaskScheduler) cog.getScheduler();
            scheduler.setSchedulingStrategy(strategy);
            return ABSUnit.UNIT;
        }
    });
}

Example 4 with TaskInfo

use of abs.backend.java.scheduling.SimpleTaskScheduler.TaskInfo in project abstools by abstools.

the class SwingWrapperProxy method awaitGUITaskChoose.

private synchronized TaskInfo awaitGUITaskChoose() {
    while (choosenTask == null) {
        try {
            wait();
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
    }
    TaskInfo t = choosenTask;
    choosenTask = null;
    return t;
}

Example 5 with TaskInfo

use of abs.backend.java.scheduling.SimpleTaskScheduler.TaskInfo in project abstools by abstools.

the class RecordingSchedulerStrategy method schedule.

@Override
public TaskInfo schedule(TaskScheduler scheduler, List<TaskInfo> scheduableTasks) {
    int cogId = scheduler.getCOG().getID();
    TaskInfo choosenTask = schedulingStrat.schedule(scheduler, scheduableTasks);
    long taskId = choosenTask.id;
    String suff = scheduableTasks.size() == 1 ? " (NO CHOICE)" : "";
    System.out.println("COG " + cogId + " (" + scheduler.getCOG().getInitialClass().getName() + "): Scheduled task " + taskId + " from {" + tasksToStringList(cogId, scheduableTasks) + "}" + suff);
    return choosenTask;
}