Integrator.h

/*********************************************************************
 * @file  Integrator.h
 * 
 * @brief Declaration of Integrator class with factory-like method
 * GetIntegrator()
 *
 * Defines hashable Integrator::Conf struct that allows operator
 * resuse. Common CPU/GPU kernels implemented in Integrator/kernels.h
 * with various BD/MD integrators implemented on different backends in
 * Integrator/CUDA.h and Integrator/CPU.h
 *********************************************************************/
#pragma once

#include <cassert>
#include <iostream>
#include <map>
#include "PatchOp.h"

class Integrator : public BasePatchOp {
public:
    virtual void compute(Patch* patch) = 0;
    int num_patches() const { return 1; };

    // Following relates to lazy initialized factory method
    struct Conf {
	enum Object   { Particle, RigidBody };
	enum Algorithm { BD, MD };
	enum Backend   { Default, CUDA, CPU };    

	Object object_type;
	Algorithm algorithm;
	Backend backend;

	explicit operator int() const {return object_type*16 + algorithm*4 + backend;};
    };
        
    static Integrator* GetIntegrator(Conf& conf);
	    	
protected:
    static std::map<Conf, Integrator*> _integrators;

};

#include "Integrator/kernels.h"
#include "Integrator/CUDA.h"
#include "Integrator/CPU.h"