AirLoopHVACUnitaryHeatPumpAirToAirMultiSpeed: restrict addToNode + fi…

…xup clone method
NREL · Dec 18, 2024 · 93c150f · 93c150f
1 parent 33fd849
commit 93c150f
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 8 deletions.
diff --git a/src/model/AirLoopHVACUnitaryHeatPumpAirToAirMultiSpeed.cpp b/src/model/AirLoopHVACUnitaryHeatPumpAirToAirMultiSpeed.cpp
@@ -7,12 +7,14 @@
 #include "Model_Impl.hpp"
 #include "AirLoopHVACUnitaryHeatPumpAirToAirMultiSpeed.hpp"
 #include "AirLoopHVACUnitaryHeatPumpAirToAirMultiSpeed_Impl.hpp"
+#include "AirLoopHVAC.hpp"
+#include "Node.hpp"
 #include "Schedule.hpp"
 #include "Schedule_Impl.hpp"
-#include "ThermalZone.hpp"
-#include "ThermalZone_Impl.hpp"
 #include "ScheduleTypeLimits.hpp"
 #include "ScheduleTypeRegistry.hpp"
+#include "ThermalZone.hpp"
+#include "ThermalZone_Impl.hpp"
 
 #include "../utilities/core/Assert.hpp"
 #include "../utilities/data/DataEnums.hpp"
@@ -595,8 +597,18 @@ namespace model {
       return OS_AirLoopHVAC_UnitaryHeatPump_AirToAir_MultiSpeedFields::AirOutletNode;
     }
 
+    bool AirLoopHVACUnitaryHeatPumpAirToAirMultiSpeed_Impl::addToNode(Node& node) {
+      if (boost::optional<AirLoopHVAC> airLoop = node.airLoopHVAC()) {
+        if (airLoop->supplyComponent(node.handle())) {
+          return StraightComponent_Impl::addToNode(node);
+        }
+      }
+
+      return false;
+    }
+
     ModelObject AirLoopHVACUnitaryHeatPumpAirToAirMultiSpeed_Impl::clone(Model model) const {
-      auto modelObjectClone = ModelObject_Impl::clone(model).cast<AirLoopHVACUnitaryHeatPumpAirToAirMultiSpeed>();
+      auto modelObjectClone = StraightComponent_Impl::clone(model).cast<AirLoopHVACUnitaryHeatPumpAirToAirMultiSpeed>();
 
       if (boost::optional<HVACComponent> supplyFan = this->supplyAirFan()) {
         modelObjectClone.setSupplyAirFan(supplyFan->clone(model).cast<HVACComponent>());

diff --git a/src/model/AirLoopHVACUnitaryHeatPumpAirToAirMultiSpeed_Impl.hpp b/src/model/AirLoopHVACUnitaryHeatPumpAirToAirMultiSpeed_Impl.hpp
@@ -43,6 +43,14 @@ namespace model {
 
       virtual std::vector<ScheduleTypeKey> getScheduleTypeKeys(const Schedule& schedule) const override;
 
+      virtual std::vector<ModelObject> children() const override;
+
+      virtual unsigned inletPort() const override;
+      virtual unsigned outletPort() const override;
+      virtual bool addToNode(Node& node) override;
+
+      virtual ModelObject clone(Model model) const override;
+
       virtual ComponentType componentType() const override;
       virtual std::vector<FuelType> coolingFuelTypes() const override;
       virtual std::vector<FuelType> heatingFuelTypes() const override;
@@ -236,11 +244,6 @@ namespace model {
       /** @name Other */
       //@{
 
-      virtual unsigned inletPort() const override;
-      virtual unsigned outletPort() const override;
-      std::vector<ModelObject> children() const override;
-      ModelObject clone(Model model) const override;
-
       //@}
      protected:
      private: