test_basic_objects.py

import sys
import os
PROJECT_ROOT = os.path.abspath(os.path.join(
                  os.path.dirname(__file__),
                  os.pardir)
)
sys.path.append(PROJECT_ROOT)
from RFEM.enums import *
from RFEM.initModel import Model
from RFEM.BasicObjects.material import Material
from RFEM.BasicObjects.section import Section
from RFEM.BasicObjects.thickness import Thickness
from RFEM.BasicObjects.node import Node
from RFEM.BasicObjects.line import Line
from RFEM.BasicObjects.member import Member
from RFEM.BasicObjects.surface import Surface
from RFEM.BasicObjects.surfaceSet import SurfaceSet
from RFEM.BasicObjects.opening import Opening
from RFEM.BasicObjects.lineSet import LineSet
from RFEM.BasicObjects.memberSet import MemberSet
from RFEM.BasicObjects.solid import Solid
from RFEM.BasicObjects.solidSet import SolidSet
from RFEM.Calculate.meshSettings import GetModelInfo

if Model.clientModel is None:
    Model()

def test_line_init():

    Model.clientModel.service.delete_all()
    Model.clientModel.service.begin_modification()

    Node(1, 2, 0, 0)
    Node(2, 4, 0, 0)

    Line(1, '1 2')

    Model.clientModel.service.finish_modification()

    line = Line.GetLine(1)

    assert line.no == 1
    assert line.length == 2

def test_line_polyline():

    Model.clientModel.service.delete_all()
    Model.clientModel.service.begin_modification()

    Node(1, 0, 0, 0)
    Node(2, 5, 0, 0)

    Line.Polyline(1, '1 2')

    Model.clientModel.service.finish_modification()

    line = Line.GetLine(1)

    assert line.no == 1
    assert line.length == 5

def test_line_arc():

    Model.clientModel.service.delete_all()
    Model.clientModel.service.begin_modification()

    Node(1, 2, 0, 0)
    Node(2, 4, 0, 0)

    Line.Arc(1, [1, 2], [3, 3, 0], LineArcAlphaAdjustmentTarget.ALPHA_ADJUSTMENT_TARGET_BEGINNING_OF_ARC)

    Model.clientModel.service.finish_modification()

    line = Line.GetLine(1)

    assert line.no == 1
    assert line.type == "TYPE_ARC"

def test_line_circle():

    Model.clientModel.service.delete_all()
    Model.clientModel.service.begin_modification()

    Line.Circle(1, [5, 6, 7], 3, [0, 0, 1])

    Model.clientModel.service.finish_modification()

    line = Line.GetLine(1)

    assert line.no == 1
    assert line.circle_radius == 3
    assert line.circle_center_coordinate_1 == 5
    assert line.circle_center_coordinate_2 == 6
    assert line.circle_center_coordinate_3 == 7

def test_line_ellipse():

    Model.clientModel.service.delete_all()
    Model.clientModel.service.begin_modification()

    Node(1,2,0,0)
    Node(2,-2,0,0)
    Line.Ellipse(2,[1,2],[0,1,0])

    Model.clientModel.service.finish_modification()

    line = Line.GetLine(2)

    assert round(line.length, 4) == 9.6884

def test_line_parabola():

    Model.clientModel.service.delete_all()
    Model.clientModel.service.begin_modification()

    Node(1,2,0,0)
    Node(2,-2,0,0)
    Line.Parabola(3,[1,2],[0,1,0],0.17453)

    Model.clientModel.service.finish_modification()

    line = Line.GetLine(3)

    assert round(line.length, 3) == 4.605

def test_line_spline():

    Model.clientModel.service.delete_all()
    Model.clientModel.service.begin_modification()

    Node(1,2,0,0)
    Node(2,1,1,0)
    Node(3,0,-1,0)
    Node(4,-1,1,0)
    Node(5,-2,0,0)
    Line.Spline(1, "1-5")

    Model.clientModel.service.finish_modification()

    line = Line.GetLine(1)

    assert round(line.length, 4) == 8.42290

