/******************************************************************************
 *
 * MantaFlow fluid solver framework
 * Copyright 2011 Tobias Pfaff, Nils Thuerey 
 *
 * This program is free software, distributed under the terms of the
 * Apache License, Version 2.0 
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Vortex particles
 * (warning, the vortex methods are currently experimental, and not fully supported!)
 *
 ******************************************************************************/

#ifndef _VORTEXPART_H
#define _VORTEXPART_H

#include "particle.h"

namespace Manta {
class Mesh;
	
struct VortexParticleData {
	VortexParticleData() : pos(0.0),vorticity(0.0),sigma(0),flag(0) {}
	VortexParticleData(const Vec3& p, const Vec3& v, Real sig) : pos(p),vorticity(v),sigma(sig),flag(0) {}
	Vec3 pos, vorticity;
	Real sigma;
	int flag;    
	static ParticleBase::SystemType getType() { return ParticleBase::VORTEX; }
};

//! Vortex particles
PYTHON() class VortexParticleSystem : public ParticleSystem<VortexParticleData> {
public:
	PYTHON() VortexParticleSystem(FluidSolver* parent);
  
	PYTHON() void advectSelf(Real scale=1.0, int integrationMode=IntRK4);
	PYTHON() void applyToMesh(Mesh& mesh, Real scale=1.0, int integrationMode=IntRK4);
	
	virtual ParticleBase* clone();
};

} // namespace


#endif