make FaultCollection constructible from variables

also add accessor and equality operator
This commit is contained in:
Arne Morten Kvarving
2019-12-16 14:27:16 +01:00
parent d5d5a01503
commit 6a1acc15b8
2 changed files with 18 additions and 0 deletions

View File

@@ -36,6 +36,7 @@ class FaultCollection {
public:
FaultCollection();
FaultCollection(const GRIDSection& gridSection, const GridDims& grid);
FaultCollection(const OrderedMap<std::string, Fault>& faults);
size_t size() const;
bool hasFault(const std::string& faultName) const;
@@ -48,6 +49,10 @@ public:
void addFault(const std::string& faultName);
void setTransMult(const std::string& faultName , double transMult);
const OrderedMap<std::string, Fault>& getFaults() const;
bool operator==(const FaultCollection& data) const;
private:
void addFaultFaces(const GridDims& grid,
const DeckRecord& faultRecord,

View File

@@ -56,6 +56,11 @@ namespace Opm {
}
}
FaultCollection::FaultCollection(const OrderedMap<std::string, Fault>& faults) :
m_faults(faults)
{
}
void FaultCollection::addFaultFaces(const GridDims& grid,
const DeckRecord& faultRecord,
@@ -113,4 +118,12 @@ namespace Opm {
Fault& fault = getFault( faultName );
fault.setTransMult( transMult );
}
const OrderedMap<std::string, Fault>& FaultCollection::getFaults() const {
return m_faults;
}
bool FaultCollection::operator==(const FaultCollection& data) const {
return this->getFaults() == data.getFaults();
}
}