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 "TStopwatch.h"
#include "TString.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, const UInt_t nExp, const UInt_t firstExp = 0, const Bool_t useAsymmErrors = kFALSE, const Bool_t doTwoStageFit = kFALSE );


                virtual void withinAsymErrorCalc(const Bool_t inAsymErrCalc);

                // Need to unshadow the query method defined in the base class
                using LauFitObject::withinAsymErrorCalc;


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


                virtual Double_t getTotNegLogLikelihood();

        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();

                void checkInitFitParams();

                Bool_t finalise();

                Bool_t writeOutResults();


        private:
                LauSimFitMaster(const LauSimFitMaster& rhs);

                LauSimFitMaster& operator=(const LauSimFitMaster& rhs);

                std::vector<StoreConstraints> storeCon_;

                const UInt_t nSlaves_;

                const UInt_t reqPort_;

                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< std::vector<UInt_t> > slaveFreeIndices_;

                std::vector<Double_t*> vectorPar_;

                std::vector<Double_t> vectorRes_;

                TStopwatch timer_; 

                TStopwatch cumulTimer_;

                LauFitNtuple* fitNtuple_;

                ClassDef(LauSimFitMaster,0);
};

#endif