LauSimFitMaster.hh

Go to the documentation of this file.

/*
Copyright 2013 University of Warwick

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

    http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/

/*
Laura++ package authors:
John Back
Paul Harrison
Thomas Latham
*/

#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