coProcessor3D.h

Go to the documentation of this file.

/* This file is part of the Palabos library.
 *
 * Copyright (C) 2011 FlowKit Sarl
 * Avenue de Chailly 23
 * 1012 Lausanne, Switzerland
 * E-mail contact: contact@flowkit.com
 *
 * The most recent release of Palabos can be downloaded at 
 * <http://www.palabos.org/>
 *
 * The library Palabos is free software: you can redistribute it and/or
 * modify it under the terms of the GNU Affero General Public License as
 * published by the Free Software Foundation, either version 3 of the
 * License, or (at your option) any later version.
 *
 * The library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
*/

#ifndef CO_PROCESSOR_3D_H
#define CO_PROCESSOR_3D_H

#include "core/globalDefs.h"
#include "core/geometry3D.h"
#include "atomicBlock/blockLattice3D.h"
#include <map>

namespace plb {


template<typename T>
class CoProcessor3D {
public:
    virtual ~CoProcessor3D() { }


    virtual int addDomain(plint nx, plint ny, plint nz, T omega, int& domainHandle) =0;


    virtual int send(int domainHandle, Box3D const& subDomain, std::vector<char> const& data) =0;


    virtual int receive(int domainHandle, Box3D const& subDomain, std::vector<char>& data) const =0;



    virtual int collideAndStream(int domainHandle) =0;
};

template<typename T>
class D3Q19ExampleCoProcessor3D : public CoProcessor3D<T>
{
public:
    virtual int addDomain(plint nx, plint ny, plint nz, T omega, int& domainHandle);
    virtual int send(int domainHandle, Box3D const& subDomain, std::vector<char> const& data);
    virtual int receive(int domainHandle, Box3D const& subDomain, std::vector<char>& data) const;
    virtual int collideAndStream(int domainHandle);
private:
    std::map<int, BlockLattice3D<T,descriptors::D3Q19Descriptor> > domains;
};


template<typename T>
class D3Q19CudaCoProcessor3D : public CoProcessor3D<T>
{
public:
    D3Q19CudaCoProcessor3D();
    ~D3Q19CudaCoProcessor3D();
    virtual int addDomain(plint nx, plint ny, plint nz, T omega, int& domainHandle);
//    virtual int addDomain(Box3D const& domain, T omega);
    virtual int send(int domainHandle, Box3D const& subDomain, std::vector<char> const& data);
    virtual int receive(int domainHandle, Box3D const& subDomain, std::vector<char>& data) const;
    virtual int collideAndStream(int domainHandle);
private:
    /*
    PyObject *createCUDASimulation(void);
    void prepareSimulationToRun(void);

    PyObject *geometryClass;
    PyObject *simulationClass;
    PyObject *simulation;
    PyObject *pyCUDA;
    PyObject *copyFunctionDtoH;
    PyObject *copyFunctionHtoD;
    PyObject *buffer;
    float *data;
    int nx;
    int ny;
    int nz;
    int basis;
    int size;
    int floatSize;
    int strideXSailfish;
    int strideYSailfish;
    int strideZSailfish;
    int strideYPalabos;
    int strideZPalabos;
    */
};

namespace global {

template<typename T>
inline CoProcessor3D<T>& defaultCoProcessor3D() {
    //  D3Q19ExampleCoProcessor3D singleton by a
    //  D3Q19CudaCoProcessor3D singleton to get your code running.
    static D3Q19CudaCoProcessor3D<T> singleton;
    return singleton;
};

}

}  // namespace plb

#endif  // CO_PROCESSOR_3D_H