69 lines
1.4 KiB
C++
69 lines
1.4 KiB
C++
|
#include <iostream>
|
||
|
|
#include "sphSmooth.hpp"
|
||
|
|
#include "yorick.hpp"
|
||
|
|
#include "mykdtree.hpp"
|
||
|
|
|
||
|
|
using namespace std;
|
||
|
|
using namespace CosmoTool;
|
||
|
|
|
||
|
|
#define NX 1024
|
||
|
|
#define ND 2
|
||
|
|
|
||
|
|
typedef SPHSmooth<char,ND> MySmooth;
|
||
|
|
typedef MySmooth::SPHTree MyTree;
|
||
|
|
typedef MySmooth::SPHNode MyNode;
|
||
|
|
typedef MySmooth::SPHCell MyCell;
|
||
|
|
|
||
|
|
double unit_fun(const char& c)
|
||
|
|
{
|
||
|
|
return 1.0;
|
||
|
|
}
|
||
|
|
|
||
|
|
int main()
|
||
|
|
{
|
||
|
|
uint32_t Ncells = 10000;
|
||
|
|
MyCell *cells = new MyCell[Ncells];
|
||
|
|
|
||
|
|
for (int i = 0; i < Ncells; i++)
|
||
|
|
{
|
||
|
|
cells[i].active = true;
|
||
|
|
for (int l = 0; l < ND; l++)
|
||
|
|
cells[i].coord[l] = drand48();
|
||
|
|
cells[i].val.weight = 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
MyTree tree(cells, Ncells);
|
||
|
|
MySmooth smooth(&tree, 16);
|
||
|
|
|
||
|
|
for (uint32_t iy = 0; iy < NX; iy++)
|
||
|
|
{
|
||
|
|
cout << "iy=" << iy << endl;
|
||
|
|
for (uint32_t ix = 0; ix < NX; ix++)
|
||
|
|
{
|
||
|
|
MyTree::coords c = { 1.0*ix/NX, 1.0*iy/NX };
|
||
|
|
smooth.fetchNeighbours(c);
|
||
|
|
smooth.addGridSite(c);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
uint32_t dims[] = { NX, NX };
|
||
|
|
ProgressiveOutput<ComputePrecision> out =
|
||
|
|
ProgressiveOutput<ComputePrecision>::saveArrayProgressive("out.nc", dims, 2);
|
||
|
|
for (uint32_t iy = 0; iy < NX; iy++)
|
||
|
|
{
|
||
|
|
cout << "iy=" << iy << endl;
|
||
|
|
for (uint32_t ix = 0; ix < NX; ix++)
|
||
|
|
{
|
||
|
|
MyTree::coords c = { 1.0*ix/NX, 1.0*iy/NX };
|
||
|
|
smooth.fetchNeighbours(c);
|
||
|
|
out.put(smooth.computeSmoothedValue(c, unit_fun));
|
||
|
|
|
||
|
|
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
return 0;
|
||
|
|
}
|