qpOASES_e_QProblemB.c

Go to the documentation of this file.

/*
 *      This file is part of qpOASES.
 *
 *      qpOASES -- An Implementation of the Online Active Set Strategy.
 *      Copyright (C) 2007-2015 by Hans Joachim Ferreau, Andreas Potschka,
 *      Christian Kirches et al. All rights reserved.
 *
 *      qpOASES is free software; you can redistribute it and/or
 *      modify it under the terms of the GNU Lesser General Public
 *      License as published by the Free Software Foundation; either
 *      version 2.1 of the License, or (at your option) any later version.
 *
 *      qpOASES is distributed in the hope that it will be useful,
 *      but WITHOUT ANY WARRANTY; without even the implied warranty of
 *      MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
 *      See the GNU Lesser General Public License for more details.
 *
 *      You should have received a copy of the GNU Lesser General Public
 *      License along with qpOASES; if not, write to the Free Software
 *      Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
 *
 */


#include <stdlib.h>

#include <qpOASES_e.h>
#include "qpOASES_e_simulink_utils.c"


#ifdef __cplusplus
extern "C" {
#endif


#define S_FUNCTION_NAME   qpOASES_e_QProblemB   
#define S_FUNCTION_LEVEL  2                                             
#define MDL_START                                                               
#include "simstruc.h"


/* SETTINGS */
#define SAMPLINGTIME    -1                                              
#define NCONTROLINPUTS  2                                               
#define MAXITER         100                                             
#define HESSIANTYPE     HST_UNKNOWN                             
static void mdlInitializeSizes (SimStruct *S)   /* Init sizes array */
{
        int nU = NCONTROLINPUTS;

        /* Specify the number of continuous and discrete states */
        ssSetNumContStates(S, 0);
        ssSetNumDiscStates(S, 0);

        /* Specify the number of parameters */
        ssSetNumSFcnParams(S, 1); /* H */
        if ( ssGetNumSFcnParams(S) != ssGetSFcnParamsCount(S) )
                return;

        /* Specify the number of intput ports */
        if ( !ssSetNumInputPorts(S, 3) )
                return;

        /* Specify the number of output ports */
        if ( !ssSetNumOutputPorts(S, 4) )
                return;

        /* Specify dimension information for the input ports */
        ssSetInputPortVectorDimension(S, 0, DYNAMICALLY_SIZED); /* g */
        ssSetInputPortVectorDimension(S, 1, DYNAMICALLY_SIZED); /* lb */
        ssSetInputPortVectorDimension(S, 2, DYNAMICALLY_SIZED); /* ub */

        /* Specify dimension information for the output ports */
        ssSetOutputPortVectorDimension(S, 0, nU );  /* uOpt */
    ssSetOutputPortVectorDimension(S, 1, 1 );   /* fval */
        ssSetOutputPortVectorDimension(S, 2, 1 );   /* exitflag */
        ssSetOutputPortVectorDimension(S, 3, 1 );   /* iter */

        /* Specify the direct feedthrough status */
        ssSetInputPortDirectFeedThrough(S, 0, 1);
        ssSetInputPortDirectFeedThrough(S, 1, 1);
        ssSetInputPortDirectFeedThrough(S, 2, 1);

        /* One sample time */
        ssSetNumSampleTimes(S, 1);

        /* global variables:
     * 0: problem
     * 1: H
     * 2: g
     * 3: lb
     * 4: ub
     */

        /* Specify the size of the block's pointer work vector */
    ssSetNumPWork(S, 5);
}


#if defined(MATLAB_MEX_FILE)

#define MDL_SET_INPUT_PORT_DIMENSION_INFO
#define MDL_SET_OUTPUT_PORT_DIMENSION_INFO

static void mdlSetInputPortDimensionInfo(SimStruct *S, int_T port, const DimsInfo_T *dimsInfo)
{
        if ( !ssSetInputPortDimensionInfo(S, port, dimsInfo) )
                return;
}

static void mdlSetOutputPortDimensionInfo(SimStruct *S, int_T port, const DimsInfo_T *dimsInfo)
{
        if ( !ssSetOutputPortDimensionInfo(S, port, dimsInfo) )
                return;
}

#endif


static void mdlInitializeSampleTimes(SimStruct *S)
{
        ssSetSampleTime(S, 0, SAMPLINGTIME);
        ssSetOffsetTime(S, 0, 0.0);
}


static void mdlStart(SimStruct *S)
{
        USING_NAMESPACE_QPOASES

        int nU = NCONTROLINPUTS;
        int size_g, size_lb, size_ub;
        int size_H, nRows_H, nCols_H;
        int nV;

        static QProblemB problem;
        static Options problemOptions;
        real_t* count;


        /* get block inputs dimensions */
        const mxArray* in_H = ssGetSFcnParam(S, 0);

        if ( mxIsEmpty(in_H) == 1 )
        {
                if ( ( HESSIANTYPE != HST_ZERO ) && ( HESSIANTYPE != HST_IDENTITY ) )
                {
                        #ifndef __SUPPRESSANYOUTPUT__
                        mexErrMsgTxt( "ERROR (qpOASES): Hessian can only be empty if type is set to HST_ZERO or HST_IDENTITY!" );
                        #endif
                        return;
                }
                
            nRows_H = 0;
                nCols_H = 0;
                size_H  = 0;
        }
        else
        {
            nRows_H = (int)mxGetM(in_H);
                nCols_H = (int)mxGetN(in_H);
                size_H  = nRows_H * nCols_H;
        }

        size_g   = ssGetInputPortWidth(S, 0);
        size_lb  = ssGetInputPortWidth(S, 1);
        size_ub  = ssGetInputPortWidth(S, 2);


        /* dimension checks */
        nV = size_g;

        if ( MAXITER < 0 )
        {
                #ifndef __SUPPRESSANYOUTPUT__
                mexErrMsgTxt( "ERROR (qpOASES): Maximum number of iterations must not be negative!" );
                #endif
                return;
        }

        if ( nV <= 0 )
        {
                #ifndef __SUPPRESSANYOUTPUT__
                mexErrMsgTxt( "ERROR (qpOASES): Dimension mismatch!" );
                #endif
                return;
        }

        if ( ( size_H != nV*nV ) && ( size_H != 0 ) )
        {
                #ifndef __SUPPRESSANYOUTPUT__
                mexErrMsgTxt( "ERROR (qpOASES): Dimension mismatch in H!" );
                #endif
                return;
        }

        if ( nRows_H != nCols_H )
        {
                #ifndef __SUPPRESSANYOUTPUT__
                mexErrMsgTxt( "ERROR (qpOASES): Hessian matrix must be square matrix!" );
                #endif
                return;
        }

        if ( ( nU < 1 ) || ( nU > nV ) )
        {
                #ifndef __SUPPRESSANYOUTPUT__
                mexErrMsgTxt( "ERROR (qpOASES): Invalid number of control inputs!" );
                #endif
                return;
        }

        if ( ( size_lb != nV ) && ( size_lb != 0 ) )
        {
                #ifndef __SUPPRESSANYOUTPUT__
                mexErrMsgTxt( "ERROR (qpOASES): Dimension mismatch in lb!" );
                #endif
                return;
        }

        if ( ( size_ub != nV ) && ( size_ub != 0 ) )
        {
                #ifndef __SUPPRESSANYOUTPUT__
                mexErrMsgTxt( "ERROR (qpOASES): Dimension mismatch in ub!" );
                #endif
                return;
        }


        /* allocate QProblemB object */
        QProblemBCON( &problem,nV,HESSIANTYPE );

        Options_setToMPC( &problemOptions );
        QProblemB_setOptions( &problem,problemOptions );
        #ifndef __DEBUG__
        QProblemB_setPrintLevel( &problem,PL_LOW );
        #endif
        #ifdef __SUPPRESSANYOUTPUT__
        QProblemB_setPrintLevel( &problem,PL_NONE );
        #endif

        ssGetPWork(S)[0] = (void*)(&problem);

        /* allocate memory for QP data ... */
        if ( size_H > 0 )
                ssGetPWork(S)[1] = (void *) calloc( size_H, sizeof(real_t) );   /* H */
        else
                ssGetPWork(S)[1] = 0;

        ssGetPWork(S)[2] = (void *) calloc( size_g, sizeof(real_t) );           /* g */

        if ( size_lb > 0 )
                ssGetPWork(S)[3] = (void *) calloc( size_lb, sizeof(real_t) );  /* lb */
        else
                ssGetPWork(S)[3] = 0;

        if ( size_ub > 0 )
                ssGetPWork(S)[4] = (void *) calloc( size_ub, sizeof(real_t) );  /* ub */
        else
                ssGetPWork(S)[4] = 0;
}


static void mdlOutputs(SimStruct *S, int_T tid)
{
        USING_NAMESPACE_QPOASES

        int i;
        int nV;
        returnValue status;

        int nWSR = MAXITER;
        int nU   = NCONTROLINPUTS;

        InputRealPtrsType in_g, in_lb, in_ub;

        QProblemB* problem;
        real_t *H, *g, *lb, *ub;

        real_t xOpt[NVMAX];

        real_T *out_uOpt, *out_objVal, *out_status, *out_nWSR;

        int nWSR_retry;


        /* get pointers to block inputs ... */
        const mxArray* in_H = ssGetSFcnParam(S, 0);
        in_g  = ssGetInputPortRealSignalPtrs(S, 0);
        in_lb = ssGetInputPortRealSignalPtrs(S, 1);
        in_ub = ssGetInputPortRealSignalPtrs(S, 2);

        /* ... and to the QP data */
        problem = (QProblemB*)(ssGetPWork(S)[0]);

        H  = (real_t*)(ssGetPWork(S)[1]);
        g  = (real_t*)(ssGetPWork(S)[2]);
        lb = (real_t*)(ssGetPWork(S)[3]);
        ub = (real_t*)(ssGetPWork(S)[4]);


        /* setup QP data */
        nV = ssGetInputPortWidth(S, 1); /* nV = size_g */

        if ( H != 0 )
        {
                /* no conversion from FORTRAN to C as Hessian is symmetric! */
                for ( i=0; i<nV*nV; ++i )
                        H[i] = (mxGetPr(in_H))[i];
        }

        for ( i=0; i<nV; ++i )
                g[i] = (*in_g)[i];

        if ( lb != 0 )
        {
                for ( i=0; i<nV; ++i )
                        lb[i] = (*in_lb)[i];
        }

        if ( ub != 0 )
        {
                for ( i=0; i<nV; ++i )
                        ub[i] = (*in_ub)[i];
        }

        if ( QProblemB_getCount( problem ) == 0 )
        {
                /* initialise and solve first QP */
                status = QProblemB_init( problem,H,g,lb,ub, &nWSR,0 );
                QProblemB_getPrimalSolution( problem,xOpt );
        }
        else
        {
                /* solve neighbouring QP using hotstart technique */
                status = QProblemB_hotstart( problem,g,lb,ub, &nWSR,0 );
                if ( ( status != SUCCESSFUL_RETURN ) && ( status != RET_MAX_NWSR_REACHED ) )
                {
                        /* if an error occurs, reset problem data structures ... */
                        QProblemB_reset( problem );
            
            /* ... and initialise/solve again with remaining number of iterations. */
            nWSR_retry = MAXITER - nWSR;
                        status = QProblemB_init( problem,H,g,lb,ub, &nWSR_retry,0 );
            nWSR += nWSR_retry;
                }

                /* obtain optimal solution */
                QProblemB_getPrimalSolution( problem,xOpt );
        }

        /* generate block output: status information ... */
        out_uOpt   = ssGetOutputPortRealSignal(S, 0);
        out_objVal = ssGetOutputPortRealSignal(S, 1);
        out_status = ssGetOutputPortRealSignal(S, 2);
        out_nWSR   = ssGetOutputPortRealSignal(S, 3);

        for ( i=0; i<nU; ++i )
                out_uOpt[i] = (real_T)(xOpt[i]);

        out_objVal[0] = (real_T)(QProblemB_getObjVal( problem ));
        out_status[0] = (real_t)(qpOASES_getSimpleStatus( status,BT_FALSE ));
        out_nWSR[0]   = (real_T)(nWSR);

        removeNaNs( out_uOpt,nU );
        removeInfs( out_uOpt,nU );
        removeNaNs( out_objVal,1 );
        removeInfs( out_objVal,1 );
}


static void mdlTerminate(SimStruct *S)
{
        USING_NAMESPACE_QPOASES

        int i;

        /* reset global message handler */
        MessageHandling_reset( qpOASES_getGlobalMessageHandler() );

        for ( i=1; i<5; ++i )
        {
                if ( ssGetPWork(S)[i] != 0 )
                        free( ssGetPWork(S)[i] );
        }
}


#ifdef  MATLAB_MEX_FILE
#include "simulink.c"
#else
#include "cg_sfun.h"
#endif


#ifdef __cplusplus
}
#endif


/*
 *      end of file
 */