[393] | 1 | #include <stat/merger.h> |
---|
| 2 | |
---|
[407] | 3 | #ifdef MEX |
---|
| 4 | #include <itpp/itmex.h> |
---|
| 5 | #include <mex/mex_logger.h> |
---|
| 6 | #include <mex/mex_parser.h> |
---|
| 7 | #include <mex/config2mxstruct.h> |
---|
| 8 | #endif |
---|
| 9 | |
---|
[411] | 10 | /*! \file |
---|
| 11 | \brief Merging static pdfs |
---|
| 12 | Merger is a program/mex-function for static merging of given pdfs on given support. |
---|
| 13 | |
---|
| 14 | In command line, it expected config file with the following structure: |
---|
| 15 | \code |
---|
| 16 | Sources = (... any epdf or mpdf ...); |
---|
| 17 | Support = {grid=([low1d,high1d],[low2d,high2d],...); |
---|
| 18 | nbins=[bins1d,bins2d,... ]}; |
---|
| 19 | // OR |
---|
| 20 | Support = {pdf={class='epdf',...}; |
---|
| 21 | nsamples=2}; |
---|
| 22 | Merger = {class="merger_base",...} |
---|
| 23 | \endcode |
---|
| 24 | |
---|
| 25 | In mex-file, the above structures (Sources,Support,Merger) are input arguments. Type >>merger for help. |
---|
| 26 | */ |
---|
| 27 | |
---|
[393] | 28 | using namespace bdm; |
---|
| 29 | |
---|
[411] | 30 | #ifdef MEX |
---|
[407] | 31 | void mexFunction(int n_output, mxArray *output[], int n_input, const mxArray *input[]) |
---|
[393] | 32 | { |
---|
[407] | 33 | // Check the number of inputs and output arguments |
---|
| 34 | if(n_input!=3) mexErrMsgTxt("Usage:\n" |
---|
| 35 | "result=merger(sources, support, merger)\n" |
---|
| 36 | " sources= { struct('class','epdf'),... }; % cell of pdfs (epdfs or mpdfs) to be merged,\n" |
---|
| 37 | " support= struct(\n" |
---|
| 38 | " grid = {[dim1_start,dim1_end], [dim2_start, dim2_end]...} %support boundary \n" |
---|
| 39 | " nbins = [bins_in_dim1, bins_in_dim2,...] %fixed \n" |
---|
| 40 | " === OR ==\n" |
---|
| 41 | " pdf = struct('class','epdf'); % pdf to draw samples from\n" |
---|
| 42 | " nsamples= 100; % number of samples\n" |
---|
| 43 | " );\n" |
---|
| 44 | " If all elements are present, (grid,nbins) is used;\n" |
---|
| 45 | " merger = struct('class','merger_*'); % object to be used for merging,\n\n" |
---|
| 46 | "see documentation of classes epdf, mpdf, merger_base and their offsprings in BDM."); |
---|
| 47 | |
---|
| 48 | // LOAD CONFIG |
---|
| 49 | UImxArray Cfg; |
---|
| 50 | Cfg.addList(input[0],"Sources"); |
---|
| 51 | Cfg.addGroup(input[1],"Support"); |
---|
| 52 | Cfg.addGroup(input[2],"Merger"); |
---|
| 53 | |
---|
| 54 | //DBG |
---|
| 55 | Cfg.writeFile("merger.cfg"); |
---|
| 56 | #else |
---|
| 57 | int main() |
---|
| 58 | { |
---|
| 59 | UIFile Cfg("merger.cfg"); |
---|
| 60 | #endif |
---|
[393] | 61 | // Sources |
---|
| 62 | Array<mpdf*> Sources; |
---|
| 63 | //abuse Mer to store sources |
---|
| 64 | Setting& _Sources=Cfg.lookup("Sources"); |
---|
| 65 | int Slen=_Sources.getLength(); |
---|
| 66 | Sources.set_size(Slen); |
---|
| 67 | for (int i=0; i<Slen; i++){ |
---|
| 68 | try{ |
---|
| 69 | mpdf* mtmp = UI::build<mpdf>(_Sources,i); |
---|
| 70 | Sources(i)=mtmp; |
---|
| 71 | } |
---|
[400] | 72 | catch (UIException){ |
---|
[393] | 73 | // it is not mpdf - see if it is epdf |
---|
| 74 | try { |
---|
| 75 | epdf* etmp = UI::build<epdf>(_Sources,i); |
---|
| 76 | if (etmp){ |
---|
| 77 | Sources(i) = new mepdf(etmp, true); |
---|
[407] | 78 | } |
---|
[393] | 79 | } |
---|
[400] | 80 | catch (UIException e) |
---|
| 81 | { |
---|
| 82 | it_error("No mpdfs or epdfs found! " + string(e.what())); |
---|
| 83 | } |
---|
| 84 | catch (std::exception e) { |
---|
| 85 | it_error("Error in UI at "+_Sources[i].getPath()); |
---|
[407] | 86 | } |
---|
[393] | 87 | } |
---|
[400] | 88 | catch (std::exception e) { |
---|
| 89 | it_error("Error in UI at "+_Sources[i].getPath()); |
---|
[407] | 90 | } |
---|
[393] | 91 | } |
---|
| 92 | |
---|
| 93 | merger_base* Merger=UI::build<merger_base>(Cfg,"Merger"); |
---|
| 94 | |
---|
| 95 | // Support |
---|
| 96 | Setting & _Supp=Cfg.lookup("Support"); |
---|
| 97 | |
---|
| 98 | if (_Supp.exists("grid") && _Supp.exists("nbins")) { |
---|
| 99 | Array<vec> bounds (0); |
---|
| 100 | UI::get (bounds, _Supp, "grid"); |
---|
| 101 | ivec nbins(0); |
---|
| 102 | UI::get (nbins, _Supp, "nbins"); |
---|
| 103 | Merger->set_support (bounds,nbins); |
---|
| 104 | |
---|
| 105 | }else { |
---|
| 106 | if (_Supp.exists("pdf") && _Supp.exists("nsamples")){ |
---|
| 107 | epdf *g0=UI::build<epdf> (_Supp, "pdf"); |
---|
| 108 | int npoints=100; |
---|
| 109 | _Supp.lookupValue("nsamples",npoints); |
---|
| 110 | Merger->set_support (*g0,npoints); |
---|
| 111 | delete g0; |
---|
| 112 | } |
---|
| 113 | else it_error("Use either [grid,nbins] or [pdf,nsamples]."); |
---|
| 114 | } |
---|
| 115 | // COMPUTE RESULTS |
---|
| 116 | Merger->set_sources(Sources,true); // takes care of deletion of sources |
---|
| 117 | Merger->merge(); |
---|
| 118 | |
---|
[407] | 119 | // save results |
---|
[429] | 120 | Array<vec> source_vals(Sources.length()); |
---|
| 121 | for (int i=0;i<Sources.length();i++){ |
---|
| 122 | datalink_m2e dl; |
---|
| 123 | dl.set_connection(Sources(i)->_rv(), Sources(i)->_rvc(), Merger->_rv()); |
---|
| 124 | |
---|
| 125 | vec ll(Merger->_Smp()._samples().length()); |
---|
| 126 | for(int j=0; j<Merger->_Smp()._samples().length(); j++){ |
---|
| 127 | vec dt=dl.pushdown(Merger->_Smp()._samples()(j)); |
---|
| 128 | vec dtc=dl.get_cond(Merger->_Smp()._samples()(j)); |
---|
| 129 | ll(j)=Sources(i)->evallogcond(dt,dtc); |
---|
| 130 | } |
---|
| 131 | |
---|
| 132 | vec sll = exp(ll); |
---|
| 133 | |
---|
| 134 | source_vals(i)=sll/sum(sll); |
---|
| 135 | } |
---|
| 136 | |
---|
| 137 | vec ll(Merger->_Smp()._samples().length()); |
---|
| 138 | for(int j=0; j<Merger->_Smp()._samples().length(); j++){ |
---|
| 139 | ll(j)=Merger->evallog(Merger->_Smp()._samples()(j)); |
---|
| 140 | } |
---|
| 141 | |
---|
| 142 | vec sll = exp(ll); |
---|
| 143 | |
---|
| 144 | vec mix_val=sll/sum(sll); |
---|
| 145 | |
---|
[407] | 146 | |
---|
| 147 | #ifdef MEX |
---|
| 148 | mxArray* tmp ; |
---|
| 149 | // Save results |
---|
| 150 | if (n_output>0){ |
---|
| 151 | tmp = mxCreateStructMatrix(1,1,0,NULL); |
---|
| 152 | //support |
---|
| 153 | Array<vec> &samples=Merger->_Smp()._samples(); |
---|
| 154 | if (samples.size()>0){ |
---|
| 155 | mxArray* fld=mxCreateDoubleMatrix(samples(0).length(), samples.size(), mxREAL); |
---|
| 156 | Arrayvec2mxArray(samples,fld); |
---|
| 157 | mxReplaceFieldNM(tmp, "support", fld); |
---|
| 158 | } |
---|
| 159 | |
---|
| 160 | //weights |
---|
| 161 | vec &w = Merger->_Smp()._w(); |
---|
| 162 | mxArray* fldw=mxCreateDoubleMatrix(1, w.length(), mxREAL); |
---|
| 163 | vec2mxArray(w,fldw); |
---|
| 164 | mxReplaceFieldNM(tmp, "weights", fldw); |
---|
[429] | 165 | |
---|
| 166 | //mixture values |
---|
| 167 | mxArray* fldm=mxCreateDoubleMatrix(1, w.length(), mxREAL); |
---|
| 168 | vec2mxArray(mix_val,fldm); |
---|
| 169 | mxReplaceFieldNM(tmp, "mix", fldm); |
---|
[407] | 170 | |
---|
| 171 | // sources |
---|
| 172 | char srcstr[20]; |
---|
| 173 | for (int i=0;i<Sources.length();i++){ |
---|
| 174 | sprintf(srcstr,"source%d",i+1); |
---|
[429] | 175 | mxArray* fldw=mxCreateDoubleMatrix(1, source_vals(i).length(), mxREAL); |
---|
| 176 | vec2mxArray(source_vals(i),fldw); |
---|
[407] | 177 | mxReplaceFieldNM(tmp, srcstr, fldw); |
---|
| 178 | } |
---|
| 179 | |
---|
| 180 | output[0] = tmp; |
---|
| 181 | } |
---|
| 182 | #endif |
---|
[393] | 183 | } |
---|