Subversion Repositories shark

/*
 * Project: S.Ha.R.K.
 *
 * Coordinators:
 *   Giorgio Buttazzo    <giorgio@sssup.it>
 *   Paolo Gai           <pj@gandalf.sssup.it>
 *
 * Authors     :
 *   Paolo Gai           <pj@gandalf.sssup.it>
 *   (see the web pages for full authors list)
 *
 * ReTiS Lab (Scuola Superiore S.Anna - Pisa - Italy)
 *
 * http://www.sssup.it
 * http://retis.sssup.it
 * http://shark.sssup.it
 */

/*
 * Copyright (C) 2001 Paolo Gai
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 *
 */

#include "edfstar.h"
#include <ll/stdio.h>
#include <ll/string.h>
#include <kernel/model.h>
#include <kernel/descr.h>
#include <kernel/var.h>
#include <kernel/func.h>

#include <tracer.h>

#define EDFSTAR_CHANGE_LEVEL 1

/* for iqueues */
/* #include "iqueue.h" Now iqueues are the only queue type available
   into the kernel */
#include <kernel/iqueue.h>

/* for BUDGET_TASK_MODEL */
#include "fsf_contract.h"
#include "fsf_server.h"

/*
 * DEBUG stuffs begin
 */

//#define EDFSTAR_DEBUG

#ifdef EDFSTAR_DEBUG

static __inline__ fake_printf(char *fmt, ...) {}

//#define edfstar_printf fake_printf
//#define edfstar_printf2 fake_printf
//#define edfstar_printf3 fake_printf

#define edfstar_printf kern_printf
#define edfstar_printf2 kern_printf
#define edfstar_printf3 kern_printf
#endif

/*
 * DEBUG stuffs end
 */

/* Status used in the level */
#define EDFSTAR_READY         MODULE_STATUS_BASE    /* - Ready status        */
#define EDFSTAR_IDLE          MODULE_STATUS_BASE+4  /* to wait the deadline  */

/* flags */
#define EDFSTAR_FLAG_NOPREEMPT   4
#define EDFSTAR_FLAG_NORAISEEXC  2
#define EDFSTAR_FLAG_SPORADIC    1


/* the level redefinition for the Earliest Deadline First level */
typedef struct {
  level_des l;     /* the standard level descriptor          */

  TIME period[MAX_PROC]; /* The task periods; the deadlines are
                       stored in the priority field           */
  int deadline_timer[MAX_PROC];
                   /* The task deadline timers               */

  struct timespec deadline_timespec[MAX_PROC];

  int dline_miss[MAX_PROC]; /* Deadline miss counter */
  int wcet_miss[MAX_PROC];  /* Wcet miss counter */

  int nact[MAX_PROC];       /* Wcet miss counter */

  int flag[MAX_PROC];
                   /* used to manage the JOB_TASK_MODEL and the
                       periodicity                            */

  IQUEUE ready;     /* the ready queue                        */

  PID activated;   /* the task that has been inserted into the
                       master module */

  int budget[MAX_PROC];

  int scheduling_level;

  int cap_lev;
  struct timespec cap_lasttime;


} EDFSTAR_level_des;

static void capacity_handler(void *l)
{
  EDFSTAR_level_des *lev = l;
  lev->cap_lev = NIL;
  event_need_reschedule();
}

static void EDFSTAR_check_preemption(EDFSTAR_level_des *lev)
{
  PID first;

  #ifdef EDFSTAR_DEBUG
    edfstar_printf("(E:chk)");
  #endif
  /* check if the task is preempteble or not */
  if (lev->activated != NIL && lev->flag[lev->activated] & EDFSTAR_FLAG_NOPREEMPT) return;

  if ((first = iq_query_first(&lev->ready)) != lev->activated) {
    if (lev->activated != NIL)
      level_table[ lev->scheduling_level ]->
        private_extract(lev->scheduling_level, lev->activated);

    lev->activated = first;

    if (first != NIL) {
      BUDGET_TASK_MODEL b;
      budget_task_default_model(b, lev->budget[first]);

      level_table[ lev->scheduling_level ]->
        private_insert(lev->scheduling_level, first, (TASK_MODEL *)&b);
    }
  }
}

