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