Changeset 388 for library/bdm/stat/merger.h

Timestamp:

06/19/09 11:43:48 (15 years ago)

Author:

smidl

Message:

mergers

Files:

• library/bdm/stat/merger.h (modified) (8 diffs)

library/bdm/stat/merger.h

@@ -75 +75 @@
 
                 //!Empty constructor
-                merger_base () : compositepdf() {DBG=false;dbg_file=NULL;};
+                merger_base () : compositepdf() {DBG = false;dbg_file = NULL;};
                 //!Constructor from sources
-                merger_base (const Array<mpdf*> &S) {set_sources (S);};
-                //! Function setting the main internal structures 
-                void set_sources (const Array<mpdf*> &Sources) {
-                        compositepdf::set_elements (Sources);
+                merger_base (const Array<mpdf*> &S, bool own=false) {set_sources (S,own);};
+                //! Function setting the main internal structures
+                void set_sources (const Array<mpdf*> &Sources, bool own) {
+                        compositepdf::set_elements (Sources,own);
                         //set sizes
                         dls.set_size (Sources.length());
@@ -109 +109 @@
                         };
                 }
+                //! Rectangular support  each vector of XYZ specifies (begining-end) interval for each dimension. Same number of points, \c dimsize, in each dimension. 
+                void set_support (const Array<vec> &XYZ, const int dimsize) {
+                        set_support(XYZ,dimsize*ones_i(XYZ.length()));
+                }
+                //! Rectangular support  each vector of XYZ specifies (begining-end) interval for each dimension. \c gridsize specifies number of points is each dimension.
+                void set_support (const Array<vec> &XYZ, const ivec &gridsize) {
+                        int dim = XYZ.length();  //check with internal dim!!
+                        Npoints = prod (gridsize); 
+                        eSmp.set_parameters (Npoints, false);
+                        Array<vec> &samples = eSmp._samples();
+                        eSmp._w() = ones (Npoints) / Npoints; //unifrom size of bins
+                        //set samples
+                        ivec ind = zeros_i (dim);      //indeces of dimensions in for cycle;
+                        vec smpi (dim);            // ith sample
+                        vec steps =zeros(dim);            // ith sample
+                        // first corner
+                        for (int j = 0; j < dim; j++) {
+                                smpi (j) = XYZ (j) (0); /* beginning of the interval*/ 
+                                it_assert(gridsize(j)!=0.0,"Zeros in gridsize!");
+                                steps (j) = (smpi(j) - XYZ(j)(1))/gridsize(j);
+                        }
+                        // fill samples
+                        int act_dim=0; //active dimension
+                        for (int i = 0; i < Npoints; i++) {
+                                // copy 
+                                samples(i) = smpi; 
+                                // go through all dimensions
+                                for (int j = 0;j < dim;j++) {
+                                        if (ind (j) == gridsize (j) - 1) { //j-th index is full
+                                                ind (j) = 0; //shift back
+                                                smpi(j) = XYZ(j)(0);
+                                                
+                                                ind (j + 1) ++; //increase the next dimension;
+                                                smpi(j+1) += steps(j+1);
+                                                
+                                                if (ind (j + 1) < gridsize (j + 1) - 1) break;
+                                        } else {
+                                                ind (j) ++; 
+                                                smpi(j) +=steps(j);
+                                                break;
+                                        }
+                                }
+                        }
+                }
                 //! set debug file
