brmlife/main.cc
2011-11-27 02:24:04 +01:00

98 lines
2.1 KiB
C++

#include <cstdlib>
#include <cstdio>
#include <ctime>
#include <cstring>
#include <iostream>
#include <list>
#include <fcntl.h>
#include <netinet/in.h>
#include <signal.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <unistd.h>
#include "agent.h"
#include "connection.h"
#include "main.h"
#include "map.h"
int tick_id = 0;
int
main(int argc, char *argv[])
{
class map map(40, 20);
srandom(time(NULL));
int lfd = socket(AF_INET, SOCK_STREAM, 0);
struct sockaddr_in sin;
memset(&sin, 0, sizeof(sin));
sin.sin_family = AF_INET;
sin.sin_port = htons(27753);
sin.sin_addr.s_addr = INADDR_ANY;
int optval = 1;
setsockopt(lfd, SOL_SOCKET, SO_REUSEADDR, &optval, sizeof(optval));
bind(lfd, (struct sockaddr *) &sin, sizeof(sin));
listen(lfd, 10);
signal(SIGPIPE, SIG_IGN);
int flags = fcntl(lfd, F_GETFL, 0);
fcntl(lfd, F_SETFL, flags | O_NONBLOCK);
std::list<class agent *> agents;
int aid = 0;
/* Main tick loop. */
while (true) {
std::cout << "tick " << tick_id << '\n';
/* Drop disconnected agents. */
for (std::list<class agent *>::iterator agent = agents.begin(); agent != agents.end(); agent++)
{
next_agent:
if ((*agent)->conn && (*agent)->conn->error) {
delete *agent;
agent = agents.erase(agent);
if (agent != agents.end())
goto next_agent;
}
}
/* Accept new agents. */
int cfd = accept(lfd, NULL, NULL);
if (cfd >= 0) {
class connection *conn = new class connection(cfd);
agents.push_back(new class agent(aid++, map.agent_startpos(), conn));
}
/* Collect and take actions. */
for (std::list<class agent *>::iterator agent = agents.begin(); agent != agents.end(); agent++)
(*agent)->on_action_takes();
/* Run on_tick everywhere. */
map.on_tick();
for (std::list<class agent *>::iterator agent = agents.begin(); agent != agents.end(); agent++)
(*agent)->on_tick();
/* Update agents' senses. */
for (std::list<class agent *>::iterator agent = agents.begin(); agent != agents.end(); agent++)
(*agent)->on_senses_update();
/* Finish a tick. */
map.print_map();
std::cout << '\n';
usleep(1000000);
tick_id++;
}
return 0;
}