forked from bartvdbraak/blender
117 lines
3.0 KiB
C++
117 lines
3.0 KiB
C++
|
|
||
|
|
||
|
// DO NOT EDIT !
|
||
|
// This file is generated using the MantaFlow preprocessor (prep generate).
|
||
|
|
||
|
/******************************************************************************
|
||
|
*
|
||
|
* 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 sheets
|
||
|
* (warning, the vortex methods are currently experimental, and not fully supported!)
|
||
|
*
|
||
|
******************************************************************************/
|
||
|
|
||
|
#include "vortexsheet.h"
|
||
|
#include "solvana.h"
|
||
|
|
||
|
using namespace std;
|
||
|
namespace Manta {
|
||
|
|
||
|
// *****************************************************************************
|
||
|
// VorticityChannel class members
|
||
|
|
||
|
// *****************************************************************************
|
||
|
// VortexSheet Mesh class members
|
||
|
|
||
|
VortexSheetMesh::VortexSheetMesh(FluidSolver *parent) : Mesh(parent), mTexOffset(0.0f)
|
||
|
{
|
||
|
addTriChannel(&mVorticity);
|
||
|
addNodeChannel(&mTex1);
|
||
|
addNodeChannel(&mTex2);
|
||
|
addNodeChannel(&mTurb);
|
||
|
}
|
||
|
|
||
|
Mesh *VortexSheetMesh::clone()
|
||
|
{
|
||
|
VortexSheetMesh *nm = new VortexSheetMesh(mParent);
|
||
|
*nm = *this;
|
||
|
nm->setName(getName());
|
||
|
return nm;
|
||
|
}
|
||
|
|
||
|
void VortexSheetMesh::calcVorticity()
|
||
|
{
|
||
|
for (size_t tri = 0; tri < mTris.size(); tri++) {
|
||
|
VortexSheetInfo &v = mVorticity.data[tri];
|
||
|
Vec3 e0 = getEdge(tri, 0), e1 = getEdge(tri, 1), e2 = getEdge(tri, 2);
|
||
|
Real area = getFaceArea(tri);
|
||
|
|
||
|
if (area < 1e-10) {
|
||
|
v.smokeAmount = 0;
|
||
|
v.vorticity = 0;
|
||
|
}
|
||
|
else {
|
||
|
v.smokeAmount = 0;
|
||
|
v.vorticity = (v.circulation[0] * e0 + v.circulation[1] * e1 + v.circulation[2] * e2) / area;
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
void VortexSheetMesh::calcCirculation()
|
||
|
{
|
||
|
for (size_t tri = 0; tri < mTris.size(); tri++) {
|
||
|
VortexSheetInfo &v = mVorticity.data[tri];
|
||
|
Vec3 e0 = getEdge(tri, 0), e1 = getEdge(tri, 1), e2 = getEdge(tri, 2);
|
||
|
Real area = getFaceArea(tri);
|
||
|
|
||
|
if (area < 1e-10 || normSquare(v.vorticity) < 1e-10) {
|
||
|
v.circulation = 0;
|
||
|
continue;
|
||
|
}
|
||
|
|
||
|
float cx, cy, cz;
|
||
|
SolveOverconstraint34(e0.x,
|
||
|
e0.y,
|
||
|
e0.z,
|
||
|
e1.x,
|
||
|
e1.y,
|
||
|
e1.z,
|
||
|
e2.x,
|
||
|
e2.y,
|
||
|
e2.z,
|
||
|
v.vorticity.x,
|
||
|
v.vorticity.y,
|
||
|
v.vorticity.z,
|
||
|
cx,
|
||
|
cy,
|
||
|
cz);
|
||
|
v.circulation = Vec3(cx, cy, cz) * area;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
void VortexSheetMesh::resetTex1()
|
||
|
{
|
||
|
for (size_t i = 0; i < mNodes.size(); i++)
|
||
|
mTex1.data[i] = mNodes[i].pos + mTexOffset;
|
||
|
}
|
||
|
|
||
|
void VortexSheetMesh::resetTex2()
|
||
|
{
|
||
|
for (size_t i = 0; i < mNodes.size(); i++)
|
||
|
mTex2.data[i] = mNodes[i].pos + mTexOffset;
|
||
|
}
|
||
|
|
||
|
void VortexSheetMesh::reinitTexCoords()
|
||
|
{
|
||
|
resetTex1();
|
||
|
resetTex2();
|
||
|
}
|
||
|
|
||
|
}; // namespace Manta
|