static void EDFSTAR_timer_deadline(void *par);

static void EDFSTAR_internal_activate(EDFSTAR_level_des *lev, PID p,
                                      struct timespec *t)
{

  #ifdef EDFSTAR_DEBUG
    edfstar_printf("(E:iact)");
  #endif

  ADDUSEC2TIMESPEC(lev->period[p], t);

  *iq_query_timespec(p, &lev->ready) = *t;
  lev->deadline_timespec[p] = *t;

  /* Insert task in the correct position */
  proc_table[p].status = EDFSTAR_READY;
  iq_timespec_insert(p,&lev->ready);
  proc_table[p].control &= ~CONTROL_CAP;

  /* check for preemption */
  EDFSTAR_check_preemption(lev);
}

static void EDFSTAR_timer_deadline(void *par)
{
  PID p = (PID) par;
  EDFSTAR_level_des *lev;

  lev = (EDFSTAR_level_des *)level_table[proc_table[p].task_level];
  lev->deadline_timer[p] = NIL;  

  switch (proc_table[p].status) {
    case EDFSTAR_IDLE:
      /* set the request time */
      if (!(lev->flag[p] & EDFSTAR_FLAG_SPORADIC))
        EDFSTAR_internal_activate(lev,p,iq_query_timespec(p, &lev->ready));

      event_need_reschedule();
      break;

    default:
      #ifdef EDFSTAR_DEBUG
        kern_printf("(E:Dl:%d)",p);
      #endif
      /* else, a deadline miss occurred!!! */
      lev->dline_miss[p]++;
      TRACER_LOGEVENT(FTrace_EVT_task_deadline_miss,proc_table[p].context,proc_table[p].task_level);

      /* the task is into another state */
      if (!(lev->flag[p] & EDFSTAR_FLAG_SPORADIC)) {
        lev->nact[p]++;
        ADDUSEC2TIMESPEC(lev->period[p], &lev->deadline_timespec[p]);
      }
  }

  /* Set the deadline timer */
  if (!(lev->flag[p] & EDFSTAR_FLAG_SPORADIC))
    lev->deadline_timer[p] = kern_event_post(&lev->deadline_timespec[p],
                                             EDFSTAR_timer_deadline,
                                             (void *)p);

}

static void EDFSTAR_timer_guest_deadline(void *par)
{
  PID p = (PID) par;

  #ifdef EDFSTAR_DEBUG
    edfstar_printf("(E:gdl)");
  #endif

  kern_raise(XDEADLINE_MISS,p);
}

static int EDFSTAR_public_create(LEVEL l, PID p, TASK_MODEL *m)
{
  EDFSTAR_level_des *lev = (EDFSTAR_level_des *)(level_table[l]);

  /* if the EDFSTAR_task_create is called, then the pclass must be a
     valid pclass. */
  HARD_TASK_MODEL *h;

  if (m->pclass != HARD_PCLASS) return -1;
  if (m->level != 0 && m->level != l) return -1;
  h = (HARD_TASK_MODEL *)m;
  if (!h->wcet || !h->mit) return -1;
  /* now we know that m is a valid model */

  #ifdef EDFSTAR_DEBUG
    edfstar_printf("(E:Crt)");
  #endif

  lev->period[p] = h->mit;

  lev->flag[p] = 0;

  if (h->periodicity == APERIODIC)
       lev->flag[p] |= EDFSTAR_FLAG_SPORADIC;

  lev->deadline_timer[p] = -1;
  lev->dline_miss[p]     = 0;
  lev->wcet_miss[p]      = 0;
  lev->nact[p]           = 0;

  /* Enable wcet check */
  proc_table[p].avail_time = h->wcet;
  proc_table[p].wcet       = h->wcet;

  return 0; /* OK, also if the task cannot be guaranteed... */
}


static void EDFSTAR_account_capacity(EDFSTAR_level_des *lev, PID p)
{
  struct timespec ty;
  TIME tx;

  SUBTIMESPEC(&schedule_time, &lev->cap_lasttime, &ty);
  tx = TIMESPEC2USEC(&ty);

  proc_table[p].avail_time -= tx;
}

