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 +175 -0
- qnty/dimension.py +113 -14
- qnty/equation.py +216 -0
- qnty/expression.py +480 -0
- qnty/prefixes.py +229 -0
- qnty/unit.py +76 -8
- qnty/unit_types/__init__.py +0 -0
- qnty/unit_types/base.py +47 -0
- qnty/units.py +8078 -33
- qnty/variable.py +69 -29
- qnty/variable_types/__init__.py +0 -0
- qnty/variable_types/base.py +58 -0
- qnty/variable_types/expression_variable.py +68 -0
- qnty/variable_types/typed_variable.py +92 -0
- qnty/variables.py +2291 -19
- qnty/variables.pyi +6097 -0
- {qnty-0.0.1.dist-info → qnty-0.0.3.dist-info}/METADATA +109 -22
- qnty-0.0.3.dist-info/RECORD +19 -0
- qnty/setters.py +0 -89
- qnty-0.0.1.dist-info/RECORD +0 -10
- {qnty-0.0.1.dist-info → qnty-0.0.3.dist-info}/WHEEL +0 -0
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
|
-
|
76
|
-
|
77
|
-
|
78
|
-
|
79
|
-
|
80
|
-
|
81
|
-
|
82
|
-
|
83
|
-
|
84
|
-
|
85
|
-
|
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})"
|