Compare commits
20 Commits
c04722c2bb
...
main
| Author | SHA1 | Date | |
|---|---|---|---|
| 40846bba78 | |||
| 4d681e3373 | |||
| 3a9efd5598 | |||
| fa1930f9d7 | |||
|
|
b64afe8773 | ||
|
|
f3ebba4931 | ||
|
|
79e1abb2ff | ||
|
|
554eff9b55 | ||
|
|
42db99759f | ||
|
|
69920acd61 | ||
|
|
647d0caa1c | ||
|
|
e69b29a259 | ||
|
|
9a59e031ed | ||
|
|
adedbcc37c | ||
|
|
eb76521060 | ||
|
|
b1fb123026 | ||
|
|
52580d8cde | ||
|
|
07915295cb | ||
|
|
d56758d0b3 | ||
|
|
7ded15d596 |
52
.clangd
Normal file
52
.clangd
Normal file
@@ -0,0 +1,52 @@
|
|||||||
|
CompileFlags:
|
||||||
|
CompilationDatabase: build
|
||||||
|
Add:
|
||||||
|
- -I/home/rigoni/devel/cmt/ulib/src
|
||||||
|
- -isystem/home/share/micromamba/envs/mutom/include
|
||||||
|
- -isystem/home/share/micromamba/envs/mutom/include/eigen3
|
||||||
|
- -isystem/home/share/micromamba/envs/mutom/targets/x86_64-linux/include
|
||||||
|
- -isystem/home/share/micromamba/envs/mutom/lib/gcc/x86_64-conda-linux-gnu/14.3.0/include/c++
|
||||||
|
- -isystem/isystem/home/share/micromamba/envs/mutom/lib/gcc/x86_64-conda-linux-gnu/14.3.0/include/c++/x86_64-conda-linux-gnu
|
||||||
|
- -isystem/home/share/micromamba/envs/mutom/x86_64-conda-linux-gnu/sysroot/usr/include
|
||||||
|
- "--gcc-toolchain=/home/share/micromamba/envs/mutom"
|
||||||
|
- -D_ULIB_DETAIL_SIGNAL_EMIT
|
||||||
|
- -DUSE_CUDA
|
||||||
|
- -std=c++17
|
||||||
|
- "-D__host__="
|
||||||
|
- "-D__device__="
|
||||||
|
- "-D__global__="
|
||||||
|
- "-D__constant__="
|
||||||
|
- "-D__shared__="
|
||||||
|
- "-D__align__(x)="
|
||||||
|
- "-D__forceinline__=inline"
|
||||||
|
- "-D__launch_bounds__(x)="
|
||||||
|
|
||||||
|
Diagnostics:
|
||||||
|
UnusedIncludes: None
|
||||||
|
MissingIncludes: None
|
||||||
|
|
||||||
|
---
|
||||||
|
If:
|
||||||
|
PathExclude: [/home/rigoni/devel/cmt/ulib/src/.*]
|
||||||
|
Diagnostics:
|
||||||
|
Suppress: ["*"]
|
||||||
|
|
||||||
|
---
|
||||||
|
If:
|
||||||
|
PathMatch: [.*\.cu, .*/src/Math/testing/VoxRaytracerTest.cpp, .*/src/Math/VoxRaytracer.cpp, .*/src/Math/VoxImage.cpp]
|
||||||
|
CompileFlags:
|
||||||
|
Add:
|
||||||
|
- "-x"
|
||||||
|
- "cuda"
|
||||||
|
- "--cuda-path=/home/share/micromamba/envs/mutom"
|
||||||
|
- "--cuda-gpu-arch=sm_61"
|
||||||
|
- "--gcc-toolchain=/home/share/micromamba/envs/mutom"
|
||||||
|
- "-L/home/share/micromamba/envs/mutom/lib"
|
||||||
|
- "-lcudart"
|
||||||
|
- "-lcuda"
|
||||||
|
- "-U__host__"
|
||||||
|
- "-U__device__"
|
||||||
|
- "-U__global__"
|
||||||
|
- "-U__constant__"
|
||||||
|
- "-U__shared__"
|
||||||
|
- "-U__forceinline__"
|
||||||
42
.gitea/workflows/publish-docs.yaml
Normal file
42
.gitea/workflows/publish-docs.yaml
Normal file
@@ -0,0 +1,42 @@
|
|||||||
|
name: MkDocs Subpath Deploy
|
||||||
|
|
||||||
|
on:
|
||||||
|
workflow_dispatch: # trigger manuale
|
||||||
|
push:
|
||||||
|
branches:
|
||||||
|
- main # Trigger on main branch
|
||||||
|
|
||||||
|
jobs:
|
||||||
|
build-and-deploy:
|
||||||
|
runs-on: mildpub # Runner that can access to SSH_YFINPUB_HOST
|
||||||
|
|
||||||
|
steps:
|
||||||
|
- name: Checkout del codice
|
||||||
|
uses: actions/checkout@v4
|
||||||
|
|
||||||
|
- name: Configura Python
|
||||||
|
uses: actions/setup-python@v5
|
||||||
|
with:
|
||||||
|
python-version: '3.x'
|
||||||
|
|
||||||
|
- name: Installa dipendenze
|
||||||
|
run: |
|
||||||
|
python -m pip install --upgrade pip
|
||||||
|
pip install mkdocs-material
|
||||||
|
pip install -r docs/docker/requirements.txt
|
||||||
|
|
||||||
|
- name: Build del sito
|
||||||
|
run: mkdocs build
|
||||||
|
|
||||||
|
- name: Deploy via SSH (SCP)
|
||||||
|
uses: https://github.com/appleboy/scp-action@master
|
||||||
|
with:
|
||||||
|
host: ${{ vars.SSH_YFINPUB_HOST }}
|
||||||
|
username: ${{ vars.SSH_YFINPUB_USER }}
|
||||||
|
key: ${{ secrets.MILD_PUB }}
|
||||||
|
port: 22
|
||||||
|
source: "site/*"
|
||||||
|
# Il percorso sul server deve corrispondere alla tua sottopagina
|
||||||
|
target: "/var/www/docs/cmt/uLib/"
|
||||||
|
strip_components: 1 # Rimuove la cartella "site/" e mette solo il contenuto
|
||||||
|
rm: true # Pulisce la cartella prima di copiare (opzionale, stile Vercel)
|
||||||
12
.gitignore
vendored
12
.gitignore
vendored
@@ -1,3 +1,15 @@
|
|||||||
CMakeFiles/
|
CMakeFiles/
|
||||||
build/
|
build/
|
||||||
.cache/
|
.cache/
|
||||||
|
build_warnings*.log
|
||||||
|
final_build.log
|
||||||
|
cmake_configure.log
|
||||||
|
compile_commands.json
|
||||||
|
|
||||||
|
dist/
|
||||||
|
build_python/
|
||||||
|
src/Python/uLib/*.so*
|
||||||
|
src/Python/uLib/*.pyd
|
||||||
|
src/Python/uLib/*.pyc
|
||||||
|
src/Python/uLib/__pycache__
|
||||||
|
src/Python/uLib/.nfs*
|
||||||
|
|||||||
32
.vscode/settings.json
vendored
32
.vscode/settings.json
vendored
@@ -1,8 +1,32 @@
|
|||||||
{
|
{
|
||||||
"clangd.fallbackFlags": [
|
"clangd.fallbackFlags": [
|
||||||
"-I${workspaceFolder}/src",
|
"-I/home/rigoni/devel/cmt/ulib/src",
|
||||||
"-I/home/share/micromamba/envs/mutom/include",
|
"-isystem/home/share/micromamba/envs/mutom/include",
|
||||||
"-I/home/rigoni/.conan2/p/eigen5481853932f72/p/include/eigen3"
|
"-isystem/home/share/micromamba/envs/mutom/include/eigen3",
|
||||||
|
"-isystem/home/share/micromamba/envs/mutom/targets/x86_64-linux/include",
|
||||||
|
"-isystem/home/share/micromamba/envs/mutom/lib/gcc/x86_64-conda-linux-gnu/14.3.0/include/c++",
|
||||||
|
"-isystem/home/share/micromamba/envs/mutom/lib/gcc/x86_64-conda-linux-gnu/14.3.0/include/c++/x86_64-conda-linux-gnu",
|
||||||
|
"-isystem/home/share/micromamba/envs/mutom/x86_64-conda-linux-gnu/sysroot/usr/include",
|
||||||
|
"--gcc-toolchain=/home/share/micromamba/envs/mutom",
|
||||||
|
"-D__host__=",
|
||||||
|
"-D__device__=",
|
||||||
|
"-D__global__=",
|
||||||
|
"-D__constant__=",
|
||||||
|
"-D__shared__=",
|
||||||
|
"-DUSE_CUDA",
|
||||||
|
"-D__CUDACC__"
|
||||||
],
|
],
|
||||||
"clangd.semanticHighlighting.enable": true
|
"clangd.semanticHighlighting.enable": true,
|
||||||
|
"clangd.arguments": [
|
||||||
|
"--compile-commands-dir=build",
|
||||||
|
"--query-driver=/home/share/micromamba/envs/mutom/bin/g++,/home/share/micromamba/envs/mutom/bin/gcc,/home/share/micromamba/envs/mutom/bin/nvcc",
|
||||||
|
"--suppress-system-warnings",
|
||||||
|
"--all-scopes-completion",
|
||||||
|
"--completion-style=detailed",
|
||||||
|
"--header-insertion=never",
|
||||||
|
"-j=4",
|
||||||
|
"--pch-storage=memory",
|
||||||
|
"--background-index",
|
||||||
|
"--log=verbose"
|
||||||
|
]
|
||||||
}
|
}
|
||||||
@@ -82,7 +82,7 @@ ENDMACRO(uLib_add_target)
|
|||||||
# TESTS and LIBRARIES must be defined
|
# TESTS and LIBRARIES must be defined
|
||||||
macro(uLib_add_tests name)
|
macro(uLib_add_tests name)
|
||||||
foreach(tn ${TESTS})
|
foreach(tn ${TESTS})
|
||||||
add_executable(${tn} EXCLUDE_FROM_ALL ${tn}.cpp)
|
add_executable(${tn} ${tn}.cpp)
|
||||||
add_test(NAME ${tn} COMMAND ${tn})
|
add_test(NAME ${tn} COMMAND ${tn})
|
||||||
|
|
||||||
target_link_libraries(${tn} ${LIBRARIES})
|
target_link_libraries(${tn} ${LIBRARIES})
|
||||||
@@ -91,7 +91,9 @@ macro(uLib_add_tests name)
|
|||||||
|
|
||||||
# custom target to compile all tests
|
# custom target to compile all tests
|
||||||
add_custom_target(all-${name}-tests)
|
add_custom_target(all-${name}-tests)
|
||||||
add_dependencies(all-${name}-tests ${TESTS})
|
if(TESTS)
|
||||||
|
add_dependencies(all-${name}-tests ${TESTS})
|
||||||
|
endif()
|
||||||
endmacro(uLib_add_tests name)
|
endmacro(uLib_add_tests name)
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -4,11 +4,31 @@
|
|||||||
################################################################################
|
################################################################################
|
||||||
|
|
||||||
cmake_minimum_required (VERSION 3.26)
|
cmake_minimum_required (VERSION 3.26)
|
||||||
|
if(POLICY CMP0167)
|
||||||
|
cmake_policy(SET CMP0167 NEW)
|
||||||
|
endif()
|
||||||
|
|
||||||
|
|
||||||
## -------------------------------------------------------------------------- ##
|
## -------------------------------------------------------------------------- ##
|
||||||
|
|
||||||
project(uLib)
|
project(uLib)
|
||||||
|
|
||||||
|
# CUDA Toolkit seems to be missing locally. Toggle ON if nvcc is made available.
|
||||||
|
option(USE_CUDA "Enable CUDA support" ON)
|
||||||
|
if(USE_CUDA)
|
||||||
|
set(CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS} -allow-unsupported-compiler")
|
||||||
|
set(CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS} --expt-relaxed-constexpr")
|
||||||
|
set(CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS} -Wno-deprecated-gpu-targets")
|
||||||
|
set(CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS} -Xcudafe \"--diag_suppress=20012\"")
|
||||||
|
set(CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS} -Xcudafe \"--diag_suppress=20014\"")
|
||||||
|
set(CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS} -Xcudafe \"--diag_suppress=20015\"")
|
||||||
|
find_package(CUDAToolkit REQUIRED)
|
||||||
|
enable_language(CUDA)
|
||||||
|
set(CMAKE_CUDA_ARCHITECTURES 61)
|
||||||
|
include_directories(${CMAKE_CUDA_TOOLKIT_INCLUDE_DIRECTORIES})
|
||||||
|
add_compile_definitions(USE_CUDA)
|
||||||
|
endif()
|
||||||
|
|
||||||
# The version number.
|
# The version number.
|
||||||
set(PROJECT_VERSION_MAJOR 0)
|
set(PROJECT_VERSION_MAJOR 0)
|
||||||
set(PROJECT_VERSION_MINOR 6)
|
set(PROJECT_VERSION_MINOR 6)
|
||||||
@@ -76,6 +96,7 @@ set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${CMAKE_CXX_WARNING_OPTION}")
|
|||||||
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -UULIB_SERIALIZATION_ON -Wno-cpp")
|
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -UULIB_SERIALIZATION_ON -Wno-cpp")
|
||||||
|
|
||||||
# CTEST framework
|
# CTEST framework
|
||||||
|
include(CTest)
|
||||||
enable_testing()
|
enable_testing()
|
||||||
|
|
||||||
|
|
||||||
@@ -85,7 +106,8 @@ set(Boost_USE_STATIC_LIBS OFF)
|
|||||||
set(Boost_USE_MULTITHREADED ON)
|
set(Boost_USE_MULTITHREADED ON)
|
||||||
set(Boost_USE_STATIC_RUNTIME OFF)
|
set(Boost_USE_STATIC_RUNTIME OFF)
|
||||||
message(STATUS "CMAKE_PREFIX_PATH is ${CMAKE_PREFIX_PATH}")
|
message(STATUS "CMAKE_PREFIX_PATH is ${CMAKE_PREFIX_PATH}")
|
||||||
find_package(Boost 1.45.0 COMPONENTS program_options REQUIRED)
|
|
||||||
|
find_package(Boost 1.45.0 COMPONENTS program_options serialization unit_test_framework REQUIRED)
|
||||||
include_directories(${Boost_INCLUDE_DIRS})
|
include_directories(${Boost_INCLUDE_DIRS})
|
||||||
|
|
||||||
find_package(Eigen3 CONFIG REQUIRED)
|
find_package(Eigen3 CONFIG REQUIRED)
|
||||||
@@ -98,6 +120,7 @@ include(${ROOT_USE_FILE})
|
|||||||
find_package(VTK REQUIRED)
|
find_package(VTK REQUIRED)
|
||||||
# include(${VTK_USE_FILE})
|
# include(${VTK_USE_FILE})
|
||||||
|
|
||||||
|
find_package(pybind11 REQUIRED)
|
||||||
|
|
||||||
|
|
||||||
option(CENTOS_SUPPORT "VTK definitions for CentOS" OFF)
|
option(CENTOS_SUPPORT "VTK definitions for CentOS" OFF)
|
||||||
@@ -190,6 +213,8 @@ add_subdirectory(${SRC_DIR}/Root)
|
|||||||
include_directories(${SRC_DIR}/Vtk)
|
include_directories(${SRC_DIR}/Vtk)
|
||||||
add_subdirectory(${SRC_DIR}/Vtk)
|
add_subdirectory(${SRC_DIR}/Vtk)
|
||||||
|
|
||||||
|
add_subdirectory(${SRC_DIR}/Python)
|
||||||
|
|
||||||
#add_subdirectory("${SRC_DIR}/utils/make_recipe")
|
#add_subdirectory("${SRC_DIR}/utils/make_recipe")
|
||||||
|
|
||||||
## Documentation and packages
|
## Documentation and packages
|
||||||
|
|||||||
16
CMakePresets.json
Normal file
16
CMakePresets.json
Normal file
@@ -0,0 +1,16 @@
|
|||||||
|
{
|
||||||
|
"version": 8,
|
||||||
|
"configurePresets": [
|
||||||
|
{
|
||||||
|
"name": "andrea",
|
||||||
|
"displayName": "Custom configure preset",
|
||||||
|
"description": "Sets Ninja generator, build and install directory",
|
||||||
|
"generator": "Ninja",
|
||||||
|
"binaryDir": "${sourceDir}/out/build/${presetName}",
|
||||||
|
"cacheVariables": {
|
||||||
|
"CMAKE_BUILD_TYPE": "Debug",
|
||||||
|
"CMAKE_INSTALL_PREFIX": "${sourceDir}/out/install/${presetName}"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
]
|
||||||
|
}
|
||||||
@@ -61,3 +61,10 @@ cmake --preset conan-release
|
|||||||
```bash
|
```bash
|
||||||
cmake --build build -j10
|
cmake --build build -j10
|
||||||
```
|
```
|
||||||
|
|
||||||
|
### Make python package
|
||||||
|
|
||||||
|
```bash
|
||||||
|
micromamba run -n mutom env USE_CUDA=ON poetry install
|
||||||
|
```
|
||||||
|
|
||||||
|
|||||||
54
build_python.py
Normal file
54
build_python.py
Normal file
@@ -0,0 +1,54 @@
|
|||||||
|
import os
|
||||||
|
import subprocess
|
||||||
|
import sys
|
||||||
|
import shutil
|
||||||
|
|
||||||
|
def build(setup_kwargs):
|
||||||
|
"""
|
||||||
|
Build the C++ extension using CMake.
|
||||||
|
This function is called by poetry-core during the build process.
|
||||||
|
The binary is placed directly inside the uLib directory in src/Python.
|
||||||
|
"""
|
||||||
|
# Root of the whole project where this build_extension.py is located
|
||||||
|
project_root = os.path.abspath(os.path.dirname(__file__))
|
||||||
|
|
||||||
|
# Where the extension should go
|
||||||
|
package_dir = os.path.join(project_root, "src/Python/uLib")
|
||||||
|
|
||||||
|
# Ensure package directory exists
|
||||||
|
os.makedirs(package_dir, exist_ok=True)
|
||||||
|
|
||||||
|
# Temporary build directory
|
||||||
|
build_temp = os.path.join(project_root, "build_python")
|
||||||
|
os.makedirs(build_temp, exist_ok=True)
|
||||||
|
|
||||||
|
print(f"--- Running CMake build in {build_temp} ---")
|
||||||
|
print(f"Project root: {project_root}")
|
||||||
|
print(f"Target binary dir: {package_dir}")
|
||||||
|
|
||||||
|
# Determine if CUDA should be enabled
|
||||||
|
use_cuda = os.environ.get("USE_CUDA", "OFF").upper()
|
||||||
|
if use_cuda in ["ON", "1", "TRUE", "YES"]:
|
||||||
|
use_cuda = "ON"
|
||||||
|
else:
|
||||||
|
use_cuda = "OFF"
|
||||||
|
|
||||||
|
# CMake configuration
|
||||||
|
cmake_args = [
|
||||||
|
f"-DCMAKE_LIBRARY_OUTPUT_DIRECTORY={package_dir}",
|
||||||
|
f"-DPYTHON_EXECUTABLE={sys.executable}",
|
||||||
|
"-DCMAKE_BUILD_TYPE=Release",
|
||||||
|
f"-DUSE_CUDA={use_cuda}",
|
||||||
|
"-G", "Unix Makefiles",
|
||||||
|
]
|
||||||
|
|
||||||
|
# Use micromamba to ensure Boost and VTK are found during the build
|
||||||
|
subprocess.check_call(["cmake", project_root] + cmake_args, cwd=build_temp)
|
||||||
|
subprocess.check_call(["cmake", "--build", ".", "--parallel", "--target", "uLib_python"], cwd=build_temp)
|
||||||
|
|
||||||
|
# Ensure the package is found by poetry during the wheel creation process.
|
||||||
|
# Return setup_kwargs for poetry-core.
|
||||||
|
return setup_kwargs
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
build({})
|
||||||
@@ -1,6 +1,7 @@
|
|||||||
[requires]
|
[requires]
|
||||||
eigen/3.4.0
|
eigen/3.4.0
|
||||||
boost/1.83.0
|
boost/1.83.0
|
||||||
|
pybind11/3.0.2
|
||||||
|
|
||||||
[generators]
|
[generators]
|
||||||
CMakeDeps
|
CMakeDeps
|
||||||
|
|||||||
41
docs/assets/css/extensions/tabbed.css
Normal file
41
docs/assets/css/extensions/tabbed.css
Normal file
@@ -0,0 +1,41 @@
|
|||||||
|
.tabbed-set {
|
||||||
|
display: flex;
|
||||||
|
position: relative;
|
||||||
|
flex-wrap: wrap;
|
||||||
|
}
|
||||||
|
|
||||||
|
.tabbed-set .highlight {
|
||||||
|
background: #ddd;
|
||||||
|
}
|
||||||
|
|
||||||
|
.tabbed-set .tabbed-content {
|
||||||
|
display: none;
|
||||||
|
order: 99;
|
||||||
|
width: 100%;
|
||||||
|
}
|
||||||
|
|
||||||
|
.tabbed-set label {
|
||||||
|
width: auto;
|
||||||
|
margin: 0 0.5em;
|
||||||
|
padding: 0.25em;
|
||||||
|
font-size: 120%;
|
||||||
|
cursor: pointer;
|
||||||
|
color: #ffffff !important;
|
||||||
|
}
|
||||||
|
|
||||||
|
.tabbed-set input {
|
||||||
|
position: absolute;
|
||||||
|
opacity: 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
.tabbed-set input:nth-child(n+1) {
|
||||||
|
color: #333333;
|
||||||
|
}
|
||||||
|
|
||||||
|
.tabbed-set input:nth-child(n+1):checked + label {
|
||||||
|
color: cyan !important;
|
||||||
|
}
|
||||||
|
|
||||||
|
.tabbed-set input:nth-child(n+1):checked + label + .tabbed-content {
|
||||||
|
display: block;
|
||||||
|
}
|
||||||
17
docs/assets/css/extra.css
Normal file
17
docs/assets/css/extra.css
Normal file
@@ -0,0 +1,17 @@
|
|||||||
|
@import "extensions/tabbed.css";
|
||||||
|
|
||||||
|
.md-grid {
|
||||||
|
max-width: 100%;
|
||||||
|
}
|
||||||
|
|
||||||
|
.md-main__inner {
|
||||||
|
margin-top: 0;
|
||||||
|
padding-top: 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
.md-sidebar--secondary {
|
||||||
|
right: 1.5rem;
|
||||||
|
top: 4.8rem;
|
||||||
|
transform: none;
|
||||||
|
width: 18rem;
|
||||||
|
}
|
||||||
30
docs/docker/Dockerfile
Normal file
30
docs/docker/Dockerfile
Normal file
@@ -0,0 +1,30 @@
|
|||||||
|
# Stage 1: Build the static site using MkDocs
|
||||||
|
FROM python:3.9-slim-buster as builder
|
||||||
|
|
||||||
|
# Set the working directory
|
||||||
|
WORKDIR /app
|
||||||
|
|
||||||
|
# Copy the requirements file
|
||||||
|
COPY requirements.txt .
|
||||||
|
|
||||||
|
# Install the Python dependencies
|
||||||
|
RUN pip install --no-cache-dir -r requirements.txt
|
||||||
|
|
||||||
|
# Copy the rest of the application files
|
||||||
|
COPY ../.. .
|
||||||
|
|
||||||
|
# Build the MkDocs site
|
||||||
|
RUN mkdocs build
|
||||||
|
|
||||||
|
|
||||||
|
# Stage 2: Serve the static files with Nginx
|
||||||
|
FROM nginx:alpine
|
||||||
|
|
||||||
|
# Copy the built site from the builder stage
|
||||||
|
COPY --from=builder /app/site /usr/share/nginx/html
|
||||||
|
|
||||||
|
# Expose port 80 for the web server
|
||||||
|
EXPOSE 80
|
||||||
|
|
||||||
|
# Command to run Nginx in the foreground
|
||||||
|
CMD ["nginx", "-g", "daemon off;"]
|
||||||
14
docs/docker/Dockerfile.dev
Normal file
14
docs/docker/Dockerfile.dev
Normal file
@@ -0,0 +1,14 @@
|
|||||||
|
# Dockerfile for development with live-reloading
|
||||||
|
FROM python:3.9-slim-buster
|
||||||
|
|
||||||
|
WORKDIR /app
|
||||||
|
|
||||||
|
# Copy and install dependencies
|
||||||
|
COPY requirements.txt .
|
||||||
|
RUN pip install --no-cache-dir -r requirements.txt
|
||||||
|
|
||||||
|
# Expose the port MkDocs serve will run on
|
||||||
|
EXPOSE 8000
|
||||||
|
|
||||||
|
# Command to run the development server
|
||||||
|
CMD ["mkdocs", "serve", "--dev-addr", "0.0.0.0:8000"]
|
||||||
13
docs/docker/docker-compose.yml
Normal file
13
docs/docker/docker-compose.yml
Normal file
@@ -0,0 +1,13 @@
|
|||||||
|
version: '3.8'
|
||||||
|
|
||||||
|
services:
|
||||||
|
mkdocs:
|
||||||
|
build:
|
||||||
|
context: .
|
||||||
|
dockerfile: Dockerfile.dev
|
||||||
|
ports:
|
||||||
|
- "8000:8000"
|
||||||
|
volumes:
|
||||||
|
- ../..:/app
|
||||||
|
environment:
|
||||||
|
- GIT_DISCOVERY_ACROSS_FILESYSTEM=1
|
||||||
17
docs/docker/requirements.txt
Normal file
17
docs/docker/requirements.txt
Normal file
@@ -0,0 +1,17 @@
|
|||||||
|
# ------------------------------------------------------------------
|
||||||
|
# MkDocs runtime dependencies for the docs Docker image
|
||||||
|
# ------------------------------------------------------------------
|
||||||
|
|
||||||
|
# Core: theme (provides mkdocs itself as a transitive dep)
|
||||||
|
mkdocs-material==9.7.1
|
||||||
|
|
||||||
|
# pymdownx.* extensions used in mkdocs.yml:
|
||||||
|
# arithmatex, highlight, superfences, tabbed, details, blocks.caption
|
||||||
|
# (also a hard dep of mkdocs-material, pinned here for reproducibility)
|
||||||
|
pymdown-extensions>=10.0
|
||||||
|
|
||||||
|
# Markdown math rendering support (arithmatex generic mode)
|
||||||
|
# JS side is loaded via CDN (polyfill.io + MathJax), no extra Python pkg needed
|
||||||
|
|
||||||
|
# Optional: PDF export plugin (exporter: block, currently commented out in mkdocs.yml)
|
||||||
|
mkdocs-exporter
|
||||||
1
docs/docker/runtime.txt
Normal file
1
docs/docker/runtime.txt
Normal file
@@ -0,0 +1 @@
|
|||||||
|
3.7
|
||||||
63
docs/index.md
Normal file
63
docs/index.md
Normal file
@@ -0,0 +1,63 @@
|
|||||||
|
# uLib
|
||||||
|
|
||||||
|
[](https://zenodo.org/badge/latestdoi/36926725)
|
||||||
|
|
||||||
|
**uLib** is the base toolkit library for the **CMT (Cosmic Muon Tomography)** project, developed at the University of Padova and INFN Sezione di Padova, Italy.
|
||||||
|
|
||||||
|
It provides:
|
||||||
|
|
||||||
|
- **Core** – object model, timers, configuration, UUID utilities.
|
||||||
|
- **Math** – linear algebra (Eigen3), structured grids, voxel images, ray-tracing, image filters.
|
||||||
|
- **Python bindings** – full pybind11 interface for scripting and analysis workflows.
|
||||||
|
- Optional **CUDA** acceleration for voxel filtering (transparent RAM ↔ VRAM management).
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## Quick Start
|
||||||
|
|
||||||
|
=== "Users (pip / poetry)"
|
||||||
|
```bash
|
||||||
|
# Activate your conda/micromamba environment first
|
||||||
|
micromamba activate mutom
|
||||||
|
|
||||||
|
poetry install # CPU build
|
||||||
|
USE_CUDA=ON poetry install # GPU build
|
||||||
|
```
|
||||||
|
|
||||||
|
=== "Developers (CMake)"
|
||||||
|
```bash
|
||||||
|
conan install . --output-folder=build --build=missing
|
||||||
|
cmake --preset conan-release
|
||||||
|
cmake --build build --target uLib_python -j$(nproc)
|
||||||
|
|
||||||
|
export PYTHONPATH="$(pwd)/build/src/Python:$(pwd)/src/Python"
|
||||||
|
```
|
||||||
|
|
||||||
|
Then in Python:
|
||||||
|
|
||||||
|
```python
|
||||||
|
import uLib
|
||||||
|
|
||||||
|
# Core
|
||||||
|
timer = uLib.Core.Timer()
|
||||||
|
timer.Start()
|
||||||
|
|
||||||
|
# Math
|
||||||
|
grid = uLib.Math.StructuredGrid([10, 10, 10])
|
||||||
|
grid.SetSpacing([1.0, 1.0, 1.0])
|
||||||
|
|
||||||
|
img = uLib.Math.VoxImage([10, 10, 10])
|
||||||
|
img.SetValue(0, 3.14)
|
||||||
|
print(img.GetValue(0))
|
||||||
|
```
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## Documentation Sections
|
||||||
|
|
||||||
|
| Section | Description |
|
||||||
|
|---|---|
|
||||||
|
| [Python – Installation](python/installation.md) | Environment setup, user install, developer build |
|
||||||
|
| [Python – API Usage](python/usage.md) | Full API reference with examples |
|
||||||
|
| [Python – Developer Guide](python/developer_guide.md) | Adding bindings, running tests, build details |
|
||||||
|
| [C++ Build – Usage & CUDA](usage/usage.md) | CMake build, CUDA configuration |
|
||||||
179
docs/python/developer_guide.md
Normal file
179
docs/python/developer_guide.md
Normal file
@@ -0,0 +1,179 @@
|
|||||||
|
# Developer Guide – Python Bindings
|
||||||
|
|
||||||
|
This guide is aimed at contributors who want to extend or modify the Python bindings for `uLib`.
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## Repository Layout
|
||||||
|
|
||||||
|
```
|
||||||
|
ulib/
|
||||||
|
├── src/
|
||||||
|
│ └── Python/
|
||||||
|
│ ├── module.cpp # pybind11 module entry point
|
||||||
|
│ ├── core_bindings.cpp # uLib::Core bindings
|
||||||
|
│ ├── math_bindings.cpp # uLib::Math bindings
|
||||||
|
│ ├── math_filters_bindings.cpp# VoxImageFilter bindings
|
||||||
|
│ ├── CMakeLists.txt # builds uLib_python shared lib
|
||||||
|
│ ├── testing/ # Python unit tests
|
||||||
|
│ │ ├── pybind_test.py
|
||||||
|
│ │ ├── core_pybind_test.py
|
||||||
|
│ │ ├── math_pybind_test.py
|
||||||
|
│ │ └── math_filters_test.py
|
||||||
|
│ └── uLib/ # Python package (uLib_python.so lands here)
|
||||||
|
│ └── __init__.py
|
||||||
|
├── build_python.py # poetry build hook (calls CMake)
|
||||||
|
├── pyproject.toml # poetry metadata
|
||||||
|
└── condaenv.yml # conda/micromamba environment
|
||||||
|
```
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## Adding a New Binding
|
||||||
|
|
||||||
|
All bindings live in the four source files listed above. The module entry point `module.cpp` calls `init_core()`, `init_math()`, and `init_math_filters()` in order.
|
||||||
|
|
||||||
|
### 1. Pick (or create) the right binding file
|
||||||
|
|
||||||
|
| C++ header location | Binding file |
|
||||||
|
|---|---|
|
||||||
|
| `src/Core/` | `core_bindings.cpp` |
|
||||||
|
| `src/Math/` (geometry, grids, VoxImage) | `math_bindings.cpp` |
|
||||||
|
| `src/Math/VoxImageFilter*.hpp` | `math_filters_bindings.cpp` |
|
||||||
|
|
||||||
|
### 2. Add the `#include` directive
|
||||||
|
|
||||||
|
```cpp
|
||||||
|
// math_bindings.cpp
|
||||||
|
#include "Math/MyNewClass.h"
|
||||||
|
```
|
||||||
|
|
||||||
|
### 3. Write the pybind11 binding inside the appropriate `init_*` function
|
||||||
|
|
||||||
|
```cpp
|
||||||
|
void init_math(py::module_ &m) {
|
||||||
|
// ... existing bindings ...
|
||||||
|
|
||||||
|
py::class_<MyNewClass>(m, "MyNewClass")
|
||||||
|
.def(py::init<>())
|
||||||
|
.def("MyMethod", &MyNewClass::MyMethod)
|
||||||
|
.def("AnotherMethod", &MyNewClass::AnotherMethod,
|
||||||
|
py::arg("x"), py::arg("y") = 0.0f);
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
### 4. Rebuild only the Python target
|
||||||
|
|
||||||
|
```bash
|
||||||
|
cmake --build build --target uLib_python -j$(nproc)
|
||||||
|
```
|
||||||
|
|
||||||
|
### 5. Write a Python test
|
||||||
|
|
||||||
|
Add a new test class to the relevant test file (or create a new one under `src/Python/testing/`):
|
||||||
|
|
||||||
|
```python
|
||||||
|
# src/Python/testing/math_pybind_test.py
|
||||||
|
class TestMyNewClass(unittest.TestCase):
|
||||||
|
def test_basic(self):
|
||||||
|
obj = uLib.Math.MyNewClass()
|
||||||
|
result = obj.MyMethod()
|
||||||
|
self.assertAlmostEqual(result, expected_value)
|
||||||
|
```
|
||||||
|
|
||||||
|
Register the test in `src/Python/CMakeLists.txt` if you add a new file:
|
||||||
|
|
||||||
|
```cmake
|
||||||
|
add_test(NAME pybind_my_new
|
||||||
|
COMMAND ${Python3_EXECUTABLE} ${CMAKE_CURRENT_SOURCE_DIR}/testing/my_new_test.py)
|
||||||
|
set_tests_properties(pybind_my_new PROPERTIES
|
||||||
|
ENVIRONMENT "PYTHONPATH=$<TARGET_FILE_DIR:uLib_python>:${PROJECT_SOURCE_DIR}/src/Python")
|
||||||
|
```
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## Build System Details
|
||||||
|
|
||||||
|
### CMakeLists.txt (`src/Python/`)
|
||||||
|
|
||||||
|
`pybind11_add_module` compiles the shared library `uLib_python` and links it against the C++ static/shared libraries `uLibCore` and `uLibMath`. The install target copies the `.so` into the standard library directory.
|
||||||
|
|
||||||
|
```cmake
|
||||||
|
pybind11_add_module(uLib_python
|
||||||
|
module.cpp core_bindings.cpp math_bindings.cpp math_filters_bindings.cpp)
|
||||||
|
|
||||||
|
target_link_libraries(uLib_python PRIVATE uLibCore uLibMath)
|
||||||
|
```
|
||||||
|
|
||||||
|
### poetry / build_python.py
|
||||||
|
|
||||||
|
`pyproject.toml` declares `build_python.py` as the custom build hook. When `poetry install` or `poetry build` is invoked it:
|
||||||
|
|
||||||
|
1. Calls `cmake <root> -DCMAKE_LIBRARY_OUTPUT_DIRECTORY=<pkg_dir> ...` in `build_python/`.
|
||||||
|
2. Builds only the `uLib_python` target.
|
||||||
|
3. The resulting `.so` is placed inside `src/Python/uLib/` so it is picked up by Poetry as a package data file.
|
||||||
|
|
||||||
|
The `USE_CUDA` environment variable gates CUDA support at build time:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
USE_CUDA=ON poetry install # with CUDA
|
||||||
|
USE_CUDA=OFF poetry install # CPU only (default)
|
||||||
|
```
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## Running All Tests
|
||||||
|
|
||||||
|
```bash
|
||||||
|
# From the repository root, with PYTHONPATH set:
|
||||||
|
export PYTHONPATH="$(pwd)/build/src/Python:$(pwd)/src/Python"
|
||||||
|
|
||||||
|
python -m pytest src/Python/testing/ -v
|
||||||
|
```
|
||||||
|
|
||||||
|
Or through CMake's test runner (after building the full project):
|
||||||
|
|
||||||
|
```bash
|
||||||
|
cd build
|
||||||
|
ctest --output-on-failure -R pybind
|
||||||
|
```
|
||||||
|
|
||||||
|
Expected output (all passing):
|
||||||
|
|
||||||
|
```
|
||||||
|
Start 1: pybind_general
|
||||||
|
1/4 Test #1: pybind_general ............. Passed
|
||||||
|
Start 2: pybind_core
|
||||||
|
2/4 Test #2: pybind_core ................ Passed
|
||||||
|
Start 3: pybind_math
|
||||||
|
3/4 Test #3: pybind_math ................ Passed
|
||||||
|
Start 4: pybind_math_filters
|
||||||
|
4/4 Test #4: pybind_math_filters ........ Passed
|
||||||
|
```
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## Memory Management Notes
|
||||||
|
|
||||||
|
`uLib::Vector<T>` has explicit GPU memory management. When wrapping methods that return references to internal data, use `py::return_value_policy::reference_internal` to avoid dangling references:
|
||||||
|
|
||||||
|
```cpp
|
||||||
|
.def("Data", &VoxImage<Voxel>::Data,
|
||||||
|
py::return_value_policy::reference_internal)
|
||||||
|
```
|
||||||
|
|
||||||
|
For objects held by `std::unique_ptr` without Python-side deletion, use `py::nodelete`:
|
||||||
|
|
||||||
|
```cpp
|
||||||
|
py::class_<Abstract::VoxImageFilter,
|
||||||
|
std::unique_ptr<Abstract::VoxImageFilter, py::nodelete>>(m, "AbstractVoxImageFilter")
|
||||||
|
```
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## Useful References
|
||||||
|
|
||||||
|
- [pybind11 documentation](https://pybind11.readthedocs.io)
|
||||||
|
- [pybind11 – STL containers](https://pybind11.readthedocs.io/en/stable/advanced/cast/stl.html)
|
||||||
|
- [pybind11 – Eigen integration](https://pybind11.readthedocs.io/en/stable/advanced/cast/eigen.html)
|
||||||
|
- [CMake – pybind11 integration](https://pybind11.readthedocs.io/en/stable/compiling.html)
|
||||||
146
docs/python/installation.md
Normal file
146
docs/python/installation.md
Normal file
@@ -0,0 +1,146 @@
|
|||||||
|
# Python Installation
|
||||||
|
|
||||||
|
The `uLib` Python package exposes the Core and Math C++ libraries via [pybind11](https://pybind11.readthedocs.io) bindings. There are two ways to install it: as an **end user** (pre-built wheel / pip) or as a **developer** (editable build from source).
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## Prerequisites
|
||||||
|
|
||||||
|
`uLib` depends on native C++ libraries that must be compiled. Ensure the following are available in your environment before installing:
|
||||||
|
|
||||||
|
| Dependency | Minimum version | Notes |
|
||||||
|
|---|---|---|
|
||||||
|
| Python | 3.9 | |
|
||||||
|
| CMake | 3.12 | |
|
||||||
|
| pybind11 | 2.6.0 | |
|
||||||
|
| Conan | 2.x | for Eigen3 / Boost |
|
||||||
|
| micromamba / conda | any | recommended – provides ROOT, VTK |
|
||||||
|
|
||||||
|
### Creating the `mutom` Conda/Micromamba Environment
|
||||||
|
|
||||||
|
A ready-to-use environment definition is provided as `condaenv.yml` at the repository root.
|
||||||
|
|
||||||
|
=== "Micromamba"
|
||||||
|
```bash
|
||||||
|
micromamba env create -f condaenv.yml
|
||||||
|
micromamba activate mutom
|
||||||
|
```
|
||||||
|
|
||||||
|
=== "Conda"
|
||||||
|
```bash
|
||||||
|
conda env create -f condaenv.yml
|
||||||
|
conda activate mutom
|
||||||
|
```
|
||||||
|
|
||||||
|
The environment installs CMake, Conan, ROOT, VTK, and the compiler toolchain.
|
||||||
|
|
||||||
|
> **CUDA (optional)**
|
||||||
|
> If you want GPU-accelerated voxel filtering, you also need NVCC inside the environment:
|
||||||
|
> ```bash
|
||||||
|
> micromamba install cuda-nvcc -c conda-forge
|
||||||
|
> ```
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## User Installation (wheel / pip)
|
||||||
|
|
||||||
|
Once the native dependencies are present in your environment, install the package with Poetry or pip:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
# Activate your environment first
|
||||||
|
micromamba activate mutom
|
||||||
|
|
||||||
|
# Build and install (CUDA disabled by default)
|
||||||
|
poetry install
|
||||||
|
|
||||||
|
# Build and install with CUDA support
|
||||||
|
USE_CUDA=ON poetry install
|
||||||
|
```
|
||||||
|
|
||||||
|
After installation the module is importable from anywhere in the environment:
|
||||||
|
|
||||||
|
```python
|
||||||
|
import uLib
|
||||||
|
print(dir(uLib.Core))
|
||||||
|
print(dir(uLib.Math))
|
||||||
|
```
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## Developer Installation (editable / in-tree build)
|
||||||
|
|
||||||
|
For development you typically want to skip the packaging layer and work directly against the CMake build tree.
|
||||||
|
|
||||||
|
### Step 1 – Install Conan dependencies
|
||||||
|
|
||||||
|
```bash
|
||||||
|
conan profile detect # first time only
|
||||||
|
conan install . --output-folder=build --build=missing
|
||||||
|
```
|
||||||
|
|
||||||
|
### Step 2 – Configure and build
|
||||||
|
|
||||||
|
```bash
|
||||||
|
# Standard release build
|
||||||
|
cmake --preset conan-release
|
||||||
|
|
||||||
|
# …or manually
|
||||||
|
cmake -B build \
|
||||||
|
-DCMAKE_TOOLCHAIN_FILE=build/conan_toolchain.cmake \
|
||||||
|
-DCMAKE_BUILD_TYPE=Release \
|
||||||
|
-DUSE_CUDA=OFF # set to ON when a GPU is available
|
||||||
|
|
||||||
|
cmake --build build --target uLib_python -j$(nproc)
|
||||||
|
```
|
||||||
|
|
||||||
|
The shared library (`uLib_python*.so`) is written to `build/src/Python/`.
|
||||||
|
|
||||||
|
### Step 3 – Make the module importable
|
||||||
|
|
||||||
|
Point `PYTHONPATH` at the build output **and** the Python source directory (the latter carries the `uLib/__init__.py` that stitches sub-modules together):
|
||||||
|
|
||||||
|
```bash
|
||||||
|
export PYTHONPATH="$(pwd)/build/src/Python:$(pwd)/src/Python:$PYTHONPATH"
|
||||||
|
python -c "import uLib; print(uLib.__version__)"
|
||||||
|
```
|
||||||
|
|
||||||
|
Or, for a one-shot check:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
PYTHONPATH="build/src/Python:src/Python" python src/Python/testing/pybind_test.py
|
||||||
|
```
|
||||||
|
|
||||||
|
### Step 4 – Run the tests
|
||||||
|
|
||||||
|
CMake registers the Python tests alongside the C++ ones; use `ctest` from the build directory:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
cd build
|
||||||
|
ctest --output-on-failure -R pybind
|
||||||
|
```
|
||||||
|
|
||||||
|
Individual test scripts can also be run directly once `PYTHONPATH` is set:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
python src/Python/testing/core_pybind_test.py
|
||||||
|
python src/Python/testing/math_pybind_test.py
|
||||||
|
python src/Python/testing/math_filters_test.py
|
||||||
|
```
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## Verifying the Installation
|
||||||
|
|
||||||
|
```python
|
||||||
|
import uLib
|
||||||
|
|
||||||
|
# Core module
|
||||||
|
obj = uLib.Core.Object()
|
||||||
|
timer = uLib.Core.Timer()
|
||||||
|
timer.Start()
|
||||||
|
elapsed = timer.StopWatch() # float, seconds
|
||||||
|
|
||||||
|
# Math module
|
||||||
|
v3 = uLib.Math.Vector3f([1.0, 0.0, 0.0])
|
||||||
|
print(v3[0]) # 1.0
|
||||||
|
```
|
||||||
373
docs/python/usage.md
Normal file
373
docs/python/usage.md
Normal file
@@ -0,0 +1,373 @@
|
|||||||
|
# Python API Usage
|
||||||
|
|
||||||
|
The `uLib` Python package is split into two sub-modules mirroring the C++ library:
|
||||||
|
|
||||||
|
| Sub-module | Contents |
|
||||||
|
|---|---|
|
||||||
|
| `uLib.Core` | Low-level utilities: `Object`, `Timer`, `Options`, `TypeRegister` |
|
||||||
|
| `uLib.Math` | Geometry, grids, voxel images, ray-tracing, image filters |
|
||||||
|
|
||||||
|
```python
|
||||||
|
import uLib
|
||||||
|
# Sub-modules are accessible as attributes
|
||||||
|
uLib.Core # core utilities
|
||||||
|
uLib.Math # mathematical structures
|
||||||
|
```
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## uLib.Core
|
||||||
|
|
||||||
|
### Object
|
||||||
|
|
||||||
|
Base class for uLib objects; exposed to Python for type-hierarchy purposes.
|
||||||
|
|
||||||
|
```python
|
||||||
|
obj = uLib.Core.Object()
|
||||||
|
copy = obj.DeepCopy()
|
||||||
|
```
|
||||||
|
|
||||||
|
### Timer
|
||||||
|
|
||||||
|
Precision wall-clock timer.
|
||||||
|
|
||||||
|
```python
|
||||||
|
import time
|
||||||
|
timer = uLib.Core.Timer()
|
||||||
|
timer.Start()
|
||||||
|
time.sleep(0.5)
|
||||||
|
elapsed = timer.StopWatch() # returns elapsed seconds as float
|
||||||
|
print(f"Elapsed: {elapsed:.3f} s")
|
||||||
|
```
|
||||||
|
|
||||||
|
### Options
|
||||||
|
|
||||||
|
Wraps Boost.ProgramOptions for INI-style configuration files.
|
||||||
|
|
||||||
|
```python
|
||||||
|
opt = uLib.Core.Options("My Program")
|
||||||
|
opt.parse_config_file("config.ini") # load settings
|
||||||
|
n = opt.count("my_key") # check if key exists
|
||||||
|
opt.save_config_file("out.ini")
|
||||||
|
```
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## uLib.Math – Linear Algebra
|
||||||
|
|
||||||
|
The math module exposes Eigen3 vectors and matrices as well-typed Python objects with NumPy interoperability.
|
||||||
|
|
||||||
|
### Fixed-size Vectors
|
||||||
|
|
||||||
|
```python
|
||||||
|
import numpy as np
|
||||||
|
import uLib
|
||||||
|
|
||||||
|
# Construct from list
|
||||||
|
v3 = uLib.Math.Vector3f([1.0, 2.0, 3.0])
|
||||||
|
print(v3[0], v3[1], v3[2]) # 1.0 2.0 3.0
|
||||||
|
|
||||||
|
# Construct from NumPy array
|
||||||
|
arr = np.array([4.0, 5.0, 6.0], dtype=np.float32)
|
||||||
|
v3b = uLib.Math.Vector3f(arr)
|
||||||
|
|
||||||
|
# Zero-initialise
|
||||||
|
v4d = uLib.Math.Vector4d() # all zeros
|
||||||
|
|
||||||
|
# Available types
|
||||||
|
# Vector1f / 2f / 3f / 4f (float32)
|
||||||
|
# Vector1d / 2d / 3d / 4d (float64)
|
||||||
|
# Vector1i / 2i / 3i / 4i (int32)
|
||||||
|
```
|
||||||
|
|
||||||
|
### Fixed-size Matrices
|
||||||
|
|
||||||
|
```python
|
||||||
|
# 2-by-2 float matrix
|
||||||
|
m2f = uLib.Math.Matrix2f()
|
||||||
|
m2f[0, 0] = 1; m2f[0, 1] = 2
|
||||||
|
m2f[1, 0] = 3; m2f[1, 1] = 4
|
||||||
|
|
||||||
|
# From list (row-major)
|
||||||
|
m4f = uLib.Math.Matrix4f([1,0,0,0, 0,1,0,0, 0,0,1,0, 0,0,0,1])
|
||||||
|
|
||||||
|
# From NumPy (2-D array)
|
||||||
|
mat = np.eye(3, dtype=np.float32)
|
||||||
|
m3f = uLib.Math.Matrix3f(mat)
|
||||||
|
|
||||||
|
# Dynamic matrices
|
||||||
|
mXf = uLib.Math.MatrixXf(4, 4) # 4×4 float, zeros
|
||||||
|
```
|
||||||
|
|
||||||
|
### Homogeneous Types
|
||||||
|
|
||||||
|
```python
|
||||||
|
# HPoint3f – a 3-D point in homogeneous coordinates (w = 1)
|
||||||
|
p = uLib.Math.HPoint3f(1.0, 2.0, 3.0)
|
||||||
|
|
||||||
|
# HVector3f – a free vector (w = 0)
|
||||||
|
v = uLib.Math.HVector3f(0.0, 1.0, 0.0)
|
||||||
|
|
||||||
|
# HLine3f – a parametric ray
|
||||||
|
line = uLib.Math.HLine3f()
|
||||||
|
line.origin = uLib.Math.HPoint3f(0, 0, 0)
|
||||||
|
line.direction = uLib.Math.HVector3f(0, 0, 1)
|
||||||
|
```
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## uLib.Math – Transforms and Geometry
|
||||||
|
|
||||||
|
### AffineTransform
|
||||||
|
|
||||||
|
A rigid-body / affine transform stored as a 4×4 matrix.
|
||||||
|
|
||||||
|
```python
|
||||||
|
tf = uLib.Math.AffineTransform()
|
||||||
|
|
||||||
|
tf.SetPosition([1.0, 0.0, 0.0]) # translate
|
||||||
|
tf.Translate([0.0, 1.0, 0.0]) # cumulative translate
|
||||||
|
tf.Scale([2.0, 2.0, 2.0]) # uniform scale
|
||||||
|
tf.Rotate(uLib.Math.Vector3f([0, 0, 3.14159])) # Euler angles (rad)
|
||||||
|
|
||||||
|
mat4 = tf.GetWorldMatrix() # 4×4 matrix
|
||||||
|
pos = tf.GetPosition() # Vector3f
|
||||||
|
```
|
||||||
|
|
||||||
|
### Geometry
|
||||||
|
|
||||||
|
Inherits `AffineTransform`; converts points between world and local frames.
|
||||||
|
|
||||||
|
```python
|
||||||
|
geo = uLib.Math.Geometry()
|
||||||
|
geo.SetPosition([1.0, 1.0, 1.0])
|
||||||
|
|
||||||
|
world_pt = uLib.Math.Vector4f([2.0, 3.0, 2.0, 1.0])
|
||||||
|
local_pt = geo.GetLocalPoint(world_pt)
|
||||||
|
back = geo.GetWorldPoint(local_pt)
|
||||||
|
# back ≈ [2, 3, 2, 1]
|
||||||
|
```
|
||||||
|
|
||||||
|
### ContainerBox
|
||||||
|
|
||||||
|
An axis-aligned bounding box with an associated transform.
|
||||||
|
|
||||||
|
```python
|
||||||
|
box = uLib.Math.ContainerBox()
|
||||||
|
box.SetOrigin([-1.0, -1.0, -1.0])
|
||||||
|
box.SetSize([2.0, 2.0, 2.0])
|
||||||
|
print(box.GetSize()) # [2, 2, 2]
|
||||||
|
```
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## uLib.Math – Structured Grids
|
||||||
|
|
||||||
|
### StructuredGrid (3-D)
|
||||||
|
|
||||||
|
A 3-D voxel grid (origin, spacing, and integer dimensions).
|
||||||
|
|
||||||
|
```python
|
||||||
|
dims = uLib.Math.Vector3i([10, 10, 10])
|
||||||
|
grid = uLib.Math.StructuredGrid(dims)
|
||||||
|
grid.SetSpacing([1.0, 1.0, 1.0])
|
||||||
|
grid.SetOrigin([0.0, 0.0, 0.0])
|
||||||
|
|
||||||
|
print(grid.GetSpacing()) # [1, 1, 1]
|
||||||
|
print(grid.IsInsideBounds([5, 5, 5, 1])) # True
|
||||||
|
idx = grid.Find([2.5, 2.5, 2.5]) # returns grid cell index
|
||||||
|
```
|
||||||
|
|
||||||
|
### Structured2DGrid / Structured4DGrid
|
||||||
|
|
||||||
|
```python
|
||||||
|
grid2d = uLib.Math.Structured2DGrid()
|
||||||
|
grid2d.SetDims([100, 100])
|
||||||
|
grid2d.SetPhysicalSpace([0, 0], [1, 1])
|
||||||
|
print(grid2d.GetSpacing())
|
||||||
|
```
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## uLib.Math – VoxImage
|
||||||
|
|
||||||
|
`VoxImage` is a 3-D voxel volume where each cell stores a `Voxel` ( `.Value` + `.Count`).
|
||||||
|
|
||||||
|
```python
|
||||||
|
dims = uLib.Math.Vector3i([20, 20, 20])
|
||||||
|
img = uLib.Math.VoxImage(dims)
|
||||||
|
img.SetSpacing([0.5, 0.5, 0.5])
|
||||||
|
|
||||||
|
# Access by linear index
|
||||||
|
img.SetValue(0, 42.0)
|
||||||
|
print(img.GetValue(0)) # 42.0
|
||||||
|
|
||||||
|
# Access by 3-D index
|
||||||
|
img.SetValue(uLib.Math.Vector3i([1, 1, 1]), 7.5)
|
||||||
|
print(img.GetValue(uLib.Math.Vector3i([1, 1, 1]))) # 7.5
|
||||||
|
|
||||||
|
# Clipping / masking helpers
|
||||||
|
cropped = img.clipImage(uLib.Math.Vector3i([2, 2, 2]),
|
||||||
|
uLib.Math.Vector3i([18, 18, 18]))
|
||||||
|
masked = img.maskImage(0.0, 100.0, 0.0) # mask outside [0, 100]
|
||||||
|
|
||||||
|
# I/O
|
||||||
|
img.ExportToVti("output.vti")
|
||||||
|
img.ImportFromVti("output.vti")
|
||||||
|
```
|
||||||
|
|
||||||
|
### Voxel (element type)
|
||||||
|
|
||||||
|
```python
|
||||||
|
vox = uLib.Math.Voxel()
|
||||||
|
vox.Value = 1.5
|
||||||
|
vox.Count = 3
|
||||||
|
|
||||||
|
data = img.Data() # returns the underlying Vector_Voxel
|
||||||
|
vox0 = data[0]
|
||||||
|
print(vox0.Value, vox0.Count)
|
||||||
|
```
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## uLib.Math – VoxRaytracer
|
||||||
|
|
||||||
|
Performs ray-tracing through a `StructuredGrid` and returns per-voxel chord lengths.
|
||||||
|
|
||||||
|
```python
|
||||||
|
import numpy as np
|
||||||
|
import uLib
|
||||||
|
|
||||||
|
grid = uLib.Math.StructuredGrid([10, 10, 10])
|
||||||
|
grid.SetSpacing([1.0, 1.0, 1.0])
|
||||||
|
grid.SetOrigin([0.0, 0.0, 0.0])
|
||||||
|
|
||||||
|
rt = uLib.Math.VoxRaytracer(grid)
|
||||||
|
|
||||||
|
# Trace a ray between two homogeneous points (x, y, z, w=1)
|
||||||
|
p1 = np.array([0.5, 0.5, -1.0, 1.0], dtype=np.float32)
|
||||||
|
p2 = np.array([0.5, 0.5, 11.0, 1.0], dtype=np.float32)
|
||||||
|
result = rt.TraceBetweenPoints(p1, p2)
|
||||||
|
|
||||||
|
print("Voxels crossed:", result.Count())
|
||||||
|
print("Total length :", result.TotalLength())
|
||||||
|
|
||||||
|
elements = result.Data()
|
||||||
|
for i in range(result.Count()):
|
||||||
|
print(f" vox_id={elements[i].vox_id} L={elements[i].L:.4f}")
|
||||||
|
```
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## uLib.Math – Image Filters
|
||||||
|
|
||||||
|
All filters share the same interface: construct with a kernel size, attach a `VoxImage`, optionally set parameters, then call `.Run()`.
|
||||||
|
|
||||||
|
```python
|
||||||
|
import uLib
|
||||||
|
|
||||||
|
dims = uLib.Math.Vector3i([10, 10, 10])
|
||||||
|
img = uLib.Math.VoxImage(dims)
|
||||||
|
for i in range(10**3):
|
||||||
|
img.SetValue(i, float(i))
|
||||||
|
|
||||||
|
kernel_dims = uLib.Math.Vector3i([3, 3, 3])
|
||||||
|
```
|
||||||
|
|
||||||
|
### Linear (Gaussian / Box) Filter
|
||||||
|
|
||||||
|
```python
|
||||||
|
filt = uLib.Math.VoxFilterAlgorithmLinear(kernel_dims)
|
||||||
|
filt.SetImage(img)
|
||||||
|
filt.SetKernelNumericXZY([1.0] * 27) # uniform box kernel, length = product of dims
|
||||||
|
filt.Run()
|
||||||
|
```
|
||||||
|
|
||||||
|
### ABTrim Filter
|
||||||
|
|
||||||
|
Applies alpha-beta trimming to remove outliers before averaging.
|
||||||
|
|
||||||
|
```python
|
||||||
|
filt = uLib.Math.VoxFilterAlgorithmAbtrim(kernel_dims)
|
||||||
|
filt.SetImage(img)
|
||||||
|
filt.SetKernelNumericXZY([1.0] * 27)
|
||||||
|
filt.SetABTrim(2, 2) # trim 2 low and 2 high values
|
||||||
|
filt.Run()
|
||||||
|
```
|
||||||
|
|
||||||
|
### Bilateral Filter
|
||||||
|
|
||||||
|
Edge-preserving smoothing controlled by a spatial sigma (from the kernel shape) and an intensity sigma.
|
||||||
|
|
||||||
|
```python
|
||||||
|
filt = uLib.Math.VoxFilterAlgorithmBilateral(kernel_dims)
|
||||||
|
filt.SetImage(img)
|
||||||
|
filt.SetKernelNumericXZY([1.0] * 27)
|
||||||
|
filt.SetIntensitySigma(0.3)
|
||||||
|
filt.Run()
|
||||||
|
```
|
||||||
|
|
||||||
|
### Threshold Filter
|
||||||
|
|
||||||
|
Zeros voxels below a threshold.
|
||||||
|
|
||||||
|
```python
|
||||||
|
filt = uLib.Math.VoxFilterAlgorithmThreshold(kernel_dims)
|
||||||
|
filt.SetImage(img)
|
||||||
|
filt.SetKernelNumericXZY([1.0] * 27)
|
||||||
|
filt.SetThreshold(0.5)
|
||||||
|
filt.Run()
|
||||||
|
```
|
||||||
|
|
||||||
|
### Median Filter
|
||||||
|
|
||||||
|
```python
|
||||||
|
filt = uLib.Math.VoxFilterAlgorithmMedian(kernel_dims)
|
||||||
|
filt.SetImage(img)
|
||||||
|
filt.SetKernelNumericXZY([1.0] * 27)
|
||||||
|
filt.Run()
|
||||||
|
```
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## uLib.Math – Accumulators
|
||||||
|
|
||||||
|
Accumulators collect scalar samples and return a summary statistic.
|
||||||
|
|
||||||
|
```python
|
||||||
|
# Arithmetic mean
|
||||||
|
acc = uLib.Math.Accumulator_Mean_f()
|
||||||
|
acc(10.0)
|
||||||
|
acc(20.0)
|
||||||
|
mean = acc() # 15.0
|
||||||
|
|
||||||
|
# Alpha-beta trimmed mean
|
||||||
|
acc2 = uLib.Math.Accumulator_ABTrim_f()
|
||||||
|
acc2.SetABTrim(0.1, 0.1) # trim bottom 10 % and top 10 %
|
||||||
|
acc2 += 1.0
|
||||||
|
acc2 += 9999.0 # outlier
|
||||||
|
acc2 += 5.0
|
||||||
|
result = acc2() # trimmed mean ≈ 3.0
|
||||||
|
```
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## Dynamic Vectors (`uLib.Math.Vector_*`)
|
||||||
|
|
||||||
|
Typed dynamic arrays backed by `uLib::Vector<T>` with optional CUDA memory management.
|
||||||
|
|
||||||
|
```python
|
||||||
|
# Integer vector
|
||||||
|
vi = uLib.Math.Vector_i()
|
||||||
|
vi.append(1); vi.append(2); vi.append(3)
|
||||||
|
print(len(vi), vi[0])
|
||||||
|
|
||||||
|
# Float vector with CUDA management
|
||||||
|
vf = uLib.Math.Vector_f()
|
||||||
|
vf.append(1.5)
|
||||||
|
vf.MoveToVRAM() # copy to GPU (no-op when CUDA is absent)
|
||||||
|
vf.MoveToRAM() # copy back to CPU
|
||||||
|
|
||||||
|
# Other types: Vector_ui, Vector_l, Vector_ul, Vector_d
|
||||||
|
# Compound element types: Vector_Vector3f, Vector_Vector4f, Vector_Voxel …
|
||||||
|
```
|
||||||
60
docs/usage/usage.md
Normal file
60
docs/usage/usage.md
Normal file
@@ -0,0 +1,60 @@
|
|||||||
|
# Usage and Installation Guide
|
||||||
|
|
||||||
|
## Requirements
|
||||||
|
|
||||||
|
### Compiling with CUDA Support
|
||||||
|
|
||||||
|
The library supports running VoxImage filtering operations directly on CUDA cores via transparent RAM/VRAM memory transfers.
|
||||||
|
|
||||||
|
By default, the `CMakeLists.txt` build system sets `USE_CUDA=ON` and will attempt to locate `nvcc` and the NVIDIA CUDA Toolkit. If the toolkit is missing, `CMake` will fail unless you explicitly configure the project with `-DUSE_CUDA=OFF`.
|
||||||
|
|
||||||
|
### 1. Installing CUDA Environment via Micromamba
|
||||||
|
|
||||||
|
If you are developing inside an isolated Conda/Micromamba environment (e.g., `mutom`), you can inject the CUDA compilers directly into your environment rather than relying on global system dependencies:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
# Add the conda-forge channel if not already available
|
||||||
|
micromamba config append channels conda-forge
|
||||||
|
|
||||||
|
# Install nvcc and the necessary CUDA toolkit components
|
||||||
|
micromamba install cuda-nvcc
|
||||||
|
```
|
||||||
|
|
||||||
|
Verify your installation:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
nvcc --version
|
||||||
|
```
|
||||||
|
|
||||||
|
### 2. Building the Project
|
||||||
|
|
||||||
|
Configure and compile the project using standard CMake flows:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
mkdir -p build && cd build
|
||||||
|
|
||||||
|
# Configure CMake
|
||||||
|
# (Optional) Explicitly toggle CUDA: cmake -DUSE_CUDA=ON ..
|
||||||
|
cmake ..
|
||||||
|
|
||||||
|
# Compile the project and tests
|
||||||
|
make -j $(nproc)
|
||||||
|
```
|
||||||
|
|
||||||
|
### 3. Validating CUDA Support
|
||||||
|
|
||||||
|
You can verify that the CUDA kernels are launching correctly and allocating device memory through `DataAllocator` by running the mathematical unit tests.
|
||||||
|
|
||||||
|
```bash
|
||||||
|
# From the build directory
|
||||||
|
./src/Math/testing/VoxImageFilterTest
|
||||||
|
|
||||||
|
# Output should show:
|
||||||
|
# "Data correctly stayed in VRAM after CUDA execution!"
|
||||||
|
```
|
||||||
|
|
||||||
|
## How It Works Under The Hood
|
||||||
|
|
||||||
|
The `DataAllocator<T>` container automatically wraps memory allocations to transparently map to CPU RAM, or GPU VRAM. Standard iteration automatically pulls data backwards using implicit `MoveToRAM()` calls.
|
||||||
|
|
||||||
|
Filters using `#ifdef USE_CUDA` explicitly dictate `<buffer>.MoveToVRAM()` allocating directly on device bounds seamlessly. Fallbacks to Host compute iterations handle themselves automatically. Chaining specific filters together safely chains continuous VRAM operations avoiding costly Host copies in between iterations.
|
||||||
561
mkdocs.yml
Normal file
561
mkdocs.yml
Normal file
@@ -0,0 +1,561 @@
|
|||||||
|
# site_name: uLib Documentation
|
||||||
|
# site_description: CMT Cosmic Muon Tomography – uLib toolkit
|
||||||
|
# site_author: Andrea Rigoni Garola
|
||||||
|
# repo_url: https://github.com/cmt/ulib
|
||||||
|
# docs_dir: docs
|
||||||
|
|
||||||
|
# +--------------------------------------------------------------------------------------------------------+
|
||||||
|
# | |
|
||||||
|
# | This is the main file used by MkDocs to build the pages. |
|
||||||
|
# | It contains a lot of information and settings for you to read and use. |
|
||||||
|
# | Comments may contain "Read More" URLs to more in-depth documentation about the option for you to read. |
|
||||||
|
# | |
|
||||||
|
# | You can check out https://www.mkdocs.org/user-guide/configuration/ for a more detailed explanation of |
|
||||||
|
# | all the options MkDocs offers by default. |
|
||||||
|
# | |
|
||||||
|
# +------------------------------------------------- NOTE -------------------------------------------------+
|
||||||
|
# | |
|
||||||
|
# | Some of the options listed here are only available through the usage of Material for MkDocs. |
|
||||||
|
# | Those options will usually have a link to the docs of this Theme and also mention "Material" as name. |
|
||||||
|
# | The actual name of the theme is "Material for MkDocs" and "Material" is used for simplicity reasons. |
|
||||||
|
# | |
|
||||||
|
# +--------------------------------------------------------------------------------------------------------+
|
||||||
|
|
||||||
|
# +--------------------------------------------------------------------------------------------------------+
|
||||||
|
# | |
|
||||||
|
# | Main Page Settings for MkDocs. |
|
||||||
|
# | Those settings are site name, site description, Site author and also Site URL (Canonical URL) |
|
||||||
|
# | |
|
||||||
|
# | Read More: |
|
||||||
|
# | - https://www.mkdocs.org/user-guide/configuration/#site_name |
|
||||||
|
# | - https://www.mkdocs.org/user-guide/configuration/#site_description |
|
||||||
|
# | - https://www.mkdocs.org/user-guide/configuration/#site_author |
|
||||||
|
# | - https://www.mkdocs.org/user-guide/configuration/#site_url |
|
||||||
|
# | |
|
||||||
|
# +--------------------------------------------------------------------------------------------------------+
|
||||||
|
site_name: OpenCMT uLib Documentation
|
||||||
|
site_url: https://docs.mildstone.org/uLib/ # <--- project subfolder
|
||||||
|
use_directory_urls: true
|
||||||
|
site_description: 'Documentation for OpenCMT uLib'
|
||||||
|
site_author: 'Andrea Rigoni Garola'
|
||||||
|
|
||||||
|
# +--------------------------------------------------------------------------------------------------------+
|
||||||
|
# | |
|
||||||
|
# | This setting allows you to define your own Copyright notice. |
|
||||||
|
# | The text is treated as HTML code so you can use things like <a> tags or © to display the |
|
||||||
|
# | Copyright icon. |
|
||||||
|
# | |
|
||||||
|
# | Where or IF the Copyright is displayed depends on the theme you use. |
|
||||||
|
# | |
|
||||||
|
# | Read More: |
|
||||||
|
# | - https://www.mkdocs.org/user-guide/configuration/#copyright |
|
||||||
|
# | |
|
||||||
|
# +--------------------------------------------------------------------------------------------------------+
|
||||||
|
copyright: |
|
||||||
|
© Author
|
||||||
|
|
||||||
|
# +--------------------------------------------------------------------------------------------------------+
|
||||||
|
# | |
|
||||||
|
# | The base folder to use. |
|
||||||
|
# | Any Markdown files you put into this folder will be turned into a static HTML page once you build or |
|
||||||
|
# | publish your page. |
|
||||||
|
# | |
|
||||||
|
# | It is also used as the base directory for other settings like the "extra_css" or "extra_javascript" |
|
||||||
|
# | option. |
|
||||||
|
# | |
|
||||||
|
# +--------------------------------------------------------------------------------------------------------+
|
||||||
|
docs_dir: docs/
|
||||||
|
|
||||||
|
|
||||||
|
# +--------------------------------------------------------------------------------------------------------+
|
||||||
|
# | |
|
||||||
|
# | These options allow to define a Repository to link to. |
|
||||||
|
# | The result will, depending on the theme, be a link somewhere shown on the page that links to the |
|
||||||
|
# | Repository with the specified repo_name. |
|
||||||
|
# | |
|
||||||
|
# | This will also enable a "edit" button on the page itself that allows the direct editing of the page. |
|
||||||
|
# | You can disable this by setting "edit_uri" to an empty String. |
|
||||||
|
# | |
|
||||||
|
# | Read More: |
|
||||||
|
# | - https://www.mkdocs.org/user-guide/configuration/#repo_name |
|
||||||
|
# | - https://www.mkdocs.org/user-guide/configuration/#repo_url |
|
||||||
|
# | - https://www.mkdocs.org/user-guide/configuration/#edit_uri |
|
||||||
|
# | |
|
||||||
|
# +--------------------------------------------------------------------------------------------------------+
|
||||||
|
repo_name: OpenCMT/uLib
|
||||||
|
repo_url: https://gitea.mildstone.org/OpenCMT/uLib.git
|
||||||
|
#edit_uri: tree/master/docs # Uncomment to define a different URI/URL for the "edit" option
|
||||||
|
|
||||||
|
# +--------------------------------------------------------------------------------------------------------+
|
||||||
|
# | |
|
||||||
|
# | The "nav" option is where you define the navigation to show in MkDocs. |
|
||||||
|
# | |
|
||||||
|
# | Depending on the theme you use will the resulting Navigation look different. |
|
||||||
|
# | |
|
||||||
|
# | You can set different types of navigations. Either just the path, the path with a separate title or |
|
||||||
|
# | an external URL. |
|
||||||
|
# | |
|
||||||
|
# | Read More: |
|
||||||
|
# | - https://www.mkdocs.org/user-guide/configuration/#documentation-layout |
|
||||||
|
# | |
|
||||||
|
# +--------------------------------------------------------------------------------------------------------+
|
||||||
|
|
||||||
|
nav:
|
||||||
|
- Home: index.md
|
||||||
|
- Python:
|
||||||
|
- Installation: python/installation.md
|
||||||
|
- API Usage: python/usage.md
|
||||||
|
- Developer Guide: python/developer_guide.md
|
||||||
|
- C++ Build:
|
||||||
|
- Usage & CUDA: usage/usage.md
|
||||||
|
|
||||||
|
# +--------------------------------------------------------------------------------------------------------+
|
||||||
|
# | |
|
||||||
|
# | The "theme" section allows you to define what theme to use. |
|
||||||
|
# | It is also used for theme-specific options, but also for advanced stuff such as theme-extensions, if |
|
||||||
|
# | the theme actually supports it. |
|
||||||
|
# | |
|
||||||
|
# | Read More: |
|
||||||
|
# | - https://www.mkdocs.org/user-guide/configuration/#theme |
|
||||||
|
# | |
|
||||||
|
# +--------------------------------------------------------------------------------------------------------+
|
||||||
|
theme:
|
||||||
|
|
||||||
|
# +------------------------------------------------------------------------------------------------------+
|
||||||
|
# | |
|
||||||
|
# | The "name" option is where you define the theme to use. |
|
||||||
|
# | |
|
||||||
|
# | Note that not all themes are included by default and will require you to install them first. |
|
||||||
|
# | The Material theme is one of them. See the "Read More" link for instructions on how to install it. |
|
||||||
|
# | |
|
||||||
|
# | Read More: |
|
||||||
|
# | - https://squidfunk.github.io/mkdocs-material/getting-started/ |
|
||||||
|
# | |
|
||||||
|
# +------------------------------------------------------------------------------------------------------+
|
||||||
|
name: 'material'
|
||||||
|
|
||||||
|
|
||||||
|
# +------------------------------------------------------------------------------------------------------+
|
||||||
|
# | |
|
||||||
|
# | The Material theme allows "theme-extsnions", meaning that you can override parts of it by either |
|
||||||
|
# | overriding a particular file, or only parts (blocks) of it. |
|
||||||
|
# | |
|
||||||
|
# | If you want to override parts of Material, uncomment the "custom_dir" option below and set the |
|
||||||
|
# | folder (relative to the mkdocs.yml file) where your theme extensions will be located at. |
|
||||||
|
# | |
|
||||||
|
# | Read More: |
|
||||||
|
# | - https://www.mkdocs.org/user-guide/configuration/#custom_dir |
|
||||||
|
# | - https://squidfunk.github.io/mkdocs-material/customization/#extending-the-theme |
|
||||||
|
# | |
|
||||||
|
# +------------------------------------------------------------------------------------------------------+
|
||||||
|
#custom_dir: 'theme'
|
||||||
|
|
||||||
|
# +------------------------------------------------------------------------------------------------------+
|
||||||
|
# | |
|
||||||
|
# | The "favicon" option allows you to set your own image/icon to use in the browser-tab. |
|
||||||
|
# | |
|
||||||
|
# | Pretty much all image types are supported, but it's recommended to use a PNG, SVG or ICO image for |
|
||||||
|
# | the favicon. |
|
||||||
|
# | |
|
||||||
|
# | The directory is relative to the "docs_dir". |
|
||||||
|
# | |
|
||||||
|
# | Example: Having a favicon.png in docs/assets/images will result in the "favicon" setting showing |
|
||||||
|
# | 'assets/images/favicon.png' |
|
||||||
|
# | |
|
||||||
|
# | Read More: |
|
||||||
|
# | - https://squidfunk.github.io/mkdocs-material/setup/changing-the-logo-and-icons/#favicon |
|
||||||
|
# | |
|
||||||
|
# +------------------------------------------------------------------------------------------------------+
|
||||||
|
#favicon: 'assets/images/favicon.png'
|
||||||
|
|
||||||
|
# +------------------------------------------------------------------------------------------------------+
|
||||||
|
# | |
|
||||||
|
# | The "palette" section is a Material option and allows you to define specific style options such as |
|
||||||
|
# | Color-Scheme, and primary and secondary Color. |
|
||||||
|
# | |
|
||||||
|
# | You can also define multiple palettes that can have different Color Schemses and primary and/or |
|
||||||
|
# | secondary Colors. |
|
||||||
|
# | |
|
||||||
|
# | Read More: |
|
||||||
|
# | - https://squidfunk.github.io/mkdocs-material/setup/changing-the-colors/ |
|
||||||
|
# | - https://squidfunk.github.io/mkdocs-material/setup/changing-the-colors/#color-palette-toggle |
|
||||||
|
# | |
|
||||||
|
# +------------------------------------------------------------------------------------------------------+
|
||||||
|
palette:
|
||||||
|
|
||||||
|
# Palette toggle for light mode
|
||||||
|
- media: "(prefers-color-scheme: light)"
|
||||||
|
scheme: default
|
||||||
|
primary: 'indigo'
|
||||||
|
accent: 'indigo'
|
||||||
|
toggle:
|
||||||
|
icon: material/brightness-7
|
||||||
|
name: Switch to dark mode
|
||||||
|
|
||||||
|
# Palette toggle for dark mode
|
||||||
|
- media: "(prefers-color-scheme: dark)"
|
||||||
|
scheme: slate
|
||||||
|
primary: 'indigo'
|
||||||
|
accent: 'indigo'
|
||||||
|
toggle:
|
||||||
|
icon: material/brightness-4
|
||||||
|
name: Switch to light mode
|
||||||
|
|
||||||
|
# +------------------------------------------------------------------------------------------------------+
|
||||||
|
# | |
|
||||||
|
# | With the "font" option can you set a different font to use. |
|
||||||
|
# | |
|
||||||
|
# | Material supports all Google fonts, but you can also define your own ones if you choose so. |
|
||||||
|
# | |
|
||||||
|
# | The "text" option is used for the regular font while "code" is used for code blocks, inline code and |
|
||||||
|
# | similar. |
|
||||||
|
# | |
|
||||||
|
# | Read More: |
|
||||||
|
# | - https://squidfunk.github.io/mkdocs-material/setup/changing-the-fonts/ |
|
||||||
|
# | |
|
||||||
|
# +------------------------------------------------------------------------------------------------------+
|
||||||
|
#font:
|
||||||
|
# text: 'Roboto'
|
||||||
|
# code: 'Roboto Mono'
|
||||||
|
|
||||||
|
# +------------------------------------------------------------------------------------------------------+
|
||||||
|
# | |
|
||||||
|
# | Material suppors more than 40 different languages which you can set using the "language" option |
|
||||||
|
# | below. |
|
||||||
|
# | |
|
||||||
|
# | The default language is "en" (English). |
|
||||||
|
# | |
|
||||||
|
# | You can also enable/set a "selector" to allow switching between languages. |
|
||||||
|
# | See the "alternate" option in the "extra" section below for more information on this topic. |
|
||||||
|
# | |
|
||||||
|
# | Read More: |
|
||||||
|
# | - https://squidfunk.github.io/mkdocs-material/setup/changing-the-language/ |
|
||||||
|
# | |
|
||||||
|
# +------------------------------------------------------------------------------------------------------+
|
||||||
|
#language: 'en'
|
||||||
|
|
||||||
|
# +------------------------------------------------------------------------------------------------------+
|
||||||
|
# | |
|
||||||
|
# | The "direction" option is commonly used together with the "language" option. |
|
||||||
|
# | |
|
||||||
|
# | It allows you to change the text direction from the default left-to-right (ltr) to right-to-left |
|
||||||
|
# | (rtl) which is used in certain languages. |
|
||||||
|
# | |
|
||||||
|
# | Read More: |
|
||||||
|
# | - https://squidfunk.github.io/mkdocs-material/setup/changing-the-language/#directionality |
|
||||||
|
# | |
|
||||||
|
# +------------------------------------------------------------------------------------------------------+
|
||||||
|
#direction: 'ltr'
|
||||||
|
|
||||||
|
# +------------------------------------------------------------------------------------------------------+
|
||||||
|
# | |
|
||||||
|
# | The "features" option allows you to enable specific features of Material, by adding them to the |
|
||||||
|
# | list. |
|
||||||
|
# | |
|
||||||
|
# | Features are in the format <category>.<name>. As an example, the feature to enable tabs is called |
|
||||||
|
# | navigation.tabs. |
|
||||||
|
# | |
|
||||||
|
# | The list below contains all known features of Material. |
|
||||||
|
# | |
|
||||||
|
# | Features marked with a * are currently Insiders-only. (Last update: 11th December 2021) |
|
||||||
|
# | https://squidfunk.github.io/mkdocs-material/insiders/ |
|
||||||
|
# | |
|
||||||
|
# | Read More: |
|
||||||
|
# | - https://squidfunk.github.io/mkdocs-material/setup/setting-up-navigation/ |
|
||||||
|
# | |
|
||||||
|
# +------------------------------------------------------------------------------------------------------+
|
||||||
|
features:
|
||||||
|
# Announce
|
||||||
|
#
|
||||||
|
#- announce.dismiss # Adds a "X" button to dismiss a news banner/mark it as read.*
|
||||||
|
|
||||||
|
# Header
|
||||||
|
#
|
||||||
|
#- header.autohide # Hide header when user scrolls past a specific point.
|
||||||
|
|
||||||
|
# Navigation:
|
||||||
|
#
|
||||||
|
#- navigation.expand # Expand all collapsable sections.
|
||||||
|
#- navigation.instant # Instant loading pages.
|
||||||
|
#- navigation.indexes # Attach pages directly to Sections. Incompatible with "toc.integrate"
|
||||||
|
#- navigation.sections # Render top sections as groups.
|
||||||
|
- navigation.tabs # Render top sections as tabs at the top.
|
||||||
|
#- navigation.tabs.sticky # Tabs won't disappear when scrolling down. Requires "navigation.tabs".
|
||||||
|
#- navigation.top # Adds a "Back to top" that is shown when scrolling up.
|
||||||
|
#- navigation.tracking # Updates the url with highlighted section anchor.
|
||||||
|
|
||||||
|
# Search
|
||||||
|
#
|
||||||
|
#- search.highlight # Search will highlight the searched word(s) on the page.*
|
||||||
|
#- search.share # Adds an option to share a search query link.*
|
||||||
|
#- search.suggest # Search will suggest the likeliest completion for a word.*
|
||||||
|
|
||||||
|
# Table of Contents
|
||||||
|
#
|
||||||
|
#- toc.integrate # Include the TOC sections in the left navugation.
|
||||||
|
|
||||||
|
# +------------------------------------------------------------------------------------------------------+
|
||||||
|
# | |
|
||||||
|
# | The "icon" section allows you to define a icon to use for the logo and/or repository. |
|
||||||
|
# | |
|
||||||
|
# | To use already available icons will you need to set the right path for it, depending on which you |
|
||||||
|
# | want to use. |
|
||||||
|
# | |
|
||||||
|
# | Available icons: |
|
||||||
|
# | - FontAwesome |
|
||||||
|
# | - Brands: fontawesome/brands/... (https://fontawesome.com/icons?d=gallery&p=2&s=brands&m=free) |
|
||||||
|
# | - Regular: fontawesome/regular/... (https://fontawesome.com/icons?d=gallery&p=2&s=regular&m=free) |
|
||||||
|
# | - Solid: fontawesome/solid/... (https://fontawesome.com/icons?d=gallery&p=2&s=solid&m=free) |
|
||||||
|
# | |
|
||||||
|
# | - Material Design Icons: material/... (https://materialdesignicons.com/) |
|
||||||
|
# | |
|
||||||
|
# | - Octicons: octicons/... (https://primer.style/octicons/) |
|
||||||
|
# | |
|
||||||
|
# | You can also define your own Image for the logo. To do that, remove the "logo" option from "icon" |
|
||||||
|
# | instead add a "logo" option on the same level as the "icon" one, where you then set the path |
|
||||||
|
# | (relative to the "docs_dir") to the icon to use. Supported are all images types, including SVG. |
|
||||||
|
# | |
|
||||||
|
# | Read More: |
|
||||||
|
# | - https://squidfunk.github.io/mkdocs-material/setup/changing-the-logo-and-icons/#logo |
|
||||||
|
# | |
|
||||||
|
# +------------------------------------------------------------------------------------------------------+
|
||||||
|
icon:
|
||||||
|
logo: 'material/library'
|
||||||
|
repo: 'material/library'
|
||||||
|
|
||||||
|
# +------------------------------------------------------------------------------------------------------+
|
||||||
|
# | |
|
||||||
|
# | The "admonition" option allows you to set a different icon for each admonition type. |
|
||||||
|
# | |
|
||||||
|
# | This is currently a Insiders-only feature. (Last update: 7th October 2021) |
|
||||||
|
# | https://squidfunk.github.io/mkdocs-material/insiders/ |
|
||||||
|
# | |
|
||||||
|
# | Supported are all bundled icons: |
|
||||||
|
# | - FontAwesome |
|
||||||
|
# | - Brands: fontawesome/brands/... (https://fontawesome.com/icons?d=gallery&p=2&s=brands&m=free) |
|
||||||
|
# | - Regular: fontawesome/regular/... (https://fontawesome.com/icons?d=gallery&p=2&s=regular&m=free) |
|
||||||
|
# | - Solid: fontawesome/solid/... (https://fontawesome.com/icons?d=gallery&p=2&s=solid&m=free) |
|
||||||
|
# | |
|
||||||
|
# | - Material Design Icons: material/... (https://materialdesignicons.com/) |
|
||||||
|
# | |
|
||||||
|
# | - Octicons: octicons/... (https://primer.style/octicons/) |
|
||||||
|
# | |
|
||||||
|
# | You can also create and use your own icons. See the documentation for more information. |
|
||||||
|
# | |
|
||||||
|
# | Read More: |
|
||||||
|
# | - https://squidfunk.github.io/mkdocs-material/reference/admonitions/#changing-the-icons |
|
||||||
|
# | |
|
||||||
|
# +------------------------------------------------------------------------------------------------------+
|
||||||
|
#admonition:
|
||||||
|
# note: 'octicons/tag-16'
|
||||||
|
# abstract: 'octicons/checklist-16'
|
||||||
|
# info: 'octicons/info-16'
|
||||||
|
# tip: 'octicons/squirrel-16'
|
||||||
|
# success: 'octicons/check-16'
|
||||||
|
# question: 'octicons/question-16'
|
||||||
|
# warning: 'octicons/alert-16'
|
||||||
|
# failure: 'octicons/x-circle-16'
|
||||||
|
# danger: 'octicons/zap-16'
|
||||||
|
# bug: 'octicons/bug-16'
|
||||||
|
# example: 'octicons/beaker-16'
|
||||||
|
# quote: 'octicons/quote-16'
|
||||||
|
|
||||||
|
# +--------------------------------------------------------------------------------------------------------+
|
||||||
|
# | |
|
||||||
|
# | With the "extra_css" option can you add your own (S)CSS files to enhance the documentation. |
|
||||||
|
# | |
|
||||||
|
# | The path to the file is relative to the "docs_dir". |
|
||||||
|
# | |
|
||||||
|
# | Read More: |
|
||||||
|
# | - https://www.mkdocs.org/user-guide/configuration/#extra_css |
|
||||||
|
# | |
|
||||||
|
# +--------------------------------------------------------------------------------------------------------+
|
||||||
|
extra_css:
|
||||||
|
- assets/css/extra.css
|
||||||
|
|
||||||
|
# +--------------------------------------------------------------------------------------------------------+
|
||||||
|
# | |
|
||||||
|
# | Similar to the "extra_css" option does the "extra_javascript" option allow you to set custom JS files |
|
||||||
|
# | to add extra featurues. |
|
||||||
|
# | |
|
||||||
|
# | The path to the file is relative to the "docs_dir". |
|
||||||
|
# | |
|
||||||
|
# | Read More: |
|
||||||
|
# | - https://www.mkdocs.org/user-guide/configuration/#extra_javascript |
|
||||||
|
# | |
|
||||||
|
# +--------------------------------------------------------------------------------------------------------+
|
||||||
|
extra_javascript:
|
||||||
|
- https://polyfill.io/v3/polyfill.min.js?features=es6
|
||||||
|
- https://cdn.jsdelivr.net/npm/mathjax@3/es5/tex-mml-chtml.js
|
||||||
|
|
||||||
|
# +--------------------------------------------------------------------------------------------------------+
|
||||||
|
# | |
|
||||||
|
# | The "extra" section contains pretty much anything you want, as long as it is a valid key-value pair. |
|
||||||
|
# | |
|
||||||
|
# | Material uses this section for different custom settings that wouldn't fit in the theme section. |
|
||||||
|
# | |
|
||||||
|
# | Read More: |
|
||||||
|
# | - https://www.mkdocs.org/user-guide/configuration/#extra |
|
||||||
|
# | |
|
||||||
|
# +--------------------------------------------------------------------------------------------------------+
|
||||||
|
extra:
|
||||||
|
# +------------------------------------------------------------------------------------------------------+
|
||||||
|
# | |
|
||||||
|
# | The social section allows you to set a list of entries which would be displayed in the footer of the |
|
||||||
|
# | page. |
|
||||||
|
# | |
|
||||||
|
# | Each entry has the exact same options: |
|
||||||
|
# | - icon: Path to the SVG icon to use. See "icon" section for available icon sets. |
|
||||||
|
# | - link: URL to which the icon should link. |
|
||||||
|
# | - name: Optional Name that would be displayed as title on hover. |
|
||||||
|
# | |
|
||||||
|
# | Read More: |
|
||||||
|
# | - https://squidfunk.github.io/mkdocs-material/setup/setting-up-the-footer/#social-links |
|
||||||
|
# | |
|
||||||
|
# +------------------------------------------------------------------------------------------------------+
|
||||||
|
social:
|
||||||
|
- icon: 'fontawesome/brands/github'
|
||||||
|
link: 'https://github.com/Andre601/mkdocs-template'
|
||||||
|
|
||||||
|
# +------------------------------------------------------------------------------------------------------+
|
||||||
|
# | |
|
||||||
|
# | Allows you to hide the "Made with Material for MkDocs" text in the footer of the pages by setting |
|
||||||
|
# | this to "true". |
|
||||||
|
# | |
|
||||||
|
# | Read More: |
|
||||||
|
# | - https://squidfunk.github.io/mkdocs-material/setup/setting-up-the-footer/#generator-notice |
|
||||||
|
# | |
|
||||||
|
# +------------------------------------------------------------------------------------------------------+
|
||||||
|
#generator: true
|
||||||
|
|
||||||
|
# +------------------------------------------------------------------------------------------------------+
|
||||||
|
# | |
|
||||||
|
# | The "manifest" option allows you to define a .manifest file to use. |
|
||||||
|
# | |
|
||||||
|
# | A .manifest file makes the doc act like a web-application and tells it how to behave when installed. |
|
||||||
|
# | |
|
||||||
|
# | Read More: |
|
||||||
|
# | - https://squidfunk.github.io/mkdocs-material/reference/meta-tags/#adding-a-web-app-manifest |
|
||||||
|
# | |
|
||||||
|
# +------------------------------------------------------------------------------------------------------+
|
||||||
|
#manifest: manifest.webmanifest
|
||||||
|
|
||||||
|
# +------------------------------------------------------------------------------------------------------+
|
||||||
|
# | |
|
||||||
|
# | The "alternate" option can be used to create a selector to switch languages. |
|
||||||
|
# | |
|
||||||
|
# | Using this requires you to create a specific, more complicated MkDocs setup. |
|
||||||
|
# | |
|
||||||
|
# | A Setup Guide for multi-language docs can be found here: |
|
||||||
|
# | https://github.com/squidfunk/mkdocs-material/discussions/2346 |
|
||||||
|
# | |
|
||||||
|
# | Read More: |
|
||||||
|
# | - https://squidfunk.github.io/mkdocs-material/setup/changing-the-language/#site-language-selector |
|
||||||
|
# | |
|
||||||
|
# +------------------------------------------------------------------------------------------------------+
|
||||||
|
#alternate:
|
||||||
|
|
||||||
|
# +--------------------------------------------------------------------------------------------------------+
|
||||||
|
# | |
|
||||||
|
# | MkDocs allows the usage of Markdown extensions which can do various things. |
|
||||||
|
# | |
|
||||||
|
# | Material includes the pymdownx extension which provides a lot of useful features to use. |
|
||||||
|
# | |
|
||||||
|
# | Note that some extensions may use specific settings that you need to set. |
|
||||||
|
# | Please check out the official documentation of PyMdownx for more information: |
|
||||||
|
# | https://facelessuser.github.io/pymdown-extensions/ |
|
||||||
|
# | |
|
||||||
|
# | Material already provides required CSS and JS values for the PyMdownX Extensions, which means you do |
|
||||||
|
# | not need to set them up yourself. |
|
||||||
|
# | |
|
||||||
|
# | Read More: |
|
||||||
|
# | - https://www.mkdocs.org/user-guide/configuration/#markdown_extensions |
|
||||||
|
# | |
|
||||||
|
# +--------------------------------------------------------------------------------------------------------+
|
||||||
|
markdown_extensions:
|
||||||
|
- markdown.extensions.admonition:
|
||||||
|
- markdown.extensions.codehilite:
|
||||||
|
guess_lang: false
|
||||||
|
- markdown.extensions.toc:
|
||||||
|
permalink: true
|
||||||
|
- pymdownx.arithmatex:
|
||||||
|
generic: true
|
||||||
|
- attr_list
|
||||||
|
- md_in_html
|
||||||
|
- pymdownx.blocks.caption
|
||||||
|
- admonition
|
||||||
|
- pymdownx.highlight:
|
||||||
|
anchor_linenums: true
|
||||||
|
- pymdownx.superfences
|
||||||
|
- pymdownx.tabbed:
|
||||||
|
alternate_style: true
|
||||||
|
- pymdownx.details
|
||||||
|
- attr_list
|
||||||
|
- tables
|
||||||
|
|
||||||
|
#- pymdownx.b64:
|
||||||
|
#- pymdownx.betterem:
|
||||||
|
#- pymdownx.caret:
|
||||||
|
#- pymdownx.critic:
|
||||||
|
#- pymdownx.details:
|
||||||
|
#- pymdownx.emoji:
|
||||||
|
#- pymdownx.escapeall:
|
||||||
|
#- pymdownx.extra:
|
||||||
|
#- pymdownx.extrarawhtml:
|
||||||
|
#- pymdownx.highlight:
|
||||||
|
#- pymdownx.inlinehilite:
|
||||||
|
#- pymdownx.keys:
|
||||||
|
#- pymdownx.magiclink:
|
||||||
|
#- pymdownx.mark:
|
||||||
|
#- pymdownx.pathconverter:
|
||||||
|
#- pymdownx.progressbar:
|
||||||
|
#- pymdownx.smartsymbols:
|
||||||
|
#- pymdownx.snippets:
|
||||||
|
#- pymdownx.striphtml:
|
||||||
|
#- pymdownx.superfences:
|
||||||
|
#- pymdownx.tabbed:
|
||||||
|
#- pymdownx.tasklist:
|
||||||
|
#- pymdownx.tilde:
|
||||||
|
# - exporter:
|
||||||
|
# formats:
|
||||||
|
# pdf:
|
||||||
|
# enabled: !ENV [MKDOCS_EXPORTER_PDF, true]
|
||||||
|
# concurrency: 8
|
||||||
|
# stylesheets:
|
||||||
|
# - resources/stylesheets/pdf.scss
|
||||||
|
# covers:
|
||||||
|
# front: resources/templates/covers/front.html.j2
|
||||||
|
# back: resources/templates/covers/back.html.j2
|
||||||
|
# aggregator:
|
||||||
|
# enabled: true
|
||||||
|
# output: .well-known/site.pdf
|
||||||
|
# covers: all
|
||||||
|
|
||||||
|
# theme:
|
||||||
|
# name: material
|
||||||
|
# palette:
|
||||||
|
# - scheme: default
|
||||||
|
# primary: indigo
|
||||||
|
# accent: blue
|
||||||
|
# toggle:
|
||||||
|
# icon: material/brightness-7
|
||||||
|
# name: Switch to dark mode
|
||||||
|
# - scheme: slate
|
||||||
|
# primary: indigo
|
||||||
|
# accent: blue
|
||||||
|
# toggle:
|
||||||
|
# icon: material/brightness-4
|
||||||
|
# name: Switch to light mode
|
||||||
|
# features:
|
||||||
|
# - navigation.tabs
|
||||||
|
# - navigation.sections
|
||||||
|
# - navigation.top
|
||||||
|
# - content.code.copy
|
||||||
|
# - content.tabs.link
|
||||||
|
|
||||||
|
# plugins:
|
||||||
|
# - search
|
||||||
|
|
||||||
|
# markdown_extensions:
|
||||||
|
|
||||||
|
|
||||||
7
poetry.lock
generated
Normal file
7
poetry.lock
generated
Normal file
@@ -0,0 +1,7 @@
|
|||||||
|
# This file is automatically @generated by Poetry 2.3.1 and should not be changed by hand.
|
||||||
|
package = []
|
||||||
|
|
||||||
|
[metadata]
|
||||||
|
lock-version = "2.1"
|
||||||
|
python-versions = ">=3.9"
|
||||||
|
content-hash = "db9b4c08b159b17b239e26c67ead7c37b82d9f9eb06550245ae3134c095f98f7"
|
||||||
15
pyproject.toml
Normal file
15
pyproject.toml
Normal file
@@ -0,0 +1,15 @@
|
|||||||
|
[tool.poetry]
|
||||||
|
name = "uLib"
|
||||||
|
version = "0.6.0"
|
||||||
|
description = "CMT Cosmic Muon Tomography project uLib python bindings"
|
||||||
|
authors = ["Andrea Rigoni Garola <andrea.rigoni@pd.infn.it>"]
|
||||||
|
readme = "README.md"
|
||||||
|
packages = [{ include = "uLib", from = "src/Python" }]
|
||||||
|
build = "build_python.py"
|
||||||
|
|
||||||
|
[tool.poetry.dependencies]
|
||||||
|
python = ">=3.9"
|
||||||
|
|
||||||
|
[build-system]
|
||||||
|
requires = ["poetry-core>=2.0.0", "pybind11>=2.6.0", "cmake>=3.12"]
|
||||||
|
build-backend = "poetry.core.masonry.api"
|
||||||
File diff suppressed because it is too large
Load Diff
@@ -1,10 +1,38 @@
|
|||||||
|
|
||||||
set(HEADERS Options.h
|
set(HEADERS
|
||||||
StaticInterface.h)
|
Archives.h
|
||||||
|
Array.h
|
||||||
|
Collection.h
|
||||||
|
DataAllocator.h
|
||||||
|
Debug.h
|
||||||
|
Export.h
|
||||||
|
Function.h
|
||||||
|
Macros.h
|
||||||
|
Mpl.h
|
||||||
|
Object.h
|
||||||
|
Options.h
|
||||||
|
Serializable.h
|
||||||
|
Signal.h
|
||||||
|
Singleton.h
|
||||||
|
SmartPointer.h
|
||||||
|
StaticInterface.h
|
||||||
|
StringReader.h
|
||||||
|
Types.h
|
||||||
|
Uuid.h
|
||||||
|
Vector.h
|
||||||
|
)
|
||||||
|
|
||||||
set(SOURCES Options.cpp)
|
set(SOURCES
|
||||||
|
Archives.cpp
|
||||||
|
Debug.cpp
|
||||||
|
Object.cpp
|
||||||
|
Options.cpp
|
||||||
|
Serializable.cpp
|
||||||
|
Signal.cpp
|
||||||
|
Uuid.cpp
|
||||||
|
)
|
||||||
|
|
||||||
set(LIBRARIES ${Boost_PROGRAM_OPTIONS_LIBRARY})
|
set(LIBRARIES Boost::program_options Boost::serialization)
|
||||||
|
|
||||||
set(libname ${PACKAGE_LIBPREFIX}Core)
|
set(libname ${PACKAGE_LIBPREFIX}Core)
|
||||||
set(ULIB_SHARED_LIBRARIES ${ULIB_SHARED_LIBRARIES} ${libname} PARENT_SCOPE)
|
set(ULIB_SHARED_LIBRARIES ${ULIB_SHARED_LIBRARIES} ${libname} PARENT_SCOPE)
|
||||||
@@ -14,6 +42,10 @@ add_library(${libname} SHARED ${SOURCES})
|
|||||||
set_target_properties(${libname} PROPERTIES
|
set_target_properties(${libname} PROPERTIES
|
||||||
VERSION ${PROJECT_VERSION}
|
VERSION ${PROJECT_VERSION}
|
||||||
SOVERSION ${PROJECT_SOVERSION})
|
SOVERSION ${PROJECT_SOVERSION})
|
||||||
|
if(USE_CUDA)
|
||||||
|
set(LIBRARIES ${LIBRARIES} CUDA::cudart)
|
||||||
|
endif()
|
||||||
|
|
||||||
target_link_libraries(${libname} ${LIBRARIES})
|
target_link_libraries(${libname} ${LIBRARIES})
|
||||||
|
|
||||||
install(TARGETS ${libname}
|
install(TARGETS ${libname}
|
||||||
@@ -23,5 +55,7 @@ install(TARGETS ${libname}
|
|||||||
|
|
||||||
install(FILES ${HEADERS} DESTINATION ${INSTALL_INC_DIR}/Core)
|
install(FILES ${HEADERS} DESTINATION ${INSTALL_INC_DIR}/Core)
|
||||||
|
|
||||||
|
if(BUILD_TESTING)
|
||||||
|
include(uLibTargetMacros)
|
||||||
|
add_subdirectory(testing)
|
||||||
|
endif()
|
||||||
|
|||||||
260
src/Core/DataAllocator.h
Normal file
260
src/Core/DataAllocator.h
Normal file
@@ -0,0 +1,260 @@
|
|||||||
|
/*//////////////////////////////////////////////////////////////////////////////
|
||||||
|
// CMT Cosmic Muon Tomography project //////////////////////////////////////////
|
||||||
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
|
|
||||||
|
Copyright (c) 2014, Universita' degli Studi di Padova, INFN sez. di Padova
|
||||||
|
All rights reserved
|
||||||
|
|
||||||
|
Authors: Andrea Rigoni Garola < andrea.rigoni@pd.infn.it >
|
||||||
|
|
||||||
|
------------------------------------------------------------------
|
||||||
|
This library is free software; you can redistribute it and/or
|
||||||
|
modify it under the terms of the GNU Lesser General Public
|
||||||
|
License as published by the Free Software Foundation; either
|
||||||
|
version 3.0 of the License, or (at your option) any later version.
|
||||||
|
|
||||||
|
This library is distributed in the hope that it will be useful,
|
||||||
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
||||||
|
Lesser General Public License for more details.
|
||||||
|
|
||||||
|
You should have received a copy of the GNU Lesser General Public
|
||||||
|
License along with this library.
|
||||||
|
|
||||||
|
//////////////////////////////////////////////////////////////////////////////*/
|
||||||
|
|
||||||
|
#ifndef U_MATH_DATAALLOCATOR_H
|
||||||
|
#define U_MATH_DATAALLOCATOR_H
|
||||||
|
|
||||||
|
#include <algorithm>
|
||||||
|
#include <cstring>
|
||||||
|
#include <iostream>
|
||||||
|
#include <stdexcept>
|
||||||
|
#include <vector>
|
||||||
|
|
||||||
|
#ifdef USE_CUDA
|
||||||
|
#include <cuda_runtime.h>
|
||||||
|
#include <thrust/device_vector.h>
|
||||||
|
#endif
|
||||||
|
|
||||||
|
namespace uLib {
|
||||||
|
|
||||||
|
enum class MemoryDevice { RAM, VRAM };
|
||||||
|
|
||||||
|
template <typename T> class DataAllocator {
|
||||||
|
public:
|
||||||
|
DataAllocator(size_t size = 0, bool owns_objects = true)
|
||||||
|
: m_Size(size), m_RamData(nullptr), m_VramData(nullptr),
|
||||||
|
m_Device(MemoryDevice::RAM), m_OwnsObjects(owns_objects) {
|
||||||
|
if (m_Size > 0) {
|
||||||
|
if (m_OwnsObjects)
|
||||||
|
m_RamData = new T[m_Size]();
|
||||||
|
else
|
||||||
|
m_RamData = static_cast<T *>(::operator new(m_Size * sizeof(T)));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
DataAllocator(const DataAllocator<T> &other)
|
||||||
|
: m_Size(other.m_Size), m_RamData(nullptr), m_VramData(nullptr),
|
||||||
|
m_Device(other.m_Device), m_OwnsObjects(other.m_OwnsObjects) {
|
||||||
|
if (m_Size > 0) {
|
||||||
|
if (other.m_RamData) {
|
||||||
|
if (m_OwnsObjects)
|
||||||
|
m_RamData = new T[m_Size];
|
||||||
|
else
|
||||||
|
m_RamData = static_cast<T *>(::operator new(m_Size * sizeof(T)));
|
||||||
|
std::memcpy(m_RamData, other.m_RamData, m_Size * sizeof(T));
|
||||||
|
}
|
||||||
|
#ifdef USE_CUDA
|
||||||
|
if (other.m_VramData) {
|
||||||
|
cudaMalloc((void **)&m_VramData, m_Size * sizeof(T));
|
||||||
|
cudaMemcpy(m_VramData, other.m_VramData, m_Size * sizeof(T),
|
||||||
|
cudaMemcpyDeviceToDevice);
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
~DataAllocator() {
|
||||||
|
if (m_RamData) {
|
||||||
|
if (m_OwnsObjects)
|
||||||
|
delete[] m_RamData;
|
||||||
|
else
|
||||||
|
::operator delete(m_RamData);
|
||||||
|
}
|
||||||
|
#ifdef USE_CUDA
|
||||||
|
if (m_VramData) {
|
||||||
|
cudaFree(m_VramData);
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
DataAllocator &operator=(const DataAllocator &other) {
|
||||||
|
if (this != &other) {
|
||||||
|
m_OwnsObjects = other.m_OwnsObjects;
|
||||||
|
resize(other.m_Size);
|
||||||
|
m_Device = other.m_Device;
|
||||||
|
if (other.m_RamData) {
|
||||||
|
if (!m_RamData) {
|
||||||
|
if (m_OwnsObjects)
|
||||||
|
m_RamData = new T[m_Size];
|
||||||
|
else
|
||||||
|
m_RamData = static_cast<T *>(::operator new(m_Size * sizeof(T)));
|
||||||
|
}
|
||||||
|
std::memcpy(m_RamData, other.m_RamData, m_Size * sizeof(T));
|
||||||
|
}
|
||||||
|
#ifdef USE_CUDA
|
||||||
|
if (other.m_VramData) {
|
||||||
|
if (!m_VramData)
|
||||||
|
cudaMalloc((void **)&m_VramData, m_Size * sizeof(T));
|
||||||
|
cudaMemcpy(m_VramData, other.m_VramData, m_Size * sizeof(T),
|
||||||
|
cudaMemcpyDeviceToDevice);
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
return *this;
|
||||||
|
}
|
||||||
|
|
||||||
|
void MoveToRAM() {
|
||||||
|
if (m_Device == MemoryDevice::RAM)
|
||||||
|
return;
|
||||||
|
if (!m_RamData && m_Size > 0) {
|
||||||
|
if (m_OwnsObjects)
|
||||||
|
m_RamData = new T[m_Size]();
|
||||||
|
else
|
||||||
|
m_RamData = static_cast<T *>(::operator new(m_Size * sizeof(T)));
|
||||||
|
}
|
||||||
|
#ifdef USE_CUDA
|
||||||
|
if (m_VramData && m_Size > 0) {
|
||||||
|
cudaMemcpy(m_RamData, m_VramData, m_Size * sizeof(T),
|
||||||
|
cudaMemcpyDeviceToHost);
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
m_Device = MemoryDevice::RAM;
|
||||||
|
}
|
||||||
|
|
||||||
|
void MoveToVRAM() {
|
||||||
|
if (m_Device == MemoryDevice::VRAM)
|
||||||
|
return;
|
||||||
|
#ifdef USE_CUDA
|
||||||
|
if (!m_VramData && m_Size > 0) {
|
||||||
|
cudaMalloc((void **)&m_VramData, m_Size * sizeof(T));
|
||||||
|
}
|
||||||
|
if (m_RamData && m_Size > 0) {
|
||||||
|
cudaMemcpy(m_VramData, m_RamData, m_Size * sizeof(T),
|
||||||
|
cudaMemcpyHostToDevice);
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
m_Device = MemoryDevice::VRAM;
|
||||||
|
}
|
||||||
|
|
||||||
|
void resize(size_t size) {
|
||||||
|
if (m_Size == size)
|
||||||
|
return;
|
||||||
|
|
||||||
|
T *newRam = nullptr;
|
||||||
|
T *newVram = nullptr;
|
||||||
|
|
||||||
|
if (size > 0) {
|
||||||
|
if (m_OwnsObjects)
|
||||||
|
newRam = new T[size]();
|
||||||
|
else
|
||||||
|
newRam = static_cast<T *>(::operator new(size * sizeof(T)));
|
||||||
|
|
||||||
|
if (m_RamData) {
|
||||||
|
std::memcpy(newRam, m_RamData, std::min(m_Size, size) * sizeof(T));
|
||||||
|
}
|
||||||
|
|
||||||
|
#ifdef USE_CUDA
|
||||||
|
cudaMalloc((void **)&newVram, size * sizeof(T));
|
||||||
|
if (m_VramData) {
|
||||||
|
cudaMemcpy(newVram, m_VramData, std::min(m_Size, size) * sizeof(T),
|
||||||
|
cudaMemcpyDeviceToDevice);
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
if (m_RamData) {
|
||||||
|
if (m_OwnsObjects)
|
||||||
|
delete[] m_RamData;
|
||||||
|
else
|
||||||
|
::operator delete(m_RamData);
|
||||||
|
}
|
||||||
|
#ifdef USE_CUDA
|
||||||
|
if (m_VramData)
|
||||||
|
cudaFree(m_VramData);
|
||||||
|
#endif
|
||||||
|
|
||||||
|
m_Size = size;
|
||||||
|
m_RamData = newRam;
|
||||||
|
m_VramData = newVram;
|
||||||
|
}
|
||||||
|
|
||||||
|
size_t size() const { return m_Size; }
|
||||||
|
|
||||||
|
T &at(size_t index) {
|
||||||
|
MoveToRAM();
|
||||||
|
if (index >= m_Size)
|
||||||
|
throw std::out_of_range("Index out of range");
|
||||||
|
return m_RamData[index];
|
||||||
|
}
|
||||||
|
|
||||||
|
const T &at(size_t index) const {
|
||||||
|
const_cast<DataAllocator *>(this)->MoveToRAM();
|
||||||
|
if (index >= m_Size)
|
||||||
|
throw std::out_of_range("Index out of range");
|
||||||
|
return m_RamData[index];
|
||||||
|
}
|
||||||
|
|
||||||
|
T &operator[](size_t index) {
|
||||||
|
MoveToRAM();
|
||||||
|
return m_RamData[index];
|
||||||
|
}
|
||||||
|
|
||||||
|
const T &operator[](size_t index) const {
|
||||||
|
const_cast<DataAllocator *>(this)->MoveToRAM();
|
||||||
|
return m_RamData[index];
|
||||||
|
}
|
||||||
|
|
||||||
|
T *data() { return (m_Device == MemoryDevice::RAM) ? m_RamData : m_VramData; }
|
||||||
|
const T *data() const {
|
||||||
|
return (m_Device == MemoryDevice::RAM) ? m_RamData : m_VramData;
|
||||||
|
}
|
||||||
|
|
||||||
|
T *GetRAMData() { return m_RamData; }
|
||||||
|
const T *GetRAMData() const { return m_RamData; }
|
||||||
|
|
||||||
|
T *GetVRAMData() { return m_VramData; }
|
||||||
|
const T *GetVRAMData() const { return m_VramData; }
|
||||||
|
|
||||||
|
MemoryDevice GetDevice() const { return m_Device; }
|
||||||
|
|
||||||
|
// Iterator support for RAM operations
|
||||||
|
T *begin() {
|
||||||
|
MoveToRAM();
|
||||||
|
return m_RamData;
|
||||||
|
}
|
||||||
|
T *end() {
|
||||||
|
MoveToRAM();
|
||||||
|
return m_RamData + m_Size;
|
||||||
|
}
|
||||||
|
const T *begin() const {
|
||||||
|
const_cast<DataAllocator *>(this)->MoveToRAM();
|
||||||
|
return m_RamData;
|
||||||
|
}
|
||||||
|
const T *end() const {
|
||||||
|
const_cast<DataAllocator *>(this)->MoveToRAM();
|
||||||
|
return m_RamData + m_Size;
|
||||||
|
}
|
||||||
|
|
||||||
|
private:
|
||||||
|
size_t m_Size;
|
||||||
|
T *m_RamData;
|
||||||
|
T *m_VramData;
|
||||||
|
MemoryDevice m_Device;
|
||||||
|
bool m_OwnsObjects;
|
||||||
|
};
|
||||||
|
|
||||||
|
} // namespace uLib
|
||||||
|
|
||||||
|
#endif // U_MATH_DATAALLOCATOR_H
|
||||||
@@ -23,95 +23,85 @@
|
|||||||
|
|
||||||
//////////////////////////////////////////////////////////////////////////////*/
|
//////////////////////////////////////////////////////////////////////////////*/
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
#ifndef U_CORE_EXPORT_H
|
#ifndef U_CORE_EXPORT_H
|
||||||
#define U_CORE_EXPORT_H
|
#define U_CORE_EXPORT_H
|
||||||
|
|
||||||
#include <utility>
|
|
||||||
#include <cstddef> // NULL
|
#include <cstddef> // NULL
|
||||||
#include <iostream>
|
#include <iostream>
|
||||||
|
#include <utility>
|
||||||
|
|
||||||
#include <boost/config.hpp>
|
#include <boost/config.hpp>
|
||||||
#include <boost/static_assert.hpp>
|
|
||||||
#include <boost/preprocessor/stringize.hpp>
|
#include <boost/preprocessor/stringize.hpp>
|
||||||
|
#include <boost/static_assert.hpp>
|
||||||
#include <boost/type_traits/is_polymorphic.hpp>
|
#include <boost/type_traits/is_polymorphic.hpp>
|
||||||
|
|
||||||
#include <boost/mpl/assert.hpp>
|
|
||||||
#include <boost/mpl/and.hpp>
|
#include <boost/mpl/and.hpp>
|
||||||
#include <boost/mpl/not.hpp>
|
#include <boost/mpl/assert.hpp>
|
||||||
#include <boost/mpl/bool.hpp>
|
#include <boost/mpl/bool.hpp>
|
||||||
|
#include <boost/mpl/not.hpp>
|
||||||
|
|
||||||
#include <boost/serialization/extended_type_info.hpp> // for guid_defined only
|
|
||||||
#include <boost/serialization/static_warning.hpp>
|
|
||||||
#include <boost/serialization/assume_abstract.hpp>
|
#include <boost/serialization/assume_abstract.hpp>
|
||||||
|
#include <boost/serialization/extended_type_info.hpp> // for guid_defined only
|
||||||
#include <boost/serialization/force_include.hpp>
|
#include <boost/serialization/force_include.hpp>
|
||||||
#include <boost/serialization/singleton.hpp>
|
#include <boost/serialization/singleton.hpp>
|
||||||
|
#include <boost/serialization/static_warning.hpp>
|
||||||
|
|
||||||
#include <boost/archive/detail/register_archive.hpp>
|
#include <boost/archive/detail/register_archive.hpp>
|
||||||
|
|
||||||
|
|
||||||
#include "Core/Archives.h"
|
#include "Core/Archives.h"
|
||||||
|
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
namespace uLib {
|
namespace uLib {
|
||||||
namespace Archive {
|
namespace Archive {
|
||||||
namespace detail {
|
namespace detail {
|
||||||
namespace extra_detail {
|
namespace extra_detail {
|
||||||
|
|
||||||
template<class T>
|
template <class T> struct guid_initializer {
|
||||||
struct guid_initializer
|
void export_guid(boost::mpl::false_) const {
|
||||||
{
|
// generates the statically-initialized objects whose constructors
|
||||||
void export_guid(mpl::false_) const {
|
// register the information allowing serialization of T objects
|
||||||
// generates the statically-initialized objects whose constructors
|
// through pointers to their base classes.
|
||||||
// register the information allowing serialization of T objects
|
boost::archive::detail::instantiate_ptr_serialization(
|
||||||
// through pointers to their base classes.
|
(T *)0, 0, uLib::Archive::detail::adl_tag());
|
||||||
boost::archive::detail::
|
}
|
||||||
instantiate_ptr_serialization((T*)0, 0,
|
void export_guid(boost::mpl::true_) const {}
|
||||||
uLib::Archive::detail::adl_tag());
|
guid_initializer const &export_guid() const {
|
||||||
}
|
BOOST_STATIC_WARNING(boost::is_polymorphic<T>::value);
|
||||||
void export_guid(mpl::true_) const {
|
// note: exporting an abstract base class will have no effect
|
||||||
}
|
// and cannot be used to instantitiate serialization code
|
||||||
guid_initializer const & export_guid() const {
|
// (one might be using this in a DLL to instantiate code)
|
||||||
BOOST_STATIC_WARNING(boost::is_polymorphic< T >::value);
|
// BOOST_STATIC_WARNING(! boost::serialization::is_abstract< T >::value);
|
||||||
// note: exporting an abstract base class will have no effect
|
export_guid(boost::serialization::is_abstract<T>());
|
||||||
// and cannot be used to instantitiate serialization code
|
return *this;
|
||||||
// (one might be using this in a DLL to instantiate code)
|
}
|
||||||
//BOOST_STATIC_WARNING(! boost::serialization::is_abstract< T >::value);
|
|
||||||
export_guid(boost::serialization::is_abstract< T >());
|
|
||||||
return *this;
|
|
||||||
}
|
|
||||||
};
|
};
|
||||||
|
|
||||||
template<typename T>
|
template <typename T> struct init_guid;
|
||||||
struct init_guid;
|
|
||||||
|
|
||||||
} // anonymous
|
} // namespace extra_detail
|
||||||
} // namespace detail
|
} // namespace detail
|
||||||
} // namespace Archive
|
} // namespace Archive
|
||||||
} // namespace uLib
|
} // namespace uLib
|
||||||
|
|
||||||
|
#define ULIB_CLASS_EXPORT_IMPLEMENT(T) \
|
||||||
|
namespace uLib { \
|
||||||
#define ULIB_CLASS_EXPORT_IMPLEMENT(T) \
|
namespace Archive { \
|
||||||
namespace uLib { \
|
namespace detail { \
|
||||||
namespace Archive { \
|
namespace extra_detail { \
|
||||||
namespace detail { \
|
template <> struct init_guid<T> { \
|
||||||
namespace extra_detail { \
|
static guid_initializer<T> const &g; \
|
||||||
template<> \
|
}; \
|
||||||
struct init_guid< T > { \
|
guid_initializer<T> const &init_guid<T>::g = \
|
||||||
static guid_initializer< T > const & g; \
|
::boost::serialization::singleton< \
|
||||||
}; \
|
guid_initializer<T>>::get_mutable_instance() \
|
||||||
guid_initializer< T > const & init_guid< T >::g = \
|
.export_guid(); \
|
||||||
::boost::serialization::singleton< \
|
} \
|
||||||
guid_initializer< T > \
|
} \
|
||||||
>::get_mutable_instance().export_guid(); \
|
} \
|
||||||
}}}} \
|
} \
|
||||||
/**/
|
/**/
|
||||||
|
|
||||||
#endif // EXPORT_H
|
#endif // EXPORT_H
|
||||||
|
|||||||
@@ -110,11 +110,13 @@ bool Object::addSignalImpl(SignalBase *sig, GenericMFPtr fptr,
|
|||||||
const char *name) {
|
const char *name) {
|
||||||
ObjectPrivate::Signal s = {fptr, std::string(name), sig};
|
ObjectPrivate::Signal s = {fptr, std::string(name), sig};
|
||||||
d->sigv.push_back(s);
|
d->sigv.push_back(s);
|
||||||
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
bool Object::addSlotImpl(GenericMFPtr fptr, const char *name) {
|
bool Object::addSlotImpl(GenericMFPtr fptr, const char *name) {
|
||||||
ObjectPrivate::Slot s = {fptr, std::string(name)};
|
ObjectPrivate::Slot s = {fptr, std::string(name)};
|
||||||
d->slov.push_back(s);
|
d->slov.push_back(s);
|
||||||
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
SignalBase *Object::findSignalImpl(const GenericMFPtr &fptr) const {
|
SignalBase *Object::findSignalImpl(const GenericMFPtr &fptr) const {
|
||||||
|
|||||||
@@ -107,6 +107,7 @@ public:
|
|||||||
// if(sig && slo)
|
// if(sig && slo)
|
||||||
// return Object::connect(sig,slo->operator ()(),receiver);
|
// return Object::connect(sig,slo->operator ()(),receiver);
|
||||||
// else return false;
|
// else return false;
|
||||||
|
return false;
|
||||||
}
|
}
|
||||||
|
|
||||||
// Qt5 style connector //
|
// Qt5 style connector //
|
||||||
@@ -115,15 +116,16 @@ public:
|
|||||||
connect(typename FunctionPointer<Func1>::Object *sender, Func1 sigf,
|
connect(typename FunctionPointer<Func1>::Object *sender, Func1 sigf,
|
||||||
typename FunctionPointer<Func2>::Object *receiver, Func2 slof) {
|
typename FunctionPointer<Func2>::Object *receiver, Func2 slof) {
|
||||||
SignalBase *sigb = sender->findOrAddSignal(sigf);
|
SignalBase *sigb = sender->findOrAddSignal(sigf);
|
||||||
typedef boost::signals2::signal<
|
ConnectSignal<typename FunctionPointer<Func1>::SignalSignature>(sigb, slof,
|
||||||
typename FunctionPointer<Func2>::SignalSignature>
|
receiver);
|
||||||
SigT;
|
return true;
|
||||||
ConnectSignal(sigb, slof, receiver);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
template <typename FuncT>
|
template <typename FuncT>
|
||||||
static inline bool connect(SignalBase *sigb, FuncT slof, Object *receiver) {
|
static inline bool connect(SignalBase *sigb, FuncT slof, Object *receiver) {
|
||||||
ConnectSignal(sigb, slof, receiver);
|
ConnectSignal<typename FunctionPointer<FuncT>::SignalSignature>(sigb, slof,
|
||||||
|
receiver);
|
||||||
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
template <typename FuncT>
|
template <typename FuncT>
|
||||||
@@ -139,7 +141,7 @@ public:
|
|||||||
}
|
}
|
||||||
|
|
||||||
template <typename FuncT> inline bool addSlot(FuncT fun, const char *name) {
|
template <typename FuncT> inline bool addSlot(FuncT fun, const char *name) {
|
||||||
this->addSlotImpl(GenericMFPtr(fun), name);
|
return this->addSlotImpl(GenericMFPtr(fun), name);
|
||||||
}
|
}
|
||||||
|
|
||||||
template <typename FuncT>
|
template <typename FuncT>
|
||||||
|
|||||||
@@ -23,21 +23,19 @@
|
|||||||
|
|
||||||
//////////////////////////////////////////////////////////////////////////////*/
|
//////////////////////////////////////////////////////////////////////////////*/
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
#ifndef U_CORE_SIGNAL_H
|
#ifndef U_CORE_SIGNAL_H
|
||||||
#define U_CORE_SIGNAL_H
|
#define U_CORE_SIGNAL_H
|
||||||
|
|
||||||
#include <boost/typeof/typeof.hpp>
|
#include <boost/typeof/typeof.hpp>
|
||||||
|
|
||||||
#include <boost/signals2/signal.hpp>
|
#include <boost/signals2/signal.hpp>
|
||||||
#include <boost/signals2/slot.hpp>
|
|
||||||
#include <boost/signals2/signal_type.hpp>
|
#include <boost/signals2/signal_type.hpp>
|
||||||
|
#include <boost/signals2/slot.hpp>
|
||||||
|
|
||||||
#include "Function.h"
|
#include "Function.h"
|
||||||
|
#include <boost/bind/bind.hpp>
|
||||||
|
|
||||||
|
using namespace boost::placeholders;
|
||||||
|
|
||||||
|
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
@@ -48,12 +46,15 @@
|
|||||||
#define slots
|
#define slots
|
||||||
#define signals /*virtual void init_signals();*/ public
|
#define signals /*virtual void init_signals();*/ public
|
||||||
#define emit
|
#define emit
|
||||||
#define SLOT(a) BOOST_STRINGIZE(a)
|
#define SLOT(a) BOOST_STRINGIZE(a)
|
||||||
#define SIGNAL(a) BOOST_STRINGIZE(a)
|
#define SIGNAL(a) BOOST_STRINGIZE(a)
|
||||||
|
|
||||||
#define _ULIB_DETAIL_SIGNAL_EMIT(_name,...) \
|
#define _ULIB_DETAIL_SIGNAL_EMIT(_name, ...) \
|
||||||
static BOOST_AUTO(sig,this->findOrAddSignal(&_name)); \
|
do { \
|
||||||
sig->operator()(__VA_ARGS__);
|
BOOST_AUTO(sig, this->findOrAddSignal(&_name)); \
|
||||||
|
if (sig) \
|
||||||
|
sig->operator()(__VA_ARGS__); \
|
||||||
|
} while (0)
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Utility macro to implement signal emission implementa una delle seguenti:
|
* Utility macro to implement signal emission implementa una delle seguenti:
|
||||||
@@ -66,103 +67,101 @@
|
|||||||
* // cast automatico //
|
* // cast automatico //
|
||||||
* static BOOST_AUTO(sig,this->findOrAddSignal(&Ob1::V0));
|
* static BOOST_AUTO(sig,this->findOrAddSignal(&Ob1::V0));
|
||||||
* sig->operator()();
|
* sig->operator()();
|
||||||
*/
|
*/
|
||||||
#define ULIB_SIGNAL_EMIT(_name,...) _ULIB_DETAIL_SIGNAL_EMIT(_name,__VA_ARGS__)
|
#define ULIB_SIGNAL_EMIT(_name, ...) \
|
||||||
|
_ULIB_DETAIL_SIGNAL_EMIT(_name, __VA_ARGS__)
|
||||||
|
|
||||||
namespace uLib {
|
namespace uLib {
|
||||||
|
|
||||||
|
|
||||||
// A boost::signal wrapper structure ///////////////////////////////////////////
|
// A boost::signal wrapper structure ///////////////////////////////////////////
|
||||||
|
|
||||||
// TODO ...
|
// TODO ...
|
||||||
|
|
||||||
typedef boost::signals2::signal_base SignalBase;
|
typedef boost::signals2::signal_base SignalBase;
|
||||||
|
|
||||||
template <typename T>
|
template <typename T> struct Signal {
|
||||||
struct Signal {
|
typedef boost::signals2::signal<T> type;
|
||||||
typedef boost::signals2::signal<T> type;
|
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
|
|
||||||
|
|
||||||
namespace detail {
|
namespace detail {
|
||||||
|
|
||||||
|
template <typename FuncT, typename SigSignature, int arity>
|
||||||
template <typename FuncT, int arity>
|
|
||||||
struct ConnectSignal {};
|
struct ConnectSignal {};
|
||||||
|
|
||||||
template <typename FuncT>
|
template <typename FuncT, typename SigSignature>
|
||||||
struct ConnectSignal< FuncT, 0 > {
|
struct ConnectSignal<FuncT, SigSignature, 0> {
|
||||||
static void connect(SignalBase *sigb, FuncT slof, typename FunctionPointer<FuncT>::Object *receiver) {
|
static void connect(SignalBase *sigb, FuncT slof,
|
||||||
typedef typename Signal<typename FunctionPointer<FuncT>::SignalSignature>::type SigT;
|
typename FunctionPointer<FuncT>::Object *receiver) {
|
||||||
reinterpret_cast<SigT*>(sigb)->connect(slof);
|
typedef typename Signal<SigSignature>::type SigT;
|
||||||
}
|
reinterpret_cast<SigT *>(sigb)->connect(slof);
|
||||||
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
template <typename FuncT>
|
template <typename FuncT, typename SigSignature>
|
||||||
struct ConnectSignal< FuncT, 1 > {
|
struct ConnectSignal<FuncT, SigSignature, 1> {
|
||||||
static void connect(SignalBase *sigb, FuncT slof, typename FunctionPointer<FuncT>::Object *receiver) {
|
static void connect(SignalBase *sigb, FuncT slof,
|
||||||
typedef typename Signal<typename FunctionPointer<FuncT>::SignalSignature>::type SigT;
|
typename FunctionPointer<FuncT>::Object *receiver) {
|
||||||
reinterpret_cast<SigT*>(sigb)->connect(boost::bind(slof,receiver));
|
typedef typename Signal<SigSignature>::type SigT;
|
||||||
}
|
reinterpret_cast<SigT *>(sigb)->connect(boost::bind(slof, receiver));
|
||||||
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
template <typename FuncT>
|
template <typename FuncT, typename SigSignature>
|
||||||
struct ConnectSignal< FuncT, 2 > {
|
struct ConnectSignal<FuncT, SigSignature, 2> {
|
||||||
static void connect(SignalBase *sigb, FuncT slof, typename FunctionPointer<FuncT>::Object *receiver) {
|
static void connect(SignalBase *sigb, FuncT slof,
|
||||||
typedef typename Signal<typename FunctionPointer<FuncT>::SignalSignature>::type SigT;
|
typename FunctionPointer<FuncT>::Object *receiver) {
|
||||||
reinterpret_cast<SigT*>(sigb)->connect(boost::bind(slof,receiver,_1));
|
typedef typename Signal<SigSignature>::type SigT;
|
||||||
}
|
reinterpret_cast<SigT *>(sigb)->connect(boost::bind(slof, receiver, _1));
|
||||||
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
template <typename FuncT>
|
template <typename FuncT, typename SigSignature>
|
||||||
struct ConnectSignal< FuncT, 3 > {
|
struct ConnectSignal<FuncT, SigSignature, 3> {
|
||||||
static void connect(SignalBase *sigb, FuncT slof, typename FunctionPointer<FuncT>::Object *receiver) {
|
static void connect(SignalBase *sigb, FuncT slof,
|
||||||
typedef typename Signal<typename FunctionPointer<FuncT>::SignalSignature>::type SigT;
|
typename FunctionPointer<FuncT>::Object *receiver) {
|
||||||
reinterpret_cast<SigT*>(sigb)->connect(boost::bind(slof,receiver,_1,_2));
|
typedef typename Signal<SigSignature>::type SigT;
|
||||||
}
|
reinterpret_cast<SigT *>(sigb)->connect(
|
||||||
|
boost::bind(slof, receiver, _1, _2));
|
||||||
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
template <typename FuncT>
|
template <typename FuncT, typename SigSignature>
|
||||||
struct ConnectSignal< FuncT, 4 > {
|
struct ConnectSignal<FuncT, SigSignature, 4> {
|
||||||
static void connect(SignalBase *sigb, FuncT slof, typename FunctionPointer<FuncT>::Object *receiver) {
|
static void connect(SignalBase *sigb, FuncT slof,
|
||||||
typedef typename Signal<typename FunctionPointer<FuncT>::SignalSignature>::type SigT;
|
typename FunctionPointer<FuncT>::Object *receiver) {
|
||||||
reinterpret_cast<SigT*>(sigb)->connect(boost::bind(slof,receiver,_1,_2,_3));
|
typedef typename Signal<SigSignature>::type SigT;
|
||||||
}
|
reinterpret_cast<SigT *>(sigb)->connect(
|
||||||
|
boost::bind(slof, receiver, _1, _2, _3));
|
||||||
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
template <typename FuncT>
|
template <typename FuncT, typename SigSignature>
|
||||||
struct ConnectSignal< FuncT, 5 > {
|
struct ConnectSignal<FuncT, SigSignature, 5> {
|
||||||
static void connect(SignalBase *sigb, FuncT slof, typename FunctionPointer<FuncT>::Object *receiver) {
|
static void connect(SignalBase *sigb, FuncT slof,
|
||||||
typedef typename Signal<typename FunctionPointer<FuncT>::SignalSignature>::type SigT;
|
typename FunctionPointer<FuncT>::Object *receiver) {
|
||||||
reinterpret_cast<SigT*>(sigb)->connect(boost::bind(slof,receiver,_1,_2,_3,_4));
|
typedef typename Signal<SigSignature>::type SigT;
|
||||||
}
|
reinterpret_cast<SigT *>(sigb)->connect(
|
||||||
|
boost::bind(slof, receiver, _1, _2, _3, _4));
|
||||||
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
|
} // namespace detail
|
||||||
|
|
||||||
} // detail
|
template <typename FuncT> SignalBase *NewSignal(FuncT f) {
|
||||||
|
return new
|
||||||
|
typename Signal<typename FunctionPointer<FuncT>::SignalSignature>::type;
|
||||||
|
|
||||||
template <typename FuncT>
|
|
||||||
SignalBase *NewSignal(FuncT f) {
|
|
||||||
// seems to work wow !
|
|
||||||
return new Signal<void()>::type;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
template <typename FuncT>
|
template <typename SigSignature, typename FuncT>
|
||||||
void ConnectSignal(SignalBase *sigb, FuncT slof, typename FunctionPointer<FuncT>::Object *receiver)
|
void ConnectSignal(SignalBase *sigb, FuncT slof,
|
||||||
{
|
typename FunctionPointer<FuncT>::Object *receiver) {
|
||||||
detail::ConnectSignal< FuncT, FunctionPointer<FuncT>::arity >::connect(sigb,slof,receiver);
|
detail::ConnectSignal<FuncT, SigSignature,
|
||||||
|
FunctionPointer<FuncT>::arity>::connect(sigb, slof,
|
||||||
|
receiver);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
} // namespace uLib
|
||||||
|
|
||||||
|
|
||||||
} // uLib
|
|
||||||
|
|
||||||
#endif // SIGNAL_H
|
#endif // SIGNAL_H
|
||||||
|
|||||||
119
src/Core/Uuid.h
119
src/Core/Uuid.h
@@ -23,69 +23,49 @@
|
|||||||
|
|
||||||
//////////////////////////////////////////////////////////////////////////////*/
|
//////////////////////////////////////////////////////////////////////////////*/
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
#ifndef U_CORE_UUID_H
|
#ifndef U_CORE_UUID_H
|
||||||
#define U_CORE_UUID_H
|
#define U_CORE_UUID_H
|
||||||
|
|
||||||
#include <iostream>
|
#include <iostream>
|
||||||
#include <vector>
|
#include <vector>
|
||||||
|
|
||||||
#include <boost/uuid/uuid.hpp>
|
|
||||||
#include <boost/uuid/name_generator.hpp>
|
#include <boost/uuid/name_generator.hpp>
|
||||||
#include <boost/uuid/random_generator.hpp>
|
#include <boost/uuid/random_generator.hpp>
|
||||||
|
#include <boost/uuid/uuid.hpp>
|
||||||
#include <boost/uuid/uuid_io.hpp>
|
#include <boost/uuid/uuid_io.hpp>
|
||||||
|
|
||||||
#include "Core/Mpl.h"
|
#include "Core/Mpl.h"
|
||||||
#include "Core/Object.h"
|
#include "Core/Object.h"
|
||||||
|
|
||||||
|
|
||||||
namespace uLib {
|
namespace uLib {
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
// Object Registration //
|
// Object Registration //
|
||||||
|
|
||||||
|
|
||||||
typedef boost::uuids::uuid uuid_t;
|
typedef boost::uuids::uuid uuid_t;
|
||||||
|
|
||||||
extern uuid_t uLib_dns_uuid;
|
extern uuid_t uLib_dns_uuid;
|
||||||
|
|
||||||
template < typename T >
|
template <typename T> class type_id : public boost::uuids::uuid {
|
||||||
class type_id : public boost::uuids::uuid {
|
|
||||||
public:
|
public:
|
||||||
type_id() :
|
type_id()
|
||||||
m_size(sizeof(T)),
|
: m_size(sizeof(T)),
|
||||||
uuid(boost::uuids::name_generator(uLib_dns_uuid)(typeid(T).name()))
|
uuid(boost::uuids::name_generator(uLib_dns_uuid)(typeid(T).name())) {
|
||||||
{
|
std::cout << "Request for register new type\n"
|
||||||
std::cout << "Request for register new type\n" <<
|
<< "name: " << typeid(T).name() << "\n"
|
||||||
"name: " << typeid(T).name() << "\n" <<
|
<< "uuid: " << to_string(*this) << "\n";
|
||||||
"uuid: " << to_string(*this) << "\n";
|
}
|
||||||
}
|
|
||||||
|
|
||||||
explicit type_id(boost::uuids::uuid const& u)
|
explicit type_id(boost::uuids::uuid const &u) : boost::uuids::uuid(u) {}
|
||||||
: boost::uuids::uuid(u) {}
|
|
||||||
|
|
||||||
operator boost::uuids::uuid() {
|
unsigned int size() const { return m_size; }
|
||||||
return static_cast<boost::uuids::uuid&>(*this);
|
|
||||||
}
|
|
||||||
|
|
||||||
operator boost::uuids::uuid() const {
|
|
||||||
return static_cast<boost::uuids::uuid const&>(*this);
|
|
||||||
}
|
|
||||||
|
|
||||||
unsigned int size() const { return m_size; }
|
|
||||||
|
|
||||||
private:
|
private:
|
||||||
unsigned int m_size;
|
unsigned int m_size;
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
@@ -94,70 +74,57 @@ private:
|
|||||||
namespace detail {
|
namespace detail {
|
||||||
|
|
||||||
class TypeRegister {
|
class TypeRegister {
|
||||||
typedef boost::uuids::name_generator IDGen_t;
|
typedef boost::uuids::name_generator IDGen_t;
|
||||||
|
|
||||||
public:
|
public:
|
||||||
struct RegisterEntry {
|
struct RegisterEntry {
|
||||||
uuid_t id;
|
uuid_t id;
|
||||||
int size;
|
int size;
|
||||||
};
|
};
|
||||||
|
|
||||||
TypeRegister(uuid_t const &dns) :
|
TypeRegister(uuid_t const &dns) : gen(dns) {}
|
||||||
gen(dns) {}
|
|
||||||
|
|
||||||
|
|
||||||
template< typename T >
|
|
||||||
RegisterEntry * AddType(T *t = NULL) {
|
|
||||||
RegisterEntry en = { gen(typeid(T).name()), sizeof(T) };
|
|
||||||
for(int i=0; i < m_registry.size(); ++i)
|
|
||||||
if(en.id == m_registry[i].id) return &(m_registry[i]);
|
|
||||||
m_registry.push_back(en);
|
|
||||||
return &m_registry.back();
|
|
||||||
}
|
|
||||||
|
|
||||||
void PrintSelf(std::ostream &o) {
|
|
||||||
std::cout << "RegisterController: \n";
|
|
||||||
for (int i=0; i<m_registry.size(); ++i)
|
|
||||||
o << "type [" << i << "]: "
|
|
||||||
<< to_string(m_registry[i].id) << " "
|
|
||||||
<< m_registry[i].size << "\n";
|
|
||||||
o << "\n";
|
|
||||||
}
|
|
||||||
|
|
||||||
|
template <typename T> RegisterEntry *AddType(T *t = NULL) {
|
||||||
|
RegisterEntry en = {gen(typeid(T).name()), sizeof(T)};
|
||||||
|
for (int i = 0; i < m_registry.size(); ++i)
|
||||||
|
if (en.id == m_registry[i].id)
|
||||||
|
return &(m_registry[i]);
|
||||||
|
m_registry.push_back(en);
|
||||||
|
return &m_registry.back();
|
||||||
|
}
|
||||||
|
|
||||||
|
void PrintSelf(std::ostream &o) {
|
||||||
|
std::cout << "RegisterController: \n";
|
||||||
|
for (int i = 0; i < m_registry.size(); ++i)
|
||||||
|
o << "type [" << i << "]: " << to_string(m_registry[i].id) << " "
|
||||||
|
<< m_registry[i].size << "\n";
|
||||||
|
o << "\n";
|
||||||
|
}
|
||||||
|
|
||||||
private:
|
private:
|
||||||
IDGen_t gen;
|
IDGen_t gen;
|
||||||
std::vector<RegisterEntry> m_registry;
|
std::vector<RegisterEntry> m_registry;
|
||||||
|
|
||||||
};
|
};
|
||||||
|
|
||||||
} // detail
|
} // namespace detail
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
class TypeRegister : public detail::TypeRegister {
|
class TypeRegister : public detail::TypeRegister {
|
||||||
public:
|
public:
|
||||||
typedef detail::TypeRegister BaseClass;
|
typedef detail::TypeRegister BaseClass;
|
||||||
typedef detail::TypeRegister::RegisterEntry Entry;
|
typedef detail::TypeRegister::RegisterEntry Entry;
|
||||||
|
|
||||||
static TypeRegister* Controller();
|
static TypeRegister *Controller();
|
||||||
|
|
||||||
private:
|
private:
|
||||||
TypeRegister(); // Blocks constructor
|
TypeRegister(); // Blocks constructor
|
||||||
static TypeRegister *s_Instance; // Singleton instance
|
static TypeRegister *s_Instance; // Singleton instance
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
// OBJECT REGISTER //
|
// OBJECT REGISTER //
|
||||||
|
|
||||||
|
} // namespace uLib
|
||||||
|
|
||||||
} // uLib
|
|
||||||
|
|
||||||
#endif // UUID_H
|
#endif // UUID_H
|
||||||
|
|||||||
@@ -23,156 +23,433 @@
|
|||||||
|
|
||||||
//////////////////////////////////////////////////////////////////////////////*/
|
//////////////////////////////////////////////////////////////////////////////*/
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
#ifndef U_CORE_VECTOR_H
|
#ifndef U_CORE_VECTOR_H
|
||||||
#define U_CORE_VECTOR_H
|
#define U_CORE_VECTOR_H
|
||||||
|
|
||||||
#include <vector>
|
|
||||||
#include <iostream>
|
#include <iostream>
|
||||||
|
#include <map>
|
||||||
|
#include <mutex>
|
||||||
|
#include <vector>
|
||||||
|
|
||||||
|
#include <Core/DataAllocator.h>
|
||||||
|
|
||||||
#include <Core/StaticInterface.h>
|
|
||||||
#include <Core/SmartPointer.h>
|
|
||||||
#include <Core/CommaInitializer.h>
|
#include <Core/CommaInitializer.h>
|
||||||
|
#include <Core/SmartPointer.h>
|
||||||
|
#include <Core/StaticInterface.h>
|
||||||
|
|
||||||
|
#ifdef USE_CUDA
|
||||||
|
#include <thrust/device_ptr.h>
|
||||||
|
#include <thrust/device_vector.h>
|
||||||
|
#endif
|
||||||
|
|
||||||
namespace uLib {
|
namespace uLib {
|
||||||
|
|
||||||
// Vector Implemetation ... wraps std::vector
|
// MetaAllocator Implementation ...
|
||||||
template <typename T>
|
template <typename T> class MetaAllocator {
|
||||||
class Vector : public std::vector<T, std::allocator<T> >
|
|
||||||
{
|
|
||||||
typedef std::vector< T,std::allocator<T> > BaseClass;
|
|
||||||
typedef std::allocator<T> Allocator;
|
|
||||||
public:
|
|
||||||
typedef T TypeData;
|
|
||||||
typedef __gnu_cxx::__normal_iterator<T*, BaseClass > Iterator;
|
|
||||||
typedef __gnu_cxx::__normal_iterator<const T*, BaseClass> ConstIterator;
|
|
||||||
|
|
||||||
typedef CommaInitializer< Vector<T> , T > VectorCommaInit;
|
|
||||||
|
|
||||||
Vector(unsigned int size) : BaseClass(size) {}
|
|
||||||
Vector(unsigned int size, T &value) : BaseClass(size,value) {}
|
|
||||||
Vector() : BaseClass(0) {}
|
|
||||||
|
|
||||||
inline VectorCommaInit operator <<(T scalar) {
|
|
||||||
return VectorCommaInit(this, scalar);
|
|
||||||
}
|
|
||||||
|
|
||||||
inline void PrintSelf(std::ostream &o);
|
|
||||||
|
|
||||||
void remove_element(unsigned int index) {
|
|
||||||
std::swap(this->at(index),this->back());
|
|
||||||
this->pop_back();
|
|
||||||
}
|
|
||||||
|
|
||||||
void remove_element(T &t) {
|
|
||||||
std::swap(t, this->back());
|
|
||||||
this->pop_back();
|
|
||||||
}
|
|
||||||
|
|
||||||
};
|
|
||||||
|
|
||||||
template<typename T>
|
|
||||||
void Vector<T>::PrintSelf(std::ostream &o)
|
|
||||||
{
|
|
||||||
o << " *** uLib Vector *** \n";
|
|
||||||
o << " n. of items = " << this->size() << "\n";
|
|
||||||
for(int i=0; i< this->size(); ++i)
|
|
||||||
o << (T)this->at(i) << " ";
|
|
||||||
o << "\n";
|
|
||||||
}
|
|
||||||
|
|
||||||
template <typename T>
|
|
||||||
std::ostream & operator << (std::ostream &o, const Vector<T> &v) {
|
|
||||||
for(int i=0; i< v.size(); ++i)
|
|
||||||
o << (T)v.at(i) << " ";
|
|
||||||
o << "\n";
|
|
||||||
return o;
|
|
||||||
}
|
|
||||||
|
|
||||||
template <typename T>
|
|
||||||
std::ofstream & operator << (std::ofstream &o, const Vector<T> &v) {
|
|
||||||
for(int i=0; i< v.size(); ++i)
|
|
||||||
o << (T)v.at(i) << " ";
|
|
||||||
return o;
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
template < typename T >
|
|
||||||
std::istream & operator >> (std::istream &is, Vector<T> &v) {
|
|
||||||
T value;
|
|
||||||
while( is >> value ) {
|
|
||||||
if(is.fail()) v.push_back(0);
|
|
||||||
else v.push_back( value );
|
|
||||||
}
|
|
||||||
return is;
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
// Smart pointer Vector Implementation //
|
|
||||||
|
|
||||||
|
|
||||||
template <typename T>
|
|
||||||
class SmartVector : public SmartPointer< Vector<T> > {
|
|
||||||
typedef SmartPointer< Vector<T> > Base;
|
|
||||||
public:
|
public:
|
||||||
|
using value_type = T;
|
||||||
|
using pointer = T *;
|
||||||
|
using const_pointer = const T *;
|
||||||
|
using reference = T &;
|
||||||
|
using const_reference = const T &;
|
||||||
|
using size_type = std::size_t;
|
||||||
|
using difference_type = std::ptrdiff_t;
|
||||||
|
|
||||||
SmartVector() : Base(new Vector<T>()) { }
|
template <class U> struct rebind {
|
||||||
SmartVector( const SmartVector ©) : Base(copy) { }
|
using other = MetaAllocator<U>;
|
||||||
SmartVector(unsigned int size) : Base(new Vector<T>((int)size)) { }
|
};
|
||||||
|
|
||||||
virtual ~SmartVector() {}
|
MetaAllocator() noexcept = default;
|
||||||
|
|
||||||
T& operator[](int p) {
|
template <class U>
|
||||||
return Base::get()->at(p);
|
constexpr MetaAllocator(const MetaAllocator<U> &) noexcept {}
|
||||||
|
|
||||||
|
T *allocate(std::size_t n) {
|
||||||
|
if (n == 0)
|
||||||
|
return nullptr;
|
||||||
|
|
||||||
|
DataAllocator<T> *da = new DataAllocator<T>(n, false);
|
||||||
|
T *ptr = da->GetRAMData();
|
||||||
|
|
||||||
|
std::lock_guard<std::mutex> lock(GetMutex());
|
||||||
|
GetAllocationMap()[ptr] = da;
|
||||||
|
return ptr;
|
||||||
}
|
}
|
||||||
|
|
||||||
void swap_elements(unsigned int first, unsigned int second) {
|
void deallocate(T *p, std::size_t /*n*/) noexcept {
|
||||||
std::swap(Base::get()->at(first),Base::get()->at(second));
|
if (!p)
|
||||||
|
return;
|
||||||
|
|
||||||
|
std::lock_guard<std::mutex> lock(GetMutex());
|
||||||
|
auto &map = GetAllocationMap();
|
||||||
|
auto it = map.find(p);
|
||||||
|
if (it != map.end()) {
|
||||||
|
delete it->second;
|
||||||
|
map.erase(it);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
static DataAllocator<T> *GetDataAllocator(T *p) {
|
||||||
|
if (!p)
|
||||||
|
return nullptr;
|
||||||
|
std::lock_guard<std::mutex> lock(GetMutex());
|
||||||
|
auto &map = GetAllocationMap();
|
||||||
|
auto it = map.find(p);
|
||||||
|
if (it != map.end()) {
|
||||||
|
return it->second;
|
||||||
|
}
|
||||||
|
return nullptr;
|
||||||
|
}
|
||||||
|
|
||||||
|
private:
|
||||||
|
static std::map<T *, DataAllocator<T> *> &GetAllocationMap() {
|
||||||
|
static std::map<T *, DataAllocator<T> *> allocMap;
|
||||||
|
return allocMap;
|
||||||
|
}
|
||||||
|
|
||||||
|
static std::mutex &GetMutex() {
|
||||||
|
static std::mutex mtx;
|
||||||
|
return mtx;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
template <class T, class U>
|
||||||
|
bool operator==(const MetaAllocator<T> &, const MetaAllocator<U> &) {
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
template <class T, class U>
|
||||||
|
bool operator!=(const MetaAllocator<T> &, const MetaAllocator<U> &) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Vector Implemetation ... wraps std::vector
|
||||||
|
template <typename T> class Vector : public std::vector<T, MetaAllocator<T>> {
|
||||||
|
typedef std::vector<T, MetaAllocator<T>> BaseClass;
|
||||||
|
typedef MetaAllocator<T> Allocator;
|
||||||
|
|
||||||
|
public:
|
||||||
|
typedef T TypeData;
|
||||||
|
typedef __gnu_cxx::__normal_iterator<T *, BaseClass> Iterator;
|
||||||
|
typedef __gnu_cxx::__normal_iterator<const T *, BaseClass> ConstIterator;
|
||||||
|
|
||||||
|
typedef CommaInitializer<Vector<T>, T> VectorCommaInit;
|
||||||
|
typedef typename BaseClass::iterator iterator;
|
||||||
|
typedef typename BaseClass::const_iterator const_iterator;
|
||||||
|
typedef typename BaseClass::size_type size_type;
|
||||||
|
typedef typename BaseClass::reference reference;
|
||||||
|
|
||||||
|
Vector(unsigned int size) : BaseClass(size) {}
|
||||||
|
Vector(unsigned int size, T &value) : BaseClass(size, value) {}
|
||||||
|
Vector() : BaseClass(0) {}
|
||||||
|
Vector(std::initializer_list<T> init) : BaseClass(init) {}
|
||||||
|
|
||||||
|
inline VectorCommaInit operator<<(T scalar) {
|
||||||
|
return VectorCommaInit(this, scalar);
|
||||||
|
}
|
||||||
|
|
||||||
|
void MoveToVRAM() {
|
||||||
|
if (auto alloc = MetaAllocator<T>::GetDataAllocator(BaseClass::data())) {
|
||||||
|
alloc->MoveToVRAM();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void MoveToRAM() {
|
||||||
|
if (auto alloc = MetaAllocator<T>::GetDataAllocator(BaseClass::data())) {
|
||||||
|
alloc->MoveToRAM();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
T *GetVRAMData() {
|
||||||
|
if (auto alloc = MetaAllocator<T>::GetDataAllocator(BaseClass::data())) {
|
||||||
|
return alloc->GetVRAMData();
|
||||||
|
}
|
||||||
|
return nullptr;
|
||||||
|
}
|
||||||
|
|
||||||
|
const T *GetVRAMData() const {
|
||||||
|
if (auto alloc = MetaAllocator<T>::GetDataAllocator(
|
||||||
|
const_cast<T *>(BaseClass::data()))) {
|
||||||
|
return alloc->GetVRAMData();
|
||||||
|
}
|
||||||
|
return nullptr;
|
||||||
|
}
|
||||||
|
|
||||||
|
#ifdef USE_CUDA
|
||||||
|
/// Returns a thrust::device_ptr to the VRAM data (valid after MoveToVRAM()).
|
||||||
|
/// thrust::device_ptr<T> is itself a random-access iterator compatible with
|
||||||
|
/// all thrust algorithms (thrust::transform, thrust::sort,
|
||||||
|
/// thrust::for_each…).
|
||||||
|
thrust::device_ptr<T> DeviceData() {
|
||||||
|
if (auto alloc = MetaAllocator<T>::GetDataAllocator(BaseClass::data())) {
|
||||||
|
return thrust::device_pointer_cast(alloc->GetVRAMData());
|
||||||
|
}
|
||||||
|
return thrust::device_ptr<T>(nullptr);
|
||||||
|
}
|
||||||
|
|
||||||
|
thrust::device_ptr<const T> DeviceData() const {
|
||||||
|
if (auto alloc = MetaAllocator<T>::GetDataAllocator(
|
||||||
|
const_cast<T *>(BaseClass::data()))) {
|
||||||
|
return thrust::device_pointer_cast(
|
||||||
|
static_cast<const T *>(alloc->GetVRAMData()));
|
||||||
|
}
|
||||||
|
return thrust::device_ptr<const T>(nullptr);
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Device-side begin iterator (valid after MoveToVRAM()).
|
||||||
|
thrust::device_ptr<T> DeviceBegin() { return DeviceData(); }
|
||||||
|
|
||||||
|
/// Device-side end iterator (valid after MoveToVRAM()).
|
||||||
|
thrust::device_ptr<T> DeviceEnd() {
|
||||||
|
return DeviceData() + static_cast<std::ptrdiff_t>(BaseClass::size());
|
||||||
|
}
|
||||||
|
|
||||||
|
thrust::device_ptr<const T> DeviceBegin() const { return DeviceData(); }
|
||||||
|
|
||||||
|
thrust::device_ptr<const T> DeviceEnd() const {
|
||||||
|
return DeviceData() + static_cast<std::ptrdiff_t>(BaseClass::size());
|
||||||
|
}
|
||||||
|
#endif // USE_CUDA
|
||||||
|
|
||||||
|
inline void PrintSelf(std::ostream &o);
|
||||||
|
|
||||||
|
// Overrides for auto-sync //
|
||||||
|
T &operator[](size_t i) {
|
||||||
|
this->MoveToRAM();
|
||||||
|
return BaseClass::operator[](i);
|
||||||
|
}
|
||||||
|
const T &operator[](size_t i) const {
|
||||||
|
const_cast<Vector *>(this)->MoveToRAM();
|
||||||
|
return BaseClass::operator[](i);
|
||||||
|
}
|
||||||
|
T &at(size_t i) {
|
||||||
|
this->MoveToRAM();
|
||||||
|
return BaseClass::at(i);
|
||||||
|
}
|
||||||
|
const T &at(size_t i) const {
|
||||||
|
const_cast<Vector *>(this)->MoveToRAM();
|
||||||
|
return BaseClass::at(i);
|
||||||
|
}
|
||||||
|
T &front() {
|
||||||
|
this->MoveToRAM();
|
||||||
|
return BaseClass::front();
|
||||||
|
}
|
||||||
|
const T &front() const {
|
||||||
|
const_cast<Vector *>(this)->MoveToRAM();
|
||||||
|
return BaseClass::front();
|
||||||
|
}
|
||||||
|
T &back() {
|
||||||
|
this->MoveToRAM();
|
||||||
|
return BaseClass::back();
|
||||||
|
}
|
||||||
|
const T &back() const {
|
||||||
|
const_cast<Vector *>(this)->MoveToRAM();
|
||||||
|
return BaseClass::back();
|
||||||
|
}
|
||||||
|
T *data() noexcept {
|
||||||
|
this->MoveToRAM();
|
||||||
|
return BaseClass::data();
|
||||||
|
}
|
||||||
|
const T *data() const noexcept {
|
||||||
|
const_cast<Vector *>(this)->MoveToRAM();
|
||||||
|
return BaseClass::data();
|
||||||
|
}
|
||||||
|
Iterator begin() noexcept {
|
||||||
|
this->MoveToRAM();
|
||||||
|
return BaseClass::begin();
|
||||||
|
}
|
||||||
|
ConstIterator begin() const noexcept {
|
||||||
|
const_cast<Vector *>(this)->MoveToRAM();
|
||||||
|
return BaseClass::begin();
|
||||||
|
}
|
||||||
|
Iterator end() noexcept {
|
||||||
|
this->MoveToRAM();
|
||||||
|
return BaseClass::end();
|
||||||
|
}
|
||||||
|
ConstIterator end() const noexcept {
|
||||||
|
const_cast<Vector *>(this)->MoveToRAM();
|
||||||
|
return BaseClass::end();
|
||||||
|
}
|
||||||
|
auto rbegin() noexcept {
|
||||||
|
this->MoveToRAM();
|
||||||
|
return BaseClass::rbegin();
|
||||||
|
}
|
||||||
|
auto rbegin() const noexcept {
|
||||||
|
const_cast<Vector *>(this)->MoveToRAM();
|
||||||
|
return BaseClass::rbegin();
|
||||||
|
}
|
||||||
|
auto rend() noexcept {
|
||||||
|
this->MoveToRAM();
|
||||||
|
return BaseClass::rend();
|
||||||
|
}
|
||||||
|
auto rend() const noexcept {
|
||||||
|
const_cast<Vector *>(this)->MoveToRAM();
|
||||||
|
return BaseClass::rend();
|
||||||
|
}
|
||||||
|
|
||||||
|
void push_back(const T &x) {
|
||||||
|
this->MoveToRAM();
|
||||||
|
BaseClass::push_back(x);
|
||||||
|
}
|
||||||
|
void push_back(T &&x) {
|
||||||
|
this->MoveToRAM();
|
||||||
|
BaseClass::push_back(std::move(x));
|
||||||
|
}
|
||||||
|
template <typename... Args> reference emplace_back(Args &&...args) {
|
||||||
|
this->MoveToRAM();
|
||||||
|
return BaseClass::emplace_back(std::forward<Args>(args)...);
|
||||||
|
}
|
||||||
|
void pop_back() {
|
||||||
|
this->MoveToRAM();
|
||||||
|
BaseClass::pop_back();
|
||||||
|
}
|
||||||
|
|
||||||
|
template <typename... Args>
|
||||||
|
iterator emplace(const_iterator pos, Args &&...args) {
|
||||||
|
this->MoveToRAM();
|
||||||
|
return BaseClass::emplace(pos, std::forward<Args>(args)...);
|
||||||
|
}
|
||||||
|
iterator insert(const_iterator pos, const T &x) {
|
||||||
|
this->MoveToRAM();
|
||||||
|
return BaseClass::insert(pos, x);
|
||||||
|
}
|
||||||
|
iterator insert(const_iterator pos, T &&x) {
|
||||||
|
this->MoveToRAM();
|
||||||
|
return BaseClass::insert(pos, std::move(x));
|
||||||
|
}
|
||||||
|
template <typename InputIt>
|
||||||
|
iterator insert(const_iterator pos, InputIt first, InputIt last) {
|
||||||
|
this->MoveToRAM();
|
||||||
|
return BaseClass::insert(pos, first, last);
|
||||||
|
}
|
||||||
|
iterator erase(const_iterator pos) {
|
||||||
|
this->MoveToRAM();
|
||||||
|
return BaseClass::erase(pos);
|
||||||
|
}
|
||||||
|
iterator erase(const_iterator first, const_iterator last) {
|
||||||
|
this->MoveToRAM();
|
||||||
|
return BaseClass::erase(first, last);
|
||||||
|
}
|
||||||
|
|
||||||
|
void resize(size_t n) {
|
||||||
|
this->MoveToRAM();
|
||||||
|
BaseClass::resize(n);
|
||||||
|
}
|
||||||
|
void resize(size_t n, const T &x) {
|
||||||
|
this->MoveToRAM();
|
||||||
|
BaseClass::resize(n, x);
|
||||||
|
}
|
||||||
|
void reserve(size_t n) {
|
||||||
|
this->MoveToRAM();
|
||||||
|
BaseClass::reserve(n);
|
||||||
|
}
|
||||||
|
void clear() noexcept {
|
||||||
|
this->MoveToRAM();
|
||||||
|
BaseClass::clear();
|
||||||
|
}
|
||||||
|
template <typename InputIt> void assign(InputIt first, InputIt last) {
|
||||||
|
this->MoveToRAM();
|
||||||
|
BaseClass::assign(first, last);
|
||||||
|
}
|
||||||
|
void assign(size_type count, const T &value) {
|
||||||
|
this->MoveToRAM();
|
||||||
|
BaseClass::assign(count, value);
|
||||||
}
|
}
|
||||||
|
|
||||||
void remove_element(unsigned int index) {
|
void remove_element(unsigned int index) {
|
||||||
std::swap(Base::get()->at(index),Base::get()->back());
|
this->MoveToRAM();
|
||||||
Base::get()->pop_back();
|
std::swap(this->at(index), this->back());
|
||||||
|
this->pop_back();
|
||||||
|
}
|
||||||
|
|
||||||
|
void remove_element(T &t) {
|
||||||
|
this->MoveToRAM();
|
||||||
|
std::swap(t, this->back());
|
||||||
|
this->pop_back();
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
template <typename T> void Vector<T>::PrintSelf(std::ostream &o) {
|
||||||
|
o << " *** uLib Vector *** \n";
|
||||||
|
o << " n. of items = " << this->size() << "\n";
|
||||||
|
for (int i = 0; i < this->size(); ++i)
|
||||||
|
o << (T)this->at(i) << " ";
|
||||||
|
o << "\n";
|
||||||
|
}
|
||||||
|
|
||||||
|
template <typename T>
|
||||||
|
std::ostream &operator<<(std::ostream &o, const Vector<T> &v) {
|
||||||
|
for (int i = 0; i < v.size(); ++i)
|
||||||
|
o << (T)v.at(i) << " ";
|
||||||
|
o << "\n";
|
||||||
|
return o;
|
||||||
|
}
|
||||||
|
|
||||||
|
template <typename T>
|
||||||
|
std::ofstream &operator<<(std::ofstream &o, const Vector<T> &v) {
|
||||||
|
for (int i = 0; i < v.size(); ++i)
|
||||||
|
o << (T)v.at(i) << " ";
|
||||||
|
return o;
|
||||||
|
}
|
||||||
|
|
||||||
|
template <typename T> std::istream &operator>>(std::istream &is, Vector<T> &v) {
|
||||||
|
T value;
|
||||||
|
while (is >> value) {
|
||||||
|
if (is.fail())
|
||||||
|
v.push_back(0);
|
||||||
|
else
|
||||||
|
v.push_back(value);
|
||||||
|
}
|
||||||
|
return is;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Smart pointer Vector Implementation //
|
||||||
|
|
||||||
|
template <typename T> class SmartVector : public SmartPointer<Vector<T>> {
|
||||||
|
typedef SmartPointer<Vector<T>> Base;
|
||||||
|
|
||||||
|
public:
|
||||||
|
SmartVector() : Base(new Vector<T>()) {}
|
||||||
|
SmartVector(const SmartVector ©) : Base(copy) {}
|
||||||
|
SmartVector(unsigned int size) : Base(new Vector<T>((int)size)) {}
|
||||||
|
|
||||||
|
virtual ~SmartVector() {}
|
||||||
|
|
||||||
|
T &operator[](int p) { return Base::get()->at(p); }
|
||||||
|
|
||||||
|
void swap_elements(unsigned int first, unsigned int second) {
|
||||||
|
std::swap(Base::get()->at(first), Base::get()->at(second));
|
||||||
|
}
|
||||||
|
|
||||||
|
void remove_element(unsigned int index) {
|
||||||
|
std::swap(Base::get()->at(index), Base::get()->back());
|
||||||
|
Base::get()->pop_back();
|
||||||
}
|
}
|
||||||
|
|
||||||
void remove_element(T &t) {
|
void remove_element(T &t) {
|
||||||
std::swap(t, Base::get()->back());
|
std::swap(t, Base::get()->back());
|
||||||
Base::get()->pop_back();
|
Base::get()->pop_back();
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
// ------ Utils ------------------------------------------------------------- //
|
// ------ Utils ------------------------------------------------------------- //
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
// RIFARE con iteratore !
|
// RIFARE con iteratore !
|
||||||
template <typename _Tp, class _CmpT>
|
template <typename _Tp, class _CmpT>
|
||||||
inline const unsigned long
|
inline unsigned long VectorSplice(const _Tp &_it, const _Tp &_end,
|
||||||
VectorSplice(const _Tp &_it, const _Tp &_end, const float value, _CmpT _comp)
|
const float value, _CmpT _comp) {
|
||||||
{
|
_Tp it = _it;
|
||||||
_Tp it = _it;
|
_Tp end = _end - 1;
|
||||||
_Tp end = _end-1;
|
for (; it != end;) {
|
||||||
for(it; it != end;)
|
if (_comp(*it, value))
|
||||||
{
|
++it;
|
||||||
if (_comp(*it,value)) ++it;
|
else if (_comp(*end, value))
|
||||||
else if(_comp(*end,value)) std::swap(*it,*end--);
|
std::swap(*it, *end--);
|
||||||
else --end;
|
else
|
||||||
}
|
--end;
|
||||||
return it - _it;
|
}
|
||||||
|
return it - _it;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
} // namespace uLib
|
||||||
|
|
||||||
} // uLib
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
#endif // VECTOR_H
|
#endif // VECTOR_H
|
||||||
|
|||||||
@@ -8,7 +8,7 @@ set( TESTS
|
|||||||
ObjectCopyTest
|
ObjectCopyTest
|
||||||
StaticInterfaceTest
|
StaticInterfaceTest
|
||||||
CommaInitTest
|
CommaInitTest
|
||||||
DebugTTreeDumpTest
|
# DebugTTreeDumpTest
|
||||||
BoostTest
|
BoostTest
|
||||||
BoostAccumulatorTest
|
BoostAccumulatorTest
|
||||||
PropertiesTest
|
PropertiesTest
|
||||||
@@ -19,14 +19,20 @@ set( TESTS
|
|||||||
UuidTest
|
UuidTest
|
||||||
TypeIntrospectionTraversal
|
TypeIntrospectionTraversal
|
||||||
OptionsTest
|
OptionsTest
|
||||||
|
PingPongTest
|
||||||
|
VectorMetaAllocatorTest
|
||||||
)
|
)
|
||||||
|
|
||||||
set(LIBRARIES
|
set(LIBRARIES
|
||||||
${PACKAGE_LIBPREFIX}Core
|
${PACKAGE_LIBPREFIX}Core
|
||||||
${PACKAGE_LIBPREFIX}Math
|
${PACKAGE_LIBPREFIX}Math
|
||||||
${Boost_SERIALIZATION_LIBRARY}
|
Boost::serialization
|
||||||
${Boost_SIGNALS_LIBRARY}
|
Boost::program_options
|
||||||
${Boost_PROGRAM_OPTIONS_LIBRARY}
|
|
||||||
${ROOT_LIBRARIES}
|
${ROOT_LIBRARIES}
|
||||||
)
|
)
|
||||||
uLib_add_tests(${uLib-module})
|
uLib_add_tests(Core)
|
||||||
|
|
||||||
|
if(USE_CUDA)
|
||||||
|
set_source_files_properties(VectorMetaAllocatorTest.cpp PROPERTIES LANGUAGE CUDA)
|
||||||
|
endif()
|
||||||
|
|
||||||
|
|||||||
52
src/Core/testing/PingPongTest.cpp
Normal file
52
src/Core/testing/PingPongTest.cpp
Normal file
@@ -0,0 +1,52 @@
|
|||||||
|
#include "Core/Object.h"
|
||||||
|
#include "Core/Signal.h"
|
||||||
|
#include "testing-prototype.h"
|
||||||
|
#include <iostream>
|
||||||
|
|
||||||
|
using namespace uLib;
|
||||||
|
|
||||||
|
class Ping : public Object {
|
||||||
|
public:
|
||||||
|
signals:
|
||||||
|
void PingSignal(int count);
|
||||||
|
public slots:
|
||||||
|
void OnPong(int count) {
|
||||||
|
std::cout << "Ping received Pong " << count << std::endl;
|
||||||
|
if (count > 0)
|
||||||
|
ULIB_SIGNAL_EMIT(Ping::PingSignal, count - 1);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
void Ping::PingSignal(int count) { ULIB_SIGNAL_EMIT(Ping::PingSignal, count); }
|
||||||
|
|
||||||
|
class Pong : public Object {
|
||||||
|
public:
|
||||||
|
signals:
|
||||||
|
void PongSignal(int count);
|
||||||
|
public slots:
|
||||||
|
void OnPing(int count) {
|
||||||
|
std::cout << "Pong received Ping " << count << std::endl;
|
||||||
|
if (count > 0)
|
||||||
|
ULIB_SIGNAL_EMIT(Pong::PongSignal, count - 1);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
void Pong::PongSignal(int count) { ULIB_SIGNAL_EMIT(Pong::PongSignal, count); }
|
||||||
|
|
||||||
|
int main() {
|
||||||
|
BEGIN_TESTING(PingPong);
|
||||||
|
Ping ping;
|
||||||
|
Pong pong;
|
||||||
|
|
||||||
|
std::cout << "Connecting ping to pong" << std::endl;
|
||||||
|
Object::connect(&ping, &Ping::PingSignal, &pong, &Pong::OnPing);
|
||||||
|
|
||||||
|
std::cout << "Connecting pong to ping" << std::endl;
|
||||||
|
Object::connect(&pong, &Pong::PongSignal, &ping, &Ping::OnPong);
|
||||||
|
|
||||||
|
std::cout << "Emitting PingSignal(5)" << std::endl;
|
||||||
|
ping.PingSignal(5);
|
||||||
|
|
||||||
|
END_TESTING;
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
@@ -23,13 +23,10 @@
|
|||||||
|
|
||||||
//////////////////////////////////////////////////////////////////////////////*/
|
//////////////////////////////////////////////////////////////////////////////*/
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
#include <iostream>
|
|
||||||
#include <fstream>
|
#include <fstream>
|
||||||
#include <typeinfo>
|
#include <iostream>
|
||||||
#include <string>
|
#include <string>
|
||||||
|
#include <typeinfo>
|
||||||
|
|
||||||
#include "Core/Object.h"
|
#include "Core/Object.h"
|
||||||
|
|
||||||
@@ -37,78 +34,43 @@
|
|||||||
|
|
||||||
using namespace uLib;
|
using namespace uLib;
|
||||||
|
|
||||||
|
struct A : Object {
|
||||||
|
uLibTypeMacro(A, Object) A() : numa(5552368) {}
|
||||||
|
int numa;
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
struct A : Object {
|
|
||||||
uLibTypeMacro(A,Object)
|
|
||||||
A() : numa(5552368) {}
|
|
||||||
int numa;
|
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|
||||||
ULIB_SERIALIZABLE_OBJECT(A)
|
ULIB_SERIALIZABLE_OBJECT(A)
|
||||||
ULIB_SERIALIZE_OBJECT(A,Object) {
|
ULIB_SERIALIZE_OBJECT(A, Object) { ar &AR(numa); }
|
||||||
ar & AR(numa);
|
|
||||||
}
|
|
||||||
|
|
||||||
struct B : virtual Object {
|
struct B : virtual Object {
|
||||||
uLibTypeMacro(B,Object)
|
uLibTypeMacro(B, Object) B() : numb(5552369) {}
|
||||||
B() : numb(5552369) {}
|
int numb;
|
||||||
int numb;
|
|
||||||
};
|
};
|
||||||
|
|
||||||
ULIB_SERIALIZABLE_OBJECT(B)
|
ULIB_SERIALIZABLE_OBJECT(B)
|
||||||
ULIB_SERIALIZE_OBJECT(B,Object) { ar & AR(numb); }
|
ULIB_SERIALIZE_OBJECT(B, Object) { ar &AR(numb); }
|
||||||
|
|
||||||
|
|
||||||
struct C : B {
|
struct C : B {
|
||||||
uLibTypeMacro(C,B)
|
uLibTypeMacro(C, B) C() : numc(5552370) {}
|
||||||
C() : numc(5552370) {}
|
int numc;
|
||||||
int numc;
|
|
||||||
};
|
};
|
||||||
|
|
||||||
ULIB_SERIALIZABLE_OBJECT(C)
|
ULIB_SERIALIZABLE_OBJECT(C)
|
||||||
ULIB_SERIALIZE_OBJECT(C,B) { ar & AR(numc); }
|
ULIB_SERIALIZE_OBJECT(C, B) { ar &AR(numc); }
|
||||||
|
|
||||||
struct D : A,B {
|
struct D : A, B {
|
||||||
uLibTypeMacro(D,A,B)
|
uLibTypeMacro(D, A, B)
|
||||||
|
|
||||||
D() : numd(5552371) {}
|
D()
|
||||||
int numd;
|
: numd(5552371) {}
|
||||||
|
int numd;
|
||||||
};
|
};
|
||||||
|
|
||||||
ULIB_SERIALIZABLE_OBJECT(D)
|
ULIB_SERIALIZABLE_OBJECT(D)
|
||||||
ULIB_SERIALIZE_OBJECT(D,A,B) { ar & AR(numd); }
|
ULIB_SERIALIZE_OBJECT(D, A, B) { ar &AR(numd); }
|
||||||
|
|
||||||
|
int main() {
|
||||||
|
A o;
|
||||||
|
|
||||||
|
Archive::xml_oarchive(std::cout) << NVP(o);
|
||||||
main() {
|
|
||||||
A o; o.init_properties();
|
|
||||||
|
|
||||||
Archive::xml_oarchive(std::cout) << NVP(o);
|
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -23,20 +23,16 @@
|
|||||||
|
|
||||||
//////////////////////////////////////////////////////////////////////////////*/
|
//////////////////////////////////////////////////////////////////////////////*/
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
#include <iostream>
|
|
||||||
#include <fstream>
|
#include <fstream>
|
||||||
|
#include <iostream>
|
||||||
|
|
||||||
#include "Core/Object.h"
|
|
||||||
#include "Core/Archives.h"
|
#include "Core/Archives.h"
|
||||||
|
#include "Core/Object.h"
|
||||||
|
|
||||||
#include "testing-prototype.h"
|
#include "testing-prototype.h"
|
||||||
|
|
||||||
using namespace uLib;
|
using namespace uLib;
|
||||||
|
|
||||||
|
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
@@ -45,63 +41,51 @@ using namespace uLib;
|
|||||||
|
|
||||||
struct V3f {
|
struct V3f {
|
||||||
|
|
||||||
float x,y,z;
|
float x, y, z;
|
||||||
V3f()
|
V3f() { x = y = z = 0; }
|
||||||
{ x = y = z =0; }
|
|
||||||
|
|
||||||
V3f(float x, float y, float z) :
|
V3f(float x, float y, float z) : x(x), y(y), z(z) {}
|
||||||
x(x), y(y), z(z) {}
|
|
||||||
|
|
||||||
template <class Archive>
|
template <class Archive> void serialize(Archive &ar, unsigned int v) {
|
||||||
void serialize (Archive &ar,unsigned int v) {
|
ar & "<" & NVP(x) & NVP(y) & NVP(z) & ">";
|
||||||
ar
|
}
|
||||||
& "<" & NVP(x) & NVP(y) & NVP(z) & ">";
|
|
||||||
}
|
|
||||||
};
|
};
|
||||||
|
|
||||||
ULIB_CLASS_EXPORT_KEY(V3f);
|
ULIB_CLASS_EXPORT_KEY(V3f);
|
||||||
ULIB_CLASS_EXPORT_IMPLEMENT(V3f);
|
ULIB_CLASS_EXPORT_IMPLEMENT(V3f);
|
||||||
|
|
||||||
|
inline std::ostream &operator<<(std::ostream &o, const V3f &v) {
|
||||||
inline std::ostream &
|
Archive::hrt_oarchive(o) << v;
|
||||||
operator <<(std::ostream &o, const V3f &v) {
|
return o;
|
||||||
Archive::hrt_oarchive(o) << v;
|
|
||||||
return o;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
inline std::istream &
|
inline std::istream &operator>>(std::istream &is, V3f &v) {
|
||||||
operator >>(std::istream &is, V3f &v) {
|
Archive::hrt_iarchive(is) >> v;
|
||||||
Archive::hrt_iarchive(is) >> v;
|
return is;
|
||||||
return is;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
int test_V3f() {
|
int test_V3f() {
|
||||||
// testing human readble archive with simple serializable structure //
|
// testing human readble archive with simple serializable structure //
|
||||||
|
|
||||||
V3f v1(1,2,3),v2,v3,v4;
|
V3f v1(1, 2, 3), v2, v3, v4;
|
||||||
std::cout << "v --> " << v1 << "\n";
|
std::cout << "v --> " << v1 << "\n";
|
||||||
|
|
||||||
std::stringstream ss; ss << v1;
|
std::stringstream ss;
|
||||||
std::cout << "ss.v --> " << ss.str() << "\n";
|
ss << v1;
|
||||||
|
std::cout << "ss.v --> " << ss.str() << "\n";
|
||||||
|
|
||||||
Archive::hrt_iarchive ar(ss); ar >> v2;
|
Archive::hrt_iarchive ar(ss);
|
||||||
std::cout << "v2 --> " << v2 << "\n";
|
ar >> v2;
|
||||||
|
std::cout << "v2 --> " << v2 << "\n";
|
||||||
|
|
||||||
std::stringstream("<2 3 4>") >> v3;
|
std::stringstream("<2 3 4>") >> v3;
|
||||||
std::cout << "v3 --> " << v3 << "\n";
|
std::cout << "v3 --> " << v3 << "\n";
|
||||||
|
|
||||||
// std::cout << "insert V3f string to parse: "; std::cin >> v4;
|
// std::cout << "insert V3f string to parse: "; std::cin >> v4;
|
||||||
// std::cout << "v4 --> " << v4 << "\n";
|
// std::cout << "v4 --> " << v4 << "\n";
|
||||||
return (1);
|
return (1);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
@@ -109,104 +93,72 @@ int test_V3f() {
|
|||||||
// OBJECT SERIALIZATION //
|
// OBJECT SERIALIZATION //
|
||||||
|
|
||||||
class A : public virtual Object {
|
class A : public virtual Object {
|
||||||
uLibTypeMacro(A,Object)
|
uLibTypeMacro(A, Object) ULIB_SERIALIZE_ACCESS public : A() : m_a(5552368) {}
|
||||||
ULIB_SERIALIZE_ACCESS
|
|
||||||
public:
|
|
||||||
A() : m_a(5552368) {}
|
|
||||||
|
|
||||||
properties() {
|
void init_properties();
|
||||||
std::string p_a;
|
std::string p_a;
|
||||||
};
|
|
||||||
|
uLibRefMacro(a, int);
|
||||||
|
|
||||||
uLibRefMacro (a,int);
|
|
||||||
private:
|
private:
|
||||||
int m_a;
|
int m_a;
|
||||||
};
|
};
|
||||||
|
|
||||||
void A::init_properties() {
|
void A::init_properties() { p_a = "A property string"; }
|
||||||
$_init();
|
|
||||||
$$.p_a = "A property string";
|
|
||||||
}
|
|
||||||
|
|
||||||
ULIB_SERIALIZABLE_OBJECT(A)
|
ULIB_SERIALIZABLE_OBJECT(A)
|
||||||
ULIB_SERIALIZE_OBJECT(A,Object) {
|
ULIB_SERIALIZE_OBJECT(A, Object) {
|
||||||
ar
|
ar & "Object A : " & "--> m_a = " & AR(m_a) & "\n" & "Object A properties: " &
|
||||||
& "Object A : "
|
"---> p_a = " & AR(p_a) & "\n";
|
||||||
& "--> m_a = " & AR(m_a)
|
|
||||||
& "\n";
|
|
||||||
}
|
}
|
||||||
|
|
||||||
ULIB_SERIALIZE_OBJECT_PROPS(A) {
|
|
||||||
ar
|
|
||||||
& "Object A properties: "
|
|
||||||
& "---> p_a = " & AR(p_a) & "\n";
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
int testing_xml_class() {
|
int testing_xml_class() {
|
||||||
|
|
||||||
A a; a.init_properties();
|
A a;
|
||||||
|
a.init_properties();
|
||||||
|
|
||||||
{
|
{
|
||||||
std::ofstream file("test.xml");
|
std::ofstream file("test.xml");
|
||||||
Archive::xml_oarchive(file) << NVP(a);
|
Archive::xml_oarchive(file) << NVP(a);
|
||||||
}
|
}
|
||||||
a.a() = 0;
|
a.a() = 0;
|
||||||
a.$$.p_a = "zero string";
|
a.p_a = "zero string";
|
||||||
{
|
{
|
||||||
std::ifstream file("test.xml");
|
std::ifstream file("test.xml");
|
||||||
Archive::xml_iarchive(file) >> NVP(a);
|
Archive::xml_iarchive(file) >> NVP(a);
|
||||||
}
|
}
|
||||||
|
|
||||||
Archive::xml_oarchive(std::cout) << NVP(a);
|
Archive::xml_oarchive(std::cout) << NVP(a);
|
||||||
return ( a.a() == 5552368 && a.$$.p_a == "A property string" );
|
return (a.a() == 5552368 && a.p_a == "A property string");
|
||||||
}
|
}
|
||||||
|
|
||||||
int testing_hrt_class() {
|
int testing_hrt_class() {
|
||||||
|
|
||||||
A a; a.init_properties();
|
A a;
|
||||||
|
a.init_properties();
|
||||||
|
|
||||||
{
|
{
|
||||||
std::ofstream file("test.xml");
|
std::ofstream file("test.xml");
|
||||||
Archive::hrt_oarchive(file) << NVP(a);
|
Archive::hrt_oarchive(file) << NVP(a);
|
||||||
}
|
}
|
||||||
a.a() = 0;
|
a.a() = 0;
|
||||||
a.$$.p_a = "zero string";
|
a.p_a = "zero string";
|
||||||
{
|
{
|
||||||
// ERRORE FIX !
|
// ERRORE FIX !
|
||||||
// std::ifstream file("test.xml");
|
// std::ifstream file("test.xml");
|
||||||
// Archive::hrt_iarchive(file) >> NVP(a);
|
// Archive::hrt_iarchive(file) >> NVP(a);
|
||||||
}
|
}
|
||||||
|
|
||||||
Archive::hrt_oarchive(std::cout) << NVP(a);
|
Archive::hrt_oarchive(std::cout) << NVP(a);
|
||||||
return ( a.a() == 5552368 && a.$$.p_a == "A property string" );
|
return (a.a() == 5552368 && a.p_a == "A property string");
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
int main() {
|
int main() {
|
||||||
BEGIN_TESTING(Serialize Test);
|
BEGIN_TESTING(Serialize Test);
|
||||||
|
|
||||||
TEST1( test_V3f() );
|
TEST1(test_V3f());
|
||||||
TEST1( testing_xml_class() );
|
TEST1(testing_xml_class());
|
||||||
// testing_hrt_class(); ///// << ERRORE in HRT with properties
|
// testing_hrt_class(); ///// << ERRORE in HRT with properties
|
||||||
|
|
||||||
END_TESTING;
|
END_TESTING;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -23,93 +23,63 @@
|
|||||||
|
|
||||||
//////////////////////////////////////////////////////////////////////////////*/
|
//////////////////////////////////////////////////////////////////////////////*/
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
#include <iostream>
|
#include <iostream>
|
||||||
#include <typeinfo>
|
#include <typeinfo>
|
||||||
|
|
||||||
|
|
||||||
#include "testing-prototype.h"
|
|
||||||
#include "Core/Types.h"
|
|
||||||
#include "Core/Object.h"
|
#include "Core/Object.h"
|
||||||
#include "Core/Signal.h"
|
#include "Core/Signal.h"
|
||||||
|
#include "Core/Types.h"
|
||||||
|
#include "testing-prototype.h"
|
||||||
|
|
||||||
using namespace uLib;
|
using namespace uLib;
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
class Ob1 : public Object {
|
class Ob1 : public Object {
|
||||||
public:
|
public:
|
||||||
|
|
||||||
|
|
||||||
signals:
|
signals:
|
||||||
void V0();
|
void V0();
|
||||||
|
|
||||||
int V1(int a);
|
|
||||||
|
|
||||||
|
void V1(int a);
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
// should be done by moc //
|
// should be done by moc //
|
||||||
void Ob1::V0() {
|
void Ob1::V0() { ULIB_SIGNAL_EMIT(Ob1::V0); }
|
||||||
ULIB_SIGNAL_EMIT(Ob1::V0);
|
|
||||||
}
|
|
||||||
|
|
||||||
int Ob1::V1(int a) {
|
|
||||||
ULIB_SIGNAL_EMIT(Ob1::V1,a);
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
void Ob1::V1(int a) { ULIB_SIGNAL_EMIT(Ob1::V1, a); }
|
||||||
|
|
||||||
class Ob2 : public Object {
|
class Ob2 : public Object {
|
||||||
|
|
||||||
public slots:
|
public slots:
|
||||||
void PrintV0() {
|
void PrintV0() { std::cout << "Ob2 prints V0\n" << std::flush; }
|
||||||
std::cout << "Ob2 prints V0\n" << std::flush;
|
|
||||||
}
|
|
||||||
};
|
};
|
||||||
|
|
||||||
class Ob3 : public Object {
|
class Ob3 : public Object {
|
||||||
|
|
||||||
public slots:
|
public slots:
|
||||||
void PrintV0() {
|
void PrintV0() { std::cout << "Ob3 prints V0\n" << std::flush; }
|
||||||
std::cout << "Ob3 prints V0\n" << std::flush;
|
|
||||||
}
|
|
||||||
|
|
||||||
void PrintNumber(int n) {
|
void PrintNumber(int n) {
|
||||||
std::cout << "Ob3 is printing number: " << n << "\n";
|
std::cout << "Ob3 is printing number: " << n << "\n";
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
int main() {
|
int main() {
|
||||||
BEGIN_TESTING(Signals);
|
BEGIN_TESTING(Signals);
|
||||||
|
|
||||||
Ob1 ob1;
|
Ob1 ob1;
|
||||||
Ob2 ob2;
|
Ob2 ob2;
|
||||||
Ob3 ob3;
|
Ob3 ob3;
|
||||||
|
|
||||||
Object::connect(&ob1,&Ob1::V0,&ob2,&Ob2::PrintV0);
|
Object::connect(&ob1, &Ob1::V0, &ob2, &Ob2::PrintV0);
|
||||||
Object::connect(&ob1,&Ob1::V0,&ob3,&Ob3::PrintV0);
|
Object::connect(&ob1, &Ob1::V0, &ob3, &Ob3::PrintV0);
|
||||||
Object::connect(&ob1,&Ob1::V1,&ob3,&Ob3::PrintNumber);
|
Object::connect(&ob1, &Ob1::V1, &ob3, &Ob3::PrintNumber);
|
||||||
|
|
||||||
// not working yet
|
// not working yet
|
||||||
// Object::connect(&ob1,SIGNAL(V0(),&ob2,SLOT(PrintV0())
|
// Object::connect(&ob1,SIGNAL(V0(),&ob2,SLOT(PrintV0())
|
||||||
|
|
||||||
ob1.PrintSelf(std::cout);
|
ob1.PrintSelf(std::cout);
|
||||||
|
|
||||||
emit ob1.V0();
|
emit ob1.V0();
|
||||||
emit ob1.V1(5552368);
|
emit ob1.V1(5552368);
|
||||||
|
|
||||||
END_TESTING;
|
END_TESTING;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -23,90 +23,73 @@
|
|||||||
|
|
||||||
//////////////////////////////////////////////////////////////////////////////*/
|
//////////////////////////////////////////////////////////////////////////////*/
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
#include <iostream>
|
#include <iostream>
|
||||||
|
|
||||||
#include "testing-prototype.h"
|
#include "testing-prototype.h"
|
||||||
#include <Core/StaticInterface.h>
|
#include <Core/StaticInterface.h>
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
namespace uLib {
|
namespace uLib {
|
||||||
|
|
||||||
//// INTERFACE TO COMPLEX CLASS /////
|
//// INTERFACE TO COMPLEX CLASS /////
|
||||||
|
|
||||||
namespace Interface {
|
namespace Interface {
|
||||||
struct Test {
|
struct Test {
|
||||||
MAKE_TRAITS
|
MAKE_TRAITS
|
||||||
template<class Self> void check_structural() {
|
template <class Self> void check_structural() {
|
||||||
uLibCheckFunction(Self,test,bool,int,float);
|
uLibCheckFunction(Self, test, bool, int, float);
|
||||||
uLibCheckMember(Self,testmemb,int);
|
uLibCheckMember(Self, testmemb, int);
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
}
|
} // namespace Interface
|
||||||
|
|
||||||
struct Test {
|
struct Test {
|
||||||
bool test(int i, float f){}
|
bool test(int i, float f) { return true; }
|
||||||
int testmemb;
|
int testmemb;
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
//// INTERFAC TO SIMPLE CLASS ///////////
|
//// INTERFAC TO SIMPLE CLASS ///////////
|
||||||
|
|
||||||
namespace Interface {
|
namespace Interface {
|
||||||
struct Simple {
|
struct Simple {
|
||||||
MAKE_TRAITS
|
MAKE_TRAITS
|
||||||
template<class Self> void check_structural() {
|
template <class Self> void check_structural() {
|
||||||
uLibCheckMember(Self,memb1,int);
|
uLibCheckMember(Self, memb1, int);
|
||||||
uLibCheckMember(Self,memb2,float);
|
uLibCheckMember(Self, memb2, float);
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
}
|
} // namespace Interface
|
||||||
|
|
||||||
struct Simple {
|
struct Simple {
|
||||||
int memb1;
|
int memb1;
|
||||||
float memb2;
|
float memb2;
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|
||||||
/////////////////////////
|
/////////////////////////
|
||||||
|
|
||||||
template <class T>
|
template <class T> class UseTest {
|
||||||
class UseTest {
|
|
||||||
public:
|
public:
|
||||||
UseTest() {
|
UseTest() {
|
||||||
Interface::IsA<T,Interface::Test>();
|
Interface::IsA<T, Interface::Test>();
|
||||||
T t;
|
T t;
|
||||||
int i; float f;
|
int i;
|
||||||
t.test(i,f);
|
float f;
|
||||||
}
|
t.test(i, f);
|
||||||
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
template <class T>
|
template <class T> class UseSimple {
|
||||||
class UseSimple {
|
|
||||||
public:
|
public:
|
||||||
UseSimple() {
|
UseSimple() { Interface::IsA<T, Interface::Simple>(); }
|
||||||
Interface::IsA<T,Interface::Simple>();
|
|
||||||
}
|
|
||||||
};
|
};
|
||||||
|
|
||||||
|
} // namespace uLib
|
||||||
|
|
||||||
|
int main() {
|
||||||
|
BEGIN_TESTING(Static Interface);
|
||||||
|
|
||||||
|
uLib::UseTest<uLib::Test> u;
|
||||||
|
|
||||||
|
uLib::UseSimple<uLib::Simple> s;
|
||||||
|
|
||||||
|
END_TESTING;
|
||||||
}
|
}
|
||||||
|
|
||||||
int main()
|
|
||||||
{
|
|
||||||
BEGIN_TESTING(Static Interface);
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
uLib::UseTest<uLib::Test> u;
|
|
||||||
|
|
||||||
uLib::UseSimple<uLib::Simple> s;
|
|
||||||
|
|
||||||
END_TESTING;
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|||||||
97
src/Core/testing/VectorMetaAllocatorTest.cpp
Normal file
97
src/Core/testing/VectorMetaAllocatorTest.cpp
Normal file
@@ -0,0 +1,97 @@
|
|||||||
|
/*//////////////////////////////////////////////////////////////////////////////
|
||||||
|
// CMT Cosmic Muon Tomography project //////////////////////////////////////////
|
||||||
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
|
|
||||||
|
Copyright (c) 2014, Universita' degli Studi di Padova, INFN sez. di Padova
|
||||||
|
All rights reserved
|
||||||
|
|
||||||
|
Authors: Andrea Rigoni Garola < andrea.rigoni@pd.infn.it >
|
||||||
|
|
||||||
|
//////////////////////////////////////////////////////////////////////////////*/
|
||||||
|
|
||||||
|
#include "testing-prototype.h"
|
||||||
|
#include <Core/Vector.h>
|
||||||
|
|
||||||
|
#ifdef USE_CUDA
|
||||||
|
#include <thrust/device_ptr.h>
|
||||||
|
#include <thrust/transform.h>
|
||||||
|
|
||||||
|
struct DoubleFunctor {
|
||||||
|
__host__ __device__ int operator()(int x) const { return x * 2; }
|
||||||
|
};
|
||||||
|
#endif
|
||||||
|
|
||||||
|
int main() {
|
||||||
|
BEGIN_TESTING(VectorMetaAllocator);
|
||||||
|
|
||||||
|
uLib::Vector<int> v;
|
||||||
|
std::cout << "Pushing elements...\n";
|
||||||
|
v << 1, 2, 3, 4, 5;
|
||||||
|
|
||||||
|
std::cout << "Initial RAM contents: ";
|
||||||
|
for (size_t i = 0; i < v.size(); ++i) {
|
||||||
|
std::cout << v[i] << " ";
|
||||||
|
if (v[i] != (int)(i + 1)) {
|
||||||
|
std::cout << "\nError: Value mismatch at index " << i << "\n";
|
||||||
|
exit(1);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
std::cout << "\n";
|
||||||
|
|
||||||
|
#ifdef USE_CUDA
|
||||||
|
std::cout << "Moving to VRAM...\n";
|
||||||
|
v.MoveToVRAM();
|
||||||
|
|
||||||
|
int *vram_ptr = v.GetVRAMData();
|
||||||
|
if (vram_ptr) {
|
||||||
|
std::cout << "Successfully obtained VRAM pointer: " << vram_ptr << "\n";
|
||||||
|
} else {
|
||||||
|
std::cout << "Error: Failed to obtain VRAM pointer!\n";
|
||||||
|
exit(1);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Verify DeviceData() matches GetVRAMData()
|
||||||
|
{
|
||||||
|
thrust::device_ptr<int> dev_ptr = v.DeviceData();
|
||||||
|
if (dev_ptr.get() != vram_ptr) {
|
||||||
|
std::cout << "Error: DeviceData() does not match GetVRAMData()!\n";
|
||||||
|
exit(1);
|
||||||
|
}
|
||||||
|
std::cout << "DeviceData() matches GetVRAMData(). OK\n";
|
||||||
|
}
|
||||||
|
|
||||||
|
// Use thrust::transform via DeviceBegin()/DeviceEnd() to double all elements
|
||||||
|
// on device
|
||||||
|
std::cout << "Doubling elements on device via thrust::transform...\n";
|
||||||
|
thrust::transform(v.DeviceBegin(), v.DeviceEnd(), v.DeviceBegin(),
|
||||||
|
DoubleFunctor{});
|
||||||
|
|
||||||
|
std::cout << "Moving back to RAM...\n";
|
||||||
|
v.MoveToRAM();
|
||||||
|
|
||||||
|
std::cout << "RAM contents after VRAM trip + thrust transform: ";
|
||||||
|
for (size_t i = 0; i < v.size(); ++i) {
|
||||||
|
std::cout << v[i] << " ";
|
||||||
|
if (v[i] != (int)((i + 1) * 2)) {
|
||||||
|
std::cout << "\nError: Data corrupted after RAM->VRAM->thrust->RAM trip "
|
||||||
|
"at index "
|
||||||
|
<< i << "\n";
|
||||||
|
exit(1);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
std::cout << "\n";
|
||||||
|
#else
|
||||||
|
std::cout << "USE_CUDA not defined, skipping VRAM tests.\n";
|
||||||
|
#endif
|
||||||
|
|
||||||
|
std::cout << "Scaling vector...\n";
|
||||||
|
for (size_t i = 0; i < v.size(); ++i)
|
||||||
|
v[i] *= 10;
|
||||||
|
|
||||||
|
std::cout << "Final contents: ";
|
||||||
|
for (size_t i = 0; i < v.size(); ++i)
|
||||||
|
std::cout << v[i] << " ";
|
||||||
|
std::cout << "\n";
|
||||||
|
|
||||||
|
END_TESTING;
|
||||||
|
}
|
||||||
@@ -23,62 +23,47 @@
|
|||||||
|
|
||||||
//////////////////////////////////////////////////////////////////////////////*/
|
//////////////////////////////////////////////////////////////////////////////*/
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
#include <Core/Vector.h>
|
|
||||||
#include "testing-prototype.h"
|
#include "testing-prototype.h"
|
||||||
|
#include <Core/Vector.h>
|
||||||
|
|
||||||
#include <algorithm>
|
#include <algorithm>
|
||||||
|
|
||||||
template < typename T >
|
template <typename T> struct __Cmp {
|
||||||
struct __Cmp {
|
bool operator()(const T &data, const float value) { return data <= value; }
|
||||||
bool operator()(const T &data, const float value) {
|
|
||||||
return data <= value;
|
|
||||||
}
|
|
||||||
};
|
};
|
||||||
|
|
||||||
|
template <typename _Tp, typename _CmpT>
|
||||||
|
inline const unsigned long VectorSplice(const _Tp &_it, const _Tp &_end,
|
||||||
|
const float value, _CmpT _comp) {
|
||||||
|
|
||||||
|
_Tp it = _it;
|
||||||
|
_Tp end = _end - 1;
|
||||||
template<typename _Tp, typename _CmpT>
|
for (it; it != end;) {
|
||||||
inline const unsigned long
|
if (_comp(*it, value))
|
||||||
VectorSplice(const _Tp &_it, const _Tp &_end, const float value, _CmpT _comp)
|
it++;
|
||||||
{
|
else if (_comp(*end, value)) {
|
||||||
|
std::swap(*it, *end--);
|
||||||
_Tp it = _it;
|
} else
|
||||||
_Tp end = _end-1;
|
--end;
|
||||||
for(it; it != end; )
|
}
|
||||||
{
|
return it - _it;
|
||||||
if ( _comp(*it, value) ) it++;
|
|
||||||
else if( _comp(*end, value) )
|
|
||||||
{
|
|
||||||
std::swap(*it,*end--);
|
|
||||||
}
|
|
||||||
else --end;
|
|
||||||
}
|
|
||||||
return it - _it;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
int main() {
|
||||||
|
BEGIN_TESTING(Vector);
|
||||||
|
|
||||||
int main()
|
uLib::Vector<float> v;
|
||||||
{
|
v << 5, 4, 3, 2, 6, 1, 2, 3, 65, 7, 32, 23, 4, 3, 45, 4, 34, 3, 4, 4, 3, 3, 4,
|
||||||
BEGIN_TESTING(Vector);
|
2, 2, 3;
|
||||||
|
|
||||||
uLib::Vector<float> v;
|
int id = ::VectorSplice(v.begin(), v.end(), 3, __Cmp<float>());
|
||||||
v << 5,4,3,2,6,1,2,3,65,7,32,23,4,3,45,4,34,3,4,4,3,3,4,2,2,3;
|
|
||||||
|
|
||||||
|
std::cout << "id: " << id << "\n";
|
||||||
|
std::cout << "vector: ";
|
||||||
|
for (uLib::Vector<float>::Iterator it = v.begin(); it != v.end(); it++)
|
||||||
|
std::cout << *it << " ";
|
||||||
|
std::cout << std::endl;
|
||||||
|
// std::sort(v.begin(),v.end(),LT<float>());
|
||||||
|
|
||||||
|
END_TESTING;
|
||||||
int id = VectorSplice(v.begin(),v.end(),3,__Cmp<float>());
|
|
||||||
|
|
||||||
std::cout << "id: " << id << "\n";
|
|
||||||
std::cout << "vector: ";
|
|
||||||
for(uLib::Vector<float>::Iterator it = v.begin(); it!=v.end(); it++)
|
|
||||||
std::cout << *it <<" ";
|
|
||||||
std::cout << std::endl;
|
|
||||||
// std::sort(v.begin(),v.end(),LT<float>());
|
|
||||||
|
|
||||||
|
|
||||||
END_TESTING;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -5,3 +5,8 @@ set(ULIB_SELECTED_MODULES ${ULIB_SELECTED_MODULES} Detectors PARENT_SCOPE)
|
|||||||
|
|
||||||
install(FILES ${HEADERS}
|
install(FILES ${HEADERS}
|
||||||
DESTINATION ${INSTALL_INC_DIR}/Detectors)
|
DESTINATION ${INSTALL_INC_DIR}/Detectors)
|
||||||
|
|
||||||
|
if(BUILD_TESTING)
|
||||||
|
include(uLibTargetMacros)
|
||||||
|
add_subdirectory(testing)
|
||||||
|
endif()
|
||||||
|
|||||||
@@ -1,18 +1,15 @@
|
|||||||
# TESTS
|
# TESTS
|
||||||
set( TESTS
|
set( TESTS
|
||||||
GDMLSolidTest
|
# GDMLSolidTest
|
||||||
HierarchicalEncodingTest
|
HierarchicalEncodingTest
|
||||||
)
|
)
|
||||||
|
|
||||||
#set(LIBRARIES
|
set(LIBRARIES
|
||||||
# ${PACKAGE_LIBPREFIX}Core
|
${PACKAGE_LIBPREFIX}Core
|
||||||
# ${PACKAGE_LIBPREFIX}Math
|
${PACKAGE_LIBPREFIX}Math
|
||||||
# ${PACKAGE_LIBPREFIX}Detectors
|
Boost::serialization
|
||||||
# ${Boost_SERIALIZATION_LIBRARY}
|
Boost::program_options
|
||||||
# ${Boost_SIGNALS_LIBRARY}
|
Eigen3::Eigen
|
||||||
# ${Boost_PROGRAM_OPTIONS_LIBRARY}
|
${ROOT_LIBRARIES}
|
||||||
# ${Eigen_LIBRARY}
|
)
|
||||||
# ${Geant4_LIBRARIES}
|
uLib_add_tests(Detectors)
|
||||||
# ${ROOT_LIBRARIES}
|
|
||||||
#)
|
|
||||||
uLib_add_tests(${uLib-module})
|
|
||||||
|
|||||||
@@ -1,35 +0,0 @@
|
|||||||
# SUBDIRS = .
|
|
||||||
|
|
||||||
include $(top_srcdir)/Common.am
|
|
||||||
|
|
||||||
DISTSOURCES = vtkviewport.cpp main.cpp
|
|
||||||
DISTHEADERS_MOC =
|
|
||||||
DISTHEADERS_NO_MOC =
|
|
||||||
FORMS = vtkviewport.ui
|
|
||||||
|
|
||||||
FORMHEADERS = $(FORMS:.ui=.h)
|
|
||||||
MOC_CC = $(FORMS:.ui=.moc.cpp) $(DISTHEADERS_MOC:.h=.moc.cpp)
|
|
||||||
|
|
||||||
bin_PROGRAMS = QTVtkViewport
|
|
||||||
|
|
||||||
BUILT_SOURCES = $(FORMHEADERS) $(MOC_CC)
|
|
||||||
CLEANFILES = $(BUILT_SOURCES)
|
|
||||||
|
|
||||||
EXTRA_DIST = $(FORMS)
|
|
||||||
QTVtkViewport_SOURCES = $(DISTSOURCES) $(DISTHEADERS_MOC) $(DISTHEADERS_NO_MOC)
|
|
||||||
nodist_QTVtkViewport_SOURCES = $(MOC_CC)
|
|
||||||
|
|
||||||
QTVtkViewport_LDADD = $(top_srcdir)/libmutom.la
|
|
||||||
|
|
||||||
|
|
||||||
.ui.h: $(FORMS)
|
|
||||||
$(UIC) -o ui_$@ $<
|
|
||||||
|
|
||||||
.ui.hpp: $(FORMS_HPP)
|
|
||||||
$(UIC) -o $@ $<
|
|
||||||
|
|
||||||
.h.moc.cpp:
|
|
||||||
$(MOC) -o $@ $<
|
|
||||||
|
|
||||||
|
|
||||||
SUFFIXES = .h .ui .moc.cpp
|
|
||||||
@@ -1,18 +0,0 @@
|
|||||||
#-------------------------------------------------
|
|
||||||
#
|
|
||||||
# Project created by QtCreator 2012-08-30T18:59:53
|
|
||||||
#
|
|
||||||
#-------------------------------------------------
|
|
||||||
|
|
||||||
QT += core gui
|
|
||||||
|
|
||||||
TARGET = QVTKViewport2
|
|
||||||
TEMPLATE = app
|
|
||||||
|
|
||||||
|
|
||||||
SOURCES += main.cpp\
|
|
||||||
vtkviewport.cpp
|
|
||||||
|
|
||||||
HEADERS += vtkviewport.h
|
|
||||||
|
|
||||||
FORMS += vtkviewport.ui
|
|
||||||
@@ -1,453 +0,0 @@
|
|||||||
<?xml version="1.0" encoding="UTF-8"?>
|
|
||||||
<!DOCTYPE QtCreatorProject>
|
|
||||||
<!-- Written by Qt Creator 2.4.1, 2012-09-03T09:43:56. -->
|
|
||||||
<qtcreator>
|
|
||||||
<data>
|
|
||||||
<variable>ProjectExplorer.Project.ActiveTarget</variable>
|
|
||||||
<value type="int">0</value>
|
|
||||||
</data>
|
|
||||||
<data>
|
|
||||||
<variable>ProjectExplorer.Project.EditorSettings</variable>
|
|
||||||
<valuemap type="QVariantMap">
|
|
||||||
<value type="bool" key="EditorConfiguration.AutoIndent">true</value>
|
|
||||||
<value type="bool" key="EditorConfiguration.AutoSpacesForTabs">false</value>
|
|
||||||
<valuemap type="QVariantMap" key="EditorConfiguration.CodeStyle.0">
|
|
||||||
<value type="QString" key="language">Cpp</value>
|
|
||||||
<valuemap type="QVariantMap" key="value">
|
|
||||||
<value type="QString" key="CurrentPreferences">CppGlobal</value>
|
|
||||||
</valuemap>
|
|
||||||
</valuemap>
|
|
||||||
<valuemap type="QVariantMap" key="EditorConfiguration.CodeStyle.1">
|
|
||||||
<value type="QString" key="language">QmlJS</value>
|
|
||||||
<valuemap type="QVariantMap" key="value">
|
|
||||||
<value type="QString" key="CurrentPreferences">QmlJSGlobal</value>
|
|
||||||
</valuemap>
|
|
||||||
</valuemap>
|
|
||||||
<value type="int" key="EditorConfiguration.CodeStyle.Count">2</value>
|
|
||||||
<value type="QByteArray" key="EditorConfiguration.Codec">System</value>
|
|
||||||
<value type="bool" key="EditorConfiguration.ConstrainTooltips">false</value>
|
|
||||||
<value type="int" key="EditorConfiguration.IndentSize">4</value>
|
|
||||||
<value type="bool" key="EditorConfiguration.MouseNavigation">true</value>
|
|
||||||
<value type="int" key="EditorConfiguration.PaddingMode">1</value>
|
|
||||||
<value type="bool" key="EditorConfiguration.ScrollWheelZooming">true</value>
|
|
||||||
<value type="int" key="EditorConfiguration.SmartBackspaceBehavior">0</value>
|
|
||||||
<value type="bool" key="EditorConfiguration.SpacesForTabs">true</value>
|
|
||||||
<value type="int" key="EditorConfiguration.TabKeyBehavior">1</value>
|
|
||||||
<value type="int" key="EditorConfiguration.TabSize">8</value>
|
|
||||||
<value type="bool" key="EditorConfiguration.UseGlobal">true</value>
|
|
||||||
<value type="int" key="EditorConfiguration.Utf8BomBehavior">1</value>
|
|
||||||
<value type="bool" key="EditorConfiguration.addFinalNewLine">true</value>
|
|
||||||
<value type="bool" key="EditorConfiguration.cleanIndentation">true</value>
|
|
||||||
<value type="bool" key="EditorConfiguration.cleanWhitespace">true</value>
|
|
||||||
<value type="bool" key="EditorConfiguration.inEntireDocument">false</value>
|
|
||||||
</valuemap>
|
|
||||||
</data>
|
|
||||||
<data>
|
|
||||||
<variable>ProjectExplorer.Project.PluginSettings</variable>
|
|
||||||
<valuemap type="QVariantMap"/>
|
|
||||||
</data>
|
|
||||||
<data>
|
|
||||||
<variable>ProjectExplorer.Project.Target.0</variable>
|
|
||||||
<valuemap type="QVariantMap">
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DefaultDisplayName">Desktop</value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DisplayName">Desktop</value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.Id">Qt4ProjectManager.Target.DesktopTarget</value>
|
|
||||||
<value type="int" key="ProjectExplorer.Target.ActiveBuildConfiguration">0</value>
|
|
||||||
<value type="int" key="ProjectExplorer.Target.ActiveDeployConfiguration">0</value>
|
|
||||||
<value type="int" key="ProjectExplorer.Target.ActiveRunConfiguration">0</value>
|
|
||||||
<valuemap type="QVariantMap" key="ProjectExplorer.Target.BuildConfiguration.0">
|
|
||||||
<value type="QString" key="ProjectExplorer.BuildCOnfiguration.ToolChain">ProjectExplorer.ToolChain.Gcc:/usr/bin/g++.x86-linux-generic-elf-64bit.gdb</value>
|
|
||||||
<valuemap type="QVariantMap" key="ProjectExplorer.BuildConfiguration.BuildStepList.0">
|
|
||||||
<valuemap type="QVariantMap" key="ProjectExplorer.BuildStepList.Step.0">
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DefaultDisplayName">qmake</value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DisplayName"></value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.Id">QtProjectManager.QMakeBuildStep</value>
|
|
||||||
<value type="bool" key="QtProjectManager.QMakeBuildStep.LinkQmlDebuggingLibrary">false</value>
|
|
||||||
<value type="bool" key="QtProjectManager.QMakeBuildStep.LinkQmlDebuggingLibraryAuto">true</value>
|
|
||||||
<value type="QString" key="QtProjectManager.QMakeBuildStep.QMakeArguments"></value>
|
|
||||||
<value type="bool" key="QtProjectManager.QMakeBuildStep.QMakeForced">false</value>
|
|
||||||
</valuemap>
|
|
||||||
<valuemap type="QVariantMap" key="ProjectExplorer.BuildStepList.Step.1">
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DefaultDisplayName">Make</value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DisplayName"></value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.Id">Qt4ProjectManager.MakeStep</value>
|
|
||||||
<value type="bool" key="Qt4ProjectManager.MakeStep.Clean">false</value>
|
|
||||||
<value type="QString" key="Qt4ProjectManager.MakeStep.MakeArguments"></value>
|
|
||||||
<value type="QString" key="Qt4ProjectManager.MakeStep.MakeCommand"></value>
|
|
||||||
</valuemap>
|
|
||||||
<value type="int" key="ProjectExplorer.BuildStepList.StepsCount">2</value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DefaultDisplayName">Build</value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DisplayName"></value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.Id">ProjectExplorer.BuildSteps.Build</value>
|
|
||||||
</valuemap>
|
|
||||||
<valuemap type="QVariantMap" key="ProjectExplorer.BuildConfiguration.BuildStepList.1">
|
|
||||||
<valuemap type="QVariantMap" key="ProjectExplorer.BuildStepList.Step.0">
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DefaultDisplayName">Make</value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DisplayName"></value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.Id">Qt4ProjectManager.MakeStep</value>
|
|
||||||
<value type="bool" key="Qt4ProjectManager.MakeStep.Clean">true</value>
|
|
||||||
<value type="QString" key="Qt4ProjectManager.MakeStep.MakeArguments">clean</value>
|
|
||||||
<value type="QString" key="Qt4ProjectManager.MakeStep.MakeCommand"></value>
|
|
||||||
</valuemap>
|
|
||||||
<value type="int" key="ProjectExplorer.BuildStepList.StepsCount">1</value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DefaultDisplayName">Clean</value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DisplayName"></value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.Id">ProjectExplorer.BuildSteps.Clean</value>
|
|
||||||
</valuemap>
|
|
||||||
<value type="int" key="ProjectExplorer.BuildConfiguration.BuildStepListCount">2</value>
|
|
||||||
<value type="bool" key="ProjectExplorer.BuildConfiguration.ClearSystemEnvironment">false</value>
|
|
||||||
<valuelist type="QVariantList" key="ProjectExplorer.BuildConfiguration.UserEnvironmentChanges"/>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DefaultDisplayName">Desktop Qt 4.7.4 for GCC (Qt SDK) Release</value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DisplayName"></value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.Id">Qt4ProjectManager.Qt4BuildConfiguration</value>
|
|
||||||
<value type="int" key="Qt4ProjectManager.Qt4BuildConfiguration.BuildConfiguration">0</value>
|
|
||||||
<value type="QString" key="Qt4ProjectManager.Qt4BuildConfiguration.BuildDirectory">/home/andrea/devel/MuonTomography/mutom/uLib/trunk/src/Gui/Qt/QVTKViewport2</value>
|
|
||||||
<value type="int" key="Qt4ProjectManager.Qt4BuildConfiguration.QtVersionId">3</value>
|
|
||||||
<value type="bool" key="Qt4ProjectManager.Qt4BuildConfiguration.UseShadowBuild">false</value>
|
|
||||||
</valuemap>
|
|
||||||
<valuemap type="QVariantMap" key="ProjectExplorer.Target.BuildConfiguration.1">
|
|
||||||
<value type="QString" key="ProjectExplorer.BuildCOnfiguration.ToolChain">ProjectExplorer.ToolChain.Gcc:/usr/bin/g++.x86-linux-generic-elf-64bit.gdb</value>
|
|
||||||
<valuemap type="QVariantMap" key="ProjectExplorer.BuildConfiguration.BuildStepList.0">
|
|
||||||
<valuemap type="QVariantMap" key="ProjectExplorer.BuildStepList.Step.0">
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DefaultDisplayName">qmake</value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DisplayName"></value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.Id">QtProjectManager.QMakeBuildStep</value>
|
|
||||||
<value type="bool" key="QtProjectManager.QMakeBuildStep.LinkQmlDebuggingLibrary">false</value>
|
|
||||||
<value type="bool" key="QtProjectManager.QMakeBuildStep.LinkQmlDebuggingLibraryAuto">true</value>
|
|
||||||
<value type="QString" key="QtProjectManager.QMakeBuildStep.QMakeArguments"></value>
|
|
||||||
<value type="bool" key="QtProjectManager.QMakeBuildStep.QMakeForced">false</value>
|
|
||||||
</valuemap>
|
|
||||||
<valuemap type="QVariantMap" key="ProjectExplorer.BuildStepList.Step.1">
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DefaultDisplayName">Make</value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DisplayName"></value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.Id">Qt4ProjectManager.MakeStep</value>
|
|
||||||
<value type="bool" key="Qt4ProjectManager.MakeStep.Clean">false</value>
|
|
||||||
<value type="QString" key="Qt4ProjectManager.MakeStep.MakeArguments"></value>
|
|
||||||
<value type="QString" key="Qt4ProjectManager.MakeStep.MakeCommand"></value>
|
|
||||||
</valuemap>
|
|
||||||
<value type="int" key="ProjectExplorer.BuildStepList.StepsCount">2</value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DefaultDisplayName">Build</value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DisplayName"></value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.Id">ProjectExplorer.BuildSteps.Build</value>
|
|
||||||
</valuemap>
|
|
||||||
<valuemap type="QVariantMap" key="ProjectExplorer.BuildConfiguration.BuildStepList.1">
|
|
||||||
<valuemap type="QVariantMap" key="ProjectExplorer.BuildStepList.Step.0">
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DefaultDisplayName">Make</value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DisplayName"></value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.Id">Qt4ProjectManager.MakeStep</value>
|
|
||||||
<value type="bool" key="Qt4ProjectManager.MakeStep.Clean">true</value>
|
|
||||||
<value type="QString" key="Qt4ProjectManager.MakeStep.MakeArguments">clean</value>
|
|
||||||
<value type="QString" key="Qt4ProjectManager.MakeStep.MakeCommand"></value>
|
|
||||||
</valuemap>
|
|
||||||
<value type="int" key="ProjectExplorer.BuildStepList.StepsCount">1</value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DefaultDisplayName">Clean</value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DisplayName"></value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.Id">ProjectExplorer.BuildSteps.Clean</value>
|
|
||||||
</valuemap>
|
|
||||||
<value type="int" key="ProjectExplorer.BuildConfiguration.BuildStepListCount">2</value>
|
|
||||||
<value type="bool" key="ProjectExplorer.BuildConfiguration.ClearSystemEnvironment">false</value>
|
|
||||||
<valuelist type="QVariantList" key="ProjectExplorer.BuildConfiguration.UserEnvironmentChanges"/>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DefaultDisplayName">Desktop Qt 4.7.4 for GCC (Qt SDK) Debug</value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DisplayName"></value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.Id">Qt4ProjectManager.Qt4BuildConfiguration</value>
|
|
||||||
<value type="int" key="Qt4ProjectManager.Qt4BuildConfiguration.BuildConfiguration">2</value>
|
|
||||||
<value type="QString" key="Qt4ProjectManager.Qt4BuildConfiguration.BuildDirectory">/home/andrea/devel/MuonTomography/mutom/uLib/trunk/src/Gui/Qt/QVTKViewport2</value>
|
|
||||||
<value type="int" key="Qt4ProjectManager.Qt4BuildConfiguration.QtVersionId">3</value>
|
|
||||||
<value type="bool" key="Qt4ProjectManager.Qt4BuildConfiguration.UseShadowBuild">false</value>
|
|
||||||
</valuemap>
|
|
||||||
<valuemap type="QVariantMap" key="ProjectExplorer.Target.BuildConfiguration.2">
|
|
||||||
<value type="QString" key="ProjectExplorer.BuildCOnfiguration.ToolChain">ProjectExplorer.ToolChain.Gcc:/usr/bin/g++.x86-linux-generic-elf-64bit.gdb</value>
|
|
||||||
<valuemap type="QVariantMap" key="ProjectExplorer.BuildConfiguration.BuildStepList.0">
|
|
||||||
<valuemap type="QVariantMap" key="ProjectExplorer.BuildStepList.Step.0">
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DefaultDisplayName">qmake</value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DisplayName"></value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.Id">QtProjectManager.QMakeBuildStep</value>
|
|
||||||
<value type="bool" key="QtProjectManager.QMakeBuildStep.LinkQmlDebuggingLibrary">false</value>
|
|
||||||
<value type="bool" key="QtProjectManager.QMakeBuildStep.LinkQmlDebuggingLibraryAuto">true</value>
|
|
||||||
<value type="QString" key="QtProjectManager.QMakeBuildStep.QMakeArguments"></value>
|
|
||||||
<value type="bool" key="QtProjectManager.QMakeBuildStep.QMakeForced">false</value>
|
|
||||||
</valuemap>
|
|
||||||
<valuemap type="QVariantMap" key="ProjectExplorer.BuildStepList.Step.1">
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DefaultDisplayName">Make</value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DisplayName"></value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.Id">Qt4ProjectManager.MakeStep</value>
|
|
||||||
<value type="bool" key="Qt4ProjectManager.MakeStep.Clean">false</value>
|
|
||||||
<value type="QString" key="Qt4ProjectManager.MakeStep.MakeArguments"></value>
|
|
||||||
<value type="QString" key="Qt4ProjectManager.MakeStep.MakeCommand"></value>
|
|
||||||
</valuemap>
|
|
||||||
<value type="int" key="ProjectExplorer.BuildStepList.StepsCount">2</value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DefaultDisplayName">Build</value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DisplayName"></value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.Id">ProjectExplorer.BuildSteps.Build</value>
|
|
||||||
</valuemap>
|
|
||||||
<valuemap type="QVariantMap" key="ProjectExplorer.BuildConfiguration.BuildStepList.1">
|
|
||||||
<valuemap type="QVariantMap" key="ProjectExplorer.BuildStepList.Step.0">
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DefaultDisplayName">Make</value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DisplayName"></value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.Id">Qt4ProjectManager.MakeStep</value>
|
|
||||||
<value type="bool" key="Qt4ProjectManager.MakeStep.Clean">true</value>
|
|
||||||
<value type="QString" key="Qt4ProjectManager.MakeStep.MakeArguments">clean</value>
|
|
||||||
<value type="QString" key="Qt4ProjectManager.MakeStep.MakeCommand"></value>
|
|
||||||
</valuemap>
|
|
||||||
<value type="int" key="ProjectExplorer.BuildStepList.StepsCount">1</value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DefaultDisplayName">Clean</value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DisplayName"></value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.Id">ProjectExplorer.BuildSteps.Clean</value>
|
|
||||||
</valuemap>
|
|
||||||
<value type="int" key="ProjectExplorer.BuildConfiguration.BuildStepListCount">2</value>
|
|
||||||
<value type="bool" key="ProjectExplorer.BuildConfiguration.ClearSystemEnvironment">false</value>
|
|
||||||
<valuelist type="QVariantList" key="ProjectExplorer.BuildConfiguration.UserEnvironmentChanges"/>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DefaultDisplayName">Qt 4.7.0 in PATH (System) Release</value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DisplayName"></value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.Id">Qt4ProjectManager.Qt4BuildConfiguration</value>
|
|
||||||
<value type="int" key="Qt4ProjectManager.Qt4BuildConfiguration.BuildConfiguration">0</value>
|
|
||||||
<value type="QString" key="Qt4ProjectManager.Qt4BuildConfiguration.BuildDirectory">/home/andrea/devel/MuonTomography/mutom/uLib/trunk/src/Gui/Qt/QVTKViewport2</value>
|
|
||||||
<value type="int" key="Qt4ProjectManager.Qt4BuildConfiguration.QtVersionId">4</value>
|
|
||||||
<value type="bool" key="Qt4ProjectManager.Qt4BuildConfiguration.UseShadowBuild">false</value>
|
|
||||||
</valuemap>
|
|
||||||
<valuemap type="QVariantMap" key="ProjectExplorer.Target.BuildConfiguration.3">
|
|
||||||
<value type="QString" key="ProjectExplorer.BuildCOnfiguration.ToolChain">ProjectExplorer.ToolChain.Gcc:/usr/bin/g++.x86-linux-generic-elf-64bit.gdb</value>
|
|
||||||
<valuemap type="QVariantMap" key="ProjectExplorer.BuildConfiguration.BuildStepList.0">
|
|
||||||
<valuemap type="QVariantMap" key="ProjectExplorer.BuildStepList.Step.0">
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DefaultDisplayName">qmake</value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DisplayName"></value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.Id">QtProjectManager.QMakeBuildStep</value>
|
|
||||||
<value type="bool" key="QtProjectManager.QMakeBuildStep.LinkQmlDebuggingLibrary">false</value>
|
|
||||||
<value type="bool" key="QtProjectManager.QMakeBuildStep.LinkQmlDebuggingLibraryAuto">true</value>
|
|
||||||
<value type="QString" key="QtProjectManager.QMakeBuildStep.QMakeArguments"></value>
|
|
||||||
<value type="bool" key="QtProjectManager.QMakeBuildStep.QMakeForced">false</value>
|
|
||||||
</valuemap>
|
|
||||||
<valuemap type="QVariantMap" key="ProjectExplorer.BuildStepList.Step.1">
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DefaultDisplayName">Make</value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DisplayName"></value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.Id">Qt4ProjectManager.MakeStep</value>
|
|
||||||
<value type="bool" key="Qt4ProjectManager.MakeStep.Clean">false</value>
|
|
||||||
<value type="QString" key="Qt4ProjectManager.MakeStep.MakeArguments"></value>
|
|
||||||
<value type="QString" key="Qt4ProjectManager.MakeStep.MakeCommand"></value>
|
|
||||||
</valuemap>
|
|
||||||
<value type="int" key="ProjectExplorer.BuildStepList.StepsCount">2</value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DefaultDisplayName">Build</value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DisplayName"></value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.Id">ProjectExplorer.BuildSteps.Build</value>
|
|
||||||
</valuemap>
|
|
||||||
<valuemap type="QVariantMap" key="ProjectExplorer.BuildConfiguration.BuildStepList.1">
|
|
||||||
<valuemap type="QVariantMap" key="ProjectExplorer.BuildStepList.Step.0">
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DefaultDisplayName">Make</value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DisplayName"></value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.Id">Qt4ProjectManager.MakeStep</value>
|
|
||||||
<value type="bool" key="Qt4ProjectManager.MakeStep.Clean">true</value>
|
|
||||||
<value type="QString" key="Qt4ProjectManager.MakeStep.MakeArguments">clean</value>
|
|
||||||
<value type="QString" key="Qt4ProjectManager.MakeStep.MakeCommand"></value>
|
|
||||||
</valuemap>
|
|
||||||
<value type="int" key="ProjectExplorer.BuildStepList.StepsCount">1</value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DefaultDisplayName">Clean</value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DisplayName"></value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.Id">ProjectExplorer.BuildSteps.Clean</value>
|
|
||||||
</valuemap>
|
|
||||||
<value type="int" key="ProjectExplorer.BuildConfiguration.BuildStepListCount">2</value>
|
|
||||||
<value type="bool" key="ProjectExplorer.BuildConfiguration.ClearSystemEnvironment">false</value>
|
|
||||||
<valuelist type="QVariantList" key="ProjectExplorer.BuildConfiguration.UserEnvironmentChanges"/>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DefaultDisplayName">Qt 4.7.0 in PATH (System) Debug</value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DisplayName"></value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.Id">Qt4ProjectManager.Qt4BuildConfiguration</value>
|
|
||||||
<value type="int" key="Qt4ProjectManager.Qt4BuildConfiguration.BuildConfiguration">2</value>
|
|
||||||
<value type="QString" key="Qt4ProjectManager.Qt4BuildConfiguration.BuildDirectory">/home/andrea/devel/MuonTomography/mutom/uLib/trunk/src/Gui/Qt/QVTKViewport2</value>
|
|
||||||
<value type="int" key="Qt4ProjectManager.Qt4BuildConfiguration.QtVersionId">4</value>
|
|
||||||
<value type="bool" key="Qt4ProjectManager.Qt4BuildConfiguration.UseShadowBuild">false</value>
|
|
||||||
</valuemap>
|
|
||||||
<valuemap type="QVariantMap" key="ProjectExplorer.Target.BuildConfiguration.4">
|
|
||||||
<value type="QString" key="ProjectExplorer.BuildCOnfiguration.ToolChain">ProjectExplorer.ToolChain.Gcc:/usr/bin/g++.x86-linux-generic-elf-64bit.gdb</value>
|
|
||||||
<valuemap type="QVariantMap" key="ProjectExplorer.BuildConfiguration.BuildStepList.0">
|
|
||||||
<valuemap type="QVariantMap" key="ProjectExplorer.BuildStepList.Step.0">
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DefaultDisplayName">qmake</value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DisplayName"></value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.Id">QtProjectManager.QMakeBuildStep</value>
|
|
||||||
<value type="bool" key="QtProjectManager.QMakeBuildStep.LinkQmlDebuggingLibrary">false</value>
|
|
||||||
<value type="bool" key="QtProjectManager.QMakeBuildStep.LinkQmlDebuggingLibraryAuto">true</value>
|
|
||||||
<value type="QString" key="QtProjectManager.QMakeBuildStep.QMakeArguments"></value>
|
|
||||||
<value type="bool" key="QtProjectManager.QMakeBuildStep.QMakeForced">false</value>
|
|
||||||
</valuemap>
|
|
||||||
<valuemap type="QVariantMap" key="ProjectExplorer.BuildStepList.Step.1">
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DefaultDisplayName">Make</value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DisplayName"></value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.Id">Qt4ProjectManager.MakeStep</value>
|
|
||||||
<value type="bool" key="Qt4ProjectManager.MakeStep.Clean">false</value>
|
|
||||||
<value type="QString" key="Qt4ProjectManager.MakeStep.MakeArguments"></value>
|
|
||||||
<value type="QString" key="Qt4ProjectManager.MakeStep.MakeCommand"></value>
|
|
||||||
</valuemap>
|
|
||||||
<value type="int" key="ProjectExplorer.BuildStepList.StepsCount">2</value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DefaultDisplayName">Build</value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DisplayName"></value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.Id">ProjectExplorer.BuildSteps.Build</value>
|
|
||||||
</valuemap>
|
|
||||||
<valuemap type="QVariantMap" key="ProjectExplorer.BuildConfiguration.BuildStepList.1">
|
|
||||||
<valuemap type="QVariantMap" key="ProjectExplorer.BuildStepList.Step.0">
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DefaultDisplayName">Make</value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DisplayName"></value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.Id">Qt4ProjectManager.MakeStep</value>
|
|
||||||
<value type="bool" key="Qt4ProjectManager.MakeStep.Clean">true</value>
|
|
||||||
<value type="QString" key="Qt4ProjectManager.MakeStep.MakeArguments">clean</value>
|
|
||||||
<value type="QString" key="Qt4ProjectManager.MakeStep.MakeCommand"></value>
|
|
||||||
</valuemap>
|
|
||||||
<value type="int" key="ProjectExplorer.BuildStepList.StepsCount">1</value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DefaultDisplayName">Clean</value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DisplayName"></value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.Id">ProjectExplorer.BuildSteps.Clean</value>
|
|
||||||
</valuemap>
|
|
||||||
<value type="int" key="ProjectExplorer.BuildConfiguration.BuildStepListCount">2</value>
|
|
||||||
<value type="bool" key="ProjectExplorer.BuildConfiguration.ClearSystemEnvironment">false</value>
|
|
||||||
<valuelist type="QVariantList" key="ProjectExplorer.BuildConfiguration.UserEnvironmentChanges"/>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DefaultDisplayName">Qt 4.7.0 (System) Release</value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DisplayName"></value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.Id">Qt4ProjectManager.Qt4BuildConfiguration</value>
|
|
||||||
<value type="int" key="Qt4ProjectManager.Qt4BuildConfiguration.BuildConfiguration">0</value>
|
|
||||||
<value type="QString" key="Qt4ProjectManager.Qt4BuildConfiguration.BuildDirectory">/home/andrea/devel/MuonTomography/mutom/uLib/trunk/src/Gui/Qt/QVTKViewport2</value>
|
|
||||||
<value type="int" key="Qt4ProjectManager.Qt4BuildConfiguration.QtVersionId">5</value>
|
|
||||||
<value type="bool" key="Qt4ProjectManager.Qt4BuildConfiguration.UseShadowBuild">false</value>
|
|
||||||
</valuemap>
|
|
||||||
<valuemap type="QVariantMap" key="ProjectExplorer.Target.BuildConfiguration.5">
|
|
||||||
<value type="QString" key="ProjectExplorer.BuildCOnfiguration.ToolChain">ProjectExplorer.ToolChain.Gcc:/usr/bin/g++.x86-linux-generic-elf-64bit.gdb</value>
|
|
||||||
<valuemap type="QVariantMap" key="ProjectExplorer.BuildConfiguration.BuildStepList.0">
|
|
||||||
<valuemap type="QVariantMap" key="ProjectExplorer.BuildStepList.Step.0">
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DefaultDisplayName">qmake</value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DisplayName"></value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.Id">QtProjectManager.QMakeBuildStep</value>
|
|
||||||
<value type="bool" key="QtProjectManager.QMakeBuildStep.LinkQmlDebuggingLibrary">false</value>
|
|
||||||
<value type="bool" key="QtProjectManager.QMakeBuildStep.LinkQmlDebuggingLibraryAuto">true</value>
|
|
||||||
<value type="QString" key="QtProjectManager.QMakeBuildStep.QMakeArguments"></value>
|
|
||||||
<value type="bool" key="QtProjectManager.QMakeBuildStep.QMakeForced">false</value>
|
|
||||||
</valuemap>
|
|
||||||
<valuemap type="QVariantMap" key="ProjectExplorer.BuildStepList.Step.1">
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DefaultDisplayName">Make</value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DisplayName"></value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.Id">Qt4ProjectManager.MakeStep</value>
|
|
||||||
<value type="bool" key="Qt4ProjectManager.MakeStep.Clean">false</value>
|
|
||||||
<value type="QString" key="Qt4ProjectManager.MakeStep.MakeArguments"></value>
|
|
||||||
<value type="QString" key="Qt4ProjectManager.MakeStep.MakeCommand"></value>
|
|
||||||
</valuemap>
|
|
||||||
<value type="int" key="ProjectExplorer.BuildStepList.StepsCount">2</value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DefaultDisplayName">Build</value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DisplayName"></value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.Id">ProjectExplorer.BuildSteps.Build</value>
|
|
||||||
</valuemap>
|
|
||||||
<valuemap type="QVariantMap" key="ProjectExplorer.BuildConfiguration.BuildStepList.1">
|
|
||||||
<valuemap type="QVariantMap" key="ProjectExplorer.BuildStepList.Step.0">
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DefaultDisplayName">Make</value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DisplayName"></value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.Id">Qt4ProjectManager.MakeStep</value>
|
|
||||||
<value type="bool" key="Qt4ProjectManager.MakeStep.Clean">true</value>
|
|
||||||
<value type="QString" key="Qt4ProjectManager.MakeStep.MakeArguments">clean</value>
|
|
||||||
<value type="QString" key="Qt4ProjectManager.MakeStep.MakeCommand"></value>
|
|
||||||
</valuemap>
|
|
||||||
<value type="int" key="ProjectExplorer.BuildStepList.StepsCount">1</value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DefaultDisplayName">Clean</value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DisplayName"></value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.Id">ProjectExplorer.BuildSteps.Clean</value>
|
|
||||||
</valuemap>
|
|
||||||
<value type="int" key="ProjectExplorer.BuildConfiguration.BuildStepListCount">2</value>
|
|
||||||
<value type="bool" key="ProjectExplorer.BuildConfiguration.ClearSystemEnvironment">false</value>
|
|
||||||
<valuelist type="QVariantList" key="ProjectExplorer.BuildConfiguration.UserEnvironmentChanges"/>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DefaultDisplayName">Qt 4.7.0 (System) Debug</value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DisplayName"></value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.Id">Qt4ProjectManager.Qt4BuildConfiguration</value>
|
|
||||||
<value type="int" key="Qt4ProjectManager.Qt4BuildConfiguration.BuildConfiguration">2</value>
|
|
||||||
<value type="QString" key="Qt4ProjectManager.Qt4BuildConfiguration.BuildDirectory">/home/andrea/devel/MuonTomography/mutom/uLib/trunk/src/Gui/Qt/QVTKViewport2</value>
|
|
||||||
<value type="int" key="Qt4ProjectManager.Qt4BuildConfiguration.QtVersionId">5</value>
|
|
||||||
<value type="bool" key="Qt4ProjectManager.Qt4BuildConfiguration.UseShadowBuild">false</value>
|
|
||||||
</valuemap>
|
|
||||||
<value type="int" key="ProjectExplorer.Target.BuildConfigurationCount">6</value>
|
|
||||||
<valuemap type="QVariantMap" key="ProjectExplorer.Target.DeployConfiguration.0">
|
|
||||||
<valuemap type="QVariantMap" key="ProjectExplorer.BuildConfiguration.BuildStepList.0">
|
|
||||||
<value type="int" key="ProjectExplorer.BuildStepList.StepsCount">0</value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DefaultDisplayName">Deploy</value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DisplayName"></value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.Id">ProjectExplorer.BuildSteps.Deploy</value>
|
|
||||||
</valuemap>
|
|
||||||
<value type="int" key="ProjectExplorer.BuildConfiguration.BuildStepListCount">1</value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DefaultDisplayName">No deployment</value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DisplayName"></value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.Id">ProjectExplorer.DefaultDeployConfiguration</value>
|
|
||||||
</valuemap>
|
|
||||||
<value type="int" key="ProjectExplorer.Target.DeployConfigurationCount">1</value>
|
|
||||||
<valuemap type="QVariantMap" key="ProjectExplorer.Target.RunConfiguration.0">
|
|
||||||
<value type="bool" key="Analyzer.Project.UseGlobal">true</value>
|
|
||||||
<value type="bool" key="Analyzer.Project.UseGlobal">true</value>
|
|
||||||
<valuelist type="QVariantList" key="Analyzer.Valgrind.AddedSuppressionFiles"/>
|
|
||||||
<valuelist type="QVariantList" key="Analyzer.Valgrind.AddedSuppressionFiles"/>
|
|
||||||
<value type="bool" key="Analyzer.Valgrind.Callgrind.CollectBusEvents">false</value>
|
|
||||||
<value type="bool" key="Analyzer.Valgrind.Callgrind.CollectBusEvents">false</value>
|
|
||||||
<value type="bool" key="Analyzer.Valgrind.Callgrind.CollectSystime">false</value>
|
|
||||||
<value type="bool" key="Analyzer.Valgrind.Callgrind.CollectSystime">false</value>
|
|
||||||
<value type="bool" key="Analyzer.Valgrind.Callgrind.EnableBranchSim">false</value>
|
|
||||||
<value type="bool" key="Analyzer.Valgrind.Callgrind.EnableBranchSim">false</value>
|
|
||||||
<value type="bool" key="Analyzer.Valgrind.Callgrind.EnableCacheSim">false</value>
|
|
||||||
<value type="bool" key="Analyzer.Valgrind.Callgrind.EnableCacheSim">false</value>
|
|
||||||
<value type="bool" key="Analyzer.Valgrind.Callgrind.EnableEventToolTips">true</value>
|
|
||||||
<value type="bool" key="Analyzer.Valgrind.Callgrind.EnableEventToolTips">true</value>
|
|
||||||
<value type="double" key="Analyzer.Valgrind.Callgrind.MinimumCostRatio">0.01</value>
|
|
||||||
<value type="double" key="Analyzer.Valgrind.Callgrind.MinimumCostRatio">0.01</value>
|
|
||||||
<value type="double" key="Analyzer.Valgrind.Callgrind.VisualisationMinimumCostRatio">10</value>
|
|
||||||
<value type="double" key="Analyzer.Valgrind.Callgrind.VisualisationMinimumCostRatio">10</value>
|
|
||||||
<value type="bool" key="Analyzer.Valgrind.FilterExternalIssues">false</value>
|
|
||||||
<value type="bool" key="Analyzer.Valgrind.FilterExternalIssues">false</value>
|
|
||||||
<value type="int" key="Analyzer.Valgrind.NumCallers">25</value>
|
|
||||||
<value type="int" key="Analyzer.Valgrind.NumCallers">25</value>
|
|
||||||
<valuelist type="QVariantList" key="Analyzer.Valgrind.RemovedSuppressionFiles"/>
|
|
||||||
<valuelist type="QVariantList" key="Analyzer.Valgrind.RemovedSuppressionFiles"/>
|
|
||||||
<value type="bool" key="Analyzer.Valgrind.TrackOrigins">true</value>
|
|
||||||
<value type="bool" key="Analyzer.Valgrind.TrackOrigins">true</value>
|
|
||||||
<value type="QString" key="Analyzer.Valgrind.ValgrindExecutable">valgrind</value>
|
|
||||||
<value type="QString" key="Analyzer.Valgrind.ValgrindExecutable">valgrind</value>
|
|
||||||
<valuelist type="QVariantList" key="Analyzer.Valgrind.VisibleErrorKinds">
|
|
||||||
<value type="int">2</value>
|
|
||||||
<value type="int">3</value>
|
|
||||||
<value type="int">4</value>
|
|
||||||
<value type="int">5</value>
|
|
||||||
<value type="int">6</value>
|
|
||||||
<value type="int">7</value>
|
|
||||||
<value type="int">8</value>
|
|
||||||
<value type="int">9</value>
|
|
||||||
<value type="int">10</value>
|
|
||||||
</valuelist>
|
|
||||||
<valuelist type="QVariantList" key="Analyzer.Valgrind.VisibleErrorKinds">
|
|
||||||
<value type="int">2</value>
|
|
||||||
<value type="int">3</value>
|
|
||||||
<value type="int">4</value>
|
|
||||||
<value type="int">5</value>
|
|
||||||
<value type="int">6</value>
|
|
||||||
<value type="int">7</value>
|
|
||||||
<value type="int">8</value>
|
|
||||||
<value type="int">9</value>
|
|
||||||
<value type="int">10</value>
|
|
||||||
</valuelist>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DefaultDisplayName">QVTKViewport2</value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.DisplayName"></value>
|
|
||||||
<value type="QString" key="ProjectExplorer.ProjectConfiguration.Id">Qt4ProjectManager.Qt4RunConfiguration</value>
|
|
||||||
<value type="int" key="Qt4ProjectManager.Qt4RunConfiguration.BaseEnvironmentBase">2</value>
|
|
||||||
<value type="QString" key="Qt4ProjectManager.Qt4RunConfiguration.CommandLineArguments"></value>
|
|
||||||
<value type="QString" key="Qt4ProjectManager.Qt4RunConfiguration.ProFile">QVTKViewport2.pro</value>
|
|
||||||
<value type="bool" key="Qt4ProjectManager.Qt4RunConfiguration.UseDyldImageSuffix">false</value>
|
|
||||||
<value type="bool" key="Qt4ProjectManager.Qt4RunConfiguration.UseTerminal">false</value>
|
|
||||||
<valuelist type="QVariantList" key="Qt4ProjectManager.Qt4RunConfiguration.UserEnvironmentChanges"/>
|
|
||||||
<value type="QString" key="Qt4ProjectManager.Qt4RunConfiguration.UserWorkingDirectory"></value>
|
|
||||||
<value type="uint" key="RunConfiguration.QmlDebugServerPort">3768</value>
|
|
||||||
<value type="bool" key="RunConfiguration.UseCppDebugger">true</value>
|
|
||||||
<value type="bool" key="RunConfiguration.UseQmlDebugger">false</value>
|
|
||||||
<value type="bool" key="RunConfiguration.UseQmlDebuggerAuto">false</value>
|
|
||||||
</valuemap>
|
|
||||||
<value type="int" key="ProjectExplorer.Target.RunConfigurationCount">1</value>
|
|
||||||
</valuemap>
|
|
||||||
</data>
|
|
||||||
<data>
|
|
||||||
<variable>ProjectExplorer.Project.TargetCount</variable>
|
|
||||||
<value type="int">1</value>
|
|
||||||
</data>
|
|
||||||
<data>
|
|
||||||
<variable>ProjectExplorer.Project.Updater.EnvironmentId</variable>
|
|
||||||
<value type="QString">{f8bb0047-7f6e-45df-9cc8-e746abebf883}</value>
|
|
||||||
</data>
|
|
||||||
<data>
|
|
||||||
<variable>ProjectExplorer.Project.Updater.FileVersion</variable>
|
|
||||||
<value type="int">10</value>
|
|
||||||
</data>
|
|
||||||
</qtcreator>
|
|
||||||
@@ -1,38 +0,0 @@
|
|||||||
/*//////////////////////////////////////////////////////////////////////////////
|
|
||||||
// CMT Cosmic Muon Tomography project //////////////////////////////////////////
|
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
|
||||||
|
|
||||||
Copyright (c) 2014, Universita' degli Studi di Padova, INFN sez. di Padova
|
|
||||||
All rights reserved
|
|
||||||
|
|
||||||
Authors: Andrea Rigoni Garola < andrea.rigoni@pd.infn.it >
|
|
||||||
|
|
||||||
------------------------------------------------------------------
|
|
||||||
This library is free software; you can redistribute it and/or
|
|
||||||
modify it under the terms of the GNU Lesser General Public
|
|
||||||
License as published by the Free Software Foundation; either
|
|
||||||
version 3.0 of the License, or (at your option) any later version.
|
|
||||||
|
|
||||||
This library is distributed in the hope that it will be useful,
|
|
||||||
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
||||||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
||||||
Lesser General Public License for more details.
|
|
||||||
|
|
||||||
You should have received a copy of the GNU Lesser General Public
|
|
||||||
License along with this library.
|
|
||||||
|
|
||||||
//////////////////////////////////////////////////////////////////////////////*/
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
#include <QtGui/QApplication>
|
|
||||||
#include "vtkviewport.h"
|
|
||||||
|
|
||||||
int main(int argc, char *argv[])
|
|
||||||
{
|
|
||||||
QApplication a(argc, argv);
|
|
||||||
VtkViewport w;
|
|
||||||
w.show();
|
|
||||||
|
|
||||||
return a.exec();
|
|
||||||
}
|
|
||||||
@@ -1,41 +0,0 @@
|
|||||||
/*//////////////////////////////////////////////////////////////////////////////
|
|
||||||
// CMT Cosmic Muon Tomography project //////////////////////////////////////////
|
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
|
||||||
|
|
||||||
Copyright (c) 2014, Universita' degli Studi di Padova, INFN sez. di Padova
|
|
||||||
All rights reserved
|
|
||||||
|
|
||||||
Authors: Andrea Rigoni Garola < andrea.rigoni@pd.infn.it >
|
|
||||||
|
|
||||||
------------------------------------------------------------------
|
|
||||||
This library is free software; you can redistribute it and/or
|
|
||||||
modify it under the terms of the GNU Lesser General Public
|
|
||||||
License as published by the Free Software Foundation; either
|
|
||||||
version 3.0 of the License, or (at your option) any later version.
|
|
||||||
|
|
||||||
This library is distributed in the hope that it will be useful,
|
|
||||||
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
||||||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
||||||
Lesser General Public License for more details.
|
|
||||||
|
|
||||||
You should have received a copy of the GNU Lesser General Public
|
|
||||||
License along with this library.
|
|
||||||
|
|
||||||
//////////////////////////////////////////////////////////////////////////////*/
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
#include "vtkviewport.h"
|
|
||||||
#include "ui_vtkviewport.h"
|
|
||||||
|
|
||||||
VtkViewport::VtkViewport(QWidget *parent) :
|
|
||||||
QWidget(parent),
|
|
||||||
ui(new Ui::VtkViewport)
|
|
||||||
{
|
|
||||||
ui->setupUi(this);
|
|
||||||
}
|
|
||||||
|
|
||||||
VtkViewport::~VtkViewport()
|
|
||||||
{
|
|
||||||
delete ui;
|
|
||||||
}
|
|
||||||
@@ -1,49 +0,0 @@
|
|||||||
/*//////////////////////////////////////////////////////////////////////////////
|
|
||||||
// CMT Cosmic Muon Tomography project //////////////////////////////////////////
|
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
|
||||||
|
|
||||||
Copyright (c) 2014, Universita' degli Studi di Padova, INFN sez. di Padova
|
|
||||||
All rights reserved
|
|
||||||
|
|
||||||
Authors: Andrea Rigoni Garola < andrea.rigoni@pd.infn.it >
|
|
||||||
|
|
||||||
------------------------------------------------------------------
|
|
||||||
This library is free software; you can redistribute it and/or
|
|
||||||
modify it under the terms of the GNU Lesser General Public
|
|
||||||
License as published by the Free Software Foundation; either
|
|
||||||
version 3.0 of the License, or (at your option) any later version.
|
|
||||||
|
|
||||||
This library is distributed in the hope that it will be useful,
|
|
||||||
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
||||||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
||||||
Lesser General Public License for more details.
|
|
||||||
|
|
||||||
You should have received a copy of the GNU Lesser General Public
|
|
||||||
License along with this library.
|
|
||||||
|
|
||||||
//////////////////////////////////////////////////////////////////////////////*/
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
#ifndef VTKVIEWPORT_H
|
|
||||||
#define VTKVIEWPORT_H
|
|
||||||
|
|
||||||
#include <QWidget>
|
|
||||||
|
|
||||||
namespace Ui {
|
|
||||||
class VtkViewport;
|
|
||||||
}
|
|
||||||
|
|
||||||
class VtkViewport : public QWidget
|
|
||||||
{
|
|
||||||
Q_OBJECT
|
|
||||||
|
|
||||||
public:
|
|
||||||
explicit VtkViewport(QWidget *parent = 0);
|
|
||||||
~VtkViewport();
|
|
||||||
|
|
||||||
private:
|
|
||||||
Ui::VtkViewport *ui;
|
|
||||||
};
|
|
||||||
|
|
||||||
#endif // VTKVIEWPORT_H
|
|
||||||
@@ -1,32 +0,0 @@
|
|||||||
<?xml version="1.0" encoding="UTF-8"?>
|
|
||||||
<ui version="4.0">
|
|
||||||
<class>VtkViewport</class>
|
|
||||||
<widget class="QWidget" name="VtkViewport">
|
|
||||||
<property name="geometry">
|
|
||||||
<rect>
|
|
||||||
<x>0</x>
|
|
||||||
<y>0</y>
|
|
||||||
<width>580</width>
|
|
||||||
<height>536</height>
|
|
||||||
</rect>
|
|
||||||
</property>
|
|
||||||
<property name="windowTitle">
|
|
||||||
<string>Form</string>
|
|
||||||
</property>
|
|
||||||
<widget class="QLabel" name="label">
|
|
||||||
<property name="geometry">
|
|
||||||
<rect>
|
|
||||||
<x>140</x>
|
|
||||||
<y>170</y>
|
|
||||||
<width>161</width>
|
|
||||||
<height>41</height>
|
|
||||||
</rect>
|
|
||||||
</property>
|
|
||||||
<property name="text">
|
|
||||||
<string>Hello World !</string>
|
|
||||||
</property>
|
|
||||||
</widget>
|
|
||||||
</widget>
|
|
||||||
<resources/>
|
|
||||||
<connections/>
|
|
||||||
</ui>
|
|
||||||
@@ -43,9 +43,15 @@ set(ULIB_SELECTED_MODULES ${ULIB_SELECTED_MODULES} Math PARENT_SCOPE)
|
|||||||
add_library(${libname} SHARED ${SOURCES})
|
add_library(${libname} SHARED ${SOURCES})
|
||||||
set_target_properties(${libname} PROPERTIES
|
set_target_properties(${libname} PROPERTIES
|
||||||
VERSION ${PROJECT_VERSION}
|
VERSION ${PROJECT_VERSION}
|
||||||
SOVERSION ${PROJECT_SOVERSION})
|
SOVERSION ${PROJECT_SOVERSION}
|
||||||
|
CXX_STANDARD 17
|
||||||
|
CUDA_STANDARD 17)
|
||||||
target_link_libraries(${libname} ${LIBRARIES})
|
target_link_libraries(${libname} ${LIBRARIES})
|
||||||
|
|
||||||
|
if(USE_CUDA)
|
||||||
|
set_source_files_properties(VoxRaytracer.cpp VoxImage.cpp PROPERTIES LANGUAGE CUDA)
|
||||||
|
endif()
|
||||||
|
|
||||||
|
|
||||||
install(TARGETS ${libname}
|
install(TARGETS ${libname}
|
||||||
EXPORT "${PROJECT_NAME}Targets"
|
EXPORT "${PROJECT_NAME}Targets"
|
||||||
@@ -54,7 +60,8 @@ install(TARGETS ${libname}
|
|||||||
|
|
||||||
install(FILES ${HEADERS} DESTINATION ${INSTALL_INC_DIR}/Math)
|
install(FILES ${HEADERS} DESTINATION ${INSTALL_INC_DIR}/Math)
|
||||||
|
|
||||||
# TESTING
|
if(BUILD_TESTING)
|
||||||
# include(uLibTargetMacros)
|
include(uLibTargetMacros)
|
||||||
# add_subdirectory(testing)
|
add_subdirectory(testing)
|
||||||
|
endif()
|
||||||
|
|
||||||
|
|||||||
@@ -47,6 +47,7 @@
|
|||||||
#ifndef ULIB_DENSEMATRIX_H
|
#ifndef ULIB_DENSEMATRIX_H
|
||||||
#define ULIB_DENSEMATRIX_H
|
#define ULIB_DENSEMATRIX_H
|
||||||
|
|
||||||
|
// #include <Eigen/src/Core/Matrix.h>
|
||||||
#include <stdlib.h>
|
#include <stdlib.h>
|
||||||
|
|
||||||
#include <Eigen/Dense>
|
#include <Eigen/Dense>
|
||||||
@@ -114,6 +115,21 @@ typedef unsigned long Scalarul;
|
|||||||
typedef float Scalarf;
|
typedef float Scalarf;
|
||||||
typedef double Scalard;
|
typedef double Scalard;
|
||||||
|
|
||||||
|
typedef Eigen::Matrix<int, 1, 1> Vector1i;
|
||||||
|
typedef Eigen::Vector2i Vector2i;
|
||||||
|
typedef Eigen::Vector3i Vector3i;
|
||||||
|
typedef Eigen::Vector4i Vector4i;
|
||||||
|
|
||||||
|
typedef Eigen::Matrix<float, 1, 1> Vector1f;
|
||||||
|
typedef Eigen::Vector2f Vector2f;
|
||||||
|
typedef Eigen::Vector3f Vector3f;
|
||||||
|
typedef Eigen::Vector4f Vector4f;
|
||||||
|
|
||||||
|
typedef Eigen::Matrix<double, 1, 1> Vector1d;
|
||||||
|
typedef Eigen::Vector2d Vector2d;
|
||||||
|
typedef Eigen::Vector3d Vector3d;
|
||||||
|
typedef Eigen::Vector4d Vector4d;
|
||||||
|
|
||||||
typedef Eigen::Matrix<int, 1, 1> Matrix1i;
|
typedef Eigen::Matrix<int, 1, 1> Matrix1i;
|
||||||
typedef Eigen::Matrix2i Matrix2i;
|
typedef Eigen::Matrix2i Matrix2i;
|
||||||
typedef Eigen::Matrix3i Matrix3i;
|
typedef Eigen::Matrix3i Matrix3i;
|
||||||
@@ -124,15 +140,15 @@ typedef Eigen::Matrix2f Matrix2f;
|
|||||||
typedef Eigen::Matrix3f Matrix3f;
|
typedef Eigen::Matrix3f Matrix3f;
|
||||||
typedef Eigen::Matrix4f Matrix4f;
|
typedef Eigen::Matrix4f Matrix4f;
|
||||||
|
|
||||||
typedef Eigen::Matrix<int, 1, 1> Vector1i;
|
typedef Eigen::Matrix<double, 1, 1> Matrix1d;
|
||||||
typedef Eigen::Vector2i Vector2i;
|
typedef Eigen::Matrix2d Matrix2d;
|
||||||
typedef Eigen::Vector3i Vector3i;
|
typedef Eigen::Matrix3d Matrix3d;
|
||||||
typedef Eigen::Vector4i Vector4i;
|
typedef Eigen::Matrix4d Matrix4d;
|
||||||
|
|
||||||
|
typedef Eigen::MatrixXi MatrixXi;
|
||||||
|
typedef Eigen::MatrixXf MatrixXf;
|
||||||
|
typedef Eigen::MatrixXd MatrixXd;
|
||||||
|
|
||||||
typedef Eigen::Matrix<float, 1, 1> Vector1f;
|
|
||||||
typedef Eigen::Vector2f Vector2f;
|
|
||||||
typedef Eigen::Vector3f Vector3f;
|
|
||||||
typedef Eigen::Vector4f Vector4f;
|
|
||||||
|
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
// Vector String interaction ///////////////////////////////////////////////////
|
// Vector String interaction ///////////////////////////////////////////////////
|
||||||
@@ -175,7 +191,7 @@ std::string VectorxT_ToString(const Eigen::Matrix<T, size, 1> &vec) {
|
|||||||
// }
|
// }
|
||||||
|
|
||||||
template <typename T, int size>
|
template <typename T, int size>
|
||||||
void operator>>(std::string &str, Eigen::Matrix<T, size, 1> &vec) {
|
void operator >> (std::string &str, Eigen::Matrix<T, size, 1> &vec) {
|
||||||
VectorxT_StringTo(vec, str);
|
VectorxT_StringTo(vec, str);
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -188,6 +204,9 @@ public:
|
|||||||
typedef Eigen::Matrix<Scalarf, 4, 1> BaseClass;
|
typedef Eigen::Matrix<Scalarf, 4, 1> BaseClass;
|
||||||
|
|
||||||
_HPoint3f() : BaseClass(0, 0, 0, p) {}
|
_HPoint3f() : BaseClass(0, 0, 0, p) {}
|
||||||
|
_HPoint3f(int rows, int cols) : BaseClass() {
|
||||||
|
this->operator()(3) = p;
|
||||||
|
}
|
||||||
_HPoint3f(float x, float y, float z) : BaseClass(x, y, z, p) {}
|
_HPoint3f(float x, float y, float z) : BaseClass(x, y, z, p) {}
|
||||||
_HPoint3f(Vector3f &in) : BaseClass(in.homogeneous()) {
|
_HPoint3f(Vector3f &in) : BaseClass(in.homogeneous()) {
|
||||||
this->operator()(3) = p;
|
this->operator()(3) = p;
|
||||||
|
|||||||
@@ -23,8 +23,6 @@
|
|||||||
|
|
||||||
//////////////////////////////////////////////////////////////////////////////*/
|
//////////////////////////////////////////////////////////////////////////////*/
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
#ifndef U_MATH_VOXIMAGE_H
|
#ifndef U_MATH_VOXIMAGE_H
|
||||||
#define U_MATH_VOXIMAGE_H
|
#define U_MATH_VOXIMAGE_H
|
||||||
|
|
||||||
@@ -36,6 +34,8 @@
|
|||||||
#include <stdlib.h>
|
#include <stdlib.h>
|
||||||
#include <vector>
|
#include <vector>
|
||||||
|
|
||||||
|
#include <Core/DataAllocator.h>
|
||||||
|
|
||||||
namespace uLib {
|
namespace uLib {
|
||||||
|
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
@@ -46,36 +46,36 @@ namespace Abstract {
|
|||||||
|
|
||||||
class VoxImage : public uLib::StructuredGrid {
|
class VoxImage : public uLib::StructuredGrid {
|
||||||
public:
|
public:
|
||||||
typedef uLib::StructuredGrid BaseClass;
|
typedef uLib::StructuredGrid BaseClass;
|
||||||
|
|
||||||
virtual float GetValue(const Vector3i &id) const = 0;
|
virtual float GetValue(const Vector3i &id) const = 0;
|
||||||
virtual float GetValue(const int id) const = 0;
|
virtual float GetValue(const int id) const = 0;
|
||||||
virtual void SetValue(const Vector3i &id, float value) = 0;
|
virtual void SetValue(const Vector3i &id, float value) = 0;
|
||||||
virtual void SetValue(const int id, float value) = 0;
|
virtual void SetValue(const int id, float value) = 0;
|
||||||
|
|
||||||
virtual void SetDims(const Vector3i &size) = 0;
|
virtual void SetDims(const Vector3i &size) = 0;
|
||||||
|
|
||||||
void ExportToVtk(const char *file, bool density_type = 0);
|
void ExportToVtk(const char *file, bool density_type = 0);
|
||||||
|
|
||||||
// use this function to export to VTK binary format
|
// use this function to export to VTK binary format
|
||||||
void ExportToVti (const char *file, bool density_type = 0, bool compressed = 0);
|
void ExportToVti(const char *file, bool density_type = 0,
|
||||||
|
bool compressed = 0);
|
||||||
// this function has been deprecated in favor of ExportToVti
|
|
||||||
// but it is kept for backward compatibility and because it
|
|
||||||
// does not depend on vtk library
|
|
||||||
void ExportToVtkXml(const char *file, bool density_type = 0);
|
|
||||||
|
|
||||||
int ImportFromVtk(const char *file, bool density_type = 0);
|
// this function has been deprecated in favor of ExportToVti
|
||||||
|
// but it is kept for backward compatibility and because it
|
||||||
|
// does not depend on vtk library
|
||||||
|
void ExportToVtkXml(const char *file, bool density_type = 0);
|
||||||
|
|
||||||
int ImportFromVti(const char *file, bool density_type = 0);
|
int ImportFromVtk(const char *file, bool density_type = 0);
|
||||||
|
|
||||||
|
int ImportFromVti(const char *file, bool density_type = 0);
|
||||||
|
|
||||||
|
virtual ~VoxImage() {}
|
||||||
protected:
|
protected:
|
||||||
|
VoxImage(const Vector3i &size) : BaseClass(size) {}
|
||||||
virtual ~VoxImage() {}
|
|
||||||
VoxImage(const Vector3i &size) : BaseClass(size) {}
|
|
||||||
};
|
};
|
||||||
|
|
||||||
}
|
} // namespace Abstract
|
||||||
|
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
// VOXEL ////////////////////////////////////////////////////////////////////
|
// VOXEL ////////////////////////////////////////////////////////////////////
|
||||||
@@ -83,421 +83,416 @@ protected:
|
|||||||
|
|
||||||
namespace Interface {
|
namespace Interface {
|
||||||
struct Voxel {
|
struct Voxel {
|
||||||
template<class Self> void check_structural() {
|
template <class Self> void check_structural() {
|
||||||
uLibCheckMember(Self,Value, Scalarf);
|
uLibCheckMember(Self, Value, Scalarf);
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
}
|
} // namespace Interface
|
||||||
|
|
||||||
struct Voxel {
|
struct Voxel {
|
||||||
Scalarf Value;
|
Scalarf Value = 0.0f;
|
||||||
|
Scalari Count = 0;
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
// VOX IMAGE /////////////////////////////////////////////////////////////////
|
// VOX IMAGE /////////////////////////////////////////////////////////////////
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
|
|
||||||
|
template <typename T> class VoxImage : public Abstract::VoxImage {
|
||||||
template< typename T >
|
|
||||||
class VoxImage : public Abstract::VoxImage {
|
|
||||||
public:
|
public:
|
||||||
typedef Abstract::VoxImage BaseClass;
|
typedef Abstract::VoxImage BaseClass;
|
||||||
|
|
||||||
VoxImage();
|
VoxImage();
|
||||||
|
|
||||||
VoxImage(const Vector3i &size);
|
VoxImage(const Vector3i &size);
|
||||||
|
|
||||||
VoxImage(const VoxImage<T> ©) :
|
VoxImage(const VoxImage<T> ©) : BaseClass(copy) {
|
||||||
BaseClass(copy)
|
this->m_Data = copy.m_Data;
|
||||||
{
|
}
|
||||||
this->m_Data = copy.m_Data;
|
|
||||||
|
inline DataAllocator<T> &Data() { return this->m_Data; }
|
||||||
|
inline const DataAllocator<T> &ConstData() const { return m_Data; }
|
||||||
|
|
||||||
|
inline const T &At(int i) const { return m_Data.at(i); }
|
||||||
|
inline const T &At(const Vector3i &id) const { return m_Data.at(Map(id)); }
|
||||||
|
inline T &operator[](unsigned int i) { return m_Data[i]; }
|
||||||
|
inline T &operator[](const Vector3i &id) { return m_Data[Map(id)]; }
|
||||||
|
|
||||||
|
// this implements Abstract interface //
|
||||||
|
inline Scalarf GetValue(const Vector3i &id) const {
|
||||||
|
return this->At(id).Value;
|
||||||
|
}
|
||||||
|
inline Scalarf GetValue(const int id) const { return this->At(id).Value; }
|
||||||
|
|
||||||
|
inline void SetValue(const Vector3i &id, Scalarf value) {
|
||||||
|
this->operator[](id).Value = value;
|
||||||
|
}
|
||||||
|
inline void SetValue(const int id, float value) {
|
||||||
|
this->operator[](id).Value = value;
|
||||||
|
}
|
||||||
|
|
||||||
|
inline void SetDims(const Vector3i &size) {
|
||||||
|
this->m_Data.resize(size.prod());
|
||||||
|
StructuredGrid::SetDims(size);
|
||||||
|
}
|
||||||
|
|
||||||
|
inline VoxImage<T> clipImage(const Vector3i begin, const Vector3i end) const;
|
||||||
|
inline VoxImage<T> clipImage(const HPoint3f begin, const HPoint3f end) const;
|
||||||
|
inline VoxImage<T> clipImage(const float density) const;
|
||||||
|
inline VoxImage<T> clipImage(const float densityMin,
|
||||||
|
const float densityMax) const;
|
||||||
|
|
||||||
|
inline VoxImage<T> maskImage(const HPoint3f begin, const HPoint3f end,
|
||||||
|
float value) const;
|
||||||
|
inline VoxImage<T> maskImage(const float threshold, float belowValue = 0,
|
||||||
|
float aboveValue = 0) const;
|
||||||
|
inline VoxImage<T> fixVoxels(const float threshold, float tolerance) const;
|
||||||
|
inline VoxImage<T> fixVoxels(const float threshold, float tolerance,
|
||||||
|
const HPoint3f begin, const HPoint3f end) const;
|
||||||
|
inline VoxImage<T> fixVoxelsAroundPlane(const float threshold,
|
||||||
|
float tolerance, const HPoint3f begin,
|
||||||
|
const HPoint3f end,
|
||||||
|
bool aboveAir) const;
|
||||||
|
inline VoxImage<T> fixVoxels(const HPoint3f begin, const HPoint3f end) const;
|
||||||
|
inline VoxImage<T> Abs() const;
|
||||||
|
|
||||||
|
inline void SelectScalarfComponent(T &element, Scalarf *scalar);
|
||||||
|
|
||||||
|
inline void InitVoxels(T t);
|
||||||
|
|
||||||
|
// MATH OPERATORS //
|
||||||
|
inline void operator*=(Scalarf scalar) {
|
||||||
|
for (unsigned int i = 0; i < m_Data.size(); ++i)
|
||||||
|
m_Data[i].Value *= scalar;
|
||||||
|
}
|
||||||
|
inline void operator+=(Scalarf scalar) {
|
||||||
|
for (unsigned int i = 0; i < m_Data.size(); ++i)
|
||||||
|
m_Data[i].Value += scalar;
|
||||||
|
}
|
||||||
|
inline void operator/=(Scalarf scalar) {
|
||||||
|
for (unsigned int i = 0; i < m_Data.size(); ++i)
|
||||||
|
m_Data[i].Value /= scalar;
|
||||||
|
}
|
||||||
|
inline void operator-=(Scalarf scalar) {
|
||||||
|
for (unsigned int i = 0; i < m_Data.size(); ++i)
|
||||||
|
m_Data[i].Value -= scalar;
|
||||||
|
}
|
||||||
|
|
||||||
|
// MATH VoxImage Operators //
|
||||||
|
template <typename S> void operator+=(VoxImage<S> &sibling) {
|
||||||
|
if (this->GetDims() != sibling.GetDims()) {
|
||||||
|
// printf("Warning when adding VoxImages: I'm NOT doing it!\n");
|
||||||
|
return;
|
||||||
|
} // WARNING! You must Warn the user!
|
||||||
|
for (unsigned int i = 0; i < m_Data.size(); ++i) {
|
||||||
|
m_Data[i].Value += sibling.At(i).Value;
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
inline std::vector<T> & Data() { return this->m_Data; }
|
template <typename S> void operator-=(VoxImage<S> &sibling) {
|
||||||
inline const std::vector<T>& ConstData() const { return m_Data; }
|
if (this->GetDims() != sibling.GetDims()) {
|
||||||
|
// printf("Warning when subtracting VoxImages: I'm NOT doing it!\n");
|
||||||
inline const T& At(int i) const { return m_Data.at(i); }
|
return;
|
||||||
inline const T& At(const Vector3i &id) const { return m_Data.at(Map(id)); }
|
} // WARNING! You must Warn the user!
|
||||||
inline T& operator[](unsigned int i) { return m_Data[i]; }
|
for (unsigned int i = 0; i < m_Data.size(); ++i) {
|
||||||
inline T& operator[](const Vector3i &id) { return m_Data[Map(id)]; }
|
m_Data[i].Value -= sibling.At(i).Value;
|
||||||
|
|
||||||
// this implements Abstract interface //
|
|
||||||
inline Scalarf GetValue(const Vector3i &id) const {
|
|
||||||
return this->At(id).Value;
|
|
||||||
}
|
|
||||||
inline Scalarf GetValue(const int id) const {
|
|
||||||
return this->At(id).Value;
|
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
inline void SetValue(const Vector3i &id, Scalarf value) {
|
template <typename S> void operator*=(VoxImage<S> &sibling) {
|
||||||
this->operator [](id).Value = value;
|
if (this->GetDims() != sibling.GetDims()) {
|
||||||
}
|
// printf("Warning when multiplying VoxImages: I'm NOT doing it!\n");
|
||||||
inline void SetValue(const int id, float value) {
|
return;
|
||||||
this->operator [](id).Value = value;
|
} // WARNING! You must Warn the user!
|
||||||
|
for (unsigned int i = 0; i < m_Data.size(); ++i) {
|
||||||
|
m_Data[i].Value *= sibling.At(i).Value;
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
inline void SetDims(const Vector3i &size) {
|
template <typename S> void operator/=(VoxImage<S> &sibling) {
|
||||||
this->m_Data.resize(size.prod());
|
if (this->GetDims() != sibling.GetDims()) {
|
||||||
BaseClass::BaseClass::SetDims(size); // FIX horrible coding style !
|
// printf("Warning when dividing VoxImages: I'm NOT doing it!\n");
|
||||||
}
|
return;
|
||||||
|
} // WARNING! You must Warn the user!
|
||||||
inline VoxImage<T> clipImage(const Vector3i begin, const Vector3i end) const;
|
for (unsigned int i = 0; i < m_Data.size(); ++i) {
|
||||||
inline VoxImage<T> clipImage(const HPoint3f begin, const HPoint3f end) const;
|
m_Data[i].Value /= sibling.At(i).Value;
|
||||||
inline VoxImage<T> clipImage(const float density) const;
|
|
||||||
inline VoxImage<T> clipImage(const float densityMin, const float densityMax) const;
|
|
||||||
|
|
||||||
inline VoxImage<T> maskImage(const HPoint3f begin, const HPoint3f end, float value) const;
|
|
||||||
inline VoxImage<T> maskImage(const float threshold, float belowValue=0, float aboveValue=0) const;
|
|
||||||
inline VoxImage<T> fixVoxels(const float threshold, float tolerance) const;
|
|
||||||
inline VoxImage<T> fixVoxels(const float threshold, float tolerance, const HPoint3f begin, const HPoint3f end) const;
|
|
||||||
inline VoxImage<T> fixVoxelsAroundPlane(const float threshold, float tolerance, const HPoint3f begin, const HPoint3f end, bool aboveAir) const;
|
|
||||||
inline VoxImage<T> fixVoxels(const HPoint3f begin, const HPoint3f end) const;
|
|
||||||
inline VoxImage<T> Abs() const;
|
|
||||||
|
|
||||||
inline void SelectScalarfComponent(T &element, Scalarf *scalar);
|
|
||||||
|
|
||||||
inline void InitVoxels(T t);
|
|
||||||
|
|
||||||
// MATH OPERATORS //
|
|
||||||
inline void operator *=(Scalarf scalar) {
|
|
||||||
for(unsigned int i = 0; i < m_Data.size(); ++i)
|
|
||||||
m_Data[i].Value *= scalar;
|
|
||||||
}
|
|
||||||
inline void operator +=(Scalarf scalar) {
|
|
||||||
for(unsigned int i = 0; i < m_Data.size(); ++i)
|
|
||||||
m_Data[i].Value += scalar;
|
|
||||||
}
|
|
||||||
inline void operator /=(Scalarf scalar) {
|
|
||||||
for(unsigned int i = 0; i < m_Data.size(); ++i)
|
|
||||||
m_Data[i].Value /= scalar;
|
|
||||||
}
|
|
||||||
inline void operator -=(Scalarf scalar) {
|
|
||||||
for(unsigned int i = 0; i < m_Data.size(); ++i)
|
|
||||||
m_Data[i].Value -= scalar;
|
|
||||||
}
|
|
||||||
|
|
||||||
// MATH VoxImage Operators //
|
|
||||||
template <typename S>
|
|
||||||
void operator +=(VoxImage<S> &sibling) {
|
|
||||||
if (this->GetDims() != sibling.GetDims()) {
|
|
||||||
//printf("Warning when adding VoxImages: I'm NOT doing it!\n");
|
|
||||||
return;
|
|
||||||
}// WARNING! You must Warn the user!
|
|
||||||
for(unsigned int i = 0; i < m_Data.size(); ++i) {
|
|
||||||
m_Data[i].Value += sibling.At(i).Value;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
template <typename S>
|
|
||||||
void operator -=(VoxImage<S> &sibling) {
|
|
||||||
if (this->GetDims() != sibling.GetDims()) {
|
|
||||||
//printf("Warning when subtracting VoxImages: I'm NOT doing it!\n");
|
|
||||||
return;
|
|
||||||
}// WARNING! You must Warn the user!
|
|
||||||
for(unsigned int i = 0; i < m_Data.size(); ++i) {
|
|
||||||
m_Data[i].Value -= sibling.At(i).Value;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
template <typename S>
|
|
||||||
void operator *=(VoxImage<S> &sibling) {
|
|
||||||
if (this->GetDims() != sibling.GetDims()) {
|
|
||||||
//printf("Warning when multiplying VoxImages: I'm NOT doing it!\n");
|
|
||||||
return;
|
|
||||||
}// WARNING! You must Warn the user!
|
|
||||||
for(unsigned int i = 0; i < m_Data.size(); ++i) {
|
|
||||||
m_Data[i].Value *= sibling.At(i).Value;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
template <typename S>
|
|
||||||
void operator /=(VoxImage<S> &sibling) {
|
|
||||||
if (this->GetDims() != sibling.GetDims()) {
|
|
||||||
//printf("Warning when dividing VoxImages: I'm NOT doing it!\n");
|
|
||||||
return;
|
|
||||||
}// WARNING! You must Warn the user!
|
|
||||||
for(unsigned int i = 0; i < m_Data.size(); ++i) {
|
|
||||||
m_Data[i].Value /= sibling.At(i).Value;
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
private:
|
private:
|
||||||
std::vector<T> m_Data;
|
DataAllocator<T> m_Data;
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|
||||||
template<typename T>
|
|
||||||
VoxImage<T>::VoxImage() :
|
|
||||||
m_Data(0),
|
|
||||||
BaseClass(Vector3i(0,0,0))
|
|
||||||
{ Interface::IsA <T,Interface::Voxel>(); /* structural check for T */ }
|
|
||||||
|
|
||||||
template<typename T>
|
|
||||||
VoxImage<T>::VoxImage(const Vector3i &size) :
|
|
||||||
m_Data(size.prod()),
|
|
||||||
BaseClass(size)
|
|
||||||
{ Interface::IsA <T,Interface::Voxel>(); /* structural check for T */ }
|
|
||||||
|
|
||||||
|
|
||||||
template <typename T>
|
template <typename T>
|
||||||
VoxImage<T> VoxImage<T>::clipImage(const Vector3i begin, const Vector3i end) const
|
VoxImage<T>::VoxImage() : m_Data(0), BaseClass(Vector3i(0, 0, 0)) {
|
||||||
{
|
Interface::IsA<T, Interface::Voxel>(); /* structural check for T */
|
||||||
Vector3i dim = (end-begin)+Vector3i(1,1,1);
|
|
||||||
VoxImage<T> out(*this);
|
|
||||||
out.SetDims(dim);
|
|
||||||
out.SetPosition(this->GetPosition() + this->GetSpacing().cwiseProduct(begin.cast<float>()) );
|
|
||||||
|
|
||||||
for(uint x = 0; x<dim(0); ++x )
|
|
||||||
for(uint y = 0; y<dim(1); ++y )
|
|
||||||
for(uint z = 0; z<dim(2); ++z )
|
|
||||||
{
|
|
||||||
Vector3i id = Vector3i(x,y,z);
|
|
||||||
out[id] = this->At(begin + id);
|
|
||||||
}
|
|
||||||
return out;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
template <typename T>
|
template <typename T>
|
||||||
VoxImage<T> VoxImage<T>::clipImage(const HPoint3f begin, const HPoint3f end) const
|
VoxImage<T>::VoxImage(const Vector3i &size)
|
||||||
{
|
: m_Data(size.prod()), BaseClass(size) {
|
||||||
Vector3i v1 = this->Find(begin);
|
Interface::IsA<T, Interface::Voxel>(); /* structural check for T */
|
||||||
Vector3i v2 = this->Find(end);
|
|
||||||
return this->clipImage(v1,v2);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
template <typename T>
|
template <typename T>
|
||||||
VoxImage<T> VoxImage<T>::clipImage(const float density) const
|
VoxImage<T> VoxImage<T>::clipImage(const Vector3i begin,
|
||||||
{
|
const Vector3i end) const {
|
||||||
Vector3i v1 = this->GetDims();
|
Vector3i dim = (end - begin) + Vector3i(1, 1, 1);
|
||||||
Vector3i v2 = Vector3i(0,0,0);
|
VoxImage<T> out(*this);
|
||||||
for(uint i=0; i< this->m_Data.size(); ++i) {
|
out.SetDims(dim);
|
||||||
if( this->GetValue(i) >= density ) {
|
out.SetPosition(this->GetPosition() +
|
||||||
Vector3i id = this->UnMap(i);
|
this->GetSpacing().cwiseProduct(begin.cast<float>()));
|
||||||
v1 = v1.array().min(id.array());
|
|
||||||
v2 = v2.array().max(id.array());
|
for (uint x = 0; x < dim(0); ++x)
|
||||||
}
|
for (uint y = 0; y < dim(1); ++y)
|
||||||
|
for (uint z = 0; z < dim(2); ++z) {
|
||||||
|
Vector3i id = Vector3i(x, y, z);
|
||||||
|
out[id] = this->At(begin + id);
|
||||||
|
}
|
||||||
|
return out;
|
||||||
|
}
|
||||||
|
|
||||||
|
template <typename T>
|
||||||
|
VoxImage<T> VoxImage<T>::clipImage(const HPoint3f begin,
|
||||||
|
const HPoint3f end) const {
|
||||||
|
Vector3i v1 = this->Find(begin);
|
||||||
|
Vector3i v2 = this->Find(end);
|
||||||
|
return this->clipImage(v1, v2);
|
||||||
|
}
|
||||||
|
|
||||||
|
template <typename T>
|
||||||
|
VoxImage<T> VoxImage<T>::clipImage(const float density) const {
|
||||||
|
Vector3i v1 = this->GetDims();
|
||||||
|
Vector3i v2 = Vector3i(0, 0, 0);
|
||||||
|
for (uint i = 0; i < this->m_Data.size(); ++i) {
|
||||||
|
if (this->GetValue(i) >= density) {
|
||||||
|
Vector3i id = this->UnMap(i);
|
||||||
|
v1 = v1.array().min(id.array());
|
||||||
|
v2 = v2.array().max(id.array());
|
||||||
}
|
}
|
||||||
return this->clipImage(v1,v2);
|
}
|
||||||
|
return this->clipImage(v1, v2);
|
||||||
}
|
}
|
||||||
|
|
||||||
template <typename T>
|
template <typename T>
|
||||||
VoxImage<T> VoxImage<T>::clipImage(const float densityMin, const float densityMax) const
|
VoxImage<T> VoxImage<T>::clipImage(const float densityMin,
|
||||||
{
|
const float densityMax) const {
|
||||||
Vector3i v1 = this->GetDims();
|
Vector3i v1 = this->GetDims();
|
||||||
Vector3i v2 = Vector3i(0,0,0);
|
Vector3i v2 = Vector3i(0, 0, 0);
|
||||||
for(uint i=0; i< this->m_Data.size(); ++i) {
|
for (uint i = 0; i < this->m_Data.size(); ++i) {
|
||||||
if( this->GetValue(i) >= densityMin && this->GetValue(i) <= densityMax) {
|
if (this->GetValue(i) >= densityMin && this->GetValue(i) <= densityMax) {
|
||||||
Vector3i id = this->UnMap(i);
|
Vector3i id = this->UnMap(i);
|
||||||
v1 = v1.array().min(id.array());
|
v1 = v1.array().min(id.array());
|
||||||
v2 = v2.array().max(id.array());
|
v2 = v2.array().max(id.array());
|
||||||
}
|
|
||||||
}
|
}
|
||||||
return this->clipImage(v1,v2);
|
}
|
||||||
|
return this->clipImage(v1, v2);
|
||||||
}
|
}
|
||||||
|
|
||||||
template <typename T>
|
template <typename T>
|
||||||
VoxImage<T> VoxImage<T>::maskImage(const HPoint3f begin, const HPoint3f end, float value) const
|
VoxImage<T> VoxImage<T>::maskImage(const HPoint3f begin, const HPoint3f end,
|
||||||
{
|
float value) const {
|
||||||
VoxImage<T> out(*this);
|
VoxImage<T> out(*this);
|
||||||
out.SetDims(this->GetDims());
|
out.SetDims(this->GetDims());
|
||||||
out.SetPosition(this->GetPosition());
|
out.SetPosition(this->GetPosition());
|
||||||
|
|
||||||
Vector3i voxB = this->Find(begin);
|
Vector3i voxB = this->Find(begin);
|
||||||
Vector3i voxE = this->Find(end);
|
Vector3i voxE = this->Find(end);
|
||||||
|
|
||||||
Vector3i ID;
|
Vector3i ID;
|
||||||
|
|
||||||
for(int ix=voxB(0); ix<voxE(0); ix++)
|
for (int ix = voxB(0); ix < voxE(0); ix++)
|
||||||
for(int iy=voxB(1); iy<voxE(1); iy++)
|
for (int iy = voxB(1); iy < voxE(1); iy++)
|
||||||
for(int iz=voxB(2); iz<voxE(2); iz++){
|
for (int iz = voxB(2); iz < voxE(2); iz++) {
|
||||||
ID << ix,iy,iz;
|
ID << ix, iy, iz;
|
||||||
out.SetValue(ID,value*1.E-6);
|
out.SetValue(ID, value * 1.E-6);
|
||||||
}
|
}
|
||||||
|
|
||||||
return out;
|
return out;
|
||||||
}
|
}
|
||||||
|
|
||||||
template <typename T>
|
template <typename T>
|
||||||
VoxImage<T> VoxImage<T>::maskImage(const float threshold, float belowValue, float aboveValue) const
|
VoxImage<T> VoxImage<T>::maskImage(const float threshold, float belowValue,
|
||||||
{
|
float aboveValue) const {
|
||||||
std::cout << "VoxImage: maskImage, fixing voxels under threshold " << threshold;
|
std::cout << "VoxImage: maskImage, fixing voxels under threshold "
|
||||||
if(belowValue)
|
<< threshold;
|
||||||
std::cout << " at value " << belowValue;
|
if (belowValue)
|
||||||
else
|
std::cout << " at value " << belowValue;
|
||||||
std::cout << " at -value";
|
else
|
||||||
std::cout << ", voxels above threshold at value ";
|
std::cout << " at -value";
|
||||||
if(aboveValue)
|
std::cout << ", voxels above threshold at value ";
|
||||||
std::cout << aboveValue;
|
if (aboveValue)
|
||||||
else
|
std::cout << aboveValue;
|
||||||
std::cout << "found";
|
else
|
||||||
|
std::cout << "found";
|
||||||
|
|
||||||
|
VoxImage<T> out(*this);
|
||||||
|
out.SetDims(this->GetDims());
|
||||||
|
out.SetPosition(this->GetPosition());
|
||||||
|
|
||||||
VoxImage<T> out(*this);
|
for (uint i = 0; i < this->m_Data.size(); ++i) {
|
||||||
out.SetDims(this->GetDims());
|
// skip negative voxels: they are already frozen
|
||||||
out.SetPosition(this->GetPosition());
|
if (this->GetValue(i) >= 0) {
|
||||||
|
// voxels under threshold
|
||||||
for(uint i=0; i< this->m_Data.size(); ++i) {
|
if (this->GetValue(i) <= threshold * 1.E-6) {
|
||||||
// skip negative voxels: they are already frozen
|
if (belowValue) {
|
||||||
if( this->GetValue(i) >= 0 ){
|
// std::cout << "vox " << i << ", " <<
|
||||||
// voxels under threshold
|
// this->GetValue(i); std::cout << " ----> set to " <<
|
||||||
if( this->GetValue(i) <= threshold*1.E-6 ){
|
// -1.*belowValue*1.E-6 << std::endl;
|
||||||
if(belowValue){
|
out.SetValue(i, -1. * belowValue * 1.E-6);
|
||||||
// std::cout << "vox " << i << ", " << this->GetValue(i);
|
} else
|
||||||
// std::cout << " ----> set to " << -1.*belowValue*1.E-6 << std::endl;
|
out.SetValue(i, -1. * this->GetValue(i));
|
||||||
out.SetValue(i,-1.*belowValue*1.E-6);}
|
}
|
||||||
else
|
// voxels over threshold
|
||||||
out.SetValue(i,-1.*this->GetValue(i));
|
else {
|
||||||
}
|
if (aboveValue)
|
||||||
// voxels over threshold
|
out.SetValue(i, aboveValue * 1.E-6);
|
||||||
else{
|
else
|
||||||
if(aboveValue)
|
out.SetValue(i, this->GetValue(i));
|
||||||
out.SetValue(i,aboveValue*1.E-6);
|
}
|
||||||
else
|
|
||||||
out.SetValue(i,this->GetValue(i));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
return out;
|
}
|
||||||
|
return out;
|
||||||
}
|
}
|
||||||
|
|
||||||
template <typename T>
|
template <typename T>
|
||||||
VoxImage<T> VoxImage<T>::fixVoxels(const float threshold, float tolerance) const
|
VoxImage<T> VoxImage<T>::fixVoxels(const float threshold,
|
||||||
{
|
float tolerance) const {
|
||||||
std::cout << "VoxImage: fixing voxels with value " << threshold << std::endl;
|
std::cout << "VoxImage: fixing voxels with value " << threshold << std::endl;
|
||||||
|
|
||||||
VoxImage<T> out(*this);
|
VoxImage<T> out(*this);
|
||||||
out.SetDims(this->GetDims());
|
out.SetDims(this->GetDims());
|
||||||
out.SetPosition(this->GetPosition());
|
out.SetPosition(this->GetPosition());
|
||||||
|
|
||||||
for(uint i=0; i< this->m_Data.size(); ++i) {
|
for (uint i = 0; i < this->m_Data.size(); ++i) {
|
||||||
|
// voxels around threshold
|
||||||
|
if (fabs(this->GetValue(i) - threshold * 1.E-6) < tolerance * 1.E-6) {
|
||||||
|
// std::cout << "vox " << i << ", " << this->GetValue(i);
|
||||||
|
// std::cout << " ----> set to " << -1.*this->GetValue(i) <<
|
||||||
|
// std::endl;
|
||||||
|
out.SetValue(i, -1. * this->GetValue(i));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return out;
|
||||||
|
}
|
||||||
|
|
||||||
|
template <typename T> VoxImage<T> VoxImage<T>::Abs() const {
|
||||||
|
std::cout << "VoxImage: set abs voxels value " << std::endl;
|
||||||
|
|
||||||
|
VoxImage<T> out(*this);
|
||||||
|
out.SetDims(this->GetDims());
|
||||||
|
out.SetPosition(this->GetPosition());
|
||||||
|
|
||||||
|
for (uint i = 0; i < this->m_Data.size(); ++i)
|
||||||
|
out.SetValue(i, fabs(this->GetValue(i)));
|
||||||
|
|
||||||
|
return out;
|
||||||
|
}
|
||||||
|
|
||||||
|
template <typename T>
|
||||||
|
VoxImage<T> VoxImage<T>::fixVoxels(const float threshold, float tolerance,
|
||||||
|
const HPoint3f begin,
|
||||||
|
const HPoint3f end) const {
|
||||||
|
VoxImage<T> out(*this);
|
||||||
|
out.SetDims(this->GetDims());
|
||||||
|
out.SetPosition(this->GetPosition());
|
||||||
|
|
||||||
|
Vector3i voxB = this->Find(begin);
|
||||||
|
Vector3i voxE = this->Find(end);
|
||||||
|
|
||||||
|
Vector3i ID;
|
||||||
|
|
||||||
|
for (int ix = voxB(0); ix < voxE(0); ix++)
|
||||||
|
for (int iy = voxB(1); iy < voxE(1); iy++)
|
||||||
|
for (int iz = voxB(2); iz < voxE(2); iz++) {
|
||||||
|
ID << ix, iy, iz;
|
||||||
// voxels around threshold
|
// voxels around threshold
|
||||||
if( fabs(this->GetValue(i) - threshold*1.E-6) < tolerance* 1.E-6 ){
|
if (fabs(this->GetValue(ID) - threshold * 1.E-6) < tolerance * 1.E-6) {
|
||||||
// std::cout << "vox " << i << ", " << this->GetValue(i);
|
out.SetValue(ID, -1. * this->GetValue(ID));
|
||||||
// std::cout << " ----> set to " << -1.*this->GetValue(i) << std::endl;
|
|
||||||
out.SetValue(i,-1.*this->GetValue(i));
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return out;
|
|
||||||
|
return out;
|
||||||
}
|
}
|
||||||
|
|
||||||
template <typename T>
|
template <typename T>
|
||||||
VoxImage<T> VoxImage<T>::Abs() const
|
VoxImage<T> VoxImage<T>::fixVoxels(const HPoint3f begin,
|
||||||
{
|
const HPoint3f end) const {
|
||||||
std::cout << "VoxImage: set abs voxels value " << std::endl;
|
VoxImage<T> out(*this);
|
||||||
|
out.SetDims(this->GetDims());
|
||||||
|
out.SetPosition(this->GetPosition());
|
||||||
|
|
||||||
VoxImage<T> out(*this);
|
Vector3i voxB = this->Find(begin);
|
||||||
out.SetDims(this->GetDims());
|
Vector3i voxE = this->Find(end);
|
||||||
out.SetPosition(this->GetPosition());
|
|
||||||
|
|
||||||
for(uint i=0; i< this->m_Data.size(); ++i)
|
Vector3i ID;
|
||||||
out.SetValue(i,fabs(this->GetValue(i)));
|
|
||||||
|
|
||||||
return out;
|
for (int ix = voxB(0); ix < voxE(0); ix++)
|
||||||
|
for (int iy = voxB(1); iy < voxE(1); iy++)
|
||||||
|
for (int iz = voxB(2); iz < voxE(2); iz++) {
|
||||||
|
ID << ix, iy, iz;
|
||||||
|
// voxels around threshold
|
||||||
|
out.SetValue(ID, -1. * this->GetValue(ID));
|
||||||
|
}
|
||||||
|
return out;
|
||||||
}
|
}
|
||||||
|
|
||||||
template <typename T>
|
template <typename T>
|
||||||
VoxImage<T> VoxImage<T>::fixVoxels( const float threshold, float tolerance, const HPoint3f begin, const HPoint3f end) const
|
VoxImage<T> VoxImage<T>::fixVoxelsAroundPlane(const float threshold,
|
||||||
{
|
float tolerance, const HPoint3f B,
|
||||||
VoxImage<T> out(*this);
|
const HPoint3f E,
|
||||||
out.SetDims(this->GetDims());
|
bool aboveAir) const {
|
||||||
out.SetPosition(this->GetPosition());
|
VoxImage<T> out(*this);
|
||||||
|
Vector3i dim = this->GetDims();
|
||||||
|
out.SetDims(dim);
|
||||||
|
out.SetPosition(this->GetPosition());
|
||||||
|
|
||||||
Vector3i voxB = this->Find(begin);
|
HPoint3f Bcoll = this->GetPosition().homogeneous();
|
||||||
Vector3i voxE = this->Find(end);
|
|
||||||
|
|
||||||
Vector3i ID;
|
Vector3i ID;
|
||||||
|
for (int ix = 0; ix < dim(0); ix++)
|
||||||
|
for (int iy = 0; iy < dim(1); iy++)
|
||||||
|
for (int iz = 0; iz < dim(2); iz++) {
|
||||||
|
ID << ix, iy, iz;
|
||||||
|
|
||||||
for(int ix=voxB(0); ix<voxE(0); ix++)
|
// B, E voxel position
|
||||||
for(int iy=voxB(1); iy<voxE(1); iy++)
|
Vector3i iv(ix, iy, iz);
|
||||||
for(int iz=voxB(2); iz<voxE(2); iz++){
|
Vector3f v =
|
||||||
ID << ix,iy,iz;
|
Vector3f(iv.cast<float>().cwiseProduct(this->GetSpacing()));
|
||||||
// voxels around threshold
|
HPoint3f Bvox = Bcoll + HPoint3f(v);
|
||||||
if( fabs(this->GetValue(ID) - threshold*1.E-6) < tolerance*1.E-6 ){
|
HPoint3f Evox = Bvox + this->GetSpacing().homogeneous();
|
||||||
out.SetValue(ID,-1.*this->GetValue(ID));
|
HPoint3f V = Bvox + 0.5 * (this->GetSpacing().homogeneous());
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return out;
|
// if distance point (x0,y0) from line by points (x1,y1) and (x2,y2) is
|
||||||
|
// less than tolerance
|
||||||
|
float x1 = B[1];
|
||||||
|
float y1 = B[2];
|
||||||
|
float x2 = E[1];
|
||||||
|
float y2 = E[2];
|
||||||
|
float x0 = V[1];
|
||||||
|
float y0 = V[2];
|
||||||
|
float dist = fabs((x2 - x1) * (y1 - y0) - ((x1 - x0) * (y2 - y1))) /
|
||||||
|
sqrt((x2 - x1) * (x2 - x1) + ((y2 - y1) * (y2 - y1)));
|
||||||
|
float distSign = (x2 - x1) * (y1 - y0) - ((x1 - x0) * (y2 - y1));
|
||||||
|
|
||||||
|
// set voxel air value
|
||||||
|
if (dist < tolerance) {
|
||||||
|
// std::cout << "voxel " << iv << ", line " << dist << ", tolerance "
|
||||||
|
// << tolerance << std::endl;
|
||||||
|
out.SetValue(ID, threshold * 1.E-6);
|
||||||
|
} else
|
||||||
|
out.SetValue(ID, this->GetValue(ID));
|
||||||
|
|
||||||
|
if ((distSign > 0 && aboveAir) || (distSign < 0 && !aboveAir))
|
||||||
|
out.SetValue(ID, threshold * 1.E-6);
|
||||||
|
}
|
||||||
|
return out;
|
||||||
}
|
}
|
||||||
|
|
||||||
template <typename T>
|
template <typename T> void VoxImage<T>::InitVoxels(T t) {
|
||||||
VoxImage<T> VoxImage<T>::fixVoxels(const HPoint3f begin, const HPoint3f end) const
|
std::fill(m_Data.begin(), m_Data.end(), t); // warning... stl function //
|
||||||
{
|
|
||||||
VoxImage<T> out(*this);
|
|
||||||
out.SetDims(this->GetDims());
|
|
||||||
out.SetPosition(this->GetPosition());
|
|
||||||
|
|
||||||
Vector3i voxB = this->Find(begin);
|
|
||||||
Vector3i voxE = this->Find(end);
|
|
||||||
|
|
||||||
Vector3i ID;
|
|
||||||
|
|
||||||
for(int ix=voxB(0); ix<voxE(0); ix++)
|
|
||||||
for(int iy=voxB(1); iy<voxE(1); iy++)
|
|
||||||
for(int iz=voxB(2); iz<voxE(2); iz++){
|
|
||||||
ID << ix,iy,iz;
|
|
||||||
// voxels around threshold
|
|
||||||
out.SetValue(ID,-1.*this->GetValue(ID));
|
|
||||||
}
|
|
||||||
return out;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
} // namespace uLib
|
||||||
template <typename T>
|
|
||||||
VoxImage<T> VoxImage<T>::fixVoxelsAroundPlane( const float threshold, float tolerance, const HPoint3f B, const HPoint3f E, bool aboveAir) const
|
|
||||||
{
|
|
||||||
VoxImage<T> out(*this);
|
|
||||||
Vector3i dim = this->GetDims();
|
|
||||||
out.SetDims(dim);
|
|
||||||
out.SetPosition(this->GetPosition());
|
|
||||||
|
|
||||||
HPoint3f Bcoll = this->GetPosition().homogeneous();
|
|
||||||
|
|
||||||
Vector3i ID;
|
|
||||||
for(int ix=0; ix<dim(0); ix++)
|
|
||||||
for(int iy=0; iy<dim(1); iy++)
|
|
||||||
for(int iz=0; iz<dim(2); iz++){
|
|
||||||
ID << ix,iy,iz;
|
|
||||||
|
|
||||||
// B, E voxel position
|
|
||||||
Vector3i iv(ix,iy,iz);
|
|
||||||
Vector3f v = Vector3f(iv.cast<float>().cwiseProduct(this->GetSpacing()));
|
|
||||||
HPoint3f Bvox = Bcoll + HPoint3f(v);
|
|
||||||
HPoint3f Evox = Bvox + this->GetSpacing().homogeneous();
|
|
||||||
HPoint3f V = Bvox + 0.5*(this->GetSpacing().homogeneous());
|
|
||||||
|
|
||||||
// if distance point (x0,y0) from line by points (x1,y1) and (x2,y2) is less than tolerance
|
|
||||||
float x1 = B[1];
|
|
||||||
float y1 = B[2];
|
|
||||||
float x2 = E[1];
|
|
||||||
float y2 = E[2];
|
|
||||||
float x0 = V[1];
|
|
||||||
float y0 = V[2];
|
|
||||||
float dist = fabs( (x2-x1)*(y1-y0) - ((x1-x0)*(y2-y1))) / sqrt( (x2-x1)*(x2-x1)+((y2-y1)*(y2-y1)));
|
|
||||||
float distSign = (x2-x1)*(y1-y0) - ((x1-x0)*(y2-y1));
|
|
||||||
|
|
||||||
// set voxel air value
|
|
||||||
if(dist < tolerance){
|
|
||||||
//std::cout << "voxel " << iv << ", line " << dist << ", tolerance " << tolerance << std::endl;
|
|
||||||
out.SetValue(ID,threshold*1.E-6);
|
|
||||||
}
|
|
||||||
else
|
|
||||||
out.SetValue(ID,this->GetValue(ID));
|
|
||||||
|
|
||||||
if((distSign>0 && aboveAir) || (distSign<0 && !aboveAir) )
|
|
||||||
out.SetValue(ID,threshold*1.E-6);
|
|
||||||
}
|
|
||||||
return out;
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
template<typename T>
|
|
||||||
void VoxImage<T>::InitVoxels(T t)
|
|
||||||
{
|
|
||||||
std::fill( m_Data.begin(), m_Data.end(), t ); // warning... stl function //
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
#endif // VOXIMAGE_H
|
#endif // VOXIMAGE_H
|
||||||
|
|||||||
@@ -23,8 +23,6 @@
|
|||||||
|
|
||||||
//////////////////////////////////////////////////////////////////////////////*/
|
//////////////////////////////////////////////////////////////////////////////*/
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
#ifndef VOXIMAGEFILTER_H
|
#ifndef VOXIMAGEFILTER_H
|
||||||
#define VOXIMAGEFILTER_H
|
#define VOXIMAGEFILTER_H
|
||||||
|
|
||||||
@@ -33,96 +31,83 @@
|
|||||||
|
|
||||||
#include "Math/VoxImage.h"
|
#include "Math/VoxImage.h"
|
||||||
|
|
||||||
|
|
||||||
namespace uLib {
|
namespace uLib {
|
||||||
|
|
||||||
|
|
||||||
namespace Interface {
|
namespace Interface {
|
||||||
struct VoxImageFilterShape {
|
struct VoxImageFilterShape {
|
||||||
template <class Self> void check_structural() {
|
template <class Self> void check_structural() {
|
||||||
uLibCheckFunction(Self,operator(),float,float);
|
uLibCheckFunction(Self, operator(), float, float);
|
||||||
uLibCheckFunction(Self,operator(),float,const Vector3f&);
|
uLibCheckFunction(Self, operator(), float, const Vector3f &);
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
}
|
} // namespace Interface
|
||||||
|
|
||||||
template < typename VoxelT > class Kernel;
|
|
||||||
|
|
||||||
|
template <typename VoxelT> class Kernel;
|
||||||
|
|
||||||
namespace Abstract {
|
namespace Abstract {
|
||||||
class VoxImageFilter {
|
class VoxImageFilter {
|
||||||
public:
|
public:
|
||||||
virtual void Run() = 0;
|
virtual void Run() = 0;
|
||||||
|
|
||||||
virtual void SetImage(Abstract::VoxImage *image) = 0;
|
virtual void SetImage(Abstract::VoxImage *image) = 0;
|
||||||
|
|
||||||
protected:
|
protected:
|
||||||
virtual ~VoxImageFilter() {}
|
virtual ~VoxImageFilter() {}
|
||||||
};
|
};
|
||||||
}
|
} // namespace Abstract
|
||||||
|
|
||||||
|
template <typename VoxelT, typename AlgorithmT>
|
||||||
template < typename VoxelT, typename AlgorithmT >
|
class VoxImageFilter : public Abstract::VoxImageFilter {
|
||||||
class VoxImageFilter : public Abstract::VoxImageFilter
|
|
||||||
{
|
|
||||||
|
|
||||||
public:
|
public:
|
||||||
VoxImageFilter(const Vector3i &size);
|
VoxImageFilter(const Vector3i &size);
|
||||||
|
|
||||||
void Run();
|
void Run();
|
||||||
|
|
||||||
void SetKernelNumericXZY(const std::vector<float> &numeric);
|
void SetKernelNumericXZY(const std::vector<float> &numeric);
|
||||||
|
|
||||||
void SetKernelSpherical(float (*shape)(float));
|
void SetKernelSpherical(float (*shape)(float));
|
||||||
|
|
||||||
template < class ShapeT >
|
template <class ShapeT> void SetKernelSpherical(ShapeT shape);
|
||||||
void SetKernelSpherical( ShapeT shape );
|
|
||||||
|
|
||||||
void SetKernelWeightFunction(float (*shape)(const Vector3f &));
|
void SetKernelWeightFunction(float (*shape)(const Vector3f &));
|
||||||
|
|
||||||
template < class ShapeT >
|
template <class ShapeT> void SetKernelWeightFunction(ShapeT shape);
|
||||||
void SetKernelWeightFunction( ShapeT shape );
|
|
||||||
|
|
||||||
inline Kernel<VoxelT> GetKernelData() const { return this->m_KernelData; }
|
inline const Kernel<VoxelT> &GetKernelData() const {
|
||||||
|
return this->m_KernelData;
|
||||||
|
}
|
||||||
|
inline Kernel<VoxelT> &GetKernelData() { return this->m_KernelData; }
|
||||||
|
|
||||||
inline VoxImage<VoxelT>* GetImage() const { return this->m_Image; }
|
inline VoxImage<VoxelT> *GetImage() const { return this->m_Image; }
|
||||||
|
|
||||||
void SetImage(Abstract::VoxImage *image);
|
void SetImage(Abstract::VoxImage *image);
|
||||||
|
|
||||||
protected:
|
protected:
|
||||||
|
float Convolve(const VoxImage<VoxelT> &buffer, int index); // remove //
|
||||||
|
|
||||||
float Convolve(const VoxImage<VoxelT> &buffer, int index); // remove //
|
void SetKernelOffset();
|
||||||
|
|
||||||
void SetKernelOffset();
|
float Distance2(const Vector3i &v);
|
||||||
|
|
||||||
float Distance2(const Vector3i &v);
|
// protected members for algorithm access //
|
||||||
|
Kernel<VoxelT> m_KernelData;
|
||||||
// protected members for algorithm access //
|
VoxImage<VoxelT> *m_Image;
|
||||||
Kernel<VoxelT> m_KernelData;
|
|
||||||
VoxImage<VoxelT> *m_Image;
|
|
||||||
|
|
||||||
private:
|
private:
|
||||||
AlgorithmT *t_Algoritm;
|
AlgorithmT *t_Algoritm;
|
||||||
|
|
||||||
};
|
};
|
||||||
|
|
||||||
|
} // namespace uLib
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
#endif // VOXIMAGEFILTER_H
|
#endif // VOXIMAGEFILTER_H
|
||||||
|
|
||||||
#include "VoxImageFilter.hpp"
|
#include "VoxImageFilter.hpp"
|
||||||
|
|
||||||
#include "VoxImageFilterLinear.hpp"
|
#include "VoxImageFilter2ndStat.hpp"
|
||||||
#include "VoxImageFilterThreshold.hpp"
|
|
||||||
#include "VoxImageFilterMedian.hpp"
|
|
||||||
#include "VoxImageFilterABTrim.hpp"
|
#include "VoxImageFilterABTrim.hpp"
|
||||||
#include "VoxImageFilterBilateral.hpp"
|
#include "VoxImageFilterBilateral.hpp"
|
||||||
#include "VoxImageFilter2ndStat.hpp"
|
|
||||||
#include "VoxImageFilterCustom.hpp"
|
#include "VoxImageFilterCustom.hpp"
|
||||||
|
#include "VoxImageFilterLinear.hpp"
|
||||||
|
#include "VoxImageFilterMedian.hpp"
|
||||||
|
#include "VoxImageFilterThreshold.hpp"
|
||||||
|
|||||||
@@ -23,280 +23,238 @@
|
|||||||
|
|
||||||
//////////////////////////////////////////////////////////////////////////////*/
|
//////////////////////////////////////////////////////////////////////////////*/
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
#ifndef VOXIMAGEFILTER_HPP
|
#ifndef VOXIMAGEFILTER_HPP
|
||||||
#define VOXIMAGEFILTER_HPP
|
#define VOXIMAGEFILTER_HPP
|
||||||
|
|
||||||
#include <Math/Dense.h>
|
|
||||||
#include "Math/StructuredData.h"
|
#include "Math/StructuredData.h"
|
||||||
#include "Math/VoxImage.h"
|
#include "Math/VoxImage.h"
|
||||||
#include "VoxImageFilter.h"
|
#include "VoxImageFilter.h"
|
||||||
|
#include <Math/Dense.h>
|
||||||
|
|
||||||
namespace uLib {
|
namespace uLib {
|
||||||
|
|
||||||
|
|
||||||
// KERNEL //////////////////////////////////////////////////////////////////////
|
// KERNEL //////////////////////////////////////////////////////////////////////
|
||||||
|
|
||||||
template < typename T >
|
template <typename T> class Kernel : public StructuredData {
|
||||||
class Kernel : public StructuredData {
|
typedef StructuredData BaseClass;
|
||||||
typedef StructuredData BaseClass;
|
|
||||||
public:
|
public:
|
||||||
Kernel(const Vector3i &size);
|
Kernel(const Vector3i &size);
|
||||||
|
|
||||||
inline T& operator[](const Vector3i &id) { return m_Data[Map(id)]; }
|
inline T &operator[](const Vector3i &id) { return m_Data[Map(id)]; }
|
||||||
inline T& operator[](const int &id) { return m_Data[id]; }
|
inline T &operator[](const int &id) { return m_Data[id]; }
|
||||||
inline int GetCenterData() const;
|
inline int GetCenterData() const;
|
||||||
|
|
||||||
inline std::vector<T> & Data() { return this->m_Data; }
|
inline DataAllocator<T> &Data() { return this->m_Data; }
|
||||||
|
|
||||||
inline const std::vector<T>& ConstData() const { return this->m_Data; }
|
inline const DataAllocator<T> &ConstData() const { return this->m_Data; }
|
||||||
|
|
||||||
void PrintSelf(std::ostream &o) const;
|
void PrintSelf(std::ostream &o) const;
|
||||||
|
|
||||||
private:
|
private:
|
||||||
std::vector<T> m_Data;
|
DataAllocator<T> m_Data;
|
||||||
};
|
};
|
||||||
|
|
||||||
template < typename T >
|
template <typename T>
|
||||||
Kernel<T>::Kernel(const Vector3i &size) :
|
Kernel<T>::Kernel(const Vector3i &size) : BaseClass(size), m_Data(size.prod()) {
|
||||||
BaseClass(size),
|
Interface::IsA<T, Interface::Voxel>();
|
||||||
m_Data(size.prod())
|
|
||||||
{
|
|
||||||
Interface::IsA<T,Interface::Voxel>();
|
|
||||||
}
|
}
|
||||||
|
|
||||||
template < typename T >
|
template <typename T> inline int Kernel<T>::GetCenterData() const {
|
||||||
inline int Kernel<T>::GetCenterData() const
|
static int center = Map(this->GetDims() / 2);
|
||||||
{
|
return center;
|
||||||
static int center = Map(this->GetDims() / 2);
|
|
||||||
return center;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
template < typename T >
|
template <typename T> void Kernel<T>::PrintSelf(std::ostream &o) const {
|
||||||
void Kernel<T>::PrintSelf(std::ostream &o) const
|
o << " Filter Kernel Dump [XZ_Y]: \n";
|
||||||
{
|
Vector3i index;
|
||||||
o << " Filter Kernel Dump [XZ_Y]: \n";
|
o << "\n Value: \n\n"
|
||||||
Vector3i index;
|
<< "------------------------------------------------- \n";
|
||||||
o << "\n Value: \n\n"
|
for (int y = 0; y < this->GetDims()(1); ++y) {
|
||||||
<< "------------------------------------------------- \n";
|
o << "[y=" << y << "]\n";
|
||||||
for (int y = 0 ; y < this->GetDims()(1); ++y ) {
|
for (int z = 0; z < this->GetDims()(2); ++z) {
|
||||||
o << "[y=" << y << "]\n";
|
for (int x = 0; x < this->GetDims()(0); ++x) {
|
||||||
for (int z = 0 ; z < this->GetDims()(2); ++z ) {
|
index << x, y, z;
|
||||||
for (int x = 0 ; x < this->GetDims()(0); ++x ) {
|
o << m_Data[Map(index)].Value << " ";
|
||||||
index << x,y,z;
|
}
|
||||||
o << m_Data[Map(index)].Value << " ";
|
o << "\n";
|
||||||
} o << "\n";
|
|
||||||
} o << " --------------------------------------------------- \n";
|
|
||||||
}
|
}
|
||||||
o << "\n Offset: \n"
|
o << " --------------------------------------------------- \n";
|
||||||
<< "------------------------------------------------- \n";
|
}
|
||||||
for (int y = 0 ; y < this->GetDims()(1); ++y ) {
|
o << "\n Offset: \n"
|
||||||
o << "[y=" << y << "]\n";
|
<< "------------------------------------------------- \n";
|
||||||
for (int z = 0 ; z < this->GetDims()(2); ++z ) {
|
for (int y = 0; y < this->GetDims()(1); ++y) {
|
||||||
for (int x = 0 ; x < this->GetDims()(0); ++x ) {
|
o << "[y=" << y << "]\n";
|
||||||
index << x,y,z;
|
for (int z = 0; z < this->GetDims()(2); ++z) {
|
||||||
o << m_Data[Map(index)].Count << " ";
|
for (int x = 0; x < this->GetDims()(0); ++x) {
|
||||||
} o << "\n";
|
index << x, y, z;
|
||||||
} o << " --------------------------------------------------- \n";
|
o << m_Data[Map(index)].Count << " ";
|
||||||
|
}
|
||||||
|
o << "\n";
|
||||||
}
|
}
|
||||||
|
o << " --------------------------------------------------- \n";
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
|
|
||||||
|
#define _TPL_ template <typename VoxelT, typename AlgorithmT>
|
||||||
|
#define _TPLT_ VoxelT, AlgorithmT
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
#define _TPL_ template < typename VoxelT , typename AlgorithmT >
|
|
||||||
#define _TPLT_ VoxelT,AlgorithmT
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
_TPL_
|
_TPL_
|
||||||
VoxImageFilter<_TPLT_>::VoxImageFilter(const Vector3i &size) :
|
VoxImageFilter<_TPLT_>::VoxImageFilter(const Vector3i &size)
|
||||||
m_KernelData(size),
|
: m_KernelData(size), t_Algoritm(static_cast<AlgorithmT *>(this)) {}
|
||||||
t_Algoritm(static_cast<AlgorithmT *>(this))
|
|
||||||
{
|
|
||||||
|
|
||||||
|
_TPL_
|
||||||
|
void VoxImageFilter<_TPLT_>::Run() {
|
||||||
|
VoxImage<VoxelT> buffer = *m_Image;
|
||||||
|
#pragma omp parallel for
|
||||||
|
for (int i = 0; i < m_Image->Data().size(); ++i)
|
||||||
|
m_Image->operator[](i).Value = this->t_Algoritm->Evaluate(buffer, i);
|
||||||
|
#pragma omp barrier
|
||||||
}
|
}
|
||||||
|
|
||||||
_TPL_
|
_TPL_
|
||||||
void VoxImageFilter<_TPLT_>::Run()
|
void VoxImageFilter<_TPLT_>::SetKernelOffset() {
|
||||||
{
|
Vector3i id(0, 0, 0);
|
||||||
VoxImage<VoxelT> buffer = *m_Image;
|
for (int z = 0; z < m_KernelData.GetDims()(2); ++z) {
|
||||||
#pragma omp parallel for
|
for (int x = 0; x < m_KernelData.GetDims()(0); ++x) {
|
||||||
for(int i=0 ; i < m_Image->Data().size() ; ++i)
|
for (int y = 0; y < m_KernelData.GetDims()(1); ++y) {
|
||||||
m_Image->operator [](i).Value = this->t_Algoritm->Evaluate(buffer,i);
|
id << x, y, z;
|
||||||
#pragma omp barrier
|
m_KernelData[id].Count = id.transpose() * m_Image->GetIncrements();
|
||||||
}
|
}
|
||||||
|
|
||||||
_TPL_
|
|
||||||
void VoxImageFilter<_TPLT_>::SetKernelOffset()
|
|
||||||
{
|
|
||||||
Vector3i id(0,0,0);
|
|
||||||
for( int z=0 ; z < m_KernelData.GetDims()(2); ++z ) {
|
|
||||||
for( int x=0 ; x < m_KernelData.GetDims()(0); ++x ) {
|
|
||||||
for( int y=0 ; y < m_KernelData.GetDims()(1); ++y ) {
|
|
||||||
id << x,y,z;
|
|
||||||
m_KernelData[id].Count = id.transpose() * m_Image->GetIncrements();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
_TPL_
|
_TPL_
|
||||||
float VoxImageFilter<_TPLT_>::Distance2(const Vector3i &v)
|
float VoxImageFilter<_TPLT_>::Distance2(const Vector3i &v) {
|
||||||
{
|
Vector3i tmp = v;
|
||||||
Vector3i tmp = v;
|
const Vector3i &dim = this->m_KernelData.GetDims();
|
||||||
const Vector3i &dim = this->m_KernelData.GetDims();
|
Vector3i center = dim / 2;
|
||||||
Vector3i center = dim / 2;
|
tmp = tmp - center;
|
||||||
tmp = tmp - center;
|
center = center.cwiseProduct(center);
|
||||||
center = center.cwiseProduct(center);
|
tmp = tmp.cwiseProduct(tmp);
|
||||||
tmp = tmp.cwiseProduct(tmp);
|
return (float)(tmp.sum()) /
|
||||||
return (float)(tmp.sum()) / (float)( center.sum() + 0.25 *
|
(float)(center.sum() +
|
||||||
(3 - (dim(0) % 2) - (dim(1) % 2) - (dim(2) % 2)));
|
0.25 * (3 - (dim(0) % 2) - (dim(1) % 2) - (dim(2) % 2)));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
_TPL_
|
_TPL_
|
||||||
void VoxImageFilter<_TPLT_>::SetKernelNumericXZY(const std::vector<float> &numeric)
|
void VoxImageFilter<_TPLT_>::SetKernelNumericXZY(
|
||||||
{
|
const std::vector<float> &numeric) {
|
||||||
// set data order //
|
// set data order //
|
||||||
StructuredData::Order order = m_KernelData.GetDataOrder();
|
StructuredData::Order order = m_KernelData.GetDataOrder();
|
||||||
//m_KernelData.SetDataOrder(StructuredData::XZY);
|
// m_KernelData.SetDataOrder(StructuredData::XZY);
|
||||||
Vector3i id;
|
Vector3i id;
|
||||||
int index = 0;
|
int index = 0;
|
||||||
for( int y=0 ; y < m_KernelData.GetDims()(1); ++y ) {
|
for (int y = 0; y < m_KernelData.GetDims()(1); ++y) {
|
||||||
for( int z=0 ; z < m_KernelData.GetDims()(2); ++z ) {
|
for (int z = 0; z < m_KernelData.GetDims()(2); ++z) {
|
||||||
for( int x=0 ; x < m_KernelData.GetDims()(0); ++x ) {
|
for (int x = 0; x < m_KernelData.GetDims()(0); ++x) {
|
||||||
id << x,y,z;
|
id << x, y, z;
|
||||||
m_KernelData[id].Value = numeric[index++];
|
m_KernelData[id].Value = numeric[index++];
|
||||||
}
|
}
|
||||||
}
|
|
||||||
}
|
}
|
||||||
//m_KernelData.SetDataOrder(order);
|
}
|
||||||
|
// m_KernelData.SetDataOrder(order);
|
||||||
}
|
}
|
||||||
|
|
||||||
_TPL_
|
_TPL_
|
||||||
void VoxImageFilter<_TPLT_>::SetKernelSpherical(float(* shape)(float))
|
void VoxImageFilter<_TPLT_>::SetKernelSpherical(float (*shape)(float)) {
|
||||||
{
|
Vector3i id;
|
||||||
Vector3i id;
|
for (int y = 0; y < m_KernelData.GetDims()(1); ++y) {
|
||||||
for( int y=0 ; y < m_KernelData.GetDims()(1); ++y ) {
|
for (int z = 0; z < m_KernelData.GetDims()(2); ++z) {
|
||||||
for( int z=0 ; z < m_KernelData.GetDims()(2); ++z ) {
|
for (int x = 0; x < m_KernelData.GetDims()(0); ++x) {
|
||||||
for( int x=0 ; x < m_KernelData.GetDims()(0); ++x ) {
|
id << x, y, z;
|
||||||
id << x,y,z;
|
m_KernelData[id].Value = shape(this->Distance2(id));
|
||||||
m_KernelData[id].Value = shape(this->Distance2(id));
|
}
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
_TPL_ template <class ShapeT>
|
_TPL_ template <class ShapeT>
|
||||||
void VoxImageFilter<_TPLT_>::SetKernelSpherical(ShapeT shape)
|
void VoxImageFilter<_TPLT_>::SetKernelSpherical(ShapeT shape) {
|
||||||
{
|
Interface::IsA<ShapeT, Interface::VoxImageFilterShape>();
|
||||||
Interface::IsA<ShapeT,Interface::VoxImageFilterShape>();
|
Vector3i id;
|
||||||
Vector3i id;
|
for (int y = 0; y < m_KernelData.GetDims()(1); ++y) {
|
||||||
for( int y=0 ; y < m_KernelData.GetDims()(1); ++y ) {
|
for (int z = 0; z < m_KernelData.GetDims()(2); ++z) {
|
||||||
for( int z=0 ; z < m_KernelData.GetDims()(2); ++z ) {
|
for (int x = 0; x < m_KernelData.GetDims()(0); ++x) {
|
||||||
for( int x=0 ; x < m_KernelData.GetDims()(0); ++x ) {
|
id << x, y, z;
|
||||||
id << x,y,z;
|
m_KernelData[id].Value = shape(this->Distance2(id));
|
||||||
m_KernelData[id].Value = shape(this->Distance2(id));
|
}
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
_TPL_
|
_TPL_
|
||||||
void VoxImageFilter<_TPLT_>::SetKernelWeightFunction(float (*shape)(const Vector3f &))
|
void VoxImageFilter<_TPLT_>::SetKernelWeightFunction(
|
||||||
{
|
float (*shape)(const Vector3f &)) {
|
||||||
const Vector3i &dim = m_KernelData.GetDims();
|
const Vector3i &dim = m_KernelData.GetDims();
|
||||||
Vector3i id;
|
Vector3i id;
|
||||||
Vector3f pt;
|
Vector3f pt;
|
||||||
for( int y=0 ; y < dim(1); ++y ) {
|
for (int y = 0; y < dim(1); ++y) {
|
||||||
for( int z=0 ; z < dim(2); ++z ) {
|
for (int z = 0; z < dim(2); ++z) {
|
||||||
for( int x=0 ; x < dim(0); ++x ) {
|
for (int x = 0; x < dim(0); ++x) {
|
||||||
// get voxels centroid coords from kernel center //
|
// get voxels centroid coords from kernel center //
|
||||||
id << x,y,z;
|
id << x, y, z;
|
||||||
pt << id(0) - dim(0)/2 + 0.5 * !(dim(0) % 2),
|
pt << id(0) - dim(0) / 2 + 0.5 * !(dim(0) % 2),
|
||||||
id(1) - dim(1)/2 + 0.5 * !(dim(1) % 2),
|
id(1) - dim(1) / 2 + 0.5 * !(dim(1) % 2),
|
||||||
id(2) - dim(2)/2 + 0.5 * !(dim(2) % 2);
|
id(2) - dim(2) / 2 + 0.5 * !(dim(2) % 2);
|
||||||
// compute function using given shape //
|
// compute function using given shape //
|
||||||
m_KernelData[id].Value = shape(pt);
|
m_KernelData[id].Value = shape(pt);
|
||||||
}
|
}
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
_TPL_ template < class ShapeT >
|
_TPL_ template <class ShapeT>
|
||||||
void VoxImageFilter<_TPLT_>::SetKernelWeightFunction(ShapeT shape)
|
void VoxImageFilter<_TPLT_>::SetKernelWeightFunction(ShapeT shape) {
|
||||||
{
|
Interface::IsA<ShapeT, Interface::VoxImageFilterShape>();
|
||||||
Interface::IsA<ShapeT,Interface::VoxImageFilterShape>();
|
const Vector3i &dim = m_KernelData.GetDims();
|
||||||
const Vector3i &dim = m_KernelData.GetDims();
|
Vector3i id;
|
||||||
Vector3i id;
|
Vector3f pt;
|
||||||
Vector3f pt;
|
for (int y = 0; y < dim(1); ++y) {
|
||||||
for( int y=0 ; y < dim(1); ++y ) {
|
for (int z = 0; z < dim(2); ++z) {
|
||||||
for( int z=0 ; z < dim(2); ++z ) {
|
for (int x = 0; x < dim(0); ++x) {
|
||||||
for( int x=0 ; x < dim(0); ++x ) {
|
// get voxels centroid coords from kernel center //
|
||||||
// get voxels centroid coords from kernel center //
|
id << x, y, z;
|
||||||
id << x,y,z;
|
pt << id(0) - dim(0) / 2 + 0.5 * !(dim(0) % 2),
|
||||||
pt << id(0) - dim(0)/2 + 0.5 * !(dim(0) % 2),
|
id(1) - dim(1) / 2 + 0.5 * !(dim(1) % 2),
|
||||||
id(1) - dim(1)/2 + 0.5 * !(dim(1) % 2),
|
id(2) - dim(2) / 2 + 0.5 * !(dim(2) % 2);
|
||||||
id(2) - dim(2)/2 + 0.5 * !(dim(2) % 2);
|
// compute function using given shape //
|
||||||
// compute function using given shape //
|
m_KernelData[id].Value = shape(pt);
|
||||||
m_KernelData[id].Value = shape(pt);
|
}
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
_TPL_
|
_TPL_
|
||||||
void VoxImageFilter<_TPLT_>::SetImage(Abstract::VoxImage *image)
|
void VoxImageFilter<_TPLT_>::SetImage(Abstract::VoxImage *image) {
|
||||||
{
|
this->m_Image = reinterpret_cast<VoxImage<VoxelT> *>(image);
|
||||||
this->m_Image = reinterpret_cast<VoxImage<VoxelT> *> (image);
|
this->SetKernelOffset();
|
||||||
this->SetKernelOffset();
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
_TPL_
|
_TPL_
|
||||||
float VoxImageFilter<_TPLT_>::Convolve(const VoxImage<VoxelT> &buffer, int index)
|
float VoxImageFilter<_TPLT_>::Convolve(const VoxImage<VoxelT> &buffer,
|
||||||
{
|
int index) {
|
||||||
const std::vector<VoxelT> &vbuf = buffer.ConstData();
|
const DataAllocator<VoxelT> &vbuf = buffer.ConstData();
|
||||||
const std::vector<VoxelT> &vker = m_KernelData.ConstData();
|
const DataAllocator<VoxelT> &vker = m_KernelData.ConstData();
|
||||||
int vox_size = vbuf.size();
|
int vox_size = vbuf.size();
|
||||||
int ker_size = vker.size();
|
int ker_size = vker.size();
|
||||||
int pos;
|
int pos;
|
||||||
float conv = 0, ksum = 0;
|
float conv = 0, ksum = 0;
|
||||||
for (int ik = 0; ik < ker_size; ++ik) {
|
for (int ik = 0; ik < ker_size; ++ik) {
|
||||||
pos = index + vker[ik].Count - vker[m_KernelData.GetCenterData()].Count;
|
pos = index + vker[ik].Count - vker[m_KernelData.GetCenterData()].Count;
|
||||||
pos = (pos + vox_size) % vox_size;
|
pos = (pos + vox_size) % vox_size;
|
||||||
conv += vbuf[pos].Value * vker[ik].Value;
|
conv += vbuf[pos].Value * vker[ik].Value;
|
||||||
ksum += vker[ik].Value;
|
ksum += vker[ik].Value;
|
||||||
}
|
}
|
||||||
return conv / ksum;
|
return conv / ksum;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
#undef _TPLT_
|
#undef _TPLT_
|
||||||
#undef _TPL_
|
#undef _TPL_
|
||||||
|
|
||||||
|
} // namespace uLib
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
#endif // VOXIMAGEFILTER_HPP
|
#endif // VOXIMAGEFILTER_HPP
|
||||||
|
|||||||
@@ -23,14 +23,12 @@
|
|||||||
|
|
||||||
//////////////////////////////////////////////////////////////////////////////*/
|
//////////////////////////////////////////////////////////////////////////////*/
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
#ifndef VOXIMAGEFILTER2NDSTAT_HPP
|
#ifndef VOXIMAGEFILTER2NDSTAT_HPP
|
||||||
#define VOXIMAGEFILTER2NDSTAT_HPP
|
#define VOXIMAGEFILTER2NDSTAT_HPP
|
||||||
|
|
||||||
#include <Math/Dense.h>
|
|
||||||
#include "Math/VoxImage.h"
|
#include "Math/VoxImage.h"
|
||||||
#include "VoxImageFilter.h"
|
#include "VoxImageFilter.h"
|
||||||
|
#include <Math/Dense.h>
|
||||||
|
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
///// VOXIMAGE FILTER ABTRIM /////////////////////////////////////////////////
|
///// VOXIMAGE FILTER ABTRIM /////////////////////////////////////////////////
|
||||||
@@ -39,45 +37,42 @@
|
|||||||
namespace uLib {
|
namespace uLib {
|
||||||
|
|
||||||
template <typename VoxelT>
|
template <typename VoxelT>
|
||||||
class VoxFilterAlgorithm2ndStat :
|
class VoxFilterAlgorithm2ndStat
|
||||||
public VoxImageFilter<VoxelT, VoxFilterAlgorithm2ndStat<VoxelT> > {
|
: public VoxImageFilter<VoxelT, VoxFilterAlgorithm2ndStat<VoxelT>> {
|
||||||
|
|
||||||
public:
|
public:
|
||||||
typedef VoxImageFilter<VoxelT, VoxFilterAlgorithm2ndStat<VoxelT> > BaseClass;
|
typedef VoxImageFilter<VoxelT, VoxFilterAlgorithm2ndStat<VoxelT>> BaseClass;
|
||||||
VoxFilterAlgorithm2ndStat(const Vector3i &size) :
|
VoxFilterAlgorithm2ndStat(const Vector3i &size) : BaseClass(size) {}
|
||||||
BaseClass(size)
|
|
||||||
{ }
|
|
||||||
|
|
||||||
float Evaluate(const VoxImage<VoxelT> &buffer, int index)
|
float Evaluate(const VoxImage<VoxelT> &buffer, int index) {
|
||||||
{
|
const DataAllocator<VoxelT> &vbuf = buffer.ConstData();
|
||||||
const std::vector<VoxelT> &vbuf = buffer.ConstData();
|
const DataAllocator<VoxelT> &vker = this->m_KernelData.ConstData();
|
||||||
const std::vector<VoxelT> &vker = this->m_KernelData.ConstData();
|
int vox_size = vbuf.size();
|
||||||
int vox_size = vbuf.size();
|
int ker_size = vker.size();
|
||||||
int ker_size = vker.size();
|
int pos;
|
||||||
int pos;
|
|
||||||
|
|
||||||
// mean //
|
// mean //
|
||||||
float conv = 0, ksum = 0;
|
float conv = 0, ksum = 0;
|
||||||
for (int ik = 0; ik < ker_size; ++ik) {
|
for (int ik = 0; ik < ker_size; ++ik) {
|
||||||
pos = index + vker[ik].Count - vker[this->m_KernelData.GetCenterData()].Count;
|
pos = index + vker[ik].Count -
|
||||||
pos = (pos + vox_size) % vox_size;
|
vker[this->m_KernelData.GetCenterData()].Count;
|
||||||
conv += vbuf[pos].Value * vker[ik].Value;
|
pos = (pos + vox_size) % vox_size;
|
||||||
ksum += vker[ik].Value;
|
conv += vbuf[pos].Value * vker[ik].Value;
|
||||||
}
|
ksum += vker[ik].Value;
|
||||||
float mean = conv / ksum;
|
|
||||||
|
|
||||||
// rms //
|
|
||||||
conv = 0;
|
|
||||||
for (int ik = 0; ik < ker_size; ++ik) {
|
|
||||||
pos = index + vker[ik].Count - vker[this->m_KernelData.GetCenterData()].Count;
|
|
||||||
pos = (pos + vox_size) % vox_size;
|
|
||||||
conv += pow((vbuf[pos].Value * vker[ik].Value) - mean , 2);
|
|
||||||
}
|
|
||||||
return conv / (vker.size() - 1) ;
|
|
||||||
}
|
}
|
||||||
|
float mean = conv / ksum;
|
||||||
|
|
||||||
|
// rms //
|
||||||
|
conv = 0;
|
||||||
|
for (int ik = 0; ik < ker_size; ++ik) {
|
||||||
|
pos = index + vker[ik].Count -
|
||||||
|
vker[this->m_KernelData.GetCenterData()].Count;
|
||||||
|
pos = (pos + vox_size) % vox_size;
|
||||||
|
conv += pow((vbuf[pos].Value * vker[ik].Value) - mean, 2);
|
||||||
|
}
|
||||||
|
return conv / (vker.size() - 1);
|
||||||
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
}
|
} // namespace uLib
|
||||||
#endif // VOXIMAGEFILTER2NDSTAT_HPP
|
#endif // VOXIMAGEFILTER2NDSTAT_HPP
|
||||||
|
|||||||
@@ -23,14 +23,12 @@
|
|||||||
|
|
||||||
//////////////////////////////////////////////////////////////////////////////*/
|
//////////////////////////////////////////////////////////////////////////////*/
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
#ifndef VOXIMAGEFILTERABTRIM_HPP
|
#ifndef VOXIMAGEFILTERABTRIM_HPP
|
||||||
#define VOXIMAGEFILTERABTRIM_HPP
|
#define VOXIMAGEFILTERABTRIM_HPP
|
||||||
|
|
||||||
#include <Math/Dense.h>
|
|
||||||
#include "Math/VoxImage.h"
|
#include "Math/VoxImage.h"
|
||||||
#include "VoxImageFilter.h"
|
#include "VoxImageFilter.h"
|
||||||
|
#include <Math/Dense.h>
|
||||||
|
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
///// VOXIMAGE FILTER ABTRIM /////////////////////////////////////////////////
|
///// VOXIMAGE FILTER ABTRIM /////////////////////////////////////////////////
|
||||||
@@ -38,142 +36,257 @@
|
|||||||
|
|
||||||
namespace uLib {
|
namespace uLib {
|
||||||
|
|
||||||
|
#if defined(USE_CUDA) && defined(__CUDACC__)
|
||||||
template <typename VoxelT>
|
template <typename VoxelT>
|
||||||
class VoxFilterAlgorithmAbtrim :
|
__global__ void ABTrimFilterKernel(const VoxelT *in, VoxelT *out,
|
||||||
public VoxImageFilter<VoxelT, VoxFilterAlgorithmAbtrim<VoxelT> > {
|
const VoxelT *kernel, int vox_size,
|
||||||
|
int ker_size, int center_count, int mAtrim,
|
||||||
|
int mBtrim) {
|
||||||
|
int index = blockIdx.x * blockDim.x + threadIdx.x;
|
||||||
|
if (index < vox_size) {
|
||||||
|
// Allocate space for sorting
|
||||||
|
extern __shared__ char shared_mem[];
|
||||||
|
VoxelT *mfh =
|
||||||
|
(VoxelT *)&shared_mem[threadIdx.x * ker_size * sizeof(VoxelT)];
|
||||||
|
|
||||||
struct KernelSortAscending
|
for (int i = 0; i < ker_size; ++i) {
|
||||||
{
|
mfh[i].Count = i;
|
||||||
bool operator()(const VoxelT& e1, const VoxelT& e2)
|
}
|
||||||
{ return e1.Value < e2.Value; }
|
|
||||||
};
|
for (int ik = 0; ik < ker_size; ik++) {
|
||||||
|
int pos = index + kernel[ik].Count - center_count;
|
||||||
|
if (pos < 0) {
|
||||||
|
pos += vox_size * ((-pos / vox_size) + 1);
|
||||||
|
}
|
||||||
|
pos = pos % vox_size;
|
||||||
|
mfh[ik].Value = in[pos].Value;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Simple bubble sort for small arrays
|
||||||
|
for (int i = 0; i < ker_size - 1; i++) {
|
||||||
|
for (int j = 0; j < ker_size - i - 1; j++) {
|
||||||
|
if (mfh[j].Value > mfh[j + 1].Value) {
|
||||||
|
VoxelT temp = mfh[j];
|
||||||
|
mfh[j] = mfh[j + 1];
|
||||||
|
mfh[j + 1] = temp;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
float ker_sum = 0;
|
||||||
|
float fconv = 0;
|
||||||
|
for (int ik = 0; ik < mAtrim; ik++) {
|
||||||
|
ker_sum += kernel[mfh[ik].Count].Value;
|
||||||
|
}
|
||||||
|
for (int ik = mAtrim; ik < ker_size - mBtrim; ik++) {
|
||||||
|
fconv += mfh[ik].Value * kernel[mfh[ik].Count].Value;
|
||||||
|
ker_sum += kernel[mfh[ik].Count].Value;
|
||||||
|
}
|
||||||
|
for (int ik = ker_size - mBtrim; ik < ker_size; ik++) {
|
||||||
|
ker_sum += kernel[mfh[ik].Count].Value;
|
||||||
|
}
|
||||||
|
|
||||||
|
out[index].Value = fconv / ker_sum;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
|
template <typename VoxelT>
|
||||||
|
class VoxFilterAlgorithmAbtrim
|
||||||
|
: public VoxImageFilter<VoxelT, VoxFilterAlgorithmAbtrim<VoxelT>> {
|
||||||
|
|
||||||
|
struct KernelSortAscending {
|
||||||
|
bool operator()(const VoxelT &e1, const VoxelT &e2) {
|
||||||
|
return e1.Value < e2.Value;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
public:
|
public:
|
||||||
typedef VoxImageFilter<VoxelT, VoxFilterAlgorithmAbtrim<VoxelT> > BaseClass;
|
typedef VoxImageFilter<VoxelT, VoxFilterAlgorithmAbtrim<VoxelT>> BaseClass;
|
||||||
VoxFilterAlgorithmAbtrim(const Vector3i &size) :
|
VoxFilterAlgorithmAbtrim(const Vector3i &size) : BaseClass(size) {
|
||||||
BaseClass(size)
|
mAtrim = 0;
|
||||||
{
|
mBtrim = 0;
|
||||||
mAtrim = 0;
|
}
|
||||||
mBtrim = 0;
|
|
||||||
|
#if defined(USE_CUDA) && defined(__CUDACC__)
|
||||||
|
void Run() {
|
||||||
|
if (this->m_Image->Data().GetDevice() == MemoryDevice::VRAM ||
|
||||||
|
this->m_KernelData.Data().GetDevice() == MemoryDevice::VRAM) {
|
||||||
|
|
||||||
|
this->m_Image->Data().MoveToVRAM();
|
||||||
|
this->m_KernelData.Data().MoveToVRAM();
|
||||||
|
|
||||||
|
VoxImage<VoxelT> buffer = *(this->m_Image);
|
||||||
|
buffer.Data().MoveToVRAM();
|
||||||
|
|
||||||
|
int vox_size = buffer.Data().size();
|
||||||
|
int ker_size = this->m_KernelData.Data().size();
|
||||||
|
|
||||||
|
VoxelT *d_img_out = this->m_Image->Data().GetVRAMData();
|
||||||
|
const VoxelT *d_img_in = buffer.Data().GetVRAMData();
|
||||||
|
const VoxelT *d_kernel = this->m_KernelData.Data().GetVRAMData();
|
||||||
|
int center_count =
|
||||||
|
this->m_KernelData[this->m_KernelData.GetCenterData()].Count;
|
||||||
|
|
||||||
|
int threadsPerBlock = 256;
|
||||||
|
int blocksPerGrid = (vox_size + threadsPerBlock - 1) / threadsPerBlock;
|
||||||
|
size_t shared_mem_size = threadsPerBlock * ker_size * sizeof(VoxelT);
|
||||||
|
|
||||||
|
ABTrimFilterKernel<<<blocksPerGrid, threadsPerBlock, shared_mem_size>>>(
|
||||||
|
d_img_in, d_img_out, d_kernel, vox_size, ker_size, center_count,
|
||||||
|
mAtrim, mBtrim);
|
||||||
|
cudaDeviceSynchronize();
|
||||||
|
} else {
|
||||||
|
BaseClass::Run();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
|
float Evaluate(const VoxImage<VoxelT> &buffer, int index) {
|
||||||
|
const DataAllocator<VoxelT> &vbuf = buffer.ConstData();
|
||||||
|
const DataAllocator<VoxelT> &vker = this->m_KernelData.ConstData();
|
||||||
|
int vox_size = vbuf.size();
|
||||||
|
int ker_size = vker.size();
|
||||||
|
int pos;
|
||||||
|
|
||||||
|
std::vector<VoxelT> mfh(ker_size);
|
||||||
|
for (int i = 0; i < ker_size; ++i)
|
||||||
|
mfh[i].Count = i; // index key for ordering function
|
||||||
|
for (int ik = 0; ik < ker_size; ik++) {
|
||||||
|
pos = index + vker[ik].Count -
|
||||||
|
vker[this->m_KernelData.GetCenterData()].Count;
|
||||||
|
pos = (pos + vox_size) % vox_size;
|
||||||
|
mfh[ik].Value = vbuf[pos].Value;
|
||||||
}
|
}
|
||||||
|
|
||||||
float Evaluate(const VoxImage<VoxelT> &buffer, int index)
|
std::sort(mfh.begin(), mfh.end(), KernelSortAscending());
|
||||||
{
|
float ker_sum = 0;
|
||||||
const std::vector<VoxelT> &vbuf = buffer.ConstData();
|
float fconv = 0;
|
||||||
const std::vector<VoxelT> &vker = this->m_KernelData.ConstData();
|
for (int ik = 0; ik < mAtrim; ik++)
|
||||||
int vox_size = vbuf.size();
|
ker_sum += vker[mfh[ik].Count].Value;
|
||||||
int ker_size = vker.size();
|
for (int ik = mAtrim; ik < ker_size - mBtrim; ik++) {
|
||||||
int pos;
|
fconv += mfh[ik].Value * vker[mfh[ik].Count].Value; // convloution //
|
||||||
|
ker_sum += vker[mfh[ik].Count].Value;
|
||||||
std::vector<VoxelT> mfh(ker_size);
|
|
||||||
for (int i = 0; i < ker_size; ++i)
|
|
||||||
mfh[i].Count = i; //index key for ordering function
|
|
||||||
for (int ik = 0; ik < ker_size; ik++) {
|
|
||||||
pos = index + vker[ik].Count - vker[this->m_KernelData.GetCenterData()].Count;
|
|
||||||
pos = (pos + vox_size) % vox_size;
|
|
||||||
mfh[ik].Value = vbuf[pos].Value;
|
|
||||||
}
|
|
||||||
|
|
||||||
std::sort(mfh.begin(), mfh.end(), KernelSortAscending());
|
|
||||||
float ker_sum = 0;
|
|
||||||
float fconv = 0;
|
|
||||||
for (int ik = 0; ik < mAtrim; ik++)
|
|
||||||
ker_sum += vker[ mfh[ik].Count ].Value;
|
|
||||||
for (int ik = mAtrim; ik < ker_size - mBtrim; ik++) {
|
|
||||||
fconv += mfh[ik].Value * vker[ mfh[ik].Count ].Value; // convloution //
|
|
||||||
ker_sum += vker[ mfh[ik].Count ].Value;
|
|
||||||
}
|
|
||||||
for (int ik = ker_size - mBtrim; ik < ker_size; ik++)
|
|
||||||
ker_sum += vker[ mfh[ik].Count ].Value;
|
|
||||||
|
|
||||||
return fconv / ker_sum;
|
|
||||||
}
|
}
|
||||||
|
for (int ik = ker_size - mBtrim; ik < ker_size; ik++)
|
||||||
|
ker_sum += vker[mfh[ik].Count].Value;
|
||||||
|
|
||||||
inline void SetABTrim(int a, int b) { mAtrim = a; mBtrim = b; }
|
return fconv / ker_sum;
|
||||||
|
}
|
||||||
|
|
||||||
|
inline void SetABTrim(int a, int b) {
|
||||||
|
mAtrim = a;
|
||||||
|
mBtrim = b;
|
||||||
|
}
|
||||||
|
|
||||||
private:
|
private:
|
||||||
int mAtrim;
|
int mAtrim;
|
||||||
int mBtrim;
|
int mBtrim;
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
// Roberspierre Filter //
|
// Roberspierre Filter //
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
template <typename VoxelT>
|
template <typename VoxelT>
|
||||||
class VoxFilterAlgorithmSPR :
|
class VoxFilterAlgorithmSPR
|
||||||
public VoxImageFilter<VoxelT, VoxFilterAlgorithmSPR<VoxelT> > {
|
: public VoxImageFilter<VoxelT, VoxFilterAlgorithmSPR<VoxelT>> {
|
||||||
|
|
||||||
struct KernelSortAscending
|
struct KernelSortAscending {
|
||||||
{
|
bool operator()(const VoxelT &e1, const VoxelT &e2) {
|
||||||
bool operator()(const VoxelT& e1, const VoxelT& e2)
|
return e1.Value < e2.Value;
|
||||||
{ return e1.Value < e2.Value; }
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
public:
|
public:
|
||||||
typedef VoxImageFilter<VoxelT, VoxFilterAlgorithmSPR<VoxelT> > BaseClass;
|
typedef VoxImageFilter<VoxelT, VoxFilterAlgorithmSPR<VoxelT>> BaseClass;
|
||||||
VoxFilterAlgorithmSPR(const Vector3i &size) :
|
VoxFilterAlgorithmSPR(const Vector3i &size) : BaseClass(size) {
|
||||||
BaseClass(size)
|
mAtrim = 0;
|
||||||
{
|
mBtrim = 0;
|
||||||
mAtrim = 0;
|
}
|
||||||
mBtrim = 0;
|
|
||||||
|
#if defined(USE_CUDA) && defined(__CUDACC__)
|
||||||
|
void Run() {
|
||||||
|
if (this->m_Image->Data().GetDevice() == MemoryDevice::VRAM ||
|
||||||
|
this->m_KernelData.Data().GetDevice() == MemoryDevice::VRAM) {
|
||||||
|
|
||||||
|
this->m_Image->Data().MoveToVRAM();
|
||||||
|
this->m_KernelData.Data().MoveToVRAM();
|
||||||
|
|
||||||
|
VoxImage<VoxelT> buffer = *(this->m_Image);
|
||||||
|
buffer.Data().MoveToVRAM();
|
||||||
|
|
||||||
|
int vox_size = buffer.Data().size();
|
||||||
|
int ker_size = this->m_KernelData.Data().size();
|
||||||
|
|
||||||
|
VoxelT *d_img_out = this->m_Image->Data().GetVRAMData();
|
||||||
|
const VoxelT *d_img_in = buffer.Data().GetVRAMData();
|
||||||
|
const VoxelT *d_kernel = this->m_KernelData.Data().GetVRAMData();
|
||||||
|
int center_count =
|
||||||
|
this->m_KernelData[this->m_KernelData.GetCenterData()].Count;
|
||||||
|
|
||||||
|
int threadsPerBlock = 256;
|
||||||
|
int blocksPerGrid = (vox_size + threadsPerBlock - 1) / threadsPerBlock;
|
||||||
|
size_t shared_mem_size = threadsPerBlock * ker_size * sizeof(VoxelT);
|
||||||
|
|
||||||
|
ABTrimFilterKernel<<<blocksPerGrid, threadsPerBlock, shared_mem_size>>>(
|
||||||
|
d_img_in, d_img_out, d_kernel, vox_size, ker_size, center_count,
|
||||||
|
mAtrim, mBtrim);
|
||||||
|
cudaDeviceSynchronize();
|
||||||
|
} else {
|
||||||
|
BaseClass::Run();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
|
float Evaluate(const VoxImage<VoxelT> &buffer, int index) {
|
||||||
|
const DataAllocator<VoxelT> &vbuf = buffer.ConstData();
|
||||||
|
const DataAllocator<VoxelT> &vker = this->m_KernelData.ConstData();
|
||||||
|
int vox_size = vbuf.size();
|
||||||
|
int ker_size = vker.size();
|
||||||
|
int pos;
|
||||||
|
|
||||||
|
std::vector<VoxelT> mfh(ker_size);
|
||||||
|
for (int i = 0; i < ker_size; ++i)
|
||||||
|
mfh[i].Count = i; // index key for ordering function
|
||||||
|
for (int ik = 0; ik < ker_size; ik++) {
|
||||||
|
pos = index + vker[ik].Count -
|
||||||
|
vker[this->m_KernelData.GetCenterData()].Count;
|
||||||
|
pos = (pos + vox_size) % vox_size;
|
||||||
|
mfh[ik].Value = vbuf[pos].Value;
|
||||||
}
|
}
|
||||||
|
|
||||||
float Evaluate(const VoxImage<VoxelT> &buffer, int index)
|
std::sort(mfh.begin(), mfh.end(), KernelSortAscending());
|
||||||
{
|
float spr = vbuf[index].Value;
|
||||||
const std::vector<VoxelT> &vbuf = buffer.ConstData();
|
if ((mAtrim > 0 && spr <= mfh[mAtrim - 1].Value) ||
|
||||||
const std::vector<VoxelT> &vker = this->m_KernelData.ConstData();
|
(mBtrim > 0 && spr >= mfh[ker_size - mBtrim].Value)) {
|
||||||
int vox_size = vbuf.size();
|
float ker_sum = 0;
|
||||||
int ker_size = vker.size();
|
float fconv = 0;
|
||||||
int pos;
|
for (int ik = 0; ik < mAtrim; ik++)
|
||||||
|
ker_sum += vker[mfh[ik].Count].Value;
|
||||||
|
for (int ik = mAtrim; ik < ker_size - mBtrim; ik++) {
|
||||||
|
fconv += mfh[ik].Value * vker[mfh[ik].Count].Value;
|
||||||
|
ker_sum += vker[mfh[ik].Count].Value;
|
||||||
|
}
|
||||||
|
for (int ik = ker_size - mBtrim; ik < ker_size; ik++)
|
||||||
|
ker_sum += vker[mfh[ik].Count].Value;
|
||||||
|
|
||||||
std::vector<VoxelT> mfh(ker_size);
|
return fconv / ker_sum;
|
||||||
for (int i = 0; i < ker_size; ++i)
|
} else
|
||||||
mfh[i].Count = i; //index key for ordering function
|
return spr;
|
||||||
for (int ik = 0; ik < ker_size; ik++) {
|
}
|
||||||
pos = index + vker[ik].Count -
|
|
||||||
vker[this->m_KernelData.GetCenterData()].Count;
|
|
||||||
pos = (pos + vox_size) % vox_size;
|
|
||||||
mfh[ik].Value = vbuf[pos].Value;
|
|
||||||
}
|
|
||||||
|
|
||||||
std::sort(mfh.begin(), mfh.end(), KernelSortAscending());
|
inline void SetABTrim(int a, int b) {
|
||||||
float spr = vbuf[index].Value;
|
mAtrim = a;
|
||||||
if( (mAtrim > 0 && spr <= mfh[mAtrim-1].Value) ||
|
mBtrim = b;
|
||||||
(mBtrim > 0 && spr >= mfh[ker_size - mBtrim].Value) )
|
}
|
||||||
{
|
|
||||||
float ker_sum = 0;
|
|
||||||
float fconv = 0;
|
|
||||||
for (int ik = 0; ik < mAtrim; ik++)
|
|
||||||
ker_sum += vker[ mfh[ik].Count ].Value;
|
|
||||||
for (int ik = mAtrim; ik < ker_size - mBtrim; ik++) {
|
|
||||||
fconv += mfh[ik].Value * vker[ mfh[ik].Count ].Value;
|
|
||||||
ker_sum += vker[ mfh[ik].Count ].Value;
|
|
||||||
}
|
|
||||||
for (int ik = ker_size - mBtrim; ik < ker_size; ik++)
|
|
||||||
ker_sum += vker[ mfh[ik].Count ].Value;
|
|
||||||
|
|
||||||
return fconv / ker_sum;
|
|
||||||
}
|
|
||||||
else
|
|
||||||
return spr;
|
|
||||||
}
|
|
||||||
|
|
||||||
inline void SetABTrim(int a, int b) { mAtrim = a; mBtrim = b; }
|
|
||||||
|
|
||||||
private:
|
private:
|
||||||
int mAtrim;
|
int mAtrim;
|
||||||
int mBtrim;
|
int mBtrim;
|
||||||
};
|
};
|
||||||
|
|
||||||
|
} // namespace uLib
|
||||||
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
#endif // VOXIMAGEFILTERABTRIM_HPP
|
#endif // VOXIMAGEFILTERABTRIM_HPP
|
||||||
|
|||||||
@@ -23,14 +23,12 @@
|
|||||||
|
|
||||||
//////////////////////////////////////////////////////////////////////////////*/
|
//////////////////////////////////////////////////////////////////////////////*/
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
#ifndef VOXIMAGEFILTERBILATERAL_HPP
|
#ifndef VOXIMAGEFILTERBILATERAL_HPP
|
||||||
#define VOXIMAGEFILTERBILATERAL_HPP
|
#define VOXIMAGEFILTERBILATERAL_HPP
|
||||||
|
|
||||||
#include <Math/Dense.h>
|
|
||||||
#include "Math/VoxImage.h"
|
#include "Math/VoxImage.h"
|
||||||
#include "VoxImageFilter.h"
|
#include "VoxImageFilter.h"
|
||||||
|
#include <Math/Dense.h>
|
||||||
|
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
///// VOXIMAGE FILTER LINEAR /////////////////////////////////////////////////
|
///// VOXIMAGE FILTER LINEAR /////////////////////////////////////////////////
|
||||||
@@ -38,115 +36,119 @@
|
|||||||
|
|
||||||
namespace uLib {
|
namespace uLib {
|
||||||
|
|
||||||
|
|
||||||
template <typename VoxelT>
|
template <typename VoxelT>
|
||||||
class VoxFilterAlgorithmBilateral :
|
class VoxFilterAlgorithmBilateral
|
||||||
public VoxImageFilter<VoxelT, VoxFilterAlgorithmBilateral<VoxelT> > {
|
: public VoxImageFilter<VoxelT, VoxFilterAlgorithmBilateral<VoxelT>> {
|
||||||
public:
|
public:
|
||||||
typedef VoxImageFilter<VoxelT, VoxFilterAlgorithmBilateral<VoxelT> > BaseClass;
|
typedef VoxImageFilter<VoxelT, VoxFilterAlgorithmBilateral<VoxelT>> BaseClass;
|
||||||
VoxFilterAlgorithmBilateral(const Vector3i &size) : BaseClass(size) {
|
VoxFilterAlgorithmBilateral(const Vector3i &size) : BaseClass(size) {
|
||||||
m_sigma = 1;
|
m_sigma = 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
float Evaluate(const VoxImage<VoxelT> &buffer, int index)
|
float Evaluate(const VoxImage<VoxelT> &buffer, int index) {
|
||||||
{
|
const DataAllocator<VoxelT> &vbuf = buffer.ConstData();
|
||||||
const std::vector<VoxelT> &vbuf = buffer.ConstData();
|
const DataAllocator<VoxelT> &vker = this->m_KernelData.ConstData();
|
||||||
const std::vector<VoxelT> &vker = this->m_KernelData.ConstData();
|
int vox_size = vbuf.size();
|
||||||
int vox_size = vbuf.size();
|
int ker_size = vker.size();
|
||||||
int ker_size = vker.size();
|
int pos;
|
||||||
int pos;
|
float conv = 0, ksum = 0;
|
||||||
float conv = 0, ksum = 0;
|
float gamma_smooth;
|
||||||
float gamma_smooth;
|
for (int ik = 0; ik < ker_size; ++ik) {
|
||||||
for (int ik = 0; ik < ker_size; ++ik) {
|
// if (ik==this->m_KernelData.GetCenterData()) continue;
|
||||||
// if (ik==this->m_KernelData.GetCenterData()) continue;
|
pos = index + vker[ik].Count -
|
||||||
pos = index + vker[ik].Count - vker[this->m_KernelData.GetCenterData()].Count;
|
vker[this->m_KernelData.GetCenterData()].Count;
|
||||||
pos = (pos + vox_size) % vox_size;
|
pos = (pos + vox_size) % vox_size;
|
||||||
gamma_smooth = compute_gauss( fabs(vbuf[index].Value - vbuf[pos].Value) * 1.E6 );
|
gamma_smooth =
|
||||||
conv += vbuf[pos].Value * vker[ik].Value * gamma_smooth;
|
compute_gauss(fabs(vbuf[index].Value - vbuf[pos].Value) * 1.E6);
|
||||||
ksum += vker[ik].Value * gamma_smooth;
|
conv += vbuf[pos].Value * vker[ik].Value * gamma_smooth;
|
||||||
}
|
ksum += vker[ik].Value * gamma_smooth;
|
||||||
return conv / ksum;
|
|
||||||
}
|
}
|
||||||
|
return conv / ksum;
|
||||||
|
}
|
||||||
|
|
||||||
inline void SetIntensitySigma(const float s) { m_sigma = s; }
|
inline void SetIntensitySigma(const float s) { m_sigma = s; }
|
||||||
|
|
||||||
private:
|
private:
|
||||||
inline float compute_gauss(const float x) {
|
inline float compute_gauss(const float x) {
|
||||||
return 1/(sqrt(2*M_PI)* m_sigma) * exp(-0.5*(x*x)/(m_sigma*m_sigma));
|
return 1 / (sqrt(2 * M_PI) * m_sigma) *
|
||||||
}
|
exp(-0.5 * (x * x) / (m_sigma * m_sigma));
|
||||||
|
}
|
||||||
|
|
||||||
Scalarf m_sigma;
|
Scalarf m_sigma;
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|
||||||
template <typename VoxelT>
|
template <typename VoxelT>
|
||||||
class VoxFilterAlgorithmBilateralTrim :
|
class VoxFilterAlgorithmBilateralTrim
|
||||||
public VoxImageFilter<VoxelT, VoxFilterAlgorithmBilateralTrim<VoxelT> > {
|
: public VoxImageFilter<VoxelT, VoxFilterAlgorithmBilateralTrim<VoxelT>> {
|
||||||
|
|
||||||
typedef std::pair<float,float> FPair;
|
typedef std::pair<float, float> FPair;
|
||||||
|
|
||||||
struct KernelSortAscending
|
struct KernelSortAscending {
|
||||||
{
|
bool operator()(const FPair &e1, const FPair &e2) {
|
||||||
bool operator()(const FPair& e1, const FPair& e2)
|
return e1.second < e2.second;
|
||||||
{ return e1.second < e2.second; }
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
public:
|
public:
|
||||||
typedef VoxImageFilter<VoxelT, VoxFilterAlgorithmBilateralTrim<VoxelT> > BaseClass;
|
typedef VoxImageFilter<VoxelT, VoxFilterAlgorithmBilateralTrim<VoxelT>>
|
||||||
VoxFilterAlgorithmBilateralTrim(const Vector3i &size) : BaseClass(size) {
|
BaseClass;
|
||||||
m_sigma = 1;
|
VoxFilterAlgorithmBilateralTrim(const Vector3i &size) : BaseClass(size) {
|
||||||
mAtrim = 0;
|
m_sigma = 1;
|
||||||
mBtrim = 0;
|
mAtrim = 0;
|
||||||
|
mBtrim = 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
float Evaluate(const VoxImage<VoxelT> &buffer, int index) {
|
||||||
|
const DataAllocator<VoxelT> &vbuf = buffer.ConstData();
|
||||||
|
const DataAllocator<VoxelT> &vker = this->m_KernelData.ConstData();
|
||||||
|
int img_size = vbuf.size();
|
||||||
|
int ker_size = vker.size();
|
||||||
|
int pos;
|
||||||
|
|
||||||
|
std::vector<FPair> mfh(ker_size);
|
||||||
|
for (int i = 0; i < ker_size; ++i)
|
||||||
|
mfh[i].first = vker[i].Value; // kernel value in first
|
||||||
|
for (int ik = 0; ik < ker_size; ik++) {
|
||||||
|
pos = index + vker[ik].Count -
|
||||||
|
vker[this->m_KernelData.GetCenterData()].Count;
|
||||||
|
pos = (pos + img_size) % img_size;
|
||||||
|
mfh[ik].second = vbuf[pos].Value; // image value in second
|
||||||
}
|
}
|
||||||
|
std::sort(mfh.begin(), mfh.end(), KernelSortAscending());
|
||||||
|
|
||||||
float Evaluate(const VoxImage<VoxelT> &buffer, int index)
|
float conv = 0, ksum = 0;
|
||||||
{
|
float gamma_smooth;
|
||||||
const std::vector<VoxelT> &vbuf = buffer.ConstData();
|
// for (int ik = 0; ik < mAtrim; ik++)
|
||||||
const std::vector<VoxelT> &vker = this->m_KernelData.ConstData();
|
// ksum += mfh[ik].first;
|
||||||
int img_size = vbuf.size();
|
for (int ik = mAtrim; ik < ker_size - mBtrim; ik++) {
|
||||||
int ker_size = vker.size();
|
gamma_smooth =
|
||||||
int pos;
|
compute_gauss(fabs(vbuf[index].Value - mfh[ik].second) * 1.E6);
|
||||||
|
conv += mfh[ik].first * mfh[ik].second * gamma_smooth;
|
||||||
|
ksum += mfh[ik].first * gamma_smooth;
|
||||||
|
|
||||||
std::vector<FPair> mfh(ker_size);
|
|
||||||
for (int i = 0; i < ker_size; ++i)
|
|
||||||
mfh[i].first = vker[i].Value; // kernel value in first
|
|
||||||
for (int ik = 0; ik < ker_size; ik++) {
|
|
||||||
pos = index + vker[ik].Count - vker[this->m_KernelData.GetCenterData()].Count;
|
|
||||||
pos = (pos + img_size) % img_size;
|
|
||||||
mfh[ik].second = vbuf[pos].Value; // image value in second
|
|
||||||
}
|
|
||||||
std::sort(mfh.begin(), mfh.end(), KernelSortAscending());
|
|
||||||
|
|
||||||
float conv = 0, ksum = 0;
|
|
||||||
float gamma_smooth;
|
|
||||||
// for (int ik = 0; ik < mAtrim; ik++)
|
|
||||||
// ksum += mfh[ik].first;
|
|
||||||
for (int ik = mAtrim; ik < ker_size - mBtrim; ik++) {
|
|
||||||
gamma_smooth = compute_gauss( fabs(vbuf[index].Value - mfh[ik].second) * 1.E6 );
|
|
||||||
conv += mfh[ik].first * mfh[ik].second * gamma_smooth;
|
|
||||||
ksum += mfh[ik].first * gamma_smooth;
|
|
||||||
}
|
|
||||||
// for (int ik = ker_size - mBtrim; ik < ker_size; ik++)
|
|
||||||
// ksum += mfh[ik].first;
|
|
||||||
|
|
||||||
return conv / ksum;
|
|
||||||
}
|
}
|
||||||
|
// for (int ik = ker_size - mBtrim; ik < ker_size; ik++)
|
||||||
|
// ksum += mfh[ik].first;
|
||||||
|
|
||||||
inline void SetIntensitySigma(const float s) { m_sigma = s; }
|
return conv / ksum;
|
||||||
inline void SetABTrim(int a, int b) { mAtrim = a; mBtrim = b; }
|
}
|
||||||
|
|
||||||
|
inline void SetIntensitySigma(const float s) { m_sigma = s; }
|
||||||
|
inline void SetABTrim(int a, int b) {
|
||||||
|
mAtrim = a;
|
||||||
|
mBtrim = b;
|
||||||
|
}
|
||||||
|
|
||||||
private:
|
private:
|
||||||
inline float compute_gauss(const float x) {
|
inline float compute_gauss(const float x) {
|
||||||
return 1/(sqrt(2*M_PI)* m_sigma) * exp(-0.5*(x*x)/(m_sigma*m_sigma));
|
return 1 / (sqrt(2 * M_PI) * m_sigma) *
|
||||||
}
|
exp(-0.5 * (x * x) / (m_sigma * m_sigma));
|
||||||
|
}
|
||||||
|
|
||||||
Scalarf m_sigma;
|
Scalarf m_sigma;
|
||||||
int mAtrim;
|
int mAtrim;
|
||||||
int mBtrim;
|
int mBtrim;
|
||||||
};
|
};
|
||||||
|
|
||||||
}
|
} // namespace uLib
|
||||||
|
|
||||||
#endif // VOXIMAGEFILTERBILATERAL_HPP
|
#endif // VOXIMAGEFILTERBILATERAL_HPP
|
||||||
|
|||||||
@@ -23,14 +23,12 @@
|
|||||||
|
|
||||||
//////////////////////////////////////////////////////////////////////////////*/
|
//////////////////////////////////////////////////////////////////////////////*/
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
#ifndef VOXIMAGEFILTERCUSTOM_HPP
|
#ifndef VOXIMAGEFILTERCUSTOM_HPP
|
||||||
#define VOXIMAGEFILTERCUSTOM_HPP
|
#define VOXIMAGEFILTERCUSTOM_HPP
|
||||||
|
|
||||||
#include <Math/Dense.h>
|
|
||||||
#include "Math/VoxImage.h"
|
#include "Math/VoxImage.h"
|
||||||
#include "VoxImageFilter.h"
|
#include "VoxImageFilter.h"
|
||||||
|
#include <Math/Dense.h>
|
||||||
|
|
||||||
#define likely(expr) __builtin_expect(!!(expr), 1)
|
#define likely(expr) __builtin_expect(!!(expr), 1)
|
||||||
|
|
||||||
@@ -41,50 +39,50 @@
|
|||||||
namespace uLib {
|
namespace uLib {
|
||||||
|
|
||||||
template <typename VoxelT>
|
template <typename VoxelT>
|
||||||
class VoxFilterAlgorithmCustom :
|
class VoxFilterAlgorithmCustom
|
||||||
public VoxImageFilter<VoxelT, VoxFilterAlgorithmCustom<VoxelT> > {
|
: public VoxImageFilter<VoxelT, VoxFilterAlgorithmCustom<VoxelT>> {
|
||||||
|
|
||||||
|
typedef float (*FunctionPt)(const std::vector<Scalarf> &);
|
||||||
|
|
||||||
typedef float (* FunctionPt)(const std::vector<Scalarf> &);
|
|
||||||
public:
|
public:
|
||||||
typedef VoxImageFilter<VoxelT, VoxFilterAlgorithmCustom<VoxelT> > BaseClass;
|
typedef VoxImageFilter<VoxelT, VoxFilterAlgorithmCustom<VoxelT>> BaseClass;
|
||||||
VoxFilterAlgorithmCustom(const Vector3i &size) :
|
VoxFilterAlgorithmCustom(const Vector3i &size)
|
||||||
BaseClass(size), m_CustomEvaluate(NULL)
|
: BaseClass(size), m_CustomEvaluate(NULL) {}
|
||||||
{}
|
|
||||||
|
|
||||||
float Evaluate(const VoxImage<VoxelT> &buffer, int index)
|
float Evaluate(const VoxImage<VoxelT> &buffer, int index) {
|
||||||
{
|
if (likely(m_CustomEvaluate)) {
|
||||||
if(likely(m_CustomEvaluate)) {
|
const DataAllocator<VoxelT> &vbuf = buffer.ConstData();
|
||||||
const std::vector<VoxelT> &vbuf = buffer.ConstData();
|
const DataAllocator<VoxelT> &vker = this->m_KernelData.ConstData();
|
||||||
const std::vector<VoxelT> &vker = this->m_KernelData.ConstData();
|
int vox_size = vbuf.size();
|
||||||
int vox_size = vbuf.size();
|
int ker_size = vker.size();
|
||||||
int ker_size = vker.size();
|
int pos;
|
||||||
int pos;
|
|
||||||
|
|
||||||
float ker_sum = 0;
|
float ker_sum = 0;
|
||||||
std::vector<Scalarf> mfh(ker_size);
|
std::vector<Scalarf> mfh(ker_size);
|
||||||
for (int ik = 0; ik < ker_size; ik++) {
|
for (int ik = 0; ik < ker_size; ik++) {
|
||||||
pos = index + vker[ik].Count - vker[this->m_KernelData.GetCenterData()].Count;
|
pos = index + vker[ik].Count -
|
||||||
pos = (pos + vox_size) % vox_size;
|
vker[this->m_KernelData.GetCenterData()].Count;
|
||||||
mfh[ik] = vbuf[pos].Value * vker[ik].Value;
|
pos = (pos + vox_size) % vox_size;
|
||||||
ker_sum += vker[ik].Value;
|
mfh[ik] = vbuf[pos].Value * vker[ik].Value;
|
||||||
}
|
ker_sum += vker[ik].Value;
|
||||||
|
}
|
||||||
return this->m_CustomEvaluate(mfh);
|
|
||||||
}
|
|
||||||
else
|
|
||||||
std::cerr << "Custom evaluate function is NULL \n" <<
|
|
||||||
"No operation performed by filter.\n";
|
|
||||||
|
|
||||||
|
return this->m_CustomEvaluate(mfh);
|
||||||
|
} else {
|
||||||
|
std::cerr << "Custom evaluate function is NULL \n"
|
||||||
|
<< "No operation performed by filter.\n";
|
||||||
|
return 0;
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
inline void SetCustomEvaluate(FunctionPt funPt) { this->m_CustomEvaluate = funPt; }
|
inline void SetCustomEvaluate(FunctionPt funPt) {
|
||||||
|
this->m_CustomEvaluate = funPt;
|
||||||
|
}
|
||||||
|
|
||||||
private:
|
private:
|
||||||
FunctionPt m_CustomEvaluate;
|
FunctionPt m_CustomEvaluate;
|
||||||
};
|
};
|
||||||
|
|
||||||
}
|
} // namespace uLib
|
||||||
|
|
||||||
|
|
||||||
#endif // VOXIMAGEFILTERCUSTOM_HPP
|
#endif // VOXIMAGEFILTERCUSTOM_HPP
|
||||||
|
|||||||
@@ -23,14 +23,12 @@
|
|||||||
|
|
||||||
//////////////////////////////////////////////////////////////////////////////*/
|
//////////////////////////////////////////////////////////////////////////////*/
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
#ifndef VOXIMAGEFILTERLINEAR_HPP
|
#ifndef VOXIMAGEFILTERLINEAR_HPP
|
||||||
#define VOXIMAGEFILTERLINEAR_HPP
|
#define VOXIMAGEFILTERLINEAR_HPP
|
||||||
|
|
||||||
#include <Math/Dense.h>
|
|
||||||
#include "Math/VoxImage.h"
|
#include "Math/VoxImage.h"
|
||||||
#include "VoxImageFilter.h"
|
#include "VoxImageFilter.h"
|
||||||
|
#include <Math/Dense.h>
|
||||||
|
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
///// VOXIMAGE FILTER LINEAR /////////////////////////////////////////////////
|
///// VOXIMAGE FILTER LINEAR /////////////////////////////////////////////////
|
||||||
@@ -38,32 +36,86 @@
|
|||||||
|
|
||||||
namespace uLib {
|
namespace uLib {
|
||||||
|
|
||||||
|
#if defined(USE_CUDA) && defined(__CUDACC__)
|
||||||
|
template <typename VoxelT>
|
||||||
|
__global__ void LinearFilterKernel(const VoxelT *in, VoxelT *out,
|
||||||
|
const VoxelT *kernel, int vox_size,
|
||||||
|
int ker_size, int center_count) {
|
||||||
|
int index = blockIdx.x * blockDim.x + threadIdx.x;
|
||||||
|
if (index < vox_size) {
|
||||||
|
float conv = 0;
|
||||||
|
float ksum = 0;
|
||||||
|
for (int ik = 0; ik < ker_size; ++ik) {
|
||||||
|
int pos = index + kernel[ik].Count - center_count;
|
||||||
|
if (pos < 0) {
|
||||||
|
pos += vox_size * ((-pos / vox_size) + 1);
|
||||||
|
}
|
||||||
|
pos = pos % vox_size;
|
||||||
|
conv += in[pos].Value * kernel[ik].Value;
|
||||||
|
ksum += kernel[ik].Value;
|
||||||
|
}
|
||||||
|
out[index].Value = conv / ksum;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
template <typename VoxelT>
|
template <typename VoxelT>
|
||||||
class VoxFilterAlgorithmLinear :
|
class VoxFilterAlgorithmLinear
|
||||||
public VoxImageFilter<VoxelT, VoxFilterAlgorithmLinear<VoxelT> > {
|
: public VoxImageFilter<VoxelT, VoxFilterAlgorithmLinear<VoxelT>> {
|
||||||
public:
|
public:
|
||||||
typedef VoxImageFilter<VoxelT, VoxFilterAlgorithmLinear<VoxelT> > BaseClass;
|
typedef VoxImageFilter<VoxelT, VoxFilterAlgorithmLinear<VoxelT>> BaseClass;
|
||||||
VoxFilterAlgorithmLinear(const Vector3i &size) : BaseClass(size) {}
|
VoxFilterAlgorithmLinear(const Vector3i &size) : BaseClass(size) {}
|
||||||
|
|
||||||
float Evaluate(const VoxImage<VoxelT> &buffer, int index)
|
#if defined(USE_CUDA) && defined(__CUDACC__)
|
||||||
{
|
void Run() {
|
||||||
const std::vector<VoxelT> &vbuf = buffer.ConstData();
|
if (this->m_Image->Data().GetDevice() == MemoryDevice::VRAM ||
|
||||||
const std::vector<VoxelT> &vker = this->m_KernelData.ConstData();
|
this->m_KernelData.Data().GetDevice() == MemoryDevice::VRAM) {
|
||||||
int vox_size = vbuf.size();
|
|
||||||
int ker_size = vker.size();
|
this->m_Image->Data().MoveToVRAM();
|
||||||
int pos;
|
this->m_KernelData.Data().MoveToVRAM();
|
||||||
float conv = 0, ksum = 0;
|
|
||||||
for (int ik = 0; ik < ker_size; ++ik) {
|
VoxImage<VoxelT> buffer = *(this->m_Image);
|
||||||
pos = index + vker[ik].Count - vker[this->m_KernelData.GetCenterData()].Count;
|
buffer.Data().MoveToVRAM();
|
||||||
pos = (pos + vox_size) % vox_size;
|
|
||||||
conv += vbuf[pos].Value * vker[ik].Value;
|
int vox_size = buffer.Data().size();
|
||||||
ksum += vker[ik].Value;
|
int ker_size = this->m_KernelData.Data().size();
|
||||||
}
|
|
||||||
return conv / ksum;
|
VoxelT *d_img_out = this->m_Image->Data().GetVRAMData();
|
||||||
|
const VoxelT *d_img_in = buffer.Data().GetVRAMData();
|
||||||
|
const VoxelT *d_kernel = this->m_KernelData.Data().GetVRAMData();
|
||||||
|
int center_count =
|
||||||
|
this->m_KernelData[this->m_KernelData.GetCenterData()].Count;
|
||||||
|
|
||||||
|
int threadsPerBlock = 256;
|
||||||
|
int blocksPerGrid = (vox_size + threadsPerBlock - 1) / threadsPerBlock;
|
||||||
|
|
||||||
|
LinearFilterKernel<<<blocksPerGrid, threadsPerBlock>>>(
|
||||||
|
d_img_in, d_img_out, d_kernel, vox_size, ker_size, center_count);
|
||||||
|
cudaDeviceSynchronize();
|
||||||
|
} else {
|
||||||
|
BaseClass::Run();
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
|
float Evaluate(const VoxImage<VoxelT> &buffer, int index) {
|
||||||
|
const DataAllocator<VoxelT> &vbuf = buffer.ConstData();
|
||||||
|
const DataAllocator<VoxelT> &vker = this->m_KernelData.ConstData();
|
||||||
|
int vox_size = vbuf.size();
|
||||||
|
int ker_size = vker.size();
|
||||||
|
int pos;
|
||||||
|
float conv = 0, ksum = 0;
|
||||||
|
for (int ik = 0; ik < ker_size; ++ik) {
|
||||||
|
pos = index + vker[ik].Count -
|
||||||
|
vker[this->m_KernelData.GetCenterData()].Count;
|
||||||
|
pos = (pos + vox_size) % vox_size;
|
||||||
|
conv += vbuf[pos].Value * vker[ik].Value;
|
||||||
|
ksum += vker[ik].Value;
|
||||||
|
}
|
||||||
|
return conv / ksum;
|
||||||
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
}
|
} // namespace uLib
|
||||||
|
|
||||||
#endif // VOXIMAGEFILTERLINEAR_HPP
|
#endif // VOXIMAGEFILTERLINEAR_HPP
|
||||||
|
|||||||
@@ -23,14 +23,12 @@
|
|||||||
|
|
||||||
//////////////////////////////////////////////////////////////////////////////*/
|
//////////////////////////////////////////////////////////////////////////////*/
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
#ifndef VOXIMAGEFILTERMEDIAN_HPP
|
#ifndef VOXIMAGEFILTERMEDIAN_HPP
|
||||||
#define VOXIMAGEFILTERMEDIAN_HPP
|
#define VOXIMAGEFILTERMEDIAN_HPP
|
||||||
|
|
||||||
#include <Math/Dense.h>
|
|
||||||
#include "Math/VoxImage.h"
|
#include "Math/VoxImage.h"
|
||||||
#include "VoxImageFilter.h"
|
#include "VoxImageFilter.h"
|
||||||
|
#include <Math/Dense.h>
|
||||||
|
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
///// VOXIMAGE FILTER MEDIAN /////////////////////////////////////////////////
|
///// VOXIMAGE FILTER MEDIAN /////////////////////////////////////////////////
|
||||||
@@ -39,37 +37,38 @@
|
|||||||
namespace uLib {
|
namespace uLib {
|
||||||
|
|
||||||
template <typename VoxelT>
|
template <typename VoxelT>
|
||||||
class VoxFilterAlgorithmMedian :
|
class VoxFilterAlgorithmMedian
|
||||||
public VoxImageFilter<VoxelT, VoxFilterAlgorithmMedian<VoxelT> > {
|
: public VoxImageFilter<VoxelT, VoxFilterAlgorithmMedian<VoxelT>> {
|
||||||
public:
|
public:
|
||||||
typedef VoxImageFilter<VoxelT, VoxFilterAlgorithmMedian<VoxelT> > BaseClass;
|
typedef VoxImageFilter<VoxelT, VoxFilterAlgorithmMedian<VoxelT>> BaseClass;
|
||||||
VoxFilterAlgorithmMedian(const Vector3i &size) : BaseClass(size) {}
|
VoxFilterAlgorithmMedian(const Vector3i &size) : BaseClass(size) {}
|
||||||
|
|
||||||
float Evaluate(const VoxImage<VoxelT> &buffer, int index)
|
float Evaluate(const VoxImage<VoxelT> &buffer, int index) {
|
||||||
{
|
const DataAllocator<VoxelT> &vbuf = buffer.ConstData();
|
||||||
const std::vector<VoxelT> &vbuf = buffer.ConstData();
|
const DataAllocator<VoxelT> &vker = this->m_KernelData.ConstData();
|
||||||
const std::vector<VoxelT> &vker = this->m_KernelData.ConstData();
|
int vox_size = vbuf.size();
|
||||||
int vox_size = vbuf.size();
|
int ker_size = vker.size();
|
||||||
int ker_size = vker.size();
|
int pos;
|
||||||
int pos;
|
|
||||||
|
|
||||||
std::vector<float> mfh(ker_size);
|
std::vector<float> mfh(ker_size);
|
||||||
for (int ik = 0; ik < ker_size; ik++) {
|
for (int ik = 0; ik < ker_size; ik++) {
|
||||||
pos = index + vker[ik].Count - vker[this->m_KernelData.GetCenterData()].Count;
|
pos = index + vker[ik].Count -
|
||||||
pos = (pos + vox_size) % vox_size;
|
vker[this->m_KernelData.GetCenterData()].Count;
|
||||||
mfh[ik] = vbuf[pos].Value * vker[ik].Value;
|
pos = (pos + vox_size) % vox_size;
|
||||||
}
|
mfh[ik] = vbuf[pos].Value * vker[ik].Value;
|
||||||
std::sort(mfh.begin(), mfh.end());
|
|
||||||
pos = 0;
|
|
||||||
// count zeroes in filter kernel to move it out of median //
|
|
||||||
for (int i = 0; i < ker_size; ++i)
|
|
||||||
if (vker[i].Value == 0.0) pos++;
|
|
||||||
// median //
|
|
||||||
pos += (ker_size - pos) / 2;
|
|
||||||
return mfh[pos];
|
|
||||||
}
|
}
|
||||||
|
std::sort(mfh.begin(), mfh.end());
|
||||||
|
pos = 0;
|
||||||
|
// count zeroes in filter kernel to move it out of median //
|
||||||
|
for (int i = 0; i < ker_size; ++i)
|
||||||
|
if (vker[i].Value == 0.0)
|
||||||
|
pos++;
|
||||||
|
// median //
|
||||||
|
pos += (ker_size - pos) / 2;
|
||||||
|
return mfh[pos];
|
||||||
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
}
|
} // namespace uLib
|
||||||
|
|
||||||
#endif // VOXIMAGEFILTERMEDIAN_HPP
|
#endif // VOXIMAGEFILTERMEDIAN_HPP
|
||||||
|
|||||||
@@ -39,48 +39,48 @@ namespace uLib {
|
|||||||
////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
|
|
||||||
void VoxRaytracer::RayData::AddElement(Id_t id, float L) {
|
void VoxRaytracer::RayData::AddElement(Id_t id, float L) {
|
||||||
|
if (m_Count >= m_Data.size()) {
|
||||||
|
size_t new_size = m_Data.size() == 0 ? 128 : m_Data.size() * 2;
|
||||||
|
m_Data.resize(new_size);
|
||||||
|
}
|
||||||
Element el = {id, L};
|
Element el = {id, L};
|
||||||
m_Data.push_back(el);
|
m_Data[m_Count] = el;
|
||||||
|
m_Count++;
|
||||||
m_TotalLength += L;
|
m_TotalLength += L;
|
||||||
}
|
}
|
||||||
|
|
||||||
void VoxRaytracer::RayData::AppendRay(const VoxRaytracer::RayData &in) {
|
void VoxRaytracer::RayData::AppendRay(const VoxRaytracer::RayData &in) {
|
||||||
if (unlikely(!in.m_Data.size())) {
|
if (unlikely(in.m_Count == 0)) {
|
||||||
std::cout << "Warinig: PoCA on exit border!\n";
|
std::cout << "Warinig: PoCA on exit border!\n";
|
||||||
return;
|
return;
|
||||||
} else if (unlikely(!m_Data.size())) {
|
} else if (unlikely(m_Count == 0)) {
|
||||||
m_Data = in.m_Data;
|
m_Data.resize(in.m_Count);
|
||||||
|
for (size_t i = 0; i < in.m_Count; ++i) {
|
||||||
|
m_Data[i] = in.m_Data[i];
|
||||||
|
}
|
||||||
|
m_Count = in.m_Count;
|
||||||
|
m_TotalLength = in.m_TotalLength;
|
||||||
std::cout << "Warinig: PoCA on entrance border!\n";
|
std::cout << "Warinig: PoCA on entrance border!\n";
|
||||||
return;
|
return;
|
||||||
} else {
|
} else {
|
||||||
// Opzione 1) un voxel in piu' //
|
// Opzione 1) un voxel in piu' //
|
||||||
if (in.m_Data.size() > 0) {
|
if (in.m_Count > 0) {
|
||||||
m_Data.insert(m_Data.end(), in.m_Data.begin(), in.m_Data.end());
|
if (m_Count + in.m_Count > m_Data.size()) {
|
||||||
|
m_Data.resize(m_Count + in.m_Count);
|
||||||
|
}
|
||||||
|
for (size_t i = 0; i < in.m_Count; ++i) {
|
||||||
|
m_Data[m_Count + i] = in.m_Data[i];
|
||||||
|
}
|
||||||
|
m_Count += in.m_Count;
|
||||||
}
|
}
|
||||||
// Opzione 2) merge dei voxel nel poca.
|
|
||||||
// RayData::Element &e1 = m_Data.back();
|
|
||||||
// const RayData::Element &e2 = in.m_Data.front();
|
|
||||||
// if(e1.vox_id == e2.vox_id)
|
|
||||||
// {
|
|
||||||
// m_Data.reserve(m_Data.size() + in.m_Data.size() - 1);
|
|
||||||
// e1.L += e2.L; //fix//
|
|
||||||
// m_Data.insert(m_Data.end(), in.m_Data.begin()+1,
|
|
||||||
// in.m_Data.end());
|
|
||||||
// }
|
|
||||||
// else {
|
|
||||||
// m_Data.reserve(m_Data.size() + in.m_Data.size());
|
|
||||||
// m_Data.insert(m_Data.end(), in.m_Data.begin(),
|
|
||||||
// in.m_Data.end());
|
|
||||||
// }
|
|
||||||
m_TotalLength += in.m_TotalLength;
|
m_TotalLength += in.m_TotalLength;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
void VoxRaytracer::RayData::PrintSelf(std::ostream &o) {
|
void VoxRaytracer::RayData::PrintSelf(std::ostream &o) {
|
||||||
o << "Ray: total lenght " << m_TotalLength << "\n";
|
o << "Ray: total lenght " << m_TotalLength << "\n";
|
||||||
std::vector<Element>::iterator it;
|
for (size_t i = 0; i < m_Count; ++i)
|
||||||
for (it = m_Data.begin(); it < m_Data.end(); ++it)
|
o << "[ " << m_Data[i].vox_id << ", " << m_Data[i].L << "] \n";
|
||||||
o << "[ " << (*it).vox_id << ", " << (*it).L << "] \n";
|
|
||||||
}
|
}
|
||||||
|
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
@@ -144,14 +144,21 @@ VoxRaytracer::RayData
|
|||||||
VoxRaytracer::TraceBetweenPoints(const HPoint3f &in,
|
VoxRaytracer::TraceBetweenPoints(const HPoint3f &in,
|
||||||
const HPoint3f &out) const {
|
const HPoint3f &out) const {
|
||||||
RayData ray;
|
RayData ray;
|
||||||
|
|
||||||
|
// get the local points and the direction vector
|
||||||
|
// local to image means in the normalized voxel space where the size
|
||||||
|
// of the voxel is 1 in all dimensions
|
||||||
Vector4f pt1 = m_Image->GetLocalPoint(in);
|
Vector4f pt1 = m_Image->GetLocalPoint(in);
|
||||||
Vector4f pt2 = m_Image->GetLocalPoint(out);
|
Vector4f pt2 = m_Image->GetLocalPoint(out);
|
||||||
Vector4f s = pt2 - pt1;
|
Vector4f s = pt2 - pt1;
|
||||||
|
|
||||||
|
// l is the total length of the ray in normalized voxel space
|
||||||
float l = s.head(3).norm();
|
float l = s.head(3).norm();
|
||||||
|
|
||||||
|
// L is the length of the ray between two grid lines in grid
|
||||||
Vector3f L(l / s(0), l / s(1), l / s(2));
|
Vector3f L(l / s(0), l / s(1), l / s(2));
|
||||||
|
|
||||||
// Vector3f scale; // FIXXX
|
// Vector3f scale; // TODO: FIX Scaling
|
||||||
// scale << (m_Image->GetWorldMatrix() * Vector4f(1,0,0,0)).norm(),
|
// scale << (m_Image->GetWorldMatrix() * Vector4f(1,0,0,0)).norm(),
|
||||||
// (m_Image->GetWorldMatrix() * Vector4f(0,1,0,0)).norm(),
|
// (m_Image->GetWorldMatrix() * Vector4f(0,1,0,0)).norm(),
|
||||||
// (m_Image->GetWorldMatrix() * Vector4f(0,0,1,0)).norm();
|
// (m_Image->GetWorldMatrix() * Vector4f(0,0,1,0)).norm();
|
||||||
@@ -174,21 +181,23 @@ VoxRaytracer::TraceBetweenPoints(const HPoint3f &in,
|
|||||||
float d;
|
float d;
|
||||||
while (l > 0) {
|
while (l > 0) {
|
||||||
|
|
||||||
|
// find which is the minimum of the offsets to the next grid line
|
||||||
|
// it will be also the actual normalized voxel ray length
|
||||||
d = offset.minCoeff(&id);
|
d = offset.minCoeff(&id);
|
||||||
|
|
||||||
|
// see if the voxel is inside the grid (we are still inside image)
|
||||||
if (m_Image->IsInsideGrid(vid)) {
|
if (m_Image->IsInsideGrid(vid)) {
|
||||||
|
// add the voxel to the ray with mapping id and length scaled
|
||||||
ray.AddElement(m_Image->Map(vid), d * m_scale(id));
|
ray.AddElement(m_Image->Map(vid), d * m_scale(id));
|
||||||
}
|
}
|
||||||
|
|
||||||
// nan check //
|
// move to the next voxel
|
||||||
// if(unlikely(!isFinite(d * scale(id)))) {
|
|
||||||
// std:: cout << "NAN in raytracer\n";
|
|
||||||
// exit(1);
|
|
||||||
// }
|
|
||||||
|
|
||||||
vid(id) += (int)fast_sign(s(id));
|
vid(id) += (int)fast_sign(s(id));
|
||||||
|
|
||||||
|
// update the remaining length
|
||||||
l -= d;
|
l -= d;
|
||||||
|
|
||||||
|
// update the offsets
|
||||||
offset.array() -= d;
|
offset.array() -= d;
|
||||||
offset(id) = fmin(L(id), l);
|
offset(id) = fmin(L(id), l);
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -23,71 +23,101 @@
|
|||||||
|
|
||||||
//////////////////////////////////////////////////////////////////////////////*/
|
//////////////////////////////////////////////////////////////////////////////*/
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
#ifndef VOXRAYTRACER_H
|
#ifndef VOXRAYTRACER_H
|
||||||
#define VOXRAYTRACER_H
|
#define VOXRAYTRACER_H
|
||||||
|
|
||||||
|
#include <Core/DataAllocator.h>
|
||||||
|
#include <Core/Vector.h>
|
||||||
#include <math.h>
|
#include <math.h>
|
||||||
#include <vector>
|
#include <vector>
|
||||||
|
|
||||||
#include "Math/StructuredGrid.h"
|
#include "Math/StructuredGrid.h"
|
||||||
|
#include "Math/VoxImage.h"
|
||||||
|
|
||||||
|
#ifdef USE_CUDA
|
||||||
|
#include <cuda_runtime.h>
|
||||||
|
#endif
|
||||||
|
|
||||||
namespace uLib {
|
namespace uLib {
|
||||||
|
|
||||||
class VoxRaytracer {
|
class VoxRaytracer {
|
||||||
|
|
||||||
public:
|
public:
|
||||||
class RayData {
|
class RayData {
|
||||||
public:
|
public:
|
||||||
RayData() : m_TotalLength(0) {}
|
RayData() : m_TotalLength(0), m_Count(0) {}
|
||||||
|
|
||||||
typedef struct {
|
struct Element {
|
||||||
Id_t vox_id;
|
Id_t vox_id;
|
||||||
Scalarf L;
|
Scalarf L;
|
||||||
} Element;
|
~Element() {}
|
||||||
|
|
||||||
inline void AddElement(Id_t id, float L);
|
|
||||||
|
|
||||||
void AppendRay ( const RayData &in);
|
|
||||||
|
|
||||||
inline const std::vector<Element>& Data() const { return this->m_Data; }
|
|
||||||
|
|
||||||
inline const Scalarf& TotalLength() const { return this->m_TotalLength; }
|
|
||||||
|
|
||||||
void PrintSelf(std::ostream &o);
|
|
||||||
|
|
||||||
private:
|
|
||||||
std::vector<Element> m_Data;
|
|
||||||
Scalarf m_TotalLength;
|
|
||||||
};
|
};
|
||||||
|
|
||||||
|
inline void AddElement(Id_t id, float L);
|
||||||
|
|
||||||
public:
|
void AppendRay(const RayData &in);
|
||||||
VoxRaytracer(StructuredGrid &image) : m_Image(&image) {
|
|
||||||
m_scale <<
|
|
||||||
(m_Image->GetWorldMatrix() * Vector4f(1,0,0,0)).norm(),
|
|
||||||
(m_Image->GetWorldMatrix() * Vector4f(0,1,0,0)).norm(),
|
|
||||||
(m_Image->GetWorldMatrix() * Vector4f(0,0,1,0)).norm();
|
|
||||||
}
|
|
||||||
|
|
||||||
bool GetEntryPoint(const HLine3f &line, HPoint3f &pt);
|
inline uLib::Vector<Element> &Data() { return this->m_Data; }
|
||||||
|
|
||||||
bool GetExitPoint(const HLine3f &line, HPoint3f &pt);
|
inline const uLib::Vector<Element> &Data() const { return this->m_Data; }
|
||||||
|
|
||||||
RayData TraceBetweenPoints(const HPoint3f &in, const HPoint3f &out) const;
|
inline size_t Count() const { return this->m_Count; }
|
||||||
|
|
||||||
RayData TraceLine(const HLine3f &line) const;
|
inline const Scalarf &TotalLength() const { return this->m_TotalLength; }
|
||||||
|
|
||||||
inline StructuredGrid* GetImage() const { return this->m_Image; }
|
inline void SetCount(size_t c) { this->m_Count = c; }
|
||||||
|
|
||||||
|
inline void SetTotalLength(Scalarf tl) { this->m_TotalLength = tl; }
|
||||||
|
|
||||||
|
void PrintSelf(std::ostream &o);
|
||||||
|
|
||||||
|
private:
|
||||||
|
uLib::Vector<Element> m_Data;
|
||||||
|
Scalarf m_TotalLength;
|
||||||
|
size_t m_Count;
|
||||||
|
};
|
||||||
|
|
||||||
|
public:
|
||||||
|
VoxRaytracer(StructuredGrid &image) : m_Image(&image) {
|
||||||
|
m_scale << (m_Image->GetWorldMatrix() * Vector4f(1, 0, 0, 0)).norm(),
|
||||||
|
(m_Image->GetWorldMatrix() * Vector4f(0, 1, 0, 0)).norm(),
|
||||||
|
(m_Image->GetWorldMatrix() * Vector4f(0, 0, 1, 0)).norm();
|
||||||
|
}
|
||||||
|
|
||||||
|
bool GetEntryPoint(const HLine3f &line, HPoint3f &pt);
|
||||||
|
|
||||||
|
bool GetExitPoint(const HLine3f &line, HPoint3f &pt);
|
||||||
|
|
||||||
|
RayData TraceBetweenPoints(const HPoint3f &in, const HPoint3f &out) const;
|
||||||
|
|
||||||
|
RayData TraceLine(const HLine3f &line) const;
|
||||||
|
|
||||||
|
inline StructuredGrid *GetImage() const { return this->m_Image; }
|
||||||
|
|
||||||
|
#ifdef USE_CUDA
|
||||||
|
template <typename VoxelT>
|
||||||
|
void AccumulateLinesCUDA(const HLine3f *lines, size_t num_lines,
|
||||||
|
VoxImage<VoxelT> &image);
|
||||||
|
|
||||||
|
void TraceLineCUDA(const HLine3f *lines, size_t num_lines, RayData *out_rays,
|
||||||
|
int max_elements_per_ray = 128,
|
||||||
|
float *kernel_time_ms = nullptr);
|
||||||
|
|
||||||
|
void TraceBetweenPointsCUDA(const HPoint3f *in_pts, const HPoint3f *out_pts,
|
||||||
|
size_t num_lines, RayData *out_rays,
|
||||||
|
int max_elements_per_ray = 128,
|
||||||
|
float *kernel_time_ms = nullptr);
|
||||||
|
#endif
|
||||||
|
|
||||||
private:
|
private:
|
||||||
StructuredGrid *m_Image;
|
StructuredGrid *m_Image;
|
||||||
Vector3f m_scale;
|
Vector3f m_scale;
|
||||||
};
|
};
|
||||||
|
|
||||||
}
|
} // namespace uLib
|
||||||
|
|
||||||
|
|
||||||
|
#ifdef USE_CUDA
|
||||||
|
#include "Math/VoxRaytracerCUDA.hpp"
|
||||||
|
#endif
|
||||||
|
|
||||||
#endif // VOXRAYTRACER_H
|
#endif // VOXRAYTRACER_H
|
||||||
|
|||||||
548
src/Math/VoxRaytracerCUDA.hpp
Normal file
548
src/Math/VoxRaytracerCUDA.hpp
Normal file
@@ -0,0 +1,548 @@
|
|||||||
|
#ifndef VOXRAYTRACERCUDA_H
|
||||||
|
#define VOXRAYTRACERCUDA_H
|
||||||
|
|
||||||
|
#ifdef USE_CUDA
|
||||||
|
|
||||||
|
#include "Math/VoxImage.h"
|
||||||
|
#include "Math/VoxRaytracer.h"
|
||||||
|
#include <cuda_runtime.h>
|
||||||
|
|
||||||
|
namespace uLib {
|
||||||
|
|
||||||
|
#ifdef __CUDACC__
|
||||||
|
template <typename VoxelT>
|
||||||
|
__global__ void
|
||||||
|
RaytraceAccumulateKernel(const float *lines_data, int num_lines,
|
||||||
|
VoxelT *d_image, int dim0, int dim1, int dim2,
|
||||||
|
const float *inv_world_matrix_data, float scale0,
|
||||||
|
float scale1, float scale2) {
|
||||||
|
int idx = blockIdx.x * blockDim.x + threadIdx.x;
|
||||||
|
if (idx >= num_lines)
|
||||||
|
return;
|
||||||
|
|
||||||
|
const float *line_ptr = &lines_data[idx * 8];
|
||||||
|
|
||||||
|
float o_vec[4] = {line_ptr[0], line_ptr[1], line_ptr[2], line_ptr[3]};
|
||||||
|
float d_vec[4] = {line_ptr[4], line_ptr[5], line_ptr[6], line_ptr[7]};
|
||||||
|
|
||||||
|
float pt[4] = {0, 0, 0, 0};
|
||||||
|
float s[4] = {0, 0, 0, 0};
|
||||||
|
|
||||||
|
for (int i = 0; i < 4; ++i) {
|
||||||
|
for (int j = 0; j < 4; ++j) {
|
||||||
|
float m_val = inv_world_matrix_data[i + j * 4];
|
||||||
|
pt[i] += m_val * o_vec[j];
|
||||||
|
s[i] += m_val * d_vec[j];
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
float l = sqrtf(s[0] * s[0] + s[1] * s[1] + s[2] * s[2]);
|
||||||
|
if (l == 0)
|
||||||
|
return;
|
||||||
|
|
||||||
|
float L[3];
|
||||||
|
L[0] = l / s[0];
|
||||||
|
L[1] = l / s[1];
|
||||||
|
L[2] = l / s[2];
|
||||||
|
|
||||||
|
float offset[3];
|
||||||
|
for (int i = 0; i < 3; ++i) {
|
||||||
|
float fpt_i = floorf(pt[i]);
|
||||||
|
offset[i] = (s[i] >= 0) ? (1.0f - (pt[i] - fpt_i)) : (pt[i] - fpt_i);
|
||||||
|
offset[i] = fabsf(offset[i] * L[i]);
|
||||||
|
L[i] = fabsf(L[i]);
|
||||||
|
}
|
||||||
|
|
||||||
|
int id;
|
||||||
|
float d;
|
||||||
|
int vid[3] = {(int)floorf(pt[0]), (int)floorf(pt[1]), (int)floorf(pt[2])};
|
||||||
|
|
||||||
|
float scale_arr[3] = {scale0, scale1, scale2};
|
||||||
|
|
||||||
|
while (vid[0] >= 0 && vid[0] < dim0 && vid[1] >= 0 && vid[1] < dim1 &&
|
||||||
|
vid[2] >= 0 && vid[2] < dim2) {
|
||||||
|
|
||||||
|
d = offset[0];
|
||||||
|
id = 0;
|
||||||
|
if (offset[1] < d) {
|
||||||
|
d = offset[1];
|
||||||
|
id = 1;
|
||||||
|
}
|
||||||
|
if (offset[2] < d) {
|
||||||
|
d = offset[2];
|
||||||
|
id = 2;
|
||||||
|
}
|
||||||
|
|
||||||
|
float L_intersect = d * scale_arr[id];
|
||||||
|
size_t vox_index = vid[0] * dim1 * dim2 + vid[1] * dim2 + vid[2];
|
||||||
|
|
||||||
|
atomicAdd(&(d_image[vox_index].Value), L_intersect);
|
||||||
|
|
||||||
|
float sign_s = (s[id] >= 0) ? 1.0f : -1.0f;
|
||||||
|
vid[id] += (int)sign_s;
|
||||||
|
|
||||||
|
offset[0] -= d;
|
||||||
|
offset[1] -= d;
|
||||||
|
offset[2] -= d;
|
||||||
|
offset[id] = L[id];
|
||||||
|
}
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
|
template <typename VoxelT>
|
||||||
|
void VoxRaytracer::AccumulateLinesCUDA(const HLine3f *lines, size_t num_lines,
|
||||||
|
VoxImage<VoxelT> &image) {
|
||||||
|
if (num_lines == 0)
|
||||||
|
return;
|
||||||
|
|
||||||
|
image.Data().MoveToVRAM();
|
||||||
|
|
||||||
|
float *d_lines = nullptr;
|
||||||
|
size_t lines_size = num_lines * sizeof(HLine3f);
|
||||||
|
cudaMalloc(&d_lines, lines_size);
|
||||||
|
cudaMemcpy(d_lines, lines, lines_size, cudaMemcpyHostToDevice);
|
||||||
|
int threadsPerBlock = 256;
|
||||||
|
int blocksPerGrid = (num_lines + threadsPerBlock - 1) / threadsPerBlock;
|
||||||
|
|
||||||
|
Vector3i dims = image.GetDims();
|
||||||
|
Matrix4f inv_world_matrix = image.GetWorldMatrix().inverse();
|
||||||
|
|
||||||
|
float *d_inv_world;
|
||||||
|
cudaMalloc(&d_inv_world, 16 * sizeof(float));
|
||||||
|
cudaMemcpy(d_inv_world, inv_world_matrix.data(), 16 * sizeof(float),
|
||||||
|
cudaMemcpyHostToDevice);
|
||||||
|
|
||||||
|
#ifdef __CUDACC__
|
||||||
|
RaytraceAccumulateKernel<<<blocksPerGrid, threadsPerBlock>>>(
|
||||||
|
d_lines, num_lines, image.Data().GetVRAMData(), dims(0), dims(1), dims(2),
|
||||||
|
d_inv_world, m_scale(0), m_scale(1), m_scale(2));
|
||||||
|
cudaDeviceSynchronize();
|
||||||
|
|
||||||
|
cudaError_t err = cudaGetLastError();
|
||||||
|
if (err != cudaSuccess) {
|
||||||
|
std::cerr << "CUDA Error in AccumulateLinesCUDA: "
|
||||||
|
<< cudaGetErrorString(err) << std::endl;
|
||||||
|
}
|
||||||
|
#else
|
||||||
|
std::cerr << "RaytraceAccumulateKernel requires NVCC!" << std::endl;
|
||||||
|
#endif
|
||||||
|
|
||||||
|
cudaFree(d_lines);
|
||||||
|
cudaFree(d_inv_world);
|
||||||
|
}
|
||||||
|
|
||||||
|
#ifdef __CUDACC__
|
||||||
|
|
||||||
|
__global__ void TraceBetweenPointsKernel(
|
||||||
|
const float *in_pts_data, const float *out_pts_data, int num_lines,
|
||||||
|
VoxRaytracer::RayData::Element **d_out_elements, size_t *d_out_counts,
|
||||||
|
float *d_out_lengths, int max_elements, int dim0, int dim1, int dim2,
|
||||||
|
const float *inv_world_matrix_data, float scale0, float scale1,
|
||||||
|
float scale2, int inc0, int inc1, int inc2) {
|
||||||
|
int idx = blockIdx.x * blockDim.x + threadIdx.x;
|
||||||
|
if (idx >= num_lines)
|
||||||
|
return;
|
||||||
|
|
||||||
|
VoxRaytracer::RayData::Element *ray_out = d_out_elements[idx];
|
||||||
|
size_t count = 0;
|
||||||
|
float tot_len = 0.0f;
|
||||||
|
|
||||||
|
const float *in_ptr = &in_pts_data[idx * 4];
|
||||||
|
const float *out_ptr = &out_pts_data[idx * 4];
|
||||||
|
|
||||||
|
float pt1[4] = {0, 0, 0, 0}, pt2[4] = {0, 0, 0, 0};
|
||||||
|
for (int i = 0; i < 4; ++i) {
|
||||||
|
for (int j = 0; j < 4; ++j) {
|
||||||
|
float m_val = inv_world_matrix_data[i + j * 4];
|
||||||
|
pt1[i] += m_val * in_ptr[j];
|
||||||
|
pt2[i] += m_val * out_ptr[j];
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
float s[4];
|
||||||
|
for (int i = 0; i < 4; ++i)
|
||||||
|
s[i] = pt2[i] - pt1[i];
|
||||||
|
|
||||||
|
float l = sqrtf(s[0] * s[0] + s[1] * s[1] + s[2] * s[2]);
|
||||||
|
if (l == 0) {
|
||||||
|
d_out_counts[idx] = count;
|
||||||
|
d_out_lengths[idx] = tot_len;
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
float L[3];
|
||||||
|
L[0] = fabsf(l / s[0]);
|
||||||
|
L[1] = fabsf(l / s[1]);
|
||||||
|
L[2] = fabsf(l / s[2]);
|
||||||
|
|
||||||
|
float offset[3];
|
||||||
|
for (int i = 0; i < 3; ++i) {
|
||||||
|
float fpt_i = floorf(pt1[i]);
|
||||||
|
offset[i] = (s[i] >= 0) ? (1.0f - (pt1[i] - fpt_i)) : (pt1[i] - fpt_i);
|
||||||
|
offset[i] = fabsf(offset[i] * L[i]);
|
||||||
|
}
|
||||||
|
|
||||||
|
int vid[3] = {(int)floorf(pt1[0]), (int)floorf(pt1[1]), (int)floorf(pt1[2])};
|
||||||
|
int vid_out[3] = {(int)floorf(pt2[0]), (int)floorf(pt2[1]),
|
||||||
|
(int)floorf(pt2[2])};
|
||||||
|
|
||||||
|
float scale_arr[3] = {scale0, scale1, scale2};
|
||||||
|
|
||||||
|
if (vid[0] == vid_out[0] && vid[1] == vid_out[1] && vid[2] == vid_out[2]) {
|
||||||
|
if (vid[0] >= 0 && vid[0] < dim0 && vid[1] >= 0 && vid[1] < dim1 &&
|
||||||
|
vid[2] >= 0 && vid[2] < dim2) {
|
||||||
|
if (count < max_elements) {
|
||||||
|
int map_id = vid[0] * inc0 + vid[1] * inc1 + vid[2] * inc2;
|
||||||
|
ray_out[count].vox_id = map_id;
|
||||||
|
ray_out[count].L = l;
|
||||||
|
tot_len += l;
|
||||||
|
count++;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
d_out_counts[idx] = count;
|
||||||
|
d_out_lengths[idx] = tot_len;
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
int id;
|
||||||
|
float d;
|
||||||
|
while (l > 0) {
|
||||||
|
d = offset[0];
|
||||||
|
id = 0;
|
||||||
|
if (offset[1] < d) {
|
||||||
|
d = offset[1];
|
||||||
|
id = 1;
|
||||||
|
}
|
||||||
|
if (offset[2] < d) {
|
||||||
|
d = offset[2];
|
||||||
|
id = 2;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (vid[0] >= 0 && vid[0] < dim0 && vid[1] >= 0 && vid[1] < dim1 &&
|
||||||
|
vid[2] >= 0 && vid[2] < dim2) {
|
||||||
|
if (count < max_elements) {
|
||||||
|
int map_id = vid[0] * inc0 + vid[1] * inc1 + vid[2] * inc2;
|
||||||
|
ray_out[count].vox_id = map_id;
|
||||||
|
ray_out[count].L = d * scale_arr[id];
|
||||||
|
tot_len += d * scale_arr[id];
|
||||||
|
count++;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
float sign_s = (s[id] >= 0) ? 1.0f : -1.0f;
|
||||||
|
vid[id] += (int)sign_s;
|
||||||
|
|
||||||
|
l -= d;
|
||||||
|
|
||||||
|
offset[0] -= d;
|
||||||
|
offset[1] -= d;
|
||||||
|
offset[2] -= d;
|
||||||
|
offset[id] = fminf(L[id], l);
|
||||||
|
}
|
||||||
|
d_out_counts[idx] = count;
|
||||||
|
d_out_lengths[idx] = tot_len;
|
||||||
|
}
|
||||||
|
|
||||||
|
__global__ void TraceLineKernel(const float *lines_data, int num_lines,
|
||||||
|
VoxRaytracer::RayData::Element **d_out_elements,
|
||||||
|
size_t *d_out_counts, float *d_out_lengths,
|
||||||
|
int max_elements, int dim0, int dim1, int dim2,
|
||||||
|
const float *inv_world_matrix_data,
|
||||||
|
float scale0, float scale1, float scale2,
|
||||||
|
int inc0, int inc1, int inc2) {
|
||||||
|
int idx = blockIdx.x * blockDim.x + threadIdx.x;
|
||||||
|
if (idx >= num_lines)
|
||||||
|
return;
|
||||||
|
|
||||||
|
VoxRaytracer::RayData::Element *ray_out = d_out_elements[idx];
|
||||||
|
size_t count = 0;
|
||||||
|
float tot_len = 0.0f;
|
||||||
|
|
||||||
|
const float *line_ptr = &lines_data[idx * 8];
|
||||||
|
float o_vec[4] = {line_ptr[0], line_ptr[1], line_ptr[2], line_ptr[3]};
|
||||||
|
float d_vec[4] = {line_ptr[4], line_ptr[5], line_ptr[6], line_ptr[7]};
|
||||||
|
|
||||||
|
float pt[4] = {0, 0, 0, 0}, s[4] = {0, 0, 0, 0};
|
||||||
|
for (int i = 0; i < 4; ++i) {
|
||||||
|
for (int j = 0; j < 4; ++j) {
|
||||||
|
float m_val = inv_world_matrix_data[i + j * 4];
|
||||||
|
pt[i] += m_val * o_vec[j];
|
||||||
|
s[i] += m_val * d_vec[j];
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
float l = sqrtf(s[0] * s[0] + s[1] * s[1] + s[2] * s[2]);
|
||||||
|
if (l == 0) {
|
||||||
|
d_out_counts[idx] = count;
|
||||||
|
d_out_lengths[idx] = tot_len;
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
float L[3];
|
||||||
|
L[0] = fabsf(l / s[0]);
|
||||||
|
L[1] = fabsf(l / s[1]);
|
||||||
|
L[2] = fabsf(l / s[2]);
|
||||||
|
|
||||||
|
float offset[3];
|
||||||
|
for (int i = 0; i < 3; ++i) {
|
||||||
|
float fpt_i = floorf(pt[i]);
|
||||||
|
offset[i] = (s[i] >= 0) ? (1.0f - (pt[i] - fpt_i)) : (pt[i] - fpt_i);
|
||||||
|
offset[i] = fabsf(offset[i] * L[i]);
|
||||||
|
}
|
||||||
|
|
||||||
|
int id;
|
||||||
|
float d;
|
||||||
|
int vid[3] = {(int)floorf(pt[0]), (int)floorf(pt[1]), (int)floorf(pt[2])};
|
||||||
|
float scale_arr[3] = {scale0, scale1, scale2};
|
||||||
|
|
||||||
|
while (vid[0] >= 0 && vid[0] < dim0 && vid[1] >= 0 && vid[1] < dim1 &&
|
||||||
|
vid[2] >= 0 && vid[2] < dim2) {
|
||||||
|
d = offset[0];
|
||||||
|
id = 0;
|
||||||
|
if (offset[1] < d) {
|
||||||
|
d = offset[1];
|
||||||
|
id = 1;
|
||||||
|
}
|
||||||
|
if (offset[2] < d) {
|
||||||
|
d = offset[2];
|
||||||
|
id = 2;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (count < max_elements) {
|
||||||
|
int map_id = vid[0] * inc0 + vid[1] * inc1 + vid[2] * inc2;
|
||||||
|
ray_out[count].vox_id = map_id;
|
||||||
|
ray_out[count].L = d * scale_arr[id];
|
||||||
|
tot_len += d * scale_arr[id];
|
||||||
|
count++;
|
||||||
|
}
|
||||||
|
|
||||||
|
float sign_s = (s[id] >= 0) ? 1.0f : -1.0f;
|
||||||
|
vid[id] += (int)sign_s;
|
||||||
|
|
||||||
|
offset[0] -= d;
|
||||||
|
offset[1] -= d;
|
||||||
|
offset[2] -= d;
|
||||||
|
offset[id] = L[id];
|
||||||
|
}
|
||||||
|
d_out_counts[idx] = count;
|
||||||
|
d_out_lengths[idx] = tot_len;
|
||||||
|
}
|
||||||
|
|
||||||
|
#endif // __CUDACC__
|
||||||
|
|
||||||
|
inline void VoxRaytracer::TraceLineCUDA(const HLine3f *lines, size_t num_lines,
|
||||||
|
RayData *out_rays,
|
||||||
|
int max_elements_per_ray,
|
||||||
|
float *kernel_time_ms) {
|
||||||
|
if (num_lines == 0)
|
||||||
|
return;
|
||||||
|
|
||||||
|
float *d_lines = nullptr;
|
||||||
|
bool alloc_lines = false;
|
||||||
|
|
||||||
|
cudaPointerAttributes ptr_attr;
|
||||||
|
cudaPointerGetAttributes(&ptr_attr, lines);
|
||||||
|
if (ptr_attr.type == cudaMemoryTypeDevice) {
|
||||||
|
d_lines = (float *)lines;
|
||||||
|
} else {
|
||||||
|
alloc_lines = true;
|
||||||
|
size_t lines_size = num_lines * sizeof(HLine3f);
|
||||||
|
cudaMalloc(&d_lines, lines_size);
|
||||||
|
cudaMemcpy(d_lines, lines, lines_size, cudaMemcpyHostToDevice);
|
||||||
|
}
|
||||||
|
|
||||||
|
std::vector<RayData::Element *> h_out_elements(num_lines);
|
||||||
|
for (size_t i = 0; i < num_lines; ++i) {
|
||||||
|
out_rays[i].Data().resize(max_elements_per_ray);
|
||||||
|
out_rays[i].Data().MoveToVRAM();
|
||||||
|
h_out_elements[i] = out_rays[i].Data().GetVRAMData();
|
||||||
|
}
|
||||||
|
|
||||||
|
RayData::Element **d_out_elements;
|
||||||
|
cudaMalloc(&d_out_elements, num_lines * sizeof(RayData::Element *));
|
||||||
|
cudaMemcpy(d_out_elements, h_out_elements.data(),
|
||||||
|
num_lines * sizeof(RayData::Element *), cudaMemcpyHostToDevice);
|
||||||
|
|
||||||
|
size_t *d_out_counts;
|
||||||
|
float *d_out_lengths;
|
||||||
|
cudaMalloc(&d_out_counts, num_lines * sizeof(size_t));
|
||||||
|
cudaMalloc(&d_out_lengths, num_lines * sizeof(float));
|
||||||
|
|
||||||
|
int threadsPerBlock = 256;
|
||||||
|
int blocksPerGrid = (num_lines + threadsPerBlock - 1) / threadsPerBlock;
|
||||||
|
|
||||||
|
Vector3i dims = m_Image->GetDims();
|
||||||
|
Vector3i incs = m_Image->GetIncrements();
|
||||||
|
Matrix4f inv_world_matrix = m_Image->GetWorldMatrix().inverse();
|
||||||
|
|
||||||
|
float *d_inv_world;
|
||||||
|
cudaMalloc(&d_inv_world, 16 * sizeof(float));
|
||||||
|
cudaMemcpy(d_inv_world, inv_world_matrix.data(), 16 * sizeof(float),
|
||||||
|
cudaMemcpyHostToDevice);
|
||||||
|
|
||||||
|
#ifdef __CUDACC__
|
||||||
|
cudaEvent_t start, stop;
|
||||||
|
if (kernel_time_ms) {
|
||||||
|
cudaEventCreate(&start);
|
||||||
|
cudaEventCreate(&stop);
|
||||||
|
cudaEventRecord(start);
|
||||||
|
}
|
||||||
|
|
||||||
|
TraceLineKernel<<<blocksPerGrid, threadsPerBlock>>>(
|
||||||
|
d_lines, num_lines, d_out_elements, d_out_counts, d_out_lengths,
|
||||||
|
max_elements_per_ray, dims(0), dims(1), dims(2), d_inv_world, m_scale(0),
|
||||||
|
m_scale(1), m_scale(2), incs(0), incs(1), incs(2));
|
||||||
|
|
||||||
|
if (kernel_time_ms) {
|
||||||
|
cudaEventRecord(stop);
|
||||||
|
cudaEventSynchronize(stop);
|
||||||
|
cudaEventElapsedTime(kernel_time_ms, start, stop);
|
||||||
|
cudaEventDestroy(start);
|
||||||
|
cudaEventDestroy(stop);
|
||||||
|
} else {
|
||||||
|
cudaDeviceSynchronize();
|
||||||
|
}
|
||||||
|
|
||||||
|
cudaError_t err = cudaGetLastError();
|
||||||
|
if (err != cudaSuccess) {
|
||||||
|
std::cerr << "CUDA Error in TraceLineCUDA: " << cudaGetErrorString(err)
|
||||||
|
<< std::endl;
|
||||||
|
}
|
||||||
|
#else
|
||||||
|
std::cerr << "TraceLineKernel requires NVCC!" << std::endl;
|
||||||
|
#endif
|
||||||
|
|
||||||
|
std::vector<size_t> h_out_counts(num_lines);
|
||||||
|
std::vector<float> h_out_lengths(num_lines);
|
||||||
|
cudaMemcpy(h_out_counts.data(), d_out_counts, num_lines * sizeof(size_t),
|
||||||
|
cudaMemcpyDeviceToHost);
|
||||||
|
cudaMemcpy(h_out_lengths.data(), d_out_lengths, num_lines * sizeof(float),
|
||||||
|
cudaMemcpyDeviceToHost);
|
||||||
|
|
||||||
|
for (size_t i = 0; i < num_lines; ++i) {
|
||||||
|
out_rays[i].SetCount(h_out_counts[i]);
|
||||||
|
out_rays[i].SetTotalLength(h_out_lengths[i]);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (alloc_lines) {
|
||||||
|
cudaFree(d_lines);
|
||||||
|
}
|
||||||
|
cudaFree(d_out_elements);
|
||||||
|
cudaFree(d_out_counts);
|
||||||
|
cudaFree(d_out_lengths);
|
||||||
|
cudaFree(d_inv_world);
|
||||||
|
}
|
||||||
|
|
||||||
|
inline void VoxRaytracer::TraceBetweenPointsCUDA(
|
||||||
|
const HPoint3f *in_pts, const HPoint3f *out_pts, size_t num_lines,
|
||||||
|
RayData *out_rays, int max_elements_per_ray, float *kernel_time_ms) {
|
||||||
|
if (num_lines == 0)
|
||||||
|
return;
|
||||||
|
|
||||||
|
float *d_in_pts = nullptr;
|
||||||
|
float *d_out_pts = nullptr;
|
||||||
|
bool alloc_pts = false;
|
||||||
|
|
||||||
|
cudaPointerAttributes ptr_attr;
|
||||||
|
cudaPointerGetAttributes(&ptr_attr, in_pts);
|
||||||
|
if (ptr_attr.type == cudaMemoryTypeDevice) {
|
||||||
|
d_in_pts = (float *)in_pts;
|
||||||
|
d_out_pts = (float *)out_pts;
|
||||||
|
} else {
|
||||||
|
alloc_pts = true;
|
||||||
|
size_t pts_size = num_lines * sizeof(HPoint3f);
|
||||||
|
cudaMalloc(&d_in_pts, pts_size);
|
||||||
|
cudaMalloc(&d_out_pts, pts_size);
|
||||||
|
cudaMemcpy(d_in_pts, in_pts, pts_size, cudaMemcpyHostToDevice);
|
||||||
|
cudaMemcpy(d_out_pts, out_pts, pts_size, cudaMemcpyHostToDevice);
|
||||||
|
}
|
||||||
|
|
||||||
|
std::vector<RayData::Element *> h_out_elements(num_lines);
|
||||||
|
for (size_t i = 0; i < num_lines; ++i) {
|
||||||
|
out_rays[i].Data().resize(max_elements_per_ray);
|
||||||
|
out_rays[i].Data().MoveToVRAM();
|
||||||
|
h_out_elements[i] = out_rays[i].Data().GetVRAMData();
|
||||||
|
}
|
||||||
|
|
||||||
|
RayData::Element **d_out_elements;
|
||||||
|
cudaMalloc(&d_out_elements, num_lines * sizeof(RayData::Element *));
|
||||||
|
cudaMemcpy(d_out_elements, h_out_elements.data(),
|
||||||
|
num_lines * sizeof(RayData::Element *), cudaMemcpyHostToDevice);
|
||||||
|
|
||||||
|
size_t *d_out_counts;
|
||||||
|
float *d_out_lengths;
|
||||||
|
cudaMalloc(&d_out_counts, num_lines * sizeof(size_t));
|
||||||
|
cudaMalloc(&d_out_lengths, num_lines * sizeof(float));
|
||||||
|
|
||||||
|
int threadsPerBlock = 256;
|
||||||
|
int blocksPerGrid = (num_lines + threadsPerBlock - 1) / threadsPerBlock;
|
||||||
|
|
||||||
|
Vector3i dims = m_Image->GetDims();
|
||||||
|
Vector3i incs = m_Image->GetIncrements();
|
||||||
|
Matrix4f inv_world_matrix = m_Image->GetWorldMatrix().inverse();
|
||||||
|
|
||||||
|
float *d_inv_world;
|
||||||
|
cudaMalloc(&d_inv_world, 16 * sizeof(float));
|
||||||
|
cudaMemcpy(d_inv_world, inv_world_matrix.data(), 16 * sizeof(float),
|
||||||
|
cudaMemcpyHostToDevice);
|
||||||
|
|
||||||
|
#ifdef __CUDACC__
|
||||||
|
cudaEvent_t start, stop;
|
||||||
|
if (kernel_time_ms) {
|
||||||
|
cudaEventCreate(&start);
|
||||||
|
cudaEventCreate(&stop);
|
||||||
|
cudaEventRecord(start);
|
||||||
|
}
|
||||||
|
|
||||||
|
TraceBetweenPointsKernel<<<blocksPerGrid, threadsPerBlock>>>(
|
||||||
|
d_in_pts, d_out_pts, num_lines, d_out_elements, d_out_counts,
|
||||||
|
d_out_lengths, max_elements_per_ray, dims(0), dims(1), dims(2),
|
||||||
|
d_inv_world, m_scale(0), m_scale(1), m_scale(2), incs(0), incs(1),
|
||||||
|
incs(2));
|
||||||
|
|
||||||
|
if (kernel_time_ms) {
|
||||||
|
cudaEventRecord(stop);
|
||||||
|
cudaEventSynchronize(stop);
|
||||||
|
cudaEventElapsedTime(kernel_time_ms, start, stop);
|
||||||
|
cudaEventDestroy(start);
|
||||||
|
cudaEventDestroy(stop);
|
||||||
|
} else {
|
||||||
|
cudaDeviceSynchronize();
|
||||||
|
}
|
||||||
|
|
||||||
|
cudaError_t err = cudaGetLastError();
|
||||||
|
if (err != cudaSuccess) {
|
||||||
|
std::cerr << "CUDA Error in TraceBetweenPointsCUDA: "
|
||||||
|
<< cudaGetErrorString(err) << std::endl;
|
||||||
|
}
|
||||||
|
#else
|
||||||
|
std::cerr << "TraceBetweenPointsKernel requires NVCC!" << std::endl;
|
||||||
|
#endif
|
||||||
|
|
||||||
|
std::vector<size_t> h_out_counts(num_lines);
|
||||||
|
std::vector<float> h_out_lengths(num_lines);
|
||||||
|
cudaMemcpy(h_out_counts.data(), d_out_counts, num_lines * sizeof(size_t),
|
||||||
|
cudaMemcpyDeviceToHost);
|
||||||
|
cudaMemcpy(h_out_lengths.data(), d_out_lengths, num_lines * sizeof(float),
|
||||||
|
cudaMemcpyDeviceToHost);
|
||||||
|
|
||||||
|
for (size_t i = 0; i < num_lines; ++i) {
|
||||||
|
out_rays[i].SetCount(h_out_counts[i]);
|
||||||
|
out_rays[i].SetTotalLength(h_out_lengths[i]);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (alloc_pts) {
|
||||||
|
cudaFree(d_in_pts);
|
||||||
|
cudaFree(d_out_pts);
|
||||||
|
}
|
||||||
|
cudaFree(d_out_elements);
|
||||||
|
cudaFree(d_out_counts);
|
||||||
|
cudaFree(d_out_lengths);
|
||||||
|
cudaFree(d_inv_world);
|
||||||
|
}
|
||||||
|
|
||||||
|
} // namespace uLib
|
||||||
|
|
||||||
|
#endif // USE_CUDA
|
||||||
|
|
||||||
|
#endif // VOXRAYTRACERCUDA_H
|
||||||
@@ -23,88 +23,90 @@
|
|||||||
|
|
||||||
//////////////////////////////////////////////////////////////////////////////*/
|
//////////////////////////////////////////////////////////////////////////////*/
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
#include "Math/Accumulator.h"
|
#include "Math/Accumulator.h"
|
||||||
#include "testing-prototype.h"
|
#include "testing-prototype.h"
|
||||||
|
|
||||||
#include <TRandom.h>
|
#include <TRandom.h>
|
||||||
|
#include <iostream>
|
||||||
|
#include <vector>
|
||||||
|
|
||||||
|
// #include <boost/accumulators/framework/accumulator_set.hpp>
|
||||||
//#include <boost/accumulators/framework/accumulator_set.hpp>
|
// #include <boost/accumulators/statistics/count.hpp>
|
||||||
//#include <boost/accumulators/statistics/count.hpp>
|
// #include <boost/accumulators/accumulators.hpp>
|
||||||
//#include <boost/accumulators/accumulators.hpp>
|
|
||||||
|
|
||||||
using namespace uLib;
|
using namespace uLib;
|
||||||
|
|
||||||
int test_ABTrim() {
|
int test_ABTrim() {
|
||||||
|
|
||||||
Accumulator_ABTrim<float> acc;
|
Accumulator_ABTrim<float> acc;
|
||||||
|
|
||||||
acc.SetABTrim(1,1);
|
acc.SetABTrim(1, 1);
|
||||||
|
|
||||||
std::vector<float> v;
|
std::vector<float> v;
|
||||||
for(float tmpf : {1,5,5,5,300}) v.push_back(tmpf);
|
for (float tmpf : {1, 5, 5, 5, 300})
|
||||||
//v << 1,5,5,5,300;
|
v.push_back(tmpf);
|
||||||
|
// v << 1,5,5,5,300;
|
||||||
|
|
||||||
for(std::vector<float>::iterator itr=v.begin(); itr<v.end(); itr++)
|
for (std::vector<float>::iterator itr = v.begin(); itr < v.end(); itr++)
|
||||||
acc += *itr;
|
acc += *itr;
|
||||||
|
|
||||||
// TODO missing operator <<
|
// TODO missing operator <<
|
||||||
//std::cout << "Accumulating Trim(1,1) vector: "
|
// std::cout << "Accumulating Trim(1,1) vector: "
|
||||||
// << v << " ... out = " << acc() << "\n";
|
// << v << " ... out = " << acc() << "\n";
|
||||||
|
|
||||||
return( acc() == 15.0 );
|
|
||||||
|
|
||||||
|
return (acc() == 15.0);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
int test_Mean() {
|
int test_Mean() {
|
||||||
Accumulator_Mean<float> mean;
|
Accumulator_Mean<float> mean;
|
||||||
TRandom rnd;
|
TRandom rnd;
|
||||||
const int c = 10000000;
|
const int c = 10000000;
|
||||||
std::vector<float> v;
|
std::vector<float> v;
|
||||||
v.reserve(c);
|
v.reserve(c);
|
||||||
for(int i=0;i<c;++i) v.push_back( rnd.Gaus(2000,5) );
|
for (int i = 0; i < c; ++i)
|
||||||
|
v.push_back(rnd.Gaus(2000, 5));
|
||||||
float m = 0;
|
|
||||||
for(int i=0;i<c;++i) m += v[i];
|
|
||||||
m /= c;
|
|
||||||
std::cout << "simple mean: " << m << "\n";
|
|
||||||
|
|
||||||
for(int i=0;i<c;++i) mean(v[i]);
|
|
||||||
std::cout << "mean pass: " << mean() << "\n";
|
|
||||||
mean.AddPass();
|
|
||||||
for(int i=0;i<c;++i) mean(v[i]);
|
|
||||||
std::cout << "mean pass: " << mean() << "\n";
|
|
||||||
mean.AddPass();
|
|
||||||
for(int i=0;i<c;++i) mean(v[i]);
|
|
||||||
std::cout << "mean pass: " << mean() << "\n";
|
|
||||||
mean.AddPass();
|
|
||||||
for(int i=0;i<c;++i) mean(v[i]);
|
|
||||||
std::cout << "mean pass: " << mean() << "\n";
|
|
||||||
mean.AddPass();
|
|
||||||
for(int i=0;i<c;++i) mean(v[i]);
|
|
||||||
std::cout << "mean pass: " << mean() << "\n";
|
|
||||||
mean.AddPass();
|
|
||||||
for(int i=0;i<c;++i) mean(v[i]);
|
|
||||||
std::cout << "mean pass: " << mean() << "\n";
|
|
||||||
mean.AddPass();
|
|
||||||
for(int i=0;i<c;++i) mean(v[i]);
|
|
||||||
std::cout << "mean pass: " << mean() << "\n";
|
|
||||||
|
|
||||||
|
float m = 0;
|
||||||
|
for (int i = 0; i < c; ++i)
|
||||||
|
m += v[i];
|
||||||
|
m /= c;
|
||||||
|
std::cout << "simple mean: " << m << "\n";
|
||||||
|
|
||||||
|
for (int i = 0; i < c; ++i)
|
||||||
|
mean(v[i]);
|
||||||
|
std::cout << "mean pass: " << mean() << "\n";
|
||||||
|
mean.AddPass();
|
||||||
|
for (int i = 0; i < c; ++i)
|
||||||
|
mean(v[i]);
|
||||||
|
std::cout << "mean pass: " << mean() << "\n";
|
||||||
|
mean.AddPass();
|
||||||
|
for (int i = 0; i < c; ++i)
|
||||||
|
mean(v[i]);
|
||||||
|
std::cout << "mean pass: " << mean() << "\n";
|
||||||
|
mean.AddPass();
|
||||||
|
for (int i = 0; i < c; ++i)
|
||||||
|
mean(v[i]);
|
||||||
|
std::cout << "mean pass: " << mean() << "\n";
|
||||||
|
mean.AddPass();
|
||||||
|
for (int i = 0; i < c; ++i)
|
||||||
|
mean(v[i]);
|
||||||
|
std::cout << "mean pass: " << mean() << "\n";
|
||||||
|
mean.AddPass();
|
||||||
|
for (int i = 0; i < c; ++i)
|
||||||
|
mean(v[i]);
|
||||||
|
std::cout << "mean pass: " << mean() << "\n";
|
||||||
|
mean.AddPass();
|
||||||
|
for (int i = 0; i < c; ++i)
|
||||||
|
mean(v[i]);
|
||||||
|
std::cout << "mean pass: " << mean() << "\n";
|
||||||
|
return 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
int main(void) {
|
int main(void) {
|
||||||
BEGIN_TESTING(Accumulator);
|
BEGIN_TESTING(Accumulator);
|
||||||
|
|
||||||
//TEST1( test_ABTrim() );
|
// TEST1( test_ABTrim() );
|
||||||
test_Mean();
|
test_Mean();
|
||||||
|
|
||||||
END_TESTING;
|
END_TESTING;
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -5,6 +5,7 @@ set(TESTS
|
|||||||
ContainerBoxTest
|
ContainerBoxTest
|
||||||
VoxImageTest
|
VoxImageTest
|
||||||
VoxRaytracerTest
|
VoxRaytracerTest
|
||||||
|
VoxRaytracerTestExtended
|
||||||
StructuredDataTest
|
StructuredDataTest
|
||||||
VoxImageFilterTest
|
VoxImageFilterTest
|
||||||
PolicyTest
|
PolicyTest
|
||||||
@@ -17,6 +18,13 @@ set(TESTS
|
|||||||
set(LIBRARIES
|
set(LIBRARIES
|
||||||
${PACKAGE_LIBPREFIX}Core
|
${PACKAGE_LIBPREFIX}Core
|
||||||
${PACKAGE_LIBPREFIX}Math
|
${PACKAGE_LIBPREFIX}Math
|
||||||
|
Boost::serialization
|
||||||
|
Eigen3::Eigen
|
||||||
)
|
)
|
||||||
|
|
||||||
uLib_add_tests(Math)
|
uLib_add_tests(Math)
|
||||||
|
|
||||||
|
if(USE_CUDA)
|
||||||
|
set_source_files_properties(VoxImageTest.cpp VoxImageCopyTest.cpp VoxImageFilterTest.cpp VoxRaytracerTest.cpp VoxRaytracerTestExtended.cpp PROPERTIES LANGUAGE CUDA)
|
||||||
|
set_source_files_properties(VoxRaytracerTest.cpp VoxRaytracerTestExtended.cpp PROPERTIES CXX_STANDARD 17 CUDA_STANDARD 17)
|
||||||
|
endif()
|
||||||
|
|||||||
@@ -23,30 +23,25 @@
|
|||||||
|
|
||||||
//////////////////////////////////////////////////////////////////////////////*/
|
//////////////////////////////////////////////////////////////////////////////*/
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
#include "testing-prototype.h"
|
#include "testing-prototype.h"
|
||||||
|
|
||||||
#include "Math/TriangleMesh.h"
|
#include "Math/TriangleMesh.h"
|
||||||
|
#include <iostream>
|
||||||
|
|
||||||
using namespace uLib;
|
using namespace uLib;
|
||||||
|
|
||||||
int main()
|
int main() {
|
||||||
{
|
BEGIN_TESTING(Triangle Mesh);
|
||||||
BEGIN_TESTING(Triangle Mesh);
|
|
||||||
|
|
||||||
TriangleMesh mesh;
|
TriangleMesh mesh;
|
||||||
|
|
||||||
mesh.AddPoint(Vector3f(0,0,0));
|
mesh.AddPoint(Vector3f(0, 0, 0));
|
||||||
mesh.AddPoint(Vector3f(0,1,0));
|
mesh.AddPoint(Vector3f(0, 1, 0));
|
||||||
mesh.AddPoint(Vector3f(1,0,0));
|
mesh.AddPoint(Vector3f(1, 0, 0));
|
||||||
|
|
||||||
mesh.AddTriangle(Vector3i(0,1,2));
|
mesh.AddTriangle(Vector3i(0, 1, 2));
|
||||||
|
|
||||||
|
mesh.PrintSelf(std::cout);
|
||||||
|
|
||||||
mesh.PrintSelf(std::cout);
|
END_TESTING;
|
||||||
|
|
||||||
END_TESTING;
|
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -23,55 +23,44 @@
|
|||||||
|
|
||||||
//////////////////////////////////////////////////////////////////////////////*/
|
//////////////////////////////////////////////////////////////////////////////*/
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
#include "testing-prototype.h"
|
#include "testing-prototype.h"
|
||||||
|
|
||||||
#include "Math/VoxImage.h"
|
#include "Math/VoxImage.h"
|
||||||
|
|
||||||
using namespace uLib;
|
using namespace uLib;
|
||||||
|
|
||||||
|
|
||||||
struct TestVoxel {
|
struct TestVoxel {
|
||||||
Scalarf Value;
|
Scalarf Value;
|
||||||
unsigned int Count;
|
unsigned int Count;
|
||||||
};
|
};
|
||||||
|
|
||||||
int main() {
|
int main() {
|
||||||
BEGIN_TESTING(Math VoxImage Copy);
|
BEGIN_TESTING(Math VoxImage Copy);
|
||||||
|
|
||||||
{
|
{
|
||||||
VoxImage<TestVoxel> img(Vector3i(10,10,10));
|
VoxImage<TestVoxel> img(Vector3i(10, 10, 10));
|
||||||
TestVoxel zero = {0,0};
|
TestVoxel zero = {0.f, 0};
|
||||||
img.InitVoxels(zero);
|
img.InitVoxels(zero);
|
||||||
TestVoxel nonzero = {5.552368, 0};
|
TestVoxel nonzero = {5.552368f, 0};
|
||||||
img[Vector3i(5,1,7)] = nonzero;
|
img[Vector3i(5, 1, 7)] = nonzero;
|
||||||
img[img.Find(HPoint3f(3,3,3))].Value = 5.552369;
|
img[img.Find(HPoint3f(3, 3, 3))].Value = 5.552369;
|
||||||
TEST1( img.GetValue(Vector3i(5,1,7)) == 5.552368f );
|
TEST1(img.GetValue(Vector3i(5, 1, 7)) == 5.552368f);
|
||||||
|
|
||||||
|
img.SetOrigin(Vector3f(4, 5, 6));
|
||||||
|
|
||||||
img.SetOrigin(Vector3f(4,5,6));
|
std::cout << "\n";
|
||||||
|
|
||||||
std::cout << "\n";
|
img.PrintSelf(std::cout);
|
||||||
|
|
||||||
img.PrintSelf(std::cout);
|
VoxImage<TestVoxel> img2 = img;
|
||||||
|
img2.PrintSelf(std::cout);
|
||||||
|
|
||||||
VoxImage<TestVoxel> img2 = img;
|
TEST1(img.GetOrigin() == img2.GetOrigin());
|
||||||
img2.PrintSelf(std::cout);
|
TEST1(img.GetSpacing() == img2.GetSpacing());
|
||||||
|
|
||||||
TEST1( img.GetOrigin() == img2.GetOrigin() );
|
img2 = img;
|
||||||
TEST1( img.GetSpacing() == img2.GetSpacing() );
|
}
|
||||||
|
|
||||||
img2 = img;
|
std::cout << "returns " << _fail << "\n";
|
||||||
|
END_TESTING;
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
std::cout << "returns " << _fail << "\n";
|
|
||||||
END_TESTING;
|
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -23,128 +23,191 @@
|
|||||||
|
|
||||||
//////////////////////////////////////////////////////////////////////////////*/
|
//////////////////////////////////////////////////////////////////////////////*/
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
#include "testing-prototype.h"
|
|
||||||
#include "Math/StructuredGrid.h"
|
#include "Math/StructuredGrid.h"
|
||||||
|
#include "testing-prototype.h"
|
||||||
|
|
||||||
#include "Math/VoxImage.h"
|
#include "Math/VoxImage.h"
|
||||||
#include "Math/VoxImageFilter.h"
|
#include "Math/VoxImageFilter.h"
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
using namespace uLib;
|
using namespace uLib;
|
||||||
|
|
||||||
struct TestVoxel {
|
struct TestVoxel {
|
||||||
Scalarf Value;
|
Scalarf Value;
|
||||||
unsigned int Count;
|
unsigned int Count;
|
||||||
};
|
};
|
||||||
|
|
||||||
float GaussianShape(float d)
|
float GaussianShape(float d) {
|
||||||
{
|
// normalized manually .. fix //
|
||||||
// normalized manually .. fix //
|
return 4.5 * exp(-d * 4.5);
|
||||||
return 4.5 * exp(-d * 4.5);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
class GaussianShapeClass : public Interface::VoxImageFilterShape {
|
class GaussianShapeClass : public Interface::VoxImageFilterShape {
|
||||||
public:
|
public:
|
||||||
GaussianShapeClass(float sigma) :
|
GaussianShapeClass(float sigma) : m_sigma(sigma) {}
|
||||||
m_sigma(sigma)
|
|
||||||
{}
|
|
||||||
|
|
||||||
float operator ()(float d) {
|
float operator()(float d) { return (1 / m_sigma) * exp(-d / m_sigma); }
|
||||||
return (1/m_sigma) * exp(-d/m_sigma);
|
|
||||||
}
|
|
||||||
|
|
||||||
private:
|
private:
|
||||||
float m_sigma;
|
float m_sigma;
|
||||||
};
|
};
|
||||||
|
|
||||||
|
static float MaxInVector(const std::vector<float> &v) {
|
||||||
static float MaxInVector(const std::vector<float> &v)
|
float max = 0;
|
||||||
{
|
for (int i = 0; i < v.size(); ++i)
|
||||||
float max = 0;
|
if (v.at(i) > max)
|
||||||
for(int i=0; i<v.size(); ++i)
|
max = v.at(i);
|
||||||
if(v.at(i) > max) max = v.at(i);
|
return max;
|
||||||
return max;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
int main() {
|
||||||
|
BEGIN_TESTING(VoxImageFilters);
|
||||||
|
|
||||||
int main()
|
VoxImage<TestVoxel> image(Vector3i(20, 30, 40));
|
||||||
{
|
image[Vector3i(10, 10, 10)].Value = 1;
|
||||||
BEGIN_TESTING(VoxImageFilters);
|
// image[Vector3i(10,10,8)].Value = 1;
|
||||||
|
image.ExportToVtk("test_filter_original.vtk", 0);
|
||||||
|
|
||||||
VoxImage<TestVoxel> image(Vector3i(20,30,40));
|
////////////////////////////////////////////////////////////////////////////
|
||||||
image[Vector3i(10,10,10)].Value = 1;
|
////////////////////////////////////////////////////////////////////////////
|
||||||
//image[Vector3i(10,10,8)].Value = 1;
|
////////////////////////////////////////////////////////////////////////////
|
||||||
image.ExportToVtk("test_filter_original.vtk",0);
|
// RPS //
|
||||||
|
|
||||||
|
{
|
||||||
|
VoxFilterAlgorithmSPR<TestVoxel> filter(Vector3i(2, 3, 4));
|
||||||
|
|
||||||
|
VoxImage<TestVoxel> filtered = image;
|
||||||
|
|
||||||
////////////////////////////////////////////////////////////////////////////
|
std::vector<float> values;
|
||||||
////////////////////////////////////////////////////////////////////////////
|
for (int i = 0; i < filter.GetKernelData().GetDims().prod(); ++i) {
|
||||||
////////////////////////////////////////////////////////////////////////////
|
values.push_back(1.);
|
||||||
// RPS //
|
std::cout << values[i] << " ";
|
||||||
|
}
|
||||||
|
std::cout << "\n";
|
||||||
|
|
||||||
{
|
filter.SetImage(&filtered);
|
||||||
VoxFilterAlgorithmSPR<TestVoxel> filter(Vector3i(2,3,4));
|
|
||||||
|
|
||||||
VoxImage<TestVoxel> filtered = image;
|
filter.SetKernelNumericXZY(values);
|
||||||
|
|
||||||
std::vector<float> values;
|
filter.SetABTrim(0, 2);
|
||||||
for(int i=0; i < filter.GetKernelData().GetDims().prod(); ++i) {
|
|
||||||
values.push_back(1.);
|
|
||||||
std::cout << values[i] << " ";
|
|
||||||
}
|
|
||||||
std::cout << "\n";
|
|
||||||
|
|
||||||
filter.SetImage(&filtered);
|
filter.GetKernelData().PrintSelf(std::cout);
|
||||||
|
|
||||||
filter.SetKernelNumericXZY(values);
|
filter.Run();
|
||||||
|
|
||||||
filter.SetABTrim(0,2);
|
filtered.ExportToVtk("filter_RPS_out.vtk", 0);
|
||||||
|
}
|
||||||
|
|
||||||
filter.GetKernelData().PrintSelf(std::cout);
|
{
|
||||||
|
|
||||||
filter.Run();
|
VoxImage<TestVoxel> image(Vector3i(20, 30, 40));
|
||||||
|
image[Vector3i(10, 10, 10)].Value = 1;
|
||||||
|
image[Vector3i(9, 10, 8)].Value = 2;
|
||||||
|
image.ExportToVtk("test_filter_max_original.vtk", 0);
|
||||||
|
|
||||||
filtered.ExportToVtk("filter_RPS_out.vtk",0);
|
VoxFilterAlgorithmCustom<TestVoxel> filter(Vector3i(3, 3, 4));
|
||||||
|
|
||||||
|
std::vector<float> values;
|
||||||
|
for (int i = 0; i < filter.GetKernelData().GetDims().prod(); ++i) {
|
||||||
|
values.push_back(static_cast<float>(1));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
filter.SetImage(&image);
|
||||||
|
|
||||||
|
filter.SetKernelNumericXZY(values);
|
||||||
|
|
||||||
|
filter.SetCustomEvaluate(MaxInVector);
|
||||||
|
|
||||||
|
filter.Run();
|
||||||
|
|
||||||
{
|
image.ExportToVtk("test_filter_max.vtk", 0);
|
||||||
|
}
|
||||||
|
|
||||||
VoxImage<TestVoxel> image(Vector3i(20,30,40));
|
////////////////////////////////////////////////////////////////////////////
|
||||||
image[Vector3i(10,10,10)].Value = 1;
|
// CUDA Allocator Transfer Test //
|
||||||
image[Vector3i(9,10,8)].Value = 2;
|
{
|
||||||
image.ExportToVtk("test_filter_max_original.vtk",0);
|
VoxImage<TestVoxel> image(Vector3i(10, 10, 10));
|
||||||
|
image[Vector3i(5, 5, 5)].Value = 1;
|
||||||
|
|
||||||
|
VoxFilterAlgorithmLinear<TestVoxel> filter(Vector3i(3, 3, 3));
|
||||||
|
|
||||||
|
std::vector<float> values;
|
||||||
VoxFilterAlgorithmCustom<TestVoxel> filter(Vector3i(3,3,4));
|
for (int i = 0; i < filter.GetKernelData().GetDims().prod(); ++i) {
|
||||||
|
values.push_back(1.0f);
|
||||||
std::vector<float> values;
|
|
||||||
for(int i=0; i < filter.GetKernelData().GetDims().prod(); ++i) {
|
|
||||||
values.push_back(static_cast<float>(1));
|
|
||||||
}
|
|
||||||
|
|
||||||
filter.SetImage(&image);
|
|
||||||
|
|
||||||
filter.SetKernelNumericXZY(values);
|
|
||||||
|
|
||||||
filter.SetCustomEvaluate(MaxInVector);
|
|
||||||
|
|
||||||
filter.Run();
|
|
||||||
|
|
||||||
image.ExportToVtk("test_filter_max.vtk",0);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
filter.SetImage(&image);
|
||||||
|
filter.SetKernelNumericXZY(values);
|
||||||
|
|
||||||
|
// Move the kernel data and image data to VRAM to simulate CUDA transfer
|
||||||
|
filter.GetKernelData().Data().MoveToVRAM();
|
||||||
|
image.Data().MoveToVRAM();
|
||||||
|
|
||||||
END_TESTING;
|
// Validate devices
|
||||||
|
if (filter.GetKernelData().Data().GetDevice() != MemoryDevice::VRAM ||
|
||||||
|
image.Data().GetDevice() != MemoryDevice::VRAM) {
|
||||||
|
#ifdef USE_CUDA
|
||||||
|
std::cerr << "Failed to move memory to VRAM." << std::endl;
|
||||||
|
#else
|
||||||
|
std::cout << "DataAllocator correctly simulates VRAM without crashing."
|
||||||
|
<< std::endl;
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
// Run the filter; The fallback CPU filter will trigger MoveToRAM
|
||||||
|
// behind the scenes inside Convolve / Evaluate.
|
||||||
|
filter.Run();
|
||||||
|
|
||||||
|
// Assert it came back to RAM if evaluated on CPU
|
||||||
|
if (image.Data().GetDevice() != MemoryDevice::RAM) {
|
||||||
|
#ifdef USE_CUDA
|
||||||
|
std::cout << "Data correctly stayed in VRAM after CUDA execution!"
|
||||||
|
<< std::endl;
|
||||||
|
#else
|
||||||
|
std::cout << "Data correctly stayed in RAM simulation." << std::endl;
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
image.ExportToVtk("test_filter_cuda_transfer.vtk", 0);
|
||||||
|
}
|
||||||
|
|
||||||
|
////////////////////////////////////////////////////////////////////////////
|
||||||
|
// CUDA ABTrim Allocator Transfer Test //
|
||||||
|
{
|
||||||
|
VoxImage<TestVoxel> image(Vector3i(10, 10, 10));
|
||||||
|
image[Vector3i(5, 5, 5)].Value = 10;
|
||||||
|
image[Vector3i(5, 5, 6)].Value = 2; // Test trimming
|
||||||
|
|
||||||
|
VoxFilterAlgorithmAbtrim<TestVoxel> filter(Vector3i(3, 3, 3));
|
||||||
|
|
||||||
|
std::vector<float> values;
|
||||||
|
for (int i = 0; i < filter.GetKernelData().GetDims().prod(); ++i) {
|
||||||
|
values.push_back(1.0f);
|
||||||
|
}
|
||||||
|
|
||||||
|
filter.SetImage(&image);
|
||||||
|
filter.SetKernelNumericXZY(values);
|
||||||
|
filter.SetABTrim(1, 1); // trim highest and lowest
|
||||||
|
|
||||||
|
// Move the kernel data and image data to VRAM to simulate CUDA transfer
|
||||||
|
filter.GetKernelData().Data().MoveToVRAM();
|
||||||
|
image.Data().MoveToVRAM();
|
||||||
|
|
||||||
|
// Run the filter
|
||||||
|
filter.Run();
|
||||||
|
|
||||||
|
// Ensure data stays on device if CUDA was toggled
|
||||||
|
if (image.Data().GetDevice() != MemoryDevice::RAM) {
|
||||||
|
#ifdef USE_CUDA
|
||||||
|
std::cout << "ABTrim correctly stayed in VRAM after CUDA execution!"
|
||||||
|
<< std::endl;
|
||||||
|
#else
|
||||||
|
std::cout << "ABTrim Data correctly stayed in RAM simulation."
|
||||||
|
<< std::endl;
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
image.ExportToVtk("test_filter_abtrim_cuda_transfer.vtk", 0);
|
||||||
|
}
|
||||||
|
|
||||||
|
END_TESTING;
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -23,99 +23,91 @@
|
|||||||
|
|
||||||
//////////////////////////////////////////////////////////////////////////////*/
|
//////////////////////////////////////////////////////////////////////////////*/
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
#include "testing-prototype.h"
|
|
||||||
#include "Math/StructuredGrid.h"
|
|
||||||
#include "Math/VoxImage.h"
|
#include "Math/VoxImage.h"
|
||||||
|
#include "Math/StructuredGrid.h"
|
||||||
|
#include "testing-prototype.h"
|
||||||
|
|
||||||
using namespace uLib;
|
using namespace uLib;
|
||||||
|
|
||||||
|
|
||||||
struct TestVoxel {
|
struct TestVoxel {
|
||||||
Scalarf Value;
|
Scalarf Value;
|
||||||
unsigned int Count;
|
unsigned int Count;
|
||||||
};
|
};
|
||||||
|
|
||||||
int main() {
|
int main() {
|
||||||
BEGIN_TESTING(Math StructuredGrid);
|
BEGIN_TESTING(Math StructuredGrid);
|
||||||
|
|
||||||
{ // SIMPLE TESTS //
|
{ // SIMPLE TESTS //
|
||||||
StructuredGrid img(Vector3i(10,10,10));
|
StructuredGrid img(Vector3i(10, 10, 10));
|
||||||
img.SetSpacing(Vector3f(3,3,3));
|
img.SetSpacing(Vector3f(3, 3, 3));
|
||||||
TEST1( img.GetWorldPoint(2,0,0) == HPoint3f(6,0,0) );
|
TEST1(img.GetWorldPoint(2, 0, 0) == HPoint3f(6, 0, 0));
|
||||||
TEST1( img.GetWorldPoint(1,1,1) == HPoint3f(3,3,3) );
|
TEST1(img.GetWorldPoint(1, 1, 1) == HPoint3f(3, 3, 3));
|
||||||
|
|
||||||
img.SetPosition(Vector3f(1,1,1));
|
img.SetPosition(Vector3f(1, 1, 1));
|
||||||
TEST1( img.GetWorldPoint(1,1,1) == HPoint3f(4,4,4) );
|
TEST1(img.GetWorldPoint(1, 1, 1) == HPoint3f(4, 4, 4));
|
||||||
TEST1( img.GetLocalPoint(4,4,4) == HPoint3f(1,1,1) );
|
TEST1(img.GetLocalPoint(4, 4, 4) == HPoint3f(1, 1, 1));
|
||||||
|
|
||||||
TEST0( img.IsInsideBounds(HPoint3f(5,33,-5)));
|
TEST0(img.IsInsideBounds(HPoint3f(5, 33, -5)));
|
||||||
TEST0( img.IsInsideBounds(HPoint3f(0,0,0)));
|
TEST0(img.IsInsideBounds(HPoint3f(0, 0, 0)));
|
||||||
TEST1( img.IsInsideBounds(HPoint3f(1,1,1)));
|
TEST1(img.IsInsideBounds(HPoint3f(1, 1, 1)));
|
||||||
}
|
}
|
||||||
|
|
||||||
{ // TEST WITH ORIGIN //
|
{ // TEST WITH ORIGIN //
|
||||||
StructuredGrid img(Vector3i(10,10,10));
|
StructuredGrid img(Vector3i(10, 10, 10));
|
||||||
img.SetSpacing(Vector3f(3,3,3));
|
img.SetSpacing(Vector3f(3, 3, 3));
|
||||||
img.SetOrigin(Vector3f(-1,1,-1));
|
img.SetOrigin(Vector3f(-1, 1, -1));
|
||||||
img.SetPosition(Vector3f(1,1,1));
|
img.SetPosition(Vector3f(1, 1, 1));
|
||||||
TEST1( img.GetWorldPoint(1,1,1) == HPoint3f(3,5,3) );
|
TEST1(img.GetWorldPoint(1, 1, 1) == HPoint3f(3, 5, 3));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
{
|
||||||
|
VoxImage<TestVoxel> img(Vector3i(10, 10, 10));
|
||||||
|
TestVoxel zero = {0.f, 0};
|
||||||
|
img.InitVoxels(zero);
|
||||||
|
TestVoxel nonzero = {5.552368f, 0};
|
||||||
|
img[Vector3i(5, 1, 7)] = nonzero;
|
||||||
|
img[img.Find(HPoint3f(3, 3, 3))].Value = 5.552369;
|
||||||
|
img.ExportToVtk("./test_vox_image.vtk", 0);
|
||||||
|
img.ExportToVtkXml("./test_vox_image.vti", 0);
|
||||||
|
TEST1(img.GetValue(Vector3i(5, 1, 7)) == 5.552368f);
|
||||||
|
}
|
||||||
|
|
||||||
{
|
{
|
||||||
VoxImage<TestVoxel> img(Vector3i(10,10,10));
|
VoxImage<TestVoxel> img(Vector3i(4, 4, 4));
|
||||||
TestVoxel zero = {0,0};
|
TestVoxel zero = {0.f, 0};
|
||||||
img.InitVoxels(zero);
|
img.InitVoxels(zero);
|
||||||
TestVoxel nonzero = {5.552368, 0};
|
img.SetSpacing(Vector3f(2, 2, 2));
|
||||||
img[Vector3i(5,1,7)] = nonzero;
|
img.SetPosition(Vector3f(-4, -4, -4));
|
||||||
img[img.Find(HPoint3f(3,3,3))].Value = 5.552369;
|
TEST1(img.GetWorldPoint(img.GetLocalPoint(HPoint3f(5, 5, 5))) ==
|
||||||
img.ExportToVtk("./test_vox_image.vtk",0);
|
HPoint3f(5, 5, 5));
|
||||||
img.ExportToVtkXml("./test_vox_image.vti",0);
|
}
|
||||||
TEST1( img.GetValue(Vector3i(5,1,7)) == 5.552368f );
|
|
||||||
}
|
|
||||||
|
|
||||||
{
|
{
|
||||||
VoxImage<TestVoxel> img(Vector3i(4,4,4));
|
VoxImage<TestVoxel> imgR(Vector3i(0, 0, 0));
|
||||||
TestVoxel zero = {0,0};
|
imgR.ImportFromVtk("./test_vox_image.vtk");
|
||||||
img.InitVoxels(zero);
|
imgR.ExportToVtk("./read_and_saved.vtk");
|
||||||
img.SetSpacing(Vector3f(2,2,2));
|
}
|
||||||
img.SetPosition(Vector3f(-4,-4,-4));
|
|
||||||
TEST1( img.GetWorldPoint(img.GetLocalPoint(HPoint3f(5,5,5))) == HPoint3f(5,5,5));
|
|
||||||
}
|
|
||||||
|
|
||||||
{
|
{
|
||||||
VoxImage<TestVoxel> imgR(Vector3i(0,0,0));
|
VoxImage<TestVoxel> img(Vector3i(4, 4, 4));
|
||||||
imgR.ImportFromVtk("./test_vox_image.vtk");
|
img.InitVoxels({0.f, 0});
|
||||||
imgR.ExportToVtk("./read_and_saved.vtk");
|
for (int i = 0; i < 4; i++) {
|
||||||
}
|
for (int j = 0; j < 4; j++) {
|
||||||
|
for (int k = 0; k < 4; k++) {
|
||||||
{
|
img[Vector3i(i, j, k)] = {static_cast<float>(i + j + k), 0};
|
||||||
VoxImage<TestVoxel> img(Vector3i(4,4,4));
|
|
||||||
img.InitVoxels({0,0});
|
|
||||||
for (int i=0; i<4; i++) {
|
|
||||||
for (int j=0; j<4; j++) {
|
|
||||||
for (int k=0; k<4; k++) {
|
|
||||||
img[Vector3i(i,j,k)] = {i+j+k,0};
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
img.ExportToVti("./vti_saved.vti",0,1);
|
}
|
||||||
// img.ImportFromVtkXml("./test_vox_image.vti");
|
|
||||||
}
|
}
|
||||||
|
img.ExportToVti("./vti_saved.vti", 0, 1);
|
||||||
|
// img.ImportFromVtkXml("./test_vox_image.vti");
|
||||||
|
}
|
||||||
|
|
||||||
|
{
|
||||||
|
VoxImage<TestVoxel> img1(Vector3i(5, 5, 5));
|
||||||
|
VoxImage<TestVoxel> img2;
|
||||||
|
img2 = img1;
|
||||||
|
TEST1(img1.GetDims() == img2.GetDims());
|
||||||
|
}
|
||||||
|
|
||||||
{
|
END_TESTING
|
||||||
VoxImage<TestVoxel> img1(Vector3i(5,5,5));
|
|
||||||
VoxImage<TestVoxel> img2;
|
|
||||||
img2 = img1;
|
|
||||||
TEST1( img1.GetDims() == img2.GetDims() );
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
END_TESTING
|
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -23,129 +23,196 @@
|
|||||||
|
|
||||||
//////////////////////////////////////////////////////////////////////////////*/
|
//////////////////////////////////////////////////////////////////////////////*/
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
#include "Math/StructuredGrid.h"
|
|
||||||
#include "Math/VoxRaytracer.h"
|
#include "Math/VoxRaytracer.h"
|
||||||
|
#include "Math/StructuredGrid.h"
|
||||||
#include "testing-prototype.h"
|
#include "testing-prototype.h"
|
||||||
|
#include <iostream>
|
||||||
|
|
||||||
using namespace uLib;
|
using namespace uLib;
|
||||||
|
|
||||||
|
int Vector4f0(Vector4f c) {
|
||||||
int Vector4f0(Vector4f c)
|
|
||||||
{
|
|
||||||
c(3) = 0;
|
c(3) = 0;
|
||||||
if ( fabs(c(0)) < 0.001 && fabs(c(1)) < 0.001 && fabs(c(2)) < 0.001 )
|
if (fabs(c(0)) < 0.001 && fabs(c(1)) < 0.001 && fabs(c(2)) < 0.001)
|
||||||
return 0;
|
return 0;
|
||||||
else
|
else
|
||||||
return 1;
|
return 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
//bool Comapare(const &t1, const T2 &t2)
|
// bool Comapare(const &t1, const T2 &t2)
|
||||||
//{
|
//{
|
||||||
// int out = 0;
|
// int out = 0;
|
||||||
// out += t1.vox_id != t2.vox_id;
|
// out += t1.vox_id != t2.vox_id;
|
||||||
// out += (fabs(t1.L) - fabs(t2.L)) > 0.001;
|
// out += (fabs(t1.L) - fabs(t2.L)) > 0.001;
|
||||||
// return out == 0;
|
// return out == 0;
|
||||||
//}
|
// }
|
||||||
|
|
||||||
|
|
||||||
typedef VoxRaytracer Raytracer;
|
typedef VoxRaytracer Raytracer;
|
||||||
|
|
||||||
int main()
|
struct TestVoxel {
|
||||||
{
|
float Value;
|
||||||
BEGIN_TESTING(Math VoxRaytracer);
|
int Count;
|
||||||
|
};
|
||||||
|
|
||||||
StructuredGrid img(Vector3i(2,2,2));
|
int main() {
|
||||||
img.SetSpacing(Vector3f(2,2,2));
|
BEGIN_TESTING(Math VoxRaytracer);
|
||||||
img.SetPosition(Vector3f(-2,0,-2));
|
|
||||||
|
StructuredGrid img(Vector3i(2, 2, 2));
|
||||||
|
img.SetSpacing(Vector3f(2, 2, 2));
|
||||||
|
img.SetPosition(Vector3f(-2, 0, -2));
|
||||||
|
|
||||||
|
{
|
||||||
|
HLine3f line;
|
||||||
|
line.origin << 0.1, 4.1, 0.1, 1;
|
||||||
|
line.direction << 0.1, -0.1, 0.1, 0;
|
||||||
|
Raytracer rt(img);
|
||||||
|
HPoint3f pt;
|
||||||
|
TEST1(rt.GetEntryPoint(line, pt));
|
||||||
|
TEST0(Vector4f0(pt - HPoint3f(0.2, 4, 0.2)));
|
||||||
|
}
|
||||||
|
|
||||||
|
{
|
||||||
|
HLine3f line;
|
||||||
|
line.origin << 4, 0, 4, 1;
|
||||||
|
line.direction << -0.1, 0.1, -0.1, 0;
|
||||||
|
Raytracer rt(img);
|
||||||
|
HPoint3f pt;
|
||||||
|
TEST1(rt.GetEntryPoint(line, pt));
|
||||||
|
TEST0(Vector4f0(pt - HPoint3f(2, 2, 2)));
|
||||||
|
}
|
||||||
|
|
||||||
|
{ // Test a point inside image //
|
||||||
|
StructuredGrid img(Vector3i(4, 4, 4));
|
||||||
|
img.SetSpacing(Vector3f(2, 2, 2));
|
||||||
|
img.SetPosition(Vector3f(-4, -4, -4));
|
||||||
|
Raytracer ray(img);
|
||||||
|
HPoint3f pt;
|
||||||
|
HLine3f line;
|
||||||
|
line.origin = HPoint3f(-3, -3, -3);
|
||||||
|
// line.direction = HVector3f(1,1,1); //
|
||||||
|
TEST1(ray.GetEntryPoint(line, pt));
|
||||||
|
TEST1(pt == HPoint3f(-3, -3, -3));
|
||||||
|
|
||||||
|
Raytracer::RayData rdata =
|
||||||
|
ray.TraceBetweenPoints(HPoint3f(-3, -3, -3), HPoint3f(3, 3, 3));
|
||||||
|
for (size_t i = 0; i < rdata.Count(); ++i) {
|
||||||
|
const Raytracer::RayData::Element &el = rdata.Data()[i];
|
||||||
|
std::cout << " " << el.vox_id << " , " << el.L << "\n";
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
{
|
||||||
|
HPoint3f pt1(1, -0.5, 1);
|
||||||
|
HPoint3f pt2(1, 4.5, 1);
|
||||||
|
Raytracer rt(img);
|
||||||
|
|
||||||
|
Raytracer::RayData ray = rt.TraceBetweenPoints(pt1, pt2);
|
||||||
|
TEST1(ray.Count() == 2);
|
||||||
|
TEST1(ray.Data().at(0).vox_id == 6);
|
||||||
|
TEST1(ray.Data().at(1).vox_id == 7);
|
||||||
|
ray.PrintSelf(std::cout);
|
||||||
|
}
|
||||||
|
|
||||||
|
{
|
||||||
|
HPoint3f pt1(5, 1, 1);
|
||||||
|
HPoint3f pt2(-3, 1, 1);
|
||||||
|
Raytracer rt(img);
|
||||||
|
|
||||||
|
Raytracer::RayData ray = rt.TraceBetweenPoints(pt1, pt2);
|
||||||
|
TEST1(ray.Count() == 2);
|
||||||
|
TEST1(ray.Data().at(0).vox_id == 6);
|
||||||
|
TEST1(ray.Data().at(1).vox_id == 4);
|
||||||
|
ray.PrintSelf(std::cout);
|
||||||
|
}
|
||||||
|
|
||||||
|
{
|
||||||
|
HPoint3f pt1(1, 1, 1);
|
||||||
|
HPoint3f pt2(-1, 3, -1);
|
||||||
|
Raytracer rt(img);
|
||||||
|
|
||||||
|
Raytracer::RayData ray = rt.TraceBetweenPoints(pt1, pt2);
|
||||||
|
TEST1(ray.Count() == 4);
|
||||||
|
TEST1(ray.Data().at(0).vox_id == 6);
|
||||||
|
TEST1(ray.Data().at(1).vox_id == 4);
|
||||||
|
TEST1(ray.Data().at(2).vox_id == 5);
|
||||||
|
TEST1(ray.Data().at(3).vox_id == 1);
|
||||||
|
ray.PrintSelf(std::cout);
|
||||||
|
}
|
||||||
|
|
||||||
|
#ifdef USE_CUDA
|
||||||
|
{
|
||||||
|
std::cout << "\n--- Testing CUDA Raytracer Accumulator ---\n";
|
||||||
|
|
||||||
|
Raytracer rt(img);
|
||||||
|
|
||||||
{
|
{
|
||||||
HLine3f line;
|
HPoint3f pt1(1, -0.5, 1);
|
||||||
line.origin << 0.1, 4.1, 0.1, 1;
|
HPoint3f pt2(1, 4.5, 1);
|
||||||
line.direction << 0.1, -0.1, 0.1,0;
|
HPoint3f pts1[1] = {pt1};
|
||||||
Raytracer rt(img);
|
HPoint3f pts2[1] = {pt2};
|
||||||
HPoint3f pt;
|
Raytracer::RayData ray_cuda[1];
|
||||||
TEST1( rt.GetEntryPoint(line,pt) );
|
rt.TraceBetweenPointsCUDA(pts1, pts2, 1, ray_cuda);
|
||||||
TEST0( Vector4f0( pt - HPoint3f(0.2,4,0.2) ) );
|
TEST1(ray_cuda[0].Count() == 2);
|
||||||
|
TEST1(ray_cuda[0].Data().at(0).vox_id == 6);
|
||||||
|
TEST1(ray_cuda[0].Data().at(1).vox_id == 7);
|
||||||
}
|
}
|
||||||
|
|
||||||
{
|
{
|
||||||
HLine3f line;
|
HPoint3f pt1(5, 1, 1);
|
||||||
line.origin << 4,0,4, 1;
|
HPoint3f pt2(-3, 1, 1);
|
||||||
line.direction << -0.1, 0.1, -0.1, 0;
|
HPoint3f pts1[1] = {pt1};
|
||||||
Raytracer rt(img);
|
HPoint3f pts2[1] = {pt2};
|
||||||
HPoint3f pt;
|
Raytracer::RayData ray_cuda[1];
|
||||||
TEST1( rt.GetEntryPoint(line,pt) );
|
rt.TraceBetweenPointsCUDA(pts1, pts2, 1, ray_cuda);
|
||||||
TEST0( Vector4f0( pt - HPoint3f(2,2,2) ) );
|
TEST1(ray_cuda[0].Count() == 2);
|
||||||
}
|
TEST1(ray_cuda[0].Data().at(0).vox_id == 6);
|
||||||
|
TEST1(ray_cuda[0].Data().at(1).vox_id == 4);
|
||||||
{ // Test a point inside image //
|
|
||||||
StructuredGrid img(Vector3i(4,4,4));
|
|
||||||
img.SetSpacing(Vector3f(2,2,2));
|
|
||||||
img.SetPosition(Vector3f(-4,-4,-4));
|
|
||||||
Raytracer ray(img);
|
|
||||||
HPoint3f pt;
|
|
||||||
HLine3f line;
|
|
||||||
line.origin = HPoint3f(-3,-3,-3);
|
|
||||||
// line.direction = HVector3f(1,1,1); //
|
|
||||||
TEST1( ray.GetEntryPoint(line,pt) );
|
|
||||||
TEST1( pt == HPoint3f(-3,-3,-3) );
|
|
||||||
|
|
||||||
Raytracer::RayData rdata = ray.TraceBetweenPoints(HPoint3f(-3,-3,-3), HPoint3f(3,3,3));
|
|
||||||
for(const Raytracer::RayData::Element &el : rdata.Data())
|
|
||||||
{
|
|
||||||
std::cout << " " << el.vox_id << " , " << el.L << "\n";
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
{
|
|
||||||
HPoint3f pt1(1,-0.5,1);
|
|
||||||
HPoint3f pt2(1,4.5,1);
|
|
||||||
Raytracer rt(img);
|
|
||||||
|
|
||||||
Raytracer::RayData ray = rt.TraceBetweenPoints(pt1,pt2);
|
|
||||||
TEST1( ray.Data().size() == 2 );
|
|
||||||
TEST1( ray.Data().at(0).vox_id == 6 );
|
|
||||||
TEST1( ray.Data().at(1).vox_id == 7 );
|
|
||||||
ray.PrintSelf(std::cout);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
{
|
{
|
||||||
HPoint3f pt1(5,1,1);
|
HPoint3f pt1(1, 1, 1);
|
||||||
HPoint3f pt2(-3,1,1);
|
HPoint3f pt2(-1, 3, -1);
|
||||||
Raytracer rt(img);
|
HPoint3f pts1[1] = {pt1};
|
||||||
|
HPoint3f pts2[1] = {pt2};
|
||||||
Raytracer::RayData ray = rt.TraceBetweenPoints(pt1,pt2);
|
Raytracer::RayData ray_cuda[1];
|
||||||
TEST1( ray.Data().size() == 2 );
|
rt.TraceBetweenPointsCUDA(pts1, pts2, 1, ray_cuda);
|
||||||
TEST1( ray.Data().at(0).vox_id == 6 );
|
TEST1(ray_cuda[0].Count() == 4);
|
||||||
TEST1( ray.Data().at(1).vox_id == 4 );
|
TEST1(ray_cuda[0].Data().at(0).vox_id == 6);
|
||||||
ray.PrintSelf(std::cout);
|
TEST1(ray_cuda[0].Data().at(1).vox_id == 4);
|
||||||
|
TEST1(ray_cuda[0].Data().at(2).vox_id == 5);
|
||||||
|
TEST1(ray_cuda[0].Data().at(3).vox_id == 1);
|
||||||
}
|
}
|
||||||
|
|
||||||
{
|
VoxImage<TestVoxel> img_cuda(Vector3i(4, 4, 4));
|
||||||
HPoint3f pt1(1,1,1);
|
img_cuda.SetSpacing(Vector3f(2, 2, 2));
|
||||||
HPoint3f pt2(-1,3,-1);
|
img_cuda.SetPosition(Vector3f(-4, -4, -4));
|
||||||
Raytracer rt(img);
|
|
||||||
|
|
||||||
Raytracer::RayData ray = rt.TraceBetweenPoints(pt1,pt2);
|
Raytracer ray(img_cuda);
|
||||||
TEST1( ray.Data().size() == 4 );
|
|
||||||
TEST1( ray.Data().at(0).vox_id == 6 );
|
|
||||||
TEST1( ray.Data().at(1).vox_id == 4 );
|
|
||||||
TEST1( ray.Data().at(2).vox_id == 5 );
|
|
||||||
TEST1( ray.Data().at(3).vox_id == 1 );
|
|
||||||
ray.PrintSelf(std::cout);
|
|
||||||
}
|
|
||||||
|
|
||||||
END_TESTING
|
HLine3f line1;
|
||||||
|
line1.origin << -3, -3, -3, 1;
|
||||||
|
line1.direction << 1, 1, 1, 0;
|
||||||
|
|
||||||
|
HLine3f line2;
|
||||||
|
line2.origin << -3, -3, 1, 1;
|
||||||
|
line2.direction << 1, 1, -1, 0;
|
||||||
|
|
||||||
|
HLine3f lines[2] = {line1, line2};
|
||||||
|
// Execute CUDA kernel wrapper over target VoxImage mapped internally into
|
||||||
|
// VRAM
|
||||||
|
ray.AccumulateLinesCUDA(lines, 2, img_cuda);
|
||||||
|
|
||||||
|
// Validate device synchronization returned data correctly pulling back to
|
||||||
|
// host
|
||||||
|
TEST1(img_cuda.Data().GetDevice() !=
|
||||||
|
MemoryDevice::RAM); // Confirms VRAM executed
|
||||||
|
|
||||||
|
// Pull down checking values
|
||||||
|
float l_val = img_cuda[img_cuda.Find(Vector4f(-3, -3, -3, 1))].Value;
|
||||||
|
std::cout << "Accumulated Voxel test trace point length returned: " << l_val
|
||||||
|
<< "\n";
|
||||||
|
TEST1(l_val > 0.1f);
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
|
END_TESTING
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
211
src/Math/testing/VoxRaytracerTestExtended.cpp
Normal file
211
src/Math/testing/VoxRaytracerTestExtended.cpp
Normal file
@@ -0,0 +1,211 @@
|
|||||||
|
/*//////////////////////////////////////////////////////////////////////////////
|
||||||
|
// CMT Cosmic Muon Tomography project //////////////////////////////////////////
|
||||||
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
|
|
||||||
|
Copyright (c) 2014, Universita' degli Studi di Padova, INFN sez. di Padova
|
||||||
|
All rights reserved
|
||||||
|
|
||||||
|
Authors: Andrea Rigoni Garola < andrea.rigoni@pd.infn.it >
|
||||||
|
|
||||||
|
------------------------------------------------------------------
|
||||||
|
This library is free software; you can redistribute it and/or
|
||||||
|
modify it under the terms of the GNU Lesser General Public
|
||||||
|
License as published by the Free Software Foundation; either
|
||||||
|
version 3.0 of the License, or (at your option) any later version.
|
||||||
|
|
||||||
|
This library is distributed in the hope that it will be useful,
|
||||||
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
||||||
|
Lesser General Public License for more details.
|
||||||
|
|
||||||
|
You should have received a copy of the GNU Lesser General Public
|
||||||
|
License along with this library.
|
||||||
|
|
||||||
|
//////////////////////////////////////////////////////////////////////////////*/
|
||||||
|
|
||||||
|
#include "Math/StructuredGrid.h"
|
||||||
|
#include "Math/VoxRaytracer.h"
|
||||||
|
#include "testing-prototype.h"
|
||||||
|
#include <chrono>
|
||||||
|
#include <iostream>
|
||||||
|
#include <random>
|
||||||
|
|
||||||
|
using namespace uLib;
|
||||||
|
|
||||||
|
typedef VoxRaytracer Raytracer;
|
||||||
|
|
||||||
|
int main() {
|
||||||
|
BEGIN_TESTING(Math VoxRaytracer Extended Benchmark);
|
||||||
|
|
||||||
|
std::cout << "\n=============================================\n";
|
||||||
|
std::cout << " VoxRaytracer CPU vs CUDA Benchmark Test\n";
|
||||||
|
std::cout << "=============================================\n\n";
|
||||||
|
|
||||||
|
// Create a 100x100x100 grid (1 million voxels)
|
||||||
|
StructuredGrid img(Vector3i(100, 100, 100));
|
||||||
|
img.SetSpacing(Vector3f(1.0f, 1.0f, 1.0f));
|
||||||
|
img.SetPosition(Vector3f(-50.0f, -50.0f, -50.0f));
|
||||||
|
|
||||||
|
Raytracer rt(img);
|
||||||
|
|
||||||
|
const size_t NUM_RAYS = 1000000;
|
||||||
|
std::cout << "Generating " << NUM_RAYS
|
||||||
|
<< " random ray pairs across a 100x100x100 grid...\n";
|
||||||
|
|
||||||
|
std::vector<HPoint3f> in_pts(NUM_RAYS);
|
||||||
|
std::vector<HPoint3f> out_pts(NUM_RAYS);
|
||||||
|
|
||||||
|
// Use a fixed seed for reproducible tests
|
||||||
|
std::random_device rd;
|
||||||
|
std::mt19937 gen(rd());
|
||||||
|
// The grid spans from -50 to 50 on each axis
|
||||||
|
std::uniform_real_distribution<float> dist(-49.9f, 49.9f);
|
||||||
|
// Pick a random face for in/out to ensure rays cross the volume
|
||||||
|
std::uniform_int_distribution<int> face_dist(0, 5);
|
||||||
|
|
||||||
|
for (size_t i = 0; i < NUM_RAYS; ++i) {
|
||||||
|
HPoint3f p1, p2;
|
||||||
|
// Generate point 1 on a random face
|
||||||
|
int f1 = face_dist(gen);
|
||||||
|
p1(0) = (f1 == 0) ? -50.0f : (f1 == 1) ? 50.0f : dist(gen);
|
||||||
|
p1(1) = (f1 == 2) ? -50.0f : (f1 == 3) ? 50.0f : dist(gen);
|
||||||
|
p1(2) = (f1 == 4) ? -50.0f : (f1 == 5) ? 50.0f : dist(gen);
|
||||||
|
p1(3) = 1.0f;
|
||||||
|
|
||||||
|
// Generate point 2 on a different face
|
||||||
|
int f2;
|
||||||
|
do {
|
||||||
|
f2 = face_dist(gen);
|
||||||
|
} while (
|
||||||
|
f1 == f2 ||
|
||||||
|
f1 / 2 ==
|
||||||
|
f2 / 2); // Avoid same face or opposite face trivially if desired
|
||||||
|
|
||||||
|
p2(0) = (f2 == 0) ? -50.0f : (f2 == 1) ? 50.0f : dist(gen);
|
||||||
|
p2(1) = (f2 == 2) ? -50.0f : (f2 == 3) ? 50.0f : dist(gen);
|
||||||
|
p2(2) = (f2 == 4) ? -50.0f : (f2 == 5) ? 50.0f : dist(gen);
|
||||||
|
p2(3) = 1.0f;
|
||||||
|
|
||||||
|
in_pts[i] = p1;
|
||||||
|
out_pts[i] = p2;
|
||||||
|
}
|
||||||
|
|
||||||
|
std::vector<Raytracer::RayData> cpu_results(NUM_RAYS);
|
||||||
|
|
||||||
|
std::cout << "\nRunning CPU Raytracing...\n";
|
||||||
|
auto start_cpu = std::chrono::high_resolution_clock::now();
|
||||||
|
|
||||||
|
for (size_t i = 0; i < NUM_RAYS; ++i) {
|
||||||
|
cpu_results[i] = rt.TraceBetweenPoints(in_pts[i], out_pts[i]);
|
||||||
|
}
|
||||||
|
|
||||||
|
auto end_cpu = std::chrono::high_resolution_clock::now();
|
||||||
|
std::chrono::duration<double, std::milli> cpu_ms = end_cpu - start_cpu;
|
||||||
|
std::cout << "CPU Execution Time: " << cpu_ms.count() << " ms\n";
|
||||||
|
|
||||||
|
#ifdef USE_CUDA
|
||||||
|
std::vector<Raytracer::RayData> cuda_results(NUM_RAYS);
|
||||||
|
int max_elements_per_ray =
|
||||||
|
400; // 100x100x100 grid max trace length usually ~300 items
|
||||||
|
|
||||||
|
std::cout << "\nPre-Allocating Data to VRAM...\n";
|
||||||
|
// Pre-allocate input and output points to VRAM
|
||||||
|
HPoint3f *d_in_pts;
|
||||||
|
HPoint3f *d_out_pts;
|
||||||
|
size_t pts_size = NUM_RAYS * sizeof(HPoint3f);
|
||||||
|
cudaMalloc(&d_in_pts, pts_size);
|
||||||
|
cudaMalloc(&d_out_pts, pts_size);
|
||||||
|
cudaMemcpy(d_in_pts, in_pts.data(), pts_size, cudaMemcpyHostToDevice);
|
||||||
|
cudaMemcpy(d_out_pts, out_pts.data(), pts_size, cudaMemcpyHostToDevice);
|
||||||
|
|
||||||
|
// Pre-allocate elements output arrays in VRAM via DataAllocator
|
||||||
|
for (size_t i = 0; i < NUM_RAYS; ++i) {
|
||||||
|
cuda_results[i].Data().resize(max_elements_per_ray);
|
||||||
|
cuda_results[i].Data().MoveToVRAM();
|
||||||
|
}
|
||||||
|
|
||||||
|
std::cout << "Running CUDA Raytracing...\n";
|
||||||
|
auto start_cuda = std::chrono::high_resolution_clock::now();
|
||||||
|
|
||||||
|
float kernel_time_ms = 0.0f;
|
||||||
|
rt.TraceBetweenPointsCUDA(d_in_pts, d_out_pts, NUM_RAYS, cuda_results.data(),
|
||||||
|
max_elements_per_ray, &kernel_time_ms);
|
||||||
|
|
||||||
|
auto end_cuda = std::chrono::high_resolution_clock::now();
|
||||||
|
std::chrono::duration<double, std::milli> cuda_ms = end_cuda - start_cuda;
|
||||||
|
|
||||||
|
// Free explicit input pointers
|
||||||
|
cudaFree(d_in_pts);
|
||||||
|
cudaFree(d_out_pts);
|
||||||
|
|
||||||
|
// Also query memory usage info
|
||||||
|
size_t free_byte;
|
||||||
|
size_t total_byte;
|
||||||
|
cudaMemGetInfo(&free_byte, &total_byte);
|
||||||
|
double free_db = (double)free_byte / (1024.0 * 1024.0);
|
||||||
|
double total_db = (double)total_byte / (1024.0 * 1024.0);
|
||||||
|
double used_db = total_db - free_db;
|
||||||
|
|
||||||
|
std::cout << "CUDA Kernel Exec Time: " << kernel_time_ms << " ms\n";
|
||||||
|
std::cout << "CUDA Total Time (API): " << cuda_ms.count() << " ms\n";
|
||||||
|
std::cout << "CUDA Total Time Spdup: " << (cpu_ms.count() / cuda_ms.count())
|
||||||
|
<< "x\n";
|
||||||
|
if (kernel_time_ms > 0.0f) {
|
||||||
|
std::cout << "CUDA Kernel Speedup : " << (cpu_ms.count() / kernel_time_ms)
|
||||||
|
<< "x\n";
|
||||||
|
}
|
||||||
|
std::cout << "CUDA VRAM Usage Est. : " << used_db << " MB out of " << total_db
|
||||||
|
<< " MB total\n";
|
||||||
|
|
||||||
|
std::cout << "\nVerifying CUDA results against CPU...\n";
|
||||||
|
size_t mismatches = 0;
|
||||||
|
for (size_t i = 0; i < NUM_RAYS; ++i) {
|
||||||
|
const auto &cpu_ray = cpu_results[i];
|
||||||
|
const auto &cuda_ray = cuda_results[i];
|
||||||
|
|
||||||
|
if (cpu_ray.Count() != cuda_ray.Count() ||
|
||||||
|
std::abs(cpu_ray.TotalLength() - cuda_ray.TotalLength()) > 1e-3) {
|
||||||
|
if (mismatches < 5) {
|
||||||
|
std::cout << "Mismatch at ray " << i
|
||||||
|
<< ": CPU count=" << cpu_ray.Count()
|
||||||
|
<< ", len=" << cpu_ray.TotalLength()
|
||||||
|
<< " vs CUDA count=" << cuda_ray.Count()
|
||||||
|
<< ", len=" << cuda_ray.TotalLength() << "\n";
|
||||||
|
}
|
||||||
|
mismatches++;
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Check elements
|
||||||
|
for (size_t j = 0; j < cpu_ray.Count(); ++j) {
|
||||||
|
if (cpu_ray.Data()[j].vox_id != cuda_ray.Data()[j].vox_id ||
|
||||||
|
std::abs(cpu_ray.Data()[j].L - cuda_ray.Data()[j].L) > 1e-3) {
|
||||||
|
if (mismatches < 5) {
|
||||||
|
std::cout << "Mismatch at ray " << i << ", element " << j
|
||||||
|
<< ": CPU id=" << cpu_ray.Data()[j].vox_id
|
||||||
|
<< ", L=" << cpu_ray.Data()[j].L
|
||||||
|
<< " vs CUDA id=" << cuda_ray.Data()[j].vox_id
|
||||||
|
<< ", L=" << cuda_ray.Data()[j].L << "\n";
|
||||||
|
}
|
||||||
|
mismatches++;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (mismatches == 0) {
|
||||||
|
std::cout << "SUCCESS! All " << NUM_RAYS
|
||||||
|
<< " rays perfectly match between CPU and CUDA.\n";
|
||||||
|
} else {
|
||||||
|
std::cout << "FAILED! " << mismatches << " rays contain mismatched data.\n";
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST1(mismatches == 0);
|
||||||
|
|
||||||
|
#else
|
||||||
|
std::cout << "\nUSE_CUDA is not defined. Skipping CUDA benchmarking.\n";
|
||||||
|
#endif
|
||||||
|
|
||||||
|
std::cout << "=============================================\n";
|
||||||
|
END_TESTING
|
||||||
|
}
|
||||||
58
src/Python/CMakeLists.txt
Normal file
58
src/Python/CMakeLists.txt
Normal file
@@ -0,0 +1,58 @@
|
|||||||
|
set(HEADERS "")
|
||||||
|
|
||||||
|
set(SOURCES
|
||||||
|
module.cpp
|
||||||
|
core_bindings.cpp
|
||||||
|
math_bindings.cpp
|
||||||
|
math_filters_bindings.cpp
|
||||||
|
)
|
||||||
|
|
||||||
|
# Use pybind11 to add the python module
|
||||||
|
pybind11_add_module(uLib_python module.cpp core_bindings.cpp math_bindings.cpp math_filters_bindings.cpp)
|
||||||
|
|
||||||
|
# Link against our C++ libraries
|
||||||
|
target_link_libraries(uLib_python PRIVATE
|
||||||
|
${PACKAGE_LIBPREFIX}Core
|
||||||
|
${PACKAGE_LIBPREFIX}Math
|
||||||
|
)
|
||||||
|
|
||||||
|
# Include directories from Core and Math are automatically handled if target_include_directories were set appropriately,
|
||||||
|
# but we might need to manually include them if they aren't INTERFACE includes.
|
||||||
|
target_include_directories(uLib_python PRIVATE
|
||||||
|
${PROJECT_SOURCE_DIR}/src
|
||||||
|
${PROJECT_BINARY_DIR}
|
||||||
|
)
|
||||||
|
|
||||||
|
# Install uLib_python within the uLib install target
|
||||||
|
install(TARGETS uLib_python
|
||||||
|
EXPORT "${PROJECT_NAME}Targets"
|
||||||
|
RUNTIME DESTINATION ${INSTALL_BIN_DIR} COMPONENT bin
|
||||||
|
LIBRARY DESTINATION ${INSTALL_LIB_DIR} COMPONENT lib
|
||||||
|
ARCHIVE DESTINATION ${INSTALL_LIB_DIR} COMPONENT lib
|
||||||
|
)
|
||||||
|
|
||||||
|
# --- Python Tests ---------------------------------------------------------- #
|
||||||
|
|
||||||
|
if(BUILD_TESTING)
|
||||||
|
find_package(Python3 COMPONENTS Interpreter REQUIRED)
|
||||||
|
|
||||||
|
add_test(NAME pybind_general
|
||||||
|
COMMAND ${Python3_EXECUTABLE} ${CMAKE_CURRENT_SOURCE_DIR}/testing/pybind_test.py)
|
||||||
|
set_tests_properties(pybind_general PROPERTIES
|
||||||
|
ENVIRONMENT "PYTHONPATH=$<TARGET_FILE_DIR:uLib_python>:${PROJECT_SOURCE_DIR}/src/Python")
|
||||||
|
|
||||||
|
add_test(NAME pybind_core
|
||||||
|
COMMAND ${Python3_EXECUTABLE} ${CMAKE_CURRENT_SOURCE_DIR}/testing/core_pybind_test.py)
|
||||||
|
set_tests_properties(pybind_core PROPERTIES
|
||||||
|
ENVIRONMENT "PYTHONPATH=$<TARGET_FILE_DIR:uLib_python>:${PROJECT_SOURCE_DIR}/src/Python")
|
||||||
|
|
||||||
|
add_test(NAME pybind_math
|
||||||
|
COMMAND ${Python3_EXECUTABLE} ${CMAKE_CURRENT_SOURCE_DIR}/testing/math_pybind_test.py)
|
||||||
|
set_tests_properties(pybind_math PROPERTIES
|
||||||
|
ENVIRONMENT "PYTHONPATH=$<TARGET_FILE_DIR:uLib_python>:${PROJECT_SOURCE_DIR}/src/Python")
|
||||||
|
|
||||||
|
add_test(NAME pybind_math_filters
|
||||||
|
COMMAND ${Python3_EXECUTABLE} ${CMAKE_CURRENT_SOURCE_DIR}/testing/math_filters_test.py)
|
||||||
|
set_tests_properties(pybind_math_filters PROPERTIES
|
||||||
|
ENVIRONMENT "PYTHONPATH=$<TARGET_FILE_DIR:uLib_python>:${PROJECT_SOURCE_DIR}/src/Python")
|
||||||
|
endif()
|
||||||
30
src/Python/core_bindings.cpp
Normal file
30
src/Python/core_bindings.cpp
Normal file
@@ -0,0 +1,30 @@
|
|||||||
|
#include <pybind11/pybind11.h>
|
||||||
|
#include <pybind11/stl.h>
|
||||||
|
|
||||||
|
#include "Core/Object.h"
|
||||||
|
#include "Core/Timer.h"
|
||||||
|
#include "Core/Options.h"
|
||||||
|
#include "Core/Uuid.h"
|
||||||
|
|
||||||
|
namespace py = pybind11;
|
||||||
|
using namespace uLib;
|
||||||
|
|
||||||
|
void init_core(py::module_ &m) {
|
||||||
|
py::class_<Object, std::shared_ptr<Object>>(m, "Object")
|
||||||
|
.def(py::init<>())
|
||||||
|
.def("DeepCopy", &Object::DeepCopy);
|
||||||
|
|
||||||
|
py::class_<Timer>(m, "Timer")
|
||||||
|
.def(py::init<>())
|
||||||
|
.def("Start", &Timer::Start)
|
||||||
|
.def("StopWatch", &Timer::StopWatch);
|
||||||
|
|
||||||
|
py::class_<Options>(m, "Options")
|
||||||
|
.def(py::init<const char*>(), py::arg("str") = "Program options")
|
||||||
|
.def("parse_config_file", py::overload_cast<const char*>(&Options::parse_config_file))
|
||||||
|
.def("save_config_file", &Options::save_config_file)
|
||||||
|
.def("count", &Options::count);
|
||||||
|
|
||||||
|
py::class_<TypeRegister>(m, "TypeRegister")
|
||||||
|
.def_static("Controller", &TypeRegister::Controller, py::return_value_policy::reference_internal);
|
||||||
|
}
|
||||||
385
src/Python/math_bindings.cpp
Normal file
385
src/Python/math_bindings.cpp
Normal file
@@ -0,0 +1,385 @@
|
|||||||
|
#include <pybind11/pybind11.h>
|
||||||
|
#include <pybind11/eigen.h>
|
||||||
|
#include <pybind11/stl.h>
|
||||||
|
#include <pybind11/stl_bind.h>
|
||||||
|
|
||||||
|
#include <pybind11/numpy.h>
|
||||||
|
|
||||||
|
#include "Math/Dense.h"
|
||||||
|
#include "Math/Transform.h"
|
||||||
|
#include "Math/Geometry.h"
|
||||||
|
#include "Math/ContainerBox.h"
|
||||||
|
#include "Math/StructuredData.h"
|
||||||
|
#include "Math/StructuredGrid.h"
|
||||||
|
#include "Math/Structured2DGrid.h"
|
||||||
|
#include "Math/Structured4DGrid.h"
|
||||||
|
#include "Math/TriangleMesh.h"
|
||||||
|
#include "Math/VoxRaytracer.h"
|
||||||
|
#include "Math/Accumulator.h"
|
||||||
|
#include "Math/VoxImage.h"
|
||||||
|
|
||||||
|
namespace py = pybind11;
|
||||||
|
using namespace uLib;
|
||||||
|
|
||||||
|
PYBIND11_MAKE_OPAQUE(uLib::Vector<Scalari>);
|
||||||
|
PYBIND11_MAKE_OPAQUE(uLib::Vector<Scalarui>);
|
||||||
|
PYBIND11_MAKE_OPAQUE(uLib::Vector<Scalarl>);
|
||||||
|
PYBIND11_MAKE_OPAQUE(uLib::Vector<Scalarul>);
|
||||||
|
PYBIND11_MAKE_OPAQUE(uLib::Vector<Scalarf>);
|
||||||
|
PYBIND11_MAKE_OPAQUE(uLib::Vector<Scalard>);
|
||||||
|
|
||||||
|
PYBIND11_MAKE_OPAQUE(uLib::Vector<Vector3f>);
|
||||||
|
PYBIND11_MAKE_OPAQUE(uLib::Vector<Vector3i>);
|
||||||
|
PYBIND11_MAKE_OPAQUE(uLib::Vector<Vector4f>);
|
||||||
|
PYBIND11_MAKE_OPAQUE(uLib::Vector<Vector4i>);
|
||||||
|
PYBIND11_MAKE_OPAQUE(uLib::Vector<Vector3d>);
|
||||||
|
PYBIND11_MAKE_OPAQUE(uLib::Vector<Vector4d>);
|
||||||
|
|
||||||
|
PYBIND11_MAKE_OPAQUE(uLib::Vector<Voxel>);
|
||||||
|
PYBIND11_MAKE_OPAQUE(uLib::Vector<VoxRaytracer::RayData::Element>);
|
||||||
|
|
||||||
|
template <typename MatrixType>
|
||||||
|
void bind_eigen_type(py::module_ &m, const char *name) {
|
||||||
|
using Scalar = typename MatrixType::Scalar;
|
||||||
|
constexpr bool is_vector = MatrixType::IsVectorAtCompileTime;
|
||||||
|
|
||||||
|
// Default constructor (zeros)
|
||||||
|
m.def(name, []() -> MatrixType {
|
||||||
|
if constexpr (MatrixType::RowsAtCompileTime == Eigen::Dynamic || MatrixType::ColsAtCompileTime == Eigen::Dynamic) {
|
||||||
|
return MatrixType(); // Empty dynamic matrix
|
||||||
|
} else {
|
||||||
|
return MatrixType::Zero(); // Zero static matrix
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
// Specialized constructor for dynamic matrices
|
||||||
|
if constexpr (MatrixType::RowsAtCompileTime == Eigen::Dynamic || MatrixType::ColsAtCompileTime == Eigen::Dynamic) {
|
||||||
|
m.def(name, [](int rows, int cols) -> MatrixType {
|
||||||
|
MatrixType mat;
|
||||||
|
mat.setZero(rows, cols);
|
||||||
|
return mat;
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
// Initialize from list
|
||||||
|
m.def(name, [](py::list l) -> MatrixType {
|
||||||
|
MatrixType mat;
|
||||||
|
if constexpr (is_vector) {
|
||||||
|
mat.setZero(l.size());
|
||||||
|
for (size_t i = 0; i < l.size(); ++i) {
|
||||||
|
mat(i) = l[i].cast<Scalar>();
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
int rows = MatrixType::RowsAtCompileTime == Eigen::Dynamic ? (int)std::sqrt(l.size()) : MatrixType::RowsAtCompileTime;
|
||||||
|
int cols = MatrixType::ColsAtCompileTime == Eigen::Dynamic ? (int)std::sqrt(l.size()) : MatrixType::ColsAtCompileTime;
|
||||||
|
mat.setZero(rows, cols);
|
||||||
|
for (size_t i = 0; i < (size_t)l.size(); ++i) {
|
||||||
|
mat(i / cols, i % cols) = l[i].cast<Scalar>();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return mat;
|
||||||
|
});
|
||||||
|
|
||||||
|
// Initialize from py::array
|
||||||
|
m.def(name, [](py::array_t<Scalar, py::array::c_style | py::array::forcecast> arr) -> MatrixType {
|
||||||
|
auto buf = arr.request();
|
||||||
|
MatrixType mat;
|
||||||
|
if constexpr (is_vector) {
|
||||||
|
mat.setZero(buf.size);
|
||||||
|
Scalar* ptr = static_cast<Scalar*>(buf.ptr);
|
||||||
|
for (ssize_t i = 0; i < buf.size; ++i) mat(i) = ptr[i];
|
||||||
|
} else {
|
||||||
|
int rows = buf.shape.size() > 0 ? (int)buf.shape[0] : 1;
|
||||||
|
int cols = buf.shape.size() > 1 ? (int)buf.shape[1] : 1;
|
||||||
|
mat.setZero(rows, cols);
|
||||||
|
Scalar* ptr = static_cast<Scalar*>(buf.ptr);
|
||||||
|
for (int i = 0; i < rows; ++i) {
|
||||||
|
for (int j = 0; j < cols; ++j) {
|
||||||
|
mat(i, j) = ptr[i * cols + j];
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return mat;
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
void init_math(py::module_ &m) {
|
||||||
|
|
||||||
|
// 1. Basic Eigen Types (Vectors and Matrices)
|
||||||
|
bind_eigen_type<Vector1f>(m, "Vector1f");
|
||||||
|
bind_eigen_type<Vector2f>(m, "Vector2f");
|
||||||
|
bind_eigen_type<Vector3f>(m, "Vector3f");
|
||||||
|
bind_eigen_type<Vector4f>(m, "Vector4f");
|
||||||
|
bind_eigen_type<Vector1i>(m, "Vector1i");
|
||||||
|
bind_eigen_type<Vector2i>(m, "Vector2i");
|
||||||
|
bind_eigen_type<Vector3i>(m, "Vector3i");
|
||||||
|
bind_eigen_type<Vector4i>(m, "Vector4i");
|
||||||
|
bind_eigen_type<Vector1d>(m, "Vector1d");
|
||||||
|
bind_eigen_type<Vector2d>(m, "Vector2d");
|
||||||
|
bind_eigen_type<Vector3d>(m, "Vector3d");
|
||||||
|
bind_eigen_type<Vector4d>(m, "Vector4d");
|
||||||
|
|
||||||
|
bind_eigen_type<Matrix2f>(m, "Matrix2f");
|
||||||
|
bind_eigen_type<Matrix3f>(m, "Matrix3f");
|
||||||
|
bind_eigen_type<Matrix4f>(m, "Matrix4f");
|
||||||
|
bind_eigen_type<Matrix2i>(m, "Matrix2i");
|
||||||
|
bind_eigen_type<Matrix3i>(m, "Matrix3i");
|
||||||
|
bind_eigen_type<Matrix4i>(m, "Matrix4i");
|
||||||
|
bind_eigen_type<Matrix2d>(m, "Matrix2d");
|
||||||
|
bind_eigen_type<Matrix3d>(m, "Matrix3d");
|
||||||
|
bind_eigen_type<Matrix4d>(m, "Matrix4d");
|
||||||
|
|
||||||
|
bind_eigen_type<MatrixXi>(m, "MatrixXi");
|
||||||
|
bind_eigen_type<MatrixXf>(m, "MatrixXf");
|
||||||
|
bind_eigen_type<MatrixXd>(m, "MatrixXd");
|
||||||
|
|
||||||
|
// 2. Homogeneous types
|
||||||
|
py::class_<HPoint3f>(m, "HPoint3f")
|
||||||
|
.def(py::init<>())
|
||||||
|
.def(py::init<float, float, float>())
|
||||||
|
.def(py::init<Vector3f &>());
|
||||||
|
py::class_<HVector3f>(m, "HVector3f")
|
||||||
|
.def(py::init<>())
|
||||||
|
.def(py::init<float, float, float>())
|
||||||
|
.def(py::init<Vector3f &>());
|
||||||
|
py::class_<HLine3f>(m, "HLine3f")
|
||||||
|
.def(py::init<>())
|
||||||
|
.def_readwrite("origin", &HLine3f::origin)
|
||||||
|
.def_readwrite("direction", &HLine3f::direction);
|
||||||
|
py::class_<HError3f>(m, "HError3f")
|
||||||
|
.def(py::init<>())
|
||||||
|
.def_readwrite("position_error", &HError3f::position_error)
|
||||||
|
.def_readwrite("direction_error", &HError3f::direction_error);
|
||||||
|
|
||||||
|
// 3. Dynamic Vectors (uLib::Vector)
|
||||||
|
py::bind_vector<uLib::Vector<Scalari>>(m, "Vector_i")
|
||||||
|
.def("MoveToVRAM", &uLib::Vector<Scalari>::MoveToVRAM)
|
||||||
|
.def("MoveToRAM", &uLib::Vector<Scalari>::MoveToRAM);
|
||||||
|
py::bind_vector<uLib::Vector<Scalarui>>(m, "Vector_ui")
|
||||||
|
.def("MoveToVRAM", &uLib::Vector<Scalarui>::MoveToVRAM)
|
||||||
|
.def("MoveToRAM", &uLib::Vector<Scalarui>::MoveToRAM);
|
||||||
|
py::bind_vector<uLib::Vector<Scalarl>>(m, "Vector_l")
|
||||||
|
.def("MoveToVRAM", &uLib::Vector<Scalarl>::MoveToVRAM)
|
||||||
|
.def("MoveToRAM", &uLib::Vector<Scalarl>::MoveToRAM);
|
||||||
|
py::bind_vector<uLib::Vector<Scalarul>>(m, "Vector_ul")
|
||||||
|
.def("MoveToVRAM", &uLib::Vector<Scalarul>::MoveToVRAM)
|
||||||
|
.def("MoveToRAM", &uLib::Vector<Scalarul>::MoveToRAM);
|
||||||
|
py::bind_vector<uLib::Vector<Scalarf>>(m, "Vector_f")
|
||||||
|
.def("MoveToVRAM", &uLib::Vector<Scalarf>::MoveToVRAM)
|
||||||
|
.def("MoveToRAM", &uLib::Vector<Scalarf>::MoveToRAM);
|
||||||
|
py::bind_vector<uLib::Vector<Scalard>>(m, "Vector_d")
|
||||||
|
.def("MoveToVRAM", &uLib::Vector<Scalard>::MoveToVRAM)
|
||||||
|
.def("MoveToRAM", &uLib::Vector<Scalard>::MoveToRAM);
|
||||||
|
|
||||||
|
py::bind_vector<uLib::Vector<Vector3f>>(m, "Vector_Vector3f")
|
||||||
|
.def("MoveToVRAM", &uLib::Vector<Vector3f>::MoveToVRAM)
|
||||||
|
.def("MoveToRAM", &uLib::Vector<Vector3f>::MoveToRAM);
|
||||||
|
py::bind_vector<uLib::Vector<Vector3i>>(m, "Vector_Vector3i")
|
||||||
|
.def("MoveToVRAM", &uLib::Vector<Vector3i>::MoveToVRAM)
|
||||||
|
.def("MoveToRAM", &uLib::Vector<Vector3i>::MoveToRAM);
|
||||||
|
py::bind_vector<uLib::Vector<Vector4f>>(m, "Vector_Vector4f")
|
||||||
|
.def("MoveToVRAM", &uLib::Vector<Vector4f>::MoveToVRAM)
|
||||||
|
.def("MoveToRAM", &uLib::Vector<Vector4f>::MoveToRAM);
|
||||||
|
py::bind_vector<uLib::Vector<Vector4i>>(m, "Vector_Vector4i")
|
||||||
|
.def("MoveToVRAM", &uLib::Vector<Vector4i>::MoveToVRAM)
|
||||||
|
.def("MoveToRAM", &uLib::Vector<Vector4i>::MoveToRAM);
|
||||||
|
py::bind_vector<uLib::Vector<Vector3d>>(m, "Vector_Vector3d")
|
||||||
|
.def("MoveToVRAM", &uLib::Vector<Vector3d>::MoveToVRAM)
|
||||||
|
.def("MoveToRAM", &uLib::Vector<Vector3d>::MoveToRAM);
|
||||||
|
py::bind_vector<uLib::Vector<Vector4d>>(m, "Vector_Vector4d")
|
||||||
|
.def("MoveToVRAM", &uLib::Vector<Vector4d>::MoveToVRAM)
|
||||||
|
.def("MoveToRAM", &uLib::Vector<Vector4d>::MoveToRAM);
|
||||||
|
py::bind_vector<uLib::Vector<Voxel>>(m, "Vector_Voxel")
|
||||||
|
.def("MoveToVRAM", &uLib::Vector<Voxel>::MoveToVRAM)
|
||||||
|
.def("MoveToRAM", &uLib::Vector<Voxel>::MoveToRAM);
|
||||||
|
py::bind_vector<uLib::Vector<VoxRaytracer::RayData::Element>>(m, "Vector_VoxRaytracerRayDataElement")
|
||||||
|
.def("MoveToVRAM", &uLib::Vector<VoxRaytracer::RayData::Element>::MoveToVRAM)
|
||||||
|
.def("MoveToRAM", &uLib::Vector<VoxRaytracer::RayData::Element>::MoveToRAM);
|
||||||
|
|
||||||
|
// 4. Accumulators
|
||||||
|
py::class_<Accumulator_Mean<float>>(m, "Accumulator_Mean_f")
|
||||||
|
.def(py::init<>())
|
||||||
|
.def("AddPass", &Accumulator_Mean<float>::AddPass)
|
||||||
|
.def("__call__", py::overload_cast<const float>(&Accumulator_Mean<float>::operator()))
|
||||||
|
.def("__call__", py::overload_cast<>(&Accumulator_Mean<float>::operator(), py::const_));
|
||||||
|
|
||||||
|
py::class_<Accumulator_Mean<double>>(m, "Accumulator_Mean_d")
|
||||||
|
.def(py::init<>())
|
||||||
|
.def("AddPass", &Accumulator_Mean<double>::AddPass)
|
||||||
|
.def("__call__", py::overload_cast<const double>(&Accumulator_Mean<double>::operator()))
|
||||||
|
.def("__call__", py::overload_cast<>(&Accumulator_Mean<double>::operator(), py::const_));
|
||||||
|
|
||||||
|
py::class_<Accumulator_ABTrim<float>>(m, "Accumulator_ABTrim_f")
|
||||||
|
.def(py::init<>())
|
||||||
|
.def("SetABTrim", &Accumulator_ABTrim<float>::SetABTrim)
|
||||||
|
.def("__iadd__", [](Accumulator_ABTrim<float> &self, float val) { self += val; return &self; })
|
||||||
|
.def("__call__", &Accumulator_ABTrim<float>::operator());
|
||||||
|
|
||||||
|
py::class_<Accumulator_ABTrim<double>>(m, "Accumulator_ABTrim_d")
|
||||||
|
.def(py::init<>())
|
||||||
|
.def("SetABTrim", &Accumulator_ABTrim<double>::SetABTrim)
|
||||||
|
.def("__iadd__", [](Accumulator_ABTrim<double> &self, double val) { self += val; return &self; })
|
||||||
|
.def("__call__", &Accumulator_ABTrim<double>::operator());
|
||||||
|
|
||||||
|
py::class_<Accumulator_ABClip<float>>(m, "Accumulator_ABClip_f")
|
||||||
|
.def(py::init<>())
|
||||||
|
.def("SetABTrim", &Accumulator_ABClip<float>::SetABTrim)
|
||||||
|
.def("__iadd__", [](Accumulator_ABClip<float> &self, float val) { self += val; return &self; })
|
||||||
|
.def("__call__", &Accumulator_ABClip<float>::operator());
|
||||||
|
|
||||||
|
py::class_<Accumulator_ABClip<double>>(m, "Accumulator_ABClip_d")
|
||||||
|
.def(py::init<>())
|
||||||
|
.def("SetABTrim", &Accumulator_ABClip<double>::SetABTrim)
|
||||||
|
.def("__iadd__", [](Accumulator_ABClip<double> &self, double val) { self += val; return &self; })
|
||||||
|
.def("__call__", &Accumulator_ABClip<double>::operator());
|
||||||
|
|
||||||
|
// 5. Core Math Structures
|
||||||
|
py::class_<AffineTransform>(m, "AffineTransform")
|
||||||
|
.def(py::init<>())
|
||||||
|
.def("GetWorldMatrix", &AffineTransform::GetWorldMatrix)
|
||||||
|
.def("SetPosition", &AffineTransform::SetPosition)
|
||||||
|
.def("GetPosition", &AffineTransform::GetPosition)
|
||||||
|
.def("Translate", &AffineTransform::Translate)
|
||||||
|
.def("Scale", &AffineTransform::Scale)
|
||||||
|
.def("SetRotation", &AffineTransform::SetRotation)
|
||||||
|
.def("GetRotation", &AffineTransform::GetRotation)
|
||||||
|
.def("Rotate", py::overload_cast<const Matrix3f &>(&AffineTransform::Rotate))
|
||||||
|
.def("Rotate", py::overload_cast<const Vector3f>(&AffineTransform::Rotate))
|
||||||
|
.def("EulerYZYRotate", &AffineTransform::EulerYZYRotate)
|
||||||
|
.def("FlipAxes", &AffineTransform::FlipAxes);
|
||||||
|
|
||||||
|
py::class_<Geometry, AffineTransform>(m, "Geometry")
|
||||||
|
.def(py::init<>())
|
||||||
|
.def("GetWorldPoint", py::overload_cast<const Vector4f &>(&Geometry::GetWorldPoint, py::const_))
|
||||||
|
.def("GetLocalPoint", py::overload_cast<const Vector4f &>(&Geometry::GetLocalPoint, py::const_));
|
||||||
|
|
||||||
|
py::class_<ContainerBox, AffineTransform>(m, "ContainerBox")
|
||||||
|
.def(py::init<>())
|
||||||
|
.def("SetOrigin", &ContainerBox::SetOrigin)
|
||||||
|
.def("GetOrigin", &ContainerBox::GetOrigin)
|
||||||
|
.def("SetSize", &ContainerBox::SetSize)
|
||||||
|
.def("GetSize", &ContainerBox::GetSize)
|
||||||
|
.def("GetWorldMatrix", &ContainerBox::GetWorldMatrix)
|
||||||
|
.def("GetWorldPoint", py::overload_cast<const Vector4f &>(&ContainerBox::GetWorldPoint, py::const_))
|
||||||
|
.def("GetLocalPoint", py::overload_cast<const Vector4f &>(&ContainerBox::GetLocalPoint, py::const_));
|
||||||
|
|
||||||
|
py::enum_<StructuredData::_Order>(m, "StructuredDataOrder")
|
||||||
|
.value("CustomOrder", StructuredData::CustomOrder)
|
||||||
|
.value("XYZ", StructuredData::XYZ)
|
||||||
|
.value("XZY", StructuredData::XZY)
|
||||||
|
.value("YXZ", StructuredData::YXZ)
|
||||||
|
.value("YZX", StructuredData::YZX)
|
||||||
|
.value("ZXY", StructuredData::ZXY)
|
||||||
|
.value("ZYX", StructuredData::ZYX)
|
||||||
|
.export_values();
|
||||||
|
|
||||||
|
py::class_<StructuredData>(m, "StructuredData")
|
||||||
|
.def(py::init<const Vector3i &>())
|
||||||
|
.def("GetDims", &StructuredData::GetDims)
|
||||||
|
.def("SetDims", &StructuredData::SetDims)
|
||||||
|
.def("GetIncrements", &StructuredData::GetIncrements)
|
||||||
|
.def("SetIncrements", &StructuredData::SetIncrements)
|
||||||
|
.def("SetDataOrder", &StructuredData::SetDataOrder)
|
||||||
|
.def("GetDataOrder", &StructuredData::GetDataOrder)
|
||||||
|
.def("IsInsideGrid", &StructuredData::IsInsideGrid)
|
||||||
|
.def("Map", &StructuredData::Map)
|
||||||
|
.def("UnMap", &StructuredData::UnMap);
|
||||||
|
|
||||||
|
py::class_<StructuredGrid, ContainerBox, StructuredData>(m, "StructuredGrid")
|
||||||
|
.def(py::init<const Vector3i &>())
|
||||||
|
.def("SetSpacing", &StructuredGrid::SetSpacing)
|
||||||
|
.def("GetSpacing", &StructuredGrid::GetSpacing)
|
||||||
|
.def("IsInsideBounds", &StructuredGrid::IsInsideBounds)
|
||||||
|
.def("Find", [](StructuredGrid &self, Vector3f pt) {
|
||||||
|
return self.Find(HPoint3f(pt));
|
||||||
|
});
|
||||||
|
|
||||||
|
py::class_<Structured2DGrid>(m, "Structured2DGrid")
|
||||||
|
.def(py::init<>())
|
||||||
|
.def("SetDims", &Structured2DGrid::SetDims)
|
||||||
|
.def("GetDims", &Structured2DGrid::GetDims)
|
||||||
|
.def("IsInsideGrid", &Structured2DGrid::IsInsideGrid)
|
||||||
|
.def("Map", &Structured2DGrid::Map)
|
||||||
|
.def("UnMap", &Structured2DGrid::UnMap)
|
||||||
|
.def("SetPhysicalSpace", &Structured2DGrid::SetPhysicalSpace)
|
||||||
|
.def("GetSpacing", &Structured2DGrid::GetSpacing)
|
||||||
|
.def("GetOrigin", &Structured2DGrid::GetOrigin)
|
||||||
|
.def("IsInsideBounds", &Structured2DGrid::IsInsideBounds)
|
||||||
|
.def("PhysicsToUnitSpace", &Structured2DGrid::PhysicsToUnitSpace)
|
||||||
|
.def("UnitToPhysicsSpace", &Structured2DGrid::UnitToPhysicsSpace)
|
||||||
|
.def("SetDebug", &Structured2DGrid::SetDebug);
|
||||||
|
|
||||||
|
py::class_<Structured4DGrid>(m, "Structured4DGrid")
|
||||||
|
.def(py::init<>())
|
||||||
|
.def("SetDims", &Structured4DGrid::SetDims)
|
||||||
|
.def("GetDims", &Structured4DGrid::GetDims)
|
||||||
|
.def("IsInsideGrid", &Structured4DGrid::IsInsideGrid)
|
||||||
|
.def("Map", &Structured4DGrid::Map)
|
||||||
|
.def("UnMap", &Structured4DGrid::UnMap)
|
||||||
|
.def("SetPhysicalSpace", &Structured4DGrid::SetPhysicalSpace)
|
||||||
|
.def("GetSpacing", &Structured4DGrid::GetSpacing)
|
||||||
|
.def("GetOrigin", &Structured4DGrid::GetOrigin)
|
||||||
|
.def("IsInsideBounds", &Structured4DGrid::IsInsideBounds)
|
||||||
|
.def("PhysicsToUnitSpace", &Structured4DGrid::PhysicsToUnitSpace)
|
||||||
|
.def("UnitToPhysicsSpace", &Structured4DGrid::UnitToPhysicsSpace)
|
||||||
|
.def("SetDebug", &Structured4DGrid::SetDebug);
|
||||||
|
|
||||||
|
// 6. High-level Structures
|
||||||
|
py::class_<Voxel>(m, "Voxel")
|
||||||
|
.def(py::init<>())
|
||||||
|
.def_readwrite("Value", &Voxel::Value)
|
||||||
|
.def_readwrite("Count", &Voxel::Count);
|
||||||
|
|
||||||
|
py::class_<Abstract::VoxImage, StructuredGrid>(m, "AbstractVoxImage")
|
||||||
|
.def("GetValue", py::overload_cast<const Vector3i &>(&Abstract::VoxImage::GetValue, py::const_))
|
||||||
|
.def("GetValue", py::overload_cast<const int>(&Abstract::VoxImage::GetValue, py::const_))
|
||||||
|
.def("SetValue", py::overload_cast<const Vector3i &, float>(&Abstract::VoxImage::SetValue))
|
||||||
|
.def("SetValue", py::overload_cast<const int, float>(&Abstract::VoxImage::SetValue))
|
||||||
|
.def("ExportToVtk", &Abstract::VoxImage::ExportToVtk)
|
||||||
|
.def("ExportToVti", &Abstract::VoxImage::ExportToVti)
|
||||||
|
.def("ImportFromVtk", &Abstract::VoxImage::ImportFromVtk)
|
||||||
|
.def("ImportFromVti", &Abstract::VoxImage::ImportFromVti);
|
||||||
|
|
||||||
|
py::class_<VoxImage<Voxel>, Abstract::VoxImage>(m, "VoxImage")
|
||||||
|
.def(py::init<>())
|
||||||
|
.def(py::init<const Vector3i &>())
|
||||||
|
.def("Data", &VoxImage<Voxel>::Data, py::return_value_policy::reference_internal)
|
||||||
|
.def("InitVoxels", &VoxImage<Voxel>::InitVoxels)
|
||||||
|
.def("Abs", &VoxImage<Voxel>::Abs)
|
||||||
|
.def("clipImage", py::overload_cast<const Vector3i, const Vector3i>(&VoxImage<Voxel>::clipImage, py::const_))
|
||||||
|
.def("clipImage", py::overload_cast<const HPoint3f, const HPoint3f>(&VoxImage<Voxel>::clipImage, py::const_))
|
||||||
|
.def("clipImage", py::overload_cast<const float>(&VoxImage<Voxel>::clipImage, py::const_))
|
||||||
|
.def("maskImage", py::overload_cast<const HPoint3f, const HPoint3f, float>(&VoxImage<Voxel>::maskImage, py::const_))
|
||||||
|
.def("maskImage", py::overload_cast<const float, float, float>(&VoxImage<Voxel>::maskImage, py::const_), py::arg("threshold"), py::arg("belowValue") = 0, py::arg("aboveValue") = 0)
|
||||||
|
.def("fixVoxels", py::overload_cast<const float, float>(&VoxImage<Voxel>::fixVoxels, py::const_))
|
||||||
|
.def("__getitem__", py::overload_cast<unsigned int>(&VoxImage<Voxel>::operator[]))
|
||||||
|
.def("__getitem__", py::overload_cast<const Vector3i &>(&VoxImage<Voxel>::operator[]));
|
||||||
|
|
||||||
|
py::class_<TriangleMesh>(m, "TriangleMesh")
|
||||||
|
.def(py::init<>())
|
||||||
|
.def("AddPoint", &TriangleMesh::AddPoint)
|
||||||
|
.def("AddTriangle", py::overload_cast<const Vector3i &>(&TriangleMesh::AddTriangle))
|
||||||
|
.def("Points", &TriangleMesh::Points, py::return_value_policy::reference_internal)
|
||||||
|
.def("Triangles", &TriangleMesh::Triangles, py::return_value_policy::reference_internal);
|
||||||
|
|
||||||
|
py::class_<VoxRaytracer::RayData::Element>(m, "VoxRaytracerRayDataElement")
|
||||||
|
.def(py::init<>())
|
||||||
|
.def_readwrite("vox_id", &VoxRaytracer::RayData::Element::vox_id)
|
||||||
|
.def_readwrite("L", &VoxRaytracer::RayData::Element::L);
|
||||||
|
|
||||||
|
py::class_<VoxRaytracer::RayData>(m, "VoxRaytracerRayData")
|
||||||
|
.def(py::init<>())
|
||||||
|
.def("AppendRay", &VoxRaytracer::RayData::AppendRay)
|
||||||
|
.def("Data", py::overload_cast<>(&VoxRaytracer::RayData::Data), py::return_value_policy::reference_internal)
|
||||||
|
.def("Count", &VoxRaytracer::RayData::Count)
|
||||||
|
.def("TotalLength", &VoxRaytracer::RayData::TotalLength)
|
||||||
|
.def("SetCount", &VoxRaytracer::RayData::SetCount)
|
||||||
|
.def("SetTotalLength", &VoxRaytracer::RayData::SetTotalLength);
|
||||||
|
|
||||||
|
py::class_<VoxRaytracer>(m, "VoxRaytracer")
|
||||||
|
.def(py::init<StructuredGrid &>(), py::keep_alive<1, 2>())
|
||||||
|
.def("GetImage", &VoxRaytracer::GetImage, py::return_value_policy::reference_internal)
|
||||||
|
.def("TraceLine", &VoxRaytracer::TraceLine)
|
||||||
|
.def("TraceBetweenPoints", &VoxRaytracer::TraceBetweenPoints);
|
||||||
|
|
||||||
|
}
|
||||||
100
src/Python/math_filters_bindings.cpp
Normal file
100
src/Python/math_filters_bindings.cpp
Normal file
@@ -0,0 +1,100 @@
|
|||||||
|
#include <pybind11/pybind11.h>
|
||||||
|
#include <pybind11/eigen.h>
|
||||||
|
#include <pybind11/stl.h>
|
||||||
|
|
||||||
|
#include "Math/VoxImage.h"
|
||||||
|
#include "Math/VoxImageFilter.h"
|
||||||
|
#include "Math/VoxImageFilterLinear.hpp"
|
||||||
|
#include "Math/VoxImageFilterABTrim.hpp"
|
||||||
|
#include "Math/VoxImageFilterBilateral.hpp"
|
||||||
|
#include "Math/VoxImageFilterThreshold.hpp"
|
||||||
|
#include "Math/VoxImageFilterMedian.hpp"
|
||||||
|
#include "Math/VoxImageFilter2ndStat.hpp"
|
||||||
|
#include "Math/VoxImageFilterCustom.hpp"
|
||||||
|
|
||||||
|
namespace py = pybind11;
|
||||||
|
using namespace uLib;
|
||||||
|
|
||||||
|
template <typename Algorithm>
|
||||||
|
void bind_common_filter(py::class_<Algorithm, Abstract::VoxImageFilter> &cls) {
|
||||||
|
cls.def(py::init<const Vector3i &>())
|
||||||
|
.def("Run", &Algorithm::Run)
|
||||||
|
.def("SetKernelNumericXZY", &Algorithm::SetKernelNumericXZY)
|
||||||
|
.def("GetImage", &Algorithm::GetImage, py::return_value_policy::reference_internal)
|
||||||
|
.def("SetImage", &Algorithm::SetImage);
|
||||||
|
}
|
||||||
|
|
||||||
|
void init_math_filters(py::module_ &m) {
|
||||||
|
|
||||||
|
// Abstract::VoxImageFilter
|
||||||
|
py::class_<Abstract::VoxImageFilter, std::unique_ptr<Abstract::VoxImageFilter, py::nodelete>>(m, "AbstractVoxImageFilter")
|
||||||
|
.def("Run", &Abstract::VoxImageFilter::Run)
|
||||||
|
.def("SetImage", &Abstract::VoxImageFilter::SetImage);
|
||||||
|
|
||||||
|
// Helper macro to define standard bindings for a filter
|
||||||
|
#define BIND_FILTER(ClassName) \
|
||||||
|
{ \
|
||||||
|
auto cls = py::class_<ClassName<Voxel>, Abstract::VoxImageFilter>(m, #ClassName); \
|
||||||
|
bind_common_filter(cls); \
|
||||||
|
}
|
||||||
|
|
||||||
|
// VoxFilterAlgorithmLinear
|
||||||
|
{
|
||||||
|
auto cls = py::class_<VoxFilterAlgorithmLinear<Voxel>, Abstract::VoxImageFilter>(m, "VoxFilterAlgorithmLinear");
|
||||||
|
bind_common_filter(cls);
|
||||||
|
}
|
||||||
|
|
||||||
|
// VoxFilterAlgorithmAbtrim
|
||||||
|
{
|
||||||
|
auto cls = py::class_<VoxFilterAlgorithmAbtrim<Voxel>, Abstract::VoxImageFilter>(m, "VoxFilterAlgorithmAbtrim");
|
||||||
|
bind_common_filter(cls);
|
||||||
|
cls.def("SetABTrim", &VoxFilterAlgorithmAbtrim<Voxel>::SetABTrim);
|
||||||
|
}
|
||||||
|
|
||||||
|
// VoxFilterAlgorithmSPR
|
||||||
|
{
|
||||||
|
auto cls = py::class_<VoxFilterAlgorithmSPR<Voxel>, Abstract::VoxImageFilter>(m, "VoxFilterAlgorithmSPR");
|
||||||
|
bind_common_filter(cls);
|
||||||
|
cls.def("SetABTrim", &VoxFilterAlgorithmSPR<Voxel>::SetABTrim);
|
||||||
|
}
|
||||||
|
|
||||||
|
// VoxFilterAlgorithmBilateral
|
||||||
|
{
|
||||||
|
auto cls = py::class_<VoxFilterAlgorithmBilateral<Voxel>, Abstract::VoxImageFilter>(m, "VoxFilterAlgorithmBilateral");
|
||||||
|
bind_common_filter(cls);
|
||||||
|
cls.def("SetIntensitySigma", &VoxFilterAlgorithmBilateral<Voxel>::SetIntensitySigma);
|
||||||
|
}
|
||||||
|
|
||||||
|
// VoxFilterAlgorithmBilateralTrim
|
||||||
|
{
|
||||||
|
auto cls = py::class_<VoxFilterAlgorithmBilateralTrim<Voxel>, Abstract::VoxImageFilter>(m, "VoxFilterAlgorithmBilateralTrim");
|
||||||
|
bind_common_filter(cls);
|
||||||
|
cls.def("SetIntensitySigma", &VoxFilterAlgorithmBilateralTrim<Voxel>::SetIntensitySigma);
|
||||||
|
cls.def("SetABTrim", &VoxFilterAlgorithmBilateralTrim<Voxel>::SetABTrim);
|
||||||
|
}
|
||||||
|
|
||||||
|
// VoxFilterAlgorithmThreshold
|
||||||
|
{
|
||||||
|
auto cls = py::class_<VoxFilterAlgorithmThreshold<Voxel>, Abstract::VoxImageFilter>(m, "VoxFilterAlgorithmThreshold");
|
||||||
|
bind_common_filter(cls);
|
||||||
|
cls.def("SetThreshold", &VoxFilterAlgorithmThreshold<Voxel>::SetThreshold);
|
||||||
|
}
|
||||||
|
|
||||||
|
// VoxFilterAlgorithmMedian
|
||||||
|
{
|
||||||
|
auto cls = py::class_<VoxFilterAlgorithmMedian<Voxel>, Abstract::VoxImageFilter>(m, "VoxFilterAlgorithmMedian");
|
||||||
|
bind_common_filter(cls);
|
||||||
|
}
|
||||||
|
|
||||||
|
// VoxFilterAlgorithm2ndStat
|
||||||
|
{
|
||||||
|
auto cls = py::class_<VoxFilterAlgorithm2ndStat<Voxel>, Abstract::VoxImageFilter>(m, "VoxFilterAlgorithm2ndStat");
|
||||||
|
bind_common_filter(cls);
|
||||||
|
}
|
||||||
|
|
||||||
|
// VoxFilterAlgorithmCustom (Omit CustomEvaluate since it uses static function ptrs)
|
||||||
|
{
|
||||||
|
auto cls = py::class_<VoxFilterAlgorithmCustom<Voxel>, Abstract::VoxImageFilter>(m, "VoxFilterAlgorithmCustom");
|
||||||
|
bind_common_filter(cls);
|
||||||
|
}
|
||||||
|
}
|
||||||
20
src/Python/module.cpp
Normal file
20
src/Python/module.cpp
Normal file
@@ -0,0 +1,20 @@
|
|||||||
|
#include <pybind11/pybind11.h>
|
||||||
|
|
||||||
|
namespace py = pybind11;
|
||||||
|
|
||||||
|
void init_core(py::module_ &m);
|
||||||
|
void init_math(py::module_ &m);
|
||||||
|
void init_math_filters(py::module_ &m);
|
||||||
|
|
||||||
|
PYBIND11_MODULE(uLib_python, m) {
|
||||||
|
m.doc() = "Python bindings for uLib Core and Math libraries";
|
||||||
|
|
||||||
|
// Core submodule
|
||||||
|
py::module_ core = m.def_submodule("Core", "Core library bindings");
|
||||||
|
init_core(core);
|
||||||
|
|
||||||
|
// Math submodule
|
||||||
|
py::module_ math = m.def_submodule("Math", "Math library bindings");
|
||||||
|
init_math(math);
|
||||||
|
init_math_filters(math);
|
||||||
|
}
|
||||||
33
src/Python/testing/core_pybind_test.py
Normal file
33
src/Python/testing/core_pybind_test.py
Normal file
@@ -0,0 +1,33 @@
|
|||||||
|
import sys
|
||||||
|
import os
|
||||||
|
import unittest
|
||||||
|
import time
|
||||||
|
|
||||||
|
import uLib
|
||||||
|
|
||||||
|
class TestCoreOptions(unittest.TestCase):
|
||||||
|
def test_options(self):
|
||||||
|
opt = uLib.Core.Options("Test Options")
|
||||||
|
|
||||||
|
# Test basic config file parsing
|
||||||
|
with open("test_configuration.ini", "w") as f:
|
||||||
|
f.write("[Section]\n")
|
||||||
|
|
||||||
|
opt.parse_config_file("test_configuration.ini")
|
||||||
|
os.remove("test_configuration.ini")
|
||||||
|
|
||||||
|
class TestCoreObject(unittest.TestCase):
|
||||||
|
def test_object(self):
|
||||||
|
obj = uLib.Core.Object()
|
||||||
|
self.assertIsNotNone(obj)
|
||||||
|
|
||||||
|
class TestCoreTimer(unittest.TestCase):
|
||||||
|
def test_timer(self):
|
||||||
|
timer = uLib.Core.Timer()
|
||||||
|
timer.Start()
|
||||||
|
time.sleep(0.1)
|
||||||
|
val = timer.StopWatch()
|
||||||
|
self.assertGreater(val, 0.09)
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
unittest.main()
|
||||||
151
src/Python/testing/math_filters_test.py
Normal file
151
src/Python/testing/math_filters_test.py
Normal file
@@ -0,0 +1,151 @@
|
|||||||
|
import unittest
|
||||||
|
import numpy as np
|
||||||
|
import os
|
||||||
|
import sys
|
||||||
|
|
||||||
|
# Ensure PYTHONPATH is correct if run from root
|
||||||
|
sys.path.append(os.path.join(os.getcwd(), 'src', 'Python'))
|
||||||
|
|
||||||
|
import uLib
|
||||||
|
|
||||||
|
class TestMathFilters(unittest.TestCase):
|
||||||
|
def test_filter_creation(self):
|
||||||
|
# 1. Linear Filter
|
||||||
|
dims = [10, 10, 10]
|
||||||
|
v_dims = uLib.Math.Vector3i(dims)
|
||||||
|
|
||||||
|
linear_filter = uLib.Math.VoxFilterAlgorithmLinear(v_dims)
|
||||||
|
self.assertIsNotNone(linear_filter)
|
||||||
|
|
||||||
|
# 2. ABTrim Filter
|
||||||
|
abtrim_filter = uLib.Math.VoxFilterAlgorithmAbtrim(v_dims)
|
||||||
|
self.assertIsNotNone(abtrim_filter)
|
||||||
|
abtrim_filter.SetABTrim(1, 1)
|
||||||
|
|
||||||
|
# 3. Bilateral Filter
|
||||||
|
bilat_filter = uLib.Math.VoxFilterAlgorithmBilateral(v_dims)
|
||||||
|
self.assertIsNotNone(bilat_filter)
|
||||||
|
bilat_filter.SetIntensitySigma(0.5)
|
||||||
|
|
||||||
|
# 4. Threshold Filter
|
||||||
|
threshold_filter = uLib.Math.VoxFilterAlgorithmThreshold(v_dims)
|
||||||
|
self.assertIsNotNone(threshold_filter)
|
||||||
|
threshold_filter.SetThreshold(0.5)
|
||||||
|
|
||||||
|
# 5. Median Filter
|
||||||
|
median_filter = uLib.Math.VoxFilterAlgorithmMedian(v_dims)
|
||||||
|
self.assertIsNotNone(median_filter)
|
||||||
|
|
||||||
|
def test_filter_run(self):
|
||||||
|
# Create image
|
||||||
|
dims = [10, 10, 10]
|
||||||
|
vox_img = uLib.Math.VoxImage(dims)
|
||||||
|
for i in range(10*10*10):
|
||||||
|
vox_img.SetValue(i, 1.0)
|
||||||
|
|
||||||
|
# Linear filter
|
||||||
|
linear_filter = uLib.Math.VoxFilterAlgorithmLinear([3, 3, 3])
|
||||||
|
linear_filter.SetImage(vox_img)
|
||||||
|
|
||||||
|
# Set kernel (simple 3x3x3 all ones)
|
||||||
|
# Weights are usually normalized in linear filter logic?
|
||||||
|
# Let's just test it runs.
|
||||||
|
linear_filter.SetKernelNumericXZY([1.0] * 27)
|
||||||
|
|
||||||
|
# Run filter
|
||||||
|
linear_filter.Run()
|
||||||
|
|
||||||
|
# Value should be 1.0 (mean of all 1.0 is 1.0)
|
||||||
|
self.assertAlmostEqual(vox_img.GetValue(0), 1.0)
|
||||||
|
|
||||||
|
def test_filter_run_abtrim(self):
|
||||||
|
# Create image
|
||||||
|
dims = [10, 10, 10]
|
||||||
|
vox_img = uLib.Math.VoxImage(dims)
|
||||||
|
for i in range(10*10*10):
|
||||||
|
vox_img.SetValue(i, 1.0)
|
||||||
|
|
||||||
|
# ABTrim filter
|
||||||
|
abtrim_filter = uLib.Math.VoxFilterAlgorithmAbtrim([3, 3, 3])
|
||||||
|
abtrim_filter.SetImage(vox_img)
|
||||||
|
|
||||||
|
# Set kernel (simple 3x3x3 all ones)
|
||||||
|
# Weights are usually normalized in linear filter logic?
|
||||||
|
# Let's just test it runs.
|
||||||
|
abtrim_filter.SetKernelNumericXZY([1.0] * 27)
|
||||||
|
|
||||||
|
# Run filter
|
||||||
|
abtrim_filter.Run()
|
||||||
|
|
||||||
|
# Value should be 1.0 (mean of all 1.0 is 1.0)
|
||||||
|
self.assertAlmostEqual(vox_img.GetValue(0), 1.0)
|
||||||
|
|
||||||
|
def test_filter_run_bilateral(self):
|
||||||
|
# Create image
|
||||||
|
dims = [10, 10, 10]
|
||||||
|
vox_img = uLib.Math.VoxImage(dims)
|
||||||
|
for i in range(10*10*10):
|
||||||
|
vox_img.SetValue(i, 1.0)
|
||||||
|
|
||||||
|
# Bilateral filter
|
||||||
|
bilat_filter = uLib.Math.VoxFilterAlgorithmBilateral([3, 3, 3])
|
||||||
|
bilat_filter.SetImage(vox_img)
|
||||||
|
|
||||||
|
# Set kernel (simple 3x3x3 all ones)
|
||||||
|
# Weights are usually normalized in linear filter logic?
|
||||||
|
# Let's just test it runs.
|
||||||
|
bilat_filter.SetKernelNumericXZY([1.0] * 27)
|
||||||
|
|
||||||
|
# Run filter
|
||||||
|
bilat_filter.Run()
|
||||||
|
|
||||||
|
# Value should be 1.0 (mean of all 1.0 is 1.0)
|
||||||
|
self.assertAlmostEqual(vox_img.GetValue(0), 1.0)
|
||||||
|
|
||||||
|
def test_filter_run_threshold(self):
|
||||||
|
# Create image
|
||||||
|
dims = [10, 10, 10]
|
||||||
|
vox_img = uLib.Math.VoxImage(dims)
|
||||||
|
for i in range(10*10*10):
|
||||||
|
vox_img.SetValue(i, 1.0)
|
||||||
|
|
||||||
|
# Threshold filter
|
||||||
|
threshold_filter = uLib.Math.VoxFilterAlgorithmThreshold([3, 3, 3])
|
||||||
|
threshold_filter.SetImage(vox_img)
|
||||||
|
|
||||||
|
# Set kernel (simple 3x3x3 all ones)
|
||||||
|
# Weights are usually normalized in linear filter logic?
|
||||||
|
# Let's just test it runs.
|
||||||
|
threshold_filter.SetKernelNumericXZY([1.0] * 27)
|
||||||
|
|
||||||
|
# Run filter
|
||||||
|
threshold_filter.Run()
|
||||||
|
|
||||||
|
# Value should be 1.0 (mean of all 1.0 is 1.0)
|
||||||
|
self.assertAlmostEqual(vox_img.GetValue(0), 1.0)
|
||||||
|
|
||||||
|
def test_filter_run_median(self):
|
||||||
|
# Create image
|
||||||
|
dims = [10, 10, 10]
|
||||||
|
vox_img = uLib.Math.VoxImage(dims)
|
||||||
|
for i in range(10*10*10):
|
||||||
|
vox_img.SetValue(i, 1.0)
|
||||||
|
|
||||||
|
# Median filter
|
||||||
|
median_filter = uLib.Math.VoxFilterAlgorithmMedian([3, 3, 3])
|
||||||
|
median_filter.SetImage(vox_img)
|
||||||
|
|
||||||
|
# Set kernel (simple 3x3x3 all ones)
|
||||||
|
# Weights are usually normalized in linear filter logic?
|
||||||
|
# Let's just test it runs.
|
||||||
|
median_filter.SetKernelNumericXZY([1.0] * 27)
|
||||||
|
|
||||||
|
# Run filter
|
||||||
|
median_filter.Run()
|
||||||
|
|
||||||
|
# Value should be 1.0 (mean of all 1.0 is 1.0)
|
||||||
|
self.assertAlmostEqual(vox_img.GetValue(0), 1.0)
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
unittest.main()
|
||||||
189
src/Python/testing/math_pybind_test.py
Normal file
189
src/Python/testing/math_pybind_test.py
Normal file
@@ -0,0 +1,189 @@
|
|||||||
|
import sys
|
||||||
|
import os
|
||||||
|
import unittest
|
||||||
|
import numpy as np
|
||||||
|
|
||||||
|
import uLib
|
||||||
|
|
||||||
|
def vector4f0(v, target):
|
||||||
|
diff = np.array(v) - np.array(target)
|
||||||
|
diff[3] = 0 # ignoring w
|
||||||
|
return np.all(np.abs(diff) < 0.001)
|
||||||
|
|
||||||
|
|
||||||
|
class TestMathMatrix(unittest.TestCase):
|
||||||
|
|
||||||
|
def test_matrix(self):
|
||||||
|
def check_1234(m2f):
|
||||||
|
self.assertEqual(m2f[0, 0], 1)
|
||||||
|
self.assertEqual(m2f[0, 1], 2)
|
||||||
|
self.assertEqual(m2f[1, 0], 3)
|
||||||
|
self.assertEqual(m2f[1, 1], 4)
|
||||||
|
|
||||||
|
m2f = uLib.Math.Matrix2f()
|
||||||
|
m2f[0, 0] = 1
|
||||||
|
m2f[0, 1] = 2
|
||||||
|
m2f[1, 0] = 3
|
||||||
|
m2f[1, 1] = 4
|
||||||
|
check_1234(m2f)
|
||||||
|
|
||||||
|
m2f = uLib.Math.Matrix2f([1, 2, 3, 4])
|
||||||
|
check_1234(m2f)
|
||||||
|
|
||||||
|
# m2f = uLib.Math.Matrix2f([[1, 2], [3, 4]])
|
||||||
|
# check_1234(m2f)
|
||||||
|
|
||||||
|
m2f = uLib.Math.Matrix2f(np.array([[1, 2], [3, 4]]))
|
||||||
|
check_1234(m2f)
|
||||||
|
|
||||||
|
def test_vector2(self):
|
||||||
|
v2f = uLib.Math.Vector2f()
|
||||||
|
v2f[0] = 1
|
||||||
|
v2f[1] = 2
|
||||||
|
self.assertEqual(v2f[0], 1)
|
||||||
|
self.assertEqual(v2f[1], 2)
|
||||||
|
|
||||||
|
v2f = uLib.Math.Vector2f([1, 2])
|
||||||
|
self.assertEqual(v2f[0], 1)
|
||||||
|
self.assertEqual(v2f[1], 2)
|
||||||
|
|
||||||
|
v2f = uLib.Math.Vector2f(np.array([1, 2]))
|
||||||
|
self.assertEqual(v2f[0], 1)
|
||||||
|
self.assertEqual(v2f[1], 2)
|
||||||
|
|
||||||
|
def test_vector3(self):
|
||||||
|
v3f = uLib.Math.Vector3f()
|
||||||
|
v3f[0] = 1
|
||||||
|
v3f[1] = 2
|
||||||
|
v3f[2] = 3
|
||||||
|
self.assertEqual(v3f[0], 1)
|
||||||
|
self.assertEqual(v3f[1], 2)
|
||||||
|
self.assertEqual(v3f[2], 3)
|
||||||
|
|
||||||
|
v3f = uLib.Math.Vector3f([1, 2, 3])
|
||||||
|
self.assertEqual(v3f[0], 1)
|
||||||
|
self.assertEqual(v3f[1], 2)
|
||||||
|
self.assertEqual(v3f[2], 3)
|
||||||
|
|
||||||
|
v3f = uLib.Math.Vector3f(np.array([1, 2, 3]))
|
||||||
|
self.assertEqual(v3f[0], 1)
|
||||||
|
self.assertEqual(v3f[1], 2)
|
||||||
|
self.assertEqual(v3f[2], 3)
|
||||||
|
|
||||||
|
|
||||||
|
class TestMathGeometry(unittest.TestCase):
|
||||||
|
def test_geometry(self):
|
||||||
|
Geo = uLib.Math.Geometry()
|
||||||
|
|
||||||
|
Geo.SetPosition([1, 1, 1])
|
||||||
|
|
||||||
|
pt = Geo.GetLocalPoint([2, 3, 2, 1])
|
||||||
|
wp = Geo.GetWorldPoint(pt)
|
||||||
|
|
||||||
|
self.assertTrue(vector4f0(wp, [2, 3, 2, 1]))
|
||||||
|
|
||||||
|
Geo.Scale([2, 2, 2])
|
||||||
|
wp = Geo.GetWorldPoint([1, 1, 1, 1])
|
||||||
|
self.assertTrue(vector4f0(wp, [3, 3, 3, 1]))
|
||||||
|
|
||||||
|
class TestMathContainerBox(unittest.TestCase):
|
||||||
|
def test_container_box_local(self):
|
||||||
|
Cnt = uLib.Math.ContainerBox()
|
||||||
|
Cnt.SetOrigin([-1, -1, -1])
|
||||||
|
Cnt.SetSize([2, 2, 2])
|
||||||
|
|
||||||
|
size = Cnt.GetSize()
|
||||||
|
self.assertTrue(np.allclose(size, [2, 2, 2]))
|
||||||
|
|
||||||
|
def test_container_box_global(self):
|
||||||
|
Box = uLib.Math.ContainerBox()
|
||||||
|
Box.SetPosition([1, 1, 1])
|
||||||
|
Box.SetSize([2, 2, 2])
|
||||||
|
|
||||||
|
pt = Box.GetLocalPoint([2, 3, 2, 1])
|
||||||
|
wp = Box.GetWorldPoint(pt)
|
||||||
|
self.assertTrue(vector4f0(wp, [2, 3, 2, 1]))
|
||||||
|
|
||||||
|
class TestMathStructuredGrid(unittest.TestCase):
|
||||||
|
def test_structured_grid(self):
|
||||||
|
grid = uLib.Math.StructuredGrid([10, 10, 10])
|
||||||
|
grid.SetSpacing([1, 1, 1])
|
||||||
|
|
||||||
|
spacing = grid.GetSpacing()
|
||||||
|
self.assertTrue(np.allclose(spacing, [1, 1, 1]))
|
||||||
|
|
||||||
|
class TestMathAccumulator(unittest.TestCase):
|
||||||
|
def test_accumulator_mean(self):
|
||||||
|
acc = uLib.Math.Accumulator_Mean_f()
|
||||||
|
acc(10.0)
|
||||||
|
acc(20.0)
|
||||||
|
self.assertAlmostEqual(acc(), 15.0)
|
||||||
|
|
||||||
|
class TestMathNewTypes(unittest.TestCase):
|
||||||
|
def test_eigen_vectors(self):
|
||||||
|
v1f = uLib.Math.Vector1f()
|
||||||
|
v3d = uLib.Math.Vector3d()
|
||||||
|
m4f = uLib.Math.Matrix4f()
|
||||||
|
self.assertIsNotNone(v1f)
|
||||||
|
self.assertIsNotNone(v3d)
|
||||||
|
self.assertIsNotNone(m4f)
|
||||||
|
|
||||||
|
def test_ulib_vectors(self):
|
||||||
|
vi = uLib.Math.Vector_i()
|
||||||
|
vi.append(1)
|
||||||
|
vi.append(2)
|
||||||
|
self.assertEqual(len(vi), 2)
|
||||||
|
self.assertEqual(vi[0], 1)
|
||||||
|
self.assertEqual(vi[1], 2)
|
||||||
|
|
||||||
|
vf = uLib.Math.Vector_f()
|
||||||
|
vf.append(1.5)
|
||||||
|
self.assertAlmostEqual(vf[0], 1.5)
|
||||||
|
|
||||||
|
def test_homogeneous(self):
|
||||||
|
p = uLib.Math.HPoint3f(1.0, 2.0, 3.0)
|
||||||
|
v = uLib.Math.HVector3f(0.0, 1.0, 0.0)
|
||||||
|
self.assertIsNotNone(p)
|
||||||
|
self.assertIsNotNone(v)
|
||||||
|
|
||||||
|
def test_vox_image(self):
|
||||||
|
img = uLib.Math.VoxImage([2, 2, 2])
|
||||||
|
self.assertEqual(img.GetDims()[0], 2)
|
||||||
|
img.SetValue([0, 0, 0], 10.5)
|
||||||
|
# Note: GetValue returns float, and there might be internal scaling (1.E-6 observed in code)
|
||||||
|
# Actually in VoxImage.h: GetValue(id) returns At(id).Value
|
||||||
|
# SetValue(id, value) sets At(id).Value = value
|
||||||
|
self.assertAlmostEqual(img.GetValue([0, 0, 0]), 10.5)
|
||||||
|
|
||||||
|
class TestMathVoxRaytracer(unittest.TestCase):
|
||||||
|
def test_raytracer(self):
|
||||||
|
grid = uLib.Math.StructuredGrid([10, 10, 10])
|
||||||
|
grid.SetSpacing([1, 1, 1])
|
||||||
|
grid.SetOrigin([0, 0, 0])
|
||||||
|
|
||||||
|
rt = uLib.Math.VoxRaytracer(grid)
|
||||||
|
self.assertIsNotNone(rt)
|
||||||
|
|
||||||
|
# Test TraceBetweenPoints
|
||||||
|
p1 = np.array([0.5, 0.5, -1.0, 1.0], dtype=np.float32)
|
||||||
|
p2 = np.array([0.5, 0.5, 11.0, 1.0], dtype=np.float32)
|
||||||
|
data = rt.TraceBetweenPoints(p1, p2)
|
||||||
|
|
||||||
|
self.assertGreater(data.Count(), 0)
|
||||||
|
self.assertAlmostEqual(data.TotalLength(), 10.0)
|
||||||
|
|
||||||
|
# Check elements
|
||||||
|
elements = data.Data()
|
||||||
|
for i in range(data.Count()):
|
||||||
|
self.assertGreaterEqual(elements[i].vox_id, 0)
|
||||||
|
self.assertGreater(elements[i].L, 0)
|
||||||
|
|
||||||
|
def test_ray_data(self):
|
||||||
|
data = uLib.Math.VoxRaytracerRayData()
|
||||||
|
data.SetCount(10)
|
||||||
|
data.SetTotalLength(5.5)
|
||||||
|
self.assertEqual(data.Count(), 10)
|
||||||
|
self.assertAlmostEqual(data.TotalLength(), 5.5)
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
unittest.main()
|
||||||
46
src/Python/testing/pybind_test.py
Normal file
46
src/Python/testing/pybind_test.py
Normal file
@@ -0,0 +1,46 @@
|
|||||||
|
import sys
|
||||||
|
import os
|
||||||
|
|
||||||
|
import uLib
|
||||||
|
|
||||||
|
def test_core():
|
||||||
|
print("Testing Core module...")
|
||||||
|
obj = uLib.Core.Object()
|
||||||
|
print("Core Object created:", obj)
|
||||||
|
|
||||||
|
timer = uLib.Core.Timer()
|
||||||
|
timer.Start()
|
||||||
|
print("Core Timer started")
|
||||||
|
|
||||||
|
options = uLib.Core.Options("Test Options")
|
||||||
|
print("Core Options created:", options)
|
||||||
|
|
||||||
|
def test_math():
|
||||||
|
print("Testing Math module...")
|
||||||
|
|
||||||
|
# Test AffineTransform
|
||||||
|
transform = uLib.Math.AffineTransform()
|
||||||
|
print("AffineTransform created")
|
||||||
|
|
||||||
|
# Test Geometry
|
||||||
|
geom = uLib.Math.Geometry()
|
||||||
|
print("Geometry created")
|
||||||
|
|
||||||
|
# Test StructuredData
|
||||||
|
data = uLib.Math.StructuredData([10, 10, 10])
|
||||||
|
print("StructuredData created with dims:", data.GetDims())
|
||||||
|
|
||||||
|
# Test Structured2DGrid
|
||||||
|
grid2d = uLib.Math.Structured2DGrid()
|
||||||
|
grid2d.SetDims([100, 100])
|
||||||
|
print("Structured2DGrid created with dims:", grid2d.GetDims())
|
||||||
|
|
||||||
|
# Test TriangleMesh
|
||||||
|
mesh = uLib.Math.TriangleMesh()
|
||||||
|
print("TriangleMesh created")
|
||||||
|
|
||||||
|
print("All tests passed successfully!")
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
test_core()
|
||||||
|
test_math()
|
||||||
7
src/Python/uLib/__init__.py
Normal file
7
src/Python/uLib/__init__.py
Normal file
@@ -0,0 +1,7 @@
|
|||||||
|
try:
|
||||||
|
from .uLib_python import Core, Math
|
||||||
|
except ImportError:
|
||||||
|
# Handle cases where the binary extension is not yet built
|
||||||
|
pass
|
||||||
|
|
||||||
|
__all__ = ["Core", "Math"]
|
||||||
@@ -62,4 +62,7 @@ install(TARGETS ${libname}
|
|||||||
|
|
||||||
install(FILES ${HEADERS} DESTINATION ${INSTALL_INC_DIR}/Root)
|
install(FILES ${HEADERS} DESTINATION ${INSTALL_INC_DIR}/Root)
|
||||||
|
|
||||||
|
if(BUILD_TESTING)
|
||||||
|
include(uLibTargetMacros)
|
||||||
|
add_subdirectory(testing)
|
||||||
|
endif()
|
||||||
|
|||||||
@@ -1,16 +1,15 @@
|
|||||||
# TESTS
|
# TESTS
|
||||||
set( TESTS
|
set( TESTS
|
||||||
RootDebugTest
|
# RootDebugTest
|
||||||
muBlastMCTrackTest
|
# muBlastMCTrackTest
|
||||||
)
|
)
|
||||||
|
|
||||||
set(LIBRARIES
|
set(LIBRARIES
|
||||||
${PACKAGE_LIBPREFIX}Core
|
${PACKAGE_LIBPREFIX}Core
|
||||||
${PACKAGE_LIBPREFIX}Math
|
${PACKAGE_LIBPREFIX}Math
|
||||||
${PACKAGE_LIBPREFIX}Root
|
${PACKAGE_LIBPREFIX}Root
|
||||||
${Boost_SERIALIZATION_LIBRARY}
|
Boost::serialization
|
||||||
${Boost_SIGNALS_LIBRARY}
|
Boost::program_options
|
||||||
${Boost_PROGRAM_OPTIONS_LIBRARY}
|
|
||||||
${ROOT_LIBRARIES}
|
${ROOT_LIBRARIES}
|
||||||
)
|
)
|
||||||
uLib_add_tests(${uLib-module})
|
uLib_add_tests(Root)
|
||||||
|
|||||||
@@ -19,6 +19,11 @@ set(LIBRARIES Eigen3::Eigen
|
|||||||
${VTK_LIBRARIES}
|
${VTK_LIBRARIES}
|
||||||
${PACKAGE_LIBPREFIX}Math)
|
${PACKAGE_LIBPREFIX}Math)
|
||||||
|
|
||||||
|
if(USE_CUDA)
|
||||||
|
find_package(CUDAToolkit REQUIRED)
|
||||||
|
list(APPEND LIBRARIES CUDA::cudart)
|
||||||
|
endif()
|
||||||
|
|
||||||
set(libname ${PACKAGE_LIBPREFIX}Vtk)
|
set(libname ${PACKAGE_LIBPREFIX}Vtk)
|
||||||
set(ULIB_SHARED_LIBRARIES ${ULIB_SHARED_LIBRARIES} ${libname} PARENT_SCOPE)
|
set(ULIB_SHARED_LIBRARIES ${ULIB_SHARED_LIBRARIES} ${libname} PARENT_SCOPE)
|
||||||
set(ULIB_SELECTED_MODULES ${ULIB_SELECTED_MODULES} Vtk PARENT_SCOPE)
|
set(ULIB_SELECTED_MODULES ${ULIB_SELECTED_MODULES} Vtk PARENT_SCOPE)
|
||||||
@@ -36,3 +41,7 @@ install(TARGETS ${libname}
|
|||||||
|
|
||||||
install(FILES ${HEADERS} DESTINATION ${INSTALL_INC_DIR}/Vtk)
|
install(FILES ${HEADERS} DESTINATION ${INSTALL_INC_DIR}/Vtk)
|
||||||
|
|
||||||
|
if(BUILD_TESTING)
|
||||||
|
include(uLibTargetMacros)
|
||||||
|
add_subdirectory(testing)
|
||||||
|
endif()
|
||||||
|
|||||||
@@ -5,7 +5,7 @@ set( TESTS
|
|||||||
vtkMuonScatter
|
vtkMuonScatter
|
||||||
vtkStructuredGridTest
|
vtkStructuredGridTest
|
||||||
vtkVoxRaytracerTest
|
vtkVoxRaytracerTest
|
||||||
vtkVoxImageTest
|
# vtkVoxImageTest
|
||||||
# vtkTriangleMeshTest
|
# vtkTriangleMeshTest
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -14,5 +14,5 @@ set(LIBRARIES
|
|||||||
${PACKAGE_LIBPREFIX}Vtk
|
${PACKAGE_LIBPREFIX}Vtk
|
||||||
)
|
)
|
||||||
|
|
||||||
include(${VTK_USE_FILE})
|
# include(${VTK_USE_FILE})
|
||||||
uLib_add_tests(${uLib-module})
|
uLib_add_tests(Vtk)
|
||||||
|
|||||||
@@ -23,21 +23,17 @@
|
|||||||
|
|
||||||
//////////////////////////////////////////////////////////////////////////////*/
|
//////////////////////////////////////////////////////////////////////////////*/
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
#ifdef HAVE_CONFIG_H
|
#ifdef HAVE_CONFIG_H
|
||||||
#include "config.h"
|
#include "config.h"
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
|
|
||||||
#include "vtkVoxRaytracerRepresentation.h"
|
#include "vtkVoxRaytracerRepresentation.h"
|
||||||
|
|
||||||
#include "Math/VoxRaytracer.h"
|
#include "Math/VoxRaytracer.h"
|
||||||
|
|
||||||
//#include "vtkMuonEvent.h"
|
// #include "vtkMuonEvent.h"
|
||||||
#include "vtkMuonScatter.h"
|
#include "vtkMuonScatter.h"
|
||||||
|
|
||||||
|
|
||||||
namespace uLib {
|
namespace uLib {
|
||||||
namespace Vtk {
|
namespace Vtk {
|
||||||
|
|
||||||
@@ -45,345 +41,297 @@ namespace Vtk {
|
|||||||
////// VOX RAYTRACER REPRESENTATION ///////////////////////////////////////////
|
////// VOX RAYTRACER REPRESENTATION ///////////////////////////////////////////
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
|
|
||||||
|
vtkVoxRaytracerRepresentation::vtkVoxRaytracerRepresentation(Content &content)
|
||||||
|
: m_Content(&content), m_Assembly(vtkAssembly::New()),
|
||||||
|
m_Sphere1(vtkSphereSource::New()), m_Sphere2(vtkSphereSource::New()),
|
||||||
|
m_Line1(vtkLineSource::New()), m_Line2(vtkLineSource::New()),
|
||||||
|
m_Line3(vtkLineSource::New()), m_RayLine(vtkAppendPolyData::New()),
|
||||||
|
m_RayLineActor(vtkActor::New()),
|
||||||
|
m_RayRepresentation(vtkAppendPolyData::New()),
|
||||||
|
m_RayRepresentationActor(vtkActor::New()),
|
||||||
|
m_Transform(vtkTransform::New()) {
|
||||||
|
default_radius = content.GetImage()->GetSpacing()(0) / 4;
|
||||||
|
m_Sphere1->SetRadius(default_radius);
|
||||||
|
m_Sphere2->SetRadius(default_radius);
|
||||||
|
m_SelectedElement = m_RayLine;
|
||||||
|
|
||||||
|
InstallPipe();
|
||||||
|
}
|
||||||
|
|
||||||
vtkVoxRaytracerRepresentation::vtkVoxRaytracerRepresentation(Content &content) :
|
vtkVoxRaytracerRepresentation::~vtkVoxRaytracerRepresentation() {
|
||||||
m_Content(&content),
|
m_Assembly->Delete();
|
||||||
m_Assembly(vtkAssembly::New()),
|
m_RayLine->Delete();
|
||||||
m_Sphere1(vtkSphereSource::New()),
|
m_RayLineActor->Delete();
|
||||||
m_Sphere2(vtkSphereSource::New()),
|
m_RayRepresentationActor->Delete();
|
||||||
m_Line1(vtkLineSource::New()),
|
m_Transform->Delete();
|
||||||
m_Line2(vtkLineSource::New()),
|
}
|
||||||
m_Line3(vtkLineSource::New()),
|
|
||||||
m_RayLine(vtkAppendPolyData::New()),
|
VoxRaytracer *vtkVoxRaytracerRepresentation::GetRaytracerAlgorithm() {
|
||||||
m_RayLineActor(vtkActor::New()),
|
return m_Content;
|
||||||
m_RayRepresentation(vtkAppendPolyData::New()),
|
}
|
||||||
m_RayRepresentationActor(vtkActor::New()),
|
|
||||||
m_Transform(vtkTransform::New())
|
vtkProp *vtkVoxRaytracerRepresentation::GetProp() { return m_Assembly; }
|
||||||
{
|
|
||||||
default_radius = content.GetImage()->GetSpacing()(0)/4;
|
vtkPolyData *vtkVoxRaytracerRepresentation::GetPolyData() const {
|
||||||
m_Sphere1->SetRadius(default_radius);
|
std::cout << "get Raytracer polydata\n";
|
||||||
m_Sphere2->SetRadius(default_radius);
|
m_SelectedElement->Update();
|
||||||
|
return m_SelectedElement->GetOutput();
|
||||||
|
}
|
||||||
|
|
||||||
|
void vtkVoxRaytracerRepresentation::SetRepresentationElements(
|
||||||
|
vtkVoxRaytracerRepresentation::RepresentationElements el) {
|
||||||
|
switch (el) {
|
||||||
|
case Vtk::vtkVoxRaytracerRepresentation::RayElements:
|
||||||
m_SelectedElement = m_RayLine;
|
m_SelectedElement = m_RayLine;
|
||||||
|
break;
|
||||||
InstallPipe();
|
case Vtk::vtkVoxRaytracerRepresentation::VoxelsElements:
|
||||||
|
m_SelectedElement = m_RayRepresentation;
|
||||||
|
break;
|
||||||
|
default:
|
||||||
|
m_SelectedElement = m_RayLine;
|
||||||
|
break;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
vtkVoxRaytracerRepresentation::~vtkVoxRaytracerRepresentation()
|
void vtkVoxRaytracerRepresentation::SetMuon(MuonScatter &muon) {
|
||||||
{
|
HPoint3f pt1, pt2, src;
|
||||||
m_Assembly->Delete();
|
src = muon.LineIn().origin;
|
||||||
m_RayLine->Delete();
|
m_Content->GetEntryPoint(muon.LineIn(), pt1);
|
||||||
m_RayLineActor->Delete();
|
m_Sphere1->SetCenter(pt1(0), pt1(1), pt1(2));
|
||||||
m_RayRepresentationActor->Delete();
|
m_Line1->SetPoint1(src(0), src(1), src(2));
|
||||||
m_Transform->Delete();
|
m_Line1->SetPoint2(pt1(0), pt1(1), pt1(2));
|
||||||
|
|
||||||
|
HLine3f line_out = muon.LineOut();
|
||||||
|
src = line_out.origin;
|
||||||
|
float *direction = line_out.direction.data();
|
||||||
|
for (int i = 0; i < 3; ++i)
|
||||||
|
direction[i] *= -1;
|
||||||
|
m_Content->GetEntryPoint(line_out, pt2);
|
||||||
|
m_Sphere2->SetCenter(pt2(0), pt2(1), pt2(2));
|
||||||
|
m_Line2->SetPoint1(src(0), src(1), src(2));
|
||||||
|
m_Line2->SetPoint2(pt2(0), pt2(1), pt2(2));
|
||||||
|
|
||||||
|
m_Line3->SetPoint1(pt1(0), pt1(1), pt1(2));
|
||||||
|
m_Line3->SetPoint2(pt2(0), pt2(1), pt2(2));
|
||||||
|
|
||||||
|
// Create a vtkPoints object and store the points in it
|
||||||
|
vtkSmartPointer<vtkPoints> points = vtkSmartPointer<vtkPoints>::New();
|
||||||
|
points->InsertNextPoint(pt1(0), pt1(1), pt1(2));
|
||||||
|
points->InsertNextPoint(pt2(0), pt2(1), pt2(2));
|
||||||
|
|
||||||
|
// Create a cell array to store the lines in and add the lines to it
|
||||||
|
vtkSmartPointer<vtkCellArray> lines = vtkSmartPointer<vtkCellArray>::New();
|
||||||
|
|
||||||
|
vtkSmartPointer<vtkLine> line = vtkSmartPointer<vtkLine>::New();
|
||||||
|
line->GetPointIds()->SetId(0, 0);
|
||||||
|
line->GetPointIds()->SetId(1, 1);
|
||||||
|
lines->InsertNextCell(line);
|
||||||
|
|
||||||
|
// Create a polydata to store everything in
|
||||||
|
vtkSmartPointer<vtkPolyData> linesPolyData =
|
||||||
|
vtkSmartPointer<vtkPolyData>::New();
|
||||||
|
|
||||||
|
// Add the points to the dataset
|
||||||
|
linesPolyData->SetPoints(points);
|
||||||
|
|
||||||
|
// Add the lines to the dataset
|
||||||
|
linesPolyData->SetLines(lines);
|
||||||
|
|
||||||
|
m_RayLine->RemoveAllInputs();
|
||||||
|
#if VTK_MAJOR_VERSION <= 5
|
||||||
|
m_RayLine->AddInputConnection(linesPolyData->GetProducerPort());
|
||||||
|
#endif
|
||||||
|
m_RayLine->AddInputConnection(m_Line1->GetOutputPort());
|
||||||
|
m_RayLine->AddInputConnection(m_Sphere1->GetOutputPort());
|
||||||
|
m_RayLine->AddInputConnection(m_Line2->GetOutputPort());
|
||||||
|
m_RayLine->AddInputConnection(m_Sphere2->GetOutputPort());
|
||||||
|
|
||||||
|
vtkSmartPointer<vtkMatrix4x4> vmat = vtkSmartPointer<vtkMatrix4x4>::New();
|
||||||
|
Matrix4f mat = m_Content->GetImage()->GetWorldMatrix();
|
||||||
|
for (int i = 0; i < 4; ++i)
|
||||||
|
for (int j = 0; j < 4; ++j)
|
||||||
|
vmat->SetElement(i, j, mat(i, j));
|
||||||
|
m_Transform->SetMatrix(vmat);
|
||||||
|
|
||||||
|
this->SetRay(pt1, pt2);
|
||||||
}
|
}
|
||||||
|
|
||||||
VoxRaytracer *vtkVoxRaytracerRepresentation::GetRaytracerAlgorithm()
|
void vtkVoxRaytracerRepresentation::SetMuon(MuonScatter &muon, HPoint3f poca) {
|
||||||
{
|
HPoint3f pt1, pt2, src;
|
||||||
return m_Content;
|
src = muon.LineIn().origin;
|
||||||
|
m_Content->GetEntryPoint(muon.LineIn(), pt1);
|
||||||
|
m_Sphere1->SetCenter(pt1(0), pt1(1), pt1(2));
|
||||||
|
m_Line1->SetPoint1(src(0), src(1), src(2));
|
||||||
|
m_Line1->SetPoint2(pt1(0), pt1(1), pt1(2));
|
||||||
|
|
||||||
|
HLine3f line_out = muon.LineOut();
|
||||||
|
src = line_out.origin;
|
||||||
|
float *direction = line_out.direction.data();
|
||||||
|
for (int i = 0; i < 3; ++i)
|
||||||
|
direction[i] *= -1;
|
||||||
|
m_Content->GetEntryPoint(line_out, pt2);
|
||||||
|
m_Sphere2->SetCenter(pt2(0), pt2(1), pt2(2));
|
||||||
|
m_Line2->SetPoint1(src(0), src(1), src(2));
|
||||||
|
m_Line2->SetPoint2(pt2(0), pt2(1), pt2(2));
|
||||||
|
|
||||||
|
m_Line3->SetPoint1(pt1(0), pt1(1), pt1(2));
|
||||||
|
m_Line3->SetPoint2(pt2(0), pt2(1), pt2(2));
|
||||||
|
|
||||||
|
// Create a vtkPoints object and store the points in it
|
||||||
|
vtkSmartPointer<vtkPoints> points = vtkSmartPointer<vtkPoints>::New();
|
||||||
|
points->InsertNextPoint(pt1(0), pt1(1), pt1(2));
|
||||||
|
points->InsertNextPoint(poca(0), poca(1), poca(2));
|
||||||
|
points->InsertNextPoint(pt2(0), pt2(1), pt2(2));
|
||||||
|
|
||||||
|
// Create a cell array to store the lines in and add the lines to it
|
||||||
|
vtkSmartPointer<vtkCellArray> lines = vtkSmartPointer<vtkCellArray>::New();
|
||||||
|
|
||||||
|
for (unsigned int i = 0; i < 2; i++) {
|
||||||
|
vtkSmartPointer<vtkLine> line = vtkSmartPointer<vtkLine>::New();
|
||||||
|
line->GetPointIds()->SetId(0, i);
|
||||||
|
line->GetPointIds()->SetId(1, i + 1);
|
||||||
|
lines->InsertNextCell(line);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Create a polydata to store everything in
|
||||||
|
vtkSmartPointer<vtkPolyData> linesPolyData =
|
||||||
|
vtkSmartPointer<vtkPolyData>::New();
|
||||||
|
|
||||||
|
// Add the points to the dataset
|
||||||
|
linesPolyData->SetPoints(points);
|
||||||
|
|
||||||
|
// Add the lines to the dataset
|
||||||
|
linesPolyData->SetLines(lines);
|
||||||
|
|
||||||
|
m_RayLine->RemoveAllInputs();
|
||||||
|
#if VTK_MAJOR_VERSION <= 5
|
||||||
|
m_RayLine->AddInputConnection(linesPolyData->GetProducerPort());
|
||||||
|
#endif
|
||||||
|
m_RayLine->AddInputConnection(m_Line1->GetOutputPort());
|
||||||
|
m_RayLine->AddInputConnection(m_Sphere1->GetOutputPort());
|
||||||
|
m_RayLine->AddInputConnection(m_Line2->GetOutputPort());
|
||||||
|
m_RayLine->AddInputConnection(m_Sphere2->GetOutputPort());
|
||||||
|
|
||||||
|
vtkSmartPointer<vtkSphereSource> poca_sphere =
|
||||||
|
vtkSmartPointer<vtkSphereSource>::New();
|
||||||
|
poca_sphere->SetRadius(default_radius);
|
||||||
|
poca_sphere->SetCenter(poca(0), poca(1), poca(2));
|
||||||
|
m_RayLine->AddInputConnection(poca_sphere->GetOutputPort());
|
||||||
|
|
||||||
|
vtkSmartPointer<vtkMatrix4x4> vmat = vtkSmartPointer<vtkMatrix4x4>::New();
|
||||||
|
Matrix4f mat = m_Content->GetImage()->GetWorldMatrix();
|
||||||
|
for (int i = 0; i < 4; ++i)
|
||||||
|
for (int j = 0; j < 4; ++j)
|
||||||
|
vmat->SetElement(i, j, mat(i, j));
|
||||||
|
m_Transform->SetMatrix(vmat);
|
||||||
|
|
||||||
|
if (m_Content->GetImage()->IsInsideBounds(poca))
|
||||||
|
this->SetRay(pt1, poca, pt2);
|
||||||
|
else
|
||||||
|
this->SetRay(pt1, pt2);
|
||||||
}
|
}
|
||||||
|
|
||||||
vtkProp *vtkVoxRaytracerRepresentation::GetProp()
|
void vtkVoxRaytracerRepresentation::SetMuon(vtkMuonScatter &muon) {
|
||||||
{
|
HPoint3f poca = muon.GetPocaPoint();
|
||||||
return m_Assembly;
|
MuonScatter &mu = muon.GetContent();
|
||||||
|
this->SetMuon(mu, poca);
|
||||||
}
|
}
|
||||||
|
|
||||||
vtkPolyData *vtkVoxRaytracerRepresentation::GetPolyData() const
|
VoxRaytracer::RayData vtkVoxRaytracerRepresentation::GetRay() { return m_Ray; }
|
||||||
{
|
|
||||||
std::cout << "get Raytracer polydata\n";
|
void vtkVoxRaytracerRepresentation::SetRay(HPoint3f in, HPoint3f out) {
|
||||||
m_SelectedElement->Update();
|
m_Ray = m_Content->TraceBetweenPoints(in, out);
|
||||||
return m_SelectedElement->GetOutput();
|
this->SetRay(&m_Ray);
|
||||||
}
|
}
|
||||||
|
|
||||||
void vtkVoxRaytracerRepresentation::SetRepresentationElements(vtkVoxRaytracerRepresentation::RepresentationElements el)
|
void vtkVoxRaytracerRepresentation::SetRay(HPoint3f in, HPoint3f mid,
|
||||||
{
|
HPoint3f out) {
|
||||||
switch(el) {
|
m_Ray = m_Content->TraceBetweenPoints(in, mid);
|
||||||
case Vtk::vtkVoxRaytracerRepresentation::RayElements:
|
m_Ray.AppendRay(m_Content->TraceBetweenPoints(mid, out));
|
||||||
m_SelectedElement = m_RayLine;
|
this->SetRay(&m_Ray);
|
||||||
break;
|
|
||||||
case Vtk::vtkVoxRaytracerRepresentation::VoxelsElements:
|
|
||||||
m_SelectedElement = m_RayRepresentation;
|
|
||||||
break;
|
|
||||||
default:
|
|
||||||
m_SelectedElement = m_RayLine;
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
void vtkVoxRaytracerRepresentation::SetRay(VoxRaytracer::RayData *ray) {
|
||||||
|
vtkAppendPolyData *appender = m_RayRepresentation;
|
||||||
|
appender->RemoveAllInputs();
|
||||||
|
|
||||||
void vtkVoxRaytracerRepresentation::SetMuon(MuonScatter &muon)
|
for (size_t i = 0; i < ray->Count(); ++i) {
|
||||||
{
|
int id = ray->Data().at(i).vox_id;
|
||||||
HPoint3f pt1,pt2,src;
|
Vector3i idv = m_Content->GetImage()->UnMap(id);
|
||||||
src = muon.LineIn().origin;
|
vtkSmartPointer<vtkCubeSource> cube = vtkSmartPointer<vtkCubeSource>::New();
|
||||||
m_Content->GetEntryPoint(muon.LineIn(),pt1);
|
cube->SetBounds(idv(0), idv(0) + 1, idv(1), idv(1) + 1, idv(2), idv(2) + 1);
|
||||||
m_Sphere1->SetCenter(pt1(0),pt1(1),pt1(2));
|
cube->Update();
|
||||||
m_Line1->SetPoint1(src(0),src(1),src(2));
|
#if VTK_MAJOR_VERSION <= 5
|
||||||
m_Line1->SetPoint2(pt1(0),pt1(1),pt1(2));
|
appender->AddInput(cube->GetOutput());
|
||||||
|
#endif
|
||||||
HLine3f line_out = muon.LineOut();
|
appender->Update();
|
||||||
src = line_out.origin;
|
}
|
||||||
float *direction = line_out.direction.data();
|
|
||||||
for(int i=0;i<3;++i) direction[i] *= -1;
|
|
||||||
m_Content->GetEntryPoint(line_out,pt2);
|
|
||||||
m_Sphere2->SetCenter(pt2(0),pt2(1),pt2(2));
|
|
||||||
m_Line2->SetPoint1(src(0),src(1),src(2));
|
|
||||||
m_Line2->SetPoint2(pt2(0),pt2(1),pt2(2));
|
|
||||||
|
|
||||||
m_Line3->SetPoint1(pt1(0),pt1(1),pt1(2));
|
|
||||||
m_Line3->SetPoint2(pt2(0),pt2(1),pt2(2));
|
|
||||||
|
|
||||||
// Create a vtkPoints object and store the points in it
|
|
||||||
vtkSmartPointer<vtkPoints> points =
|
|
||||||
vtkSmartPointer<vtkPoints>::New();
|
|
||||||
points->InsertNextPoint(pt1(0),pt1(1),pt1(2));
|
|
||||||
points->InsertNextPoint(pt2(0),pt2(1),pt2(2));
|
|
||||||
|
|
||||||
// Create a cell array to store the lines in and add the lines to it
|
|
||||||
vtkSmartPointer<vtkCellArray> lines =
|
|
||||||
vtkSmartPointer<vtkCellArray>::New();
|
|
||||||
|
|
||||||
vtkSmartPointer<vtkLine> line =
|
|
||||||
vtkSmartPointer<vtkLine>::New();
|
|
||||||
line->GetPointIds()->SetId(0,0);
|
|
||||||
line->GetPointIds()->SetId(1,1);
|
|
||||||
lines->InsertNextCell(line);
|
|
||||||
|
|
||||||
// Create a polydata to store everything in
|
|
||||||
vtkSmartPointer<vtkPolyData> linesPolyData =
|
|
||||||
vtkSmartPointer<vtkPolyData>::New();
|
|
||||||
|
|
||||||
// Add the points to the dataset
|
|
||||||
linesPolyData->SetPoints(points);
|
|
||||||
|
|
||||||
// Add the lines to the dataset
|
|
||||||
linesPolyData->SetLines(lines);
|
|
||||||
|
|
||||||
m_RayLine->RemoveAllInputs();
|
|
||||||
# if VTK_MAJOR_VERSION <= 5
|
|
||||||
m_RayLine->AddInputConnection(linesPolyData->GetProducerPort());
|
|
||||||
# endif
|
|
||||||
m_RayLine->AddInputConnection(m_Line1->GetOutputPort());
|
|
||||||
m_RayLine->AddInputConnection(m_Sphere1->GetOutputPort());
|
|
||||||
m_RayLine->AddInputConnection(m_Line2->GetOutputPort());
|
|
||||||
m_RayLine->AddInputConnection(m_Sphere2->GetOutputPort());
|
|
||||||
|
|
||||||
|
|
||||||
vtkSmartPointer<vtkMatrix4x4> vmat = vtkSmartPointer<vtkMatrix4x4>::New();
|
|
||||||
Matrix4f mat = m_Content->GetImage()->GetWorldMatrix();
|
|
||||||
for(int i=0; i<4; ++i)
|
|
||||||
for(int j=0; j<4; ++j)
|
|
||||||
vmat->SetElement(i,j,mat(i,j));
|
|
||||||
m_Transform->SetMatrix(vmat);
|
|
||||||
|
|
||||||
this->SetRay(pt1,pt2);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
void vtkVoxRaytracerRepresentation::SetMuon(MuonScatter &muon, HPoint3f poca)
|
void vtkVoxRaytracerRepresentation::SetVoxelsColor(Vector4f rgba) {
|
||||||
{
|
this->SetColor(m_RayRepresentationActor, rgba);
|
||||||
HPoint3f pt1,pt2,src;
|
|
||||||
src = muon.LineIn().origin;
|
|
||||||
m_Content->GetEntryPoint(muon.LineIn(),pt1);
|
|
||||||
m_Sphere1->SetCenter(pt1(0),pt1(1),pt1(2));
|
|
||||||
m_Line1->SetPoint1(src(0),src(1),src(2));
|
|
||||||
m_Line1->SetPoint2(pt1(0),pt1(1),pt1(2));
|
|
||||||
|
|
||||||
HLine3f line_out = muon.LineOut();
|
|
||||||
src = line_out.origin;
|
|
||||||
float *direction = line_out.direction.data();
|
|
||||||
for(int i=0;i<3;++i) direction[i] *= -1;
|
|
||||||
m_Content->GetEntryPoint(line_out,pt2);
|
|
||||||
m_Sphere2->SetCenter(pt2(0),pt2(1),pt2(2));
|
|
||||||
m_Line2->SetPoint1(src(0),src(1),src(2));
|
|
||||||
m_Line2->SetPoint2(pt2(0),pt2(1),pt2(2));
|
|
||||||
|
|
||||||
m_Line3->SetPoint1(pt1(0),pt1(1),pt1(2));
|
|
||||||
m_Line3->SetPoint2(pt2(0),pt2(1),pt2(2));
|
|
||||||
|
|
||||||
// Create a vtkPoints object and store the points in it
|
|
||||||
vtkSmartPointer<vtkPoints> points =
|
|
||||||
vtkSmartPointer<vtkPoints>::New();
|
|
||||||
points->InsertNextPoint(pt1(0),pt1(1),pt1(2));
|
|
||||||
points->InsertNextPoint(poca(0),poca(1),poca(2));
|
|
||||||
points->InsertNextPoint(pt2(0),pt2(1),pt2(2));
|
|
||||||
|
|
||||||
// Create a cell array to store the lines in and add the lines to it
|
|
||||||
vtkSmartPointer<vtkCellArray> lines =
|
|
||||||
vtkSmartPointer<vtkCellArray>::New();
|
|
||||||
|
|
||||||
for(unsigned int i = 0; i < 2; i++)
|
|
||||||
{
|
|
||||||
vtkSmartPointer<vtkLine> line =
|
|
||||||
vtkSmartPointer<vtkLine>::New();
|
|
||||||
line->GetPointIds()->SetId(0,i);
|
|
||||||
line->GetPointIds()->SetId(1,i+1);
|
|
||||||
lines->InsertNextCell(line);
|
|
||||||
}
|
|
||||||
|
|
||||||
// Create a polydata to store everything in
|
|
||||||
vtkSmartPointer<vtkPolyData> linesPolyData =
|
|
||||||
vtkSmartPointer<vtkPolyData>::New();
|
|
||||||
|
|
||||||
// Add the points to the dataset
|
|
||||||
linesPolyData->SetPoints(points);
|
|
||||||
|
|
||||||
// Add the lines to the dataset
|
|
||||||
linesPolyData->SetLines(lines);
|
|
||||||
|
|
||||||
m_RayLine->RemoveAllInputs();
|
|
||||||
# if VTK_MAJOR_VERSION <= 5
|
|
||||||
m_RayLine->AddInputConnection(linesPolyData->GetProducerPort());
|
|
||||||
# endif
|
|
||||||
m_RayLine->AddInputConnection(m_Line1->GetOutputPort());
|
|
||||||
m_RayLine->AddInputConnection(m_Sphere1->GetOutputPort());
|
|
||||||
m_RayLine->AddInputConnection(m_Line2->GetOutputPort());
|
|
||||||
m_RayLine->AddInputConnection(m_Sphere2->GetOutputPort());
|
|
||||||
|
|
||||||
|
|
||||||
vtkSmartPointer<vtkSphereSource> poca_sphere =
|
|
||||||
vtkSmartPointer<vtkSphereSource>::New();
|
|
||||||
poca_sphere->SetRadius(default_radius);
|
|
||||||
poca_sphere->SetCenter(poca(0),poca(1),poca(2));
|
|
||||||
m_RayLine->AddInputConnection(poca_sphere->GetOutputPort());
|
|
||||||
|
|
||||||
|
|
||||||
vtkSmartPointer<vtkMatrix4x4> vmat = vtkSmartPointer<vtkMatrix4x4>::New();
|
|
||||||
Matrix4f mat = m_Content->GetImage()->GetWorldMatrix();
|
|
||||||
for(int i=0; i<4; ++i)
|
|
||||||
for(int j=0; j<4; ++j)
|
|
||||||
vmat->SetElement(i,j,mat(i,j));
|
|
||||||
m_Transform->SetMatrix(vmat);
|
|
||||||
|
|
||||||
if(m_Content->GetImage()->IsInsideBounds(poca))
|
|
||||||
this->SetRay(pt1,poca,pt2);
|
|
||||||
else
|
|
||||||
this->SetRay(pt1,pt2);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
void vtkVoxRaytracerRepresentation::SetRayColor(Vector4f rgba) {
|
||||||
void vtkVoxRaytracerRepresentation::SetMuon(vtkMuonScatter &muon)
|
this->SetColor(m_RayLineActor, rgba);
|
||||||
{
|
|
||||||
HPoint3f poca = muon.GetPocaPoint();
|
|
||||||
MuonScatter &mu = muon.GetContent();
|
|
||||||
this->SetMuon(mu,poca);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
void vtkVoxRaytracerRepresentation::SetColor(vtkActor *actor, Vector4f rgba) {
|
||||||
VoxRaytracer::RayData vtkVoxRaytracerRepresentation::GetRay()
|
if (!actor)
|
||||||
{
|
return;
|
||||||
return m_Ray;
|
vtkProperty *pr = actor->GetProperty();
|
||||||
|
pr->SetDiffuseColor(rgba(0), rgba(1), rgba(2));
|
||||||
|
pr->SetOpacity(rgba(3));
|
||||||
|
pr->SetDiffuse(1);
|
||||||
}
|
}
|
||||||
|
|
||||||
void vtkVoxRaytracerRepresentation::SetRay(HPoint3f in, HPoint3f out)
|
void vtkVoxRaytracerRepresentation::InstallPipe() {
|
||||||
{
|
|
||||||
m_Ray = m_Content->TraceBetweenPoints(in,out);
|
vtkSmartPointer<vtkAppendPolyData> append =
|
||||||
this->SetRay(&m_Ray);
|
vtkSmartPointer<vtkAppendPolyData>::New();
|
||||||
|
append->AddInputConnection(m_Sphere1->GetOutputPort());
|
||||||
|
append->AddInputConnection(m_Sphere2->GetOutputPort());
|
||||||
|
append->AddInputConnection(m_Line1->GetOutputPort());
|
||||||
|
append->AddInputConnection(m_Line2->GetOutputPort());
|
||||||
|
|
||||||
|
append->Update();
|
||||||
|
vtkSmartPointer<vtkPolyDataMapper> mapper =
|
||||||
|
vtkSmartPointer<vtkPolyDataMapper>::New();
|
||||||
|
|
||||||
|
mapper->SetInputConnection(append->GetOutputPort());
|
||||||
|
mapper->Update();
|
||||||
|
|
||||||
|
vtkSmartPointer<vtkActor> actor = vtkSmartPointer<vtkActor>::New();
|
||||||
|
actor->SetMapper(mapper);
|
||||||
|
actor->GetProperty()->SetColor(0.6, 0.6, 1);
|
||||||
|
this->SetProp(actor);
|
||||||
|
|
||||||
|
mapper = vtkSmartPointer<vtkPolyDataMapper>::New();
|
||||||
|
mapper->SetInputConnection(m_RayLine->GetOutputPort());
|
||||||
|
mapper->Update();
|
||||||
|
|
||||||
|
m_RayLineActor->SetMapper(mapper);
|
||||||
|
m_RayLineActor->GetProperty()->SetColor(1, 0, 0);
|
||||||
|
this->SetProp(m_RayLineActor);
|
||||||
|
|
||||||
|
vtkSmartPointer<vtkTransformPolyDataFilter> polyfilter =
|
||||||
|
vtkSmartPointer<vtkTransformPolyDataFilter>::New();
|
||||||
|
|
||||||
|
polyfilter->SetInputConnection(m_RayRepresentation->GetOutputPort());
|
||||||
|
polyfilter->SetTransform(m_Transform);
|
||||||
|
|
||||||
|
mapper = vtkSmartPointer<vtkPolyDataMapper>::New();
|
||||||
|
mapper->SetInputConnection(polyfilter->GetOutputPort());
|
||||||
|
mapper->Update();
|
||||||
|
|
||||||
|
vtkActor *vra = m_RayRepresentationActor;
|
||||||
|
vra->SetMapper(mapper);
|
||||||
|
vra->GetProperty()->SetOpacity(0.2);
|
||||||
|
vra->GetProperty()->SetEdgeVisibility(true);
|
||||||
|
vra->GetProperty()->SetColor(0.5, 0.5, 0.5);
|
||||||
|
|
||||||
|
this->SetProp(vra);
|
||||||
}
|
}
|
||||||
|
|
||||||
void vtkVoxRaytracerRepresentation::SetRay(HPoint3f in, HPoint3f mid, HPoint3f out)
|
} // namespace Vtk
|
||||||
{
|
} // namespace uLib
|
||||||
m_Ray = m_Content->TraceBetweenPoints(in,mid);
|
|
||||||
m_Ray.AppendRay( m_Content->TraceBetweenPoints(mid,out) );
|
|
||||||
this->SetRay(&m_Ray);
|
|
||||||
}
|
|
||||||
|
|
||||||
void vtkVoxRaytracerRepresentation::SetRay(VoxRaytracer::RayData *ray)
|
|
||||||
{
|
|
||||||
vtkAppendPolyData *appender = m_RayRepresentation;
|
|
||||||
appender->RemoveAllInputs();
|
|
||||||
|
|
||||||
for(int i=0; i<ray->Data().size(); ++i) {
|
|
||||||
int id = ray->Data().at(i).vox_id;
|
|
||||||
Vector3i idv = m_Content->GetImage()->UnMap(id);
|
|
||||||
vtkSmartPointer<vtkCubeSource> cube =
|
|
||||||
vtkSmartPointer<vtkCubeSource>::New();
|
|
||||||
cube->SetBounds(idv(0),idv(0)+1,idv(1),idv(1)+1,idv(2),idv(2)+1);
|
|
||||||
cube->Update();
|
|
||||||
# if VTK_MAJOR_VERSION <= 5
|
|
||||||
appender->AddInput(cube->GetOutput());
|
|
||||||
# endif
|
|
||||||
appender->Update();
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
void vtkVoxRaytracerRepresentation::SetVoxelsColor(Vector4f rgba)
|
|
||||||
{
|
|
||||||
this->SetColor(m_RayRepresentationActor,rgba);
|
|
||||||
}
|
|
||||||
|
|
||||||
void vtkVoxRaytracerRepresentation::SetRayColor(Vector4f rgba)
|
|
||||||
{
|
|
||||||
this->SetColor(m_RayLineActor,rgba);
|
|
||||||
}
|
|
||||||
|
|
||||||
void vtkVoxRaytracerRepresentation::SetColor(vtkActor *actor, Vector4f rgba)
|
|
||||||
{
|
|
||||||
if(!actor) return;
|
|
||||||
vtkProperty *pr = actor->GetProperty();
|
|
||||||
pr->SetDiffuseColor( rgba(0),
|
|
||||||
rgba(1),
|
|
||||||
rgba(2) );
|
|
||||||
pr->SetOpacity( rgba(3) );
|
|
||||||
pr->SetDiffuse(1);
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
void vtkVoxRaytracerRepresentation::InstallPipe()
|
|
||||||
{
|
|
||||||
|
|
||||||
vtkSmartPointer<vtkAppendPolyData> append =
|
|
||||||
vtkSmartPointer<vtkAppendPolyData>::New();
|
|
||||||
append->AddInputConnection(m_Sphere1->GetOutputPort());
|
|
||||||
append->AddInputConnection(m_Sphere2->GetOutputPort());
|
|
||||||
append->AddInputConnection(m_Line1->GetOutputPort());
|
|
||||||
append->AddInputConnection(m_Line2->GetOutputPort());
|
|
||||||
|
|
||||||
append->Update();
|
|
||||||
vtkSmartPointer<vtkPolyDataMapper> mapper =
|
|
||||||
vtkSmartPointer<vtkPolyDataMapper>::New();
|
|
||||||
|
|
||||||
mapper->SetInputConnection(append->GetOutputPort());
|
|
||||||
mapper->Update();
|
|
||||||
|
|
||||||
vtkSmartPointer<vtkActor> actor = vtkSmartPointer<vtkActor>::New();
|
|
||||||
actor->SetMapper(mapper);
|
|
||||||
actor->GetProperty()->SetColor(0.6,0.6,1);
|
|
||||||
this->SetProp(actor);
|
|
||||||
|
|
||||||
mapper = vtkSmartPointer<vtkPolyDataMapper>::New();
|
|
||||||
mapper->SetInputConnection(m_RayLine->GetOutputPort());
|
|
||||||
mapper->Update();
|
|
||||||
|
|
||||||
m_RayLineActor->SetMapper(mapper);
|
|
||||||
m_RayLineActor->GetProperty()->SetColor(1,0,0);
|
|
||||||
this->SetProp(m_RayLineActor);
|
|
||||||
|
|
||||||
vtkSmartPointer<vtkTransformPolyDataFilter> polyfilter =
|
|
||||||
vtkSmartPointer<vtkTransformPolyDataFilter>::New();
|
|
||||||
|
|
||||||
polyfilter->SetInputConnection(m_RayRepresentation->GetOutputPort());
|
|
||||||
polyfilter->SetTransform(m_Transform);
|
|
||||||
|
|
||||||
mapper = vtkSmartPointer<vtkPolyDataMapper>::New();
|
|
||||||
mapper->SetInputConnection(polyfilter->GetOutputPort());
|
|
||||||
mapper->Update();
|
|
||||||
|
|
||||||
vtkActor *vra = m_RayRepresentationActor;
|
|
||||||
vra->SetMapper(mapper);
|
|
||||||
vra->GetProperty()->SetOpacity(0.2);
|
|
||||||
vra->GetProperty()->SetEdgeVisibility(true);
|
|
||||||
vra->GetProperty()->SetColor(0.5,0.5,0.5);
|
|
||||||
|
|
||||||
this->SetProp(vra);
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
} // vtk
|
|
||||||
} // uLib
|
|
||||||
|
|||||||
@@ -1,12 +0,0 @@
|
|||||||
|
|
||||||
set(HEADERS
|
|
||||||
ltktypes.h
|
|
||||||
ltkmacros.h
|
|
||||||
ltkdebug.h
|
|
||||||
)
|
|
||||||
|
|
||||||
SET(SOURCES)
|
|
||||||
|
|
||||||
set(LIBRARIES)
|
|
||||||
|
|
||||||
uLib_add_shared_library(${uLib-module})
|
|
||||||
@@ -1,111 +0,0 @@
|
|||||||
/*//////////////////////////////////////////////////////////////////////////////
|
|
||||||
// CMT Cosmic Muon Tomography project //////////////////////////////////////////
|
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
|
||||||
|
|
||||||
Copyright (c) 2014, Universita' degli Studi di Padova, INFN sez. di Padova
|
|
||||||
All rights reserved
|
|
||||||
|
|
||||||
Authors: Andrea Rigoni Garola < andrea.rigoni@pd.infn.it >
|
|
||||||
|
|
||||||
------------------------------------------------------------------
|
|
||||||
This library is free software; you can redistribute it and/or
|
|
||||||
modify it under the terms of the GNU Lesser General Public
|
|
||||||
License as published by the Free Software Foundation; either
|
|
||||||
version 3.0 of the License, or (at your option) any later version.
|
|
||||||
|
|
||||||
This library is distributed in the hope that it will be useful,
|
|
||||||
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
||||||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
||||||
Lesser General Public License for more details.
|
|
||||||
|
|
||||||
You should have received a copy of the GNU Lesser General Public
|
|
||||||
License along with this library.
|
|
||||||
|
|
||||||
//////////////////////////////////////////////////////////////////////////////*/
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
#ifndef DENSEMATRIX_H
|
|
||||||
#define DENSEMATRIX_H
|
|
||||||
|
|
||||||
#include "templates.h"
|
|
||||||
#include "Vector.h"
|
|
||||||
|
|
||||||
BEGIN_NAMESPACE(ltk)
|
|
||||||
|
|
||||||
/* ////////////////////////////////////////////////////////////////////////// */
|
|
||||||
/* //////////////////////////// MATRIX ////////////////////////////////////// */
|
|
||||||
/* ////////////////////////////////////////////////////////////////////////// */
|
|
||||||
|
|
||||||
C_BEGIN_DECLS
|
|
||||||
|
|
||||||
#define LTK_MATRIX_DECLARE_FUNC(Type,ltkType,Class_name,csize,rsize) \
|
|
||||||
inline Type ltkType##_##Class_name##_get(ltkType m, \
|
|
||||||
unsigned int c, \
|
|
||||||
unsigned int r) \
|
|
||||||
{ \
|
|
||||||
assert(c + r * csize < csize * rsize); \
|
|
||||||
return *((ltkType)m + c + r * csize); \
|
|
||||||
} \
|
|
||||||
inline void ltkType##_##Class_name##_set(ltkType m, \
|
|
||||||
unsigned int c, \
|
|
||||||
unsigned int r, \
|
|
||||||
Type t) \
|
|
||||||
{ \
|
|
||||||
assert (c + r * csize < csize * rsize); \
|
|
||||||
*((ltkType)m + c + r * csize) = t; \
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
#define LTK_MATRIX_DECLARE(Type,Type_name,csize,rsize) \
|
|
||||||
typedef Type * Type_name; \
|
|
||||||
LTK_SIMPLE_ALLOC_FUNC(Type, Type_name,ltk_matrix,csize * rsize) \
|
|
||||||
LTK_MATRIX_DECLARE_FUNC(Type,Type_name,ltk_matrix,csize,rsize)
|
|
||||||
|
|
||||||
C_END_DECLS
|
|
||||||
|
|
||||||
#ifdef __cplusplus
|
|
||||||
|
|
||||||
template <typename T, unsigned int csize, unsigned int rsize> class ltkDenseMatrix
|
|
||||||
: public ltkAbstractBuffer< T, csize * rsize >
|
|
||||||
{
|
|
||||||
protected:
|
|
||||||
typedef ltkDenseMatrix<T, csize, rsize> ThisClass;
|
|
||||||
typedef ltkAbstractBuffer<T, csize * rsize> BaseClass;
|
|
||||||
public:
|
|
||||||
ltkDenseMatrix() {}
|
|
||||||
~ltkDenseMatrix() { }
|
|
||||||
|
|
||||||
inline T* getBuffer() {
|
|
||||||
return (T*)m_buffer;
|
|
||||||
}
|
|
||||||
inline const T* getBuffer() const {
|
|
||||||
return (const T*)m_buffer;
|
|
||||||
}
|
|
||||||
|
|
||||||
ThisClass & operator =(const ThisClass &src) {
|
|
||||||
if (this != &src) {
|
|
||||||
CType_ltk_matrix_copy(m_buffer, (const CType) src.getBuffer());
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
typedef struct BaseClass::CommaInit CommaInit;
|
|
||||||
inline CommaInit operator = (T scalar) {
|
|
||||||
return this->operator <<(scalar);
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
private:
|
|
||||||
LTK_MATRIX_DECLARE(T,CType, csize, rsize);
|
|
||||||
T m_buffer [csize * rsize];
|
|
||||||
};
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
#endif // __cplusplus
|
|
||||||
|
|
||||||
END_NAMESPACE
|
|
||||||
|
|
||||||
#endif // DENSEMATRIX_H
|
|
||||||
@@ -1,6 +0,0 @@
|
|||||||
include $(top_srcdir)/Common.am
|
|
||||||
|
|
||||||
library_includedir = $(includedir)/libmutom-${PACKAGE_VERSION}/ltk
|
|
||||||
library_include_HEADERS = ltktypes.h \
|
|
||||||
ltkmacros.h \
|
|
||||||
ltkdebug.h
|
|
||||||
@@ -1,18 +0,0 @@
|
|||||||
#include "Object.h"
|
|
||||||
|
|
||||||
|
|
||||||
static const struct _ltkObjectClass {
|
|
||||||
size_t class_size;
|
|
||||||
void (* constructor)(struct _Object *);
|
|
||||||
void (* destructor) (struct _Object *);
|
|
||||||
int (* GetElement) (struct _Object *);
|
|
||||||
} _ObjectClassInstance;
|
|
||||||
|
|
||||||
const struct _ltkObjectClass *ltkObjectClass = &_ObjectClassInstance;
|
|
||||||
|
|
||||||
ltkPointer Object_new(struct _ObjectClass klass)
|
|
||||||
{
|
|
||||||
struct _Object *ob = (struct _Object)malloc(klass->class_size);
|
|
||||||
klass->constructor(ob);
|
|
||||||
return ob;
|
|
||||||
}
|
|
||||||
@@ -1,78 +0,0 @@
|
|||||||
/*//////////////////////////////////////////////////////////////////////////////
|
|
||||||
// CMT Cosmic Muon Tomography project //////////////////////////////////////////
|
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
|
||||||
|
|
||||||
Copyright (c) 2014, Universita' degli Studi di Padova, INFN sez. di Padova
|
|
||||||
All rights reserved
|
|
||||||
|
|
||||||
Authors: Andrea Rigoni Garola < andrea.rigoni@pd.infn.it >
|
|
||||||
|
|
||||||
------------------------------------------------------------------
|
|
||||||
This library is free software; you can redistribute it and/or
|
|
||||||
modify it under the terms of the GNU Lesser General Public
|
|
||||||
License as published by the Free Software Foundation; either
|
|
||||||
version 3.0 of the License, or (at your option) any later version.
|
|
||||||
|
|
||||||
This library is distributed in the hope that it will be useful,
|
|
||||||
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
||||||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
||||||
Lesser General Public License for more details.
|
|
||||||
|
|
||||||
You should have received a copy of the GNU Lesser General Public
|
|
||||||
License along with this library.
|
|
||||||
|
|
||||||
//////////////////////////////////////////////////////////////////////////////*/
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
#ifndef OBJECT_H
|
|
||||||
#define OBJECT_H
|
|
||||||
|
|
||||||
#include <stdlib.h> // for std allocator //
|
|
||||||
#include <string.h> // for memcpy //
|
|
||||||
#include <assert.h>
|
|
||||||
|
|
||||||
#include "templates.h"
|
|
||||||
|
|
||||||
BEGIN_NAMESPACE(ltk)
|
|
||||||
|
|
||||||
|
|
||||||
C_BEGIN_DECLS
|
|
||||||
|
|
||||||
struct _ltkObject
|
|
||||||
{
|
|
||||||
int element;
|
|
||||||
};
|
|
||||||
|
|
||||||
struct _ltkObjectClass {
|
|
||||||
size_t class_size;
|
|
||||||
void (* constructor)(struct _ltkObject *);
|
|
||||||
void (* destructor) (struct _ltkObject *);
|
|
||||||
int (* GetElement) (struct _ltkObject *);
|
|
||||||
};
|
|
||||||
|
|
||||||
struct _ltkObject *ltk_object_new()
|
|
||||||
{
|
|
||||||
return NULL;
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
C_END_DECLS
|
|
||||||
|
|
||||||
#ifdef __cplusplus
|
|
||||||
|
|
||||||
class Object
|
|
||||||
{
|
|
||||||
struct _ltkObject *d;
|
|
||||||
typedef struct _ltkObject ObjectType;
|
|
||||||
public:
|
|
||||||
Object() : d(new ObjectType()) { d->element = 5552368; }
|
|
||||||
int GetElement() { return d->element; }
|
|
||||||
};
|
|
||||||
|
|
||||||
#endif
|
|
||||||
|
|
||||||
|
|
||||||
END_NAMESPACE
|
|
||||||
|
|
||||||
#endif //OBJECT_H
|
|
||||||
195
src/ltk/Vector.h
195
src/ltk/Vector.h
@@ -1,195 +0,0 @@
|
|||||||
/*//////////////////////////////////////////////////////////////////////////////
|
|
||||||
// CMT Cosmic Muon Tomography project //////////////////////////////////////////
|
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
|
||||||
|
|
||||||
Copyright (c) 2014, Universita' degli Studi di Padova, INFN sez. di Padova
|
|
||||||
All rights reserved
|
|
||||||
|
|
||||||
Authors: Andrea Rigoni Garola < andrea.rigoni@pd.infn.it >
|
|
||||||
|
|
||||||
------------------------------------------------------------------
|
|
||||||
This library is free software; you can redistribute it and/or
|
|
||||||
modify it under the terms of the GNU Lesser General Public
|
|
||||||
License as published by the Free Software Foundation; either
|
|
||||||
version 3.0 of the License, or (at your option) any later version.
|
|
||||||
|
|
||||||
This library is distributed in the hope that it will be useful,
|
|
||||||
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
||||||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
||||||
Lesser General Public License for more details.
|
|
||||||
|
|
||||||
You should have received a copy of the GNU Lesser General Public
|
|
||||||
License along with this library.
|
|
||||||
|
|
||||||
//////////////////////////////////////////////////////////////////////////////*/
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
/*
|
|
||||||
* File: denseVector.h
|
|
||||||
* Author: andrea
|
|
||||||
*
|
|
||||||
* Created on July 6, 2012, 3:41 PM
|
|
||||||
*/
|
|
||||||
|
|
||||||
#ifndef VECTOR_H
|
|
||||||
#define VECTOR_H
|
|
||||||
|
|
||||||
#include <stdlib.h> // for std allocator //
|
|
||||||
#include <string.h> // for memcpy //
|
|
||||||
#include <assert.h>
|
|
||||||
|
|
||||||
#include "templates.h"
|
|
||||||
|
|
||||||
BEGIN_NAMESPACE(ltk)
|
|
||||||
|
|
||||||
/* ////////////////////////////////////////////////////////////////////////// */
|
|
||||||
/* //////////////////////////// VECTOR ///////////////////////////////////// */
|
|
||||||
/* ////////////////////////////////////////////////////////////////////////// */
|
|
||||||
|
|
||||||
C_BEGIN_DECLS
|
|
||||||
|
|
||||||
#define LTK_VECTOR_DECLARE(Type,Type_name,size) \
|
|
||||||
typedef Type *Type_name; \
|
|
||||||
LTK_SIMPLE_ALLOC_FUNC(Type,Type_name,ltk_vector,size) \
|
|
||||||
inline unsigned int Type_name##_ltk_vector_size() { return size; } \
|
|
||||||
inline Type Type_name##_ltk_vector_get(Type_name vector, unsigned int i) \
|
|
||||||
{ return *(vector+i); } \
|
|
||||||
inline void Type_name##_ltk_vector_set(Type_name vector, unsigned int i, \
|
|
||||||
Type data) { *(vector+i) = data; }
|
|
||||||
|
|
||||||
C_END_DECLS
|
|
||||||
|
|
||||||
#ifdef __cplusplus
|
|
||||||
|
|
||||||
template <typename T, unsigned int size> class ltkVector
|
|
||||||
: public ltkAbstractBuffer< T, size >
|
|
||||||
{
|
|
||||||
T m_vector [size];
|
|
||||||
protected:
|
|
||||||
typedef ltkVector<T, size> ThisClass;
|
|
||||||
typedef ltkAbstractBuffer<T, size> BaseClass;
|
|
||||||
public:
|
|
||||||
ltkVector() {}
|
|
||||||
ltkVector(const ltkVector<T, size> ©) {
|
|
||||||
CType_ltk_vector_copy(m_vector, (const CType) copy.getBuffer());
|
|
||||||
}
|
|
||||||
|
|
||||||
virtual ~ltkVector() { }
|
|
||||||
|
|
||||||
inline unsigned int getSize() const {
|
|
||||||
return size;
|
|
||||||
}
|
|
||||||
inline T* getBuffer() {
|
|
||||||
return (T*)m_vector;
|
|
||||||
}
|
|
||||||
inline const T* getBuffer() const {
|
|
||||||
return (const T*)m_vector;
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
ThisClass & operator =(const ThisClass &src) {
|
|
||||||
if (this != &src) {
|
|
||||||
CType_ltk_vector_copy(m_vector, (const CType) src.getBuffer());
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
typedef struct BaseClass::CommaInit CommaInit;
|
|
||||||
inline CommaInit operator = (T scalar) {
|
|
||||||
return this->operator <<(scalar);
|
|
||||||
}
|
|
||||||
|
|
||||||
// MATH //
|
|
||||||
inline bool operator ==(const ThisClass &in) {
|
|
||||||
int test = 0;
|
|
||||||
for (int i = 0; i < size; ++i) test += this->at(i) != in(i);
|
|
||||||
return test == 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
inline bool operator !=(const ThisClass &in) {
|
|
||||||
return !this->operator ==(in);
|
|
||||||
}
|
|
||||||
|
|
||||||
inline void operator +=(const ThisClass &in) {
|
|
||||||
for (int i = 0; i < size; ++i) this->at(i) += in(i);
|
|
||||||
}
|
|
||||||
inline void operator -=(const ThisClass &in) {
|
|
||||||
for (int i = 0; i < size; ++i) this->at(i) -= in(i);
|
|
||||||
}
|
|
||||||
inline void operator *=(const ThisClass &in) {
|
|
||||||
for (int i = 0; i < size; ++i) this->at(i) *= in(i);
|
|
||||||
}
|
|
||||||
inline void operator /=(const ThisClass &in) {
|
|
||||||
for (int i = 0; i < size; ++i) this->at(i) /= in(i);
|
|
||||||
}
|
|
||||||
|
|
||||||
inline void operator +=(const T t) {
|
|
||||||
for (int i = 0; i < size; ++i) this->at(i) += t;
|
|
||||||
}
|
|
||||||
inline void operator -=(const T t) {
|
|
||||||
for (int i = 0; i < size; ++i) this->at(i) -= t;
|
|
||||||
}
|
|
||||||
inline void operator *=(const T t) {
|
|
||||||
for (int i = 0; i < size; ++i) this->at(i) *= t;
|
|
||||||
}
|
|
||||||
inline void operator /=(const T t) {
|
|
||||||
for (int i = 0; i < size; ++i) this->at(i) /= t;
|
|
||||||
}
|
|
||||||
|
|
||||||
inline ThisClass & operator + (const ThisClass &in) {
|
|
||||||
ThisClass *out = new ThisClass();
|
|
||||||
for (int i = 0; i < size; ++i) out->at(i) = this->at(i) + in(i);
|
|
||||||
return *out;
|
|
||||||
}
|
|
||||||
inline ThisClass & operator - (const ThisClass &in) {
|
|
||||||
ThisClass *out = new ThisClass();
|
|
||||||
for (int i = 0; i < size; ++i) out->at(i) = this->at(i) - in(i);
|
|
||||||
return *out;
|
|
||||||
}
|
|
||||||
|
|
||||||
// fix with constructor and product null element ------------- //
|
|
||||||
inline T operator * (const ThisClass &in) {
|
|
||||||
T out = 0;
|
|
||||||
for (int i = 0; i < size; ++i) out += this->at(i) * in(i);
|
|
||||||
return out;
|
|
||||||
}
|
|
||||||
inline T operator / (const ThisClass &in) {
|
|
||||||
T out = 0;
|
|
||||||
for (int i = 0; i < size; ++i) out += this->at(i) / in(i);
|
|
||||||
return out;
|
|
||||||
}
|
|
||||||
// ------------------------------------------------------------- //
|
|
||||||
|
|
||||||
inline ThisClass & operator +(const T t) {
|
|
||||||
ThisClass *out = new ThisClass();
|
|
||||||
for (int i = 0; i < size; ++i) out->at(i) = this->at(i) + t;
|
|
||||||
return *out;
|
|
||||||
}
|
|
||||||
inline ThisClass & operator -(const T t) {
|
|
||||||
ThisClass *out = new ThisClass();
|
|
||||||
for (int i = 0; i < size; ++i) out->at(i) = this->at(i) - t;
|
|
||||||
return *out;
|
|
||||||
}
|
|
||||||
inline ThisClass & operator *(const T t) {
|
|
||||||
ThisClass *out = new ThisClass();
|
|
||||||
for (int i = 0; i < size; ++i) out->at(i) = this->at(i) * t;
|
|
||||||
return *out;
|
|
||||||
}
|
|
||||||
inline ThisClass & operator /(const T t) {
|
|
||||||
ThisClass *out = new ThisClass();
|
|
||||||
for (int i = 0; i < size; ++i) out->at(i) = this->at(i) / t;
|
|
||||||
return *out;
|
|
||||||
}
|
|
||||||
private:
|
|
||||||
LTK_VECTOR_DECLARE(T,CType, size)
|
|
||||||
};
|
|
||||||
|
|
||||||
#endif // __cplusplus
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
END_NAMESPACE // ltk
|
|
||||||
#endif /* DENSEVECTOR_H */
|
|
||||||
|
|
||||||
@@ -1,43 +0,0 @@
|
|||||||
/*//////////////////////////////////////////////////////////////////////////////
|
|
||||||
// CMT Cosmic Muon Tomography project //////////////////////////////////////////
|
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
|
||||||
|
|
||||||
Copyright (c) 2014, Universita' degli Studi di Padova, INFN sez. di Padova
|
|
||||||
All rights reserved
|
|
||||||
|
|
||||||
Authors: Andrea Rigoni Garola < andrea.rigoni@pd.infn.it >
|
|
||||||
|
|
||||||
------------------------------------------------------------------
|
|
||||||
This library is free software; you can redistribute it and/or
|
|
||||||
modify it under the terms of the GNU Lesser General Public
|
|
||||||
License as published by the Free Software Foundation; either
|
|
||||||
version 3.0 of the License, or (at your option) any later version.
|
|
||||||
|
|
||||||
This library is distributed in the hope that it will be useful,
|
|
||||||
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
||||||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
||||||
Lesser General Public License for more details.
|
|
||||||
|
|
||||||
You should have received a copy of the GNU Lesser General Public
|
|
||||||
License along with this library.
|
|
||||||
|
|
||||||
//////////////////////////////////////////////////////////////////////////////*/
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
/*
|
|
||||||
* File: comma_init.h
|
|
||||||
* Author: andrea
|
|
||||||
*
|
|
||||||
* Created on May 24, 2012, 3:12 PM
|
|
||||||
*/
|
|
||||||
|
|
||||||
#ifndef CONTAINER_H
|
|
||||||
#define CONTAINER_H
|
|
||||||
|
|
||||||
#include <assert.h>
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
#endif /* CONTAINER_H */
|
|
||||||
|
|
||||||
@@ -1,32 +0,0 @@
|
|||||||
/*//////////////////////////////////////////////////////////////////////////////
|
|
||||||
// CMT Cosmic Muon Tomography project //////////////////////////////////////////
|
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
|
||||||
|
|
||||||
Copyright (c) 2014, Universita' degli Studi di Padova, INFN sez. di Padova
|
|
||||||
All rights reserved
|
|
||||||
|
|
||||||
Authors: Andrea Rigoni Garola < andrea.rigoni@pd.infn.it >
|
|
||||||
|
|
||||||
------------------------------------------------------------------
|
|
||||||
This library is free software; you can redistribute it and/or
|
|
||||||
modify it under the terms of the GNU Lesser General Public
|
|
||||||
License as published by the Free Software Foundation; either
|
|
||||||
version 3.0 of the License, or (at your option) any later version.
|
|
||||||
|
|
||||||
This library is distributed in the hope that it will be useful,
|
|
||||||
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
||||||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
||||||
Lesser General Public License for more details.
|
|
||||||
|
|
||||||
You should have received a copy of the GNU Lesser General Public
|
|
||||||
License along with this library.
|
|
||||||
|
|
||||||
//////////////////////////////////////////////////////////////////////////////*/
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
#include "templates.h"
|
|
||||||
#include "Vector.h"
|
|
||||||
#include "DenseMatrix.h"
|
|
||||||
@@ -1,339 +0,0 @@
|
|||||||
//#include <linux/config.h>
|
|
||||||
#include <linux/version.h>
|
|
||||||
#include <linux/init.h>
|
|
||||||
#include <linux/module.h>
|
|
||||||
#include <linux/fs.h>
|
|
||||||
#include <linux/cdev.h>
|
|
||||||
#include <linux/slab.h>
|
|
||||||
#include <linux/vmalloc.h>
|
|
||||||
#include <linux/mm.h>
|
|
||||||
#include <linux/ioctl.h>
|
|
||||||
#ifdef MODVERSIONS
|
|
||||||
# include <linux/modversions.h>
|
|
||||||
#endif
|
|
||||||
#include <asm/io.h>
|
|
||||||
#include <asm/system.h>
|
|
||||||
#include <asm/uaccess.h>
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
/* methods of the character device */
|
|
||||||
static int ltkb_open(struct inode *inode, struct file *filp);
|
|
||||||
static int ltkb_release(struct inode *inode, struct file *filp);
|
|
||||||
static int ltkb_mmap(struct file *filp, struct vm_area_struct *vma);
|
|
||||||
static ssize_t ltkb_read(struct file *filp,
|
|
||||||
char __user *buffer,
|
|
||||||
size_t count,
|
|
||||||
loff_t *offp);
|
|
||||||
static ssize_t ltkb_write(struct file *filp,
|
|
||||||
const char __user *buffer,
|
|
||||||
size_t count,
|
|
||||||
loff_t *offp);
|
|
||||||
|
|
||||||
|
|
||||||
// internal data
|
|
||||||
// length of the two memory areas
|
|
||||||
#define NPAGES 16
|
|
||||||
// pointer to the vmalloc'd area - alway page aligned
|
|
||||||
static int *vmalloc_area;
|
|
||||||
// pointer to the kmalloc'd area, rounded up to a page boundary
|
|
||||||
static int *kmalloc_area;
|
|
||||||
// original pointer for kmalloc'd area as returned by kmalloc
|
|
||||||
static void *kmalloc_ptr;
|
|
||||||
|
|
||||||
|
|
||||||
#define BUF_LEN 80
|
|
||||||
static int Device_Open = 0;
|
|
||||||
|
|
||||||
static char Message[BUF_LEN];
|
|
||||||
/*
|
|
||||||
* How far did the process reading the message get?
|
|
||||||
* Useful if the message is larger than the size of the
|
|
||||||
* buffer we get to fill in device_read.
|
|
||||||
*/
|
|
||||||
static char *Message_Ptr;
|
|
||||||
|
|
||||||
|
|
||||||
struct ltkb_dev {
|
|
||||||
struct cdev cdev; // device structure
|
|
||||||
struct semaphore semaphore; // lock semaphore for race control
|
|
||||||
char *message; // message passing from/to user
|
|
||||||
unsigned int size; // size of data stored
|
|
||||||
unsigned int acl_key; // user access control
|
|
||||||
};
|
|
||||||
|
|
||||||
struct ltkb_dev *_ltkb_devices; // devices array //
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
/////// OPEN AND RELEASE ///////////////////////////////////////////////////////
|
|
||||||
static int ltkb_open(struct inode *inode, struct file *filp)
|
|
||||||
{
|
|
||||||
// insert cdev data in dev structure and file->private_data //
|
|
||||||
struct ltkb_dev *dev;
|
|
||||||
dev = container_of (inode->i_cdev, struct ltkb_dev, cdev);
|
|
||||||
|
|
||||||
// this shares dev for all methods //
|
|
||||||
filp->private_data = dev;
|
|
||||||
|
|
||||||
pr_debug("ltk bridge opened");
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
/* character device last close method */
|
|
||||||
static int ltkb_release(struct inode *inode, struct file *filp)
|
|
||||||
{
|
|
||||||
pr_debug("ltk bridge released");
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
// -------------------------------------------------------------------------- //
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
// kmalloc instancer //
|
|
||||||
int ltkb_kmem(struct file *filp, struct vm_area_struct *vma)
|
|
||||||
{
|
|
||||||
int ret;
|
|
||||||
long length = vma->vm_end - vma->vm_start;
|
|
||||||
|
|
||||||
/* check length - do not allow larger mappings than the number of
|
|
||||||
pages allocated */
|
|
||||||
if (length > NPAGES * PAGE_SIZE)
|
|
||||||
return -EIO;
|
|
||||||
|
|
||||||
/* map the whole physically contiguous area in one piece */
|
|
||||||
if ((ret = remap_pfn_range(vma,
|
|
||||||
vma->vm_start,
|
|
||||||
virt_to_phys((void *)kmalloc_area) >> PAGE_SHIFT,
|
|
||||||
length,
|
|
||||||
vma->vm_page_prot)) < 0) {
|
|
||||||
return ret;
|
|
||||||
}
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
// vmalloc instancer //
|
|
||||||
int ltkb_vmem(struct file *filp, struct vm_area_struct *vma)
|
|
||||||
{
|
|
||||||
int ret;
|
|
||||||
long length = vma->vm_end - vma->vm_start;
|
|
||||||
unsigned long start = vma->vm_start;
|
|
||||||
char *vmalloc_area_ptr = (char *)vmalloc_area;
|
|
||||||
unsigned long pfn;
|
|
||||||
|
|
||||||
/* check length - do not allow larger mappings than the number of
|
|
||||||
pages allocated */
|
|
||||||
if (length > NPAGES * PAGE_SIZE)
|
|
||||||
return -EIO;
|
|
||||||
|
|
||||||
/* loop over all pages, map it page individually */
|
|
||||||
while (length > 0) {
|
|
||||||
pfn = vmalloc_to_pfn(vmalloc_area_ptr);
|
|
||||||
if ((ret = remap_pfn_range(vma, start, pfn, PAGE_SIZE,
|
|
||||||
PAGE_SHARED)) < 0) {
|
|
||||||
return ret;
|
|
||||||
}
|
|
||||||
start += PAGE_SIZE;
|
|
||||||
vmalloc_area_ptr += PAGE_SIZE;
|
|
||||||
length -= PAGE_SIZE;
|
|
||||||
}
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
// mmap function //
|
|
||||||
static int ltkb_mmap(struct file *filp, struct vm_area_struct *vma)
|
|
||||||
{
|
|
||||||
/* at offset 0 we map the vmalloc'd area */
|
|
||||||
if (vma->vm_pgoff == 0) {
|
|
||||||
return ltkb_vmem(filp, vma);
|
|
||||||
}
|
|
||||||
/* at offset NPAGES we map the kmalloc'd area */
|
|
||||||
if (vma->vm_pgoff == NPAGES) {
|
|
||||||
return ltkb_kmem(filp, vma);
|
|
||||||
}
|
|
||||||
/* at any other offset we return an error */
|
|
||||||
return -EIO;
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
////////////// CHAR DEVICE READ/WRITE /////////////////////////////////////////
|
|
||||||
static ssize_t ltkb_read(struct file *filp, /* see include/linux/fs.h */
|
|
||||||
char __user *buffer, /* buffer to fill with data */
|
|
||||||
size_t length, /* length of the buffer */
|
|
||||||
loff_t * offset)
|
|
||||||
{
|
|
||||||
|
|
||||||
#ifdef _DEBUG
|
|
||||||
printk(KERN_INFO "device_read(%p,%s,%d)", filp, buffer, (int)length);
|
|
||||||
#endif
|
|
||||||
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
static ssize_t ltkb_write(struct file *filp,
|
|
||||||
const char __user *buffer,
|
|
||||||
size_t length,
|
|
||||||
loff_t * offset)
|
|
||||||
{
|
|
||||||
#ifdef _DEBUG
|
|
||||||
printk(KERN_INFO "device_write(%p,%s,%d)", filp, buffer, (int)length);
|
|
||||||
#endif
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
// ------------------------------------------------------------------------- //
|
|
||||||
|
|
||||||
|
|
||||||
#define LTKB_IOC_HEADER 'b'
|
|
||||||
|
|
||||||
#define LTKB_IOC_RESET _IO(LTKB_IOC_HEADER, 0)
|
|
||||||
#define LTKB_IOC_PING _IOWR(LTKB_IOC_HEADER, 1, char)
|
|
||||||
|
|
||||||
#define LTKB_IOC_MAXNR 1
|
|
||||||
|
|
||||||
////////////// IOCTL ///////////////////////////////////////////////////////////
|
|
||||||
int ltkb_ioctl(struct inode *inode, struct file *filp,
|
|
||||||
unsigned int cmd, unsigned long arg)
|
|
||||||
{
|
|
||||||
char c = 0;
|
|
||||||
int retval = 0;
|
|
||||||
|
|
||||||
/*
|
|
||||||
* extract the type and number bitfields, and don't decode
|
|
||||||
* wrong cmds: return ENOTTY (inappropriate ioctl) before access_ok()
|
|
||||||
*/
|
|
||||||
if (_IOC_TYPE(cmd) != LTKB_IOC_HEADER) return -ENOTTY;
|
|
||||||
if (_IOC_NR(cmd) > LTKB_IOC_MAXNR) return -ENOTTY;
|
|
||||||
|
|
||||||
// TODO user access control //
|
|
||||||
|
|
||||||
|
|
||||||
switch(cmd) {
|
|
||||||
case LTKB_IOC_RESET:
|
|
||||||
if(!capable(CAP_SYS_ADMIN))
|
|
||||||
return -EPERM;
|
|
||||||
// TODO free all memory mapping//
|
|
||||||
break;
|
|
||||||
case LTKB_IOC_PING:
|
|
||||||
retval = __get_user(c,(char __user *)arg);
|
|
||||||
c++;
|
|
||||||
retval = __put_user(c,(char __user *)arg);
|
|
||||||
break;
|
|
||||||
default:
|
|
||||||
return -ENOTTY;
|
|
||||||
}
|
|
||||||
return retval;
|
|
||||||
|
|
||||||
}
|
|
||||||
// ------------------------------------------------------------------------- //
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
#ifndef LTKB_MAJOR
|
|
||||||
#define LTKB_MAJOR 0
|
|
||||||
#endif
|
|
||||||
|
|
||||||
#ifndef LTKB_NR_DEVS
|
|
||||||
#define LTKB_NR_DEVS 4
|
|
||||||
#endif
|
|
||||||
|
|
||||||
#ifndef LTKB_DEVICE_NAME
|
|
||||||
#define LTKB_DEVICE_NAME "ltkb"
|
|
||||||
#endif
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
/* the file operations, i.e. all character device methods */
|
|
||||||
static struct file_operations ltkb_fops = {
|
|
||||||
.open = ltkb_open,
|
|
||||||
.release = ltkb_release,
|
|
||||||
.read = ltkb_read,
|
|
||||||
.write = ltkb_write,
|
|
||||||
.ioctl = ltkb_ioctl,
|
|
||||||
.mmap = ltkb_mmap,
|
|
||||||
.owner = THIS_MODULE,
|
|
||||||
};
|
|
||||||
|
|
||||||
static int _ltkb_major_number = LTKB_MAJOR;
|
|
||||||
|
|
||||||
// Module init function ////////////////////////////////////////////////////////
|
|
||||||
static int __init ltkb_init(void)
|
|
||||||
{
|
|
||||||
int i,ret = 0;
|
|
||||||
dev_t dev = 0;
|
|
||||||
|
|
||||||
// REGISTER DEVICE //
|
|
||||||
if(_ltkb_major_number)
|
|
||||||
{
|
|
||||||
dev = MKDEV(_ltkb_major_number,0);
|
|
||||||
ret = register_chrdev_region(dev, LTKB_NR_DEVS, LTKB_DEVICE_NAME);
|
|
||||||
}
|
|
||||||
else {
|
|
||||||
ret = alloc_chrdev_region(&dev, 0, LTKB_NR_DEVS, LTKB_DEVICE_NAME );
|
|
||||||
_ltkb_major_number = MAJOR(dev);
|
|
||||||
}
|
|
||||||
if (ret < 0) {
|
|
||||||
printk(KERN_WARNING "could not allocate major number %d for ltkb\n",
|
|
||||||
_ltkb_major_number);
|
|
||||||
return ret;
|
|
||||||
}
|
|
||||||
|
|
||||||
// ALLOCATE DEVICES //
|
|
||||||
_ltkb_devices = kmalloc( LTKB_NR_DEVS * sizeof(struct ltkb_dev), GFP_KERNEL );
|
|
||||||
if(!_ltkb_devices) {
|
|
||||||
printk(KERN_ERR "error allocating device structure in memory");
|
|
||||||
return -ENOMEM;
|
|
||||||
}
|
|
||||||
memset(_ltkb_devices, 0, LTKB_NR_DEVS * sizeof(struct ltkb_dev));
|
|
||||||
|
|
||||||
// SETUP DEVICES //
|
|
||||||
for (i = 0 ; i < LTKB_NR_DEVS ; i++ )
|
|
||||||
{
|
|
||||||
struct ltkb_dev *dev = &_ltkb_devices[i];
|
|
||||||
int devno = MKDEV(_ltkb_major_number, i);
|
|
||||||
cdev_init (&dev->cdev, <kb_fops);
|
|
||||||
dev->cdev.owner = THIS_MODULE;
|
|
||||||
dev->cdev.ops = <kb_fops;
|
|
||||||
ret = cdev_add(&dev->cdev,devno,1);
|
|
||||||
if(ret) printk(KERN_NOTICE "Error %d adding ltkb%d device", ret, i);
|
|
||||||
init_MUTEX(&dev->semaphore);
|
|
||||||
}
|
|
||||||
printk( KERN_NOTICE "ltk bridge module loaded");
|
|
||||||
return ret;
|
|
||||||
}
|
|
||||||
|
|
||||||
// module release //
|
|
||||||
static void __exit ltkb_exit(void)
|
|
||||||
{
|
|
||||||
// remove the character deivce //
|
|
||||||
int i;
|
|
||||||
dev_t devno = MKDEV(_ltkb_major_number,0);
|
|
||||||
if(_ltkb_devices)
|
|
||||||
{
|
|
||||||
for(i=0;i<LTKB_NR_DEVS;i++)
|
|
||||||
cdev_del(&_ltkb_devices[i].cdev);
|
|
||||||
kfree(_ltkb_devices);
|
|
||||||
}
|
|
||||||
unregister_chrdev_region(devno,LTKB_NR_DEVS);
|
|
||||||
printk( KERN_NOTICE "ltk bridge module unloaded");
|
|
||||||
}
|
|
||||||
// ------------------------------------------------------------------------ //
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
//////////////////// Main Module Init ///////////////////
|
|
||||||
//
|
|
||||||
module_init(ltkb_init); //
|
|
||||||
module_exit(ltkb_exit); //
|
|
||||||
//
|
|
||||||
MODULE_DESCRIPTION("ltk bridge memory dispatcher"); //
|
|
||||||
MODULE_AUTHOR("andrea"); //
|
|
||||||
MODULE_LICENSE("Not licenced yet"); //
|
|
||||||
//
|
|
||||||
/////////////////////////////////////////////////////////
|
|
||||||
@@ -1,40 +0,0 @@
|
|||||||
#include <stdio.h>
|
|
||||||
#include <unistd.h>
|
|
||||||
#include <sys/mman.h>
|
|
||||||
#include <sys/types.h>
|
|
||||||
#include <sys/stat.h>
|
|
||||||
#include <fcntl.h>
|
|
||||||
#include <stdlib.h>
|
|
||||||
#include <sys/ioctl.h>
|
|
||||||
|
|
||||||
#define LTKB_IOC_HEADER 'b'
|
|
||||||
|
|
||||||
#define LTKB_IOC_RESET _IO(LTKB_IOC_HEADER, 0)
|
|
||||||
#define LTKB_IOC_PING _IOWR(LTKB_IOC_HEADER, 1, char)
|
|
||||||
|
|
||||||
#define LTKB_IOC_MAXNR 1
|
|
||||||
|
|
||||||
|
|
||||||
int main(void)
|
|
||||||
{
|
|
||||||
int fd;
|
|
||||||
|
|
||||||
if ((fd=open("ltkb0", O_RDWR|O_SYNC))<0)
|
|
||||||
{
|
|
||||||
perror("open");
|
|
||||||
exit(-1);
|
|
||||||
}
|
|
||||||
|
|
||||||
char buffer[200] = "hello";
|
|
||||||
write(fd, (void *)&buffer, 6);
|
|
||||||
|
|
||||||
char c = 'a';
|
|
||||||
printf("%c ",c);
|
|
||||||
ioctl(fd, LTKB_IOC_PING, &c);
|
|
||||||
printf("%c ",c);
|
|
||||||
ioctl(fd, LTKB_IOC_PING, &c);
|
|
||||||
printf("%c\n",c);
|
|
||||||
close(fd);
|
|
||||||
return(0);
|
|
||||||
}
|
|
||||||
|
|
||||||
@@ -1,21 +0,0 @@
|
|||||||
#!/bin/sh
|
|
||||||
module="ltkb"
|
|
||||||
device="ltkb"
|
|
||||||
mode="664"
|
|
||||||
|
|
||||||
/sbin/insmod ./$module.ko $* || exit 1
|
|
||||||
|
|
||||||
rm -f ./${device}[0-3]
|
|
||||||
|
|
||||||
major=$(awk "\$2==\"$module\" {print \$1}" /proc/devices)
|
|
||||||
|
|
||||||
mknod ./${device}0 c $major 0
|
|
||||||
mknod ./${device}1 c $major 1
|
|
||||||
mknod ./${device}2 c $major 2
|
|
||||||
mknod ./${device}3 c $major 3
|
|
||||||
|
|
||||||
group="mutom"
|
|
||||||
grep -q '^mutom:' /etc/group || group="adm"
|
|
||||||
|
|
||||||
chgrp $group ./${device}[0-3]
|
|
||||||
chmod $mode ./${device}[0-3]
|
|
||||||
@@ -1,4 +0,0 @@
|
|||||||
|
|
||||||
#include "ltkdebug.h"
|
|
||||||
|
|
||||||
|
|
||||||
Some files were not shown because too many files have changed in this diff Show More
Reference in New Issue
Block a user