subworld.cpp

Go to the documentation of this file.

// included by ocbbs.cpp

void BBSImpl::subworld_worker_execute() {
    // execute the same thing that execute_worker is executing. This
    // is done for all the nrnmpi_myid_bbs == -1 workers associated with
    // the specific nrnmpi_myid == 0 with nrnmpi_myid_bbs >= 0.
    // All the nrnmpi/mpispike.cpp functions can be used since the
    // proper communicators for a subworld are used by those functions.
    // The broadcast functions are particularly useful and those are
    // how execute_worker passes messages into here.

//printf("%d enter subworld_worker_execute\n", nrnmpi_myid_world);
    int info[2];
    // wait for something to do
    nrnmpi_int_broadcast(info, 2, 0);
    // info[0] = -1 means it was from a pc.context. Also -2 means
    // DONE.
//printf("%d subworld_worker_execute info %d %d\n", nrnmpi_myid_world, info[0], info[1]);
    int id = info[0];
    if (id == -2) { // DONE, so quit.
        done();
    }
    hoc_ac_ = double(id);
    int style = info[1];
    if (style == 0) { // hoc statement form
        int size;
        nrnmpi_int_broadcast(&size, 1, 0); // includes terminator
        char *s = new char[size];
        nrnmpi_char_broadcast(s, size, 0);
        hoc_obj_run(s, nil);
        delete [] s;
//printf("%d leave subworld_worker_execute\n", nrnmpi_myid_world);
        return;
    }
    int i, j;
    int npickle;
    char* s;
    Symbol* fname = 0;
    Object* ob = nil;
    char* sarg[20]; // up to 20 arguments may be strings
    int ns = 0; // number of args that are strings
    int narg = 0; // total number of args

    if (style == 3) { // python callable
        nrnmpi_int_broadcast(&npickle, 1, 0);
        s = new char[npickle];
        nrnmpi_char_broadcast(s, npickle, 0);
    }else if (style == 1) { // hoc function
        int size;
        nrnmpi_int_broadcast(&size, 1, 0); // includes terminator
//printf("%d subworld hoc function string size = %d\n", nrnmpi_myid_world, size);
        s = new char(size);
        nrnmpi_char_broadcast(s, size, 0);
        fname = hoc_lookup(s);
        if (!fname) { return; } // error raised by sender
    }else{
        return; // no others implemented, error raised by sender
    }

    // now get the args
    int argtypes;
    nrnmpi_int_broadcast(&argtypes, 1, 0);
//printf("%d subworld argtypes = %d\n", nrnmpi_myid_world, argtypes);
    for (j = argtypes; (i = j%5) != 0; j /= 5) {
        ++narg;
        if (i == 1) { // double
            double x;
            nrnmpi_dbl_broadcast(&x, 1, 0);
//printf("%d subworld scalar = %g\n", nrnmpi_myid_world, x);
            hoc_pushx(x);
        }else if (i == 2) { // string
            int size;
            nrnmpi_int_broadcast(&size, 1, 0);
            sarg[ns] = new char(size);
            nrnmpi_char_broadcast(sarg[ns], size, 0);
            hoc_pushstr(sarg+ns);
            ns++;
        }else if (i == 3) { // Vector
            int n;
            nrnmpi_int_broadcast(&n, 1, 0);
            Vect* vec = new Vect(n);
            nrnmpi_dbl_broadcast(vec->data(), n, 0);
            hoc_pushobj(vec->temp_objvar());
        }else{ // PythonObject
            int n;
            nrnmpi_int_broadcast(&n, 1, 0);
            char* s;
            s = new char[n];
            nrnmpi_char_broadcast(s, n, 0);
            Object* po = nrnpy_pickle2po(s, size_t(n));
            delete [] s;
            hoc_pushobj(hoc_temp_objptr(po));
        }
    }

    if (style == 3) {
        size_t size;
        char* rs = (*nrnpy_callpicklef)(s, size_t(npickle), narg, &size);
        assert(rs);
        delete [] rs;
    }else{
//printf("%d subworld hoc call %s narg=%d\n", nrnmpi_myid_world, fname->name, narg);
        hoc_call_objfunc(fname, narg, ob);
//printf("%d subworld return from hoc call %s\n", nrnmpi_myid_world, fname->name);
    }
    delete [] s;
    for (i=0; i < ns; ++i) {
        delete [] sarg[i];
    }
}