Added a removeUnits() member function for completeness

Author: Martin D. Weinberg

expui/Coefficients.H

@@ -239,6 +239,9 @@ namespace CoefClasses
     void setUnits
     (const std::vector<std::tuple<std::string, std::string, float>>& units);
 
+    //! Remove a unit by name
+    void removeUnits(const std::string name);
+
     //! Write units to H5
     void WriteH5Units(HighFive::File& file);
 

expui/Coefficients.cc

@@ -71,6 +71,19 @@ namespace CoefClasses
     p->times    = times;
   }
 
+  void Coefs::removeUnits(const std::string name)
+  {
+    // Lambda test for matching name
+    auto test = [&name](Unit& elem) -> bool{return elem.name==name; };
+
+    // Explanation: std::remove_if shifts elements to be removed to
+    // the end of the range and returns an iterator to the new logical
+    // end. units.erase() then removes the elements from that point to
+    // the end.
+    units.erase(std::remove_if(units.begin(), units.end(), test), units.end()); 
+  }
+  
+
   void Coefs::setUnits
   (const std::string Name, const std::string Unit, const float Value)
   {

expui/UnitValidator.cc

@@ -54,6 +54,8 @@ UnitValidator::createAllowedUnitTypes()
     allowed["L"]      = "length";
     allowed["m"]      = "mass";
     allowed["M"]      = "mass";
+    allowed["t"]      = "time";
+    allowed["T"]      = "time";
     allowed["vel"]    = "velocity";
     allowed["Vel"]    = "velocity";
     allowed["v"]      = "velocity";

pyEXP/CoefWrappers.cc

@@ -202,6 +202,19 @@ void CoefficientClasses(py::module &m) {
     void zerodata() override {
       PYBIND11_OVERRIDE_PURE(void, Coefs, zerodata,);
     }
+
+    void setUnits(const std::string name, const std::string unit, const float value) {
+      PYBIND11_OVERRIDE(void, Coefs, setUnits, name, unit, value);
+    }
+
+    void setUnits(const std::vector<std::tuple<std::string, std::string, float>>& units) {
+      PYBIND11_OVERRIDE(void, Coefs, setUnits, units);
+    }
+
+    void removeUnits(const std::string name) {
+      PYBIND11_OVERRIDE(void, Coefs, removeUnits, name);
+    }
+
   };
 
   class PySphCoefs : public SphCoefs
@@ -1167,13 +1180,27 @@ void CoefficientClasses(py::module &m) {
     .def("Times",
 	 &CoefClasses::Coefs::Times,
 	 R"(
-            Return a list of times for coefficient sets currently in the container
+         Return a list of times for coefficient sets currently in the container
 
-            Returns
-            -------
-            list(float,...)
-                list of times
-            )")
+         Returns
+         -------
+         list(float,...)
+             list of times
+         )")
+    .def("removeUnits",
+         &CoefClasses::Coefs::removeUnits,
+         R"(
+         Remove a unit from the coefficient structure.
+
+         Parameters
+         ----------
+         name : str
+             the name of physical quantity (G, Length, Mass, Time, etc)
+
+         Returns
+         -------
+         None
+         )")
     .def("setUnits",
 	 py::overload_cast<const std::string, const std::string, const float>(&CoefClasses::Coefs::setUnits),
 	 R"(