format

Author: baggepinnen

src/Mechanical/Rotational/sources.jl

@@ -1,13 +1,14 @@
 
 function PartialTorque(; name, use_support = false)
-  @named partial_element = PartialElementaryOneFlangeAndSupport2(use_support = use_support)
-  @unpack flange, phi_support = partial_element
-  @variables phi(t) [description = "Angle of flange with respect to support (= flange.phi - support.phi)"]
-  eqs = [phi ~ flange.phi - phi_support]
-  return extend(ODESystem(eqs, t; name = name), partial_element)
+    @named partial_element = PartialElementaryOneFlangeAndSupport2(use_support = use_support)
+    @unpack flange, phi_support = partial_element
+    @variables phi(t) [
+        description = "Angle of flange with respect to support (= flange.phi - support.phi)",
+    ]
+    eqs = [phi ~ flange.phi - phi_support]
+    return extend(ODESystem(eqs, t; name = name), partial_element)
 end
 
-
 """
     Torque(; name, use_support=false) 
 
@@ -34,21 +35,20 @@ function Torque(; name, use_support = false)
     return extend(ODESystem(eqs, t, [], []; name = name, systems = [tau]), partial_element)
 end
 
-
-
 function ConstantTorque(; name, tau_constant, use_support = false)
-  # NOTE: The use_support option is not specified for modelica ConstantTorque and defaults to false in the definition of PartialElementaryOneFlangeAndSupport2. If we do not set it to true, we will get a structurally singular system, and in the comment attached to the phi variable, they mention "flange.phi - support.phi", indicating that the support is used. 
-  @named partial_element = PartialTorque(; use_support)
-  @unpack flange, phi = partial_element
-  @parameters tau_constant=tau_constant [description = "Constant torque (if negative, torque is acting as load in positive direction of rotation)"]
-  @variables tau(t) [description = "Accelerating torque acting at flange (= -flange.tau)"]
-  @variables w(t) [description = "Angular velocity of flange with respect to support (= der(phi))"]
-  eqs = [
-    w ~ D(phi)
-    tau ~ -flange.tau
-    tau ~ tau_constant
-  ]
-  return extend(ODESystem(eqs, t; name = name), partial_element)
+    @named partial_element = PartialTorque(; use_support)
+    @unpack flange, phi = partial_element
+    @parameters tau_constant=tau_constant [
+        description = "Constant torque (if negative, torque is acting as load in positive direction of rotation)",
+    ]
+    @variables tau(t) [description = "Accelerating torque acting at flange (= -flange.tau)"]
+    @variables w(t) [
+        description = "Angular velocity of flange with respect to support (= der(phi))",
+    ]
+    eqs = [w ~ D(phi)
+           tau ~ -flange.tau
+           tau ~ tau_constant]
+    return extend(ODESystem(eqs, t; name = name), partial_element)
 end
 
 """

src/Mechanical/Rotational/utils.jl

@@ -18,7 +18,6 @@ Base.@doc """
     extend(ODESystem(Equation[], t, [], [], name = name), flange)
 end
 
-
 Base.@doc """
     InternalSupport(;name, tau)
 
@@ -35,11 +34,9 @@ Base.@doc """
     @named flange = Flange()
     @variables phi(t)=0 [description = "Rotation angle of support $name"]
     # tau(t), [connect = Flow, description = "Cut torque in support $name"],)
-    equations = [
-        flange.tau ~ tau
-        flange.phi ~ phi
-    ]
-    ODESystem(equations, t, [phi], [], name = name, systems=[flange]) # NOTE: tau not included since it belongs elsewhere
+    equations = [flange.tau ~ tau
+                 flange.phi ~ phi]
+    ODESystem(equations, t, [phi], [], name = name, systems = [flange]) # NOTE: tau not included since it belongs elsewhere
 end
 
 Base.@doc """