86 lines
3.3 KiB
C++
86 lines
3.3 KiB
C++
#include "../util/sim/fluidsim.hpp"
|
|
#include "../util/timing_decorator.cpp"
|
|
|
|
int main() {
|
|
fluidSim sim;
|
|
|
|
// Simulation settings
|
|
const int TOTAL_FRAMES = 10000;
|
|
const int WIDTH = 1024;
|
|
const int HEIGHT = 1024;
|
|
|
|
// Setup Camera
|
|
Camera cam;
|
|
cam.origin = Eigen::Vector3f(0.0f, 4000.0f, -5800.0f);
|
|
cam.direction = (Eigen::Vector3f(0.0f, 0.0f, 0.0f) - cam.origin).normalized();
|
|
cam.up = Eigen::Vector3f(0.0f, 1.0f, 0.0f);
|
|
cam.fov = 60.0f;
|
|
|
|
// Base particle template
|
|
fluidParticle baseParticle;
|
|
baseParticle.velocity = Eigen::Vector3f::Zero();
|
|
baseParticle.acceleration = Eigen::Vector3f::Zero();
|
|
baseParticle.forceAccumulator = Eigen::Vector3f::Zero();
|
|
baseParticle.pressure = 0.0f;
|
|
baseParticle.viscosity = 8.0f;
|
|
baseParticle.restitution = 200.0f;
|
|
|
|
std::cout << "Starting Fluid Simulation..." << std::endl;
|
|
sim.grid.setLODFalloff(0.001);
|
|
sim.grid.setLODMinDistance(4096);
|
|
|
|
for (int frameIdx = 0; frameIdx < TOTAL_FRAMES; frameIdx++) {
|
|
|
|
if (frameIdx < (TOTAL_FRAMES * 0.1f)) {
|
|
if (frameIdx % 1 == 0) {
|
|
//float t = static_cast<float>(frameIdx) / (TOTAL_FRAMES * 0.9f);
|
|
int spawnCount = 2; // + static_cast<int>(t * 4);
|
|
sim.spawnParticles(baseParticle, spawnCount);
|
|
}
|
|
}
|
|
|
|
sim.applyPhysics();
|
|
|
|
if (frameIdx % 100 == 0) {
|
|
std::cout << "Rendering Frame " << frameIdx << " / " << TOTAL_FRAMES << std::endl;
|
|
|
|
frame renderedFrame = sim.grid.renderFrame(cam, HEIGHT, WIDTH, frame::colormap::RGB, 4, 5, true, false);
|
|
|
|
std::stringstream ss;
|
|
ss << "output/frame_" << std::setw(4) << std::setfill('0') << frameIdx << ".bmp";
|
|
BMPWriter::saveBMP(ss.str(), renderedFrame);
|
|
} else if (frameIdx % 25 == 0) {
|
|
std::cout << "Rendering quick Frame " << frameIdx << " / " << TOTAL_FRAMES << std::endl;
|
|
|
|
frame renderedFrame = sim.grid.renderFrame(cam, HEIGHT, WIDTH, frame::colormap::RGB, 1, 1, true, false);
|
|
|
|
std::stringstream ss;
|
|
ss << "output/frame_" << std::setw(4) << std::setfill('0') << frameIdx << ".bmp";
|
|
BMPWriter::saveBMP(ss.str(), renderedFrame);
|
|
} else if (frameIdx % 5 == 0) {
|
|
std::cout << "Rendering ultrafast Frame " << frameIdx << " / " << TOTAL_FRAMES << std::endl;
|
|
|
|
frame renderedFrame = sim.grid.fastRenderFrame(cam, HEIGHT / 2, WIDTH / 2, frame::colormap::RGB);
|
|
renderedFrame.resize(HEIGHT, WIDTH, frame::interpolation::NEAREST);
|
|
|
|
std::stringstream ss;
|
|
ss << "output/frame_" << std::setw(4) << std::setfill('0') << frameIdx << ".bmp";
|
|
BMPWriter::saveBMP(ss.str(), renderedFrame);
|
|
}
|
|
|
|
if (frameIdx % 500 == 0) {
|
|
sim.grid.printStats();
|
|
}
|
|
}
|
|
|
|
sim.grid.printStats();
|
|
frame renderedFrame = sim.grid.renderFrame(cam, HEIGHT, WIDTH, frame::colormap::RGB, 5, 7, true, false);
|
|
|
|
std::stringstream ss;
|
|
ss << "output/frame_" << std::setw(4) << std::setfill('0') << TOTAL_FRAMES << ".bmp";
|
|
BMPWriter::saveBMP(ss.str(), renderedFrame);
|
|
|
|
std::cout << "Simulation Complete." << std::endl;
|
|
FunctionTimer::printStats(FunctionTimer::Mode::ENHANCED);
|
|
return 0;
|
|
} |