qnty 0.0.1__py3-none-any.whl → 0.0.3__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
qnty/__init__.py CHANGED
@@ -0,0 +1,175 @@
1
+ """
2
+ Qnty - High-Performance Unit System for Engineering
3
+ ====================================================
4
+
5
+ A fast, type-safe unit system library for Python with dimensional safety
6
+ and optimized unit conversions for engineering calculations.
7
+ """
8
+
9
+ from .dimension import BaseDimension, DimensionSignature
10
+ from .equation import Equation
11
+ from .expression import Expression
12
+ from .unit import registry
13
+ from .units import register_all_units
14
+ from .variable import FastQuantity, TypeSafeSetter, TypeSafeVariable
15
+ from .variables import (
16
+ AbsorbedDose,
17
+ Acceleration,
18
+ ActivationEnergy,
19
+ AmountOfSubstance,
20
+ AnglePlane,
21
+ AngleSolid,
22
+ AngularAcceleration,
23
+ AngularMomentum,
24
+ Area,
25
+ AreaPerUnitVolume,
26
+ AtomicWeight,
27
+ Concentration,
28
+ DynamicFluidity,
29
+ ElectricCapacitance,
30
+ ElectricCharge,
31
+ ElectricCurrentIntensity,
32
+ ElectricDipoleMoment,
33
+ ElectricFieldStrength,
34
+ ElectricInductance,
35
+ ElectricPotential,
36
+ ElectricResistance,
37
+ ElectricalConductance,
38
+ ElectricalPermittivity,
39
+ ElectricalResistivity,
40
+ EnergyFlux,
41
+ EnergyHeatWork,
42
+ EnergyPerUnitArea,
43
+ Force,
44
+ ForceBody,
45
+ ForcePerUnitMass,
46
+ FrequencyVoltageRatio,
47
+ FuelConsumption,
48
+ HeatOfCombustion,
49
+ HeatOfFusion,
50
+ HeatOfVaporization,
51
+ HeatTransferCoefficient,
52
+ Illuminance,
53
+ KineticEnergyOfTurbulence,
54
+ Length,
55
+ LinearMassDensity,
56
+ LinearMomentum,
57
+ LuminanceSelf,
58
+ LuminousFlux,
59
+ LuminousIntensity,
60
+ MagneticField,
61
+ MagneticFlux,
62
+ MagneticInductionFieldStrength,
63
+ MagneticMoment,
64
+ MagneticPermeability,
65
+ MagnetomotiveForce,
66
+ Mass,
67
+ MassDensity,
68
+ MassFlowRate,
69
+ MassFlux,
70
+ MassFractionOfI,
71
+ MassTransferCoefficient,
72
+ MolalityOfSoluteI,
73
+ MolarConcentrationByMass,
74
+ MolarFlowRate,
75
+ MolarFlux,
76
+ MolarHeatCapacity,
77
+ MolarityOfI,
78
+ MoleFractionOfI,
79
+ MomentOfInertia,
80
+ MomentumFlowRate,
81
+ MomentumFlux,
82
+ NormalityOfSolution,
83
+ ParticleDensity,
84
+ Permeability,
85
+ PhotonEmissionRate,
86
+ PowerPerUnitMass,
87
+ PowerPerUnitVolume,
88
+ PowerThermalDuty,
89
+ Pressure,
90
+ RadiationDoseEquivalent,
91
+ RadiationExposure,
92
+ Radioactivity,
93
+ SecondMomentOfArea,
94
+ SecondRadiationConstantPlanck,
95
+ SpecificEnthalpy,
96
+ SpecificGravity,
97
+ SpecificHeatCapacityConstantPressure,
98
+ SpecificLength,
99
+ SpecificSurface,
100
+ SpecificVolume,
101
+ Stress,
102
+ SurfaceMassDensity,
103
+ SurfaceTension,
104
+ Temperature,
105
+ ThermalConductivity,
106
+ Time,
107
+ Torque,
108
+ TurbulenceEnergyDissipationRate,
109
+ VelocityAngular,
110
+ VelocityLinear,
111
+ ViscosityDynamic,
112
+ ViscosityKinematic,
113
+ Volume,
114
+ VolumeFractionOfI,
115
+ VolumetricCalorificHeatingValue,
116
+ VolumetricCoefficientOfExpansion,
117
+ VolumetricFlowRate,
118
+ VolumetricFlux,
119
+ VolumetricMassFlowRate,
120
+ Wavenumber
121
+ )
122
+
123
+ # Register all units to the global registry
124
+ register_all_units(registry)
125
+
126
+ # Finalize registry after all registrations
127
+ registry.finalize_registration()
128
+
129
+ # Version information
130
+ __version__ = "0.0.3"
131
+
132
+ # Define public API
133
+ __all__ = [
134
+ # Core variable types (most commonly used)
135
+ "Length", "Pressure", "Temperature", "Time", "Mass", "Volume", "Area",
136
+ "Force", "EnergyHeatWork", "PowerThermalDuty",
137
+
138
+ # Core classes for advanced usage
139
+ "FastQuantity", "TypeSafeVariable", "TypeSafeSetter",
140
+ "DimensionSignature", "BaseDimension",
141
+ "Expression", "Equation",
142
+
143
+ # All other variable types (95 additional types)
144
+ "AbsorbedDose", "Acceleration", "ActivationEnergy", "AmountOfSubstance",
145
+ "AnglePlane", "AngleSolid", "AngularAcceleration", "AngularMomentum",
146
+ "AreaPerUnitVolume", "AtomicWeight", "Concentration", "DynamicFluidity",
147
+ "ElectricCapacitance", "ElectricCharge", "ElectricCurrentIntensity",
148
+ "ElectricDipoleMoment", "ElectricFieldStrength", "ElectricInductance",
149
+ "ElectricPotential", "ElectricResistance", "ElectricalConductance",
150
+ "ElectricalPermittivity", "ElectricalResistivity", "EnergyFlux",
151
+ "EnergyPerUnitArea", "ForceBody", "ForcePerUnitMass",
152
+ "FrequencyVoltageRatio", "FuelConsumption", "HeatOfCombustion",
153
+ "HeatOfFusion", "HeatOfVaporization", "HeatTransferCoefficient",
154
+ "Illuminance", "KineticEnergyOfTurbulence", "LinearMassDensity",
155
+ "LinearMomentum", "LuminanceSelf", "LuminousFlux", "LuminousIntensity",
156
+ "MagneticField", "MagneticFlux", "MagneticInductionFieldStrength",
157
+ "MagneticMoment", "MagneticPermeability", "MagnetomotiveForce",
158
+ "MassDensity", "MassFlowRate", "MassFlux", "MassFractionOfI",
159
+ "MassTransferCoefficient", "MolalityOfSoluteI", "MolarConcentrationByMass",
160
+ "MolarFlowRate", "MolarFlux", "MolarHeatCapacity", "MolarityOfI",
161
+ "MoleFractionOfI", "MomentOfInertia", "MomentumFlowRate", "MomentumFlux",
162
+ "NormalityOfSolution", "ParticleDensity", "Permeability",
163
+ "PhotonEmissionRate", "PowerPerUnitMass", "PowerPerUnitVolume",
164
+ "RadiationDoseEquivalent", "RadiationExposure", "Radioactivity",
165
+ "SecondMomentOfArea", "SecondRadiationConstantPlanck", "SpecificEnthalpy",
166
+ "SpecificGravity", "SpecificHeatCapacityConstantPressure",
167
+ "SpecificLength", "SpecificSurface", "SpecificVolume", "Stress",
168
+ "SurfaceMassDensity", "SurfaceTension", "ThermalConductivity", "Torque",
169
+ "TurbulenceEnergyDissipationRate", "VelocityAngular", "VelocityLinear",
170
+ "ViscosityDynamic", "ViscosityKinematic", "VolumeFractionOfI",
171
+ "VolumetricCalorificHeatingValue", "VolumetricCoefficientOfExpansion",
172
+ "VolumetricFlowRate", "VolumetricFlux", "VolumetricMassFlowRate",
173
+ "Wavenumber",
174
+ ]
175
+
qnty/dimension.py CHANGED
@@ -5,14 +5,15 @@ Dimension System
5
5
  Compile-time dimensional analysis using type system for ultra-fast operations.
