NumOpt 0.0.2__tar.gz → 0.0.3__tar.gz

numopt-0.0.3/NumOpt/FSI/tools.py

@@ -0,0 +1,33 @@
+import re
+from pathlib import Path
+from ..cprint import cprint_green
+
+
+def deal_name(nasfile, outfile, simplify=True):
+    def sub_func(match):
+        old_str: list = match.group().split("\n")
+        pshell_name = match.group("name")
+        if simplify:
+            pshell_name = pshell_name.split(":")[0]
+        pshell_id = match.group("id")
+        insert_str = f"$ANSA_NAME_COMMENT;{pshell_id};PSHELL;{pshell_name};"
+        old_str.insert(1, insert_str)
+        new_str = "\n".join(old_str)
+        return new_str
+
+    pattern = r"\$\*\s+Property:\s(?P<name>.*)\nPSHELL\s+(?P<id>\d+)\s+.*\n"
+    pattern = re.compile(pattern)
+
+    content = Path(nasfile).read_text()
+    new_content = re.sub(pattern, sub_func, content)
+    outfile = Path(outfile)
+    if not outfile.parent.exists():
+        outfile.parent.mkdir(exist_ok=True)
+    outfile.write_text(new_content)
+    cprint_green(f"new nastran file `{outfile.as_posix()}` is generated.")
+
+
+
+if __name__ == "__main__":
+    nasfile = "./bulk.dat"
+    deal_name(nasfile, "new_bulk.dat")

numopt-0.0.3/NumOpt/__init__.py

@@ -0,0 +1,4 @@
+from NumOpt.opti import *
+from NumOpt.cprint import *
+
+__version__ = "0.0.3"

numopt-0.0.3/NumOpt/cprint.py

@@ -0,0 +1,9 @@
+from termcolor import cprint
+from functools import partial
+
+
+cprint_green = partial(cprint, color="green", attrs=["bold"])
+cprint_magenta = partial(cprint, color="magenta", attrs=["bold"])
+cprint_blue = partial(cprint, color="blue", attrs=["bold"])
+cprint_red = partial(cprint, color="red", attrs=["bold"])
+cprint_yellow = partial(cprint, color="yellow", attrs=["bold"])

numopt-0.0.3/NumOpt/nx/tools.py

@@ -0,0 +1,89 @@
+import NXOpen
+import NXOpen.Features
+from NXOpen import ParasolidExporter
+
+import json
+import os
+from pathlib import Path
+
+
+def create_af(coords, session: NXOpen.Session):
+    part = session.Parts.Work
+
+    markid = session.SetUndoMark(NXOpen.Session.MarkVisibility.Visible, "create airfoil")
+    splineEX = part.Features.CreateStudioSplineBuilderEx(NXOpen.NXObject.Null)
+    for i in coords:
+        pt = NXOpen.Point3d(i[0] * 1000.0, i[1] * 1000.0, 0.0)
+        pt = part.Points.CreatePoint(pt)
+        gcons = splineEX.ConstraintManager.CreateGeometricConstraintData()
+        gcons.Point = pt
+        splineEX.ConstraintManager.Append(gcons)
+    spline = splineEX.Commit()
+    splineEX.Destroy()
+    session.UpdateManager.DoUpdate(markid)
+
+
+def modify_af(coords, name, session: NXOpen.Session):
+    part = session.Parts.Work
+
+    markid = session.SetUndoMark(NXOpen.Session.MarkVisibility.Visible, "modify airfoil")
+    spline = part.Features.FindObject(name)
+    splineEX = part.Features.CreateStudioSplineBuilderEx(spline)
+    for i, vi in enumerate(coords):
+        gcons = splineEX.ConstraintManager.FindItem(i)
+        pt = gcons.Point
+        pt.SetCoordinates(NXOpen.Point3d(vi[0] * 1000.0, vi[1] * 1000.0, 0.0))
+    splineEX.Evaluate()
+    spline = splineEX.Commit()
+    splineEX.Destroy()
+    session.UpdateManager.DoUpdate(markid)
+
+
+def modify_expr(expr_name: str, value: float, session: NXOpen.Session, unit: str = None, update: bool = False):
+    markid = session.SetUndoMark(NXOpen.Session.MarkVisibility.Visible, f"modify {expr_name}")
+    part = session.Parts.Work
+    expr = part.Expressions.FindObject(expr_name)
+    unit = NXOpen.Unit.Null if unit is None else part.UnitCollection.FindObject(unit)
+    part.Expressions.EditExpressionWithUnits(expr, unit, f"{value:.6f}")
+    if update:
+        session.UpdateManager.DoUpdate(markid)
+
+
+def export_solid(solid_name: list[str], outfile: str, session: NXOpen.Session):
+    markid = session.SetUndoMark(NXOpen.Session.MarkVisibility.Visible, "export parasoild")
+
+    part = session.Parts.Work
+    parasolidExporter: ParasolidExporter = session.DexManager.CreateParasolidExporter()
+    parasolidExporter.InputFile = part.FullPath
+    parasolidExporter.OutputFile = outfile
+
+    parasolidExporter.ExportSelectionBlock.SelectionScope = NXOpen.ObjectSelector.Scope.SelectedObjects
+    for i in solid_name:
+        solid = part.Bodies.FindObject(i)
+        parasolidExporter.ExportSelectionBlock.SelectionComp.Add(solid)
+    parasolidExporter.Commit()
+    session.UpdateManager.DoUpdate(markid)
+
+
+def saveAs(outfile, session: NXOpen.Session, overwrite: False):
+    outfile = Path(outfile)
+    if overwrite:
+        if outfile.exists():
+            os.unlink(outfile)
+    else:
+        if outfile.exists():
+            raise (f"{outfile} has been existed.")
+
+    markid = session.SetUndoMark(NXOpen.Session.MarkVisibility.Visible, f"save as")
+    part = session.Parts.Work
+    part.SaveAs(outfile.as_posix())
+    session.UpdateManager.DoUpdate(markid)
+
+
+if __name__ == "__main__":
+    with open("C:/Users/Zcaic/Desktop/hxy_ppt/coords.json", "r") as fin:
+        data = json.load(fin)
+
+    session = NXOpen.Session.GetSession()
+    # create_af(data["init_coords"],session=session)
+    modify_af(coords=data["init_coords"], name="SPLINE(1)", session=session)

