LauSimFitMaster.hh

Go to the documentation of this file.

// Copyright University of Warwick 2013 - 2013.
// Distributed under the Boost Software License, Version 1.0.
// (See accompanying file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)

// Authors:
// Thomas Latham
// John Back
// Paul Harrison

#ifndef LAU_SIM_FIT_MASTER
#define LAU_SIM_FIT_MASTER

#include <map>
#include <vector>

#include "TMatrixD.h"
#include "TMessage.h"
#include "TStopwatch.h"
#include "TString.h"
#include "TVectorDfwd.h"

#include "LauFitObject.hh"

class TMessage;
class TMonitor;
class TSocket;
class LauAbsRValue;
class LauParameter;
class LauFitNtuple;


class LauSimFitMaster : public LauFitObject {

        public:

                LauSimFitMaster( UInt_t numSlaves, UInt_t port = 9090 );

                virtual ~LauSimFitMaster();


                void runSimFit( const TString& fitNtupleFileName, UInt_t nExpt, UInt_t firstExpt = 0, Bool_t useAsymmErrors = kFALSE, Bool_t twoStageFit = kFALSE );


                virtual void withinAsymErrorCalc(Bool_t inAsymErrCalc) { withinAsymErrorCalc_ = inAsymErrCalc; }


                virtual void setParsFromMinuit(Double_t* par, Int_t npar);


                virtual Double_t getTotNegLogLikelihood();


                virtual void addConstraint(const TString& formula, const std::vector<TString>& pars, const Double_t mean, const Double_t width);

        protected:
                void printParInfo() const;

                void initialise();

                void initSockets();

                void getParametersFromSlaves();

                void getParametersFromSlavesFirstTime();

                void updateParametersFromSlaves();

                void checkParameter( const LauParameter* param, UInt_t index ) const;

                void addConParameters();

                Double_t getLogLikelihoodPenalty();


                Bool_t readData();


                Bool_t cacheInputData();

                void fitExpt( Bool_t useAsymmErrors, Bool_t twoStageFit );

                void checkInitFitParams();

                Bool_t finalise();

                Bool_t writeOutResults();


        private:
                // Setup a struct to store information on constrained fit parameters
                struct StoreConstraints {
                        TString formula_;
                        std::vector<TString> conPars_;
                        Double_t mean_;
                        Double_t width_;
                };

                std::vector<StoreConstraints> storeCon_;

                const UInt_t nSlaves_;

                const UInt_t reqPort_;

                UInt_t nParams_; 

                UInt_t nFreeParams_;

                Bool_t withinAsymErrorCalc_;

                UInt_t numberOKFits_;

                UInt_t numberBadFits_;

                Int_t fitStatus_;

                UInt_t iExpt_;

                Double_t NLL_;

                TMatrixD covMatrix_;

                std::vector<TMatrixD> covMatrices_;

                TMonitor* socketMonitor_;

                std::vector<TSocket*> sSlaves_;

                std::vector<TMessage*> messagesToSlaves_;

                TMessage* messageFromSlave_;

                std::map< TString, UInt_t > parIndices_;

                std::map< UInt_t, TString > parNames_;

                std::vector<LauParameter*> params_;

                std::vector<LauAbsRValue*> conVars_;

                std::vector<Double_t> parValues_;

                std::vector< std::vector<UInt_t> > slaveIndices_;

                std::vector<Double_t*> vectorPar_;

                std::vector<Double_t> vectorRes_;

                TStopwatch timer_; 

                TStopwatch cumulTimer_;

                LauFitNtuple* fitNtuple_;

                ClassDef(LauSimFitMaster,0);
};

#endif