static int EDFSTAR_public_eligible(LEVEL l, PID p)
{
  EDFSTAR_level_des *lev = (EDFSTAR_level_des *)(level_table[l]);

  #ifdef EDFSTAR_DEBUG
    edfstar_printf2("(E:eli)");
  #endif

  return level_table[ lev->scheduling_level ]->
    private_eligible(lev->scheduling_level,p);

}

static void EDFSTAR_public_dispatch(LEVEL l, PID p, int nostop)
{
  EDFSTAR_level_des *lev = (EDFSTAR_level_des *)(level_table[l]);
  struct timespec ty;

  #ifdef EDFSTAR_DEBUG
    edfstar_printf("(E:dis)");
  #endif
 
  if (!nostop || proc_table[exec].task_level==l) {
      TIMESPEC_ASSIGN(&ty, &schedule_time);
      TIMESPEC_ASSIGN(&lev->cap_lasttime, &schedule_time);

      /* ...and finally, we have to post a capacity event on exec task because the shadow_task consume
       *      *        capacity on exe task always */
      if (proc_table[exec].avail_time > 0) {
        ADDUSEC2TIMESPEC(proc_table[exec].avail_time ,&ty);
        lev->cap_lev = kern_event_post(&ty,capacity_handler, lev);
      }
      level_table[lev->scheduling_level]->private_dispatch(lev->scheduling_level, p, nostop);
  }
  else
      level_table[proc_table[exec].task_level]->public_dispatch(proc_table[exec].task_level, p, nostop);

}

static void EDFSTAR_public_epilogue(LEVEL l, PID p)
{
  EDFSTAR_level_des *lev = (EDFSTAR_level_des *)(level_table[l]);

  #ifdef EDFSTAR_DEBUG
    edfstar_printf("(E:epi ");
  #endif

  if (lev->cap_lev!=NIL) {
       kern_event_delete(lev->cap_lev);
       lev->cap_lev=NIL;
  }


  if ( proc_table[exec].task_level==l ) {
   
     if (proc_table[exec].avail_time > 0) EDFSTAR_account_capacity(lev,exec);

     /* check if the wcet is finished... */
     if (proc_table[exec].avail_time < 0) {
        /* wcet finished: disable wcet event and count wcet miss */

       #ifdef EDFSTAR_DEBUG
         edfstar_printf2("W%d",p);
       #endif
        //proc_table[p].control &= ~CONTROL_CAP;
        lev->wcet_miss[exec]++;
        proc_table[exec].avail_time = 0;
        TRACER_LOGEVENT(FTrace_EVT_task_wcet_violation,proc_table[exec].context,proc_table[exec].task_level);
     }

     #ifdef EDFSTAR_DEBUG
       edfstar_printf(")");
     #endif

     level_table[ lev->scheduling_level ]->
       private_epilogue(lev->scheduling_level,p);

     proc_table[exec].status = EDFSTAR_READY;
    } else
        level_table[proc_table[exec].task_level]->public_epilogue(proc_table[exec].task_level,p);

}

static void EDFSTAR_public_activate(LEVEL l, PID p, struct timespec *o)
{
  EDFSTAR_level_des *lev = (EDFSTAR_level_des *)(level_table[l]);
  struct timespec t;

  #ifdef EDFSTAR_DEBUG
    edfstar_printf("(E:act:%d)",p);
  #endif

  /* Test if we are trying to activate a non sleeping task    */
  /* save activation (only if needed... */
  if (proc_table[p].status != SLEEP) {
    /* a periodic task cannot be activated when it is already active */
    /* but aperiodic task can be reactivate before */
    if (lev->flag[p] & EDFSTAR_FLAG_SPORADIC) {
        if (proc_table[p].status != EDFSTAR_IDLE) {
          lev->nact[p]++;
          return;
        }
    } else {
        kern_raise(XACTIVATION,p);
    }
  }

  kern_gettime(&t);

  EDFSTAR_internal_activate(lev,p, &t);

  /* Set the deadline timer */
  lev->deadline_timer[p] = kern_event_post(&lev->deadline_timespec[p],
                                           EDFSTAR_timer_deadline,
                                           (void *)p);

}

