Skip to content

Commit aef5d80

Browse files
angranl-flexflow360-auto-hotfix-bot
authored andcommitted
Fix a bug that an extra "velocity" field appears in the UDF section (#1600)
1 parent 479ed89 commit aef5d80

File tree

3 files changed

+12
-30
lines changed

3 files changed

+12
-30
lines changed

flow360/component/simulation/outputs/output_fields.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -315,7 +315,7 @@ def get_unit_for_field(field_name: str):
315315
FIELD_TYPE_SCALAR = "scalar"
316316

317317
_FIELD_TYPE_INFO = {
318-
"velocity": {
318+
"velocity_": {
319319
"type": FIELD_TYPE_3DVECTOR,
320320
},
321321
"velocity_magnitude": {
@@ -346,9 +346,9 @@ def get_unit_for_field(field_name: str):
346346

347347
# Predefined UDF expressions
348348
PREDEFINED_UDF_EXPRESSIONS = {
349-
"velocity": "velocity[0] = primitiveVars[1] * velocityScale;"
350-
+ "velocity[1] = primitiveVars[2] * velocityScale;"
351-
+ "velocity[2] = primitiveVars[3] * velocityScale;",
349+
"velocity_": "velocity_[0] = primitiveVars[1] * velocityScale;"
350+
+ "velocity_[1] = primitiveVars[2] * velocityScale;"
351+
+ "velocity_[2] = primitiveVars[3] * velocityScale;",
352352
"velocity_magnitude": "double velocity[3];"
353353
+ "velocity[0] = primitiveVars[1];"
354354
+ "velocity[1] = primitiveVars[2];"

tests/simulation/outputs/test_output_fields.py

Lines changed: 7 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -19,31 +19,18 @@ def simulation_params():
1919
return params
2020

2121

22-
def test_generate_field_udf_no_unit(simulation_params):
23-
"""Test generating UDF expression for a field without units."""
24-
result = generate_predefined_udf("velocity", simulation_params)
25-
26-
expected = (
27-
"velocity[0] = primitiveVars[1] * velocityScale;"
28-
"velocity[1] = primitiveVars[2] * velocityScale;"
29-
"velocity[2] = primitiveVars[3] * velocityScale;"
30-
)
31-
32-
assert result == expected
33-
34-
3522
def test_generate_field_udf_with_unit(simulation_params):
3623
"""Test generating UDF expression for a field with units."""
3724
result = generate_predefined_udf("velocity_m_per_s", simulation_params)
3825

3926
expected = (
40-
"double velocity[3];"
41-
"velocity[0] = primitiveVars[1] * velocityScale;"
42-
"velocity[1] = primitiveVars[2] * velocityScale;"
43-
"velocity[2] = primitiveVars[3] * velocityScale;"
44-
"velocity_m_per_s[0] = velocity[0] * 340.2940058082124;"
45-
"velocity_m_per_s[1] = velocity[1] * 340.2940058082124;"
46-
"velocity_m_per_s[2] = velocity[2] * 340.2940058082124;"
27+
"double velocity_[3];"
28+
"velocity_[0] = primitiveVars[1] * velocityScale;"
29+
"velocity_[1] = primitiveVars[2] * velocityScale;"
30+
"velocity_[2] = primitiveVars[3] * velocityScale;"
31+
"velocity_m_per_s[0] = velocity_[0] * 340.2940058082124;"
32+
"velocity_m_per_s[1] = velocity_[1] * 340.2940058082124;"
33+
"velocity_m_per_s[2] = velocity_[2] * 340.2940058082124;"
4734
)
4835

4936
assert result == expected

tests/simulation/translator/test_output_translation.py

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1372,18 +1372,13 @@ def test_dimensioned_output_fields_translation(vel_in_km_per_hr):
13721372
"expression": "double pressure_;double gamma = 1.4;pressure_ = (usingLiquidAsMaterial) ? (primitiveVars[4] - 1.0 / gamma) * (velocityScale * velocityScale) : primitiveVars[4];pressure_pa = pressure_ * 10000000.0;",
13731373
"from_user_variables": False,
13741374
},
1375-
{
1376-
"name": "velocity",
1377-
"expression": "velocity[0] = primitiveVars[1] * velocityScale;velocity[1] = primitiveVars[2] * velocityScale;velocity[2] = primitiveVars[3] * velocityScale;",
1378-
"from_user_variables": False,
1379-
},
13801375
{
13811376
"name": "velocity_in_km_per_hr",
13821377
"expression": "double ___velocity[3];___velocity[0] = primitiveVars[1] * velocityScale;___velocity[1] = primitiveVars[2] * velocityScale;___velocity[2] = primitiveVars[3] * velocityScale;velocity_in_km_per_hr[0] = (___velocity[0] * 360.0); velocity_in_km_per_hr[1] = (___velocity[1] * 360.0); velocity_in_km_per_hr[2] = (___velocity[2] * 360.0);",
13831378
},
13841379
{
13851380
"name": "velocity_m_per_s",
1386-
"expression": "double velocity[3];velocity[0] = primitiveVars[1] * velocityScale;velocity[1] = primitiveVars[2] * velocityScale;velocity[2] = primitiveVars[3] * velocityScale;velocity_m_per_s[0] = velocity[0] * 100.0;velocity_m_per_s[1] = velocity[1] * 100.0;velocity_m_per_s[2] = velocity[2] * 100.0;",
1381+
"expression": "double velocity_[3];velocity_[0] = primitiveVars[1] * velocityScale;velocity_[1] = primitiveVars[2] * velocityScale;velocity_[2] = primitiveVars[3] * velocityScale;velocity_m_per_s[0] = velocity_[0] * 100.0;velocity_m_per_s[1] = velocity_[1] * 100.0;velocity_m_per_s[2] = velocity_[2] * 100.0;",
13871382
"from_user_variables": False,
13881383
},
13891384
{

0 commit comments

Comments
 (0)