Compare commits
4 Commits
987d783fdb
...
andrea-dev
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
e09a614fa5 | ||
|
|
7f6323403d | ||
|
|
a53b3051de | ||
|
|
c53570192f |
@@ -1,9 +0,0 @@
|
||||
---
|
||||
trigger: manual
|
||||
---
|
||||
# Context Restriction: No VTK
|
||||
When this rule is active, restrict the operational context to libraries excluding the VTK layer.
|
||||
- **Exclude Path**: `src/Vtk`
|
||||
- **Include Paths**: `src/Core`, `src/Math`, `src/HEP`, `src/Root`, `src/Python`, `src/utils`
|
||||
- **Focus**: Tomographic reconstruction algorithms, data structures in `Core`, and physical modeling in `HEP`.
|
||||
- **Constraint**: Avoid referencing `Prop3D`, `Viewport`, or any VTK-specific headers unless the user overrides this restriction.
|
||||
@@ -1,9 +0,0 @@
|
||||
---
|
||||
trigger: manual
|
||||
---
|
||||
# Context Focus: gcompose Application
|
||||
When this rule is active, prioritize the `gcompose` GUI application.
|
||||
- **Primary Path**: `app/gcompose`
|
||||
- **Focus**: `MainPanel`, `ContextPanel`, `PropertiesPanel`, and `ViewportPane`.
|
||||
- **Integration**: Wiring of Qt signals/slots between the `uLib` core model and the GUI widgets.
|
||||
- **Dependency**: Reference `src/Vtk` and `src/Core` as the underlying framework for the application.
|
||||
@@ -1,9 +0,0 @@
|
||||
---
|
||||
trigger: always_on
|
||||
---
|
||||
# Context Inclusion: VTK
|
||||
When this rule is active, include the VTK visualization layer in the operational context.
|
||||
- **Priority Path**: `src/Vtk`
|
||||
- **Focus**: `Prop3D` hierarchy, `Viewport` management, and the synchronization between domain objects and VTK props.
|
||||
- **Key Classes**: `vtkViewport`, `vtkQViewport`, `vtkObjectsContext`, and all classes in `src/Vtk/HEP/Geant`.
|
||||
- **Logic**: Ensure transformations (TRS) applied to domain objects are correctly mirrored in the visualization layer and vice versa.
|
||||
7
.agents/rules/micromamba.md
Normal file
7
.agents/rules/micromamba.md
Normal file
@@ -0,0 +1,7 @@
|
||||
---
|
||||
trigger: always_on
|
||||
---
|
||||
|
||||
build in build directory using always micromamba "mutom" env.
|
||||
build with make flag -j$(nproc).
|
||||
|
||||
@@ -1,39 +0,0 @@
|
||||
# Skill: Build uLib with Micromamba
|
||||
|
||||
This skill provides instructions for building the uLib project using the micromamba environment.
|
||||
|
||||
## Context
|
||||
- **Environment**: micromamba `uLib`
|
||||
- **Output Directory**: `build`
|
||||
- **CPU Usage**: All available cores
|
||||
|
||||
## Instructions
|
||||
|
||||
1. **Environment Setup**:
|
||||
Ensure micromamba is properly initialized and the `uLib` environment is active.
|
||||
```bash
|
||||
export MAMBA_EXE="/home/share/micromamba/bin/micromamba"
|
||||
export MAMBA_ROOT_PREFIX="/home/share/micromamba"
|
||||
eval "$(/home/share/micromamba/bin/micromamba shell hook --shell bash)"
|
||||
micromamba activate uLib
|
||||
```
|
||||
|
||||
2. **Full Rebuild (if needed)**:
|
||||
If the `build` directory does not exist or a full reconfiguration is required:
|
||||
```bash
|
||||
conan profile detect --force
|
||||
conan install . --output-folder=build --build=missing
|
||||
cmake --preset conan-release
|
||||
```
|
||||
|
||||
3. **Incremental Build**:
|
||||
Run the build command from the root directory, pointing to the `build` folder and using all cores.
|
||||
```bash
|
||||
cmake --build build -j$(nproc)
|
||||
```
|
||||
|
||||
4. **Specific Target Build**:
|
||||
To build a specific target (e.g., gcompose):
|
||||
```bash
|
||||
cmake --build build --target gcompose -j$(nproc)
|
||||
```
|
||||
2
.gitignore
vendored
2
.gitignore
vendored
@@ -18,5 +18,3 @@ test_props.xml
|
||||
test_props2.xml
|
||||
test_boost.cpp
|
||||
.claude/settings.json
|
||||
build_output.log
|
||||
configure_output.log
|
||||
|
||||
3
.vscode/gdb_wrapper.sh
vendored
3
.vscode/gdb_wrapper.sh
vendored
@@ -1,3 +0,0 @@
|
||||
#!/bin/bash
|
||||
export DISPLAY=:1001.0
|
||||
/home/share/micromamba/bin/micromamba run -n uLib /usr/bin/gdb "$@"
|
||||
31
.vscode/launch.json
vendored
31
.vscode/launch.json
vendored
@@ -1,31 +0,0 @@
|
||||
{
|
||||
// Use IntelliSense to learn about possible attributes.
|
||||
// Hover to view descriptions of existing attributes.
|
||||
// For more information, visit: https://go.microsoft.com/fwlink/?linkid=830387
|
||||
"version": "0.2.0",
|
||||
"configurations": [
|
||||
{
|
||||
"name": "gcompose",
|
||||
"type": "cppdbg",
|
||||
"request": "launch",
|
||||
"program": "${workspaceFolder}/build/app/gcompose/gcompose",
|
||||
"args": [],
|
||||
"stopAtEntry": false,
|
||||
"cwd": "${workspaceFolder}",
|
||||
"environment": [
|
||||
{ "name": "DISPLAY", "value": ":1001.0" }
|
||||
],
|
||||
"externalConsole": false,
|
||||
"MIMode": "gdb",
|
||||
"setupCommands": [
|
||||
{
|
||||
"description": "Enable pretty-printing for gdb",
|
||||
"text": "-enable-pretty-printing",
|
||||
"ignoreFailures": true
|
||||
}
|
||||
],
|
||||
"preLaunchTask": "Build gcompose",
|
||||
"miDebuggerPath": "${workspaceFolder}/.vscode/gdb_wrapper.sh"
|
||||
}
|
||||
]
|
||||
}
|
||||
17
.vscode/tasks.json
vendored
17
.vscode/tasks.json
vendored
@@ -1,17 +0,0 @@
|
||||
{
|
||||
"version": "2.0.0",
|
||||
"tasks": [
|
||||
{
|
||||
"label": "Build gcompose",
|
||||
"type": "shell",
|
||||
"command": "/home/share/micromamba/bin/micromamba run -n uLib cmake --build build --target gcompose -j$(nproc)",
|
||||
"group": {
|
||||
"kind": "build",
|
||||
"isDefault": true
|
||||
},
|
||||
"problemMatcher": [
|
||||
"$gcc"
|
||||
]
|
||||
}
|
||||
]
|
||||
}
|
||||
26
CLAUDE.md
26
CLAUDE.md
@@ -10,7 +10,7 @@ This file provides guidance to Claude Code (claude.ai/code) when working with co
|
||||
export MAMBA_EXE="/home/share/micromamba/bin/micromamba"
|
||||
export MAMBA_ROOT_PREFIX="/home/share/micromamba"
|
||||
eval "$(/home/share/micromamba/bin/micromamba shell hook --shell bash)"
|
||||
micromamba activate uLib
|
||||
micromamba activate mutom
|
||||
|
||||
# Configure (from repo root, using Conan preset — uses Ninja + ccache)
|
||||
cmake --preset conan-release
|
||||
@@ -71,32 +71,32 @@ mutomCore → mutomMath → mutomDetectors → mutomGeant
|
||||
- `ObjectsContext` is a container owning a list of `Object*` pointers; signals `ObjectAdded`/`ObjectRemoved`
|
||||
|
||||
### VTK Layer (`src/Vtk/`)
|
||||
- `Prop3D` (inherits `uLib::Object`): wraps a VTK `vtkProp` for rendering. Has `GetContent()` returning the underlying domain object. Display-only properties are registered via `ULIB_ACTIVATE_DISPLAY_PROPERTIES` macro.
|
||||
- `Viewport`: base class managing the VTK renderer, picking, selection logic. Maintains `m_Prop3Ds` vector and `m_ObjectToProp3D` map.
|
||||
- `QViewport` (inherits `QWidget` + `Viewport`): Qt-embedded VTK widget. Emits Qt signal `prop3dSelected(Prop3D*)` on click-selection via `OnSelectionChanged`.
|
||||
- `vtkObjectsContext`: wraps `ObjectsContext`, creating/destroying `Prop3D`s as objects come/go. Emits `Prop3DAdded`/`Prop3DRemoved`.
|
||||
- Display properties: `serialize_display()` + `display_properties_archive` registers selected `hrp<T>` fields as `PropertyBase*` in the prop3d's `m_DisplayProperties`. `PropertyEditor::setObject(obj, displayOnly=true)` shows only those.
|
||||
- `Puppet` (inherits `uLib::Object`): wraps a VTK `vtkProp` for rendering. Has `GetContent()` returning the underlying domain object. Display-only properties are registered via `ULIB_ACTIVATE_DISPLAY_PROPERTIES` macro.
|
||||
- `Viewport`: base class managing the VTK renderer, picking, selection logic. Maintains `m_Puppets` vector and `m_ObjectToPuppet` map.
|
||||
- `QViewport` (inherits `QWidget` + `Viewport`): Qt-embedded VTK widget. Emits Qt signal `puppetSelected(Puppet*)` on click-selection via `OnSelectionChanged`.
|
||||
- `vtkObjectsContext`: wraps `ObjectsContext`, creating/destroying `Puppet`s as objects come/go. Emits `PuppetAdded`/`PuppetRemoved`.
|
||||
- Display properties: `serialize_display()` + `display_properties_archive` registers selected `hrp<T>` fields as `PropertyBase*` in the puppet's `m_DisplayProperties`. `PropertyEditor::setObject(obj, displayOnly=true)` shows only those.
|
||||
|
||||
### gcompose GUI App (`app/gcompose/src/`)
|
||||
- `MainPanel`: top-level widget. Owns `ContextPanel` (left) and `ViewportPane` (right). Wires together viewport↔context selection via signals.
|
||||
- `ContextPanel`: tree view of `ObjectsContext`. Emits `objectSelected(Object*)`. Contains an embedded `PropertiesPanel`.
|
||||
- `PropertiesPanel`: shows `uLib::Object` properties via `PropertyEditor`.
|
||||
- `ViewportPane`: embeds `QViewport` + a slide-out "Display Properties" panel (`PropertyEditor` in display-only mode).
|
||||
- `PropertyEditor`: populates widgets from `Object::GetProperties()` (all) or `Prop3D::GetDisplayProperties()` (display-only mode).
|
||||
- `PropertyEditor`: populates widgets from `Object::GetProperties()` (all) or `Puppet::GetDisplayProperties()` (display-only mode).
|
||||
|
||||
### Selection Sync Flow
|
||||
```
|
||||
Viewport click → Viewport::SelectProp3D() → QViewport::OnSelectionChanged()
|
||||
→ emit prop3dSelected(p)
|
||||
Viewport click → Viewport::SelectPuppet() → QViewport::OnSelectionChanged()
|
||||
→ emit puppetSelected(p)
|
||||
→ MainPanel: contextPanel->selectObject(p->GetContent()) [updates tree + PropertiesPanel]
|
||||
→ MainPanel: firstPane->setObject(p) [updates Display Properties panel]
|
||||
|
||||
ContextPanel tree click → emit objectSelected(obj)
|
||||
→ MainPanel: viewport->SelectProp3D(prop3d) [visual selection in VTK]
|
||||
→ MainPanel: firstPane->setObject(prop3d) [updates Display Properties panel]
|
||||
→ MainPanel: viewport->SelectPuppet(puppet) [visual selection in VTK]
|
||||
→ MainPanel: firstPane->setObject(puppet) [updates Display Properties panel]
|
||||
```
|
||||
|
||||
### Key Patterns
|
||||
- **Two signal systems coexist**: Qt signals (`Q_OBJECT`, `connect(...)`) for GUI; `uLib::Object::connect(...)` for domain signals.
|
||||
- **Display properties** flow: `Prop3D::serialize_display()` → `display_properties_archive` → `RegisterDisplayProperty()` → `PropertyEditor(displayOnly=true)`. Must call `ULIB_ACTIVATE_DISPLAY_PROPERTIES` in the prop3d constructor.
|
||||
- **Prop3D ↔ Object map**: `Viewport::m_ObjectToProp3D` allows lookup by domain object; `vtkObjectsContext::GetProp3D(obj)` does the same.
|
||||
- **Display properties** flow: `Puppet::serialize_display()` → `display_properties_archive` → `RegisterDisplayProperty()` → `PropertyEditor(displayOnly=true)`. Must call `ULIB_ACTIVATE_DISPLAY_PROPERTIES` in the puppet constructor.
|
||||
- **Puppet ↔ Object map**: `Viewport::m_ObjectToPuppet` allows lookup by domain object; `vtkObjectsContext::GetPuppet(obj)` does the same.
|
||||
|
||||
@@ -15,32 +15,11 @@ if(POLICY CMP0167)
|
||||
cmake_policy(SET CMP0167 NEW)
|
||||
endif()
|
||||
|
||||
|
||||
## -------------------------------------------------------------------------- ##
|
||||
|
||||
project(uLib)
|
||||
|
||||
option(ULIB_USE_CCACHE "Use ccache for build acceleration" ON)
|
||||
if(ULIB_USE_CCACHE)
|
||||
find_program(CCACHE_PROGRAM ccache)
|
||||
if(CCACHE_PROGRAM)
|
||||
set(CMAKE_CXX_COMPILER_LAUNCHER "${CCACHE_PROGRAM}")
|
||||
set(CMAKE_C_COMPILER_LAUNCHER "${CCACHE_PROGRAM}")
|
||||
endif()
|
||||
else()
|
||||
set(CMAKE_CXX_COMPILER_LAUNCHER "")
|
||||
set(CMAKE_C_COMPILER_LAUNCHER "")
|
||||
endif()
|
||||
|
||||
# Applica la flag SOLO se il compilatore è GCC
|
||||
if(CMAKE_CXX_COMPILER_ID STREQUAL "GNU")
|
||||
add_compile_options(-fno-merge-constants)
|
||||
endif()
|
||||
|
||||
# Disabilita il warning se il compilatore è Clang (o AppleClang)
|
||||
if(CMAKE_CXX_COMPILER_ID MATCHES "Clang")
|
||||
add_compile_options(-Wno-ignored-optimization-argument)
|
||||
endif()
|
||||
|
||||
# CUDA Toolkit seems to be missing locally. Toggle ON if nvcc is made available.
|
||||
option(USE_CUDA "Enable CUDA support" OFF)
|
||||
if(USE_CUDA)
|
||||
@@ -142,13 +121,8 @@ find_package(Boost 1.45.0 COMPONENTS program_options serialization unit_test_fra
|
||||
include_directories(${Boost_INCLUDE_DIRS})
|
||||
|
||||
find_package(Eigen3 CONFIG REQUIRED)
|
||||
# if(NOT EIGEN3_INCLUDE_DIRS)
|
||||
# get_target_property(EIGEN3_INCLUDE_DIRS Eigen3::Eigen INTERFACE_INCLUDE_DIRECTORIES)
|
||||
# else()
|
||||
# include_directories(${EIGEN3_INCLUDE_DIRS})
|
||||
# endif()
|
||||
|
||||
find_package(OpenMP)
|
||||
get_target_property(EIGEN3_INCLUDE_DIRS Eigen3::Eigen INTERFACE_INCLUDE_DIRECTORIES)
|
||||
include_directories(${EIGEN3_INCLUDE_DIRS})
|
||||
|
||||
find_package(ROOT CONFIG REQUIRED)
|
||||
include(${ROOT_USE_FILE})
|
||||
@@ -171,8 +145,6 @@ else()
|
||||
IOXML
|
||||
IOXMLParser
|
||||
ImagingCore
|
||||
ImagingHybrid
|
||||
ImagingSources
|
||||
InteractionStyle
|
||||
InteractionWidgets
|
||||
RenderingAnnotation
|
||||
|
||||
20
README.md
20
README.md
@@ -30,13 +30,13 @@ You can create and activate the environment using either `micromamba` or `conda`
|
||||
**Using Micromamba:**
|
||||
```bash
|
||||
micromamba env create -f condaenv.yml
|
||||
micromamba activate uLib
|
||||
micromamba activate mutom
|
||||
```
|
||||
|
||||
**Using Conda:**
|
||||
```bash
|
||||
conda env create -f condaenv.yml
|
||||
conda activate uLib
|
||||
conda activate mutom
|
||||
```
|
||||
|
||||
### Configure and Build
|
||||
@@ -65,24 +65,12 @@ cmake --preset conan-release
|
||||
cmake --build build -j$(nproc)
|
||||
```
|
||||
|
||||
5. **Clean build (wipe and rebuild everything):**
|
||||
```bash
|
||||
cmake --build build --clean-first -j$(nproc)
|
||||
```
|
||||
|
||||
6. **Run tests:**
|
||||
```bash
|
||||
cmake --build build --target test -j$(nproc)
|
||||
# or equivalently:
|
||||
ctest --test-dir build --output-on-failure -j$(nproc)
|
||||
```
|
||||
|
||||
#### LLVM/Clang build (clang + lld + ccache — fastest)
|
||||
|
||||
A `fast` conan profile is provided that uses **clang**, **lld** (LLVM linker), and **ccache**. Install them into your environment first:
|
||||
|
||||
```bash
|
||||
micromamba install -n uLib -y clang clangxx lld -c conda-forge
|
||||
micromamba install -n mutom -y clang clangxx lld -c conda-forge
|
||||
```
|
||||
|
||||
Then build using the `fast` profile:
|
||||
@@ -103,6 +91,6 @@ The `fast` profile is defined at `~/.conan2/profiles/fast` and sets:
|
||||
### Make python package
|
||||
|
||||
```bash
|
||||
micromamba run -n uLib env USE_CUDA=ON poetry install
|
||||
micromamba run -n mutom env USE_CUDA=ON poetry install
|
||||
```
|
||||
|
||||
|
||||
@@ -17,8 +17,6 @@ add_executable(gcompose
|
||||
src/PropertyWidgets.cpp
|
||||
src/PropertiesPanel.h
|
||||
src/PropertiesPanel.cpp
|
||||
src/PreferencesDialog.h
|
||||
src/PreferencesDialog.cpp
|
||||
)
|
||||
|
||||
set_target_properties(gcompose PROPERTIES
|
||||
|
||||
@@ -10,7 +10,6 @@
|
||||
#include <QList>
|
||||
#include <QShortcut>
|
||||
#include <QItemSelectionModel>
|
||||
#include <functional>
|
||||
|
||||
ContextPanel::ContextPanel(QWidget* parent)
|
||||
: QWidget(parent)
|
||||
@@ -58,8 +57,6 @@ ContextPanel::ContextPanel(QWidget* parent)
|
||||
m_splitter->setSizes(sizes);
|
||||
|
||||
m_layout->addWidget(m_splitter);
|
||||
|
||||
connect(m_propertiesPanel, &PropertiesPanel::propertyUpdated, this, &ContextPanel::propertyUpdated);
|
||||
|
||||
connect(m_treeView->selectionModel(), &QItemSelectionModel::selectionChanged,
|
||||
this, &ContextPanel::onSelectionChanged);
|
||||
@@ -106,34 +103,15 @@ void ContextPanel::selectObject(uLib::Object* obj) {
|
||||
return;
|
||||
}
|
||||
|
||||
// Recursive search helper
|
||||
std::function<QModelIndex(const QModelIndex&)> findIdx = [&](const QModelIndex& parent) -> QModelIndex {
|
||||
for (int i = 0; i < m_model->rowCount(parent); ++i) {
|
||||
QModelIndex idx = m_model->index(i, 0, parent);
|
||||
if (idx.internalPointer() == obj) return idx;
|
||||
|
||||
if (m_model->rowCount(idx) > 0) {
|
||||
QModelIndex childIdx = findIdx(idx);
|
||||
if (childIdx.isValid()) return childIdx;
|
||||
}
|
||||
for (int i = 0; i < m_model->rowCount(); ++i) {
|
||||
QModelIndex idx = m_model->index(i, 0);
|
||||
if (idx.internalPointer() == obj) {
|
||||
QSignalBlocker blocker(m_treeView->selectionModel());
|
||||
m_treeView->selectionModel()->select(idx, QItemSelectionModel::ClearAndSelect | QItemSelectionModel::Rows);
|
||||
m_treeView->scrollTo(idx);
|
||||
m_propertiesPanel->setObject(obj); // Explicitly update properties too
|
||||
return;
|
||||
}
|
||||
return QModelIndex();
|
||||
};
|
||||
|
||||
QModelIndex targetIdx = findIdx(QModelIndex());
|
||||
if (targetIdx.isValid()) {
|
||||
QSignalBlocker blocker(m_treeView->selectionModel());
|
||||
|
||||
// Expand parents so the selection is visible
|
||||
QModelIndex p = targetIdx.parent();
|
||||
while (p.isValid()) {
|
||||
m_treeView->expand(p);
|
||||
p = p.parent();
|
||||
}
|
||||
|
||||
m_treeView->selectionModel()->select(targetIdx, QItemSelectionModel::ClearAndSelect | QItemSelectionModel::Rows);
|
||||
m_treeView->scrollTo(targetIdx);
|
||||
m_propertiesPanel->setObject(obj);
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
@@ -25,7 +25,6 @@ public:
|
||||
|
||||
signals:
|
||||
void objectSelected(uLib::Object* obj);
|
||||
void propertyUpdated();
|
||||
|
||||
private slots:
|
||||
void onSelectionChanged(const QItemSelection& selected, const QItemSelection& deselected);
|
||||
|
||||
@@ -6,8 +6,6 @@
|
||||
#include "Core/ObjectsContext.h"
|
||||
#include "Vtk/vtkObjectsContext.h"
|
||||
#include "Vtk/vtkQViewport.h"
|
||||
#include "Vtk/vtkViewportProperties.h"
|
||||
#include <Vtk/uLibVtkInterface.h>
|
||||
#include <QVBoxLayout>
|
||||
#include <QHBoxLayout>
|
||||
#include <QSplitter>
|
||||
@@ -15,16 +13,13 @@
|
||||
#include <QPushButton>
|
||||
#include <QMenu>
|
||||
#include <QAction>
|
||||
#include <QShortcut>
|
||||
#include <QApplication>
|
||||
#include <QFileDialog>
|
||||
#include <QFileInfo>
|
||||
#include "StyleManager.h"
|
||||
#include "Math/VoxImage.h"
|
||||
#include "PreferencesDialog.h"
|
||||
#include "Settings.h"
|
||||
|
||||
MainPanel::MainPanel(QWidget* parent) : QWidget(parent), m_context(nullptr), m_mainVtkContext(nullptr), m_viewportProps(nullptr) {
|
||||
MainPanel::MainPanel(QWidget* parent) : QWidget(parent), m_context(nullptr), m_mainVtkContext(nullptr) {
|
||||
this->setObjectName("MainPanel");
|
||||
this->setAttribute(Qt::WA_StyledBackground);
|
||||
auto* mainLayout = new QVBoxLayout(this);
|
||||
@@ -50,12 +45,17 @@ MainPanel::MainPanel(QWidget* parent) : QWidget(parent), m_context(nullptr), m_m
|
||||
fileMenu->addAction("Open", this, &MainPanel::onOpen);
|
||||
fileMenu->addAction("Save", this, &MainPanel::onSave);
|
||||
fileMenu->addAction("Save As", this, &MainPanel::onSaveAs);
|
||||
fileMenu->addSeparator();
|
||||
fileMenu->addAction("Preferences", this, &MainPanel::onPreferences);
|
||||
fileMenu->addSeparator();
|
||||
fileMenu->addAction("Exit", this, &MainPanel::onExit);
|
||||
btnFile->setMenu(fileMenu);
|
||||
|
||||
// Theme Menu Button
|
||||
auto* btnTheme = new QPushButton("Theme", menuPanel);
|
||||
btnTheme->setObjectName("MenuButton");
|
||||
auto* themeMenu = new QMenu(btnTheme);
|
||||
themeMenu->addAction("Dark", this, &MainPanel::onDarkTheme);
|
||||
themeMenu->addAction("Bright", this, &MainPanel::onBrightTheme);
|
||||
btnTheme->setMenu(themeMenu);
|
||||
|
||||
// New Menu Button
|
||||
auto* btnNew = new QPushButton("Add", menuPanel);
|
||||
btnNew->setObjectName("MenuButton");
|
||||
@@ -73,6 +73,7 @@ MainPanel::MainPanel(QWidget* parent) : QWidget(parent), m_context(nullptr), m_m
|
||||
menuLayout->addWidget(logo);
|
||||
menuLayout->addWidget(btnFile);
|
||||
menuLayout->addWidget(btnNew);
|
||||
menuLayout->addWidget(btnTheme);
|
||||
menuLayout->addStretch();
|
||||
|
||||
mainLayout->addWidget(menuPanel);
|
||||
@@ -89,24 +90,17 @@ MainPanel::MainPanel(QWidget* parent) : QWidget(parent), m_context(nullptr), m_m
|
||||
|
||||
connect(m_contextPanel, &ContextPanel::objectSelected, [this](uLib::Object* obj) {
|
||||
if (auto* viewport = qobject_cast<uLib::Vtk::QViewport*>(m_firstPane->currentViewport())) {
|
||||
uLib::Vtk::Prop3D* prop3d = nullptr;
|
||||
uLib::Vtk::Puppet* puppet = nullptr;
|
||||
if (m_mainVtkContext) {
|
||||
prop3d = m_mainVtkContext->GetProp3D(obj);
|
||||
puppet = m_mainVtkContext->GetPuppet(obj);
|
||||
}
|
||||
viewport->SelectProp3D(prop3d);
|
||||
// Update the display properties in the viewport pane itself - use the prop3d proxy if possible
|
||||
m_firstPane->setObject(prop3d ? (uLib::Object*)prop3d : obj);
|
||||
viewport->SelectPuppet(puppet);
|
||||
// Update the display properties in the viewport pane itself - use the puppet proxy if possible
|
||||
m_firstPane->setObject(puppet ? (uLib::Object*)puppet : obj);
|
||||
} else {
|
||||
m_firstPane->setObject(obj);
|
||||
}
|
||||
});
|
||||
|
||||
connect(m_contextPanel, &ContextPanel::propertyUpdated, [this](){
|
||||
auto viewports = this->findChildren<uLib::Vtk::QViewport*>();
|
||||
for (auto* vp : viewports) {
|
||||
vp->Render();
|
||||
}
|
||||
});
|
||||
|
||||
// Set initial sizes: Context(250), Viewport(600), Properties(250)
|
||||
QList<int> sizes;
|
||||
@@ -114,14 +108,6 @@ MainPanel::MainPanel(QWidget* parent) : QWidget(parent), m_context(nullptr), m_m
|
||||
m_rootSplitter->setSizes(sizes);
|
||||
|
||||
mainLayout->addWidget(m_rootSplitter, 1);
|
||||
|
||||
// Shortcuts
|
||||
auto* groupShortcut = new QShortcut(QKeySequence("Ctrl+G"), this);
|
||||
connect(groupShortcut, &QShortcut::activated, [this]() {
|
||||
if (auto* viewport = qobject_cast<uLib::Vtk::QViewport*>(m_firstPane->currentViewport())) {
|
||||
viewport->GroupSelection(m_context);
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
void MainPanel::setContext(uLib::ObjectsContext* context) {
|
||||
@@ -130,7 +116,7 @@ void MainPanel::setContext(uLib::ObjectsContext* context) {
|
||||
|
||||
if (m_mainVtkContext) {
|
||||
if (auto* viewport = qobject_cast<uLib::Vtk::QViewport*>(m_firstPane->currentViewport())) {
|
||||
viewport->RemoveProp3D(*m_mainVtkContext);
|
||||
viewport->RemovePuppet(*m_mainVtkContext);
|
||||
}
|
||||
delete m_mainVtkContext;
|
||||
m_mainVtkContext = nullptr;
|
||||
@@ -138,10 +124,10 @@ void MainPanel::setContext(uLib::ObjectsContext* context) {
|
||||
|
||||
if (context) {
|
||||
if (auto* viewport = qobject_cast<uLib::Vtk::QViewport*>(m_firstPane->currentViewport())) {
|
||||
m_mainVtkContext = new uLib::Vtk::ObjectsContext(context);
|
||||
// viewport->AddProp3D(*m_mainVtkContext); // redundant
|
||||
m_mainVtkContext = new uLib::Vtk::vtkObjectsContext(context);
|
||||
// viewport->AddPuppet(*m_mainVtkContext); // redundant
|
||||
|
||||
auto syncSelection = [this](uLib::Vtk::Prop3D* p) {
|
||||
auto syncSelection = [this](uLib::Vtk::Puppet* p) {
|
||||
if (!p) {
|
||||
m_contextPanel->clearSelection();
|
||||
m_firstPane->setObject(nullptr);
|
||||
@@ -150,14 +136,14 @@ void MainPanel::setContext(uLib::ObjectsContext* context) {
|
||||
m_firstPane->setObject(p);
|
||||
}
|
||||
};
|
||||
connect(viewport, &uLib::Vtk::QViewport::prop3dSelected, syncSelection);
|
||||
connect(viewport, &uLib::Vtk::QViewport::puppetSelected, syncSelection);
|
||||
|
||||
uLib::Object::connect(m_mainVtkContext, &uLib::Vtk::ObjectsContext::Prop3DAdded, [this](uLib::Vtk::Prop3D* p) {
|
||||
uLib::Object::connect(m_mainVtkContext, &uLib::Vtk::vtkObjectsContext::PuppetAdded, [this](uLib::Vtk::Puppet* p) {
|
||||
if (p) {
|
||||
auto panes = this->findChildren<ViewportPane*>();
|
||||
for (auto* pane : panes) {
|
||||
if (auto* vp = qobject_cast<uLib::Vtk::QViewport*>(pane->currentViewport())) {
|
||||
vp->AddProp3D(*p);
|
||||
vp->AddPuppet(*p);
|
||||
vp->ZoomAuto();
|
||||
vp->Render();
|
||||
}
|
||||
@@ -165,25 +151,25 @@ void MainPanel::setContext(uLib::ObjectsContext* context) {
|
||||
}
|
||||
});
|
||||
|
||||
uLib::Object::connect(m_mainVtkContext, &uLib::Vtk::ObjectsContext::Prop3DRemoved, [this](uLib::Vtk::Prop3D* p) {
|
||||
uLib::Object::connect(m_mainVtkContext, &uLib::Vtk::vtkObjectsContext::PuppetRemoved, [this](uLib::Vtk::Puppet* p) {
|
||||
if (p) {
|
||||
auto panes = this->findChildren<ViewportPane*>();
|
||||
for (auto* pane : panes) {
|
||||
if (auto* vp = qobject_cast<uLib::Vtk::QViewport*>(pane->currentViewport())) {
|
||||
vp->RemoveProp3D(*p);
|
||||
vp->RemovePuppet(*p);
|
||||
vp->Render();
|
||||
}
|
||||
}
|
||||
}
|
||||
});
|
||||
|
||||
// Add any prop3ds that were created during m_mainVtkContext's construction to all panes
|
||||
// Add any puppets that were created during m_mainVtkContext's construction to all panes
|
||||
auto panes = this->findChildren<ViewportPane*>();
|
||||
for (auto* obj : context->GetObjects()) {
|
||||
if (auto* p = m_mainVtkContext->GetProp3D(obj)) {
|
||||
if (auto* p = m_mainVtkContext->GetPuppet(obj)) {
|
||||
for (auto* pane : panes) {
|
||||
if (auto* vp = qobject_cast<uLib::Vtk::QViewport*>(pane->currentViewport())) {
|
||||
vp->AddProp3D(*p);
|
||||
vp->AddPuppet(*p);
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -250,27 +236,12 @@ void MainPanel::onExit() {
|
||||
qApp->quit();
|
||||
}
|
||||
|
||||
void MainPanel::onPreferences() {
|
||||
uLib::Qt::PreferencesDialog dlg(this);
|
||||
if (dlg.exec() == QDialog::Accepted) {
|
||||
// Apply theme and GUI font
|
||||
auto theme = uLib::Qt::Settings::Instance().GetTheme();
|
||||
auto guiFont = uLib::Qt::Settings::Instance().GetGuiFont();
|
||||
StyleManager::applyStyle(qApp, theme == uLib::Qt::Settings::Dark ? "dark" : "bright", guiFont);
|
||||
void MainPanel::onDarkTheme() {
|
||||
StyleManager::applyStyle(qApp, "dark");
|
||||
}
|
||||
|
||||
// Apply rendering and font preferences to all viewports
|
||||
bool throttled = uLib::Qt::Settings::Instance().GetThrottledRendering();
|
||||
auto font = uLib::Qt::Settings::Instance().GetFont();
|
||||
auto fontColor = uLib::Qt::Settings::Instance().GetFontColor();
|
||||
|
||||
auto viewports = this->findChildren<uLib::Vtk::QViewport*>();
|
||||
for (auto* vp : viewports) {
|
||||
vp->SetThrottledRendering(throttled);
|
||||
vp->SetFont(font);
|
||||
vp->SetFontColor(fontColor);
|
||||
vp->Render();
|
||||
}
|
||||
}
|
||||
void MainPanel::onBrightTheme() {
|
||||
StyleManager::applyStyle(qApp, "bright");
|
||||
}
|
||||
|
||||
MainPanel::~MainPanel() {}
|
||||
|
||||
@@ -11,8 +11,7 @@ class PropertiesPanel;
|
||||
namespace uLib {
|
||||
class ObjectsContext;
|
||||
namespace Vtk {
|
||||
class ObjectsContext;
|
||||
class ViewportProperties;
|
||||
class vtkObjectsContext;
|
||||
}
|
||||
}
|
||||
|
||||
@@ -31,7 +30,8 @@ private slots:
|
||||
void onSaveAs();
|
||||
void onExit();
|
||||
|
||||
void onPreferences();
|
||||
void onDarkTheme();
|
||||
void onBrightTheme();
|
||||
|
||||
void onCreateObject(const std::string& className);
|
||||
|
||||
@@ -40,8 +40,7 @@ private:
|
||||
ViewportPane* m_firstPane;
|
||||
ContextPanel* m_contextPanel;
|
||||
uLib::ObjectsContext* m_context;
|
||||
uLib::Vtk::ObjectsContext* m_mainVtkContext;
|
||||
uLib::Vtk::ViewportProperties* m_viewportProps;
|
||||
uLib::Vtk::vtkObjectsContext* m_mainVtkContext;
|
||||
};
|
||||
|
||||
#endif // MAINPANEL_H
|
||||
|
||||
@@ -1,184 +0,0 @@
|
||||
#include "PreferencesDialog.h"
|
||||
#include <QVBoxLayout>
|
||||
#include <QHBoxLayout>
|
||||
#include <QFormLayout>
|
||||
#include <QPushButton>
|
||||
#include <QLabel>
|
||||
#include <QGroupBox>
|
||||
#include <QColorDialog>
|
||||
#include <QFormLayout>
|
||||
|
||||
namespace uLib {
|
||||
namespace Qt {
|
||||
|
||||
PreferencesDialog::PreferencesDialog(QWidget* parent) : QDialog(parent) {
|
||||
setWindowTitle("Preferences");
|
||||
setMinimumWidth(400);
|
||||
|
||||
auto* mainLayout = new QVBoxLayout(this);
|
||||
mainLayout->setSpacing(20);
|
||||
mainLayout->setContentsMargins(20, 20, 20, 20);
|
||||
|
||||
// ── General / Rendering Settings ────────────────────────────────────────
|
||||
auto* renderingGroup = new QGroupBox("Appearance & Performance", this);
|
||||
auto* renderingLayout = new QVBoxLayout(renderingGroup);
|
||||
|
||||
auto* themeLayout = new QHBoxLayout();
|
||||
themeLayout->addWidget(new QLabel("Color Theme:"));
|
||||
m_themeCombo = new QComboBox(renderingGroup);
|
||||
m_themeCombo->addItem("Dark");
|
||||
m_themeCombo->addItem("Bright");
|
||||
m_themeCombo->setCurrentIndex(Settings::Instance().GetTheme() == Settings::Dark ? 0 : 1);
|
||||
themeLayout->addWidget(m_themeCombo);
|
||||
themeLayout->addStretch();
|
||||
|
||||
renderingLayout->addLayout(themeLayout);
|
||||
renderingLayout->addSpacing(10);
|
||||
|
||||
m_throttledRendering = new QCheckBox("Enable throttled rendering (recommended for performance)", renderingGroup);
|
||||
m_throttledRendering->setChecked(Settings::Instance().GetThrottledRendering());
|
||||
m_throttledRendering->setToolTip("Limits framerate to ~60fps to reduce CPU/GPU usage.");
|
||||
|
||||
renderingLayout->addWidget(m_throttledRendering);
|
||||
mainLayout->addWidget(renderingGroup);
|
||||
|
||||
// ── Units Settings ──────────────────────────────────────────────────────
|
||||
auto* unitsGroup = new QGroupBox("Preferred Units", this);
|
||||
auto* unitsLayout = new QFormLayout(unitsGroup);
|
||||
unitsLayout->setLabelAlignment(::Qt::AlignRight);
|
||||
unitsLayout->setSpacing(10);
|
||||
|
||||
auto addUnitRow = [&](const QString& label, Settings::Dimension dim, const QStringList& units) {
|
||||
auto* combo = new QComboBox(unitsGroup);
|
||||
combo->addItems(units);
|
||||
std::string current = Settings::Instance().GetPreferredUnit(dim);
|
||||
int idx = combo->findText(QString::fromStdString(current));
|
||||
if (idx >= 0) combo->setCurrentIndex(idx);
|
||||
|
||||
unitsLayout->addRow(label + ":", combo);
|
||||
m_unitCombos[dim] = combo;
|
||||
};
|
||||
|
||||
addUnitRow("Length", Settings::Length, {"m", "cm", "mm", "um", "nm"});
|
||||
addUnitRow("Angle", Settings::Angle, {"deg", "rad"});
|
||||
addUnitRow("Energy", Settings::Energy, {"MeV", "GeV", "eV", "keV", "TeV"});
|
||||
addUnitRow("Time", Settings::Time, {"ns", "s", "ms", "us"});
|
||||
|
||||
mainLayout->addWidget(unitsGroup);
|
||||
|
||||
// ── Font Configuration ──────────────────────────────────────────────────
|
||||
auto* fontGroup = new QGroupBox("Viewport Font Configuration", this);
|
||||
auto* fontLayout = new QFormLayout(fontGroup);
|
||||
fontLayout->setLabelAlignment(::Qt::AlignRight);
|
||||
|
||||
FontConfig currentFont = Settings::Instance().GetFont();
|
||||
m_currentFontColor = Settings::Instance().GetFontColor();
|
||||
|
||||
m_fontFamilies = new QComboBox(fontGroup);
|
||||
m_fontFamilies->addItems({"Arial", "Courier", "Times"});
|
||||
m_fontFamilies->setCurrentText(QString::fromStdString(currentFont.family));
|
||||
|
||||
m_fontSize = new QSpinBox(fontGroup);
|
||||
m_fontSize->setRange(6, 72);
|
||||
m_fontSize->setValue(currentFont.size);
|
||||
|
||||
m_fontBold = new QCheckBox("Bold", fontGroup);
|
||||
m_fontBold->setChecked(currentFont.bold);
|
||||
|
||||
m_fontItalic = new QCheckBox("Italic", fontGroup);
|
||||
m_fontItalic->setChecked(currentFont.italic);
|
||||
|
||||
m_fontColorBtn = new QPushButton(fontGroup);
|
||||
m_fontColorBtn->setFixedWidth(60);
|
||||
updateFontColorButton();
|
||||
connect(m_fontColorBtn, &QPushButton::clicked, [this](){
|
||||
QColor c = QColor::fromRgbF(m_currentFontColor.x(), m_currentFontColor.y(), m_currentFontColor.z());
|
||||
QColor selected = QColorDialog::getColor(c, this, "Select Font Color");
|
||||
if (selected.isValid()) {
|
||||
m_currentFontColor = Vector3d(selected.redF(), selected.greenF(), selected.blueF());
|
||||
updateFontColorButton();
|
||||
}
|
||||
});
|
||||
|
||||
fontLayout->addRow("Family:", m_fontFamilies);
|
||||
fontLayout->addRow("Size:", m_fontSize);
|
||||
fontLayout->addRow(m_fontBold);
|
||||
fontLayout->addRow(m_fontItalic);
|
||||
fontLayout->addRow("Color:", m_fontColorBtn);
|
||||
|
||||
mainLayout->addWidget(fontGroup);
|
||||
|
||||
// ── GUI Font Configuration ──────────────────────────────────────────────
|
||||
auto* guiFontGroup = new QGroupBox("GUI Font Configuration", this);
|
||||
auto* guiFontLayout = new QFormLayout(guiFontGroup);
|
||||
guiFontLayout->setLabelAlignment(::Qt::AlignRight);
|
||||
|
||||
FontConfig currentGuiFont = Settings::Instance().GetGuiFont();
|
||||
|
||||
m_guiFontFamilies = new QComboBox(guiFontGroup);
|
||||
m_guiFontFamilies->setEditable(true);
|
||||
m_guiFontFamilies->addItems({"Inter", "Roboto", "Segoe UI", "Arial", "Ubuntu"});
|
||||
m_guiFontFamilies->setCurrentText(QString::fromStdString(currentGuiFont.family));
|
||||
|
||||
m_guiFontSize = new QSpinBox(guiFontGroup);
|
||||
m_guiFontSize->setRange(6, 48);
|
||||
m_guiFontSize->setValue(currentGuiFont.size);
|
||||
|
||||
m_guiFontBold = new QCheckBox("Bold", guiFontGroup);
|
||||
m_guiFontBold->setChecked(currentGuiFont.bold);
|
||||
|
||||
m_guiFontItalic = new QCheckBox("Italic", guiFontGroup);
|
||||
m_guiFontItalic->setChecked(currentGuiFont.italic);
|
||||
|
||||
guiFontLayout->addRow("Family:", m_guiFontFamilies);
|
||||
guiFontLayout->addRow("Size:", m_guiFontSize);
|
||||
guiFontLayout->addRow(m_guiFontBold);
|
||||
guiFontLayout->addRow(m_guiFontItalic);
|
||||
|
||||
mainLayout->addWidget(guiFontGroup);
|
||||
|
||||
mainLayout->addStretch();
|
||||
|
||||
// ── Buttons ─────────────────────────────────────────────────────────────
|
||||
auto* buttonLayout = new QHBoxLayout();
|
||||
buttonLayout->addStretch();
|
||||
|
||||
auto* btnCancel = new QPushButton("Cancel", this);
|
||||
connect(btnCancel, &QPushButton::clicked, this, &QDialog::reject);
|
||||
|
||||
auto* btnOk = new QPushButton("Apply", this);
|
||||
btnOk->setDefault(true);
|
||||
btnOk->setObjectName("DisplayToggleBtn"); // Reusing high-contrast style
|
||||
btnOk->setMinimumWidth(100);
|
||||
connect(btnOk, &QPushButton::clicked, this, &PreferencesDialog::onAccept);
|
||||
|
||||
buttonLayout->addWidget(btnCancel);
|
||||
buttonLayout->addWidget(btnOk);
|
||||
mainLayout->addLayout(buttonLayout);
|
||||
}
|
||||
|
||||
void PreferencesDialog::onAccept() {
|
||||
Settings::Instance().SetThrottledRendering(m_throttledRendering->isChecked());
|
||||
Settings::Instance().SetTheme(m_themeCombo->currentIndex() == 0 ? Settings::Dark : Settings::Bright);
|
||||
|
||||
for (auto const& pair : m_unitCombos) {
|
||||
Settings::Instance().SetPreferredUnit(pair.first, pair.second->currentText().toStdString());
|
||||
}
|
||||
|
||||
FontConfig font(m_fontFamilies->currentText().toStdString(), m_fontSize->value(), m_fontBold->isChecked(), m_fontItalic->isChecked());
|
||||
Settings::Instance().SetFont(font);
|
||||
Settings::Instance().SetFontColor(m_currentFontColor);
|
||||
|
||||
FontConfig guiFont(m_guiFontFamilies->currentText().toStdString(), m_guiFontSize->value(), m_guiFontBold->isChecked(), m_guiFontItalic->isChecked());
|
||||
Settings::Instance().SetGuiFont(guiFont);
|
||||
|
||||
accept();
|
||||
}
|
||||
|
||||
void PreferencesDialog::updateFontColorButton() {
|
||||
QColor c = QColor::fromRgbF(m_currentFontColor.x(), m_currentFontColor.y(), m_currentFontColor.z());
|
||||
m_fontColorBtn->setStyleSheet(QString("background-color: %1; border: 1px solid #555; height: 18px;").arg(c.name()));
|
||||
}
|
||||
|
||||
} // namespace Qt
|
||||
} // namespace uLib
|
||||
@@ -1,49 +0,0 @@
|
||||
#ifndef GCOMPOSE_PREFERENCESDIALOG_H
|
||||
#define GCOMPOSE_PREFERENCESDIALOG_H
|
||||
|
||||
#include <QDialog>
|
||||
#include <QCheckBox>
|
||||
#include <QComboBox>
|
||||
#include <QSpinBox>
|
||||
#include <QPushButton>
|
||||
#include <map>
|
||||
#include <string>
|
||||
#include "Settings.h"
|
||||
|
||||
namespace uLib {
|
||||
namespace Qt {
|
||||
|
||||
class PreferencesDialog : public QDialog {
|
||||
Q_OBJECT
|
||||
public:
|
||||
explicit PreferencesDialog(QWidget* parent = nullptr);
|
||||
|
||||
private slots:
|
||||
void onAccept();
|
||||
|
||||
private:
|
||||
QCheckBox* m_throttledRendering;
|
||||
QComboBox* m_themeCombo;
|
||||
std::map<Settings::Dimension, QComboBox*> m_unitCombos;
|
||||
|
||||
// Font Configuration
|
||||
QComboBox* m_fontFamilies;
|
||||
QSpinBox* m_fontSize;
|
||||
QCheckBox* m_fontBold;
|
||||
QCheckBox* m_fontItalic;
|
||||
QPushButton* m_fontColorBtn;
|
||||
Vector3d m_currentFontColor;
|
||||
|
||||
// GUI Font Configuration
|
||||
QComboBox* m_guiFontFamilies;
|
||||
QSpinBox* m_guiFontSize;
|
||||
QCheckBox* m_guiFontBold;
|
||||
QCheckBox* m_guiFontItalic;
|
||||
|
||||
void updateFontColorButton();
|
||||
};
|
||||
|
||||
} // namespace Qt
|
||||
} // namespace uLib
|
||||
|
||||
#endif
|
||||
@@ -30,10 +30,6 @@ PropertiesPanel::PropertiesPanel(QWidget* parent) : QWidget(parent) {
|
||||
// Editor
|
||||
m_editor = new uLib::Qt::PropertyEditor(this);
|
||||
m_layout->addWidget(m_editor, 1);
|
||||
|
||||
connect(m_editor, &uLib::Qt::PropertyEditor::propertyUpdated, [this](uLib::PropertyBase*){
|
||||
emit propertyUpdated();
|
||||
});
|
||||
}
|
||||
|
||||
void PropertiesPanel::setObject(uLib::Object* obj) {
|
||||
|
||||
@@ -24,9 +24,6 @@ public:
|
||||
/** @brief Sets the object to be inspected. */
|
||||
void setObject(uLib::Object* obj);
|
||||
|
||||
signals:
|
||||
void propertyUpdated();
|
||||
|
||||
private:
|
||||
QVBoxLayout* m_layout;
|
||||
QWidget* m_titleBar;
|
||||
|
||||
@@ -11,7 +11,6 @@
|
||||
#include <QColorDialog>
|
||||
#include <QFrame>
|
||||
#include <QSlider>
|
||||
#include <QFontDialog>
|
||||
#include "Settings.h"
|
||||
|
||||
namespace uLib {
|
||||
@@ -152,7 +151,7 @@ DoublePropertyWidget::DoublePropertyWidget(Property<double>* prop, QWidget* pare
|
||||
}
|
||||
m_Edit->setValue(prop->Get());
|
||||
m_Layout->addWidget(m_Edit, 1);
|
||||
connect(m_Edit, &UnitLineEdit::valueManualChanged, [this](double val){ m_Prop->Set(val); emit updated(); });
|
||||
connect(m_Edit, &UnitLineEdit::valueManualChanged, [this](double val){ m_Prop->Set(val); });
|
||||
m_Connection = uLib::Object::connect(m_Prop, &Property<double>::Updated, [this](){
|
||||
m_Edit->setValue(m_Prop->Get());
|
||||
});
|
||||
@@ -170,7 +169,7 @@ FloatPropertyWidget::FloatPropertyWidget(Property<float>* prop, QWidget* parent)
|
||||
}
|
||||
m_Edit->setValue(prop->Get());
|
||||
m_Layout->addWidget(m_Edit, 1);
|
||||
connect(m_Edit, &UnitLineEdit::valueManualChanged, [this](double val){ m_Prop->Set((float)val); emit updated(); });
|
||||
connect(m_Edit, &UnitLineEdit::valueManualChanged, [this](double val){ m_Prop->Set((float)val); });
|
||||
m_Connection = uLib::Object::connect(m_Prop, &Property<float>::Updated, [this](){
|
||||
m_Edit->setValue((double)m_Prop->Get());
|
||||
});
|
||||
@@ -189,7 +188,7 @@ IntPropertyWidget::IntPropertyWidget(Property<int>* prop, QWidget* parent)
|
||||
}
|
||||
m_Edit->setValue(prop->Get());
|
||||
m_Layout->addWidget(m_Edit, 1);
|
||||
connect(m_Edit, &UnitLineEdit::valueManualChanged, [this](double val){ m_Prop->Set((int)val); emit updated(); });
|
||||
connect(m_Edit, &UnitLineEdit::valueManualChanged, [this](double val){ m_Prop->Set((int)val); });
|
||||
m_Connection = uLib::Object::connect(m_Prop, &Property<int>::Updated, [this](){
|
||||
m_Edit->setValue((double)m_Prop->Get());
|
||||
});
|
||||
@@ -200,7 +199,7 @@ BoolPropertyWidget::BoolPropertyWidget(Property<bool>* prop, QWidget* parent)
|
||||
m_CheckBox = new QCheckBox(this);
|
||||
m_CheckBox->setChecked(prop->Get());
|
||||
m_Layout->addWidget(m_CheckBox, 1);
|
||||
connect(m_CheckBox, &QCheckBox::toggled, [this](bool val){ if (m_Prop->Get() != val) { m_Prop->Set(val); emit updated(); } });
|
||||
connect(m_CheckBox, &QCheckBox::toggled, [this](bool val){ if (m_Prop->Get() != val) m_Prop->Set(val); });
|
||||
m_Connection = uLib::Object::connect(m_Prop, &Property<bool>::Updated, [this](){
|
||||
if (m_CheckBox->isChecked() != m_Prop->Get()) {
|
||||
QSignalBlocker blocker(m_CheckBox);
|
||||
@@ -223,7 +222,7 @@ RangePropertyWidget::RangePropertyWidget(Property<double>* prop, QWidget* parent
|
||||
m_Layout->addWidget(m_Edit, 0);
|
||||
|
||||
connect(m_Slider, &QSlider::valueChanged, this, &RangePropertyWidget::onSliderChanged);
|
||||
connect(m_Edit, &UnitLineEdit::valueManualChanged, [this](double val){ m_Prop->Set(val); emit updated(); });
|
||||
connect(m_Edit, &UnitLineEdit::valueManualChanged, [this](double val){ m_Prop->Set(val); });
|
||||
|
||||
m_Connection = uLib::Object::connect(m_Prop, &Property<double>::Updated, [this](){
|
||||
this->updateUi();
|
||||
@@ -245,7 +244,6 @@ void RangePropertyWidget::updateUi() {
|
||||
void RangePropertyWidget::onSliderChanged(int val) {
|
||||
double realVal = m_Prop->GetMin() + (val / 100.0) * (m_Prop->GetMax() - m_Prop->GetMin());
|
||||
m_Prop->Set(realVal);
|
||||
emit updated();
|
||||
}
|
||||
|
||||
ColorPropertyWidget::ColorPropertyWidget(Property<Vector3d>* prop, QWidget* parent)
|
||||
@@ -278,7 +276,6 @@ void ColorPropertyWidget::onClicked() {
|
||||
QColor selected = QColorDialog::getColor(current, this, "Select Color");
|
||||
if (selected.isValid()) {
|
||||
m_Prop->Set(Vector3d(selected.redF(), selected.greenF(), selected.blueF()));
|
||||
emit updated();
|
||||
}
|
||||
}
|
||||
|
||||
@@ -289,7 +286,7 @@ StringPropertyWidget::StringPropertyWidget(Property<std::string>* prop, QWidget*
|
||||
m_Layout->addWidget(m_LineEdit, 1);
|
||||
connect(m_LineEdit, &QLineEdit::editingFinished, [this](){
|
||||
std::string val = m_LineEdit->text().toStdString();
|
||||
if (m_Prop->Get() != val) { m_Prop->Set(val); emit updated(); }
|
||||
if (m_Prop->Get() != val) m_Prop->Set(val);
|
||||
});
|
||||
m_Connection = uLib::Object::connect(m_Prop, &Property<std::string>::Updated, [this](){
|
||||
if (m_LineEdit->text().toStdString() != m_Prop->Get()) {
|
||||
@@ -300,40 +297,6 @@ StringPropertyWidget::StringPropertyWidget(Property<std::string>* prop, QWidget*
|
||||
}
|
||||
StringPropertyWidget::~StringPropertyWidget() {}
|
||||
|
||||
FontPropertyWidget::FontPropertyWidget(Property<FontConfig>* prop, QWidget* parent)
|
||||
: PropertyWidgetBase(prop, parent), m_Prop(prop) {
|
||||
m_Button = new QPushButton(this);
|
||||
m_Button->setMinimumWidth(100);
|
||||
this->updateButtonText();
|
||||
m_Layout->addWidget(m_Button, 1);
|
||||
|
||||
connect(m_Button, &QPushButton::clicked, this, &FontPropertyWidget::onClicked);
|
||||
m_Connection = uLib::Object::connect(m_Prop, &Property<FontConfig>::Updated, [this](){
|
||||
this->updateButtonText();
|
||||
});
|
||||
}
|
||||
FontPropertyWidget::~FontPropertyWidget() {}
|
||||
|
||||
void FontPropertyWidget::updateButtonText() {
|
||||
FontConfig f = m_Prop->Get();
|
||||
m_Button->setText(QString::fromStdString(f.family) + " " + QString::number(f.size));
|
||||
}
|
||||
|
||||
void FontPropertyWidget::onClicked() {
|
||||
FontConfig current = m_Prop->Get();
|
||||
QFont font(QString::fromStdString(current.family), current.size);
|
||||
font.setBold(current.bold);
|
||||
font.setItalic(current.italic);
|
||||
|
||||
bool ok;
|
||||
QFont selected = QFontDialog::getFont(&ok, font, this, "Select Font");
|
||||
if (ok) {
|
||||
FontConfig newF(selected.family().toStdString(), selected.pointSize(), selected.bold(), selected.italic());
|
||||
m_Prop->Set(newF);
|
||||
emit updated();
|
||||
}
|
||||
}
|
||||
|
||||
class GroupHeaderWidget : public QWidget {
|
||||
public:
|
||||
GroupHeaderWidget(const QString& name, QWidget* parent = nullptr) : QWidget(parent) {
|
||||
@@ -369,9 +332,8 @@ public:
|
||||
// Get initial value
|
||||
if (auto* p = dynamic_cast<Property<int>*>(prop)) {
|
||||
m_Combo->setCurrentIndex(p->Get());
|
||||
connect(m_Combo, &QComboBox::currentIndexChanged, [this, p](int index){
|
||||
connect(m_Combo, &QComboBox::currentIndexChanged, [p](int index){
|
||||
p->Set(index);
|
||||
emit updated();
|
||||
});
|
||||
// Store connection in base m_Connection so it's auto-disconnected on destruction.
|
||||
m_Connection = uLib::Object::connect(p, &Property<int>::Updated, [this, p](){
|
||||
@@ -412,9 +374,6 @@ PropertyEditor::PropertyEditor(QWidget* parent) : QWidget(parent), m_Object(null
|
||||
registerFactory<std::string>([](PropertyBase* p, QWidget* parent){
|
||||
return new StringPropertyWidget(static_cast<Property<std::string>*>(p), parent);
|
||||
});
|
||||
registerFactory<FontConfig>([](PropertyBase* p, QWidget* parent){
|
||||
return new FontPropertyWidget(static_cast<Property<FontConfig>*>(p), parent);
|
||||
});
|
||||
|
||||
// Register EnumProperty specifically (needs to check type since it holds Property<int> but is EnumProperty)
|
||||
m_Factories[std::type_index(typeid(EnumProperty))] = [](PropertyBase* p, QWidget* parent) {
|
||||
@@ -444,10 +403,10 @@ void PropertyEditor::setObject(::uLib::Object* obj, bool displayOnly) {
|
||||
const std::vector<::uLib::PropertyBase*>* props = &obj->GetProperties();
|
||||
|
||||
if (displayOnly) {
|
||||
if (auto* prop3d = dynamic_cast<::uLib::Vtk::Prop3D*>(obj)) {
|
||||
props = &prop3d->GetDisplayProperties();
|
||||
if (auto* puppet = dynamic_cast<::uLib::Vtk::Puppet*>(obj)) {
|
||||
props = &puppet->GetDisplayProperties();
|
||||
} else {
|
||||
// If it's not a prop3d but displayOnly is requested, showing nothing or fallback?
|
||||
// If it's not a puppet but displayOnly is requested, showing nothing or fallback?
|
||||
// Fallback: core properties.
|
||||
}
|
||||
}
|
||||
@@ -503,12 +462,6 @@ void PropertyEditor::setObject(::uLib::Object* obj, bool displayOnly) {
|
||||
}
|
||||
|
||||
if (widget) {
|
||||
if (auto* propWidget = qobject_cast<PropertyWidgetBase*>(widget)) {
|
||||
connect(propWidget, &PropertyWidgetBase::updated, [this, prop](){
|
||||
emit propertyUpdated(prop);
|
||||
});
|
||||
}
|
||||
|
||||
if (!groupName.empty()) {
|
||||
// Indent grouped properties
|
||||
widget->setContentsMargins(16, 0, 0, 0);
|
||||
|
||||
@@ -17,7 +17,6 @@ class QSlider;
|
||||
#include "Core/Property.h"
|
||||
#include "Core/Object.h"
|
||||
#include "Core/Signal.h"
|
||||
#include "Core/FontConfig.h"
|
||||
#include "Math/Dense.h"
|
||||
#include "Settings.h"
|
||||
|
||||
@@ -33,9 +32,6 @@ public:
|
||||
virtual ~PropertyWidgetBase();
|
||||
PropertyBase* getProperty() const { return m_BaseProperty; }
|
||||
|
||||
signals:
|
||||
void updated();
|
||||
|
||||
protected:
|
||||
PropertyBase* m_BaseProperty;
|
||||
QHBoxLayout* m_Layout;
|
||||
@@ -125,10 +121,7 @@ public:
|
||||
connect(m_Edits[i], &UnitLineEdit::valueManualChanged, [this, i](double val){
|
||||
VecT v = m_Prop->Get();
|
||||
v(i) = (typename VecT::Scalar)val;
|
||||
if (m_Prop->Get() != v) {
|
||||
m_Prop->Set(v);
|
||||
emit updated();
|
||||
}
|
||||
if (m_Prop->Get() != v) m_Prop->Set(v);
|
||||
});
|
||||
}
|
||||
updateEdits();
|
||||
@@ -198,19 +191,6 @@ private:
|
||||
QLineEdit* m_LineEdit;
|
||||
};
|
||||
|
||||
class FontPropertyWidget : public PropertyWidgetBase {
|
||||
Q_OBJECT
|
||||
public:
|
||||
FontPropertyWidget(Property<FontConfig>* prop, QWidget* parent = nullptr);
|
||||
virtual ~FontPropertyWidget();
|
||||
private slots:
|
||||
void onClicked();
|
||||
private:
|
||||
void updateButtonText();
|
||||
Property<FontConfig>* m_Prop;
|
||||
QPushButton* m_Button;
|
||||
};
|
||||
|
||||
class PropertyEditor : public QWidget {
|
||||
Q_OBJECT
|
||||
public:
|
||||
@@ -222,9 +202,6 @@ public:
|
||||
m_Factories[std::type_index(typeid(T))] = factory;
|
||||
}
|
||||
|
||||
signals:
|
||||
void propertyUpdated(PropertyBase* prop = nullptr);
|
||||
|
||||
private:
|
||||
void clear();
|
||||
uLib::Object* m_Object;
|
||||
|
||||
@@ -88,7 +88,7 @@ void QViewportPane::toggleDisplayPanel() {
|
||||
|
||||
void QViewportPane::setObject(uLib::Object* obj) {
|
||||
m_displayEditor->setObject(obj, true);
|
||||
// Auto-show panel if it's a prop3d and we want to highlight this feature?
|
||||
// Auto-show panel if it's a puppet and we want to highlight this feature?
|
||||
// User asked for "hiding panel", so maybe we don't auto-show.
|
||||
}
|
||||
|
||||
@@ -179,9 +179,9 @@ void QViewportPane::AttemptSplit(Qt::Orientation orientation) {
|
||||
if (currentVtk) {
|
||||
auto* newVtk = qobject_cast<uLib::Vtk::QViewport*>(newPane->currentViewport());
|
||||
if (newVtk) {
|
||||
// Copy prop3ds
|
||||
for (auto* prop3d : currentVtk->getProp3Ds()) {
|
||||
newVtk->AddProp3D(*prop3d);
|
||||
// Copy puppets
|
||||
for (auto* puppet : currentVtk->getPuppets()) {
|
||||
newVtk->AddPuppet(*puppet);
|
||||
}
|
||||
// Copy camera
|
||||
if (currentVtk->GetRenderer() && newVtk->GetRenderer()) {
|
||||
|
||||
@@ -4,8 +4,6 @@
|
||||
#include <string>
|
||||
#include <map>
|
||||
#include "Math/Units.h"
|
||||
#include "Core/FontConfig.h"
|
||||
#include "Math/Dense.h"
|
||||
|
||||
namespace uLib {
|
||||
namespace Qt {
|
||||
@@ -25,11 +23,6 @@ public:
|
||||
Dimensionless
|
||||
};
|
||||
|
||||
enum Theme {
|
||||
Dark,
|
||||
Bright
|
||||
};
|
||||
|
||||
void SetPreferredUnit(Dimension dim, const std::string& unit) {
|
||||
m_PreferredUnits[dim] = unit;
|
||||
}
|
||||
@@ -71,29 +64,9 @@ public:
|
||||
return Dimensionless;
|
||||
}
|
||||
|
||||
bool GetThrottledRendering() const { return m_ThrottledRendering; }
|
||||
void SetThrottledRendering(bool enabled) { m_ThrottledRendering = enabled; }
|
||||
|
||||
Theme GetTheme() const { return m_Theme; }
|
||||
void SetTheme(Theme theme) { m_Theme = theme; }
|
||||
|
||||
FontConfig GetFont() const { return m_Font; }
|
||||
void SetFont(const FontConfig& font) { m_Font = font; }
|
||||
|
||||
FontConfig GetGuiFont() const { return m_GuiFont; }
|
||||
void SetGuiFont(const FontConfig& font) { m_GuiFont = font; }
|
||||
|
||||
Vector3d GetFontColor() const { return m_FontColor; }
|
||||
void SetFontColor(const Vector3d& color) { m_FontColor = color; }
|
||||
|
||||
private:
|
||||
Settings() : m_ThrottledRendering(true), m_Theme(Dark), m_Font("Arial", 10), m_GuiFont("Inter", 9), m_FontColor(1.0, 1.0, 1.0) {}
|
||||
Settings() {}
|
||||
std::map<Dimension, std::string> m_PreferredUnits;
|
||||
bool m_ThrottledRendering;
|
||||
Theme m_Theme;
|
||||
FontConfig m_Font;
|
||||
FontConfig m_GuiFont;
|
||||
Vector3d m_FontColor;
|
||||
};
|
||||
|
||||
} // namespace Qt
|
||||
|
||||
@@ -1,15 +1,11 @@
|
||||
#include "StyleManager.h"
|
||||
#include <QApplication>
|
||||
#include "Core/FontConfig.h"
|
||||
|
||||
static const QString DARK_THEME = R"(
|
||||
QWidget#MenuPanel { background-color: #2b2b2b; border-bottom: 1px solid #111; }
|
||||
QLabel#LogoLabel { font-weight: bold; color: #0078d7; font-size: 14px; letter-spacing: 1px; }
|
||||
QPushButton#MenuButton { background: transparent; color: #ccc; border: none; padding: 5px 10px; }
|
||||
QPushButton#MenuButton:hover { background: #3c3c3c; color: white; border-radius: 4px; }
|
||||
QPushButton { background-color: #3e3e42; color: white; border: 1px solid #555; padding: 4px 12px; border-radius: 2px; }
|
||||
QPushButton:hover { background-color: #505050; border-color: #0078d7; }
|
||||
QPushButton:pressed { background-color: #0078d7; }
|
||||
QWidget#PaneTitleBar { background-color: #333; color: white; border-bottom: 2px solid #222; }
|
||||
QLabel#TitleLabel { font-weight: bold; margin-left: 2px; }
|
||||
QToolButton#PaneCloseButton { border: none; font-weight: bold; background: transparent; color: #ccc; }
|
||||
@@ -27,11 +23,8 @@ QScrollArea > QWidget > QWidget { background: transparent; }
|
||||
|
||||
/* Property Widgets Styling */
|
||||
QLabel { color: #cccccc; }
|
||||
QDoubleSpinBox, QSpinBox, QLineEdit, QComboBox { background: #3c3c3c; color: #f1f1f1; border: 1px solid #3e3e42; padding: 2px 4px; border-radius: 2px; selection-background-color: #0078d7; }
|
||||
QDoubleSpinBox:focus, QSpinBox:focus, QLineEdit:focus, QComboBox:focus { border-color: #0078d7; }
|
||||
QComboBox::drop-down { border-left-width: 1px; border-left-color: #3e3e42; border-left-style: solid; width: 20px; border-top-right-radius: 2px; border-bottom-right-radius: 2px; }
|
||||
QComboBox::down-arrow { image: none; border: 5px solid transparent; border-top-color: #ccc; margin-top: 5px; }
|
||||
QAbstractItemView { background-color: #2b2b2b; color: white; border: 1px solid #3e3e42; selection-background-color: #0078d7; outline: 0; }
|
||||
QDoubleSpinBox, QSpinBox, QLineEdit { background: #3c3c3c; color: #f1f1f1; border: 1px solid #3e3e42; padding: 2px 4px; border-radius: 2px; selection-background-color: #0078d7; }
|
||||
QDoubleSpinBox:focus, QSpinBox:focus, QLineEdit:focus { border-color: #0078d7; }
|
||||
QCheckBox { color: #cccccc; spacing: 5px; }
|
||||
QCheckBox::indicator { width: 14px; height: 14px; border: 1px solid #3e3e42; background: #333337; border-radius: 2px; }
|
||||
QCheckBox::indicator:checked { background: #0078d7; border-color: #005a9e; }
|
||||
@@ -49,11 +42,6 @@ QScrollBar:vertical { background: #1e1e1e; width: 12px; margin: 0px; }
|
||||
QScrollBar::handle:vertical { background: #3e3e42; min-height: 20px; border-radius: 6px; margin: 2px; }
|
||||
QScrollBar::handle:vertical:hover { background: #505050; }
|
||||
QScrollBar::add-line:vertical, QScrollBar::sub-line:vertical { height: 0px; }
|
||||
|
||||
/* Dialogs & Preferences */
|
||||
QDialog { background-color: #252526; color: #f1f1f1; }
|
||||
QGroupBox { font-weight: bold; color: #0078d7; border: 1px solid #3e3e42; margin-top: 1.1em; padding-top: 0.5em; border-radius: 4px; }
|
||||
QGroupBox::title { subcontrol-origin: margin; subcontrol-position: top left; padding: 0 3px; left: 10px; }
|
||||
)";
|
||||
|
||||
static const QString BRIGHT_THEME = R"(
|
||||
@@ -61,9 +49,6 @@ QWidget#MenuPanel { background-color: #f3f3f3; border-bottom: 1px solid #ccc; }
|
||||
QLabel#LogoLabel { font-weight: bold; color: #005a9e; font-size: 14px; letter-spacing: 1px; }
|
||||
QPushButton#MenuButton { background: transparent; color: #333; border: none; padding: 5px 10px; }
|
||||
QPushButton#MenuButton:hover { background: #e5e5e5; color: black; border-radius: 4px; }
|
||||
QPushButton { background-color: #ffffff; color: #333; border: 1px solid #cccccc; padding: 4px 12px; border-radius: 2px; }
|
||||
QPushButton:hover { background-color: #f2f2f2; border-color: #0078d7; }
|
||||
QPushButton:pressed { background-color: #0078d7; color: white; }
|
||||
QWidget#PaneTitleBar { background-color: #eeeeee; color: black; border-bottom: 2px solid #ddd; }
|
||||
QLabel#TitleLabel { font-weight: bold; margin-left: 2px; }
|
||||
QToolButton#PaneCloseButton { border: none; font-weight: bold; background: transparent; color: #666; }
|
||||
@@ -81,11 +66,8 @@ QScrollArea > QWidget > QWidget { background: transparent; }
|
||||
|
||||
/* Property Widgets Styling */
|
||||
QLabel { color: #333333; }
|
||||
QDoubleSpinBox, QSpinBox, QLineEdit, QComboBox { background: #ffffff; color: #333333; border: 1px solid #cccccc; padding: 2px 4px; border-radius: 2px; selection-background-color: #0078d7; }
|
||||
QDoubleSpinBox:focus, QSpinBox:focus, QLineEdit:focus, QComboBox:focus { border-color: #0078d7; }
|
||||
QComboBox::drop-down { border-left-width: 1px; border-left-color: #cccccc; border-left-style: solid; width: 20px; border-top-right-radius: 2px; border-bottom-right-radius: 2px; }
|
||||
QComboBox::down-arrow { image: none; border: 5px solid transparent; border-top-color: #666; margin-top: 5px; }
|
||||
QAbstractItemView { background-color: #ffffff; color: #333; border: 1px solid #cccccc; selection-background-color: #0078d7; outline: 0; }
|
||||
QDoubleSpinBox, QSpinBox, QLineEdit { background: #ffffff; color: #333333; border: 1px solid #cccccc; padding: 2px 4px; border-radius: 2px; selection-background-color: #0078d7; }
|
||||
QDoubleSpinBox:focus, QSpinBox:focus, QLineEdit:focus { border-color: #0078d7; }
|
||||
QCheckBox { color: #333333; spacing: 5px; }
|
||||
QCheckBox::indicator { width: 14px; height: 14px; border: 1px solid #cccccc; background: #ffffff; border-radius: 2px; }
|
||||
QCheckBox::indicator:checked { background: #0078d7; border-color: #005a9e; }
|
||||
@@ -103,26 +85,14 @@ QScrollBar:vertical { background: #ffffff; width: 12px; margin: 0px; }
|
||||
QScrollBar::handle:vertical { background: #cccccc; min-height: 20px; border-radius: 6px; margin: 2px; }
|
||||
QScrollBar::handle:vertical:hover { background: #aaaaaa; }
|
||||
QScrollBar::add-line:vertical, QScrollBar::sub-line:vertical { height: 0px; }
|
||||
|
||||
/* Dialogs & Preferences */
|
||||
QDialog { background-color: #f3f3f3; color: #333; }
|
||||
QGroupBox { font-weight: bold; color: #005a9e; border: 1px solid #cccccc; margin-top: 1.1em; padding-top: 0.5em; border-radius: 4px; }
|
||||
QGroupBox::title { subcontrol-origin: margin; subcontrol-position: top left; padding: 0 3px; left: 10px; }
|
||||
)";
|
||||
|
||||
void StyleManager::applyStyle(QApplication* app, const QString& themeName, const uLib::FontConfig& fontCfg) {
|
||||
void StyleManager::applyStyle(QApplication* app, const QString& themeName) {
|
||||
if (!app) return;
|
||||
|
||||
QString baseStyle = (themeName == "bright") ? BRIGHT_THEME : DARK_THEME;
|
||||
|
||||
QString fontStyle = QString(
|
||||
"QWidget { font-family: '%1'; font-size: %2pt; }\n"
|
||||
).arg(QString::fromStdString(fontCfg.family))
|
||||
.arg(fontCfg.size);
|
||||
|
||||
// If bold/italic are needed globally
|
||||
if (fontCfg.bold) fontStyle += "QWidget { font-weight: bold; }\n";
|
||||
if (fontCfg.italic) fontStyle += "QWidget { font-style: italic; }\n";
|
||||
|
||||
app->setStyleSheet(fontStyle + baseStyle);
|
||||
if (themeName == "bright") {
|
||||
app->setStyleSheet(BRIGHT_THEME);
|
||||
} else {
|
||||
app->setStyleSheet(DARK_THEME); // default
|
||||
}
|
||||
}
|
||||
|
||||
@@ -5,11 +5,9 @@
|
||||
|
||||
class QApplication;
|
||||
|
||||
namespace uLib { class FontConfig; }
|
||||
|
||||
class StyleManager {
|
||||
public:
|
||||
static void applyStyle(QApplication* app, const QString& themeName, const uLib::FontConfig& font);
|
||||
static void applyStyle(QApplication* app, const QString& themeName);
|
||||
};
|
||||
|
||||
#endif // STYLEMANAGER_H
|
||||
|
||||
@@ -99,15 +99,15 @@ void ViewportPane::toggleDisplayPanel() {
|
||||
void ViewportPane::setObject(uLib::Object* obj) {
|
||||
m_displayEditor->setObject(obj, true);
|
||||
|
||||
// Check if the object is a Prop3D (meaning it has display properties)
|
||||
bool isProp3D = (dynamic_cast<::uLib::Vtk::Prop3D*>(obj) != nullptr);
|
||||
// Check if the object is a Puppet (meaning it has display properties)
|
||||
bool isPuppet = (dynamic_cast<::uLib::Vtk::Puppet*>(obj) != nullptr);
|
||||
|
||||
// Only show the "Display" toggle button if it's a prop3d
|
||||
m_toggleBtn->setVisible(isProp3D);
|
||||
// Only show the "Display" toggle button if it's a puppet
|
||||
m_toggleBtn->setVisible(isPuppet);
|
||||
|
||||
// If it's a prop3d, we might want to keep the panel state if it was already open,
|
||||
// or if it's NOT a prop3d, definitely hide the toggle and panel.
|
||||
if (!isProp3D) {
|
||||
// If it's a puppet, we might want to keep the panel state if it was already open,
|
||||
// or if it's NOT a puppet, definitely hide the toggle and panel.
|
||||
if (!isPuppet) {
|
||||
m_toggleBtn->setChecked(false);
|
||||
m_displayPanel->hide();
|
||||
}
|
||||
@@ -125,10 +125,6 @@ void ViewportPane::setViewport(QWidget* viewport, const QString& title) {
|
||||
m_areaSplitter->setStretchFactor(0, 1);
|
||||
}
|
||||
|
||||
uLib::Vtk::Viewport* ViewportPane::viewport() const {
|
||||
return dynamic_cast<uLib::Vtk::Viewport*>(m_viewport);
|
||||
}
|
||||
|
||||
void ViewportPane::addVtkViewport() {
|
||||
auto* viewport = new uLib::Vtk::QViewport(this);
|
||||
setViewport(viewport, "VTK Viewport");
|
||||
|
||||
@@ -8,7 +8,6 @@
|
||||
namespace uLib {
|
||||
class Object;
|
||||
namespace Qt { class PropertyEditor; }
|
||||
namespace Vtk { class Viewport; }
|
||||
}
|
||||
|
||||
class QSplitter;
|
||||
@@ -25,7 +24,6 @@ public:
|
||||
void addRootCanvas();
|
||||
|
||||
QWidget* currentViewport() const { return m_viewport; }
|
||||
uLib::Vtk::Viewport* viewport() const;
|
||||
|
||||
/** @brief Update the display properties for the given object. */
|
||||
void setObject(uLib::Object* obj);
|
||||
|
||||
@@ -3,8 +3,6 @@
|
||||
#include "MainPanel.h"
|
||||
#include "ViewportPane.h"
|
||||
#include "StyleManager.h"
|
||||
#include "Settings.h"
|
||||
#include "Core/FontConfig.h"
|
||||
|
||||
#include "Math/ContainerBox.h"
|
||||
#include <HEP/Geant/Scene.h>
|
||||
@@ -31,9 +29,7 @@ using namespace uLib::literals;
|
||||
|
||||
int main(int argc, char** argv) {
|
||||
QApplication app(argc, argv);
|
||||
auto theme = uLib::Qt::Settings::Instance().GetTheme();
|
||||
auto initialGuiFont = uLib::Qt::Settings::Instance().GetGuiFont();
|
||||
StyleManager::applyStyle(&app, theme == uLib::Qt::Settings::Dark ? "dark" : "bright", initialGuiFont);
|
||||
StyleManager::applyStyle(&app, "dark");
|
||||
std::cout << "Starting gcompose Qt application..." << std::endl;
|
||||
|
||||
// ContainerBox world_box(Vector3f(1, 1, 1));
|
||||
|
||||
@@ -1,6 +1,6 @@
|
||||
[requires]
|
||||
# eigen/5.0.1
|
||||
boost/1.86.0
|
||||
eigen/3.4.0
|
||||
boost/1.83.0
|
||||
# pybind11/3.0.2
|
||||
hdf5/1.14.3
|
||||
|
||||
|
||||
15
condaenv.yml
15
condaenv.yml
@@ -1,4 +1,4 @@
|
||||
name: uLib
|
||||
name: mutom
|
||||
channels:
|
||||
- conda-forge
|
||||
dependencies:
|
||||
@@ -7,15 +7,4 @@ dependencies:
|
||||
- cmake
|
||||
- conan
|
||||
- root
|
||||
- vtk=9.4 # VTK 9.4
|
||||
- pybind11
|
||||
# - boost=1.86.0 # requested by VTK 9.4
|
||||
- ninja
|
||||
- clang
|
||||
- clangxx
|
||||
- lld
|
||||
- ccache
|
||||
- OpenMP
|
||||
- Geant4
|
||||
- gdb
|
||||
- valgrind
|
||||
- vtk
|
||||
|
||||
@@ -1,48 +0,0 @@
|
||||
# Prop3D
|
||||
|
||||
`uLib::Vtk::Prop3D` is a bridge class that wraps VTK 3D representations (`vtkProp`, `vtkProp3D`) and integrates them into the `uLib` object model. It allows the framework to manage visual objects, synchronize them with underlying data models, and expose display-specific properties to the GUI.
|
||||
|
||||
## Inheritance
|
||||
`uLib::Vtk::Prop3D` : `uLib::Object`
|
||||
|
||||
## Key Functionalities
|
||||
|
||||
### VTK Integration
|
||||
The class provides access to the underlying VTK objects:
|
||||
- `GetProp()`: Returns the `vtkProp`.
|
||||
- `GetProxyProp()`: Returns the `vtkProp3D`.
|
||||
- `GetParts()` / `GetProps()`: Returns `vtkPropCollection` for composite objects.
|
||||
|
||||
### Model-View Synchronization
|
||||
`Prop3D` ensures that the visual representation stays in sync with the domain model:
|
||||
- `Update()`: Synchronizes the VTK representation based on current internal state and properties. Should be called when model data changes.
|
||||
- `SyncFromVtk()`: Updates internal state using data from the VTK representation (e.g., after user interaction via gizmos in the 3D view).
|
||||
- `GetContent()`: Returns the `uLib::Object` that this `Prop3D` represents visually.
|
||||
|
||||
### Visual Appearance
|
||||
- **Color & Opacity**: `SetColor(r, g, b)` and `SetOpacity(alpha)`.
|
||||
- **Selection**: `SetSelectable(bool)` and `SetSelected(bool)` to manage interactivity and highlighting.
|
||||
- **BBox/Scale**: `ShowBoundingBox(bool)` and `ShowScaleMeasures(bool)`.
|
||||
|
||||
### Rendering Modes
|
||||
The rendering style can be controlled via the `Representation` enum:
|
||||
- `Points`
|
||||
- `Wireframe`
|
||||
- `Surface`
|
||||
- `SurfaceWithEdges`
|
||||
- `Volume`
|
||||
- `Outline`
|
||||
- `Slice`
|
||||
|
||||
### Display Properties System
|
||||
`Prop3D` implements a system to expose specific properties (often marked as `hrp` - human readable properties) to a property editor in the GUI.
|
||||
|
||||
- `GetDisplayProperties()`: Returns the list of properties registered for display.
|
||||
- `RegisterDisplayProperty(uLib::PropertyBase*)`: Adds a property to the display list.
|
||||
- `serialize_display(...)`: A virtual method that subclasses implement to define which properties should be exposed.
|
||||
|
||||
#### Activating Display Properties
|
||||
To automatically populate the display properties list, the `ULIB_ACTIVATE_DISPLAY_PROPERTIES` macro should be called in the constructor. This triggers `serialize_display` using a `display_properties_archive`.
|
||||
|
||||
## Implementation Details
|
||||
`Prop3D` uses the Pimpl idiom (via `Prop3DData *pd`) to hide VTK-specific implementation details and reduce header dependencies.
|
||||
@@ -3,32 +3,9 @@
|
||||
Geant4 integration in uLib is done through the `HEP/Geant` module.
|
||||
The module represets a set of wrapper for geant objects that are also deriving from uLib::Object so they can be used in the uLib::Object tree and visualized with the uLib::Vtk module and driven py properties.
|
||||
|
||||
# Geant Solid Integration
|
||||
|
||||
Geant4 solids in `uLib` are encapsulated within the `uLib::Geant::Solid` hierarchy, with primary implementations such as `BoxSolid` and `TessellatedSolid`. These classes act as reactive wrappers around native Geant4 `G4VSolid` objects, bridging the gap between Geant4's static geometry and `uLib`'s dynamic property system.
|
||||
|
||||
### Property Mapping and Synchronization
|
||||
The integration relies on a mapping between `uLib` properties and Geant4 parameters. This allows geometry to be driven by Qt-based property panels or interactive VTK handles. When a property is modified, the library handles the update through the following mechanism:
|
||||
|
||||
* **Parameter Application**: If the underlying Geant4 object supports dynamic updates (e.g., changing box dimensions via `SetXHalfLength`), `uLib` applies the change directly to the existing object.
|
||||
* **Solid Re-instantiation**: In cases where Geant4 does not support on-the-fly parameter changes, the library transparently recreates the underlying `G4VSolid`. This ensures that the simulation model always reflects the current state of the design environment.
|
||||
* **Global Signaling**: Once the underlying Geant4 state is synchronized, the `uLib::Geant::Solid` emits an `updated()` signal. This triggers cascading updates across all dependent observers, such as the VTK rendering pipeline and the Geant4 scene manager, ensuring visual and logical consistency.
|
||||
|
||||
## VTK Visualization Layer
|
||||
|
||||
The visualization of Geant4 solids in VTK is managed by the `uLib::Vtk::GeantSolid` class and its specializations (e.g., `vtkBoxSolid`). This class serves as a bridge between the domain model (`uLib::Geant::Solid`) and the VTK rendering pipeline, inheriting from `uLib::Vtk::Prop3D` to leverage the framework's standard transformation and property management features.
|
||||
|
||||
### Geometry Extraction and Rendering
|
||||
Since Geant4 solids are defined by analytical or tessellated boundary representations, `GeantSolid` converts them into a format suitable for VTK:
|
||||
1. **Faceted Representation**: It retrieves the `G4Polyhedron` tessellation from the underlying `G4VSolid`.
|
||||
2. **PolyData Mapping**: The vertices and facets of the polyhedron are extracted and mapped to a `vtkPolyData` object.
|
||||
3. **Actor Configuration**: This geometry is assigned to a `vtkActor`, which is then added to the `Prop3D`. Default visual properties—such as semi-transparent surfaces and edge visibility—are applied to aid in debugging and design.
|
||||
|
||||
### Transform Synchronization
|
||||
The spatial state (Translation, Rotation, and Scale) is synchronized between the Geant4 geometry tree and the VTK viewer ensuring a consistent representation across both domains.
|
||||
|
||||
* **Geant4 to VTK (Forward)**: When a solid's placement is updated in Geant4, `GeantSolid` extracts the translation and rotation from the associated `G4VPhysicalVolume`. These are converted into a `vtkTransform` and applied to the actor, aligning the visual model with the simulation's coordinate system.
|
||||
* **VTK to Geant4 (Interactive)**: Interaction in the viewer (e.g., using a transformation handler) triggers the `SyncFromVtk()` method. This pushes the new transformation matrix back to the `uLib::Geant::Solid` domain object, which then updates the Geant4 physical volume.
|
||||
* **Feedback Loop Prevention**: To avoid infinite recursion during interactive updates, the synchronization logic utilizes temporary signal blocking. This ensures that a transform update originating from VTK does not trigger a redundant re-update of the VTK representation from the domain model.
|
||||
# Geant Solid integration
|
||||
|
||||
Geant solid in uLib is represented by the `uLib::Geant::Solid` class and mainly BoxSolid and TessellatedSolid. The solids in Geant does not have the possibility to set properties on the fly so we need to create a new solid every time we want to change the properties of a solid. This is done by creating a new `uLib::Geant::Solid` object and setting the properties of the new solid. The new solid is then added to the `uLib::Geant::Solid` object as a child. The old solid is then removed from the `uLib::Geant::Solid` object as a child. The old solid is then deleted. However id some of the properties can be set then the library will drive the change in the solid update.
|
||||
|
||||
The idea is to have a mapping of solid properties that can be used in uLib for Qt representation or vtk representation. then when the property is changed the signaling will update the property in uLib and then the solid will be updated. If the Geant property can be applied to the G4 object underneath then the update will apply the change, in case it is not possible to apply the change to the G4 object underneath then the G4 element will be recreated.
|
||||
In any case a updated singal is emitted and the related element that use that solid is updated ( for instance the scene ).
|
||||
@@ -1,21 +1,21 @@
|
||||
|
||||
# Properties and the vtk-gui representation
|
||||
|
||||
This is the rationale behind the connection between TRS properties and Prop3D Transformation.
|
||||
This is the rationale behind the connection between TRS properties and Puppet Transformation.
|
||||
|
||||
The properties from model get propoagated via Object signalling system (the Update signal) to the vtkRepresentation and to the Qt widgets so that the overall transformation of the model reflects into a modification of its representation in vtk and in the gui.
|
||||
|
||||
In addition the properties need to be adjusted also from vtk, for example if user uses handlerwidget to change the transformation this is eventually applied to Prop3D and Prop3D should propagate the transformation change to the vtk representation object (for instance vtkContainerBox) and the latter eventually propagates the change into the model.
|
||||
In addition the properties need to be adjusted also from vtk, for example if user uses handlerwidget to change the transformation this is eventually applied to Puppet and Puppet should propagate the transformation change to the vtk representation object (for instance vtkContainerBox) and the latter eventually propagates the change into the model.
|
||||
|
||||
the Prop3D or the vtk representation wrapper ( vtkContainerBox for instance is the wrapper od ContainerBox ) should not directly show the transformation of the handlerwidget but it should show the transformation of the model once applied so we are always seeing the actual aspect of the model reflected to the vtk representation and not the other way around.
|
||||
the Puppet or the vtk representation wrapper ( vtkContainerBox for instance is the wrapper od ContainerBox ) should not directly show the transformation of the handlerwidget but it should show the transformation of the model once applied so we are always seeing the actual aspect of the model reflected to the vtk representation and not the other way around.
|
||||
|
||||
So in syntesis the model is the master and the vtk representation and the gui are the slaves of any modification, but the vtkHandlerWidget is able to apply a transform that should be applied to the model and then the model should propagate the transformation change to the vtk representation and to the gui.
|
||||
|
||||
## The Prop3D
|
||||
## The Puppet
|
||||
|
||||
The prop3d is the proxy of the spatial placement of objects in the scene. Prop3Ds should have an internal ContainerBox that is shown in the scene around the content to be able to pick Prop3D from vtkViewport using the handler widget. The HandlerWidget moves the Prop3D ContainerBox (the red Highlight element whe selected) to reflect the handler current transformation, but the transformation is propagated to the derived Prop3D classes like vtkContainerBox.
|
||||
The puppet is the proxy of the spatial placement of objects in the scene. Puppets should have an internal ContainerBox that is shown in the scene around the content to be able to pick Puppet from vtkViewport using the handler widget. The HandlerWidget moves the Puppet ContainerBox (the red Highlight element whe selected) to reflect the handler current transformation, but the transformation is propagated to the derived Puppet classes like vtkContainerBox.
|
||||
|
||||
The vtkHandlerWidget should handle the transformation of the prop3d internal ContainerBox. The changes of the ContainerBox will be propagated to the derived classes and eventually to the model.
|
||||
The vtkHandlerWidget should handle the transformation of the puppet internal ContainerBox. The changes of the ContainerBox will be propagated to the derived classes and eventually to the model.
|
||||
|
||||
|
||||
|
||||
|
||||
@@ -80,9 +80,6 @@ template <class ArchiveImplementation> class polymorphic_iarchive_route;
|
||||
namespace boost {
|
||||
namespace serialization {
|
||||
template <typename T> struct hrp;
|
||||
template <typename T> struct hrp_val;
|
||||
template <typename T> struct hrp_enum;
|
||||
template <typename T> struct hrp_enum_val;
|
||||
}
|
||||
} // namespace boost
|
||||
|
||||
@@ -180,24 +177,6 @@ public:
|
||||
return *this->This();
|
||||
}
|
||||
|
||||
template <class T>
|
||||
Archive &operator>>(const boost::serialization::hrp_val<T> &t) {
|
||||
this->This()->load_override(const_cast<boost::serialization::hrp_val<T> &>(t));
|
||||
return *this->This();
|
||||
}
|
||||
|
||||
template <class T>
|
||||
Archive &operator>>(const boost::serialization::hrp_enum<T> &t) {
|
||||
this->This()->load_override(const_cast<boost::serialization::hrp_enum<T> &>(t));
|
||||
return *this->This();
|
||||
}
|
||||
|
||||
template <class T>
|
||||
Archive &operator>>(const boost::serialization::hrp_enum_val<T> &t) {
|
||||
this->This()->load_override(const_cast<boost::serialization::hrp_enum_val<T> &>(t));
|
||||
return *this->This();
|
||||
}
|
||||
|
||||
// the & operator
|
||||
template <class T> Archive &operator&(T &t) { return *(this->This()) >> t; }
|
||||
|
||||
@@ -211,21 +190,6 @@ public:
|
||||
return *(this->This()) >> t;
|
||||
}
|
||||
|
||||
template <class T>
|
||||
Archive &operator&(const boost::serialization::hrp_val<T> &t) {
|
||||
return *(this->This()) >> t;
|
||||
}
|
||||
|
||||
template <class T>
|
||||
Archive &operator&(const boost::serialization::hrp_enum<T> &t) {
|
||||
return *(this->This()) >> t;
|
||||
}
|
||||
|
||||
template <class T>
|
||||
Archive &operator&(const boost::serialization::hrp_enum_val<T> &t) {
|
||||
return *(this->This()) >> t;
|
||||
}
|
||||
|
||||
// the == operator
|
||||
template <class T> Archive &operator==(T &t) { return this->operator&(t); }
|
||||
|
||||
@@ -265,62 +229,12 @@ public:
|
||||
this->This()->save_override(t);
|
||||
return *this->This();
|
||||
}
|
||||
|
||||
template <class T> Archive &operator<<(const boost::serialization::hrp<T> &t) {
|
||||
this->This()->save_override(t);
|
||||
return *this->This();
|
||||
}
|
||||
|
||||
template <class T> Archive &operator<<(const boost::serialization::hrp_val<T> &t) {
|
||||
this->This()->save_override(t);
|
||||
return *this->This();
|
||||
}
|
||||
|
||||
template <class T> Archive &operator<<(const boost::serialization::hrp_enum<T> &t) {
|
||||
this->This()->save_override(t);
|
||||
return *this->This();
|
||||
}
|
||||
|
||||
template <class T> Archive &operator<<(const boost::serialization::hrp_enum_val<T> &t) {
|
||||
this->This()->save_override(t);
|
||||
return *this->This();
|
||||
}
|
||||
|
||||
template <class T> Archive &operator<<(const boost::serialization::nvp<T> &t) {
|
||||
this->This()->save_override(t);
|
||||
return *this->This();
|
||||
}
|
||||
|
||||
// the & operator
|
||||
template <class T> Archive &operator&(const T &t) {
|
||||
return *this->This() << t;
|
||||
}
|
||||
|
||||
template <class T>
|
||||
Archive &operator&(const boost::serialization::hrp<T> &t) {
|
||||
return *this->This() << t;
|
||||
}
|
||||
|
||||
template <class T>
|
||||
Archive &operator&(const boost::serialization::hrp_val<T> &t) {
|
||||
return *this->This() << t;
|
||||
}
|
||||
|
||||
template <class T>
|
||||
Archive &operator&(const boost::serialization::hrp_enum<T> &t) {
|
||||
return *this->This() << t;
|
||||
}
|
||||
|
||||
template <class T>
|
||||
Archive &operator&(const boost::serialization::hrp_enum_val<T> &t) {
|
||||
return *this->This() << t;
|
||||
}
|
||||
|
||||
template <class T>
|
||||
Archive &operator&(const boost::serialization::nvp<T> &t) {
|
||||
return *this->This() << t;
|
||||
}
|
||||
|
||||
// the == operator
|
||||
template <class T> Archive &operator==(T &t) { return this->operator&(t); }
|
||||
|
||||
|
||||
@@ -21,7 +21,6 @@ set(HEADERS
|
||||
StringReader.h
|
||||
Threads.h
|
||||
Monitor.h
|
||||
Property.h
|
||||
Types.h
|
||||
Uuid.h
|
||||
Vector.h
|
||||
|
||||
@@ -1,48 +0,0 @@
|
||||
#ifndef U_CORE_FONTCONFIG_H
|
||||
#define U_CORE_FONTCONFIG_H
|
||||
|
||||
#include <string>
|
||||
#include <ostream>
|
||||
#include <boost/serialization/nvp.hpp>
|
||||
|
||||
namespace uLib {
|
||||
|
||||
/**
|
||||
* @struct FontConfig
|
||||
* @brief Basic font configuration for text properties.
|
||||
*/
|
||||
struct FontConfig {
|
||||
std::string family;
|
||||
int size;
|
||||
bool bold;
|
||||
bool italic;
|
||||
|
||||
FontConfig() : family("Arial"), size(10), bold(false), italic(false) {}
|
||||
FontConfig(const std::string& fam, int sz, bool b = false, bool i = false)
|
||||
: family(fam), size(sz), bold(b), italic(i) {}
|
||||
|
||||
bool operator==(const FontConfig& other) const {
|
||||
return family == other.family && size == other.size &&
|
||||
bold == other.bold && italic == other.italic;
|
||||
}
|
||||
bool operator!=(const FontConfig& other) const { return !(*this == other); }
|
||||
|
||||
template<class Archive>
|
||||
void serialize(Archive& ar, const unsigned int version) {
|
||||
ar & boost::serialization::make_nvp("family", family);
|
||||
ar & boost::serialization::make_nvp("size", size);
|
||||
ar & boost::serialization::make_nvp("bold", bold);
|
||||
ar & boost::serialization::make_nvp("italic", italic);
|
||||
}
|
||||
};
|
||||
|
||||
inline std::ostream& operator<<(std::ostream& os, const FontConfig& f) {
|
||||
os << f.family << " " << f.size;
|
||||
if (f.bold) os << " Bold";
|
||||
if (f.italic) os << " Italic";
|
||||
return os;
|
||||
}
|
||||
|
||||
} // namespace uLib
|
||||
|
||||
#endif // U_CORE_FONTCONFIG_H
|
||||
@@ -59,41 +59,18 @@ public:
|
||||
GenericMFPtr sloptr;
|
||||
std::string slostr;
|
||||
};
|
||||
|
||||
~ObjectPrivate() {
|
||||
for (auto& s : sigv) delete s.signal;
|
||||
}
|
||||
|
||||
std::string m_InstanceName;
|
||||
std::vector<Signal> sigv;
|
||||
std::vector<Slot> slov;
|
||||
std::vector<PropertyBase*> m_Properties;
|
||||
std::vector<PropertyBase*> m_DynamicProperties;
|
||||
std::vector<PropertyBase*> m_DisplayProperties;
|
||||
bool m_SignalsBlocked;
|
||||
};
|
||||
|
||||
// Implementations of Property methods
|
||||
void Object::RegisterDisplayProperty(PropertyBase* prop) {
|
||||
if (prop) {
|
||||
for (auto* existing : d->m_DisplayProperties) {
|
||||
if (existing == prop) return;
|
||||
if (existing->GetName() == prop->GetName()) return;
|
||||
}
|
||||
d->m_DisplayProperties.push_back(prop);
|
||||
}
|
||||
}
|
||||
|
||||
const std::vector<PropertyBase*>& Object::GetDisplayProperties() const {
|
||||
return d->m_DisplayProperties;
|
||||
}
|
||||
|
||||
void Object::RegisterProperty(PropertyBase* prop) {
|
||||
if (prop) {
|
||||
for (auto* existing : d->m_Properties) {
|
||||
if (existing == prop) return;
|
||||
if (existing->GetName() == prop->GetName()) return;
|
||||
}
|
||||
d->m_Properties.push_back(prop);
|
||||
}
|
||||
}
|
||||
@@ -119,19 +96,31 @@ PropertyBase* Object::GetProperty(const std::string& name) const {
|
||||
for (auto* p : d->m_DynamicProperties) {
|
||||
if (p->GetName() == name || p->GetQualifiedName() == name) return p;
|
||||
}
|
||||
for (auto* p : d->m_DisplayProperties) {
|
||||
if (p->GetName() == name || p->GetQualifiedName() == name) return p;
|
||||
}
|
||||
return nullptr;
|
||||
}
|
||||
|
||||
void Object::NotifyPropertiesUpdated() {
|
||||
// Only notify properties in the primary list to avoid duplicates,
|
||||
// as all registered properties should be there.
|
||||
for (auto* p : d->m_Properties) p->Updated();
|
||||
for (auto* p : d->m_DynamicProperties) p->Updated();
|
||||
}
|
||||
|
||||
// In Object.h, the template serialize needs to be updated to call property serialization.
|
||||
// However, since Object::serialize is a template in the header, we might need a helper here.
|
||||
|
||||
template <class ArchiveT>
|
||||
void Object::serialize(ArchiveT &ar, const unsigned int version) {
|
||||
ar & boost::serialization::make_nvp("InstanceName", d->m_InstanceName);
|
||||
for (auto* prop : d->m_Properties) {
|
||||
prop->serialize(ar, version);
|
||||
}
|
||||
}
|
||||
|
||||
void Object::Updated() { ULIB_SIGNAL_EMIT(Object::Updated); }
|
||||
void Object::PropertyUpdated() { ULIB_SIGNAL_EMIT(Object::PropertyUpdated); }
|
||||
|
||||
template <class ArchiveT>
|
||||
void Object::save_override(ArchiveT &ar, const unsigned int version) {}
|
||||
|
||||
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
@@ -164,9 +153,6 @@ Object::~Object() {
|
||||
for (auto* p : d->m_DynamicProperties) {
|
||||
delete p;
|
||||
}
|
||||
for (auto* p : d->m_DisplayProperties) {
|
||||
delete p;
|
||||
}
|
||||
delete d;
|
||||
}
|
||||
|
||||
|
||||
@@ -78,30 +78,26 @@ public:
|
||||
Object(const Object ©);
|
||||
virtual ~Object();
|
||||
|
||||
virtual const char *GetClassName() const { return type_name(); }
|
||||
virtual const char *type_name() const { return "Object"; }
|
||||
virtual const char * GetClassName() const { return type_name(); }
|
||||
virtual const char * type_name() const { return "Object"; }
|
||||
|
||||
const std::string &GetInstanceName() const;
|
||||
void SetInstanceName(const std::string &name);
|
||||
|
||||
/** @brief Temporarily blocks all signal emissions from this object. Returns
|
||||
* previous state. */
|
||||
const std::string& GetInstanceName() const;
|
||||
void SetInstanceName(const std::string& name);
|
||||
|
||||
/** @brief Temporarily blocks all signal emissions from this object. Returns previous state. */
|
||||
bool blockSignals(bool block);
|
||||
|
||||
|
||||
/** @brief Checks if signals are currently blocked. */
|
||||
bool signalsBlocked() const;
|
||||
|
||||
////////////////////////////////////////////////////////////////////////////
|
||||
// PROPERTIES //
|
||||
virtual void RegisterProperty(PropertyBase *property);
|
||||
virtual void RegisterDynamicProperty(PropertyBase *property);
|
||||
virtual void RegisterDisplayProperty(PropertyBase *property);
|
||||
virtual const std::vector<PropertyBase *> &GetProperties() const;
|
||||
virtual const std::vector<PropertyBase *> &GetDisplayProperties() const;
|
||||
PropertyBase *GetProperty(const std::string &name) const;
|
||||
void RegisterProperty(PropertyBase* prop);
|
||||
void RegisterDynamicProperty(PropertyBase* prop);
|
||||
const std::vector<PropertyBase*>& GetProperties() const;
|
||||
PropertyBase* GetProperty(const std::string& name) const;
|
||||
|
||||
/** @brief Sends an Updated signal for all properties of this object. useful
|
||||
* for real-time UI refresh. */
|
||||
/** @brief Sends an Updated signal for all properties of this object. useful for real-time UI refresh. */
|
||||
void NotifyPropertiesUpdated();
|
||||
|
||||
////////////////////////////////////////////////////////////////////////////
|
||||
@@ -111,7 +107,7 @@ public:
|
||||
virtual void DeepCopy(const Object ©);
|
||||
|
||||
/** @brief Returns a nested context for children objects, if any. */
|
||||
virtual ObjectsContext *GetChildren() { return nullptr; }
|
||||
virtual ObjectsContext* GetChildren() { return nullptr; }
|
||||
|
||||
////////////////////////////////////////////////////////////////////////////
|
||||
// SERIALIZATION //
|
||||
@@ -119,23 +115,16 @@ public:
|
||||
template <class ArchiveT>
|
||||
void serialize(ArchiveT &ar, const unsigned int version);
|
||||
|
||||
virtual void serialize(Archive::xml_oarchive &ar,
|
||||
const unsigned int version) {}
|
||||
virtual void serialize(Archive::xml_iarchive &ar,
|
||||
const unsigned int version) {}
|
||||
virtual void serialize(Archive::text_oarchive &ar,
|
||||
const unsigned int version) {}
|
||||
virtual void serialize(Archive::text_iarchive &ar,
|
||||
const unsigned int version) {}
|
||||
virtual void serialize(Archive::hrt_oarchive &ar,
|
||||
const unsigned int version) {}
|
||||
virtual void serialize(Archive::hrt_iarchive &ar,
|
||||
const unsigned int version) {}
|
||||
virtual void serialize(Archive::log_archive &ar, const unsigned int version) {
|
||||
}
|
||||
virtual void serialize(Archive::xml_oarchive & ar, const unsigned int version) {}
|
||||
virtual void serialize(Archive::xml_iarchive & ar, const unsigned int version) {}
|
||||
virtual void serialize(Archive::text_oarchive & ar, const unsigned int version) {}
|
||||
virtual void serialize(Archive::text_iarchive & ar, const unsigned int version) {}
|
||||
virtual void serialize(Archive::hrt_oarchive & ar, const unsigned int version) {}
|
||||
virtual void serialize(Archive::hrt_iarchive & ar, const unsigned int version) {}
|
||||
virtual void serialize(Archive::log_archive & ar, const unsigned int version) {}
|
||||
|
||||
template <class ArchiveT>
|
||||
void save_override(ArchiveT &ar, const unsigned int version) {}
|
||||
void save_override(ArchiveT &ar, const unsigned int version);
|
||||
|
||||
void SaveConfig(std::ostream &os, int version = 0);
|
||||
void LoadConfig(std::istream &is, int version = 0);
|
||||
@@ -146,8 +135,9 @@ public:
|
||||
////////////////////////////////////////////////////////////////////////////
|
||||
// SIGNALS //
|
||||
|
||||
signals:
|
||||
signals:
|
||||
virtual void Updated();
|
||||
virtual void PropertyUpdated();
|
||||
|
||||
// Qt4 style connector //
|
||||
static bool connect(const Object *ob1, const char *signal_name,
|
||||
@@ -168,14 +158,14 @@ signals:
|
||||
connect(typename FunctionPointer<Func1>::Object *sender, Func1 sigf,
|
||||
typename FunctionPointer<Func2>::Object *receiver, Func2 slof) {
|
||||
SignalBase *sigb = sender->findOrAddSignal(sigf);
|
||||
return ConnectSignal<typename FunctionPointer<Func1>::SignalSignature>(
|
||||
sigb, slof, receiver);
|
||||
return ConnectSignal<typename FunctionPointer<Func1>::SignalSignature>(sigb, slof,
|
||||
receiver);
|
||||
}
|
||||
|
||||
// Lambda/Function object connector //
|
||||
template <typename Func1, typename SlotT>
|
||||
static Connection connect(typename FunctionPointer<Func1>::Object *sender,
|
||||
Func1 sigf, SlotT slof) {
|
||||
Func1 sigf, SlotT slof) {
|
||||
SignalBase *sigb = sender->findOrAddSignal(sigf);
|
||||
typedef typename FunctionPointer<Func1>::SignalSignature SigSignature;
|
||||
typedef typename Signal<SigSignature>::type SigT;
|
||||
@@ -191,10 +181,9 @@ signals:
|
||||
}
|
||||
|
||||
template <typename FuncT>
|
||||
static inline Connection connect(SignalBase *sigb, FuncT slof,
|
||||
Object *receiver) {
|
||||
return ConnectSignal<typename FunctionPointer<FuncT>::SignalSignature>(
|
||||
sigb, slof, receiver);
|
||||
static inline Connection connect(SignalBase *sigb, FuncT slof, Object *receiver) {
|
||||
return ConnectSignal<typename FunctionPointer<FuncT>::SignalSignature>(sigb, slof,
|
||||
receiver);
|
||||
}
|
||||
|
||||
template <typename FuncT>
|
||||
|
||||
@@ -1,55 +1,53 @@
|
||||
#ifndef U_CORE_OBJECTFACTORY_H
|
||||
#define U_CORE_OBJECTFACTORY_H
|
||||
|
||||
#include "Core/Object.h"
|
||||
#include <functional>
|
||||
#include <map>
|
||||
#include <string>
|
||||
#include <type_traits>
|
||||
#include <map>
|
||||
#include <vector>
|
||||
#include <functional>
|
||||
#include "Core/Object.h"
|
||||
|
||||
namespace uLib {
|
||||
|
||||
/**
|
||||
* @brief Singleton factory for dynamic Object instantiation based on class
|
||||
* name.
|
||||
* @brief Singleton factory for dynamic Object instantiation based on class name.
|
||||
*/
|
||||
class ObjectFactory {
|
||||
public:
|
||||
typedef std::function<Object *()> FactoryFunction;
|
||||
typedef std::function<Object*()> FactoryFunction;
|
||||
|
||||
/** @brief Get the singleton instance. */
|
||||
static ObjectFactory &Instance();
|
||||
/** @brief Get the singleton instance. */
|
||||
static ObjectFactory& Instance();
|
||||
|
||||
/** @brief Register a factory function for a given class name. */
|
||||
void Register(const std::string &className, FactoryFunction func);
|
||||
/** @brief Register a factory function for a given class name. */
|
||||
void Register(const std::string& className, FactoryFunction func);
|
||||
|
||||
/** @brief Create a new instance of the specified class. */
|
||||
Object *Create(const std::string &className);
|
||||
/** @brief Create a new instance of the specified class. */
|
||||
Object* Create(const std::string& className);
|
||||
|
||||
/** @brief Get the names of all registered classes. */
|
||||
std::vector<std::string> GetRegisteredClasses() const;
|
||||
/** @brief Get the names of all registered classes. */
|
||||
std::vector<std::string> GetRegisteredClasses() const;
|
||||
|
||||
private:
|
||||
ObjectFactory() = default;
|
||||
~ObjectFactory() = default;
|
||||
ObjectFactory() = default;
|
||||
~ObjectFactory() = default;
|
||||
|
||||
// Prevent copy and assignment
|
||||
ObjectFactory(const ObjectFactory &) = delete;
|
||||
ObjectFactory &operator=(const ObjectFactory &) = delete;
|
||||
// Prevent copy and assignment
|
||||
ObjectFactory(const ObjectFactory&) = delete;
|
||||
ObjectFactory& operator=(const ObjectFactory&) = delete;
|
||||
|
||||
std::map<std::string, FactoryFunction> m_factoryMap;
|
||||
std::map<std::string, FactoryFunction> m_factoryMap;
|
||||
};
|
||||
|
||||
/**
|
||||
* @brief Helper class to statically register a factory function.
|
||||
*/
|
||||
template <typename T> class ObjectRegistrar {
|
||||
template <typename T>
|
||||
class ObjectRegistrar {
|
||||
public:
|
||||
ObjectRegistrar(const std::string &className) {
|
||||
ObjectFactory::Instance().Register(className,
|
||||
[]() -> Object * { return new T(); });
|
||||
}
|
||||
ObjectRegistrar(const std::string& className) {
|
||||
ObjectFactory::Instance().Register(className, []() -> Object* { return new T(); });
|
||||
}
|
||||
};
|
||||
|
||||
#define ULIB_REG_CONCAT_IMPL(a, b) a##b
|
||||
@@ -59,70 +57,11 @@ public:
|
||||
* @brief Macro to register a class to the factory.
|
||||
* Put this in the .cpp file of the class.
|
||||
*/
|
||||
#define ULIB_REGISTER_OBJECT(className) \
|
||||
static uLib::ObjectRegistrar<className> ULIB_REG_CONCAT( \
|
||||
g_ObjectRegistrar_, __LINE__)(#className);
|
||||
#define ULIB_REGISTER_OBJECT(className) \
|
||||
static uLib::ObjectRegistrar<className> ULIB_REG_CONCAT(g_ObjectRegistrar_, __LINE__)(#className);
|
||||
|
||||
#define ULIB_REGISTER_OBJECT_NAME(className, registeredName) \
|
||||
static uLib::ObjectRegistrar<className> ULIB_REG_CONCAT( \
|
||||
g_ObjectRegistrar_, __LINE__)(registeredName);
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
// Object Wrapper
|
||||
|
||||
template <typename T> class ObjectWrapper {
|
||||
public:
|
||||
ObjectWrapper(const std::string &className) {
|
||||
ObjectFactory::Instance().Register(className,
|
||||
[]() -> Object * { return new T(); });
|
||||
}
|
||||
|
||||
ObjectWrapper(T *model) : m_model(model) {}
|
||||
ObjectWrapper(T &model) : m_model(model) {}
|
||||
|
||||
template <typename U = T,
|
||||
typename = std::enable_if_t<std::is_default_constructible_v<U>>>
|
||||
ObjectWrapper() : m_model(new T()) {}
|
||||
|
||||
ObjectWrapper(const ObjectWrapper &other) : m_model(other.m_model) {}
|
||||
|
||||
ObjectWrapper &operator=(const ObjectWrapper &other) {
|
||||
m_model = other.m_model;
|
||||
return *this;
|
||||
}
|
||||
|
||||
ObjectWrapper(ObjectWrapper &&other) noexcept
|
||||
: m_model(std::move(other.m_model)) {}
|
||||
|
||||
ObjectWrapper &operator=(ObjectWrapper &&other) noexcept {
|
||||
m_model = std::move(other.m_model);
|
||||
return *this;
|
||||
}
|
||||
|
||||
~ObjectWrapper() = default;
|
||||
|
||||
T *operator->() const { return m_model.get(); }
|
||||
|
||||
T &operator*() const { return *m_model; }
|
||||
|
||||
T *get() const { return m_model.get(); }
|
||||
|
||||
bool operator==(const ObjectWrapper &other) const {
|
||||
return m_model == other.m_model;
|
||||
}
|
||||
|
||||
bool operator!=(const ObjectWrapper &other) const {
|
||||
return m_model != other.m_model;
|
||||
}
|
||||
|
||||
explicit operator bool() const { return m_model != nullptr; }
|
||||
|
||||
protected:
|
||||
SmartPointer<T> m_model;
|
||||
};
|
||||
#define ULIB_REGISTER_OBJECT_NAME(className, registeredName) \
|
||||
static uLib::ObjectRegistrar<className> ULIB_REG_CONCAT(g_ObjectRegistrar_, __LINE__)(registeredName);
|
||||
|
||||
} // namespace uLib
|
||||
|
||||
|
||||
@@ -5,23 +5,19 @@
|
||||
#include <vector>
|
||||
#include <sstream>
|
||||
#include <typeinfo>
|
||||
#include <typeindex>
|
||||
#include <typeindex> // Added
|
||||
#include <boost/serialization/nvp.hpp>
|
||||
#include <boost/lexical_cast.hpp>
|
||||
#include <vector>
|
||||
#include <boost/type_traits/is_class.hpp>
|
||||
#include <boost/mpl/bool.hpp>
|
||||
#include <set>
|
||||
#include <boost/type_traits/is_base_of.hpp>
|
||||
#include <boost/serialization/serialization.hpp>
|
||||
#include "Core/Archives.h"
|
||||
#include "Core/Signal.h"
|
||||
#include "Core/Object.h"
|
||||
|
||||
namespace uLib {
|
||||
|
||||
namespace Archive {
|
||||
class property_register_archive;
|
||||
}
|
||||
|
||||
/**
|
||||
* @brief Base class for properties to allow runtime listing and identification.
|
||||
*/
|
||||
@@ -31,7 +27,7 @@ public:
|
||||
virtual const std::string& GetName() const = 0;
|
||||
virtual const char* GetTypeName() const = 0;
|
||||
virtual std::string GetValueAsString() const = 0;
|
||||
virtual std::type_index GetTypeIndex() const = 0;
|
||||
virtual std::type_index GetTypeIndex() const = 0; // Added
|
||||
virtual const std::string& GetUnits() const = 0;
|
||||
virtual void SetUnits(const std::string& units) = 0;
|
||||
virtual const std::vector<std::string>& GetEnumLabels() const {
|
||||
@@ -53,6 +49,9 @@ public:
|
||||
return GetGroup() + "." + GetName();
|
||||
}
|
||||
|
||||
// Signal support
|
||||
signals:
|
||||
virtual void Updated() override { ULIB_SIGNAL_EMIT(PropertyBase::Updated); }
|
||||
|
||||
// Serialization support for different uLib archives
|
||||
virtual void serialize(Archive::xml_oarchive & ar, const unsigned int version) override = 0;
|
||||
@@ -62,29 +61,42 @@ public:
|
||||
virtual void serialize(Archive::hrt_oarchive & ar, const unsigned int version) override = 0;
|
||||
virtual void serialize(Archive::hrt_iarchive & ar, const unsigned int version) override = 0;
|
||||
virtual void serialize(Archive::log_archive & ar, const unsigned int version) override = 0;
|
||||
|
||||
virtual void serialize(Archive::property_register_archive & ar, const unsigned int v) = 0;
|
||||
};
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
/**
|
||||
* @brief Template class for typed properties.
|
||||
*/
|
||||
template <typename T>
|
||||
class Property : public PropertyBase {
|
||||
public:
|
||||
// PROXY: Use an existing variable as back-end storage
|
||||
Property(Object* owner, const std::string& name, T* valuePtr, const std::string& units = "", const std::string& group = "")
|
||||
: m_owner(owner), m_name(name), m_units(units), m_group(group), m_value(valuePtr), m_own(false),
|
||||
m_HasRange(false), m_HasDefault(false), m_ReadOnly(false) {
|
||||
if (m_owner) m_owner->RegisterProperty(this);
|
||||
if (m_owner) {
|
||||
m_owner->RegisterProperty(this);
|
||||
}
|
||||
}
|
||||
|
||||
// MANAGED: Create and own internal storage
|
||||
Property(Object* owner, const std::string& name, const T& defaultValue = T(), const std::string& units = "", const std::string& group = "")
|
||||
: m_owner(owner), m_name(name), m_units(units), m_group(group), m_value(new T(defaultValue)), m_own(true),
|
||||
m_HasRange(false), m_HasDefault(true), m_Default(defaultValue), m_ReadOnly(false) {
|
||||
if (m_owner) m_owner->RegisterProperty(this);
|
||||
if (m_owner) {
|
||||
m_owner->RegisterProperty(this);
|
||||
}
|
||||
}
|
||||
|
||||
virtual ~Property() { if (m_own) delete m_value; }
|
||||
virtual ~Property() {
|
||||
if (m_own) delete m_value;
|
||||
}
|
||||
|
||||
// Identification
|
||||
virtual const std::string& GetName() const override { return m_name; }
|
||||
@@ -95,19 +107,36 @@ public:
|
||||
virtual const std::string& GetGroup() const override { return m_group; }
|
||||
virtual void SetGroup(const std::string& group) override { m_group = group; }
|
||||
|
||||
|
||||
std::string GetValueAsString() const override {
|
||||
try { return boost::lexical_cast<std::string>(*m_value); }
|
||||
catch (...) { std::stringstream ss; ss << *m_value; return ss.str(); }
|
||||
try {
|
||||
return boost::lexical_cast<std::string>(*m_value);
|
||||
} catch (const boost::bad_lexical_cast&) {
|
||||
std::stringstream ss;
|
||||
ss << *m_value;
|
||||
return ss.str();
|
||||
}
|
||||
}
|
||||
|
||||
// Accessors
|
||||
const T& Get() const { return *m_value; }
|
||||
void Set(const T& value) {
|
||||
if (!m_value) return;
|
||||
T val = value;
|
||||
if constexpr (std::is_arithmetic<T>::value) {
|
||||
if (m_HasRange) { if (val < m_Min) val = m_Min; if (val > m_Max) val = m_Max; }
|
||||
template<typename U = T>
|
||||
typename std::enable_if<std::is_arithmetic<U>::value, void>::type
|
||||
ValidateT(T& val) {
|
||||
if (m_HasRange) {
|
||||
if (val < m_Min) val = m_Min;
|
||||
if (val > m_Max) val = m_Max;
|
||||
}
|
||||
}
|
||||
|
||||
template<typename U = T>
|
||||
typename std::enable_if<!std::is_arithmetic<U>::value, void>::type
|
||||
ValidateT(T& val) {
|
||||
}
|
||||
|
||||
void Set(const T& value) {
|
||||
T val = value;
|
||||
ValidateT<T>(val);
|
||||
if (*m_value != val) {
|
||||
*m_value = val;
|
||||
ULIB_SIGNAL_EMIT(Property<T>::PropertyChanged);
|
||||
@@ -121,54 +150,65 @@ public:
|
||||
void SetReadOnly(bool ro) { m_ReadOnly = ro; }
|
||||
|
||||
virtual bool IsReadOnly() const override { return m_ReadOnly; }
|
||||
|
||||
virtual bool HasRange() const override { return m_HasRange; }
|
||||
virtual double GetMin() const override { return m_HasRange ? convert_to_double(m_Min) : 0.0; }
|
||||
virtual double GetMax() const override { return m_HasRange ? convert_to_double(m_Max) : 0.0; }
|
||||
|
||||
const T& GetMinTyped() const { return m_Min; }
|
||||
const T& GetMaxTyped() const { return m_Max; }
|
||||
|
||||
template<typename U = T>
|
||||
typename std::enable_if<std::is_arithmetic<U>::value, double>::type
|
||||
GetMinT() const { return (double)m_Min; }
|
||||
|
||||
template<typename U = T>
|
||||
typename std::enable_if<!std::is_arithmetic<U>::value, double>::type
|
||||
GetMinT() const { return 0.0; }
|
||||
|
||||
template<typename U = T>
|
||||
typename std::enable_if<std::is_arithmetic<U>::value, double>::type
|
||||
GetMaxT() const { return (double)m_Max; }
|
||||
|
||||
template<typename U = T>
|
||||
typename std::enable_if<!std::is_arithmetic<U>::value, double>::type
|
||||
GetMaxT() const { return 0.0; }
|
||||
|
||||
virtual double GetMin() const override { return GetMinT<T>(); }
|
||||
virtual double GetMax() const override { return GetMaxT<T>(); }
|
||||
|
||||
virtual bool HasDefault() const override { return m_HasDefault; }
|
||||
virtual std::string GetDefaultValueAsString() const override {
|
||||
try { return boost::lexical_cast<std::string>(m_Default); } catch (...) { return ""; }
|
||||
try { return boost::lexical_cast<std::string>(m_Default); }
|
||||
catch (...) { return ""; }
|
||||
}
|
||||
|
||||
// Operators
|
||||
// Operators for seamless usage
|
||||
operator const T&() const { return *m_value; }
|
||||
Property& operator=(const T& value) { Set(value); return *this; }
|
||||
Property& operator=(const T& value) {
|
||||
Set(value);
|
||||
return *this;
|
||||
}
|
||||
|
||||
// Signals
|
||||
signals:
|
||||
virtual void PropertyChanged() { ULIB_SIGNAL_EMIT(Property<T>::PropertyChanged); }
|
||||
|
||||
private:
|
||||
template <typename U>
|
||||
static double convert_to_double(const U& val) {
|
||||
return convert_to_double_impl(val, typename std::is_arithmetic<U>::type());
|
||||
}
|
||||
template <typename U>
|
||||
static double convert_to_double_impl(const U& val, std::true_type) { return (double)val; }
|
||||
template <typename U>
|
||||
static double convert_to_double_impl(const U& val, std::false_type) { return 0.0; }
|
||||
|
||||
public:
|
||||
// Serialization
|
||||
template <class ArchiveT>
|
||||
void serialize_helper(ArchiveT & ar, const unsigned int version) {
|
||||
ar & boost::serialization::make_hrp(m_name.c_str(), *m_value, m_units.c_str());
|
||||
void serialize_impl(ArchiveT & ar, const unsigned int version) {
|
||||
ar & boost::serialization::make_nvp(m_name.c_str(), *m_value);
|
||||
}
|
||||
|
||||
virtual void serialize(Archive::xml_oarchive & ar, const unsigned int v) override { serialize_helper(ar, v); }
|
||||
virtual void serialize(Archive::xml_iarchive & ar, const unsigned int v) override { serialize_helper(ar, v); }
|
||||
virtual void serialize(Archive::text_oarchive & ar, const unsigned int v) override { serialize_helper(ar, v); }
|
||||
virtual void serialize(Archive::text_iarchive & ar, const unsigned int v) override { serialize_helper(ar, v); }
|
||||
virtual void serialize(Archive::hrt_oarchive & ar, const unsigned int v) override { serialize_helper(ar, v); }
|
||||
virtual void serialize(Archive::hrt_iarchive & ar, const unsigned int v) override { serialize_helper(ar, v); }
|
||||
virtual void serialize(Archive::log_archive & ar, const unsigned int v) override { serialize_helper(ar, v); }
|
||||
|
||||
virtual void serialize(Archive::property_register_archive & ar, const unsigned int v) override;
|
||||
void serialize(Archive::xml_oarchive & ar, const unsigned int v) override { serialize_impl(ar, v); }
|
||||
void serialize(Archive::xml_iarchive & ar, const unsigned int v) override { serialize_impl(ar, v); }
|
||||
void serialize(Archive::text_oarchive & ar, const unsigned int v) override { serialize_impl(ar, v); }
|
||||
void serialize(Archive::text_iarchive & ar, const unsigned int v) override { serialize_impl(ar, v); }
|
||||
void serialize(Archive::hrt_oarchive & ar, const unsigned int v) override { serialize_impl(ar, v); }
|
||||
void serialize(Archive::hrt_iarchive & ar, const unsigned int v) override { serialize_impl(ar, v); }
|
||||
void serialize(Archive::log_archive & ar, const unsigned int v) override { serialize_impl(ar, v); }
|
||||
|
||||
virtual void Updated() override {
|
||||
PropertyBase::Updated();
|
||||
this->PropertyChanged();
|
||||
}
|
||||
|
||||
protected:
|
||||
private:
|
||||
std::string m_name;
|
||||
std::string m_units;
|
||||
std::string m_group;
|
||||
@@ -184,7 +224,20 @@ protected:
|
||||
};
|
||||
|
||||
/**
|
||||
* @brief Property specialized for enumerations.
|
||||
* @brief Conveninent typedefs for common property types.
|
||||
*/
|
||||
typedef Property<std::string> StringProperty;
|
||||
typedef Property<int> IntProperty;
|
||||
typedef Property<unsigned int> UIntProperty;
|
||||
typedef Property<long> LongProperty;
|
||||
typedef Property<unsigned long> ULongProperty;
|
||||
typedef Property<float> FloatProperty;
|
||||
typedef Property<double> DoubleProperty;
|
||||
typedef Property<Bool_t> BoolProperty;
|
||||
|
||||
|
||||
/**
|
||||
* @brief Property specialized for enumerations, providing labels for GUI representations.
|
||||
*/
|
||||
class EnumProperty : public Property<int> {
|
||||
public:
|
||||
@@ -195,49 +248,68 @@ public:
|
||||
const char* GetTypeName() const override { return "Enum"; }
|
||||
virtual std::type_index GetTypeIndex() const override { return std::type_index(typeid(EnumProperty)); }
|
||||
|
||||
template <class ArchiveT>
|
||||
void serialize_enum_helper(ArchiveT & ar, const unsigned int version) {
|
||||
ar & boost::serialization::make_hrp_enum(m_name.c_str(), *m_value, m_Labels, m_units.c_str());
|
||||
}
|
||||
|
||||
virtual void serialize(Archive::xml_oarchive & ar, const unsigned int v) override { serialize_enum_helper(ar, v); }
|
||||
virtual void serialize(Archive::xml_iarchive & ar, const unsigned int v) override { serialize_enum_helper(ar, v); }
|
||||
virtual void serialize(Archive::text_oarchive & ar, const unsigned int v) override { serialize_enum_helper(ar, v); }
|
||||
virtual void serialize(Archive::text_iarchive & ar, const unsigned int v) override { serialize_enum_helper(ar, v); }
|
||||
virtual void serialize(Archive::hrt_oarchive & ar, const unsigned int v) override { serialize_enum_helper(ar, v); }
|
||||
virtual void serialize(Archive::hrt_iarchive & ar, const unsigned int v) override { serialize_enum_helper(ar, v); }
|
||||
virtual void serialize(Archive::log_archive & ar, const unsigned int v) override { serialize_enum_helper(ar, v); }
|
||||
|
||||
virtual void serialize(Archive::property_register_archive & ar, const unsigned int v) override;
|
||||
|
||||
private:
|
||||
std::vector<std::string> m_Labels;
|
||||
};
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
/**
|
||||
* @brief Macro to simplify property declaration within a class.
|
||||
* Usage: ULIB_PROPERTY(float, Width, 1.0f)
|
||||
* It creates a raw member m_Width and a Property proxy Width.
|
||||
*/
|
||||
#define ULIB_PROPERTY(type, name, defaultValue) \
|
||||
type m_##name = defaultValue; \
|
||||
Property<type> name = Property<type>(this, #name, &m_##name);
|
||||
|
||||
|
||||
} // namespace uLib
|
||||
|
||||
namespace uLib {
|
||||
namespace Archive {
|
||||
class property_register_archive;
|
||||
} // namespace Archive
|
||||
} // namespace uLib
|
||||
|
||||
class property_register_archive
|
||||
: public boost::archive::detail::common_oarchive<property_register_archive> {
|
||||
protected:
|
||||
namespace boost {
|
||||
namespace archive {
|
||||
namespace detail {
|
||||
template <>
|
||||
class interface_oarchive<uLib::Archive::property_register_archive>
|
||||
: public uLib_interface_oarchive<uLib::Archive::property_register_archive> {};
|
||||
} // namespace detail
|
||||
} // namespace archive
|
||||
} // namespace boost
|
||||
|
||||
namespace uLib {
|
||||
namespace Archive {
|
||||
|
||||
/**
|
||||
* @brief A special archive that creates and registers Property proxies
|
||||
* for any member it encounters wrapped in HRP().
|
||||
*/
|
||||
class property_register_archive :
|
||||
public boost::archive::detail::common_oarchive<property_register_archive>
|
||||
{
|
||||
Object* m_Object;
|
||||
bool m_DisplayOnly;
|
||||
public:
|
||||
friend class boost::archive::detail::interface_oarchive<property_register_archive>;
|
||||
friend class boost::archive::save_access;
|
||||
|
||||
using boost::archive::detail::common_oarchive<property_register_archive>::save_override;
|
||||
typedef boost::archive::detail::common_oarchive<property_register_archive> detail_common_oarchive;
|
||||
|
||||
property_register_archive(Object* obj, bool displayOnly = false) :
|
||||
property_register_archive(Object* obj) :
|
||||
boost::archive::detail::common_oarchive<property_register_archive>(boost::archive::no_header),
|
||||
m_Object(obj), m_DisplayOnly(displayOnly) {
|
||||
if (obj) m_Visited.insert(dynamic_cast<const void*>(obj));
|
||||
}
|
||||
|
||||
template<class T> property_register_archive &operator&(const T &t) { this->save_override(t); return *this; }
|
||||
template<class T> property_register_archive &operator<<(const T &t) { this->save_override(t); return *this; }
|
||||
m_Object(obj) {}
|
||||
|
||||
std::string GetCurrentGroup() const {
|
||||
std::string group;
|
||||
@@ -248,91 +320,77 @@ public:
|
||||
return group;
|
||||
}
|
||||
|
||||
template<class T> void register_property(Property<T>& p) {
|
||||
save_property_impl(p.GetName().c_str(), const_cast<T&>(p.Get()), p.GetUnits().c_str(),
|
||||
p.HasRange(), p.GetMinTyped(), p.GetMaxTyped(), p.IsReadOnly());
|
||||
}
|
||||
|
||||
void register_enum_property(EnumProperty& p) {
|
||||
if (!m_Object) return;
|
||||
EnumProperty* newP = new EnumProperty(m_Object, p.GetName(), const_cast<int*>(&p.Get()), p.GetEnumLabels(), p.GetUnits(), GetCurrentGroup());
|
||||
newP->SetReadOnly(p.IsReadOnly());
|
||||
if (m_DisplayOnly) {
|
||||
m_Object->RegisterDisplayProperty(newP);
|
||||
Object* obj = m_Object;
|
||||
Object::connect(newP, &Object::Updated, [obj]() { obj->Updated(); });
|
||||
} else {
|
||||
m_Object->RegisterDynamicProperty(newP);
|
||||
}
|
||||
}
|
||||
|
||||
template<class T> void save_property_impl(const char* name, T& val, const char* units, bool hasRange, const T& minVal, const T& maxVal, bool isReadOnly) {
|
||||
// Core logic: encounter HRP -> Create Dynamic Property
|
||||
template<class T>
|
||||
void save_override(const boost::serialization::hrp<T> &t) {
|
||||
if (m_Object) {
|
||||
Property<T>* p = new Property<T>(m_Object, name, &val, units ? units : "", GetCurrentGroup());
|
||||
set_range_helper(p, hasRange, minVal, maxVal, typename std::is_arithmetic<T>::type());
|
||||
p->SetReadOnly(isReadOnly);
|
||||
if (m_DisplayOnly) {
|
||||
m_Object->RegisterDisplayProperty(p);
|
||||
Object* obj = m_Object;
|
||||
Object::connect(p, &Object::Updated, [obj]() { obj->Updated(); });
|
||||
} else {
|
||||
m_Object->RegisterDynamicProperty(p);
|
||||
}
|
||||
Property<T>* p = new Property<T>(m_Object, t.name(), &const_cast<boost::serialization::hrp<T>&>(t).value(), t.units() ? t.units() : "", GetCurrentGroup());
|
||||
if (t.has_range()) p->SetRange(t.min_val(), t.max_val());
|
||||
if (t.has_default()) p->SetDefault(t.default_val());
|
||||
p->SetReadOnly(t.is_read_only());
|
||||
m_Object->RegisterDynamicProperty(p);
|
||||
}
|
||||
}
|
||||
|
||||
template<class U> static void set_range_helper(Property<U>* p, bool hasRange, const U& minVal, const U& maxVal, std::true_type) { if (hasRange) p->SetRange(minVal, maxVal); }
|
||||
template<class U> static void set_range_helper(Property<U>* p, bool hasRange, const U& minVal, const U& maxVal, std::false_type) {}
|
||||
template<class T>
|
||||
void save_override(const boost::serialization::hrp_val<T> &t) {
|
||||
if (m_Object) {
|
||||
// Note: hrp_val stores by value. Property usually points to existing data.
|
||||
// But here we are registering properties from HRP wrappers.
|
||||
// If it's hrp_val, it means it's an rvalue from a getter.
|
||||
// The hrp_val wrapper itself owns the value.
|
||||
// However, the property_register_archive is temporary.
|
||||
// This is a bit tricky. Usually HRP(rvalue) is meant for read-only display.
|
||||
// Let's use the address of the value in the wrapper, but mark it read-only.
|
||||
Property<T>* p = new Property<T>(m_Object, t.name(), &const_cast<boost::serialization::hrp_val<T>&>(t).value(), t.units() ? t.units() : "", GetCurrentGroup());
|
||||
if (t.has_range()) p->SetRange(t.min_val(), t.max_val());
|
||||
if (t.has_default()) p->SetDefault(t.default_val());
|
||||
p->SetReadOnly(t.is_read_only());
|
||||
m_Object->RegisterDynamicProperty(p);
|
||||
}
|
||||
}
|
||||
|
||||
template<class T> void save_override(const boost::serialization::hrp<T> &t) {
|
||||
// To handle T correctly without deduction issues, we assume T can be passed to save_property_impl
|
||||
T dummy = T(); // Ensure we can construct T
|
||||
save_property_impl(t.name(), const_cast<boost::serialization::hrp<T>&>(t).value(), t.units(), t.has_range(), t.has_range() ? t.min_val() : dummy, t.has_range() ? t.max_val() : dummy, t.is_read_only());
|
||||
}
|
||||
template<class T> void save_override(const boost::serialization::hrp_val<T> &t) {
|
||||
T dummy = T();
|
||||
save_property_impl(t.name(), const_cast<boost::serialization::hrp_val<T>&>(t).value(), t.units(), t.has_range(), t.has_range() ? t.min_val() : dummy, t.has_range() ? t.max_val() : dummy, t.is_read_only());
|
||||
}
|
||||
template<class T> void save_override(const boost::serialization::hrp_enum<T> &t) {
|
||||
template<class T>
|
||||
void save_override(const boost::serialization::hrp_enum<T> &t) {
|
||||
if (m_Object) {
|
||||
EnumProperty* p = new EnumProperty(m_Object, t.name(), (int*)&const_cast<boost::serialization::hrp_enum<T>&>(t).value(), t.labels(), t.units() ? t.units() : "", GetCurrentGroup());
|
||||
p->SetReadOnly(t.is_read_only());
|
||||
if (m_DisplayOnly) { m_Object->RegisterDisplayProperty(p); Object* obj = m_Object; Object::connect(p, &Object::Updated, [obj]() { obj->Updated(); }); }
|
||||
else { m_Object->RegisterDynamicProperty(p); }
|
||||
}
|
||||
}
|
||||
template<class T> void save_override(const boost::serialization::hrp_enum_val<T> &t) {
|
||||
if (m_Object) {
|
||||
EnumProperty* p = new EnumProperty(m_Object, t.name(), (int*)&const_cast<boost::serialization::hrp_enum_val<T>&>(t).value(), t.labels(), t.units() ? t.units() : "", GetCurrentGroup());
|
||||
p->SetReadOnly(t.is_read_only());
|
||||
if (m_DisplayOnly) { m_Object->RegisterDisplayProperty(p); Object* obj = m_Object; Object::connect(p, &Object::Updated, [obj]() { obj->Updated(); }); }
|
||||
else { m_Object->RegisterDynamicProperty(p); }
|
||||
m_Object->RegisterDynamicProperty(p);
|
||||
}
|
||||
}
|
||||
|
||||
template<class T> void save_override(const boost::serialization::nvp<T> &t) {
|
||||
template<class T>
|
||||
void save_override(const boost::serialization::hrp_enum_val<T> &t) {
|
||||
if (m_Object) {
|
||||
EnumProperty* p = new EnumProperty(m_Object, t.name(), (int*)&const_cast<boost::serialization::hrp_enum_val<T>&>(t).value(), t.labels(), t.units() ? t.units() : "", GetCurrentGroup());
|
||||
p->SetReadOnly(t.is_read_only());
|
||||
m_Object->RegisterDynamicProperty(p);
|
||||
}
|
||||
}
|
||||
|
||||
// Handle standard NVPs by recursing (important for base classes)
|
||||
template<class T>
|
||||
void save_override(const boost::serialization::nvp<T> &t) {
|
||||
if (t.name()) m_GroupStack.push_back(t.name());
|
||||
this->save_helper(t.const_value(), typename boost::is_class<T>::type());
|
||||
if (t.name()) m_GroupStack.pop_back();
|
||||
}
|
||||
|
||||
void save_override(const std::string &t) {}
|
||||
template<class T> void save_override(T * const & t) {
|
||||
if (!t) return;
|
||||
this->save_pointer_helper(t, typename boost::is_base_of<Object, T>::type());
|
||||
// Recursion for nested classes, ignore primitives
|
||||
template<class T>
|
||||
void save_override(const T &t) {
|
||||
this->save_helper(t, typename boost::is_class<T>::type());
|
||||
}
|
||||
template<class T> void save_pointer_helper(T* t, boost::mpl::true_) {
|
||||
const void* ptr = dynamic_cast<const void*>(t);
|
||||
if (m_Visited.find(ptr) != m_Visited.end()) return;
|
||||
m_Visited.insert(ptr);
|
||||
this->save_override(*t);
|
||||
}
|
||||
template<class T> void save_pointer_helper(T* t, boost::mpl::false_) {}
|
||||
template<class T> void save_override(const T &t) { this->save_helper(t, typename boost::is_class<T>::type()); }
|
||||
template<class T> void save_helper(const T &t, boost::mpl::true_) { boost::serialization::serialize_adl(*this, const_cast<T&>(t), 0); }
|
||||
void save_helper(const std::string &t, boost::mpl::true_) {}
|
||||
template<class T> void save_helper(const T &t, boost::mpl::false_) {}
|
||||
|
||||
template<class T>
|
||||
void save_helper(const T &t, boost::mpl::true_) {
|
||||
boost::serialization::serialize_adl(*this, const_cast<T&>(t), 0);
|
||||
}
|
||||
|
||||
template<class T>
|
||||
void save_helper(const T &t, boost::mpl::false_) {}
|
||||
|
||||
// Required attribute overrides for common_oarchive
|
||||
void save_override(const boost::archive::object_id_type & t) {}
|
||||
void save_override(const boost::archive::object_reference_type & t) {}
|
||||
void save_override(const boost::archive::version_type & t) {}
|
||||
@@ -342,30 +400,30 @@ public:
|
||||
void save_override(const boost::archive::class_name_type & t) {}
|
||||
void save_override(const boost::archive::tracking_type & t) {}
|
||||
|
||||
protected:
|
||||
private:
|
||||
std::vector<std::string> m_GroupStack;
|
||||
std::set<const void*> m_Visited;
|
||||
};
|
||||
|
||||
} // namespace Archive
|
||||
} // namespace uLib
|
||||
|
||||
namespace uLib {
|
||||
|
||||
template <typename T>
|
||||
inline void Property<T>::serialize(Archive::property_register_archive & ar, const unsigned int v) {
|
||||
ar.register_property(*this);
|
||||
}
|
||||
|
||||
inline void EnumProperty::serialize(Archive::property_register_archive & ar, const unsigned int v) {
|
||||
ar.register_enum_property(*this);
|
||||
}
|
||||
|
||||
namespace Archive {
|
||||
|
||||
/**
|
||||
* @brief Convenience macro to automatically activate and register all HRP members
|
||||
* as uLib properties. Usage: ULIB_ACTIVATE_PROPERTIES(obj)
|
||||
*/
|
||||
#define ULIB_ACTIVATE_PROPERTIES(obj) \
|
||||
{ uLib::Archive::property_register_archive _ar_tmp(&(obj)); _ar_tmp & (obj); }
|
||||
|
||||
/**
|
||||
* @brief Declares a private member that automatically calls ULIB_ACTIVATE_PROPERTIES
|
||||
* in every constructor of the class. Place this macro as the last declaration
|
||||
* inside the class body (before the closing brace).
|
||||
*
|
||||
* Usage: ULIB_DECLARE_PROPERTIES(ClassName)
|
||||
*
|
||||
* This replaces per-constructor ULIB_ACTIVATE_PROPERTIES(*this) calls.
|
||||
* RegisterDynamicProperty deduplicates by qualified name, so re-registration
|
||||
* from inherited activators in a hierarchy is safe.
|
||||
*/
|
||||
#define ULIB_DECLARE_PROPERTIES(SelfType) \
|
||||
private: \
|
||||
struct _PropActivator { \
|
||||
@@ -376,31 +434,6 @@ private: \
|
||||
} _prop_activator{this};
|
||||
|
||||
} // namespace Archive
|
||||
|
||||
// Convenience macro: declares a named Property<T> member with a default value.
|
||||
// Usage inside a class body (requires 'this' to be available, so use in-class initializer):
|
||||
// ULIB_PROPERTY(int, MyProp, 42)
|
||||
#define ULIB_PROPERTY(type, name, defaultVal) \
|
||||
::uLib::Property<type> name{this, #name, (type)(defaultVal)};
|
||||
|
||||
// Common property type aliases
|
||||
typedef Property<bool> BoolProperty;
|
||||
typedef Property<int> IntProperty;
|
||||
typedef Property<float> FloatProperty;
|
||||
typedef Property<double> DoubleProperty;
|
||||
typedef Property<std::string> StringProperty;
|
||||
|
||||
template <class ArchiveT>
|
||||
void serialize_properties_helper(ArchiveT &ar, const std::vector<PropertyBase*> &props, unsigned int version) {
|
||||
for (auto* prop : props) prop->serialize(ar, version);
|
||||
}
|
||||
|
||||
template <class ArchiveT>
|
||||
void Object::serialize(ArchiveT &ar, const unsigned int version) {
|
||||
ar & boost::serialization::make_nvp("InstanceName", this->GetInstanceName());
|
||||
serialize_properties_helper(ar, this->GetProperties(), version);
|
||||
}
|
||||
|
||||
} // namespace uLib
|
||||
|
||||
#endif // U_CORE_PROPERTY_H
|
||||
|
||||
@@ -23,249 +23,89 @@
|
||||
|
||||
//////////////////////////////////////////////////////////////////////////////*/
|
||||
|
||||
|
||||
|
||||
#ifndef U_CORE_SMARTPOINTER_H
|
||||
#define U_CORE_SMARTPOINTER_H
|
||||
|
||||
#include <atomic>
|
||||
#include <functional>
|
||||
#include <type_traits>
|
||||
#include <utility>
|
||||
|
||||
namespace uLib {
|
||||
|
||||
/**
|
||||
* @brief A smart pointer implementation inspired by std::shared_ptr.
|
||||
*
|
||||
* Features modernized C++11/14/17 syntax, thread-safe reference counting,
|
||||
* move semantics, and support for custom deleters.
|
||||
*
|
||||
* NOTE: Default constructor allocates a new T following legacy behavior.
|
||||
*/
|
||||
template <typename T>
|
||||
class SmartPointer {
|
||||
typedef T element_type;
|
||||
public:
|
||||
using element_type = T;
|
||||
|
||||
/**
|
||||
* @brief Constructor from raw pointer.
|
||||
* If ptr is nullptr, a new T is allocated (legacy behavior).
|
||||
*/
|
||||
explicit SmartPointer(T* ptr = nullptr) : m_counter(nullptr) {
|
||||
if (!ptr) {
|
||||
if constexpr (std::is_default_constructible_v<T>) {
|
||||
ptr = new T();
|
||||
}
|
||||
}
|
||||
explicit
|
||||
SmartPointer(T* ptr = NULL) : m_counter(0) {
|
||||
if(!ptr) ptr = new T;
|
||||
if (ptr) m_counter = new ReferenceCounter(ptr);
|
||||
}
|
||||
|
||||
/**
|
||||
* @brief Constructor with custom deleter.
|
||||
*/
|
||||
template <typename D>
|
||||
SmartPointer(T* ptr, D deleter) : m_counter(nullptr) {
|
||||
if (ptr) m_counter = new ReferenceCounter(ptr, deleter);
|
||||
// TakeReference //
|
||||
SmartPointer(T &ref) : m_counter(new ReferenceCounter(&ref,0)) { }
|
||||
|
||||
SmartPointer(const SmartPointer& copy) throw () {
|
||||
acquire(copy.m_counter);
|
||||
}
|
||||
|
||||
/**
|
||||
* @brief Non-owning constructor from reference.
|
||||
* Uses a no-op deleter to ensure the referenced object is not destroyed.
|
||||
*/
|
||||
SmartPointer(T &ref) : m_counter(new ReferenceCounter(&ref, [](T*){}, 1)) { }
|
||||
|
||||
/**
|
||||
* @brief Copy constructor.
|
||||
*/
|
||||
SmartPointer(const SmartPointer& other) noexcept : m_counter(nullptr) {
|
||||
acquire(other.m_counter);
|
||||
SmartPointer(SmartPointer * copy) throw () {
|
||||
acquire(copy->m_counter);
|
||||
}
|
||||
|
||||
/**
|
||||
* @brief Copy constructor from a pointer to SmartPointer (Legacy support).
|
||||
*/
|
||||
SmartPointer(const SmartPointer* other) noexcept : m_counter(nullptr) {
|
||||
if (other) acquire(other->m_counter);
|
||||
}
|
||||
virtual ~SmartPointer() { release(); }
|
||||
|
||||
/**
|
||||
* @brief Move constructor.
|
||||
*/
|
||||
SmartPointer(SmartPointer&& other) noexcept : m_counter(other.m_counter) {
|
||||
other.m_counter = nullptr;
|
||||
}
|
||||
|
||||
/**
|
||||
* @brief Virtual destructor.
|
||||
*/
|
||||
virtual ~SmartPointer() { release(); }
|
||||
|
||||
/**
|
||||
* @brief Copy assignment.
|
||||
*/
|
||||
SmartPointer& operator=(const SmartPointer& other) noexcept {
|
||||
if (this != &other) {
|
||||
SmartPointer & operator=(const SmartPointer& copy) {
|
||||
if (this != ©)
|
||||
{
|
||||
release();
|
||||
acquire(other.m_counter);
|
||||
acquire(copy.m_counter);
|
||||
}
|
||||
return *this;
|
||||
}
|
||||
|
||||
/**
|
||||
* @brief Move assignment.
|
||||
*/
|
||||
SmartPointer& operator=(SmartPointer&& other) noexcept {
|
||||
if (this != &other) {
|
||||
release();
|
||||
m_counter = other.m_counter;
|
||||
other.m_counter = nullptr;
|
||||
}
|
||||
return *this;
|
||||
}
|
||||
T & operator*() const throw () { return *m_counter->ptr; }
|
||||
T * operator->() const throw () { return m_counter->ptr; }
|
||||
|
||||
/**
|
||||
* @brief Resets the smart pointer to hold a new raw pointer.
|
||||
*/
|
||||
void reset(T* ptr = nullptr) {
|
||||
release();
|
||||
if (ptr) m_counter = new ReferenceCounter(ptr);
|
||||
}
|
||||
T * get() const throw () {
|
||||
return m_counter ? m_counter->ptr : 0; }
|
||||
bool unique() const throw () {
|
||||
return (m_counter ? m_counter->count == 1 : true); }
|
||||
|
||||
/**
|
||||
* @brief Resets the smart pointer with a custom deleter.
|
||||
*/
|
||||
template <typename D>
|
||||
void reset(T* ptr, D deleter) {
|
||||
release();
|
||||
if (ptr) m_counter = new ReferenceCounter(ptr, deleter);
|
||||
}
|
||||
|
||||
/**
|
||||
* @brief Swaps contents with another SmartPointer.
|
||||
*/
|
||||
void swap(SmartPointer& other) noexcept {
|
||||
std::swap(m_counter, other.m_counter);
|
||||
}
|
||||
private:
|
||||
|
||||
/**
|
||||
* @brief Dereference operator.
|
||||
*/
|
||||
T& operator*() const noexcept { return *m_counter->ptr; }
|
||||
struct ReferenceCounter
|
||||
{
|
||||
ReferenceCounter(T* ptr = 0, unsigned c = 1) : ptr(ptr), count(c) { }
|
||||
T* ptr;
|
||||
unsigned count;
|
||||
} * m_counter;
|
||||
|
||||
/**
|
||||
* @brief Member access operator.
|
||||
*/
|
||||
T* operator->() const noexcept { return m_counter->ptr; }
|
||||
|
||||
/**
|
||||
* @brief Returns the raw pointer.
|
||||
*/
|
||||
T* get() const noexcept { return m_counter ? m_counter->ptr : nullptr; }
|
||||
|
||||
/**
|
||||
* @brief Implicit conversion to raw pointer (legacy compatibility).
|
||||
*/
|
||||
operator T*() const noexcept { return get(); }
|
||||
|
||||
/**
|
||||
* @brief Returns the number of SmartPointers sharing ownership.
|
||||
*/
|
||||
uint32_t use_count() const noexcept {
|
||||
return m_counter ? m_counter->count.load(std::memory_order_relaxed) : 0;
|
||||
}
|
||||
|
||||
/**
|
||||
* @brief Returns true if this is the only SmartPointer owning the resource.
|
||||
*/
|
||||
bool unique() const noexcept { return use_count() == 1; }
|
||||
|
||||
/**
|
||||
* @brief Boolean conversion operator.
|
||||
*/
|
||||
explicit operator bool() const noexcept { return get() != nullptr; }
|
||||
|
||||
private:
|
||||
struct ReferenceCounter {
|
||||
T* ptr;
|
||||
std::atomic<uint32_t> count;
|
||||
std::function<void(T*)> deleter;
|
||||
|
||||
ReferenceCounter(T* p, uint32_t initial_count = 1)
|
||||
: ptr(p), count(initial_count), deleter([](T* ptr_to_del) { delete ptr_to_del; }) {}
|
||||
|
||||
template <typename D>
|
||||
ReferenceCounter(T* p, D d, uint32_t initial_count = 1)
|
||||
: ptr(p), count(initial_count), deleter(d) {}
|
||||
};
|
||||
|
||||
ReferenceCounter* m_counter;
|
||||
|
||||
void acquire(ReferenceCounter* c) noexcept {
|
||||
// increment the count
|
||||
void acquire(ReferenceCounter* c) throw ()
|
||||
{
|
||||
m_counter = c;
|
||||
if (c) {
|
||||
c->count.fetch_add(1, std::memory_order_relaxed);
|
||||
if (c && c->count>0) ++c->count;
|
||||
}
|
||||
|
||||
// decrement the count, delete if it is 0
|
||||
void release() {
|
||||
if (m_counter) {
|
||||
if (--m_counter->count == 0) {
|
||||
delete m_counter->ptr;
|
||||
}
|
||||
if (m_counter->count <= 0) {
|
||||
delete m_counter;
|
||||
m_counter = NULL;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
void release() noexcept {
|
||||
if (m_counter) {
|
||||
if (m_counter->count.fetch_sub(1, std::memory_order_acq_rel) == 1) {
|
||||
if (m_counter->ptr) {
|
||||
m_counter->deleter(m_counter->ptr);
|
||||
}
|
||||
delete m_counter;
|
||||
}
|
||||
m_counter = nullptr;
|
||||
}
|
||||
}
|
||||
};
|
||||
|
||||
/**
|
||||
* @brief Global swap for SmartPointer.
|
||||
*/
|
||||
template <typename T>
|
||||
void swap(SmartPointer<T>& a, SmartPointer<T>& b) noexcept {
|
||||
a.swap(b);
|
||||
|
||||
|
||||
}
|
||||
|
||||
/**
|
||||
* @brief Equality comparison.
|
||||
*/
|
||||
template <typename T, typename U>
|
||||
bool operator==(const SmartPointer<T>& a, const SmartPointer<U>& b) noexcept {
|
||||
return a.get() == b.get();
|
||||
}
|
||||
|
||||
/**
|
||||
* @brief Inequality comparison.
|
||||
*/
|
||||
template <typename T, typename U>
|
||||
bool operator!=(const SmartPointer<T>& a, const SmartPointer<U>& b) noexcept {
|
||||
return a.get() != b.get();
|
||||
}
|
||||
|
||||
/**
|
||||
* @brief Comparison with nullptr.
|
||||
*/
|
||||
template <typename T>
|
||||
bool operator==(const SmartPointer<T>& a, std::nullptr_t) noexcept {
|
||||
return a.get() == nullptr;
|
||||
}
|
||||
|
||||
template <typename T>
|
||||
bool operator==(std::nullptr_t, const SmartPointer<T>& a) noexcept {
|
||||
return a.get() == nullptr;
|
||||
}
|
||||
|
||||
template <typename T>
|
||||
bool operator!=(const SmartPointer<T>& a, std::nullptr_t) noexcept {
|
||||
return a.get() != nullptr;
|
||||
}
|
||||
|
||||
template <typename T>
|
||||
bool operator!=(std::nullptr_t, const SmartPointer<T>& a) noexcept {
|
||||
return a.get() != nullptr;
|
||||
}
|
||||
|
||||
} // namespace uLib
|
||||
|
||||
#endif // U_CORE_SMARTPOINTER_H
|
||||
#endif // SMARTPOINTER_H
|
||||
|
||||
@@ -139,6 +139,7 @@ typedef id_t Id_t;
|
||||
typedef void *Pointer_t;
|
||||
typedef bool Bool_t; // Boolean (0=false, 1=true) (bool)
|
||||
|
||||
|
||||
//--- bit manipulation ---------------------------------------------------------
|
||||
#ifndef BIT
|
||||
#define BIT(n) (1ULL << (n))
|
||||
@@ -181,7 +182,7 @@ typedef bool Bool_t; // Boolean (0=false, 1=true) (bool)
|
||||
\
|
||||
public: \
|
||||
typedef type_info::BaseClass BaseClass; \
|
||||
virtual const char *type_name() const override { return type_info::name; } \
|
||||
virtual const char *type_name() const override { return type_info::name; } \
|
||||
/**/
|
||||
|
||||
/**
|
||||
|
||||
@@ -2,8 +2,6 @@
|
||||
# TESTS
|
||||
set( TESTS
|
||||
SmartVectorTest
|
||||
SmartPointerTest
|
||||
ObjectWrapperTest
|
||||
VectorTest
|
||||
ObjectFlagsTest
|
||||
ObjectParametersTest
|
||||
|
||||
@@ -1,28 +0,0 @@
|
||||
#include "Core/ObjectFactory.h"
|
||||
#include <iostream>
|
||||
|
||||
struct NonDefault {
|
||||
NonDefault(int) {}
|
||||
};
|
||||
|
||||
struct Default {
|
||||
Default() : value(42) {}
|
||||
int value;
|
||||
};
|
||||
|
||||
int main() {
|
||||
std::cout << "Testing ObjectWrapper with Default Constructible type..." << std::endl;
|
||||
uLib::ObjectWrapper<Default> w1;
|
||||
|
||||
std::cout << "Testing ObjectWrapper with Non-Default Constructible type..." << std::endl;
|
||||
NonDefault nd(10);
|
||||
uLib::ObjectWrapper<NonDefault> w2(nd);
|
||||
|
||||
// The following would NOT compile without SFINAE:
|
||||
// uLib::ObjectWrapper<NonDefault> w3;
|
||||
|
||||
std::cout << "Tests passed (compilation and manual instantiation)!" << std::endl;
|
||||
return 0;
|
||||
}
|
||||
|
||||
|
||||
@@ -150,58 +150,11 @@ int testing_hrt_class() {
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
class B : public virtual Object {
|
||||
uLibTypeMacro(B, Object)
|
||||
ULIB_SERIALIZE_ACCESS
|
||||
|
||||
public:
|
||||
B() : m_b(1234567), m_a(nullptr) {}
|
||||
|
||||
int m_b;
|
||||
|
||||
// reference to A (serializes through ID)
|
||||
A *m_a;
|
||||
};
|
||||
|
||||
ULIB_SERIALIZABLE_OBJECT(B)
|
||||
ULIB_SERIALIZE_OBJECT(B, Object) {
|
||||
ar & "Object B : " & "--> m_b = " & AR(m_b) & "--> m_a = " & AR(m_a);
|
||||
}
|
||||
|
||||
|
||||
|
||||
int test_referece_serialization() {
|
||||
A a;
|
||||
{
|
||||
B b,b2 ;
|
||||
b.m_a = &a;
|
||||
b2.m_a = &a;
|
||||
|
||||
std::ofstream file("test_ref.xml");
|
||||
Archive::xml_oarchive(file) << NVP(b) << NVP(b2);
|
||||
}
|
||||
|
||||
|
||||
B b,b2;
|
||||
{
|
||||
std::ifstream file("test_ref.xml");
|
||||
Archive::xml_iarchive(file) >> NVP(b) >> NVP(b2);
|
||||
}
|
||||
|
||||
return (b.m_a->a() == a.a() && b2.m_a->a() == a.a());
|
||||
}
|
||||
|
||||
|
||||
|
||||
|
||||
int main() {
|
||||
BEGIN_TESTING(Serialize Test);
|
||||
|
||||
TEST1(test_V3f());
|
||||
TEST1(testing_xml_class());
|
||||
TEST1(test_referece_serialization());
|
||||
// TEST1(testing_hrt_class());
|
||||
|
||||
END_TESTING;
|
||||
|
||||
@@ -38,7 +38,7 @@ inline const unsigned long VectorSplice(const _Tp &_it, const _Tp &_end,
|
||||
|
||||
_Tp it = _it;
|
||||
_Tp end = _end - 1;
|
||||
for (; it != end;) {
|
||||
for (it; it != end;) {
|
||||
if (_comp(*it, value))
|
||||
it++;
|
||||
else if (_comp(*end, value)) {
|
||||
|
||||
@@ -39,7 +39,7 @@ EmitterPrimary::EmitterPrimary()
|
||||
// Initial position and direction through AffineTransform
|
||||
// 10m on Z axis, pointing towards origin
|
||||
this->SetPosition(Vector3f(0, 0, 10000.0));
|
||||
// Default orientation is identity (pointing along -Z if we rotate the prop3d accordingly)
|
||||
// Default orientation is identity (pointing along -Z if we rotate the puppet accordingly)
|
||||
// But fParticleGun defaults are set here and overridden in GeneratePrimaries
|
||||
}
|
||||
|
||||
|
||||
@@ -32,7 +32,7 @@ class EmitterPrimary : public G4VUserPrimaryGeneratorAction, public AffineTransf
|
||||
virtual ~EmitterPrimary();
|
||||
|
||||
// Metodo principale chiamato all'inizio di ogni evento
|
||||
virtual void GeneratePrimaries(G4Event*) override;
|
||||
virtual void GeneratePrimaries(G4Event*);
|
||||
|
||||
virtual void Updated() override { ULIB_SIGNAL_EMIT(EmitterPrimary::Updated); }
|
||||
|
||||
@@ -51,7 +51,7 @@ class SkyPlaneEmitterPrimary : public EmitterPrimary
|
||||
SkyPlaneEmitterPrimary();
|
||||
virtual ~SkyPlaneEmitterPrimary();
|
||||
|
||||
virtual void GeneratePrimaries(G4Event*) override;
|
||||
virtual void GeneratePrimaries(G4Event*);
|
||||
|
||||
void SetPlane(const uLib::Vector3f& p0, const uLib::Vector3f& normal);
|
||||
void SetSkySize(const uLib::Vector2f& size);
|
||||
@@ -72,7 +72,7 @@ class CylinderEmitterPrimary : public EmitterPrimary
|
||||
CylinderEmitterPrimary();
|
||||
virtual ~CylinderEmitterPrimary();
|
||||
|
||||
virtual void GeneratePrimaries(G4Event*) override;
|
||||
virtual void GeneratePrimaries(G4Event*);
|
||||
|
||||
void SetRadius(float r);
|
||||
float GetRadius() const { return m_Radius; }
|
||||
@@ -101,7 +101,7 @@ class QuadMeshEmitterPrimary : public EmitterPrimary
|
||||
virtual ~QuadMeshEmitterPrimary();
|
||||
|
||||
// Metodo principale chiamato all'inizio di ogni evento
|
||||
virtual void GeneratePrimaries(G4Event*) override;
|
||||
virtual void GeneratePrimaries(G4Event*);
|
||||
|
||||
void SetMesh(uLib::QuadMesh* mesh);
|
||||
|
||||
|
||||
@@ -35,14 +35,12 @@
|
||||
#include <Geant4/G4TriangularFacet.hh>
|
||||
#include <Geant4/G4Box.hh>
|
||||
#include <Geant4/G4PVPlacement.hh>
|
||||
#include <cstddef>
|
||||
|
||||
|
||||
#include "Math/Dense.h"
|
||||
#include "Math/Transform.h"
|
||||
|
||||
#include "Solid.h"
|
||||
#include "uLibInterface.hh"
|
||||
|
||||
namespace uLib {
|
||||
namespace Geant {
|
||||
@@ -57,17 +55,18 @@ public:
|
||||
};
|
||||
|
||||
Solid::Solid()
|
||||
: m_Material(nullptr), m_Logical(nullptr) {
|
||||
}
|
||||
: m_Name("unnamed_solid"), m_Material(NULL), m_Logical(NULL), m_Physical(NULL),
|
||||
m_Position(new G4ThreeVector(0,0,0)), m_Rotation(NULL) {}
|
||||
|
||||
Solid::Solid(const char *name)
|
||||
: m_Name(name), m_Material(nullptr), m_Logical(nullptr) {
|
||||
: m_Name(name), m_Material(NULL), m_Logical(NULL), m_Physical(NULL),
|
||||
m_Position(new G4ThreeVector(0,0,0)), m_Rotation(NULL) {}
|
||||
|
||||
Solid::~Solid() {
|
||||
if (m_Position) delete m_Position;
|
||||
if (m_Rotation) delete m_Rotation;
|
||||
}
|
||||
|
||||
Solid::~Solid() {}
|
||||
|
||||
void Solid::Update() {}
|
||||
|
||||
void Solid::SetNistMaterial(const char *name) {
|
||||
G4NistManager *nist = G4NistManager::Instance();
|
||||
G4Material *mat = nist->FindOrBuildMaterial(name);
|
||||
@@ -86,12 +85,28 @@ void Solid::SetMaterial(G4Material *material) {
|
||||
}
|
||||
|
||||
void Solid::SetTransform(Matrix4f transform) {
|
||||
m_Transform.FromMatrix(transform);
|
||||
m_Transform.Updated();
|
||||
uLib::AffineTransform t;
|
||||
t.SetMatrix(transform);
|
||||
|
||||
// 2. Extract position and rotation for Geant4
|
||||
Vector3f pos = t.GetPosition();
|
||||
if (!m_Position) m_Position = new G4ThreeVector();
|
||||
*m_Position = G4ThreeVector(pos(0), pos(1), pos(2));
|
||||
|
||||
// Create a G4 rotation matrix from the 4x4 matrix
|
||||
Matrix3f m = t.GetRotation();
|
||||
if (!m_Rotation) m_Rotation = new G4RotationMatrix();
|
||||
m_Rotation->set(G4ThreeVector(m(0,0), m(1,0), m(2,0)),
|
||||
G4ThreeVector(m(0,1), m(1,1), m(2,1)),
|
||||
G4ThreeVector(m(0,2), m(1,2), m(2,2)));
|
||||
|
||||
// 3. If object is already placed, update its transformation
|
||||
if (m_Physical) {
|
||||
m_Physical->SetTranslation(*m_Position);
|
||||
m_Physical->SetRotation(m_Rotation);
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
|
||||
void Solid::SetParent(Solid *parent) {
|
||||
if (!m_Logical) {
|
||||
std::cerr << "logical volume not created for solid " << GetName() << std::endl;
|
||||
@@ -114,8 +129,8 @@ void Solid::SetParent(Solid *parent) {
|
||||
|
||||
// G4PVPlacement
|
||||
m_Physical = new G4PVPlacement(
|
||||
ToG4Transform(m_Transform), // Rotation
|
||||
GetPosition(), // Position (translation) inside the parent
|
||||
m_Rotation, // Rotation
|
||||
*m_Position, // Position (translation) inside the parent
|
||||
m_Logical, // The logical volume of this solid (the child)
|
||||
GetName(), // Name of the physical volume
|
||||
parentLogical, // The logical volume of the parent (nullptr if it's the World volume)
|
||||
@@ -163,14 +178,11 @@ void TessellatedSolid::Update() {
|
||||
|
||||
BoxSolid::BoxSolid(const char *name) :
|
||||
BaseClass(name),
|
||||
m_Density(1.0),
|
||||
m_ContainerBox(new ContainerBox()),
|
||||
m_Solid(new G4Box(name, 1, 1, 1))
|
||||
{}
|
||||
|
||||
BoxSolid::BoxSolid(const char *name, ContainerBox *box) :
|
||||
BaseClass(name),
|
||||
m_Density(1.0) {
|
||||
BoxSolid::BoxSolid(const char *name, ContainerBox *box) : BaseClass(name) {
|
||||
m_Solid = new G4Box(name, 1, 1, 1);
|
||||
m_ContainerBox = box;
|
||||
Object::connect(box, &ContainerBox::Updated, this, &BoxSolid::Update);
|
||||
@@ -180,7 +192,6 @@ BoxSolid::BoxSolid(const char *name, ContainerBox *box) :
|
||||
Update();
|
||||
}
|
||||
|
||||
|
||||
void BoxSolid::Update() {
|
||||
if (m_ContainerBox) {
|
||||
Vector3f size = m_ContainerBox->GetSize();
|
||||
|
||||
@@ -28,15 +28,11 @@
|
||||
|
||||
#include "Core/Object.h"
|
||||
#include "Geant/Matter.h"
|
||||
#include "Math/Transform.h"
|
||||
#include <Geant4/G4LogicalVolume.hh>
|
||||
#include "Math/ContainerBox.h"
|
||||
#include "Math/Dense.h"
|
||||
#include "Math/TriangleMesh.h"
|
||||
|
||||
|
||||
|
||||
|
||||
class G4Material;
|
||||
class G4LogicalVolume;
|
||||
class G4TessellatedSolid;
|
||||
@@ -46,17 +42,27 @@ namespace uLib {
|
||||
namespace Geant {
|
||||
|
||||
class Solid : public Object {
|
||||
|
||||
uLibTypeMacro(Solid, Object)
|
||||
ULIB_SERIALIZE_ACCESS
|
||||
ULIB_DECLARE_PROPERTIES(Solid)
|
||||
|
||||
public:
|
||||
uLibTypeMacro(Solid, Object)
|
||||
|
||||
Solid();
|
||||
Solid(const char *name);
|
||||
virtual ~Solid();
|
||||
|
||||
void SetNistMaterial(const char *name);
|
||||
void SetMaterial(G4Material *material);
|
||||
|
||||
void SetSizeUnit(const char *unit);
|
||||
// Implementiamo SetParent qui, per tutti.
|
||||
virtual void SetParent(Solid *parent);
|
||||
|
||||
// Setters per la posizione (necessari per il piazzamento)
|
||||
void SetTransform(Matrix4f transform);
|
||||
|
||||
uLibGetMacro(Material, G4Material *)
|
||||
uLibGetSetMacro(Logical, G4LogicalVolume *)
|
||||
uLibGetSetMacro(Physical, G4VPhysicalVolume *)
|
||||
|
||||
virtual G4VSolid* GetG4Solid() const { return nullptr; }
|
||||
|
||||
inline const char *GetName() const {
|
||||
@@ -65,70 +71,34 @@ public:
|
||||
|
||||
template < typename Ar >
|
||||
void serialize(Ar &ar, const unsigned int version) {
|
||||
ar & HRP("Name", m_Name);
|
||||
ar & HRP("Material", m_Material);
|
||||
ar & m_Name;
|
||||
}
|
||||
|
||||
|
||||
|
||||
|
||||
public slots:
|
||||
virtual void Update();
|
||||
|
||||
protected:
|
||||
|
||||
std::string m_Name;
|
||||
};
|
||||
|
||||
|
||||
class LogicalVolume : public Object {
|
||||
|
||||
uLibTypeMacro(LogicalVolume, Object)
|
||||
ULIB_SERIALIZE_ACCESS
|
||||
ULIB_DECLARE_PROPERTIES(LogicalVolume)
|
||||
|
||||
public:
|
||||
|
||||
LogicalVolume();
|
||||
LogicalVolume(const char *name);
|
||||
virtual ~LogicalVolume();
|
||||
|
||||
virtual G4VSolid* GetG4Solid() const { return nullptr; }
|
||||
|
||||
inline const char *GetName() const {
|
||||
return m_Logical ? m_Logical->GetName().c_str() : m_Name.c_str();
|
||||
}
|
||||
|
||||
template < typename Ar >
|
||||
void serialize(Ar &ar, const unsigned int version) {
|
||||
ar & HRP("Name", m_Name);
|
||||
ar & HRP("Material", m_Material);
|
||||
}
|
||||
|
||||
|
||||
public slots:
|
||||
virtual void Update();
|
||||
|
||||
protected:
|
||||
std::string m_Name;
|
||||
Material *m_Material;
|
||||
|
||||
G4Material *m_Material;
|
||||
G4LogicalVolume *m_Logical;
|
||||
G4VPhysicalVolume *m_Physical; // <-- Memorizza l'oggetto posizionato
|
||||
G4ThreeVector *m_Position; // <-- Offset rispetto al centro del padre
|
||||
G4RotationMatrix* m_Rotation; // <-- Rotazione rispetto al padre
|
||||
};
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
class TessellatedSolid : public Solid {
|
||||
public:
|
||||
uLibTypeMacro(TessellatedSolid, Solid)
|
||||
|
||||
TessellatedSolid();
|
||||
TessellatedSolid(const char *name);
|
||||
|
||||
void SetMesh(TriangleMesh &mesh);
|
||||
uLibGetMacro(Solid, G4TessellatedSolid *)
|
||||
|
||||
virtual G4VSolid* GetG4Solid() const override { return (G4VSolid*)m_Solid; }
|
||||
|
||||
const TriangleMesh& GetMesh() const { return m_Mesh; }
|
||||
@@ -147,30 +117,26 @@ private :
|
||||
|
||||
|
||||
class BoxSolid : public Solid {
|
||||
uLibTypeMacro(BoxSolid, Solid)
|
||||
ULIB_SERIALIZE_ACCESS
|
||||
ULIB_DECLARE_PROPERTIES(BoxSolid)
|
||||
|
||||
public:
|
||||
uLibTypeMacro(BoxSolid, Solid)
|
||||
|
||||
BoxSolid(const char *name = "");
|
||||
BoxSolid(const char *name, ContainerBox *box);
|
||||
|
||||
virtual G4VSolid* GetG4Solid() const override { return (G4VSolid*)m_Solid; }
|
||||
|
||||
ContainerBox* GetObject() const { return m_ContainerBox; }
|
||||
|
||||
template < typename Ar >
|
||||
void serialize(Ar &ar, const unsigned int version) {
|
||||
// ar & boost::serialization::base_object<BaseClass>(*this);
|
||||
ar & HRP("Density", m_Density, "g/cm3");
|
||||
ar & boost::serialization::base_object<BaseClass>(*this);
|
||||
ar & m_ContainerBox;
|
||||
}
|
||||
|
||||
public slots:
|
||||
void Update();
|
||||
|
||||
private:
|
||||
float m_Density;
|
||||
ContainerBox *m_ContainerBox;
|
||||
G4Box *m_Solid;
|
||||
};
|
||||
|
||||
@@ -25,32 +25,6 @@ int main() {
|
||||
TEST1(solid.GetMaterial()->GetName() == "G4_AIR");
|
||||
}
|
||||
|
||||
// Test BoxSolid //
|
||||
{
|
||||
Geant::BoxSolid boxsolid("test_boxsolid");
|
||||
boxsolid.SetNistMaterial("G4_AIR");
|
||||
TEST1(boxsolid.GetLogical() != nullptr);
|
||||
// TEST1(boxsolid.GetSolid() != nullptr);
|
||||
}
|
||||
|
||||
// Test BoxSolid with a container box //
|
||||
{
|
||||
ContainerBox box;
|
||||
|
||||
// box.SetPosition(Vector3f(1,1,1));
|
||||
// box.SetRotation(Rotation(Vector3f(0,1,0), 45_deg));
|
||||
|
||||
Geant::BoxSolid boxsolid("test_boxsolid", &box);
|
||||
boxsolid.SetNistMaterial("G4_AIR");
|
||||
TEST1(boxsolid.GetLogical() != nullptr);
|
||||
// TEST1(boxsolid.GetSolid() != nullptr);
|
||||
// TEST1(boxsolid.GetSolid()->GetXHalfLength() == 0.5);
|
||||
// TEST1(boxsolid.GetSolid()->GetYHalfLength() == 0.5);
|
||||
// TEST1(boxsolid.GetSolid()->GetZHalfLength() == 0.5);
|
||||
|
||||
|
||||
}
|
||||
|
||||
// Test TessellatedSolid with a simple mesh //
|
||||
{
|
||||
Geant::TessellatedSolid tsolid("test_tessellated");
|
||||
|
||||
@@ -1,51 +0,0 @@
|
||||
#ifndef ULIB_GEANT_ULIB_INTERFACE_HH
|
||||
#define ULIB_GEANT_ULIB_INTERFACE_HH
|
||||
|
||||
#include "Math/Transform.h"
|
||||
#include <Geant4/G4Transform3D.hh>
|
||||
#include <Geant4/G4ThreeVector.hh>
|
||||
#include <Geant4/G4RotationMatrix.hh>
|
||||
|
||||
namespace uLib {
|
||||
namespace Geant {
|
||||
|
||||
/**
|
||||
* @brief Converts a uLib::Matrix4f to a Geant4 G4Transform3D.
|
||||
*/
|
||||
inline G4Transform3D ToG4Transform(const Matrix4f& m) {
|
||||
return G4Transform3D(
|
||||
m(0, 0), m(0, 1), m(0, 2), m(0, 3),
|
||||
m(1, 0), m(1, 1), m(1, 2), m(1, 3),
|
||||
m(2, 0), m(2, 1), m(2, 2), m(2, 3)
|
||||
);
|
||||
}
|
||||
|
||||
/**
|
||||
* @brief Converts a uLib::AffineTransform to a Geant4 G4Transform3D.
|
||||
*/
|
||||
inline G4Transform3D ToG4Transform(const AffineTransform& at) {
|
||||
return ToG4Transform(at.GetMatrix());
|
||||
}
|
||||
|
||||
/**
|
||||
* @brief Converts a uLib::Vector3f to a Geant4 G4ThreeVector.
|
||||
*/
|
||||
inline G4ThreeVector ToG4Vector(const Vector3f& v) {
|
||||
return G4ThreeVector(v(0), v(1), v(2));
|
||||
}
|
||||
|
||||
/**
|
||||
* @brief Converts a uLib::Matrix3f to a Geant4 G4RotationMatrix.
|
||||
*/
|
||||
inline G4RotationMatrix ToG4Rotation(const Matrix3f& m) {
|
||||
G4RotationMatrix rot;
|
||||
rot.set(G4ThreeVector(m(0, 0), m(1, 0), m(2, 0)),
|
||||
G4ThreeVector(m(0, 1), m(1, 1), m(2, 1)),
|
||||
G4ThreeVector(m(0, 2), m(1, 2), m(2, 2)));
|
||||
return rot;
|
||||
}
|
||||
|
||||
} // namespace Geant
|
||||
} // namespace uLib
|
||||
|
||||
#endif // ULIB_GEANT_ULIB_INTERFACE_HH
|
||||
@@ -41,7 +41,7 @@ namespace uLib {
|
||||
* transformation that is applied on top of each child's own transform.
|
||||
*
|
||||
* A bounding box is automatically computed from all contained objects and
|
||||
* can be queried or shown/hidden through the VTK prop3d.
|
||||
* can be queried or shown/hidden through the VTK puppet.
|
||||
*/
|
||||
class Assembly : public ObjectsContext, public TRS {
|
||||
public:
|
||||
@@ -81,7 +81,7 @@ public:
|
||||
|
||||
/**
|
||||
* @brief Controls whether the bounding box wireframe should be shown
|
||||
* in the viewer (used by the VTK prop3d).
|
||||
* in the viewer (used by the VTK puppet).
|
||||
*/
|
||||
void SetShowBoundingBox(bool show);
|
||||
bool GetShowBoundingBox() const;
|
||||
|
||||
@@ -50,9 +50,9 @@
|
||||
// #include <Eigen/src/Core/Matrix.h>
|
||||
#include <stdlib.h>
|
||||
|
||||
#include "Core/Property.h"
|
||||
#include "Core/Types.h"
|
||||
#include <Eigen/Dense>
|
||||
#include "Core/Types.h"
|
||||
#include "Core/Property.h"
|
||||
|
||||
//// BOOST SERIALIZATION ///////////////////////////////////////////////////////
|
||||
|
||||
@@ -150,6 +150,7 @@ typedef Eigen::MatrixXi MatrixXi;
|
||||
typedef Eigen::MatrixXf MatrixXf;
|
||||
typedef Eigen::MatrixXd MatrixXd;
|
||||
|
||||
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
// Vector String interaction ///////////////////////////////////////////////////
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
@@ -191,7 +192,7 @@ std::string VectorxT_ToString(const Eigen::Matrix<T, size, 1> &vec) {
|
||||
// }
|
||||
|
||||
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);
|
||||
}
|
||||
|
||||
@@ -204,7 +205,9 @@ public:
|
||||
typedef Eigen::Matrix<Scalarf, 4, 1> BaseClass;
|
||||
|
||||
_HPoint3f() : BaseClass(0, 0, 0, p) {}
|
||||
_HPoint3f(int rows, int cols) : BaseClass() { this->operator()(3) = p; }
|
||||
_HPoint3f(int rows, int cols) : BaseClass() {
|
||||
this->operator()(3) = p;
|
||||
}
|
||||
_HPoint3f(float x, float y, float z) : BaseClass(x, y, z, p) {}
|
||||
_HPoint3f(Vector3f &in) : BaseClass(in.homogeneous()) {
|
||||
this->operator()(3) = p;
|
||||
@@ -247,24 +250,24 @@ struct _HError3f {
|
||||
HVector3f position_error;
|
||||
HVector3f direction_error;
|
||||
};
|
||||
typedef struct _HError3f HError3f;
|
||||
typedef struct _HError3f HError3f;
|
||||
|
||||
inline std::ostream &operator<<(std::ostream &stream, const HError3f &err) {
|
||||
stream << "HError3f(" << "ept[" << err.position_error.transpose()
|
||||
<< "] , edr[" << err.direction_error.transpose() << "]) ";
|
||||
return stream;
|
||||
}
|
||||
|
||||
typedef Property<Scalari> ScalariProperty;
|
||||
typedef Property<Scalarui> ScalaruiProperty;
|
||||
typedef Property<Scalarl> ScalarlProperty;
|
||||
typedef Property<Scalarul> ScalarulProperty;
|
||||
typedef Property<Scalarf> ScalarfProperty;
|
||||
typedef Property<Scalard> ScalardProperty;
|
||||
|
||||
inline std::ostream &operator<<(std::ostream &stream, const HError3f &err) {
|
||||
stream << "HError3f(" << "ept[" << err.position_error.transpose()
|
||||
<< "] , edr[" << err.direction_error.transpose() << "]) ";
|
||||
return stream;
|
||||
}
|
||||
|
||||
typedef Property<Scalari> ScalariProperty;
|
||||
typedef Property<Scalarui> ScalaruiProperty;
|
||||
typedef Property<Scalarl> ScalarlProperty;
|
||||
typedef Property<Scalarul> ScalarulProperty;
|
||||
typedef Property<Scalarf> ScalarfProperty;
|
||||
typedef Property<Scalard> ScalardProperty;
|
||||
|
||||
typedef Property<Vector1i> Vector1iProperty;
|
||||
typedef Property<Vector1f> Vector1fProperty;
|
||||
typedef Property<Vector1d> Vector1dProperty;
|
||||
typedef Property<Vector1i> Vector1iProperty;
|
||||
typedef Property<Vector1f> Vector1fProperty;
|
||||
typedef Property<Vector1d> Vector1dProperty;
|
||||
|
||||
typedef Property<Vector2i> Vector2iProperty;
|
||||
typedef Property<Vector3i> Vector3iProperty;
|
||||
@@ -291,9 +294,9 @@ typedef Property<Matrix3d> Matrix3dProperty;
|
||||
typedef Property<Matrix4d> Matrix4dProperty;
|
||||
|
||||
typedef Property<HVector3f> HVector3fProperty;
|
||||
typedef Property<HPoint3f> HPoint3fProperty;
|
||||
typedef Property<HPoint3f> HPoint3fProperty;
|
||||
|
||||
} // namespace uLib
|
||||
} // namespace uLib
|
||||
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
|
||||
@@ -1,12 +1,12 @@
|
||||
#include "Core/ObjectFactory.h"
|
||||
#include "Math/Assembly.h"
|
||||
#include "Math/ContainerBox.h"
|
||||
#include "Math/Cylinder.h"
|
||||
#include "Math/Geometry.h"
|
||||
#include "Math/QuadMesh.h"
|
||||
#include "Math/StructuredData.h"
|
||||
#include "Math/TriangleMesh.h"
|
||||
#include "Math/QuadMesh.h"
|
||||
#include "Math/VoxImage.h"
|
||||
#include "Math/Assembly.h"
|
||||
#include "Math/StructuredData.h"
|
||||
|
||||
namespace uLib {
|
||||
|
||||
@@ -14,6 +14,8 @@ ULIB_REGISTER_OBJECT(TRS)
|
||||
ULIB_REGISTER_OBJECT(ContainerBox)
|
||||
ULIB_REGISTER_OBJECT(Cylinder)
|
||||
ULIB_REGISTER_OBJECT(Assembly)
|
||||
ULIB_REGISTER_OBJECT(CylindricalGeometry)
|
||||
ULIB_REGISTER_OBJECT(SphericalGeometry)
|
||||
ULIB_REGISTER_OBJECT(TriangleMesh)
|
||||
ULIB_REGISTER_OBJECT(QuadMesh)
|
||||
ULIB_REGISTER_OBJECT_NAME(VoxImage<Voxel>, "VoxImage")
|
||||
|
||||
@@ -23,6 +23,8 @@
|
||||
|
||||
//////////////////////////////////////////////////////////////////////////////*/
|
||||
|
||||
|
||||
|
||||
/*
|
||||
* Copyright (C) 2012 Andrea Rigoni Garola <andrea.rigoni@pd.infn.it>
|
||||
*
|
||||
@@ -43,120 +45,142 @@
|
||||
*
|
||||
*/
|
||||
|
||||
|
||||
#ifndef U_TRANSFORM_H
|
||||
#define U_TRANSFORM_H
|
||||
|
||||
#include "Math/Dense.h"
|
||||
#include "Math/Units.h"
|
||||
#include <Eigen/Geometry>
|
||||
#include "Math/Units.h"
|
||||
#include "Math/Dense.h"
|
||||
|
||||
|
||||
namespace uLib {
|
||||
|
||||
using Eigen::Isometry3d;
|
||||
|
||||
|
||||
using Eigen::Isometry3f;
|
||||
using Eigen::Isometry3d;
|
||||
|
||||
using Eigen::Affine3d;
|
||||
using Eigen::Affine3f;
|
||||
using Eigen::Affine3d;
|
||||
|
||||
using Eigen::Projective3d;
|
||||
using Eigen::Projective3f;
|
||||
using Eigen::Projective3d;
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
///////// AFFINE TRANSFORM WRAPPER //////////////////////////////////////////
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
|
||||
|
||||
|
||||
class AffineTransform : virtual public Object {
|
||||
public:
|
||||
uLibTypeMacro(AffineTransform, Object) protected :
|
||||
uLibTypeMacro(AffineTransform, Object)
|
||||
protected:
|
||||
|
||||
Affine3f m_T;
|
||||
AffineTransform *m_Parent;
|
||||
Affine3f m_T;
|
||||
AffineTransform *m_Parent;
|
||||
|
||||
public:
|
||||
AffineTransform() : m_T(Matrix4f::Identity()), m_Parent(NULL) {}
|
||||
AffineTransform() :
|
||||
m_T(Matrix4f::Identity()),
|
||||
m_Parent(NULL)
|
||||
{}
|
||||
|
||||
AffineTransform(AffineTransform *parent)
|
||||
: m_T(Matrix4f::Identity()), m_Parent(parent) {}
|
||||
AffineTransform(AffineTransform *parent) :
|
||||
m_T(Matrix4f::Identity()),
|
||||
m_Parent(parent)
|
||||
{}
|
||||
|
||||
AffineTransform(const AffineTransform ©)
|
||||
: m_T(copy.m_T), m_Parent(copy.m_Parent) {}
|
||||
AffineTransform(const AffineTransform ©) :
|
||||
m_T(copy.m_T),
|
||||
m_Parent(copy.m_Parent)
|
||||
{}
|
||||
|
||||
Affine3f &GetTransform() { return m_T; }
|
||||
Affine3f& GetTransform() { return m_T; }
|
||||
|
||||
AffineTransform *GetParent() const { return this->m_Parent; }
|
||||
AffineTransform *GetParent() const { return this->m_Parent; }
|
||||
|
||||
void SetParent(AffineTransform *name) { this->m_Parent = name; }
|
||||
void SetParent(AffineTransform *name) { this->m_Parent = name; }
|
||||
|
||||
void SetMatrix(const Matrix4f &mat) { m_T.matrix() = mat; }
|
||||
Matrix4f &GetMatrix() { return m_T.matrix(); }
|
||||
const Matrix4f &GetMatrix() const { return m_T.matrix(); }
|
||||
void SetMatrix (const Matrix4f &mat) { m_T.matrix() = mat; }
|
||||
Matrix4f& GetMatrix () { return m_T.matrix(); }
|
||||
const Matrix4f& GetMatrix () const { return m_T.matrix(); }
|
||||
|
||||
Matrix4f GetWorldMatrix() const {
|
||||
if (!m_Parent)
|
||||
return m_T.matrix();
|
||||
else
|
||||
return m_Parent->GetWorldMatrix() * m_T.matrix(); // T = B * A //
|
||||
}
|
||||
Matrix4f GetWorldMatrix() const
|
||||
{
|
||||
if(!m_Parent) return m_T.matrix();
|
||||
else return m_Parent->GetWorldMatrix() * m_T.matrix(); // T = B * A //
|
||||
}
|
||||
|
||||
void SetWorldMatrix(const Matrix4f &mat) {
|
||||
if (!m_Parent)
|
||||
m_T.matrix() = mat;
|
||||
else
|
||||
m_T.matrix() = m_Parent->GetWorldMatrix().inverse() * mat;
|
||||
}
|
||||
void SetWorldMatrix(const Matrix4f &mat)
|
||||
{
|
||||
if(!m_Parent) m_T.matrix() = mat;
|
||||
else m_T.matrix() = m_Parent->GetWorldMatrix().inverse() * mat;
|
||||
}
|
||||
|
||||
void SetPosition(const Vector3f &v) { this->m_T.translation() = v; }
|
||||
void SetPosition(const Vector3f &v) { this->m_T.translation() = v; }
|
||||
|
||||
Vector3f GetPosition() const { return this->m_T.translation(); }
|
||||
Vector3f GetPosition() const { return this->m_T.translation(); }
|
||||
|
||||
void SetRotation(const Matrix3f &m) { this->m_T.linear() = m; }
|
||||
void SetRotation(const Matrix3f &m) { this->m_T.linear() = m; }
|
||||
|
||||
Matrix3f GetRotation() const { return this->m_T.rotation(); }
|
||||
Matrix3f GetRotation() const { return this->m_T.rotation(); }
|
||||
|
||||
void Translate(const Vector3f &v) { this->m_T.translate(v); }
|
||||
void Translate(const Vector3f &v) { this->m_T.translate(v); }
|
||||
|
||||
void Scale(const Vector3f &v) { this->m_T.scale(v); }
|
||||
void Scale(const Vector3f &v) { this->m_T.scale(v); }
|
||||
|
||||
Vector3f GetScale() const {
|
||||
return Vector3f(this->m_T.linear().col(0).norm(),
|
||||
this->m_T.linear().col(1).norm(),
|
||||
this->m_T.linear().col(2).norm());
|
||||
}
|
||||
Vector3f GetScale() const {
|
||||
return Vector3f(this->m_T.linear().col(0).norm(),
|
||||
this->m_T.linear().col(1).norm(),
|
||||
this->m_T.linear().col(2).norm());
|
||||
}
|
||||
|
||||
void Rotate(const Matrix3f &m) { this->m_T.rotate(m); }
|
||||
|
||||
void Rotate(const float angle, Vector3f axis) {
|
||||
axis.normalize(); // prehaps not necessary ( see eigens )
|
||||
Eigen::AngleAxisf ax(angle, axis);
|
||||
this->m_T.rotate(Eigen::Quaternion<float>(ax));
|
||||
}
|
||||
void Rotate(const Matrix3f &m) { this->m_T.rotate(m); }
|
||||
|
||||
void Rotate(const Vector3f euler_axis) {
|
||||
float angle = euler_axis.norm();
|
||||
Rotate(angle, euler_axis);
|
||||
}
|
||||
void Rotate(const float angle, Vector3f axis)
|
||||
{
|
||||
axis.normalize(); // prehaps not necessary ( see eigens )
|
||||
Eigen::AngleAxisf ax(angle,axis);
|
||||
this->m_T.rotate(Eigen::Quaternion<float>(ax));
|
||||
}
|
||||
|
||||
void PreRotate(const Matrix3f &m) { this->m_T.prerotate(m); }
|
||||
void Rotate(const Vector3f euler_axis) {
|
||||
float angle = euler_axis.norm();
|
||||
Rotate(angle,euler_axis);
|
||||
}
|
||||
|
||||
void QuaternionRotate(const Vector4f &q) {
|
||||
this->m_T.rotate(Eigen::Quaternion<float>(q));
|
||||
}
|
||||
void PreRotate(const Matrix3f &m) { this->m_T.prerotate(m); }
|
||||
|
||||
void EulerYZYRotate(const Vector3f &e) {
|
||||
Matrix3f mat;
|
||||
mat = Eigen::AngleAxisf(e.x(), Vector3f::UnitY()) *
|
||||
Eigen::AngleAxisf(e.y(), Vector3f::UnitZ()) *
|
||||
Eigen::AngleAxisf(e.z(), Vector3f::UnitY());
|
||||
m_T.rotate(mat);
|
||||
}
|
||||
void QuaternionRotate(const Vector4f &q)
|
||||
{ this->m_T.rotate(Eigen::Quaternion<float>(q)); }
|
||||
|
||||
void FlipAxes(int first, int second) {
|
||||
Matrix3f mat = Matrix3f::Identity();
|
||||
mat.col(first).swap(mat.col(second));
|
||||
m_T.rotate(mat);
|
||||
}
|
||||
void EulerYZYRotate(const Vector3f &e) {
|
||||
Matrix3f mat;
|
||||
mat = Eigen::AngleAxisf(e.x(), Vector3f::UnitY())
|
||||
* Eigen::AngleAxisf(e.y(), Vector3f::UnitZ())
|
||||
* Eigen::AngleAxisf(e.z(), Vector3f::UnitY());
|
||||
m_T.rotate(mat);
|
||||
}
|
||||
|
||||
void FlipAxes(int first, int second)
|
||||
{
|
||||
Matrix3f mat = Matrix3f::Identity();
|
||||
mat.col(first).swap(mat.col(second));
|
||||
m_T.rotate(mat);
|
||||
}
|
||||
};
|
||||
|
||||
|
||||
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
///////// TRS PARAMETERS /////////////////////////////////////////////////////
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
@@ -165,100 +189,116 @@ typedef Eigen::Affine3f AffineMatrix;
|
||||
|
||||
class TRS : public AffineTransform {
|
||||
|
||||
uLibTypeMacro(TRS, AffineTransform) ULIB_SERIALIZE_ACCESS
|
||||
uLibTypeMacro(TRS, AffineTransform)
|
||||
ULIB_SERIALIZE_ACCESS
|
||||
// ULIB_DECLARE_PROPERTIES(TRS)
|
||||
|
||||
public :
|
||||
public:
|
||||
|
||||
Vector3f position = Vector3f::Zero();
|
||||
Vector3f rotation = Vector3f::Zero();
|
||||
Vector3f scaling = Vector3f::Ones();
|
||||
Vector3f rotation = Vector3f::Zero();
|
||||
Vector3f scaling = Vector3f::Ones();
|
||||
|
||||
TRS() = default;
|
||||
TRS() = default;
|
||||
|
||||
TRS(const class AffineTransform &at) { this->FromMatrix(at.GetMatrix()); }
|
||||
TRS(const class AffineTransform& at) {
|
||||
this->FromMatrix(at.GetMatrix());
|
||||
}
|
||||
|
||||
TRS(const Matrix4f &mat) { this->FromMatrix(mat); }
|
||||
TRS(const Matrix4f& mat) {
|
||||
this->FromMatrix(mat);
|
||||
}
|
||||
|
||||
void FromMatrix(const Matrix4f &mat) {
|
||||
this->position = mat.block<3, 1>(0, 3);
|
||||
void FromMatrix(const Matrix4f& mat) {
|
||||
this->position = mat.block<3,1>(0,3);
|
||||
|
||||
Matrix3f linear = mat.block<3,3>(0,0);
|
||||
this->scaling(0) = linear.col(0).norm();
|
||||
this->scaling(1) = linear.col(1).norm();
|
||||
this->scaling(2) = linear.col(2).norm();
|
||||
|
||||
Matrix3f rot = linear;
|
||||
if (this->scaling(0) > 1e-6) rot.col(0) /= this->scaling(0);
|
||||
if (this->scaling(1) > 1e-6) rot.col(1) /= this->scaling(1);
|
||||
if (this->scaling(2) > 1e-6) rot.col(2) /= this->scaling(2);
|
||||
|
||||
Vector3f euler = rot.eulerAngles(2, 1, 0);
|
||||
this->rotation = Vector3f(euler(2), euler(1), euler(0));
|
||||
|
||||
this->SetMatrix(mat);
|
||||
this->NotifyPropertiesUpdated();
|
||||
}
|
||||
|
||||
Matrix3f linear = mat.block<3, 3>(0, 0);
|
||||
this->scaling(0) = linear.col(0).norm();
|
||||
this->scaling(1) = linear.col(1).norm();
|
||||
this->scaling(2) = linear.col(2).norm();
|
||||
void SetPosition(const Vector3f &v) {
|
||||
position = v;
|
||||
this->AffineTransform::SetPosition(v);
|
||||
}
|
||||
|
||||
void SetRotation(const Vector3f &v) {
|
||||
rotation = v;
|
||||
this->SyncMatrix();
|
||||
}
|
||||
|
||||
void SetOrientation(const Vector3f &v) { SetRotation(v); }
|
||||
|
||||
void SetScale(const Vector3f &v) {
|
||||
scaling = v;
|
||||
this->SyncMatrix();
|
||||
}
|
||||
|
||||
Matrix3f rot = linear;
|
||||
if (this->scaling(0) > 1e-6)
|
||||
rot.col(0) /= this->scaling(0);
|
||||
if (this->scaling(1) > 1e-6)
|
||||
rot.col(1) /= this->scaling(1);
|
||||
if (this->scaling(2) > 1e-6)
|
||||
rot.col(2) /= this->scaling(2);
|
||||
void SyncMatrix() {
|
||||
this->GetTransform() = GetAffineMatrix();
|
||||
}
|
||||
|
||||
Vector3f euler = rot.canonicalEulerAngles(2, 1, 0);
|
||||
this->rotation = Vector3f(euler(2), euler(1), euler(0));
|
||||
void Updated() override {
|
||||
this->SyncMatrix();
|
||||
this->NotifyPropertiesUpdated();
|
||||
this->AffineTransform::Updated();
|
||||
}
|
||||
|
||||
this->SetMatrix(mat);
|
||||
this->NotifyPropertiesUpdated();
|
||||
}
|
||||
template <class ArchiveT>
|
||||
void serialize(ArchiveT & ar, const unsigned int version) {
|
||||
ar & HRPU(position, "mm");
|
||||
ar & HRPU(rotation, "rad");
|
||||
ar & HRP(scaling);
|
||||
}
|
||||
|
||||
|
||||
void SetPosition(const Vector3f &v) {
|
||||
position = v;
|
||||
this->AffineTransform::SetPosition(v);
|
||||
}
|
||||
AffineMatrix GetAffineMatrix() const {
|
||||
AffineMatrix m = AffineMatrix::Identity();
|
||||
m.translate(position);
|
||||
m.rotate(Eigen::AngleAxisf(rotation.z(), Vector3f::UnitZ()));
|
||||
m.rotate(Eigen::AngleAxisf(rotation.y(), Vector3f::UnitY()));
|
||||
m.rotate(Eigen::AngleAxisf(rotation.x(), Vector3f::UnitX()));
|
||||
m.scale(scaling);
|
||||
return m;
|
||||
}
|
||||
|
||||
void SetRotation(const Vector3f &v) {
|
||||
rotation = v;
|
||||
this->SyncMatrix();
|
||||
}
|
||||
Matrix4f GetMatrix() const {
|
||||
return this->GetAffineMatrix().matrix();
|
||||
}
|
||||
|
||||
void SetOrientation(const Vector3f &v) { SetRotation(v); }
|
||||
|
||||
void SetScale(const Vector3f &v) {
|
||||
scaling = v;
|
||||
this->SyncMatrix();
|
||||
}
|
||||
|
||||
void SyncMatrix() { this->GetTransform() = GetAffineMatrix(); }
|
||||
|
||||
void Updated() override {
|
||||
this->SyncMatrix();
|
||||
this->NotifyPropertiesUpdated();
|
||||
this->AffineTransform::Updated();
|
||||
}
|
||||
|
||||
template <class ArchiveT>
|
||||
void serialize(ArchiveT &ar, const unsigned int version) {
|
||||
ar &HRPU(position, "mm");
|
||||
ar &HRPU(rotation, "rad");
|
||||
ar &HRP(scaling);
|
||||
}
|
||||
|
||||
AffineMatrix GetAffineMatrix() const {
|
||||
AffineMatrix m = AffineMatrix::Identity();
|
||||
m.translate(position);
|
||||
m.rotate(Eigen::AngleAxisf(rotation.z(), Vector3f::UnitZ()));
|
||||
m.rotate(Eigen::AngleAxisf(rotation.y(), Vector3f::UnitY()));
|
||||
m.rotate(Eigen::AngleAxisf(rotation.x(), Vector3f::UnitX()));
|
||||
m.scale(scaling);
|
||||
return m;
|
||||
}
|
||||
|
||||
Matrix4f GetMatrix() const { return this->GetAffineMatrix().matrix(); }
|
||||
|
||||
|
||||
|
||||
};
|
||||
|
||||
inline std::ostream &operator<<(std::ostream &os, const TRS &trs) {
|
||||
os << trs.position << " " << trs.rotation << " " << trs.scaling;
|
||||
return os;
|
||||
|
||||
|
||||
|
||||
|
||||
inline std::ostream& operator<<(std::ostream& os, const TRS& trs) {
|
||||
os << trs.position << " " << trs.rotation << " " << trs.scaling;
|
||||
return os;
|
||||
}
|
||||
|
||||
inline std::istream &operator>>(std::istream &is, TRS &trs) {
|
||||
is >> trs.position >> trs.rotation >> trs.scaling;
|
||||
return is;
|
||||
inline std::istream& operator>>(std::istream& is, TRS& trs) {
|
||||
is >> trs.position >> trs.rotation >> trs.scaling;
|
||||
return is;
|
||||
}
|
||||
|
||||
} // namespace uLib
|
||||
} // uLib
|
||||
|
||||
#endif // U_TRANSFORM_H
|
||||
|
||||
|
||||
#endif//U_TRANSFORM_H
|
||||
|
||||
@@ -13,41 +13,41 @@ namespace uLib {
|
||||
using namespace CLHEP;
|
||||
|
||||
inline namespace literals {
|
||||
constexpr double operator""_m(long double v) { return static_cast<double>(v) * CLHEP::meter; }
|
||||
constexpr double operator""_cm(long double v) { return static_cast<double>(v) * CLHEP::centimeter; }
|
||||
constexpr double operator""_mm(long double v) { return static_cast<double>(v) * CLHEP::millimeter; }
|
||||
constexpr double operator""_um(long double v) { return static_cast<double>(v) * CLHEP::micrometer; }
|
||||
constexpr double operator""_nm(long double v) { return static_cast<double>(v) * CLHEP::nanometer; }
|
||||
constexpr double operator""_km(long double v) { return static_cast<double>(v) * CLHEP::kilometer; }
|
||||
constexpr double operator"" _m(long double v) { return static_cast<double>(v) * CLHEP::meter; }
|
||||
constexpr double operator"" _cm(long double v) { return static_cast<double>(v) * CLHEP::centimeter; }
|
||||
constexpr double operator"" _mm(long double v) { return static_cast<double>(v) * CLHEP::millimeter; }
|
||||
constexpr double operator"" _um(long double v) { return static_cast<double>(v) * CLHEP::micrometer; }
|
||||
constexpr double operator"" _nm(long double v) { return static_cast<double>(v) * CLHEP::nanometer; }
|
||||
constexpr double operator"" _km(long double v) { return static_cast<double>(v) * CLHEP::kilometer; }
|
||||
|
||||
constexpr double operator""_m(unsigned long long v) { return static_cast<double>(v) * CLHEP::meter; }
|
||||
constexpr double operator""_cm(unsigned long long v) { return static_cast<double>(v) * CLHEP::centimeter; }
|
||||
constexpr double operator""_mm(unsigned long long v) { return static_cast<double>(v) * CLHEP::millimeter; }
|
||||
constexpr double operator""_um(unsigned long long v) { return static_cast<double>(v) * CLHEP::micrometer; }
|
||||
constexpr double operator""_nm(unsigned long long v) { return static_cast<double>(v) * CLHEP::nanometer; }
|
||||
constexpr double operator""_km(unsigned long long v) { return static_cast<double>(v) * CLHEP::kilometer; }
|
||||
constexpr double operator"" _m(unsigned long long v) { return static_cast<double>(v) * CLHEP::meter; }
|
||||
constexpr double operator"" _cm(unsigned long long v) { return static_cast<double>(v) * CLHEP::centimeter; }
|
||||
constexpr double operator"" _mm(unsigned long long v) { return static_cast<double>(v) * CLHEP::millimeter; }
|
||||
constexpr double operator"" _um(unsigned long long v) { return static_cast<double>(v) * CLHEP::micrometer; }
|
||||
constexpr double operator"" _nm(unsigned long long v) { return static_cast<double>(v) * CLHEP::nanometer; }
|
||||
constexpr double operator"" _km(unsigned long long v) { return static_cast<double>(v) * CLHEP::kilometer; }
|
||||
|
||||
constexpr double operator""_deg(long double v) { return static_cast<double>(v) * CLHEP::degree; }
|
||||
constexpr double operator""_rad(long double v) { return static_cast<double>(v) * CLHEP::radian; }
|
||||
constexpr double operator""_deg(unsigned long long v) { return static_cast<double>(v) * CLHEP::degree; }
|
||||
constexpr double operator""_rad(unsigned long long v) { return static_cast<double>(v) * CLHEP::radian; }
|
||||
constexpr double operator"" _deg(long double v) { return static_cast<double>(v) * CLHEP::degree; }
|
||||
constexpr double operator"" _rad(long double v) { return static_cast<double>(v) * CLHEP::radian; }
|
||||
constexpr double operator"" _deg(unsigned long long v) { return static_cast<double>(v) * CLHEP::degree; }
|
||||
constexpr double operator"" _rad(unsigned long long v) { return static_cast<double>(v) * CLHEP::radian; }
|
||||
|
||||
constexpr double operator""_ns(long double v) { return static_cast<double>(v) * CLHEP::nanosecond; }
|
||||
constexpr double operator""_s(long double v) { return static_cast<double>(v) * CLHEP::second; }
|
||||
constexpr double operator""_ms(long double v) { return static_cast<double>(v) * CLHEP::millisecond; }
|
||||
constexpr double operator""_ns(unsigned long long v) { return static_cast<double>(v) * CLHEP::nanosecond; }
|
||||
constexpr double operator""_s(unsigned long long v) { return static_cast<double>(v) * CLHEP::second; }
|
||||
constexpr double operator""_ms(unsigned long long v) { return static_cast<double>(v) * CLHEP::millisecond; }
|
||||
constexpr double operator"" _ns(long double v) { return static_cast<double>(v) * CLHEP::nanosecond; }
|
||||
constexpr double operator"" _s(long double v) { return static_cast<double>(v) * CLHEP::second; }
|
||||
constexpr double operator"" _ms(long double v) { return static_cast<double>(v) * CLHEP::millisecond; }
|
||||
constexpr double operator"" _ns(unsigned long long v) { return static_cast<double>(v) * CLHEP::nanosecond; }
|
||||
constexpr double operator"" _s(unsigned long long v) { return static_cast<double>(v) * CLHEP::second; }
|
||||
constexpr double operator"" _ms(unsigned long long v) { return static_cast<double>(v) * CLHEP::millisecond; }
|
||||
|
||||
constexpr double operator""_MeV(long double v) { return static_cast<double>(v) * CLHEP::megaelectronvolt; }
|
||||
constexpr double operator""_eV(long double v) { return static_cast<double>(v) * CLHEP::electronvolt; }
|
||||
constexpr double operator""_keV(long double v) { return static_cast<double>(v) * CLHEP::kiloelectronvolt; }
|
||||
constexpr double operator""_GeV(long double v) { return static_cast<double>(v) * CLHEP::gigaelectronvolt; }
|
||||
constexpr double operator""_TeV(long double v) { return static_cast<double>(v) * CLHEP::teraelectronvolt; }
|
||||
constexpr double operator""_MeV(unsigned long long v) { return static_cast<double>(v) * CLHEP::megaelectronvolt; }
|
||||
constexpr double operator""_eV(unsigned long long v) { return static_cast<double>(v) * CLHEP::electronvolt; }
|
||||
constexpr double operator""_keV(unsigned long long v) { return static_cast<double>(v) * CLHEP::kiloelectronvolt; }
|
||||
constexpr double operator""_GeV(unsigned long long v) { return static_cast<double>(v) * CLHEP::gigaelectronvolt; }
|
||||
constexpr double operator"" _MeV(long double v) { return static_cast<double>(v) * CLHEP::megaelectronvolt; }
|
||||
constexpr double operator"" _eV(long double v) { return static_cast<double>(v) * CLHEP::electronvolt; }
|
||||
constexpr double operator"" _keV(long double v) { return static_cast<double>(v) * CLHEP::kiloelectronvolt; }
|
||||
constexpr double operator"" _GeV(long double v) { return static_cast<double>(v) * CLHEP::gigaelectronvolt; }
|
||||
constexpr double operator"" _TeV(long double v) { return static_cast<double>(v) * CLHEP::teraelectronvolt; }
|
||||
constexpr double operator"" _MeV(unsigned long long v) { return static_cast<double>(v) * CLHEP::megaelectronvolt; }
|
||||
constexpr double operator"" _eV(unsigned long long v) { return static_cast<double>(v) * CLHEP::electronvolt; }
|
||||
constexpr double operator"" _keV(unsigned long long v) { return static_cast<double>(v) * CLHEP::kiloelectronvolt; }
|
||||
constexpr double operator"" _GeV(unsigned long long v) { return static_cast<double>(v) * CLHEP::gigaelectronvolt; }
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
@@ -33,11 +33,7 @@ set(DICTIONARY_HEADERS muCastorMCTrack.h
|
||||
SkinDetectorWriter.h)
|
||||
|
||||
set(LIBRARIES ${ROOT_LIBRARIES}
|
||||
${PACKAGE_LIBPREFIX}Math
|
||||
Eigen3::Eigen)
|
||||
|
||||
get_target_property(EIGEN3_INC Eigen3::Eigen INTERFACE_INCLUDE_DIRECTORIES)
|
||||
include_directories(${EIGEN3_INC})
|
||||
${PACKAGE_LIBPREFIX}Math)
|
||||
|
||||
set(rDictName ${PACKAGE_LIBPREFIX}RootDict)
|
||||
root_generate_dictionary(${rDictName} ${DICTIONARY_HEADERS}
|
||||
|
||||
@@ -1,84 +1,88 @@
|
||||
#ifndef muCastor_MCTRACK_H
|
||||
#define muCastor_MCTRACK_H
|
||||
|
||||
// ########################################
|
||||
// muCastorMCTrack class
|
||||
// Created at the University of Brescia, Italy
|
||||
// Date: December 2011
|
||||
// Autors: Germano Bonomi germano.bonomi@ing.unibs.it
|
||||
// Martin Subieta martin.subieta@ing.unibs.it
|
||||
// ########################################
|
||||
//########################################
|
||||
// muCastorMCTrack class
|
||||
// Created at the University of Brescia, Italy
|
||||
// Date: December 2011
|
||||
// Autors: Germano Bonomi germano.bonomi@ing.unibs.it
|
||||
// Martin Subieta martin.subieta@ing.unibs.it
|
||||
//########################################
|
||||
|
||||
#include "Detectors/MuonScatter.h"
|
||||
#include <iostream>
|
||||
#include "vector"
|
||||
#include "TObject.h"
|
||||
#include "TParticle.h"
|
||||
#include "vector"
|
||||
#include <iostream>
|
||||
#include "Detectors/MuonScatter.h"
|
||||
|
||||
class TClonesArray;
|
||||
|
||||
class muCastorMCTrack : public TObject {
|
||||
class muCastorMCTrack : public TObject
|
||||
{
|
||||
|
||||
public:
|
||||
|
||||
public:
|
||||
/** Default constructor **/
|
||||
muCastorMCTrack();
|
||||
|
||||
/** Constructor from TParticle **/
|
||||
muCastorMCTrack(TParticle *particle);
|
||||
muCastorMCTrack(TParticle* particle);
|
||||
|
||||
/** Destructor **/
|
||||
virtual ~muCastorMCTrack();
|
||||
void Reset();
|
||||
void Reset();
|
||||
|
||||
|
||||
/** Accessors **/
|
||||
Int_t GetFirstDaughter() const { return fDaughter[0]; }
|
||||
Int_t GetMother() const { return fMother[0]; }
|
||||
Int_t GetFirstDaughter() const { return fDaughter[0];}
|
||||
Int_t GetMother() const { return fMother[0];}
|
||||
|
||||
/** Modifiers **/
|
||||
virtual void AddPoint(TLorentzVector pos, TLorentzVector mom);
|
||||
virtual void SetFirstDaughter(Int_t trkid) { fDaughter[0] = trkid; }
|
||||
virtual void SetLastDaughter(Int_t trkid) { fDaughter[1] = trkid; }
|
||||
virtual void AddPoint(TLorentzVector pos, TLorentzVector mom);
|
||||
virtual void SetFirstDaughter(Int_t trkid) { fDaughter[0] = trkid; }
|
||||
virtual void SetLastDaughter(Int_t trkid) { fDaughter[1] = trkid; }
|
||||
|
||||
inline void Dump() {
|
||||
std::cout << "muCastorMCTrack\n"
|
||||
<< "PDG code: " << fPdgCode << "\n"
|
||||
<< "Momentum: " << fPx << ", " << fPy << ", " << fPz << "\n"
|
||||
<< "Position: " << fVx << ", " << fVy << ", " << fVz << "\n"
|
||||
<< "Npoints: " << fNpoints << "\n";
|
||||
return;
|
||||
}
|
||||
inline void Dump() {
|
||||
std::cout << "muCastorMCTrack\n"
|
||||
<< "PDG code: " << fPdgCode << "\n"
|
||||
<< "Momentum: " << fPx << ", " << fPy << ", " << fPz << "\n"
|
||||
<< "Position: " << fVx << ", " << fVy << ", " << fVz << "\n"
|
||||
<< "Npoints: " << fNpoints << "\n";
|
||||
return;
|
||||
}
|
||||
|
||||
public:
|
||||
/* Private variables - copying private variables of TParticle */
|
||||
|
||||
Int_t fPdgCode; // PDG code of the particle
|
||||
Int_t fMother[2]; // Indices of the mother particles
|
||||
Int_t fDaughter[2]; // Indices of the daughter particles
|
||||
/* Private variables - copying private variables of TParticle */
|
||||
|
||||
Double_t fPx; // x component of momentum
|
||||
Double_t fPy; // y component of momentum
|
||||
Double_t fPz; // z component of momentum
|
||||
Double_t fE; // Energy
|
||||
Int_t fPdgCode; // PDG code of the particle
|
||||
Int_t fMother[2]; // Indices of the mother particles
|
||||
Int_t fDaughter[2]; // Indices of the daughter particles
|
||||
|
||||
Double_t fVx; // x of production vertex
|
||||
Double_t fVy; // y of production vertex
|
||||
Double_t fVz; // z of production vertex
|
||||
Double_t fVt; // t of production vertex
|
||||
Double_t fPx; // x component of momentum
|
||||
Double_t fPy; // y component of momentum
|
||||
Double_t fPz; // z component of momentum
|
||||
Double_t fE; // Energy
|
||||
|
||||
Int_t fPointsSize; // capacity of points array
|
||||
Int_t fNpoints; // number of stored points
|
||||
Double_t *fPntPosX; //[fNpoints] array of points (x) belonging to this track
|
||||
Double_t *fPntPosY; //[fNpoints] array of points (y) belonging to this track
|
||||
Double_t *fPntPosZ; //[fNpoints] array of points (z) belonging to this track
|
||||
Double_t *fPntT; //[fNpoints] array of points (t) belonging to this track
|
||||
Double_t *fPntMomX; //[fNpoints] array of points (px) belonging to this track
|
||||
Double_t *fPntMomY; //[fNpoints] array of points (py) belonging to this track
|
||||
Double_t *fPntMomZ; //[fNpoints] array of points (pz) belonging to this track
|
||||
Double_t *fPntE; //[fNpoints] array of points (E) belonging to this track
|
||||
Double_t fVx; // x of production vertex
|
||||
Double_t fVy; // y of production vertex
|
||||
Double_t fVz; // z of production vertex
|
||||
Double_t fVt; // t of production vertex
|
||||
|
||||
Int_t fPointsSize; // capacity of points array
|
||||
Int_t fNpoints; // number of stored points
|
||||
Double_t *fPntPosX; //[fNpoints] array of points (x) belonging to this track
|
||||
Double_t *fPntPosY; //[fNpoints] array of points (y) belonging to this track
|
||||
Double_t *fPntPosZ; //[fNpoints] array of points (z) belonging to this track
|
||||
Double_t *fPntT; //[fNpoints] array of points (t) belonging to this track
|
||||
Double_t *fPntMomX; //[fNpoints] array of points (px) belonging to this track
|
||||
Double_t *fPntMomY; //[fNpoints] array of points (py) belonging to this track
|
||||
Double_t *fPntMomZ; //[fNpoints] array of points (pz) belonging to this track
|
||||
Double_t *fPntE; //[fNpoints] array of points (E) belonging to this track
|
||||
|
||||
ClassDef(muCastorMCTrack,1);
|
||||
|
||||
ClassDef(muCastorMCTrack, 1);
|
||||
};
|
||||
|
||||
uLib::MuonScatter &operator<<(uLib::MuonScatter &mu,
|
||||
const muCastorMCTrack &bsmu);
|
||||
#endif // muCastor_MCTRACK_H
|
||||
uLib::MuonScatter &operator << (uLib::MuonScatter &mu, const muCastorMCTrack &bsmu);
|
||||
#endif //muCastor_MCTRACK_H
|
||||
|
||||
@@ -4,7 +4,6 @@ set(HEADERS uLibVtkInterface.h
|
||||
vtkQViewport.h
|
||||
vtkViewport.h
|
||||
vtkObjectsContext.h
|
||||
vtkMultiSelectionProp.h
|
||||
)
|
||||
|
||||
set(SOURCES uLibVtkInterface.cxx
|
||||
@@ -13,7 +12,6 @@ set(SOURCES uLibVtkInterface.cxx
|
||||
vtkQViewport.cpp
|
||||
vtkViewport.cpp
|
||||
vtkObjectsContext.cpp
|
||||
vtkMultiSelectionProp.cpp
|
||||
)
|
||||
|
||||
## Pull in Math VTK wrappers (sets MATH_SOURCES / MATH_HEADERS)
|
||||
@@ -39,8 +37,6 @@ list(APPEND HEADERS ${HEP_GEANT_HEADERS})
|
||||
set(LIBRARIES Eigen3::Eigen
|
||||
${ROOT_LIBRARIES}
|
||||
${VTK_LIBRARIES}
|
||||
VTK::ImagingHybrid
|
||||
VTK::ImagingSources
|
||||
${PACKAGE_LIBPREFIX}Math
|
||||
${PACKAGE_LIBPREFIX}Detectors
|
||||
${PACKAGE_LIBPREFIX}Geant)
|
||||
@@ -60,7 +56,7 @@ set_target_properties(${libname} PROPERTIES
|
||||
AUTOMOC ON
|
||||
AUTOUIC ON
|
||||
AUTORCC ON)
|
||||
target_link_libraries(${libname} PUBLIC ${LIBRARIES} Qt6::Widgets)
|
||||
target_link_libraries(${libname} ${LIBRARIES} Qt6::Widgets)
|
||||
|
||||
install(TARGETS ${libname}
|
||||
EXPORT "uLibTargets"
|
||||
|
||||
@@ -24,7 +24,7 @@
|
||||
//////////////////////////////////////////////////////////////////////////////*/
|
||||
|
||||
#include "Vtk/HEP/Detectors/vtkDetectorChamber.h"
|
||||
#include "Vtk/HEP/Detectors/vtkDetectorChamber.h"
|
||||
#include "HEP/Detectors/DetectorChamber.h"
|
||||
#include "Math/Units.h"
|
||||
|
||||
#include "Vtk/uLibVtkViewer.h"
|
||||
@@ -43,20 +43,20 @@ BOOST_AUTO_TEST_CASE(vtkDetectorChamberTest) {
|
||||
d2.Scale(Vector3f(1_m, 2_m, 20_cm));
|
||||
d2.Translate(Vector3f(1_m, 0, 10_m));
|
||||
|
||||
Vtk::DetectorChamber v_d1(&d1);
|
||||
Vtk::DetectorChamber v_d2(&d2);
|
||||
v_d1.SetRepresentation(Vtk::Prop3D::Surface);
|
||||
v_d2.SetRepresentation(Vtk::Prop3D::Surface);
|
||||
Vtk::vtkDetectorChamber v_d1(&d1);
|
||||
Vtk::vtkDetectorChamber v_d2(&d2);
|
||||
v_d1.SetRepresentation(Vtk::Puppet::Surface);
|
||||
v_d2.SetRepresentation(Vtk::Puppet::Surface);
|
||||
|
||||
if (!v_d1.GetProp()) {
|
||||
BOOST_FAIL("DetectorChamber::GetProp() returned NULL");
|
||||
BOOST_FAIL("vtkDetectorChamber::GetProp() returned NULL");
|
||||
}
|
||||
|
||||
if (std::getenv("CTEST_PROJECT_NAME") == nullptr) {
|
||||
Vtk::Viewer viewer;
|
||||
viewer.SetGridAxis(Vtk::Viewport::Y);
|
||||
viewer.AddProp3D(v_d1);
|
||||
viewer.AddProp3D(v_d2);
|
||||
viewer.AddPuppet(v_d1);
|
||||
viewer.AddPuppet(v_d2);
|
||||
viewer.Start();
|
||||
}
|
||||
|
||||
|
||||
@@ -23,8 +23,8 @@
|
||||
|
||||
//////////////////////////////////////////////////////////////////////////////*/
|
||||
|
||||
#include "Vtk/HEP/Detectors/vtkDetectorChamber.h"
|
||||
#include "Vtk/HEP/Detectors/vtkMuonScatter.h"
|
||||
#include "HEP/Detectors/DetectorChamber.h"
|
||||
#include "HEP/Detectors/MuonScatter.h"
|
||||
#include "Math/Units.h"
|
||||
#include "Vtk/HEP/Detectors/vtkDetectorChamber.h"
|
||||
#include "Vtk/HEP/Detectors/vtkMuonScatter.h"
|
||||
@@ -42,10 +42,10 @@
|
||||
|
||||
using namespace uLib;
|
||||
|
||||
// A simple prop3d class to represent an arrow indicative of a projected muon hit
|
||||
class vtkArrowProp3D : public Vtk::Prop3D {
|
||||
// A simple puppet class to represent an arrow indicative of a projected muon hit
|
||||
class vtkArrowPuppet : public Vtk::Puppet {
|
||||
public:
|
||||
vtkArrowProp3D() : m_Actor(vtkActor::New()) {
|
||||
vtkArrowPuppet() : m_Actor(vtkActor::New()) {
|
||||
vtkNew<vtkArrowSource> arrow;
|
||||
vtkNew<vtkPolyDataMapper> mapper;
|
||||
mapper->SetInputConnection(arrow->GetOutputPort());
|
||||
@@ -54,7 +54,7 @@ public:
|
||||
m_Actor->GetProperty()->SetColor(1, 1, 0); // Yellow color for visibility
|
||||
}
|
||||
|
||||
virtual ~vtkArrowProp3D() { m_Actor->Delete(); }
|
||||
virtual ~vtkArrowPuppet() { m_Actor->Delete(); }
|
||||
|
||||
void SetProjection(const HLine3f &line) {
|
||||
HPoint3f p = line.origin;
|
||||
@@ -125,30 +125,30 @@ BOOST_AUTO_TEST_CASE(vtkDetectorMuonProjectionTest) {
|
||||
MuonEvent mu_proj1 = d1.ProjectMuonEvent(mu_event);
|
||||
MuonEvent mu_proj2 = d2.ProjectMuonEvent(mu_event);
|
||||
|
||||
Vtk::DetectorChamber v_d1(&d1);
|
||||
Vtk::DetectorChamber v_d2(&d2);
|
||||
Vtk::MuonScatter v_event(event);
|
||||
Vtk::vtkDetectorChamber v_d1(&d1);
|
||||
Vtk::vtkDetectorChamber v_d2(&d2);
|
||||
Vtk::vtkMuonScatter v_event(event);
|
||||
v_event.AddPocaPoint(HPoint3f(0, 0, 0));
|
||||
v_event.SetColor(1, 0, 0); // Red muon event
|
||||
|
||||
v_d1.SetRepresentation(Vtk::Prop3D::Surface);
|
||||
v_d1.SetRepresentation(Vtk::Puppet::Surface);
|
||||
v_d1.SetOpacity(0.3);
|
||||
v_d2.SetRepresentation(Vtk::Prop3D::Surface);
|
||||
v_d2.SetRepresentation(Vtk::Puppet::Surface);
|
||||
v_d2.SetOpacity(0.3);
|
||||
|
||||
// 5. Add two arrows to mark where the projection is located on the chambers
|
||||
vtkArrowProp3D v_p1, v_p2;
|
||||
vtkArrowPuppet v_p1, v_p2;
|
||||
v_p1.SetProjection(mu_proj1.LineIn());
|
||||
v_p2.SetProjection(mu_proj2.LineIn());
|
||||
|
||||
if (std::getenv("CTEST_PROJECT_NAME") == nullptr) {
|
||||
Vtk::Viewer viewer;
|
||||
viewer.SetGridAxis(Vtk::Viewport::Z);
|
||||
viewer.AddProp3D(v_d1);
|
||||
viewer.AddProp3D(v_d2);
|
||||
viewer.AddProp3D(v_event);
|
||||
viewer.AddProp3D(v_p1);
|
||||
viewer.AddProp3D(v_p2);
|
||||
viewer.AddPuppet(v_d1);
|
||||
viewer.AddPuppet(v_d2);
|
||||
viewer.AddPuppet(v_event);
|
||||
viewer.AddPuppet(v_p1);
|
||||
viewer.AddPuppet(v_p2);
|
||||
viewer.Start();
|
||||
}
|
||||
|
||||
|
||||
@@ -24,7 +24,7 @@
|
||||
//////////////////////////////////////////////////////////////////////////////*/
|
||||
|
||||
#include "Vtk/HEP/Detectors/vtkMuonScatter.h"
|
||||
#include "Vtk/HEP/Detectors/vtkMuonScatter.h"
|
||||
#include "HEP/Detectors/MuonScatter.h"
|
||||
|
||||
#include "Vtk/uLibVtkViewer.h"
|
||||
|
||||
@@ -41,7 +41,7 @@ BOOST_AUTO_TEST_CASE(vtkMuonScatterTest) {
|
||||
event.LineOut().direction << 0, -1, 0, 0;
|
||||
event.LineOut().origin << 0, -1, 0, 1;
|
||||
|
||||
Vtk::MuonScatter v_event(event);
|
||||
Vtk::vtkMuonScatter v_event(event);
|
||||
v_event.AddPocaPoint(HPoint3f(0, 0, 0));
|
||||
|
||||
v_event.SaveToXMLFile("vtk_testing_muonevent.vtp");
|
||||
@@ -49,12 +49,12 @@ BOOST_AUTO_TEST_CASE(vtkMuonScatterTest) {
|
||||
if (std::getenv("CTEST_PROJECT_NAME") == nullptr) {
|
||||
Vtk::Viewer viewer;
|
||||
|
||||
// Vtk::Tie<Vtk::MuonScatter> tms;
|
||||
// Vtk::Tie<Vtk::vtkMuonScatter> tms;
|
||||
// tms.DoAction();
|
||||
// Vtk::Tie<Vtk::Viewer> vms;
|
||||
// vms.DoAction();
|
||||
|
||||
viewer.AddProp3D(v_event);
|
||||
viewer.AddPuppet(v_event);
|
||||
viewer.Start();
|
||||
}
|
||||
|
||||
|
||||
@@ -48,8 +48,8 @@
|
||||
namespace uLib {
|
||||
namespace Vtk {
|
||||
|
||||
DetectorChamber::DetectorChamber(uLib::DetectorChamber *content)
|
||||
: ContainerBox(content) {
|
||||
vtkDetectorChamber::vtkDetectorChamber(DetectorChamber *content)
|
||||
: vtkContainerBox(content) {
|
||||
m_PlaneSource = vtkPlaneSource::New();
|
||||
|
||||
vtkNew<vtkPolyDataMapper> mapper;
|
||||
@@ -67,20 +67,20 @@ DetectorChamber::DetectorChamber(uLib::DetectorChamber *content)
|
||||
this->Update();
|
||||
}
|
||||
|
||||
DetectorChamber::~DetectorChamber() {
|
||||
vtkDetectorChamber::~vtkDetectorChamber() {
|
||||
m_PlaneSource->Delete();
|
||||
m_PlaneActor->Delete();
|
||||
}
|
||||
|
||||
DetectorChamber::Content *DetectorChamber::GetContent() const {
|
||||
return static_cast<Content *>(this->m_model.get());
|
||||
DetectorChamber *vtkDetectorChamber::GetContent() {
|
||||
return static_cast<DetectorChamber *>(m_Content);
|
||||
}
|
||||
|
||||
void DetectorChamber::Update() {
|
||||
void vtkDetectorChamber::Update() {
|
||||
this->BaseClass::Update();
|
||||
|
||||
if (!this->m_model) return;
|
||||
Content *c = this->GetContent();
|
||||
if (!m_Content) return;
|
||||
DetectorChamber *c = this->GetContent();
|
||||
Vector3f size = c->GetSize();
|
||||
HLine3f plane = c->GetProjectionPlane();
|
||||
|
||||
|
||||
@@ -32,7 +32,7 @@
|
||||
#include <vtkTransform.h>
|
||||
|
||||
#include "HEP/Detectors/DetectorChamber.h"
|
||||
#include "Vtk/Math/vtkDense.h"
|
||||
#include "Math/Dense.h"
|
||||
#include "Vtk/uLibVtkInterface.h"
|
||||
#include "Vtk/Math/vtkContainerBox.h"
|
||||
#include <vtkActor.h>
|
||||
@@ -44,17 +44,17 @@ class vtkPlaneSource;
|
||||
namespace uLib {
|
||||
namespace Vtk {
|
||||
|
||||
// class HandlerWidget; // Removed as we use vtkBoxWidget now
|
||||
// class vtkHandlerWidget; // Removed as we use vtkBoxWidget now
|
||||
|
||||
class DetectorChamber : public ContainerBox {
|
||||
typedef uLib::DetectorChamber Content;
|
||||
typedef ContainerBox BaseClass;
|
||||
class vtkDetectorChamber : public vtkContainerBox {
|
||||
typedef DetectorChamber Content;
|
||||
typedef vtkContainerBox BaseClass;
|
||||
|
||||
public:
|
||||
DetectorChamber(uLib::DetectorChamber *content);
|
||||
virtual ~DetectorChamber();
|
||||
vtkDetectorChamber(DetectorChamber *content);
|
||||
virtual ~vtkDetectorChamber();
|
||||
|
||||
Content *GetContent() const override;
|
||||
Content *GetContent();
|
||||
|
||||
virtual void Update() override;
|
||||
|
||||
|
||||
@@ -27,7 +27,7 @@
|
||||
#include "config.h"
|
||||
#endif
|
||||
|
||||
#include "Vtk/Math/vtkDense.h"
|
||||
#include "Math/Dense.h"
|
||||
#include "Vtk/HEP/Detectors/vtkMuonEvent.h"
|
||||
|
||||
namespace uLib {
|
||||
@@ -38,7 +38,7 @@ namespace Vtk {
|
||||
class vtkWidgetCallback : public vtkCommand {
|
||||
public:
|
||||
static vtkWidgetCallback *New() { return new vtkWidgetCallback; }
|
||||
void SetParent(uLib::Vtk::MuonEvent *parent) { this->parent = parent; }
|
||||
void SetParent(uLib::Vtk::vtkMuonEvent *parent) { this->parent = parent; }
|
||||
virtual void Execute(vtkObject *caller, unsigned long, void *) {
|
||||
vtkSmartPointer<vtkTransform> t = vtkSmartPointer<vtkTransform>::New();
|
||||
vtkBoxWidget *widget = reinterpret_cast<vtkBoxWidget *>(caller);
|
||||
@@ -48,30 +48,30 @@ public:
|
||||
}
|
||||
|
||||
private:
|
||||
uLib::Vtk::MuonEvent *parent;
|
||||
uLib::Vtk::vtkMuonEvent *parent;
|
||||
};
|
||||
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
///// VTK MUON EVENT /////////////////////////////////////////////////////////
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
|
||||
MuonEvent::MuonEvent(MuonEventData &content)
|
||||
vtkMuonEvent::vtkMuonEvent(MuonEventData &content)
|
||||
: m_PolyData(NULL), m_Appender(vtkAppendPolyData::New()),
|
||||
content(&content) {
|
||||
InstallPipe();
|
||||
}
|
||||
|
||||
MuonEvent::MuonEvent(const MuonEventData &content)
|
||||
vtkMuonEvent::vtkMuonEvent(const MuonEventData &content)
|
||||
: m_PolyData(NULL), m_Appender(vtkAppendPolyData::New()),
|
||||
content(const_cast<MuonEventData *>(&content)) {
|
||||
InstallPipe();
|
||||
}
|
||||
|
||||
MuonEvent::~MuonEvent() {}
|
||||
vtkMuonEvent::~vtkMuonEvent() {}
|
||||
|
||||
MuonEventData &MuonEvent::GetModel() { return *content; }
|
||||
vtkMuonEvent::Content &vtkMuonEvent::GetContent() { return *content; }
|
||||
|
||||
void MuonEvent::PrintSelf(std::ostream &o) const {
|
||||
void vtkMuonEvent::PrintSelf(std::ostream &o) const {
|
||||
o << "..:: MuonEvent ::..\n"
|
||||
"\t[in] Origin > "
|
||||
<< content->LineIn().origin.transpose() << "\n"
|
||||
@@ -83,7 +83,7 @@ void MuonEvent::PrintSelf(std::ostream &o) const {
|
||||
<< "...................\n";
|
||||
}
|
||||
|
||||
void MuonEvent::InstallPipe() {
|
||||
void vtkMuonEvent::InstallPipe() {
|
||||
|
||||
vtkAppendPolyData *appender = m_Appender;
|
||||
vtkSmartPointer<vtkPolyDataMapper> mapper =
|
||||
@@ -121,11 +121,11 @@ void MuonEvent::InstallPipe() {
|
||||
this->SetProp(actor);
|
||||
}
|
||||
|
||||
vtkPolyData *MuonEvent::GetPolyData() const {
|
||||
vtkPolyData *vtkMuonEvent::GetPolyData() const {
|
||||
return m_Appender->GetOutput();
|
||||
}
|
||||
|
||||
void MuonEvent::AddPocaPoint(HPoint3f poca) {
|
||||
void vtkMuonEvent::AddPocaPoint(HPoint3f poca) {
|
||||
m_Poca = poca;
|
||||
vtkSmartPointer<vtkSphereSource> sphere =
|
||||
vtkSmartPointer<vtkSphereSource>::New();
|
||||
@@ -139,8 +139,7 @@ void MuonEvent::AddPocaPoint(HPoint3f poca) {
|
||||
m_Appender->Update();
|
||||
}
|
||||
|
||||
HPoint3f MuonEvent::GetPocaPoint() { return m_Poca; }
|
||||
HPoint3f vtkMuonEvent::GetPocaPoint() { return m_Poca; }
|
||||
|
||||
uLib::Object* MuonEvent::GetContent() const { return (uLib::Object*)content; }
|
||||
} // namespace Vtk
|
||||
} // namespace uLib
|
||||
|
||||
@@ -41,7 +41,7 @@
|
||||
#include <vtkCommand.h>
|
||||
#include <vtkTransform.h>
|
||||
|
||||
#include "Vtk/Math/vtkDense.h"
|
||||
#include "Math/Dense.h"
|
||||
|
||||
#include "HEP/Detectors/MuonEvent.h"
|
||||
#include "Vtk/uLibVtkInterface.h"
|
||||
@@ -50,20 +50,19 @@
|
||||
namespace uLib {
|
||||
namespace Vtk {
|
||||
|
||||
class MuonEvent : public Prop3D, public Polydata {
|
||||
class vtkMuonEvent : public Puppet, public Polydata {
|
||||
typedef MuonEventData Content;
|
||||
|
||||
public:
|
||||
MuonEvent(const MuonEventData &content);
|
||||
MuonEvent(MuonEventData &content);
|
||||
~MuonEvent();
|
||||
vtkMuonEvent(const MuonEventData &content);
|
||||
vtkMuonEvent(MuonEventData &content);
|
||||
~vtkMuonEvent();
|
||||
|
||||
Content &GetModel();
|
||||
uLib::Object* GetContent() const override;
|
||||
Content &GetContent();
|
||||
|
||||
void PrintSelf(std::ostream &o) const override;
|
||||
void PrintSelf(std::ostream &o) const;
|
||||
|
||||
vtkPolyData *GetPolyData() const override;
|
||||
virtual vtkPolyData *GetPolyData() const;
|
||||
|
||||
void AddPocaPoint(HPoint3f poca);
|
||||
|
||||
@@ -74,7 +73,7 @@ public:
|
||||
private:
|
||||
void InstallPipe();
|
||||
|
||||
MuonEvent::Content *content;
|
||||
vtkMuonEvent::Content *content;
|
||||
vtkPolyData *m_PolyData;
|
||||
|
||||
vtkAppendPolyData *m_Appender;
|
||||
|
||||
@@ -27,7 +27,7 @@
|
||||
#include "config.h"
|
||||
#endif
|
||||
|
||||
#include "Vtk/Math/vtkDense.h"
|
||||
#include "Math/Dense.h"
|
||||
#include "Vtk/HEP/Detectors/vtkMuonScatter.h"
|
||||
|
||||
namespace uLib {
|
||||
@@ -37,21 +37,21 @@ namespace Vtk {
|
||||
///// VTK MUON Scatter ///////////////////////////////////////////////////////
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
|
||||
MuonScatter::MuonScatter(uLib::MuonScatter &content)
|
||||
vtkMuonScatter::vtkMuonScatter(MuonScatter &content)
|
||||
: m_Content(&content), m_LineIn(vtkLineSource::New()),
|
||||
m_LineOut(vtkLineSource::New()), m_PolyData(vtkPolyData::New()),
|
||||
m_SpherePoca(NULL), m_Asm(vtkAssembly::New()) {
|
||||
InstallPipe();
|
||||
}
|
||||
|
||||
MuonScatter::MuonScatter(const MuonScatter &content)
|
||||
: m_Content(content.m_Content),
|
||||
vtkMuonScatter::vtkMuonScatter(const MuonScatter &content)
|
||||
: m_Content(const_cast<MuonScatter *>(&content)),
|
||||
m_LineIn(vtkLineSource::New()), m_LineOut(vtkLineSource::New()),
|
||||
m_PolyData(vtkPolyData::New()), m_SpherePoca(NULL), m_Asm(vtkAssembly::New()) {
|
||||
InstallPipe();
|
||||
}
|
||||
|
||||
MuonScatter::~MuonScatter() {
|
||||
vtkMuonScatter::~vtkMuonScatter() {
|
||||
m_LineIn->Delete();
|
||||
m_LineOut->Delete();
|
||||
m_Asm->Delete();
|
||||
@@ -59,11 +59,11 @@ MuonScatter::~MuonScatter() {
|
||||
m_SpherePoca->Delete();
|
||||
}
|
||||
|
||||
MuonScatter::Content &MuonScatter::GetModel() { return *m_Content; }
|
||||
vtkMuonScatter::Content &vtkMuonScatter::GetContent() { return *m_Content; }
|
||||
|
||||
void MuonScatter::PrintSelf(std::ostream &o) const {}
|
||||
void vtkMuonScatter::PrintSelf(std::ostream &o) const {}
|
||||
|
||||
void MuonScatter::InstallPipe() {
|
||||
void vtkMuonScatter::InstallPipe() {
|
||||
if (m_Content) {
|
||||
vtkLineSource *line_in = m_LineIn;
|
||||
vtkLineSource *line_out = m_LineOut;
|
||||
@@ -99,7 +99,7 @@ void MuonScatter::InstallPipe() {
|
||||
this->SetProp(m_Asm);
|
||||
}
|
||||
|
||||
vtkPolyData *MuonScatter::GetPolyData() const {
|
||||
vtkPolyData *vtkMuonScatter::GetPolyData() const {
|
||||
vtkSmartPointer<vtkAppendPolyData> append =
|
||||
vtkSmartPointer<vtkAppendPolyData>::New();
|
||||
append->AddInputConnection(m_LineIn->GetOutputPort());
|
||||
@@ -111,7 +111,7 @@ vtkPolyData *MuonScatter::GetPolyData() const {
|
||||
return m_PolyData;
|
||||
}
|
||||
|
||||
void MuonScatter::AddPocaPoint(HPoint3f poca) {
|
||||
void vtkMuonScatter::AddPocaPoint(HPoint3f poca) {
|
||||
vtkSphereSource *sphere = vtkSphereSource::New();
|
||||
float size =
|
||||
(m_Content->LineIn().origin - m_Content->LineOut().origin).head(3).norm();
|
||||
@@ -129,7 +129,7 @@ void MuonScatter::AddPocaPoint(HPoint3f poca) {
|
||||
m_Asm->AddPart(actor);
|
||||
}
|
||||
|
||||
HPoint3f MuonScatter::GetPocaPoint() {
|
||||
HPoint3f vtkMuonScatter::GetPocaPoint() {
|
||||
double center[3];
|
||||
if (m_SpherePoca) {
|
||||
m_SpherePoca->GetCenter(center);
|
||||
@@ -139,7 +139,7 @@ HPoint3f MuonScatter::GetPocaPoint() {
|
||||
}
|
||||
}
|
||||
|
||||
void MuonScatter::ConnectInteractor(vtkRenderWindowInteractor *interactor) {
|
||||
void vtkMuonScatter::ConnectInteractor(vtkRenderWindowInteractor *interactor) {
|
||||
// TODO
|
||||
}
|
||||
|
||||
|
||||
@@ -43,7 +43,7 @@
|
||||
#include <vtkCommand.h>
|
||||
#include <vtkTransform.h>
|
||||
|
||||
#include "Vtk/Math/vtkDense.h"
|
||||
#include "Math/Dense.h"
|
||||
|
||||
#include "HEP/Detectors/MuonScatter.h"
|
||||
#include "Vtk/uLibVtkInterface.h"
|
||||
@@ -54,21 +54,20 @@ class vtkRenderWindowInteractor;
|
||||
namespace uLib {
|
||||
namespace Vtk {
|
||||
|
||||
class MuonScatter : public Prop3D, public Polydata {
|
||||
typedef uLib::MuonScatter Content;
|
||||
class vtkMuonScatter : public Puppet, public Polydata {
|
||||
typedef MuonScatter Content;
|
||||
|
||||
public:
|
||||
MuonScatter(const MuonScatter &content);
|
||||
MuonScatter(uLib::MuonScatter &content);
|
||||
vtkMuonScatter(const MuonScatter &content);
|
||||
vtkMuonScatter(MuonScatter &content);
|
||||
|
||||
~MuonScatter();
|
||||
~vtkMuonScatter();
|
||||
|
||||
Content &GetModel();
|
||||
uLib::Object* GetContent() const override { return (uLib::Object*)m_Content; }
|
||||
Content &GetContent();
|
||||
|
||||
void PrintSelf(std::ostream &o) const override;
|
||||
void PrintSelf(std::ostream &o) const;
|
||||
|
||||
virtual vtkPolyData *GetPolyData() const override;
|
||||
virtual vtkPolyData *GetPolyData() const;
|
||||
|
||||
void AddPocaPoint(HPoint3f poca);
|
||||
|
||||
@@ -77,12 +76,12 @@ public:
|
||||
void vtkStartInteractive();
|
||||
|
||||
protected:
|
||||
void ConnectInteractor(vtkRenderWindowInteractor *interactor) override;
|
||||
void ConnectInteractor(vtkRenderWindowInteractor *interactor);
|
||||
|
||||
private:
|
||||
void InstallPipe();
|
||||
|
||||
MuonScatter::Content *m_Content;
|
||||
vtkMuonScatter::Content *m_Content;
|
||||
vtkLineSource *m_LineIn;
|
||||
vtkLineSource *m_LineOut;
|
||||
vtkSphereSource *m_SpherePoca;
|
||||
|
||||
@@ -10,11 +10,11 @@
|
||||
//////////////////////////////////////////////////////////////////////////////*/
|
||||
|
||||
#include "Geant/Solid.h"
|
||||
#include "Vtk/HEP/Geant/vtkGeantEvent.h"
|
||||
#include "HEP/Geant/GeantEvent.h"
|
||||
#include "HEP/Geant/Scene.h"
|
||||
#include "HEP/Geant/EmitterPrimary.hh"
|
||||
#include "Vtk/Math/vtkCylinder.h"
|
||||
#include "Vtk/Math/vtkDense.h"
|
||||
#include "Math/Cylinder.h"
|
||||
#include "Math/Dense.h"
|
||||
#include "Math/Units.h"
|
||||
#include "Vtk/uLibVtkViewer.h"
|
||||
#include "Vtk/HEP/Geant/vtkGeantEvent.h"
|
||||
@@ -58,8 +58,8 @@ void KeyPressCallbackFunction(vtkObject* caller, long unsigned int eventId, void
|
||||
std::cout << " Collected event " << lastEvent->GetEventID()
|
||||
<< " with " << lastEvent->Path().size() << " steps." << std::endl;
|
||||
|
||||
Vtk::GeantEvent* vtkEvent = new Vtk::GeantEvent(lastEvent);
|
||||
state->viewer->AddProp3D(*vtkEvent);
|
||||
Vtk::vtkGeantEvent* vtkEvent = new Vtk::vtkGeantEvent(lastEvent);
|
||||
state->viewer->AddPuppet(*vtkEvent);
|
||||
|
||||
state->viewer->GetRenderer()->Render();
|
||||
state->viewer->GetRenderWindow()->Render();
|
||||
@@ -97,21 +97,21 @@ int main(int argc, char** argv) {
|
||||
Vtk::Viewer viewer;
|
||||
viewer.GetRenderer()->SetBackground(0.05, 0.05, 0.1);
|
||||
|
||||
Vtk::ContainerBox* vtkWorld = new Vtk::ContainerBox(scene.GetWorldBox());
|
||||
Vtk::vtkContainerBox* vtkWorld = new Vtk::vtkContainerBox(scene.GetWorldBox());
|
||||
vtkWorld->ShowScaleMeasures(true);
|
||||
vtkWorld->SetRepresentation(Vtk::Prop3D::Wireframe);
|
||||
vtkWorld->SetRepresentation(Vtk::Puppet::Wireframe);
|
||||
vtkWorld->SetSelectable(false);
|
||||
viewer.AddProp3D(*vtkWorld);
|
||||
viewer.AddPuppet(*vtkWorld);
|
||||
|
||||
Vtk::ContainerBox* vtkIron = new Vtk::ContainerBox(&iron_box);
|
||||
Vtk::vtkContainerBox* vtkIron = new Vtk::vtkContainerBox(&iron_box);
|
||||
vtkIron->SetOpacity(0.2);
|
||||
vtkIron->SetRepresentation(Vtk::Prop3D::Surface);
|
||||
viewer.AddProp3D(*vtkIron);
|
||||
vtkIron->SetRepresentation(Vtk::Puppet::Surface);
|
||||
viewer.AddPuppet(*vtkIron);
|
||||
|
||||
// Use vtkCylinderEmitterPrimary
|
||||
Vtk::vtkCylinderEmitterPrimary* vtkEmitter = new Vtk::vtkCylinderEmitterPrimary(*emitter);
|
||||
vtkEmitter->SetSelectable(false);
|
||||
viewer.AddProp3D(*vtkEmitter);
|
||||
viewer.AddPuppet(*vtkEmitter);
|
||||
|
||||
// 3. Event Handling
|
||||
AppState state = { &scene, &viewer, {} };
|
||||
|
||||
@@ -1,15 +1,15 @@
|
||||
#include "Geant/Solid.h"
|
||||
#include "Vtk/HEP/Geant/vtkGeantEvent.h"
|
||||
#include "HEP/Geant/GeantEvent.h"
|
||||
#include "HEP/Geant/Scene.h"
|
||||
#include "HEP/Geant/EmitterPrimary.hh"
|
||||
#include "Vtk/Math/vtkContainerBox.h"
|
||||
#include "Vtk/Math/vtkDense.h"
|
||||
#include "Math/ContainerBox.h"
|
||||
#include "Math/Dense.h"
|
||||
#include "Math/Units.h"
|
||||
#include "Vtk/uLibVtkViewer.h"
|
||||
#include "Vtk/HEP/Geant/vtkGeantEvent.h"
|
||||
#include "Vtk/HEP/Geant/vtkEmitterPrimary.h"
|
||||
#include "Vtk/Math/vtkContainerBox.h"
|
||||
#include "Vtk/HEP/Detectors/vtkDetectorChamber.h"
|
||||
#include "HEP/Detectors/DetectorChamber.h"
|
||||
#include "Vtk/HEP/Detectors/vtkDetectorChamber.h"
|
||||
|
||||
#include <vtkSmartPointer.h>
|
||||
@@ -51,8 +51,8 @@ void KeyPressCallbackFunction(vtkObject* caller, long unsigned int eventId, void
|
||||
<< " with " << lastEvent->Path().size() << " steps." << std::endl;
|
||||
|
||||
// Wrap it for VTK
|
||||
Vtk::GeantEvent* vtkEvent = new Vtk::GeantEvent(lastEvent);
|
||||
state->viewer->AddProp3D(*vtkEvent);
|
||||
Vtk::vtkGeantEvent* vtkEvent = new Vtk::vtkGeantEvent(lastEvent);
|
||||
state->viewer->AddPuppet(*vtkEvent);
|
||||
|
||||
// Re-render
|
||||
state->viewer->GetRenderer()->Render();
|
||||
@@ -105,35 +105,35 @@ int main(int argc, char** argv) {
|
||||
viewer.GetRenderer()->SetBackground(0.05, 0.05, 0.1);
|
||||
|
||||
// Visualize world box
|
||||
Vtk::ContainerBox* vtkWorld = new Vtk::ContainerBox(scene.GetWorldBox());
|
||||
Vtk::vtkContainerBox* vtkWorld = new Vtk::vtkContainerBox(scene.GetWorldBox());
|
||||
vtkWorld->ShowScaleMeasures(true);
|
||||
vtkWorld->SetRepresentation(Vtk::Prop3D::Wireframe);
|
||||
vtkWorld->SetRepresentation(Vtk::Puppet::Wireframe);
|
||||
vtkWorld->SetSelectable(false);
|
||||
viewer.AddProp3D(*vtkWorld);
|
||||
viewer.AddPuppet(*vtkWorld);
|
||||
|
||||
// Visualize iron cube
|
||||
Vtk::ContainerBox* vtkIron = new Vtk::ContainerBox(&iron_box);
|
||||
Vtk::vtkContainerBox* vtkIron = new Vtk::vtkContainerBox(&iron_box);
|
||||
vtkIron->SetOpacity(0.2);
|
||||
vtkIron->SetRepresentation(Vtk::Prop3D::Surface);
|
||||
viewer.AddProp3D(*vtkIron);
|
||||
vtkIron->SetRepresentation(Vtk::Puppet::Surface);
|
||||
viewer.AddPuppet(*vtkIron);
|
||||
|
||||
// Visualize Top Chamber
|
||||
Vtk::DetectorChamber* vtkTop = new Vtk::DetectorChamber(top_chamber_box);
|
||||
Vtk::vtkDetectorChamber* vtkTop = new Vtk::vtkDetectorChamber(top_chamber_box);
|
||||
vtkTop->SetOpacity(0.5);
|
||||
vtkTop->SetColor(0.2, 0.8, 0.2);
|
||||
vtkTop->SetRepresentation(Vtk::Prop3D::Surface);
|
||||
viewer.AddProp3D(*vtkTop);
|
||||
vtkTop->SetRepresentation(Vtk::Puppet::Surface);
|
||||
viewer.AddPuppet(*vtkTop);
|
||||
|
||||
// Visualize Bottom Chamber
|
||||
Vtk::DetectorChamber* vtkBottom = new Vtk::DetectorChamber(bottom_chamber_box);
|
||||
Vtk::vtkDetectorChamber* vtkBottom = new Vtk::vtkDetectorChamber(bottom_chamber_box);
|
||||
vtkBottom->SetOpacity(0.5);
|
||||
vtkBottom->SetColor(0.2, 0.8, 0.2);
|
||||
vtkBottom->SetRepresentation(Vtk::Prop3D::Surface);
|
||||
viewer.AddProp3D(*vtkBottom);
|
||||
vtkBottom->SetRepresentation(Vtk::Puppet::Surface);
|
||||
viewer.AddPuppet(*vtkBottom);
|
||||
|
||||
// Visualize Emitter
|
||||
Vtk::EmitterPrimary* vtkEmitter = new Vtk::EmitterPrimary(*emitter);
|
||||
viewer.AddProp3D(*vtkEmitter);
|
||||
Vtk::vtkEmitterPrimary* vtkEmitter = new Vtk::vtkEmitterPrimary(*emitter);
|
||||
viewer.AddPuppet(*vtkEmitter);
|
||||
|
||||
// 3. Event Handling
|
||||
AppState state = { &scene, &viewer, {} };
|
||||
|
||||
@@ -10,11 +10,11 @@
|
||||
//////////////////////////////////////////////////////////////////////////////*/
|
||||
|
||||
#include "Geant/Solid.h"
|
||||
#include "Vtk/HEP/Geant/vtkGeantEvent.h"
|
||||
#include "HEP/Geant/GeantEvent.h"
|
||||
#include "HEP/Geant/Scene.h"
|
||||
#include "HEP/Geant/EmitterPrimary.hh"
|
||||
#include "Vtk/Math/vtkContainerBox.h"
|
||||
#include "Vtk/Math/vtkDense.h"
|
||||
#include "Math/ContainerBox.h"
|
||||
#include "Math/Dense.h"
|
||||
#include "Math/Units.h"
|
||||
#include "Vtk/uLibVtkViewer.h"
|
||||
#include "Vtk/HEP/Geant/vtkGeantEvent.h"
|
||||
@@ -82,8 +82,8 @@ void KeyPressCallbackFunction(vtkObject* caller, long unsigned int eventId, void
|
||||
<< " with " << lastEvent->Path().size() << " steps." << std::endl;
|
||||
|
||||
// Wrap it for VTK
|
||||
Vtk::GeantEvent* vtkEvent = new Vtk::GeantEvent(lastEvent);
|
||||
state->viewer->AddProp3D(*vtkEvent);
|
||||
Vtk::vtkGeantEvent* vtkEvent = new Vtk::vtkGeantEvent(lastEvent);
|
||||
state->viewer->AddPuppet(*vtkEvent);
|
||||
|
||||
// Re-render
|
||||
state->viewer->GetRenderer()->Render();
|
||||
@@ -117,16 +117,16 @@ int main(int argc, char** argv) {
|
||||
viewer.GetRenderer()->SetBackground(0.05, 0.05, 0.1);
|
||||
|
||||
// Visualize world box
|
||||
Vtk::ContainerBox* vtkWorld = new Vtk::ContainerBox(scene.GetWorldBox());
|
||||
Vtk::vtkContainerBox* vtkWorld = new Vtk::vtkContainerBox(scene.GetWorldBox());
|
||||
// vtkWorld->ShowBoundingBox(true);
|
||||
vtkWorld->ShowScaleMeasures(true);
|
||||
viewer.AddProp3D(*vtkWorld);
|
||||
viewer.AddPuppet(*vtkWorld);
|
||||
|
||||
// Visualize iron cube
|
||||
Vtk::ContainerBox* vtkIron = new Vtk::ContainerBox(&iron_box);
|
||||
Vtk::vtkContainerBox* vtkIron = new Vtk::vtkContainerBox(&iron_box);
|
||||
vtkIron->SetOpacity(0.2);
|
||||
vtkIron->SetRepresentation(Vtk::Prop3D::Surface);
|
||||
viewer.AddProp3D(*vtkIron);
|
||||
vtkIron->SetRepresentation(Vtk::Puppet::Surface);
|
||||
viewer.AddPuppet(*vtkIron);
|
||||
|
||||
// 3. Event Handling
|
||||
AppState state = { &scene, &viewer, {} };
|
||||
|
||||
@@ -11,8 +11,8 @@
|
||||
|
||||
#include "Geant/Solid.h"
|
||||
#include "HEP/Geant/Scene.h"
|
||||
#include "Vtk/Math/vtkContainerBox.h"
|
||||
#include "Vtk/Math/vtkDense.h"
|
||||
#include "Math/ContainerBox.h"
|
||||
#include "Math/Dense.h"
|
||||
#include "Math/Units.h"
|
||||
#include "Vtk/uLibVtkViewer.h"
|
||||
#include "Vtk/HEP/Geant/vtkGeantScene.h"
|
||||
@@ -46,11 +46,11 @@ int main(int argc, char** argv) {
|
||||
Vtk::Viewer viewer;
|
||||
viewer.GetRenderer()->SetBackground(0.05, 0.05, 0.1);
|
||||
|
||||
Vtk::GeantScene vtkScene(&scene);
|
||||
Vtk::vtkGeantScene vtkScene(&scene);
|
||||
vtkScene.AddToViewer(viewer);
|
||||
|
||||
std::cout << "==================================================" << std::endl;
|
||||
std::cout << " GeantScene Test" << std::endl;
|
||||
std::cout << " vtkGeantScene Test" << std::endl;
|
||||
std::cout << " World box + 1 iron cube displayed" << std::endl;
|
||||
std::cout << "==================================================" << std::endl;
|
||||
|
||||
|
||||
@@ -1,15 +1,15 @@
|
||||
#include "Geant/Solid.h"
|
||||
#include "Vtk/HEP/Geant/vtkGeantEvent.h"
|
||||
#include "HEP/Geant/GeantEvent.h"
|
||||
#include "HEP/Geant/Scene.h"
|
||||
#include "HEP/Geant/EmitterPrimary.hh"
|
||||
#include "Vtk/Math/vtkContainerBox.h"
|
||||
#include "Vtk/Math/vtkDense.h"
|
||||
#include "Math/ContainerBox.h"
|
||||
#include "Math/Dense.h"
|
||||
#include "Math/Units.h"
|
||||
#include "Vtk/uLibVtkViewer.h"
|
||||
#include "Vtk/HEP/Geant/vtkGeantEvent.h"
|
||||
#include "Vtk/HEP/Geant/vtkEmitterPrimary.h"
|
||||
#include "Vtk/Math/vtkContainerBox.h"
|
||||
#include "Vtk/HEP/Detectors/vtkDetectorChamber.h"
|
||||
#include "HEP/Detectors/DetectorChamber.h"
|
||||
#include "Vtk/HEP/Detectors/vtkDetectorChamber.h"
|
||||
|
||||
#include <vtkSmartPointer.h>
|
||||
@@ -49,8 +49,8 @@ void KeyPressCallbackFunction(vtkObject* caller, long unsigned int eventId, void
|
||||
std::cout << " Collected event " << lastEvent->GetEventID()
|
||||
<< " with " << lastEvent->Path().size() << " steps." << std::endl;
|
||||
|
||||
Vtk::GeantEvent* vtkEvent = new Vtk::GeantEvent(lastEvent);
|
||||
state->viewer->AddProp3D(*vtkEvent);
|
||||
Vtk::vtkGeantEvent* vtkEvent = new Vtk::vtkGeantEvent(lastEvent);
|
||||
state->viewer->AddPuppet(*vtkEvent);
|
||||
|
||||
state->viewer->GetRenderer()->Render();
|
||||
state->viewer->GetRenderWindow()->Render();
|
||||
@@ -107,35 +107,35 @@ int main(int argc, char** argv) {
|
||||
Vtk::Viewer viewer;
|
||||
viewer.GetRenderer()->SetBackground(0.05, 0.05, 0.1);
|
||||
|
||||
Vtk::ContainerBox* vtkWorld = new Vtk::ContainerBox(scene.GetWorldBox());
|
||||
Vtk::vtkContainerBox* vtkWorld = new Vtk::vtkContainerBox(scene.GetWorldBox());
|
||||
vtkWorld->ShowScaleMeasures(true);
|
||||
vtkWorld->SetRepresentation(Vtk::Prop3D::Wireframe);
|
||||
vtkWorld->SetRepresentation(Vtk::Puppet::Wireframe);
|
||||
vtkWorld->SetSelectable(false);
|
||||
viewer.AddProp3D(*vtkWorld);
|
||||
viewer.AddPuppet(*vtkWorld);
|
||||
|
||||
Vtk::ContainerBox* vtkIron = new Vtk::ContainerBox(&iron_box);
|
||||
Vtk::vtkContainerBox* vtkIron = new Vtk::vtkContainerBox(&iron_box);
|
||||
vtkIron->SetOpacity(0.2);
|
||||
vtkIron->SetRepresentation(Vtk::Prop3D::Surface);
|
||||
viewer.AddProp3D(*vtkIron);
|
||||
vtkIron->SetRepresentation(Vtk::Puppet::Surface);
|
||||
viewer.AddPuppet(*vtkIron);
|
||||
|
||||
// Visualize Top Chamber
|
||||
Vtk::DetectorChamber* vtkTop = new Vtk::DetectorChamber(top_chamber_box);
|
||||
Vtk::vtkDetectorChamber* vtkTop = new Vtk::vtkDetectorChamber(top_chamber_box);
|
||||
vtkTop->SetOpacity(0.5);
|
||||
vtkTop->SetColor(0.2, 0.8, 0.2);
|
||||
vtkTop->SetRepresentation(Vtk::Prop3D::Surface);
|
||||
viewer.AddProp3D(*vtkTop);
|
||||
vtkTop->SetRepresentation(Vtk::Puppet::Surface);
|
||||
viewer.AddPuppet(*vtkTop);
|
||||
|
||||
// Visualize Bottom Chamber
|
||||
Vtk::DetectorChamber* vtkBottom = new Vtk::DetectorChamber(bottom_chamber_box);
|
||||
Vtk::vtkDetectorChamber* vtkBottom = new Vtk::vtkDetectorChamber(bottom_chamber_box);
|
||||
vtkBottom->SetOpacity(0.5);
|
||||
vtkBottom->SetColor(0.2, 0.8, 0.2);
|
||||
vtkBottom->SetRepresentation(Vtk::Prop3D::Surface);
|
||||
viewer.AddProp3D(*vtkBottom);
|
||||
vtkBottom->SetRepresentation(Vtk::Puppet::Surface);
|
||||
viewer.AddPuppet(*vtkBottom);
|
||||
|
||||
// Use vtkSkyPlaneEmitterPrimary instead of EmitterPrimary
|
||||
// Use vtkSkyPlaneEmitterPrimary instead of vtkEmitterPrimary
|
||||
Vtk::vtkSkyPlaneEmitterPrimary* vtkEmitter = new Vtk::vtkSkyPlaneEmitterPrimary(*emitter);
|
||||
vtkEmitter->SetSelectable(false);
|
||||
viewer.AddProp3D(*vtkEmitter);
|
||||
viewer.AddPuppet(*vtkEmitter);
|
||||
|
||||
// 3. Event Handling
|
||||
AppState state = { &scene, &viewer, {} };
|
||||
|
||||
@@ -13,8 +13,8 @@
|
||||
#include "Vtk/HEP/Geant/vtkTessellatedSolid.h"
|
||||
#include "Vtk/uLibVtkViewer.h"
|
||||
#include "HEP/Geant/Solid.h"
|
||||
#include "Vtk/Math/vtkContainerBox.h"
|
||||
#include "Vtk/Math/vtkTriangleMesh.h"
|
||||
#include "Math/ContainerBox.h"
|
||||
#include "Math/TriangleMesh.h"
|
||||
#include "Math/Units.h"
|
||||
|
||||
#include <vtkProperty.h>
|
||||
@@ -31,7 +31,7 @@ int main(int argc, char** argv) {
|
||||
Geant::BoxSolid gBox("MyBox", &box);
|
||||
gBox.Update();
|
||||
|
||||
Vtk::BoxSolid vtkBox(&gBox);
|
||||
Vtk::vtkBoxSolid vtkBox(&gBox);
|
||||
|
||||
// 2. Create a TessellatedSolid
|
||||
Geant::TessellatedSolid gTess("MyTess");
|
||||
@@ -53,7 +53,7 @@ int main(int argc, char** argv) {
|
||||
gTess.SetMesh(mesh);
|
||||
gTess.Update();
|
||||
|
||||
Vtk::TessellatedSolid vtkTess(&gTess);
|
||||
Vtk::vtkTessellatedSolid vtkTess(&gTess);
|
||||
|
||||
// 3. Visualization setup
|
||||
Vtk::Viewer viewer;
|
||||
@@ -61,8 +61,8 @@ int main(int argc, char** argv) {
|
||||
vtkTess.AddToViewer(viewer);
|
||||
|
||||
// Color them differently
|
||||
vtkBox.SetColor(0.8, 0.2, 0.2); // Redish box
|
||||
vtkTess.SetColor(0.2, 0.8, 0.2); // Greenish tess
|
||||
vtkActor::SafeDownCast(vtkBox.GetProp())->GetProperty()->SetColor(0.8, 0.2, 0.2); // Redish box
|
||||
vtkActor::SafeDownCast(vtkTess.GetProp())->GetProperty()->SetColor(0.2, 0.8, 0.2); // Greenish tess
|
||||
|
||||
// Position tessellated solid away from box
|
||||
Matrix4f trans = Matrix4f::Identity();
|
||||
|
||||
@@ -10,41 +10,47 @@
|
||||
//////////////////////////////////////////////////////////////////////////////*/
|
||||
|
||||
#include "vtkBoxSolid.h"
|
||||
#include "Core/Monitor.h"
|
||||
#include <vtkCubeSource.h>
|
||||
#include <vtkPolyDataMapper.h>
|
||||
#include <vtkActor.h>
|
||||
#include <vtkProperty.h>
|
||||
#include <vtkAssembly.h>
|
||||
#include <vtkTransform.h>
|
||||
#include <vtkMatrix4x4.h>
|
||||
#include <Geant4/G4VPhysicalVolume.hh>
|
||||
#include "Vtk/Math/vtkDense.h"
|
||||
|
||||
namespace uLib {
|
||||
namespace Vtk {
|
||||
|
||||
BoxSolid::BoxSolid(Geant::BoxSolid *content)
|
||||
: GeantSolid(content), m_BoxContent(content) {
|
||||
vtkBoxSolid::vtkBoxSolid(Geant::BoxSolid *content)
|
||||
: vtkGeantSolid(content), m_BoxContent(content) {
|
||||
// Re-run Update for box-specific pipe
|
||||
this->Update();
|
||||
}
|
||||
|
||||
BoxSolid::~BoxSolid() {
|
||||
vtkBoxSolid::~vtkBoxSolid() {}
|
||||
|
||||
void vtkBoxSolid::Update() {
|
||||
this->UpdateGeometry();
|
||||
this->UpdateTransform();
|
||||
}
|
||||
|
||||
void BoxSolid::serialize_display(uLib::Archive::display_properties_archive &ar,
|
||||
const unsigned int version) {
|
||||
// Expose Geant solid properties and underlying Box/TRS properties
|
||||
this->Prop3D::serialize_display(ar, version);
|
||||
if (m_BoxContent) {
|
||||
ar & NVP("Box", *m_BoxContent);
|
||||
if (m_BoxContent->GetObject()) {
|
||||
ar & NVP("Container", *m_BoxContent->GetObject());
|
||||
}
|
||||
void vtkBoxSolid::UpdateGeometry() {
|
||||
if (!m_BoxContent || !m_BoxContent->GetObject()) {
|
||||
// Fallback to base tessellation if no model object
|
||||
vtkGeantSolid::UpdateGeometry();
|
||||
return;
|
||||
}
|
||||
}
|
||||
|
||||
void BoxSolid::InstallPipe() {
|
||||
GeantSolid::InstallPipe();
|
||||
// Use the underlying ContainerBox for precise geometry
|
||||
Vector3f size = m_BoxContent->GetObject()->GetSize();
|
||||
|
||||
vtkNew<vtkCubeSource> cube;
|
||||
cube->SetXLength(size(0));
|
||||
cube->SetYLength(size(1));
|
||||
cube->SetZLength(size(2));
|
||||
cube->Update();
|
||||
|
||||
vtkPolyData *poly = GetPolyData();
|
||||
if (poly) {
|
||||
poly->ShallowCopy(cube->GetOutput());
|
||||
poly->Modified();
|
||||
}
|
||||
}
|
||||
|
||||
} // namespace Vtk
|
||||
|
||||
@@ -26,42 +26,24 @@
|
||||
#ifndef U_VTKBOXSOLID_H
|
||||
#define U_VTKBOXSOLID_H
|
||||
|
||||
#include "Core/Types.h"
|
||||
#include "Core/Property.h"
|
||||
#include "Core/Serializable.h"
|
||||
|
||||
#include "vtkGeantSolid.h"
|
||||
|
||||
class vtkCubeSource;
|
||||
class vtkActor;
|
||||
|
||||
namespace uLib {
|
||||
namespace Vtk {
|
||||
|
||||
/**
|
||||
* @brief VTK Prop3D for visualizing a Geant::BoxSolid.
|
||||
* @brief VTK Puppet for visualizing a Geant::BoxSolid.
|
||||
*/
|
||||
class BoxSolid : public GeantSolid {
|
||||
uLibTypeMacro(BoxSolid, uLib::Vtk::GeantSolid)
|
||||
|
||||
class vtkBoxSolid : public vtkGeantSolid {
|
||||
public:
|
||||
BoxSolid(Geant::BoxSolid *content);
|
||||
virtual ~BoxSolid();
|
||||
vtkBoxSolid(Geant::BoxSolid *content);
|
||||
virtual ~vtkBoxSolid();
|
||||
|
||||
virtual void serialize_display(uLib::Archive::display_properties_archive &ar,
|
||||
const unsigned int version = 0) override;
|
||||
|
||||
template <typename Ar>
|
||||
void serialize(Ar &ar, const unsigned int version) {
|
||||
ar & NVP("BoxSolid", *m_BoxContent);
|
||||
}
|
||||
virtual void Update() override;
|
||||
virtual void UpdateGeometry() override;
|
||||
|
||||
protected:
|
||||
virtual void InstallPipe() override;
|
||||
|
||||
Geant::BoxSolid *m_BoxContent;
|
||||
|
||||
ULIB_DECLARE_PROPERTIES(BoxSolid)
|
||||
};
|
||||
|
||||
} // namespace Vtk
|
||||
|
||||
@@ -11,12 +11,12 @@
|
||||
#include <vtkPolyData.h>
|
||||
#include <vtkProperty.h>
|
||||
#include <vtkCylinderSource.h>
|
||||
#include "Vtk/Math/vtkDense.h"
|
||||
#include "Math/vtkDense.h"
|
||||
|
||||
namespace uLib {
|
||||
namespace Vtk {
|
||||
|
||||
EmitterPrimary::EmitterPrimary(Geant::EmitterPrimary &emitter)
|
||||
vtkEmitterPrimary::vtkEmitterPrimary(Geant::EmitterPrimary &emitter)
|
||||
: m_emitter(emitter), m_Actor(vtkActor::New()) {
|
||||
|
||||
vtkNew<vtkArrowSource> arrow;
|
||||
@@ -43,16 +43,16 @@ EmitterPrimary::EmitterPrimary(Geant::EmitterPrimary &emitter)
|
||||
|
||||
this->SetProp(m_Actor);
|
||||
|
||||
Object::connect(&m_emitter, &Object::Updated, this, &EmitterPrimary::contentUpdate);
|
||||
Object::connect(&m_emitter, &Object::Updated, this, &vtkEmitterPrimary::contentUpdate);
|
||||
this->contentUpdate();
|
||||
}
|
||||
|
||||
EmitterPrimary::~EmitterPrimary() {
|
||||
Object::disconnect(&m_emitter, &Object::Updated, this, &EmitterPrimary::contentUpdate);
|
||||
vtkEmitterPrimary::~vtkEmitterPrimary() {
|
||||
Object::disconnect(&m_emitter, &Object::Updated, this, &vtkEmitterPrimary::contentUpdate);
|
||||
m_Actor->Delete();
|
||||
}
|
||||
|
||||
void EmitterPrimary::contentUpdate() {
|
||||
void vtkEmitterPrimary::contentUpdate() {
|
||||
vtkMatrix4x4 *vmat = m_Actor->GetUserMatrix();
|
||||
if (!vmat) {
|
||||
vtkNew<vtkMatrix4x4> mat;
|
||||
@@ -63,10 +63,10 @@ void EmitterPrimary::contentUpdate() {
|
||||
Matrix4f transform = m_emitter.GetWorldMatrix();
|
||||
Matrix4fToVtk(transform, vmat);
|
||||
|
||||
Prop3D::Update();
|
||||
Puppet::Update();
|
||||
}
|
||||
|
||||
void EmitterPrimary::Update() {
|
||||
void vtkEmitterPrimary::Update() {
|
||||
vtkMatrix4x4 *vmat = m_Actor->GetUserMatrix();
|
||||
if (!vmat) return;
|
||||
|
||||
@@ -79,7 +79,7 @@ void EmitterPrimary::Update() {
|
||||
// vtkSkyPlaneEmitterPrimary
|
||||
|
||||
vtkSkyPlaneEmitterPrimary::vtkSkyPlaneEmitterPrimary(Geant::SkyPlaneEmitterPrimary &emitter)
|
||||
: EmitterPrimary(emitter), m_skyEmitter(emitter), m_PlaneSource(vtkPlaneSource::New()) {
|
||||
: vtkEmitterPrimary(emitter), m_skyEmitter(emitter), m_PlaneSource(vtkPlaneSource::New()) {
|
||||
|
||||
vtkNew<vtkAppendPolyData> append;
|
||||
// Base class constructor already set an arrow. We keep it as a directional indicator.
|
||||
@@ -107,27 +107,27 @@ void vtkSkyPlaneEmitterPrimary::contentUpdate() {
|
||||
m_PlaneSource->SetPoint2(-size(0)/2.0, size(1)/2.0, 0.0);
|
||||
m_PlaneSource->Update();
|
||||
|
||||
EmitterPrimary::contentUpdate();
|
||||
vtkEmitterPrimary::contentUpdate();
|
||||
}
|
||||
|
||||
// -------------------------------------------------------------------------- //
|
||||
// vtkQuadMeshEmitterPrimary
|
||||
|
||||
vtkQuadMeshEmitterPrimary::vtkQuadMeshEmitterPrimary(Geant::QuadMeshEmitterPrimary &emitter)
|
||||
: EmitterPrimary(emitter), m_meshEmitter(emitter) {
|
||||
: vtkEmitterPrimary(emitter), m_meshEmitter(emitter) {
|
||||
this->contentUpdate();
|
||||
}
|
||||
|
||||
void vtkQuadMeshEmitterPrimary::contentUpdate() {
|
||||
// For now stick with the arrow. In the future visualize the mesh if useful.
|
||||
EmitterPrimary::contentUpdate();
|
||||
vtkEmitterPrimary::contentUpdate();
|
||||
}
|
||||
|
||||
// -------------------------------------------------------------------------- //
|
||||
// vtkCylinderEmitterPrimary
|
||||
|
||||
vtkCylinderEmitterPrimary::vtkCylinderEmitterPrimary(Geant::CylinderEmitterPrimary &emitter)
|
||||
: EmitterPrimary(emitter), m_cylinderEmitter(emitter), m_CylinderSource(vtkCylinderSource::New()) {
|
||||
: vtkEmitterPrimary(emitter), m_cylinderEmitter(emitter), m_CylinderSource(vtkCylinderSource::New()) {
|
||||
|
||||
// vtkCylinderSource is along Y by default.
|
||||
// We will update its actual dimensions in contentUpdate().
|
||||
@@ -170,7 +170,7 @@ void vtkCylinderEmitterPrimary::contentUpdate() {
|
||||
m_CylinderSource->SetCenter(0, h/2.0, 0);
|
||||
m_CylinderSource->Update();
|
||||
|
||||
EmitterPrimary::contentUpdate();
|
||||
vtkEmitterPrimary::contentUpdate();
|
||||
}
|
||||
|
||||
} // namespace Vtk
|
||||
|
||||
@@ -14,10 +14,10 @@ class vtkCylinderSource;
|
||||
namespace uLib {
|
||||
namespace Vtk {
|
||||
|
||||
class EmitterPrimary : public Prop3D {
|
||||
class vtkEmitterPrimary : public Puppet {
|
||||
public:
|
||||
EmitterPrimary(Geant::EmitterPrimary &emitter);
|
||||
virtual ~EmitterPrimary();
|
||||
vtkEmitterPrimary(Geant::EmitterPrimary &emitter);
|
||||
virtual ~vtkEmitterPrimary();
|
||||
|
||||
virtual void contentUpdate();
|
||||
virtual void Update();
|
||||
@@ -27,7 +27,7 @@ protected:
|
||||
vtkActor *m_Actor;
|
||||
};
|
||||
|
||||
class vtkSkyPlaneEmitterPrimary : public EmitterPrimary {
|
||||
class vtkSkyPlaneEmitterPrimary : public vtkEmitterPrimary {
|
||||
public:
|
||||
vtkSkyPlaneEmitterPrimary(Geant::SkyPlaneEmitterPrimary &emitter);
|
||||
virtual ~vtkSkyPlaneEmitterPrimary();
|
||||
@@ -41,7 +41,7 @@ private:
|
||||
|
||||
|
||||
|
||||
class vtkQuadMeshEmitterPrimary : public EmitterPrimary {
|
||||
class vtkQuadMeshEmitterPrimary : public vtkEmitterPrimary {
|
||||
public:
|
||||
vtkQuadMeshEmitterPrimary(Geant::QuadMeshEmitterPrimary &emitter);
|
||||
virtual void contentUpdate();
|
||||
@@ -50,7 +50,7 @@ private:
|
||||
Geant::QuadMeshEmitterPrimary &m_meshEmitter;
|
||||
};
|
||||
|
||||
class vtkCylinderEmitterPrimary : public EmitterPrimary {
|
||||
class vtkCylinderEmitterPrimary : public vtkEmitterPrimary {
|
||||
public:
|
||||
vtkCylinderEmitterPrimary(Geant::CylinderEmitterPrimary &emitter);
|
||||
virtual ~vtkCylinderEmitterPrimary();
|
||||
|
||||
@@ -35,23 +35,23 @@
|
||||
namespace uLib {
|
||||
namespace Vtk {
|
||||
|
||||
GeantEvent::GeantEvent(Content *content)
|
||||
vtkGeantEvent::vtkGeantEvent(Content *content)
|
||||
: m_MuonPath(vtkActor::New()), m_Content(content) {
|
||||
this->InstallPipe();
|
||||
this->Update();
|
||||
}
|
||||
|
||||
GeantEvent::~GeantEvent() {
|
||||
vtkGeantEvent::~vtkGeantEvent() {
|
||||
m_MuonPath->Delete();
|
||||
}
|
||||
|
||||
vtkPolyData *GeantEvent::GetPolyData() const {
|
||||
vtkPolyData *vtkGeantEvent::GetPolyData() const {
|
||||
if (!m_MuonPath || !m_MuonPath->GetMapper())
|
||||
return NULL;
|
||||
return vtkPolyData::SafeDownCast(m_MuonPath->GetMapper()->GetInput());
|
||||
}
|
||||
|
||||
void GeantEvent::Update() {
|
||||
void vtkGeantEvent::Update() {
|
||||
if (!m_Content)
|
||||
return;
|
||||
|
||||
@@ -88,7 +88,7 @@ void GeantEvent::Update() {
|
||||
}
|
||||
}
|
||||
|
||||
void GeantEvent::InstallPipe() {
|
||||
void vtkGeantEvent::InstallPipe() {
|
||||
vtkSmartPointer<vtkPolyData> polyData = vtkSmartPointer<vtkPolyData>::New();
|
||||
vtkSmartPointer<vtkPolyDataMapper> mapper =
|
||||
vtkSmartPointer<vtkPolyDataMapper>::New();
|
||||
|
||||
@@ -34,16 +34,16 @@
|
||||
namespace uLib {
|
||||
namespace Vtk {
|
||||
|
||||
class GeantEvent : public Prop3D, public Polydata {
|
||||
class vtkGeantEvent : public Puppet, public Polydata {
|
||||
typedef Geant::GeantEvent Content;
|
||||
|
||||
public:
|
||||
GeantEvent(Content *content);
|
||||
~GeantEvent();
|
||||
vtkGeantEvent(Content *content);
|
||||
~vtkGeantEvent();
|
||||
|
||||
virtual class vtkPolyData *GetPolyData() const override;
|
||||
|
||||
virtual void Update() override;
|
||||
virtual void Update();
|
||||
|
||||
protected:
|
||||
virtual void InstallPipe();
|
||||
|
||||
@@ -32,20 +32,20 @@
|
||||
namespace uLib {
|
||||
namespace Vtk {
|
||||
|
||||
GeantScene::GeantScene(Geant::Scene *scene)
|
||||
: m_Scene(scene), m_WorldProp3D(nullptr) {
|
||||
vtkGeantScene::vtkGeantScene(Geant::Scene *scene)
|
||||
: m_Scene(scene), m_WorldPuppet(nullptr) {
|
||||
if (!m_Scene)
|
||||
return;
|
||||
|
||||
// 1. Create the world box wireframe prop3d
|
||||
uLib::ContainerBox *worldBox = m_Scene->GetWorldBox();
|
||||
// 1. Create the world box wireframe puppet
|
||||
ContainerBox *worldBox = m_Scene->GetWorldBox();
|
||||
if (worldBox) {
|
||||
m_WorldProp3D = new ContainerBox(worldBox);
|
||||
m_WorldProp3D->SetRepresentation(Prop3D::Wireframe);
|
||||
m_WorldProp3D->ShowScaleMeasures(true);
|
||||
m_WorldPuppet = new vtkContainerBox(worldBox);
|
||||
m_WorldPuppet->SetRepresentation(Puppet::Wireframe);
|
||||
m_WorldPuppet->ShowScaleMeasures(true);
|
||||
}
|
||||
|
||||
// 2. Create prop3ds for each non-world solid
|
||||
// 2. Create puppets for each non-world solid
|
||||
const Geant::Solid *world = m_Scene->GetWorld();
|
||||
const Vector<Geant::Solid *> &solids = m_Scene->GetSolids();
|
||||
|
||||
@@ -54,47 +54,47 @@ GeantScene::GeantScene(Geant::Scene *scene)
|
||||
if (solid == world)
|
||||
continue;
|
||||
|
||||
// Only create a prop3d if the solid has a valid G4VSolid
|
||||
// Only create a puppet if the solid has a valid G4VSolid
|
||||
if (solid->GetG4Solid()) {
|
||||
GeantSolid *vtkSolid = nullptr;
|
||||
vtkGeantSolid *vtkSolid = nullptr;
|
||||
if (auto *box = dynamic_cast<Geant::BoxSolid *>(solid)) {
|
||||
vtkSolid = new BoxSolid(box);
|
||||
vtkSolid = new vtkBoxSolid(box);
|
||||
} else if (auto *tess = dynamic_cast<Geant::TessellatedSolid *>(solid)) {
|
||||
vtkSolid = new TessellatedSolid(tess);
|
||||
vtkSolid = new vtkTessellatedSolid(tess);
|
||||
} else {
|
||||
vtkSolid = new GeantSolid(solid);
|
||||
vtkSolid = new vtkGeantSolid(solid);
|
||||
vtkSolid->Update();
|
||||
}
|
||||
|
||||
if (vtkSolid) {
|
||||
m_SolidProp3Ds.push_back(vtkSolid);
|
||||
m_SolidPuppets.push_back(vtkSolid);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
GeantScene::~GeantScene() {
|
||||
delete m_WorldProp3D;
|
||||
for (auto *p : m_SolidProp3Ds) {
|
||||
vtkGeantScene::~vtkGeantScene() {
|
||||
delete m_WorldPuppet;
|
||||
for (auto *p : m_SolidPuppets) {
|
||||
delete p;
|
||||
}
|
||||
}
|
||||
|
||||
void GeantScene::AddToViewer(Viewport &viewer) {
|
||||
if (m_WorldProp3D) {
|
||||
viewer.AddProp3D(*m_WorldProp3D);
|
||||
void vtkGeantScene::AddToViewer(Viewport &viewer) {
|
||||
if (m_WorldPuppet) {
|
||||
viewer.AddPuppet(*m_WorldPuppet);
|
||||
}
|
||||
for (auto *p : m_SolidProp3Ds) {
|
||||
viewer.AddProp3D(*p);
|
||||
for (auto *p : m_SolidPuppets) {
|
||||
viewer.AddPuppet(*p);
|
||||
}
|
||||
}
|
||||
|
||||
void GeantScene::RemoveFromViewer(Viewport &viewer) {
|
||||
if (m_WorldProp3D) {
|
||||
viewer.RemoveProp3D(*m_WorldProp3D);
|
||||
void vtkGeantScene::RemoveFromViewer(Viewport &viewer) {
|
||||
if (m_WorldPuppet) {
|
||||
viewer.RemovePuppet(*m_WorldPuppet);
|
||||
}
|
||||
for (auto *p : m_SolidProp3Ds) {
|
||||
viewer.RemoveProp3D(*p);
|
||||
for (auto *p : m_SolidPuppets) {
|
||||
viewer.RemovePuppet(*p);
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
@@ -36,14 +36,14 @@
|
||||
namespace uLib {
|
||||
namespace Vtk {
|
||||
|
||||
class GeantSolid;
|
||||
class vtkGeantSolid;
|
||||
|
||||
/**
|
||||
* @brief VTK Prop3D representing the entire Geant::Scene.
|
||||
* @brief VTK Puppet representing the entire Geant::Scene.
|
||||
*
|
||||
* When constructed, it creates child prop3ds for the world box (as a
|
||||
* ContainerBox wireframe) and for each non-world Solid in the scene
|
||||
* (as GeantSolid surfaces).
|
||||
* When constructed, it creates child puppets for the world box (as a
|
||||
* vtkContainerBox wireframe) and for each non-world Solid in the scene
|
||||
* (as vtkGeantSolid surfaces).
|
||||
*
|
||||
* Usage:
|
||||
* @code
|
||||
@@ -53,32 +53,32 @@ class GeantSolid;
|
||||
* scene.Initialize();
|
||||
*
|
||||
* Vtk::Viewer viewer;
|
||||
* Vtk::GeantScene vtkScene(&scene);
|
||||
* Vtk::vtkGeantScene vtkScene(&scene);
|
||||
* vtkScene.AddToViewer(viewer);
|
||||
* viewer.Start();
|
||||
* @endcode
|
||||
*/
|
||||
class GeantScene : public Object {
|
||||
class vtkGeantScene : public Object {
|
||||
public:
|
||||
GeantScene(Geant::Scene *scene);
|
||||
~GeantScene();
|
||||
vtkGeantScene(Geant::Scene *scene);
|
||||
~vtkGeantScene();
|
||||
|
||||
/// Add all prop3ds (world box + solids) to a viewer.
|
||||
/// Add all puppets (world box + solids) to a viewer.
|
||||
void AddToViewer(class Viewport &viewer);
|
||||
|
||||
/// Remove all prop3ds from viewport.
|
||||
/// Remove all puppets from viewport.
|
||||
void RemoveFromViewer(class Viewport &viewer);
|
||||
|
||||
/// Get the world box prop3d
|
||||
ContainerBox* GetWorldProp3D() const { return m_WorldProp3D; }
|
||||
/// Get the world box puppet
|
||||
vtkContainerBox* GetWorldPuppet() const { return m_WorldPuppet; }
|
||||
|
||||
/// Get the solid prop3ds
|
||||
const std::vector<GeantSolid*>& GetSolidProp3Ds() const { return m_SolidProp3Ds; }
|
||||
/// Get the solid puppets
|
||||
const std::vector<vtkGeantSolid*>& GetSolidPuppets() const { return m_SolidPuppets; }
|
||||
|
||||
private:
|
||||
Geant::Scene *m_Scene;
|
||||
ContainerBox *m_WorldProp3D;
|
||||
std::vector<GeantSolid*> m_SolidProp3Ds;
|
||||
vtkContainerBox *m_WorldPuppet;
|
||||
std::vector<vtkGeantSolid*> m_SolidPuppets;
|
||||
};
|
||||
|
||||
} // namespace Vtk
|
||||
|
||||
@@ -24,7 +24,6 @@
|
||||
//////////////////////////////////////////////////////////////////////////////*/
|
||||
|
||||
#include "vtkGeantSolid.h"
|
||||
#include "Core/Monitor.h"
|
||||
|
||||
#include <vtkActor.h>
|
||||
#include <vtkPolyData.h>
|
||||
@@ -35,7 +34,6 @@
|
||||
#include <vtkSmartPointer.h>
|
||||
#include <vtkTransform.h>
|
||||
#include <vtkMatrix4x4.h>
|
||||
#include "Vtk/Math/vtkDense.h"
|
||||
|
||||
#include <Geant4/G4VSolid.hh>
|
||||
#include <Geant4/G4Polyhedron.hh>
|
||||
@@ -44,48 +42,27 @@
|
||||
namespace uLib {
|
||||
namespace Vtk {
|
||||
|
||||
GeantSolid::GeantSolid(Content *content)
|
||||
vtkGeantSolid::vtkGeantSolid(Content *content)
|
||||
: m_SolidActor(vtkActor::New()), m_Content(content) {
|
||||
this->InstallPipe();
|
||||
if (m_Content) {
|
||||
m_UpdateConnection = Object::connect(m_Content, &uLib::Object::Updated, [this]() {
|
||||
std::cout << "GeantSolid: Signal Updated received." << std::endl;
|
||||
this->Update();
|
||||
});
|
||||
// Mandatory initial update to generate geometry and apply initial transform
|
||||
this->Update();
|
||||
}
|
||||
}
|
||||
|
||||
GeantSolid::~GeantSolid() {
|
||||
vtkGeantSolid::~vtkGeantSolid() {
|
||||
m_SolidActor->Delete();
|
||||
}
|
||||
|
||||
vtkPolyData *GeantSolid::GetPolyData() const {
|
||||
vtkPolyData *vtkGeantSolid::GetPolyData() const {
|
||||
if (!m_SolidActor || !m_SolidActor->GetMapper())
|
||||
return NULL;
|
||||
return vtkPolyData::SafeDownCast(m_SolidActor->GetMapper()->GetInput());
|
||||
}
|
||||
|
||||
void GeantSolid::Update() {
|
||||
ConnectionBlock blocker(m_UpdateConnection);
|
||||
void vtkGeantSolid::Update() {
|
||||
this->UpdateGeometry();
|
||||
this->UpdateTransform();
|
||||
this->Prop3D::Update();
|
||||
}
|
||||
|
||||
void GeantSolid::SyncFromVtk() {
|
||||
if (!m_Content) return;
|
||||
this->Prop3D::SyncFromVtk();
|
||||
|
||||
if (auto* proxy = vtkProp3D::SafeDownCast(this->GetProxyProp())) {
|
||||
if (vtkMatrix4x4* mat = proxy->GetUserMatrix()) {
|
||||
m_Content->SetTransform(VtkToMatrix4f(mat));
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
void GeantSolid::UpdateGeometry() {
|
||||
void vtkGeantSolid::UpdateGeometry() {
|
||||
if (!m_Content)
|
||||
return;
|
||||
|
||||
@@ -105,7 +82,6 @@ void GeantSolid::UpdateGeometry() {
|
||||
int nVertices = polyhedron->GetNoVertices();
|
||||
for (int i = 1; i <= nVertices; ++i) {
|
||||
G4Point3D vtx = polyhedron->GetVertex(i);
|
||||
|
||||
points->InsertNextPoint(vtx.x(), vtx.y(), vtx.z());
|
||||
}
|
||||
|
||||
@@ -133,42 +109,39 @@ void GeantSolid::UpdateGeometry() {
|
||||
}
|
||||
}
|
||||
|
||||
void GeantSolid::UpdateTransform() {
|
||||
void vtkGeantSolid::UpdateTransform() {
|
||||
if (!m_Content || !m_SolidActor)
|
||||
return;
|
||||
|
||||
// Apply the Geant4 transform (position/rotation/scaling)
|
||||
G4ThreeVector pos = m_Content->GetPosition();
|
||||
const G4RotationMatrix *rot = m_Content->GetRotation();
|
||||
G4ThreeVector scale = m_Content->GetScaling();
|
||||
// Apply the Geant4 transform (position/rotation) if placed
|
||||
if (m_Content->GetPhysical()) {
|
||||
auto *phys = m_Content->GetPhysical();
|
||||
G4ThreeVector pos = phys->GetTranslation();
|
||||
const G4RotationMatrix *rot = phys->GetRotation();
|
||||
|
||||
std::cout << "GeantSolid: Updating transform. Pos: " << pos.x() << " " << pos.y() << " " << pos.z()
|
||||
<< " Scale: " << scale.x() << " " << scale.y() << " " << scale.z() << std::endl;
|
||||
vtkSmartPointer<vtkTransform> transform = vtkSmartPointer<vtkTransform>::New();
|
||||
transform->Identity();
|
||||
transform->Translate(pos.x(), pos.y(), pos.z());
|
||||
|
||||
vtkNew<vtkTransform> transform;
|
||||
transform->Identity();
|
||||
transform->Translate(pos.x(), pos.y(), pos.z());
|
||||
|
||||
if (rot) {
|
||||
G4RotationMatrix invRot = *rot;
|
||||
double elements[16] = {
|
||||
invRot.xx(), invRot.xy(), invRot.xz(), 0,
|
||||
invRot.yx(), invRot.yy(), invRot.yz(), 0,
|
||||
invRot.zx(), invRot.zy(), invRot.zz(), 0,
|
||||
0, 0, 0, 1
|
||||
};
|
||||
vtkNew<vtkMatrix4x4> mat;
|
||||
mat->DeepCopy(elements);
|
||||
transform->Concatenate(mat);
|
||||
}
|
||||
|
||||
transform->Scale(scale.x(), scale.y(), scale.z());
|
||||
if (rot) {
|
||||
// G4RotationMatrix stores the inverse of the rotation for placement
|
||||
G4RotationMatrix invRot = rot->inverse();
|
||||
double elements[16] = {
|
||||
invRot.xx(), invRot.xy(), invRot.xz(), 0,
|
||||
invRot.yx(), invRot.yy(), invRot.yz(), 0,
|
||||
invRot.zx(), invRot.zy(), invRot.zz(), 0,
|
||||
0, 0, 0, 1
|
||||
};
|
||||
vtkSmartPointer<vtkMatrix4x4> mat = vtkSmartPointer<vtkMatrix4x4>::New();
|
||||
mat->DeepCopy(elements);
|
||||
transform->Concatenate(mat);
|
||||
}
|
||||
|
||||
m_SolidActor->SetUserTransform(transform);
|
||||
m_SolidActor->Modified();
|
||||
}
|
||||
}
|
||||
|
||||
void GeantSolid::InstallPipe() {
|
||||
void vtkGeantSolid::InstallPipe() {
|
||||
vtkSmartPointer<vtkPolyData> polyData = vtkSmartPointer<vtkPolyData>::New();
|
||||
vtkSmartPointer<vtkPolyDataMapper> mapper =
|
||||
vtkSmartPointer<vtkPolyDataMapper>::New();
|
||||
|
||||
@@ -36,35 +36,31 @@ namespace uLib {
|
||||
namespace Vtk {
|
||||
|
||||
/**
|
||||
* @brief VTK Prop3D for visualizing a Geant::Solid.
|
||||
* @brief VTK Puppet for visualizing a Geant::Solid.
|
||||
*
|
||||
* Renders the G4VSolid geometry as a tessellated polydata surface.
|
||||
* Works with BoxSolid, TessellatedSolid, or any Solid that provides
|
||||
* a valid G4VSolid via GetG4Solid().
|
||||
*/
|
||||
class GeantSolid : public Prop3D, public Polydata {
|
||||
class vtkGeantSolid : public Puppet, public Polydata {
|
||||
typedef Geant::Solid Content;
|
||||
|
||||
public:
|
||||
GeantSolid(Content *content);
|
||||
~GeantSolid();
|
||||
vtkGeantSolid(Content *content);
|
||||
~vtkGeantSolid();
|
||||
|
||||
virtual class vtkPolyData *GetPolyData() const override;
|
||||
|
||||
virtual void Update() override;
|
||||
virtual void SyncFromVtk() override;
|
||||
|
||||
virtual void UpdateGeometry();
|
||||
virtual void UpdateTransform();
|
||||
|
||||
virtual uLib::Object *GetContent() const override { return (uLib::Object*)m_Content; }
|
||||
|
||||
protected:
|
||||
virtual void InstallPipe();
|
||||
|
||||
Content *m_Content;
|
||||
|
||||
vtkActor *m_SolidActor;
|
||||
uLib::Connection m_UpdateConnection;
|
||||
Content *m_Content;
|
||||
};
|
||||
|
||||
} // namespace Vtk
|
||||
|
||||
@@ -18,22 +18,22 @@
|
||||
namespace uLib {
|
||||
namespace Vtk {
|
||||
|
||||
TessellatedSolid::TessellatedSolid(Geant::TessellatedSolid *content)
|
||||
: GeantSolid(content), m_TessContent(content) {
|
||||
vtkTessellatedSolid::vtkTessellatedSolid(Geant::TessellatedSolid *content)
|
||||
: vtkGeantSolid(content), m_TessContent(content) {
|
||||
this->Update();
|
||||
}
|
||||
|
||||
TessellatedSolid::~TessellatedSolid() {}
|
||||
vtkTessellatedSolid::~vtkTessellatedSolid() {}
|
||||
|
||||
void TessellatedSolid::Update() {
|
||||
void vtkTessellatedSolid::Update() {
|
||||
this->UpdateGeometry();
|
||||
this->UpdateTransform();
|
||||
}
|
||||
|
||||
void TessellatedSolid::UpdateGeometry() {
|
||||
void vtkTessellatedSolid::UpdateGeometry() {
|
||||
if (!m_TessContent || m_TessContent->GetMesh().Points().empty()) {
|
||||
// Fallback to base tessellation if no model mesh
|
||||
GeantSolid::UpdateGeometry();
|
||||
vtkGeantSolid::UpdateGeometry();
|
||||
return;
|
||||
}
|
||||
|
||||
|
||||
@@ -18,12 +18,12 @@ namespace uLib {
|
||||
namespace Vtk {
|
||||
|
||||
/**
|
||||
* @brief VTK Prop3D for visualizing a Geant::TessellatedSolid.
|
||||
* @brief VTK Puppet for visualizing a Geant::TessellatedSolid.
|
||||
*/
|
||||
class TessellatedSolid : public GeantSolid {
|
||||
class vtkTessellatedSolid : public vtkGeantSolid {
|
||||
public:
|
||||
TessellatedSolid(Geant::TessellatedSolid *content);
|
||||
virtual ~TessellatedSolid();
|
||||
vtkTessellatedSolid(Geant::TessellatedSolid *content);
|
||||
virtual ~vtkTessellatedSolid();
|
||||
|
||||
virtual void Update() override;
|
||||
virtual void UpdateGeometry() override;
|
||||
|
||||
@@ -46,7 +46,7 @@ class vtkWidgetCallback : public vtkCommand {
|
||||
public:
|
||||
static vtkWidgetCallback *New() { return new vtkWidgetCallback; }
|
||||
|
||||
void SetTracer(Vtk::VoxRaytracerRepresentation *parent) {
|
||||
void SetTracer(Vtk::vtkVoxRaytracerRepresentation *parent) {
|
||||
this->vtk_raytr = parent;
|
||||
}
|
||||
|
||||
@@ -74,7 +74,7 @@ private:
|
||||
vtkWidgetCallback() : vtk_raytr(NULL), muon(NULL), annotation(NULL) {}
|
||||
|
||||
uLib::VoxRaytracer *raytracer;
|
||||
Vtk::VoxRaytracerRepresentation *vtk_raytr;
|
||||
Vtk::vtkVoxRaytracerRepresentation *vtk_raytr;
|
||||
uLib::MuonScatter *muon;
|
||||
vtkCornerAnnotation *annotation;
|
||||
};
|
||||
@@ -87,14 +87,14 @@ BOOST_AUTO_TEST_CASE(vtkVoxRaytracerRepresentationTest) {
|
||||
muon.LineOut().origin << 6, -4, 6, 1;
|
||||
muon.LineOut().direction << 1, -1, 1, 0;
|
||||
|
||||
Vtk::MuonScatter v_muon(muon);
|
||||
Vtk::vtkMuonScatter v_muon(muon);
|
||||
|
||||
// structured grid //
|
||||
StructuredGrid grid(Vector3i(12, 10, 12));
|
||||
grid.SetSpacing(Vector3f(1, 1, 1));
|
||||
grid.SetPosition(Vector3f(0, 0, 0));
|
||||
|
||||
Vtk::StructuredGrid v_grid(&grid);
|
||||
Vtk::vtkStructuredGrid v_grid(grid);
|
||||
|
||||
// voxraytracer //
|
||||
VoxRaytracer rt(grid);
|
||||
@@ -102,7 +102,7 @@ BOOST_AUTO_TEST_CASE(vtkVoxRaytracerRepresentationTest) {
|
||||
rt.GetEntryPoint(muon.LineIn(), pt);
|
||||
std::cout << pt.transpose() << "\n";
|
||||
|
||||
Vtk::VoxRaytracerRepresentation v_rt(rt);
|
||||
Vtk::vtkVoxRaytracerRepresentation v_rt(rt);
|
||||
v_rt.SetMuon(muon);
|
||||
v_rt.SetRayColor(Vector4f(1, 0, 0, 1));
|
||||
|
||||
@@ -110,12 +110,12 @@ BOOST_AUTO_TEST_CASE(vtkVoxRaytracerRepresentationTest) {
|
||||
// renderer //
|
||||
Vtk::Viewer viewer;
|
||||
|
||||
viewer.AddProp3D(v_grid);
|
||||
viewer.AddProp3D(v_rt);
|
||||
viewer.AddProp3D(v_muon);
|
||||
viewer.AddPuppet(v_grid);
|
||||
viewer.AddPuppet(v_rt);
|
||||
viewer.AddPuppet(v_muon);
|
||||
|
||||
// Select grid to show handler widget
|
||||
viewer.SelectProp3D(&v_grid);
|
||||
viewer.SelectPuppet(&v_grid);
|
||||
|
||||
viewer.Start();
|
||||
}
|
||||
|
||||
@@ -30,7 +30,7 @@
|
||||
|
||||
|
||||
|
||||
#include "Vtk/Math/vtkDense.h"
|
||||
#include "Math/Dense.h"
|
||||
|
||||
#include "uLibVtkInterface.h"
|
||||
#include "Detectors/MuonScatter.h"
|
||||
@@ -39,11 +39,11 @@ class vtkRenderWindowInteractor;
|
||||
|
||||
namespace uLib {
|
||||
|
||||
class MuonContainerScattering : public Abstract::uLibVtkPolydata {
|
||||
typedef uLib::MuonScatter Content;
|
||||
class vtkMuonContainerScattering : public Abstract::uLibVtkPolydata {
|
||||
typedef MuonScatter Content;
|
||||
public:
|
||||
MuonContainerScattering(const MuonScatter &content);
|
||||
~MuonScatter();
|
||||
vtkMuonContainerScattering(const MuonScatter &content);
|
||||
~vtkMuonScatter();
|
||||
|
||||
Content& GetContent();
|
||||
|
||||
|
||||
@@ -41,7 +41,7 @@ namespace Vtk {
|
||||
////// VOX RAYTRACER REPRESENTATION ///////////////////////////////////////////
|
||||
////////////////////////////////////////////////////////////////////////////////
|
||||
|
||||
VoxRaytracerRepresentation::VoxRaytracerRepresentation(Content &content)
|
||||
vtkVoxRaytracerRepresentation::vtkVoxRaytracerRepresentation(Content &content)
|
||||
: m_Content(&content),
|
||||
m_Sphere1(vtkSphereSource::New()), m_Sphere2(vtkSphereSource::New()),
|
||||
m_Line1(vtkLineSource::New()), m_Line2(vtkLineSource::New()),
|
||||
@@ -51,7 +51,6 @@ VoxRaytracerRepresentation::VoxRaytracerRepresentation(Content &content)
|
||||
m_RayRepresentationActor(vtkActor::New()),
|
||||
m_Transform(vtkTransform::New()),
|
||||
m_Asm(vtkAssembly::New()),
|
||||
m_Muon(nullptr),
|
||||
m_HasMuon(false), m_HasPoca(false) {
|
||||
default_radius = content.GetImage()->GetSpacing()(0) / 4;
|
||||
m_Sphere1->SetRadius(default_radius);
|
||||
@@ -62,49 +61,48 @@ VoxRaytracerRepresentation::VoxRaytracerRepresentation(Content &content)
|
||||
InstallPipe();
|
||||
|
||||
if (m_Content && m_Content->GetImage()) {
|
||||
Object::connect(m_Content->GetImage(), &StructuredGrid::Updated, this, &VoxRaytracerRepresentation::imageUpdate);
|
||||
Object::connect(m_Content->GetImage(), &StructuredGrid::Updated, this, &vtkVoxRaytracerRepresentation::imageUpdate);
|
||||
}
|
||||
}
|
||||
|
||||
VoxRaytracerRepresentation::~VoxRaytracerRepresentation() {
|
||||
vtkVoxRaytracerRepresentation::~vtkVoxRaytracerRepresentation() {
|
||||
m_RayLine->Delete();
|
||||
m_RayLineActor->Delete();
|
||||
m_RayRepresentationActor->Delete();
|
||||
m_Transform->Delete();
|
||||
if (m_Muon) delete m_Muon;
|
||||
}
|
||||
|
||||
uLib::VoxRaytracer *VoxRaytracerRepresentation::GetRaytracerAlgorithm() {
|
||||
uLib::VoxRaytracer *vtkVoxRaytracerRepresentation::GetRaytracerAlgorithm() {
|
||||
return m_Content;
|
||||
}
|
||||
|
||||
void VoxRaytracerRepresentation::Update() {
|
||||
void vtkVoxRaytracerRepresentation::Update() {
|
||||
this->imageUpdate();
|
||||
}
|
||||
|
||||
void VoxRaytracerRepresentation::imageUpdate() {
|
||||
if (m_HasMuon && m_Muon) {
|
||||
void vtkVoxRaytracerRepresentation::imageUpdate() {
|
||||
if (m_HasMuon) {
|
||||
if (m_HasPoca) {
|
||||
this->SetMuon(m_Muon->GetModel(), m_Poca);
|
||||
this->SetMuon(m_Muon, m_Poca);
|
||||
} else {
|
||||
this->SetMuon(*m_Muon);
|
||||
this->SetMuon(m_Muon);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
vtkPolyData *VoxRaytracerRepresentation::GetPolyData() const {
|
||||
vtkPolyData *vtkVoxRaytracerRepresentation::GetPolyData() const {
|
||||
std::cout << "get Raytracer polydata\n";
|
||||
m_SelectedElement->Update();
|
||||
return m_SelectedElement->GetOutput();
|
||||
}
|
||||
|
||||
void VoxRaytracerRepresentation::SetRepresentationElements(
|
||||
VoxRaytracerRepresentation::RepresentationElements el) {
|
||||
void vtkVoxRaytracerRepresentation::SetRepresentationElements(
|
||||
vtkVoxRaytracerRepresentation::RepresentationElements el) {
|
||||
switch (el) {
|
||||
case Vtk::VoxRaytracerRepresentation::RayElements:
|
||||
case Vtk::vtkVoxRaytracerRepresentation::RayElements:
|
||||
m_SelectedElement = m_RayLine;
|
||||
break;
|
||||
case Vtk::VoxRaytracerRepresentation::VoxelsElements:
|
||||
case Vtk::vtkVoxRaytracerRepresentation::VoxelsElements:
|
||||
m_SelectedElement = m_RayRepresentation;
|
||||
break;
|
||||
default:
|
||||
@@ -113,9 +111,8 @@ void VoxRaytracerRepresentation::SetRepresentationElements(
|
||||
}
|
||||
}
|
||||
|
||||
void VoxRaytracerRepresentation::SetMuon(uLib::MuonScatter &muon) {
|
||||
if (m_Muon) delete m_Muon;
|
||||
m_Muon = new MuonScatter(muon);
|
||||
void vtkVoxRaytracerRepresentation::SetMuon(MuonScatter &muon) {
|
||||
m_Muon = muon;
|
||||
m_HasMuon = true;
|
||||
m_HasPoca = false;
|
||||
|
||||
@@ -176,9 +173,8 @@ void VoxRaytracerRepresentation::SetMuon(uLib::MuonScatter &muon) {
|
||||
this->SetRay(pt1, pt2);
|
||||
}
|
||||
|
||||
void VoxRaytracerRepresentation::SetMuon(uLib::MuonScatter &muon, HPoint3f poca) {
|
||||
if (m_Muon) delete m_Muon;
|
||||
m_Muon = new MuonScatter(muon);
|
||||
void vtkVoxRaytracerRepresentation::SetMuon(MuonScatter &muon, HPoint3f poca) {
|
||||
m_Muon = muon;
|
||||
m_Poca = poca;
|
||||
m_HasMuon = true;
|
||||
m_HasPoca = true;
|
||||
@@ -251,27 +247,27 @@ void VoxRaytracerRepresentation::SetMuon(uLib::MuonScatter &muon, HPoint3f poca)
|
||||
this->SetRay(pt1, pt2);
|
||||
}
|
||||
|
||||
void VoxRaytracerRepresentation::SetMuon(MuonScatter &muon) {
|
||||
void vtkVoxRaytracerRepresentation::SetMuon(vtkMuonScatter &muon) {
|
||||
HPoint3f poca = muon.GetPocaPoint();
|
||||
uLib::MuonScatter &mu = muon.GetModel();
|
||||
MuonScatter &mu = muon.GetContent();
|
||||
this->SetMuon(mu, poca);
|
||||
}
|
||||
|
||||
VoxRaytracer::RayData VoxRaytracerRepresentation::GetRay() { return m_Ray; }
|
||||
VoxRaytracer::RayData vtkVoxRaytracerRepresentation::GetRay() { return m_Ray; }
|
||||
|
||||
void VoxRaytracerRepresentation::SetRay(HPoint3f in, HPoint3f out) {
|
||||
void vtkVoxRaytracerRepresentation::SetRay(HPoint3f in, HPoint3f out) {
|
||||
m_Ray = m_Content->TraceBetweenPoints(in, out);
|
||||
this->SetRay(&m_Ray);
|
||||
}
|
||||
|
||||
void VoxRaytracerRepresentation::SetRay(HPoint3f in, HPoint3f mid,
|
||||
void vtkVoxRaytracerRepresentation::SetRay(HPoint3f in, HPoint3f mid,
|
||||
HPoint3f out) {
|
||||
m_Ray = m_Content->TraceBetweenPoints(in, mid);
|
||||
m_Ray.AppendRay(m_Content->TraceBetweenPoints(mid, out));
|
||||
this->SetRay(&m_Ray);
|
||||
}
|
||||
|
||||
void VoxRaytracerRepresentation::SetRay(VoxRaytracer::RayData *ray) {
|
||||
void vtkVoxRaytracerRepresentation::SetRay(VoxRaytracer::RayData *ray) {
|
||||
vtkAppendPolyData *appender = m_RayRepresentation;
|
||||
appender->RemoveAllInputs();
|
||||
|
||||
@@ -290,15 +286,15 @@ void VoxRaytracerRepresentation::SetRay(VoxRaytracer::RayData *ray) {
|
||||
appender->Modified();
|
||||
}
|
||||
|
||||
void VoxRaytracerRepresentation::SetVoxelsColor(Vector4f rgba) {
|
||||
void vtkVoxRaytracerRepresentation::SetVoxelsColor(Vector4f rgba) {
|
||||
this->SetColor(m_RayRepresentationActor, rgba);
|
||||
}
|
||||
|
||||
void VoxRaytracerRepresentation::SetRayColor(Vector4f rgba) {
|
||||
void vtkVoxRaytracerRepresentation::SetRayColor(Vector4f rgba) {
|
||||
this->SetColor(m_RayLineActor, rgba);
|
||||
}
|
||||
|
||||
void VoxRaytracerRepresentation::SetColor(vtkActor *actor, Vector4f rgba) {
|
||||
void vtkVoxRaytracerRepresentation::SetColor(vtkActor *actor, Vector4f rgba) {
|
||||
if (!actor)
|
||||
return;
|
||||
vtkProperty *pr = actor->GetProperty();
|
||||
@@ -307,7 +303,7 @@ void VoxRaytracerRepresentation::SetColor(vtkActor *actor, Vector4f rgba) {
|
||||
pr->SetDiffuse(1);
|
||||
}
|
||||
|
||||
void VoxRaytracerRepresentation::InstallPipe() {
|
||||
void vtkVoxRaytracerRepresentation::InstallPipe() {
|
||||
|
||||
vtkSmartPointer<vtkAppendPolyData> append =
|
||||
vtkSmartPointer<vtkAppendPolyData>::New();
|
||||
|
||||
@@ -44,13 +44,13 @@
|
||||
#include <vtkTransform.h>
|
||||
#include <vtkTransformPolyDataFilter.h>
|
||||
|
||||
#include "Vtk/Math/vtkDense.h"
|
||||
#include "Math/Dense.h"
|
||||
|
||||
#include "Vtk/uLibVtkInterface.h"
|
||||
|
||||
#include "Vtk/HEP/Detectors/vtkMuonScatter.h"
|
||||
#include "HEP/Detectors/MuonScatter.h"
|
||||
|
||||
#include "Vtk/Math/vtkStructuredGrid.h"
|
||||
#include "Math/StructuredGrid.h"
|
||||
#include "Math/VoxRaytracer.h"
|
||||
|
||||
class vtkActor;
|
||||
@@ -58,12 +58,12 @@ class vtkActor;
|
||||
namespace uLib {
|
||||
namespace Vtk {
|
||||
|
||||
class VoxRaytracerRepresentation : public Prop3D {
|
||||
class vtkVoxRaytracerRepresentation : public Puppet {
|
||||
typedef VoxRaytracer Content;
|
||||
|
||||
public:
|
||||
VoxRaytracerRepresentation(Content &content);
|
||||
~VoxRaytracerRepresentation();
|
||||
vtkVoxRaytracerRepresentation(Content &content);
|
||||
~vtkVoxRaytracerRepresentation();
|
||||
|
||||
uLib::VoxRaytracer *GetRaytracerAlgorithm();
|
||||
|
||||
@@ -76,13 +76,13 @@ public:
|
||||
enum RepresentationElements { RayElements, VoxelsElements };
|
||||
void SetRepresentationElements(enum RepresentationElements el);
|
||||
|
||||
void SetMuon(uLib::MuonScatter &muon);
|
||||
void SetMuon(MuonScatter &muon);
|
||||
|
||||
void SetMuon(uLib::MuonScatter &muon, HPoint3f poca);
|
||||
void SetMuon(MuonScatter &muon, HPoint3f poca);
|
||||
|
||||
void SetMuon(class MuonEvent &muon);
|
||||
void SetMuon(class vtkMuonEvent &muon);
|
||||
|
||||
void SetMuon(class MuonScatter &muon);
|
||||
void SetMuon(class vtkMuonScatter &muon);
|
||||
|
||||
VoxRaytracer::RayData GetRay();
|
||||
|
||||
@@ -101,7 +101,7 @@ private:
|
||||
void SetColor(vtkActor *actor, Vector4f rgba);
|
||||
|
||||
VoxRaytracer *m_Content;
|
||||
MuonScatter *m_Muon;
|
||||
MuonScatter m_Muon;
|
||||
HPoint3f m_Poca;
|
||||
bool m_HasMuon;
|
||||
bool m_HasPoca;
|
||||
|
||||
@@ -9,9 +9,9 @@
|
||||
|
||||
//////////////////////////////////////////////////////////////////////////////*/
|
||||
|
||||
#include "Vtk/Math/vtkAssembly.h"
|
||||
#include "Vtk/Math/vtkContainerBox.h"
|
||||
#include "Vtk/Math/vtkCylinder.h"
|
||||
#include "Math/Assembly.h"
|
||||
#include "Math/ContainerBox.h"
|
||||
#include "Math/Cylinder.h"
|
||||
#include "Vtk/Math/vtkAssembly.h"
|
||||
#include "Vtk/vtkObjectsContext.h"
|
||||
#include "Vtk/uLibVtkViewer.h"
|
||||
@@ -50,18 +50,18 @@ int main(int argc, char **argv) {
|
||||
// ---- 3. Apply a group transform ----
|
||||
assembly.SetPosition(Vector3f(1_m, 1_m, 0));
|
||||
|
||||
// ---- 5. Visualize (create prop3ds to set properties) ----
|
||||
// ---- 5. Visualize (create puppets to set properties) ----
|
||||
Vtk::Assembly vtkAsm(&assembly);
|
||||
|
||||
Vtk::Viewer viewer;
|
||||
vtkAsm.AddToViewer(viewer); // This triggers prop3d creation via ConnectRenderer which eventually calls Prop3D::GetProp
|
||||
vtkAsm.AddToViewer(viewer); // This triggers puppet creation via ConnectRenderer which eventually calls Puppet::GetProp
|
||||
|
||||
// Explicitly update to ensure prop3ds exist and are added to assemblies
|
||||
// Explicitly update to ensure puppets exist and are added to assemblies
|
||||
vtkAsm.Update();
|
||||
|
||||
// Use the child context to find child prop3ds and set colors
|
||||
// Use the child context to find child puppets and set colors
|
||||
if (auto* childCtx = vtkAsm.GetChildrenContext()) {
|
||||
auto setProps = [](Vtk::Prop3D* p, float r, float g, float b) {
|
||||
auto setProps = [](Vtk::Puppet* p, float r, float g, float b) {
|
||||
if (!p) return;
|
||||
vtkPropCollection* props = p->GetProps();
|
||||
props->InitTraversal();
|
||||
@@ -74,12 +74,12 @@ int main(int argc, char **argv) {
|
||||
}
|
||||
};
|
||||
|
||||
setProps(childCtx->GetProp3D(&box1), 1.0, 0.0, 0.0); // Red
|
||||
setProps(childCtx->GetProp3D(&box2), 0.0, 1.0, 0.0); // Green
|
||||
setProps(childCtx->GetProp3D(&cyl), 0.0, 0.0, 1.0); // Blue
|
||||
setProps(childCtx->GetPuppet(&box1), 1.0, 0.0, 0.0); // Red
|
||||
setProps(childCtx->GetPuppet(&box2), 0.0, 1.0, 0.0); // Green
|
||||
setProps(childCtx->GetPuppet(&cyl), 0.0, 0.0, 1.0); // Blue
|
||||
}
|
||||
|
||||
std::cout << "Prop3Ds in viewport: " << viewer.getProp3Ds().size() << " (Expected 4: 1 assembly + 3 children)" << std::endl;
|
||||
std::cout << "Puppets in viewport: " << viewer.getPuppets().size() << " (Expected 4: 1 assembly + 3 children)" << std::endl;
|
||||
|
||||
// ---- 4. Query the bounding box for terminal output ----
|
||||
Vector3f bbMin, bbMax;
|
||||
|
||||
@@ -25,6 +25,7 @@
|
||||
|
||||
#include "Vtk/uLibVtkViewer.h"
|
||||
|
||||
#include "Math/ContainerBox.h"
|
||||
#include "Math/Units.h"
|
||||
#include "Vtk/Math/vtkContainerBox.h"
|
||||
|
||||
@@ -35,28 +36,21 @@ using namespace uLib;
|
||||
int main() {
|
||||
BEGIN_TESTING(vtk ContainerBox Test);
|
||||
|
||||
{
|
||||
ContainerBox* box = new ContainerBox();
|
||||
box->Scale(Vector3f(1_m, 2_m, 1_m));
|
||||
box->SetPosition(Vector3f(0, 0, 0));
|
||||
|
||||
Vtk::ContainerBox v_box(box);
|
||||
v_box.Update();
|
||||
|
||||
v_box.SetRepresentation(Vtk::Prop3D::Surface);
|
||||
v_box.SetOpacity(0.5);
|
||||
v_box.SetSelectable(true);
|
||||
}
|
||||
|
||||
Vtk::ContainerBox v_box;
|
||||
v_box.findOrAddSignal(&Object::Updated)->connect([&v_box]() {
|
||||
std::cout << "box updated: "
|
||||
<< v_box.get()->GetWorldPoint(HPoint3f(1, 1, 1)) << std::endl;
|
||||
ContainerBox box;
|
||||
box.Scale(Vector3f(1_m,5_m,1_m));
|
||||
box.SetPosition(Vector3f(0,1_m,0));
|
||||
Vtk::vtkContainerBox v_box(&box);
|
||||
v_box.SetRepresentation(Vtk::Puppet::Surface);
|
||||
v_box.SetOpacity(0.5);
|
||||
v_box.SetSelectable(true);
|
||||
|
||||
box.findOrAddSignal(&ContainerBox::Updated)->connect([&box](){
|
||||
std::cout << "box updated: " << box.GetWorldPoint(HPoint3f(1,1,1)) << std::endl;
|
||||
});
|
||||
|
||||
if (std::getenv("CTEST_PROJECT_NAME") == nullptr) {
|
||||
Vtk::Viewer v_viewer;
|
||||
v_viewer.AddProp3D(v_box);
|
||||
v_viewer.AddPuppet(v_box);
|
||||
v_viewer.Start();
|
||||
}
|
||||
|
||||
|
||||
@@ -6,7 +6,7 @@
|
||||
//////////////////////////////////////////////////////////////////////////////*/
|
||||
|
||||
#include "Vtk/uLibVtkViewer.h"
|
||||
#include "Vtk/Math/vtkContainerBox.h"
|
||||
#include "Math/ContainerBox.h"
|
||||
#include "Math/Units.h"
|
||||
#include "Vtk/Math/vtkContainerBox.h"
|
||||
#include <iostream>
|
||||
@@ -15,12 +15,12 @@ using namespace uLib;
|
||||
|
||||
int main() {
|
||||
std::cout << "Creating ContainerBox..." << std::endl;
|
||||
ContainerBox* box = new ContainerBox(Vector3f(1.0, 1.0, 1.0)); // 1x1x1 unit box
|
||||
box->SetInstanceName("MyTestBox");
|
||||
ContainerBox box(Vector3f(1.0, 1.0, 1.0)); // 1x1x1 unit box
|
||||
box.SetInstanceName("MyTestBox");
|
||||
|
||||
std::cout << "Creating VTK representation..." << std::endl;
|
||||
Vtk::ContainerBox v_box(box);
|
||||
v_box.SetRepresentation(Vtk::Prop3D::Wireframe);
|
||||
Vtk::vtkContainerBox v_box(&box);
|
||||
v_box.SetRepresentation(Vtk::Puppet::Wireframe);
|
||||
v_box.SetColor(1.0, 0.0, 0.0); // Red
|
||||
|
||||
// // 1. Initial Visualization setup (handled by Viewer)
|
||||
@@ -38,7 +38,7 @@ int main() {
|
||||
|
||||
std::cout << "Starting viewer (close window to exit)..." << std::endl;
|
||||
Vtk::Viewer viewer;
|
||||
viewer.AddProp3D(v_box);
|
||||
viewer.AddPuppet(v_box);
|
||||
viewer.Start();
|
||||
|
||||
return 0;
|
||||
|
||||
@@ -24,6 +24,7 @@
|
||||
//////////////////////////////////////////////////////////////////////////////*/
|
||||
|
||||
#include "Vtk/Math/vtkQuadMesh.h"
|
||||
#include "Math/QuadMesh.h"
|
||||
#include "Vtk/uLibVtkViewer.h"
|
||||
|
||||
#define BOOST_TEST_MODULE VtkQuadMeshTest
|
||||
@@ -38,10 +39,10 @@ BOOST_AUTO_TEST_CASE(vtkQuadMeshConstruction) {
|
||||
mesh.AddPoint(Vector3f(1, 0, 0));
|
||||
mesh.AddPoint(Vector3f(1, 1, 0));
|
||||
mesh.AddPoint(Vector3f(0, 1, 0));
|
||||
|
||||
|
||||
mesh.AddQuad(Vector4i(0, 1, 2, 3));
|
||||
|
||||
Vtk::QuadMesh v_mesh(&mesh);
|
||||
Vtk::vtkQuadMesh v_mesh(mesh);
|
||||
|
||||
Object::connect(&mesh, &QuadMesh::Updated, [&mesh]() {
|
||||
Vector3f points[4];
|
||||
@@ -49,15 +50,15 @@ BOOST_AUTO_TEST_CASE(vtkQuadMeshConstruction) {
|
||||
points[1] = mesh.GetPoint(1);
|
||||
points[2] = mesh.GetPoint(2);
|
||||
points[3] = mesh.GetPoint(3);
|
||||
std::cout << "mesh updated: " << points[0] << " " << points[1] << " "
|
||||
<< points[2] << " " << points[3] << std::endl;
|
||||
std::cout << "mesh updated: " << points[0] << " " << points[1]
|
||||
<< " " << points[2] << " " << points[3] << std::endl;
|
||||
});
|
||||
|
||||
v_mesh.Update();
|
||||
|
||||
if (std::getenv("CTEST_PROJECT_NAME") == nullptr) {
|
||||
Vtk::Viewer viewer;
|
||||
viewer.AddProp3D(v_mesh);
|
||||
viewer.AddPuppet(v_mesh);
|
||||
viewer.Start();
|
||||
}
|
||||
|
||||
|
||||
@@ -24,6 +24,7 @@
|
||||
//////////////////////////////////////////////////////////////////////////////*/
|
||||
|
||||
#include "Vtk/Math/vtkStructuredGrid.h"
|
||||
#include "Math/StructuredGrid.h"
|
||||
#include "Vtk/uLibVtkViewer.h"
|
||||
|
||||
#define BOOST_TEST_MODULE VtkStructuredGridTest
|
||||
@@ -35,11 +36,11 @@ BOOST_AUTO_TEST_CASE(vtkStructuredGridTest) {
|
||||
StructuredGrid grid(Vector3i(10, 10, 100));
|
||||
grid.SetSpacing(Vector3f(3, 1, 1));
|
||||
|
||||
Vtk::StructuredGrid grid_viewer(&grid);
|
||||
Vtk::vtkStructuredGrid grid_viewer(grid);
|
||||
|
||||
if (std::getenv("CTEST_PROJECT_NAME") == nullptr) {
|
||||
Vtk::Viewer viewer;
|
||||
viewer.AddProp3D(grid_viewer);
|
||||
viewer.AddPuppet(grid_viewer);
|
||||
viewer.Start();
|
||||
}
|
||||
|
||||
|
||||
@@ -24,6 +24,7 @@
|
||||
//////////////////////////////////////////////////////////////////////////////*/
|
||||
|
||||
#include "Vtk/Math/vtkTriangleMesh.h"
|
||||
#include "Math/TriangleMesh.h"
|
||||
#include "Vtk/uLibVtkViewer.h"
|
||||
|
||||
#define BOOST_TEST_MODULE VtkTriangleMeshTest
|
||||
@@ -39,23 +40,22 @@ BOOST_AUTO_TEST_CASE(vtkTriangleMeshConstruction) {
|
||||
mesh.AddPoint(Vector3f(1, 0, 0));
|
||||
mesh.AddTriangle(Vector3i(0, 1, 2));
|
||||
|
||||
Vtk::TriangleMesh v_mesh(&mesh);
|
||||
Vtk::vtkTriangleMesh v_mesh(mesh);
|
||||
|
||||
Object::connect(&mesh, &TriangleMesh::Updated, [&mesh]() {
|
||||
Vector3f points[3];
|
||||
points[0] = mesh.GetPoint(0);
|
||||
points[1] = mesh.GetPoint(1);
|
||||
points[2] = mesh.GetPoint(2);
|
||||
std::cout << "mesh updated: " << points[0].transpose() << " "
|
||||
<< points[1].transpose() << " " << points[2].transpose()
|
||||
<< std::endl;
|
||||
std::cout << "mesh updated: " << points[0].transpose() << " " << points[1].transpose()
|
||||
<< " " << points[2].transpose() << std::endl;
|
||||
});
|
||||
|
||||
v_mesh.Update();
|
||||
|
||||
if (std::getenv("CTEST_PROJECT_NAME") == nullptr) {
|
||||
Vtk::Viewer viewer;
|
||||
viewer.AddProp3D(v_mesh);
|
||||
viewer.AddPuppet(v_mesh);
|
||||
viewer.Start();
|
||||
}
|
||||
|
||||
@@ -66,13 +66,13 @@ BOOST_AUTO_TEST_CASE(vtkTriangleMeshConstruction) {
|
||||
BOOST_AUTO_TEST_CASE(vtkTriangleMeshConstruction2) {
|
||||
TriangleMesh mesh;
|
||||
|
||||
Vtk::TriangleMesh v_mesh(&mesh);
|
||||
Vtk::vtkTriangleMesh v_mesh(mesh);
|
||||
v_mesh.ReadFromStlFile("capelluzzo.stl");
|
||||
v_mesh.Update();
|
||||
|
||||
if (std::getenv("CTEST_PROJECT_NAME") == nullptr) {
|
||||
Vtk::Viewer viewer;
|
||||
viewer.AddProp3D(v_mesh);
|
||||
viewer.AddPuppet(v_mesh);
|
||||
viewer.Start();
|
||||
}
|
||||
|
||||
|
||||
@@ -10,139 +10,133 @@
|
||||
//////////////////////////////////////////////////////////////////////////////*/
|
||||
|
||||
#include "Vtk/Math/vtkVoxImage.h"
|
||||
#include "Math/VoxImage.h"
|
||||
#include "Vtk/uLibVtkViewer.h"
|
||||
#include <cmath>
|
||||
#include <iostream>
|
||||
#include <vtkCallbackCommand.h>
|
||||
#include <vtkProperty.h>
|
||||
#include <vtkRenderWindow.h>
|
||||
#include <vtkRenderWindowInteractor.h>
|
||||
#include <vtkRenderer.h>
|
||||
#include <vtkSmartPointer.h>
|
||||
#include <vtkCallbackCommand.h>
|
||||
#include <vtkRenderWindowInteractor.h>
|
||||
#include <vtkProperty.h>
|
||||
#include <vtkRenderer.h>
|
||||
#include <vtkRenderWindow.h>
|
||||
#include <iostream>
|
||||
#include <cmath>
|
||||
|
||||
using namespace uLib;
|
||||
|
||||
struct AppState {
|
||||
std::vector<Vtk::VoxImage *> images;
|
||||
Vtk::Viewer *viewer;
|
||||
std::vector<Vtk::vtkVoxImage*> images;
|
||||
Vtk::Viewer* viewer;
|
||||
};
|
||||
|
||||
void KeyPressCallbackFunction(vtkObject *caller, long unsigned int eventId,
|
||||
void *clientData, void *callData) {
|
||||
auto *interactor = static_cast<vtkRenderWindowInteractor *>(caller);
|
||||
auto *state = static_cast<AppState *>(clientData);
|
||||
|
||||
std::string key = interactor->GetKeySym();
|
||||
if (key == "w") {
|
||||
std::cout << "--> Switching all images to Wireframe Box" << std::endl;
|
||||
for (auto *img : state->images)
|
||||
img->SetRepresentation(Vtk::Prop3D::Wireframe);
|
||||
state->viewer->GetRenderWindow()->Render();
|
||||
} else if (key == "s") {
|
||||
std::cout << "--> Switching all images to Surface (Volume Rendering)"
|
||||
<< std::endl;
|
||||
for (auto *img : state->images)
|
||||
img->SetRepresentation(Vtk::Prop3D::Surface);
|
||||
state->viewer->GetRenderWindow()->Render();
|
||||
} else if (key >= "0" && key <= "5") {
|
||||
int preset = key[0] - '0';
|
||||
std::cout << "--> Switching all images to Rendering Preset " << preset
|
||||
<< std::endl;
|
||||
for (auto *img : state->images)
|
||||
img->setShadingPreset(preset);
|
||||
state->viewer->GetRenderWindow()->Render();
|
||||
}
|
||||
void KeyPressCallbackFunction(vtkObject* caller, long unsigned int eventId, void* clientData, void* callData) {
|
||||
auto* interactor = static_cast<vtkRenderWindowInteractor*>(caller);
|
||||
auto* state = static_cast<AppState*>(clientData);
|
||||
|
||||
std::string key = interactor->GetKeySym();
|
||||
if (key == "w") {
|
||||
std::cout << "--> Switching all images to Wireframe Box" << std::endl;
|
||||
for (auto* img : state->images) img->SetRepresentation(Vtk::Puppet::Wireframe);
|
||||
state->viewer->GetRenderWindow()->Render();
|
||||
}
|
||||
else if (key == "s") {
|
||||
std::cout << "--> Switching all images to Surface (Volume Rendering)" << std::endl;
|
||||
for (auto* img : state->images) img->SetRepresentation(Vtk::Puppet::Surface);
|
||||
state->viewer->GetRenderWindow()->Render();
|
||||
}
|
||||
else if (key >= "0" && key <= "5") {
|
||||
int preset = key[0] - '0';
|
||||
std::cout << "--> Switching all images to Rendering Preset " << preset << std::endl;
|
||||
for (auto* img : state->images) img->setShadingPreset(preset);
|
||||
state->viewer->GetRenderWindow()->Render();
|
||||
}
|
||||
}
|
||||
|
||||
int main(int argc, char **argv) {
|
||||
float factor = 1.0e6f;
|
||||
int main(int argc, char** argv) {
|
||||
float factor = 1.0e6f;
|
||||
|
||||
// --- Image 1: Spherical Shell ---
|
||||
Vector3i dims1(64, 64, 64);
|
||||
VoxImage<Voxel> img1(dims1);
|
||||
img1.SetSpacing(Vector3f(1.0, 1.0, 1.0));
|
||||
img1.SetPosition(Vector3f(-40, -32, -32));
|
||||
// --- Image 1: Spherical Shell ---
|
||||
Vector3i dims1(64, 64, 64);
|
||||
VoxImage<Voxel> img1(dims1);
|
||||
img1.SetSpacing(Vector3f(1.0, 1.0, 1.0));
|
||||
img1.SetPosition(Vector3f(-40, -32, -32));
|
||||
|
||||
for (int z = 0; z < dims1(2); ++z) {
|
||||
for (int y = 0; y < dims1(1); ++y) {
|
||||
for (int x = 0; x < dims1(0); ++x) {
|
||||
float dx = x - 32.0f;
|
||||
float dy = y - 32.0f;
|
||||
float dz = z - 32.0f;
|
||||
float dist = std::sqrt(dx * dx + dy * dy + dz * dz);
|
||||
Voxel v;
|
||||
if (dist < 25.0f && dist > 10.0f) {
|
||||
v.Value = (40.0f * (25.0f - dist) / 15.0f) / factor;
|
||||
} else {
|
||||
v.Value = 0.0f;
|
||||
for (int z = 0; z < dims1(2); ++z) {
|
||||
for (int y = 0; y < dims1(1); ++y) {
|
||||
for (int x = 0; x < dims1(0); ++x) {
|
||||
float dx = x - 32.0f;
|
||||
float dy = y - 32.0f;
|
||||
float dz = z - 32.0f;
|
||||
float dist = std::sqrt(dx*dx + dy*dy + dz*dz);
|
||||
Voxel v;
|
||||
if (dist < 25.0f && dist > 10.0f) {
|
||||
v.Value = (40.0f * (25.0f - dist) / 15.0f) / factor;
|
||||
} else {
|
||||
v.Value = 0.0f;
|
||||
}
|
||||
img1[Vector3i(x, y, z)] = v;
|
||||
}
|
||||
}
|
||||
img1[Vector3i(x, y, z)] = v;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// --- Image 2: Axes Gradient ---
|
||||
Vector3i dims2(64, 64, 64);
|
||||
VoxImage<Voxel> img2(dims2);
|
||||
img2.SetSpacing(Vector3f(1.0, 1.0, 1.0));
|
||||
img2.SetPosition(Vector3f(40, -32, -32));
|
||||
// --- Image 2: Axes Gradient ---
|
||||
Vector3i dims2(64, 64, 64);
|
||||
VoxImage<Voxel> img2(dims2);
|
||||
img2.SetSpacing(Vector3f(1.0, 1.0, 1.0));
|
||||
img2.SetPosition(Vector3f(40, -32, -32));
|
||||
|
||||
for (int z = 0; z < dims2(2); ++z) {
|
||||
for (int y = 0; y < dims2(1); ++y) {
|
||||
for (int x = 0; x < dims2(0); ++x) {
|
||||
Voxel v;
|
||||
// Linear gradient along X, Y, Z
|
||||
float val =
|
||||
(float(x) / dims2(0) + float(y) / dims2(1) + float(z) / dims2(2)) /
|
||||
3.0f;
|
||||
v.Value = (40.0f * val) / factor;
|
||||
img2[Vector3i(x, y, z)] = v;
|
||||
}
|
||||
for (int z = 0; z < dims2(2); ++z) {
|
||||
for (int y = 0; y < dims2(1); ++y) {
|
||||
for (int x = 0; x < dims2(0); ++x) {
|
||||
Voxel v;
|
||||
// Linear gradient along X, Y, Z
|
||||
float val = (float(x)/dims2(0) + float(y)/dims2(1) + float(z)/dims2(2)) / 3.0f;
|
||||
v.Value = (40.0f * val) / factor;
|
||||
img2[Vector3i(x, y, z)] = v;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Vtk::VoxImage vtk_img1(&img1);
|
||||
vtk_img1.setShadingPreset(0);
|
||||
Vtk::vtkVoxImage vtk_img1(img1);
|
||||
vtk_img1.setShadingPreset(0);
|
||||
|
||||
Vtk::VoxImage vtk_img2(&img2);
|
||||
vtk_img2.setShadingPreset(1); // Use Composite without MIP for variety
|
||||
Vtk::vtkVoxImage vtk_img2(img2);
|
||||
vtk_img2.setShadingPreset(1); // Use Composite without MIP for variety
|
||||
|
||||
Vtk::Viewer viewer;
|
||||
viewer.GetRenderer()->SetBackground(0.05, 0.05, 0.1);
|
||||
viewer.AddProp3D(vtk_img1);
|
||||
viewer.AddProp3D(vtk_img2);
|
||||
Vtk::Viewer viewer;
|
||||
viewer.GetRenderer()->SetBackground(0.05, 0.05, 0.1);
|
||||
viewer.AddPuppet(vtk_img1);
|
||||
viewer.AddPuppet(vtk_img2);
|
||||
|
||||
// Setup KeyPress Callback
|
||||
AppState state;
|
||||
state.images.push_back(&vtk_img1);
|
||||
state.images.push_back(&vtk_img2);
|
||||
state.viewer = &viewer;
|
||||
|
||||
// Setup KeyPress Callback
|
||||
AppState state;
|
||||
state.images.push_back(&vtk_img1);
|
||||
state.images.push_back(&vtk_img2);
|
||||
state.viewer = &viewer;
|
||||
vtkSmartPointer<vtkCallbackCommand> keyCallback = vtkSmartPointer<vtkCallbackCommand>::New();
|
||||
keyCallback->SetCallback(KeyPressCallbackFunction);
|
||||
keyCallback->SetClientData(&state);
|
||||
viewer.GetInteractor()->AddObserver(vtkCommand::KeyPressEvent, keyCallback);
|
||||
|
||||
vtkSmartPointer<vtkCallbackCommand> keyCallback =
|
||||
vtkSmartPointer<vtkCallbackCommand>::New();
|
||||
keyCallback->SetCallback(KeyPressCallbackFunction);
|
||||
keyCallback->SetClientData(&state);
|
||||
viewer.GetInteractor()->AddObserver(vtkCommand::KeyPressEvent, keyCallback);
|
||||
std::cout << "=========================================" << std::endl;
|
||||
std::cout << " vtkVoxImage Interactive Viewer Test" << std::endl;
|
||||
std::cout << " [LEFT] Spherical Shell (MIP)" << std::endl;
|
||||
std::cout << " [RIGHT] Axes Gradient (Composite)" << std::endl;
|
||||
std::cout << "-----------------------------------------" << std::endl;
|
||||
std::cout << " Press [w] to show Wireframe Bounding Boxes" << std::endl;
|
||||
std::cout << " Press [s] to show Volume Rendering" << std::endl;
|
||||
std::cout << " Press [0..5] to switch Rendering Presets:" << std::endl;
|
||||
std::cout << " 0: MIP (Grayscale)" << std::endl;
|
||||
std::cout << " 1: Composite (Grayscale)" << std::endl;
|
||||
std::cout << " 2: Composite (Shaded)" << std::endl;
|
||||
std::cout << " 3: CT Bone/Tissue (Bone colors)" << std::endl;
|
||||
std::cout << " 4: MIP (Rainbow)" << std::endl;
|
||||
std::cout << " 5: Additive (Total path sum)" << std::endl;
|
||||
std::cout << " Press [q] to quit" << std::endl;
|
||||
std::cout << "=========================================" << std::endl;
|
||||
|
||||
std::cout << "=========================================" << std::endl;
|
||||
std::cout << " VoxImage Interactive Viewer Test" << std::endl;
|
||||
std::cout << " [LEFT] Spherical Shell (MIP)" << std::endl;
|
||||
std::cout << " [RIGHT] Axes Gradient (Composite)" << std::endl;
|
||||
std::cout << "-----------------------------------------" << std::endl;
|
||||
std::cout << " Press [w] to show Wireframe Bounding Boxes" << std::endl;
|
||||
std::cout << " Press [s] to show Volume Rendering" << std::endl;
|
||||
std::cout << " Press [0..5] to switch Rendering Presets:" << std::endl;
|
||||
std::cout << " 0: MIP (Grayscale)" << std::endl;
|
||||
std::cout << " 1: Composite (Grayscale)" << std::endl;
|
||||
std::cout << " 2: Composite (Shaded)" << std::endl;
|
||||
std::cout << " 3: CT Bone/Tissue (Bone colors)" << std::endl;
|
||||
std::cout << " 4: MIP (Rainbow)" << std::endl;
|
||||
std::cout << " 5: Additive (Total path sum)" << std::endl;
|
||||
std::cout << " Press [q] to quit" << std::endl;
|
||||
std::cout << "=========================================" << std::endl;
|
||||
viewer.ZoomAuto();
|
||||
viewer.Start();
|
||||
|
||||
viewer.ZoomAuto();
|
||||
viewer.Start();
|
||||
|
||||
return 0;
|
||||
return 0;
|
||||
}
|
||||
|
||||
Some files were not shown because too many files have changed in this diff Show More
Reference in New Issue
Block a user