Compare commits
16 Commits
46c39bc26e
...
andrea-geo
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
6396bdfebf | ||
|
|
96ab3b0930 | ||
|
|
5c04d00d4c | ||
|
|
72e69cfca5 | ||
|
|
59a9e829fc | ||
|
|
6068b62e39 | ||
|
|
4435776484 | ||
|
|
a1c5fc2600 | ||
|
|
9118afdd13 | ||
|
|
8e6e332217 | ||
|
|
e1bd7eb44f | ||
|
|
c0c25de694 | ||
|
|
34f834d370 | ||
|
|
f3274f346b | ||
|
|
d4fd2d3914 | ||
|
|
22d0041942 |
4
.clangd
4
.clangd
@@ -1,7 +1,7 @@
|
|||||||
CompileFlags:
|
CompileFlags:
|
||||||
CompilationDatabase: build
|
CompilationDatabase: build
|
||||||
Add:
|
Add:
|
||||||
- -I/home/rigoni/devel/cmt/ulib/src
|
- -I/home/rigoni/devel/cmt/uLib/src
|
||||||
- -isystem/home/share/micromamba/envs/mutom/include
|
- -isystem/home/share/micromamba/envs/mutom/include
|
||||||
- -isystem/home/share/micromamba/envs/mutom/include/eigen3
|
- -isystem/home/share/micromamba/envs/mutom/include/eigen3
|
||||||
- -isystem/home/share/micromamba/envs/mutom/targets/x86_64-linux/include
|
- -isystem/home/share/micromamba/envs/mutom/targets/x86_64-linux/include
|
||||||
@@ -27,7 +27,7 @@ Diagnostics:
|
|||||||
|
|
||||||
---
|
---
|
||||||
If:
|
If:
|
||||||
PathExclude: [/home/rigoni/devel/cmt/ulib/src/.*]
|
PathExclude: [/home/rigoni/devel/cmt/uLib/src/.*]
|
||||||
Diagnostics:
|
Diagnostics:
|
||||||
Suppress: ["*"]
|
Suppress: ["*"]
|
||||||
|
|
||||||
|
|||||||
4
.gitignore
vendored
4
.gitignore
vendored
@@ -13,3 +13,7 @@ src/Python/uLib/*.pyd
|
|||||||
src/Python/uLib/*.pyc
|
src/Python/uLib/*.pyc
|
||||||
src/Python/uLib/__pycache__
|
src/Python/uLib/__pycache__
|
||||||
src/Python/uLib/.nfs*
|
src/Python/uLib/.nfs*
|
||||||
|
test_props.xml
|
||||||
|
test_props2.xml
|
||||||
|
test_boost.cpp
|
||||||
|
.claude/settings.json
|
||||||
|
|||||||
5
.vscode/settings.json
vendored
5
.vscode/settings.json
vendored
@@ -1,6 +1,6 @@
|
|||||||
{
|
{
|
||||||
"clangd.fallbackFlags": [
|
"clangd.fallbackFlags": [
|
||||||
"-I/home/rigoni/devel/cmt/ulib/src",
|
"-I/home/rigoni/devel/cmt/uLib/src",
|
||||||
"-isystem/home/share/micromamba/envs/mutom/include",
|
"-isystem/home/share/micromamba/envs/mutom/include",
|
||||||
"-isystem/home/share/micromamba/envs/mutom/include/eigen3",
|
"-isystem/home/share/micromamba/envs/mutom/include/eigen3",
|
||||||
"-isystem/home/share/micromamba/envs/mutom/targets/x86_64-linux/include",
|
"-isystem/home/share/micromamba/envs/mutom/targets/x86_64-linux/include",
|
||||||
@@ -19,8 +19,7 @@
|
|||||||
"clangd.semanticHighlighting.enable": true,
|
"clangd.semanticHighlighting.enable": true,
|
||||||
"clangd.arguments": [
|
"clangd.arguments": [
|
||||||
"--compile-commands-dir=build",
|
"--compile-commands-dir=build",
|
||||||
"--query-driver=/home/share/micromamba/envs/mutom/bin/g++,/home/share/micromamba/envs/mutom/bin/gcc,/home/share/micromamba/envs/mutom/bin/nvcc",
|
"--query-driver=/home/share/micromamba/envs/mutom/bin/*",
|
||||||
"--suppress-system-warnings",
|
|
||||||
"--all-scopes-completion",
|
"--all-scopes-completion",
|
||||||
"--completion-style=detailed",
|
"--completion-style=detailed",
|
||||||
"--header-insertion=never",
|
"--header-insertion=never",
|
||||||
|
|||||||
@@ -38,7 +38,7 @@ endif()
|
|||||||
|
|
||||||
# The version number.
|
# The version number.
|
||||||
set(PROJECT_VERSION_MAJOR 0)
|
set(PROJECT_VERSION_MAJOR 0)
|
||||||
set(PROJECT_VERSION_MINOR 6)
|
set(PROJECT_VERSION_MINOR 7)
|
||||||
set(PROJECT_VERSION "${PROJECT_VERSION_MAJOR}.${PROJECT_VERSION_MINOR}")
|
set(PROJECT_VERSION "${PROJECT_VERSION_MAJOR}.${PROJECT_VERSION_MINOR}")
|
||||||
set(PROJECT_SOVERSION "${PROJECT_VERSION_MAJOR}.${PROJECT_VERSION_MINOR}")
|
set(PROJECT_SOVERSION "${PROJECT_VERSION_MAJOR}.${PROJECT_VERSION_MINOR}")
|
||||||
|
|
||||||
|
|||||||
@@ -11,6 +11,12 @@
|
|||||||
"CMAKE_BUILD_TYPE": "Debug",
|
"CMAKE_BUILD_TYPE": "Debug",
|
||||||
"CMAKE_INSTALL_PREFIX": "${sourceDir}/out/install/${presetName}"
|
"CMAKE_INSTALL_PREFIX": "${sourceDir}/out/install/${presetName}"
|
||||||
}
|
}
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"name": "mutom",
|
||||||
|
"description": "",
|
||||||
|
"displayName": "",
|
||||||
|
"inherits": []
|
||||||
}
|
}
|
||||||
]
|
]
|
||||||
}
|
}
|
||||||
@@ -36,6 +36,8 @@ PropertyWidgetBase::PropertyWidgetBase(PropertyBase* prop, QWidget* parent)
|
|||||||
m_Label = new QLabel(labelText, this);
|
m_Label = new QLabel(labelText, this);
|
||||||
m_Label->setMinimumWidth(120);
|
m_Label->setMinimumWidth(120);
|
||||||
m_Layout->addWidget(m_Label);
|
m_Layout->addWidget(m_Label);
|
||||||
|
|
||||||
|
this->setEnabled(!prop->IsReadOnly());
|
||||||
}
|
}
|
||||||
PropertyWidgetBase::~PropertyWidgetBase() {
|
PropertyWidgetBase::~PropertyWidgetBase() {
|
||||||
m_Connection.disconnect();
|
m_Connection.disconnect();
|
||||||
@@ -150,7 +152,7 @@ DoublePropertyWidget::DoublePropertyWidget(Property<double>* prop, QWidget* pare
|
|||||||
m_Edit->setValue(prop->Get());
|
m_Edit->setValue(prop->Get());
|
||||||
m_Layout->addWidget(m_Edit, 1);
|
m_Layout->addWidget(m_Edit, 1);
|
||||||
connect(m_Edit, &UnitLineEdit::valueManualChanged, [this](double val){ m_Prop->Set(val); });
|
connect(m_Edit, &UnitLineEdit::valueManualChanged, [this](double val){ m_Prop->Set(val); });
|
||||||
m_Connection = uLib::Object::connect(m_Prop, &Property<double>::PropertyChanged, [this](){
|
m_Connection = uLib::Object::connect(m_Prop, &Property<double>::Updated, [this](){
|
||||||
m_Edit->setValue(m_Prop->Get());
|
m_Edit->setValue(m_Prop->Get());
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
@@ -168,7 +170,7 @@ FloatPropertyWidget::FloatPropertyWidget(Property<float>* prop, QWidget* parent)
|
|||||||
m_Edit->setValue(prop->Get());
|
m_Edit->setValue(prop->Get());
|
||||||
m_Layout->addWidget(m_Edit, 1);
|
m_Layout->addWidget(m_Edit, 1);
|
||||||
connect(m_Edit, &UnitLineEdit::valueManualChanged, [this](double val){ m_Prop->Set((float)val); });
|
connect(m_Edit, &UnitLineEdit::valueManualChanged, [this](double val){ m_Prop->Set((float)val); });
|
||||||
m_Connection = uLib::Object::connect(m_Prop, &Property<float>::PropertyChanged, [this](){
|
m_Connection = uLib::Object::connect(m_Prop, &Property<float>::Updated, [this](){
|
||||||
m_Edit->setValue((double)m_Prop->Get());
|
m_Edit->setValue((double)m_Prop->Get());
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
@@ -187,7 +189,7 @@ IntPropertyWidget::IntPropertyWidget(Property<int>* prop, QWidget* parent)
|
|||||||
m_Edit->setValue(prop->Get());
|
m_Edit->setValue(prop->Get());
|
||||||
m_Layout->addWidget(m_Edit, 1);
|
m_Layout->addWidget(m_Edit, 1);
|
||||||
connect(m_Edit, &UnitLineEdit::valueManualChanged, [this](double val){ m_Prop->Set((int)val); });
|
connect(m_Edit, &UnitLineEdit::valueManualChanged, [this](double val){ m_Prop->Set((int)val); });
|
||||||
m_Connection = uLib::Object::connect(m_Prop, &Property<int>::PropertyChanged, [this](){
|
m_Connection = uLib::Object::connect(m_Prop, &Property<int>::Updated, [this](){
|
||||||
m_Edit->setValue((double)m_Prop->Get());
|
m_Edit->setValue((double)m_Prop->Get());
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
@@ -198,7 +200,7 @@ BoolPropertyWidget::BoolPropertyWidget(Property<bool>* prop, QWidget* parent)
|
|||||||
m_CheckBox->setChecked(prop->Get());
|
m_CheckBox->setChecked(prop->Get());
|
||||||
m_Layout->addWidget(m_CheckBox, 1);
|
m_Layout->addWidget(m_CheckBox, 1);
|
||||||
connect(m_CheckBox, &QCheckBox::toggled, [this](bool val){ if (m_Prop->Get() != val) m_Prop->Set(val); });
|
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>::PropertyChanged, [this](){
|
m_Connection = uLib::Object::connect(m_Prop, &Property<bool>::Updated, [this](){
|
||||||
if (m_CheckBox->isChecked() != m_Prop->Get()) {
|
if (m_CheckBox->isChecked() != m_Prop->Get()) {
|
||||||
QSignalBlocker blocker(m_CheckBox);
|
QSignalBlocker blocker(m_CheckBox);
|
||||||
m_CheckBox->setChecked(m_Prop->Get());
|
m_CheckBox->setChecked(m_Prop->Get());
|
||||||
@@ -222,7 +224,7 @@ RangePropertyWidget::RangePropertyWidget(Property<double>* prop, QWidget* parent
|
|||||||
connect(m_Slider, &QSlider::valueChanged, this, &RangePropertyWidget::onSliderChanged);
|
connect(m_Slider, &QSlider::valueChanged, this, &RangePropertyWidget::onSliderChanged);
|
||||||
connect(m_Edit, &UnitLineEdit::valueManualChanged, [this](double val){ m_Prop->Set(val); });
|
connect(m_Edit, &UnitLineEdit::valueManualChanged, [this](double val){ m_Prop->Set(val); });
|
||||||
|
|
||||||
m_Connection = uLib::Object::connect(m_Prop, &Property<double>::PropertyChanged, [this](){
|
m_Connection = uLib::Object::connect(m_Prop, &Property<double>::Updated, [this](){
|
||||||
this->updateUi();
|
this->updateUi();
|
||||||
});
|
});
|
||||||
updateUi();
|
updateUi();
|
||||||
@@ -252,7 +254,7 @@ ColorPropertyWidget::ColorPropertyWidget(Property<Vector3d>* prop, QWidget* pare
|
|||||||
m_Layout->addWidget(m_Button, 0, ::Qt::AlignRight);
|
m_Layout->addWidget(m_Button, 0, ::Qt::AlignRight);
|
||||||
|
|
||||||
connect(m_Button, &QPushButton::clicked, this, &ColorPropertyWidget::onClicked);
|
connect(m_Button, &QPushButton::clicked, this, &ColorPropertyWidget::onClicked);
|
||||||
m_Connection = uLib::Object::connect(m_Prop, &Property<Vector3d>::PropertyChanged, [this](){
|
m_Connection = uLib::Object::connect(m_Prop, &Property<Vector3d>::Updated, [this](){
|
||||||
this->updateButtonColor();
|
this->updateButtonColor();
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
@@ -286,7 +288,7 @@ StringPropertyWidget::StringPropertyWidget(Property<std::string>* prop, QWidget*
|
|||||||
std::string val = m_LineEdit->text().toStdString();
|
std::string val = m_LineEdit->text().toStdString();
|
||||||
if (m_Prop->Get() != val) m_Prop->Set(val);
|
if (m_Prop->Get() != val) m_Prop->Set(val);
|
||||||
});
|
});
|
||||||
m_Connection = uLib::Object::connect(m_Prop, &Property<std::string>::PropertyChanged, [this](){
|
m_Connection = uLib::Object::connect(m_Prop, &Property<std::string>::Updated, [this](){
|
||||||
if (m_LineEdit->text().toStdString() != m_Prop->Get()) {
|
if (m_LineEdit->text().toStdString() != m_Prop->Get()) {
|
||||||
QSignalBlocker blocker(m_LineEdit);
|
QSignalBlocker blocker(m_LineEdit);
|
||||||
m_LineEdit->setText(QString::fromStdString(m_Prop->Get()));
|
m_LineEdit->setText(QString::fromStdString(m_Prop->Get()));
|
||||||
@@ -334,7 +336,7 @@ public:
|
|||||||
p->Set(index);
|
p->Set(index);
|
||||||
});
|
});
|
||||||
// Store connection in base m_Connection so it's auto-disconnected on destruction.
|
// Store connection in base m_Connection so it's auto-disconnected on destruction.
|
||||||
m_Connection = uLib::Object::connect(p, &Property<int>::PropertyChanged, [this, p](){
|
m_Connection = uLib::Object::connect(p, &Property<int>::Updated, [this, p](){
|
||||||
if (m_Combo->currentIndex() != p->Get()) {
|
if (m_Combo->currentIndex() != p->Get()) {
|
||||||
QSignalBlocker blocker(m_Combo);
|
QSignalBlocker blocker(m_Combo);
|
||||||
m_Combo->setCurrentIndex(p->Get());
|
m_Combo->setCurrentIndex(p->Get());
|
||||||
|
|||||||
@@ -115,6 +115,7 @@ public:
|
|||||||
if (!prefSuffix.isEmpty()) {
|
if (!prefSuffix.isEmpty()) {
|
||||||
m_Edits[i]->setUnits(prefSuffix, factor);
|
m_Edits[i]->setUnits(prefSuffix, factor);
|
||||||
}
|
}
|
||||||
|
m_Edits[i]->setEnabled(!prop->IsReadOnly());
|
||||||
m_Layout->addWidget(m_Edits[i], 1);
|
m_Layout->addWidget(m_Edits[i], 1);
|
||||||
|
|
||||||
connect(m_Edits[i], &UnitLineEdit::valueManualChanged, [this, i](double val){
|
connect(m_Edits[i], &UnitLineEdit::valueManualChanged, [this, i](double val){
|
||||||
@@ -124,7 +125,7 @@ public:
|
|||||||
});
|
});
|
||||||
}
|
}
|
||||||
updateEdits();
|
updateEdits();
|
||||||
m_Connection = uLib::Object::connect(m_Prop, &Property<VecT>::PropertyChanged, [this](){
|
m_Connection = uLib::Object::connect(m_Prop, &Property<VecT>::Updated, [this](){
|
||||||
updateEdits();
|
updateEdits();
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|||||||
84
docs/archives.md
Normal file
84
docs/archives.md
Normal file
@@ -0,0 +1,84 @@
|
|||||||
|
# Serialization and Archives Internals
|
||||||
|
|
||||||
|
This document explains the internal design of the `uLib` serialization system, which is built on top of **Boost.Serialization**. It provides custom archive implementations for various formats (XML, Text, Logging) and introduces **Human Readable Pairs (HRP)** for metadata-rich serialization.
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## Architecture Overview
|
||||||
|
|
||||||
|
The `uLib` archive system extends the standard `boost::archive` templates to add domain-specific features. The main components are:
|
||||||
|
|
||||||
|
1. **Custom Interface Layer**: Extends the default Boost archive API with additional operators and utilities.
|
||||||
|
2. **Specialized Archive Implementations**: Specialized classes for XML, Text, and Logging.
|
||||||
|
3. **HRP Support**: First-class support for `hrp` (Human Readable Pair) wrappers, which carry units, ranges, and descriptions.
|
||||||
|
4. **Static Registration System**: Macros and explicit instantiations to handle polymorphic types and compilation isolation.
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## Custom Interface Layer
|
||||||
|
|
||||||
|
All `uLib` archives use a custom interface defined in `Archives.h` via `uLib_interface_iarchive` and `uLib_interface_oarchive`. These templates add several key features:
|
||||||
|
|
||||||
|
| Feature | Operator/Method | Description |
|
||||||
|
|---|---|---|
|
||||||
|
| **Mapping Operator** | `operator==` | Aliased to `operator&` (Boost's standard mapping operator). |
|
||||||
|
| **Trace Operator** | `operator!=` | Used for trace/debug output of strings during serialization. |
|
||||||
|
| **Type Registration** | `register_type<T>()` | Registers a class type with the archive's internal serializer map. |
|
||||||
|
| **Standard IO** | `operator<<` / `operator>>` | Standard redirect for saving and loading. |
|
||||||
|
|
||||||
|
These interfaces are applied to the archives using template specialization of `boost::archive::detail::interface_iarchive` and `interface_oarchive`.
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## Archive Variants
|
||||||
|
|
||||||
|
### XML Archives (`xml_iarchive`, `xml_oarchive`)
|
||||||
|
These inherit from `boost::archive::xml_iarchive_impl` and `xml_oarchive_impl`.
|
||||||
|
- **Internals**: They override `load_override` and `save_override` to handle `boost::serialization::hrp<T>` specifically.
|
||||||
|
- **XML Mapping**: When saving an `hrp`, it uses `save_start(name)` and `save_end(name)` to wrap the value in a named XML tag.
|
||||||
|
|
||||||
|
### Text Archives (`text_iarchive`, `text_oarchive`)
|
||||||
|
Standard text-based archives used for compact serialization. They use `StringReader` to consume decorative text markers during loading.
|
||||||
|
|
||||||
|
### Human Readable Text (`hrt_iarchive`, `hrt_oarchive`)
|
||||||
|
These are "naked" text archives that suppress most of Boost's internal metadata (object IDs, class IDs, versions).
|
||||||
|
- **Goal**: Produce text output that is easy for humans to read and edit.
|
||||||
|
- **Internals**: All overrides for Boost internal types (like `object_id_type`, `version_type`, etc.) are implemented as no-ops.
|
||||||
|
|
||||||
|
### Log Archive (`log_archive`)
|
||||||
|
An XML-based output archive specifically for debug logging.
|
||||||
|
- **Internals**: It forces every object into a Name-Value Pair (NVP) even if not provided by the user, and strips all technical metadata to keep the logs clean.
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## HRP (Human Readable Pair) Integration
|
||||||
|
|
||||||
|
`hrp` is a core `uLib` wrapper (defined in `Serializable.h`) that extends Boost's `nvp`:
|
||||||
|
|
||||||
|
```cpp
|
||||||
|
// Example of HRP usage
|
||||||
|
ar & HRP2("Energy", m_energy, "MeV").range(0, 100);
|
||||||
|
```
|
||||||
|
|
||||||
|
### Internal Handling in Archives
|
||||||
|
Archives in `Archives.h` provide specific `save_override`/`load_override` for `hrp<T>`:
|
||||||
|
- **XML**: Maps the `name()` to an XML tag.
|
||||||
|
- **HRT**: Formats as `name: value [units]\n`.
|
||||||
|
- **Log**: Converts it to a standard Boost `nvp` for consistent XML logging.
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## Registration and Polymorphism
|
||||||
|
|
||||||
|
### Registration Macro
|
||||||
|
The `ULIB_SERIALIZATION_REGISTER_ARCHIVE(Archive)` macro is crucial for polymorphic serialization. It instantiates the necessary template machinery to link the custom `Archive` type with any `Serializable` class exported via `BOOST_CLASS_EXPORT`.
|
||||||
|
|
||||||
|
### Explicit Instantiation
|
||||||
|
To reduce compilation times and provide a single point of failure for link-time issues, `uLib` uses explicit instantiations in `src/Core/Archives.cpp`. This file includes the `.ipp` implementation files from Boost and instantiates the `archive_serializer_map` and implementation classes for all `uLib` archive types.
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## Utility: StringReader
|
||||||
|
The `StringReader` utility is used internally by text-based archives to parse and skip literals. For example:
|
||||||
|
- When loading a string literal from a text archive, `StringReader` consumes whitespace and ensures the stream matches the expected string, failing if there is a mismatch.
|
||||||
|
- This is vital for maintaining the structure of human-readable formats.
|
||||||
11
docs/geant_integration.md
Normal file
11
docs/geant_integration.md
Normal file
@@ -0,0 +1,11 @@
|
|||||||
|
# Geant integration
|
||||||
|
|
||||||
|
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
|
||||||
|
|
||||||
|
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,73 +0,0 @@
|
|||||||
# Transformation Flow and Synchronization System
|
|
||||||
|
|
||||||
This document describes how transformations are applied and synchronized between the interactive 3D viewport, the visualization puppets, and the underlying mathematical models within the `uLib` framework.
|
|
||||||
|
|
||||||
## Architecture Overview
|
|
||||||
|
|
||||||
The system follows a Model-View-Controller (MVC) like pattern where:
|
|
||||||
- **Model**: `uLib::AffineTransform` (or derived classes like `ContainerBox`).
|
|
||||||
- **View/Puppet**: `uLib::Vtk::Puppet` (and specialized derivations like `Vtk::Assembly`).
|
|
||||||
- **Controller/Interaction**: `vtkHandlerWidget` (the transformation gizmo).
|
|
||||||
|
|
||||||
---
|
|
||||||
|
|
||||||
## 1. Interaction Flow (Gizmo -> Model)
|
|
||||||
|
|
||||||
When a user interacts with the `vtkHandlerWidget` (dragging arrows, rings, or cubes), the following chain of events occurs:
|
|
||||||
|
|
||||||
```mermaid
|
|
||||||
sequenceDiagram
|
|
||||||
participant User
|
|
||||||
participant HW as vtkHandlerWidget
|
|
||||||
participant VP as vtkViewport
|
|
||||||
participant P as vtkPuppet
|
|
||||||
participant M as uLib Model
|
|
||||||
|
|
||||||
User->>HW: Drag handle (MouseMove)
|
|
||||||
HW->>HW: Calculate Delta Matrix (op)
|
|
||||||
HW->>HW: Total = StartState * op
|
|
||||||
HW->>HW: Decompose Total into P, O, S
|
|
||||||
HW->>P: SetPosition, SetOrientation, SetScale
|
|
||||||
HW-->>VP: Invoke InteractionEvent
|
|
||||||
VP->>P: SyncFromVtk()
|
|
||||||
P->>P: Get local matrix from VTK Prop
|
|
||||||
P->>M: SetMatrix(matrix)
|
|
||||||
M-->>M: Update local properties (P, O, S)
|
|
||||||
M-->>P: Emit Updated signal
|
|
||||||
P->>P: Puppet::Update()
|
|
||||||
P->>P: (Redundant sanity write to Prop)
|
|
||||||
```
|
|
||||||
|
|
||||||
### Key Principles:
|
|
||||||
- **Single Source of Truth**: The `uLib::AffineTransform` is the owner of the transformation state.
|
|
||||||
- **Internal TRS vs UserMatrix**: We apply transformations directly to VTK's internal `Position`, `Orientation`, and `Scale` properties. This ensures the data is "visible" to VTK actors and simplifies decomposition.
|
|
||||||
- **Cumulative Bias Avoidance**: The `HandlerWidget` calculates transformations relative to the state at the start of the click, preventing numerical drift during a single drag operation.
|
|
||||||
|
|
||||||
---
|
|
||||||
|
|
||||||
## 2. Synchronization Loop Resolution
|
|
||||||
|
|
||||||
To prevent infinite loops and "double-transformation" artifacts (especially in assemblies), the following protections are in place:
|
|
||||||
|
|
||||||
1. **Hierarchy Isolation**: The `Puppet` base class distinguishes between the **Root Property** (which receives the puppet's master transformation) and **Sub-Parts** (which only receive appearance updates like color/visibility). This prevents parts from inheriting the same displacement twice.
|
|
||||||
2. **Re-entrancy Guards**: Puppets use an `m_InUpdate` flag to prevent a feedback loop where `SyncFromVtk` triggers a Model Update, which then re-triggers the Puppet Update.
|
|
||||||
3. **Signal Blocking**: In specialized cases (like `vtkAssembly`), `m_BlockUpdate` is used to prevent the model-to-puppet push during a puppet-to-model sync.
|
|
||||||
|
|
||||||
---
|
|
||||||
|
|
||||||
## 3. Undo System (Ctrl-Z)
|
|
||||||
|
|
||||||
### Current Implementation (Delta Chain)
|
|
||||||
Currently, the system maintains a `m_TransformChain` of delta matrices.
|
|
||||||
- **Record**: After every drag, a delta matrix ($M_{delta} = M_{end} \cdot M_{start}^{-1}$) is appended to the chain.
|
|
||||||
- **Undo**: The last delta is removed, and the prop is reconstructed by reapplying the remaining chain from a `BaseMatrix`.
|
|
||||||
|
|
||||||
### Planned Improvement (TRS Snapshots)
|
|
||||||
We are migrating to a `uLib::TRS` snapshot system for Undo.
|
|
||||||
- **Record**: At the start of a drag, the current `TRS` state of the object is pushed to the `m_UndoStack`.
|
|
||||||
- **Undo**: The top `TRS` is popped and applied directly to the model.
|
|
||||||
|
|
||||||
This approach is more robust because:
|
|
||||||
- It eliminates matrix multiplication error accumulation.
|
|
||||||
- It bypasses rotation convention/order issues (Gimbal lock in deltas).
|
|
||||||
- It returns the object to exactly its previous property values.
|
|
||||||
94
docs/update_properties.md
Normal file
94
docs/update_properties.md
Normal file
@@ -0,0 +1,94 @@
|
|||||||
|
|
||||||
|
# Properties and the vtk-gui representation
|
||||||
|
|
||||||
|
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 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 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 Puppet
|
||||||
|
|
||||||
|
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 puppet internal ContainerBox. The changes of the ContainerBox will be propagated to the derived classes and eventually to the model.
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
## ACTIVATE PROPERTIES
|
||||||
|
|
||||||
|
ULIB_ACTIVATE_PROPERTIES must run after all member initialization, with the vtable pointing to the most-derived type. This is why it has to be in each constructor — in C++, virtual dispatch only works correctly after a class's vtable is installed, which happens at the start of each level's constructor body.
|
||||||
|
|
||||||
|
### Option 1 — End-of-class macro (no constructor boilerplate)
|
||||||
|
Declare a private member activator as the last member of the class. Its constructor runs after all other members, and at that point the vtable is already Derived's:
|
||||||
|
|
||||||
|
|
||||||
|
// In Property.h, add alongside ULIB_ACTIVATE_PROPERTIES:
|
||||||
|
#define ULIB_DECLARE_PROPERTIES(SelfType) \
|
||||||
|
private: \
|
||||||
|
struct _PropActivator { \
|
||||||
|
_PropActivator(SelfType* self) { \
|
||||||
|
uLib::Archive::property_register_archive ar(self); \
|
||||||
|
ar & *self; \
|
||||||
|
} \
|
||||||
|
} _prop_activator{this};
|
||||||
|
Usage in ContainerBox.h — place it just before the closing brace:
|
||||||
|
|
||||||
|
|
||||||
|
class ContainerBox : public TRS {
|
||||||
|
public:
|
||||||
|
// ... all constructors, no more ULIB_ACTIVATE_PROPERTIES(*this)
|
||||||
|
|
||||||
|
ULIB_DECLARE_PROPERTIES(ContainerBox) // ← replaces all 3 constructor calls
|
||||||
|
};
|
||||||
|
Tradeoff: Works perfectly for single-level classes. For hierarchies where multiple levels use the macro, RegisterDynamicProperty must deduplicate by name (skip if already registered). Requires one line per class in the class body, but zero lines in constructors.
|
||||||
|
|
||||||
|
### Option 2 — Lazy init via virtual InitProperties() in Object
|
||||||
|
Modify Object to call a virtual hook on first GetProperties():
|
||||||
|
|
||||||
|
|
||||||
|
// In Object.h:
|
||||||
|
class Object {
|
||||||
|
protected:
|
||||||
|
virtual void InitProperties() {} // override in derived
|
||||||
|
public:
|
||||||
|
const std::vector<PropertyBase*>& GetProperties() const {
|
||||||
|
if (!m_propertiesInitialized) {
|
||||||
|
const_cast<Object*>(this)->m_propertiesInitialized = true;
|
||||||
|
const_cast<Object*>(this)->InitProperties();
|
||||||
|
}
|
||||||
|
return m_properties;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
Then a CRTP base handles the rest without any macro:
|
||||||
|
|
||||||
|
|
||||||
|
template<typename Derived>
|
||||||
|
class PropertyObject : public Object {
|
||||||
|
protected:
|
||||||
|
void InitProperties() override {
|
||||||
|
uLib::Archive::property_register_archive ar(this);
|
||||||
|
ar & *static_cast<Derived*>(this);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
Usage — just change the base class:
|
||||||
|
|
||||||
|
|
||||||
|
class ContainerBox : public PropertyObject<ContainerBox>, public TRS { ... };
|
||||||
|
// Nothing else needed — properties activated on first GetProperties() call
|
||||||
|
Tradeoff: Most "automatic" — pure inheritance, no constructor or class-body macros. But requires modifying Object (adding m_propertiesInitialized flag + virtual hook), and lazy init means properties aren't available until first access. Also doesn't work well with multiple inheritance (which TRS likely involves).
|
||||||
|
|
||||||
|
Option 3 — CRTP doesn't work from the base constructor
|
||||||
|
Just to be explicit: a CRTP base that calls ULIB_ACTIVATE_PROPERTIES in its own constructor won't work, because when PropertyObject<Derived>'s constructor runs, the vtable is PropertyObject<Derived>'s — Derived::serialize() hasn't been installed yet. So ar & *self calls Object::serialize() (a no-op).
|
||||||
|
|
||||||
|
Recommendation
|
||||||
|
Option 1 is the least invasive and safest. Add deduplication to RegisterDynamicProperty in Object.cpp to guard against re-registration when hierarchies stack activators, then replace every ULIB_ACTIVATE_PROPERTIES(*this) in constructors with a single ULIB_DECLARE_PROPERTIES(ClassName) at the end of the class body.
|
||||||
|
|
||||||
|
Option 2 is cleaner to use but requires changing the Object interface and has the lazy-init semantic change — only worth it if you want zero-touch activation across the entire framework.
|
||||||
@@ -28,6 +28,8 @@
|
|||||||
|
|
||||||
#include <boost/archive/detail/basic_pointer_iserializer.hpp>
|
#include <boost/archive/detail/basic_pointer_iserializer.hpp>
|
||||||
#include <boost/archive/detail/basic_pointer_oserializer.hpp>
|
#include <boost/archive/detail/basic_pointer_oserializer.hpp>
|
||||||
|
#include <boost/archive/text_oarchive.hpp>
|
||||||
|
#include <cstring>
|
||||||
#include <iostream>
|
#include <iostream>
|
||||||
|
|
||||||
#include <boost/archive/text_iarchive.hpp>
|
#include <boost/archive/text_iarchive.hpp>
|
||||||
@@ -59,6 +61,8 @@ class xml_iarchive;
|
|||||||
class xml_oarchive;
|
class xml_oarchive;
|
||||||
class text_iarchive;
|
class text_iarchive;
|
||||||
class text_oarchive;
|
class text_oarchive;
|
||||||
|
class hrt_iarchive;
|
||||||
|
class hrt_oarchive;
|
||||||
class log_archive;
|
class log_archive;
|
||||||
|
|
||||||
} // namespace Archive
|
} // namespace Archive
|
||||||
@@ -150,7 +154,7 @@ public:
|
|||||||
Archive *This() { return static_cast<Archive *>(this); }
|
Archive *This() { return static_cast<Archive *>(this); }
|
||||||
|
|
||||||
template <class T>
|
template <class T>
|
||||||
const basic_pointer_iserializer *register_type(T * = NULL) {
|
const basic_pointer_iserializer *register_type(T * = nullptr) {
|
||||||
const basic_pointer_iserializer &bpis = boost::serialization::singleton<
|
const basic_pointer_iserializer &bpis = boost::serialization::singleton<
|
||||||
pointer_iserializer<Archive, T>>::get_const_instance();
|
pointer_iserializer<Archive, T>>::get_const_instance();
|
||||||
this->This()->register_basic_serializer(bpis.get_basic_serializer());
|
this->This()->register_basic_serializer(bpis.get_basic_serializer());
|
||||||
@@ -161,15 +165,36 @@ public:
|
|||||||
return *this->This();
|
return *this->This();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
template <class T>
|
||||||
|
Archive &operator>>(const boost::serialization::nvp<T> &t) {
|
||||||
|
this->This()->load_override(t);
|
||||||
|
return *this->This();
|
||||||
|
}
|
||||||
|
|
||||||
|
template <class T>
|
||||||
|
Archive &operator>>(const boost::serialization::hrp<T> &t) {
|
||||||
|
this->This()->load_override(const_cast<boost::serialization::hrp<T> &>(t));
|
||||||
|
return *this->This();
|
||||||
|
}
|
||||||
|
|
||||||
// the & operator
|
// the & operator
|
||||||
template <class T> Archive &operator&(T &t) { return *(this->This()) >> t; }
|
template <class T> Archive &operator&(T &t) { return *(this->This()) >> t; }
|
||||||
|
|
||||||
|
template <class T>
|
||||||
|
Archive &operator&(const boost::serialization::nvp<T> &t) {
|
||||||
|
return *(this->This()) >> t;
|
||||||
|
}
|
||||||
|
|
||||||
|
template <class T>
|
||||||
|
Archive &operator&(const boost::serialization::hrp<T> &t) {
|
||||||
|
return *(this->This()) >> t;
|
||||||
|
}
|
||||||
|
|
||||||
// the == operator
|
// the == operator
|
||||||
template <class T> Archive &operator==(T &t) { return this->operator&(t); }
|
template <class T> Archive &operator==(T &t) { return this->operator&(t); }
|
||||||
|
|
||||||
// the != operator for human readable access
|
// the != operator for human readable access
|
||||||
template <class T> Archive &operator!=(T &t) {
|
template <class T> Archive &operator!=(T &t) {
|
||||||
std::cerr << std::flush << "cauch string: " << t << "\n"; // REMOVE THIS !
|
|
||||||
return *this->This();
|
return *this->This();
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
@@ -191,7 +216,7 @@ public:
|
|||||||
Archive *This() { return static_cast<Archive *>(this); }
|
Archive *This() { return static_cast<Archive *>(this); }
|
||||||
|
|
||||||
template <class T>
|
template <class T>
|
||||||
const basic_pointer_oserializer *register_type(const T * = NULL) {
|
const basic_pointer_oserializer *register_type(const T * = nullptr) {
|
||||||
const basic_pointer_oserializer &bpos = boost::serialization::singleton<
|
const basic_pointer_oserializer &bpos = boost::serialization::singleton<
|
||||||
pointer_oserializer<Archive, T>>::get_const_instance();
|
pointer_oserializer<Archive, T>>::get_const_instance();
|
||||||
this->This()->register_basic_serializer(bpos.get_basic_serializer());
|
this->This()->register_basic_serializer(bpos.get_basic_serializer());
|
||||||
@@ -215,7 +240,6 @@ public:
|
|||||||
|
|
||||||
// the != operator for human readable access
|
// the != operator for human readable access
|
||||||
template <class T> Archive &operator!=(T &t) {
|
template <class T> Archive &operator!=(T &t) {
|
||||||
std::cerr << std::flush << "cauch string: " << t << "\n"; // REMOVE THIS !
|
|
||||||
return *this->This();
|
return *this->This();
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
@@ -240,23 +264,22 @@ template <>
|
|||||||
class interface_oarchive<uLib::Archive::text_oarchive>
|
class interface_oarchive<uLib::Archive::text_oarchive>
|
||||||
: public uLib_interface_oarchive<uLib::Archive::text_oarchive> {};
|
: public uLib_interface_oarchive<uLib::Archive::text_oarchive> {};
|
||||||
|
|
||||||
|
template <>
|
||||||
|
class interface_iarchive<uLib::Archive::hrt_iarchive>
|
||||||
|
: public uLib_interface_iarchive<uLib::Archive::hrt_iarchive> {};
|
||||||
|
|
||||||
|
template <>
|
||||||
|
class interface_oarchive<uLib::Archive::hrt_oarchive>
|
||||||
|
: public uLib_interface_oarchive<uLib::Archive::hrt_oarchive> {};
|
||||||
|
|
||||||
|
template <>
|
||||||
|
class interface_iarchive<uLib::Archive::log_archive>
|
||||||
|
: public uLib_interface_iarchive<uLib::Archive::log_archive> {};
|
||||||
|
|
||||||
template <>
|
template <>
|
||||||
class interface_oarchive<uLib::Archive::log_archive>
|
class interface_oarchive<uLib::Archive::log_archive>
|
||||||
: public uLib_interface_oarchive<uLib::Archive::log_archive> {};
|
: public uLib_interface_oarchive<uLib::Archive::log_archive> {};
|
||||||
|
|
||||||
//// Veritical repetition macro // FINIRE !!!!!!!!!!!!!!!!!!!!!!!!!
|
|
||||||
// #define _DECL_INTERFACE_ARCHIVE_V(vz,vn,vdata) \
|
|
||||||
// template <class TypeSeq> \
|
|
||||||
// struct inherit_nofold<TypeSeq,BOOST_PP_INC(vn)> : \
|
|
||||||
// BOOST_PP_REPEAT(BOOST_PP_INC(vn),_INERIT_NOFOLD_H,~) \
|
|
||||||
// {};
|
|
||||||
|
|
||||||
//// Multiple size declaration //
|
|
||||||
// BOOST_PP_REPEAT(ULIB_CFG_MPL_INERIT_NOFOLD_MAXSIZE,_INERIT_NOFOLD_V,~)
|
|
||||||
|
|
||||||
// #undef _INERIT_NOFOLD_H
|
|
||||||
// #undef _INERIT_NOFOLD_V
|
|
||||||
|
|
||||||
} // namespace detail
|
} // namespace detail
|
||||||
} // namespace archive
|
} // namespace archive
|
||||||
} // namespace boost
|
} // namespace boost
|
||||||
@@ -275,36 +298,6 @@ class interface_oarchive<uLib::Archive::log_archive>
|
|||||||
namespace boost {
|
namespace boost {
|
||||||
namespace archive {
|
namespace archive {
|
||||||
|
|
||||||
// template<class Archive>
|
|
||||||
// inline void load_const_override(Archive & ar, const char *t ){
|
|
||||||
// typedef typename mpl::identity<detail::load_non_pointer_type<Archive>
|
|
||||||
// >::type typex; typex::invoke(ar, t);
|
|
||||||
// }
|
|
||||||
|
|
||||||
// template<class Archive, class T>
|
|
||||||
// inline void load(Archive & ar, T &t){
|
|
||||||
// // if this assertion trips. It means we're trying to load a
|
|
||||||
// // const object with a compiler that doesn't have correct
|
|
||||||
// // funtion template ordering. On other compilers, this is
|
|
||||||
// // handled below.
|
|
||||||
// // detail::check_const_loading< T >();
|
|
||||||
// typedef
|
|
||||||
// BOOST_DEDUCED_TYPENAME mpl::eval_if<is_pointer< T >,
|
|
||||||
// mpl::identity<detail::load_pointer_type<Archive> >
|
|
||||||
// ,//else
|
|
||||||
// BOOST_DEDUCED_TYPENAME mpl::eval_if<is_array< T >,
|
|
||||||
// mpl::identity<detail::load_array_type<Archive> >
|
|
||||||
// ,//else
|
|
||||||
// BOOST_DEDUCED_TYPENAME mpl::eval_if<is_enum< T >,
|
|
||||||
// mpl::identity<detail::load_enum_type<Archive> >
|
|
||||||
// ,//else
|
|
||||||
// mpl::identity<detail::load_non_pointer_type<Archive> >
|
|
||||||
// >
|
|
||||||
// >
|
|
||||||
// >::type typex;
|
|
||||||
// typex::invoke(ar, t);
|
|
||||||
// }
|
|
||||||
|
|
||||||
} // namespace archive
|
} // namespace archive
|
||||||
} // namespace boost
|
} // namespace boost
|
||||||
|
|
||||||
@@ -312,40 +305,38 @@ namespace uLib {
|
|||||||
|
|
||||||
namespace Archive {
|
namespace Archive {
|
||||||
|
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
|
||||||
// POLYMORPHIC //
|
|
||||||
|
|
||||||
// class polymorphic_iarchive :
|
|
||||||
// public boost::archive::polymorphic_iarchive {
|
|
||||||
|
|
||||||
// public:
|
|
||||||
// void load_override(const char *t, BOOST_PFTO int)
|
|
||||||
// {
|
|
||||||
// boost::archive::load_const_override(* this->This(),
|
|
||||||
// const_cast<char*>(t));
|
|
||||||
// }
|
|
||||||
|
|
||||||
//};
|
|
||||||
|
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
// XML //
|
// XML //
|
||||||
|
|
||||||
|
// ULIB_SERIALIZATION_VERSION should be get from the build system
|
||||||
|
#ifndef ULIB_SERIALIZATION_VERSION
|
||||||
|
#define ULIB_SERIALIZATION_VERSION "0.0"
|
||||||
|
#endif
|
||||||
|
|
||||||
class xml_iarchive : public boost::archive::xml_iarchive_impl<xml_iarchive> {
|
class xml_iarchive : public boost::archive::xml_iarchive_impl<xml_iarchive> {
|
||||||
typedef xml_iarchive Archive;
|
typedef xml_iarchive Archive;
|
||||||
typedef boost::archive::xml_iarchive_impl<Archive> base;
|
typedef boost::archive::xml_iarchive_impl<Archive> base;
|
||||||
|
|
||||||
|
unsigned int m_flags;
|
||||||
|
|
||||||
// give serialization implementation access to this class
|
// give serialization implementation access to this class
|
||||||
friend class boost::archive::detail::interface_iarchive<Archive>;
|
friend class boost::archive::detail::interface_iarchive<Archive>;
|
||||||
friend class boost::archive::basic_xml_iarchive<Archive>;
|
friend class boost::archive::basic_xml_iarchive<Archive>;
|
||||||
friend class boost::archive::load_access;
|
friend class boost::archive::load_access;
|
||||||
|
|
||||||
public:
|
public:
|
||||||
xml_iarchive(std::istream &is, unsigned int flags = 0)
|
xml_iarchive(std::istream &is, unsigned int flags = 0)
|
||||||
: xml_iarchive_impl<xml_iarchive>(is, flags) {}
|
: boost::archive::xml_iarchive_impl<xml_iarchive>(
|
||||||
|
is, flags | boost::archive::no_header), m_flags(flags) {
|
||||||
|
if (0 == (flags & boost::archive::no_header)) {
|
||||||
|
std::string line;
|
||||||
|
std::getline(is, line); // <?xml ... ?>
|
||||||
|
std::getline(is, line); // <!DOCTYPE ...>
|
||||||
|
std::getline(is, line); // <ulib_serialization ...>
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
using basic_xml_iarchive::load_override;
|
using basic_xml_iarchive::load_override;
|
||||||
|
|
||||||
@@ -373,13 +364,9 @@ public:
|
|||||||
// sr >> str;
|
// sr >> str;
|
||||||
}
|
}
|
||||||
|
|
||||||
~xml_iarchive() {};
|
virtual ~xml_iarchive() {}
|
||||||
};
|
};
|
||||||
|
|
||||||
// typedef boost::archive::detail::polymorphic_iarchive_route<
|
|
||||||
// boost::archive::xml_iarchive_impl<xml_iarchive>
|
|
||||||
//> polymorphic_xml_iarchive;
|
|
||||||
|
|
||||||
template <class ArchiveImpl>
|
template <class ArchiveImpl>
|
||||||
struct polymorphic_iarchive_route
|
struct polymorphic_iarchive_route
|
||||||
: boost::archive::detail::polymorphic_iarchive_route<ArchiveImpl> {
|
: boost::archive::detail::polymorphic_iarchive_route<ArchiveImpl> {
|
||||||
@@ -389,31 +376,39 @@ struct polymorphic_iarchive_route
|
|||||||
class polymorphic_xml_iarchive
|
class polymorphic_xml_iarchive
|
||||||
: public polymorphic_iarchive_route<
|
: public polymorphic_iarchive_route<
|
||||||
boost::archive::xml_iarchive_impl<xml_iarchive>> {
|
boost::archive::xml_iarchive_impl<xml_iarchive>> {
|
||||||
// give serialization implementation access to this class
|
|
||||||
// friend class boost::archive::detail::interface_iarchive<Archive>;
|
|
||||||
// friend class boost::archive::basic_xml_iarchive<Archive>;
|
|
||||||
// friend class boost::archive::load_access;
|
|
||||||
public:
|
public:
|
||||||
virtual void load_override(const char *str) { ; }
|
virtual void load_override(const char *str) {}
|
||||||
};
|
};
|
||||||
|
|
||||||
class xml_oarchive : public boost::archive::xml_oarchive_impl<xml_oarchive> {
|
class xml_oarchive : public boost::archive::xml_oarchive_impl<xml_oarchive> {
|
||||||
typedef xml_oarchive Archive;
|
typedef xml_oarchive Archive;
|
||||||
typedef boost::archive::xml_oarchive_impl<Archive> base;
|
typedef boost::archive::xml_oarchive_impl<Archive> base;
|
||||||
|
|
||||||
|
unsigned int m_flags;
|
||||||
|
|
||||||
// give serialization implementation access to this class
|
// give serialization implementation access to this class
|
||||||
friend class boost::archive::detail::interface_oarchive<Archive>;
|
friend class boost::archive::detail::interface_oarchive<Archive>;
|
||||||
friend class boost::archive::basic_xml_oarchive<Archive>;
|
friend class boost::archive::basic_xml_oarchive<Archive>;
|
||||||
friend class boost::archive::save_access;
|
friend class boost::archive::save_access;
|
||||||
|
|
||||||
public:
|
public:
|
||||||
xml_oarchive(std::ostream &os, unsigned int flags = 0)
|
xml_oarchive(std::ostream &os, unsigned int flags = 0)
|
||||||
: boost::archive::xml_oarchive_impl<xml_oarchive>(os, flags) {}
|
: boost::archive::xml_oarchive_impl<xml_oarchive>(
|
||||||
|
os, flags | boost::archive::no_header), m_flags(flags) {
|
||||||
|
if (0 == (flags & boost::archive::no_header)) {
|
||||||
|
this->This()->put(
|
||||||
|
"<?xml version=\"1.0\" encoding=\"UTF-8\" standalone=\"yes\" ?>\n");
|
||||||
|
this->This()->put("<!DOCTYPE ulib_serialization>\n");
|
||||||
|
this->This()->put("<ulib_serialization signature=\"serialization::archive\" ");
|
||||||
|
this->write_attribute("version", (const char *)ULIB_SERIALIZATION_VERSION);
|
||||||
|
this->This()->put(">\n");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// example of implementing save_override for const char* //
|
virtual ~xml_oarchive() {
|
||||||
// void save_override(const char *t, int) {
|
if (0 == (m_flags & boost::archive::no_header)) {
|
||||||
// std::cout << "found char: " << t << "\n";
|
this->This()->put("</ulib_serialization>\n");
|
||||||
// }
|
}
|
||||||
|
}
|
||||||
|
|
||||||
using basic_xml_oarchive::save_override;
|
using basic_xml_oarchive::save_override;
|
||||||
|
|
||||||
@@ -435,8 +430,6 @@ public:
|
|||||||
// Do not save any human decoration string //
|
// Do not save any human decoration string //
|
||||||
// basic_text_oprimitive::save(str);
|
// basic_text_oprimitive::save(str);
|
||||||
}
|
}
|
||||||
|
|
||||||
~xml_oarchive() {}
|
|
||||||
};
|
};
|
||||||
|
|
||||||
// typedef boost::archive::detail::polymorphic_oarchive_route<
|
// typedef boost::archive::detail::polymorphic_oarchive_route<
|
||||||
@@ -471,15 +464,11 @@ public:
|
|||||||
sr >> str;
|
sr >> str;
|
||||||
}
|
}
|
||||||
|
|
||||||
~text_iarchive() {};
|
virtual ~text_iarchive() {}
|
||||||
};
|
};
|
||||||
|
|
||||||
typedef text_iarchive naked_text_iarchive;
|
typedef text_iarchive naked_text_iarchive;
|
||||||
|
|
||||||
// typedef boost::archive::detail::polymorphic_iarchive_route<
|
|
||||||
// naked_text_iarchive
|
|
||||||
//> polymorphic_text_iarchive;
|
|
||||||
|
|
||||||
class text_oarchive : public boost::archive::text_oarchive_impl<text_oarchive> {
|
class text_oarchive : public boost::archive::text_oarchive_impl<text_oarchive> {
|
||||||
typedef text_oarchive Archive;
|
typedef text_oarchive Archive;
|
||||||
typedef boost::archive::text_oarchive_impl<Archive> base;
|
typedef boost::archive::text_oarchive_impl<Archive> base;
|
||||||
@@ -497,13 +486,9 @@ public:
|
|||||||
|
|
||||||
void save_override(const char *str) { basic_text_oprimitive::save(str); }
|
void save_override(const char *str) { basic_text_oprimitive::save(str); }
|
||||||
|
|
||||||
~text_oarchive() {}
|
virtual ~text_oarchive() {}
|
||||||
};
|
};
|
||||||
|
|
||||||
// typedef boost::archive::detail::polymorphic_oarchive_route<
|
|
||||||
// boost::archive::text_oarchive_impl<text_oarchive>
|
|
||||||
//> polymorphic_text_oarchive;
|
|
||||||
|
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
@@ -540,7 +525,7 @@ public:
|
|||||||
sr >> str;
|
sr >> str;
|
||||||
}
|
}
|
||||||
|
|
||||||
~hrt_iarchive() {};
|
virtual ~hrt_iarchive() {}
|
||||||
};
|
};
|
||||||
|
|
||||||
class hrt_oarchive : public boost::archive::text_oarchive_impl<hrt_oarchive> {
|
class hrt_oarchive : public boost::archive::text_oarchive_impl<hrt_oarchive> {
|
||||||
@@ -576,7 +561,7 @@ public:
|
|||||||
*this << "\n";
|
*this << "\n";
|
||||||
}
|
}
|
||||||
|
|
||||||
~hrt_oarchive() {}
|
virtual ~hrt_oarchive() {}
|
||||||
};
|
};
|
||||||
|
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
@@ -605,17 +590,9 @@ public:
|
|||||||
}
|
}
|
||||||
|
|
||||||
template <class T> void save_override(const T &t) {
|
template <class T> void save_override(const T &t) {
|
||||||
base::save_override(boost::serialization::make_nvp(NULL, t));
|
base::save_override(boost::serialization::make_nvp(nullptr, t));
|
||||||
}
|
}
|
||||||
|
|
||||||
// activate this if you want to trap non nvp objects //
|
|
||||||
// template<class T>
|
|
||||||
// void save_override(T & t)
|
|
||||||
// {
|
|
||||||
// BOOST_MPL_ASSERT((boost::serialization::is_wrapper< T >));
|
|
||||||
// // this->detail_common_oarchive::save_override(t);
|
|
||||||
// }
|
|
||||||
|
|
||||||
template <class T> void save_override(const boost::serialization::nvp<T> &t) {
|
template <class T> void save_override(const boost::serialization::nvp<T> &t) {
|
||||||
base::save_override(t);
|
base::save_override(t);
|
||||||
}
|
}
|
||||||
@@ -640,11 +617,9 @@ public:
|
|||||||
log_archive(std::ostream &os, unsigned int flags = 0)
|
log_archive(std::ostream &os, unsigned int flags = 0)
|
||||||
: boost::archive::xml_oarchive_impl<log_archive>(
|
: boost::archive::xml_oarchive_impl<log_archive>(
|
||||||
os, flags | boost::archive::no_header) {}
|
os, flags | boost::archive::no_header) {}
|
||||||
};
|
|
||||||
|
|
||||||
// typedef boost::archive::detail::polymorphic_oarchive_route<
|
virtual ~log_archive() {}
|
||||||
// boost::archive::xml_oarchive_impl<log_archive>
|
};
|
||||||
//> polymorphic_log_archive;
|
|
||||||
|
|
||||||
} // namespace Archive
|
} // namespace Archive
|
||||||
|
|
||||||
@@ -658,10 +633,4 @@ ULIB_SERIALIZATION_REGISTER_ARCHIVE(uLib::Archive::hrt_iarchive)
|
|||||||
ULIB_SERIALIZATION_REGISTER_ARCHIVE(uLib::Archive::hrt_oarchive)
|
ULIB_SERIALIZATION_REGISTER_ARCHIVE(uLib::Archive::hrt_oarchive)
|
||||||
ULIB_SERIALIZATION_REGISTER_ARCHIVE(uLib::Archive::log_archive)
|
ULIB_SERIALIZATION_REGISTER_ARCHIVE(uLib::Archive::log_archive)
|
||||||
|
|
||||||
// ULIB_SERIALIZATION_REGISTER_ARCHIVE(uLib::Archive::polymorphic_xml_iarchive)
|
|
||||||
// ULIB_SERIALIZATION_REGISTER_ARCHIVE(uLib::Archive::polymorphic_xml_oarchive)
|
|
||||||
// ULIB_SERIALIZATION_REGISTER_ARCHIVE(uLib::Archive::polymorphic_text_iarchive)
|
|
||||||
// ULIB_SERIALIZATION_REGISTER_ARCHIVE(uLib::Archive::polymorphic_text_oarchive)
|
|
||||||
// ULIB_SERIALIZATION_REGISTER_ARCHIVE(uLib::Archive::polymorphic_log_archive)
|
|
||||||
|
|
||||||
#endif // U_CORE_ARCHIVES_H
|
#endif // U_CORE_ARCHIVES_H
|
||||||
|
|||||||
@@ -58,6 +58,7 @@ if(USE_CUDA)
|
|||||||
endif()
|
endif()
|
||||||
|
|
||||||
target_link_libraries(${libname} ${LIBRARIES})
|
target_link_libraries(${libname} ${LIBRARIES})
|
||||||
|
target_compile_definitions(${libname} PUBLIC ULIB_SERIALIZATION_VERSION="${PROJECT_VERSION}")
|
||||||
|
|
||||||
install(TARGETS ${libname}
|
install(TARGETS ${libname}
|
||||||
EXPORT "uLibTargets"
|
EXPORT "uLibTargets"
|
||||||
|
|||||||
@@ -38,6 +38,76 @@
|
|||||||
#include <boost/any.hpp>
|
#include <boost/any.hpp>
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @file Debug.h
|
||||||
|
* @brief Unified Debugging and Monitoring System for uLib.
|
||||||
|
*
|
||||||
|
* # Debug System Documentation
|
||||||
|
*
|
||||||
|
* The `Debug` system provides a flexible, adapter-based mechanism for monitoring
|
||||||
|
* and outputting internal variables and states without hardcoding output logic
|
||||||
|
* into core classes.
|
||||||
|
*
|
||||||
|
* ## Architecture
|
||||||
|
*
|
||||||
|
* The system follows an adapter pattern:
|
||||||
|
*
|
||||||
|
* - **`DebugAdapterInterface`**: The base interface for all output targets.
|
||||||
|
* Subclasses define how primitive types and strings are handled (e.g., printing
|
||||||
|
* to `std::cout`, writing to a log file, or updating a real-time UI widget).
|
||||||
|
*
|
||||||
|
* - **`Debug` Class**: The central registry. It stores:
|
||||||
|
* 1. A list of "Adapters" (`DebugAdapterInterface`).
|
||||||
|
* 2. A list of "Items" to monitor. Each item is a reference to a variable
|
||||||
|
* associated with a name.
|
||||||
|
*
|
||||||
|
* - **Type Safety**: Although variables are stored using `boost::any` (type erasure),
|
||||||
|
* the system automatically preserves the original type information via
|
||||||
|
* internal template adapters (`AnyCastAdapter`), ensuring that the correct
|
||||||
|
* overload of the adapter interface is called.
|
||||||
|
*
|
||||||
|
* ## Core Components
|
||||||
|
*
|
||||||
|
* | Component | Description |
|
||||||
|
* | :--- | :--- |
|
||||||
|
* | `Debug` | The main controller used to add adapters and register variables. |
|
||||||
|
* | `DebugAdapterInterface` | Virtual base for creating new output methods. |
|
||||||
|
* | `DebugAdapterText` | A simple built-in adapter for `std::ostream` output. |
|
||||||
|
*
|
||||||
|
* ## How to Use
|
||||||
|
*
|
||||||
|
* ### 1. Initialize the Debug Object
|
||||||
|
* ```cpp
|
||||||
|
* uLib::Debug dbg;
|
||||||
|
* ```
|
||||||
|
*
|
||||||
|
* ### 2. Add an Adapter
|
||||||
|
* ```cpp
|
||||||
|
* uLib::DebugAdapterText console(std::cout);
|
||||||
|
* dbg.AddAdapter(console);
|
||||||
|
* ```
|
||||||
|
*
|
||||||
|
* ### 3. Register Variables to Monitor
|
||||||
|
* Use the `operator()` to bind a variable by reference.
|
||||||
|
* ```cpp
|
||||||
|
* int frameCount = 0;
|
||||||
|
* std::string state = "Initializing";
|
||||||
|
* dbg("Frames", frameCount);
|
||||||
|
* dbg("Status", state);
|
||||||
|
* ```
|
||||||
|
*
|
||||||
|
* ### 4. Update
|
||||||
|
* Call `Update()` periodically (e.g., once per frame) to push the current
|
||||||
|
* values of all registered variables to all connected adapters.
|
||||||
|
* ```cpp
|
||||||
|
* while(running) {
|
||||||
|
* frameCount++;
|
||||||
|
* dbg.Update(); // This triggers the output
|
||||||
|
* }
|
||||||
|
* ```
|
||||||
|
*/
|
||||||
|
|
||||||
namespace uLib {
|
namespace uLib {
|
||||||
|
|
||||||
|
|
||||||
@@ -107,7 +177,6 @@ public:
|
|||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
class Debug {
|
class Debug {
|
||||||
typedef detail::DebugAdapterInterface AdapterInterface;
|
typedef detail::DebugAdapterInterface AdapterInterface;
|
||||||
typedef SmartPointer<detail::DebugAdapterInterface> Adapter;
|
typedef SmartPointer<detail::DebugAdapterInterface> Adapter;
|
||||||
|
|||||||
@@ -79,6 +79,7 @@ void Object::RegisterDynamicProperty(PropertyBase* prop) {
|
|||||||
if (prop) {
|
if (prop) {
|
||||||
for (auto* existing : d->m_DynamicProperties) {
|
for (auto* existing : d->m_DynamicProperties) {
|
||||||
if (existing == prop) return;
|
if (existing == prop) return;
|
||||||
|
if (existing->GetQualifiedName() == prop->GetQualifiedName()) return;
|
||||||
}
|
}
|
||||||
d->m_DynamicProperties.push_back(prop);
|
d->m_DynamicProperties.push_back(prop);
|
||||||
}
|
}
|
||||||
@@ -98,6 +99,11 @@ PropertyBase* Object::GetProperty(const std::string& name) const {
|
|||||||
return nullptr;
|
return nullptr;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
void Object::NotifyPropertiesUpdated() {
|
||||||
|
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.
|
// 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.
|
// However, since Object::serialize is a template in the header, we might need a helper here.
|
||||||
|
|
||||||
@@ -110,6 +116,7 @@ void Object::serialize(ArchiveT &ar, const unsigned int version) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
void Object::Updated() { ULIB_SIGNAL_EMIT(Object::Updated); }
|
void Object::Updated() { ULIB_SIGNAL_EMIT(Object::Updated); }
|
||||||
|
void Object::PropertyUpdated() { ULIB_SIGNAL_EMIT(Object::PropertyUpdated); }
|
||||||
|
|
||||||
template <class ArchiveT>
|
template <class ArchiveT>
|
||||||
void Object::save_override(ArchiveT &ar, const unsigned int version) {}
|
void Object::save_override(ArchiveT &ar, const unsigned int version) {}
|
||||||
|
|||||||
@@ -78,7 +78,8 @@ public:
|
|||||||
Object(const Object ©);
|
Object(const Object ©);
|
||||||
virtual ~Object();
|
virtual ~Object();
|
||||||
|
|
||||||
virtual const char * GetClassName() const { return "Object"; }
|
virtual const char * GetClassName() const { return type_name(); }
|
||||||
|
virtual const char * type_name() const { return "Object"; }
|
||||||
|
|
||||||
const std::string& GetInstanceName() const;
|
const std::string& GetInstanceName() const;
|
||||||
void SetInstanceName(const std::string& name);
|
void SetInstanceName(const std::string& name);
|
||||||
@@ -96,6 +97,9 @@ public:
|
|||||||
const std::vector<PropertyBase*>& GetProperties() const;
|
const std::vector<PropertyBase*>& GetProperties() const;
|
||||||
PropertyBase* GetProperty(const std::string& name) 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. */
|
||||||
|
void NotifyPropertiesUpdated();
|
||||||
|
|
||||||
////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////
|
||||||
// PARAMETERS //
|
// PARAMETERS //
|
||||||
|
|
||||||
@@ -133,6 +137,7 @@ public:
|
|||||||
|
|
||||||
signals:
|
signals:
|
||||||
virtual void Updated();
|
virtual void Updated();
|
||||||
|
virtual void PropertyUpdated();
|
||||||
|
|
||||||
// Qt4 style connector //
|
// Qt4 style connector //
|
||||||
static bool connect(const Object *ob1, const char *signal_name,
|
static bool connect(const Object *ob1, const char *signal_name,
|
||||||
|
|||||||
@@ -14,7 +14,7 @@ public:
|
|||||||
ObjectsContext();
|
ObjectsContext();
|
||||||
virtual ~ObjectsContext();
|
virtual ~ObjectsContext();
|
||||||
|
|
||||||
virtual const char * GetClassName() const { return "ObjectsContext"; }
|
uLibTypeMacro(ObjectsContext, Object)
|
||||||
virtual ObjectsContext* GetChildren() override { return this; }
|
virtual ObjectsContext* GetChildren() override { return this; }
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
|||||||
@@ -42,6 +42,7 @@ public:
|
|||||||
virtual double GetMax() const { return 0; }
|
virtual double GetMax() const { return 0; }
|
||||||
virtual bool HasDefault() const { return false; }
|
virtual bool HasDefault() const { return false; }
|
||||||
virtual std::string GetDefaultValueAsString() const { return ""; }
|
virtual std::string GetDefaultValueAsString() const { return ""; }
|
||||||
|
virtual bool IsReadOnly() const = 0;
|
||||||
|
|
||||||
std::string GetQualifiedName() const {
|
std::string GetQualifiedName() const {
|
||||||
if (GetGroup().empty()) return GetName();
|
if (GetGroup().empty()) return GetName();
|
||||||
@@ -53,15 +54,22 @@ public:
|
|||||||
virtual void Updated() override { ULIB_SIGNAL_EMIT(PropertyBase::Updated); }
|
virtual void Updated() override { ULIB_SIGNAL_EMIT(PropertyBase::Updated); }
|
||||||
|
|
||||||
// Serialization support for different uLib archives
|
// Serialization support for different uLib archives
|
||||||
virtual void serialize(Archive::xml_oarchive & ar, const unsigned int version) = 0;
|
virtual void serialize(Archive::xml_oarchive & ar, const unsigned int version) override = 0;
|
||||||
virtual void serialize(Archive::xml_iarchive & ar, const unsigned int version) = 0;
|
virtual void serialize(Archive::xml_iarchive & ar, const unsigned int version) override = 0;
|
||||||
virtual void serialize(Archive::text_oarchive & ar, const unsigned int version) = 0;
|
virtual void serialize(Archive::text_oarchive & ar, const unsigned int version) override = 0;
|
||||||
virtual void serialize(Archive::text_iarchive & ar, const unsigned int version) = 0;
|
virtual void serialize(Archive::text_iarchive & ar, const unsigned int version) override = 0;
|
||||||
virtual void serialize(Archive::hrt_oarchive & ar, const unsigned int version) = 0;
|
virtual void serialize(Archive::hrt_oarchive & ar, const unsigned int version) override = 0;
|
||||||
virtual void serialize(Archive::hrt_iarchive & ar, const unsigned int version) = 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) = 0;
|
virtual void serialize(Archive::log_archive & ar, const unsigned int version) override = 0;
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* @brief Template class for typed properties.
|
* @brief Template class for typed properties.
|
||||||
*/
|
*/
|
||||||
@@ -71,7 +79,7 @@ public:
|
|||||||
// PROXY: Use an existing variable as back-end storage
|
// 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 = "")
|
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_owner(owner), m_name(name), m_units(units), m_group(group), m_value(valuePtr), m_own(false),
|
||||||
m_HasRange(false), m_HasDefault(false) {
|
m_HasRange(false), m_HasDefault(false), m_ReadOnly(false) {
|
||||||
if (m_owner) {
|
if (m_owner) {
|
||||||
m_owner->RegisterProperty(this);
|
m_owner->RegisterProperty(this);
|
||||||
}
|
}
|
||||||
@@ -80,7 +88,7 @@ public:
|
|||||||
// MANAGED: Create and own internal storage
|
// 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 = "")
|
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_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_HasRange(false), m_HasDefault(true), m_Default(defaultValue), m_ReadOnly(false) {
|
||||||
if (m_owner) {
|
if (m_owner) {
|
||||||
m_owner->RegisterProperty(this);
|
m_owner->RegisterProperty(this);
|
||||||
}
|
}
|
||||||
@@ -139,6 +147,9 @@ public:
|
|||||||
|
|
||||||
void SetRange(const T& min, const T& max) { m_Min = min; m_Max = max; m_HasRange = true; }
|
void SetRange(const T& min, const T& max) { m_Min = min; m_Max = max; m_HasRange = true; }
|
||||||
void SetDefault(const T& def) { m_Default = def; m_HasDefault = true; }
|
void SetDefault(const T& def) { m_Default = def; m_HasDefault = true; }
|
||||||
|
void SetReadOnly(bool ro) { m_ReadOnly = ro; }
|
||||||
|
|
||||||
|
virtual bool IsReadOnly() const override { return m_ReadOnly; }
|
||||||
|
|
||||||
virtual bool HasRange() const override { return m_HasRange; }
|
virtual bool HasRange() const override { return m_HasRange; }
|
||||||
|
|
||||||
@@ -209,6 +220,7 @@ private:
|
|||||||
T m_Max;
|
T m_Max;
|
||||||
bool m_HasDefault;
|
bool m_HasDefault;
|
||||||
T m_Default;
|
T m_Default;
|
||||||
|
bool m_ReadOnly;
|
||||||
};
|
};
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@@ -223,6 +235,7 @@ typedef Property<float> FloatProperty;
|
|||||||
typedef Property<double> DoubleProperty;
|
typedef Property<double> DoubleProperty;
|
||||||
typedef Property<Bool_t> BoolProperty;
|
typedef Property<Bool_t> BoolProperty;
|
||||||
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* @brief Property specialized for enumerations, providing labels for GUI representations.
|
* @brief Property specialized for enumerations, providing labels for GUI representations.
|
||||||
*/
|
*/
|
||||||
@@ -239,6 +252,16 @@ private:
|
|||||||
std::vector<std::string> m_Labels;
|
std::vector<std::string> m_Labels;
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* @brief Macro to simplify property declaration within a class.
|
* @brief Macro to simplify property declaration within a class.
|
||||||
* Usage: ULIB_PROPERTY(float, Width, 1.0f)
|
* Usage: ULIB_PROPERTY(float, Width, 1.0f)
|
||||||
@@ -304,6 +327,25 @@ public:
|
|||||||
Property<T>* p = new Property<T>(m_Object, t.name(), &const_cast<boost::serialization::hrp<T>&>(t).value(), t.units() ? t.units() : "", GetCurrentGroup());
|
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_range()) p->SetRange(t.min_val(), t.max_val());
|
||||||
if (t.has_default()) p->SetDefault(t.default_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_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);
|
m_Object->RegisterDynamicProperty(p);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -312,6 +354,16 @@ public:
|
|||||||
void save_override(const boost::serialization::hrp_enum<T> &t) {
|
void save_override(const boost::serialization::hrp_enum<T> &t) {
|
||||||
if (m_Object) {
|
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());
|
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());
|
||||||
|
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());
|
||||||
m_Object->RegisterDynamicProperty(p);
|
m_Object->RegisterDynamicProperty(p);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -352,12 +404,34 @@ private:
|
|||||||
std::vector<std::string> m_GroupStack;
|
std::vector<std::string> m_GroupStack;
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* @brief Convenience macro to automatically activate and register all HRP members
|
* @brief Convenience macro to automatically activate and register all HRP members
|
||||||
* as uLib properties. Usage: ULIB_ACTIVATE_PROPERTIES(obj)
|
* as uLib properties. Usage: ULIB_ACTIVATE_PROPERTIES(obj)
|
||||||
*/
|
*/
|
||||||
#define ULIB_ACTIVATE_PROPERTIES(obj) \
|
#define ULIB_ACTIVATE_PROPERTIES(obj) \
|
||||||
{ uLib::Archive::property_register_archive _ar_tmp(&(obj)); (obj).serialize(_ar_tmp, 0); }
|
{ 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 { \
|
||||||
|
_PropActivator(SelfType* self) { \
|
||||||
|
uLib::Archive::property_register_archive _ar(self); \
|
||||||
|
_ar & *self; \
|
||||||
|
} \
|
||||||
|
} _prop_activator{this};
|
||||||
|
|
||||||
} // namespace Archive
|
} // namespace Archive
|
||||||
} // namespace uLib
|
} // namespace uLib
|
||||||
|
|||||||
@@ -71,15 +71,16 @@ namespace serialization {
|
|||||||
// ACCESS 2 //
|
// ACCESS 2 //
|
||||||
template <class T> struct access2 {};
|
template <class T> struct access2 {};
|
||||||
|
|
||||||
// NON FUNZIONA ... SISTEMARE !!!! // ------------------------------------------
|
|
||||||
template <class T>
|
template <class T>
|
||||||
class hrp : public boost::serialization::wrapper_traits<hrp<T>> {
|
class hrp : public boost::serialization::wrapper_traits<hrp<T>> {
|
||||||
|
|
||||||
const char *m_name;
|
const char *m_name;
|
||||||
const char *m_units;
|
const char *m_units;
|
||||||
T &m_value;
|
T &m_value;
|
||||||
|
|
||||||
bool m_has_range;
|
bool m_has_range;
|
||||||
T m_min;
|
T m_min, m_max;
|
||||||
T m_max;
|
|
||||||
bool m_has_default;
|
bool m_has_default;
|
||||||
T m_default;
|
T m_default;
|
||||||
|
|
||||||
@@ -101,6 +102,8 @@ public:
|
|||||||
bool has_default() const { return m_has_default; }
|
bool has_default() const { return m_has_default; }
|
||||||
const T& default_val() const { return m_default; }
|
const T& default_val() const { return m_default; }
|
||||||
|
|
||||||
|
static constexpr bool is_read_only() { return false; }
|
||||||
|
|
||||||
BOOST_SERIALIZATION_SPLIT_MEMBER()
|
BOOST_SERIALIZATION_SPLIT_MEMBER()
|
||||||
|
|
||||||
template <class Archivex>
|
template <class Archivex>
|
||||||
@@ -114,11 +117,6 @@ public:
|
|||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
template <class T>
|
|
||||||
inline hrp<T> make_hrp(const char *name, T &t, const char* units = nullptr) {
|
|
||||||
return hrp<T>(name, t, units);
|
|
||||||
}
|
|
||||||
|
|
||||||
template <class T>
|
template <class T>
|
||||||
class hrp_enum : public boost::serialization::wrapper_traits<hrp_enum<T>> {
|
class hrp_enum : public boost::serialization::wrapper_traits<hrp_enum<T>> {
|
||||||
const char *m_name;
|
const char *m_name;
|
||||||
@@ -142,6 +140,8 @@ public:
|
|||||||
bool has_default() const { return m_has_default; }
|
bool has_default() const { return m_has_default; }
|
||||||
const T& default_val() const { return m_default; }
|
const T& default_val() const { return m_default; }
|
||||||
|
|
||||||
|
static constexpr bool is_read_only() { return false; }
|
||||||
|
|
||||||
BOOST_SERIALIZATION_SPLIT_MEMBER()
|
BOOST_SERIALIZATION_SPLIT_MEMBER()
|
||||||
|
|
||||||
template <class Archivex>
|
template <class Archivex>
|
||||||
@@ -155,13 +155,121 @@ public:
|
|||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
|
template <class T>
|
||||||
|
class hrp_val : public boost::serialization::wrapper_traits<hrp_val<T>> {
|
||||||
|
const char *m_name;
|
||||||
|
const char *m_units;
|
||||||
|
T m_value;
|
||||||
|
bool m_has_range;
|
||||||
|
T m_min, m_max;
|
||||||
|
bool m_has_default;
|
||||||
|
T m_default;
|
||||||
|
|
||||||
|
public:
|
||||||
|
explicit hrp_val(const char *name_, T t, const char* units_ = nullptr)
|
||||||
|
: m_name(name_), m_units(units_), m_value(t), m_has_range(false), m_has_default(false) {}
|
||||||
|
|
||||||
|
hrp_val& range(const T& min_val, const T& max_val) { m_min = min_val; m_max = max_val; m_has_range = true; return *this; }
|
||||||
|
hrp_val& set_default(const T& def_val) { m_default = def_val; m_has_default = true; return *this; }
|
||||||
|
|
||||||
|
const char *name() const { return this->m_name; }
|
||||||
|
const char *units() const { return this->m_units; }
|
||||||
|
T &value() { return this->m_value; }
|
||||||
|
const T &const_value() const { return this->m_value; }
|
||||||
|
|
||||||
|
bool has_range() const { return m_has_range; }
|
||||||
|
const T& min_val() const { return m_min; }
|
||||||
|
const T& max_val() const { return m_max; }
|
||||||
|
bool has_default() const { return m_has_default; }
|
||||||
|
const T& default_val() const { return m_default; }
|
||||||
|
|
||||||
|
static constexpr bool is_read_only() { return true; }
|
||||||
|
|
||||||
|
BOOST_SERIALIZATION_SPLIT_MEMBER()
|
||||||
|
|
||||||
|
template <class Archivex>
|
||||||
|
void save(Archivex &ar, const unsigned int /* version */) const {
|
||||||
|
ar << boost::serialization::make_nvp(m_name, m_value);
|
||||||
|
}
|
||||||
|
|
||||||
|
template <class Archivex>
|
||||||
|
void load(Archivex &ar, const unsigned int /* version */) {
|
||||||
|
// Only for output archives
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
template <class T>
|
||||||
|
class hrp_enum_val : public boost::serialization::wrapper_traits<hrp_enum_val<T>> {
|
||||||
|
const char *m_name;
|
||||||
|
const char *m_units;
|
||||||
|
T m_value;
|
||||||
|
std::vector<std::string> m_labels;
|
||||||
|
bool m_has_default;
|
||||||
|
T m_default;
|
||||||
|
|
||||||
|
public:
|
||||||
|
explicit hrp_enum_val(const char *name_, T t, const std::vector<std::string>& labels, const char* units_ = nullptr)
|
||||||
|
: m_name(name_), m_units(units_), m_value(t), m_labels(labels), m_has_default(false) {}
|
||||||
|
|
||||||
|
hrp_enum_val& set_default(const T& def_val) { m_default = def_val; m_has_default = true; return *this; }
|
||||||
|
|
||||||
|
const char *name() const { return this->m_name; }
|
||||||
|
const char *units() const { return this->m_units; }
|
||||||
|
T &value() { return this->m_value; }
|
||||||
|
const std::vector<std::string>& labels() const { return m_labels; }
|
||||||
|
|
||||||
|
bool has_default() const { return m_has_default; }
|
||||||
|
const T& default_val() const { return m_default; }
|
||||||
|
|
||||||
|
static constexpr bool is_read_only() { return true; }
|
||||||
|
|
||||||
|
BOOST_SERIALIZATION_SPLIT_MEMBER()
|
||||||
|
|
||||||
|
template <class Archivex>
|
||||||
|
void save(Archivex &ar, const unsigned int /* version */) const {
|
||||||
|
ar << boost::serialization::make_nvp(m_name, m_value);
|
||||||
|
}
|
||||||
|
|
||||||
|
template <class Archivex>
|
||||||
|
void load(Archivex &ar, const unsigned int /* version */) {
|
||||||
|
// Only for output archives
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
template <class T>
|
||||||
|
inline hrp<T> make_hrp(const char *name, T &t, const char* units = nullptr) {
|
||||||
|
return hrp<T>(name, t, units);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Specialization for rvalues (value-based storage)
|
||||||
|
template <class T>
|
||||||
|
inline hrp_val<T> make_hrp(const char *name, T &&t, const char* units = nullptr) {
|
||||||
|
return hrp_val<T>(name, t, units);
|
||||||
|
}
|
||||||
|
|
||||||
template <class T>
|
template <class T>
|
||||||
inline hrp_enum<T> make_hrp_enum(const char *name, T &t, const std::vector<std::string>& labels, const char* units = nullptr) {
|
inline hrp_enum<T> make_hrp_enum(const char *name, T &t, const std::vector<std::string>& labels, const char* units = nullptr) {
|
||||||
return hrp_enum<T>(name, t, labels, units);
|
return hrp_enum<T>(name, t, labels, units);
|
||||||
}
|
}
|
||||||
|
|
||||||
#define HRP(name) boost::serialization::make_hrp(BOOST_PP_STRINGIZE(name), name)
|
// Specialization for rvalues (value-based storage)
|
||||||
#define HRPU(name, units) boost::serialization::make_hrp(BOOST_PP_STRINGIZE(name), name, units)
|
template <class T>
|
||||||
|
inline hrp_enum_val<T> make_hrp_enum(const char *name, T &&t, const std::vector<std::string>& labels, const char* units = nullptr) {
|
||||||
|
return hrp_enum_val<T>(name, t, labels, units);
|
||||||
|
}
|
||||||
|
|
||||||
|
template <class T>
|
||||||
|
inline hrp<T> make_nvp(const char *name, T &t, const char* units) {
|
||||||
|
return hrp<T>(name, t, units);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Specialization for rvalues (value-based storage)
|
||||||
|
template <class T>
|
||||||
|
inline hrp_val<T> make_nvp(const char *name, T &&t, const char* units) {
|
||||||
|
return hrp_val<T>(name, t, units);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
} // namespace serialization
|
} // namespace serialization
|
||||||
} // namespace boost
|
} // namespace boost
|
||||||
@@ -181,7 +289,41 @@ namespace uLib {
|
|||||||
|
|
||||||
#define _AR_OP(r, data, elem) data &BOOST_SERIALIZATION_BASE_OBJECT_NVP(elem);
|
#define _AR_OP(r, data, elem) data &BOOST_SERIALIZATION_BASE_OBJECT_NVP(elem);
|
||||||
|
|
||||||
#define NVP(data) BOOST_SERIALIZATION_NVP(data)
|
// NAME VALUE PAIR //
|
||||||
|
#define NVP_GET_MACRO(_1, _2, _3, NAME, ...) NAME
|
||||||
|
#define NVP(...) NVP_GET_MACRO(__VA_ARGS__, NVP3, NVP2, NVP1)(__VA_ARGS__)
|
||||||
|
|
||||||
|
#define NVP1(data) BOOST_SERIALIZATION_NVP(data)
|
||||||
|
#define NVP2(name, data) boost::serialization::make_nvp(name, data)
|
||||||
|
#define NVP3(name, data, units) boost::serialization::make_nvp(name, data, units)
|
||||||
|
|
||||||
|
|
||||||
|
// HUMAN READABLE PROPERTY //
|
||||||
|
#define HRP_GET_MACRO(_1, _2, _3, _4, _5, _6, NAME, ...) NAME
|
||||||
|
#define HRP(...) HRP_GET_MACRO(__VA_ARGS__, HRP6, HRP5, HRP4, HRP3, HRP2, HRP1)(__VA_ARGS__)
|
||||||
|
|
||||||
|
#define HRP1(data) boost::serialization::make_hrp(BOOST_PP_STRINGIZE(data), data)
|
||||||
|
#define HRP2(name, data) boost::serialization::make_hrp(name, data)
|
||||||
|
#define HRP3(name, data, units) boost::serialization::make_hrp(name, data, units)
|
||||||
|
#define HRP4(name, data, units, default) boost::serialization::make_hrp(name, data, units).set_default(default)
|
||||||
|
#define HRP5(name, data, units, min, max) boost::serialization::make_hrp(name, data, units).range(min, max)
|
||||||
|
#define HRP6(name, data, units, default, min, max) boost::serialization::make_hrp(name, data, units).set_default(default).range(min, max)
|
||||||
|
|
||||||
|
#define HRPE(name, data, labels) boost::serialization::make_hrp_enum(name, data, labels)
|
||||||
|
|
||||||
|
// LEFT FOR BACKWARD COMPATIBILITY
|
||||||
|
#define HRPU(name, units) boost::serialization::make_hrp(BOOST_PP_STRINGIZE(name), name, units)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
namespace serialization {
|
||||||
|
using boost::serialization::make_nvp;
|
||||||
|
using boost::serialization::make_hrp;
|
||||||
|
using boost::serialization::make_hrp_enum;
|
||||||
|
} // serialization
|
||||||
|
|
||||||
|
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
@@ -209,7 +351,7 @@ namespace uLib {
|
|||||||
#define ULIB_SERIALIZE_OBJECT(_Ob, ...) \
|
#define ULIB_SERIALIZE_OBJECT(_Ob, ...) \
|
||||||
_ULIB_DETAIL_UNINTRUSIVE_SERIALIZE_OBJECT(_Ob, __VA_ARGS__)
|
_ULIB_DETAIL_UNINTRUSIVE_SERIALIZE_OBJECT(_Ob, __VA_ARGS__)
|
||||||
#define AR(_name) _ULIB_DETAIL_UNINTRUSIVE_AR_(_name)
|
#define AR(_name) _ULIB_DETAIL_UNINTRUSIVE_AR_(_name)
|
||||||
#define HR(_name) _ULIB_DETAIL_UNINTRUSIVE_AR_(_name)
|
#define HR(_name) _ULIB_DETAIL_UNINTRUSIVE_HR_(_name)
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
#define ULIB_SERIALIZE_ACCESS \
|
#define ULIB_SERIALIZE_ACCESS \
|
||||||
@@ -223,13 +365,13 @@ namespace uLib {
|
|||||||
BOOST_CLASS_EXPORT_KEY(_FullNamespaceClass)
|
BOOST_CLASS_EXPORT_KEY(_FullNamespaceClass)
|
||||||
|
|
||||||
#define _SERIALIZE_IMPL_SEQ \
|
#define _SERIALIZE_IMPL_SEQ \
|
||||||
(uLib::Archive::text_iarchive)(uLib::Archive::text_oarchive)( \
|
(uLib::Archive::text_iarchive) \
|
||||||
uLib::Archive:: \
|
(uLib::Archive::text_oarchive) \
|
||||||
hrt_iarchive)(uLib::Archive:: \
|
(uLib::Archive::hrt_iarchive) \
|
||||||
hrt_oarchive)(uLib::Archive:: \
|
(uLib::Archive::hrt_oarchive) \
|
||||||
xml_iarchive)(uLib::Archive:: \
|
(uLib::Archive::xml_iarchive) \
|
||||||
xml_oarchive)(uLib::Archive:: \
|
(uLib::Archive::xml_oarchive) \
|
||||||
log_archive)
|
(uLib::Archive::log_archive)
|
||||||
|
|
||||||
/** Solving virtual class check problem */
|
/** Solving virtual class check problem */
|
||||||
#define _ULIB_DETAIL_SPECIALIZE_IS_VIRTUAL_BASE(_Base, _Derived) \
|
#define _ULIB_DETAIL_SPECIALIZE_IS_VIRTUAL_BASE(_Base, _Derived) \
|
||||||
@@ -300,6 +442,11 @@ namespace uLib {
|
|||||||
template <class ArchiveT> \
|
template <class ArchiveT> \
|
||||||
void _Ob::save_override(ArchiveT &ar, const unsigned int version)
|
void _Ob::save_override(ArchiveT &ar, const unsigned int version)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
@@ -404,7 +551,8 @@ namespace uLib {
|
|||||||
void serialize_parents(ArchiveT &ar, _Ob &ob, const unsigned int v) { \
|
void serialize_parents(ArchiveT &ar, _Ob &ob, const unsigned int v) { \
|
||||||
/* PP serialize */ BOOST_PP_SEQ_FOR_EACH( \
|
/* PP serialize */ BOOST_PP_SEQ_FOR_EACH( \
|
||||||
_UNAR_OP, ob, BOOST_PP_TUPLE_TO_SEQ((__VA_ARGS__))); \
|
_UNAR_OP, ob, BOOST_PP_TUPLE_TO_SEQ((__VA_ARGS__))); \
|
||||||
/* MPL serialize */ /*uLib::mpl::for_each<_Ob::BaseList>(uLib::detail::Serializable::serialize_baseobject<_Ob,ArchiveT>(ob,ar) );*/ } \
|
/* MPL serialize */ /*uLib::mpl::for_each<_Ob::BaseList> \
|
||||||
|
(uLib::detail::Serializable::serialize_baseobject<_Ob,ArchiveT>(ob,ar) );*/ }\
|
||||||
template <class ArchiveT> \
|
template <class ArchiveT> \
|
||||||
inline void load_construct_data(ArchiveT &ar, _Ob *ob, \
|
inline void load_construct_data(ArchiveT &ar, _Ob *ob, \
|
||||||
const unsigned int file_version) { \
|
const unsigned int file_version) { \
|
||||||
@@ -429,8 +577,16 @@ namespace uLib {
|
|||||||
void boost::serialization::access2<_Ob>::save_override( \
|
void boost::serialization::access2<_Ob>::save_override( \
|
||||||
ArchiveT &ar, _Ob &ob, const unsigned int version)
|
ArchiveT &ar, _Ob &ob, const unsigned int version)
|
||||||
|
|
||||||
|
|
||||||
#define _ULIB_DETAIL_UNINTRUSIVE_AR_(name) \
|
#define _ULIB_DETAIL_UNINTRUSIVE_AR_(name) \
|
||||||
boost::serialization::make_nvp(BOOST_PP_STRINGIZE(name), ob.name)
|
boost::serialization::make_nvp(BOOST_PP_STRINGIZE(name), ob.name)
|
||||||
|
#define _ULIB_DETAIL_UNINTRUSIVE_HR_(name) \
|
||||||
|
boost::serialization::make_hrp(BOOST_PP_STRINGIZE(name), ob.name)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
|
|||||||
@@ -95,6 +95,8 @@ namespace uLib {
|
|||||||
|
|
||||||
typedef boost::signals2::signal_base SignalBase;
|
typedef boost::signals2::signal_base SignalBase;
|
||||||
typedef boost::signals2::connection Connection;
|
typedef boost::signals2::connection Connection;
|
||||||
|
typedef boost::signals2::shared_connection_block ConnectionBlock;
|
||||||
|
|
||||||
|
|
||||||
template <typename T> struct Signal {
|
template <typename T> struct Signal {
|
||||||
typedef boost::signals2::signal<T> type;
|
typedef boost::signals2::signal<T> type;
|
||||||
|
|||||||
@@ -29,6 +29,7 @@ set( TESTS
|
|||||||
OpenMPTest
|
OpenMPTest
|
||||||
TeamTest
|
TeamTest
|
||||||
AffinityTest
|
AffinityTest
|
||||||
|
ReadOnlyPropertyTest
|
||||||
)
|
)
|
||||||
|
|
||||||
set(LIBRARIES
|
set(LIBRARIES
|
||||||
|
|||||||
@@ -23,74 +23,138 @@
|
|||||||
|
|
||||||
//////////////////////////////////////////////////////////////////////////////*/
|
//////////////////////////////////////////////////////////////////////////////*/
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
#include <iostream>
|
#include <iostream>
|
||||||
|
#include <fstream>
|
||||||
#include <boost/signals2/signal.hpp>
|
#include <string>
|
||||||
|
|
||||||
|
|
||||||
#include "Core/Object.h"
|
#include "Core/Object.h"
|
||||||
|
#include "Core/Property.h"
|
||||||
|
#include "Core/Archives.h"
|
||||||
|
#include "Core/Serializable.h"
|
||||||
#include "testing-prototype.h"
|
#include "testing-prototype.h"
|
||||||
|
|
||||||
#define emit
|
using namespace uLib;
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
template <typename T, bool copyable = true>
|
|
||||||
class property
|
|
||||||
{
|
|
||||||
typedef boost::signals2::signal<void(const property<T>& )> signal_t;
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief A test class to demonstrate property registration via SERIALIZE_OBJECT.
|
||||||
|
*/
|
||||||
|
class TestObject : public Object {
|
||||||
public:
|
public:
|
||||||
property() : m_changed(new signal_t) {}
|
uLibTypeMacro(TestObject, Object)
|
||||||
property(const T in) : value(in) , m_changed(new signal_t) {}
|
|
||||||
|
|
||||||
inline operator T const & () const { return value; }
|
TestObject() : m_Value(10.5f), m_Status("Initialized"), m_Counter(0) {}
|
||||||
inline operator T & () { return value; }
|
|
||||||
inline T & operator = (const T &i) { value = i; return value; }
|
float m_Value;
|
||||||
template <typename T2> T2 & operator = (const T2 &i) { T2 &guard = value; } // Assign exact identical types only.
|
std::string m_Status;
|
||||||
inline signal_t & valueChanged() { return *m_changed; }
|
int m_Counter;
|
||||||
|
|
||||||
|
// Static properties (registered in constructor/initializer)
|
||||||
|
ULIB_PROPERTY(int, StaticProp, 42)
|
||||||
|
|
||||||
|
ULIB_SERIALIZE_ACCESS
|
||||||
|
|
||||||
|
template <typename Ar>
|
||||||
|
void serialize(Ar& ar, unsigned int version) {
|
||||||
|
ar & HRP("value", m_Value, "mm").range(0, 100).set_default(1.);
|
||||||
|
ar & HRP("status", m_Status);
|
||||||
|
ar & HRP("counter", m_Counter);
|
||||||
|
}
|
||||||
|
|
||||||
private:
|
|
||||||
T value;
|
|
||||||
boost::shared_ptr<signal_t> m_changed;
|
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|
||||||
//template <typename T>
|
class TestObject2 : public TestObject {
|
||||||
//class property <T,false> {
|
public:
|
||||||
// typedef boost::signals2::signal<void( T )> signal_t;
|
uLibTypeMacro(TestObject2, TestObject)
|
||||||
|
|
||||||
//public:
|
TestObject2() : TestObject(), m_Value2(20.5f) {}
|
||||||
// property() : m_changed() {}
|
|
||||||
// property(const T in) : value(in) , m_changed() {}
|
|
||||||
|
|
||||||
// inline operator T const & () const { return value; }
|
float m_Value2;
|
||||||
// inline operator T & () { valueChanged()(value); return value; }
|
|
||||||
// inline T & operator = (const T &i) { value = i; valueChanged()(value); return value; }
|
|
||||||
// template <typename T2> T2 & operator = (const T2 &i) { T2 &guard = value; } // Assign exact identical types only.
|
|
||||||
// inline signal_t &valueChanged() { return m_changed; }
|
|
||||||
|
|
||||||
//private:
|
|
||||||
// property(const property<T> &);
|
|
||||||
// property<T> &operator = (const property<T>&);
|
|
||||||
|
|
||||||
// T value;
|
|
||||||
// signal_t m_changed;
|
|
||||||
//};
|
|
||||||
|
|
||||||
// test generic void function slot //
|
|
||||||
void PrintSlot(const property<int> &i) { std::cout << "slot called, new value = " << i << "!\n"; }
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
int main()
|
|
||||||
{
|
|
||||||
|
|
||||||
|
ULIB_SERIALIZE_ACCESS
|
||||||
|
};
|
||||||
|
|
||||||
|
ULIB_SERIALIZABLE_OBJECT(TestObject2)
|
||||||
|
ULIB_SERIALIZE_OBJECT(TestObject2, TestObject) {
|
||||||
|
std::cout << "Serializing TestObject2" << std::endl;
|
||||||
|
// ar & boost::serialization::make_hrp("value2", ob.m_Value2, "mm").set_default(1.);
|
||||||
|
ar & HRP("value2", ob.m_Value2, "mm").set_default(1.);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
int main() {
|
||||||
|
BEGIN_TESTING(Properties Serialization)
|
||||||
|
|
||||||
|
TestObject obj;
|
||||||
|
|
||||||
|
// 1. Initial state: check static property
|
||||||
|
ASSERT_EQUAL(obj.StaticProp, 42);
|
||||||
|
|
||||||
|
// 2. Activate dynamic properties via the property_register_archive
|
||||||
|
// This calls the serialize method with a special archive that populates m_DynamicProperties
|
||||||
|
ULIB_ACTIVATE_PROPERTIES(obj);
|
||||||
|
|
||||||
|
const auto& props = obj.GetProperties();
|
||||||
|
// This is problematic because GetProperties currently returns d->m_Properties (only static)
|
||||||
|
|
||||||
|
// For now, let's just assert on the dynamic property presence if possible
|
||||||
|
PropertyBase* pVal = obj.GetProperty("value");
|
||||||
|
ASSERT_NOT_NULL(pVal);
|
||||||
|
ASSERT_EQUAL(pVal->GetValueAsString(), "10.5");
|
||||||
|
ASSERT_EQUAL(pVal->GetUnits(), "mm");
|
||||||
|
|
||||||
|
// Check other dynamic properties
|
||||||
|
ASSERT_NOT_NULL(obj.GetProperty("status"));
|
||||||
|
ASSERT_NOT_NULL(obj.GetProperty("counter"));
|
||||||
|
|
||||||
|
// 4. Serialization round-trip (XML)
|
||||||
|
{
|
||||||
|
std::ofstream ofs("test_props.xml");
|
||||||
|
Archive::xml_oarchive(ofs) << NVP("test_obj", obj);
|
||||||
|
}
|
||||||
|
|
||||||
|
TestObject obj2;
|
||||||
|
obj2.m_Value = 0;
|
||||||
|
obj2.m_Status = "";
|
||||||
|
{
|
||||||
|
std::ifstream ifs("test_props.xml");
|
||||||
|
Archive::xml_iarchive(ifs) >> NVP("test_obj", obj2);
|
||||||
|
}
|
||||||
|
|
||||||
|
ASSERT_EQUAL(obj2.m_Value, 10.5f);
|
||||||
|
ASSERT_EQUAL(obj2.m_Status, "Initialized");
|
||||||
|
|
||||||
|
TestObject2 obj3;
|
||||||
|
obj3.m_Value = 12.5;
|
||||||
|
obj3.m_Status = "Initialized";
|
||||||
|
obj3.m_Value2 = 22.5;
|
||||||
|
|
||||||
|
ULIB_ACTIVATE_PROPERTIES(obj3);
|
||||||
|
|
||||||
|
PropertyBase* pVal3 = obj3.GetProperty("value2");
|
||||||
|
ASSERT_NOT_NULL(pVal3);
|
||||||
|
ASSERT_EQUAL(pVal3->GetValueAsString(), "22.5");
|
||||||
|
ASSERT_EQUAL(pVal3->GetUnits(), "mm");
|
||||||
|
|
||||||
|
// 5. Serialization round-trip (XML)
|
||||||
|
{
|
||||||
|
std::ofstream ofs("test_props2.xml");
|
||||||
|
Archive::xml_oarchive(ofs) << NVP("test_obj2", obj3);
|
||||||
|
}
|
||||||
|
|
||||||
|
TestObject2 obj4;
|
||||||
|
obj4.m_Value = 0;
|
||||||
|
obj4.m_Status = "";
|
||||||
|
obj4.m_Value2 = 0;
|
||||||
|
ULIB_ACTIVATE_PROPERTIES(obj4);
|
||||||
|
{
|
||||||
|
std::ifstream ifs("test_props2.xml");
|
||||||
|
Archive::xml_iarchive(ifs) >> NVP("test_obj2", obj4);
|
||||||
|
}
|
||||||
|
ASSERT_EQUAL(obj4.m_Value, 12.5f);
|
||||||
|
ASSERT_EQUAL(obj4.m_Status, "Initialized");
|
||||||
|
ASSERT_EQUAL(obj4.m_Value2, 22.5f);
|
||||||
|
|
||||||
|
|
||||||
|
END_TESTING
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -8,13 +8,12 @@ using namespace uLib;
|
|||||||
|
|
||||||
class TestObject : public Object {
|
class TestObject : public Object {
|
||||||
public:
|
public:
|
||||||
|
uLibTypeMacro(TestObject, Object)
|
||||||
TestObject() : Object(),
|
TestObject() : Object(),
|
||||||
IntProp(this, "IntProp", 10),
|
IntProp(this, "IntProp", 10),
|
||||||
StringProp(this, "StringProp", "Initial")
|
StringProp(this, "StringProp", "Initial")
|
||||||
{}
|
{}
|
||||||
|
|
||||||
virtual const char* GetClassName() const override { return "TestObject"; }
|
|
||||||
|
|
||||||
Property<int> IntProp;
|
Property<int> IntProp;
|
||||||
Property<std::string> StringProp;
|
Property<std::string> StringProp;
|
||||||
};
|
};
|
||||||
|
|||||||
@@ -9,10 +9,9 @@ using namespace uLib;
|
|||||||
|
|
||||||
class TestObject : public Object {
|
class TestObject : public Object {
|
||||||
public:
|
public:
|
||||||
|
uLibTypeMacro(TestObject, Object)
|
||||||
TestObject() : Object() {}
|
TestObject() : Object() {}
|
||||||
|
|
||||||
virtual const char* GetClassName() const override { return "TestObject"; }
|
|
||||||
|
|
||||||
// Use new typedefs
|
// Use new typedefs
|
||||||
StringProperty StringProp = StringProperty(this, "StringProp", "Initial");
|
StringProperty StringProp = StringProperty(this, "StringProp", "Initial");
|
||||||
IntProperty IntProp = IntProperty(this, "IntProp", 42);
|
IntProperty IntProp = IntProperty(this, "IntProp", 42);
|
||||||
|
|||||||
82
src/Core/testing/ReadOnlyPropertyTest.cpp
Normal file
82
src/Core/testing/ReadOnlyPropertyTest.cpp
Normal file
@@ -0,0 +1,82 @@
|
|||||||
|
#include <iostream>
|
||||||
|
#include <vector>
|
||||||
|
#include <string>
|
||||||
|
#include <cassert>
|
||||||
|
#include "Core/Object.h"
|
||||||
|
#include "Core/Property.h"
|
||||||
|
#include "Core/Serializable.h"
|
||||||
|
|
||||||
|
#include "Core/Serializable.h"
|
||||||
|
|
||||||
|
using namespace uLib;
|
||||||
|
|
||||||
|
class ReadOnlyTestObject : public Object {
|
||||||
|
public:
|
||||||
|
int m_value;
|
||||||
|
int getValue() const { return m_value; }
|
||||||
|
|
||||||
|
enum State { State1, State2 };
|
||||||
|
State m_state;
|
||||||
|
State getState() const { return m_state; }
|
||||||
|
|
||||||
|
ReadOnlyTestObject() : m_value(10), m_state(State1) {
|
||||||
|
ULIB_ACTIVATE_PROPERTIES(*this);
|
||||||
|
}
|
||||||
|
|
||||||
|
template<class Archive>
|
||||||
|
void serialize(Archive & ar, const unsigned int version) {
|
||||||
|
// Lvalue reference - should be NOT read-only
|
||||||
|
ar & HRP("lvalue_prop", m_value);
|
||||||
|
|
||||||
|
// Rvalue from getter - should be read-only
|
||||||
|
ar & HRP("rvalue_prop", getValue());
|
||||||
|
|
||||||
|
// Enum lvalue - should be NOT read-only
|
||||||
|
ar & boost::serialization::make_hrp_enum("lvalue_enum", (int&)m_state, {"State1", "State2"});
|
||||||
|
|
||||||
|
// Enum rvalue - should be read-only
|
||||||
|
ar & boost::serialization::make_hrp_enum("rvalue_enum", (int)getState(), {"State1", "State2"});
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
int main() {
|
||||||
|
std::cout << "Testing Read-Only Property Feature..." << std::endl;
|
||||||
|
|
||||||
|
ReadOnlyTestObject obj;
|
||||||
|
const auto& props = obj.GetProperties();
|
||||||
|
|
||||||
|
std::cout << "Registered Properties in ReadOnlyTestObject:" << std::endl;
|
||||||
|
bool found_lvalue = false;
|
||||||
|
bool found_rvalue = false;
|
||||||
|
bool found_lvalue_enum = false;
|
||||||
|
bool found_rvalue_enum = false;
|
||||||
|
|
||||||
|
for (auto* p : props) {
|
||||||
|
bool ro = p->IsReadOnly();
|
||||||
|
std::cout << " - Name: " << p->GetName()
|
||||||
|
<< " | Type: " << p->GetTypeName()
|
||||||
|
<< " | ReadOnly: " << (ro ? "YES" : "NO") << std::endl;
|
||||||
|
|
||||||
|
if (p->GetName() == "lvalue_prop") {
|
||||||
|
if (ro) { std::cerr << "FAIL: lvalue_prop should NOT be read-only" << std::endl; return 1; }
|
||||||
|
found_lvalue = true;
|
||||||
|
} else if (p->GetName() == "rvalue_prop") {
|
||||||
|
if (!ro) { std::cerr << "FAIL: rvalue_prop SHOULD be read-only" << std::endl; return 1; }
|
||||||
|
found_rvalue = true;
|
||||||
|
} else if (p->GetName() == "lvalue_enum") {
|
||||||
|
if (ro) { std::cerr << "FAIL: lvalue_enum should NOT be read-only" << std::endl; return 1; }
|
||||||
|
found_lvalue_enum = true;
|
||||||
|
} else if (p->GetName() == "rvalue_enum") {
|
||||||
|
if (!ro) { std::cerr << "FAIL: rvalue_enum SHOULD be read-only" << std::endl; return 1; }
|
||||||
|
found_rvalue_enum = true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (found_lvalue && found_rvalue && found_lvalue_enum && found_rvalue_enum) {
|
||||||
|
std::cout << "TEST PASSED SUCCESSFULLY!" << std::endl;
|
||||||
|
return 0;
|
||||||
|
} else {
|
||||||
|
std::cerr << "TEST FAILED: Some properties were not found!" << std::endl;
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -70,7 +70,30 @@ ULIB_SERIALIZABLE_OBJECT(D)
|
|||||||
ULIB_SERIALIZE_OBJECT(D, A, B) { ar & AR(numd); }
|
ULIB_SERIALIZE_OBJECT(D, A, B) { ar & AR(numd); }
|
||||||
|
|
||||||
int main() {
|
int main() {
|
||||||
A o;
|
BEGIN_TESTING(DreadDiamond Serialization)
|
||||||
|
|
||||||
Archive::xml_oarchive(std::cout) << NVP(o);
|
D o;
|
||||||
|
C c;
|
||||||
|
c.numb = 123;
|
||||||
|
{
|
||||||
|
std::ofstream file("test.xml");
|
||||||
|
Archive::xml_oarchive(file) << NVP("dd_test", o) << NVP("c", c);
|
||||||
|
}
|
||||||
|
{
|
||||||
|
D o2;
|
||||||
|
C c2;
|
||||||
|
std::ifstream file("test.xml");
|
||||||
|
Archive::xml_iarchive(file) >> NVP("dd_test", o2) >> NVP("c", c2);
|
||||||
|
|
||||||
|
// D //
|
||||||
|
ASSERT_EQUAL(o.numa, o2.numa);
|
||||||
|
ASSERT_EQUAL(o.numb, o2.numb);
|
||||||
|
ASSERT_EQUAL(o.numd, o2.numd);
|
||||||
|
|
||||||
|
// C //
|
||||||
|
ASSERT_EQUAL(c.numb, c2.numb);
|
||||||
|
ASSERT_EQUAL(c.numc, c2.numc);
|
||||||
|
}
|
||||||
|
|
||||||
|
END_TESTING
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -143,22 +143,19 @@ int testing_hrt_class() {
|
|||||||
}
|
}
|
||||||
a.a() = 0;
|
a.a() = 0;
|
||||||
a.p_a = "zero string";
|
a.p_a = "zero string";
|
||||||
{
|
|
||||||
// ERRORE FIX !
|
|
||||||
// std::ifstream file("test.xml");
|
|
||||||
// Archive::hrt_iarchive(file) >> NVP(a);
|
|
||||||
}
|
|
||||||
|
|
||||||
Archive::hrt_oarchive(std::cout) << NVP(a);
|
Archive::hrt_oarchive(std::cout) << NVP(a);
|
||||||
return (a.a() == 5552368 && a.p_a == "A property string");
|
return (a.a() == 5552368 && a.p_a == "A property string");
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
int main() {
|
int main() {
|
||||||
BEGIN_TESTING(Serialize Test);
|
BEGIN_TESTING(Serialize Test);
|
||||||
|
|
||||||
TEST1(test_V3f());
|
TEST1(test_V3f());
|
||||||
TEST1(testing_xml_class());
|
TEST1(testing_xml_class());
|
||||||
// testing_hrt_class(); ///// << ERRORE in HRT with properties
|
// TEST1(testing_hrt_class());
|
||||||
|
|
||||||
END_TESTING;
|
END_TESTING;
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -33,5 +33,7 @@ printf("..:: Testing " #name " ::..\n");
|
|||||||
|
|
||||||
#define TEST1(val) _fail += (val)==0
|
#define TEST1(val) _fail += (val)==0
|
||||||
#define TEST0(val) _fail += (val)!=0
|
#define TEST0(val) _fail += (val)!=0
|
||||||
|
#define ASSERT_EQUAL(a,b) if((a)!=(b)) { printf("Assertion failed: " #a " != " #b " at line %d\n", __LINE__); _fail++; }
|
||||||
|
#define ASSERT_NOT_NULL(ptr) if((ptr)==NULL) { printf("Assertion failed: " #ptr " is NULL at line %d\n", __LINE__); _fail++; }
|
||||||
#define END_TESTING return _fail;
|
#define END_TESTING return _fail;
|
||||||
|
|
||||||
|
|||||||
@@ -40,12 +40,10 @@ namespace uLib {
|
|||||||
|
|
||||||
class DetectorChamber : public ContainerBox {
|
class DetectorChamber : public ContainerBox {
|
||||||
|
|
||||||
typedef ContainerBox BaseClass;
|
|
||||||
|
|
||||||
|
|
||||||
public:
|
public:
|
||||||
|
uLibTypeMacro(DetectorChamber, ContainerBox)
|
||||||
|
|
||||||
|
|
||||||
virtual const char * GetClassName() const { return "DetectorChamber"; }
|
|
||||||
|
|
||||||
DetectorChamber() : BaseClass() {
|
DetectorChamber() : BaseClass() {
|
||||||
m_ProjectionPlane.origin = HPoint3f(0, 0, 0);
|
m_ProjectionPlane.origin = HPoint3f(0, 0, 0);
|
||||||
|
|||||||
@@ -27,6 +27,8 @@ set(SOURCES
|
|||||||
Scene.cpp
|
Scene.cpp
|
||||||
Solid.cpp
|
Solid.cpp
|
||||||
EmitterPrimary.cpp
|
EmitterPrimary.cpp
|
||||||
|
Matter.cpp
|
||||||
|
GeantRegistration.cpp
|
||||||
DetectorConstruction.cpp
|
DetectorConstruction.cpp
|
||||||
PhysicsList.cpp
|
PhysicsList.cpp
|
||||||
ActionInitialization.cpp
|
ActionInitialization.cpp
|
||||||
|
|||||||
@@ -26,8 +26,7 @@ namespace Geant {
|
|||||||
class EmitterPrimary : public G4VUserPrimaryGeneratorAction, public AffineTransform
|
class EmitterPrimary : public G4VUserPrimaryGeneratorAction, public AffineTransform
|
||||||
{
|
{
|
||||||
public:
|
public:
|
||||||
|
uLibTypeMacro(EmitterPrimary, Object)
|
||||||
virtual const char* GetClassName() const override { return "Geant.EmitterPrimary"; }
|
|
||||||
|
|
||||||
EmitterPrimary();
|
EmitterPrimary();
|
||||||
virtual ~EmitterPrimary();
|
virtual ~EmitterPrimary();
|
||||||
@@ -47,8 +46,7 @@ class EmitterPrimary : public G4VUserPrimaryGeneratorAction, public AffineTransf
|
|||||||
class SkyPlaneEmitterPrimary : public EmitterPrimary
|
class SkyPlaneEmitterPrimary : public EmitterPrimary
|
||||||
{
|
{
|
||||||
public:
|
public:
|
||||||
|
uLibTypeMacro(SkyPlaneEmitterPrimary, EmitterPrimary)
|
||||||
virtual const char* GetClassName() const override { return "Geant.SkyPlaneEmitterPrimary"; }
|
|
||||||
|
|
||||||
SkyPlaneEmitterPrimary();
|
SkyPlaneEmitterPrimary();
|
||||||
virtual ~SkyPlaneEmitterPrimary();
|
virtual ~SkyPlaneEmitterPrimary();
|
||||||
@@ -69,8 +67,7 @@ class SkyPlaneEmitterPrimary : public EmitterPrimary
|
|||||||
class CylinderEmitterPrimary : public EmitterPrimary
|
class CylinderEmitterPrimary : public EmitterPrimary
|
||||||
{
|
{
|
||||||
public:
|
public:
|
||||||
|
uLibTypeMacro(CylinderEmitterPrimary, EmitterPrimary)
|
||||||
virtual const char* GetClassName() const override { return "Geant.CylinderEmitterPrimary"; }
|
|
||||||
|
|
||||||
CylinderEmitterPrimary();
|
CylinderEmitterPrimary();
|
||||||
virtual ~CylinderEmitterPrimary();
|
virtual ~CylinderEmitterPrimary();
|
||||||
@@ -98,8 +95,7 @@ class CylinderEmitterPrimary : public EmitterPrimary
|
|||||||
class QuadMeshEmitterPrimary : public EmitterPrimary
|
class QuadMeshEmitterPrimary : public EmitterPrimary
|
||||||
{
|
{
|
||||||
public:
|
public:
|
||||||
|
uLibTypeMacro(QuadMeshEmitterPrimary, EmitterPrimary)
|
||||||
virtual const char* GetClassName() const override { return "Geant.QuadMeshEmitterPrimary"; }
|
|
||||||
|
|
||||||
QuadMeshEmitterPrimary();
|
QuadMeshEmitterPrimary();
|
||||||
virtual ~QuadMeshEmitterPrimary();
|
virtual ~QuadMeshEmitterPrimary();
|
||||||
|
|||||||
@@ -50,8 +50,7 @@ class SteppingAction;
|
|||||||
class GeantEvent : public Object {
|
class GeantEvent : public Object {
|
||||||
|
|
||||||
public:
|
public:
|
||||||
|
uLibTypeMacro(GeantEvent, Object)
|
||||||
virtual const char* GetClassName() const override { return "Geant.GeantEvent"; }
|
|
||||||
|
|
||||||
/// A single interaction step along the muon path.
|
/// A single interaction step along the muon path.
|
||||||
struct Delta {
|
struct Delta {
|
||||||
|
|||||||
22
src/HEP/Geant/GeantRegistration.cpp
Normal file
22
src/HEP/Geant/GeantRegistration.cpp
Normal file
@@ -0,0 +1,22 @@
|
|||||||
|
#include "Core/ObjectFactory.h"
|
||||||
|
#include "HEP/Geant/Matter.h"
|
||||||
|
#include "HEP/Geant/Solid.h"
|
||||||
|
#include "HEP/Geant/Scene.h"
|
||||||
|
#include "HEP/Geant/EmitterPrimary.hh"
|
||||||
|
#include "HEP/Geant/GeantEvent.h"
|
||||||
|
|
||||||
|
namespace uLib {
|
||||||
|
namespace Geant {
|
||||||
|
|
||||||
|
ULIB_REGISTER_OBJECT(Material)
|
||||||
|
ULIB_REGISTER_OBJECT(Solid)
|
||||||
|
ULIB_REGISTER_OBJECT(TessellatedSolid)
|
||||||
|
ULIB_REGISTER_OBJECT(BoxSolid)
|
||||||
|
ULIB_REGISTER_OBJECT(Scene)
|
||||||
|
ULIB_REGISTER_OBJECT(SkyPlaneEmitterPrimary)
|
||||||
|
ULIB_REGISTER_OBJECT(CylinderEmitterPrimary)
|
||||||
|
ULIB_REGISTER_OBJECT(QuadMeshEmitterPrimary)
|
||||||
|
ULIB_REGISTER_OBJECT(GeantEvent)
|
||||||
|
|
||||||
|
} // namespace Geant
|
||||||
|
} // namespace uLib
|
||||||
21
src/HEP/Geant/Matter.cpp
Normal file
21
src/HEP/Geant/Matter.cpp
Normal file
@@ -0,0 +1,21 @@
|
|||||||
|
|
||||||
|
#include "HEP/Geant/Matter.h"
|
||||||
|
#include <Geant4/G4Material.hh>
|
||||||
|
#include <Geant4/G4NistManager.hh>
|
||||||
|
|
||||||
|
using namespace uLib::Geant;
|
||||||
|
|
||||||
|
Material::Material() : m_G4Data(nullptr) {}
|
||||||
|
|
||||||
|
Material::Material(const char *name) : m_G4Data(nullptr) {
|
||||||
|
this->SetFromNist(name);
|
||||||
|
}
|
||||||
|
|
||||||
|
Material::~Material() {
|
||||||
|
if(m_G4Data) delete m_G4Data;
|
||||||
|
}
|
||||||
|
|
||||||
|
void Material::SetFromNist(const char *name) {
|
||||||
|
G4NistManager* man = G4NistManager::Instance();
|
||||||
|
m_G4Data = man->FindOrBuildMaterial(name);
|
||||||
|
}
|
||||||
@@ -29,9 +29,10 @@
|
|||||||
#define MATTER_H
|
#define MATTER_H
|
||||||
|
|
||||||
#include "Core/Object.h"
|
#include "Core/Object.h"
|
||||||
|
#include <Geant4/G4Material.hh>
|
||||||
|
#include <Geant4/G4NistManager.hh>
|
||||||
|
|
||||||
class G4Element;
|
class G4Element;
|
||||||
class G4Material;
|
|
||||||
|
|
||||||
namespace uLib {
|
namespace uLib {
|
||||||
namespace Geant {
|
namespace Geant {
|
||||||
@@ -55,18 +56,53 @@ private:
|
|||||||
//// MATERIAL //////////////////////////////////////////////////////////////////
|
//// MATERIAL //////////////////////////////////////////////////////////////////
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
|
|
||||||
|
// TODO: finish from G4NistMaterialBuilder
|
||||||
|
|
||||||
class Material : public Object {
|
class Material : public Object {
|
||||||
public:
|
public:
|
||||||
|
uLibTypeMacro(Material, Object)
|
||||||
|
|
||||||
virtual const char* GetClassName() const override { return "Geant.Material"; }
|
enum State {
|
||||||
|
Undefined = 0,
|
||||||
uLibRefMacro(G4Data,G4Material *)
|
Solid,
|
||||||
private:
|
Liquid,
|
||||||
G4Material *m_G4Data;
|
Gas
|
||||||
};
|
};
|
||||||
|
|
||||||
|
Material();
|
||||||
|
Material(const char *name);
|
||||||
|
~Material();
|
||||||
|
|
||||||
|
void SetFromNist(const char *name);
|
||||||
|
|
||||||
|
template <typename Ar>
|
||||||
|
void serialize(Ar &ar) {
|
||||||
|
ar & HRP("name", m_G4Data->GetName());
|
||||||
|
ar & HRP("density", m_G4Data->GetDensity());
|
||||||
|
ar & serialization::make_hrp_enum("state", m_G4Data->GetState(), {"Undefined", "Solid", "Liquid", "Gas"});
|
||||||
|
}
|
||||||
|
|
||||||
|
G4Material *GetG4Material() { return m_G4Data; }
|
||||||
|
|
||||||
|
private:
|
||||||
|
G4Material *m_G4Data;
|
||||||
|
|
||||||
|
};
|
||||||
|
|
||||||
|
// class MaterialCompound : public Material {
|
||||||
|
// public:
|
||||||
|
|
||||||
|
// MaterialCompound(const char *name) {}
|
||||||
|
|
||||||
|
// void AddMaterial(Material *m, double fractionmass) { m_Materials.push_back(std::make_pair(m, fractionmass)); }
|
||||||
|
// void AddElement(Element *e, double fractionmass) { m_Elements.push_back(std::make_pair(e, fractionmass)); }
|
||||||
|
// void SetDensity(double density) { m_Density = density; }
|
||||||
|
|
||||||
|
// private:
|
||||||
|
// std::vector<std::pair<Material *, double>> m_Materials;
|
||||||
|
// std::vector<std::pair<Element *, double>> m_Elements;
|
||||||
|
// double m_Density;
|
||||||
|
// };
|
||||||
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -43,8 +43,7 @@ class EmitterPrimary;
|
|||||||
|
|
||||||
class Scene : public Object {
|
class Scene : public Object {
|
||||||
public:
|
public:
|
||||||
|
uLibTypeMacro(Scene, Object)
|
||||||
virtual const char* GetClassName() const override { return "Geant.Scene"; }
|
|
||||||
|
|
||||||
Scene();
|
Scene();
|
||||||
~Scene();
|
~Scene();
|
||||||
|
|||||||
@@ -146,6 +146,9 @@ void Solid::SetParent(Solid *parent) {
|
|||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
TessellatedSolid::TessellatedSolid()
|
||||||
|
: BaseClass("unnamed_tessellated"), m_Solid(new G4TessellatedSolid("unnamed_tessellated")) {}
|
||||||
|
|
||||||
TessellatedSolid::TessellatedSolid(const char *name)
|
TessellatedSolid::TessellatedSolid(const char *name)
|
||||||
: BaseClass(name), m_Solid(new G4TessellatedSolid(name)) {
|
: BaseClass(name), m_Solid(new G4TessellatedSolid(name)) {
|
||||||
}
|
}
|
||||||
@@ -173,9 +176,15 @@ void TessellatedSolid::Update() {
|
|||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
BoxSolid::BoxSolid(const char *name) :
|
||||||
|
BaseClass(name),
|
||||||
|
m_ContainerBox(new ContainerBox()),
|
||||||
|
m_Solid(new G4Box(name, 1, 1, 1))
|
||||||
|
{}
|
||||||
|
|
||||||
BoxSolid::BoxSolid(const char *name, ContainerBox *box) : BaseClass(name) {
|
BoxSolid::BoxSolid(const char *name, ContainerBox *box) : BaseClass(name) {
|
||||||
m_Solid = new G4Box(name, 1, 1, 1);
|
m_Solid = new G4Box(name, 1, 1, 1);
|
||||||
m_Object = box;
|
m_ContainerBox = box;
|
||||||
Object::connect(box, &ContainerBox::Updated, this, &BoxSolid::Update);
|
Object::connect(box, &ContainerBox::Updated, this, &BoxSolid::Update);
|
||||||
if (m_Logical) {
|
if (m_Logical) {
|
||||||
m_Logical->SetSolid(m_Solid);
|
m_Logical->SetSolid(m_Solid);
|
||||||
@@ -184,16 +193,16 @@ BoxSolid::BoxSolid(const char *name, ContainerBox *box) : BaseClass(name) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
void BoxSolid::Update() {
|
void BoxSolid::Update() {
|
||||||
if (m_Object) {
|
if (m_ContainerBox) {
|
||||||
Vector3f size = m_Object->GetSize();
|
Vector3f size = m_ContainerBox->GetSize();
|
||||||
m_Solid->SetXHalfLength(size(0) * 0.5);
|
m_Solid->SetXHalfLength(size(0) * 0.5);
|
||||||
m_Solid->SetYHalfLength(size(1) * 0.5);
|
m_Solid->SetYHalfLength(size(1) * 0.5);
|
||||||
m_Solid->SetZHalfLength(size(2) * 0.5);
|
m_Solid->SetZHalfLength(size(2) * 0.5);
|
||||||
|
|
||||||
// Geant4 placement is relative to center. uLib Box is anchored at corner.
|
// Geant4 placement is relative to center. uLib Box is anchored at corner.
|
||||||
// 1. Get position and rotation (clean, without scale)
|
// 1. Get position and rotation (clean, without scale)
|
||||||
Vector3f pos = m_Object->GetPosition();
|
Vector3f pos = m_ContainerBox->GetPosition();
|
||||||
Matrix3f rot = m_Object->GetRotation();
|
Matrix3f rot = m_ContainerBox->GetRotation();
|
||||||
|
|
||||||
// 2. Center = Corner + Rotation * (Half-Size)
|
// 2. Center = Corner + Rotation * (Half-Size)
|
||||||
// We must rotate the offset vector because uLib box can be rotated.
|
// We must rotate the offset vector because uLib box can be rotated.
|
||||||
@@ -205,6 +214,9 @@ void BoxSolid::Update() {
|
|||||||
|
|
||||||
this->SetTransform(t.GetMatrix());
|
this->SetTransform(t.GetMatrix());
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -43,8 +43,7 @@ namespace Geant {
|
|||||||
|
|
||||||
class Solid : public Object {
|
class Solid : public Object {
|
||||||
public:
|
public:
|
||||||
|
uLibTypeMacro(Solid, Object)
|
||||||
virtual const char* GetClassName() const override { return "Geant.Solid"; }
|
|
||||||
|
|
||||||
Solid();
|
Solid();
|
||||||
Solid(const char *name);
|
Solid(const char *name);
|
||||||
@@ -52,6 +51,7 @@ public:
|
|||||||
|
|
||||||
void SetNistMaterial(const char *name);
|
void SetNistMaterial(const char *name);
|
||||||
void SetMaterial(G4Material *material);
|
void SetMaterial(G4Material *material);
|
||||||
|
|
||||||
void SetSizeUnit(const char *unit);
|
void SetSizeUnit(const char *unit);
|
||||||
// Implementiamo SetParent qui, per tutti.
|
// Implementiamo SetParent qui, per tutti.
|
||||||
virtual void SetParent(Solid *parent);
|
virtual void SetParent(Solid *parent);
|
||||||
@@ -69,6 +69,14 @@ public:
|
|||||||
return m_Logical ? m_Logical->GetName().c_str() : m_Name.c_str();
|
return m_Logical ? m_Logical->GetName().c_str() : m_Name.c_str();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
template < typename Ar >
|
||||||
|
void serialize(Ar &ar, const unsigned int version) {
|
||||||
|
ar & m_Name;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
protected:
|
protected:
|
||||||
|
|
||||||
std::string m_Name;
|
std::string m_Name;
|
||||||
@@ -84,11 +92,10 @@ protected:
|
|||||||
|
|
||||||
|
|
||||||
class TessellatedSolid : public Solid {
|
class TessellatedSolid : public Solid {
|
||||||
typedef Solid BaseClass;
|
|
||||||
public:
|
public:
|
||||||
|
uLibTypeMacro(TessellatedSolid, Solid)
|
||||||
|
|
||||||
virtual const char* GetClassName() const override { return "Geant.TessellatedSolid"; }
|
TessellatedSolid();
|
||||||
|
|
||||||
TessellatedSolid(const char *name);
|
TessellatedSolid(const char *name);
|
||||||
void SetMesh(TriangleMesh &mesh);
|
void SetMesh(TriangleMesh &mesh);
|
||||||
uLibGetMacro(Solid, G4TessellatedSolid *)
|
uLibGetMacro(Solid, G4TessellatedSolid *)
|
||||||
@@ -110,22 +117,27 @@ private :
|
|||||||
|
|
||||||
|
|
||||||
class BoxSolid : public Solid {
|
class BoxSolid : public Solid {
|
||||||
typedef Solid BaseClass;
|
|
||||||
|
|
||||||
public:
|
public:
|
||||||
|
uLibTypeMacro(BoxSolid, Solid)
|
||||||
|
|
||||||
virtual const char* GetClassName() const override { return "Geant.BoxSolid"; }
|
BoxSolid(const char *name = "");
|
||||||
|
|
||||||
BoxSolid(const char *name, ContainerBox *box);
|
BoxSolid(const char *name, ContainerBox *box);
|
||||||
virtual G4VSolid* GetG4Solid() const override { return (G4VSolid*)m_Solid; }
|
virtual G4VSolid* GetG4Solid() const override { return (G4VSolid*)m_Solid; }
|
||||||
|
|
||||||
ContainerBox* GetObject() const { return m_Object; }
|
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 & m_ContainerBox;
|
||||||
|
}
|
||||||
|
|
||||||
public slots:
|
public slots:
|
||||||
void Update();
|
void Update();
|
||||||
|
|
||||||
private:
|
private:
|
||||||
ContainerBox *m_Object;
|
ContainerBox *m_ContainerBox;
|
||||||
G4Box *m_Solid;
|
G4Box *m_Solid;
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|||||||
@@ -21,17 +21,16 @@ namespace uLib {
|
|||||||
|
|
||||||
Assembly::Assembly()
|
Assembly::Assembly()
|
||||||
: ObjectsContext(),
|
: ObjectsContext(),
|
||||||
AffineTransform(),
|
TRS(),
|
||||||
m_BBoxMin(Vector3f::Zero()),
|
m_BBoxMin(Vector3f::Zero()),
|
||||||
m_BBoxMax(Vector3f::Zero()),
|
m_BBoxMax(Vector3f::Zero()),
|
||||||
m_ShowBoundingBox(false),
|
m_ShowBoundingBox(false),
|
||||||
m_GroupSelection(true) {
|
m_GroupSelection(true) {
|
||||||
ULIB_ACTIVATE_PROPERTIES(*this);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
Assembly::Assembly(const Assembly ©)
|
Assembly::Assembly(const Assembly ©)
|
||||||
: ObjectsContext(copy),
|
: ObjectsContext(copy),
|
||||||
AffineTransform(copy),
|
TRS(copy),
|
||||||
m_BBoxMin(copy.m_BBoxMin),
|
m_BBoxMin(copy.m_BBoxMin),
|
||||||
m_BBoxMax(copy.m_BBoxMax),
|
m_BBoxMax(copy.m_BBoxMax),
|
||||||
m_ShowBoundingBox(copy.m_ShowBoundingBox),
|
m_ShowBoundingBox(copy.m_ShowBoundingBox),
|
||||||
@@ -55,6 +54,10 @@ void Assembly::AddObject(Object *obj) {
|
|||||||
this->ComputeBoundingBox();
|
this->ComputeBoundingBox();
|
||||||
this->Updated(); // Signal that assembly itself changed (AABB-wise)
|
this->Updated(); // Signal that assembly itself changed (AABB-wise)
|
||||||
});
|
});
|
||||||
|
|
||||||
|
// Parent -> Child propagation for world matrix updates
|
||||||
|
Object::connect(this, &Object::Updated, obj, &Object::Updated);
|
||||||
|
|
||||||
this->ComputeBoundingBox();
|
this->ComputeBoundingBox();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -43,10 +43,10 @@ namespace uLib {
|
|||||||
* A bounding box is automatically computed from all contained objects and
|
* A bounding box is automatically computed from all contained objects and
|
||||||
* can be queried or shown/hidden through the VTK puppet.
|
* can be queried or shown/hidden through the VTK puppet.
|
||||||
*/
|
*/
|
||||||
class Assembly : public ObjectsContext, public AffineTransform {
|
class Assembly : public ObjectsContext, public TRS {
|
||||||
public:
|
public:
|
||||||
uLibTypeMacro(Assembly, ObjectsContext, AffineTransform)
|
uLibTypeMacro(Assembly, ObjectsContext, TRS)
|
||||||
virtual const char *GetClassName() const override { return "Assembly"; }
|
|
||||||
|
|
||||||
Assembly();
|
Assembly();
|
||||||
Assembly(const Assembly ©);
|
Assembly(const Assembly ©);
|
||||||
@@ -54,7 +54,7 @@ public:
|
|||||||
|
|
||||||
template <class ArchiveT>
|
template <class ArchiveT>
|
||||||
void serialize(ArchiveT & ar, const unsigned int version) {
|
void serialize(ArchiveT & ar, const unsigned int version) {
|
||||||
ar & boost::serialization::make_nvp("AffineTransform", boost::serialization::base_object<AffineTransform>(*this));
|
ar & boost::serialization::make_nvp("TRS", boost::serialization::base_object<TRS>(*this));
|
||||||
ar & boost::serialization::make_hrp("GroupSelection", m_GroupSelection);
|
ar & boost::serialization::make_hrp("GroupSelection", m_GroupSelection);
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -96,10 +96,13 @@ public:
|
|||||||
|
|
||||||
signals:
|
signals:
|
||||||
virtual void Updated() override {
|
virtual void Updated() override {
|
||||||
if (m_InUpdated) return; // break signal recursion
|
if (m_InUpdated) return;
|
||||||
m_InUpdated = true;
|
m_InUpdated = true;
|
||||||
|
|
||||||
|
// Synchronize TRS part
|
||||||
|
this->TRS::Updated();
|
||||||
|
|
||||||
this->ComputeBoundingBox();
|
this->ComputeBoundingBox();
|
||||||
ULIB_SIGNAL_EMIT(Object::Updated);
|
|
||||||
m_InUpdated = false;
|
m_InUpdated = false;
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -110,6 +113,8 @@ private:
|
|||||||
bool m_GroupSelection;
|
bool m_GroupSelection;
|
||||||
bool m_InUpdated = false;
|
bool m_InUpdated = false;
|
||||||
std::map<Object*, Connection> m_ChildConnections;
|
std::map<Object*, Connection> m_ChildConnections;
|
||||||
|
|
||||||
|
ULIB_DECLARE_PROPERTIES(Assembly)
|
||||||
};
|
};
|
||||||
|
|
||||||
} // namespace uLib
|
} // namespace uLib
|
||||||
|
|||||||
@@ -29,32 +29,31 @@
|
|||||||
#include "Geometry.h"
|
#include "Geometry.h"
|
||||||
#include "Core/Object.h"
|
#include "Core/Object.h"
|
||||||
#include "Core/Property.h"
|
#include "Core/Property.h"
|
||||||
|
#include "Core/Serializable.h"
|
||||||
#include "Math/Dense.h"
|
#include "Math/Dense.h"
|
||||||
#include "Math/Transform.h"
|
#include "Math/Transform.h"
|
||||||
#include <utility>
|
#include <utility>
|
||||||
|
|
||||||
|
#include <iostream>
|
||||||
|
|
||||||
namespace uLib {
|
namespace uLib {
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* @brief Represents an oriented bounding box (OBB) within a hierarchical
|
* @brief Represents an oriented bounding box (OBB) within a hierarchical
|
||||||
* transformation system.
|
* transformation system.
|
||||||
*
|
*
|
||||||
* ContainerBox inherits from AffineTransform, which defines its parent
|
* ContainerBox inherits from TRS, which defines its parent
|
||||||
* coordinate system. It contains an internal local transformation (m_LocalT)
|
* coordinate system. It contains an internal local transformation (m_LocalT)
|
||||||
* that defines the box's specific origin and size relative to its own
|
* that defines the box's specific origin and size relative to its own
|
||||||
* coordinate system.
|
* coordinate system.
|
||||||
*/
|
*/
|
||||||
class ContainerBox : public AffineTransform {
|
class ContainerBox : public TRS {
|
||||||
|
|
||||||
|
uLibTypeMacro(ContainerBox, TRS)
|
||||||
|
ULIB_SERIALIZE_ACCESS
|
||||||
|
ULIB_DECLARE_PROPERTIES(ContainerBox)
|
||||||
|
|
||||||
public:
|
public:
|
||||||
uLibTypeMacro(ContainerBox, AffineTransform)
|
|
||||||
|
|
||||||
virtual const char * GetClassName() const override { return "ContainerBox"; }
|
|
||||||
|
|
||||||
////////////////////////////////////////////////////////////////////////////
|
|
||||||
// PROPERTIES //
|
|
||||||
Vector3f Size;
|
|
||||||
Vector3f Origin;
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* @brief Default constructor.
|
* @brief Default constructor.
|
||||||
@@ -64,7 +63,6 @@ public:
|
|||||||
: m_LocalT(this), // BaseClass is Parent of m_LocalTransform
|
: m_LocalT(this), // BaseClass is Parent of m_LocalTransform
|
||||||
Size(1.0f, 1.0f, 1.0f),
|
Size(1.0f, 1.0f, 1.0f),
|
||||||
Origin(0.0f, 0.0f, 0.0f) {
|
Origin(0.0f, 0.0f, 0.0f) {
|
||||||
ULIB_ACTIVATE_PROPERTIES(*this);
|
|
||||||
this->Sync();
|
this->Sync();
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -76,7 +74,6 @@ public:
|
|||||||
: m_LocalT(this),
|
: m_LocalT(this),
|
||||||
Size(size),
|
Size(size),
|
||||||
Origin(0.0f, 0.0f, 0.0f) {
|
Origin(0.0f, 0.0f, 0.0f) {
|
||||||
ULIB_ACTIVATE_PROPERTIES(*this);
|
|
||||||
this->Sync();
|
this->Sync();
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -86,20 +83,20 @@ public:
|
|||||||
*/
|
*/
|
||||||
ContainerBox(const ContainerBox ©)
|
ContainerBox(const ContainerBox ©)
|
||||||
: m_LocalT(this), // Reset parent to the new object
|
: m_LocalT(this), // Reset parent to the new object
|
||||||
AffineTransform(copy),
|
TRS(copy),
|
||||||
Size(copy.Size),
|
Size(copy.Size),
|
||||||
Origin(copy.Origin) {
|
Origin(copy.Origin) {
|
||||||
ULIB_ACTIVATE_PROPERTIES(*this);
|
|
||||||
this->Sync();
|
this->Sync();
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
// /**
|
||||||
* @brief Serialization template for property registration and persistence.
|
// * @brief Serialization template for property registration and persistence.
|
||||||
*/
|
// */
|
||||||
template <class ArchiveT>
|
template <class ArchiveT>
|
||||||
void serialize(ArchiveT & ar, const unsigned int version) {
|
void serialize(ArchiveT & ar, const unsigned int version) {
|
||||||
ar & HRP(Size);
|
ar & HRP(Size);
|
||||||
ar & HRP(Origin);
|
ar & HRP(Origin);
|
||||||
|
ar & NVP("TRS", boost::serialization::base_object<TRS>(*this));
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@@ -214,8 +211,13 @@ signals:
|
|||||||
|
|
||||||
/** Signal emitted when properties change */
|
/** Signal emitted when properties change */
|
||||||
virtual void Updated() override {
|
virtual void Updated() override {
|
||||||
|
// 1. Synchronize local box part (Size/Origin -> m_LocalT)
|
||||||
this->Sync();
|
this->Sync();
|
||||||
ULIB_SIGNAL_EMIT(Object::Updated);
|
|
||||||
|
// 2. Synchronize TRS part (position/rotation/scaling -> m_T) and emit signal
|
||||||
|
this->TRS::Updated();
|
||||||
|
|
||||||
|
// std::cout << "ContainerBox::Updated()" << std::endl;
|
||||||
}
|
}
|
||||||
|
|
||||||
private:
|
private:
|
||||||
@@ -227,9 +229,13 @@ private:
|
|||||||
|
|
||||||
|
|
||||||
private:
|
private:
|
||||||
|
Vector3f Size;
|
||||||
|
Vector3f Origin;
|
||||||
AffineTransform m_LocalT;
|
AffineTransform m_LocalT;
|
||||||
|
|
||||||
};
|
};
|
||||||
|
|
||||||
} // namespace uLib
|
} // namespace uLib
|
||||||
|
|
||||||
|
|
||||||
#endif // CONTAINERBOX_H
|
#endif // CONTAINERBOX_H
|
||||||
|
|||||||
@@ -39,10 +39,12 @@ namespace uLib {
|
|||||||
* The cylinder orientation is defined by the Axis property (0=X, 1=Y, 2=Z).
|
* The cylinder orientation is defined by the Axis property (0=X, 1=Y, 2=Z).
|
||||||
* By default, it is aligned with the Y axis (Axis=1).
|
* By default, it is aligned with the Y axis (Axis=1).
|
||||||
*/
|
*/
|
||||||
class Cylinder : public AffineTransform {
|
class Cylinder : public TRS {
|
||||||
|
|
||||||
|
uLibTypeMacro(Cylinder, TRS)
|
||||||
|
ULIB_DECLARE_PROPERTIES(Cylinder)
|
||||||
|
|
||||||
public:
|
public:
|
||||||
uLibTypeMacro(Cylinder, AffineTransform)
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* @brief PROPERTIES
|
* @brief PROPERTIES
|
||||||
@@ -51,13 +53,12 @@ public:
|
|||||||
float Height;
|
float Height;
|
||||||
int Axis;
|
int Axis;
|
||||||
|
|
||||||
virtual const char * GetClassName() const override { return "Cylinder"; }
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* @brief Default constructor. Aligns with Y by default.
|
* @brief Default constructor. Aligns with Y by default.
|
||||||
*/
|
*/
|
||||||
Cylinder() : m_LocalT(this), Radius(1.0), Height(1.0), Axis(1) {
|
Cylinder() : m_LocalT(this), Radius(1.0), Height(1.0), Axis(1) {
|
||||||
ULIB_ACTIVATE_PROPERTIES(*this);
|
|
||||||
this->Sync();
|
this->Sync();
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -66,7 +67,6 @@ public:
|
|||||||
*/
|
*/
|
||||||
Cylinder(float radius, float height, int axis = 1)
|
Cylinder(float radius, float height, int axis = 1)
|
||||||
: m_LocalT(this), Radius(radius), Height(height), Axis(axis) {
|
: m_LocalT(this), Radius(radius), Height(height), Axis(axis) {
|
||||||
ULIB_ACTIVATE_PROPERTIES(*this);
|
|
||||||
this->Sync();
|
this->Sync();
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -74,8 +74,7 @@ public:
|
|||||||
* @brief Copy constructor.
|
* @brief Copy constructor.
|
||||||
*/
|
*/
|
||||||
Cylinder(const Cylinder ©)
|
Cylinder(const Cylinder ©)
|
||||||
: m_LocalT(this), AffineTransform(copy), Radius(copy.Radius), Height(copy.Height), Axis(copy.Axis) {
|
: m_LocalT(this), TRS(copy), Radius(copy.Radius), Height(copy.Height), Axis(copy.Axis) {
|
||||||
ULIB_ACTIVATE_PROPERTIES(*this);
|
|
||||||
this->Sync();
|
this->Sync();
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -86,7 +85,8 @@ public:
|
|||||||
void serialize(ArchiveT & ar, const unsigned int version) {
|
void serialize(ArchiveT & ar, const unsigned int version) {
|
||||||
ar & HRP(Radius);
|
ar & HRP(Radius);
|
||||||
ar & HRP(Height);
|
ar & HRP(Height);
|
||||||
ar & HRP(Axis);
|
ar & boost::serialization::make_hrp_enum("Axis", Axis, {"X", "Y", "Z"});
|
||||||
|
ar & NVP("TRS", boost::serialization::base_object<TRS>(*this));
|
||||||
}
|
}
|
||||||
|
|
||||||
/** Sets the radius of the cylinder */
|
/** Sets the radius of the cylinder */
|
||||||
@@ -176,8 +176,11 @@ public:
|
|||||||
signals:
|
signals:
|
||||||
/** Signal emitted when properties change */
|
/** Signal emitted when properties change */
|
||||||
virtual void Updated() override {
|
virtual void Updated() override {
|
||||||
|
// 1. Synchronize local cylinder part (Radius/Height/Axis -> m_LocalT)
|
||||||
this->Sync();
|
this->Sync();
|
||||||
ULIB_SIGNAL_EMIT(Object::Updated);
|
|
||||||
|
// 2. Synchronize TRS part (position/rotation/scaling -> m_T) and emit signal
|
||||||
|
this->TRS::Updated();
|
||||||
}
|
}
|
||||||
|
|
||||||
private:
|
private:
|
||||||
|
|||||||
@@ -35,11 +35,21 @@
|
|||||||
|
|
||||||
namespace uLib {
|
namespace uLib {
|
||||||
|
|
||||||
class Geometry : public AffineTransform {
|
|
||||||
public:
|
|
||||||
uLibTypeMacro(Geometry, AffineTransform)
|
|
||||||
|
|
||||||
virtual const char * GetClassName() const override { return "Geometry"; }
|
class Geometry : virtual public Object {
|
||||||
|
protected:
|
||||||
|
Geometry* m_Parent = nullptr;
|
||||||
|
|
||||||
|
public:
|
||||||
|
uLibTypeMacro(Geometry, Object)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
virtual void SetParent(Geometry* p) { m_Parent = p; }
|
||||||
|
virtual Geometry* GetParent() const { return m_Parent; }
|
||||||
|
|
||||||
|
virtual bool IsLinear() const { return false; }
|
||||||
|
virtual bool IsPure() const { return false; }
|
||||||
|
|
||||||
virtual Vector3f ToLinear(const Vector3f& curved_space) const {
|
virtual Vector3f ToLinear(const Vector3f& curved_space) const {
|
||||||
return curved_space;
|
return curved_space;
|
||||||
@@ -49,38 +59,120 @@ public:
|
|||||||
return cartesian_space;
|
return cartesian_space;
|
||||||
}
|
}
|
||||||
|
|
||||||
inline Vector4f GetWorldPoint(const Vector4f v) const {
|
virtual Vector4f GetWorldPoint(const Vector4f v) const = 0;
|
||||||
Vector3f lin = ToLinear(Vector3f(v.x(), v.y(), v.z()));
|
virtual Vector4f GetLocalPoint(const Vector4f v) const = 0;
|
||||||
return this->GetWorldMatrix() * Vector4f(lin.x(), lin.y(), lin.z(), v.w());
|
|
||||||
|
virtual Vector4f GetWorldPoint(const float x, const float y, const float z) const {
|
||||||
|
return GetWorldPoint(Vector4f(x,y,z,1));
|
||||||
}
|
}
|
||||||
|
|
||||||
inline Vector4f GetWorldPoint(const float x, const float y, const float z) {
|
virtual Vector4f GetLocalPoint(const float x, const float y, const float z) const {
|
||||||
return this->GetWorldPoint(Vector4f(x,y,z,1));
|
return GetLocalPoint(Vector4f(x,y,z,1));
|
||||||
}
|
}
|
||||||
|
|
||||||
inline Vector4f GetLocalPoint(const Vector4f v) const {
|
virtual Vector4f GetWorldPoint(const Vector3f v) const {
|
||||||
Vector4f loc_lin = this->GetWorldMatrix().inverse() * v;
|
return GetWorldPoint(Vector4f(v.x(), v.y(), v.z(), 1.0f));
|
||||||
Vector3f curv = FromLinear(Vector3f(loc_lin.x(), loc_lin.y(), loc_lin.z()));
|
|
||||||
return Vector4f(curv.x(), curv.y(), curv.z(), loc_lin.w());
|
|
||||||
}
|
}
|
||||||
|
|
||||||
inline Vector4f GetLocalPoint(const float x, const float y, const float z) {
|
virtual Vector4f GetLocalPoint(const Vector3f v) const {
|
||||||
return this->GetLocalPoint(Vector4f(x,y,z,1));
|
return GetLocalPoint(Vector4f(v.x(), v.y(), v.z(), 1.0f));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
virtual void Translate(Vector3f t) = 0;
|
||||||
|
virtual void Rotate(Vector3f r) = 0;
|
||||||
|
virtual void Scale(Vector3f s) = 0;
|
||||||
|
|
||||||
};
|
};
|
||||||
|
|
||||||
class CylindricalGeometry : public Geometry {
|
|
||||||
|
|
||||||
|
class LinearGeometry : public Geometry {
|
||||||
|
protected:
|
||||||
|
Affine3f m_T = Affine3f::Identity();
|
||||||
|
|
||||||
public:
|
public:
|
||||||
uLibTypeMacro(CylindricalGeometry, Geometry)
|
uLibTypeMacro(LinearGeometry, Geometry)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
virtual bool IsLinear() const override { return true; }
|
||||||
|
virtual bool IsPure() const override { return true; }
|
||||||
|
|
||||||
|
virtual Vector4f GetWorldPoint(const Vector4f v) const override {
|
||||||
|
Vector3f lin_v = ToLinear(v.head<3>());
|
||||||
|
Vector4f v_lin(lin_v.x(), lin_v.y(), lin_v.z(), v.w());
|
||||||
|
|
||||||
|
Affine3f combined = m_T;
|
||||||
|
const Geometry* curr = m_Parent;
|
||||||
|
while (curr && curr->IsLinear() && curr->IsPure()) {
|
||||||
|
combined = static_cast<const LinearGeometry*>(curr)->m_T * combined;
|
||||||
|
curr = curr->GetParent();
|
||||||
|
}
|
||||||
|
|
||||||
|
Vector4f v_res = combined.matrix() * v_lin;
|
||||||
|
if (curr) return curr->GetWorldPoint(v_res);
|
||||||
|
return v_res;
|
||||||
|
}
|
||||||
|
|
||||||
|
virtual Vector4f GetLocalPoint(const Vector4f v) const override {
|
||||||
|
Vector4f v_parent = m_Parent ? m_Parent->GetLocalPoint(v) : v;
|
||||||
|
Vector4f v_loc_lin = m_T.inverse().matrix() * v_parent;
|
||||||
|
Vector3f v_curv = FromLinear(v_loc_lin.head<3>());
|
||||||
|
return Vector4f(v_curv.x(), v_curv.y(), v_curv.z(), v_loc_lin.w());
|
||||||
|
}
|
||||||
|
|
||||||
|
virtual void Translate(Vector3f t) override {
|
||||||
|
m_T.translate(t);
|
||||||
|
}
|
||||||
|
|
||||||
|
virtual void Rotate(Vector3f r) override {
|
||||||
|
this->EulerYZYRotate(r);
|
||||||
|
}
|
||||||
|
|
||||||
|
virtual void Scale(Vector3f s) override {
|
||||||
|
m_T.scale(s);
|
||||||
|
}
|
||||||
|
|
||||||
|
void SetPosition(const Vector3f& v) { m_T.translation() = v; }
|
||||||
|
Vector3f GetPosition() const { return m_T.translation(); }
|
||||||
|
|
||||||
|
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);
|
||||||
|
}
|
||||||
|
|
||||||
|
const Affine3f& GetTransform() const { return m_T; }
|
||||||
|
void SetTransform(const Affine3f& t) { m_T = t; }
|
||||||
|
};
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
class CylindricalGeometry : public LinearGeometry {
|
||||||
|
public:
|
||||||
|
uLibTypeMacro(CylindricalGeometry, LinearGeometry)
|
||||||
CylindricalGeometry() {}
|
CylindricalGeometry() {}
|
||||||
|
|
||||||
Vector3f ToLinear(const Vector3f& cylindrical) const {
|
|
||||||
|
|
||||||
|
virtual bool IsPure() const override { return false; }
|
||||||
|
|
||||||
|
Vector3f ToLinear(const Vector3f& cylindrical) const override {
|
||||||
return Vector3f(cylindrical.x() * std::cos(cylindrical.y()),
|
return Vector3f(cylindrical.x() * std::cos(cylindrical.y()),
|
||||||
cylindrical.x() * std::sin(cylindrical.y()),
|
cylindrical.x() * std::sin(cylindrical.y()),
|
||||||
cylindrical.z());
|
cylindrical.z());
|
||||||
}
|
}
|
||||||
|
|
||||||
Vector3f FromLinear(const Vector3f& linear) const {
|
Vector3f FromLinear(const Vector3f& linear) const override {
|
||||||
float r = std::sqrt(linear.x() * linear.x() + linear.y() * linear.y());
|
float r = std::sqrt(linear.x() * linear.x() + linear.y() * linear.y());
|
||||||
float phi = std::atan2(linear.y(), linear.x());
|
float phi = std::atan2(linear.y(), linear.x());
|
||||||
return Vector3f(r, phi, linear.z());
|
return Vector3f(r, phi, linear.z());
|
||||||
@@ -88,14 +180,16 @@ public:
|
|||||||
|
|
||||||
};
|
};
|
||||||
|
|
||||||
class SphericalGeometry : public Geometry {
|
class SphericalGeometry : public LinearGeometry {
|
||||||
public:
|
public:
|
||||||
uLibTypeMacro(SphericalGeometry, Geometry)
|
uLibTypeMacro(SphericalGeometry, LinearGeometry)
|
||||||
SphericalGeometry() {}
|
SphericalGeometry() {}
|
||||||
|
|
||||||
virtual const char * GetClassName() const override { return "SphericalGeometry"; }
|
|
||||||
|
|
||||||
Vector3f ToLinear(const Vector3f& spherical) const {
|
|
||||||
|
virtual bool IsPure() const override { return false; }
|
||||||
|
|
||||||
|
Vector3f ToLinear(const Vector3f& spherical) const override {
|
||||||
float r = spherical.x();
|
float r = spherical.x();
|
||||||
float theta = spherical.y();
|
float theta = spherical.y();
|
||||||
float phi = spherical.z();
|
float phi = spherical.z();
|
||||||
@@ -104,7 +198,7 @@ public:
|
|||||||
r * std::cos(theta));
|
r * std::cos(theta));
|
||||||
}
|
}
|
||||||
|
|
||||||
Vector3f FromLinear(const Vector3f& linear) const {
|
Vector3f FromLinear(const Vector3f& linear) const override {
|
||||||
float r = linear.norm();
|
float r = linear.norm();
|
||||||
float theta = (r == 0.0f) ? 0.0f : std::acos(linear.z() / r);
|
float theta = (r == 0.0f) ? 0.0f : std::acos(linear.z() / r);
|
||||||
float phi = std::atan2(linear.y(), linear.x());
|
float phi = std::atan2(linear.y(), linear.x());
|
||||||
@@ -113,14 +207,16 @@ public:
|
|||||||
|
|
||||||
};
|
};
|
||||||
|
|
||||||
class ToroidalGeometry : public Geometry {
|
class ToroidalGeometry : public LinearGeometry {
|
||||||
public:
|
public:
|
||||||
uLibTypeMacro(ToroidalGeometry, Geometry)
|
uLibTypeMacro(ToroidalGeometry, LinearGeometry)
|
||||||
ToroidalGeometry(float Rtor) : m_Rtor(Rtor) {}
|
ToroidalGeometry(float Rtor) : m_Rtor(Rtor) {}
|
||||||
|
|
||||||
virtual const char * GetClassName() const override { return "ToroidalGeometry"; }
|
|
||||||
|
|
||||||
Vector3f ToLinear(const Vector3f& toroidal) const {
|
|
||||||
|
virtual bool IsPure() const override { return false; }
|
||||||
|
|
||||||
|
Vector3f ToLinear(const Vector3f& toroidal) const override {
|
||||||
float r = toroidal.x();
|
float r = toroidal.x();
|
||||||
float theta = toroidal.y();
|
float theta = toroidal.y();
|
||||||
float phi = toroidal.z();
|
float phi = toroidal.z();
|
||||||
@@ -129,7 +225,7 @@ public:
|
|||||||
r * std::sin(theta));
|
r * std::sin(theta));
|
||||||
}
|
}
|
||||||
|
|
||||||
Vector3f FromLinear(const Vector3f& linear) const {
|
Vector3f FromLinear(const Vector3f& linear) const override {
|
||||||
float phi = std::atan2(linear.y(), linear.x());
|
float phi = std::atan2(linear.y(), linear.x());
|
||||||
float r_xy = std::sqrt(linear.x() * linear.x() + linear.y() * linear.y());
|
float r_xy = std::sqrt(linear.x() * linear.x() + linear.y() * linear.y());
|
||||||
float delta_r = r_xy - m_Rtor;
|
float delta_r = r_xy - m_Rtor;
|
||||||
|
|||||||
@@ -1,42 +0,0 @@
|
|||||||
SUBDIRS = .
|
|
||||||
|
|
||||||
include $(top_srcdir)/Common.am
|
|
||||||
|
|
||||||
library_includedir = $(includedir)/libmutom-${PACKAGE_VERSION}/Math
|
|
||||||
library_include_HEADERS = ContainerBox.h \
|
|
||||||
Dense.h \
|
|
||||||
Geometry.h \
|
|
||||||
Transform.h \
|
|
||||||
StructuredData.h\
|
|
||||||
StructuredGrid.h\
|
|
||||||
VoxImage.h \
|
|
||||||
VoxRaytracer.h \
|
|
||||||
Utils.h \
|
|
||||||
VoxImageFilter.h\
|
|
||||||
VoxImageFilter.hpp \
|
|
||||||
VoxImageFilterLinear.hpp \
|
|
||||||
VoxImageFilterMedian.hpp \
|
|
||||||
VoxImageFilterABTrim.hpp \
|
|
||||||
VoxImageFilterBilateral.hpp \
|
|
||||||
VoxImageFilterThreshold.hpp \
|
|
||||||
VoxImageFilter2ndStat.hpp \
|
|
||||||
VoxImageFilterCustom.hpp \
|
|
||||||
Accumulator.h \
|
|
||||||
TriangleMesh.h
|
|
||||||
|
|
||||||
|
|
||||||
_MATH_SOURCES = \
|
|
||||||
VoxRaytracer.cpp \
|
|
||||||
StructuredData.cpp \
|
|
||||||
StructuredGrid.cpp \
|
|
||||||
VoxImage.cpp \
|
|
||||||
TriangleMesh.cpp \
|
|
||||||
Dense.cpp
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
noinst_LTLIBRARIES = libmutommath.la
|
|
||||||
libmutommath_la_SOURCES = ${_MATH_SOURCES}
|
|
||||||
|
|
||||||
|
|
||||||
@@ -10,6 +10,7 @@
|
|||||||
|
|
||||||
namespace uLib {
|
namespace uLib {
|
||||||
|
|
||||||
|
ULIB_REGISTER_OBJECT(TRS)
|
||||||
ULIB_REGISTER_OBJECT(ContainerBox)
|
ULIB_REGISTER_OBJECT(ContainerBox)
|
||||||
ULIB_REGISTER_OBJECT(Cylinder)
|
ULIB_REGISTER_OBJECT(Cylinder)
|
||||||
ULIB_REGISTER_OBJECT(Assembly)
|
ULIB_REGISTER_OBJECT(Assembly)
|
||||||
|
|||||||
@@ -36,7 +36,7 @@ class Polydata : public Object {
|
|||||||
|
|
||||||
public:
|
public:
|
||||||
|
|
||||||
virtual const char * GetClassName() const { return "Polydata"; }
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -34,12 +34,12 @@
|
|||||||
|
|
||||||
namespace uLib {
|
namespace uLib {
|
||||||
|
|
||||||
class QuadMesh : public AffineTransform
|
class QuadMesh : public TRS
|
||||||
{
|
{
|
||||||
public:
|
public:
|
||||||
uLibTypeMacro(QuadMesh, AffineTransform)
|
uLibTypeMacro(QuadMesh, TRS)
|
||||||
|
|
||||||
|
|
||||||
virtual const char * GetClassName() const override { return "QuadMesh"; }
|
|
||||||
|
|
||||||
void PrintSelf(std::ostream &o);
|
void PrintSelf(std::ostream &o);
|
||||||
|
|
||||||
|
|||||||
@@ -57,21 +57,153 @@
|
|||||||
namespace uLib {
|
namespace uLib {
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
using Eigen::Isometry3f;
|
||||||
|
using Eigen::Isometry3d;
|
||||||
|
|
||||||
|
using Eigen::Affine3f;
|
||||||
|
using Eigen::Affine3d;
|
||||||
|
|
||||||
|
using Eigen::Projective3f;
|
||||||
|
using Eigen::Projective3d;
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
|
///////// AFFINE TRANSFORM WRAPPER //////////////////////////////////////////
|
||||||
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
class AffineTransform : virtual public Object {
|
||||||
|
public:
|
||||||
|
uLibTypeMacro(AffineTransform, Object)
|
||||||
|
protected:
|
||||||
|
|
||||||
|
Affine3f m_T;
|
||||||
|
AffineTransform *m_Parent;
|
||||||
|
|
||||||
|
public:
|
||||||
|
AffineTransform() :
|
||||||
|
m_T(Matrix4f::Identity()),
|
||||||
|
m_Parent(NULL)
|
||||||
|
{}
|
||||||
|
|
||||||
|
AffineTransform(AffineTransform *parent) :
|
||||||
|
m_T(Matrix4f::Identity()),
|
||||||
|
m_Parent(parent)
|
||||||
|
{}
|
||||||
|
|
||||||
|
AffineTransform(const AffineTransform ©) :
|
||||||
|
m_T(copy.m_T),
|
||||||
|
m_Parent(copy.m_Parent)
|
||||||
|
{}
|
||||||
|
|
||||||
|
Affine3f& GetTransform() { return m_T; }
|
||||||
|
|
||||||
|
AffineTransform *GetParent() const { return this->m_Parent; }
|
||||||
|
|
||||||
|
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(); }
|
||||||
|
|
||||||
|
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 SetPosition(const Vector3f &v) { this->m_T.translation() = v; }
|
||||||
|
|
||||||
|
Vector3f GetPosition() const { return this->m_T.translation(); }
|
||||||
|
|
||||||
|
void SetRotation(const Matrix3f &m) { this->m_T.linear() = m; }
|
||||||
|
|
||||||
|
Matrix3f GetRotation() const { return this->m_T.rotation(); }
|
||||||
|
|
||||||
|
void Translate(const Vector3f &v) { this->m_T.translate(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());
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
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 Vector3f euler_axis) {
|
||||||
|
float angle = euler_axis.norm();
|
||||||
|
Rotate(angle,euler_axis);
|
||||||
|
}
|
||||||
|
|
||||||
|
void PreRotate(const Matrix3f &m) { this->m_T.prerotate(m); }
|
||||||
|
|
||||||
|
void QuaternionRotate(const Vector4f &q)
|
||||||
|
{ this->m_T.rotate(Eigen::Quaternion<float>(q)); }
|
||||||
|
|
||||||
|
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 /////////////////////////////////////////////////////
|
///////// TRS PARAMETERS /////////////////////////////////////////////////////
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
|
|
||||||
typedef Eigen::Affine3f AffineMatrix;
|
typedef Eigen::Affine3f AffineMatrix;
|
||||||
|
|
||||||
class TRS {
|
class TRS : public AffineTransform {
|
||||||
|
|
||||||
|
uLibTypeMacro(TRS, AffineTransform)
|
||||||
|
ULIB_SERIALIZE_ACCESS
|
||||||
|
// ULIB_DECLARE_PROPERTIES(TRS)
|
||||||
|
|
||||||
public:
|
public:
|
||||||
|
|
||||||
Vector3f position = Vector3f::Zero();
|
Vector3f position = Vector3f::Zero();
|
||||||
Vector3f rotation = Vector3f::Zero();
|
Vector3f rotation = Vector3f::Zero();
|
||||||
Vector3f scaling = Vector3f::Ones();
|
Vector3f scaling = Vector3f::Ones();
|
||||||
|
|
||||||
TRS() = default;
|
TRS() = default;
|
||||||
|
|
||||||
TRS(const class AffineTransform& at);
|
TRS(const class AffineTransform& at) {
|
||||||
|
this->FromMatrix(at.GetMatrix());
|
||||||
|
}
|
||||||
|
|
||||||
TRS(const Matrix4f& mat) {
|
TRS(const Matrix4f& mat) {
|
||||||
this->FromMatrix(mat);
|
this->FromMatrix(mat);
|
||||||
@@ -90,221 +222,82 @@ public:
|
|||||||
if (this->scaling(1) > 1e-6) rot.col(1) /= this->scaling(1);
|
if (this->scaling(1) > 1e-6) rot.col(1) /= this->scaling(1);
|
||||||
if (this->scaling(2) > 1e-6) rot.col(2) /= this->scaling(2);
|
if (this->scaling(2) > 1e-6) rot.col(2) /= this->scaling(2);
|
||||||
|
|
||||||
// Decompose to Euler angles matching VTK (M = Rz * Ry * Rx)
|
|
||||||
// Store internally as RADIANS (standard for uLib properties)
|
|
||||||
Vector3f euler = rot.eulerAngles(2, 1, 0);
|
Vector3f euler = rot.eulerAngles(2, 1, 0);
|
||||||
this->rotation = Vector3f(euler(2), euler(1), euler(0));
|
this->rotation = Vector3f(euler(2), euler(1), euler(0));
|
||||||
|
|
||||||
|
this->SetMatrix(mat);
|
||||||
|
this->NotifyPropertiesUpdated();
|
||||||
|
}
|
||||||
|
|
||||||
|
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();
|
||||||
|
}
|
||||||
|
|
||||||
|
void SyncMatrix() {
|
||||||
|
this->GetTransform() = GetAffineMatrix();
|
||||||
|
}
|
||||||
|
|
||||||
|
void Updated() override {
|
||||||
|
this->SyncMatrix();
|
||||||
|
this->NotifyPropertiesUpdated();
|
||||||
|
this->AffineTransform::Updated();
|
||||||
}
|
}
|
||||||
|
|
||||||
template <class ArchiveT>
|
template <class ArchiveT>
|
||||||
void serialize(ArchiveT & ar, const unsigned int version) {
|
void serialize(ArchiveT & ar, const unsigned int version) {
|
||||||
ar & HRPU(position, "mm");
|
ar & HRPU(position, "mm");
|
||||||
ar & HRPU(rotation, "deg"); // Metadata informs UI to convert to/from degrees
|
ar & HRPU(rotation, "rad");
|
||||||
ar & HRP(scaling);
|
ar & HRP(scaling);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
AffineMatrix GetAffineMatrix() const {
|
AffineMatrix GetAffineMatrix() const {
|
||||||
AffineMatrix t = AffineMatrix::Identity();
|
AffineMatrix m = AffineMatrix::Identity();
|
||||||
t.translate(position);
|
m.translate(position);
|
||||||
|
m.rotate(Eigen::AngleAxisf(rotation.z(), Vector3f::UnitZ()));
|
||||||
// rotation is in Radians here
|
m.rotate(Eigen::AngleAxisf(rotation.y(), Vector3f::UnitY()));
|
||||||
t.rotate(Eigen::AngleAxisf(rotation.z(), Vector3f::UnitZ()));
|
m.rotate(Eigen::AngleAxisf(rotation.x(), Vector3f::UnitX()));
|
||||||
t.rotate(Eigen::AngleAxisf(rotation.y(), Vector3f::UnitY()));
|
m.scale(scaling);
|
||||||
t.rotate(Eigen::AngleAxisf(rotation.x(), Vector3f::UnitX()));
|
return m;
|
||||||
|
|
||||||
t.scale(scaling);
|
|
||||||
return t;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
Matrix4f GetMatrix() const {
|
||||||
|
return this->GetAffineMatrix().matrix();
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
};
|
};
|
||||||
|
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
|
||||||
///////// AFFINE TRANSFORM WRAPPER //////////////////////////////////////////
|
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
|
||||||
|
|
||||||
class AffineTransform : virtual public Object {
|
|
||||||
public:
|
|
||||||
uLibTypeMacro(AffineTransform, Object)
|
|
||||||
|
|
||||||
TRS Transform;
|
|
||||||
|
|
||||||
private:
|
|
||||||
void NotifyProperties() {
|
inline std::ostream& operator<<(std::ostream& os, const TRS& trs) {
|
||||||
PropertyBase *p;
|
os << trs.position << " " << trs.rotation << " " << trs.scaling;
|
||||||
if ((p = this->GetProperty("Transform.position"))) p->Updated();
|
return os;
|
||||||
if ((p = this->GetProperty("Transform.rotation"))) p->Updated();
|
|
||||||
if ((p = this->GetProperty("Transform.scaling"))) p->Updated();
|
|
||||||
}
|
}
|
||||||
|
|
||||||
protected:
|
inline std::istream& operator>>(std::istream& is, TRS& trs) {
|
||||||
Eigen::Affine3f m_T;
|
is >> trs.position >> trs.rotation >> trs.scaling;
|
||||||
AffineTransform *m_Parent;
|
return is;
|
||||||
|
|
||||||
public:
|
|
||||||
AffineTransform() :
|
|
||||||
m_T(Matrix4f::Identity()),
|
|
||||||
m_Parent(NULL)
|
|
||||||
{
|
|
||||||
ULIB_ACTIVATE_PROPERTIES(*this);
|
|
||||||
this->Sync();
|
|
||||||
}
|
}
|
||||||
|
|
||||||
virtual ~AffineTransform() {}
|
} // uLib
|
||||||
|
|
||||||
AffineTransform(AffineTransform *parent) :
|
|
||||||
m_T(Matrix4f::Identity()),
|
|
||||||
m_Parent(parent)
|
|
||||||
{
|
|
||||||
ULIB_ACTIVATE_PROPERTIES(*this);
|
|
||||||
this->Sync();
|
|
||||||
}
|
|
||||||
|
|
||||||
AffineTransform(const AffineTransform ©) :
|
|
||||||
m_T(copy.m_T),
|
|
||||||
m_Parent(copy.m_Parent),
|
|
||||||
Transform(copy.Transform)
|
|
||||||
{
|
|
||||||
ULIB_ACTIVATE_PROPERTIES(*this);
|
|
||||||
this->Sync();
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* @brief Registration of properties in groups.
|
|
||||||
*/
|
|
||||||
template <class ArchiveT>
|
|
||||||
void serialize(ArchiveT & ar, const unsigned int version) {
|
|
||||||
ar & boost::serialization::make_nvp("Transform", Transform);
|
|
||||||
}
|
|
||||||
|
|
||||||
Eigen::Affine3f& GetTransform() { return m_T; }
|
|
||||||
|
|
||||||
AffineTransform *GetParent() const { return this->m_Parent; }
|
|
||||||
|
|
||||||
void SetParent(AffineTransform *name) { this->m_Parent = name; }
|
|
||||||
|
|
||||||
void SetMatrix (Matrix4f mat) {
|
|
||||||
m_T.matrix() = mat;
|
|
||||||
this->UpdatePropertiesFromMatrix();
|
|
||||||
}
|
|
||||||
|
|
||||||
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 //
|
|
||||||
}
|
|
||||||
|
|
||||||
void SetPosition(const Vector3f v) {
|
|
||||||
this->Transform.position = v;
|
|
||||||
this->Updated();
|
|
||||||
this->NotifyProperties();
|
|
||||||
}
|
|
||||||
Vector3f GetPosition() const { return this->Transform.position; }
|
|
||||||
|
|
||||||
void SetOrientation(const Vector3f v) {
|
|
||||||
this->Transform.rotation = v;
|
|
||||||
this->Updated();
|
|
||||||
this->NotifyProperties();
|
|
||||||
}
|
|
||||||
Vector3f GetOrientation() const { return this->Transform.rotation; }
|
|
||||||
|
|
||||||
void SetScale(const Vector3f v) {
|
|
||||||
this->Transform.scaling = v;
|
|
||||||
this->Updated();
|
|
||||||
this->NotifyProperties();
|
|
||||||
}
|
|
||||||
Vector3f GetScale() const { return this->Transform.scaling; }
|
|
||||||
|
|
||||||
void SetRotation(const Matrix3f m) {
|
|
||||||
this->m_T.linear() = m;
|
|
||||||
this->UpdatePropertiesFromMatrix();
|
|
||||||
}
|
|
||||||
|
|
||||||
Matrix3f GetRotation() const { return this->m_T.rotation(); }
|
|
||||||
|
|
||||||
void Translate(const Vector3f v) {
|
|
||||||
this->Transform.position += v;
|
|
||||||
this->Sync();
|
|
||||||
}
|
|
||||||
|
|
||||||
void Scale(const Vector3f v) {
|
|
||||||
this->Transform.scaling = this->Transform.scaling.cwiseProduct(v);
|
|
||||||
this->Sync();
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
void Rotate(const Matrix3f m) {
|
|
||||||
this->m_T.rotate(m);
|
|
||||||
this->UpdatePropertiesFromMatrix();
|
|
||||||
}
|
|
||||||
|
|
||||||
void Rotate(const float angle, Vector3f axis)
|
|
||||||
{
|
|
||||||
axis.normalize();
|
|
||||||
Eigen::AngleAxisf ax(angle,axis);
|
|
||||||
this->m_T.rotate(Eigen::Quaternion<float>(ax));
|
|
||||||
this->UpdatePropertiesFromMatrix();
|
|
||||||
}
|
|
||||||
|
|
||||||
void Rotate(const Vector3f euler_axis) {
|
|
||||||
float angle = euler_axis.norm();
|
|
||||||
Rotate(angle,euler_axis);
|
|
||||||
}
|
|
||||||
|
|
||||||
void PreRotate(const Matrix3f m) { this->m_T.prerotate(m); this->UpdatePropertiesFromMatrix(); }
|
|
||||||
|
|
||||||
void QuaternionRotate(const Vector4f q)
|
|
||||||
{ this->m_T.rotate(Eigen::Quaternion<float>(q)); this->UpdatePropertiesFromMatrix(); }
|
|
||||||
|
|
||||||
void EulerYZYRotate(const Vector3f e) {
|
|
||||||
this->Transform.rotation = e;
|
|
||||||
this->Sync();
|
|
||||||
}
|
|
||||||
|
|
||||||
void FlipAxes(int first, int second)
|
|
||||||
{
|
|
||||||
Matrix3f mat = Matrix3f::Identity();
|
|
||||||
mat.col(first).swap(mat.col(second));
|
|
||||||
m_T.rotate(mat);
|
|
||||||
this->UpdatePropertiesFromMatrix();
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* @brief Decomposes the internal matrix m_T back into Position, Orientation, and Scale properties.
|
|
||||||
*/
|
|
||||||
void UpdatePropertiesFromMatrix() {
|
|
||||||
this->Transform.FromMatrix(this->GetMatrix());
|
|
||||||
|
|
||||||
PropertyBase *p;
|
|
||||||
if ((p = this->GetProperty("Transform.position"))) p->Updated();
|
|
||||||
if ((p = this->GetProperty("Transform.rotation"))) p->Updated();
|
|
||||||
if ((p = this->GetProperty("Transform.scaling"))) p->Updated();
|
|
||||||
}
|
|
||||||
|
|
||||||
signals:
|
|
||||||
/** Signal emitted when properties change */
|
|
||||||
virtual void Updated() override {
|
|
||||||
this->Sync();
|
|
||||||
ULIB_SIGNAL_EMIT(Object::Updated);
|
|
||||||
}
|
|
||||||
|
|
||||||
private:
|
|
||||||
void Sync() {
|
|
||||||
m_T.matrix() = this->Transform.GetAffineMatrix().matrix();
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
inline TRS::TRS(const AffineTransform& at) {
|
|
||||||
this->position = at.GetPosition();
|
|
||||||
this->rotation = at.GetOrientation();
|
|
||||||
this->scaling = at.GetScale();
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -37,12 +37,12 @@
|
|||||||
|
|
||||||
namespace uLib {
|
namespace uLib {
|
||||||
|
|
||||||
class TriangleMesh : public AffineTransform
|
class TriangleMesh : public TRS
|
||||||
{
|
{
|
||||||
public:
|
public:
|
||||||
uLibTypeMacro(TriangleMesh, AffineTransform)
|
uLibTypeMacro(TriangleMesh, TRS)
|
||||||
|
|
||||||
|
|
||||||
virtual const char * GetClassName() const override { return "TriangleMesh"; }
|
|
||||||
|
|
||||||
void PrintSelf(std::ostream &o);
|
void PrintSelf(std::ostream &o);
|
||||||
|
|
||||||
|
|||||||
@@ -47,7 +47,7 @@ namespace Abstract {
|
|||||||
class VoxImage : public uLib::StructuredGrid {
|
class VoxImage : public uLib::StructuredGrid {
|
||||||
public:
|
public:
|
||||||
|
|
||||||
virtual const char * GetClassName() const { return "VoxImage"; }
|
|
||||||
|
|
||||||
typedef uLib::StructuredGrid BaseClass;
|
typedef uLib::StructuredGrid BaseClass;
|
||||||
|
|
||||||
|
|||||||
@@ -61,7 +61,7 @@ class VoxImageFilter : public Abstract::VoxImageFilter, public Object {
|
|||||||
|
|
||||||
public:
|
public:
|
||||||
|
|
||||||
virtual const char * GetClassName() const { return "VoxImageFilter"; }
|
|
||||||
|
|
||||||
VoxImageFilter(const Vector3i &size);
|
VoxImageFilter(const Vector3i &size);
|
||||||
|
|
||||||
|
|||||||
@@ -98,15 +98,14 @@ template <typename T> void Kernel<T>::PrintSelf(std::ostream &o) const {
|
|||||||
|
|
||||||
////////////////////////////////////////////////////////////////////////////////
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
|
|
||||||
#define _TPL_ template <typename VoxelT, typename AlgorithmT>
|
|
||||||
#define _TPLT_ VoxelT, AlgorithmT
|
|
||||||
|
|
||||||
_TPL_
|
|
||||||
VoxImageFilter<_TPLT_>::VoxImageFilter(const Vector3i &size)
|
template <typename VoxelT, typename AlgorithmT>
|
||||||
|
VoxImageFilter<VoxelT, AlgorithmT>::VoxImageFilter(const Vector3i &size)
|
||||||
: m_KernelData(size), t_Algoritm(static_cast<AlgorithmT *>(this)) {}
|
: m_KernelData(size), t_Algoritm(static_cast<AlgorithmT *>(this)) {}
|
||||||
|
|
||||||
_TPL_
|
template <typename VoxelT, typename AlgorithmT>
|
||||||
void VoxImageFilter<_TPLT_>::Run() {
|
void VoxImageFilter<VoxelT, AlgorithmT>::Run() {
|
||||||
VoxImage<VoxelT> buffer = *m_Image;
|
VoxImage<VoxelT> buffer = *m_Image;
|
||||||
#pragma omp parallel for
|
#pragma omp parallel for
|
||||||
for (int i = 0; i < m_Image->Data().size(); ++i)
|
for (int i = 0; i < m_Image->Data().size(); ++i)
|
||||||
@@ -114,8 +113,8 @@ void VoxImageFilter<_TPLT_>::Run() {
|
|||||||
#pragma omp barrier
|
#pragma omp barrier
|
||||||
}
|
}
|
||||||
|
|
||||||
_TPL_
|
template <typename VoxelT, typename AlgorithmT>
|
||||||
void VoxImageFilter<_TPLT_>::SetKernelOffset() {
|
void VoxImageFilter<VoxelT, AlgorithmT>::SetKernelOffset() {
|
||||||
Vector3i id(0, 0, 0);
|
Vector3i id(0, 0, 0);
|
||||||
for (int z = 0; z < m_KernelData.GetDims()(2); ++z) {
|
for (int z = 0; z < m_KernelData.GetDims()(2); ++z) {
|
||||||
for (int x = 0; x < m_KernelData.GetDims()(0); ++x) {
|
for (int x = 0; x < m_KernelData.GetDims()(0); ++x) {
|
||||||
@@ -127,8 +126,8 @@ void VoxImageFilter<_TPLT_>::SetKernelOffset() {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
_TPL_
|
template <typename VoxelT, typename AlgorithmT>
|
||||||
float VoxImageFilter<_TPLT_>::Distance2(const Vector3i &v) {
|
float VoxImageFilter<VoxelT, AlgorithmT>::Distance2(const Vector3i &v) {
|
||||||
Vector3i tmp = v;
|
Vector3i tmp = v;
|
||||||
const Vector3i &dim = this->m_KernelData.GetDims();
|
const Vector3i &dim = this->m_KernelData.GetDims();
|
||||||
Vector3i center = dim / 2;
|
Vector3i center = dim / 2;
|
||||||
@@ -140,8 +139,8 @@ float VoxImageFilter<_TPLT_>::Distance2(const Vector3i &v) {
|
|||||||
0.25 * (3 - (dim(0) % 2) - (dim(1) % 2) - (dim(2) % 2)));
|
0.25 * (3 - (dim(0) % 2) - (dim(1) % 2) - (dim(2) % 2)));
|
||||||
}
|
}
|
||||||
|
|
||||||
_TPL_
|
template <typename VoxelT, typename AlgorithmT>
|
||||||
void VoxImageFilter<_TPLT_>::SetKernelNumericXZY(
|
void VoxImageFilter<VoxelT, AlgorithmT>::SetKernelNumericXZY(
|
||||||
const std::vector<float> &numeric) {
|
const std::vector<float> &numeric) {
|
||||||
// set data order //
|
// set data order //
|
||||||
StructuredData::Order order = m_KernelData.GetDataOrder();
|
StructuredData::Order order = m_KernelData.GetDataOrder();
|
||||||
@@ -159,8 +158,8 @@ void VoxImageFilter<_TPLT_>::SetKernelNumericXZY(
|
|||||||
// m_KernelData.SetDataOrder(order);
|
// m_KernelData.SetDataOrder(order);
|
||||||
}
|
}
|
||||||
|
|
||||||
_TPL_
|
template <typename VoxelT, typename AlgorithmT>
|
||||||
void VoxImageFilter<_TPLT_>::SetKernelSpherical(float (*shape)(float)) {
|
void VoxImageFilter<VoxelT, AlgorithmT>::SetKernelSpherical(float (*shape)(float)) {
|
||||||
Vector3i id;
|
Vector3i id;
|
||||||
for (int y = 0; y < m_KernelData.GetDims()(1); ++y) {
|
for (int y = 0; y < m_KernelData.GetDims()(1); ++y) {
|
||||||
for (int z = 0; z < m_KernelData.GetDims()(2); ++z) {
|
for (int z = 0; z < m_KernelData.GetDims()(2); ++z) {
|
||||||
@@ -172,8 +171,8 @@ void VoxImageFilter<_TPLT_>::SetKernelSpherical(float (*shape)(float)) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
_TPL_ template <class ShapeT>
|
template <typename VoxelT, typename AlgorithmT> template <class ShapeT>
|
||||||
void VoxImageFilter<_TPLT_>::SetKernelSpherical(ShapeT shape) {
|
void VoxImageFilter<VoxelT, AlgorithmT>::SetKernelSpherical(ShapeT shape) {
|
||||||
Interface::IsA<ShapeT, Interface::VoxImageFilterShape>();
|
Interface::IsA<ShapeT, Interface::VoxImageFilterShape>();
|
||||||
Vector3i id;
|
Vector3i id;
|
||||||
for (int y = 0; y < m_KernelData.GetDims()(1); ++y) {
|
for (int y = 0; y < m_KernelData.GetDims()(1); ++y) {
|
||||||
@@ -186,8 +185,8 @@ void VoxImageFilter<_TPLT_>::SetKernelSpherical(ShapeT shape) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
_TPL_
|
template <typename VoxelT, typename AlgorithmT>
|
||||||
void VoxImageFilter<_TPLT_>::SetKernelWeightFunction(
|
void VoxImageFilter<VoxelT, AlgorithmT>::SetKernelWeightFunction(
|
||||||
float (*shape)(const Vector3f &)) {
|
float (*shape)(const Vector3f &)) {
|
||||||
const Vector3i &dim = m_KernelData.GetDims();
|
const Vector3i &dim = m_KernelData.GetDims();
|
||||||
Vector3i id;
|
Vector3i id;
|
||||||
@@ -207,8 +206,8 @@ void VoxImageFilter<_TPLT_>::SetKernelWeightFunction(
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
_TPL_ template <class ShapeT>
|
template <typename VoxelT, typename AlgorithmT> template <class ShapeT>
|
||||||
void VoxImageFilter<_TPLT_>::SetKernelWeightFunction(ShapeT shape) {
|
void VoxImageFilter<VoxelT, AlgorithmT>::SetKernelWeightFunction(ShapeT shape) {
|
||||||
Interface::IsA<ShapeT, Interface::VoxImageFilterShape>();
|
Interface::IsA<ShapeT, Interface::VoxImageFilterShape>();
|
||||||
const Vector3i &dim = m_KernelData.GetDims();
|
const Vector3i &dim = m_KernelData.GetDims();
|
||||||
Vector3i id;
|
Vector3i id;
|
||||||
@@ -228,14 +227,14 @@ void VoxImageFilter<_TPLT_>::SetKernelWeightFunction(ShapeT shape) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
_TPL_
|
template <typename VoxelT, typename AlgorithmT>
|
||||||
void VoxImageFilter<_TPLT_>::SetImage(Abstract::VoxImage *image) {
|
void VoxImageFilter<VoxelT, AlgorithmT>::SetImage(Abstract::VoxImage *image) {
|
||||||
this->m_Image = reinterpret_cast<VoxImage<VoxelT> *>(image);
|
this->m_Image = reinterpret_cast<VoxImage<VoxelT> *>(image);
|
||||||
this->SetKernelOffset();
|
this->SetKernelOffset();
|
||||||
}
|
}
|
||||||
|
|
||||||
_TPL_
|
template <typename VoxelT, typename AlgorithmT>
|
||||||
float VoxImageFilter<_TPLT_>::Convolve(const VoxImage<VoxelT> &buffer,
|
float VoxImageFilter<VoxelT, AlgorithmT>::Convolve(const VoxImage<VoxelT> &buffer,
|
||||||
int index) {
|
int index) {
|
||||||
const DataAllocator<VoxelT> &vbuf = buffer.ConstData();
|
const DataAllocator<VoxelT> &vbuf = buffer.ConstData();
|
||||||
const DataAllocator<VoxelT> &vker = m_KernelData.ConstData();
|
const DataAllocator<VoxelT> &vker = m_KernelData.ConstData();
|
||||||
@@ -252,8 +251,8 @@ float VoxImageFilter<_TPLT_>::Convolve(const VoxImage<VoxelT> &buffer,
|
|||||||
return conv / ksum;
|
return conv / ksum;
|
||||||
}
|
}
|
||||||
|
|
||||||
#undef _TPLT_
|
|
||||||
#undef _TPL_
|
|
||||||
|
|
||||||
} // namespace uLib
|
} // namespace uLib
|
||||||
|
|
||||||
|
|||||||
@@ -53,7 +53,7 @@ int main()
|
|||||||
///////////////// GEOMETRY TESTING ///////////////////////////////////////////
|
///////////////// GEOMETRY TESTING ///////////////////////////////////////////
|
||||||
//////////////////////////////////////////////////////////////////////////////
|
//////////////////////////////////////////////////////////////////////////////
|
||||||
|
|
||||||
Geometry Geo;
|
LinearGeometry Geo;
|
||||||
|
|
||||||
Geo.SetPosition(Vector3f(1,1,1));
|
Geo.SetPosition(Vector3f(1,1,1));
|
||||||
Geo.EulerYZYRotate(Vector3f(0,0,0));
|
Geo.EulerYZYRotate(Vector3f(0,0,0));
|
||||||
@@ -77,7 +77,7 @@ int main()
|
|||||||
|
|
||||||
Geo.EulerYZYRotate(Vector3f(0,0,M_PI_2));
|
Geo.EulerYZYRotate(Vector3f(0,0,M_PI_2));
|
||||||
wp = Geo.GetWorldPoint(HPoint3f(1,1,1));
|
wp = Geo.GetWorldPoint(HPoint3f(1,1,1));
|
||||||
// std::cout << "Geometry matrix\n" << Geo.GetTransform() << "\n";
|
// std::cout << "Geometry matrix\n" << Geo.GetTransform().matrix() << "\n";
|
||||||
// std::cout << "World 1,1,1 coords\n" << wp << "\n";
|
// std::cout << "World 1,1,1 coords\n" << wp << "\n";
|
||||||
TEST0( Vector4f0(wp - HPoint3f(0,2,2)) );
|
TEST0( Vector4f0(wp - HPoint3f(0,2,2)) );
|
||||||
|
|
||||||
@@ -122,6 +122,27 @@ int main()
|
|||||||
TEST0( Vector4f0(recovered.homogeneous() - tor_pt.homogeneous()) );
|
TEST0( Vector4f0(recovered.homogeneous() - tor_pt.homogeneous()) );
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// PARENT GEOMETRY TESTING
|
||||||
|
{
|
||||||
|
LinearGeometry parent;
|
||||||
|
parent.Translate(Vector3f(10, 0, 0));
|
||||||
|
|
||||||
|
LinearGeometry child;
|
||||||
|
child.SetParent(&parent);
|
||||||
|
child.Translate(Vector3f(0, 5, 0));
|
||||||
|
|
||||||
|
HPoint3f wp = child.GetWorldPoint(HPoint3f(1, 1, 1));
|
||||||
|
TEST0( Vector4f0(wp - HPoint3f(11, 6, 1)) );
|
||||||
|
|
||||||
|
CylindricalGeometry cparent;
|
||||||
|
LinearGeometry grandchild;
|
||||||
|
grandchild.SetParent(&cparent);
|
||||||
|
grandchild.Translate(Vector3f(1, 0, 0));
|
||||||
|
|
||||||
|
HPoint3f gp = grandchild.GetWorldPoint(HPoint3f(1, M_PI_2, 0));
|
||||||
|
TEST0( Vector4f0(gp - HPoint3f(0, 2, 0)) );
|
||||||
|
}
|
||||||
|
|
||||||
END_TESTING;
|
END_TESTING;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -31,8 +31,10 @@
|
|||||||
static int _fail = 0; \
|
static int _fail = 0; \
|
||||||
printf("..:: Testing " #name " ::..\n");
|
printf("..:: Testing " #name " ::..\n");
|
||||||
|
|
||||||
#define TEST1(val) if ((val)==0) { printf("Assertion failed: %s != 0\n", #val); _fail++; }
|
#define TEST1(val) _fail += (val)==0
|
||||||
#define TEST0(val) if ((val)!=0) { printf("Assertion failed: %s != 0\n", #val); _fail++; }
|
#define TEST0(val) _fail += (val)!=0
|
||||||
|
#define ASSERT_EQUAL(a,b) if((a)!=(b)) { printf("Assertion failed: " #a " != " #b " at line %d\n", __LINE__); _fail++; }
|
||||||
|
#define ASSERT_NOT_NULL(ptr) if((ptr)==NULL) { printf("Assertion failed: " #ptr " is NULL at line %d\n", __LINE__); _fail++; }
|
||||||
#define END_TESTING return _fail;
|
#define END_TESTING return _fail;
|
||||||
|
|
||||||
#define ASSERT_EQ(a, b) if ((a) != (b)) { printf("Assertion failed: %s != %s\n", #a, #b); _fail++; }
|
#define ASSERT_EQ(a, b) if ((a) != (b)) { printf("Assertion failed: %s != %s\n", #a, #b); _fail++; }
|
||||||
@@ -162,6 +162,12 @@ void init_math(py::module_ &m) {
|
|||||||
.def_readwrite("direction_error", &HError3f::direction_error);
|
.def_readwrite("direction_error", &HError3f::direction_error);
|
||||||
|
|
||||||
// 3. Dynamic Vectors (uLib::Vector)
|
// 3. Dynamic Vectors (uLib::Vector)
|
||||||
|
py::class_<TRS, AffineTransform, std::shared_ptr<TRS>>(m, "TRS")
|
||||||
|
.def(py::init<>())
|
||||||
|
.def_readwrite("position", &TRS::position)
|
||||||
|
.def_readwrite("rotation", &TRS::rotation)
|
||||||
|
.def_readwrite("scaling", &TRS::scaling);
|
||||||
|
|
||||||
py::bind_vector<uLib::Vector<Scalari>>(m, "Vector_i")
|
py::bind_vector<uLib::Vector<Scalari>>(m, "Vector_i")
|
||||||
.def("MoveToVRAM", &uLib::Vector<Scalari>::MoveToVRAM)
|
.def("MoveToVRAM", &uLib::Vector<Scalari>::MoveToVRAM)
|
||||||
.def("MoveToRAM", &uLib::Vector<Scalari>::MoveToRAM);
|
.def("MoveToRAM", &uLib::Vector<Scalari>::MoveToRAM);
|
||||||
@@ -278,23 +284,46 @@ void init_math(py::module_ &m) {
|
|||||||
.def("SetRotation", &AffineTransform::SetRotation)
|
.def("SetRotation", &AffineTransform::SetRotation)
|
||||||
.def("GetRotation", &AffineTransform::GetRotation)
|
.def("GetRotation", &AffineTransform::GetRotation)
|
||||||
.def("Rotate",
|
.def("Rotate",
|
||||||
py::overload_cast<const Matrix3f>(&AffineTransform::Rotate))
|
py::overload_cast<const Matrix3f&>(&AffineTransform::Rotate))
|
||||||
.def("Rotate",
|
.def("Rotate",
|
||||||
py::overload_cast<float, Vector3f>(&AffineTransform::Rotate))
|
py::overload_cast<float, Vector3f>(&AffineTransform::Rotate))
|
||||||
.def("Rotate", py::overload_cast<Vector3f>(&AffineTransform::Rotate))
|
.def("Rotate", py::overload_cast<Vector3f>(&AffineTransform::Rotate))
|
||||||
.def("EulerYZYRotate", &AffineTransform::EulerYZYRotate)
|
.def("EulerYZYRotate", &AffineTransform::EulerYZYRotate)
|
||||||
.def("FlipAxes", &AffineTransform::FlipAxes);
|
.def("FlipAxes", &AffineTransform::FlipAxes)
|
||||||
|
.def("SetWorldMatrix", &AffineTransform::SetWorldMatrix);
|
||||||
|
|
||||||
py::class_<Geometry, AffineTransform, std::shared_ptr<Geometry>>(m, "Geometry")
|
py::class_<TRS, AffineTransform, std::shared_ptr<TRS>>(m, "TRS")
|
||||||
.def(py::init<>())
|
.def(py::init<>())
|
||||||
.def("GetWorldPoint", py::overload_cast<const Vector4f>(
|
.def(py::init<const Matrix4f &>())
|
||||||
&Geometry::GetWorldPoint, py::const_))
|
.def_readwrite("position", &TRS::position)
|
||||||
.def("GetWorldPoint",
|
.def_readwrite("rotation", &TRS::rotation)
|
||||||
py::overload_cast<float, float, float>(&Geometry::GetWorldPoint))
|
.def_readwrite("scaling", &TRS::scaling)
|
||||||
.def("GetLocalPoint", py::overload_cast<const Vector4f>(
|
.def("SetPosition", &TRS::SetPosition)
|
||||||
&Geometry::GetLocalPoint, py::const_))
|
.def("SetRotation", &TRS::SetRotation)
|
||||||
.def("GetLocalPoint",
|
.def("SetOrientation", &TRS::SetOrientation)
|
||||||
py::overload_cast<float, float, float>(&Geometry::GetLocalPoint));
|
.def("SetScale", &TRS::SetScale)
|
||||||
|
.def("FromMatrix", &TRS::FromMatrix)
|
||||||
|
.def("GetMatrix", &TRS::GetMatrix);
|
||||||
|
|
||||||
|
py::class_<Geometry, Object, std::shared_ptr<Geometry>>(m, "Geometry")
|
||||||
|
.def("GetParent", &Geometry::GetParent)
|
||||||
|
.def("SetParent", &Geometry::SetParent)
|
||||||
|
.def("GetWorldPoint", py::overload_cast<const Vector4f>(&Geometry::GetWorldPoint, py::const_))
|
||||||
|
.def("GetWorldPoint", py::overload_cast<float, float, float>(&Geometry::GetWorldPoint, py::const_))
|
||||||
|
.def("GetLocalPoint", py::overload_cast<const Vector4f>(&Geometry::GetLocalPoint, py::const_))
|
||||||
|
.def("GetLocalPoint", py::overload_cast<float, float, float>(&Geometry::GetLocalPoint, py::const_));
|
||||||
|
|
||||||
|
py::class_<LinearGeometry, Geometry, std::shared_ptr<LinearGeometry>>(m, "LinearGeometry")
|
||||||
|
.def(py::init<>())
|
||||||
|
.def("Translate", &LinearGeometry::Translate)
|
||||||
|
.def("Rotate", &LinearGeometry::Rotate)
|
||||||
|
.def("Scale", &LinearGeometry::Scale)
|
||||||
|
.def("SetPosition", &LinearGeometry::SetPosition)
|
||||||
|
.def("GetPosition", &LinearGeometry::GetPosition)
|
||||||
|
.def("EulerYZYRotate", &LinearGeometry::EulerYZYRotate)
|
||||||
|
.def("FlipAxes", &LinearGeometry::FlipAxes)
|
||||||
|
.def("GetTransform", &LinearGeometry::GetTransform)
|
||||||
|
.def("SetTransform", &LinearGeometry::SetTransform);
|
||||||
|
|
||||||
py::class_<ContainerBox, AffineTransform, Object, std::shared_ptr<ContainerBox>>(
|
py::class_<ContainerBox, AffineTransform, Object, std::shared_ptr<ContainerBox>>(
|
||||||
m, "ContainerBox")
|
m, "ContainerBox")
|
||||||
@@ -427,7 +456,7 @@ void init_math(py::module_ &m) {
|
|||||||
.def("__getitem__",
|
.def("__getitem__",
|
||||||
py::overload_cast<const Vector3i &>(&VoxImage<Voxel>::operator[]));
|
py::overload_cast<const Vector3i &>(&VoxImage<Voxel>::operator[]));
|
||||||
|
|
||||||
py::class_<TriangleMesh>(m, "TriangleMesh")
|
py::class_<TriangleMesh, TRS, std::shared_ptr<TriangleMesh>>(m, "TriangleMesh")
|
||||||
.def(py::init<>())
|
.def(py::init<>())
|
||||||
.def("AddPoint", &TriangleMesh::AddPoint)
|
.def("AddPoint", &TriangleMesh::AddPoint)
|
||||||
.def("AddTriangle",
|
.def("AddTriangle",
|
||||||
@@ -439,7 +468,7 @@ void init_math(py::module_ &m) {
|
|||||||
.def("GetTriangle", &TriangleMesh::GetTriangle)
|
.def("GetTriangle", &TriangleMesh::GetTriangle)
|
||||||
.def("GetNormal", &TriangleMesh::GetNormal);
|
.def("GetNormal", &TriangleMesh::GetNormal);
|
||||||
|
|
||||||
py::class_<QuadMesh>(m, "QuadMesh")
|
py::class_<QuadMesh, TRS, std::shared_ptr<QuadMesh>>(m, "QuadMesh")
|
||||||
.def(py::init<>())
|
.def(py::init<>())
|
||||||
.def("AddPoint", &QuadMesh::AddPoint)
|
.def("AddPoint", &QuadMesh::AddPoint)
|
||||||
.def("AddQuad",
|
.def("AddQuad",
|
||||||
|
|||||||
@@ -64,7 +64,7 @@ vtkDetectorChamber::vtkDetectorChamber(DetectorChamber *content)
|
|||||||
|
|
||||||
this->SetProp(m_PlaneActor);
|
this->SetProp(m_PlaneActor);
|
||||||
|
|
||||||
this->contentUpdate();
|
this->Update();
|
||||||
}
|
}
|
||||||
|
|
||||||
vtkDetectorChamber::~vtkDetectorChamber() {
|
vtkDetectorChamber::~vtkDetectorChamber() {
|
||||||
@@ -76,8 +76,8 @@ DetectorChamber *vtkDetectorChamber::GetContent() {
|
|||||||
return static_cast<DetectorChamber *>(m_Content);
|
return static_cast<DetectorChamber *>(m_Content);
|
||||||
}
|
}
|
||||||
|
|
||||||
void vtkDetectorChamber::contentUpdate() {
|
void vtkDetectorChamber::Update() {
|
||||||
this->BaseClass::contentUpdate();
|
this->BaseClass::Update();
|
||||||
|
|
||||||
if (!m_Content) return;
|
if (!m_Content) return;
|
||||||
DetectorChamber *c = this->GetContent();
|
DetectorChamber *c = this->GetContent();
|
||||||
|
|||||||
@@ -56,7 +56,7 @@ public:
|
|||||||
|
|
||||||
Content *GetContent();
|
Content *GetContent();
|
||||||
|
|
||||||
virtual void contentUpdate() override;
|
virtual void Update() override;
|
||||||
|
|
||||||
protected:
|
protected:
|
||||||
vtkActor *m_PlaneActor;
|
vtkActor *m_PlaneActor;
|
||||||
|
|||||||
@@ -40,20 +40,21 @@ namespace Vtk {
|
|||||||
vtkMuonScatter::vtkMuonScatter(MuonScatter &content)
|
vtkMuonScatter::vtkMuonScatter(MuonScatter &content)
|
||||||
: m_Content(&content), m_LineIn(vtkLineSource::New()),
|
: m_Content(&content), m_LineIn(vtkLineSource::New()),
|
||||||
m_LineOut(vtkLineSource::New()), m_PolyData(vtkPolyData::New()),
|
m_LineOut(vtkLineSource::New()), m_PolyData(vtkPolyData::New()),
|
||||||
m_SpherePoca(NULL) {
|
m_SpherePoca(NULL), m_Asm(vtkAssembly::New()) {
|
||||||
InstallPipe();
|
InstallPipe();
|
||||||
}
|
}
|
||||||
|
|
||||||
vtkMuonScatter::vtkMuonScatter(const MuonScatter &content)
|
vtkMuonScatter::vtkMuonScatter(const MuonScatter &content)
|
||||||
: m_Content(const_cast<MuonScatter *>(&content)),
|
: m_Content(const_cast<MuonScatter *>(&content)),
|
||||||
m_LineIn(vtkLineSource::New()), m_LineOut(vtkLineSource::New()),
|
m_LineIn(vtkLineSource::New()), m_LineOut(vtkLineSource::New()),
|
||||||
m_PolyData(vtkPolyData::New()), m_SpherePoca(NULL) {
|
m_PolyData(vtkPolyData::New()), m_SpherePoca(NULL), m_Asm(vtkAssembly::New()) {
|
||||||
InstallPipe();
|
InstallPipe();
|
||||||
}
|
}
|
||||||
|
|
||||||
vtkMuonScatter::~vtkMuonScatter() {
|
vtkMuonScatter::~vtkMuonScatter() {
|
||||||
m_LineIn->Delete();
|
m_LineIn->Delete();
|
||||||
m_LineOut->Delete();
|
m_LineOut->Delete();
|
||||||
|
m_Asm->Delete();
|
||||||
if (m_SpherePoca)
|
if (m_SpherePoca)
|
||||||
m_SpherePoca->Delete();
|
m_SpherePoca->Delete();
|
||||||
}
|
}
|
||||||
@@ -87,13 +88,15 @@ void vtkMuonScatter::InstallPipe() {
|
|||||||
mapper->SetInputConnection(m_LineIn->GetOutputPort());
|
mapper->SetInputConnection(m_LineIn->GetOutputPort());
|
||||||
vtkSmartPointer<vtkActor> actor = vtkSmartPointer<vtkActor>::New();
|
vtkSmartPointer<vtkActor> actor = vtkSmartPointer<vtkActor>::New();
|
||||||
actor->SetMapper(mapper);
|
actor->SetMapper(mapper);
|
||||||
this->SetProp(actor);
|
m_Asm->AddPart(actor);
|
||||||
|
|
||||||
mapper = vtkSmartPointer<vtkPolyDataMapper>::New();
|
mapper = vtkSmartPointer<vtkPolyDataMapper>::New();
|
||||||
mapper->SetInputConnection(m_LineOut->GetOutputPort());
|
mapper->SetInputConnection(m_LineOut->GetOutputPort());
|
||||||
actor = vtkSmartPointer<vtkActor>::New();
|
actor = vtkSmartPointer<vtkActor>::New();
|
||||||
actor->SetMapper(mapper);
|
actor->SetMapper(mapper);
|
||||||
this->SetProp(actor);
|
m_Asm->AddPart(actor);
|
||||||
|
|
||||||
|
this->SetProp(m_Asm);
|
||||||
}
|
}
|
||||||
|
|
||||||
vtkPolyData *vtkMuonScatter::GetPolyData() const {
|
vtkPolyData *vtkMuonScatter::GetPolyData() const {
|
||||||
@@ -123,7 +126,7 @@ void vtkMuonScatter::AddPocaPoint(HPoint3f poca) {
|
|||||||
mapper->SetInputConnection(m_SpherePoca->GetOutputPort());
|
mapper->SetInputConnection(m_SpherePoca->GetOutputPort());
|
||||||
vtkSmartPointer<vtkActor> actor = vtkSmartPointer<vtkActor>::New();
|
vtkSmartPointer<vtkActor> actor = vtkSmartPointer<vtkActor>::New();
|
||||||
actor->SetMapper(mapper);
|
actor->SetMapper(mapper);
|
||||||
this->SetProp(actor);
|
m_Asm->AddPart(actor);
|
||||||
}
|
}
|
||||||
|
|
||||||
HPoint3f vtkMuonScatter::GetPocaPoint() {
|
HPoint3f vtkMuonScatter::GetPocaPoint() {
|
||||||
|
|||||||
@@ -27,6 +27,7 @@
|
|||||||
#define VTKMUONSCATTER_H
|
#define VTKMUONSCATTER_H
|
||||||
|
|
||||||
#include <vtkActor.h>
|
#include <vtkActor.h>
|
||||||
|
#include <vtkAssembly.h>
|
||||||
#include <vtkAppendPolyData.h>
|
#include <vtkAppendPolyData.h>
|
||||||
#include <vtkLineSource.h>
|
#include <vtkLineSource.h>
|
||||||
#include <vtkPolyDataMapper.h>
|
#include <vtkPolyDataMapper.h>
|
||||||
@@ -85,6 +86,7 @@ private:
|
|||||||
vtkLineSource *m_LineOut;
|
vtkLineSource *m_LineOut;
|
||||||
vtkSphereSource *m_SpherePoca;
|
vtkSphereSource *m_SpherePoca;
|
||||||
vtkPolyData *m_PolyData;
|
vtkPolyData *m_PolyData;
|
||||||
|
vtkAssembly *m_Asm;
|
||||||
};
|
};
|
||||||
|
|
||||||
} // namespace Vtk
|
} // namespace Vtk
|
||||||
|
|||||||
@@ -50,6 +50,7 @@ vtkVoxRaytracerRepresentation::vtkVoxRaytracerRepresentation(Content &content)
|
|||||||
m_RayRepresentation(vtkAppendPolyData::New()),
|
m_RayRepresentation(vtkAppendPolyData::New()),
|
||||||
m_RayRepresentationActor(vtkActor::New()),
|
m_RayRepresentationActor(vtkActor::New()),
|
||||||
m_Transform(vtkTransform::New()),
|
m_Transform(vtkTransform::New()),
|
||||||
|
m_Asm(vtkAssembly::New()),
|
||||||
m_HasMuon(false), m_HasPoca(false) {
|
m_HasMuon(false), m_HasPoca(false) {
|
||||||
default_radius = content.GetImage()->GetSpacing()(0) / 4;
|
default_radius = content.GetImage()->GetSpacing()(0) / 4;
|
||||||
m_Sphere1->SetRadius(default_radius);
|
m_Sphere1->SetRadius(default_radius);
|
||||||
@@ -313,20 +314,19 @@ void vtkVoxRaytracerRepresentation::InstallPipe() {
|
|||||||
|
|
||||||
vtkSmartPointer<vtkPolyDataMapper> mapper =
|
vtkSmartPointer<vtkPolyDataMapper> mapper =
|
||||||
vtkSmartPointer<vtkPolyDataMapper>::New();
|
vtkSmartPointer<vtkPolyDataMapper>::New();
|
||||||
|
|
||||||
mapper->SetInputConnection(append->GetOutputPort());
|
mapper->SetInputConnection(append->GetOutputPort());
|
||||||
|
|
||||||
vtkSmartPointer<vtkActor> actor = vtkActor::New();
|
vtkSmartPointer<vtkActor> actor = vtkActor::New();
|
||||||
actor->SetMapper(mapper);
|
actor->SetMapper(mapper);
|
||||||
actor->GetProperty()->SetColor(0.6, 0.6, 1);
|
actor->GetProperty()->SetColor(0.6, 0.6, 1);
|
||||||
this->SetProp(actor);
|
m_Asm->AddPart(actor);
|
||||||
|
|
||||||
mapper = vtkSmartPointer<vtkPolyDataMapper>::New();
|
mapper = vtkSmartPointer<vtkPolyDataMapper>::New();
|
||||||
mapper->SetInputConnection(m_RayLine->GetOutputPort());
|
mapper->SetInputConnection(m_RayLine->GetOutputPort());
|
||||||
|
|
||||||
m_RayLineActor->SetMapper(mapper);
|
m_RayLineActor->SetMapper(mapper);
|
||||||
m_RayLineActor->GetProperty()->SetColor(1, 0, 0);
|
m_RayLineActor->GetProperty()->SetColor(1, 0, 0);
|
||||||
this->SetProp(m_RayLineActor);
|
m_Asm->AddPart(m_RayLineActor);
|
||||||
|
|
||||||
vtkSmartPointer<vtkTransformPolyDataFilter> polyfilter =
|
vtkSmartPointer<vtkTransformPolyDataFilter> polyfilter =
|
||||||
vtkSmartPointer<vtkTransformPolyDataFilter>::New();
|
vtkSmartPointer<vtkTransformPolyDataFilter>::New();
|
||||||
@@ -343,7 +343,9 @@ void vtkVoxRaytracerRepresentation::InstallPipe() {
|
|||||||
vra->GetProperty()->SetEdgeVisibility(true);
|
vra->GetProperty()->SetEdgeVisibility(true);
|
||||||
vra->GetProperty()->SetColor(0.5, 0.5, 0.5);
|
vra->GetProperty()->SetColor(0.5, 0.5, 0.5);
|
||||||
|
|
||||||
this->SetProp(vra);
|
m_Asm->AddPart(vra);
|
||||||
|
|
||||||
|
this->SetProp(m_Asm);
|
||||||
}
|
}
|
||||||
|
|
||||||
} // namespace Vtk
|
} // namespace Vtk
|
||||||
|
|||||||
@@ -107,6 +107,7 @@ private:
|
|||||||
bool m_HasPoca;
|
bool m_HasPoca;
|
||||||
|
|
||||||
Scalarf default_radius;
|
Scalarf default_radius;
|
||||||
|
vtkSmartPointer<vtkAssembly> m_Asm;
|
||||||
vtkAppendPolyData *m_RayLine;
|
vtkAppendPolyData *m_RayLine;
|
||||||
vtkActor *m_RayLineActor;
|
vtkActor *m_RayLineActor;
|
||||||
vtkActor *m_RayRepresentationActor;
|
vtkActor *m_RayRepresentationActor;
|
||||||
|
|||||||
@@ -36,6 +36,8 @@ printf("..:: Testing " #name " ::..\n");
|
|||||||
|
|
||||||
#define TEST1(val) _fail += (val)==0
|
#define TEST1(val) _fail += (val)==0
|
||||||
#define TEST0(val) _fail += (val)!=0
|
#define TEST0(val) _fail += (val)!=0
|
||||||
|
#define ASSERT_EQUAL(a,b) if((a)!=(b)) { printf("Assertion failed: " #a " != " #b " at line %d\n", __LINE__); _fail++; }
|
||||||
|
#define ASSERT_NOT_NULL(ptr) if((ptr)==NULL) { printf("Assertion failed: " #ptr " is NULL at line %d\n", __LINE__); _fail++; }
|
||||||
#define END_TESTING return _fail;
|
#define END_TESTING return _fail;
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -35,12 +35,11 @@ Assembly::Assembly(uLib::Assembly *content)
|
|||||||
m_ChildContext(nullptr),
|
m_ChildContext(nullptr),
|
||||||
m_BBoxActor(nullptr),
|
m_BBoxActor(nullptr),
|
||||||
m_VtkAsm(nullptr),
|
m_VtkAsm(nullptr),
|
||||||
m_InUpdate(false),
|
m_InUpdate(false) {
|
||||||
m_BlockUpdate(false) {
|
|
||||||
this->InstallPipe();
|
this->InstallPipe();
|
||||||
if (m_Content) {
|
if (m_Content) {
|
||||||
Object::connect(m_Content, &uLib::Assembly::Updated,
|
Object::connect(m_Content, &uLib::Assembly::Updated,
|
||||||
this, &Assembly::contentUpdate);
|
this, &Assembly::Update);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -54,6 +53,7 @@ Assembly::~Assembly() {
|
|||||||
void Assembly::InstallPipe() {
|
void Assembly::InstallPipe() {
|
||||||
// 1. Create the VTK library assembly that groups everything
|
// 1. Create the VTK library assembly that groups everything
|
||||||
m_VtkAsm = ::vtkAssembly::New();
|
m_VtkAsm = ::vtkAssembly::New();
|
||||||
|
m_VtkAsm->PickableOff();
|
||||||
this->SetProp(m_VtkAsm);
|
this->SetProp(m_VtkAsm);
|
||||||
|
|
||||||
// 2. Create the bounding-box wireframe actor
|
// 2. Create the bounding-box wireframe actor
|
||||||
@@ -78,36 +78,33 @@ void Assembly::InstallPipe() {
|
|||||||
// 3. Build a child-objects context (auto-creates puppets for each child)
|
// 3. Build a child-objects context (auto-creates puppets for each child)
|
||||||
if (m_Content) {
|
if (m_Content) {
|
||||||
m_ChildContext = new vtkObjectsContext(m_Content);
|
m_ChildContext = new vtkObjectsContext(m_Content);
|
||||||
// The vtkObjectsContext's own prop is already a ::vtkAssembly;
|
// Link the children context's assembly into our group assembly
|
||||||
// nest it inside ours so everything moves together.
|
if (auto* childProp = vtkProp3D::SafeDownCast(m_ChildContext->GetProp())) {
|
||||||
if (auto *childProp = vtkProp3D::SafeDownCast(m_ChildContext->GetProp()))
|
|
||||||
m_VtkAsm->AddPart(childProp);
|
m_VtkAsm->AddPart(childProp);
|
||||||
}
|
}
|
||||||
|
|
||||||
// 4. Apply initial transform
|
|
||||||
this->UpdateTransform();
|
|
||||||
this->UpdateBoundingBox();
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// ------------------------------------------------------------------ //
|
// 4. Force initial visual sync
|
||||||
void Assembly::contentUpdate() {
|
this->Update();
|
||||||
if (m_InUpdate) return;
|
|
||||||
m_InUpdate = true;
|
|
||||||
m_BlockUpdate = false;
|
|
||||||
this->UpdateTransform();
|
|
||||||
this->UpdateBoundingBox();
|
|
||||||
if (m_ChildContext)
|
|
||||||
m_ChildContext->Update();
|
|
||||||
|
|
||||||
Puppet::Update();
|
|
||||||
m_InUpdate = false;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// ------------------------------------------------------------------ //
|
// ------------------------------------------------------------------ //
|
||||||
void Assembly::Update() {
|
void Assembly::Update() {
|
||||||
if (m_InUpdate) return;
|
if (m_InUpdate) return;
|
||||||
m_InUpdate = true;
|
m_InUpdate = true;
|
||||||
this->contentUpdate();
|
|
||||||
|
if (m_Content && m_VtkAsm) {
|
||||||
|
// Apply world matrix from the assembly content
|
||||||
|
vtkNew<vtkMatrix4x4> m;
|
||||||
|
Matrix4fToVtk(m_Content->GetMatrix(), m);
|
||||||
|
m_VtkAsm->SetUserMatrix(m);
|
||||||
|
m_VtkAsm->Modified();
|
||||||
|
}
|
||||||
|
|
||||||
|
this->Puppet::Update();
|
||||||
|
this->UpdateBoundingBox();
|
||||||
|
if (m_ChildContext)
|
||||||
|
m_ChildContext->Update();
|
||||||
m_InUpdate = false;
|
m_InUpdate = false;
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -117,14 +114,11 @@ void Assembly::SyncFromVtk() {
|
|||||||
|
|
||||||
m_InUpdate = true;
|
m_InUpdate = true;
|
||||||
|
|
||||||
double pos[3], ori[3], scale[3];
|
// VTK -> Model: Update world matrix (accounting for model parents)
|
||||||
m_VtkAsm->GetPosition(pos);
|
if (vtkProp3D* proxy = this->GetProxyProp()) {
|
||||||
m_VtkAsm->GetOrientation(ori);
|
m_Content->SetWorldMatrix(VtkToMatrix4f(proxy->GetUserMatrix()));
|
||||||
m_VtkAsm->GetScale(scale);
|
m_Content->FromMatrix(m_Content->GetMatrix());
|
||||||
|
}
|
||||||
m_Content->SetPosition(Vector3f(pos[0], pos[1], pos[2]));
|
|
||||||
m_Content->SetOrientation(Vector3f(ori[0], ori[1], ori[2]) * CLHEP::degree);
|
|
||||||
m_Content->SetScale(Vector3f(scale[0], scale[1], scale[2]));
|
|
||||||
|
|
||||||
this->UpdateBoundingBox();
|
this->UpdateBoundingBox();
|
||||||
if (m_ChildContext)
|
if (m_ChildContext)
|
||||||
@@ -135,14 +129,6 @@ void Assembly::SyncFromVtk() {
|
|||||||
m_InUpdate = false;
|
m_InUpdate = false;
|
||||||
}
|
}
|
||||||
|
|
||||||
// ------------------------------------------------------------------ //
|
|
||||||
void Assembly::UpdateTransform() {
|
|
||||||
if (!m_Content || !m_VtkAsm) return;
|
|
||||||
|
|
||||||
this->ApplyTransform(m_VtkAsm);
|
|
||||||
m_VtkAsm->Modified();
|
|
||||||
}
|
|
||||||
|
|
||||||
// ------------------------------------------------------------------ //
|
// ------------------------------------------------------------------ //
|
||||||
void Assembly::UpdateBoundingBox() {
|
void Assembly::UpdateBoundingBox() {
|
||||||
if (!m_Content || !m_BBoxActor) return;
|
if (!m_Content || !m_BBoxActor) return;
|
||||||
|
|||||||
@@ -36,7 +36,7 @@ class vtkObjectsContext; // forward
|
|||||||
*/
|
*/
|
||||||
class Assembly : public Puppet {
|
class Assembly : public Puppet {
|
||||||
public:
|
public:
|
||||||
virtual const char *GetClassName() const override { return "Vtk.Assembly"; }
|
uLibTypeMacro(Assembly, Puppet)
|
||||||
|
|
||||||
Assembly(uLib::Assembly *content);
|
Assembly(uLib::Assembly *content);
|
||||||
virtual ~Assembly();
|
virtual ~Assembly();
|
||||||
@@ -50,14 +50,14 @@ public:
|
|||||||
virtual uLib::Object* GetContent() const override { return (uLib::Object*)m_Content; }
|
virtual uLib::Object* GetContent() const override { return (uLib::Object*)m_Content; }
|
||||||
virtual uLib::ObjectsContext* GetChildren() override { return (uLib::ObjectsContext*)m_Content; }
|
virtual uLib::ObjectsContext* GetChildren() override { return (uLib::ObjectsContext*)m_Content; }
|
||||||
|
|
||||||
/** @brief Called when the model signals an update (model→VTK push). */
|
/**
|
||||||
void contentUpdate();
|
* @brief Returns the puppet managing child objects.
|
||||||
|
*/
|
||||||
|
|
||||||
/** @brief Returns the puppet managing child objects. */
|
/** @brief Returns the puppet managing child objects. */
|
||||||
vtkObjectsContext *GetChildrenContext() const;
|
vtkObjectsContext *GetChildrenContext() const;
|
||||||
|
|
||||||
private:
|
private:
|
||||||
void UpdateTransform();
|
|
||||||
void UpdateBoundingBox();
|
void UpdateBoundingBox();
|
||||||
void InstallPipe();
|
void InstallPipe();
|
||||||
|
|
||||||
@@ -66,7 +66,6 @@ private:
|
|||||||
vtkActor *m_BBoxActor;
|
vtkActor *m_BBoxActor;
|
||||||
::vtkAssembly *m_VtkAsm; // VTK library assembly — NOT this class
|
::vtkAssembly *m_VtkAsm; // VTK library assembly — NOT this class
|
||||||
bool m_InUpdate; // re-entrancy guard
|
bool m_InUpdate; // re-entrancy guard
|
||||||
bool m_BlockUpdate; // block feedback from contentUpdate→Update
|
|
||||||
};
|
};
|
||||||
|
|
||||||
} // namespace Vtk
|
} // namespace Vtk
|
||||||
|
|||||||
@@ -50,16 +50,26 @@ namespace Vtk {
|
|||||||
struct ContainerBoxData {
|
struct ContainerBoxData {
|
||||||
vtkSmartPointer<vtkActor> m_Cube;
|
vtkSmartPointer<vtkActor> m_Cube;
|
||||||
vtkSmartPointer<vtkActor> m_Axes;
|
vtkSmartPointer<vtkActor> m_Axes;
|
||||||
|
vtkSmartPointer<vtkAssembly> m_VtkAsm;
|
||||||
|
vtkSmartPointer<vtkMatrix4x4> m_Affine;
|
||||||
|
uLib::Connection m_UpdateSignal;
|
||||||
|
|
||||||
ContainerBoxData() : m_Cube(vtkSmartPointer<vtkActor>::New()), m_Axes(vtkSmartPointer<vtkActor>::New()) {}
|
ContainerBoxData() : m_Cube(vtkSmartPointer<vtkActor>::New()),
|
||||||
|
m_Axes(vtkSmartPointer<vtkActor>::New()),
|
||||||
|
m_VtkAsm(vtkSmartPointer<vtkAssembly>::New()),
|
||||||
|
m_Affine(vtkSmartPointer<vtkMatrix4x4>::New()) {}
|
||||||
~ContainerBoxData() {
|
~ContainerBoxData() {
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
vtkContainerBox::vtkContainerBox(vtkContainerBox::Content *content)
|
vtkContainerBox::vtkContainerBox(vtkContainerBox::Content *content)
|
||||||
: d(new ContainerBoxData()), m_Content(content) {
|
: d(new ContainerBoxData()), m_Content(content) {
|
||||||
this->InstallPipe();
|
this->InstallPipe();
|
||||||
Object::connect(m_Content, &Content::Updated, this, &vtkContainerBox::contentUpdate);
|
d->m_UpdateSignal = Object::connect(m_Content, &uLib::Object::Updated, this, &vtkContainerBox::Update);
|
||||||
}
|
}
|
||||||
|
|
||||||
vtkContainerBox::~vtkContainerBox() {
|
vtkContainerBox::~vtkContainerBox() {
|
||||||
@@ -72,47 +82,50 @@ vtkPolyData *vtkContainerBox::GetPolyData() const {
|
|||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
void vtkContainerBox::contentUpdate() {
|
void vtkContainerBox::Update() {
|
||||||
RecursiveMutex::ScopedLock lock(this->m_UpdateMutex);
|
RecursiveMutex::ScopedLock lock(this->m_UpdateMutex);
|
||||||
if (!m_Content)
|
if (!m_Content) return;
|
||||||
return;
|
|
||||||
|
|
||||||
vtkProp3D* root = vtkProp3D::SafeDownCast(this->GetProp());
|
vtkProp3D* root = vtkProp3D::SafeDownCast(this->GetProp());
|
||||||
if (!root) return;
|
if (root) {
|
||||||
|
// Apply local full matrix (TRS * LocalBox) so that nested assemblies work correctly
|
||||||
d->m_Cube->SetUserMatrix(nullptr);
|
Matrix4f fullLocal = m_Content->GetMatrix() * m_Content->GetLocalMatrix();
|
||||||
d->m_Axes->SetUserMatrix(nullptr);
|
vtkNew<vtkMatrix4x4> m;
|
||||||
|
Matrix4fToVtk(fullLocal, m);
|
||||||
TRS trs(*m_Content);
|
root->SetUserMatrix(m);
|
||||||
this->ApplyTransform(root);
|
|
||||||
|
|
||||||
root->Modified();
|
root->Modified();
|
||||||
m_BlockUpdate = false;
|
|
||||||
Puppet::Update();
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Delegate rest of update (appearance, render, etc)
|
||||||
void vtkContainerBox::Update() {
|
ConnectionBlock blocker(d->m_UpdateSignal);
|
||||||
this->contentUpdate();
|
this->Puppet::Update();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
void vtkContainerBox::SyncFromVtk() {
|
void vtkContainerBox::SyncFromVtk() {
|
||||||
RecursiveMutex::ScopedLock lock(this->m_UpdateMutex);
|
RecursiveMutex::ScopedLock lock(this->m_UpdateMutex);
|
||||||
if (!m_Content) return;
|
if (!m_Content) return;
|
||||||
|
|
||||||
vtkProp3D* assembly = vtkProp3D::SafeDownCast(this->GetProp());
|
vtkProp3D* root = this->GetProxyProp();
|
||||||
if (!assembly) return;
|
if (!root) return;
|
||||||
|
|
||||||
double pos[3], ori[3], scale[3];
|
// VTK -> Model: Extract new world TRS from proxy, which matches the model's TRS center
|
||||||
assembly->GetPosition(pos);
|
vtkMatrix4x4* rootMat = root->GetUserMatrix();
|
||||||
assembly->GetOrientation(ori);
|
// if (rootMat) {
|
||||||
assembly->GetScale(scale);
|
// std::cout << "[vtkContainerBox::SyncFromVtk] Read Proxy UserMatrix:" << std::endl;
|
||||||
|
// rootMat->Print(std::cout);
|
||||||
|
// }
|
||||||
|
|
||||||
m_Content->SetPosition(Vector3f(pos[0], pos[1], pos[2]));
|
Matrix4f vtkWorld = VtkToMatrix4f(rootMat);
|
||||||
m_Content->SetOrientation(Vector3f(ori[0], ori[1], ori[2]) * CLHEP::degree);
|
|
||||||
m_Content->SetScale(Vector3f(scale[0], scale[1], scale[2]));
|
|
||||||
|
|
||||||
m_Content->Updated(); // Notify change
|
// Synchronize TRS property members from the updated local matrix
|
||||||
|
m_Content->FromMatrix(vtkWorld);
|
||||||
|
|
||||||
|
// std::cout << "[vtkContainerBox::SyncFromVtk] New Model WorldMatrix:" << std::endl << m_Content->GetWorldMatrix() << std::endl;
|
||||||
|
|
||||||
|
// Since we modified the model, notify observers, but block the loop back to VTK
|
||||||
|
// ConnectionBlock blocker(d->m_UpdateSignal);
|
||||||
|
m_Content->Updated();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
@@ -154,17 +167,16 @@ void vtkContainerBox::InstallPipe() {
|
|||||||
mapper->SetInputConnection(axes->GetOutputPort());
|
mapper->SetInputConnection(axes->GetOutputPort());
|
||||||
mapper->Update();
|
mapper->Update();
|
||||||
|
|
||||||
this->SetProp(d->m_Cube);
|
d->m_VtkAsm->AddPart(d->m_Cube);
|
||||||
this->SetProp(d->m_Axes);
|
d->m_VtkAsm->AddPart(d->m_Axes);
|
||||||
|
this->SetProp(d->m_VtkAsm);
|
||||||
|
|
||||||
vtkProp3D* root = vtkProp3D::SafeDownCast(this->GetProp());
|
vtkProp3D* root = d->m_VtkAsm;
|
||||||
if (root) {
|
if (root) {
|
||||||
TRS trs(*c);
|
d->m_Affine = Matrix4fToVtk(m_Content->GetMatrix());
|
||||||
root->SetPosition(trs.position.x(), trs.position.y(), trs.position.z());
|
root->SetUserMatrix(d->m_Affine);
|
||||||
root->SetOrientation(trs.rotation.x(), trs.rotation.y(), trs.rotation.z());
|
|
||||||
root->SetScale(trs.scaling.x(), trs.scaling.y(), trs.scaling.z());
|
|
||||||
root->SetUserMatrix(nullptr);
|
|
||||||
}
|
}
|
||||||
|
this->Update();
|
||||||
}
|
}
|
||||||
|
|
||||||
} // namespace Vtk
|
} // namespace Vtk
|
||||||
|
|||||||
@@ -29,7 +29,6 @@
|
|||||||
#include "Math/ContainerBox.h"
|
#include "Math/ContainerBox.h"
|
||||||
#include "uLibVtkInterface.h"
|
#include "uLibVtkInterface.h"
|
||||||
#include "vtkPolydata.h"
|
#include "vtkPolydata.h"
|
||||||
#include <boost/signals2/connection.hpp>
|
|
||||||
|
|
||||||
class vtkActor;
|
class vtkActor;
|
||||||
|
|
||||||
@@ -47,9 +46,14 @@ public:
|
|||||||
|
|
||||||
virtual class vtkPolyData *GetPolyData() const;
|
virtual class vtkPolyData *GetPolyData() const;
|
||||||
|
|
||||||
virtual void contentUpdate();
|
/**
|
||||||
|
* @brief Updates the VTK representation from the internal state.
|
||||||
|
*/
|
||||||
virtual void Update() override;
|
virtual void Update() override;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Synchronizes the model from the VTK representation (VTK→model).
|
||||||
|
*/
|
||||||
virtual void SyncFromVtk() override;
|
virtual void SyncFromVtk() override;
|
||||||
|
|
||||||
virtual uLib::Object* GetContent() const override { return (uLib::Object*)m_Content; }
|
virtual uLib::Object* GetContent() const override { return (uLib::Object*)m_Content; }
|
||||||
@@ -60,6 +64,7 @@ protected:
|
|||||||
struct ContainerBoxData *d;
|
struct ContainerBoxData *d;
|
||||||
Content *m_Content;
|
Content *m_Content;
|
||||||
bool m_BlockUpdate = false;
|
bool m_BlockUpdate = false;
|
||||||
|
|
||||||
};
|
};
|
||||||
|
|
||||||
} // namespace Vtk
|
} // namespace Vtk
|
||||||
|
|||||||
@@ -40,7 +40,7 @@ namespace Vtk {
|
|||||||
vtkCylinder::vtkCylinder(vtkCylinder::Content *content)
|
vtkCylinder::vtkCylinder(vtkCylinder::Content *content)
|
||||||
: m_Content(content), m_Actor(nullptr), m_VtkAsm(nullptr) {
|
: m_Content(content), m_Actor(nullptr), m_VtkAsm(nullptr) {
|
||||||
this->InstallPipe();
|
this->InstallPipe();
|
||||||
Object::connect(m_Content, &Content::Updated, this, &vtkCylinder::contentUpdate);
|
m_UpdateSignal = Object::connect(m_Content, &uLib::Object::Updated, this, &vtkCylinder::Update);
|
||||||
}
|
}
|
||||||
|
|
||||||
vtkCylinder::~vtkCylinder() {
|
vtkCylinder::~vtkCylinder() {
|
||||||
@@ -48,16 +48,16 @@ vtkCylinder::~vtkCylinder() {
|
|||||||
if (m_VtkAsm) m_VtkAsm->Delete();
|
if (m_VtkAsm) m_VtkAsm->Delete();
|
||||||
}
|
}
|
||||||
|
|
||||||
void vtkCylinder::contentUpdate() {
|
void vtkCylinder::Update() {
|
||||||
if (!m_Content)
|
if (!m_Content)
|
||||||
return;
|
return;
|
||||||
|
|
||||||
vtkProp3D* root = vtkProp3D::SafeDownCast(this->GetProp());
|
vtkProp3D* root = vtkProp3D::SafeDownCast(this->GetProp());
|
||||||
if (!root) return;
|
if (root) {
|
||||||
|
// 1. Placement handled specifically from content (use TRS GetMatrix, not World)
|
||||||
// 1. Placement handled by base Puppet class via Sync / Update logic
|
vtkNew<vtkMatrix4x4> m;
|
||||||
// Update internal pd->m_Transform from content
|
Matrix4fToVtk(m_Content->GetMatrix(), m);
|
||||||
Puppet::Update();
|
root->SetUserMatrix(m);
|
||||||
|
|
||||||
// 2. Shape-local properties (Radius, Height, Axis alignment) go to the internal actor
|
// 2. Shape-local properties (Radius, Height, Axis alignment) go to the internal actor
|
||||||
// These are relative to the root assembly
|
// These are relative to the root assembly
|
||||||
@@ -80,27 +80,32 @@ void vtkCylinder::contentUpdate() {
|
|||||||
root->Modified();
|
root->Modified();
|
||||||
}
|
}
|
||||||
|
|
||||||
void vtkCylinder::Update() {
|
// Delegate rest of update (appearance, render, etc)
|
||||||
this->contentUpdate();
|
ConnectionBlock blocker(m_UpdateSignal);
|
||||||
|
this->Puppet::Update();
|
||||||
}
|
}
|
||||||
|
|
||||||
void vtkCylinder::SyncFromVtk() {
|
void vtkCylinder::SyncFromVtk() {
|
||||||
if (!m_Content) return;
|
if (!m_Content) return;
|
||||||
|
|
||||||
vtkProp3D* assembly = vtkProp3D::SafeDownCast(this->GetProp());
|
vtkProp3D* root = this->GetProxyProp();
|
||||||
if (!assembly) return;
|
if (!root) return;
|
||||||
|
|
||||||
double pos[3], ori[3], scale[3];
|
// VTK -> Model: Extract new world TRS from proxy
|
||||||
assembly->GetPosition(pos);
|
vtkMatrix4x4* rootMat = root->GetUserMatrix();
|
||||||
assembly->GetOrientation(ori);
|
if (rootMat) {
|
||||||
assembly->GetScale(scale);
|
std::cout << "[vtkCylinder::SyncFromVtk] Read Proxy UserMatrix:" << std::endl;
|
||||||
|
rootMat->Print(std::cout);
|
||||||
|
}
|
||||||
|
|
||||||
m_Content->SetPosition(Vector3f(pos[0], pos[1], pos[2]));
|
Matrix4f vtkWorld = VtkToMatrix4f(rootMat);
|
||||||
// Convert VTK degrees to model radians
|
|
||||||
m_Content->SetOrientation(Vector3f(ori[0], ori[1], ori[2]) * CLHEP::degree);
|
|
||||||
m_Content->SetScale(Vector3f(scale[0], scale[1], scale[2]));
|
|
||||||
|
|
||||||
m_Content->Updated(); // Notify change
|
// Directly sync model from the world matrix
|
||||||
|
m_Content->FromMatrix(vtkWorld);
|
||||||
|
|
||||||
|
std::cout << "[vtkCylinder::SyncFromVtk] New Model WorldMatrix:" << std::endl << m_Content->GetWorldMatrix() << std::endl;
|
||||||
|
|
||||||
|
m_Content->Updated();
|
||||||
}
|
}
|
||||||
|
|
||||||
void vtkCylinder::InstallPipe() {
|
void vtkCylinder::InstallPipe() {
|
||||||
@@ -127,7 +132,7 @@ void vtkCylinder::InstallPipe() {
|
|||||||
|
|
||||||
m_VtkAsm->AddPart(m_Actor);
|
m_VtkAsm->AddPart(m_Actor);
|
||||||
|
|
||||||
this->contentUpdate();
|
this->Update();
|
||||||
}
|
}
|
||||||
|
|
||||||
} // namespace Vtk
|
} // namespace Vtk
|
||||||
|
|||||||
@@ -48,10 +48,7 @@ public:
|
|||||||
vtkCylinder(Content *content);
|
vtkCylinder(Content *content);
|
||||||
virtual ~vtkCylinder();
|
virtual ~vtkCylinder();
|
||||||
|
|
||||||
/** Synchronizes the VTK actor with the uLib model matrix */
|
/** Synchronizes the VTK actor with the uLib model matrix and vice-versa */
|
||||||
virtual void contentUpdate();
|
|
||||||
|
|
||||||
/** Synchronizes the uLib model matrix with the VTK actor (e.g., after UI manipulation) */
|
|
||||||
virtual void Update() override;
|
virtual void Update() override;
|
||||||
|
|
||||||
/** Synchronizes the uLib model matrix with the VTK actor specifically for gizmo interactions */
|
/** Synchronizes the uLib model matrix with the VTK actor specifically for gizmo interactions */
|
||||||
@@ -66,6 +63,7 @@ protected:
|
|||||||
vtkActor *m_Actor;
|
vtkActor *m_Actor;
|
||||||
::vtkAssembly *m_VtkAsm;
|
::vtkAssembly *m_VtkAsm;
|
||||||
Content *m_Content;
|
Content *m_Content;
|
||||||
|
uLib::Connection m_UpdateSignal;
|
||||||
};
|
};
|
||||||
|
|
||||||
} // namespace Vtk
|
} // namespace Vtk
|
||||||
|
|||||||
@@ -41,6 +41,7 @@
|
|||||||
#include <vtkPiecewiseFunction.h>
|
#include <vtkPiecewiseFunction.h>
|
||||||
#include <vtkSmartVolumeMapper.h>
|
#include <vtkSmartVolumeMapper.h>
|
||||||
#include <vtkVolumeProperty.h>
|
#include <vtkVolumeProperty.h>
|
||||||
|
#include <vtkMatrix4x4.h>
|
||||||
|
|
||||||
#include <vtkActor.h>
|
#include <vtkActor.h>
|
||||||
#include <vtkPolyDataMapper.h>
|
#include <vtkPolyDataMapper.h>
|
||||||
@@ -49,6 +50,7 @@
|
|||||||
#include <Math/VoxImage.h>
|
#include <Math/VoxImage.h>
|
||||||
|
|
||||||
#include "Vtk/Math/vtkVoxImage.h"
|
#include "Vtk/Math/vtkVoxImage.h"
|
||||||
|
#include "Vtk/Math/vtkDense.h"
|
||||||
|
|
||||||
#include <vtkAutoInit.h>
|
#include <vtkAutoInit.h>
|
||||||
VTK_MODULE_INIT(vtkRenderingVolumeOpenGL2);
|
VTK_MODULE_INIT(vtkRenderingVolumeOpenGL2);
|
||||||
@@ -124,20 +126,30 @@ void vtkVoxImage::SetContent() {
|
|||||||
|
|
||||||
vtkVoxImage::vtkVoxImage(Content &content)
|
vtkVoxImage::vtkVoxImage(Content &content)
|
||||||
: m_Content(content), m_Actor(vtkVolume::New()),
|
: m_Content(content), m_Actor(vtkVolume::New()),
|
||||||
|
m_Asm(vtkAssembly::New()),
|
||||||
m_Image(vtkImageData::New()), m_Outline(vtkCubeSource::New()),
|
m_Image(vtkImageData::New()), m_Outline(vtkCubeSource::New()),
|
||||||
m_OutlineActor(vtkActor::New()),
|
m_OutlineActor(vtkActor::New()),
|
||||||
m_Reader(NULL), m_Writer(NULL), writer_factor(1.E6),
|
m_Reader(NULL), m_Writer(NULL), writer_factor(1.E6),
|
||||||
m_Window(40/1.E6), m_Level(20/1.E6), m_ShadingPreset(0) {
|
m_Window(1.0), m_Level(0.5), m_ShadingPreset(0) {
|
||||||
|
// Transfer functions
|
||||||
|
m_ColorFun = vtkColorTransferFunction::New();
|
||||||
|
m_OpacityFun = vtkPiecewiseFunction::New();
|
||||||
|
m_UpdateConnection = Object::connect(&m_Content, &uLib::Object::Updated, this, &vtkVoxImage::Update);
|
||||||
|
|
||||||
GetContent();
|
GetContent();
|
||||||
InstallPipe();
|
InstallPipe();
|
||||||
|
RescaleShaderRange();
|
||||||
ULIB_ACTIVATE_DISPLAY_PROPERTIES;
|
ULIB_ACTIVATE_DISPLAY_PROPERTIES;
|
||||||
}
|
}
|
||||||
|
|
||||||
vtkVoxImage::~vtkVoxImage() {
|
vtkVoxImage::~vtkVoxImage() {
|
||||||
m_Image->Delete();
|
m_Image->Delete();
|
||||||
m_Actor->Delete();
|
m_Actor->Delete();
|
||||||
|
m_Asm->Delete();
|
||||||
m_Outline->Delete();
|
m_Outline->Delete();
|
||||||
m_OutlineActor->Delete();
|
m_OutlineActor->Delete();
|
||||||
|
m_ColorFun->Delete();
|
||||||
|
m_OpacityFun->Delete();
|
||||||
}
|
}
|
||||||
|
|
||||||
vtkImageData *vtkVoxImage::GetImageData() {
|
vtkImageData *vtkVoxImage::GetImageData() {
|
||||||
@@ -179,6 +191,7 @@ void vtkVoxImage::ReadFromVKTFile(const char *fname) {
|
|||||||
|
|
||||||
m_Image->DeepCopy(vtkscale->GetOutput());
|
m_Image->DeepCopy(vtkscale->GetOutput());
|
||||||
SetContent();
|
SetContent();
|
||||||
|
RescaleShaderRange();
|
||||||
} else {
|
} else {
|
||||||
std::cerr << "Error: file does not contain structured points\n";
|
std::cerr << "Error: file does not contain structured points\n";
|
||||||
}
|
}
|
||||||
@@ -198,115 +211,134 @@ void vtkVoxImage::ReadFromXMLFile(const char *fname) {
|
|||||||
|
|
||||||
m_Image->DeepCopy(vtkscale->GetOutput());
|
m_Image->DeepCopy(vtkscale->GetOutput());
|
||||||
SetContent();
|
SetContent();
|
||||||
|
RescaleShaderRange();
|
||||||
}
|
}
|
||||||
|
|
||||||
void vtkVoxImage::setShadingPreset(int blendType) {
|
void vtkVoxImage::setShadingPreset(int blendType) {
|
||||||
m_ShadingPreset = blendType;
|
m_ShadingPreset = blendType;
|
||||||
vtkSmartVolumeMapper *mapper = (vtkSmartVolumeMapper *)m_Actor->GetMapper();
|
vtkSmartVolumeMapper *mapper = (vtkSmartVolumeMapper *)m_Actor->GetMapper();
|
||||||
|
if (!mapper) return;
|
||||||
vtkVolumeProperty *property = m_Actor->GetProperty();
|
vtkVolumeProperty *property = m_Actor->GetProperty();
|
||||||
|
|
||||||
static vtkColorTransferFunction *colorFun = vtkColorTransferFunction::New();
|
|
||||||
static vtkPiecewiseFunction *opacityFun = vtkPiecewiseFunction::New();
|
|
||||||
|
|
||||||
float window = m_Window;
|
float window = m_Window;
|
||||||
float level = m_Level;
|
float level = m_Level;
|
||||||
|
|
||||||
property->SetColor(colorFun);
|
property->SetColor(m_ColorFun);
|
||||||
property->SetScalarOpacity(opacityFun);
|
property->SetScalarOpacity(m_OpacityFun);
|
||||||
property->SetInterpolationTypeToLinear();
|
property->SetInterpolationTypeToLinear();
|
||||||
|
|
||||||
if (blendType != 6) {
|
m_ColorFun->RemoveAllPoints();
|
||||||
colorFun->RemoveAllPoints();
|
m_OpacityFun->RemoveAllPoints();
|
||||||
opacityFun->RemoveAllPoints();
|
|
||||||
}
|
|
||||||
|
|
||||||
switch (blendType) {
|
switch (blendType) {
|
||||||
case 0:
|
case 0: // MIP
|
||||||
colorFun->AddRGBSegment(0.0, 1.0, 1.0, 1.0, 255.0, 1.0, 1.0, 1.0);
|
m_ColorFun->AddRGBPoint(level - 0.5 * window, 0, 0, 0);
|
||||||
opacityFun->AddSegment(level - 0.5 * window, 0.0, level + 0.5 * window,
|
m_ColorFun->AddRGBPoint(level + 0.5 * window, 1, 1, 1);
|
||||||
1.0);
|
m_OpacityFun->AddSegment(level - 0.5 * window, 0.0, level + 0.5 * window, 1.0);
|
||||||
mapper->SetBlendModeToMaximumIntensity();
|
mapper->SetBlendModeToMaximumIntensity();
|
||||||
break;
|
break;
|
||||||
case 1:
|
case 1: // Composite
|
||||||
colorFun->AddRGBSegment(level - 0.5 * window, 0.0, 0.0, 0.0,
|
m_ColorFun->AddRGBPoint(level - 0.5 * window, 0, 0, 0);
|
||||||
level + 0.5 * window, 1.0, 1.0, 1.0);
|
m_ColorFun->AddRGBPoint(level + 0.5 * window, 1, 1, 1);
|
||||||
opacityFun->AddSegment(level - 0.5 * window, 0.0, level + 0.5 * window,
|
m_OpacityFun->AddSegment(level - 0.5 * window, 0.0, level + 0.5 * window, 1.0);
|
||||||
1.0);
|
|
||||||
mapper->SetBlendModeToComposite();
|
mapper->SetBlendModeToComposite();
|
||||||
property->ShadeOff();
|
property->ShadeOff();
|
||||||
break;
|
break;
|
||||||
case 2:
|
case 2: // Composite Shaded
|
||||||
colorFun->AddRGBSegment(0.0, 1.0, 1.0, 1.0, 255.0, 1.0, 1.0, 1.0);
|
m_ColorFun->AddRGBPoint(level - 0.5 * window, 0, 0, 0);
|
||||||
opacityFun->AddSegment(level - 0.5 * window, 0.0, level + 0.5 * window,
|
m_ColorFun->AddRGBPoint(level + 0.5 * window, 1, 1, 1);
|
||||||
1.0);
|
m_OpacityFun->AddSegment(level - 0.5 * window, 0.0, level + 0.5 * window, 1.0);
|
||||||
mapper->SetBlendModeToComposite();
|
mapper->SetBlendModeToComposite();
|
||||||
property->ShadeOn();
|
property->ShadeOn();
|
||||||
break;
|
break;
|
||||||
case 3:
|
case 3: // Rainbow MIP
|
||||||
colorFun->AddRGBPoint(-3024, 0, 0, 0, 0.5, 0.0);
|
m_ColorFun->AddRGBPoint(level - 0.5 * window, 0, 0, 1);
|
||||||
colorFun->AddRGBPoint(-1000, .62, .36, .18, 0.5, 0.0);
|
m_ColorFun->AddRGBPoint(level, 0, 1, 0);
|
||||||
colorFun->AddRGBPoint(-500, .88, .60, .29, 0.33, 0.45);
|
m_ColorFun->AddRGBPoint(level + 0.5 * window, 1, 1, 0);
|
||||||
colorFun->AddRGBPoint(3071, .83, .66, 1, 0.5, 0.0);
|
m_ColorFun->AddRGBPoint(level + window, 1, 0, 0);
|
||||||
opacityFun->AddPoint(-3024, 0, 0.5, 0.0);
|
m_OpacityFun->AddSegment(level - 0.5 * window, 0.0, level + 0.5 * window, 1.0);
|
||||||
opacityFun->AddPoint(-1000, 0, 0.5, 0.0);
|
|
||||||
opacityFun->AddPoint(-500, 1.0, 0.33, 0.45);
|
|
||||||
opacityFun->AddPoint(3071, 1.0, 0.5, 0.0);
|
|
||||||
mapper->SetBlendModeToComposite();
|
|
||||||
property->ShadeOn();
|
|
||||||
property->SetAmbient(0.1);
|
|
||||||
property->SetDiffuse(0.9);
|
|
||||||
property->SetSpecular(0.2);
|
|
||||||
property->SetSpecularPower(10.0);
|
|
||||||
property->SetScalarOpacityUnitDistance(0.8919);
|
|
||||||
break;
|
|
||||||
case 4:
|
|
||||||
colorFun->AddRGBPoint(0.0, 0, 0, 1); // Blue
|
|
||||||
colorFun->AddRGBPoint(level, 0, 1, 0); // Green
|
|
||||||
colorFun->AddRGBPoint(level + 0.5*window, 1, 1, 0); // Yellow
|
|
||||||
colorFun->AddRGBPoint(level + window, 1, 0, 0); // Red
|
|
||||||
opacityFun->AddSegment(level - 0.5 * window, 0.0, level + 0.5 * window, 1.0);
|
|
||||||
mapper->SetBlendModeToMaximumIntensity();
|
mapper->SetBlendModeToMaximumIntensity();
|
||||||
break;
|
break;
|
||||||
case 5:
|
case 4: // Additive
|
||||||
colorFun->AddRGBSegment(0.0, 1.0, 1.0, 1.0, 255.0, 1.0, 1.0, 1.0);
|
m_ColorFun->AddRGBPoint(level - 0.5 * window, 0, 0, 0);
|
||||||
opacityFun->AddSegment(level - 0.5 * window, 0.0, level + 0.5 * window, 1.0);
|
m_ColorFun->AddRGBPoint(level + 0.5 * window, 1, 1, 1);
|
||||||
|
m_OpacityFun->AddSegment(level - 0.5 * window, 0.0, level + 0.5 * window, 1.0);
|
||||||
mapper->SetBlendModeToAdditive();
|
mapper->SetBlendModeToAdditive();
|
||||||
break;
|
break;
|
||||||
default:
|
default:
|
||||||
vtkGenericWarningMacro("Unknown blend type.");
|
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
void vtkVoxImage::RescaleShaderRange() {
|
||||||
|
double range[2];
|
||||||
|
m_Image->GetScalarRange(range);
|
||||||
|
m_Level = (range[0] + range[1]) / 2.0;
|
||||||
|
m_Window = range[1] - range[0];
|
||||||
|
if (m_Window <= 1e-9)
|
||||||
|
m_Window = 1.0;
|
||||||
|
setShadingPreset(m_ShadingPreset);
|
||||||
|
}
|
||||||
|
|
||||||
void vtkVoxImage::SetRepresentation(Representation mode) {
|
void vtkVoxImage::SetRepresentation(Representation mode) {
|
||||||
|
Puppet::SetRepresentation(mode); // Ensure base class data state is updated
|
||||||
|
|
||||||
if (mode == Wireframe) {
|
if (mode == Wireframe) {
|
||||||
m_Actor->SetVisibility(0);
|
m_Actor->SetVisibility(0);
|
||||||
m_OutlineActor->SetVisibility(1);
|
m_OutlineActor->SetVisibility(1);
|
||||||
} else if (mode == Surface) {
|
m_OutlineActor->GetProperty()->SetRepresentationToWireframe();
|
||||||
|
} else if (mode == Volume) {
|
||||||
m_Actor->SetVisibility(1);
|
m_Actor->SetVisibility(1);
|
||||||
m_OutlineActor->SetVisibility(1); // Keep outline visible as boundary
|
m_OutlineActor->SetVisibility(1);
|
||||||
|
m_OutlineActor->GetProperty()->SetRepresentationToWireframe();
|
||||||
} else {
|
} else {
|
||||||
Puppet::SetRepresentation(mode);
|
// Other representations (Points, Surface, etc) are handled by basic Puppet
|
||||||
|
// behavior which affects the m_Asm parts.
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
void vtkVoxImage::serialize_display(uLib::Archive::display_properties_archive & ar, const unsigned int version) {
|
void vtkVoxImage::serialize_display(uLib::Archive::display_properties_archive & ar, const unsigned int version) {
|
||||||
// Call base class if it has display properties
|
// Call base class to show Transform and Appearance properties
|
||||||
Puppet::serialize_display(ar, version);
|
// Puppet::serialize_display(ar, version);
|
||||||
|
|
||||||
// Use the member variables if they are available
|
// Use the member variables for volume rendering parameters
|
||||||
ar & boost::serialization::make_hrp("Window", m_Window);
|
ar & boost::serialization::make_hrp("Window", m_Window);
|
||||||
ar & boost::serialization::make_hrp("Level", m_Level);
|
ar & boost::serialization::make_hrp("Level", m_Level);
|
||||||
ar & boost::serialization::make_hrp_enum("Shading", m_ShadingPreset, {"MIP", "Composite", "Composite Shaded", "MIP Bone", "MIP Hot", "Additive"});
|
ar & boost::serialization::make_hrp_enum("Shading", m_ShadingPreset,
|
||||||
|
{"MIP", "Composite", "Composite Shaded", "MIP Bone", "MIP Hot", "Additive"});
|
||||||
|
}
|
||||||
|
|
||||||
|
void vtkVoxImage::SyncFromVtk() {
|
||||||
|
if (auto *root = this->GetProxyProp()) {
|
||||||
|
vtkMatrix4x4 *rootMat = root->GetUserMatrix();
|
||||||
|
if (rootMat) {
|
||||||
|
Matrix4f vtkLocal = VtkToMatrix4f(rootMat);
|
||||||
|
// Synchronize TRS from VTK, compensating for local volume offset
|
||||||
|
m_Content.FromMatrix(vtkLocal); // * m_Content.GetLocalMatrix().inverse());
|
||||||
|
m_Content.Updated();
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
void vtkVoxImage::Update() {
|
void vtkVoxImage::Update() {
|
||||||
|
if (auto *root = vtkProp3D::SafeDownCast(this->GetProp())) {
|
||||||
|
vtkNew<vtkMatrix4x4> m;
|
||||||
|
Matrix4fToVtk(m_Content.GetMatrix(), m); // * m_Content.GetLocalMatrix(), m);
|
||||||
|
root->SetUserMatrix(m);
|
||||||
|
root->Modified();
|
||||||
|
// std::cout << "[vtkVoxImage::Update] Set Proxy UserMatrix:" << std::endl;
|
||||||
|
// std::cout << m_Content.GetMatrix() << std::endl;
|
||||||
|
}
|
||||||
setShadingPreset(m_ShadingPreset);
|
setShadingPreset(m_ShadingPreset);
|
||||||
m_Actor->Update();
|
m_Actor->Update();
|
||||||
m_Outline->SetBounds(m_Image->GetBounds());
|
m_Outline->SetBounds(m_Image->GetBounds());
|
||||||
m_Outline->Update();
|
m_Outline->Update();
|
||||||
|
|
||||||
|
ConnectionBlock blocker(m_UpdateConnection);
|
||||||
|
this->Puppet::Update();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
void vtkVoxImage::InstallPipe() {
|
void vtkVoxImage::InstallPipe() {
|
||||||
vtkSmartPointer<vtkSmartVolumeMapper> mapper =
|
vtkSmartPointer<vtkSmartVolumeMapper> mapper =
|
||||||
vtkSmartPointer<vtkSmartVolumeMapper>::New();
|
vtkSmartPointer<vtkSmartVolumeMapper>::New();
|
||||||
@@ -318,7 +350,7 @@ void vtkVoxImage::InstallPipe() {
|
|||||||
mapper->Update();
|
mapper->Update();
|
||||||
|
|
||||||
m_Actor->SetMapper(mapper);
|
m_Actor->SetMapper(mapper);
|
||||||
this->setShadingPreset(0);
|
this->setShadingPreset(m_ShadingPreset);
|
||||||
mapper->Update();
|
mapper->Update();
|
||||||
|
|
||||||
m_Outline->SetBounds(m_Image->GetBounds());
|
m_Outline->SetBounds(m_Image->GetBounds());
|
||||||
@@ -330,11 +362,12 @@ void vtkVoxImage::InstallPipe() {
|
|||||||
m_OutlineActor->GetProperty()->SetRepresentationToWireframe();
|
m_OutlineActor->GetProperty()->SetRepresentationToWireframe();
|
||||||
m_OutlineActor->GetProperty()->SetAmbient(0.7);
|
m_OutlineActor->GetProperty()->SetAmbient(0.7);
|
||||||
|
|
||||||
this->SetProp(m_Actor);
|
m_Asm->AddPart(m_Actor);
|
||||||
this->SetProp(m_OutlineActor);
|
m_Asm->AddPart(m_OutlineActor);
|
||||||
|
this->SetProp(m_Asm);
|
||||||
|
|
||||||
// Default look
|
// Default look
|
||||||
this->SetRepresentation(Surface);
|
this->SetRepresentation(Volume);
|
||||||
}
|
}
|
||||||
|
|
||||||
} // namespace Vtk
|
} // namespace Vtk
|
||||||
|
|||||||
@@ -31,6 +31,7 @@
|
|||||||
#include <vtkVolume.h>
|
#include <vtkVolume.h>
|
||||||
#include <vtkXMLImageDataReader.h>
|
#include <vtkXMLImageDataReader.h>
|
||||||
#include <vtkXMLImageDataWriter.h>
|
#include <vtkXMLImageDataWriter.h>
|
||||||
|
#include <vtkAssembly.h>
|
||||||
|
|
||||||
#include <Math/VoxImage.h>
|
#include <Math/VoxImage.h>
|
||||||
|
|
||||||
@@ -38,6 +39,8 @@
|
|||||||
|
|
||||||
class vtkImageData;
|
class vtkImageData;
|
||||||
class vtkActor;
|
class vtkActor;
|
||||||
|
class vtkColorTransferFunction;
|
||||||
|
class vtkPiecewiseFunction;
|
||||||
|
|
||||||
namespace uLib {
|
namespace uLib {
|
||||||
namespace Vtk {
|
namespace Vtk {
|
||||||
@@ -54,6 +57,8 @@ public:
|
|||||||
|
|
||||||
void SetContent();
|
void SetContent();
|
||||||
|
|
||||||
|
vtkProp3D *GetProp() override { return m_Asm; }
|
||||||
|
|
||||||
vtkImageData *GetImageData();
|
vtkImageData *GetImageData();
|
||||||
|
|
||||||
void SaveToXMLFile(const char *fname);
|
void SaveToXMLFile(const char *fname);
|
||||||
@@ -64,8 +69,10 @@ public:
|
|||||||
|
|
||||||
void setShadingPreset(int blendType = 2);
|
void setShadingPreset(int blendType = 2);
|
||||||
void SetRepresentation(Representation mode);
|
void SetRepresentation(Representation mode);
|
||||||
|
void RescaleShaderRange();
|
||||||
|
|
||||||
void Update() override;
|
void Update() override;
|
||||||
|
void SyncFromVtk() override;
|
||||||
void serialize_display(uLib::Archive::display_properties_archive & ar, const unsigned int version = 0) override;
|
void serialize_display(uLib::Archive::display_properties_archive & ar, const unsigned int version = 0) override;
|
||||||
|
|
||||||
protected:
|
protected:
|
||||||
@@ -77,6 +84,7 @@ private:
|
|||||||
vtkImageData *m_Image;
|
vtkImageData *m_Image;
|
||||||
vtkCubeSource *m_Outline;
|
vtkCubeSource *m_Outline;
|
||||||
vtkActor *m_OutlineActor;
|
vtkActor *m_OutlineActor;
|
||||||
|
vtkAssembly *m_Asm;
|
||||||
|
|
||||||
vtkXMLImageDataReader *m_Reader;
|
vtkXMLImageDataReader *m_Reader;
|
||||||
vtkXMLImageDataWriter *m_Writer;
|
vtkXMLImageDataWriter *m_Writer;
|
||||||
@@ -86,6 +94,11 @@ private:
|
|||||||
float m_Window;
|
float m_Window;
|
||||||
float m_Level;
|
float m_Level;
|
||||||
int m_ShadingPreset;
|
int m_ShadingPreset;
|
||||||
|
|
||||||
|
Connection m_UpdateConnection;
|
||||||
|
|
||||||
|
class vtkColorTransferFunction *m_ColorFun;
|
||||||
|
class vtkPiecewiseFunction *m_OpacityFun;
|
||||||
};
|
};
|
||||||
|
|
||||||
} // namespace Vtk
|
} // namespace Vtk
|
||||||
|
|||||||
@@ -4,6 +4,7 @@ set(TESTS
|
|||||||
vtkHandlerWidget
|
vtkHandlerWidget
|
||||||
PuppetPropertyTest
|
PuppetPropertyTest
|
||||||
PuppetParentingTest
|
PuppetParentingTest
|
||||||
|
vtkQViewportTest
|
||||||
# vtkVoxImageTest
|
# vtkVoxImageTest
|
||||||
# vtkTriangleMeshTest
|
# vtkTriangleMeshTest
|
||||||
)
|
)
|
||||||
|
|||||||
@@ -64,7 +64,7 @@ int main() {
|
|||||||
|
|
||||||
// 3. Move the parent and verify the child follows
|
// 3. Move the parent and verify the child follows
|
||||||
assembly->SetPosition(Vector3f(100, 0, 0));
|
assembly->SetPosition(Vector3f(100, 0, 0));
|
||||||
assembly->Update();
|
assembly->Updated();
|
||||||
|
|
||||||
// In VTK assemblies, the child's absolute matrix should reflect the parent's transform
|
// In VTK assemblies, the child's absolute matrix should reflect the parent's transform
|
||||||
vtkProp3D* box1Prop = vtkProp3D::SafeDownCast(box1Puppet->GetProp());
|
vtkProp3D* box1Prop = vtkProp3D::SafeDownCast(box1Puppet->GetProp());
|
||||||
|
|||||||
@@ -36,6 +36,8 @@ printf("..:: Testing " #name " ::..\n");
|
|||||||
|
|
||||||
#define TEST1(val) _fail += (val)==0
|
#define TEST1(val) _fail += (val)==0
|
||||||
#define TEST0(val) _fail += (val)!=0
|
#define TEST0(val) _fail += (val)!=0
|
||||||
|
#define ASSERT_EQUAL(a,b) if((a)!=(b)) { printf("Assertion failed: " #a " != " #b " at line %d\n", __LINE__); _fail++; }
|
||||||
|
#define ASSERT_NOT_NULL(ptr) if((ptr)==NULL) { printf("Assertion failed: " #ptr " is NULL at line %d\n", __LINE__); _fail++; }
|
||||||
#define END_TESTING return _fail;
|
#define END_TESTING return _fail;
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
87
src/Vtk/testing/vtkQViewportTest.cpp
Normal file
87
src/Vtk/testing/vtkQViewportTest.cpp
Normal file
@@ -0,0 +1,87 @@
|
|||||||
|
/*//////////////////////////////////////////////////////////////////////////////
|
||||||
|
// CMT Cosmic Muon Tomography project //////////////////////////////////////////
|
||||||
|
////////////////////////////////////////////////////////////////////////////////
|
||||||
|
|
||||||
|
Copyright (c) 2014, Universita' degli Studi di Padova, INFN sez. di Padova
|
||||||
|
All rights reserved
|
||||||
|
|
||||||
|
Authors: Andrea Rigoni Garola < andrea.rigoni@pd.infn.it >
|
||||||
|
|
||||||
|
------------------------------------------------------------------
|
||||||
|
This library is free software; you can redistribute it and/or
|
||||||
|
modify it under the terms of the GNU Lesser General Public
|
||||||
|
License as published by the Free Software Foundation; either
|
||||||
|
version 3.0 of the License, or (at your option) any later version.
|
||||||
|
|
||||||
|
This library is distributed in the hope that it will be useful,
|
||||||
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
||||||
|
Lesser General Public License for more details.
|
||||||
|
|
||||||
|
You should have received a copy of the GNU Lesser General Public
|
||||||
|
License along with this library.
|
||||||
|
|
||||||
|
//////////////////////////////////////////////////////////////////////////////*/
|
||||||
|
|
||||||
|
#ifdef HAVE_CONFIG_H
|
||||||
|
#include "config.h"
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#include <QApplication>
|
||||||
|
#include <QtGlobal>
|
||||||
|
#include <Vtk/vtkQViewport.h>
|
||||||
|
|
||||||
|
#include <vtkSmartPointer.h>
|
||||||
|
#include <vtkCubeSource.h>
|
||||||
|
#include <vtkPolyDataMapper.h>
|
||||||
|
#include <vtkActor.h>
|
||||||
|
#include <vtkProperty.h>
|
||||||
|
#include <vtkRenderer.h>
|
||||||
|
|
||||||
|
#include "testing-prototype.h"
|
||||||
|
|
||||||
|
using namespace uLib;
|
||||||
|
|
||||||
|
int main(int argc, char** argv)
|
||||||
|
{
|
||||||
|
// Force X11 on Linux to avoid Wayland connection issues in headless/X11 environments
|
||||||
|
#if defined(Q_OS_LINUX)
|
||||||
|
qputenv("QT_QPA_PLATFORM", "xcb");
|
||||||
|
#endif
|
||||||
|
|
||||||
|
BEGIN_TESTING(vtk QViewport Test);
|
||||||
|
|
||||||
|
QApplication app(argc, argv);
|
||||||
|
app.processEvents();
|
||||||
|
|
||||||
|
Vtk::QViewport viewport;
|
||||||
|
viewport.resize(800, 600);
|
||||||
|
viewport.show();
|
||||||
|
|
||||||
|
vtkSmartPointer<vtkCubeSource> cube = vtkSmartPointer<vtkCubeSource>::New();
|
||||||
|
cube->SetXLength(10);
|
||||||
|
cube->SetYLength(10);
|
||||||
|
cube->SetZLength(10);
|
||||||
|
cube->Update();
|
||||||
|
|
||||||
|
vtkSmartPointer<vtkPolyDataMapper> mapper = vtkSmartPointer<vtkPolyDataMapper>::New();
|
||||||
|
mapper->SetInputConnection(cube->GetOutputPort());
|
||||||
|
|
||||||
|
vtkSmartPointer<vtkActor> actor = vtkSmartPointer<vtkActor>::New();
|
||||||
|
actor->SetMapper(mapper);
|
||||||
|
actor->GetProperty()->SetColor(1, 0, 0);
|
||||||
|
|
||||||
|
viewport.addProp(actor);
|
||||||
|
viewport.Render();
|
||||||
|
|
||||||
|
ASSERT_NOT_NULL(viewport.GetRenderWindow());
|
||||||
|
ASSERT_NOT_NULL(viewport.GetRenderer());
|
||||||
|
|
||||||
|
if (std::getenv("CTEST_PROJECT_NAME") == nullptr) {
|
||||||
|
// Run application for a while to see the result
|
||||||
|
return app.exec();
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
END_TESTING;
|
||||||
|
}
|
||||||
@@ -58,6 +58,7 @@
|
|||||||
#include <vtkPolyData.h>
|
#include <vtkPolyData.h>
|
||||||
#include <vtkFeatureEdges.h>
|
#include <vtkFeatureEdges.h>
|
||||||
#include <vtkTransform.h>
|
#include <vtkTransform.h>
|
||||||
|
#include <vtkCubeSource.h>
|
||||||
#include <vtkRenderWindow.h>
|
#include <vtkRenderWindow.h>
|
||||||
|
|
||||||
#include "uLibVtkInterface.h"
|
#include "uLibVtkInterface.h"
|
||||||
@@ -65,6 +66,7 @@
|
|||||||
#include "Math/Dense.h"
|
#include "Math/Dense.h"
|
||||||
#include "Vtk/Math/vtkDense.h"
|
#include "Vtk/Math/vtkDense.h"
|
||||||
#include "Core/Property.h"
|
#include "Core/Property.h"
|
||||||
|
#include "Math/Transform.h"
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
@@ -80,9 +82,10 @@ namespace Vtk {
|
|||||||
|
|
||||||
class PuppetData {
|
class PuppetData {
|
||||||
public:
|
public:
|
||||||
PuppetData() :
|
PuppetData(Puppet* owner) :
|
||||||
|
m_Puppet(owner),
|
||||||
m_Renderers(vtkSmartPointer<vtkRendererCollection>::New()),
|
m_Renderers(vtkSmartPointer<vtkRendererCollection>::New()),
|
||||||
m_Assembly(vtkSmartPointer<vtkAssembly>::New()),
|
m_Prop(nullptr),
|
||||||
m_ShowBoundingBox(false),
|
m_ShowBoundingBox(false),
|
||||||
m_ShowScaleMeasures(false),
|
m_ShowScaleMeasures(false),
|
||||||
m_Representation(Puppet::Surface),
|
m_Representation(Puppet::Surface),
|
||||||
@@ -99,9 +102,10 @@ public:
|
|||||||
// No manual Delete needed for smart pointers
|
// No manual Delete needed for smart pointers
|
||||||
}
|
}
|
||||||
|
|
||||||
|
Puppet *m_Puppet;
|
||||||
// members //
|
// members //
|
||||||
vtkSmartPointer<vtkRendererCollection> m_Renderers;
|
vtkSmartPointer<vtkRendererCollection> m_Renderers;
|
||||||
vtkSmartPointer<vtkAssembly> m_Assembly;
|
vtkSmartPointer<vtkProp3D> m_Prop;
|
||||||
|
|
||||||
vtkSmartPointer<vtkOutlineSource> m_OutlineSource;
|
vtkSmartPointer<vtkOutlineSource> m_OutlineSource;
|
||||||
vtkSmartPointer<vtkActor> m_OutlineActor;
|
vtkSmartPointer<vtkActor> m_OutlineActor;
|
||||||
@@ -118,6 +122,8 @@ public:
|
|||||||
bool m_Selected;
|
bool m_Selected;
|
||||||
bool m_Visibility;
|
bool m_Visibility;
|
||||||
bool m_Dragable;
|
bool m_Dragable;
|
||||||
|
|
||||||
|
//
|
||||||
TRS m_Transform;
|
TRS m_Transform;
|
||||||
|
|
||||||
void ApplyAppearance(vtkProp *p) {
|
void ApplyAppearance(vtkProp *p) {
|
||||||
@@ -128,11 +134,11 @@ public:
|
|||||||
|
|
||||||
vtkActor *actor = vtkActor::SafeDownCast(p);
|
vtkActor *actor = vtkActor::SafeDownCast(p);
|
||||||
if (actor) {
|
if (actor) {
|
||||||
if (m_Representation != -1) {
|
if (m_Representation != -1 && m_Representation != Puppet::Volume) {
|
||||||
if (m_Representation == Puppet::SurfaceWithEdges) {
|
if (m_Representation == Puppet::SurfaceWithEdges) {
|
||||||
actor->GetProperty()->SetRepresentation(VTK_SURFACE);
|
actor->GetProperty()->SetRepresentation(VTK_SURFACE);
|
||||||
actor->GetProperty()->SetEdgeVisibility(1);
|
actor->GetProperty()->SetEdgeVisibility(1);
|
||||||
} else {
|
} else if (m_Representation != Puppet::Outline && m_Representation != Puppet::Slice) {
|
||||||
actor->GetProperty()->SetRepresentation(m_Representation);
|
actor->GetProperty()->SetRepresentation(m_Representation);
|
||||||
actor->GetProperty()->SetEdgeVisibility(0);
|
actor->GetProperty()->SetEdgeVisibility(0);
|
||||||
}
|
}
|
||||||
@@ -148,15 +154,18 @@ public:
|
|||||||
} else if (vtkAssembly *asm_p = vtkAssembly::SafeDownCast(p)) {
|
} else if (vtkAssembly *asm_p = vtkAssembly::SafeDownCast(p)) {
|
||||||
// Recursively apply to parts of the assembly
|
// Recursively apply to parts of the assembly
|
||||||
vtkProp3DCollection *parts = asm_p->GetParts();
|
vtkProp3DCollection *parts = asm_p->GetParts();
|
||||||
|
if (parts) {
|
||||||
parts->InitTraversal();
|
parts->InitTraversal();
|
||||||
for (int i = 0; i < parts->GetNumberOfItems(); ++i) {
|
for (int i = 0; i < parts->GetNumberOfItems(); ++i) {
|
||||||
this->ApplyAppearance(parts->GetNextProp3D());
|
this->ApplyAppearance(parts->GetNextProp3D());
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
void ApplyTransform(vtkProp3D* p3d) {
|
void ApplyTransform(vtkProp3D* p3d) {
|
||||||
if (p3d) {
|
if (p3d) {
|
||||||
|
p3d->SetUserMatrix(nullptr);
|
||||||
p3d->SetPosition(m_Transform.position.x(), m_Transform.position.y(), m_Transform.position.z());
|
p3d->SetPosition(m_Transform.position.x(), m_Transform.position.y(), m_Transform.position.z());
|
||||||
|
|
||||||
// Convert Model Radians to VTK Degrees
|
// Convert Model Radians to VTK Degrees
|
||||||
@@ -165,51 +174,87 @@ public:
|
|||||||
m_Transform.rotation.z() / CLHEP::degree);
|
m_Transform.rotation.z() / CLHEP::degree);
|
||||||
|
|
||||||
p3d->SetScale(m_Transform.scaling.x(), m_Transform.scaling.y(), m_Transform.scaling.z());
|
p3d->SetScale(m_Transform.scaling.x(), m_Transform.scaling.y(), m_Transform.scaling.z());
|
||||||
p3d->SetUserMatrix(nullptr);
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
void UpdateHighlight() {
|
void UpdateHighlight() {
|
||||||
if (m_Selected) {
|
if (m_Selected) {
|
||||||
if (!m_HighlightActor) {
|
|
||||||
vtkSmartPointer<vtkFeatureEdges> edges = vtkSmartPointer<vtkFeatureEdges>::New();
|
|
||||||
edges->BoundaryEdgesOn();
|
|
||||||
edges->FeatureEdgesOn();
|
|
||||||
edges->SetFeatureAngle(30);
|
|
||||||
edges->NonManifoldEdgesOn();
|
|
||||||
edges->ManifoldEdgesOff();
|
|
||||||
|
|
||||||
// Find first polydata in assembly to highlight
|
// Find first polydata in assembly to highlight
|
||||||
vtkPropCollection *parts = m_Assembly->GetParts();
|
vtkPolyData* polydata = nullptr;
|
||||||
|
if (vtkActor *actor = vtkActor::SafeDownCast(m_Prop)) {
|
||||||
|
if (actor->GetMapper()) {
|
||||||
|
polydata = vtkPolyData::SafeDownCast(actor->GetMapper()->GetDataSetInput());
|
||||||
|
}
|
||||||
|
} else if (vtkAssembly *asm_p = vtkAssembly::SafeDownCast(m_Prop)) {
|
||||||
|
vtkPropCollection *parts = asm_p->GetParts();
|
||||||
|
if (parts) {
|
||||||
parts->InitTraversal();
|
parts->InitTraversal();
|
||||||
for (int i = 0; i < parts->GetNumberOfItems(); ++i) {
|
for (int i = 0; i < parts->GetNumberOfItems(); ++i) {
|
||||||
vtkActor *actor = vtkActor::SafeDownCast(parts->GetNextProp());
|
vtkActor *a = vtkActor::SafeDownCast(parts->GetNextProp());
|
||||||
if (actor && actor->GetMapper() && actor->GetMapper()->GetDataSetInput()) {
|
if (a && a->GetMapper()) {
|
||||||
edges->SetInputData(vtkPolyData::SafeDownCast(actor->GetMapper()->GetDataSetInput()));
|
polydata = vtkPolyData::SafeDownCast(a->GetMapper()->GetDataSetInput());
|
||||||
break;
|
if (polydata) break;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!polydata) {
|
||||||
|
if (m_HighlightActor) {
|
||||||
|
m_Renderers->InitTraversal();
|
||||||
|
for (int i = 0; i < m_Renderers->GetNumberOfItems(); ++i) {
|
||||||
|
m_Renderers->GetNextItem()->RemoveActor(m_HighlightActor);
|
||||||
|
}
|
||||||
|
m_HighlightActor = nullptr;
|
||||||
|
}
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!m_HighlightActor) {
|
||||||
|
vtkSmartPointer<vtkCubeSource> cube = vtkSmartPointer<vtkCubeSource>::New();
|
||||||
|
double bounds[6];
|
||||||
|
polydata->GetBounds(bounds);
|
||||||
|
// Add a small padding to prevent z-fighting
|
||||||
|
double maxDim = std::max({bounds[1]-bounds[0], bounds[3]-bounds[2], bounds[5]-bounds[4]});
|
||||||
|
double pad = maxDim * 0.02;
|
||||||
|
if(pad < 1e-4) pad = 0.05;
|
||||||
|
cube->SetBounds(bounds[0]-pad, bounds[1]+pad,
|
||||||
|
bounds[2]-pad, bounds[3]+pad,
|
||||||
|
bounds[4]-pad, bounds[5]+pad);
|
||||||
|
|
||||||
m_HighlightActor = vtkSmartPointer<vtkActor>::New();
|
m_HighlightActor = vtkSmartPointer<vtkActor>::New();
|
||||||
vtkSmartPointer<vtkPolyDataMapper> mapper = vtkSmartPointer<vtkPolyDataMapper>::New();
|
vtkSmartPointer<vtkPolyDataMapper> mapper = vtkSmartPointer<vtkPolyDataMapper>::New();
|
||||||
mapper->SetInputConnection(edges->GetOutputPort());
|
mapper->SetInputConnection(cube->GetOutputPort());
|
||||||
m_HighlightActor->SetMapper(mapper);
|
m_HighlightActor->SetMapper(mapper);
|
||||||
m_HighlightActor->GetProperty()->SetColor(1.0, 0.5, 0.0); // Orange
|
m_HighlightActor->GetProperty()->SetRepresentationToWireframe();
|
||||||
|
m_HighlightActor->GetProperty()->SetColor(1.0, 0.0, 0.0); // Red
|
||||||
m_HighlightActor->GetProperty()->SetLineWidth(2.0);
|
m_HighlightActor->GetProperty()->SetLineWidth(2.0);
|
||||||
m_HighlightActor->GetProperty()->SetLighting(0);
|
m_HighlightActor->GetProperty()->SetLighting(0);
|
||||||
}
|
|
||||||
|
|
||||||
// Update highlight matrix from the root prop
|
|
||||||
vtkProp3D* root = nullptr;
|
|
||||||
if (m_Assembly->GetParts()->GetNumberOfItems() == 1) {
|
|
||||||
root = vtkProp3D::SafeDownCast(m_Assembly->GetParts()->GetLastProp());
|
|
||||||
} else {
|
} else {
|
||||||
root = m_Assembly;
|
if (auto* mapper = vtkPolyDataMapper::SafeDownCast(m_HighlightActor->GetMapper())) {
|
||||||
|
if (auto* cube = vtkCubeSource::SafeDownCast(mapper->GetInputAlgorithm())) {
|
||||||
|
double bounds[6];
|
||||||
|
polydata->GetBounds(bounds);
|
||||||
|
double maxDim = std::max({bounds[1]-bounds[0], bounds[3]-bounds[2], bounds[5]-bounds[4]});
|
||||||
|
double pad = maxDim * 0.02;
|
||||||
|
if(pad < 1e-4) pad = 0.05;
|
||||||
|
cube->SetBounds(bounds[0]-pad, bounds[1]+pad,
|
||||||
|
bounds[2]-pad, bounds[3]+pad,
|
||||||
|
bounds[4]-pad, bounds[5]+pad);
|
||||||
|
cube->Modified();
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if (root) {
|
// Update highlight matrix from the model world matrix
|
||||||
// Now that we use internal TRS, the prop's total matrix is GetMatrix()
|
if (m_Puppet) {
|
||||||
m_HighlightActor->SetUserMatrix(root->GetMatrix());
|
if (auto* content = m_Puppet->GetContent()) {
|
||||||
|
if (auto* tr = dynamic_cast<uLib::TRS*>(content)) {
|
||||||
|
vtkNew<vtkMatrix4x4> vwm;
|
||||||
|
Matrix4fToVtk(tr->GetWorldMatrix(), vwm);
|
||||||
|
m_HighlightActor->SetUserMatrix(vwm);
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
m_Renderers->InitTraversal();
|
m_Renderers->InitTraversal();
|
||||||
@@ -242,7 +287,7 @@ public:
|
|||||||
|
|
||||||
|
|
||||||
|
|
||||||
Puppet::Puppet() : Object(), pd(new PuppetData) {
|
Puppet::Puppet() : Object(), pd(new PuppetData(this)) {
|
||||||
ULIB_ACTIVATE_DISPLAY_PROPERTIES;
|
ULIB_ACTIVATE_DISPLAY_PROPERTIES;
|
||||||
for (auto* p : this->GetDisplayProperties()) {
|
for (auto* p : this->GetDisplayProperties()) {
|
||||||
uLib::Object::connect(p, &uLib::PropertyBase::Updated, this, &Puppet::Update);
|
uLib::Object::connect(p, &uLib::PropertyBase::Updated, this, &Puppet::Update);
|
||||||
@@ -256,25 +301,28 @@ Puppet::~Puppet()
|
|||||||
|
|
||||||
vtkProp *Puppet::GetProp()
|
vtkProp *Puppet::GetProp()
|
||||||
{
|
{
|
||||||
if (pd->m_Assembly->GetParts()->GetNumberOfItems() == 1)
|
return pd->m_Prop;
|
||||||
return pd->m_Assembly->GetParts()->GetLastProp();
|
}
|
||||||
else
|
|
||||||
return pd->m_Assembly;
|
vtkProp3D *Puppet::GetProxyProp()
|
||||||
|
{
|
||||||
|
// The handler should manipulate the highlight actor if it exists
|
||||||
|
if (pd->m_HighlightActor) {
|
||||||
|
return pd->m_HighlightActor;
|
||||||
|
}
|
||||||
|
return vtkProp3D::SafeDownCast(this->GetProp());
|
||||||
}
|
}
|
||||||
|
|
||||||
void Puppet::SetProp(vtkProp *prop)
|
void Puppet::SetProp(vtkProp *prop)
|
||||||
{
|
{
|
||||||
if(prop) {
|
if(prop) {
|
||||||
prop->SetPickable(pd->m_Selectable);
|
prop->SetPickable(pd->m_Selectable);
|
||||||
if (auto* p3d = vtkProp3D::SafeDownCast(prop)) {
|
pd->m_Prop = vtkProp3D::SafeDownCast(prop);
|
||||||
pd->m_Assembly->AddPart(p3d);
|
|
||||||
}
|
|
||||||
pd->ApplyAppearance(prop);
|
pd->ApplyAppearance(prop);
|
||||||
|
|
||||||
// For the first actor added, seed the tracked display values from the VTK
|
// For the first actor added, seed the tracked display values from the VTK
|
||||||
// actor's current state so the display properties panel shows meaningful
|
// actor's current state so the display properties panel shows meaningful
|
||||||
// initial values instead of the -1 "not-overriding" sentinels.
|
// initial values instead of the -1 "not-overriding" sentinels.
|
||||||
if (pd->m_Assembly->GetParts()->GetNumberOfItems() == 1) {
|
|
||||||
if (auto* actor = vtkActor::SafeDownCast(prop)) {
|
if (auto* actor = vtkActor::SafeDownCast(prop)) {
|
||||||
vtkProperty* vp = actor->GetProperty();
|
vtkProperty* vp = actor->GetProperty();
|
||||||
if (pd->m_Representation < 0)
|
if (pd->m_Representation < 0)
|
||||||
@@ -289,7 +337,6 @@ void Puppet::SetProp(vtkProp *prop)
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
void Puppet::RemoveProp(vtkProp *prop)
|
void Puppet::RemoveProp(vtkProp *prop)
|
||||||
{
|
{
|
||||||
@@ -309,12 +356,18 @@ void Puppet::ApplyTransform(vtkProp3D* p3d)
|
|||||||
|
|
||||||
vtkPropCollection *Puppet::GetParts()
|
vtkPropCollection *Puppet::GetParts()
|
||||||
{
|
{
|
||||||
return pd->m_Assembly->GetParts();
|
if (auto* asm_p = vtkAssembly::SafeDownCast(pd->m_Prop)) {
|
||||||
|
return asm_p->GetParts();
|
||||||
|
}
|
||||||
|
return nullptr;
|
||||||
}
|
}
|
||||||
|
|
||||||
vtkPropCollection *Puppet::GetProps()
|
vtkPropCollection *Puppet::GetProps()
|
||||||
{
|
{
|
||||||
return pd->m_Assembly->GetParts();
|
if (auto* asm_p = vtkAssembly::SafeDownCast(pd->m_Prop)) {
|
||||||
|
return asm_p->GetParts();
|
||||||
|
}
|
||||||
|
return nullptr;
|
||||||
}
|
}
|
||||||
|
|
||||||
void Puppet::ConnectRenderer(vtkRenderer *renderer)
|
void Puppet::ConnectRenderer(vtkRenderer *renderer)
|
||||||
@@ -368,7 +421,8 @@ vtkRendererCollection *Puppet::GetRenderers() const
|
|||||||
void Puppet::PrintSelf(std::ostream &o) const
|
void Puppet::PrintSelf(std::ostream &o) const
|
||||||
{
|
{
|
||||||
o << "Props Assembly: \n";
|
o << "Props Assembly: \n";
|
||||||
pd->m_Assembly->PrintSelf(o,vtkIndent(1));
|
if (pd->m_Prop)
|
||||||
|
pd->m_Prop->PrintSelf(o,vtkIndent(1));
|
||||||
|
|
||||||
o << "Connected Renderers: \n";
|
o << "Connected Renderers: \n";
|
||||||
pd->m_Renderers->PrintSelf(o,vtkIndent(1));
|
pd->m_Renderers->PrintSelf(o,vtkIndent(1));
|
||||||
@@ -388,9 +442,11 @@ void Puppet::ShowBoundingBox(bool show)
|
|||||||
pd->m_OutlineActor->GetProperty()->SetColor(1.0, 1.0, 1.0);
|
pd->m_OutlineActor->GetProperty()->SetColor(1.0, 1.0, 1.0);
|
||||||
}
|
}
|
||||||
|
|
||||||
double* bounds = pd->m_Assembly->GetBounds();
|
if (pd->m_Prop) {
|
||||||
|
double* bounds = pd->m_Prop->GetBounds();
|
||||||
pd->m_OutlineSource->SetBounds(bounds);
|
pd->m_OutlineSource->SetBounds(bounds);
|
||||||
pd->m_OutlineSource->Update();
|
pd->m_OutlineSource->Update();
|
||||||
|
}
|
||||||
|
|
||||||
pd->m_Renderers->InitTraversal();
|
pd->m_Renderers->InitTraversal();
|
||||||
for (int i = 0; i < pd->m_Renderers->GetNumberOfItems(); ++i) {
|
for (int i = 0; i < pd->m_Renderers->GetNumberOfItems(); ++i) {
|
||||||
@@ -420,8 +476,10 @@ void Puppet::ShowScaleMeasures(bool show)
|
|||||||
pd->m_CubeAxesActor->GetProperty()->SetColor(1.0, 1.0, 1.0);
|
pd->m_CubeAxesActor->GetProperty()->SetColor(1.0, 1.0, 1.0);
|
||||||
}
|
}
|
||||||
|
|
||||||
double* bounds = pd->m_Assembly->GetBounds();
|
if (pd->m_Prop) {
|
||||||
|
double* bounds = pd->m_Prop->GetBounds();
|
||||||
pd->m_CubeAxesActor->SetBounds(bounds);
|
pd->m_CubeAxesActor->SetBounds(bounds);
|
||||||
|
}
|
||||||
|
|
||||||
pd->m_Renderers->InitTraversal();
|
pd->m_Renderers->InitTraversal();
|
||||||
for (int i = 0; i < pd->m_Renderers->GetNumberOfItems(); ++i) {
|
for (int i = 0; i < pd->m_Renderers->GetNumberOfItems(); ++i) {
|
||||||
@@ -443,12 +501,7 @@ void Puppet::ShowScaleMeasures(bool show)
|
|||||||
void Puppet::SetRepresentation(Representation mode)
|
void Puppet::SetRepresentation(Representation mode)
|
||||||
{
|
{
|
||||||
pd->m_Representation = static_cast<int>(mode);
|
pd->m_Representation = static_cast<int>(mode);
|
||||||
|
pd->ApplyAppearance(pd->m_Prop);
|
||||||
vtkProp3DCollection *props = pd->m_Assembly->GetParts();
|
|
||||||
props->InitTraversal();
|
|
||||||
for (int i = 0; i < props->GetNumberOfItems(); ++i) {
|
|
||||||
pd->ApplyAppearance(props->GetNextProp3D());
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
void Puppet::SetRepresentation(const char *mode)
|
void Puppet::SetRepresentation(const char *mode)
|
||||||
@@ -468,23 +521,13 @@ void Puppet::SetColor(double r, double g, double b)
|
|||||||
pd->m_Color[0] = r;
|
pd->m_Color[0] = r;
|
||||||
pd->m_Color[1] = g;
|
pd->m_Color[1] = g;
|
||||||
pd->m_Color[2] = b;
|
pd->m_Color[2] = b;
|
||||||
|
pd->ApplyAppearance(pd->m_Prop);
|
||||||
vtkProp3DCollection *props = pd->m_Assembly->GetParts();
|
|
||||||
props->InitTraversal();
|
|
||||||
for (int i = 0; i < props->GetNumberOfItems(); ++i) {
|
|
||||||
pd->ApplyAppearance(props->GetNextProp3D());
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
void Puppet::SetOpacity(double alpha)
|
void Puppet::SetOpacity(double alpha)
|
||||||
{
|
{
|
||||||
pd->m_Opacity = alpha;
|
pd->m_Opacity = alpha;
|
||||||
|
pd->ApplyAppearance(pd->m_Prop);
|
||||||
vtkProp3DCollection *props = pd->m_Assembly->GetParts();
|
|
||||||
props->InitTraversal();
|
|
||||||
for (int i = 0; i < props->GetNumberOfItems(); ++i) {
|
|
||||||
pd->ApplyAppearance(props->GetNextProp3D());
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
@@ -496,11 +539,7 @@ void Puppet::SetOpacity(double alpha)
|
|||||||
void Puppet::SetSelectable(bool selectable)
|
void Puppet::SetSelectable(bool selectable)
|
||||||
{
|
{
|
||||||
pd->m_Selectable = selectable;
|
pd->m_Selectable = selectable;
|
||||||
vtkProp3DCollection *props = pd->m_Assembly->GetParts();
|
pd->ApplyAppearance(pd->m_Prop);
|
||||||
props->InitTraversal();
|
|
||||||
for (int i = 0; i < props->GetNumberOfItems(); ++i) {
|
|
||||||
props->GetNextProp3D()->SetPickable(selectable);
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
bool Puppet::IsSelectable() const
|
bool Puppet::IsSelectable() const
|
||||||
@@ -523,39 +562,28 @@ bool Puppet::IsSelected() const
|
|||||||
|
|
||||||
void Puppet::Update()
|
void Puppet::Update()
|
||||||
{
|
{
|
||||||
vtkProp* root = this->GetProp();
|
// Derived classes should have updated the transform if they override Update()
|
||||||
if (root) {
|
// or we can apply base transform if it's default:
|
||||||
// Handle transformation synchronization from content
|
// pd->ApplyTransform(pd->m_Prop);
|
||||||
if (auto* content = dynamic_cast<uLib::AffineTransform*>(GetContent())) {
|
|
||||||
pd->m_Transform = *content; // Uses TRS(const AffineTransform&)
|
|
||||||
}
|
|
||||||
|
|
||||||
if (auto* p3d = vtkProp3D::SafeDownCast(root)) {
|
pd->ApplyAppearance(pd->m_Prop);
|
||||||
pd->ApplyTransform(p3d);
|
|
||||||
}
|
|
||||||
pd->ApplyAppearance(root);
|
|
||||||
}
|
|
||||||
|
|
||||||
vtkProp3DCollection *props = pd->m_Assembly->GetParts();
|
|
||||||
props->InitTraversal();
|
|
||||||
for (int i = 0; i < props->GetNumberOfItems(); ++i) {
|
|
||||||
pd->ApplyAppearance(props->GetNextProp3D());
|
|
||||||
}
|
|
||||||
|
|
||||||
if (pd->m_Selected) {
|
if (pd->m_Selected) {
|
||||||
pd->UpdateHighlight();
|
pd->UpdateHighlight();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if (pd->m_Prop) {
|
||||||
if (pd->m_ShowBoundingBox) {
|
if (pd->m_ShowBoundingBox) {
|
||||||
double* bounds = pd->m_Assembly->GetBounds();
|
double* bounds = pd->m_Prop->GetBounds();
|
||||||
pd->m_OutlineSource->SetBounds(bounds);
|
pd->m_OutlineSource->SetBounds(bounds);
|
||||||
pd->m_OutlineSource->Update();
|
pd->m_OutlineSource->Update();
|
||||||
}
|
}
|
||||||
|
|
||||||
if (pd->m_ShowScaleMeasures) {
|
if (pd->m_ShowScaleMeasures) {
|
||||||
double* bounds = pd->m_Assembly->GetBounds();
|
double* bounds = pd->m_Prop->GetBounds();
|
||||||
pd->m_CubeAxesActor->SetBounds(bounds);
|
pd->m_CubeAxesActor->SetBounds(bounds);
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// Notify that the object has been updated (important for UI refresh)
|
// Notify that the object has been updated (important for UI refresh)
|
||||||
this->Object::Updated();
|
this->Object::Updated();
|
||||||
@@ -570,50 +598,17 @@ void Puppet::Update()
|
|||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
void Puppet::SyncFromVtk()
|
|
||||||
{
|
|
||||||
vtkProp* root = this->GetProp();
|
|
||||||
if (auto* p3d = vtkProp3D::SafeDownCast(root)) {
|
|
||||||
// Handle content synchronization if it's an AffineTransform
|
|
||||||
if (auto* content = dynamic_cast<uLib::AffineTransform*>(GetContent())) {
|
|
||||||
double pos[3], ori[3], scale[3];
|
|
||||||
p3d->GetPosition(pos);
|
|
||||||
p3d->GetOrientation(ori);
|
|
||||||
p3d->GetScale(scale);
|
|
||||||
|
|
||||||
// Convert VTK Degrees to Model Radians
|
|
||||||
content->SetPosition(Vector3f(pos[0], pos[1], pos[2]));
|
|
||||||
content->SetOrientation(Vector3f(ori[0], ori[1], ori[2]) * CLHEP::degree);
|
|
||||||
content->SetScale(Vector3f(scale[0], scale[1], scale[2]));
|
|
||||||
|
|
||||||
// Re-sync internal puppet properties from the now-updated content
|
|
||||||
pd->m_Transform = *content;
|
|
||||||
}
|
|
||||||
else {
|
|
||||||
// Update internal puppet TRS directly from VTK components
|
|
||||||
double pos[3], ori[3], scale[3];
|
|
||||||
p3d->GetPosition(pos);
|
|
||||||
p3d->GetOrientation(ori);
|
|
||||||
p3d->GetScale(scale);
|
|
||||||
pd->m_Transform.position = Vector3f(pos[0], pos[1], pos[2]);
|
|
||||||
// Convert VTK Degrees to internal Radians
|
|
||||||
pd->m_Transform.rotation = Vector3f(ori[0], ori[1], ori[2]) * CLHEP::degree;
|
|
||||||
pd->m_Transform.scaling = Vector3f(scale[0], scale[1], scale[2]);
|
|
||||||
}
|
|
||||||
|
|
||||||
// Notify puppet properties updated
|
|
||||||
if (auto* propPos = this->GetProperty("Position")) propPos->Updated();
|
|
||||||
if (auto* propOri = this->GetProperty("Orientation")) propOri->Updated();
|
|
||||||
if (auto* propScale = this->GetProperty("Scale")) propScale->Updated();
|
|
||||||
|
|
||||||
this->Object::Updated();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
void Puppet::ConnectInteractor(vtkRenderWindowInteractor *interactor)
|
void Puppet::ConnectInteractor(vtkRenderWindowInteractor *interactor)
|
||||||
{
|
{
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
// ------------------------------------------------------ //
|
||||||
|
// SERIALIZE DISPLAY PROPERTIES
|
||||||
|
|
||||||
|
|
||||||
struct TransformProxy {
|
struct TransformProxy {
|
||||||
PuppetData* pd;
|
PuppetData* pd;
|
||||||
template<class Archive>
|
template<class Archive>
|
||||||
@@ -628,10 +623,13 @@ struct AppearanceProxy {
|
|||||||
void serialize(Archive & ar, const unsigned int version) {
|
void serialize(Archive & ar, const unsigned int version) {
|
||||||
ar & boost::serialization::make_hrp("Color", pd->m_Color, "color");
|
ar & boost::serialization::make_hrp("Color", pd->m_Color, "color");
|
||||||
ar & boost::serialization::make_hrp("Opacity", pd->m_Opacity).range(0.0, 1.0).set_default(1.0);
|
ar & boost::serialization::make_hrp("Opacity", pd->m_Opacity).range(0.0, 1.0).set_default(1.0);
|
||||||
ar & boost::serialization::make_hrp_enum("Representation", pd->m_Representation, {"Points", "Wireframe", "Surface", "SurfaceWithEdges", "Volume", "Outline", "Slice"});
|
ar & boost::serialization::make_hrp_enum("Representation",
|
||||||
|
pd->m_Representation, {"Points", "Wireframe", "Surface", "SurfaceWithEdges", "Volume", "Outline", "Slice"});
|
||||||
ar & boost::serialization::make_hrp("Visibility", pd->m_Visibility);
|
ar & boost::serialization::make_hrp("Visibility", pd->m_Visibility);
|
||||||
ar & boost::serialization::make_hrp("Pickable", pd->m_Selectable);
|
ar & boost::serialization::make_hrp("Pickable", pd->m_Selectable);
|
||||||
ar & boost::serialization::make_hrp("Dragable", pd->m_Dragable);
|
ar & boost::serialization::make_hrp("Dragable", pd->m_Dragable);
|
||||||
|
ar & boost::serialization::make_hrp("ShowBoundingBox", pd->m_ShowBoundingBox);
|
||||||
|
ar & boost::serialization::make_hrp("ShowScaleMeasures", pd->m_ShowScaleMeasures);
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
@@ -643,13 +641,6 @@ void Puppet::serialize_display(Archive::display_properties_archive & ar, const u
|
|||||||
ar & boost::serialization::make_nvp("Transform", transform);
|
ar & boost::serialization::make_nvp("Transform", transform);
|
||||||
}
|
}
|
||||||
|
|
||||||
void Puppet::serialize(Archive::xml_oarchive & ar, const unsigned int v) { }
|
|
||||||
void Puppet::serialize(Archive::xml_iarchive & ar, const unsigned int v) { }
|
|
||||||
void Puppet::serialize(Archive::text_oarchive & ar, const unsigned int v) { }
|
|
||||||
void Puppet::serialize(Archive::text_iarchive & ar, const unsigned int v) { }
|
|
||||||
void Puppet::serialize(Archive::hrt_oarchive & ar, const unsigned int v) { }
|
|
||||||
void Puppet::serialize(Archive::hrt_iarchive & ar, const unsigned int v) { }
|
|
||||||
void Puppet::serialize(Archive::log_archive & ar, const unsigned int v) { }
|
|
||||||
|
|
||||||
} // namespace Vtk
|
} // namespace Vtk
|
||||||
} // namespace uLib
|
} // namespace uLib
|
||||||
|
|||||||
@@ -26,15 +26,15 @@
|
|||||||
#ifndef ULIBVTKINTERFACE_H
|
#ifndef ULIBVTKINTERFACE_H
|
||||||
#define ULIBVTKINTERFACE_H
|
#define ULIBVTKINTERFACE_H
|
||||||
|
|
||||||
|
#include "Core/Monitor.h"
|
||||||
|
#include "Core/Object.h"
|
||||||
|
#include "Core/Property.h"
|
||||||
|
#include <boost/mpl/bool.hpp>
|
||||||
|
#include <boost/serialization/serialization.hpp>
|
||||||
|
#include <boost/type_traits/is_class.hpp>
|
||||||
#include <iomanip>
|
#include <iomanip>
|
||||||
#include <ostream>
|
#include <ostream>
|
||||||
#include <vector>
|
#include <vector>
|
||||||
#include <boost/type_traits/is_class.hpp>
|
|
||||||
#include <boost/mpl/bool.hpp>
|
|
||||||
#include <boost/serialization/serialization.hpp>
|
|
||||||
#include "Core/Object.h"
|
|
||||||
#include "Core/Property.h"
|
|
||||||
#include "Core/Monitor.h"
|
|
||||||
|
|
||||||
// vtk classes forward declaration //
|
// vtk classes forward declaration //
|
||||||
class vtkProp;
|
class vtkProp;
|
||||||
@@ -46,20 +46,27 @@ class vtkRendererCollection;
|
|||||||
class vtkRenderWindowInteractor;
|
class vtkRenderWindowInteractor;
|
||||||
|
|
||||||
namespace uLib {
|
namespace uLib {
|
||||||
namespace Archive { class display_properties_archive; }
|
namespace Archive {
|
||||||
namespace Vtk { class Puppet; class Viewer; }
|
class display_properties_archive;
|
||||||
}
|
}
|
||||||
|
namespace Vtk {
|
||||||
|
class Puppet;
|
||||||
|
class Viewer;
|
||||||
|
} // namespace Vtk
|
||||||
|
} // namespace uLib
|
||||||
|
|
||||||
namespace uLib {
|
namespace uLib {
|
||||||
namespace Vtk {
|
namespace Vtk {
|
||||||
|
|
||||||
class Puppet : public uLib::Object {
|
class Puppet : public uLib::Object {
|
||||||
|
|
||||||
uLibTypeMacro(Puppet, uLib::Object)
|
uLibTypeMacro(Puppet, uLib::Object)
|
||||||
public:
|
|
||||||
Puppet();
|
public : Puppet();
|
||||||
virtual ~Puppet();
|
virtual ~Puppet();
|
||||||
|
|
||||||
virtual vtkProp *GetProp();
|
virtual vtkProp *GetProp();
|
||||||
|
virtual vtkProp3D *GetProxyProp();
|
||||||
|
|
||||||
virtual vtkPropCollection *GetParts();
|
virtual vtkPropCollection *GetParts();
|
||||||
|
|
||||||
@@ -85,10 +92,31 @@ public:
|
|||||||
void SetSelected(bool selected = true);
|
void SetSelected(bool selected = true);
|
||||||
bool IsSelected() const;
|
bool IsSelected() const;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief Synchronizes the VTK representation with the internal state and properties.
|
||||||
|
*
|
||||||
|
* This method should be called whenever the underlying model or display properties
|
||||||
|
* are modified to ensure the visual representation in VTK is consistent.
|
||||||
|
*/
|
||||||
virtual void Update();
|
virtual void Update();
|
||||||
virtual void SyncFromVtk();
|
|
||||||
|
|
||||||
enum Representation { Points = 0, Wireframe = 1, Surface = 2, SurfaceWithEdges = 3, Volume = 4, Outline = 5, Slice = 6 };
|
/**
|
||||||
|
* @brief Synchronizes the internal state and properties from the VTK representation.
|
||||||
|
*
|
||||||
|
* This method should be called when the VTK representation has been modified
|
||||||
|
* (e.g., via a gizmo) and the changes need to be pushed back to the model.
|
||||||
|
*/
|
||||||
|
virtual void SyncFromVtk() {}
|
||||||
|
|
||||||
|
enum Representation {
|
||||||
|
Points = 0,
|
||||||
|
Wireframe = 1,
|
||||||
|
Surface = 2,
|
||||||
|
SurfaceWithEdges = 3,
|
||||||
|
Volume = 4,
|
||||||
|
Outline = 5,
|
||||||
|
Slice = 6
|
||||||
|
};
|
||||||
void SetRepresentation(Representation mode);
|
void SetRepresentation(Representation mode);
|
||||||
void SetRepresentation(const char *mode);
|
void SetRepresentation(const char *mode);
|
||||||
|
|
||||||
@@ -99,18 +127,15 @@ public:
|
|||||||
|
|
||||||
vtkRendererCollection *GetRenderers() const;
|
vtkRendererCollection *GetRenderers() const;
|
||||||
|
|
||||||
virtual void serialize(Archive::xml_oarchive & ar, const unsigned int version) override;
|
const std::vector<uLib::PropertyBase *> &GetDisplayProperties() const {
|
||||||
virtual void serialize(Archive::xml_iarchive & ar, const unsigned int version) override;
|
return m_DisplayProperties;
|
||||||
virtual void serialize(Archive::text_oarchive & ar, const unsigned int version) override;
|
}
|
||||||
virtual void serialize(Archive::text_iarchive & ar, const unsigned int version) override;
|
void RegisterDisplayProperty(uLib::PropertyBase *prop) {
|
||||||
virtual void serialize(Archive::hrt_oarchive & ar, const unsigned int version) override;
|
m_DisplayProperties.push_back(prop);
|
||||||
virtual void serialize(Archive::hrt_iarchive & ar, const unsigned int version) override;
|
}
|
||||||
virtual void serialize(Archive::log_archive & ar, const unsigned int version) override;
|
|
||||||
|
|
||||||
const std::vector<uLib::PropertyBase*>& GetDisplayProperties() const { return m_DisplayProperties; }
|
virtual void serialize_display(uLib::Archive::display_properties_archive &ar,
|
||||||
void RegisterDisplayProperty(uLib::PropertyBase* prop) { m_DisplayProperties.push_back(prop); }
|
const unsigned int version = 0);
|
||||||
|
|
||||||
virtual void serialize_display(uLib::Archive::display_properties_archive & ar, const unsigned int version = 0);
|
|
||||||
|
|
||||||
virtual void ConnectInteractor(class vtkRenderWindowInteractor *interactor);
|
virtual void ConnectInteractor(class vtkRenderWindowInteractor *interactor);
|
||||||
|
|
||||||
@@ -139,15 +164,6 @@ private:
|
|||||||
} // namespace Vtk
|
} // namespace Vtk
|
||||||
} // namespace uLib
|
} // namespace uLib
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
// -------------------------------------------------------------------------- //
|
// -------------------------------------------------------------------------- //
|
||||||
// DISPLAY PROPERTIES SERIALIZE
|
// DISPLAY PROPERTIES SERIALIZE
|
||||||
// -------------------------------------------------------------------------- //
|
// -------------------------------------------------------------------------- //
|
||||||
@@ -156,50 +172,66 @@ namespace uLib {
|
|||||||
namespace Archive {
|
namespace Archive {
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* @brief Specialized archive for registering display-only properties in Puppets.
|
* @brief Specialized archive for registering display-only properties in
|
||||||
|
* Puppets.
|
||||||
*/
|
*/
|
||||||
class display_properties_archive : public boost::archive::detail::common_oarchive<display_properties_archive> {
|
class display_properties_archive
|
||||||
|
: public boost::archive::detail::common_oarchive<
|
||||||
|
display_properties_archive> {
|
||||||
public:
|
public:
|
||||||
display_properties_archive(Vtk::Puppet* puppet) :
|
display_properties_archive(Vtk::Puppet *puppet)
|
||||||
boost::archive::detail::common_oarchive<display_properties_archive>(boost::archive::no_header),
|
: boost::archive::detail::common_oarchive<display_properties_archive>(
|
||||||
|
boost::archive::no_header),
|
||||||
m_Puppet(puppet) {}
|
m_Puppet(puppet) {}
|
||||||
|
|
||||||
std::string GetCurrentGroup() const {
|
std::string GetCurrentGroup() const {
|
||||||
std::string group;
|
std::string group;
|
||||||
for (const auto &g : m_GroupStack) {
|
for (const auto &g : m_GroupStack) {
|
||||||
if (!group.empty()) group += ".";
|
if (!group.empty())
|
||||||
|
group += ".";
|
||||||
group += g;
|
group += g;
|
||||||
}
|
}
|
||||||
return group;
|
return group;
|
||||||
}
|
}
|
||||||
|
|
||||||
template<class T>
|
template <class T> void save_override(const boost::serialization::hrp<T> &t) {
|
||||||
void save_override(const boost::serialization::hrp<T> &t) {
|
|
||||||
if (m_Puppet) {
|
if (m_Puppet) {
|
||||||
uLib::Property<T>* p = new uLib::Property<T>(m_Puppet, t.name(), &const_cast<boost::serialization::hrp<T>&>(t).value(), t.units() ? t.units() : "", GetCurrentGroup());
|
uLib::Property<T> *p = new uLib::Property<T>(
|
||||||
if (t.has_range()) p->SetRange(t.min_val(), t.max_val());
|
m_Puppet, t.name(),
|
||||||
if (t.has_default()) p->SetDefault(t.default_val());
|
&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());
|
||||||
|
|
||||||
m_Puppet->RegisterDisplayProperty(p);
|
m_Puppet->RegisterDisplayProperty(p);
|
||||||
Vtk::Puppet *puppet = m_Puppet;
|
Vtk::Puppet *puppet = m_Puppet;
|
||||||
uLib::Object::connect(p, &uLib::PropertyBase::Updated, [puppet](){ puppet->Update(); });
|
uLib::Object::connect(p, &uLib::PropertyBase::Updated,
|
||||||
|
[puppet]() { puppet->Update(); });
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
template <class T>
|
template <class T>
|
||||||
void save_override(const boost::serialization::hrp_enum<T> &t) {
|
void save_override(const boost::serialization::hrp_enum<T> &t) {
|
||||||
if (m_Puppet) {
|
if (m_Puppet) {
|
||||||
uLib::EnumProperty* p = new uLib::EnumProperty(m_Puppet, t.name(), (int*)&const_cast<boost::serialization::hrp_enum<T>&>(t).value(), t.labels(), t.units() ? t.units() : "", GetCurrentGroup());
|
uLib::EnumProperty *p = new uLib::EnumProperty(
|
||||||
|
m_Puppet, t.name(),
|
||||||
|
(int *)&const_cast<boost::serialization::hrp_enum<T> &>(t).value(),
|
||||||
|
t.labels(), t.units() ? t.units() : "", GetCurrentGroup());
|
||||||
m_Puppet->RegisterDisplayProperty(p);
|
m_Puppet->RegisterDisplayProperty(p);
|
||||||
Vtk::Puppet *puppet = m_Puppet;
|
Vtk::Puppet *puppet = m_Puppet;
|
||||||
uLib::Object::connect(p, &uLib::PropertyBase::Updated, [puppet](){ puppet->Update(); });
|
uLib::Object::connect(p, &uLib::PropertyBase::Updated,
|
||||||
|
[puppet]() { puppet->Update(); });
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
template <class T> void save_override(const boost::serialization::nvp<T> &t) {
|
template <class T> void save_override(const boost::serialization::nvp<T> &t) {
|
||||||
if (t.name()) m_GroupStack.push_back(t.name());
|
if (t.name())
|
||||||
|
m_GroupStack.push_back(t.name());
|
||||||
this->save_helper(t.const_value(), typename boost::is_class<T>::type());
|
this->save_helper(t.const_value(), typename boost::is_class<T>::type());
|
||||||
if (t.name()) m_GroupStack.pop_back();
|
if (t.name())
|
||||||
|
m_GroupStack.pop_back();
|
||||||
}
|
}
|
||||||
|
|
||||||
// Recursion for nested classes, ignore primitives
|
// Recursion for nested classes, ignore primitives
|
||||||
@@ -207,13 +239,11 @@ public:
|
|||||||
this->save_helper(t, typename boost::is_class<T>::type());
|
this->save_helper(t, typename boost::is_class<T>::type());
|
||||||
}
|
}
|
||||||
|
|
||||||
template<class T>
|
template <class T> void save_helper(const T &t, boost::mpl::true_) {
|
||||||
void save_helper(const T &t, boost::mpl::true_) {
|
|
||||||
boost::serialization::serialize_adl(*this, const_cast<T &>(t), 0);
|
boost::serialization::serialize_adl(*this, const_cast<T &>(t), 0);
|
||||||
}
|
}
|
||||||
|
|
||||||
template<class T>
|
template <class T> void save_helper(const T &t, boost::mpl::false_) {}
|
||||||
void save_helper(const T &t, boost::mpl::false_) {}
|
|
||||||
|
|
||||||
void save_override(const boost::archive::object_id_type &t) {}
|
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::object_reference_type &t) {}
|
||||||
@@ -231,7 +261,8 @@ private:
|
|||||||
|
|
||||||
} // namespace Archive
|
} // namespace Archive
|
||||||
|
|
||||||
// This macro MUST be defined after both Puppet and display_properties_archive are fully defined.
|
// This macro MUST be defined after both Puppet and display_properties_archive
|
||||||
|
// are fully defined.
|
||||||
#define ULIB_ACTIVATE_DISPLAY_PROPERTIES \
|
#define ULIB_ACTIVATE_DISPLAY_PROPERTIES \
|
||||||
{ \
|
{ \
|
||||||
uLib::Archive::display_properties_archive dar(this); \
|
uLib::Archive::display_properties_archive dar(this); \
|
||||||
|
|||||||
@@ -72,6 +72,7 @@ struct ViewerData {
|
|||||||
vtkRenderWindow *m_RenderWindow;
|
vtkRenderWindow *m_RenderWindow;
|
||||||
vtkSmartPointer<vtkRenderWindowInteractor> m_Interactor;
|
vtkSmartPointer<vtkRenderWindowInteractor> m_Interactor;
|
||||||
vtkSmartPointer<vtkButtonWidget> m_GridButton;
|
vtkSmartPointer<vtkButtonWidget> m_GridButton;
|
||||||
|
vtkSmartPointer<vtkButtonWidget> m_ProjButton;
|
||||||
|
|
||||||
ViewerData() : m_RenderWindow(vtkRenderWindow::New()) {}
|
ViewerData() : m_RenderWindow(vtkRenderWindow::New()) {}
|
||||||
~ViewerData() {
|
~ViewerData() {
|
||||||
@@ -97,6 +98,11 @@ Viewer::~Viewer() {
|
|||||||
dv->m_GridButton->SetInteractor(nullptr);
|
dv->m_GridButton->SetInteractor(nullptr);
|
||||||
dv->m_GridButton = nullptr;
|
dv->m_GridButton = nullptr;
|
||||||
}
|
}
|
||||||
|
if (dv->m_ProjButton) {
|
||||||
|
dv->m_ProjButton->Off();
|
||||||
|
dv->m_ProjButton->SetInteractor(nullptr);
|
||||||
|
dv->m_ProjButton = nullptr;
|
||||||
|
}
|
||||||
if (this->GetRenderWindow()) {
|
if (this->GetRenderWindow()) {
|
||||||
this->GetRenderWindow()->RemoveAllObservers();
|
this->GetRenderWindow()->RemoveAllObservers();
|
||||||
}
|
}
|
||||||
@@ -123,6 +129,7 @@ void Viewer::InstallPipe() {
|
|||||||
// Setup native grid button
|
// Setup native grid button
|
||||||
if (!std::getenv("CTEST_PROJECT_NAME")) {
|
if (!std::getenv("CTEST_PROJECT_NAME")) {
|
||||||
SetupGridButton();
|
SetupGridButton();
|
||||||
|
SetupProjButton();
|
||||||
}
|
}
|
||||||
|
|
||||||
// BUT we want to override the style with our custom NoSpin version
|
// BUT we want to override the style with our custom NoSpin version
|
||||||
@@ -238,6 +245,88 @@ void Viewer::UpdateGridButtonPosition() {
|
|||||||
rep->PlaceWidget(bds);
|
rep->PlaceWidget(bds);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
void Viewer::SetupProjButton() {
|
||||||
|
if (!dv->m_RenderWindow || !dv->m_RenderWindow->GetInteractor()) return;
|
||||||
|
|
||||||
|
vtkNew<vtkImageCanvasSource2D> canvas;
|
||||||
|
canvas->SetScalarTypeToUnsignedChar();
|
||||||
|
canvas->SetNumberOfScalarComponents(4);
|
||||||
|
canvas->SetExtent(0, 63, 0, 63, 0, 0);
|
||||||
|
|
||||||
|
// State 0: Perspective (gray trapezoid-like lines)
|
||||||
|
canvas->SetDrawColor(0, 0, 0, 0);
|
||||||
|
canvas->FillBox(0, 63, 0, 63);
|
||||||
|
canvas->SetDrawColor(120, 120, 120, 255);
|
||||||
|
canvas->DrawSegment(16, 16, 48, 16);
|
||||||
|
canvas->DrawSegment(48, 16, 56, 48);
|
||||||
|
canvas->DrawSegment(56, 48, 8, 48);
|
||||||
|
canvas->DrawSegment(8, 48, 16, 16);
|
||||||
|
canvas->Update();
|
||||||
|
|
||||||
|
vtkNew<vtkImageData> imgPersp;
|
||||||
|
imgPersp->DeepCopy(canvas->GetOutput());
|
||||||
|
|
||||||
|
// State 1: Orthographic (white rectangle)
|
||||||
|
canvas->SetDrawColor(0, 0, 0, 0);
|
||||||
|
canvas->FillBox(0, 63, 0, 63);
|
||||||
|
canvas->SetDrawColor(255, 255, 255, 255);
|
||||||
|
canvas->DrawSegment(12, 16, 52, 16);
|
||||||
|
canvas->DrawSegment(52, 16, 52, 48);
|
||||||
|
canvas->DrawSegment(52, 48, 12, 48);
|
||||||
|
canvas->DrawSegment(12, 48, 12, 16);
|
||||||
|
canvas->Update();
|
||||||
|
|
||||||
|
vtkNew<vtkImageData> imgOrtho;
|
||||||
|
imgOrtho->DeepCopy(canvas->GetOutput());
|
||||||
|
|
||||||
|
vtkNew<vtkTexturedButtonRepresentation2D> rep;
|
||||||
|
rep->SetNumberOfStates(2);
|
||||||
|
rep->SetButtonTexture(0, imgPersp);
|
||||||
|
rep->SetButtonTexture(1, imgOrtho);
|
||||||
|
|
||||||
|
dv->m_ProjButton = vtkSmartPointer<vtkButtonWidget>::New();
|
||||||
|
dv->m_ProjButton->SetInteractor(dv->m_RenderWindow->GetInteractor());
|
||||||
|
dv->m_ProjButton->SetRepresentation(rep);
|
||||||
|
|
||||||
|
UpdateProjButtonPosition();
|
||||||
|
|
||||||
|
vtkNew<vtkCallbackCommand> resizeCallback;
|
||||||
|
resizeCallback->SetClientData(this);
|
||||||
|
resizeCallback->SetCallback([](vtkObject*, unsigned long, void* clientdata, void*){
|
||||||
|
static_cast<Viewer*>(clientdata)->UpdateProjButtonPosition();
|
||||||
|
});
|
||||||
|
dv->m_RenderWindow->AddObserver(vtkCommand::ModifiedEvent, resizeCallback);
|
||||||
|
|
||||||
|
vtkNew<vtkCallbackCommand> stateCallback;
|
||||||
|
stateCallback->SetClientData(this);
|
||||||
|
stateCallback->SetCallback([](vtkObject* caller, unsigned long, void* clientdata, void*){
|
||||||
|
auto* btn = vtkButtonWidget::SafeDownCast(caller);
|
||||||
|
auto* v = static_cast<Viewer*>(clientdata);
|
||||||
|
auto* r = vtkTexturedButtonRepresentation2D::SafeDownCast(btn->GetRepresentation());
|
||||||
|
v->SetParallelProjection(r->GetState() == 1);
|
||||||
|
});
|
||||||
|
dv->m_ProjButton->AddObserver(vtkCommand::StateChangedEvent, stateCallback);
|
||||||
|
dv->m_ProjButton->On();
|
||||||
|
|
||||||
|
rep->SetState(GetParallelProjection() ? 1 : 0);
|
||||||
|
}
|
||||||
|
|
||||||
|
void Viewer::UpdateProjButtonPosition() {
|
||||||
|
if (!dv->m_ProjButton || !dv->m_RenderWindow) return;
|
||||||
|
auto* rep = vtkTexturedButtonRepresentation2D::SafeDownCast(dv->m_ProjButton->GetRepresentation());
|
||||||
|
if (!rep) return;
|
||||||
|
|
||||||
|
int *sz = dv->m_RenderWindow->GetSize();
|
||||||
|
if (sz[0] == 0 || sz[1] == 0) return;
|
||||||
|
|
||||||
|
int margin_right = 23;
|
||||||
|
int margin_top = 220; // below the grid button (170 + 50)
|
||||||
|
int btnSz = 100;
|
||||||
|
double bds[6] = { (double)sz[0] - btnSz - margin_right, (double)sz[0] - margin_right,
|
||||||
|
(double)sz[1] - margin_top - btnSz/2.0, (double)sz[1] - margin_top + btnSz/2.0, 0, 0 };
|
||||||
|
rep->PlaceWidget(bds);
|
||||||
|
}
|
||||||
|
|
||||||
void Viewer::Start() {
|
void Viewer::Start() {
|
||||||
if (std::getenv("CTEST_PROJECT_NAME")) return;
|
if (std::getenv("CTEST_PROJECT_NAME")) return;
|
||||||
dv->m_RenderWindow->GetInteractor()->Start();
|
dv->m_RenderWindow->GetInteractor()->Start();
|
||||||
|
|||||||
@@ -38,6 +38,9 @@ private:
|
|||||||
void SetupGridButton();
|
void SetupGridButton();
|
||||||
void UpdateGridButtonPosition();
|
void UpdateGridButtonPosition();
|
||||||
|
|
||||||
|
void SetupProjButton();
|
||||||
|
void UpdateProjButtonPosition();
|
||||||
|
|
||||||
struct ViewerData *dv;
|
struct ViewerData *dv;
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|||||||
@@ -547,16 +547,14 @@ void vtkHandlerWidget::OnMouseMove() {
|
|||||||
total->Concatenate(op);
|
total->Concatenate(op);
|
||||||
|
|
||||||
if (this->Prop3D) {
|
if (this->Prop3D) {
|
||||||
double p[3], r[3], s[3];
|
vtkNew<vtkMatrix4x4> result;
|
||||||
total->GetPosition(p);
|
total->GetMatrix(result);
|
||||||
total->GetOrientation(r);
|
this->Prop3D->SetUserMatrix(result);
|
||||||
total->GetScale(s);
|
|
||||||
this->Prop3D->SetPosition(p);
|
// Reset individual TRS components so UserMatrix is the single source of truth
|
||||||
// VTK GetOrientation already returned degrees, so r is in degrees.
|
this->Prop3D->SetPosition(0, 0, 0);
|
||||||
// We apply it directly back to VTK.
|
this->Prop3D->SetOrientation(0, 0, 0);
|
||||||
this->Prop3D->SetOrientation(r);
|
this->Prop3D->SetScale(1, 1, 1);
|
||||||
this->Prop3D->SetScale(s);
|
|
||||||
this->Prop3D->SetUserMatrix(nullptr);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
this->Prop3D->Modified();
|
this->Prop3D->Modified();
|
||||||
|
|||||||
@@ -3,7 +3,9 @@
|
|||||||
#include "Vtk/Math/vtkCylinder.h"
|
#include "Vtk/Math/vtkCylinder.h"
|
||||||
#include "Vtk/Math/vtkAssembly.h"
|
#include "Vtk/Math/vtkAssembly.h"
|
||||||
#include "Vtk/Math/vtkVoxImage.h"
|
#include "Vtk/Math/vtkVoxImage.h"
|
||||||
|
|
||||||
#include "HEP/Detectors/vtkDetectorChamber.h"
|
#include "HEP/Detectors/vtkDetectorChamber.h"
|
||||||
|
#include "HEP/Geant/vtkBoxSolid.h"
|
||||||
|
|
||||||
#include <vtkAssembly.h>
|
#include <vtkAssembly.h>
|
||||||
#include <vtkPropCollection.h>
|
#include <vtkPropCollection.h>
|
||||||
@@ -107,19 +109,27 @@ void vtkObjectsContext::Update() {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
void vtkObjectsContext::SyncFromVtk() {
|
||||||
|
for (auto const& [obj, puppet] : m_Puppets) {
|
||||||
|
puppet->SyncFromVtk();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
Puppet* vtkObjectsContext::CreatePuppet(uLib::Object* obj) {
|
Puppet* vtkObjectsContext::CreatePuppet(uLib::Object* obj) {
|
||||||
if (!obj) return nullptr;
|
if (!obj) return nullptr;
|
||||||
|
|
||||||
if (auto* box = dynamic_cast<uLib::ContainerBox*>(obj)) {
|
if (auto* vox = dynamic_cast<uLib::Abstract::VoxImage*>(obj)) {
|
||||||
|
return new vtkVoxImage(*vox);
|
||||||
|
} else if (auto* box = dynamic_cast<uLib::ContainerBox*>(obj)) {
|
||||||
return new vtkContainerBox(box);
|
return new vtkContainerBox(box);
|
||||||
} else if (auto* chamber = dynamic_cast<uLib::DetectorChamber*>(obj)) {
|
} else if (auto* chamber = dynamic_cast<uLib::DetectorChamber*>(obj)) {
|
||||||
return new vtkDetectorChamber(chamber);
|
return new vtkDetectorChamber(chamber);
|
||||||
} else if (auto* cylinder = dynamic_cast<uLib::Cylinder*>(obj)) {
|
} else if (auto* cylinder = dynamic_cast<uLib::Cylinder*>(obj)) {
|
||||||
return new vtkCylinder(cylinder);
|
return new vtkCylinder(cylinder);
|
||||||
} else if (auto* vox = dynamic_cast<uLib::Abstract::VoxImage*>(obj)) {
|
|
||||||
return new vtkVoxImage(*vox);
|
|
||||||
} else if (auto* assembly = dynamic_cast<uLib::Assembly*>(obj)) {
|
} else if (auto* assembly = dynamic_cast<uLib::Assembly*>(obj)) {
|
||||||
return new Assembly(assembly);
|
return new Assembly(assembly);
|
||||||
|
} else if (auto* box = dynamic_cast<uLib::Geant::BoxSolid*>(obj)) {
|
||||||
|
return new vtkBoxSolid(box);
|
||||||
}
|
}
|
||||||
|
|
||||||
// Fallback if we don't know the exact class but it might be a context itself
|
// Fallback if we don't know the exact class but it might be a context itself
|
||||||
|
|||||||
@@ -15,7 +15,7 @@ namespace Vtk {
|
|||||||
*/
|
*/
|
||||||
class vtkObjectsContext : public Puppet {
|
class vtkObjectsContext : public Puppet {
|
||||||
public:
|
public:
|
||||||
virtual const char* GetClassName() const override { return "vtkObjectsContext"; }
|
uLibTypeMacro(vtkObjectsContext, Puppet)
|
||||||
vtkObjectsContext(uLib::ObjectsContext *context);
|
vtkObjectsContext(uLib::ObjectsContext *context);
|
||||||
virtual ~vtkObjectsContext();
|
virtual ~vtkObjectsContext();
|
||||||
|
|
||||||
@@ -30,6 +30,9 @@ public:
|
|||||||
/** @brief Updates all managed puppets. */
|
/** @brief Updates all managed puppets. */
|
||||||
virtual void Update() override;
|
virtual void Update() override;
|
||||||
|
|
||||||
|
/** @brief Synchronizes all managed puppets back to their models. */
|
||||||
|
virtual void SyncFromVtk() override;
|
||||||
|
|
||||||
public:
|
public:
|
||||||
virtual void PuppetAdded(Puppet* puppet);
|
virtual void PuppetAdded(Puppet* puppet);
|
||||||
virtual void PuppetRemoved(Puppet* puppet);
|
virtual void PuppetRemoved(Puppet* puppet);
|
||||||
|
|||||||
@@ -19,6 +19,7 @@ QViewport::QViewport(QWidget* parent)
|
|||||||
, Viewport()
|
, Viewport()
|
||||||
, m_VtkWidget(nullptr)
|
, m_VtkWidget(nullptr)
|
||||||
, m_GridButton(nullptr)
|
, m_GridButton(nullptr)
|
||||||
|
, m_ProjButton(nullptr)
|
||||||
{
|
{
|
||||||
// Build the layout – zero margins so VTK fills the entire widget
|
// Build the layout – zero margins so VTK fills the entire widget
|
||||||
auto* layout = new QVBoxLayout(this);
|
auto* layout = new QVBoxLayout(this);
|
||||||
@@ -58,6 +59,36 @@ QViewport::QViewport(QWidget* parent)
|
|||||||
m_GridButton->setChecked(true); // Grid is on by default
|
m_GridButton->setChecked(true); // Grid is on by default
|
||||||
connect(m_GridButton, &QPushButton::clicked, this, &QViewport::onGridButtonClicked);
|
connect(m_GridButton, &QPushButton::clicked, this, &QViewport::onGridButtonClicked);
|
||||||
|
|
||||||
|
// Projection Toggle Button (below grid button)
|
||||||
|
m_ProjButton = new QPushButton(m_VtkWidget);
|
||||||
|
m_ProjButton->setText("P");
|
||||||
|
m_ProjButton->setFixedSize(40, 40);
|
||||||
|
m_ProjButton->setToolTip("Toggle Perspective / Orthographic");
|
||||||
|
m_ProjButton->setStyleSheet(
|
||||||
|
"QPushButton {"
|
||||||
|
" border-radius: 20px;"
|
||||||
|
" background-color: rgba(40, 40, 40, 180);"
|
||||||
|
" color: white;"
|
||||||
|
" font-size: 22px;"
|
||||||
|
" border: 1.5px solid rgba(255, 255, 255, 60);"
|
||||||
|
"}"
|
||||||
|
"QPushButton:hover {"
|
||||||
|
" background-color: rgba(70, 70, 70, 200);"
|
||||||
|
" border: 1.5px solid rgba(255, 255, 255, 100);"
|
||||||
|
"}"
|
||||||
|
"QPushButton:checked {"
|
||||||
|
" background-color: rgba(0, 120, 215, 200);"
|
||||||
|
" color: white;"
|
||||||
|
" border: 1.5px solid rgba(255, 255, 255, 120);"
|
||||||
|
"}"
|
||||||
|
"QPushButton:pressed {"
|
||||||
|
" background-color: rgba(0, 90, 160, 220);"
|
||||||
|
"}"
|
||||||
|
);
|
||||||
|
m_ProjButton->setCheckable(true);
|
||||||
|
m_ProjButton->setChecked(false); // Perspective by default
|
||||||
|
connect(m_ProjButton, &QPushButton::clicked, this, &QViewport::onProjButtonClicked);
|
||||||
|
|
||||||
// After the Qt widget exists but before the first paint,
|
// After the Qt widget exists but before the first paint,
|
||||||
// attach the renderer and configure the pipeline.
|
// attach the renderer and configure the pipeline.
|
||||||
SetupPipeline();
|
SetupPipeline();
|
||||||
@@ -81,7 +112,6 @@ void QViewport::SetupPipeline()
|
|||||||
|
|
||||||
void QViewport::Render()
|
void QViewport::Render()
|
||||||
{
|
{
|
||||||
UpdateGrid();
|
|
||||||
if (m_VtkWidget && m_VtkWidget->renderWindow())
|
if (m_VtkWidget && m_VtkWidget->renderWindow())
|
||||||
m_VtkWidget->renderWindow()->Render();
|
m_VtkWidget->renderWindow()->Render();
|
||||||
}
|
}
|
||||||
@@ -101,6 +131,11 @@ void QViewport::onGridButtonClicked()
|
|||||||
SetGridVisible(m_GridButton->isChecked());
|
SetGridVisible(m_GridButton->isChecked());
|
||||||
}
|
}
|
||||||
|
|
||||||
|
void QViewport::onProjButtonClicked()
|
||||||
|
{
|
||||||
|
SetParallelProjection(m_ProjButton->isChecked());
|
||||||
|
}
|
||||||
|
|
||||||
void QViewport::OnSelectionChanged(Puppet* p)
|
void QViewport::OnSelectionChanged(Puppet* p)
|
||||||
{
|
{
|
||||||
emit puppetSelected(p);
|
emit puppetSelected(p);
|
||||||
@@ -116,6 +151,11 @@ void QViewport::resizeEvent(QResizeEvent* event)
|
|||||||
int y = 160;
|
int y = 160;
|
||||||
m_GridButton->move(x, y);
|
m_GridButton->move(x, y);
|
||||||
}
|
}
|
||||||
|
if (m_ProjButton) {
|
||||||
|
int x = width() - m_ProjButton->width() - 10;
|
||||||
|
int y = 210;
|
||||||
|
m_ProjButton->move(x, y);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
} // namespace Vtk
|
} // namespace Vtk
|
||||||
|
|||||||
@@ -51,12 +51,14 @@ protected:
|
|||||||
|
|
||||||
private slots:
|
private slots:
|
||||||
void onGridButtonClicked();
|
void onGridButtonClicked();
|
||||||
|
void onProjButtonClicked();
|
||||||
|
|
||||||
private:
|
private:
|
||||||
void SetupPipeline();
|
void SetupPipeline();
|
||||||
|
|
||||||
QVTKOpenGLNativeWidget* m_VtkWidget;
|
QVTKOpenGLNativeWidget* m_VtkWidget;
|
||||||
QPushButton* m_GridButton;
|
QPushButton* m_GridButton;
|
||||||
|
QPushButton* m_ProjButton;
|
||||||
};
|
};
|
||||||
|
|
||||||
} // namespace Vtk
|
} // namespace Vtk
|
||||||
|
|||||||
@@ -505,11 +505,11 @@ void Viewport::SelectPuppet(Puppet* prop)
|
|||||||
|
|
||||||
if (pv->m_HandlerWidget) {
|
if (pv->m_HandlerWidget) {
|
||||||
if (prop) {
|
if (prop) {
|
||||||
vtkProp3D* prop3d = vtkProp3D::SafeDownCast(prop->GetProp());
|
vtkProp3D* prop3d = prop->GetProxyProp();
|
||||||
if (prop3d) {
|
if (prop3d) {
|
||||||
pv->m_HandlerWidget->SetProp3D(prop3d);
|
pv->m_HandlerWidget->SetProp3D(prop3d);
|
||||||
pv->m_HandlerWidget->SetEnabled(1);
|
pv->m_HandlerWidget->SetEnabled(1);
|
||||||
pv->m_HandlerWidget->PlaceWidget(prop3d->GetBounds());
|
pv->m_HandlerWidget->PlaceWidget(prop3d->GetBounds()); //TODO: FIX !
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
pv->m_HandlerWidget->SetEnabled(0);
|
pv->m_HandlerWidget->SetEnabled(0);
|
||||||
@@ -537,6 +537,21 @@ bool Viewport::GetGridVisible() const
|
|||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
void Viewport::SetParallelProjection(bool parallel)
|
||||||
|
{
|
||||||
|
if (pv->m_Renderer && pv->m_Renderer->GetActiveCamera()) {
|
||||||
|
pv->m_Renderer->GetActiveCamera()->SetParallelProjection(parallel ? 1 : 0);
|
||||||
|
Render();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
bool Viewport::GetParallelProjection() const
|
||||||
|
{
|
||||||
|
if (pv->m_Renderer && pv->m_Renderer->GetActiveCamera())
|
||||||
|
return pv->m_Renderer->GetActiveCamera()->GetParallelProjection() != 0;
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
void Viewport::SetGridAxis(Axis axis)
|
void Viewport::SetGridAxis(Axis axis)
|
||||||
{
|
{
|
||||||
m_GridAxis = axis;
|
m_GridAxis = axis;
|
||||||
|
|||||||
@@ -73,6 +73,10 @@ public:
|
|||||||
void SetGridAxis(Axis axis);
|
void SetGridAxis(Axis axis);
|
||||||
Axis GetGridAxis() const { return m_GridAxis; }
|
Axis GetGridAxis() const { return m_GridAxis; }
|
||||||
|
|
||||||
|
// Projection control
|
||||||
|
void SetParallelProjection(bool parallel);
|
||||||
|
bool GetParallelProjection() const;
|
||||||
|
|
||||||
protected:
|
protected:
|
||||||
void SetupPipeline(vtkRenderWindowInteractor* iren);
|
void SetupPipeline(vtkRenderWindowInteractor* iren);
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user