def setUp(self):
# construct MP
self.MP = multiPointSparse(gcomm)
for setName in SET_NAMES:
comm_size = COMM_SIZES[setName]
self.MP.addProcessorSet(setName,
nMembers=len(comm_size),
memberSizes=comm_size)
self.comm, self.setComm, self.setFlags, self.groupFlags, self.ptID = self.MP.createCommunicators(
)
for setName in SET_NAMES:
self.MP.addProcSetObjFunc(setName, SET_FUNC_HANDLES[setName][0])
self.MP.addProcSetSensFunc(setName, SET_FUNC_HANDLES[setName][1])
# construct optProb
optProb = Optimization("multipoint test", self.MP.obj)
for dv in DVS:
optProb.addVar(dv)
optProb.addObj("total_drag")
self.MP.setObjCon(objCon)
self.MP.setOptProb(optProb)
parser.add_argument("--opt",
type=str,
default="SLSQP",
choices=["IPOPT", "SLSQP", "SNOPT"])
parser.add_argument("--gridFile", type=str, default="wing_vol.cgns")
parser.add_argument("--optOptions",
type=ast.literal_eval,
default={},
help="additional optimizer options to be added")
args = parser.parse_args()
# rst args (end)
# ======================================================================
# Create multipoint communication object
# ======================================================================
# rst multipoint (beg)
MP = multiPointSparse(MPI.COMM_WORLD)
MP.addProcessorSet("cruise", nMembers=1, memberSizes=MPI.COMM_WORLD.size)
comm, setComm, setFlags, groupFlags, ptID = MP.createCommunicators()
if not os.path.exists(args.output):
if comm.rank == 0:
os.mkdir(args.output)
# rst multipoint (end)
# ======================================================================
# ADflow Set-up
# ======================================================================
# rst adflow (beg)
aeroOptions = {
# I/O Parameters
"gridFile": args.gridFile,