static void EDFSTAR_public_unblock(LEVEL l, PID p)
{
  EDFSTAR_level_des *lev = (EDFSTAR_level_des *)(level_table[l]);

  #ifdef EDFSTAR_DEBUG
    edfstar_printf("(E:ins)");
  #endif

  /* Insert task in the correct position */
  proc_table[p].status = EDFSTAR_READY;
  iq_timespec_insert(p,&lev->ready);

  /* and check for preemption! */
  EDFSTAR_check_preemption(lev);

}

static void EDFSTAR_public_block(LEVEL l, PID p)
{

  EDFSTAR_level_des *lev = (EDFSTAR_level_des *)(level_table[l]);

  #ifdef EDFSTAR_DEBUG
   edfstar_printf("(E:ext)");
  #endif

  /* the task is blocked on a synchronization primitive. we have to
     remove it from the master module -and- from the local queue! */
  iq_extract(p,&lev->ready);

  /* and finally, a preemption check! (it will also call guest_end) */
  EDFSTAR_check_preemption(lev);
}

static int EDFSTAR_public_message(LEVEL l, PID p, void *m)
{

  EDFSTAR_level_des *lev = (EDFSTAR_level_des *)(level_table[l]);
  struct timespec temp;

  #ifdef EDFSTAR_DEBUG
    edfstar_printf("(E:ecy ");
  #endif

  switch ((long)(m)) {

    /* Task EndCycle */
    case (long)(NULL):

       /* we call guest_end directly here because the same task may
         be reinserted in the queue before calling the preemption check! */
       level_table[ lev->scheduling_level ]->
         private_extract(lev->scheduling_level,p);
       lev->activated = NIL;

       iq_extract(p,&lev->ready);

       /* we reset the capacity counters... */
       proc_table[p].avail_time = proc_table[p].wcet;

       if (lev->nact[p] > 0) {
   
         #ifdef EDFSTAR_DEBUG
           edfstar_printf2("E%d",p);
         #endif
     
         /* Pending activation: reactivate the thread!!! */
         lev->nact[p]--;
     
         /* see also EDFSTAR_timer_deadline */
         kern_gettime(&temp);
     
         EDFSTAR_internal_activate(lev,p, &temp);
     
         /* check if the deadline has already expired */
         temp = *iq_query_timespec(p, &lev->ready);
         if (TIMESPEC_A_LT_B(&temp, &schedule_time)) {
           /* count the deadline miss */
           lev->dline_miss[p]++;
           kern_event_delete(lev->deadline_timer[p]);
           lev->deadline_timer[p] = NIL;
         }

       } else {

         #ifdef EDFSTAR_DEBUG
           edfstar_printf("e%d",p);
         #endif
     
         /* the task has terminated his job before it consume the wcet. All OK! */
         if (lev->flag[p] & EDFSTAR_FLAG_SPORADIC)
                proc_table[p].status = SLEEP;
         else
                proc_table[p].status = EDFSTAR_IDLE;
         
         if (lev->flag[p] & EDFSTAR_FLAG_SPORADIC && lev->deadline_timer[p] != NIL) {
           kern_event_delete(lev->deadline_timer[p]);
           lev->deadline_timer[p] = NIL;
         }
     
         /* and finally, a preemption check! */
         EDFSTAR_check_preemption(lev);
     
         /* when the deadline timer fire, it recognize the situation and set
           correctly all the stuffs (like reactivation, etc... ) */
       }

    #ifdef EDFSTAR_DEBUG
      edfstar_printf(")");
    #endif

    TRACER_LOGEVENT(FTrace_EVT_task_end_cycle,proc_table[p].context,proc_table[p].task_level);
    jet_update_endcycle(); /* Update the Jet data... */
    break;

  default:

    break;
     
  }
  return 0;
}

static void EDFSTAR_public_end(LEVEL l, PID p)
{
  EDFSTAR_level_des *lev = (EDFSTAR_level_des *)(level_table[l]);

  #ifdef EDFSTAR_DEBUG
    edfstar_printf("(E:end)");
  #endif

  iq_extract(p,&lev->ready);

  /* we finally put the task in the ready queue */
  proc_table[p].status = FREE;
 
  iq_insertfirst(p,&freedesc);
  lev->activated=NIL;
 
  if (lev->deadline_timer[p] != -1) {
    kern_event_delete(lev->deadline_timer[p]);
    lev->deadline_timer[p] = NIL;
  }

  /* and finally, a preemption check! (it will also call guest_end) */
  EDFSTAR_check_preemption(lev);
}

