Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
0bf0d2503d374b3c249e9e946c13259d5b816ac0
[simgrid.git] / src / surf / cpu_ti.cpp
1 /* Copyright (c) 2013-2015. The SimGrid Team.
2  * All rights reserved.                                                     */
3
4 /* This program is free software; you can redistribute it and/or modify it
5  * under the terms of the license (GNU LGPL) which comes with this package. */
6
7 #include "cpu_ti.hpp"
8 #include "xbt/heap.h"
9 #include "src/surf/trace_mgr.hpp"
10
11 #ifndef SURF_MODEL_CPUTI_H_
12 #define SURF_MODEL_CPUTI_H_
13
14 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(surf_cpu_ti, surf_cpu,
15                                 "Logging specific to the SURF CPU TRACE INTEGRATION module");
16
17 namespace simgrid {
18 namespace surf {
19   
20 static inline
21 void cpu_ti_action_update_index_heap(void *action, int i)
22 {
23   ((simgrid::surf::CpuTiAction*)action)->updateIndexHeap(i);
24 }
25
26 /*********
27  * Trace *
28  *********/
29
30 CpuTiTrace::CpuTiTrace(tmgr_trace_t speedTrace)
31 {
32   s_tmgr_event_t val;
33   unsigned int cpt;
34   double integral = 0;
35   double time = 0;
36   int i = 0;
37   timePoints_ = (double*) xbt_malloc0(sizeof(double) *
38                   (xbt_dynar_length(speedTrace->event_list) + 1));
39   integral_ = (double*) xbt_malloc0(sizeof(double) *
40                   (xbt_dynar_length(speedTrace->event_list) + 1));
41   nbPoints_ = xbt_dynar_length(speedTrace->event_list) + 1;
42   xbt_dynar_foreach(speedTrace->event_list, cpt, val) {
43     timePoints_[i] = time;
44     integral_[i] = integral;
45     integral += val.delta * val.value;
46     time += val.delta;
47     i++;
48   }
49   timePoints_[i] = time;
50   integral_[i] = integral;
51 }
52
53 CpuTiTrace::~CpuTiTrace()
54 {
55   xbt_free(timePoints_);
56   xbt_free(integral_);
57 }
58
59 CpuTiTgmr::~CpuTiTgmr()
60 {
61   if (trace_)
62     delete trace_;
63 }
64
65 /**
66 * \brief Integrate trace
67 *
68 * Wrapper around surf_cpu_integrate_trace_simple() to get
69 * the cyclic effect.
70 *
71 * \param trace Trace structure.
72 * \param a      Begin of interval
73 * \param b      End of interval
74 * \return the integrate value. -1 if an error occurs.
75 */
76 double CpuTiTgmr::integrate(double a, double b)
77 {
78   double first_chunk;
79   double middle_chunk;
80   double last_chunk;
81   int a_index, b_index;
82
83   if ((a < 0.0) || (a > b)) {
84     XBT_CRITICAL
85         ("Error, invalid integration interval [%.2f,%.2f]. You probably have a task executing with negative computation amount. Check your code.",
86          a, b);
87     xbt_abort();
88   }
89   if (a == b)
90     return 0.0;
91
92   if (type_ == TRACE_FIXED) {
93     return ((b - a) * value_);
94   }
95
96   if (ceil(a / lastTime_) == a / lastTime_)
97     a_index = 1 + (int) (ceil(a / lastTime_));
98   else
99     a_index = (int) (ceil(a / lastTime_));
100
101   b_index = (int) (floor(b / lastTime_));
102
103   if (a_index > b_index) {      /* Same chunk */
104     return trace_->integrateSimple(a - (a_index -
105                                               1) * lastTime_,
106                                          b -
107                                          (b_index) *
108                                          lastTime_);
109   }
110
111   first_chunk = trace_->integrateSimple(a - (a_index -
112                                                    1) *
113                                               lastTime_,
114                                               lastTime_);
115   middle_chunk = (b_index - a_index) * total_;
116   last_chunk = trace_->integrateSimple(0.0,
117                                              b -
118                                              (b_index) *
119                                              lastTime_);
120
121   XBT_DEBUG("first_chunk=%.2f  middle_chunk=%.2f  last_chunk=%.2f\n",
122          first_chunk, middle_chunk, last_chunk);
123
124   return (first_chunk + middle_chunk + last_chunk);
125 }
126
127 /**
128  * \brief Auxiliary function to compute the integral between a and b.
129  *     It simply computes the integrals at point a and b and returns the difference between them.
130  * \param a  Initial point
131  * \param b  Final point
132 */
133 double CpuTiTrace::integrateSimple(double a, double b)
134 {
135   return integrateSimplePoint(b) - integrateSimplePoint(a);
136 }
137
138 /**
139  * \brief Auxiliary function to compute the integral at point a.
140  * \param a        point
141  */
142 double CpuTiTrace::integrateSimplePoint(double a)
143 {
144   double integral = 0;
145   int ind;
146   double a_aux = a;
147   ind = binarySearch(timePoints_, a, 0, nbPoints_ - 1);
148   integral += integral_[ind];
149   XBT_DEBUG("a %f ind %d integral %f ind + 1 %f ind %f time +1 %f time %f",
150        a, ind, integral, integral_[ind + 1], integral_[ind],
151        timePoints_[ind + 1], timePoints_[ind]);
152   double_update(&a_aux, timePoints_[ind], sg_maxmin_precision*sg_surf_precision);
153   if (a_aux > 0)
154     integral +=((integral_[ind + 1] -
155           integral_[ind]) / (timePoints_[ind + 1] - timePoints_[ind])) * (a - timePoints_[ind]);
156   XBT_DEBUG("Integral a %f = %f", a, integral);
157
158   return integral;
159 }
160
161 /**
162 * \brief Computes the time needed to execute "amount" on cpu.
163 *
164 * Here, amount can span multiple trace periods
165 *
166 * \param a        Initial time
167 * \param amount  Amount to be executed
168 * \return  End time
169 */
170 double CpuTiTgmr::solve(double a, double amount)
171 {
172   int quotient;
173   double reduced_b;
174   double reduced_amount;
175   double reduced_a;
176   double b;
177
178 /* Fix very small negative numbers */
179   if ((a < 0.0) && (a > -EPSILON)) {
180     a = 0.0;
181   }
182   if ((amount < 0.0) && (amount > -EPSILON)) {
183     amount = 0.0;
184   }
185
186 /* Sanity checks */
187   if ((a < 0.0) || (amount < 0.0)) {
188     XBT_CRITICAL
189         ("Error, invalid parameters [a = %.2f, amount = %.2f]. You probably have a task executing with negative computation amount. Check your code.",
190          a, amount);
191     xbt_abort();
192   }
193
194 /* At this point, a and amount are positive */
195
196   if (amount < EPSILON)
197     return a;
198
199 /* Is the trace fixed ? */
200   if (type_ == TRACE_FIXED) {
201     return (a + (amount / value_));
202   }
203
204   XBT_DEBUG("amount %f total %f", amount, total_);
205 /* Reduce the problem to one where amount <= trace_total */
206   quotient = (int) (floor(amount / total_));
207   reduced_amount = (total_) * ((amount / total_) -
208                                      floor(amount / total_));
209   reduced_a = a - (lastTime_) * (int) (floor(a / lastTime_));
210
211   XBT_DEBUG("Quotient: %d reduced_amount: %f reduced_a: %f", quotient,
212          reduced_amount, reduced_a);
213
214 /* Now solve for new_amount which is <= trace_total */
215 /*
216    fprintf(stderr,"reduced_a = %.2f\n",reduced_a);
217    fprintf(stderr,"reduced_amount = %.2f\n",reduced_amount);
218  */
219   reduced_b = solveSomewhatSimple(reduced_a, reduced_amount);
220
221 /* Re-map to the original b and amount */
222   b = (lastTime_) * (int) (floor(a / lastTime_)) +
223       (quotient * lastTime_) + reduced_b;
224   return b;
225 }
226
227 /**
228 * \brief Auxiliary function to solve integral
229 *
230 * Here, amount is <= trace->total
231 * and a <=trace->last_time
232 *
233 */
234 double CpuTiTgmr::solveSomewhatSimple(double a, double amount)
235 {
236   double amount_till_end;
237   double b;
238
239   XBT_DEBUG("Solve integral: [%.2f, amount=%.2f]", a, amount);
240   amount_till_end = integrate(a, lastTime_);
241 /*
242    fprintf(stderr,"amount_till_end=%.2f\n",amount_till_end);
243  */
244
245   if (amount_till_end > amount) {
246     b = trace_->solveSimple(a, amount);
247   } else {
248     b = lastTime_ + trace_->solveSimple(0.0, amount - amount_till_end);
249   }
250   return b;
251 }
252
253 /**
254  * \brief Auxiliary function to solve integral.
255  *  It returns the date when the requested amount of flops is available
256  * \param trace    Trace structure
257  * \param a        Initial point
258  * \param amount  Amount of flops
259  * \return The date when amount is available.
260 */
261 double CpuTiTrace::solveSimple(double a, double amount)
262 {
263   double integral_a;
264   int ind;
265   double time;
266   integral_a = integrateSimplePoint(a);
267   ind = binarySearch(integral_, integral_a + amount, 0, nbPoints_ - 1);
268   time = timePoints_[ind];
269   time +=
270       (integral_a + amount -
271        integral_[ind]) / ((integral_[ind + 1] -
272                                  integral_[ind]) /
273                                 (timePoints_[ind + 1] -
274                                  timePoints_[ind]));
275
276   return time;
277 }
278
279 /**
280 * \brief Auxiliary function to update the CPU speed scale.
281 *
282 *  This function uses the trace structure to return the speed scale at the determined time a.
283 * \param trace    Trace structure to search the updated speed scale
284 * \param a        Time
285 * \return CPU speed scale
286 */
287 double CpuTiTgmr::getPowerScale(double a)
288 {
289   double reduced_a;
290   int point;
291   s_tmgr_event_t val;
292
293   reduced_a = a - floor(a / lastTime_) * lastTime_;
294   point = trace_->binarySearch(trace_->timePoints_, reduced_a, 0,
295                                 trace_->nbPoints_ - 1);
296   xbt_dynar_get_cpy(speedTrace_->event_list, point, &val);
297   return val.value;
298 }
299
300 /**
301 * \brief Creates a new integration trace from a tmgr_trace_t
302 *
303 * \param  speedTrace    CPU availability trace
304 * \param  value          Percentage of CPU speed available (useful to fixed tracing)
305 * \param  spacing        Initial spacing
306 * \return  Integration trace structure
307 */
308 CpuTiTgmr::CpuTiTgmr(tmgr_trace_t speedTrace, double value)
309 {
310   double total_time = 0.0;
311   s_tmgr_event_t val;
312   unsigned int cpt;
313   trace_ = 0;
314
315 /* no availability file, fixed trace */
316   if (!speedTrace) {
317     type_ = TRACE_FIXED;
318     value_ = value;
319     XBT_DEBUG("No availability trace. Constant value = %f", value);
320     return;
321   }
322
323   /* only one point available, fixed trace */
324   if (xbt_dynar_length(speedTrace->event_list) == 1) {
325     xbt_dynar_get_cpy(speedTrace->event_list, 0, &val);
326     type_ = TRACE_FIXED;
327     value_ = val.value;
328     return;
329   }
330
331   type_ = TRACE_DYNAMIC;
332   speedTrace_ = speedTrace;
333
334   /* count the total time of trace file */
335   xbt_dynar_foreach(speedTrace->event_list, cpt, val) {
336     total_time += val.delta;
337   }
338   trace_ = new CpuTiTrace(speedTrace);
339   lastTime_ = total_time;
340   total_ = trace_->integrateSimple(0, total_time);
341
342   XBT_DEBUG("Total integral %f, last_time %f ",
343             total_, lastTime_);
344 }
345
346 /**
347  * \brief Binary search in array.
348  *  It returns the first point of the interval in which "a" is.
349  * \param array    Array
350  * \param a        Value to search
351  * \param low     Low bound to search in array
352  * \param high    Upper bound to search in array
353  * \return Index of point
354 */
355 int CpuTiTrace::binarySearch(double *array, double a, int low, int high)
356 {
357   xbt_assert(low < high, "Wrong parameters: low (%d) should be smaller than"
358       " high (%d)", low, high);
359
360   int mid;
361   do {
362     mid = low + (high - low) / 2;
363     XBT_DEBUG("a %f low %d high %d mid %d value %f", a, low, high, mid,
364         array[mid]);
365
366     if (array[mid] > a)
367       high = mid;
368     else
369       low = mid;
370   }
371   while (low < high - 1);
372
373   return low;
374 }
375
376 }
377 }
378
379 /*********
380  * Model *
381  *********/
382
383 void surf_cpu_model_init_ti()
384 {
385   xbt_assert(!surf_cpu_model_pm,"CPU model already initialized. This should not happen.");
386   xbt_assert(!surf_cpu_model_vm,"CPU model already initialized. This should not happen.");
387
388   surf_cpu_model_pm = new simgrid::surf::CpuTiModel();
389   xbt_dynar_push(all_existing_models, &surf_cpu_model_pm);
390
391   surf_cpu_model_vm = new simgrid::surf::CpuTiModel();
392   xbt_dynar_push(all_existing_models, &surf_cpu_model_vm);
393 }
394
395 namespace simgrid {
396 namespace surf {
397
398 CpuTiModel::CpuTiModel() : CpuModel()
399 {
400   runningActionSetThatDoesNotNeedBeingChecked_ = new ActionList();
401
402   modifiedCpu_ = new CpuTiList();
403
404   tiActionHeap_ = xbt_heap_new(8, NULL);
405   xbt_heap_set_update_callback(tiActionHeap_,
406                                cpu_ti_action_update_index_heap);
407 }
408
409 CpuTiModel::~CpuTiModel()
410 {
411   surf_cpu_model_pm = NULL;
412   delete runningActionSetThatDoesNotNeedBeingChecked_;
413   delete modifiedCpu_;
414   xbt_heap_free(tiActionHeap_);
415 }
416
417 Cpu *CpuTiModel::createCpu(simgrid::s4u::Host *host,
418     xbt_dynar_t speedPeak,
419     tmgr_trace_t speedTrace,
420     int core)
421 {
422   xbt_assert(core==1,"Multi-core not handled with this model yet");
423   xbt_assert(xbt_dynar_getfirst_as(speedPeak, double) > 0.0,
424       "Speed has to be >0.0. Did you forget to specify the mandatory speed attribute?");
425   return new CpuTi(this, host, speedPeak, speedTrace, core);
426 }
427
428 double CpuTiModel::next_occuring_event(double now)
429 {
430   double min_action_duration = -1;
431
432 /* iterates over modified cpus to update share resources */
433   for(CpuTiList::iterator it(modifiedCpu_->begin()), itend(modifiedCpu_->end())
434      ; it != itend ;) {
435     CpuTi *ti = &*it;
436     ++it;
437     ti->updateActionsFinishTime(now);
438   }
439
440 /* get the min next event if heap not empty */
441   if (xbt_heap_size(tiActionHeap_) > 0)
442     min_action_duration = xbt_heap_maxkey(tiActionHeap_) - now;
443
444   XBT_DEBUG("Share resources, min next event date: %f", min_action_duration);
445
446   return min_action_duration;
447 }
448
449 void CpuTiModel::updateActionsState(double now, double /*delta*/)
450 {
451   while ((xbt_heap_size(tiActionHeap_) > 0)
452          && (xbt_heap_maxkey(tiActionHeap_) <= now)) {
453     CpuTiAction *action = (CpuTiAction*) xbt_heap_pop(tiActionHeap_);
454     XBT_DEBUG("Action %p: finish", action);
455     action->finish();
456     /* set the remains to 0 due to precision problems when updating the remaining amount */
457     action->setRemains(0);
458     action->setState(SURF_ACTION_DONE);
459     /* update remaining amount of all actions */
460     action->cpu_->updateRemainingAmount(surf_get_clock());
461   }
462 }
463
464 /************
465  * Resource *
466  ************/
467 CpuTi::CpuTi(CpuTiModel *model, simgrid::s4u::Host *host, xbt_dynar_t speedPeak,
468         tmgr_trace_t speedTrace, int core)
469   : Cpu(model, host, NULL, core, 0)
470 {
471   xbt_assert(core==1,"Multi-core not handled by this model yet");
472   coresAmount_ = core;
473
474   availTrace_ = new CpuTiTgmr(speedTrace, 1/*scale*/);
475
476   actionSet_ = new ActionTiList();
477
478   xbt_dynar_get_cpy(speedPeak, 0, &speed_.peak);
479   XBT_DEBUG("CPU create: peak=%f", speed_.peak);
480
481   if (speedTrace && xbt_dynar_length(speedTrace->event_list) > 1) {
482     s_tmgr_event_t val;
483     // add a fake trace event if periodicity == 0
484     xbt_dynar_get_cpy(speedTrace->event_list,  xbt_dynar_length(speedTrace->event_list) - 1, &val);
485     if (val.delta == 0)
486       speed_.event = future_evt_set->add_trace(tmgr_empty_trace_new(), availTrace_->lastTime_, this);
487   }
488 }
489
490 CpuTi::~CpuTi()
491 {
492   modified(false);
493   delete availTrace_;
494   delete actionSet_;
495 }
496 void CpuTi::setSpeedTrace(tmgr_trace_t trace)
497 {
498   if (availTrace_)
499     delete availTrace_;
500
501   availTrace_ = new CpuTiTgmr(trace, speed_.scale);
502
503   /* add a fake trace event if periodicity == 0 */
504   if (trace && xbt_dynar_length(trace->event_list) > 1) {
505     s_tmgr_event_t val;
506     xbt_dynar_get_cpy(trace->event_list, xbt_dynar_length(trace->event_list) - 1, &val);
507     if (val.delta == 0)
508       speed_.event = future_evt_set->add_trace(tmgr_empty_trace_new(), 0.0, this);
509   }
510 }
511
512 void CpuTi::apply_event(tmgr_trace_iterator_t event, double value)
513 {
514   if (event == speed_.event) {
515     tmgr_trace_t speedTrace;
516     CpuTiTgmr *trace;
517     s_tmgr_event_t val;
518
519     XBT_DEBUG("Finish trace date: value %f", value);
520     /* update remaining of actions and put in modified cpu swag */
521     updateRemainingAmount(surf_get_clock());
522
523     modified(true);
524
525     speedTrace = availTrace_->speedTrace_;
526     xbt_dynar_get_cpy(speedTrace->event_list, xbt_dynar_length(speedTrace->event_list) - 1, &val);
527     delete availTrace_;
528     speed_.scale = val.value;
529
530     trace = new CpuTiTgmr(TRACE_FIXED, val.value);
531     XBT_DEBUG("value %f", val.value);
532
533     availTrace_ = trace;
534
535     tmgr_trace_event_unref(&speed_.event);
536
537   } else if (event == stateEvent_) {
538     if (value > 0) {
539       if(isOff())
540         xbt_dynar_push_as(host_that_restart, char*, (char *)getName());
541       turnOn();
542     } else {
543       turnOff();
544       double date = surf_get_clock();
545
546       /* put all action running on cpu to failed */
547       for(ActionTiList::iterator it(actionSet_->begin()), itend(actionSet_->end())
548           ; it != itend ; ++it) {
549
550         CpuTiAction *action = &*it;
551         if (action->getState() == SURF_ACTION_RUNNING
552          || action->getState() == SURF_ACTION_READY
553          || action->getState() == SURF_ACTION_NOT_IN_THE_SYSTEM) {
554           action->setFinishTime(date);
555           action->setState(SURF_ACTION_FAILED);
556           if (action->indexHeap_ >= 0) {
557             CpuTiAction *heap_act = (CpuTiAction*)
558                 xbt_heap_remove(static_cast<CpuTiModel*>(getModel())->tiActionHeap_, action->indexHeap_);
559             if (heap_act != action)
560               DIE_IMPOSSIBLE;
561           }
562         }
563       }
564     }
565     tmgr_trace_event_unref(&stateEvent_);
566
567   } else {
568     xbt_die("Unknown event!\n");
569   }
570 }
571
572 void CpuTi::updateActionsFinishTime(double now)
573 {
574   CpuTiAction *action;
575   double sum_priority = 0.0, total_area, min_finish = -1;
576
577   /* update remaining amount of actions */
578   updateRemainingAmount(now);
579
580   for(ActionTiList::iterator it(actionSet_->begin()), itend(actionSet_->end()) ; it != itend ; ++it) {
581     action = &*it;
582     /* action not running, skip it */
583     if (action->getStateSet() != surf_cpu_model_pm->getRunningActionSet())
584       continue;
585
586     /* bogus priority, skip it */
587     if (action->getPriority() <= 0)
588       continue;
589
590     /* action suspended, skip it */
591     if (action->suspended_ != 0)
592       continue;
593
594     sum_priority += 1.0 / action->getPriority();
595   }
596   sumPriority_ = sum_priority;
597
598   for(ActionTiList::iterator it(actionSet_->begin()), itend(actionSet_->end()) ; it != itend ; ++it) {
599     action = &*it;
600     min_finish = -1;
601     /* action not running, skip it */
602     if (action->getStateSet() !=
603         surf_cpu_model_pm->getRunningActionSet())
604       continue;
605
606     /* verify if the action is really running on cpu */
607     if (action->suspended_ == 0 && action->getPriority() > 0) {
608       /* total area needed to finish the action. Used in trace integration */
609       total_area =
610           (action->getRemains()) * sum_priority *
611            action->getPriority();
612
613       total_area /= speed_.peak;
614
615       action->setFinishTime(availTrace_->solve(now, total_area));
616       /* verify which event will happen before (max_duration or finish time) */
617       if (action->getMaxDuration() != NO_MAX_DURATION &&
618           action->getStartTime() + action->getMaxDuration() < action->m_finish)
619         min_finish = action->getStartTime() + action->getMaxDuration();
620       else
621         min_finish = action->m_finish;
622     } else {
623       /* put the max duration time on heap */
624       if (action->getMaxDuration() != NO_MAX_DURATION)
625         min_finish = action->getStartTime() + action->getMaxDuration();
626     }
627     /* add in action heap */
628     XBT_DEBUG("action(%p) index %d", action, action->indexHeap_);
629     if (action->indexHeap_ >= 0) {
630       CpuTiAction *heap_act = (CpuTiAction*)
631           xbt_heap_remove(static_cast<CpuTiModel*>(getModel())->tiActionHeap_, action->indexHeap_);
632       if (heap_act != action)
633         DIE_IMPOSSIBLE;
634     }
635     if (min_finish != NO_MAX_DURATION)
636       xbt_heap_push(static_cast<CpuTiModel*>(getModel())->tiActionHeap_, action, min_finish);
637
638     XBT_DEBUG
639         ("Update finish time: Cpu(%s) Action: %p, Start Time: %f Finish Time: %f Max duration %f",
640          getName(), action, action->getStartTime(),
641          action->m_finish,
642          action->getMaxDuration());
643   }
644   /* remove from modified cpu */
645   modified(false);
646 }
647
648 bool CpuTi::isUsed()
649 {
650   return !actionSet_->empty();
651 }
652
653 double CpuTi::getAvailableSpeed()
654 {
655   speed_.scale = availTrace_->getPowerScale(surf_get_clock());
656   return Cpu::getAvailableSpeed();
657 }
658
659 /** @brief Update the remaining amount of actions */
660 void CpuTi::updateRemainingAmount(double now)
661 {
662
663   /* already updated */
664   if (lastUpdate_ >= now)
665     return;
666
667   /* compute the integration area */
668   double area_total = availTrace_->integrate(lastUpdate_, now) * speed_.peak;
669   XBT_DEBUG("Flops total: %f, Last update %f", area_total, lastUpdate_);
670
671   for(ActionTiList::iterator it(actionSet_->begin()), itend(actionSet_->end()) ; it != itend ; ++it) {
672     CpuTiAction *action = &*it;
673     /* action not running, skip it */
674     if (action->getStateSet() != getModel()->getRunningActionSet())
675       continue;
676
677     /* bogus priority, skip it */
678     if (action->getPriority() <= 0)
679       continue;
680
681     /* action suspended, skip it */
682     if (action->suspended_ != 0)
683       continue;
684
685     /* action don't need update */
686     if (action->getStartTime() >= now)
687       continue;
688
689     /* skip action that are finishing now */
690     if (action->m_finish >= 0 && action->m_finish <= now)
691       continue;
692
693     /* update remaining */
694     action->updateRemains(area_total / (sumPriority_ * action->getPriority()));
695     XBT_DEBUG("Update remaining action(%p) remaining %f", action, action->m_remains);
696   }
697   lastUpdate_ = now;
698 }
699
700 CpuAction *CpuTi::execution_start(double size)
701 {
702   XBT_IN("(%s,%g)", getName(), size);
703   CpuTiAction *action = new CpuTiAction(static_cast<CpuTiModel*>(getModel()), size, isOff(), this);
704
705   actionSet_->push_back(*action);
706
707   XBT_OUT();
708   return action;
709 }
710
711
712 CpuAction *CpuTi::sleep(double duration)
713 {
714   if (duration > 0)
715     duration = MAX(duration, sg_surf_precision);
716
717   XBT_IN("(%s,%g)", getName(), duration);
718   CpuTiAction *action = new CpuTiAction(static_cast<CpuTiModel*>(getModel()), 1.0, isOff(), this);
719
720   action->m_maxDuration = duration;
721   action->suspended_ = 2;
722   if (duration == NO_MAX_DURATION) {
723    /* Move to the *end* of the corresponding action set. This convention
724       is used to speed up update_resource_state  */
725   action->getStateSet()->erase(action->getStateSet()->iterator_to(*action));
726     action->p_stateSet = static_cast<CpuTiModel*>(getModel())->runningActionSetThatDoesNotNeedBeingChecked_;
727     action->getStateSet()->push_back(*action);
728   }
729
730   actionSet_->push_back(*action);
731
732   XBT_OUT();
733   return action;
734 }
735
736 void CpuTi::modified(bool modified){
737   CpuTiList *modifiedCpu = static_cast<CpuTiModel*>(getModel())->modifiedCpu_;
738   if (modified) {
739     if (!cpu_ti_hook.is_linked()) {
740       modifiedCpu->push_back(*this);
741     }
742   } else {
743     if (cpu_ti_hook.is_linked()) {
744       modifiedCpu->erase(modifiedCpu->iterator_to(*this));
745     }
746   }
747 }
748
749 /**********
750  * Action *
751  **********/
752
753 CpuTiAction::CpuTiAction(CpuTiModel *model_, double cost, bool failed, CpuTi *cpu)
754  : CpuAction(model_, cost, failed)
755 {
756   cpu_ = cpu;
757   indexHeap_ = -1;
758   cpu_->modified(true);
759 }
760
761 void CpuTiAction::updateIndexHeap(int i)
762 {
763   indexHeap_ = i;
764 }
765
766 void CpuTiAction::setState(e_surf_action_state_t state)
767 {
768   CpuAction::setState(state);
769   cpu_->modified(true);
770 }
771
772 int CpuTiAction::unref()
773 {
774   m_refcount--;
775   if (!m_refcount) {
776     if (action_hook.is_linked())
777       getStateSet()->erase(getStateSet()->iterator_to(*this));
778     /* remove from action_set */
779     if (action_ti_hook.is_linked())
780       cpu_->actionSet_->erase(cpu_->actionSet_->iterator_to(*this));
781     /* remove from heap */
782     xbt_heap_remove(static_cast<CpuTiModel*>(getModel())->tiActionHeap_, this->indexHeap_);
783     cpu_->modified(true);
784     delete this;
785     return 1;
786   }
787   return 0;
788 }
789
790 void CpuTiAction::cancel()
791 {
792   this->setState(SURF_ACTION_FAILED);
793   xbt_heap_remove(getModel()->getActionHeap(), this->indexHeap_);
794   cpu_->modified(true);
795   return;
796 }
797
798 void CpuTiAction::suspend()
799 {
800   XBT_IN("(%p)", this);
801   if (suspended_ != 2) {
802     suspended_ = 1;
803     xbt_heap_remove(getModel()->getActionHeap(), indexHeap_);
804     cpu_->modified(true);
805   }
806   XBT_OUT();
807 }
808
809 void CpuTiAction::resume()
810 {
811   XBT_IN("(%p)", this);
812   if (suspended_ != 2) {
813     suspended_ = 0;
814     cpu_->modified(true);
815   }
816   XBT_OUT();
817 }
818
819 void CpuTiAction::setMaxDuration(double duration)
820 {
821   double min_finish;
822
823   XBT_IN("(%p,%g)", this, duration);
824
825   m_maxDuration = duration;
826
827   if (duration >= 0)
828     min_finish = (getStartTime() + getMaxDuration()) < getFinishTime() ?
829                  (getStartTime() + getMaxDuration()) : getFinishTime();
830   else
831     min_finish = getFinishTime();
832
833 /* add in action heap */
834   if (indexHeap_ >= 0) {
835     CpuTiAction *heap_act = (CpuTiAction*)
836         xbt_heap_remove(getModel()->getActionHeap(), indexHeap_);
837     if (heap_act != this)
838       DIE_IMPOSSIBLE;
839   }
840   xbt_heap_push(getModel()->getActionHeap(), this, min_finish);
841
842   XBT_OUT();
843 }
844
845 void CpuTiAction::setPriority(double priority)
846 {
847   XBT_IN("(%p,%g)", this, priority);
848   m_priority = priority;
849   cpu_->modified(true);
850   XBT_OUT();
851 }
852
853 double CpuTiAction::getRemains()
854 {
855   XBT_IN("(%p)", this);
856   cpu_->updateRemainingAmount(surf_get_clock());
857   XBT_OUT();
858   return m_remains;
859 }
860
861 }
862 }
863
864 #endif /* SURF_MODEL_CPUTI_H_ */