diff --git a/opm/parser/eclipse/EclipseState/Schedule/Completion.cpp b/opm/parser/eclipse/EclipseState/Schedule/Completion.cpp index 6670ab9a0..084fd994b 100644 --- a/opm/parser/eclipse/EclipseState/Schedule/Completion.cpp +++ b/opm/parser/eclipse/EclipseState/Schedule/Completion.cpp @@ -246,16 +246,15 @@ namespace Opm { return m_segment_number; } - void Completion::setSegmentNumber(const int segment_number){ - m_segment_number = segment_number; - } - double Completion::getCenterDepth() const { return m_center_depth; } - void Completion::setCenterDepth(const double center_depth){ - m_center_depth = center_depth; + void Completion::attachSegment(int segmentNumber , double centerDepth) { + assert(segmentNumber > 0); + + m_segment_number = segmentNumber; + m_center_depth = centerDepth; } diff --git a/opm/parser/eclipse/EclipseState/Schedule/Completion.hpp b/opm/parser/eclipse/EclipseState/Schedule/Completion.hpp index b34bfdada..ea9b62dde 100644 --- a/opm/parser/eclipse/EclipseState/Schedule/Completion.hpp +++ b/opm/parser/eclipse/EclipseState/Schedule/Completion.hpp @@ -59,9 +59,8 @@ namespace Opm { double getSkinFactor() const; void fixDefaultIJ(int wellHeadI , int wellHeadJ); int getSegmentNumber() const; - void setSegmentNumber(const int segment_number); double getCenterDepth() const; - void setCenterDepth(const double center_depth); + void attachSegment(const int segmentNumber , const double centerDepth); WellCompletion::DirectionEnum getDirection() const; diff --git a/opm/parser/eclipse/EclipseState/Schedule/Compsegs.cpp b/opm/parser/eclipse/EclipseState/Schedule/Compsegs.cpp index ca16c3db5..b1c88632a 100644 --- a/opm/parser/eclipse/EclipseState/Schedule/Compsegs.cpp +++ b/opm/parser/eclipse/EclipseState/Schedule/Compsegs.cpp @@ -192,8 +192,7 @@ namespace Opm { } CompletionPtr new_completion = std::make_shared(completion_set->get(ic)); - new_completion->setSegmentNumber(compsegs[i_comp]->m_segment_number); - new_completion->setCenterDepth(compsegs[i_comp]->m_center_depth); + new_completion->attachSegment(compsegs[i_comp]->m_segment_number, compsegs[i_comp]->m_center_depth); completion_set->add(new_completion); }