source.cc

Go to the documentation of this file.

#include "stage.hh"
using namespace Stg;


const double FLAGSZ = 0.25;

typedef struct 
{
  unsigned int capacity;
  unsigned int interval;
} info_t;

int Update( Model* mod, info_t* info )
{
  if((  mod->GetWorld()->GetUpdateCount() % info->interval  == 0 )
          && ( mod->GetFlagCount() < info->capacity) )
                mod->PushFlag( new Model::Flag( Color( 1,1,0 ), FLAGSZ ) );

  return 0; // run again
}

void split( const std::string& text, 
                                const std::string& separators, 
                                std::vector<std::string>& words)
{
  int n = text.length();
  int start, stop;
  start = text.find_first_not_of(separators);
  while ((start >= 0) && (start < n))
         {
                stop = text.find_first_of(separators, start);
                if ((stop < 0) || (stop > n)) stop = n;
                words.push_back(text.substr(start, stop - start));
                start = text.find_first_not_of(separators, stop+1);
         }
}


// Stage calls this when the model starts up
extern "C" int Init( Model* mod, CtrlArgs* args )
{  
  // tokenize the argument string into words
  std::vector<std::string> words;
  split( args->worldfile, std::string(" \t"), words );
  
  // expect an interval as the 1th argument
  assert( words.size() == 3 );
  assert( words[1].size() > 0 );
  
  info_t* info = new info_t;
  info->interval = atoi( words[1].c_str() );
  info->capacity = atoi( words[2].c_str() );

  mod->AddCallback( Model::CB_UPDATE, (model_callback_t)Update, info );  
  mod->Subscribe();
  return 0; //ok
}