void InitPsi(ref RenderTexture psi1, ref RenderTexture psi2)
{
CS.SetTexture(kernelInitPsi, "Psi1", psi1);
CS.SetTexture(kernelInitPsi, "Psi2", psi2);
//var volecity2 = Vector3.Dot(GlobalVelocity, GlobalVelocity);
//float omega = volecity2 / (2 * isf.hbar);
// CS.SetFloat("omega_t", omega * isf.current_tick * isf.estimate_dt);
CS.SetFloat("omega_t", 0);
CS.SetVector("velocity", GlobalVelocity / isf.hbar);
DispatchCS(kernelInitPsi);
InitCylinderPsi(ref psi1, Direction1, Center1, Radius1, Thickness);
InitCylinderPsi(ref psi1, Direction2, Center2, Radius2, Thickness);
// fft.ExportComplex3D(psi1, "test/leap.psi1.json");
isf.Normalize(ref psi1, ref psi2);
isf.PressureProject(ref psi1, ref psi2);
}
// Update is called once per frame
void Update()
{
isf.current_tick += 1;
isf.ShroedingerIntegration(ref psi1, ref psi2);
isf.Normalize(ref psi1, ref psi2);
isf.PressureProject(ref psi1, ref psi2);
InitilizeNozzlePsi(ref psi1, ref psi2);
// UpdateObstacle();
isf.VelocityOneForm(ref psi1, ref psi2, isf.hbar);
isf.ReprojectToGrid();
particles.Emulate();
ClampParticles();
particles.DoRender();
}
// Update is called once per frame
void Update()
{
isf.current_tick += 1;
isf.ShroedingerIntegration(ref psi1, ref psi2);
isf.Normalize(ref psi1, ref psi2);
isf.PressureProject(ref psi1, ref psi2);
//DispatchCS(kernelFlushPsiMask);
//ComputePsiMask();
//UpdateGlabalMaskedPsi(ref psi1, ref psi2);
UpdateGlabalMaskedPsi(ref psi1, ref psi2);
// UpdatePsi(ref psi1, ref psi2);
isf.VelocityOneForm(ref psi1, ref psi2, isf.hbar);
isf.ReprojectToGrid();
particles.Emulate();
// Clamp();
particles.DoRender();
}
void TakeTestN(Vector3Int N)
{
isf.N = N;
isf.InitISF();
psi1 = FFT.CreateRenderTexture3D(N[0], N[1], N[2], RenderTextureFormat.RGFloat);
psi2 = FFT.CreateRenderTexture3D(N[0], N[1], N[2], RenderTextureFormat.RGFloat);
isf.InitializePsi(ref psi1, ref psi2);
var watch = new System.Diagnostics.Stopwatch();
watch.Start();
for (int i = 0; i < 5; ++i)
{
isf.current_tick += 1;
isf.ShroedingerIntegration(ref psi1, ref psi2);
isf.Normalize(ref psi1, ref psi2);
isf.PressureProject(ref psi1, ref psi2);
fft.ExportComplex3D(psi1, null);
Debug.Log(" At: " + i.ToString());
}
watch.Stop();
BenchmarkSingleResult ben = new BenchmarkSingleResult();
ben.milli = watch.ElapsedMilliseconds / 5.0;
ben.Nx = N.x;
ben.Ny = N.y;
ben.Nz = N.z;
ben.scale = N.x * N.y * N.z;
result.results.Add(ben);
}
static void Main(string[] args)
{
//PARAMETERS
int[] vol_size = { 10, 5, 5 }; // box size
int[] vol_res = { 64, 32, 32 }; // volume resolution
float hbar = (float)0.1; // Planck constant
float dt = 1 / (float)24; // time step
int tmax = 85;
float[] background_vel = { (float)-0.2, 0, 0 };
float r1 = (float)1.5;
float r2 = (float)0.9;
float[] n1 = { -1, 0, 0 };
float[] n2 = { -1, 0, 0 };
float[] cen1 = { vol_size[0] / 2f, vol_size[1] / 2f, vol_size[2] / 2f };
float[] cen2 = { vol_size[0] / 2f, vol_size[1] / 2f, vol_size[2] / 2f };
int n_particles = 10000;
//INITIALISATION
ISF.Init(vol_size, vol_res, hbar, dt);
Particles.init(n_particles);
//init psi
float[] kvec = { background_vel[0] / hbar, background_vel[1] / hbar, background_vel[2] / hbar };
float phase;
var tmp1 = new cuFloatComplex[ISF.properties.resx, ISF.properties.resy, ISF.properties.resz];
var tmp2 = new cuFloatComplex[ISF.properties.resx, ISF.properties.resy, ISF.properties.resz];
Complex tmp;
for (int i = 0; i < vol_res[0]; i++)
{
for (int j = 0; j < vol_res[1]; j++)
{
for (int k = 0; k < vol_res[2]; k++)
{
phase = kvec[0] * ISF.properties.px[i, j, k] +
kvec[1] * ISF.properties.py[i, j, k] +
kvec[2] * ISF.properties.pz[i, j, k];
tmp = Complex.Exp(Complex.ImaginaryOne * phase);
tmp1[i, j, k] = new cuFloatComplex((float)tmp.Real, (float)tmp.Imaginary);
tmp2[i, j, k] = new cuFloatComplex((float)(tmp.Real * 0.01), (float)(tmp.Imaginary * 0.01));
}
}
}
float d = ISF.properties.dx * 5;
ISF.add_circle(tmp1, cen1, n1, r1, d);
ISF.add_circle(tmp1, cen2, n2, r2, d);
ISF.psi1.CopyToDevice(tmp1);
ISF.psi2.CopyToDevice(tmp2);
ISF.Normalize();
ISF.PressureProject();
//init particles
var x = new float[n_particles];
var y = new float[n_particles];
var z = new float[n_particles];
Random rnd = new Random();
for (int i = 0; i < n_particles; i++)
{
y[i] = (float)(rnd.NextDouble() * 4 + 0.5);
z[i] = (float)(rnd.NextDouble() * 4 + 0.5);
x[i] = 5;
}
Particles.add_particles(x, y, z, n_particles);
Velocity vel = new Velocity(ISF.properties.resx, ISF.properties.resy, ISF.properties.resz);
//MAIN ITERATION
Console.Out.WriteLine("Start");
int itermax = (int)Math.Ceiling(tmax / dt);
for (int i = 0; i < 100; i++)
{
//incompressible Schroedinger flow
ISF.update_space();
//particle update
ISF.update_velocities(vel);
Particles.calculate_movement(vel);
}
float[] xx = Particles.x;
float[] yy = Particles.y;
float[] zz = Particles.z;
for (int i = 0; i < 20; i++)
{
Console.Out.WriteLine(xx[i] + " " + yy[i] + " " + zz[i]);
}
}