LauSimFitCoordinator.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_COORDINATOR
#define LAU_SIM_FIT_COORDINATOR
 
#include "LauFitObject.hh"
 
#include "TMatrixD.h"
#include "TStopwatch.h"
#include "TString.h"
 
#include <map>
#include <vector>
 
class TMessage;
class TMonitor;
class TSocket;
class LauAbsRValue;
class LauParameter;
class LauFitNtuple;
 
class LauSimFitCoordinator : public LauFitObject {
 
  public:
 
    LauSimFitCoordinator( UInt_t numTasks, UInt_t port = 9090 );
 
    virtual ~LauSimFitCoordinator();
 
 
    void runSimFit( const TString& fitNtupleFileName,
                    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 getParametersFromTasks();
 
    void getParametersFromTasksFirstTime();
 
    void updateParametersFromTasks();
 
    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:
    LauSimFitCoordinator( const LauSimFitCoordinator& rhs );
 
    LauSimFitCoordinator& operator=( const LauSimFitCoordinator& rhs );
 
    const UInt_t nTasks_;
 
    const UInt_t reqPort_;
 
    std::vector<TMatrixD> covMatrices_;
 
    TMonitor* socketMonitor_;
 
    std::vector<TSocket*> socketTasks_;
 
    std::vector<TMessage*> messagesToTasks_;
 
    TMessage* messageFromTask_;
 
    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>> taskIndices_;
 
    std::vector<std::vector<UInt_t>> taskFreeIndices_;
 
    std::vector<Double_t*> vectorPar_;
 
    std::vector<Double_t> vectorRes_;
 
    TStopwatch timer_;
 
    TStopwatch cumulTimer_;
 
    LauFitNtuple* fitNtuple_;
 
    ClassDef( LauSimFitCoordinator, 0 );
};
 
#endif