def test_line_elipticalArc():

    Model.clientModel.service.delete_all()
    Model.clientModel.service.begin_modification()

    Line.EllipticalArc(1)
    Line.EllipticalArc(2, [2,0,0], [-2,0,0], [0,-3,0], 0.17453, 2.79252)

    Model.clientModel.service.finish_modification()

    line = Line.GetLine(2)
    assert line.type == 'TYPE_ELLIPTICAL_ARC'

def test_line_NURBS():

    Model.clientModel.service.delete_all()
    Model.clientModel.service.begin_modification()

    Node(1,0,0,0)
    Node(2,8,0,0)
    Node(3,3,0,-4)
    Node(4,5,0,4)

    Line.NURBS(1, '1 2', [[0,0,0], [2,0,-2], [4,0,0], [6,0,2], [8,0,0]], [1, 1, 1, 1, 1], 4)
    Line.NURBS(2, '1 3 4 2')

    Model.clientModel.service.finish_modification()
    line1 = Line.GetLine(1)
    line2 = Line.GetLine(2)

    assert line1.nurbs_order == 4
    assert line1.definition_nodes == '1 2'
    assert line2.type == "TYPE_NURBS"
    assert line2.definition_nodes == '1 2'

def test_lineSetInit():

    Model.clientModel.service.delete_all()
    Model.clientModel.service.begin_modification()

    Node(1, 2, 0, 0)
    Node(2, 4, 0, 0)
    Node(3, 6, 0, 0)

    Line(1, '1 2')
    Line(2, '2 3')

    LineSet(1, '1 2')

    Model.clientModel.service.finish_modification()

    line_set = LineSet.GetLineSet(1)

    assert line_set.length == 4

def test_lineSetContinuous():

    Model.clientModel.service.delete_all()
    Model.clientModel.service.begin_modification()

    Node(1, 2, 0, 0)
    Node(2, 4, 0, 0)
    Node(3, 6, 1, 0)

    Line(1, '1 2')
    Line(2, '2 3')

    LineSet.ContinuousLines(1, '1 2')

    Model.clientModel.service.finish_modification()

    line_set = LineSet.GetLineSet(1)

    assert round(line_set.length, 5) == 4.23607

def test_lineSetGroup():

    Model.clientModel.service.delete_all()
    Model.clientModel.service.begin_modification()

    Node(1, 2, 0, 0)
    Node(2, 4, 1, 0)
    Node(3, 6, 0, 0)

    Line(1, '1 2')
    Line(2, '2 3')

    LineSet.GroupOfLines(2, '1 2')

    Model.clientModel.service.finish_modification()

    line_set = LineSet.GetLineSet(2)

    assert round(line_set.length, 6) == 4.472136

def test_line_delete():

    Model.clientModel.service.delete_all()
    Model.clientModel.service.begin_modification()

    Node(1, 0, 0, 0)
    Node(2, 0, 0, -5)
    Node(3, 5, 0, -5)
    Node(4, 5, 0, 0)

    Line(1, '1 2')
    Line(2, '2 3')
    Line(3, '3 4')
    Line(4, '4 1')

    Line.DeleteLine('2 4')

    Model.clientModel.service.finish_modification()

    modelInfo = GetModelInfo()

    assert modelInfo.property_line_count == 2

def test_material():

    Model.clientModel.service.delete_all()
    Model.clientModel.service.begin_modification()

    Material(1, 'S235')

    Model.clientModel.service.finish_modification()

    material = Material.GetMaterial(1)

    assert material.no == 1
    assert material.name == 'S235 | CYS EN 1993-1-1:2009-03'

def test_userDefinedMaterial():

    Model.clientModel.service.delete_all()
    Model.clientModel.service.begin_modification()

    Material.UserDefinedMaterial(2, 'S235JR', MaterialType.TYPE_STEEL, MaterialModel.MODEL_ORTHOTROPIC_3D, 220000000000)

    Model.clientModel.service.finish_modification()

    material = Material.GetMaterial(2)

    assert material.no == 2
    assert material.material_type == 'TYPE_STEEL'
    assert material.material_model == 'MODEL_ORTHOTROPIC_3D'
    assert material.temperature[0][0].row['elasticity_modulus_global'] == 220000000000

def test_node():

    Model.clientModel.service.delete_all()
    Model.clientModel.service.begin_modification()

    Node(1, 2, 0, 0)

    Model.clientModel.service.finish_modification()

    node = Node.GetNode(1)

    assert node.no == 1
    assert node.coordinate_1 == 2