-                void set_debug_file (const string fname) { 
-                        if (DBG) delete dbg_file; 
-                        dbg_file = new it_file (fname); 
+                void set_debug_file (const string fname) {
+                        if (DBG) delete dbg_file;
+                        dbg_file = new it_file (fname);
                         if (dbg_file) DBG = true;
                 }
                 //! Set internal parameters used in approximation
-                void set_method (MERGER_METHOD MTH, double beta0=0.0) {
-                        METHOD = MTH; 
+                void set_method (MERGER_METHOD MTH, double beta0 = 0.0) {
+                        METHOD = MTH;
                         beta = beta0;
                 }
                 //! Set support points from a pdf by drawing N samples
-                void set_support(const epdf &overall, int N){
-                        it_assert_debug ( rv.equal ( overall._rv() ),"Incompatible parameter overall!" );
-                        eSmp.set_statistics(&overall,N);
-                        Npoints=N;
-                }
-                
+                void set_support (const epdf &overall, int N) {
+                        eSmp.set_statistics (&overall, N);
+                        Npoints = N;
+                }
+
                 //! Destructor
                 virtual ~merger_base() {
@@ -136 +179 @@
                 };
                 //!@}
-                
+
                 //! \name Mathematical operations
                 //!@{
-                
+
                 //!Merge given sources in given points
                 void merge () {
-                        if (eSmp._w().length() ==0) {it_error("Empty support points use set_support" );}
+                        validate();
+
                         //check if sources overlap:
                         bool OK = true;
@@ -164 +208 @@
                                 eSmp._w() = wtmp / sum (wtmp);
                         } else {
-                                it_error("Sources are not compatible - use merger_mix");
-                        }
-                        ;
+                                it_error ("Sources are not compatible - use merger_mix");
+                        }
                 };
 
@@ -172 +215 @@
                 //! Merge log-likelihood values in points using method specified by parameter METHOD
                 vec merge_points (mat &lW);
-                
-                
+
+
                 //! sample from merged density
 //! weight w is a
@@ -216 +259 @@
                 //! Access function
                 eEmp& _Smp() {return eSmp;}
-                
+
+                //! load from setting
+                void from_setting (const Setting& set) {
+                        // get support
+                        // find which method to use
+                        string meth_str;
+                        UI::get<string> (meth_str, set, "method");
+                        if (!strcmp (meth_str.c_str(), "arithmetic"))
+                                set_method (ARITHMETIC);
+                        else {
+                                if (!strcmp (meth_str.c_str(), "geometric"))
+                                        set_method (GEOMETRIC);
+                                else if (!strcmp (meth_str.c_str(), "lognormal")) {
+                                        set_method (GEOMETRIC);
+                                        set.lookupValue( "beta",beta);
+                                }
+                        }
+                        validate();
+                }
+
+                void validate() {
+                        it_assert (eSmp._w().length() > 0, "Empty support, use set_support().");
+                        it_assert (dim == eSmp._samples() (0).length(), "Support points and rv are not compatible!");
+                        it_assert (isnamed(),"mergers must be named");
+                }
                 //!@}
 };
+UIREGISTER(merger_base);
 
 class merger_mix : public merger_base
@@ -234 +302 @@
                 //!@{
                 merger_mix () {};
-                merger_mix (const Array<mpdf*> &S) {set_sources(S);};
+                merger_mix (const Array<mpdf*> &S,bool own=false) {set_sources (S,own);};
                 //! Set sources and prepare all internal structures
-                void set_sources (const Array<mpdf*> &S) {
-                        merger_base::set_sources(S);
+                void set_sources (const Array<mpdf*> &S, bool own) {
+                        merger_base::set_sources (S,own);
                         Nsources = S.length();
                 }
                 //! Set internal parameters used in approximation
-                void set_parameters (int Ncoms0=10, double effss_coef0 = 0.5) {
+                void set_parameters (int Ncoms0 = 10, double effss_coef0 = 0.5) {
                         Ncoms = Ncoms0;
                         effss_coef = effss_coef0;
                 }
                 //!@}
-                
+
                 //! \name Mathematical operations
                 //!@{
-                
+
                 //!Merge values using mixture approximation
                 void merge ();
@@ -269 +337 @@
                 //! Access function
                 emix* proposal() {emix* tmp = Mix.epredictor(); tmp->set_rv (rv); return tmp;}
-        //! @}
+                //! from_settings
+                void from_settings(const Setting& set){
+                        merger_base::from_setting(set);
+                        set.lookupValue("ncoms",Ncoms);
+                }
                 
+                //! @}
+
 };
+UIREGISTER(merger_mix);
 
 }