/* Guest Functions
   These functions manages a JOB_TASK_MODEL, that is used to put
   a guest task in the EDFSTAR ready queue. */

static void EDFSTAR_private_insert(LEVEL l, PID p, TASK_MODEL *m)
{
  EDFSTAR_level_des *lev = (EDFSTAR_level_des *)(level_table[l]);
  JOB_TASK_MODEL *job;

  if (m->pclass != JOB_PCLASS || (m->level != 0 && m->level != l) ) {
    kern_raise(XINVALID_TASK, p);
    return;
  }

  job = (JOB_TASK_MODEL *)m;

  /* if the EDFSTAR_guest_create is called, then the pclass must be a
     valid pclass. */

  *iq_query_timespec(p, &lev->ready) = job->deadline;
 
  lev->deadline_timer[p] = -1;
  lev->dline_miss[p]     = 0;
  lev->wcet_miss[p]      = 0;
  lev->nact[p]           = 0;

  if (job->noraiseexc)
    lev->flag[p] |= EDFSTAR_FLAG_NORAISEEXC;
  else {
    lev->flag[p] &= ~EDFSTAR_FLAG_NORAISEEXC;
    lev->deadline_timer[p] = kern_event_post(iq_query_timespec(p, &lev->ready),
                                             EDFSTAR_timer_guest_deadline,
                                             (void *)p);
  }

  lev->period[p] = job->period;

  /* Insert task in the correct position */
  iq_timespec_insert(p,&lev->ready);
  proc_table[p].status = EDFSTAR_READY;

  /* check for preemption */
  EDFSTAR_check_preemption(lev);

  /* there is no bandwidth guarantee at this level, it is performed
     by the level that inserts guest tasks... */
}

static void EDFSTAR_private_dispatch(LEVEL l, PID p, int nostop)
{
  EDFSTAR_level_des *lev = (EDFSTAR_level_des *)(level_table[l]);

  level_table[ lev->scheduling_level ]->
    private_dispatch(lev->scheduling_level,p,nostop);
}

static void EDFSTAR_private_epilogue(LEVEL l, PID p)
{
  EDFSTAR_level_des *lev = (EDFSTAR_level_des *)(level_table[l]);

  /* the task has been preempted. it returns into the ready queue... */
  level_table[ lev->scheduling_level ]->
    private_epilogue(lev->scheduling_level,p);

  proc_table[p].status = EDFSTAR_READY;
}

static void EDFSTAR_private_extract(LEVEL l, PID p)
{
  EDFSTAR_level_des *lev = (EDFSTAR_level_des *)(level_table[l]);

#ifdef EDFSTAR_DEBUG
  //kern_printf("EDFSTAR_guest_end: dline timer %d\n",lev->deadline_timer[p]);
#endif

  iq_extract(p, &lev->ready);

  /* we remove the deadline timer, because the slice is finished */
  if (lev->deadline_timer[p] != NIL) {
#ifdef EDFSTAR_DEBUG
//    kern_printf("EDFSTAR_guest_end: dline timer %d\n",lev->deadline_timer[p]);
#endif
    kern_event_delete(lev->deadline_timer[p]);
    lev->deadline_timer[p] = NIL;
  }

  /* and finally, a preemption check! (it will also call guest_end() */
  EDFSTAR_check_preemption(lev);
}

/* Registration functions */

/* Registration function:
    int flags                 the init flags ... see EDFSTAR.h */