{numopt-0.0.2 → numopt-0.0.3}/NumOpt/opti.py

@@ -1,16 +1,8 @@
 import aerosandbox.numpy as anp
 import aerosandbox as asb 
 import casadi as cas
-from termcolor import cprint
-from functools import partial
 from typing import Callable, Any, Dict
-
-cprint_green = partial(cprint, color="green", attrs=["bold"])
-cprint_magenta = partial(cprint, color="magenta", attrs=["bold"])
-cprint_blue = partial(cprint, color="blue", attrs=["bold"])
-cprint_red = partial(cprint, color="red", attrs=["bold"])
-cprint_yellow = partial(cprint, color="yellow", attrs=["bold"])
-
+from .cprint import cprint_yellow
 
 # def trape(y, x):
 #     y = anp.array(y)

{numopt-0.0.2 → numopt-0.0.3}/NumOpt.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: NumOpt
-Version: 0.0.2
+Version: 0.0.3
 Summary: pti is a Python package that helps you design and optimize engineered systems.
 Home-page: https://github.com/Zcaic/NumOpt.git
 Author: Zcaic

{numopt-0.0.2 → numopt-0.0.3}/NumOpt.egg-info/SOURCES.txt

@@ -3,9 +3,14 @@ README.md
 pyproject.toml
 setup.py
 NumOpt/__init__.py
+NumOpt/cprint.py
 NumOpt/opti.py
 NumOpt.egg-info/PKG-INFO
 NumOpt.egg-info/SOURCES.txt
 NumOpt.egg-info/dependency_links.txt
 NumOpt.egg-info/requires.txt
-NumOpt.egg-info/top_level.txt
+NumOpt.egg-info/top_level.txt
+NumOpt/FSI/__init__.py
+NumOpt/FSI/tools.py
+NumOpt/nx/__init__.py
+NumOpt/nx/tools.py

{numopt-0.0.2 → numopt-0.0.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: NumOpt
-Version: 0.0.2
+Version: 0.0.3
 Summary: pti is a Python package that helps you design and optimize engineered systems.
 Home-page: https://github.com/Zcaic/NumOpt.git
 Author: Zcaic

numopt-0.0.2/NumOpt/__init__.py

@@ -1,3 +0,0 @@
-from NumOpt.opti import *
-
-__version__ = "0.0.2"