Make casting decimal types neater

angularsen · Jan 9, 2024 · d6cb7d1 · d6cb7d1
1 parent ef44d00
commit d6cb7d1
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/CodeGen/Generators/UnitsNetGen/QuantityGenerator.cs b/CodeGen/Generators/UnitsNetGen/QuantityGenerator.cs
@@ -17,8 +17,6 @@ internal class QuantityGenerator : GeneratorBase
         private readonly string _valueType;
         private readonly Unit _baseUnit;
 
-        private readonly string[] _decimalTypes = { "BitRate", "Information", "Power" };
-
         public QuantityGenerator(Quantity quantity)
         {
             _quantity = quantity ?? throw new ArgumentNullException(nameof(quantity));
@@ -794,8 +792,8 @@ private void GenerateRelationalOperators()
                         rightPart = "value";
                     }
 
-                    var leftCast = _decimalTypes.Contains(relation.LeftQuantity.Name) ? "(double)" : "";
-                    var rightCast = _decimalTypes.Contains(relation.RightQuantity.Name) ? "(double)" : "";
+                    var leftCast = relation.LeftQuantity.ValueType is "decimal" ? "(double)" : string.Empty;
+                    var rightCast = relation.RightQuantity.ValueType is "decimal" ? "(double)" : string.Empty;
 
                     var expression = $"{leftCast}{leftPart} {relation.Operator} {rightCast}{rightPart}";