def test_member_init():

     Model.clientModel.service.delete_all()
     Model.clientModel.service.begin_modification()

     Node(1, 0, 0, 0)
     Node(2, 5, 0, 0)
     Material(1, 'S235')
     Section(1, 'IPE 300', 1)
     Member(1,  1, 2, 0, 1, 1)

     Model.clientModel.service.finish_modification()

     member = Member.GetMember(1)

     assert member.analytical_length == 5
     assert member.section_start == 1

def test_member_types():

    Model.clientModel.service.delete_all()
    Model.clientModel.service.begin_modification()

    Node(1, 0, 0, 0)
    Node(2, 1, 0, 0)
    Node(3, 2, 0, 0)
    Node(4, 3, 0, 0)
    Node(5, 4, 0, 0)
    Node(6, 5, 0, 0)
    Node(7, 6, 0, 0)
    Node(8, 7, 0, 0)
    Node(9, 8, 0, 0)
    Node(10, 9, 0, 0)
    Node(11, 10, 0, 0)
    Node(12, 11, 0, 0)
    Node(13, 12, 0, 0)
    Node(14, 13, 0, 0)

    Material(1, 'S235')

    Section(1, 'IPE 300', 1)

    Member.Beam(1, 1, 2)
    Member.Rigid(2, 2, 3)
    Member.Truss(3, 3, 4)
    Member.TrussOnlyN(4, 4, 5)
    Member.Tension(5, 5, 6)
    Member.Compression(6, 6, 7)
    Member.Buckling(7, 7, 8)
    Member.Cable(8, 8, 9)
    Member.DefinableStiffness(9, 9, 10)
    Member.CouplingRigidRigid(10, 10, 11)
    Member.CouplingRigidHinge(11, 11, 12)
    Member.CouplingHingeRigid(12, 12, 13)
    Member.CouplingHingeHinge(13, 13, 14)


    Model.clientModel.service.finish_modification()

    assert Member.GetMember(1).type == MemberType.TYPE_BEAM.name
    assert Member.GetMember(2).type == MemberType.TYPE_RIGID.name
    assert Member.GetMember(3).type == MemberType.TYPE_TRUSS.name
    assert Member.GetMember(4).type == MemberType.TYPE_TRUSS_ONLY_N.name
    assert Member.GetMember(5).type == MemberType.TYPE_TENSION.name
    assert Member.GetMember(6).type == MemberType.TYPE_COMPRESSION.name
    assert Member.GetMember(7).type == MemberType.TYPE_BUCKLING.name
    assert Member.GetMember(8).type == MemberType.TYPE_CABLE.name
    assert Member.GetMember(9).type == MemberType.TYPE_DEFINABLE_STIFFNESS.name
    assert Member.GetMember(10).type == MemberType.TYPE_COUPLING_RIGID_RIGID.name
    assert Member.GetMember(11).type == MemberType.TYPE_COUPLING_RIGID_HINGE.name
    assert Member.GetMember(12).type == MemberType.TYPE_COUPLING_HINGE_RIGID.name
    assert Member.GetMember(13).type == MemberType.TYPE_COUPLING_HINGE_HINGE.name

def test_member_set():

    Model.clientModel.service.delete_all()
    Model.clientModel.service.begin_modification()

    Node(1, 0, 0, 0)
    Node(2, 5, 0, 0)
    Node(3, 10, 0, 0)

    Material(1, 'S235')

    Section(1, 'IPE 300', 1)

    Member(1, 1, 2, 0, 1, 1)
    Member(2, 2, 3, 0, 1, 1)

    MemberSet(1, '1 2', SetType.SET_TYPE_GROUP)

    Model.clientModel.service.finish_modification()

    member_set = MemberSet.GetMemberSet(1)

    assert member_set.members == '1 2'
    assert member_set.length == 10

def test_member_delete():

    Model.clientModel.service.delete_all()
    Model.clientModel.service.begin_modification()

    Node(1, 0, 0, 0)
    Node(2, 5, 0, 0)
    Node(3, 10, 0, 0)

    Material(1, 'S235')

    Section(1, 'IPE 300', 1)

    Member(1, 1, 2, 0, 1, 1)
    Member(2, 2, 3, 0, 1, 1)

    Member.DeleteMember('1')

    Model.clientModel.service.finish_modification()

    modelInfo = GetModelInfo()

    assert modelInfo.property_member_count == 1

