 |
ImFusion SDK 4.3
|
|
ImFusion SDK 4.3
|
#include <ImFusion/CT/MLOperations/ConeBeamSimulationOperation.h>
ML operation for simulating X-ray projections from CT volumes. More...
Inheritance diagram for ConeBeamSimulationOperation:ML operation for simulating X-ray projections from CT volumes.
Forward projects CT volumes to create DRRs (digitally reconstructed radiographs). Supports label volume projection and configurable cone-beam geometry. For each geometry, a set of frames representing a cone-beam geometry is created. The entire geometry is typically centered around the center of the CT volume and aligned with it, but this can be adjusted using the offset parameters.
Public Member Functions | |
| void | process (ML::DataItem &item) override |
| Main function that processes a DataItem in-place, may throw OperationException The data to be processed is determined by the active fields (if set) and the processing policy otherwise (if set), otherwise all fields are processed. | |
| Public Member Functions inherited from Operation | |
| Operation (std::string name, ProcessingPolicy processingPolicy) | |
| Derived classes must specify a name and whether, by default, the operation should also be applied to label maps. | |
| virtual std::shared_ptr< SharedImageSet > | process (std::shared_ptr< SharedImageSet > input) |
| Utility function to directly apply an operation on a SharedImageSet irrespective of active fields and processing policy. | |
| virtual std::shared_ptr< KeypointSet > | process (std::shared_ptr< KeypointSet > input) |
| Utility function to directly apply an operation on a set of KeypointSet irrespective of active fields and processing policy. | |
| virtual std::shared_ptr< BoundingBoxSet > | process (std::shared_ptr< BoundingBoxSet > input) |
| Utility function to directly apply an operation on a set of BoundingBoxSet irrespective of active fields and processing policy. | |
| virtual std::unique_ptr< SharedImageSet > | process (std::unique_ptr< SharedImageSet > input) final |
| Utility function with unique_ptr as input and output to avoid breaking changes (must not be overridden) | |
| void | addTemporaryPreProcessHook (PreProcessHook hook) |
| Add a temporary hook that will only be active for the next process call and then automatically removed after the DataItem is processed. | |
| void | addTemporaryPostProcessHook (PostProcessHook hook) |
| Add a temporary hook that will only be active for the next process call and then automatically removed after the DataItem is processed. | |
| virtual bool | checkRequiredItemsTypes (const DataItem &item) const |
| Checks if the data item holds fields with types required by the operation. | |
| const std::string & | name () const |
| Returns the name of the operation. | |
| virtual bool | configure (const Properties &properties) |
| Configure an Operation with the given properties and return whether the configuration was successful. | |
| virtual Properties | configuration () const |
| Return the Operation configuration. | |
| void | setActiveFields (std::optional< std::unordered_set< std::string > > activeFields) |
| std::optional< std::unordered_set< std::string > > | activeFields () const |
| virtual void | setProcessingPolicy (ProcessingPolicy policy) |
| ProcessingPolicy | processingPolicy () const |
| std::optional< uint32_t > | seed () const |
| Return the current seed (if there is one) | |
| virtual void | seedRandomEngine (uint32_t seed) |
| Set a seed for the random generator. | |
| ML::ComputingDevice | computingDevice () const |
| Get the computing device. | |
| virtual void | setComputingDevice (ML::ComputingDevice device) |
| Set the computing device selection strategy. | |
| void | configFailed (const std::string &missingParam) const |
| Helper function to show an error message due to a bad configuration. | |
| void | logDeprecatedParam (const std::string &oldName, const std::string &newName) const |
| Helper function to print a warning because of a deprecated parameter has been specified. | |
| bool | errorOnUnexpectedBehaviour () const |
| Get the policy whether to treat unexpected behavior warnings as errors. | |
| virtual void | setErrorOnUnexpectedBehaviour (bool error) |
| Treat unexpected behavior warnings as errors. | |
| void | registerParameter (ParameterBase *param) |
| Register a parameter so that the operation knows about it, and can automatically configure it (unless it has been marked as manually configured parameter via the Operation::setManuallyConfiguredParameters method). | |
| std::vector< ParameterBase * > | parameters () const |
| Return the list of registered parameters. | |
| virtual bool | doesNotModifyInput () const |
| Returns whether the operation is guaranteed to not modify its input element, either because it computes its output out-of-place or because it is a no-op. | |
| virtual bool | supportsInversion () const |
Whether the operation supports inversion. Returns False unless the operation derives from InvertibleOperation. | |
| const std::string & | recordIdentifier () const |
| Get the operation's record identifier. | |
| virtual void | setRecordIdentifier (const std::string &recordIdentifier) |
| Set the operation's record identifier. | |
| const std::string & | logDomain () const |
| Log domain for AdvancedParameter. | |
| std::optional< std::unordered_set< std::string > > | selectedFields (const DataItem &item) |
| Get the fields on which the operation will be applied. | |
Public Attributes | |
| ML::OpParam< std::vector< int > > | p_labelValues = {"label_values", std::vector<int>{1}, this, ML::ParamRequired::No} |
| Label values to include in the output label image. | |
| ML::OpParam< double > | p_detSizeY = {"det_size_y", 500.0, this, ML::ParamRequired::No, "detSizeY"} |
| Detector size in Y. | |
| ML::OpParam< double > | p_detSizeX = {"det_size_x", 500.0, this, ML::ParamRequired::No, "detSizeX"} |
| Detector size in X. | |
| ML::OpParam< bool > | p_detSizeYIsRelative = {"detSizeY_is_relative", false, this, ML::ParamRequired::No} |
| Whether the detector size in Y is relative to the detector size in X. | |
| ML::OpParam< vec3 > | p_offsetTrans = {"offset_trans", vec3::Zero(), this, ML::ParamRequired::No} |
| Translational offset of the geometry in x,y,z relative to the extent of the volume. | |
| ML::OpParam< vec3 > | p_offsetRot = {"offset_rot", vec3::Zero(), this, ML::ParamRequired::No} |
| Rotational offset of the geometry (euler angles in degrees). | |
| ML::OpParam< double > | p_detRotation = {"det_rotation", 0.0, this, ML::ParamRequired::No} |
| Rotation of the detector (in degrees in 2D on the detector), i.e. this corresponds to a rotation around the viewing axis for each frame separately. | |
| ML::OpParam< int > | p_simulationPreset = {"simulation_preset", 4, this, ML::ParamRequired::No} |
| Simulation preset to use from the ConeBeamSimulation class. | |
| ML::OpParam< int > | p_numFrames = {"num_frames", 1, this, ML::ParamRequired::No} |
| Number of images to simulate. | |
| ML::OpParam< int > | p_width = {"width", 512, this, ML::ParamRequired::No} |
| Output image width in pixels. | |
| ML::OpParam< int > | p_height = {"height", 512, this, ML::ParamRequired::No} |
| Output image height in pixels. | |
| ML::OpParam< bool > | p_replace = {"replace", true, this, ML::ParamRequired::No} |
| Whether to replace the input DataElements with the output. | |
| ML::OpParam< bool > | p_logConvertedProjections = {"log_converted_projections", true, this, ML::ParamRequired::No} |
| If true, the output DRRs have log-converted attenuation values. | |
| ML::OpParam< std::string > | p_directionFieldName = {"direction_field_name", "", this, ML::ParamRequired::No} |
| If not empty, the field of this name will be treated as a "direction" output, i.e. | |
| ML::OpParam< std::string > | p_keypointsFieldName = {"keypoints_field_name", "", this, ML::ParamRequired::No} |
| If not empty, keypoint field of this name will be treated as an input around which the output images will be centered. | |
Additional Inherited Members | |
| Public Types inherited from Operation | |
| enum | ProcessingPolicy { EverythingExceptLabels = 0 , Everything , OnlyLabels } |
| Policy used by default when selecting on the fields on which the operation will be applied Note that "Labels" here refers to the target tag, not to be confused with Data::Modality::LABEL. More... | |
| using | PreProcessHook = std::function<void(Operation&, DataElement*)> |
| using | PostProcessHook = std::function<void(Operation&, DataElement*)> |
| Static Public Member Functions inherited from Operation | |
| static std::string | processingPolicyToString (const ProcessingPolicy policy) |
| static ProcessingPolicy | stringToProcessingPolicy (const std::string &s) |
| static std::unique_ptr< Operation > | createFromFactories (const Operation::Specs &specs) |
| Helper function to create an operation from any factory (both C++ and Python factories are tested) | |
| Protected Member Functions inherited from Operation | |
| virtual std::shared_ptr< SharedImageSet > | processImages (std::shared_ptr< SharedImageSet > input) const |
| Virtual function that every subclass should override. | |
| virtual std::shared_ptr< BoundingBoxSet > | processBoxes (std::shared_ptr< BoundingBoxSet > input) const |
| Virtual function that every subclass should override. | |
| virtual std::shared_ptr< KeypointSet > | processPoints (std::shared_ptr< KeypointSet > input) const |
| Virtual function that every subclass should override. | |
| virtual std::shared_ptr< SharedImageSet > | processVectors (std::shared_ptr< SharedImageSet > input) const |
| Virtual function that every subclass should override. | |
| virtual std::shared_ptr< TensorSet > | processTensors (std::shared_ptr< TensorSet > input) const |
| void | throwOperationError (const std::string &msg) const |
| Helper function to throw an exception due to a runtime error. | |
| void | warnOperationUnexpectedBehaviour (const std::string &msg) const |
| Helper function to warn about a behavior different than asked. | |
| bool | inputIsEmptyOrNull (const SharedImageSet *input) const |
| Helper function to check if the input is empty or null and print a warning if so. | |
| virtual bool | useGPU (const SharedImageSet *input) const |
| Helper function that returns which device the operation should choose based on its configuration and a given input. | |
| void | prepareInputForDevice (SharedImageSet &input) const |
| Make sure the input is on the correct device for the current device strategy. | |
| virtual bool | allowChannelBatchOnGPU () const |
| Whether the operation implements channel-batching so that it can still be run on input with more than 4 channels. | |
| bool | configureOnly (const Properties &properties, const std::vector< ParameterBase * > ¶mSelection) noexcept |
| Helper method to configure only a subset of parameters. | |
| void | setManuallyConfiguredParameters (const std::vector< ParameterBase * > &manuallyConfiguredParams) |
| Disable the auto-configuration of some of the parameters so that they can be parsed manually (in the configure function of the derived class). | |
| void | applyPreProcessHooks (DataElement *element) |
| Apply the current pre-process hooks to an element. | |
| void | applyPostProcessHooks (DataElement *element) |
| Apply the current post-process hooks to an element. | |
| Protected Attributes inherited from Operation | |
| std::string | m_name |
| ProcessingPolicy | m_processingFieldsPolicy |
| Flag specifying whether labels should be processed by default (when no active field has been specified) | |
| std::optional< std::unordered_set< std::string > > | m_activeFields |
| Run the Operation only on those fields (if empty, will select suitable fields based on the current processing policy) when calling process() | |
| std::unordered_set< std::string > | m_alreadyWarned |
| Set of warnings about unexpected behavior that have already been printed. | |
| bool | m_errorOnUnexpectedBehaviour = false |
| Whether to throw an exception instead of warning about unexpected behavior. | |
| std::optional< uint32_t > | m_seed |
| Random::Generator | m_randGenerator |
| ML::ComputingDevice | m_device = ML::ComputingDevice::GPUIfOpenGl |
| std::vector< ParameterBase * > | m_params |
| All registered parameters. | |
| std::vector< ParameterBase * > | m_manuallyConfiguredParams |
| All registered parameters that should not be configured automatically. | |
| std::unordered_map< std::string, ElementType > | m_requiredFieldsTypes |
| List of all required field types used by checkRequiredItemsTypes. | |
| std::string | m_recordIdentifier |
| User-provided unique identifier for this operation, used for recording processing history and inversion. | |
| std::vector< PreProcessHook > | m_preProcessHooks |
| std::vector< PostProcessHook > | m_postProcessHooks |
| std::vector< PreProcessHook > | m_temporaryPreProcessHooks |
| Pre-process hooks that will be used for the next processing call only. | |
| std::vector< PostProcessHook > | m_temporaryPostProcessHooks |
| Post-process hooks that will be used for the next processing call only. | |
|
overridevirtual |
Main function that processes a DataItem in-place, may throw OperationException The data to be processed is determined by the active fields (if set) and the processing policy otherwise (if set), otherwise all fields are processed.
Reimplemented from Operation.
| ML::OpParam<std::string> p_directionFieldName = {"direction_field_name", "", this, ML::ParamRequired::No} |
If not empty, the field of this name will be treated as a "direction" output, i.e.
a 12-channel field where the first 6 values are the (x,y) coordinates of the standard coordinate system in 3D (orthographically) projected to the DRRs, and the next 6 values are their squares (viewed as complex numbers).
