Iterate_ODE.h

/*
*	Copyright (c) 2016 Michael Schellenberger Costa mschellenbergercosta@gmail.com
*
*	Permission is hereby granted, free of charge, to any person obtaining a copy
*	of this software and associated documentation files (the "Software"), to deal
*	in the Software without restriction, including without limitation the rights
*	to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
*	copies of the Software, and to permit persons to whom the Software is
*	furnished to do so, subject to the following conditions:
*
*	The above copyright notice and this permission notice shall be included in
*	all copies or substantial portions of the Software.
*
*	THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
*	IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
*	FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
*	AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
*	LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
*	OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
*	THE SOFTWARE.
*/
#ifndef ODE_H
#define ODE_H
#include <vector>
#include "Inhibitory_Neuron.h"
#include "Pyramidal_Neuron.h"
#include "Reticular_Neuron.h"
#include "Thalamocortical_Neuron.h"

void Iterate_ODE(std::vector<Pyramidal_Neuron>& PY,
                 std::vector<Inhibitory_Neuron>& IN,
                 std::vector<Thalamocortical_Neuron>& TC,
                 std::vector<Reticular_Neuron>& RE) {
    /* Parameters for the parallelization */
    extern const int N_Cores;

    /* First get all the RK terms */
    for (unsigned i=0; i < 4; i++) {
        #pragma omp parallel for num_threads(N_Cores) schedule(dynamic)
        for(auto it = PY.begin(); it < PY.end(); ++it)
            it->set_RK(i);

        #pragma omp parallel for num_threads(N_Cores) schedule(dynamic)
        for(auto it = IN.begin(); it < IN.end(); ++it)
            it->set_RK(i);

        #pragma omp parallel for num_threads(N_Cores) schedule(dynamic)
        for(auto it = TC.begin(); it < TC.end(); ++it)
            it->set_RK(i);

        #pragma omp parallel for num_threads(N_Cores) schedule(dynamic)
        for(auto it = RE.begin(); it < RE.end(); ++it)
            it->set_RK(i);
    }

    /* Add the RK terms up*/
    #pragma omp parallel for num_threads(N_Cores) schedule(dynamic)
    for(auto it = PY.begin(); it < PY.end(); ++it)
        it->add_RK();

    #pragma omp parallel for num_threads(N_Cores) schedule(dynamic)
    for(auto it = IN.begin(); it < IN.end(); ++it)
        it->add_RK();

    #pragma omp parallel for num_threads(N_Cores) schedule(dynamic)
    for(auto it = TC.begin(); it < TC.end(); ++it)
        it->add_RK();

    #pragma omp parallel for num_threads(N_Cores) schedule(dynamic)
    for(auto it = RE.begin(); it < RE.end(); ++it)
        it->add_RK();
}

#endif // ODE_H