def __init__(self,
fileName,
comm=MPI.COMM_WORLD,
scale=1.0,
comps=[],
projTol=0.01):
if comm.rank == 0:
print("Initializing DVGeometryVSP")
t0 = time.time()
super().__init__(fileName=fileName,
comm=comm,
scale=scale,
projTol=projTol)
self.exportComps = []
# Clear the vsp model
openvsp.ClearVSPModel()
t1 = time.time()
# read the model
openvsp.ReadVSPFile(fileName)
t2 = time.time()
if self.comm.rank == 0:
print("Loading the vsp model took:", (t2 - t1))
# List of all componets returned from VSP. Note that this
# order is important. It is the order that we use to map the
# actual geom_id by using the geom_names
allComps = openvsp.FindGeoms()
allNames = []
for c in allComps:
allNames.append(openvsp.GetContainerName(c))
if not comps:
# no components specified, we use all
self.allComps = allComps[:]
else:
# we get the vsp comp IDs from the comps list
self.allComps = []
for c in comps:
self.allComps.append(allComps[allNames.index(c)])
# we need the names and bounding boxes of components
self.compNames = []
self.bbox = OrderedDict()
self.bboxuv = self._getuv()
for c in self.allComps:
self.compNames.append(openvsp.GetContainerName(c))
self.bbox[c] = self._getBBox(c)
# Now, we need to form our own quad meshes for fast projections
if comm.rank == 0:
print("Building a quad mesh for fast projections.")
self._getQuads()
if comm.rank == 0:
t3 = time.time()
print("Initialized DVGeometry VSP in", (t3 - t0), "seconds.")
def test_plate_degen_geom(self):
vsp.VSPRenew()
vsp.ClearVSPModel()
prop_id = vsp.AddGeom("PROP")
vsp.Update()
dg_mgr = vsp.run_degen_geom(set_index=vsp.SET_ALL)
dg_mgr.degen_objs[prop_id].plot_copies_plates()
plt.savefig(os.path.join(self.OUTPUT_FOLDER, "plates.png"), dpi=300)
def test_get_prop_info(self):
import matplotlib.pyplot as plt
vsp.VSPRenew()
vsp.ClearVSPModel()
prop_id = vsp.AddGeom("PROP")
valid_location = np.array([-30.0, 60.0, 0.0]).reshape((3, 1))
vsp.SetParmVal(prop_id, "Y_Rel_Location", "XForm", valid_location[1,
0])
vsp.SetParmVal(prop_id, "X_Rel_Location", "XForm", valid_location[0,
0])
vsp.SetParmVal(prop_id, "Y_Rel_Rotation", "XForm", 0.0)
vsp.SetParmVal(prop_id, "X_Rel_Rotation", "XForm", 0.0)
vsp.SetParmVal(prop_id, "Sym_Planar_Flag", "Sym",
vsp.SYM_XZ | vsp.SYM_YZ)
valid_location_sym = np.array(valid_location)
valid_location_sym[1] *= -1.0
valid_location_aft = np.array(valid_location)
valid_location_aft[0] *= -1.0
valid_location_aft_sym = np.array(valid_location_sym)
valid_location_aft_sym[0] *= -1.0
vsp.Update()
prop_info = vsp.get_propeller_thrust_vectors(vsp.SET_ALL)
# Plot prop objects
vsp.plot_propeller_info(prop_info, vector_scale=30.0, markersize=5)
plt.draw()
plt.savefig(os.path.join(self.OUTPUT_FOLDER, "prop_plots.png"),
dpi=300)
valid_thrust_dir = np.array([-1.0, 0.0, 0.0]).reshape((3, 1))
valid_thrust_dir_aft_props = np.array([1.0, 0.0, 0.0]).reshape((3, 1))
# Check thrust vectors
npt.assert_allclose(prop_info[0].thrust_vector, valid_thrust_dir)
npt.assert_allclose(prop_info[1].thrust_vector, valid_thrust_dir)
npt.assert_allclose(prop_info[2].thrust_vector,
valid_thrust_dir_aft_props)
npt.assert_allclose(prop_info[3].thrust_vector,
valid_thrust_dir_aft_props)
# Check origins
npt.assert_allclose(prop_info[0].hub_center, valid_location)
npt.assert_allclose(prop_info[1].hub_center, valid_location_sym)
npt.assert_allclose(prop_info[2].hub_center, valid_location_aft)
npt.assert_allclose(prop_info[3].hub_center, valid_location_aft_sym)
# Check rotation directions
self.assertTrue(prop_info[0].rotation_direction == 1)
self.assertTrue(prop_info[1].rotation_direction == -1)
self.assertTrue(prop_info[2].rotation_direction == -1)
self.assertTrue(prop_info[3].rotation_direction == 1)
def test_surface_patches(self):
vsp.VSPRenew()
vsp.ClearVSPModel()
vsp.AddGeom("PROP")
vsp.Update()
components = vsp.export_surface_patches(vsp.SET_ALL,
remove_degenerate=True)
vsp.plot_surface_components(components)
def test_copy_paste_ids(self):
vsp.VSPRenew()
vsp.ClearVSPModel()
prop_id1 = vsp.AddGeom("PROP")
vsp.Update()
vsp.CopyGeomToClipboard(prop_id1)
prop_id2 = vsp.PasteGeomClipboard()
self.assertTrue(prop_id1 != prop_id2[0])
def test_surface_patches(self):
vsp.VSPRenew()
vsp.ClearVSPModel()
vsp.AddGeom("PROP")
vsp.Update()
components = vsp.export_surface_patches(vsp.SET_ALL,
remove_degenerate=True)
vsp.plot_surface_components(components)
plt.savefig(os.path.join(self.OUTPUT_FOLDER, "surface_patches.png"),
dpi=300)
self.assertTrue(True)
def test_parse_degen_geom(self):
vsp.VSPRenew()
vsp.ClearVSPModel()
wing_id = vsp.AddGeom("WING")
vsp.Update()
vsp.SetIntAnalysisInput("DegenGeom", "WriteCSVFlag", [0], 0)
vsp.SetIntAnalysisInput("DegenGeom", "WriteMFileFlag", [0], 0)
degen_results_id = vsp.ExecAnalysis("DegenGeom")
for degen_obj in vsp.parse_degen_geom(degen_results_id):
print(degen_obj)
def test_simple_prop_degen(self):
vsp.VSPRenew()
vsp.ClearVSPModel()
prop_id = vsp.AddGeom("PROP")
vsp.Update()
print(vsp.GetAnalysisInputNames("DegenGeom"))
vsp.SetAnalysisInputDefaults("DegenGeom")
vsp.PrintAnalysisInputs("DegenGeom")
vsp.SetIntAnalysisInput("DegenGeom", "WriteCSVFlag", [0], 0)
vsp.SetIntAnalysisInput("DegenGeom", "WriteMFileFlag", [0], 0)
vsp.PrintAnalysisInputs("DegenGeom")
self.assertTrue(True)
def create_geom():
vsp.ClearVSPModel()
vsp.DeleteAllResults()
wing_id = vsp.AddGeom("WING")
rotor_id = vsp.AddGeom("PROP")
vsp.SetParmVal(wing_id, "X_Rel_Location", "XForm", 10.0)
vsp.Update()
dg_mgr = vsp.run_degen_geom(set_index=vsp.SET_ALL)
vsp.WriteVSPFile(os.path.join(path, "testfile.vsp3"))
return dg_mgr, wing_id, rotor_id
def test_parasite_drag(self):
import numpy as np
vsp.VSPRenew()
vsp.ClearVSPModel()
wing_id = vsp.AddGeom("WING")
pod_id = vsp.AddGeom("POD")
sref = vsp.GetParmVal(wing_id, "TotalArea", "WingGeom")
res = vsp.parasitedrag_sweep(speeds=np.linspace(10, 250, 10),
alts_ft=[0, 10000, 20000, 50000],
sref=sref,
length_unit=vsp.LEN_FT,
speed_unit=vsp.V_UNIT_MPH)
res.plot()
def genX3d(file=None, set=vsp.SET_ALL, dims=[1000, 400], **kwargs):
if file is not None:
vsp.ClearVSPModel()
vsp.ReadVSPFile(file)
vsp.Update()
with RunManager(**kwargs):
vsp.ExportFile("prop.x3d", set, vsp.EXPORT_X3D)
with open("prop.x3d", "r") as f:
x3d_str = f.read()
x3d_str = x3d_str[0:26] + " width=\"{}px\" height=\"{}px\"".format(
dims[0], dims[1]) + x3d_str[26:-1]
return x3d_str
def test_double_mat(self):
vsp.VSPRenew()
vsp.ClearVSPModel()
pod_id = vsp.AddGeom("POD")
vsp.Update()
degen_results_id = vsp.ExecAnalysis("DegenGeom")
degen_ids = vsp.GetStringResults(degen_results_id, "Degen_DegenGeoms")
for degen_id in degen_ids:
surf_id = vsp.GetStringResults(degen_id, "surf", 0)[0]
vsp.PrintResults(surf_id)
areas = vsp.GetDoubleMatResults(surf_id, "area")
print(areas[0][0])
self.assertTrue(True)
def test_degen_transform_mat(self):
import degen_geom as dg
vsp.VSPRenew()
vsp.ClearVSPModel()
prop_id = vsp.AddGeom("PROP")
vsp.SetParmVal(prop_id, "Y_Rel_Location", "XForm", 60.0)
vsp.SetParmVal(prop_id, "Y_Rel_Rotation", "XForm", 90.0)
vsp.SetParmVal(prop_id, "X_Rel_Rotation", "XForm", 10.0)
vsp.SetParmVal(prop_id, "Sym_Planar_Flag", "Sym", vsp.SYM_XZ)
vsp.Update()
vsp.SetIntAnalysisInput("DegenGeom", "WriteCSVFlag", [0], 0)
vsp.SetIntAnalysisInput("DegenGeom", "WriteMFileFlag", [0], 0)
degen_results_id = vsp.ExecAnalysis("DegenGeom")
degen_objects = vsp.parse_degen_geom(degen_results_id)
degen_mgr = dg.DegenGeomMgr(degen_objects)
dg_prop_comp = degen_mgr.degen_objs[prop_id]
orig_copy = dg_prop_comp.copies[0]
origin = orig_copy[0].transmat.get_translations()
self.assertAlmostEqual(origin[0], 0.0, places=6)
self.assertAlmostEqual(origin[1], 60.0, places=6)
self.assertAlmostEqual(origin[2], 0.0, places=6)
angles = orig_copy[0].transmat.get_angles()
self.assertAlmostEqual(angles[0], 10.0, places=6)
self.assertAlmostEqual(angles[1], 90.0, places=6)
self.assertAlmostEqual(angles[2], 0.0, places=6)
sym_copy = dg_prop_comp.copies[1]
origin = sym_copy[0].transmat.get_translations()
self.assertAlmostEqual(origin[0], 0.0, places=6)
self.assertAlmostEqual(origin[1], -60.0, places=6)
self.assertAlmostEqual(origin[2], 0.0, places=6)
angles = sym_copy[0].transmat.get_angles()
self.assertAlmostEqual(angles[0], -180.0 + 10.0, places=6)
self.assertAlmostEqual(angles[1], 90.0, places=6)
self.assertAlmostEqual(angles[2], 0.0, places=6)
self.assertTrue(True)
def test_plate_degen_geom_area(self):
vsp.VSPRenew()
vsp.ClearVSPModel()
prop_id = vsp.AddGeom("PROP")
vsp.Update()
dg_mgr = vsp.run_degen_geom(set_index=vsp.SET_ALL)
prop_dg = dg_mgr.degen_objs[prop_id]
for comp in prop_dg.copies.values():
for surf in comp:
for plate in surf.plates:
area_brute_force = plate._compute_areas_brute_force().sum()
area_vectorized = plate.compute_areas().sum()
self.assertAlmostEqual(area_brute_force,
area_vectorized,
delta=1.0e-1)
def test_parasite_drag(self):
import numpy as np
vsp.VSPRenew()
vsp.ClearVSPModel()
wing_id = vsp.AddGeom("WING")
pod_id = vsp.AddGeom("POD")
sref = vsp.GetParmVal(wing_id, "TotalArea", "WingGeom")
res = vsp.parasitedrag_sweep(speeds=np.linspace(10, 250, 10),
alts_ft=[0, 10000, 20000, 50000],
sref=sref,
length_unit=vsp.LEN_FT,
speed_unit=vsp.V_UNIT_MPH)
ax = res.plot()
plt.legend()
plt.savefig(os.path.join(self.OUTPUT_FOLDER, "parasite.png"), dpi=300)
self.assertTrue(True)
def test_2(self, train=False, refDeriv=False):
"""
Test 2: OpenVSP wing test
"""
# we skip parallel tests for now
if not train and self.N_PROCS > 1:
self.skipTest("Skipping the parallel test for now.")
def sample_uv(nu, nv):
# function to create sample uv from the surface and save these points.
u = np.linspace(0, 1, nu + 1)
v = np.linspace(0, 1, nv + 1)
uu, vv = np.meshgrid(u, v)
# print (uu.flatten(), vv.flatten())
uv = np.array((uu.flatten(), vv.flatten()))
return uv
refFile = os.path.join(self.base_path, "ref/test_DVGeometryVSP_02.ref")
with BaseRegTest(refFile, train=train) as handler:
handler.root_print("Test 2: OpenVSP NACA 0012 wing")
vspFile = os.path.join(self.base_path,
"../../input_files/naca0012.vsp3")
DVGeo = DVGeometryVSP(vspFile)
dh = 1e-6
openvsp.ClearVSPModel()
openvsp.ReadVSPFile(vspFile)
geoms = openvsp.FindGeoms()
DVGeo = DVGeometryVSP(vspFile)
comp = "WingGeom"
# loop over sections
# normally, there are 9 sections so we should loop over range(9) for the full test
# to have it run faster, we just pick 2 sections
for i in [0, 5]:
# Twist
DVGeo.addVariable(comp,
"XSec_%d" % i,
"Twist",
lower=-10.0,
upper=10.0,
scale=1e-2,
scaledStep=False,
dh=dh)
# loop over coefs
# normally, there are 7 coeffs so we should loop over range(7) for the full test
# to have it run faster, we just pick 2 sections
for j in [0, 4]:
# CST Airfoil shape variables
group = "UpperCoeff_%d" % i
var = "Au_%d" % j
DVGeo.addVariable(comp,
group,
var,
lower=-0.1,
upper=0.5,
scale=1e-3,
scaledStep=False,
dh=dh)
group = "LowerCoeff_%d" % i
var = "Al_%d" % j
DVGeo.addVariable(comp,
group,
var,
lower=-0.5,
upper=0.1,
scale=1e-3,
scaledStep=False,
dh=dh)
# now lets generate ourselves a quad mesh of these cubes.
uv_g = sample_uv(8, 8)
# total number of points
ntot = uv_g.shape[1]
# rank on this proc
rank = MPI.COMM_WORLD.rank
# first, equally divide
nuv = ntot // MPI.COMM_WORLD.size
# then, add the remainder
if rank < ntot % MPI.COMM_WORLD.size:
nuv += 1
# allocate the uv array on this proc
uv = np.zeros((2, nuv))
# print how mant points we have
MPI.COMM_WORLD.Barrier()
# loop over the points and save all that this proc owns
ii = 0
for i in range(ntot):
if i % MPI.COMM_WORLD.size == rank:
uv[:, ii] = uv_g[:, i]
ii += 1
# get the coordinates
nNodes = len(uv[0, :])
openvsp.CompVecPnt01(geoms[0], 0, uv[0, :], uv[1, :])
# extract node coordinates and save them in a numpy array
coor = np.zeros((nNodes, 3))
for i in range(nNodes):
pnt = openvsp.CompPnt01(geoms[0], 0, uv[0, i], uv[1, i])
coor[i, :] = (pnt.x(), pnt.y(), pnt.z())
# Add this pointSet to DVGeo
DVGeo.addPointSet(coor, "test_points")
# We will have nNodes*3 many functions of interest...
dIdpt = np.zeros((nNodes * 3, nNodes, 3))
# set the seeds to one in the following fashion:
# first function of interest gets the first coordinate of the first point
# second func gets the second coord of first point etc....
for i in range(nNodes):
for j in range(3):
dIdpt[i * 3 + j, i, j] = 1
# first get the dvgeo result
funcSens = DVGeo.totalSensitivity(dIdpt.copy(), "test_points")
# now perturb the design with finite differences and compute FD gradients
DVs = DVGeo.getValues()
funcSensFD = {}
for x in DVs:
# perturb the design
xRef = DVs[x].copy()
DVs[x] += dh
DVGeo.setDesignVars(DVs)
# get the new points
coorNew = DVGeo.update("test_points")
# calculate finite differences
funcSensFD[x] = (coorNew.flatten() - coor.flatten()) / dh
# set back the DV
DVs[x] = xRef.copy()
# now loop over the values and compare
# when this is run with multiple procs, VSP sometimes has a bug
# that leads to different procs having different spanwise
# u-v distributions. as a result, the final values can differ up to 1e-5 levels
# this issue does not come up if this tests is ran with a single proc
biggest_deriv = 1e-16
for x in DVs:
err = np.array(funcSens[x].squeeze()) - np.array(funcSensFD[x])
maxderiv = np.max(np.abs(funcSens[x].squeeze()))
normalizer = np.median(np.abs(funcSensFD[x].squeeze()))
if np.abs(normalizer) < 1:
normalizer = np.ones(1)
normalized_error = err / normalizer
if maxderiv > biggest_deriv:
biggest_deriv = maxderiv
handler.assert_allclose(normalized_error,
0.0,
name=f"{x}_grad_normalized_error",
rtol=1e0,
atol=5e-5)
# make sure that at least one derivative is nonzero
self.assertGreater(biggest_deriv, 0.005)
def test_degen_geom(self):
# Test analysis manager
vsp.VSPRenew()
vsp.ClearVSPModel()
# Print all types
print(vsp.GetGeomTypes())
prop_id = vsp.AddGeom("PROP")
blank_id = vsp.AddGeom("BLANK")
disk_id = vsp.AddGeom("Disk")
vsp.SetParmVal(blank_id, "Point_Mass_Flag", "Mass", 1)
vsp.SetParmVal(blank_id, "Point_Mass", "Mass", 5.0)
wing_id = vsp.AddGeom("WING")
pod_id = vsp.AddGeom("POD")
vsp.AddSubSurf(pod_id, vsp.SS_RECTANGLE)
vsp.AddSubSurf(wing_id, vsp.SS_CONTROL)
vsp.Update()
# Run Degen Geom
print(vsp.FindGeoms())
print(vsp.GetAnalysisInputNames("DegenGeom"))
vsp.SetAnalysisInputDefaults("DegenGeom")
vsp.SetIntAnalysisInput("DegenGeom", "WriteMFileFlag", [0], 0)
vsp.SetIntAnalysisInput("DegenGeom", "WriteCSVFlag", [0], 0)
vsp.PrintAnalysisInputs("DegenGeom")
degen_results_id = vsp.ExecAnalysis("DegenGeom")
print(vsp.GetAllResultsNames())
vsp.PrintResults(degen_results_id)
blank_ids = vsp.GetStringResults(degen_results_id, "Degen_BlankGeoms")
degen_ids = vsp.GetStringResults(degen_results_id, "Degen_DegenGeoms")
for blank_id in blank_ids:
vsp.PrintResults(blank_id)
for degen_id in degen_ids:
vsp.PrintResults(degen_id)
t = vsp.GetStringResults(degen_id, "type", 0)[0]
if t == "DISK":
disk_id = vsp.GetStringResults(degen_id, "disk", 0)[0]
vsp.PrintResults(disk_id)
surf_id = vsp.GetStringResults(degen_id, "surf", 0)[0]
vsp.PrintResults(surf_id)
areas = vsp.GetDoubleMatResults(surf_id, "area")
plate_ids = vsp.GetStringResults(degen_id, "plates")
for plate_id in plate_ids:
vsp.PrintResults(plate_id)
stick_ids = vsp.GetStringResults(degen_id, "sticks")
for stick_id in stick_ids:
vsp.PrintResults(stick_id)
if t != "DISK":
point_id = vsp.GetStringResults(degen_id, "point")[0]
vsp.PrintResults(point_id)
subsurf_ids = vsp.GetStringResults(degen_id, "subsurfs")
for ss_id in subsurf_ids:
vsp.PrintResults(ss_id)
hinge_ids = vsp.GetStringResults(degen_id, "hinges")
for hinge_id in hinge_ids:
vsp.PrintResults(hinge_id)
self.assertTrue(True)