def test_opening():

    Model.clientModel.service.delete_all()
    Model.clientModel.service.begin_modification()

    Node(1, 0, 0, 0)
    Node(2, 4, 0, 0)
    Node(3, 0, 4, 0)
    Node(4, 4, 4, 0)

    Node(5, 2, 2, 0)
    Node(6, 3, 2, 0)
    Node(7, 3, 3, 0)
    Node(8, 2, 3, 0)

    Line(1, '1 2')
    Line(2, '2 4')
    Line(3, '4 3')
    Line(4, '3 1')

    Line(5, '5 6')
    Line(6, '5 8')
    Line(7, '8 7')
    Line(8, '7 6')

    Material(1, 'S235')
    Thickness(1, '20 mm', 1, 0.02)

    Surface(1, '1 2 3 4', 1)

    Opening(1, '5 6 7 8')

    Model.clientModel.service.finish_modification()

    opening = Opening.GetOpening(1)

    assert opening.area == 1
    assert opening.center_of_opening_x == 2.5

def test_section():

    Model.clientModel.service.delete_all()
    Model.clientModel.service.begin_modification()

    Material(1, 'S235')
    Section(1, 'IPE 300')

    Model.clientModel.service.finish_modification()

    section = Section.GetSection(1)

    assert section.no == 1
    assert section.name == 'IPE 300 | -- | British Steel'

## Solid Class should be updated.

## SolidSet Class should be updated.

## Surface Class should be update. Thickness no can't be assigned.

def test_solid():

    Model.clientModel.service.delete_all()
    Model.clientModel.service.begin_modification()

    Node(1, 0.0, 0.0, 0.0)
    Node(2, 10.0, 0.0, 0.0)
    Node(3, 10.0, 10.0, 0.0)
    Node(4, 0.0, 10.0, 0.0)

    Node(5, 0.0, 0.0, -5.0)
    Node(6, 10.0, 0.0, -5.0)
    Node(7, 10.0, 10.0, -5.0)
    Node(8, 0.0, 10.0, -5.0)

    Line(1, '1 2')
    Line(2, '2 3')
    Line(3, '3 4')
    Line(4, '4 1')

    Line(5, '5 6')
    Line(6, '6 7')
    Line(7, '7 8')
    Line(8, '8 5')

    Line(9, '1 5')
    Line(10, '2 6')
    Line(11, '3 7')
    Line(12, '4 8')

    Surface(1, '1-4', 1)
    Surface(2, '5-8', 1)
    Surface(3, '1 9 5 10', 1)
    Surface(4, '2 10 6 11', 1)
    Surface(5, '3 11 7 12', 1)
    Surface(6, '4 12 8 9', 1)

    Solid(1, '1 2 3 4 5 6', 1)

    Model.clientModel.service.finish_modification()

    solid = Solid.GetSolid(1)

    assert solid['no'] ==  1
    assert solid['surface_area'] == 400.0

