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