Merge branch 'master' of bitbucket.org:glavaux/cosmotool
This commit is contained in:
commit
0124431cd9
@ -31,13 +31,15 @@ find_library(NETCDF_LIBRARY netcdf)
|
|||||||
find_library(GSL_LIBRARY gsl)
|
find_library(GSL_LIBRARY gsl)
|
||||||
find_library(GSLCBLAS_LIBRARY gslcblas)
|
find_library(GSLCBLAS_LIBRARY gslcblas)
|
||||||
|
|
||||||
|
find_package(Boost 1.53)
|
||||||
|
|
||||||
if (ENABLE_SHARP)
|
if (ENABLE_SHARP)
|
||||||
SET(SHARP_SOURCE ${CMAKE_SOURCE_DIR}/external/sharp)
|
SET(SHARP_SOURCE ${CMAKE_SOURCE_DIR}/external/sharp)
|
||||||
SET(DEP_BUILD ${CMAKE_SOURCE_DIR}/external/sharp/auto)
|
SET(DEP_BUILD ${CMAKE_SOURCE_DIR}/external/sharp/auto)
|
||||||
ExternalProject_Add(sharp
|
ExternalProject_Add(sharp
|
||||||
SOURCE_DIR ${SHARP_SOURCE}
|
SOURCE_DIR ${SHARP_SOURCE}
|
||||||
BUILD_IN_SOURCE 1
|
BUILD_IN_SOURCE 1
|
||||||
CONFIGURE_COMMAND ${SHARP_SOURCE}/configure --prefix=${DEP_BUILD}
|
CONFIGURE_COMMAND ${SHARP_SOURCE}/configure "CC=${CMAKE_C_COMPILER}" "CXX=${CMAKE_CXX_COMPILER}" --prefix=${DEP_BUILD}
|
||||||
BUILD_COMMAND ${CMAKE_MAKE_PROGRAM}
|
BUILD_COMMAND ${CMAKE_MAKE_PROGRAM}
|
||||||
INSTALL_COMMAND echo "No install"
|
INSTALL_COMMAND echo "No install"
|
||||||
)
|
)
|
||||||
@ -67,12 +69,28 @@ pkg_check_modules(FFTW3 fftw3>=3.3)
|
|||||||
pkg_check_modules(FFTW3F fftw3f>=3.3)
|
pkg_check_modules(FFTW3F fftw3f>=3.3)
|
||||||
pkg_check_modules(EIGEN3 eigen3)
|
pkg_check_modules(EIGEN3 eigen3)
|
||||||
|
|
||||||
|
|
||||||
|
find_library(FFTW3F_LIBRARY_FULL fftw3f PATHS ${FFTW3F_LIBDIR} NO_DEFAULT_PATH)
|
||||||
|
find_library(FFTW3_LIBRARY_FULL fftw3 PATHS ${FFTW3_LIBDIR} NO_DEFAULT_PATH)
|
||||||
|
|
||||||
include(FindPackageHandleStandardArgs)
|
include(FindPackageHandleStandardArgs)
|
||||||
set(NETCDF_FIND_REQUIRED TRUE)
|
set(NETCDF_FIND_REQUIRED TRUE)
|
||||||
set(GSL_FIND_REQUIRED TRUE)
|
set(GSL_FIND_REQUIRED TRUE)
|
||||||
FIND_PACKAGE_HANDLE_STANDARD_ARGS(NetCDF DEFAULT_MSG NETCDF_LIBRARY NETCDFCPP_LIBRARY NETCDF_INCLUDE_PATH)
|
FIND_PACKAGE_HANDLE_STANDARD_ARGS(NetCDF DEFAULT_MSG NETCDF_LIBRARY NETCDFCPP_LIBRARY NETCDF_INCLUDE_PATH)
|
||||||
FIND_PACKAGE_HANDLE_STANDARD_ARGS(GSL DEFAULT_MSG GSL_LIBRARY GSLCBLAS_LIBRARY GSL_INCLUDE_PATH)
|
FIND_PACKAGE_HANDLE_STANDARD_ARGS(GSL DEFAULT_MSG GSL_LIBRARY GSLCBLAS_LIBRARY GSL_INCLUDE_PATH)
|
||||||
|
|
||||||
|
IF(ENABLE_OPENMP)
|
||||||
|
|
||||||
|
IF (NOT OPENMP_FOUND)
|
||||||
|
MESSAGE(ERROR "No known compiler option for enabling OpenMP")
|
||||||
|
ENDIF(NOT OPENMP_FOUND)
|
||||||
|
|
||||||
|
SET(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} ${OpenMP_C_FLAGS}")
|
||||||
|
SET(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${OpenMP_CXX_FLAGS}")
|
||||||
|
SET(CMAKE_EXE_LINKER_FLAGS "${CMAKE_EXE_LINKED_FLAGS} ${OpenMP_C_FLAGS}")
|
||||||
|
ENDIF(ENABLE_OPENMP)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
set(GSL_LIBRARIES ${GSL_LIBRARY} ${GSLCBLAS_LIBRARY})
|
set(GSL_LIBRARIES ${GSL_LIBRARY} ${GSLCBLAS_LIBRARY})
|
||||||
|
|
||||||
|
@ -1,5 +1,7 @@
|
|||||||
SET(tolink ${GSL_LIBRARIES} CosmoTool ${CosmoTool_LIBS})
|
SET(tolink ${GSL_LIBRARIES} CosmoTool ${CosmoTool_LIBS})
|
||||||
include_directories(${CMAKE_SOURCE_DIR}/src ${FFTW3_INCLUDE_DIRS} ${EIGEN3_INCLUDE_DIRS} ${NETCDF_INCLUDE_PATH} ${GSL_INCLUDE_PATH})
|
include_directories(${CMAKE_SOURCE_DIR}/src)
|
||||||
|
include_directories(${FFTW3_INCLUDE_DIRS}
|
||||||
|
${EIGEN3_INCLUDE_DIRS} ${NETCDF_INCLUDE_PATH} ${GSL_INCLUDE_PATH})
|
||||||
|
|
||||||
IF(SHARP_INCLUDE_PATH)
|
IF(SHARP_INCLUDE_PATH)
|
||||||
include_directories(BEFORE ${SHARP_INCLUDE_PATH})
|
include_directories(BEFORE ${SHARP_INCLUDE_PATH})
|
||||||
@ -11,9 +13,6 @@ target_link_libraries(testBQueue ${tolink})
|
|||||||
add_executable(testInterpolate testInterpolate.cpp)
|
add_executable(testInterpolate testInterpolate.cpp)
|
||||||
target_link_libraries(testInterpolate ${tolink})
|
target_link_libraries(testInterpolate ${tolink})
|
||||||
|
|
||||||
add_executable(testSmooth testSmooth.cpp)
|
|
||||||
target_link_libraries(testSmooth ${tolink})
|
|
||||||
|
|
||||||
add_executable(testkd testkd.cpp)
|
add_executable(testkd testkd.cpp)
|
||||||
target_link_libraries(testkd ${tolink})
|
target_link_libraries(testkd ${tolink})
|
||||||
|
|
||||||
@ -34,7 +33,14 @@ target_link_libraries(testPool ${tolink})
|
|||||||
|
|
||||||
if (HDF5_FOUND)
|
if (HDF5_FOUND)
|
||||||
add_executable(testReadFlash testReadFlash.cpp)
|
add_executable(testReadFlash testReadFlash.cpp)
|
||||||
target_link_libraries(testReadFlash ${tolink})
|
target_link_libraries(testReadFlash ${tolink})
|
||||||
|
|
||||||
|
add_executable(testHDF5 testHDF5.cpp)
|
||||||
|
target_link_libraries(testHDF5 ${tolink})
|
||||||
|
|
||||||
|
add_executable(gadgetToArray gadgetToArray.cpp)
|
||||||
|
target_link_libraries(gadgetToArray ${tolink})
|
||||||
|
|
||||||
endif (HDF5_FOUND)
|
endif (HDF5_FOUND)
|
||||||
|
|
||||||
|
|
||||||
@ -48,18 +54,36 @@ add_executable(testBSP testBSP.cpp)
|
|||||||
target_link_libraries(testBSP ${tolink})
|
target_link_libraries(testBSP ${tolink})
|
||||||
|
|
||||||
if (FFTW3_FOUND AND FFTW3F_FOUND AND EIGEN3_FOUND)
|
if (FFTW3_FOUND AND FFTW3F_FOUND AND EIGEN3_FOUND)
|
||||||
|
IF (FFTW3F_LIBRARY_FULL)
|
||||||
|
SET(FFTW3_LIB ${FFTW3F_LIBRARY_FULL})
|
||||||
|
ENDIF (FFTW3F_LIBRARY_FULL)
|
||||||
|
IF (FFTW3_LIBRARY_FULL)
|
||||||
|
SET(FFTW3_LIB ${FFTW3_LIB} ${FFTW3_LIBRARY_FULL})
|
||||||
|
ENDIF (FFTW3_LIBRARY_FULL)
|
||||||
add_executable(test_fft_calls test_fft_calls.cpp)
|
add_executable(test_fft_calls test_fft_calls.cpp)
|
||||||
target_link_libraries(test_fft_calls ${tolink} ${FFTW3_LIBRARIES} ${FFTW3F_LIBRARIES})
|
target_link_libraries(test_fft_calls ${tolink} ${FFTW3_LIB})
|
||||||
endif (FFTW3_FOUND AND FFTW3F_FOUND AND EIGEN3_FOUND)
|
endif (FFTW3_FOUND AND FFTW3F_FOUND AND EIGEN3_FOUND)
|
||||||
|
|
||||||
if (ENABLE_SHARP AND SHARP_LIBRARY AND SHARP_INCLUDE_PATH AND EIGEN3_FOUND)
|
if (ENABLE_SHARP AND SHARP_LIBRARY AND SHARP_INCLUDE_PATH AND EIGEN3_FOUND)
|
||||||
include_directories(${SHARP_INCLUDE_PATH})
|
include_directories(${SHARP_INCLUDE_PATH})
|
||||||
add_executable(test_healpix_calls test_healpix_calls.cpp)
|
add_executable(test_healpix_calls test_healpix_calls.cpp)
|
||||||
target_link_libraries(test_healpix_calls ${tolink} ${SHARP_LIBRARIES})
|
target_link_libraries(test_healpix_calls ${tolink} ${SHARP_LIBRARIES})
|
||||||
set_target_properties(test_healpix_calls PROPERTIES COMPILE_FLAGS ${OpenMP_CXX_FLAGS} LINK_FLAGS ${OpenMP_CXX_FLAGS})
|
set_target_properties(test_healpix_calls PROPERTIES COMPILE_FLAGS "${OpenMP_CXX_FLAGS}" LINK_FLAGS "${OpenMP_CXX_FLAGS}")
|
||||||
add_dependencies(test_healpix_calls sharp)
|
add_dependencies(test_healpix_calls sharp)
|
||||||
endif (ENABLE_SHARP AND SHARP_LIBRARY AND SHARP_INCLUDE_PATH AND EIGEN3_FOUND)
|
endif (ENABLE_SHARP AND SHARP_LIBRARY AND SHARP_INCLUDE_PATH AND EIGEN3_FOUND)
|
||||||
|
|
||||||
add_executable(test_cosmopower test_cosmopower.cpp)
|
add_executable(test_cosmopower test_cosmopower.cpp)
|
||||||
target_link_libraries(test_cosmopower ${tolink})
|
target_link_libraries(test_cosmopower ${tolink})
|
||||||
|
|
||||||
|
if (Boost_FOUND)
|
||||||
|
include_directories(${Boost_INCLUDE_DIRS})
|
||||||
|
|
||||||
|
add_executable(testSmooth testSmooth.cpp)
|
||||||
|
target_link_libraries(testSmooth ${tolink})
|
||||||
|
|
||||||
|
add_executable(simple3DFilter simple3DFilter.cpp)
|
||||||
|
target_link_libraries(simple3DFilter ${tolink})
|
||||||
|
endif (Boost_FOUND)
|
||||||
|
|
||||||
|
add_executable(gadgetToDensity gadgetToDensity.cpp)
|
||||||
|
target_link_libraries(gadgetToDensity ${tolink})
|
||||||
|
57
sample/gadgetToArray.cpp
Normal file
57
sample/gadgetToArray.cpp
Normal file
@ -0,0 +1,57 @@
|
|||||||
|
#include <cmath>
|
||||||
|
#include <iostream>
|
||||||
|
#include <cstdlib>
|
||||||
|
#include "cic.hpp"
|
||||||
|
#include "loadGadget.hpp"
|
||||||
|
#include "miniargs.hpp"
|
||||||
|
#include <H5Cpp.h>
|
||||||
|
#include "hdf5_array.hpp"
|
||||||
|
|
||||||
|
using namespace CosmoTool;
|
||||||
|
using namespace std;
|
||||||
|
|
||||||
|
int main(int argc, char **argv)
|
||||||
|
{
|
||||||
|
typedef boost::multi_array<float, 2> array_type;
|
||||||
|
uint32_t res;
|
||||||
|
char *fname;
|
||||||
|
int id;
|
||||||
|
|
||||||
|
MiniArgDesc desc[] = {
|
||||||
|
{ "SNAPSHOT", &fname, MINIARG_STRING },
|
||||||
|
{ 0, 0, MINIARG_NULL }
|
||||||
|
};
|
||||||
|
|
||||||
|
if (!parseMiniArgs(argc, argv, desc))
|
||||||
|
return 1;
|
||||||
|
|
||||||
|
H5::H5File f("density.h5", H5F_ACC_TRUNC);
|
||||||
|
|
||||||
|
|
||||||
|
SimuData *p = loadGadgetMulti(fname, 0, 0);
|
||||||
|
double L0 = p->BoxSize/1000;
|
||||||
|
array_type parts(boost::extents[p->TotalNumPart][6]);
|
||||||
|
uint64_t q = 0;
|
||||||
|
|
||||||
|
try {
|
||||||
|
for (int cpuid=0;;cpuid++) {
|
||||||
|
p = loadGadgetMulti(fname, cpuid, NEED_POSITION|NEED_VELOCITY);
|
||||||
|
for (uint32_t i = 0; i < p->NumPart; i++)
|
||||||
|
{
|
||||||
|
parts[q][0] = p->Pos[0][i]/1000;
|
||||||
|
parts[q][1] = p->Pos[1][i]/1000;
|
||||||
|
parts[q][2] = p->Pos[2][i]/1000;
|
||||||
|
parts[q][3] = p->Vel[0][i];
|
||||||
|
parts[q][4] = p->Vel[1][i];
|
||||||
|
parts[q][5] = p->Vel[2][i];
|
||||||
|
q++;
|
||||||
|
}
|
||||||
|
|
||||||
|
delete p;
|
||||||
|
}
|
||||||
|
} catch (const NoSuchFileException& e) {}
|
||||||
|
|
||||||
|
hdf5_write_array(f, "particles", parts);
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
62
sample/gadgetToDensity.cpp
Normal file
62
sample/gadgetToDensity.cpp
Normal file
@ -0,0 +1,62 @@
|
|||||||
|
#include <cmath>
|
||||||
|
#include <iostream>
|
||||||
|
#include <cstdlib>
|
||||||
|
#include "cic.hpp"
|
||||||
|
#include "loadGadget.hpp"
|
||||||
|
#include "miniargs.hpp"
|
||||||
|
#include "yorick.hpp"
|
||||||
|
|
||||||
|
using namespace std;
|
||||||
|
using namespace CosmoTool;
|
||||||
|
|
||||||
|
int main(int argc, char **argv)
|
||||||
|
{
|
||||||
|
uint32_t res;
|
||||||
|
char *fname;
|
||||||
|
int id;
|
||||||
|
|
||||||
|
MiniArgDesc desc[] = {
|
||||||
|
{ "SNAPSHOT", &fname, MINIARG_STRING },
|
||||||
|
{ "ID", &id, MINIARG_INT },
|
||||||
|
{ "RESOLUTION", &res, MINIARG_INT },
|
||||||
|
{ 0, 0, MINIARG_NULL }
|
||||||
|
};
|
||||||
|
|
||||||
|
if (!parseMiniArgs(argc, argv, desc))
|
||||||
|
return 1;
|
||||||
|
|
||||||
|
SimuData *p = loadGadgetMulti(fname, 0, 0);
|
||||||
|
double L0 = p->BoxSize/1000;
|
||||||
|
CICFilter filter(res, L0);
|
||||||
|
|
||||||
|
delete p;
|
||||||
|
|
||||||
|
try {
|
||||||
|
for (int cpuid=0;;cpuid++) {
|
||||||
|
p = loadGadgetMulti(fname, cpuid, NEED_POSITION);
|
||||||
|
for (uint32_t i = 0; i < p->NumPart; i++)
|
||||||
|
{
|
||||||
|
CICParticles a;
|
||||||
|
|
||||||
|
a.mass = 1.0;
|
||||||
|
a.coords[0] = p->Pos[0][i]/1000;
|
||||||
|
a.coords[1] = p->Pos[1][i]/1000;
|
||||||
|
a.coords[2] = p->Pos[2][i]/1000;
|
||||||
|
filter.putParticles(&a, 1);
|
||||||
|
}
|
||||||
|
|
||||||
|
delete p;
|
||||||
|
}
|
||||||
|
} catch (const NoSuchFileException& e) {}
|
||||||
|
|
||||||
|
CICType *denField;
|
||||||
|
uint32_t Ntot;
|
||||||
|
filter.getDensityField(denField, Ntot);
|
||||||
|
|
||||||
|
cout << "L0=" << L0 << endl;
|
||||||
|
cout << "Saving density field" << endl;
|
||||||
|
uint32_t dimList[] = { res, res, res};
|
||||||
|
saveArray("densityField.nc", denField, dimList, 3);
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
204
sample/simple3DFilter.cpp
Normal file
204
sample/simple3DFilter.cpp
Normal file
@ -0,0 +1,204 @@
|
|||||||
|
#include <cassert>
|
||||||
|
#include "yorick.hpp"
|
||||||
|
#include "sphSmooth.hpp"
|
||||||
|
#include "mykdtree.hpp"
|
||||||
|
#include "miniargs.hpp"
|
||||||
|
#include <H5Cpp.h>
|
||||||
|
#include "hdf5_array.hpp"
|
||||||
|
#include <iostream>
|
||||||
|
|
||||||
|
using namespace std;
|
||||||
|
using namespace CosmoTool;
|
||||||
|
|
||||||
|
#define N_SPH 16
|
||||||
|
|
||||||
|
struct VCoord{
|
||||||
|
float v[3];
|
||||||
|
};
|
||||||
|
|
||||||
|
template<int i>
|
||||||
|
ComputePrecision getVelocity(const VCoord& v)
|
||||||
|
{
|
||||||
|
return v.v[i];
|
||||||
|
}
|
||||||
|
|
||||||
|
ComputePrecision getUnity(const VCoord& v)
|
||||||
|
{
|
||||||
|
return 1.0;
|
||||||
|
}
|
||||||
|
|
||||||
|
typedef SPHSmooth<VCoord> MySmooth;
|
||||||
|
typedef MySmooth::SPHTree MyTree;
|
||||||
|
typedef MyTree::Cell MyCell;
|
||||||
|
|
||||||
|
int main(int argc, char **argv)
|
||||||
|
{
|
||||||
|
typedef boost::multi_array<float, 2> array_type;
|
||||||
|
typedef boost::multi_array<float, 3> array3_type;
|
||||||
|
typedef boost::multi_array<float, 4> array4_type;
|
||||||
|
|
||||||
|
char *fname1, *fname2;
|
||||||
|
double rLimit, boxsize, rLimit2, cx, cy, cz;
|
||||||
|
int Nres;
|
||||||
|
|
||||||
|
MiniArgDesc args[] = {
|
||||||
|
{ "INPUT DATA1", &fname1, MINIARG_STRING },
|
||||||
|
{ "RADIUS LIMIT", &rLimit, MINIARG_DOUBLE },
|
||||||
|
{ "BOXSIZE", &boxsize, MINIARG_DOUBLE },
|
||||||
|
{ "RESOLUTION", &Nres, MINIARG_INT },
|
||||||
|
{ "CX", &cx, MINIARG_DOUBLE },
|
||||||
|
{ "CY", &cy, MINIARG_DOUBLE },
|
||||||
|
{ "CZ", &cz, MINIARG_DOUBLE },
|
||||||
|
{ 0, 0, MINIARG_NULL }
|
||||||
|
};
|
||||||
|
|
||||||
|
if (!parseMiniArgs(argc, argv, args))
|
||||||
|
return 1;
|
||||||
|
|
||||||
|
H5::H5File in_f(fname1, 0);
|
||||||
|
H5::H5File out_f("fields.h5", H5F_ACC_TRUNC);
|
||||||
|
array_type v1_data;
|
||||||
|
uint32_t N1_points, N2_points;
|
||||||
|
|
||||||
|
array3_type bins(boost::extents[Nres][Nres][Nres]);
|
||||||
|
|
||||||
|
rLimit2 = rLimit*rLimit;
|
||||||
|
|
||||||
|
hdf5_read_array(in_f, "particles", v1_data);
|
||||||
|
assert(v1_data.shape()[1] == 6);
|
||||||
|
|
||||||
|
N1_points = v1_data.shape()[0];
|
||||||
|
|
||||||
|
cout << "Got " << N1_points << " in the first file." << endl;
|
||||||
|
|
||||||
|
MyCell *allCells_1 = new MyCell[N1_points];
|
||||||
|
|
||||||
|
for (long i = 0; i < Nres*Nres*Nres; i++)
|
||||||
|
bins.data()[i] = 0;
|
||||||
|
|
||||||
|
cout << "Shuffling data in cells..." << endl;
|
||||||
|
for (int i = 0 ; i < N1_points; i++)
|
||||||
|
{
|
||||||
|
for (int j = 0; j < 3; j++)
|
||||||
|
allCells_1[i].coord[j] = v1_data[i][j];
|
||||||
|
for (int k = 0; k < 3; k++)
|
||||||
|
allCells_1[i].val.pValue.v[k] = v1_data[i][3+k];
|
||||||
|
allCells_1[i].active = true;
|
||||||
|
allCells_1[i].val.weight = 0.0;
|
||||||
|
|
||||||
|
long rx = floor((allCells_1[i].coord[0]+cx)*Nres/boxsize+0.5);
|
||||||
|
long ry = floor((allCells_1[i].coord[1]+cy)*Nres/boxsize+0.5);
|
||||||
|
long rz = floor((allCells_1[i].coord[2]+cz)*Nres/boxsize+0.5);
|
||||||
|
|
||||||
|
if (rx < 0 || rx >= Nres || ry < 0 || ry >= Nres || rz < 0 || rz >= Nres)
|
||||||
|
continue;
|
||||||
|
|
||||||
|
bins[rx][ry][rz]++;
|
||||||
|
}
|
||||||
|
v1_data.resize(boost::extents[1][1]);
|
||||||
|
|
||||||
|
hdf5_write_array(out_f, "num_in_cell", bins);
|
||||||
|
|
||||||
|
cout << "Building trees..." << endl;
|
||||||
|
MyTree tree1(allCells_1, N1_points);
|
||||||
|
|
||||||
|
cout << "Creating smoothing filter..." << endl;
|
||||||
|
|
||||||
|
array3_type out_den_1(boost::extents[Nres][Nres][Nres]);
|
||||||
|
array4_type out_v3d_1(boost::extents[Nres][Nres][Nres][3]);
|
||||||
|
array3_type out_rad_1(boost::extents[Nres][Nres][Nres]);
|
||||||
|
|
||||||
|
cout << "Weighing..." << endl;
|
||||||
|
|
||||||
|
#pragma omp parallel
|
||||||
|
{
|
||||||
|
MySmooth smooth1(&tree1, N_SPH);
|
||||||
|
|
||||||
|
#pragma omp for schedule(dynamic)
|
||||||
|
for (int rz = 0; rz < Nres; rz++)
|
||||||
|
{
|
||||||
|
double pz = (rz)*boxsize/Nres-cz;
|
||||||
|
|
||||||
|
cout << rz << " / " << Nres << endl;
|
||||||
|
for (int ry = 0; ry < Nres; ry++)
|
||||||
|
{
|
||||||
|
double py = (ry)*boxsize/Nres-cy;
|
||||||
|
for (int rx = 0; rx < Nres; rx++)
|
||||||
|
{
|
||||||
|
double px = (rx)*boxsize/Nres-cx;
|
||||||
|
|
||||||
|
MyTree::coords c = { px, py, pz };
|
||||||
|
|
||||||
|
double r2 = c[0]*c[0]+c[1]*c[1]+c[2]*c[2];
|
||||||
|
if (r2 > rLimit2)
|
||||||
|
{
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
uint32_t numInCell = bins[rx][ry][rz];
|
||||||
|
if (numInCell > N_SPH)
|
||||||
|
smooth1.fetchNeighbours(c, numInCell);
|
||||||
|
else
|
||||||
|
smooth1.fetchNeighbours(c);
|
||||||
|
#pragma omp critical
|
||||||
|
smooth1.addGridSite(c);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
cout << "Interpolating..." << endl;
|
||||||
|
#pragma omp parallel
|
||||||
|
{
|
||||||
|
MySmooth smooth1(&tree1, N_SPH);
|
||||||
|
|
||||||
|
#pragma omp for schedule(dynamic)
|
||||||
|
for (int rz = 0; rz < Nres; rz++)
|
||||||
|
{
|
||||||
|
double pz = (rz)*boxsize/Nres-cz;
|
||||||
|
|
||||||
|
cout << rz << " / " << Nres << endl;
|
||||||
|
for (int ry = 0; ry < Nres; ry++)
|
||||||
|
{
|
||||||
|
double py = (ry)*boxsize/Nres-cy;
|
||||||
|
for (int rx = 0; rx < Nres; rx++)
|
||||||
|
{
|
||||||
|
double px = (rx)*boxsize/Nres-cx;
|
||||||
|
|
||||||
|
MyTree::coords c = { px, py, pz };
|
||||||
|
|
||||||
|
double r2 = c[0]*c[0]+c[1]*c[1]+c[2]*c[2];
|
||||||
|
if (r2 > rLimit2)
|
||||||
|
{
|
||||||
|
out_v3d_1[rx][ry][rz][0] = 0;
|
||||||
|
out_v3d_1[rx][ry][rz][1] = 0;
|
||||||
|
out_v3d_1[rx][ry][rz][2] = 0;
|
||||||
|
out_den_1[rx][ry][rz] = 0;
|
||||||
|
out_rad_1[rx][ry][rz] = -1;
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
uint32_t numInCell = bins[rx][ry][rz];
|
||||||
|
if (numInCell > N_SPH)
|
||||||
|
smooth1.fetchNeighbours(c, numInCell);
|
||||||
|
else
|
||||||
|
smooth1.fetchNeighbours(c);
|
||||||
|
|
||||||
|
float val;
|
||||||
|
|
||||||
|
out_rad_1[rx][ry][rz] = smooth1.getSmoothingLen();
|
||||||
|
out_v3d_1[rx][ry][rz][0] = smooth1.computeSmoothedValue(c, getVelocity<0>);
|
||||||
|
out_v3d_1[rx][ry][rz][1] = smooth1.computeSmoothedValue(c, getVelocity<1>);
|
||||||
|
out_v3d_1[rx][ry][rz][2] = smooth1.computeSmoothedValue(c, getVelocity<2>);
|
||||||
|
out_den_1[rx][ry][rz] = smooth1.computeSmoothedValue(c, getUnity);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
hdf5_write_array(out_f, "radii", out_rad_1);
|
||||||
|
hdf5_write_array(out_f, "velocity", out_v3d_1);
|
||||||
|
hdf5_write_array(out_f, "density", out_den_1);
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
};
|
65
sample/testHDF5.cpp
Normal file
65
sample/testHDF5.cpp
Normal file
@ -0,0 +1,65 @@
|
|||||||
|
#include <iostream>
|
||||||
|
#include <H5Cpp.h>
|
||||||
|
#include "hdf5_array.hpp"
|
||||||
|
|
||||||
|
using namespace std;
|
||||||
|
|
||||||
|
int main()
|
||||||
|
{
|
||||||
|
typedef boost::multi_array<float, 2> array_type;
|
||||||
|
typedef boost::multi_array<float, 3> array3_type;
|
||||||
|
typedef boost::multi_array<std::complex<double>, 2> arrayc_type;
|
||||||
|
typedef array_type::index index;
|
||||||
|
|
||||||
|
H5::H5File f("test.h5", H5F_ACC_TRUNC);
|
||||||
|
|
||||||
|
H5::Group g = f.createGroup("test_group");
|
||||||
|
|
||||||
|
array_type A(boost::extents[2][3]);
|
||||||
|
array_type B;
|
||||||
|
array3_type C(boost::extents[2][3][4]);
|
||||||
|
arrayc_type D, E;
|
||||||
|
|
||||||
|
int values = 0;
|
||||||
|
for (index i = 0; i != 2; i++)
|
||||||
|
for (index j = 0; j != 3; j++)
|
||||||
|
A[i][j] = values++;
|
||||||
|
|
||||||
|
CosmoTool::hdf5_write_array(g, "test_data", A);
|
||||||
|
|
||||||
|
CosmoTool::hdf5_read_array(g, "test_data", B);
|
||||||
|
|
||||||
|
int verify = 0;
|
||||||
|
for (index i = 0; i != 2; i++)
|
||||||
|
for (index j = 0; j != 3; j++)
|
||||||
|
if (B[i][j] != verify++) {
|
||||||
|
std::cout << "Invalid array content" << endl;
|
||||||
|
abort();
|
||||||
|
}
|
||||||
|
|
||||||
|
try
|
||||||
|
{
|
||||||
|
CosmoTool::hdf5_read_array(g, "test_data", C);
|
||||||
|
std::cout << "Did not throw InvalidDimensions" << endl;
|
||||||
|
abort();
|
||||||
|
}
|
||||||
|
catch (const CosmoTool::InvalidDimensions&)
|
||||||
|
{}
|
||||||
|
|
||||||
|
D.resize(boost::extents[2][3]);
|
||||||
|
D = A;
|
||||||
|
|
||||||
|
CosmoTool::hdf5_write_array(g, "test_data_c", D);
|
||||||
|
|
||||||
|
CosmoTool::hdf5_read_array(g, "test_data_c", E);
|
||||||
|
|
||||||
|
verify = 0;
|
||||||
|
for (index i = 0; i != 2; i++)
|
||||||
|
for (index j = 0; j != 3; j++)
|
||||||
|
if (E[i][j].real() != verify++) {
|
||||||
|
std::cout << "Invalid array content" << endl;
|
||||||
|
abort();
|
||||||
|
}
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
@ -85,14 +85,16 @@ int main()
|
|||||||
uint32_t dims[] = { NX, NX };
|
uint32_t dims[] = { NX, NX };
|
||||||
ProgressiveOutput<ComputePrecision> out =
|
ProgressiveOutput<ComputePrecision> out =
|
||||||
ProgressiveOutput<ComputePrecision>::saveArrayProgressive("out.nc", dims, 2);
|
ProgressiveOutput<ComputePrecision>::saveArrayProgressive("out.nc", dims, 2);
|
||||||
|
//#pragma omp parallel for schedule(static)
|
||||||
for (uint32_t iy = 0; iy < NX; iy++)
|
for (uint32_t iy = 0; iy < NX; iy++)
|
||||||
{
|
{
|
||||||
|
MySmooth::SPHState state;
|
||||||
cout << "iy=" << iy << endl;
|
cout << "iy=" << iy << endl;
|
||||||
for (uint32_t ix = 0; ix < NX; ix++)
|
for (uint32_t ix = 0; ix < NX; ix++)
|
||||||
{
|
{
|
||||||
MyTree::coords c = { 1.0*ix/NX, 1.0*iy/NX };
|
MyTree::coords c = { 1.0*ix/NX, 1.0*iy/NX };
|
||||||
smooth.fetchNeighbours(c);
|
smooth.fetchNeighbours(c, &state);
|
||||||
out.put(smooth.computeSmoothedValue(c, unit_fun));
|
out.put(smooth.computeSmoothedValue(c, unit_fun, &state));
|
||||||
|
|
||||||
|
|
||||||
}
|
}
|
||||||
|
@ -89,5 +89,6 @@ int main(int argc, char **argv)
|
|||||||
test_2d<float>(128,128);
|
test_2d<float>(128,128);
|
||||||
test_2d<float>(128,131);
|
test_2d<float>(128,131);
|
||||||
test_2d<float>(128,130);
|
test_2d<float>(128,130);
|
||||||
|
test_2d<float>(131,128);
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
@ -9,6 +9,7 @@ SET(CosmoTool_SRCS
|
|||||||
miniargs.cpp
|
miniargs.cpp
|
||||||
growthFactor.cpp
|
growthFactor.cpp
|
||||||
cosmopower.cpp
|
cosmopower.cpp
|
||||||
|
cic.cpp
|
||||||
)
|
)
|
||||||
|
|
||||||
IF(FOUND_NETCDF3)
|
IF(FOUND_NETCDF3)
|
||||||
|
@ -212,7 +212,7 @@ double CosmoPower::powerBDM(double k)
|
|||||||
|
|
||||||
double CosmoPower::powerTest(double k)
|
double CosmoPower::powerTest(double k)
|
||||||
{
|
{
|
||||||
return 1/(1+k*k);
|
return normPower;//1/(1+k*k);
|
||||||
}
|
}
|
||||||
|
|
||||||
/*
|
/*
|
||||||
|
@ -55,6 +55,7 @@ namespace CosmoTool
|
|||||||
protected:
|
protected:
|
||||||
SpectrumFunction() {}
|
SpectrumFunction() {}
|
||||||
public:
|
public:
|
||||||
|
typedef T type;
|
||||||
typedef Eigen::Array<T, 1, Eigen::Dynamic> VecType;
|
typedef Eigen::Array<T, 1, Eigen::Dynamic> VecType;
|
||||||
typedef Eigen::Map<VecType, Eigen::Aligned> MapType;
|
typedef Eigen::Map<VecType, Eigen::Aligned> MapType;
|
||||||
typedef Eigen::Map<VecType const, Eigen::Aligned> ConstMapType;
|
typedef Eigen::Map<VecType const, Eigen::Aligned> ConstMapType;
|
||||||
@ -88,6 +89,7 @@ namespace CosmoTool
|
|||||||
FourierMap() {}
|
FourierMap() {}
|
||||||
|
|
||||||
public:
|
public:
|
||||||
|
typedef T type;
|
||||||
typedef Eigen::Array<T, 1, Eigen::Dynamic> VecType;
|
typedef Eigen::Array<T, 1, Eigen::Dynamic> VecType;
|
||||||
typedef Eigen::Map<VecType, Eigen::Aligned> MapType;
|
typedef Eigen::Map<VecType, Eigen::Aligned> MapType;
|
||||||
typedef Eigen::Map<VecType const, Eigen::Aligned> ConstMapType;
|
typedef Eigen::Map<VecType const, Eigen::Aligned> ConstMapType;
|
||||||
@ -190,6 +192,7 @@ namespace CosmoTool
|
|||||||
class MapUtilityFunction
|
class MapUtilityFunction
|
||||||
{
|
{
|
||||||
public:
|
public:
|
||||||
|
typedef T type;
|
||||||
typedef SpectrumFunction<T> Spectrum;
|
typedef SpectrumFunction<T> Spectrum;
|
||||||
typedef boost::shared_ptr<Spectrum> Spectrum_ptr;
|
typedef boost::shared_ptr<Spectrum> Spectrum_ptr;
|
||||||
typedef FourierMap<std::complex<T> > FMap;
|
typedef FourierMap<std::complex<T> > FMap;
|
||||||
|
@ -36,6 +36,7 @@ knowledge of the CeCILL license and that you accept its terms.
|
|||||||
#ifndef __DETAILS_EUCLIDIAN_MAPS
|
#ifndef __DETAILS_EUCLIDIAN_MAPS
|
||||||
#define __DETAILS_EUCLIDIAN_MAPS
|
#define __DETAILS_EUCLIDIAN_MAPS
|
||||||
|
|
||||||
|
#include <cmath>
|
||||||
|
|
||||||
namespace CosmoTool
|
namespace CosmoTool
|
||||||
{
|
{
|
||||||
@ -172,7 +173,7 @@ namespace CosmoTool
|
|||||||
}
|
}
|
||||||
|
|
||||||
template<typename Array2>
|
template<typename Array2>
|
||||||
void get_Kvec(long p, Array2& kvec) const
|
void get_Kvec_p(long p, Array2& kvec) const
|
||||||
{
|
{
|
||||||
const DimArray& dims = this->getDims();
|
const DimArray& dims = this->getDims();
|
||||||
DimArray d(delta_k.size());
|
DimArray d(delta_k.size());
|
||||||
@ -212,7 +213,7 @@ namespace CosmoTool
|
|||||||
return std::sqrt(k2);
|
return std::sqrt(k2);
|
||||||
}
|
}
|
||||||
|
|
||||||
double get_K(long p) const
|
double get_K_p(long p) const
|
||||||
{
|
{
|
||||||
const DimArray& dims = this->getDims();
|
const DimArray& dims = this->getDims();
|
||||||
DimArray d(delta_k.size());
|
DimArray d(delta_k.size());
|
||||||
@ -246,7 +247,7 @@ namespace CosmoTool
|
|||||||
{
|
{
|
||||||
long idx = q0+dims[0]*p;
|
long idx = q0+dims[0]*p;
|
||||||
assert(idx < this->size());
|
assert(idx < this->size());
|
||||||
result += 2*(conj(d1[idx]) * d2[idx]).real();
|
result += T(2)*(std::conj(d1[idx]) * d2[idx]).real();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
if (even0)
|
if (even0)
|
||||||
@ -254,8 +255,8 @@ namespace CosmoTool
|
|||||||
for (long p = 0; p < plane_size; p++)
|
for (long p = 0; p < plane_size; p++)
|
||||||
{
|
{
|
||||||
long q0 = N0*p, q1 = (p+1)*N0-1;
|
long q0 = N0*p, q1 = (p+1)*N0-1;
|
||||||
result += conj(d1[q0]) * d2[q0];
|
result += T(2)*std::conj(d1[q0]) * d2[q0];
|
||||||
result += conj(d1[q1]) * d2[q1];
|
result += T(2)*std::conj(d1[q1]) * d2[q1];
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return result;
|
return result;
|
||||||
|
@ -115,7 +115,7 @@ namespace CosmoTool
|
|||||||
|
|
||||||
for (long p = 1; p < rand_map.size(); p++)
|
for (long p = 1; p < rand_map.size(); p++)
|
||||||
{
|
{
|
||||||
double A_k = std::sqrt(0.5*V*f(rand_map.get_K(p)));
|
double A_k = std::sqrt(0.5*V*f(rand_map.get_K_p(p)));
|
||||||
d[p] = std::complex<T>(gsl_ran_gaussian(rng, A_k),
|
d[p] = std::complex<T>(gsl_ran_gaussian(rng, A_k),
|
||||||
gsl_ran_gaussian(rng, A_k));
|
gsl_ran_gaussian(rng, A_k));
|
||||||
}
|
}
|
||||||
@ -138,7 +138,7 @@ namespace CosmoTool
|
|||||||
plane_size *= dims[q];
|
plane_size *= dims[q];
|
||||||
}
|
}
|
||||||
|
|
||||||
for (long p = 1; p < plane_size/2; p++)
|
for (long p = 1; p < plane_size/2+1; p++)
|
||||||
{
|
{
|
||||||
long q = (p+1)*dims[0]-1;
|
long q = (p+1)*dims[0]-1;
|
||||||
long q2 = (plane_size-p+1)*dims[0]-1;
|
long q2 = (plane_size-p+1)*dims[0]-1;
|
||||||
@ -147,6 +147,13 @@ namespace CosmoTool
|
|||||||
d[q] = conj(d[q2]);
|
d[q] = conj(d[q2]);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
for (long p = 1; p < plane_size/2+1; p++)
|
||||||
|
{
|
||||||
|
long q = (p)*dims[0];
|
||||||
|
long q2 = (plane_size-p)*dims[0];
|
||||||
|
d[q] = conj(d[q2]);
|
||||||
|
}
|
||||||
|
|
||||||
if (alleven)
|
if (alleven)
|
||||||
{
|
{
|
||||||
long q = 0;
|
long q = 0;
|
||||||
@ -164,7 +171,7 @@ namespace CosmoTool
|
|||||||
std::complex<T> *d = m.data();
|
std::complex<T> *d = m.data();
|
||||||
|
|
||||||
for (long p = 0; p < m_c.size(); p++)
|
for (long p = 0; p < m_c.size(); p++)
|
||||||
d[p] *= f(m_c.get_K(p));
|
d[p] *= f(m_c.get_K_p(p));
|
||||||
}
|
}
|
||||||
|
|
||||||
template<typename T>
|
template<typename T>
|
||||||
@ -174,7 +181,7 @@ namespace CosmoTool
|
|||||||
std::complex<T> *d = m.data();
|
std::complex<T> *d = m.data();
|
||||||
|
|
||||||
for (long p = 0; p < m_c.size(); p++)
|
for (long p = 0; p < m_c.size(); p++)
|
||||||
d[p] *= std::sqrt(f(m_c.get_K(p)));
|
d[p] *= std::sqrt(f(m_c.get_K_p(p)));
|
||||||
}
|
}
|
||||||
|
|
||||||
template<typename T>
|
template<typename T>
|
||||||
@ -185,7 +192,7 @@ namespace CosmoTool
|
|||||||
|
|
||||||
for (long p = 0; p < m_c.size(); p++)
|
for (long p = 0; p < m_c.size(); p++)
|
||||||
{
|
{
|
||||||
T A = f(m_c.get_K(p));
|
T A = f(m_c.get_K_p(p));
|
||||||
if (A==0)
|
if (A==0)
|
||||||
d[p] = 0;
|
d[p] = 0;
|
||||||
else
|
else
|
||||||
@ -201,7 +208,7 @@ namespace CosmoTool
|
|||||||
|
|
||||||
for (long p = 0; p < m_c.size(); p++)
|
for (long p = 0; p < m_c.size(); p++)
|
||||||
{
|
{
|
||||||
T A = std::sqrt(f(m_c.get_K(p)));
|
T A = std::sqrt(f(m_c.get_K_p(p)));
|
||||||
if (A == 0)
|
if (A == 0)
|
||||||
d[p] = 0;
|
d[p] = 0;
|
||||||
else
|
else
|
||||||
|
@ -122,12 +122,22 @@ namespace CosmoTool
|
|||||||
calls::execute(m_synthesis);
|
calls::execute(m_synthesis);
|
||||||
realMap->scale(1/volume);
|
realMap->scale(1/volume);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
void synthesis_unnormed()
|
||||||
|
{
|
||||||
|
calls::execute(m_synthesis);
|
||||||
|
}
|
||||||
|
|
||||||
void analysis()
|
void analysis()
|
||||||
{
|
{
|
||||||
calls::execute(m_analysis);
|
calls::execute(m_analysis);
|
||||||
fourierMap->scale(volume/N);
|
fourierMap->scale(volume/N);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
void analysis_unnormed()
|
||||||
|
{
|
||||||
|
calls::execute(m_analysis);
|
||||||
|
}
|
||||||
|
|
||||||
void synthesis_conjugate()
|
void synthesis_conjugate()
|
||||||
{
|
{
|
||||||
|
@ -67,7 +67,7 @@ namespace CosmoTool
|
|||||||
sharp_make_triangular_alm_info (Lmax, Mmax, 1, &ainfo);
|
sharp_make_triangular_alm_info (Lmax, Mmax, 1, &ainfo);
|
||||||
}
|
}
|
||||||
|
|
||||||
HealpixFourierTransform(long nSide, long Lmax, long Mmax, const std::valarray<double>& weights, int iterate = 0 )
|
HealpixFourierTransform(long nSide, long Lmax, long Mmax, int iterate, const std::valarray<double>& weights )
|
||||||
: realMap(nSide), fourierMap(Lmax, Mmax), ainfo(0), ginfo(0), m_iterate(iterate)
|
: realMap(nSide), fourierMap(Lmax, Mmax), ainfo(0), ginfo(0), m_iterate(iterate)
|
||||||
{
|
{
|
||||||
sharp_make_weighted_healpix_geom_info (nSide, 1, &weights[0], &ginfo);
|
sharp_make_weighted_healpix_geom_info (nSide, 1, &weights[0], &ginfo);
|
||||||
|
175
src/hdf5_array.hpp
Normal file
175
src/hdf5_array.hpp
Normal file
@ -0,0 +1,175 @@
|
|||||||
|
#ifndef __COSMO_HDF5_ARRAY_HPP
|
||||||
|
#define __COSMO_HDF5_ARRAY_HPP
|
||||||
|
|
||||||
|
|
||||||
|
#include <string>
|
||||||
|
#include <stdint.h>
|
||||||
|
#include <boost/static_assert.hpp>
|
||||||
|
#include <boost/utility.hpp>
|
||||||
|
#include <boost/mpl/assert.hpp>
|
||||||
|
#include <boost/multi_array.hpp>
|
||||||
|
#include <boost/type_traits/is_same.hpp>
|
||||||
|
#include <boost/type_traits/is_floating_point.hpp>
|
||||||
|
#include <vector>
|
||||||
|
#include <H5Cpp.h>
|
||||||
|
|
||||||
|
namespace CosmoTool {
|
||||||
|
|
||||||
|
//!_______________________________________________________________________________________
|
||||||
|
//!
|
||||||
|
//! map types to HDF5 types
|
||||||
|
//!
|
||||||
|
//!
|
||||||
|
//! Leo Goodstadt (04 March 2013), improved with enable_if by Guilhem Lavaux (May 2014)
|
||||||
|
//!_______________________________________________________________________________________
|
||||||
|
|
||||||
|
template<typename T, class Enable = void> struct get_hdf5_data_type
|
||||||
|
{
|
||||||
|
static H5::PredType type()
|
||||||
|
{
|
||||||
|
BOOST_MPL_ASSERT_MSG(0, Unknown_HDF5_data_type, ());
|
||||||
|
return H5::PredType::NATIVE_DOUBLE;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
#define HDF5_TYPE(tl, thdf5) \
|
||||||
|
template<typename T> struct get_hdf5_data_type<T, typename boost::enable_if<boost::is_same<T, tl> >::type> \
|
||||||
|
{ static H5::PredType type() { return H5::PredType::thdf5; }; }
|
||||||
|
|
||||||
|
HDF5_TYPE(char , NATIVE_CHAR);
|
||||||
|
HDF5_TYPE(long long , NATIVE_LLONG);
|
||||||
|
HDF5_TYPE(unsigned long long, NATIVE_ULLONG);
|
||||||
|
HDF5_TYPE(int8_t , NATIVE_INT8);
|
||||||
|
HDF5_TYPE(uint8_t , NATIVE_UINT8);
|
||||||
|
HDF5_TYPE(int16_t , NATIVE_INT16);
|
||||||
|
HDF5_TYPE(uint16_t , NATIVE_UINT16);
|
||||||
|
HDF5_TYPE(int32_t , NATIVE_INT32);
|
||||||
|
HDF5_TYPE(uint32_t , NATIVE_UINT32);
|
||||||
|
HDF5_TYPE(int64_t , NATIVE_INT64);
|
||||||
|
HDF5_TYPE(uint64_t , NATIVE_UINT64);
|
||||||
|
HDF5_TYPE(float , NATIVE_FLOAT);
|
||||||
|
HDF5_TYPE(double , NATIVE_DOUBLE);
|
||||||
|
HDF5_TYPE(long double , NATIVE_LDOUBLE);
|
||||||
|
|
||||||
|
#undef HDF5_TYPE
|
||||||
|
|
||||||
|
//!_______________________________________________________________________________________
|
||||||
|
//!
|
||||||
|
//! write_hdf5 multi_array
|
||||||
|
//!
|
||||||
|
//! \author leo Goodstadt (04 March 2013)
|
||||||
|
//!
|
||||||
|
//!_______________________________________________________________________________________
|
||||||
|
template<typename T, std::size_t DIMENSIONS, typename hdf5_data_type>
|
||||||
|
void hdf5_write_array(H5::CommonFG& fg, const std::string& data_set_name,
|
||||||
|
const boost::multi_array<T, DIMENSIONS>& data,
|
||||||
|
const hdf5_data_type& datatype)
|
||||||
|
{
|
||||||
|
std::vector<hsize_t> dimensions(data.shape(), data.shape() + DIMENSIONS);
|
||||||
|
H5::DataSpace dataspace(DIMENSIONS, dimensions.data());
|
||||||
|
|
||||||
|
H5::DataSet dataset = fg.createDataSet(data_set_name, datatype, dataspace);
|
||||||
|
|
||||||
|
dataset.write(data.data(), datatype);
|
||||||
|
}
|
||||||
|
|
||||||
|
/* HDF5 complex type */
|
||||||
|
template<typename T>
|
||||||
|
class hdf5_ComplexType
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
H5::CompType type;
|
||||||
|
|
||||||
|
hdf5_ComplexType()
|
||||||
|
: type(sizeof(std::complex<T>))
|
||||||
|
{
|
||||||
|
get_hdf5_data_type<T> hdf_data_type;
|
||||||
|
type.insertMember("r", 0, hdf_data_type.type());
|
||||||
|
type.insertMember("i", sizeof(T), hdf_data_type.type());
|
||||||
|
type.pack();
|
||||||
|
}
|
||||||
|
|
||||||
|
static const hdf5_ComplexType<T> *ctype()
|
||||||
|
{
|
||||||
|
static hdf5_ComplexType<T> singleton;
|
||||||
|
|
||||||
|
return &singleton;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
template<typename T, std::size_t DIMENSIONS>
|
||||||
|
void hdf5_write_array(H5::CommonFG& fg, const std::string& data_set_name, const boost::multi_array<T, DIMENSIONS>& data )
|
||||||
|
{
|
||||||
|
get_hdf5_data_type<T> hdf_data_type;
|
||||||
|
hdf5_write_array(fg, data_set_name, data, hdf_data_type.type());
|
||||||
|
}
|
||||||
|
|
||||||
|
template<typename T, std::size_t DIMENSIONS>
|
||||||
|
void hdf5_write_array(H5::CommonFG& fg, const std::string& data_set_name, const boost::multi_array<std::complex<T>, DIMENSIONS>& data )
|
||||||
|
{
|
||||||
|
hdf5_write_array(fg, data_set_name, data, hdf5_ComplexType<T>::ctype()->type);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
// HDF5 array reader
|
||||||
|
//
|
||||||
|
// Author Guilhem Lavaux (May 2014)
|
||||||
|
|
||||||
|
class InvalidDimensions: virtual std::exception {
|
||||||
|
};
|
||||||
|
|
||||||
|
template<std::size_t r>
|
||||||
|
struct hdf5_extent_gen {
|
||||||
|
typedef typename boost::detail::multi_array::extent_gen<r> type;
|
||||||
|
|
||||||
|
static inline type build(hsize_t *d)
|
||||||
|
{
|
||||||
|
return (hdf5_extent_gen<r-1>::build(d))[d[r-1]];
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
template<>
|
||||||
|
struct hdf5_extent_gen<0> {
|
||||||
|
static inline boost::multi_array_types::extent_gen build(hsize_t *d)
|
||||||
|
{
|
||||||
|
return boost::extents;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
template<typename T, std::size_t DIMENSIONS, typename hdf5_data_type>
|
||||||
|
void hdf5_read_array(H5::CommonFG& fg, const std::string& data_set_name,
|
||||||
|
boost::multi_array<T, DIMENSIONS>& data,
|
||||||
|
const hdf5_data_type& datatype)
|
||||||
|
{
|
||||||
|
H5::DataSet dataset = fg.openDataSet(data_set_name);
|
||||||
|
H5::DataSpace dataspace = dataset.getSpace();
|
||||||
|
std::vector<hsize_t> dimensions(DIMENSIONS);
|
||||||
|
|
||||||
|
if (dataspace.getSimpleExtentNdims() != DIMENSIONS)
|
||||||
|
{
|
||||||
|
throw InvalidDimensions();
|
||||||
|
}
|
||||||
|
|
||||||
|
dataspace.getSimpleExtentDims(dimensions.data());
|
||||||
|
data.resize(hdf5_extent_gen<DIMENSIONS>::build(dimensions.data()));
|
||||||
|
dataset.read(data.data(), datatype);
|
||||||
|
}
|
||||||
|
|
||||||
|
template<typename T, std::size_t DIMENSIONS>
|
||||||
|
void hdf5_read_array(H5::CommonFG& fg, const std::string& data_set_name, boost::multi_array<T, DIMENSIONS>& data )
|
||||||
|
{
|
||||||
|
get_hdf5_data_type<T> hdf_data_type;
|
||||||
|
hdf5_read_array(fg, data_set_name, data, hdf_data_type.type());
|
||||||
|
}
|
||||||
|
|
||||||
|
template<typename T, std::size_t DIMENSIONS>
|
||||||
|
void hdf5_read_array(H5::CommonFG& fg, const std::string& data_set_name, boost::multi_array<std::complex<T>, DIMENSIONS>& data )
|
||||||
|
{
|
||||||
|
hdf5_read_array(fg, data_set_name, data, hdf5_ComplexType<T>::ctype()->type);
|
||||||
|
}
|
||||||
|
|
||||||
|
};
|
||||||
|
|
||||||
|
#endif
|
||||||
|
|
||||||
|
|
@ -309,7 +309,7 @@ CosmoTool::SimuData *CosmoTool::loadRamsesSimu(const char *basename, int outputI
|
|||||||
|
|
||||||
double hubble = info.aexp*info.aexp/info.unit_t / (1e5/CM_IN_MPC);
|
double hubble = info.aexp*info.aexp/info.unit_t / (1e5/CM_IN_MPC);
|
||||||
double L0 = info.boxSize*info.unitLength*hubble/100/CM_IN_MPC/info.aexp;
|
double L0 = info.boxSize*info.unitLength*hubble/100/CM_IN_MPC/info.aexp;
|
||||||
double unit_vel = L0*hubble/info.aexp;
|
double unit_vel = L0*100/info.aexp;
|
||||||
|
|
||||||
while (1)
|
while (1)
|
||||||
{
|
{
|
||||||
|
@ -35,7 +35,7 @@ knowledge of the CeCILL license and that you accept its terms.
|
|||||||
|
|
||||||
#ifndef __COSMOTOOL_SPH_SMOOTH_HPP
|
#ifndef __COSMOTOOL_SPH_SMOOTH_HPP
|
||||||
#define __COSMOTOOL_SPH_SMOOTH_HPP
|
#define __COSMOTOOL_SPH_SMOOTH_HPP
|
||||||
|
#include <boost/shared_ptr.hpp>
|
||||||
#include "config.hpp"
|
#include "config.hpp"
|
||||||
#include "mykdtree.hpp"
|
#include "mykdtree.hpp"
|
||||||
|
|
||||||
@ -60,37 +60,48 @@ namespace CosmoTool
|
|||||||
typedef void (*runParticleValue)(ValType& t);
|
typedef void (*runParticleValue)(ValType& t);
|
||||||
|
|
||||||
public:
|
public:
|
||||||
|
struct SPHState
|
||||||
|
{
|
||||||
|
boost::shared_ptr<SPHCell *[]> ngb;
|
||||||
|
boost::shared_ptr<CoordType[]> distances;
|
||||||
|
typename SPHTree::coords currentCenter;
|
||||||
|
int currentNgb;
|
||||||
|
ComputePrecision smoothRadius;
|
||||||
|
};
|
||||||
|
|
||||||
|
|
||||||
SPHSmooth(SPHTree *tree, uint32_t Nsph);
|
SPHSmooth(SPHTree *tree, uint32_t Nsph);
|
||||||
virtual ~SPHSmooth();
|
virtual ~SPHSmooth();
|
||||||
|
|
||||||
void fetchNeighbours(const typename SPHTree::coords& c);
|
void fetchNeighbours(const typename SPHTree::coords& c, SPHState *state = 0);
|
||||||
|
|
||||||
void fetchNeighbours(const typename SPHTree::coords& c, uint32_t newNsph);
|
void fetchNeighbours(const typename SPHTree::coords& c, uint32_t newNsph);
|
||||||
void fetchNeighboursOnVolume(const typename SPHTree::coords& c, ComputePrecision radius);
|
void fetchNeighboursOnVolume(const typename SPHTree::coords& c, ComputePrecision radius);
|
||||||
const typename SPHTree::coords& getCurrentCenter() const
|
const typename SPHTree::coords& getCurrentCenter() const
|
||||||
{
|
{
|
||||||
return currentCenter;
|
return internal.currentCenter;
|
||||||
}
|
}
|
||||||
|
|
||||||
ComputePrecision computeSmoothedValue(const typename SPHTree::coords& c,
|
ComputePrecision computeSmoothedValue(const typename SPHTree::coords& c,
|
||||||
computeParticleValue fun);
|
computeParticleValue fun, SPHState *state = 0);
|
||||||
ComputePrecision computeInterpolatedValue(const typename SPHTree::coords& c,
|
ComputePrecision computeInterpolatedValue(const typename SPHTree::coords& c,
|
||||||
computeParticleValue fun);
|
computeParticleValue fun, SPHState *state = 0);
|
||||||
ComputePrecision getMaxDistance(const typename SPHTree::coords& c,
|
ComputePrecision getMaxDistance(const typename SPHTree::coords& c,
|
||||||
SPHNode *node) const;
|
SPHNode *node) const;
|
||||||
|
|
||||||
ComputePrecision getSmoothingLen() const
|
ComputePrecision getSmoothingLen() const
|
||||||
{
|
{
|
||||||
return smoothRadius;
|
return internal.smoothRadius;
|
||||||
}
|
}
|
||||||
|
|
||||||
// TO USE WITH EXTREME CARE !
|
// TO USE WITH EXTREME CARE !
|
||||||
void setSmoothingLen(ComputePrecision len)
|
void setSmoothingLen(ComputePrecision len)
|
||||||
{
|
{
|
||||||
smoothRadius = len;
|
internal.smoothRadius = len;
|
||||||
}
|
}
|
||||||
// END
|
// END
|
||||||
|
|
||||||
void runForEachNeighbour(runParticleValue fun);
|
void runForEachNeighbour(runParticleValue fun, SPHState *state = 0);
|
||||||
void addGridSite(const typename SPHTree::coords& c);
|
void addGridSite(const typename SPHTree::coords& c);
|
||||||
|
|
||||||
bool hasNeighbours() const;
|
bool hasNeighbours() const;
|
||||||
@ -100,32 +111,26 @@ namespace CosmoTool
|
|||||||
SPHTree *getTree() { return tree; }
|
SPHTree *getTree() { return tree; }
|
||||||
|
|
||||||
void changeNgb(uint32_t newMax) {
|
void changeNgb(uint32_t newMax) {
|
||||||
delete[] ngb;
|
internal.ngb = boost::shared_ptr<SPHCell *>(new SPHCell *[newMax]);
|
||||||
delete[] distances;
|
internal.distances = boost::shared_ptr<CoordType>(new CoordType[newMax]);
|
||||||
ngb = new SPHCell *[newMax];
|
|
||||||
distances = new CoordType[newMax];
|
|
||||||
maxNgb = newMax;
|
maxNgb = newMax;
|
||||||
}
|
}
|
||||||
|
|
||||||
uint32_t getCurrent() const { return currentNgb; }
|
uint32_t getCurrent() const { return internal.currentNgb; }
|
||||||
|
|
||||||
uint32_t getNgb() const { return maxNgb; }
|
uint32_t getNgb() const { return maxNgb; }
|
||||||
|
|
||||||
protected:
|
protected:
|
||||||
SPHCell **ngb;
|
SPHState internal;
|
||||||
CoordType *distances;
|
|
||||||
uint32_t Nsph;
|
uint32_t Nsph;
|
||||||
uint32_t deltaNsph;
|
uint32_t deltaNsph;
|
||||||
uint32_t maxNgb;
|
uint32_t maxNgb;
|
||||||
uint32_t currentNgb;
|
|
||||||
SPHTree *tree;
|
SPHTree *tree;
|
||||||
ComputePrecision smoothRadius;
|
|
||||||
typename SPHTree::coords currentCenter;
|
|
||||||
|
|
||||||
ComputePrecision computeWValue(const typename SPHTree::coords & c,
|
ComputePrecision computeWValue(const typename SPHTree::coords & c,
|
||||||
SPHCell& cell,
|
SPHCell& cell,
|
||||||
CoordType d,
|
CoordType d,
|
||||||
computeParticleValue fun);
|
computeParticleValue fun, SPHState *state);
|
||||||
void runUnrollNode(SPHNode *node,
|
void runUnrollNode(SPHNode *node,
|
||||||
runParticleValue fun);
|
runParticleValue fun);
|
||||||
};
|
};
|
||||||
|
@ -1,4 +1,5 @@
|
|||||||
#include <cmath>
|
#include <cmath>
|
||||||
|
#include "algo.hpp"
|
||||||
|
|
||||||
namespace CosmoTool {
|
namespace CosmoTool {
|
||||||
|
|
||||||
@ -7,29 +8,27 @@ SPHSmooth<ValType,Ndims>::SPHSmooth(SPHTree *tree, uint32_t Nsph)
|
|||||||
{
|
{
|
||||||
this->Nsph = Nsph;
|
this->Nsph = Nsph;
|
||||||
this->tree = tree;
|
this->tree = tree;
|
||||||
this->currentNgb = 0;
|
internal.currentNgb = 0;
|
||||||
|
|
||||||
this->maxNgb = Nsph;
|
this->maxNgb = Nsph;
|
||||||
ngb = new SPHCell *[maxNgb];
|
internal.ngb = boost::shared_ptr<SPHCell *[]>(new SPHCell *[maxNgb]);
|
||||||
distances = new CoordType[maxNgb];
|
internal.distances = boost::shared_ptr<CoordType[]>(new CoordType[maxNgb]);
|
||||||
}
|
}
|
||||||
|
|
||||||
template<typename ValType, int Ndims>
|
template<typename ValType, int Ndims>
|
||||||
SPHSmooth<ValType,Ndims>::~SPHSmooth()
|
SPHSmooth<ValType,Ndims>::~SPHSmooth()
|
||||||
{
|
{
|
||||||
delete[] ngb;
|
|
||||||
delete[] distances;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
template<typename ValType, int Ndims>
|
template<typename ValType, int Ndims>
|
||||||
ComputePrecision SPHSmooth<ValType,Ndims>::computeWValue(const typename SPHTree::coords& c,
|
ComputePrecision SPHSmooth<ValType,Ndims>::computeWValue(const typename SPHTree::coords& c,
|
||||||
SPHCell& cell,
|
SPHCell& cell,
|
||||||
CoordType d,
|
CoordType d,
|
||||||
computeParticleValue fun)
|
computeParticleValue fun, SPHState *state)
|
||||||
{
|
{
|
||||||
CoordType weight;
|
CoordType weight;
|
||||||
|
|
||||||
d /= smoothRadius;
|
d /= state->smoothRadius;
|
||||||
weight = getKernel(d);
|
weight = getKernel(d);
|
||||||
|
|
||||||
if (cell.val.weight != 0)
|
if (cell.val.weight != 0)
|
||||||
@ -47,86 +46,91 @@ SPHSmooth<ValType,Ndims>::fetchNeighbours(const typename SPHTree::coords& c, uin
|
|||||||
|
|
||||||
if (requested > maxNgb)
|
if (requested > maxNgb)
|
||||||
{
|
{
|
||||||
delete[] ngb;
|
|
||||||
delete[] distances;
|
|
||||||
|
|
||||||
maxNgb = requested;
|
maxNgb = requested;
|
||||||
ngb = new SPHCell *[maxNgb];
|
internal.ngb = boost::shared_ptr<SPHCell*[]>(new SPHCell *[maxNgb]);
|
||||||
distances = new CoordType[maxNgb];
|
internal.distances = boost::shared_ptr<CoordType[]>(new CoordType[maxNgb]);
|
||||||
}
|
}
|
||||||
|
|
||||||
memcpy(currentCenter, c, sizeof(c));
|
memcpy(internal.currentCenter, c, sizeof(c));
|
||||||
tree->getNearestNeighbours(c, requested, ngb, distances);
|
tree->getNearestNeighbours(c, requested, internal.ngb.get(), internal.distances.get());
|
||||||
|
|
||||||
currentNgb = 0;
|
internal.currentNgb = 0;
|
||||||
for (uint32_t i = 0; i < requested && ngb[i] != 0; i++,currentNgb++)
|
for (uint32_t i = 0; i < requested && internal.ngb[i] != 0; i++,internal.currentNgb++)
|
||||||
{
|
{
|
||||||
distances[i] = sqrt(distances[i]);
|
internal.distances[i] = sqrt(internal.distances[i]);
|
||||||
d2 = distances[i];
|
d2 = internal.distances[i];
|
||||||
if (d2 > max_dist)
|
if (d2 > max_dist)
|
||||||
max_dist = d2;
|
max_dist = d2;
|
||||||
}
|
}
|
||||||
|
|
||||||
smoothRadius = max_dist / 2;
|
internal.smoothRadius = max_dist / 2;
|
||||||
}
|
}
|
||||||
|
|
||||||
template<typename ValType, int Ndims>
|
template<typename ValType, int Ndims>
|
||||||
void
|
void SPHSmooth<ValType,Ndims>::fetchNeighbours(const typename SPHTree::coords& c, SPHState *state)
|
||||||
SPHSmooth<ValType,Ndims>::fetchNeighbours(const typename SPHTree::coords& c)
|
|
||||||
{
|
{
|
||||||
ComputePrecision d2, max_dist = 0;
|
ComputePrecision d2, max_dist = 0;
|
||||||
uint32_t requested = Nsph;
|
uint32_t requested = Nsph;
|
||||||
|
|
||||||
memcpy(currentCenter, c, sizeof(c));
|
if (state != 0) {
|
||||||
tree->getNearestNeighbours(c, requested, ngb, distances);
|
state->distances = boost::shared_ptr<CoordType[]>(new CoordType[Nsph]);
|
||||||
|
state->ngb = boost::shared_ptr<SPHCell *[]>(new SPHCell *[Nsph]);
|
||||||
|
} else
|
||||||
|
state = &internal;
|
||||||
|
|
||||||
|
memcpy(state->currentCenter, c, sizeof(c));
|
||||||
|
|
||||||
|
tree->getNearestNeighbours(c, requested, state->ngb.get(), state->distances.get());
|
||||||
|
|
||||||
currentNgb = 0;
|
state->currentNgb = 0;
|
||||||
for (uint32_t i = 0; i < requested && ngb[i] != 0; i++,currentNgb++)
|
for (uint32_t i = 0; i < requested && state->ngb[i] != 0; i++,state->currentNgb++)
|
||||||
{
|
{
|
||||||
distances[i] = sqrt(distances[i]);
|
d2 = internal.distances[i] = sqrt(internal.distances[i]);
|
||||||
d2 = distances[i];
|
|
||||||
if (d2 > max_dist)
|
if (d2 > max_dist)
|
||||||
max_dist = d2;
|
max_dist = d2;
|
||||||
}
|
}
|
||||||
|
|
||||||
smoothRadius = max_dist / 2;
|
state->smoothRadius = max_dist / 2;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
template<typename ValType, int Ndims>
|
template<typename ValType, int Ndims>
|
||||||
void
|
void
|
||||||
SPHSmooth<ValType,Ndims>::fetchNeighboursOnVolume(const typename SPHTree::coords& c,
|
SPHSmooth<ValType,Ndims>::fetchNeighboursOnVolume(const typename SPHTree::coords& c,
|
||||||
ComputePrecision radius)
|
ComputePrecision radius)
|
||||||
{
|
{
|
||||||
uint32_t numPart;
|
uint32_t numPart;
|
||||||
ComputePrecision d2, max_dist = 0;
|
ComputePrecision d2, max_dist = 0;
|
||||||
|
|
||||||
memcpy(currentCenter, c, sizeof(c));
|
memcpy(internal.currentCenter, c, sizeof(c));
|
||||||
|
|
||||||
currentNgb = tree->getIntersection(c, radius, ngb, distances,
|
internal.currentNgb = tree->getIntersection(c, radius, internal.ngb, internal.distances,
|
||||||
maxNgb);
|
maxNgb);
|
||||||
|
|
||||||
for (uint32_t i = 0; i < currentNgb; i++)
|
for (uint32_t i = 0; i < internal.currentNgb; i++)
|
||||||
{
|
{
|
||||||
distances[i] = sqrt(distances[i]);
|
d2 = internal.distances[i] = sqrt(internal.distances[i]);
|
||||||
d2 = distances[i];
|
|
||||||
if (d2 > max_dist)
|
if (d2 > max_dist)
|
||||||
max_dist = d2;
|
max_dist = d2;
|
||||||
}
|
}
|
||||||
smoothRadius = max_dist / 2;
|
internal.smoothRadius = max_dist / 2;
|
||||||
}
|
}
|
||||||
|
|
||||||
template<typename ValType, int Ndims>
|
template<typename ValType, int Ndims>
|
||||||
ComputePrecision
|
ComputePrecision
|
||||||
SPHSmooth<ValType,Ndims>::computeSmoothedValue(const typename SPHTree::coords& c,
|
SPHSmooth<ValType,Ndims>::computeSmoothedValue(const typename SPHTree::coords& c,
|
||||||
computeParticleValue fun)
|
computeParticleValue fun, SPHState *state)
|
||||||
{
|
{
|
||||||
|
if (state == 0)
|
||||||
|
state = &internal;
|
||||||
|
|
||||||
ComputePrecision outputValue = 0;
|
ComputePrecision outputValue = 0;
|
||||||
ComputePrecision max_dist = 0;
|
ComputePrecision max_dist = 0;
|
||||||
ComputePrecision r3 = smoothRadius * smoothRadius * smoothRadius;
|
ComputePrecision r3 = cube(state->smoothRadius);
|
||||||
|
|
||||||
for (uint32_t i = 0; i < currentNgb; i++)
|
for (uint32_t i = 0; i < state->currentNgb; i++)
|
||||||
{
|
{
|
||||||
outputValue += computeWValue(c, *ngb[i], distances[i], fun);
|
outputValue += computeWValue(c, *state->ngb[i], state->distances[i], fun, state);
|
||||||
}
|
}
|
||||||
|
|
||||||
return outputValue / r3;
|
return outputValue / r3;
|
||||||
@ -141,27 +145,33 @@ ComputePrecision interpolateOne(const ValType& t)
|
|||||||
// WARNING ! Cell's weight must be 1 !!!
|
// WARNING ! Cell's weight must be 1 !!!
|
||||||
template<typename ValType, int Ndims>
|
template<typename ValType, int Ndims>
|
||||||
ComputePrecision SPHSmooth<ValType,Ndims>::computeInterpolatedValue(const typename SPHTree::coords& c,
|
ComputePrecision SPHSmooth<ValType,Ndims>::computeInterpolatedValue(const typename SPHTree::coords& c,
|
||||||
computeParticleValue fun)
|
computeParticleValue fun, SPHState *state)
|
||||||
{
|
{
|
||||||
|
if (state == 0)
|
||||||
|
state = &internal;
|
||||||
|
|
||||||
ComputePrecision outputValue = 0;
|
ComputePrecision outputValue = 0;
|
||||||
ComputePrecision max_dist = 0;
|
ComputePrecision max_dist = 0;
|
||||||
ComputePrecision weight = 0;
|
ComputePrecision weight = 0;
|
||||||
|
|
||||||
for (uint32_t i = 0; i < currentNgb; i++)
|
for (uint32_t i = 0; i < state->currentNgb; i++)
|
||||||
{
|
{
|
||||||
outputValue += computeWValue(c, *ngb[i], distances[i], fun);
|
outputValue += computeWValue(c, *state->ngb[i], state->distances[i], fun);
|
||||||
weight += computeWValue(c, *ngb[i], distances[i], interpolateOne);
|
weight += computeWValue(c, *state->ngb[i], state->distances[i], interpolateOne);
|
||||||
}
|
}
|
||||||
|
|
||||||
return (outputValue == 0) ? 0 : (outputValue / weight);
|
return (outputValue == 0) ? 0 : (outputValue / weight);
|
||||||
}
|
}
|
||||||
|
|
||||||
template<typename ValType, int Ndims>
|
template<typename ValType, int Ndims>
|
||||||
void SPHSmooth<ValType,Ndims>::runForEachNeighbour(runParticleValue fun)
|
void SPHSmooth<ValType,Ndims>::runForEachNeighbour(runParticleValue fun, SPHState *state)
|
||||||
{
|
{
|
||||||
for (uint32_t i = 0; i < currentNgb; i++)
|
if (state == 0)
|
||||||
|
state = &internal;
|
||||||
|
|
||||||
|
for (uint32_t i = 0; i < state->currentNgb; i++)
|
||||||
{
|
{
|
||||||
fun(ngb[i]);
|
fun(state->ngb[i]);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -172,13 +182,13 @@ void SPHSmooth<ValType,Ndims>::addGridSite(const typename SPHTree::coords& c)
|
|||||||
ComputePrecision outputValue = 0;
|
ComputePrecision outputValue = 0;
|
||||||
ComputePrecision max_dist = 0;
|
ComputePrecision max_dist = 0;
|
||||||
|
|
||||||
ComputePrecision r3 = smoothRadius * smoothRadius * smoothRadius;
|
ComputePrecision r3 = cube(internal.smoothRadius);
|
||||||
|
|
||||||
for (uint32_t i = 0; i < currentNgb; i++)
|
for (uint32_t i = 0; i < internal.currentNgb; i++)
|
||||||
{
|
{
|
||||||
ComputePrecision d = distances[i];
|
ComputePrecision d = internal.distances[i];
|
||||||
SPHCell& cell = *ngb[i];
|
SPHCell& cell = *internal.ngb[i];
|
||||||
cell.val.weight += getKernel(d/smoothRadius) / r3;
|
cell.val.weight += getKernel(d/internal.smoothRadius) / r3;
|
||||||
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -202,7 +212,7 @@ SPHSmooth<ValType,Ndims>::getKernel(ComputePrecision x) const
|
|||||||
template<typename ValType, int Ndims>
|
template<typename ValType, int Ndims>
|
||||||
bool SPHSmooth<ValType,Ndims>::hasNeighbours() const
|
bool SPHSmooth<ValType,Ndims>::hasNeighbours() const
|
||||||
{
|
{
|
||||||
return (currentNgb != 0);
|
return (internal.currentNgb != 0);
|
||||||
}
|
}
|
||||||
|
|
||||||
template<class ValType1, class ValType2, int Ndims>
|
template<class ValType1, class ValType2, int Ndims>
|
||||||
|
@ -218,7 +218,7 @@ namespace CosmoTool
|
|||||||
|
|
||||||
template<typename T>
|
template<typename T>
|
||||||
void saveArray(const std::string& fname,
|
void saveArray(const std::string& fname,
|
||||||
T *array, uint32_t *dimList, uint32_t rank);
|
const T *array, uint32_t *dimList, uint32_t rank);
|
||||||
|
|
||||||
template<typename T>
|
template<typename T>
|
||||||
void loadArray(const std::string& fname,
|
void loadArray(const std::string& fname,
|
||||||
|
@ -235,7 +235,7 @@ namespace CosmoTool {
|
|||||||
|
|
||||||
template<typename T>
|
template<typename T>
|
||||||
void saveArray(const std::string& fname,
|
void saveArray(const std::string& fname,
|
||||||
T *array, uint32_t *dimList, uint32_t rank)
|
const T *array, uint32_t *dimList, uint32_t rank)
|
||||||
{
|
{
|
||||||
NcFile f(fname.c_str(), NcFile::Replace, 0, 0, NcFile::Netcdf4);
|
NcFile f(fname.c_str(), NcFile::Replace, 0, 0, NcFile::Netcdf4);
|
||||||
|
|
||||||
@ -300,10 +300,10 @@ namespace CosmoTool {
|
|||||||
double*& array, uint32_t *&dimList, uint32_t& rank);
|
double*& array, uint32_t *&dimList, uint32_t& rank);
|
||||||
|
|
||||||
template void saveArray<int>(const std::string& fname,
|
template void saveArray<int>(const std::string& fname,
|
||||||
int *array, uint32_t *dimList, uint32_t rank);
|
const int *array, uint32_t *dimList, uint32_t rank);
|
||||||
template void saveArray<float>(const std::string& fname,
|
template void saveArray<float>(const std::string& fname,
|
||||||
float *array, uint32_t *dimList, uint32_t rank);
|
const float *array, uint32_t *dimList, uint32_t rank);
|
||||||
template void saveArray<double>(const std::string& fname,
|
template void saveArray<double>(const std::string& fname,
|
||||||
double *array, uint32_t *dimList, uint32_t rank);
|
const double *array, uint32_t *dimList, uint32_t rank);
|
||||||
|
|
||||||
}
|
}
|
||||||
|
@ -203,7 +203,7 @@ namespace CosmoTool {
|
|||||||
|
|
||||||
template<typename T>
|
template<typename T>
|
||||||
void saveArray(const std::string& fname,
|
void saveArray(const std::string& fname,
|
||||||
T *array, uint32_t *dimList, uint32_t rank)
|
const T *array, uint32_t *dimList, uint32_t rank)
|
||||||
{
|
{
|
||||||
NcFile f(fname.c_str(), NcFile::replace);
|
NcFile f(fname.c_str(), NcFile::replace);
|
||||||
|
|
||||||
@ -263,10 +263,10 @@ namespace CosmoTool {
|
|||||||
double*& array, uint32_t *&dimList, uint32_t& rank);
|
double*& array, uint32_t *&dimList, uint32_t& rank);
|
||||||
|
|
||||||
template void saveArray<int>(const std::string& fname,
|
template void saveArray<int>(const std::string& fname,
|
||||||
int *array, uint32_t *dimList, uint32_t rank);
|
const int *array, uint32_t *dimList, uint32_t rank);
|
||||||
template void saveArray<float>(const std::string& fname,
|
template void saveArray<float>(const std::string& fname,
|
||||||
float *array, uint32_t *dimList, uint32_t rank);
|
const float *array, uint32_t *dimList, uint32_t rank);
|
||||||
template void saveArray<double>(const std::string& fname,
|
template void saveArray<double>(const std::string& fname,
|
||||||
double *array, uint32_t *dimList, uint32_t rank);
|
const double *array, uint32_t *dimList, uint32_t rank);
|
||||||
|
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user