def test_solid_set():

    Model.clientModel.service.delete_all()
    Model.clientModel.service.begin_modification()

    # Solid 1
    Node(9, 0.0, 20.0, 0.0)
    Node(10, 10.0, 20.0, 0.0)
    Node(11, 10.0, 30.0, 0.0)
    Node(12, 0.0, 30.0, 0.0)

    Node(13, 0.0, 20.0, -5.0)
    Node(14, 10.0, 20.0, -5.0)
    Node(15, 10.0, 30.0, -5.0)
    Node(16, 0.0, 30.0, -5.0)

    Line(13, '9 10')
    Line(14, '10 11')
    Line(15, '11 12')
    Line(16, '12 9')

    Line(17, '13 14')
    Line(18, '14 15')
    Line(19, '15 16')
    Line(20, '16 13')

    Line(21, '9 13')
    Line(22, '10 14')
    Line(23, '11 15')
    Line(24, '12 16')

    Surface(7, '13-16', 1)
    Surface(8, '17-20', 1)
    Surface(9, '13 22 17 21', 1)
    Surface(10, '15 23 19 24', 1)
    Surface(11, '16 21 20 24', 1)
    Surface(12, '14 22 18 23', 1)

    # Solid 2
    Node(17, 20.0, 20.0, 0.0)
    Node(18, 20.0, 30.0, 0.0)
    Node(19, 20.0, 20.0, -5.0)
    Node(20, 20.0, 30.0, -5.0)

    Line(25, '10 17')
    Line(26, '17 18')
    Line(27, '18 11')
    Line(28, '14 19')
    Line(29, '19 20')
    Line(30, '20 15')
    Line(31, '17 19')
    Line(32, '18 20')

    Surface(13, '25 26 27 14', 1)
    Surface(14, '28 29 30 18', 1)
    Surface(15, '25 31 28 22', 1)
    Surface(16, '27 32 30 23', 1)
    Surface(17, '26 31 29 32', 1)

    Solid(1, '7 8 9 10 11 12', 1 )
    Solid(2, '13 14 15 16 17 12', 1 )

    SolidSet(1, '1 2')

    Model.clientModel.service.finish_modification()

    solidset = SolidSet.GetSolidSet(1)

    assert solidset['no'] ==  1
    assert solidset['solids'] == '1 2'

def test_surface():

    Model.clientModel.service.delete_all()
    Model.clientModel.service.begin_modification()

    Node(1, 0, 0, 0)
    Node(2, 4, 0, 0)
    Node(3, 0, 4, 0)
    Node(4, 4, 4, 0)

    Line(1, '1 2')
    Line(2, '2 4')
    Line(3, '4 3')
    Line(4, '3 1')

    Material(1, 'S235')
    Thickness(1, '20 mm', 1, 0.02)

    Surface(1, '1 2 3 4', 1)

    Model.clientModel.service.finish_modification()

    surface = Surface.GetSurface(1)

    assert surface['no'] ==  1
    assert surface['boundary_lines'] == '1 2 3 4'

def test_surface_set():

    Model.clientModel.service.delete_all()
    Model.clientModel.service.begin_modification()

    Node(1, 0, 0, 0)
    Node(2, 4, 0, 0)
    Node(3, 0, 4, 4)
    Node(4, 4, 4, 4)

    Line(1, '1 2')
    Line(2, '2 4')
    Line(3, '4 3')
    Line(4, '3 1')

    Surface(1, '1 2 3 4', 1)

    Node(7, 0, 4, 0)
    Node(8, 4, 4, 0)

    Line(6, '2 8')
    Line(7, '8 7')
    Line(8, '7 1')

    Surface(2, '1 6 7 8', 1)
    SurfaceSet(1, '1 2')

    Model.clientModel.service.finish_modification()

    surfaceset = SurfaceSet.GetSurfaceSet(1)

    assert surfaceset['no'] ==  1
    assert surfaceset['surfaces'] == '1 2'

def test_thickness_init():

    Model.clientModel.service.delete_all()
    Model.clientModel.service.begin_modification()

    Material(1, 'S235')

    Thickness(1, '20 mm', 1, 0.02)

    Model.clientModel.service.finish_modification()

    thickness = Thickness.GetThickness(1)

    assert thickness.type == "TYPE_UNIFORM"
    assert thickness.uniform_thickness == 0.02

def test_thickness_uniform():

    Model.clientModel.service.delete_all()
    Model.clientModel.service.begin_modification()

    Material(1, 'S235')

    Thickness.Uniform(1, '1', 1, [0.03])

    Model.clientModel.service.finish_modification()

    thickness = Thickness.GetThickness(1)

    assert thickness.type == "TYPE_UNIFORM"
    assert thickness.uniform_thickness == 0.03

def test_thickness_3nodes():

    Model.clientModel.service.delete_all()
    Model.clientModel.service.begin_modification()

    Material(1, 'S235')

    Node(1, 4, 0, 0)
    Node(2, 2, 2, 0)
    Node(3, 3, 4, 0)

    Thickness.Variable_3Nodes(1, '2', 1, [0.2, 1, 0.1, 2, 0.05, 3])

    Model.clientModel.service.finish_modification()

    thickness = Thickness.GetThickness(1)

    assert thickness.type == "TYPE_VARIABLE_THREE_NODES"
    assert thickness.thickness_1 == 0.2

