Exemplo n.º 1
0
    def __init__(self, num_elem=10, num_cp=5, x_pts=None, surr_file=None):
        """Initialize this segment trajectory problem.

        num_elem: int
            number of computations points in the mission profile

        num_cp: int
            number of control points for the splines

        x_pts: 1d array
            array containing the x locations of the spline control points.

        surr_file: Name of file for generating the Tripan surrogate models.
        """

        self.num_elem = num_elem
        self.num_pt = num_cp
        self.x_pts = x_pts

        # Generate jacobians for b-splines using MBI package
        self.jac_h, self.jac_gamma = setup_MBI(num_elem + 1, num_cp, x_pts)

        # Setup the surrogate models
        self.CL_arr, self.CD_arr, self.CM_arr, self.num = \
            setup_surrogate(surr_file)

        super(MissionSegment, self).__init__()
Exemplo n.º 2
0
    def __init__(self, num_elem=10, num_cp=5, x_pts=None, surr_file=None):
        """Initialize this segment trajectory problem.

        num_elem: int
            number of computations points in the mission profile

        num_cp: int
            number of control points for the splines

        x_pts: 1d array
            array containing the x locations of the spline control points.

        surr_file: Name of file for generating the Tripan surrogate models.
        """

        self.num_elem = num_elem
        self.num_pt = num_cp
        self.x_pts = x_pts

        # Generate jacobians for b-splines using MBI package
        self.jac_h, self.jac_gamma = setup_MBI(num_elem+1, num_cp, x_pts)

        # Setup the surrogate models
        self.CL_arr, self.CD_arr, self.CM_arr, self.num = \
            setup_surrogate(surr_file)

        super(MissionSegment, self).__init__()
Exemplo n.º 3
0
    def test_ASysTripanCDSurrogate(self):

        surr = '../crm_surr'
        CL_arr, CD_arr, CM_arr, num = setup_surrogate(surr)
        compname = 'SysTripanCDSurrogate'

        self.arg_dict['num'] = num
        self.arg_dict['CD'] = CD_arr

        self.setup(compname, self.arg_dict)

        # Scale some inputs
        val = self.model.comp.get('M')
        shape1 = val.shape
        self.model.comp.set('M', 0.8 + .1*np.random.random(shape1))

        self.run_model()
        self.compare_derivatives()
Exemplo n.º 4
0
    def test_ASysTripanCDSurrogate(self):

        surr = '../crm_surr'
        CL_arr, CD_arr, CM_arr, num = setup_surrogate(surr)
        compname = 'SysTripanCDSurrogate'

        self.arg_dict['num'] = num
        self.arg_dict['CD'] = CD_arr

        self.setup(compname, self.arg_dict)

        # Scale some inputs
        val = self.model.comp.get('M')
        shape1 = val.shape
        self.model.comp.set('M', 0.8 + .1 * np.random.random(shape1))

        self.run_model()
        self.compare_derivatives()
Exemplo n.º 5
0
    def __init__(self, problem_file):
        super(AllocationProblem, self).__init__()

        data = {}
        execfile(problem_file, data)
        misc_data = data['misc_data']
        ac_data = data['ac_data']
        rt_data = data['rt_data']

        num_routes = rt_data['number']
        num_existing_ac = len(ac_data['existing_ac'])
        num_new_ac = len(ac_data['new_ac'])
        num_ac = num_existing_ac + num_new_ac

        self.num_routes = num_routes
        self.num_existing_ac = num_existing_ac
        self.num_new_ac = num_new_ac
        self.num_ac = num_ac

        ac_aerosurr = {}
        for ac_name in ac_data['new_ac']:
            data = {}
            execfile(ac_name+'_params.py', data)
            params = data['params']

            surr = {}
            surr['CL'], surr['CD'], surr['CM'], surr['nums'] = setup_surrogate(params['surr'])
            ac_aerosurr[ac_name] = surr

        self.gamma_lb = np.tan(-35.0 * (np.pi/180.0))/1e-1
        self.gamma_ub = np.tan(35.0 * (np.pi/180.0))/1e-1

        for irt in xrange(num_routes):
            num_elem = rt_data['num_elem'][irt]
            num_cp = rt_data['num_cp'][irt]
            x_range = rt_data['range'][irt] * 1.852
            altitude = 10 * np.sin(np.pi * np.linspace(0,1,num_elem+1))

            x_init = x_range * 1e3 * (1-np.cos(np.linspace(0, 1, num_cp)*np.pi))/2/1e6
            M_init = np.ones(num_cp)*0.82
            h_init = 10 * np.sin(np.pi * x_init / (x_range/1e3))

            jac_h, jac_gamma = setup_MBI(num_elem+1, num_cp, x_init)

            #altitude = 10 * np.sin(np.pi * np.linspace(0,1,num_elem+1))

            for iac in xrange(num_new_ac):
                ac_name = ac_data['new_ac'][iac]
                seg_name = 'Seg_%03i_%03i'%(irt,iac)

                seg = self.add(seg_name, MissionSegment(num_elem=num_elem, num_cp=num_cp,
                                                        x_pts=x_init, params_file=ac_name+'_params.py',
                                                        aero_surr=ac_aerosurr[ac_name],
                                                        jac_h=jac_h, jac_gamma=jac_gamma))

                seg.h_pt = h_init
                seg.M_pt = M_init
                #seg.set_init_h_pt(altitude)
                seg.driver.system_type = 'serial'
                seg.coupled_solver.system_type = 'serial'
                seg.driver.gradient_options.iprint = 0
                seg.coupled_solver.gradient_options.iprint = 0
                seg.coupled_solver.iprint = 0