6
6
  """
7
7
 
8
- from typing import final
9
8
  from dataclasses import dataclass
10
9
  from enum import IntEnum
10
+ from typing import final
11
+
11
12
 
12
13
  class BaseDimension(IntEnum):
13
14
  """Base dimensions as prime numbers for efficient bit operations."""
14
15
  LENGTH = 2
15
- MASS = 3
16
+ MASS = 3
16
17
  TIME = 5
17
18
  CURRENT = 7
18
19
  TEMPERATURE = 11
@@ -71,15 +72,113 @@ class DimensionSignature:
71
72
  return hash(self._signature)
72
73
 
73
74
 
74
- # Pre-defined dimension constants
75
- DIMENSIONLESS = DimensionSignature.create()
76
- LENGTH = DimensionSignature.create(length=1)
77
- MASS = DimensionSignature.create(mass=1)
78
- TIME = DimensionSignature.create(time=1)
79
- AREA = DimensionSignature.create(length=2)
80
- VOLUME = DimensionSignature.create(length=3)
81
- VELOCITY = DimensionSignature.create(length=1, time=-1)
82
- ACCELERATION = DimensionSignature.create(length=1, time=-2)
83
- FORCE = DimensionSignature.create(mass=1, length=1, time=-2)
84
- PRESSURE = DimensionSignature.create(mass=1, length=-1, time=-2)
85
- ENERGY = DimensionSignature.create(mass=1, length=2, time=-2)
75
+ # Pre-defined dimension constants (alphabetically ordered)
76
+ ABSORBED_DOSE = DimensionSignature.create(length=2, time=-2) # L^2 T^-2
77
+ ACCELERATION = DimensionSignature.create(length=1, time=-2) # L T^-2
78
+ ACTIVATION_ENERGY = DimensionSignature.create(amount=-1, length=2, time=-2) # N^-1 L^2 T^-2
79
+ AMOUNT = DimensionSignature.create(amount=1) # N
80
+ AMOUNT_OF_SUBSTANCE = DimensionSignature.create(amount=1) # N
81
+ ANGLE_PLANE = DimensionSignature.create() # Dimensionless
82
+ ANGLE_SOLID = DimensionSignature.create() # Dimensionless
83
+ ANGULAR_ACCELERATION = DimensionSignature.create(time=-2) # T^-2
84
+ ANGULAR_MOMENTUM = DimensionSignature.create(length=2, mass=1, time=-1) # L^2 M T^-1
85
+ AREA = DimensionSignature.create(length=2) # L^2
86
+ AREA_PER_UNIT_VOLUME = DimensionSignature.create(length=-1) # L^-1
87
+ ATOMIC_WEIGHT = DimensionSignature.create(amount=-1, mass=1) # N^-1 M
88
+ CONCENTRATION = DimensionSignature.create(length=-3, mass=1) # L^-3 M
89
+ CURRENT = DimensionSignature.create(current=1) # A
90
+ DIMENSIONLESS = DimensionSignature.create() # Dimensionless
91
+ DYNAMIC_FLUIDITY = DimensionSignature.create(length=1, mass=-1, time=1) # L M^-1 T
92
+ ELECTRICAL_CONDUCTANCE = DimensionSignature.create(current=2, length=-2, mass=-1, time=3) # A^2 L^-2 M^-1 T^3
93
+ ELECTRICAL_PERMITTIVITY = DimensionSignature.create(current=2, length=-3, mass=-1, time=4) # A^2 L^-3 M^-1 T^4
94
+ ELECTRICAL_RESISTIVITY = DimensionSignature.create(current=-2, length=3, mass=1, time=-3) # A^-2 L^3 M T^-3
95
+ ELECTRIC_CAPACITANCE = DimensionSignature.create(current=2, length=-2, mass=-1, time=4) # A^2 L^-2 M^-1 T^4
96
+ ELECTRIC_CHARGE = DimensionSignature.create(amount=-1, current=1, time=1) # N^-1 A T
97
+ ELECTRIC_CURRENT_INTENSITY = DimensionSignature.create(current=1) # A
98
+ ELECTRIC_DIPOLE_MOMENT = DimensionSignature.create(current=1, length=1, time=1) # A L T
99
+ ELECTRIC_FIELD_STRENGTH = DimensionSignature.create(current=-1, length=1, mass=1, time=-3) # A^-1 L M T^-3
100
+ ELECTRIC_INDUCTANCE = DimensionSignature.create(current=-2, length=2, mass=1, time=-2) # A^-2 L^2 M T^-2
101
+ ELECTRIC_POTENTIAL = DimensionSignature.create(current=-1, length=2, mass=1, time=-3) # A^-1 L^2 M T^-3
102
+ ELECTRIC_RESISTANCE = DimensionSignature.create(current=-2, length=2, mass=1, time=-3) # A^-2 L^2 M T^-3
103
+ ENERGY_FLUX = DimensionSignature.create(mass=1, time=-3) # M T^-3
104
+ ENERGY_HEAT_WORK = DimensionSignature.create(length=2, mass=1, time=-2) # L^2 M T^-2
105
+ ENERGY_PER_UNIT_AREA = DimensionSignature.create(mass=1, time=-2) # M T^-2
106
+ FORCE = DimensionSignature.create(length=1, mass=1, time=-2) # L M T^-2
107
+ FORCE_BODY = DimensionSignature.create(length=-2, mass=1, time=-2) # L^-2 M T^-2
108
+ FORCE_PER_UNIT_MASS = DimensionSignature.create(length=1, time=-2) # L T^-2
109
+ FREQUENCY_VOLTAGE_RATIO = DimensionSignature.create(current=1, length=-2, mass=-1, time=3) # A L^-2 M^-1 T^3
110
+ FUEL_CONSUMPTION = DimensionSignature.create(length=-2) # L^-2
111
+ HEAT_OF_COMBUSTION = DimensionSignature.create(length=2, time=-2) # L^2 T^-2
112
+ HEAT_OF_FUSION = DimensionSignature.create(length=2, time=-2) # L^2 T^-2
113
+ HEAT_OF_VAPORIZATION = DimensionSignature.create(length=2, time=-2) # L^2 T^-2
114
+ HEAT_TRANSFER_COEFFICIENT = DimensionSignature.create(mass=1, temp=-1, time=-3) # M Θ^-1 T^-3
115
+ ILLUMINANCE = DimensionSignature.create(length=-2, luminosity=1) # L^-2 J
116
+ KINETIC_ENERGY_OF_TURBULENCE = DimensionSignature.create(length=2, time=-2) # L^2 T^-2
117
+ LENGTH = DimensionSignature.create(length=1) # L
118
+ LINEAR_MASS_DENSITY = DimensionSignature.create(length=-1, mass=1) # L^-1 M
119
+ LINEAR_MOMENTUM = DimensionSignature.create(length=1, mass=1, time=-1) # L M T^-1
120
+ LUMINANCE_SELF = DimensionSignature.create(length=-2, luminosity=1) # L^-2 J
121
+ LUMINOSITY = DimensionSignature.create(luminosity=1) # J
122
+ LUMINOUS_FLUX = DimensionSignature.create(luminosity=1) # J
123
+ LUMINOUS_INTENSITY = DimensionSignature.create(luminosity=1) # J
124
+ MAGNETIC_FIELD = DimensionSignature.create(current=1, length=-1) # A L^-1
125
+ MAGNETIC_FLUX = DimensionSignature.create(current=-1, length=2, mass=1, time=-2) # A^-1 L^2 M T^-2
126
+ MAGNETIC_INDUCTION_FIELD_STRENGTH = DimensionSignature.create(current=-1, mass=1, time=-2) # A^-1 M T^-2
127
+ MAGNETIC_MOMENT = DimensionSignature.create(current=1, length=2) # A L^2
128
+ MAGNETIC_PERMEABILITY = DimensionSignature.create(current=-2, length=2, mass=1, time=-2) # A^-2 L^2 M T^-2
129
+ MAGNETOMOTIVE_FORCE = DimensionSignature.create(current=1) # A
130
+ MASS = DimensionSignature.create(mass=1) # M
131
+ MASS_DENSITY = DimensionSignature.create(length=-3, mass=1) # L^-3 M
132
+ MASS_FLOW_RATE = DimensionSignature.create(mass=1, time=-1) # M T^-1
133
+ MASS_FLUX = DimensionSignature.create(length=-2, mass=1, time=-1) # L^-2 M T^-1
134
+ MASS_FRACTION_OF_I = DimensionSignature.create() # Dimensionless
135
+ MASS_TRANSFER_COEFFICIENT = DimensionSignature.create(length=-2, mass=1, time=-1) # L^-2 M T^-1
136
+ MOLALITY_OF_SOLUTE_I = DimensionSignature.create(amount=1, mass=-1) # N M^-1
137
+ MOLARITY_OF_I = DimensionSignature.create(amount=1, length=-3) # N L^-3
138
+ MOLAR_CONCENTRATION_BY_MASS = DimensionSignature.create(amount=1) # N
139
+ MOLAR_FLOW_RATE = DimensionSignature.create(amount=1, time=-1) # N T^-1
140
+ MOLAR_FLUX = DimensionSignature.create(amount=1, length=-2, time=-1) # N L^-2 T^-1
141
+ MOLAR_HEAT_CAPACITY = DimensionSignature.create(amount=-1, length=2, temp=-1, time=-2) # N^-1 L^2 Θ^-1 T^-2
142
+ MOLE_FRACTION_OF_I = DimensionSignature.create() # Dimensionless
143
+ MOMENTUM_FLOW_RATE = DimensionSignature.create(length=1, mass=1, time=-2) # L M T^-2
144
+ MOMENTUM_FLUX = DimensionSignature.create(length=-1, mass=1, time=-2) # L^-1 M T^-2
145
+ MOMENT_OF_INERTIA = DimensionSignature.create(length=2, mass=1) # L^2 M
146
+ NORMALITY_OF_SOLUTION = DimensionSignature.create(amount=1, length=-3) # N L^-3
147
+ PARTICLE_DENSITY = DimensionSignature.create(length=-3) # L^-3
148
+ PERMEABILITY = DimensionSignature.create(length=2) # L^2
149
+ PHOTON_EMISSION_RATE = DimensionSignature.create(length=-2, time=-1) # L^-2 T^-1
150
+ POWER_PER_UNIT_MASS = DimensionSignature.create(length=2, time=-3) # L^2 T^-3
151
+ POWER_PER_UNIT_VOLUME = DimensionSignature.create(length=-1, mass=1, time=-3) # L^-1 M T^-3
152
+ POWER_THERMAL_DUTY = DimensionSignature.create(length=2, mass=1, time=-3) # L^2 M T^-3
153
+ PRESSURE = DimensionSignature.create(length=-1, mass=1, time=-2) # L^-1 M T^-2
154
+ RADIATION_DOSE_EQUIVALENT = DimensionSignature.create(length=2, time=-2) # L^2 T^-2
155
+ RADIATION_EXPOSURE = DimensionSignature.create(current=1, mass=-1, time=1) # A M^-1 T
156
+ RADIOACTIVITY = DimensionSignature.create(time=-1) # T^-1
157
+ SECOND_MOMENT_OF_AREA = DimensionSignature.create(length=4) # L^4
158
+ SECOND_RADIATION_CONSTANT_PLANCK = DimensionSignature.create(length=1, temp=1) # L Θ
159
+ SPECIFIC_ENTHALPY = DimensionSignature.create(length=2, time=-2) # L^2 T^-2
160
+ SPECIFIC_GRAVITY = DimensionSignature.create() # Dimensionless
161
+ SPECIFIC_HEAT_CAPACITY_CONSTANT_PRESSURE = DimensionSignature.create(length=2, mass=1, temp=-1, time=-2) # L^2 M Θ^-1 T^-2
162
+ SPECIFIC_LENGTH = DimensionSignature.create(length=1, mass=-1) # L M^-1
163
+ SPECIFIC_SURFACE = DimensionSignature.create(length=2, mass=-1) # L^2 M^-1
164
+ SPECIFIC_VOLUME = DimensionSignature.create(length=3, mass=-1) # L^3 M^-1
165
+ STRESS = DimensionSignature.create(length=-1, mass=1, time=-2) # L^-1 M T^-2
166
+ SURFACE_MASS_DENSITY = DimensionSignature.create(length=-2, mass=1) # L^-2 M
167
+ SURFACE_TENSION = DimensionSignature.create(mass=1, time=-2) # M T^-2
168
+ TEMPERATURE = DimensionSignature.create(temp=1) # Θ
169
+ THERMAL_CONDUCTIVITY = DimensionSignature.create(length=1, mass=1, temp=1, time=-3) # L M Θ T^-3
170
+ TIME = DimensionSignature.create(time=1) # T
171
+ TORQUE = DimensionSignature.create(length=2, mass=1, time=-2) # L^2 M T^-2
172
+ TURBULENCE_ENERGY_DISSIPATION_RATE = DimensionSignature.create(length=2, time=-3) # L^2 T^-3
173
+ VELOCITY_ANGULAR = DimensionSignature.create(time=-1) # T^-1
174
+ VELOCITY_LINEAR = DimensionSignature.create(length=1, time=-1) # L T^-1
175
+ VISCOSITY_DYNAMIC = DimensionSignature.create(length=-1, mass=1, time=-1) # L^-1 M T^-1
176
+ VISCOSITY_KINEMATIC = DimensionSignature.create(length=2, time=-1) # L^2 T^-1
177
+ VOLUME = DimensionSignature.create(length=3) # L^3
178
+ VOLUMETRIC_CALORIFIC_HEATING_VALUE = DimensionSignature.create(length=-1, mass=1, time=-2) # L^-1 M T^-2
179
+ VOLUMETRIC_COEFFICIENT_OF_EXPANSION = DimensionSignature.create(length=-3, mass=1, temp=-1) # L^-3 M Θ^-1
180
+ VOLUMETRIC_FLOW_RATE = DimensionSignature.create(length=3, time=-1) # L^3 T^-1
181
+ VOLUMETRIC_FLUX = DimensionSignature.create(length=1, time=-1) # L T^-1
182
+ VOLUMETRIC_MASS_FLOW_RATE = DimensionSignature.create(length=-3, mass=1, time=-1) # L^-3 M T^-1
183
+ VOLUME_FRACTION_OF_I = DimensionSignature.create() # Dimensionless
184
+ WAVENUMBER = DimensionSignature.create(length=-1) # L^-1
qnty/equation.py ADDED
@@ -0,0 +1,216 @@
1
+ """
2
+ Equation System
3
+ ===============
4
+
5
+ Mathematical equations for qnty variables with solving capabilities.
6
+ """
7
+
8
+ from __future__ import annotations
9
+
10
+ from typing import cast
11
+
12
+ from .expression import Expression, VariableReference
13
+ from .variable import TypeSafeVariable
14
+
15
+
16
+ class Equation:
17
+ """Represents a mathematical equation with left-hand side equal to right-hand side."""
18
+
19
+ def __init__(self, name: str, lhs: TypeSafeVariable | Expression, rhs: Expression):
20
+ self.name = name
21
+
22
+ # Convert Variable to VariableReference if needed
23
+ # Use duck typing to avoid circular import
24
+ if hasattr(lhs, 'name') and hasattr(lhs, 'quantity') and hasattr(lhs, 'is_known'):
25
+ # It's a TypeSafeVariable-like object
26
+ self.lhs = VariableReference(cast('TypeSafeVariable', lhs))
27
+ else:
28
+ # It's already an Expression
29
+ self.lhs = cast(Expression, lhs)
30
+
31
+ self.rhs = rhs
32
+ self.variables = self.get_all_variables()
33
+
34
+ def get_all_variables(self) -> set[str]:
35
+ """Get all variable names used in this equation."""
36
+ # Both lhs and rhs should be Expressions after __init__ conversion
37
+ lhs_vars = self.lhs.get_variables()
38
+ rhs_vars = self.rhs.get_variables()
39
+ return lhs_vars | rhs_vars
40
+
41
+ def get_unknown_variables(self, known_vars: set[str]) -> set[str]:
42
+ """Get variables that are unknown (not in known_vars set)."""
43
+ return self.variables - known_vars
44
+
45
+ def get_known_variables(self, known_vars: set[str]) -> set[str]:
46
+ """Get variables that are known (in known_vars set)."""
47
+ return self.variables & known_vars
48
+
49
+ def can_solve_for(self, target_var: str, known_vars: set[str]) -> bool:
50
+ """Check if this equation can solve for target_var given known_vars."""
51
+ if target_var not in self.variables:
52
+ return False
53
+ # Direct assignment case: lhs is the variable
54
+ if isinstance(self.lhs, VariableReference) and self.lhs.name == target_var:
55
+ rhs_vars = self.rhs.get_variables()
56
+ return rhs_vars.issubset(known_vars)
57
+ unknown_vars = self.get_unknown_variables(known_vars)
58
+ # Can solve if target_var is the only unknown
59
+ return unknown_vars == {target_var}
60
+
61
+ def solve_for(self, target_var: str, variable_values: dict[str, TypeSafeVariable]) -> TypeSafeVariable:
62
+ """
63
+ Solve the equation for target_var.
64
+ Returns the target variable with updated quantity.
65
+ """
66
+ if target_var not in self.variables:
67
+ raise ValueError(f"Variable '{target_var}' not found in equation")
68
+
69
+ # Handle direct assignment: target = expression
70
+ if isinstance(self.lhs, VariableReference) and self.lhs.name == target_var:
71
+ # Direct assignment: target_var = rhs
72
+ result_qty = self.rhs.evaluate(variable_values)
73
+
74
+ # Update existing variable object to preserve references
75
+ var_obj = variable_values.get(target_var)
76
+ if var_obj is not None:
77
+ # Convert result to the target variable's original unit if it had one
78
+ if var_obj.quantity is not None and var_obj.quantity.unit is not None:
79
+ # Convert to the target variable's defined unit
80
+ try:
81
+ result_qty = result_qty.to(var_obj.quantity.unit)
82
+ except Exception:
83
+ # If conversion fails, keep the calculated unit
84
+ pass
85
+
86
+ var_obj.quantity = result_qty
87
+ var_obj.is_known = True
88
+ return var_obj
89
+
90
+ # Create new variable if not found - this shouldn't happen in normal usage
91
+ raise ValueError(f"Variable '{target_var}' not found in variable_values")
92
+
93
+ # For more complex equations, we would need algebraic manipulation
94
+ # Currently focusing on direct assignment which covers most engineering cases
95
+ raise NotImplementedError(f"Cannot solve for {target_var} in equation {self}. "
96
+ f"Only direct assignment equations (var = expression) are supported.")
97
+
98
+ def check_residual(self, variable_values: dict[str, TypeSafeVariable], tolerance: float = 1e-10) -> bool:
99
+ """
100
+ Check if equation is satisfied by evaluating residual (LHS - RHS).
101
+ Returns True if |residual| < tolerance, accounting for units.
102
+ """
103
+ try:
104
+ # Both lhs and rhs should be Expressions after __init__ conversion
105
+ lhs_value = self.lhs.evaluate(variable_values)
106
+ rhs_value = self.rhs.evaluate(variable_values)
107
+
108
+ # Check dimensional compatibility
109
+ if lhs_value._dimension_sig != rhs_value._dimension_sig:
110
+ return False
111
+
112
+ # Convert to same units for comparison
113
+ rhs_converted = rhs_value.to(lhs_value.unit)
114
+ residual = abs(lhs_value.value - rhs_converted.value)
115
+
116
+ return residual < tolerance
117
+ except Exception:
118
+ return False
119
+
120
+ def __str__(self) -> str:
121
+ return f"{self.lhs} = {self.rhs}"
122
+
123
+ def __repr__(self) -> str:
124
+ return f"Equation(name='{self.name}', lhs={self.lhs!r}, rhs={self.rhs!r})"
125
+
126
+
127
+ class EquationSystem:
128
+ """System of equations that can be solved together."""
129
+
130
+ def __init__(self, equations: list[Equation] | None = None):
131
+ self.equations = equations or []
132
+ self.variables = {} # Dict[str, TypeSafeVariable]
133
+
134
+ def add_equation(self, equation: Equation):
135
+ """Add an equation to the system."""
136
+ self.equations.append(equation)
137
+
138
+ def add_variable(self, variable: TypeSafeVariable):
139
+ """Add a variable to the system."""
140
+ self.variables[variable.name] = variable
141
+
142
+ def get_known_variables(self) -> set[str]:
143
+ """Get names of all known variables."""
144
+ return {name for name, var in self.variables.items() if var.is_known and var.quantity is not None}
145
+
146
+ def get_unknown_variables(self) -> set[str]:
147
+ """Get names of all unknown variables."""
148
+ return {name for name, var in self.variables.items() if not var.is_known or var.quantity is None}
149
+
150
+ def can_solve_any(self) -> bool:
151
+ """Check if any equation can be solved with current known variables."""
152
+ known_vars = self.get_known_variables()
153
+ unknown_vars = self.get_unknown_variables()
154
+
155
+ for equation in self.equations:
156
+ for unknown_var in unknown_vars:
157
+ if equation.can_solve_for(unknown_var, known_vars):
158
+ return True
159
+ return False
160
+
161
+ def solve_step(self) -> bool:
162
+ """Solve one step - find and solve one equation. Returns True if progress made."""
163
+ known_vars = self.get_known_variables()
164
+ unknown_vars = self.get_unknown_variables()
165
+
166
+ # Find an equation that can be solved
167
+ for equation in self.equations:
168
+ for unknown_var in unknown_vars:
169
+ if equation.can_solve_for(unknown_var, known_vars):
170
+ # Solve for this variable
171
+ equation.solve_for(unknown_var, self.variables)
172
+ return True # Progress made
173
+
174
+ return False # No progress possible
175
+
176
+ def solve(self, max_iterations: int = 100) -> bool:
177
+ """Solve the system iteratively. Returns True if fully solved."""
178
+ for _ in range(max_iterations):
179
+ if not self.can_solve_any():
180
+ break
181
+ if not self.solve_step():
182
+ break
183
+
184
+ # Check if all variables are known
185
+ unknown_vars = self.get_unknown_variables()
186
+ return len(unknown_vars) == 0
187
+
188
+ def get_solving_order(self) -> list[str]:
189
+ """Get the order in which variables can be solved."""
190
+ order = []
191
+ temp_system = EquationSystem(self.equations.copy())
192
+ temp_system.variables = self.variables.copy()
193
+
194
+ while temp_system.can_solve_any():
195
+ known_vars = temp_system.get_known_variables()
196
+ unknown_vars = temp_system.get_unknown_variables()
197
+
198
+ # Find next solvable variable
199
+ for equation in temp_system.equations:
200
+ for unknown_var in unknown_vars:
201
+ if equation.can_solve_for(unknown_var, known_vars):
202
+ order.append(unknown_var)
203
+ # Mark as known for next iteration
204
+ temp_system.variables[unknown_var].is_known = True
205
+ break
206
+ else:
207
+ continue
208
+ break
209
+
210
+ return order
211
+
212
+ def __str__(self) -> str:
213
+ return f"EquationSystem({len(self.equations)} equations, {len(self.variables)} variables)"
214
+
215
+ def __repr__(self) -> str:
216
+ return f"EquationSystem(equations={self.equations!r})"