def test_thickness_2nodes():

    Model.clientModel.service.delete_all()
    Model.clientModel.service.begin_modification()

    Material(1, 'S235')

    Node(1, 4, 0, 0)
    Node(2, 2, 2, 0)

    Thickness.Variable_2NodesAndDirection(1, '3', 1, [0.2, 1, 0.1, 2, ThicknessDirection.THICKNESS_DIRECTION_IN_X])
    Model.clientModel.service.finish_modification()

    thickness = Thickness.GetThickness(1)

    assert thickness.type == "TYPE_VARIABLE_TWO_NODES_AND_DIRECTION"
    assert thickness.thickness_2 == 0.1

def test_thickness_4corners():

    Model.clientModel.service.delete_all()
    Model.clientModel.service.begin_modification()

    Material(1, 'S235')

    Node(1, 0, 0, 0)
    Node(2, 3, 0, 0)
    Node(3, 3, 3, 0)
    Node(4, 0, 3, 0)

    Thickness.Variable_4SurfaceCorners(1, '4', 1, [0.2, 1, 0.15, 2, 0.1, 3, 0.05, 4])

    Model.clientModel.service.finish_modification()

    thickness = Thickness.GetThickness(1)

    assert thickness.type == "TYPE_VARIABLE_FOUR_SURFACE_CORNERS"
    assert thickness.thickness_2 == 0.15

def test_thickness_circle():

    Model.clientModel.service.delete_all()
    Model.clientModel.service.begin_modification()

    Material(1, 'S235')

    Thickness.Variable_Circle(1, '5', 1, [0.2, 0.1])

    Model.clientModel.service.finish_modification()

    thickness = Thickness.GetThickness(1)

    assert thickness.type == "TYPE_VARIABLE_CIRCLE"
    assert thickness.thickness_circle_line == 0.1

def test_thickness_shape_orthotropy():

    Model.clientModel.service.reset()
    Model.clientModel.service.begin_modification()

    Material(1, 'S275')

    Thickness.ShapeOrthotropy(1, 'EFFECTIVE_THICKNESS', 1, ThicknessOrthotropyType.EFFECTIVE_THICKNESS, 7)
    Thickness.ShapeOrthotropy(2, 'COUPLING', 1, ThicknessOrthotropyType.COUPLING, 8, parameters=[0.15,0.18,0.16])
    Thickness.ShapeOrthotropy(3, 'UNIDIRECTIONAL_RIBBED_PLATE', 1, ThicknessOrthotropyType.UNIDIRECTIONAL_RIBBED_PLATE, 9, parameters=[0.15, 0.15,0.18,0.16])
    Thickness.ShapeOrthotropy(4, 'BIDIRECTIONAL_RIBBED_PLATE', 1, ThicknessOrthotropyType.BIDIRECTIONAL_RIBBED_PLATE, 10, parameters=[0.15, 0.3,0.25, 0.8,0.75,0.2,0.15])
    Thickness.ShapeOrthotropy(5, 'TRAPEZOIDAL_SHEET', 1, ThicknessOrthotropyType.TRAPEZOIDAL_SHEET, 11, parameters=[0.15, 0.45,0.4,0.14,0.14])
    Thickness.ShapeOrthotropy(6, 'HOLLOW_CORE_SLAB', 1, ThicknessOrthotropyType.HOLLOW_CORE_SLAB, 12, parameters=[0.2, 0.15, 0.1])
    Thickness.ShapeOrthotropy(7, 'GRILLAGE', 1, ThicknessOrthotropyType.GRILLAGE, 13,[ThicknessShapeOrthotropySelfWeightDefinitionType.SELF_WEIGHT_DEFINED_VIA_FICTITIOUS_THICKNESS, 0.15], [0.15,0.8,0.75,0.2,0.15])

    Model.clientModel.service.finish_modification()

def test_get_thickness():

    Model.clientModel.service.delete_all()
    Model.clientModel.service.begin_modification()

    Material(1, 'S235')

    Thickness(1, '20 mm', 1, 0.02)

    Model.clientModel.service.finish_modification()

    thickness = Thickness.GetThickness(1)

    assert thickness['type'] == "TYPE_UNIFORM"
    assert thickness['uniform_thickness'] == 0.02