#                seg.coupled_solver.gradient_options.iprint = 0
                #seg.coupled_solver.pre_setup()

        self.add('pax_flt', Array(10*np.ones((num_routes, num_ac)), iotype='in'))
        self.add('flt_day', Array(0.1*np.ones((num_routes, num_ac)), iotype='in'))

        cost_fuel = np.zeros((num_routes, num_ac))
        prc_pax = np.zeros((num_routes, num_ac))
        cost_nf = np.zeros((num_routes, num_ac))
        maintenance = np.zeros((num_routes, num_ac))
        turnaround = np.zeros((num_routes, num_ac))
        for irt in xrange(num_routes):
            for iac in xrange(num_ac):
                if iac < num_existing_ac:
                    ac_name = ac_data['existing_ac'][iac]
                else:
                    inac = iac - num_existing_ac
                    ac_name = ac_data['new_ac'][inac]
                cost_fuel[irt, iac] = misc_data['cost/fuel'] * 2.2 / 9.81
                prc_pax[irt, iac] = ac_data['ticket price', ac_name][irt]
                cost_nf[irt, iac] = ac_data['flight cost no fuel', ac_name][irt]
                maintenance[irt, iac] = ac_data['MH', ac_name] + 1
                turnaround[irt, iac] = misc_data['turnaround']

        self.add('SysProfit', Profit(num_routes, num_ac,
                                     prc_pax, cost_fuel, cost_nf))
        self.add('SysPaxCon', PaxCon(num_routes, num_ac))
        self.add('SysAircraftCon', AircraftCon(num_routes, num_ac,
                                               maintenance, turnaround))

        for iac in xrange(num_existing_ac):
            ac_name = ac_data['existing_ac'][iac]
            self.SysProfit.fuelburn[:, iac] = ac_data['fuel', ac_name] / 1e5
            self.SysAircraftCon.time[:, iac] = ac_data['block time', ac_name] * 3.6e3 / 1e4

        self.connect('pax_flt', 'SysProfit.pax_flt')
        self.connect('flt_day', 'SysProfit.flt_day')

        self.connect('pax_flt', 'SysPaxCon.pax_flt')
        self.connect('flt_day', 'SysPaxCon.flt_day')

        self.connect('flt_day', 'SysAircraftCon.flt_day')

        for irt in xrange(num_routes):
            for inac in xrange(num_new_ac):
                iac = inac + num_existing_ac
                seg_name = 'Seg_%03i_%03i'%(irt,inac)

                self.connect('pax_flt[%i, %i]'%(irt,iac), seg_name+'.pax_flt')
                self.connect(seg_name+'.fuelburn',
                             'SysProfit.fuelburn[%i, %i]'%(irt,iac))
                self.connect(seg_name+'.time',
                             'SysAircraftCon.time[%i, %i]'%(irt,iac))

        self.create_passthrough('SysProfit.profit')
        self.create_passthrough('SysPaxCon.pax_con')
        self.create_passthrough('SysAircraftCon.ac_con')

        pax_upper = np.zeros((num_routes, num_ac))
        demand = np.zeros(num_routes)
        ac_avail = np.zeros(num_ac)
        for irt in xrange(num_routes):
            for iac in xrange(num_ac):
                if iac < num_existing_ac:
                    ac_name = ac_data['existing_ac'][iac]
                else:
                    inac = iac - num_existing_ac
                    ac_name = ac_data['new_ac'][inac]
                pax_upper[irt, iac] = ac_data['capacity', ac_name]
        for irt in xrange(num_routes):
            demand[irt] = rt_data['demand'][irt]
        for iac in xrange(num_ac):
            if iac < num_existing_ac:
                ac_name = ac_data['existing_ac'][iac]
            else:
                inac = iac - num_existing_ac
                ac_name = ac_data['new_ac'][inac]
            ac_avail[iac] = 12 * ac_data['number', ac_name]

        self.add('pax_upper', Array(np.zeros((num_routes, num_ac)), iotype='out'))
        self.add('demand', Array(np.zeros(num_routes), iotype='out'))
        self.add('ac_avail', Array(np.zeros(num_ac), iotype='out'))

        self.pax_upper = pax_upper
        self.demand = demand
        self.ac_avail = ac_avail

        self.add('missions', Driver())
        for irt in xrange(self.num_routes):
            for inac in xrange(self.num_new_ac):
                seg_name = 'Seg_%03i_%03i'%(irt,inac)
                self.missions.workflow.add(seg_name)
        self.missions.system_type = 'parallel'
        #self.missions.gradient_options.lin_solver = "linear_gs"
        self.missions.gradient_options.lin_solver = "petsc_ksp"
        self.missions.gradient_options.iprint = 0


        self.add('stuff', Driver())
        self.stuff.system_type = "serial"
        self.stuff.workflow.add(['SysProfit', 'SysPaxCon', 'SysAircraftCon'])

        self.driver.workflow.add(['missions', 'stuff'])