mirror of
https://gitlab.com/scemama/qmcchem.git
synced 2024-09-02 00:03:44 +02:00
272 lines
6.0 KiB
Promela
272 lines
6.0 KiB
Promela
|
#define NPROC 2
|
||
|
#define BUFSIZE 4
|
||
|
#define not_here false
|
||
|
|
||
|
mtype = { NONE, TERMINATE, OK, TEST, ERROR, PROPERTY, WALKERS, EZFIO, GETWALKERS, REGISTER,
|
||
|
EZFIO_REPLY, UNREGISTER, STOPPING, STOPPED, QUEUED, RUNNING };
|
||
|
|
||
|
typedef message_req {
|
||
|
mtype m = NONE;
|
||
|
byte value = 0;
|
||
|
chan reply = [BUFSIZE] of { mtype };
|
||
|
}
|
||
|
|
||
|
typedef message_pull {
|
||
|
mtype m = NONE;
|
||
|
byte value = 0;
|
||
|
}
|
||
|
|
||
|
chan dataserver_pull = [NPROC] of { message_pull };
|
||
|
chan dataserver_req = [NPROC] of { message_req };
|
||
|
|
||
|
byte dataserver_status_pub;
|
||
|
bit http_address = 0;
|
||
|
bit killall_qmc = 0;
|
||
|
bit killall_dataserver = 0;
|
||
|
byte dataserver_status = QUEUED;
|
||
|
byte dataserver_status_n_connected = 0;
|
||
|
|
||
|
|
||
|
|
||
|
|
||
|
/* qmcchem process */
|
||
|
active proctype qmcchem() {
|
||
|
byte reply = NONE;
|
||
|
byte dataserver_pid;
|
||
|
byte i,j;
|
||
|
message_req msg;
|
||
|
|
||
|
dataserver_pid = run dataserver();
|
||
|
|
||
|
/* Wait until ZMQ socket is open */
|
||
|
(http_address == 1);
|
||
|
do
|
||
|
:: (reply == OK) -> break
|
||
|
:: (reply == NONE) ->
|
||
|
msg.m = TEST;
|
||
|
dataserver_req ! msg;
|
||
|
msg.reply ? reply ;
|
||
|
assert (reply == OK || reply == NONE)
|
||
|
od;
|
||
|
printf("Dataserver is ready.\n");
|
||
|
|
||
|
/* Start the QMC processes */
|
||
|
|
||
|
printf("qmcchem: Starting qmc processes.\n");
|
||
|
atomic {
|
||
|
i=0;
|
||
|
do
|
||
|
:: (i < NPROC) ->
|
||
|
run qmc(); i++
|
||
|
:: else -> break
|
||
|
od;
|
||
|
}
|
||
|
printf("qmcchem: qmc processes started.\n");
|
||
|
|
||
|
}
|
||
|
|
||
|
|
||
|
|
||
|
|
||
|
|
||
|
|
||
|
|
||
|
/* dataserver process */
|
||
|
proctype dataserver() {
|
||
|
byte reply = 0;
|
||
|
byte request = 0;
|
||
|
byte cont = 0;
|
||
|
byte reply_pid = 0;
|
||
|
message_req msg;
|
||
|
|
||
|
/* Simulate initialization */
|
||
|
http_address = 1;
|
||
|
dataserver_req ? msg;
|
||
|
msg.reply ! NONE ;
|
||
|
|
||
|
/* Status thread */
|
||
|
run dataserver_status_thread();
|
||
|
run dataserver_main_thread();
|
||
|
}
|
||
|
|
||
|
#define delay 5
|
||
|
#define stop_time 100
|
||
|
|
||
|
|
||
|
proctype dataserver_status_thread() {
|
||
|
byte count=0;
|
||
|
byte n_connected = 0;
|
||
|
byte time=0;
|
||
|
|
||
|
dataserver_status_pub = dataserver_status;
|
||
|
do
|
||
|
:: (dataserver_status == STOPPED) -> break
|
||
|
:: else ->
|
||
|
time = (time < stop_time -> time+1 : time);
|
||
|
count++;
|
||
|
if
|
||
|
:: (count != delay) -> skip
|
||
|
:: else ->
|
||
|
count = 0;
|
||
|
if
|
||
|
:: (dataserver_status == RUNNING &&
|
||
|
n_connected == dataserver_status_n_connected &&
|
||
|
time >= stop_time) ->
|
||
|
dataserver_status = STOPPING;
|
||
|
printf("Stop time reached : STOPPING\n")
|
||
|
|
||
|
:: (dataserver_status == STOPPING &&
|
||
|
n_connected != dataserver_status_n_connected &&
|
||
|
dataserver_status_n_connected == 0) ->
|
||
|
dataserver_status = STOPPED;
|
||
|
printf("No more connected clients : STOPPED\n")
|
||
|
|
||
|
:: (n_connected != dataserver_status_n_connected &&
|
||
|
dataserver_status_n_connected > 0) ->
|
||
|
n_connected = dataserver_status_n_connected;
|
||
|
|
||
|
:: else -> skip
|
||
|
fi
|
||
|
fi
|
||
|
dataserver_status_pub = dataserver_status;
|
||
|
od
|
||
|
printf ("End of dataserver_status_thread\n");
|
||
|
}
|
||
|
|
||
|
|
||
|
proctype dataserver_main_thread() {
|
||
|
byte time = 0;
|
||
|
mtype reply;
|
||
|
dataserver_status = QUEUED;
|
||
|
message_req msg;
|
||
|
message_pull pull;
|
||
|
|
||
|
/* Inform main process that the qmc processes can start (blocking recv) */
|
||
|
dataserver_req ? msg;
|
||
|
assert (msg.m == TEST);
|
||
|
msg.reply ! OK;
|
||
|
|
||
|
do
|
||
|
:: (dataserver_status == STOPPED && (!dataserver_pull ?[pull]) && (!dataserver_req ?[msg])) -> break
|
||
|
:: else ->
|
||
|
do
|
||
|
:: (dataserver_pull ?[pull]) ->
|
||
|
dataserver_pull ? pull
|
||
|
printf("pull: "); printm(pull.m); printf("\n");
|
||
|
if
|
||
|
:: (pull.m == ERROR) -> skip;
|
||
|
:: (pull.m == WALKERS) -> skip
|
||
|
:: (pull.m == PROPERTY) -> skip;
|
||
|
fi
|
||
|
:: else -> break
|
||
|
od
|
||
|
if
|
||
|
:: (dataserver_req ?[msg]) ->
|
||
|
dataserver_req ? msg;
|
||
|
printf("req : "); printm(msg.m); printf("\n");
|
||
|
if
|
||
|
:: (msg.m == TEST) -> reply = OK
|
||
|
:: (msg.m == EZFIO) -> reply = EZFIO_REPLY
|
||
|
:: (msg.m == GETWALKERS) -> reply = WALKERS
|
||
|
:: (msg.m == REGISTER && dataserver_status == QUEUED ) ->
|
||
|
dataserver_status_n_connected++;
|
||
|
dataserver_status = RUNNING;
|
||
|
reply = OK;
|
||
|
printf("Status changed to RUNNING\n")
|
||
|
:: (msg.m == REGISTER && dataserver_status == RUNNING ) ->
|
||
|
dataserver_status_n_connected++;
|
||
|
reply = OK
|
||
|
:: (msg.m == REGISTER &&
|
||
|
(dataserver_status == STOPPED || dataserver_status == STOPPING) ) ->
|
||
|
dataserver_status_n_connected++; reply = ERROR;
|
||
|
printf("dataserver_req: register failed \n")
|
||
|
:: (msg.m == UNREGISTER) ->
|
||
|
dataserver_status_n_connected--;
|
||
|
reply = OK;
|
||
|
if
|
||
|
:: (dataserver_status_n_connected == 0) ->
|
||
|
dataserver_status = STOPPED
|
||
|
printf("Status changed to STOPPED\n")
|
||
|
:: else -> skip
|
||
|
fi
|
||
|
:: else -> skip
|
||
|
fi;
|
||
|
msg.reply ! reply
|
||
|
:: else -> skip
|
||
|
fi
|
||
|
od
|
||
|
}
|
||
|
|
||
|
/* qmc processes */
|
||
|
proctype qmc() {
|
||
|
byte status;
|
||
|
mtype reply;
|
||
|
message_req msg;
|
||
|
message_pull pull;
|
||
|
|
||
|
/* Init */
|
||
|
status = dataserver_status_pub;
|
||
|
|
||
|
msg.m = REGISTER;
|
||
|
dataserver_req ! msg;
|
||
|
end: msg.reply ? reply;
|
||
|
if
|
||
|
:: (reply == ERROR) -> goto exit;
|
||
|
:: else -> assert (reply == OK);
|
||
|
fi;
|
||
|
|
||
|
msg.m = EZFIO;
|
||
|
dataserver_req ! msg;
|
||
|
msg.reply ? reply;
|
||
|
if
|
||
|
:: (reply == ERROR) -> goto exit;
|
||
|
:: else -> assert (reply == EZFIO_REPLY);
|
||
|
fi;
|
||
|
|
||
|
msg.m = GETWALKERS;
|
||
|
dataserver_req ! msg;
|
||
|
msg.reply ? reply;
|
||
|
if
|
||
|
:: (reply == ERROR) -> goto exit;
|
||
|
:: else -> assert (reply == WALKERS);
|
||
|
fi;
|
||
|
|
||
|
|
||
|
|
||
|
/* Equilibration */
|
||
|
(dataserver_status_pub == RUNNING);
|
||
|
|
||
|
msg.m = EZFIO;
|
||
|
dataserver_req ! msg;
|
||
|
msg.reply ? reply;
|
||
|
if
|
||
|
:: (reply == ERROR) -> goto exit;
|
||
|
:: else -> assert (reply == EZFIO_REPLY);
|
||
|
fi;
|
||
|
|
||
|
status = dataserver_status_pub;
|
||
|
|
||
|
/* Cycles */
|
||
|
do
|
||
|
:: (status != RUNNING) -> break
|
||
|
:: else ->
|
||
|
pull.m = PROPERTY; pull.value = 0;
|
||
|
dataserver_pull ! pull;
|
||
|
pull.m = PROPERTY; pull.value =1 ;
|
||
|
dataserver_pull ! pull;
|
||
|
pull.m = WALKERS;
|
||
|
dataserver_pull ! pull;
|
||
|
status = dataserver_status_pub;
|
||
|
od;
|
||
|
|
||
|
/* Termination */
|
||
|
msg.m = UNREGISTER;
|
||
|
dataserver_req ! msg;
|
||
|
msg.reply ? reply;
|
||
|
assert (reply == OK);
|
||
|
|
||
|
exit: skip
|
||
|
}
|
||
|
|
||
|
|