root/applications/bdmtoolbox/mex/merger.cpp @ 622

#include <stat/merger.h>

#ifdef MEX
#include <itpp/itmex.h>
#include <mex/mex_logger.h>
#include <mex/mex_parser.h>
#include <mex/config2mxstruct.h>
#endif

/*! \file
\brief Merging static pdfs
Merger is a program/mex-function for static merging of given pdfs on given support.

In command line, it expected config file with the following structure:
\code
Sources = (... any epdf or mpdf  ...);
Support = {grid=([low1d,high1d],[low2d,high2d],...);
                   nbins=[bins1d,bins2d,... ]};
// OR
Support = {pdf={class='epdf',...};
                   nsamples=2};
Merger = {class="merger_base",...}
\endcode

In mex-file, the above structures (Sources,Support,Merger) are input arguments. Type >>merger for help.
*/

using namespace bdm;

#ifdef MEX
void mexFunction ( int n_output, mxArray *output[], int n_input, const mxArray *input[] ) {
        // Check the number of inputs and output arguments
        if ( n_input != 3 ) mexErrMsgTxt ( "Usage:\n"
                                                   "result=merger(sources, support, merger)\n"
                                                   "  sources= { struct('class','epdf'),... };  % cell of pdfs (epdfs or mpdfs) to be merged,\n"
                                                   "  support= struct(\n"
                                                   "           grid    = {[dim1_start,dim1_end], [dim2_start, dim2_end]...}  %support boundary \n"
                                                   "           nbins   = [bins_in_dim1, bins_in_dim2,...]                    %fixed \n"
                                                   "         === OR ==\n"
                                                   "           pdf     = struct('class','epdf'); % pdf to draw samples from\n"
                                                   "           nsamples= 100;                    % number of samples\n"
                                                   "           );\n"
                                                   "        If all elements are present,  (grid,nbins) is used;\n"
                                                   "  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." );
        RV::clear_all();
        // LOAD CONFIG
        UImxArray Cfg;
        Cfg.addList ( input[0], "Sources" );
        Cfg.addGroup ( input[1], "Support" );
        Cfg.addGroup ( input[2], "Merger" );

        //DBG
        Cfg.writeFile ( "merger.cfg" );
#else
int main() {
        UIFile Cfg ( "merger.cfg" );
#endif
        // Sources
        Array<shared_ptr<mpdf> > Sources;
        //abuse Mer to store sources
        Setting& _Sources = Cfg.lookup ( "Sources" );
        int Slen = _Sources.getLength();
        Sources.set_size ( Slen );
        for ( int i = 0; i < Slen; i++ ) {
                try {
                        shared_ptr<mpdf> mtmp = UI::build<mpdf> ( _Sources, i );
                        Sources ( i ) = mtmp;
                } catch ( UIException ) {
                        // it is not mpdf - see if it is epdf
                        try {
                                shared_ptr<epdf> etmp = UI::build<epdf> ( _Sources, i );
                                if ( etmp ) {
                                        Sources ( i ) = new mepdf ( etmp ); // hopefully OK
                                }
                        } catch ( UIException &e ) {
                                it_error ( "No mpdfs or epdfs found! " + string ( e.what() ) );
                        } catch ( std::exception e ) {
                                it_error ( "Error in UI at " + _Sources[i].getPath() );
                        }
                } catch ( std::exception &e ) {
                        it_error ( "Error in UI at " + _Sources[i].getPath() );
                }
        }

        shared_ptr<merger_base> Merger = UI::build<merger_base> ( Cfg, "Merger" );

        // Support
        try {
                shared_ptr<rectangular_support> RecSup = UI::build<rectangular_support> ( Cfg, "Support" );
                Merger->set_support ( *RecSup );
        } catch ( UIException &e ) {
                shared_ptr<discrete_support> DisSup = UI::build<discrete_support> ( Cfg, "Support" );
                Merger->set_support ( *DisSup );
        }
// COMPUTE RESULTS
        Merger->set_sources ( Sources );
        Merger->merge();

// save results
        Array<vec> source_vals ( Sources.length() );
        for ( int i = 0;i < Sources.length();i++ ) {
                datalink_m2e dl;
                dl.set_connection ( Sources ( i )->_rv(), Sources ( i )->_rvc(), Merger->_rv() );

                vec ll ( Merger->_Smp()._samples().length() );
                for ( int j = 0; j < Merger->_Smp()._samples().length(); j++ ) {
                        vec dt = dl.pushdown ( Merger->_Smp()._samples() ( j ) );
                        vec dtc = dl.get_cond ( Merger->_Smp()._samples() ( j ) );
                        ll ( j ) = Sources ( i )->evallogcond ( dt, dtc );
                }

                vec sll = exp ( ll );

                source_vals ( i ) = sll / sum ( sll );
        }

        merger_mix* MerMix=dynamic_cast<merger_mix*> ( Merger.get() );
        vec mix_val;

        if ( MerMix ) {
                vec ll ( Merger->_Smp()._samples().length() );
                for ( int j = 0; j < Merger->_Smp()._samples().length(); j++ ) {
                        ll ( j ) = Merger->evallog ( Merger->_Smp()._samples() ( j ) );
                }

                vec sll = exp ( ll );

                mix_val = sll / sum ( sll );
        }

#ifdef MEX
        mxArray* tmp ;
        // Save results
        if ( n_output > 0 ) {
                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 );

                //mixture values
                if ( mix_val.length() >0 ) {
                        mxArray* fldm = mxCreateDoubleMatrix ( 1, w.length(), mxREAL );
                        vec2mxArray ( mix_val, fldm );
                        mxReplaceFieldNM ( tmp, "mix", fldm );
                }
                // sources
                char srcstr[20];
                for ( int i = 0;i < Sources.length();i++ ) {
                        sprintf ( srcstr, "source%d", i + 1 );
                        mxArray* fldw = mxCreateDoubleMatrix ( 1, source_vals ( i ).length(), mxREAL );
                        vec2mxArray ( source_vals ( i ), fldw );
                        mxReplaceFieldNM ( tmp, srcstr, fldw );
                }

                output[0] = tmp;
        }
#endif
}