root/applications/bdmtoolbox/mex/merger_mx.cpp @ 384

#include <itpp/itmex.h>
#include <estim/merger.h>
#include "config2mxstruct.h"
#include "mexlog.h"
#include "mexparse.h"

using namespace bdm;

void mexFunction(int n_output, mxArray *output[], int n_input, const mxArray *input[])
{
    // Check the number of inputs and output arguments
        if(n_input!=1) mexErrMsgTxt("Configuration structure expected in the form:\n"  
                "S.sources = { };                          % cell of pdfs (epdfs or mpdfs) to be merged,\n"
                "S.merger= struct('class','merger_*');     % object to be used for merging,\n\n"
                "see documentation of classes epdf, mpdf, merger_base and their offsprings in BDM.");

        // LOAD CONFIG
        UImxArray F (input[0]);

        Array<mpdf*> Sources;

        Setting& _Sources=F.lookup("sources");
        int Slen=_Sources.getLength();
        Sources.set_size(Slen);
        for (int i=0; i<Slen; i++){
                mpdf* mtmp = UI::build<mpdf>(_Sources,i);
                if (mtmp) { 
                        Sources(i)=mtmp;
                } else{ // source is not mpdf
                        epdf* etmp = UI::build<epdf>(_Sources,i);
                        if (etmp){
                                Sources(i) = new mepdf(etmp, true);
                        }
                }

        }
        
        merger_base* Merger=UI::build<merger_base>(F.getRoot(),"merger");

// COMPUTE RESULTS

        Merger->merge();
        
        if (n_output>0){
                mxArray* tmp = mxCreateStructMatrix(1,1,0,NULL);
                //support
                Array<vec> &samples=Merger->_Smp()._samples();
                if (samples.size()>0){
                        mxArray* fld=mxCreateDoubleMatrix(samples(0).length(), samples.size(), mxREAL);
                Arrayvec2mxArray(samples,fld);
                mxReplaceFieldNM(tmp, "support", fld);
                }

                //weights
                vec &w = Merger->_Smp()._w();
                mxArray* fldw=mxCreateDoubleMatrix(1, w.length(), mxREAL);
                vec2mxArray(w,fldw);
                mxReplaceFieldNM(tmp, "weights", fldw);
                
                output[0] = tmp;
        }
}