LEVEL EDFSTAR_register_level(int master)
{
  LEVEL l;            /* the level that we register */
  EDFSTAR_level_des *lev;  /* for readableness only */
  PID i;              /* a counter */

#ifdef EDFSTAR_DEBUG
  printk("EDFSTAR_register_level\n");
#endif

  /* request an entry in the level_table */
  l = level_alloc_descriptor(sizeof(EDFSTAR_level_des));

  lev = (EDFSTAR_level_des *)level_table[l];

  /* fill the standard descriptor */
  lev->l.private_insert   = EDFSTAR_private_insert;
  lev->l.private_extract  = EDFSTAR_private_extract;
  lev->l.private_dispatch = EDFSTAR_private_dispatch;
  lev->l.private_epilogue = EDFSTAR_private_epilogue;

  lev->l.public_guarantee = NULL;
  lev->l.public_eligible  = EDFSTAR_public_eligible;
  lev->l.public_create    = EDFSTAR_public_create;
  lev->l.public_end       = EDFSTAR_public_end;
  lev->l.public_dispatch  = EDFSTAR_public_dispatch;
  lev->l.public_epilogue  = EDFSTAR_public_epilogue;
  lev->l.public_activate  = EDFSTAR_public_activate;
  lev->l.public_unblock   = EDFSTAR_public_unblock;
  lev->l.public_block     = EDFSTAR_public_block;
  lev->l.public_message   = EDFSTAR_public_message;

  /* fill the EDFSTAR descriptor part */
  for(i=0; i<MAX_PROC; i++) {
    lev->period[i]         = 0;
    lev->deadline_timer[i] = -1;
    lev->flag[i]           = 0;
    lev->dline_miss[i]     = 0;
    lev->wcet_miss[i]      = 0;
    lev->nact[i]           = 0;
    lev->budget[i]         = NIL;
  }

  iq_init(&lev->ready, NULL, IQUEUE_NO_PRIORITY);
  lev->activated = NIL;

  lev->scheduling_level = master;
  lev->cap_lev = NIL;
  NULL_TIMESPEC(&lev->cap_lasttime);

  return l;
}

int EDFSTAR_get_dline_miss(PID p)
{
  LEVEL l = proc_table[p].task_level;
  EDFSTAR_level_des *lev = (EDFSTAR_level_des *)(level_table[l]);
 
  return lev->dline_miss[p];
}

int EDFSTAR_get_wcet_miss(PID p)
{
  LEVEL l = proc_table[p].task_level;
  EDFSTAR_level_des *lev = (EDFSTAR_level_des *)(level_table[l]);

  return lev->wcet_miss[p];
}

int EDFSTAR_get_nact(PID p)
{
  LEVEL l = proc_table[p].task_level;
  EDFSTAR_level_des *lev = (EDFSTAR_level_des *)(level_table[l]);

  return lev->nact[p];
}

int EDFSTAR_reset_dline_miss(PID p)
{
  LEVEL l = proc_table[p].task_level;
  EDFSTAR_level_des *lev = (EDFSTAR_level_des *)(level_table[l]);

  lev->dline_miss[p] = 0;
  return 0;
}

int EDFSTAR_reset_wcet_miss(PID p)
{
  LEVEL l = proc_table[p].task_level;
  EDFSTAR_level_des *lev = (EDFSTAR_level_des *)(level_table[l]);

  lev->wcet_miss[p] = 0;
  return 0;
}

int EDFSTAR_setbudget(LEVEL l, PID p, int budget)
{

  EDFSTAR_level_des *lev = (EDFSTAR_level_des *)(level_table[l]);
 
  lev->budget[p] = budget;

  return 0;

}

int EDFSTAR_getbudget(LEVEL l, PID p)
{

  EDFSTAR_level_des *lev = (EDFSTAR_level_des *)(level_table[l]);

  return lev->budget[p];

}

void EDFSTAR_set_nopreemtive_current(LEVEL l) {
 
  EDFSTAR_level_des *lev = (EDFSTAR_level_des *)(level_table[l]);
 
  lev->flag[lev->activated]|=EDFSTAR_FLAG_NOPREEMPT;
}

void EDFSTAR_unset_nopreemtive_current(LEVEL l) {

  EDFSTAR_level_des *lev = (EDFSTAR_level_des *)(level_table[l]);

  lev->flag[lev->activated]&=~EDFSTAR_FLAG_NOPREEMPT;
}
 
int EDFSTAR_budget_has_thread(LEVEL l, int budget)
{

  EDFSTAR_level_des *lev = (EDFSTAR_level_des *)(level_table[l]);
  int i;

  for(i = 0; i< MAX_PROC; i++)
    if (lev->budget[i] == budget) return 1;

  return 0;

}