def _get_P(self, nP, airfoil): P = numpy.zeros((airfoil.shape[0],4,3),order='F') for j in range(4): P[:,j,:2] = airfoil[:,:] P[:,j,2] = j bse = BSEmodel([P]) bse.set_bspline_option('num_pt', 0, 'u', airfoil.shape[0]) bse.set_bspline_option('num_cp', 0, 'u', int(airfoil.shape[0]/3)) bse.set_bspline_option('num_pt', 0, 'v', 4) bse.set_bspline_option('num_cp', 0, 'v', 4) bse.assemble() cp = bse.vec['cp_str'].array pt = bse.vec['pt_str'].array jac = bse.jac['d(pt_str)/d(cp_str)'] bse.vec['pt_str'].surfs[0][:, :, :] = P mtx = jac.T.dot(jac) rhs = jac.T.dot(pt) for dim in xrange(3): cp[:, dim] = scipy.sparse.linalg.cg(mtx, rhs[:, dim])[0] fit = numpy.array(cp) bse.set_bspline_option('num_pt', 0, 'u', nP) bse.assemble() cp = bse.vec['cp_str'].array pt = bse.vec['pt_str'].array cp[:, :] = fit[:, :] bse.apply_jacobian('pt_str', 'd(pt_str)/d(cp_str)', 'cp_str') return bse.vec['pt_str'].surfs[0][:, 0, :]
def _get_P(self, nP, airfoils, name, bunch_LE, bunch_TE): def bunch_start(ind, a): ind[:] = ind**a def bunch_end(ind, a): ind[:] = 1 - (1-ind)**a airfoil = airfoils[name] P = numpy.zeros((airfoil.shape[0],4,3),order='F') for j in range(4): P[:,j,:2] = airfoil[:,:] P[:,j,2] = j bse = BSEmodel([P]) bse.set_bspline_option('num_pt', 0, 'u', airfoil.shape[0]) bse.set_bspline_option('num_cp', 0, 'u', int(airfoil.shape[0]/3)) bse.set_bspline_option('num_pt', 0, 'v', 4) bse.set_bspline_option('num_cp', 0, 'v', 4) bse.assemble() cp = bse.vec['cp_str'].array pt = bse.vec['pt_str'].array jac = bse.jac['d(pt_str)/d(cp_str)'] bse.vec['pt_str'].surfs[0][:, :, :] = P mtx = jac.T.dot(jac) rhs = jac.T.dot(pt) for dim in range(3): cp[:, dim] = scipy.sparse.linalg.cg(mtx, rhs[:, dim])[0] fit = numpy.array(cp) bse.set_bspline_option('num_pt', 0, 'u', nP) bse.assemble() cp = bse.vec['cp_str'].array pt = bse.vec['pt_str'].array cp[:, :] = fit[:, :] surfs = numpy.zeros(nP).astype(int) ind_u = numpy.linspace(0, 1, nP) if name == 'upp': bunch_start(ind_u, bunch_TE) bunch_end(ind_u, bunch_LE) elif name == 'low': bunch_start(ind_u, bunch_LE) bunch_end(ind_u, bunch_TE) ind_v = numpy.zeros(nP) bse.add_jacobian('new', surfs, ind_u, ind_v, 3) bse.apply_jacobian('new', 'd(new)/d(cp_str)', 'cp_str') return bse.vec['new'].array[:, :]
def _get_P(self, nP, airfoils, name, bunch_LE, bunch_TE): def bunch_start(ind, a): ind[:] = ind**a def bunch_end(ind, a): ind[:] = 1 - (1-ind)**a airfoil = airfoils[name] P = numpy.zeros((airfoil.shape[0],4,3),order='F') for j in range(4): P[:,j,:2] = airfoil[:,:] P[:,j,2] = j bse = BSEmodel([P]) bse.set_bspline_option('num_pt', 0, 'u', airfoil.shape[0]) bse.set_bspline_option('num_cp', 0, 'u', int(airfoil.shape[0]/3)) bse.set_bspline_option('num_pt', 0, 'v', 4) bse.set_bspline_option('num_cp', 0, 'v', 4) bse.assemble() cp = bse.vec['cp_str'].array pt = bse.vec['pt_str'].array jac = bse.jac['d(pt_str)/d(cp_str)'] bse.vec['pt_str'].surfs[0][:, :, :] = P mtx = jac.T.dot(jac) rhs = jac.T.dot(pt) for dim in xrange(3): cp[:, dim] = scipy.sparse.linalg.cg(mtx, rhs[:, dim])[0] fit = numpy.array(cp) bse.set_bspline_option('num_pt', 0, 'u', nP) bse.assemble() cp = bse.vec['cp_str'].array pt = bse.vec['pt_str'].array cp[:, :] = fit[:, :] surfs = numpy.zeros(nP).astype(int) ind_u = numpy.linspace(0, 1, nP) if name == 'upp': bunch_start(ind_u, bunch_TE) bunch_end(ind_u, bunch_LE) elif name == 'low': bunch_start(ind_u, bunch_LE) bunch_end(ind_u, bunch_TE) ind_v = numpy.zeros(nP) bse.add_jacobian('new', surfs, ind_u, ind_v, 3) bse.apply_jacobian('new', 'd(new)/d(cp_str)', 'cp_str') return bse.vec['new'].array[:, :]