Skip to content

Commit 676c0ef

Browse files
committed
fix: add wrapper to convert between a feature and structural frame
1 parent 97122a5 commit 676c0ef

File tree

1 file changed

+24
-1
lines changed

1 file changed

+24
-1
lines changed

LoopStructural/modelling/features/_feature_converters.py

Lines changed: 24 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
from LoopStructural.modelling.features.fold import FoldEvent
2-
from LoopStructural.modelling.features.builders import FoldedFeatureBuilder
2+
from LoopStructural.modelling.features.builders import FoldedFeatureBuilder, StructuralFrameBuilder
33
def add_fold_to_feature(feature, fold_frame,**kwargs):
44
fold = FoldEvent(fold_frame, name=f"Fold_{feature.name}", invert_norm=kwargs.get('invert_fold_norm', False))
55

@@ -11,3 +11,26 @@ def add_fold_to_feature(feature, fold_frame,**kwargs):
1111
feature = builder.feature
1212
feature.fold = fold
1313
return feature
14+
15+
def convert_feature_to_structural_frame(feature, **kwargs):
16+
"""
17+
Convert a geological feature to a structural frame by adding the feature to the frame
18+
19+
Parameters
20+
----------
21+
feature : GeologicalFeature
22+
the geological feature to convert
23+
24+
Returns
25+
-------
26+
StructuralFrame
27+
the updated structural frame with the feature added
28+
"""
29+
builder = feature.builder
30+
31+
new_builder = StructuralFrameBuilder.from_feature_builder(
32+
builder,
33+
**kwargs
34+
)
35+
return new_builder.frame
36+

0 commit comments

Comments
 (0)