src/surf/cpu_ti.cpp

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