/*
<img src="gif/TFitEditor.gif">
*/
//End_Html
#include "TFitEditor.h"
#include "TROOT.h"
#include "TClass.h"
#include "TCanvas.h"
#include "TGTab.h"
#include "TGLabel.h"
#include "TG3DLine.h"
#include "TGComboBox.h"
#include "TGTextEntry.h"
#include "TGFont.h"
#include "TGGC.h"
#include "TGButtonGroup.h"
#include "TGNumberEntry.h"
#include "TGDoubleSlider.h"
#include "TGStatusBar.h"
#include "TFitParametersDialog.h"
#include "TGMsgBox.h"
#include "TAxis.h"
#include "TGraph.h"
#include "TGraph2D.h"
#include "TH1.h"
#include "TF1.h"
#include "TTimer.h"
#include "THStack.h"
#include "TVirtualFitter.h"
#include "TMath.h"
enum EFitPanel {
kFP_FLIST, kFP_GAUS, kFP_GAUSN, kFP_EXPO, kFP_LAND, kFP_LANDN,
kFP_POL0, kFP_POL1, kFP_POL2, kFP_POL3, kFP_POL4, kFP_POL5,
kFP_POL6, kFP_POL7, kFP_POL8, kFP_POL9, kFP_USER,
kFP_NONE, kFP_ADD, kFP_CONV, kFP_FILE, kFP_PARS, kFP_RBUST, kFP_EMPW1,
kFP_INTEG, kFP_IMERR, kFP_USERG, kFP_ADDLS, kFP_ALLW1, kFP_IFITR, kFP_NOCHI,
kFP_MLIST, kFP_MCHIS, kFP_MBINL, kFP_MUBIN, kFP_MUSER, kFP_MLINF, kFP_MUSR,
kFP_DSAME, kFP_DNONE, kFP_DADVB, kFP_DNOST, kFP_PDEF, kFP_PVER, kFP_PQET,
kFP_XMIN, kFP_XMAX, kFP_YMIN, kFP_YMAX, kFP_ZMIN, kFP_ZMAX,
kFP_LMIN, kFP_LMIN2, kFP_LFUM, kFP_MIGRAD,kFP_SIMPLX,kFP_FUMILI,
kFP_MERR, kFP_MTOL, kFP_MITR,
kFP_FIT, kFP_RESET, kFP_CLOSE
};
ClassImp(TFitEditor)
TFitEditor *TFitEditor::fgFitDialog = 0;
void TFitEditor::Open(TVirtualPad* pad, TObject *obj)
{
if (!fgFitDialog) {
TFitEditor::GetFP() = new TFitEditor(pad, obj);
} else {
fgFitDialog->Show(pad, obj);
}
}
TFitEditor::TFitEditor(TVirtualPad* pad, TObject *obj) :
TGMainFrame(gClient->GetRoot(), 20, 20),
fCanvas (0),
fParentPad (0),
fFitObject (0),
fDim (0),
fXaxis (0),
fYaxis (0),
fZaxis (0),
fXmin (0),
fXmax (0),
fYmin (0),
fYmax (0),
fZmin (0),
fZmax (0),
fPlus ('+'),
fFunction (""),
fFitOption (""),
fDrawOption (""),
fFitFunc (0)
{
SetCleanup(kDeepCleanup);
fFitFuncList = new TList();
TString name = obj->GetName();
name.Append("::");
name.Append(obj->ClassName());
fObjLabelParent = new TGHorizontalFrame(this, 80, 20);
TGLabel *label = new TGLabel(fObjLabelParent,"Current selection: ");
fObjLabelParent->AddFrame(label, new TGLayoutHints(kLHintsLeft, 1, 1, 0, 0));
fObjLabel = new TGLabel(fObjLabelParent, Form("%s", name.Data()));
fObjLabelParent->AddFrame(fObjLabel, new TGLayoutHints(kLHintsLeft, 1, 1, 0, 0));
AddFrame(fObjLabelParent, new TGLayoutHints(kLHintsTop, 1, 1, 10, 10));
Pixel_t color;
gClient->GetColorByName("#ff0000", color);
fObjLabel->SetTextColor(color, kFALSE);
fObjLabel->SetTextJustify(kTextLeft);
fTab = new TGTab(this, 10, 10);
AddFrame(fTab, new TGLayoutHints(kLHintsExpandY | kLHintsExpandX));
fTab->SetCleanup(kDeepCleanup);
fTab->Associate(this);
TGHorizontalFrame *cf1 = new TGHorizontalFrame(this, 250, 20, kFixedWidth);
cf1->SetCleanup(kDeepCleanup);
fFitButton = new TGTextButton(cf1, "&Fit", kFP_FIT);
fFitButton->Associate(this);
cf1->AddFrame(fFitButton, new TGLayoutHints(kLHintsTop |
kLHintsExpandX, 2, 2, 2, 2));
fResetButton = new TGTextButton(cf1, "&Reset", kFP_RESET);
fResetButton->Associate(this);
cf1->AddFrame(fResetButton, new TGLayoutHints(kLHintsTop |
kLHintsExpandX, 3, 2, 2, 2));
fCloseButton = new TGTextButton(cf1, "&Close", kFP_CLOSE);
fCloseButton->Associate(this);
cf1->AddFrame(fCloseButton, new TGLayoutHints(kLHintsTop |
kLHintsExpandX, 3, 2, 2, 2));
AddFrame(cf1, new TGLayoutHints(kLHintsNormal |
kLHintsRight, 0, 5, 5, 5));
int parts[] = { 20, 20, 20, 20, 20 };
fStatusBar = new TGStatusBar(this, 10, 10);
fStatusBar->SetParts(parts, 5);
AddFrame(fStatusBar, new TGLayoutHints(kLHintsBottom |
kLHintsLeft |
kLHintsExpandX));
CreateGeneralTab();
CreateMinimizationTab();
gROOT->GetListOfCleanups()->Add(this);
MapSubwindows();
fGeneral->HideFrame(fSliderYParent);
fGeneral->HideFrame(fSliderZParent);
TGDimension size = GetDefaultSize();
SetWMSize(size.fWidth, size.fHeight);
SetWMSizeHints(size.fWidth, size.fHeight, size.fWidth, size.fHeight, 0, 0);
SetWindowName("Fit Panel");
SetIconName("Fit Panel");
SetClassHints("Fit Panel", "Fit Panel");
SetMWMHints(kMWMDecorAll | kMWMDecorResizeH | kMWMDecorMaximize |
kMWMDecorMinimize | kMWMDecorMenu,
kMWMFuncAll | kMWMFuncResize | kMWMFuncMaximize |
kMWMFuncMinimize,
kMWMInputModeless);
if (pad && obj) {
fParentPad = (TPad *)pad;
fFitObject = (TObject *)obj;
fDrawOption = GetDrawOption();
SetCanvas(pad->GetCanvas());
pad->GetCanvas()->Selected(pad, obj, kButton1Down);
} else {
Error("FitPanel", "need to have an object drawn first");
return;
}
UInt_t dw = fClient->GetDisplayWidth();
UInt_t cw = pad->GetCanvas()->GetWindowWidth();
UInt_t cx = (UInt_t)pad->GetCanvas()->GetWindowTopX();
UInt_t cy = (UInt_t)pad->GetCanvas()->GetWindowTopY();
if (cw + size.fWidth < dw) {
Int_t gedx = 0, gedy = 0;
gedx = cx+cw+4;
gedy = cy-20;
MoveResize(gedx, gedy,size.fWidth, size.fHeight);
SetWMPosition(gedx, gedy);
}
Resize(size);
MapWindow();
gVirtualX->RaiseWindow(GetId());
}
TFitEditor::~TFitEditor()
{
DisconnectSlots();
fCloseButton->Disconnect("Clicked()");
TQObject::Disconnect("TCanvas", "Selected(TVirtualPad *, TObject *, Int_t)");
gROOT->GetListOfCleanups()->Remove(this);
if (fFitFunc) delete fFitFunc;
Cleanup();
delete fLayoutNone;
delete fLayoutAdd;
delete fLayoutConv;
fgFitDialog = 0;
}
void TFitEditor::CreateGeneralTab()
{
fTabContainer = fTab->AddTab("General");
fGeneral = new TGCompositeFrame(fTabContainer, 10, 10, kVerticalFrame);
fTabContainer->AddFrame(fGeneral, new TGLayoutHints(kLHintsTop |
kLHintsExpandX,
5, 5, 2, 2));
TGGroupFrame *gf1 = new TGGroupFrame(fGeneral, "Function", kFitWidth);
TGCompositeFrame *tf1 = new TGCompositeFrame(gf1, 350, 26,
kHorizontalFrame);
TGVerticalFrame *tf11 = new TGVerticalFrame(tf1);
TGLabel *label1 = new TGLabel(tf11,"Predefined:");
tf11->AddFrame(label1, new TGLayoutHints(kLHintsNormal, 0, 0, 2, 0));
fFuncList = BuildFunctionList(tf11, kFP_FLIST);
fFuncList->Resize(80, 20);
fFuncList->Select(1, kFALSE);
TGListBox *lb = fFuncList->GetListBox();
lb->Resize(lb->GetWidth(), 200);
tf11->AddFrame(fFuncList, new TGLayoutHints(kLHintsNormal, 0, 0, 5, 0));
fFuncList->Associate(this);
tf1->AddFrame(tf11);
TGHButtonGroup *bgr = new TGHButtonGroup(tf1,"Operation");
bgr->SetRadioButtonExclusive();
fNone = new TGRadioButton(bgr, "Nop", kFP_NONE);
fNone->SetToolTipText("No operation defined");
fNone->SetState(kButtonDown, kFALSE);
fAdd = new TGRadioButton(bgr, "Add", kFP_ADD);
fAdd->SetToolTipText("Addition");
fConv = new TGRadioButton(bgr, "Conv", kFP_CONV);
fConv->SetToolTipText("Convolution (not implemented yet)");
fConv->SetState(kButtonDisabled);
fLayoutNone = new TGLayoutHints(kLHintsLeft,0,5,3,-10);
fLayoutAdd = new TGLayoutHints(kLHintsLeft,10,5,3,-10);
fLayoutConv = new TGLayoutHints(kLHintsLeft,10,5,3,-10);
bgr->SetLayoutHints(fLayoutNone,fNone);
bgr->SetLayoutHints(fLayoutAdd,fAdd);
bgr->SetLayoutHints(fLayoutConv,fConv);
bgr->Show();
bgr->ChangeOptions(kFitWidth | kHorizontalFrame);
tf1->AddFrame(bgr, new TGLayoutHints(kLHintsNormal, 15, 0, 3, 0));
gf1->AddFrame(tf1, new TGLayoutHints(kLHintsNormal | kLHintsExpandX));
TGCompositeFrame *tf2 = new TGCompositeFrame(gf1, 350, 26,
kHorizontalFrame);
fEnteredFunc = new TGTextEntry(tf2, new TGTextBuffer(50), kFP_FILE);
fEnteredFunc->SetMaxLength(250);
fEnteredFunc->SetAlignment(kTextLeft);
TGTextLBEntry *te = (TGTextLBEntry *)fFuncList->GetSelectedEntry();
fFunction = te->GetTitle();
fEnteredFunc->SetText(fFunction.Data());
fEnteredFunc->SetToolTipText("Enter file_name/function_name or a function expression");
fEnteredFunc->Resize(250,fEnteredFunc->GetDefaultHeight());
tf2->AddFrame(fEnteredFunc, new TGLayoutHints(kLHintsLeft |
kLHintsCenterY |
kLHintsExpandX, 2, 2, 2, 2));
gf1->AddFrame(tf2, new TGLayoutHints(kLHintsNormal |
kLHintsExpandX, 0, 0, 2, 0));
TGHorizontalFrame *s1 = new TGHorizontalFrame(gf1);
TGLabel *label21 = new TGLabel(s1, "Selected: ");
s1->AddFrame(label21, new TGLayoutHints(kLHintsNormal |
kLHintsCenterY, 2, 2, 2, 0));
TGHorizontal3DLine *hlines = new TGHorizontal3DLine(s1);
s1->AddFrame(hlines, new TGLayoutHints(kLHintsCenterY | kLHintsExpandX));
gf1->AddFrame(s1, new TGLayoutHints(kLHintsExpandX));
TGCompositeFrame *tf4 = new TGCompositeFrame(gf1, 350, 26,
kHorizontalFrame);
TGTextLBEntry *txt = (TGTextLBEntry *)fFuncList->GetSelectedEntry();
fSelLabel = new TGLabel(tf4, Form("%s", txt->GetTitle()));
tf4->AddFrame(fSelLabel, new TGLayoutHints(kLHintsNormal |
kLHintsCenterY, 0, 6, 2, 0));
Pixel_t color;
gClient->GetColorByName("#336666", color);
fSelLabel->SetTextColor(color, kFALSE);
TGCompositeFrame *tf5 = new TGCompositeFrame(tf4, 120, 20,
kHorizontalFrame | kFixedWidth);
fSetParam = new TGTextButton(tf5, "Set Parameters...", kFP_PARS);
tf5->AddFrame(fSetParam, new TGLayoutHints(kLHintsRight |
kLHintsCenterY |
kLHintsExpandX));
fSetParam->SetToolTipText("Open a dialog for parameter(s) settings");
tf4->AddFrame(tf5, new TGLayoutHints(kLHintsRight |
kLHintsTop, 5, 0, 2, 2));
gf1->AddFrame(tf4, new TGLayoutHints(kLHintsNormal |
kLHintsExpandX, 5, 0, 0, 0));
fGeneral->AddFrame(gf1, new TGLayoutHints(kLHintsExpandX, 5, 5, 0, 0));
TGGroupFrame *gf = new TGGroupFrame(fGeneral, "Fit Settings", kFitWidth);
TGHorizontalFrame *h1 = new TGHorizontalFrame(gf);
TGLabel *label4 = new TGLabel(h1, "Method");
h1->AddFrame(label4, new TGLayoutHints(kLHintsNormal |
kLHintsCenterY, 2, 2, 0, 0));
TGHorizontal3DLine *hline1 = new TGHorizontal3DLine(h1);
h1->AddFrame(hline1, new TGLayoutHints(kLHintsCenterY | kLHintsExpandX));
gf->AddFrame(h1, new TGLayoutHints(kLHintsExpandX));
TGHorizontalFrame *h2 = new TGHorizontalFrame(gf);
TGVerticalFrame *v1 = new TGVerticalFrame(h2);
fMethodList = BuildMethodList(v1, kFP_MLIST);
fMethodList->Select(1, kFALSE);
fMethodList->Resize(130, 20);
lb = fMethodList->GetListBox();
Int_t lbe = lb->GetNumberOfEntries();
lb->Resize(lb->GetWidth(), lbe*16);
v1->AddFrame(fMethodList, new TGLayoutHints(kLHintsLeft, 0, 0, 2, 5));
fLinearFit = new TGCheckButton(v1, "Linear fit", kFP_MLINF);
fLinearFit->Associate(this);
fLinearFit->SetToolTipText("Perform Linear fitter if selected");
v1->AddFrame(fLinearFit, new TGLayoutHints(kLHintsNormal, 0, 0, 2, 2));
TGHorizontalFrame *v1h = new TGHorizontalFrame(v1);
TGLabel *label41 = new TGLabel(v1h, "Robust:");
v1h->AddFrame(label41, new TGLayoutHints(kLHintsNormal |
kLHintsCenterY, 25, 5, 5, 2));
fRobustValue = new TGNumberEntry(v1h, 1., 5, kFP_RBUST,
TGNumberFormat::kNESRealTwo,
TGNumberFormat::kNEAPositive,
TGNumberFormat::kNELLimitMinMax,0.,1.);
v1h->AddFrame(fRobustValue, new TGLayoutHints(kLHintsLeft));
v1->AddFrame(v1h, new TGLayoutHints(kLHintsNormal));
fRobustValue->SetState(kFALSE);
fRobustValue->GetNumberEntry()->SetToolTipText("Available only for graphs");
h2->AddFrame(v1, new TGLayoutHints(kLHintsExpandX | kLHintsExpandY));
TGVerticalFrame *v2 = new TGVerticalFrame(h2);
TGCompositeFrame *v21 = new TGCompositeFrame(v2, 120, 20,
kHorizontalFrame | kFixedWidth);
fUserButton = new TGTextButton(v21, "User-Defined...", kFP_MUSR);
v21->AddFrame(fUserButton, new TGLayoutHints(kLHintsRight |
kLHintsCenterY |
kLHintsExpandX));
fUserButton->SetToolTipText("Open a dialog for entering a user-defined method");
fUserButton->SetState(kButtonDisabled);
v2->AddFrame(v21, new TGLayoutHints(kLHintsRight | kLHintsTop));
fNoChi2 = new TGCheckButton(v2, "No Chi-square", kFP_NOCHI);
fNoChi2->Associate(this);
fNoChi2->SetToolTipText("'C'- do not calculate Chi-square (for Linear fitter)");
v2->AddFrame(fNoChi2, new TGLayoutHints(kLHintsNormal, 0, 0, 34, 2));
h2->AddFrame(v2, new TGLayoutHints(kLHintsExpandX | kLHintsExpandY, 20, 0, 0, 0));
gf->AddFrame(h2, new TGLayoutHints(kLHintsExpandX, 20, 0, 0, 0));
TGHorizontalFrame *h3 = new TGHorizontalFrame(gf);
TGLabel *label5 = new TGLabel(h3, "Fit Options");
h3->AddFrame(label5, new TGLayoutHints(kLHintsNormal |
kLHintsCenterY, 2, 2, 0, 0));
TGHorizontal3DLine *hline2 = new TGHorizontal3DLine(h3);
h3->AddFrame(hline2, new TGLayoutHints(kLHintsCenterY | kLHintsExpandX));
gf->AddFrame(h3, new TGLayoutHints(kLHintsExpandX));
TGHorizontalFrame *h = new TGHorizontalFrame(gf);
TGVerticalFrame *v3 = new TGVerticalFrame(h);
fIntegral = new TGCheckButton(v3, "Integral", kFP_INTEG);
fIntegral->Associate(this);
fIntegral->SetToolTipText("'I'- use integral of function instead of value in bin center");
v3->AddFrame(fIntegral, new TGLayoutHints(kLHintsNormal, 0, 0, 2, 2));
fBestErrors = new TGCheckButton(v3, "Best errors", kFP_IMERR);
fBestErrors->Associate(this);
fBestErrors->SetToolTipText("'E'- better errors estimation using Minos technique");
v3->AddFrame(fBestErrors, new TGLayoutHints(kLHintsNormal, 0, 0, 2, 2));
fAllWeights1 = new TGCheckButton(v3, "All weights = 1", kFP_ALLW1);
fAllWeights1->Associate(this);
fAllWeights1->SetToolTipText("'W'- all weights=1 for non empty bins; error bars ignored");
v3->AddFrame(fAllWeights1, new TGLayoutHints(kLHintsNormal, 0, 0, 2, 2));
fEmptyBinsWghts1 = new TGCheckButton(v3, "Empty bins, weights=1", kFP_EMPW1);
fEmptyBinsWghts1->Associate(this);
fEmptyBinsWghts1->SetToolTipText("'WW'- all weights=1 including empty bins; error bars ignored");
v3->AddFrame(fEmptyBinsWghts1, new TGLayoutHints(kLHintsNormal, 0, 0, 2, 2));
h->AddFrame(v3, new TGLayoutHints(kLHintsExpandX | kLHintsExpandY));
TGVerticalFrame *v4 = new TGVerticalFrame(h);
fUseRange = new TGCheckButton(v4, "Use range", kFP_USERG);
fUseRange->Associate(this);
fUseRange->SetToolTipText("'R'- fit only data within the specified function range");
v4->AddFrame(fUseRange, new TGLayoutHints(kLHintsNormal, 0, 0, 2, 2));
fImproveResults = new TGCheckButton(v4, "Improve fit results", kFP_IFITR);
fImproveResults->Associate(this);
fImproveResults->SetToolTipText("'M'- after minimum is found, search for a new one");
v4->AddFrame(fImproveResults, new TGLayoutHints(kLHintsNormal, 0, 0, 2, 2));
fAdd2FuncList = new TGCheckButton(v4, "Add to list", kFP_ADDLS);
fAdd2FuncList->Associate(this);
fAdd2FuncList->SetToolTipText("'+'- add function to the list without deleting the previous");
v4->AddFrame(fAdd2FuncList, new TGLayoutHints(kLHintsNormal, 0, 0, 2, 2));
h->AddFrame(v4, new TGLayoutHints(kLHintsExpandX | kLHintsExpandY, 20, 0, 0, 0));
gf->AddFrame(h, new TGLayoutHints(kLHintsExpandX, 20, 0, 0, 0));
TGHorizontalFrame *h5 = new TGHorizontalFrame(gf);
TGLabel *label6 = new TGLabel(h5, "Draw Options");
h5->AddFrame(label6, new TGLayoutHints(kLHintsNormal |
kLHintsCenterY, 2, 2, 2, 2));
TGHorizontal3DLine *hline3 = new TGHorizontal3DLine(h5);
h5->AddFrame(hline3, new TGLayoutHints(kLHintsCenterY | kLHintsExpandX));
gf->AddFrame(h5, new TGLayoutHints(kLHintsExpandX));
TGHorizontalFrame *h6 = new TGHorizontalFrame(gf);
TGVerticalFrame *v5 = new TGVerticalFrame(h6);
fDrawSame = new TGCheckButton(v5, "SAME", kFP_DSAME);
fDrawSame->Associate(this);
fDrawSame->SetToolTipText("Superimpose on previous picture in the same pad");
v5->AddFrame(fDrawSame, new TGLayoutHints(kLHintsNormal, 0, 0, 2, 2));
fNoDrawing = new TGCheckButton(v5, "No drawing", kFP_DNONE);
fNoDrawing->Associate(this);
fNoDrawing->SetToolTipText("'0'- do not draw function graphics");
v5->AddFrame(fNoDrawing, new TGLayoutHints(kLHintsNormal, 0, 0, 2, 2));
fNoStoreDrawing = new TGCheckButton(v5, "Do not store/draw", kFP_DNOST);
fNoStoreDrawing->Associate(this);
fNoStoreDrawing->SetToolTipText("'N'- do not store the function, do not draw it");
v5->AddFrame(fNoStoreDrawing, new TGLayoutHints(kLHintsNormal, 0, 0, 2, 2));
h6->AddFrame(v5, new TGLayoutHints(kLHintsExpandX | kLHintsExpandY));
TGVerticalFrame *v6 = new TGVerticalFrame(h6);
TGCompositeFrame *v61 = new TGCompositeFrame(v6, 120, 20,
kHorizontalFrame | kFixedWidth);
fDrawAdvanced = new TGTextButton(v61, "Advanced...", kFP_DADVB);
v61->AddFrame(fDrawAdvanced, new TGLayoutHints(kLHintsRight |
kLHintsCenterY |
kLHintsExpandX));
fDrawAdvanced->SetToolTipText("Open a dialog for advanced draw options");
fDrawAdvanced->SetState(kButtonDisabled);
v6->AddFrame(v61, new TGLayoutHints(kLHintsRight | kLHintsTop,
0, 0, (4+fDrawSame->GetHeight())*2, 0));
h6->AddFrame(v6, new TGLayoutHints(kLHintsExpandX | kLHintsExpandY));
gf->AddFrame(h6, new TGLayoutHints(kLHintsExpandX, 20, 0, 2, 0));
fGeneral->AddFrame(gf, new TGLayoutHints(kLHintsExpandX |
kLHintsExpandY, 5, 5, 0, 0));
fSliderXParent = new TGHorizontalFrame(fGeneral);
TGLabel *label8 = new TGLabel(fSliderXParent, "X:");
fSliderXParent->AddFrame(label8, new TGLayoutHints(kLHintsLeft |
kLHintsCenterY, 0, 5, 0, 0));
fSliderX = new TGDoubleHSlider(fSliderXParent, 1, kDoubleScaleBoth);
fSliderX->SetScale(5);
fSliderXParent->AddFrame(fSliderX, new TGLayoutHints(kLHintsExpandX |
kLHintsCenterY));
fGeneral->AddFrame(fSliderXParent, new TGLayoutHints(kLHintsExpandX, 5, 5, 0, 0));
fSliderYParent = new TGHorizontalFrame(fGeneral);
TGLabel *label9 = new TGLabel(fSliderYParent, "Y:");
fSliderYParent->AddFrame(label9, new TGLayoutHints(kLHintsLeft |
kLHintsCenterY, 0, 5, 0, 0));
fSliderY = new TGDoubleHSlider(fSliderYParent, 1, kDoubleScaleBoth);
fSliderY->SetScale(5);
fSliderYParent->AddFrame(fSliderY, new TGLayoutHints(kLHintsExpandX |
kLHintsCenterY));
fGeneral->AddFrame(fSliderYParent, new TGLayoutHints(kLHintsExpandX, 5, 5, 0, 0));
fSliderZParent = new TGHorizontalFrame(fGeneral);
TGLabel *label10 = new TGLabel(fSliderZParent, "Z:");
fSliderZParent->AddFrame(label10, new TGLayoutHints(kLHintsLeft |
kLHintsCenterY, 0, 5, 0, 0));
fSliderZ = new TGDoubleHSlider(fSliderZParent, 1, kDoubleScaleBoth);
fSliderZ->SetScale(5);
fSliderZParent->AddFrame(fSliderZ, new TGLayoutHints(kLHintsExpandX |
kLHintsCenterY));
fGeneral->AddFrame(fSliderZParent, new TGLayoutHints(kLHintsExpandX, 5, 5, 0, 0));
}
void TFitEditor::CreateMinimizationTab()
{
fTabContainer = fTab->AddTab("Minimization");
fMinimization = new TGCompositeFrame(fTabContainer, 10, 10, kVerticalFrame);
fTabContainer->AddFrame(fMinimization, new TGLayoutHints(kLHintsTop |
kLHintsExpandX,
5, 5, 2, 2));
MakeTitle(fMinimization, "Library");
TGHorizontalFrame *hl = new TGHorizontalFrame(fMinimization);
fLibMinuit = new TGRadioButton(hl, "Minuit", kFP_LMIN);
fLibMinuit->Associate(this);
fLibMinuit->SetToolTipText("Use minimization from libMinuit (default)");
hl->AddFrame(fLibMinuit, new TGLayoutHints(kLHintsNormal, 40, 0, 0, 1));
fLibMinuit->SetState(kButtonDown);
fStatusBar->SetText("LIB Minuit",0);
fLibMinuit2 = new TGRadioButton(hl, "Minuit2", kFP_LMIN2);
fLibMinuit2->Associate(this);
fLibMinuit2->SetToolTipText("New C++ version of Minuit");
hl->AddFrame(fLibMinuit2, new TGLayoutHints(kLHintsNormal, 35, 0, 0, 1));
fLibFumili = new TGRadioButton(hl, "Fumili", kFP_LFUM);
fLibFumili->Associate(this);
fLibFumili->SetToolTipText("Use minimization from libFumili");
hl->AddFrame(fLibFumili, new TGLayoutHints(kLHintsNormal, 30, 0, 0, 1));
fMinimization->AddFrame(hl, new TGLayoutHints(kLHintsExpandX, 20, 0, 5, 1));
MakeTitle(fMinimization, "Method");
TGHorizontalFrame *hm = new TGHorizontalFrame(fMinimization);
fMigrad = new TGRadioButton(hm, "MIGRAD", kFP_MIGRAD);
fMigrad->Associate(this);
fMigrad->SetToolTipText("Use MIGRAD as minimization method");
hm->AddFrame(fMigrad, new TGLayoutHints(kLHintsNormal, 40, 0, 0, 1));
fMigrad->SetState(kButtonDown);
fStatusBar->SetText("MIGRAD",1);
fSimplex = new TGRadioButton(hm, "SIMPLEX", kFP_SIMPLX);
fSimplex->Associate(this);
fSimplex->SetToolTipText("Use SIMPLEX as minimization method");
fSimplex->SetState(kButtonDisabled);
hm->AddFrame(fSimplex, new TGLayoutHints(kLHintsNormal, 20, 0, 0, 1));
fFumili = new TGRadioButton(hm, "FUMILI", kFP_FUMILI);
fFumili->Associate(this);
fFumili->SetToolTipText("Use FUMILI as minimization method");
fFumili->SetState(kButtonDisabled);
hm->AddFrame(fFumili, new TGLayoutHints(kLHintsNormal, 18, 0, 0, 1));
fMinimization->AddFrame(hm, new TGLayoutHints(kLHintsExpandX, 20, 0, 5, 1));
MakeTitle(fMinimization, "Settings");
TGLabel *hslabel1 = new TGLabel(fMinimization,"Use ENTER key to validate a new value or click");
fMinimization->AddFrame(hslabel1, new TGLayoutHints(kLHintsNormal, 61, 0, 5, 1));
TGLabel *hslabel2 = new TGLabel(fMinimization,"on Reset button to set the defaults.");
fMinimization->AddFrame(hslabel2, new TGLayoutHints(kLHintsNormal, 61, 0, 1, 10));
TGHorizontalFrame *hs = new TGHorizontalFrame(fMinimization);
TGVerticalFrame *hsv1 = new TGVerticalFrame(hs, 180, 10, kFixedWidth);
TGLabel *errlabel = new TGLabel(hsv1,"Error definition (default = 1): ");
hsv1->AddFrame(errlabel, new TGLayoutHints(kLHintsLeft | kLHintsCenterY,
1, 1, 5, 7));
TGLabel *tollabel = new TGLabel(hsv1,"Max tolerance (precision): ");
hsv1->AddFrame(tollabel, new TGLayoutHints(kLHintsLeft | kLHintsCenterY,
1, 1, 5, 7));
TGLabel *itrlabel = new TGLabel(hsv1,"Max number of iterations: ");
hsv1->AddFrame(itrlabel, new TGLayoutHints(kLHintsLeft | kLHintsCenterY,
1, 1, 5, 5));
hs->AddFrame(hsv1, new TGLayoutHints(kLHintsNormal, 60, 0, 0, 0));
TGVerticalFrame *hsv2 = new TGVerticalFrame(hs, 90,10, kFixedWidth);
fErrorScale = new TGNumberEntryField(hsv2, kFP_MERR, 1.0,
TGNumberFormat::kNESRealTwo,
TGNumberFormat::kNEAPositive,
TGNumberFormat::kNELLimitMinMax,0.,100.);
hsv2->AddFrame(fErrorScale, new TGLayoutHints(kLHintsLeft | kLHintsExpandX,
1, 1, 0, 3));
fTolerance = new TGNumberEntryField(hsv2, kFP_MTOL, 1.0E-9,
TGNumberFormat::kNESReal,
TGNumberFormat::kNEAPositive,
TGNumberFormat::kNELLimitMinMax, 0., 1.);
fTolerance->SetNumber(TVirtualFitter::GetPrecision());
hsv2->AddFrame(fTolerance, new TGLayoutHints(kLHintsLeft | kLHintsExpandX,
1, 1, 3, 3));
fIterations = new TGNumberEntryField(hsv2, kFP_MITR, 5000,
TGNumberFormat::kNESInteger,
TGNumberFormat::kNEAPositive,
TGNumberFormat::kNELNoLimits);
fIterations->SetNumber(TVirtualFitter::GetMaxIterations());
hsv2->AddFrame(fIterations, new TGLayoutHints(kLHintsLeft | kLHintsExpandX,
1, 1, 3, 3));
hs->AddFrame(hsv2, new TGLayoutHints(kLHintsNormal, 0, 0, 0, 0));
fMinimization->AddFrame(hs, new TGLayoutHints(kLHintsExpandX, 0, 0, 1, 1));
fStatusBar->SetText("Itr: 5000",2);
MakeTitle(fMinimization, "Print Options");
TGHorizontalFrame *h8 = new TGHorizontalFrame(fMinimization);
fOptDefault = new TGRadioButton(h8, "Default", kFP_PDEF);
fOptDefault->Associate(this);
fOptDefault->SetToolTipText("Default is between Verbose and Quiet");
h8->AddFrame(fOptDefault, new TGLayoutHints(kLHintsNormal, 40, 0, 0, 1));
fOptDefault->SetState(kButtonDown);
fStatusBar->SetText("Prn: DEF",3);
fOptVerbose = new TGRadioButton(h8, "Verbose", kFP_PVER);
fOptVerbose->Associate(this);
fOptVerbose->SetToolTipText("'V'- print results after each iteration");
h8->AddFrame(fOptVerbose, new TGLayoutHints(kLHintsNormal, 30, 0, 0, 1));
fOptQuiet = new TGRadioButton(h8, "Quiet", kFP_PQET);
fOptQuiet->Associate(this);
fOptQuiet->SetToolTipText("'Q'- no print");
h8->AddFrame(fOptQuiet, new TGLayoutHints(kLHintsNormal, 25, 0, 0, 1));
fMinimization->AddFrame(h8, new TGLayoutHints(kLHintsExpandX, 20, 0, 5, 1));
}
void TFitEditor::ConnectSlots()
{
fFuncList->Connect("Selected(Int_t)", "TFitEditor", this, "DoFunction(Int_t)");
fEnteredFunc->Connect("ReturnPressed()", "TFitEditor", this, "DoEnteredFunction()");
fSetParam->Connect("Clicked()", "TFitEditor", this, "DoSetParameters()");
fNone->Connect("Toggled(Bool_t)","TFitEditor", this, "DoNoOperation(Bool_t)");
fAdd->Connect("Toggled(Bool_t)","TFitEditor", this, "DoAddition(Bool_t)");
fIntegral->Connect("Toggled(Bool_t)","TFitEditor",this,"DoIntegral()");
fBestErrors->Connect("Toggled(Bool_t)","TFitEditor",this,"DoBestErrors()");
fUseRange->Connect("Toggled(Bool_t)","TFitEditor",this,"DoUseRange()");
fAdd2FuncList->Connect("Toggled(Bool_t)","TFitEditor",this,"DoAddtoList()");
fAllWeights1->Connect("Toggled(Bool_t)","TFitEditor",this,"DoAllWeights1()");
fEmptyBinsWghts1->Connect("Toggled(Bool_t)","TFitEditor",this,"DoEmptyBinsAllWeights1()");
fImproveResults->Connect("Toggled(Bool_t)","TFitEditor",this,"DoImproveResults()");
fLinearFit->Connect("Toggled(Bool_t)","TFitEditor",this,"DoLinearFit()");
fNoChi2->Connect("Toggled(Bool_t)","TFitEditor",this,"DoNoChi2()");
fRobustValue->Connect("ValueSet(Long_t)", "TFitEditor", this, "DoRobust()");
(fRobustValue->GetNumberEntry())->Connect("ReturnPressed()", "TFitEditor",
this, "DoRobust()");
fNoStoreDrawing->Connect("Toggled(Bool_t)","TFitEditor",this,"DoNoStoreDrawing()");
fNoDrawing->Connect("Toggled(Bool_t)","TFitEditor",this,"DoNoDrawing()");
fDrawSame->Connect("Toggled(Bool_t)","TFitEditor",this,"DoDrawSame()");
fMethodList->Connect("Selected(Int_t)", "TFitEditor", this, "DoMethod(Int_t)");
fFitButton->Connect("Clicked()", "TFitEditor", this, "DoFit()");
fResetButton->Connect("Clicked()", "TFitEditor", this, "DoReset()");
fCloseButton->Connect("Clicked()", "TFitEditor", this, "DoClose()");
fUserButton->Connect("Clicked()", "TFitEditor", this, "DoUserDialog()");
fDrawAdvanced->Connect("Clicked()", "TFitEditor", this, "DoAdvancedOptions()");
if (fDim > 0) {
fSliderX->Connect("PositionChanged()","TFitEditor",this, "DoSliderXMoved()");
fSliderX->Connect("Pressed()","TFitEditor",this, "DoSliderXPressed()");
fSliderX->Connect("Released()","TFitEditor",this, "DoSliderXReleased()");
}
if (fDim > 1) {
fSliderY->Connect("PositionChanged()","TFitEditor",this, "DoSliderYMoved()");
fSliderY->Connect("Pressed()","TFitEditor",this, "DoSliderYPressed()");
fSliderY->Connect("Released()","TFitEditor",this, "DoSliderYReleased()");
}
if (fDim > 2) {
fSliderZ->Connect("PositionChanged()","TFitEditor",this, "DoSliderZMoved()");
fSliderZ->Connect("Pressed()","TFitEditor",this, "DoSliderZPressed()");
fSliderZ->Connect("Released()","TFitEditor",this, "DoSliderZReleased()");
}
fParentPad->Connect("RangeAxisChanged()", "TFitEditor", this, "UpdateGUI()");
fLibMinuit->Connect("Toggled(Bool_t)","TFitEditor",this,"DoLibrary(Bool_t)");
fLibMinuit2->Connect("Toggled(Bool_t)","TFitEditor",this,"DoLibrary(Bool_t)");
fLibFumili->Connect("Toggled(Bool_t)","TFitEditor",this,"DoLibrary(Bool_t)");
fMigrad->Connect("Toggled(Bool_t)","TFitEditor",this,"DoMinMethod(Bool_t)");
fFumili->Connect("Toggled(Bool_t)","TFitEditor",this,"DoMinMethod(Bool_t)");
fErrorScale->Connect("ReturnPressed()", "TFitEditor", this, "DoErrorsDef()");
fTolerance->Connect("ReturnPressed()", "TFitEditor", this, "DoMaxTolerance()");
fIterations->Connect("ReturnPressed()", "TFitEditor", this, "DoMaxIterations()");
fOptDefault->Connect("Toggled(Bool_t)","TFitEditor",this,"DoPrintOpt(Bool_t)");
fOptVerbose->Connect("Toggled(Bool_t)","TFitEditor",this,"DoPrintOpt(Bool_t)");
fOptQuiet->Connect("Toggled(Bool_t)","TFitEditor",this,"DoPrintOpt(Bool_t)");
}
void TFitEditor::DisconnectSlots()
{
Disconnect("CloseWindow()");
fFuncList->Disconnect("Selected(Int_t)");
fEnteredFunc->Disconnect("ReturnPressed()");
fSetParam->Disconnect("Clicked()");
fNone->Disconnect("Toggled(Bool_t)");
fAdd->Disconnect("Toggled(Bool_t)");
fIntegral->Disconnect("Toggled(Bool_t)");
fBestErrors->Disconnect("Toggled(Bool_t)");
fUseRange->Disconnect("Toggled(Bool_t)");
fAdd2FuncList->Disconnect("Toggled(Bool_t)");
fAllWeights1->Disconnect("Toggled(Bool_t)");
fEmptyBinsWghts1->Disconnect("Toggled(Bool_t)");
fImproveResults->Disconnect("Toggled(Bool_t)");
fLinearFit->Disconnect("Toggled(Bool_t)");
fNoChi2->Disconnect("Toggled(Bool_t)");
fRobustValue->Disconnect("ValueSet(Long_t)");
(fRobustValue->GetNumberEntry())->Disconnect("ReturnPressed()");
fNoStoreDrawing->Disconnect("Toggled(Bool_t)");
fNoDrawing->Disconnect("Toggled(Bool_t)");
fDrawSame->Disconnect("Toggled(Bool_t)");
fMethodList->Disconnect("Selected(Int_t)");
fFitButton->Disconnect("Clicked()");
fResetButton->Disconnect("Clicked()");
fUserButton->Disconnect("Clicked()");
fDrawAdvanced->Disconnect("Clicked()");
if (fDim > 0) {
fSliderX->Disconnect("PositionChanged()");
fSliderX->Disconnect("Pressed()");
fSliderX->Disconnect("Released()");
}
if (fDim > 1) {
fSliderY->Disconnect("PositionChanged()");
fSliderY->Disconnect("Pressed()");
fSliderY->Disconnect("Released()");
}
if (fDim > 2) {
fSliderZ->Disconnect("PositionChanged()");
fSliderZ->Disconnect("Pressed()");
fSliderZ->Disconnect("Released()");
}
fLibMinuit->Disconnect("Toggled(Bool_t)");
fLibMinuit2->Disconnect("Toggled(Bool_t)");
fLibFumili->Disconnect("Toggled(Bool_t)");
fMigrad->Disconnect("Toggled(Bool_t)");
fFumili->Disconnect("Toggled(Bool_t)");
fErrorScale->Disconnect("ReturnPressed()");
fTolerance->Disconnect("ReturnPressed()");
fIterations->Disconnect("ReturnPressed()");
fOptDefault->Disconnect("Toggled(Bool_t)");
fOptVerbose->Disconnect("Toggled(Bool_t)");
fOptQuiet->Disconnect("Toggled(Bool_t)");
}
void TFitEditor::SetCanvas(TCanvas *newcan)
{
if (!newcan || (fCanvas == newcan)) return;
fCanvas = newcan;
fParentPad = fCanvas->GetSelectedPad();
if (!fParentPad) fParentPad = fCanvas;
newcan->Connect("Selected(TVirtualPad*,TObject*,Int_t)", "TFitEditor",
this, "SetFitObject(TVirtualPad *, TObject *, Int_t)");
ConnectToCanvas();
}
void TFitEditor::ConnectToCanvas()
{
TQObject::Connect("TCanvas", "Selected(TVirtualPad *, TObject *, Int_t)",
"TFitEditor",this,
"SetFitObject(TVirtualPad *, TObject *, Int_t)");
TQObject::Connect("TCanvas", "Closed()", "TFitEditor", this, "DoNoSelection()");
}
void TFitEditor::Hide()
{
if (fgFitDialog) {
fgFitDialog->UnmapWindow();
}
if (fParentPad) {
fParentPad->Disconnect("RangeAxisChanged()");
DoReset();
}
fCanvas = 0;
fParentPad = 0;
fFitObject = 0;
gROOT->GetListOfCleanups()->Remove(this);
}
void TFitEditor::Show(TVirtualPad* pad, TObject *obj)
{
if (!gROOT->GetListOfCleanups()->FindObject(this))
gROOT->GetListOfCleanups()->Add(this);
if (!fgFitDialog->IsMapped()) {
fgFitDialog->MapWindow();
gVirtualX->RaiseWindow(GetId());
}
SetCanvas(pad->GetCanvas());
fCanvas->Selected(pad, obj, kButton1Down);
}
void TFitEditor::CloseWindow()
{
Hide();
}
TFitEditor *&TFitEditor::GetFP()
{
return fgFitDialog;
}
void TFitEditor::Terminate()
{
TQObject::Disconnect("TCanvas", "Closed()");
delete fgFitDialog;
fgFitDialog = 0;
}
void TFitEditor::UpdateGUI()
{
if (!fFitObject) return;
if (fDim > 0) {
fSliderX->Disconnect("PositionChanged()");
fSliderX->Disconnect("Pressed()");
fSliderX->Disconnect("Released()");
switch (fType) {
case kObjectHisto: {
fXaxis = ((TH1*)fFitObject)->GetXaxis();
fYaxis = ((TH1*)fFitObject)->GetYaxis();
fZaxis = ((TH1*)fFitObject)->GetZaxis();
fXrange = fXaxis->GetNbins();
fXmin = fXaxis->GetFirst();
fXmax = fXaxis->GetLast();
break;
}
case kObjectGraph: {
TGraph *gr = (TGraph*)fFitObject;
TH1F *hist = gr->GetHistogram();
if (hist) {
fXaxis = hist->GetXaxis();
fYaxis = hist->GetYaxis();
fZaxis = hist->GetZaxis();
fXrange = fXaxis->GetNbins();
fXmin = fXaxis->GetFirst();
fXmax = fXaxis->GetLast();
}
break;
}
case kObjectGraph2D: {
break;
}
case kObjectHStack: {
TH1 *hist = (TH1 *)((THStack *)fFitObject)->GetHists()->First();
fXaxis = hist->GetXaxis();
fYaxis = hist->GetYaxis();
fZaxis = hist->GetZaxis();
fXrange = fXaxis->GetNbins();
fXmin = fXaxis->GetFirst();
fXmax = fXaxis->GetLast();
break;
}
case kObjectTree: {
break;
}
}
fSliderX->SetRange(1,fXrange);
if (!fXmin && !fXmax)
fSliderX->SetPosition(1,fXrange);
else
fSliderX->SetPosition(fXmin,fXmax);
fSliderX->SetScale(5);
fSliderX->Connect("PositionChanged()","TFitEditor",this, "DoSliderXMoved()");
fSliderX->Connect("Pressed()","TFitEditor",this, "DoSliderXPressed()");
fSliderX->Connect("Released()","TFitEditor",this, "DoSliderXReleased()");
}
}
void TFitEditor::SetFitObject(TVirtualPad *pad, TObject *obj, Int_t event)
{
if (event != kButton1Down) return;
if (!pad || !obj) {
DoNoSelection();
return;
}
if (!SetObjectType(obj)) return;
fParentPad = pad;
fFitObject = obj;
fDrawOption = GetDrawOption();
ShowObjectName(obj);
UpdateGUI();
ConnectSlots();
Bool_t hasf = HasFitFunction(obj);
if (hasf) {
fFunction = fFitFunc->GetTitle();
fEnteredFunc->SetText(fFunction.Data());
TGLBEntry *en = fFuncList->FindEntry(fFunction.Data());
if (en) fFuncList->Select(en->EntryId());
} else {
TGTextLBEntry *te = (TGTextLBEntry *)fFuncList->GetSelectedEntry();
if (fNone->GetState() == kButtonDown)
fFunction = te->GetTitle();
else if (fAdd->GetState() == kButtonDown) {
fFunction += '+';
fFunction +=te->GetTitle();
}
fEnteredFunc->SetText(fFunction.Data());
fEnteredFunc->SelectAll();
if (!fFitFunc) {
fFitFunc = new TF1("fitFunc",fFunction.Data(),fXmin,fXmax);
}
}
if (fSetParam->GetState() == kButtonDisabled)
fSetParam->SetEnabled(kTRUE);
if (fFitButton->GetState() == kButtonDisabled)
fFitButton->SetEnabled(kTRUE);
if (fResetButton->GetState() == kButtonDisabled)
fResetButton->SetEnabled(kTRUE);
DoLinearFit();
}
void TFitEditor::DoNoSelection()
{
if (gROOT->GetListOfCanvases()->IsEmpty()) {
Terminate();
return;
}
DisconnectSlots();
fParentPad = 0;
fFitObject = 0;
fObjLabel->SetText("No object selected");
fObjLabelParent->Resize(GetDefaultSize());
Layout();
fSetParam->SetEnabled(kFALSE);
fFitButton->SetEnabled(kFALSE);
fResetButton->SetEnabled(kFALSE);
}
void TFitEditor::RecursiveRemove(TObject* obj)
{
if (obj == fFitObject) {
fFitObject = 0;
DisconnectSlots();
fObjLabel->SetText("No object selected");
fObjLabelParent->Resize(GetDefaultSize());
Layout();
fFitButton->SetEnabled(kFALSE);
fResetButton->SetEnabled(kFALSE);
fSetParam->SetEnabled(kFALSE);
TQObject::Connect("TCanvas", "Selected(TVirtualPad *, TObject *, Int_t)",
"TFitEditor",this,
"SetFitObject(TVirtualPad *, TObject *, Int_t)");
TQObject::Connect("TCanvas", "Closed()", "TFitEditor", this,
"DoNoSelection()");
return;
}
if (obj == fParentPad) {
fFitObject = 0;
fParentPad = 0;
DisconnectSlots();
fObjLabel->SetText("No object selected");
fObjLabelParent->Resize(GetDefaultSize());
Layout();
fFitButton->SetEnabled(kFALSE);
fResetButton->SetEnabled(kFALSE);
fSetParam->SetEnabled(kFALSE);
}
}
TGComboBox* TFitEditor::BuildFunctionList(TGFrame* parent, Int_t id)
{
TGComboBox *c = new TGComboBox(parent, id);
c->AddEntry("gaus" , kFP_GAUS);
c->AddEntry("gausn", kFP_GAUSN);
c->AddEntry("expo", kFP_EXPO);
c->AddEntry("landau", kFP_LAND);
c->AddEntry("landaun",kFP_LANDN);
c->AddEntry("pol0", kFP_POL0);
c->AddEntry("pol1", kFP_POL1);
c->AddEntry("pol2", kFP_POL2);
c->AddEntry("pol3", kFP_POL3);
c->AddEntry("pol4", kFP_POL4);
c->AddEntry("pol5", kFP_POL5);
c->AddEntry("pol6", kFP_POL6);
c->AddEntry("pol7", kFP_POL7);
c->AddEntry("pol8", kFP_POL8);
c->AddEntry("pol9", kFP_POL9);
c->AddEntry("user", kFP_USER);
fLastEntryId = kFP_USER;
if (!gROOT->GetFunction("gaus")) {
Float_t xmin = 1.;
Float_t xmax = 2.;
new TF1("gaus","gaus",xmin,xmax);
new TF1("landau","landau",xmin,xmax);
new TF1("expo","expo",xmin,xmax);
for (Int_t i=0; i<10; i++) {
new TF1(Form("pol%d",i),Form("pol%d",i),xmin,xmax);
}
}
return c;
}
TGComboBox* TFitEditor::BuildMethodList(TGFrame* parent, Int_t id)
{
TGComboBox *c = new TGComboBox(parent, id);
c->AddEntry("Chi-square", kFP_MCHIS);
c->AddEntry("Binned Likelihood", kFP_MBINL);
c->Select(kFP_MCHIS);
return c;
}
void TFitEditor::DoAddtoList()
{
if (fAdd2FuncList->GetState() == kButtonDown)
fFitOption += '+';
else {
Int_t eq = fFitOption.First('+');
fFitOption.Remove(eq, 1);
}
}
void TFitEditor::DoAdvancedOptions()
{
new TGMsgBox(fClient->GetRoot(), GetMainFrame(),
"Info", "Advanced option dialog is not implemented yet",
kMBIconAsterisk,kMBOk, 0);
}
void TFitEditor::DoEmptyBinsAllWeights1()
{
if (fEmptyBinsWghts1->GetState() == kButtonDown) {
if (fAllWeights1->GetState() == kButtonDown) {
fAllWeights1->SetState(kButtonUp, kTRUE);
}
fFitOption += "WW";
} else {
Int_t eq = fFitOption.First("WW");
fFitOption.Remove(eq, 2);
}
}
void TFitEditor::DoAllWeights1()
{
if (fAllWeights1->GetState() == kButtonDown) {
if (fEmptyBinsWghts1->GetState() == kButtonDown) {
fEmptyBinsWghts1->SetState(kButtonUp, kTRUE);
}
fFitOption += 'W';
} else {
Int_t eq = fFitOption.First('W');
fFitOption.Remove(eq, 1);
}
}
void TFitEditor::DoClose()
{
Hide();
}
void TFitEditor::DoDrawSame()
{
fFitOption.ToUpper();
if (fDrawSame->GetState() == kButtonDown) {
if (fDrawOption.Contains("SAME"))
return;
else
fDrawOption += "SAME";
} else {
if (fDrawOption.Contains("SAME"))
fDrawOption.ReplaceAll("SAME", "");
}
}
void TFitEditor::DoFit()
{
if (!fFitObject) return;
if (!fParentPad) return;
if (!fFitFunc) {
new TGMsgBox(fClient->GetRoot(), GetMainFrame(),
"Error", Form("Function with name '%s' does not exist",
fFunction.Data()), kMBIconExclamation, kMBClose, 0);
return;
}
fFitButton->SetState(kButtonEngaged);
if (gPad) gPad->GetVirtCanvas()->SetCursor(kWatch);
gVirtualX->SetCursor(GetId(), gVirtualX->CreateCursor(kWatch));
fParentPad->Disconnect("RangeAxisChanged()");
TVirtualPad *save = 0;
save = gPad;
gPad = fParentPad;
fParentPad->cd();
fParentPad->GetCanvas()->SetCursor(kWatch);
Double_t xmin = 0;
Double_t xmax = 0;
switch (fType) {
case kObjectHisto: {
TH1 *h1 = (TH1*)fFitObject;
xmin = fXaxis->GetBinLowEdge((Int_t)(fSliderX->GetMinPosition()));
xmax = fXaxis->GetBinUpEdge((Int_t)(fSliderX->GetMaxPosition()));
fFitFunc->SetRange(xmin,xmax);
fDrawOption = GetDrawOption();
h1->Fit(fFitFunc, fFitOption.Data(), fDrawOption.Data(), xmin, xmax);
break;
}
case kObjectGraph: {
TGraph *gr = (TGraph*)fFitObject;
TH1F *hist = gr->GetHistogram();
if (hist) {
xmin = fXaxis->GetBinLowEdge((Int_t)(fSliderX->GetMinPosition()));
xmax = fXaxis->GetBinUpEdge((Int_t)(fSliderX->GetMaxPosition()));
Int_t npoints = gr->GetN();
Double_t *gx = gr->GetX();
Double_t gxmin, gxmax;
const Int_t imin = TMath::LocMin(npoints,gx);
const Int_t imax = TMath::LocMax(npoints,gx);
gxmin = gx[imin];
gxmax = gx[imax];
if (xmin < gxmin) xmin = gxmin;
if (xmax > gxmax) xmax = gxmax;
}
fFitFunc->SetRange(xmin,xmax);
fDrawOption = GetDrawOption();
gr->Fit(fFitFunc, fFitOption.Data(), fDrawOption.Data(), xmin, xmax);
break;
}
case kObjectGraph2D: {
break;
}
case kObjectHStack: {
break;
}
case kObjectTree: {
break;
}
}
fParentPad->Modified();
fParentPad->Update();
fParentPad->GetCanvas()->SetCursor(kPointer);
fParentPad->Connect("RangeAxisChanged()", "TFitEditor", this, "UpdateGUI()");
if (save) gPad = save;
if (fSetParam->GetState() == kButtonDisabled &&
fLinearFit->GetState() == kButtonUp)
fSetParam->SetState(kButtonUp);
if (gPad) gPad->GetVirtCanvas()->SetCursor(kPointer);
gVirtualX->SetCursor(GetId(), gVirtualX->CreateCursor(kPointer));
fFitButton->SetState(kButtonUp);
}
Int_t TFitEditor::CheckFunctionString(const char *fname)
{
fFitFunc = (TF1 *)gROOT->GetListOfFunctions()->FindObject(fname);
if (fFitFunc)
return fFitFunc->Compile();
TF1 *form = 0;
form = new TF1("fitFunc", fname, fXmin, fXmax);
if (form) {
return form->Compile();
}
delete form;
return -1;
}
void TFitEditor::DoAddition(Bool_t on)
{
Int_t sel = fFuncList->GetSelected();
static Bool_t first = kFALSE;
if (sel > kFP_USER) {
fEnteredFunc->Clear();
fFunction = "";
fFuncList->Select(kFP_GAUS, kTRUE);
return;
}
TString s = fEnteredFunc->GetText();
if (on) {
if (!first) {
s += "(0)";
fEnteredFunc->SetText(s.Data());
first = kTRUE;
fSelLabel->SetText(fFunction.Data());
((TGCompositeFrame *)fSelLabel->GetParent())->Layout();
}
} else {
first = kFALSE;
}
}
void TFitEditor::DoNoOperation(Bool_t on)
{
TGTextLBEntry *te = (TGTextLBEntry *)fFuncList->GetSelectedEntry();
if (on) {
fEnteredFunc->SetText(te->GetTitle());
}
fFunction = fEnteredFunc->GetText();
fSelLabel->SetText(fFunction.Data());
((TGCompositeFrame *)fSelLabel->GetParent())->Layout();
fFitFunc = (TF1 *) gROOT->GetListOfFunctions()->FindObject(fFunction.Data());
if (!fFitFunc)
fFitFunc = new TF1("fitFunc",fFunction.Data(),fXmin,fXmax);
}
void TFitEditor::DoFunction(Int_t)
{
TGTextLBEntry *te = (TGTextLBEntry *)fFuncList->GetSelectedEntry();
if (fNone->GetState() == kButtonDown) {
fEnteredFunc->SetText(te->GetTitle());
} else if (fAdd->GetState() == kButtonDown) {
Int_t np = 0;
TString s = "";
if (!strcmp(fEnteredFunc->GetText(), "")) {
fFunction = te->GetTitle();
} else {
s = fEnteredFunc->GetTitle();
TFormula tmp("tmp", fFunction.Data());
np = tmp.GetNpar();
}
if (np)
s += Form("+%s(%d)", te->GetTitle(), np);
else
s += Form("%s(%d)", te->GetTitle(), np);
fEnteredFunc->SetText(s.Data());
}
fFunction = fEnteredFunc->GetText();
if (fFunction.Contains("pol") || fFunction.Contains("++")) {
fLinearFit->SetState(kButtonDown, kTRUE);
} else {
fLinearFit->SetState(kButtonUp, kTRUE);
}
fEnteredFunc->SelectAll();
fSelLabel->SetText(fFunction.Data());
((TGCompositeFrame *)fSelLabel->GetParent())->Layout();
fFitFunc = (TF1 *) gROOT->GetListOfFunctions()->FindObject(fFunction.Data());
if (!fFitFunc)
fFitFunc = new TF1("fitFunc",fFunction.Data(),fXmin,fXmax);
}
void TFitEditor::DoEnteredFunction()
{
if (!strcmp(fEnteredFunc->GetText(), "")) return;
Int_t ok = CheckFunctionString(fEnteredFunc->GetText());
if (ok != 0) {
new TGMsgBox(fClient->GetRoot(), GetMainFrame(),
"Error...", "Verify the entered function string!",
kMBIconStop,kMBOk, 0);
}
fFunction = fEnteredFunc->GetText();
fFitFunc = (TF1 *) gROOT->GetListOfFunctions()->FindObject(fFunction.Data());
fSelLabel->SetText(fFunction.Data());
((TGCompositeFrame *)fSelLabel->GetParent())->Layout();
if (fFunction.Contains("++")) {
fLinearFit->SetState(kButtonDown, kTRUE);
fAdd->SetState(kButtonDown, kTRUE);
} else if (fFunction.Contains('+')) {
fAdd->SetState(kButtonDown, kTRUE);
} else {
fNone->SetState(kButtonDown, kTRUE);
}
}
void TFitEditor::DoImproveResults()
{
if (fImproveResults->GetState() == kButtonDown)
fFitOption += 'M';
else if (fFitOption.Contains('M'))
fFitOption.ReplaceAll('M', "");
}
void TFitEditor::DoBestErrors()
{
if (fBestErrors->GetState() == kButtonDown)
fFitOption += 'E';
else if (fFitOption.Contains('E'))
fFitOption.ReplaceAll('E', "");
}
void TFitEditor::DoIntegral()
{
if (fIntegral->GetState() == kButtonDown)
fFitOption += 'I';
else if (fFitOption.Contains('I'))
fFitOption.ReplaceAll('I', "");
}
void TFitEditor::DoLinearFit()
{
if (fLinearFit->GetState() == kButtonDown) {
fPlus = "++";
if (fFitOption.Contains('F'))
fFitOption.ReplaceAll('F', "");
fSetParam->SetState(kButtonDisabled);
fBestErrors->SetState(kButtonDisabled);
fImproveResults->SetState(kButtonDisabled);
fRobustValue->SetState(kTRUE);
} else {
fPlus = '+';
if (fFunction.Contains("pol") || fFunction.Contains("++"))
fFitOption += 'F';
fSetParam->SetState(kButtonUp);
fBestErrors->SetState(kButtonUp);
fImproveResults->SetState(kButtonUp);
fRobustValue->SetState(kFALSE);
}
}
void TFitEditor::DoMethod(Int_t id)
{
if (id == kFP_MCHIS) {
if (fFitOption.Contains('L'))
fFitOption.ReplaceAll('L', "");
} else {
fFitOption += 'L';
}
}
void TFitEditor::DoNoChi2()
{
if (fNoChi2->GetState() == kButtonDown)
fFitOption += 'C';
else if (fFitOption.Contains('C'))
fFitOption.ReplaceAll('C', "");
if (fLinearFit->GetState() == kButtonUp)
fLinearFit->SetState(kButtonDown, kTRUE);
}
void TFitEditor::DoNoDrawing()
{
if (fNoDrawing->GetState() == kButtonDown)
fFitOption += '0';
else if (fFitOption.Contains('0'))
fFitOption.ReplaceAll('0', "");
}
void TFitEditor::DoNoStoreDrawing()
{
if (fNoStoreDrawing->GetState() == kButtonDown)
fFitOption += 'N';
else if (fFitOption.Contains('N'))
fFitOption.ReplaceAll('N', "");
if (fNoDrawing->GetState() == kButtonUp)
fNoDrawing->SetState(kButtonDown);
}
void TFitEditor::DoPrintOpt(Bool_t on)
{
TGButton *btn = (TGButton *) gTQSender;
Int_t id = btn->WidgetId();
switch (id) {
case kFP_PDEF:
if (on) {
fOptDefault->SetState(kButtonDown);
fOptVerbose->SetState(kButtonUp);
fOptQuiet->SetState(kButtonUp);
if (fFitOption.Contains('Q')) {
fFitOption.ReplaceAll('Q', "");
}
if (fFitOption.Contains('V')) {
fFitOption.ReplaceAll('V', "");
}
}
fStatusBar->SetText("Prn: DEF",3);
break;
case kFP_PVER:
if (on) {
fOptVerbose->SetState(kButtonDown);
fOptDefault->SetState(kButtonUp);
fOptQuiet->SetState(kButtonUp);
if (fFitOption.Contains('Q')) {
fFitOption.ReplaceAll('Q', "");
}
fFitOption += 'V';
}
fStatusBar->SetText("Prn: VER",3);
break;
case kFP_PQET:
if (on) {
fOptQuiet->SetState(kButtonDown);
fOptDefault->SetState(kButtonUp);
fOptVerbose->SetState(kButtonUp);
if (fFitOption.Contains('V')) {
fFitOption.ReplaceAll('V', "");
}
fFitOption += 'Q';
}
fStatusBar->SetText("Prn: QT",3);
default:
break;
}
}
void TFitEditor::DoReset()
{
fParentPad->Modified();
fParentPad->Update();
fFitOption = "";
fDrawOption = GetDrawOption();
fFunction = "gaus";
if (fFitFunc) {
fFitFunc->SetParent(0);
fFitFunc = (TF1 *) gROOT->GetListOfFunctions()->FindObject(fFunction.Data());
if (!fFitFunc)
fFitFunc = new TF1("fitFunc",fFunction.Data(),fXmin,fXmax);
}
if (fXmin > 1 || fXmax < fXrange) {
fSliderX->SetRange(fXmin,fXmax);
fSliderX->SetPosition(fXmin, fXmax);
} else {
fSliderX->SetRange(1,fXrange);
fSliderX->SetPosition(fXmin,fXmax);
}
fPlus = '+';
if (fLinearFit->GetState() == kButtonDown)
fLinearFit->SetState(kButtonUp, kTRUE);
if (fBestErrors->GetState() == kButtonDown)
fBestErrors->SetState(kButtonUp, kFALSE);
if (fUseRange->GetState() == kButtonDown)
fUseRange->SetState(kButtonUp, kFALSE);
if (fAllWeights1->GetState() == kButtonDown)
fAllWeights1->SetState(kButtonUp, kFALSE);
if (fEmptyBinsWghts1->GetState() == kButtonDown)
fEmptyBinsWghts1->SetState(kButtonUp, kFALSE);
if (fImproveResults->GetState() == kButtonDown)
fImproveResults->SetState(kButtonUp, kFALSE);
if (fAdd2FuncList->GetState() == kButtonDown)
fAdd2FuncList->SetState(kButtonUp, kFALSE);
if (fNoChi2->GetState() == kButtonDown)
fNoChi2->SetState(kButtonUp, kFALSE);
if (fDrawSame->GetState() == kButtonDown)
fDrawSame->SetState(kButtonUp, kFALSE);
if (fNoDrawing->GetState() == kButtonDown)
fNoDrawing->SetState(kButtonUp, kFALSE);
if (fNoStoreDrawing->GetState() == kButtonDown)
fNoStoreDrawing->SetState(kButtonUp, kFALSE);
fNone->SetState(kButtonDown, kTRUE);
fFuncList->Select(1, kTRUE);
fEnteredFunc->SetText("gaus");
if (fLibMinuit->GetState() != kButtonDown)
fLibMinuit->SetState(kButtonDown, kTRUE);
if (fMigrad->GetState() != kButtonDown)
fMigrad->SetState(kButtonDown, kTRUE);
if (fOptDefault->GetState() != kButtonDown)
fOptDefault->SetState(kButtonDown, kTRUE);
if (fErrorScale->GetNumber() != 1.0) {
fErrorScale->SetNumber(1.0);
fErrorScale->ReturnPressed();
}
if (fTolerance->GetNumber() != 1e-6) {
fTolerance->SetNumber(1e-6);
fTolerance->ReturnPressed();
}
if (fIterations->GetNumber() != 5000) {
fIterations->SetIntNumber(5000);
fIterations->ReturnPressed();
}
}
void TFitEditor::DoRobust()
{
if (fType != kObjectGraph) return;
fRobustValue->SetState(kTRUE);
if (fFitOption.Contains("ROB")) {
Int_t pos = fFitOption.Index("=");
fFitOption.Replace(pos+1, 4, Form("%g",fRobustValue->GetNumber()));
} else {
fFitOption += Form("ROB=%g", fRobustValue->GetNumber());
}
}
void TFitEditor::DoBound(Bool_t on)
{
TString s = fFitOption;
if (s.Contains("ROB")) {
s.ReplaceAll("ROB", "H");
}
if (on) {
if (s.Contains('B'))
return;
else
fFitOption += 'B';
} else {
if (s.Contains('B')) {
Int_t pos = fFitOption.First('B');
Int_t rob = fFitOption.Index("ROB");
if (pos != rob+2)
fFitOption.Remove(pos, 1);
}
}
}
void TFitEditor::DoSetParameters()
{
if (!fFitFunc) {
printf("SetParamters - create fit function %s\n",fFunction.Data());
fFitFunc = new TF1("fitFunc",Form("%s",fFunction.Data()), fXmin, fXmax);
}
fParentPad->Disconnect("RangeAxisChanged()");
Double_t xmin, xmax;
fFitFunc->GetRange(xmin, xmax);
Int_t ret = 0;
new TFitParametersDialog(gClient->GetDefaultRoot(), GetMainFrame(),
fFitFunc, fParentPad, xmin, xmax, &ret);
TGTextLBEntry *te = (TGTextLBEntry *)fFuncList->GetSelectedEntry();
if ((fNone->GetState() == kButtonDown) &&
strcmp(te->GetTitle(), "user")) {
if (ret == kFPDBounded) {
DoBound(kTRUE);
} else {
DoBound(kFALSE);
}
}
fParentPad->Connect("RangeAxisChanged()", "TFitEditor", this, "UpdateGUI()");
}
void TFitEditor::DoSliderXPressed()
{
if (!fParentPad) return;
TVirtualPad *save = 0;
save = gPad;
gPad = fParentPad;
fParentPad->cd();
fParentPad->GetCanvas()->FeedbackMode(kFALSE);
fParentPad->SetLineWidth(1);
fParentPad->SetLineColor(2);
Float_t xleft = 0;
Double_t xright = 0;
switch (fType) {
case kObjectHisto: {
xleft = fXaxis->GetBinLowEdge((Int_t)((fSliderX->GetMinPosition())+0.5));
xright = fXaxis->GetBinUpEdge((Int_t)((fSliderX->GetMaxPosition())+0.5));
break;
}
case kObjectGraph: {
xleft = fXaxis->GetBinLowEdge((Int_t)((fSliderX->GetMinPosition())+0.5));
xright = fXaxis->GetBinUpEdge((Int_t)((fSliderX->GetMaxPosition())+0.5));
break;
}
case kObjectGraph2D: {
break;
}
case kObjectHStack: {
break;
}
case kObjectTree: {
break;
}
}
Float_t ymin = fParentPad->GetUymin();
Float_t ymax = fParentPad->GetUymax();
fPx1old = fParentPad->XtoAbsPixel(xleft);
fPy1old = fParentPad->YtoAbsPixel(ymin);
fPx2old = fParentPad->XtoAbsPixel(xright);
fPy2old = fParentPad->YtoAbsPixel(ymax);
gVirtualX->DrawBox(fPx1old, fPy1old, fPx2old, fPy2old, TVirtualX::kHollow);
if(save) gPad = save;
}
void TFitEditor::DoSliderXMoved()
{
Int_t px1,py1,px2,py2;
Float_t xleft = 0;
Double_t xright = 0;
TVirtualPad *save = 0;
save = gPad;
gPad = fParentPad;
gPad->cd();
switch (fType) {
case kObjectHisto: {
xleft = fXaxis->GetBinLowEdge((Int_t)((fSliderX->GetMinPosition())+0.5));
xright = fXaxis->GetBinUpEdge((Int_t)((fSliderX->GetMaxPosition())+0.5));
break;
}
case kObjectGraph: {
xleft = fXaxis->GetBinLowEdge((Int_t)((fSliderX->GetMinPosition())+0.5));
xright = fXaxis->GetBinUpEdge((Int_t)((fSliderX->GetMaxPosition())+0.5));
break;
}
case kObjectGraph2D: {
break;
}
case kObjectHStack: {
break;
}
case kObjectTree: {
break;
}
}
Float_t ymin = gPad->GetUymin();
Float_t ymax = gPad->GetUymax();
px1 = gPad->XtoAbsPixel(xleft);
py1 = gPad->YtoAbsPixel(ymin);
px2 = gPad->XtoAbsPixel(xright);
py2 = gPad->YtoAbsPixel(ymax);
gPad->GetCanvas()->FeedbackMode(kTRUE);
gPad->SetLineWidth(1);
gPad->SetLineColor(2);
gVirtualX->DrawBox(fPx1old, fPy1old, fPx2old, fPy2old, TVirtualX::kHollow);
gVirtualX->DrawBox(px1, py1, px2, py2, TVirtualX::kHollow);
fPx1old = px1;
fPy1old = py1;
fPx2old = px2 ;
fPy2old = py2;
if(save) gPad = save;
}
void TFitEditor::DoSliderXReleased()
{
gVirtualX->Update(0);
}
void TFitEditor::DoSliderYPressed()
{
fParentPad->cd();
switch (fType) {
case kObjectHisto: {
if (!fParentPad) return;
fParentPad->cd();
fParentPad->GetCanvas()->FeedbackMode(kFALSE);
fParentPad->SetLineWidth(1);
fParentPad->SetLineColor(2);
Float_t ybottom = fYaxis->GetBinLowEdge((Int_t)((fSliderY->GetMinPosition())+0.5));
Float_t ytop = fYaxis->GetBinUpEdge((Int_t)((fSliderY->GetMaxPosition())+0.5));
Float_t xmin = fParentPad->GetUxmin();
Float_t xmax = fParentPad->GetUxmax();
fPx1old = fParentPad->XtoAbsPixel(xmin);
fPy1old = fParentPad->YtoAbsPixel(ybottom);
fPx2old = fParentPad->XtoAbsPixel(xmax);
fPy2old = fParentPad->YtoAbsPixel(ytop);
gVirtualX->DrawBox(fPx1old, fPy1old, fPx2old, fPy2old, TVirtualX::kHollow);
}
case kObjectGraph: {
break;
}
case kObjectGraph2D: {
break;
}
case kObjectHStack: {
break;
}
case kObjectTree: {
break;
}
}
}
void TFitEditor::DoSliderYMoved()
{
switch (fType) {
case kObjectHisto: {
Int_t px1,py1,px2,py2;
Float_t ybottom = fYaxis->GetBinLowEdge((Int_t)((fSliderY->GetMinPosition())+0.5));
Float_t ytop = fYaxis->GetBinUpEdge((Int_t)((fSliderY->GetMaxPosition())+0.5));
Float_t xmin = fParentPad->GetUxmin();
Float_t xmax = fParentPad->GetUxmax();
px1 = fParentPad->XtoAbsPixel(xmin);
py1 = fParentPad->YtoAbsPixel(ybottom);
px2 = fParentPad->XtoAbsPixel(xmax);
py2 = fParentPad->YtoAbsPixel(ytop);
fParentPad->GetCanvas()->FeedbackMode(kTRUE);
fParentPad->cd();
fParentPad->SetLineWidth(1);
fParentPad->SetLineColor(2);
gVirtualX->DrawBox(fPx1old, fPy1old, fPx2old, fPy2old, TVirtualX::kHollow);
gVirtualX->DrawBox(px1, py1, px2, py2, TVirtualX::kHollow);
fPx1old = px1;
fPy1old = py1;
fPx2old = px2 ;
fPy2old = py2;
gVirtualX->Update(0);
}
case kObjectGraph: {
break;
}
case kObjectGraph2D: {
break;
}
case kObjectHStack: {
break;
}
case kObjectTree: {
break;
}
}
}
void TFitEditor::DoSliderYReleased()
{
fParentPad->Modified();
fParentPad->Update();
}
void TFitEditor::DoSliderZPressed()
{
}
void TFitEditor::DoSliderZMoved()
{
}
void TFitEditor::DoSliderZReleased()
{
}
void TFitEditor::DoUserDialog()
{
new TGMsgBox(fClient->GetRoot(), GetMainFrame(),
"Info", "Dialog of user method is not implemented yet",
kMBIconAsterisk,kMBOk, 0);
}
void TFitEditor::DoUseRange()
{
if (fUseRange->GetState() == kButtonDown)
fFitOption.Insert(0,'R');
else {
Int_t pos = fFitOption.First('R');
Int_t rob = fFitOption.Index("ROB");
if (pos != rob)
fFitOption.Remove(pos, 1);
}
}
void TFitEditor::SetFunction(const char *function)
{
fFunction = function;
}
Bool_t TFitEditor::SetObjectType(TObject* obj)
{
Bool_t set = kFALSE;
if (obj->InheritsFrom("TGraph")) {
fType = kObjectGraph;
set = kTRUE;
fDim = 1;
if (fMethodList->FindEntry("Binned Likelihood"))
fMethodList->RemoveEntry(kFP_MBINL);
if (!fMethodList->FindEntry("Chi-square"))
fMethodList->AddEntry("Chi-square", kFP_MCHIS);
fMethodList->Select(kFP_MCHIS, kFALSE);
fRobustValue->SetState(kTRUE);
fRobustValue->GetNumberEntry()->SetToolTipText("Set robust value");
} else if (obj->InheritsFrom("TGraph2D")) {
fType = kObjectGraph2D;
set = kTRUE;
fDim = 2;
if (fMethodList->FindEntry("Unbinned Likelihood"))
fMethodList->RemoveEntry(kFP_MUBIN);
if (!fMethodList->FindEntry("Chi-square"))
fMethodList->AddEntry("Chi-square", kFP_MCHIS);
fMethodList->Select(kFP_MCHIS, kFALSE);
} else if (obj->InheritsFrom("THStack")) {
fType = kObjectHStack;
set = kTRUE;
TH1 *hist = (TH1 *)((THStack *)obj)->GetHists()->First();
fDim = hist->GetDimension();
if (fMethodList->FindEntry("Unbinned Likelihood"))
fMethodList->RemoveEntry(kFP_MUBIN);
if (!fMethodList->FindEntry("Chi-square"))
fMethodList->AddEntry("Chi-square", kFP_MCHIS);
fMethodList->Select(kFP_MCHIS, kFALSE);
} else if (obj->InheritsFrom("TTree")) {
fType = kObjectTree;
set = kTRUE;
fDim = -1;
fMethodList->SetEnabled(kFALSE);
} else if (obj->InheritsFrom("TH1")){
fType = kObjectHisto;
set = kTRUE;
fDim = ((TH1*)obj)->GetDimension();
if (!fMethodList->FindEntry("Binned Likelihood"))
fMethodList->AddEntry("Binned Likelihood", kFP_MBINL);
if (!fMethodList->FindEntry("Chi-square"))
fMethodList->AddEntry("Chi-square", kFP_MCHIS);
fMethodList->Select(kFP_MCHIS, kFALSE);
}
return set;
}
void TFitEditor::ShowObjectName(TObject* obj)
{
TString name;
if (obj) {
name = obj->GetName();
name.Append("::");
name.Append(obj->ClassName());
} else {
name = "No object selected";
}
fObjLabel->SetText(name.Data());
fObjLabelParent->Resize(GetDefaultSize());
Layout();
}
Option_t *TFitEditor::GetDrawOption() const
{
if (!fParentPad) return "";
TListIter next(fParentPad->GetListOfPrimitives());
TObject *obj;
while ((obj = next())) {
if (obj == fFitObject) return next.GetOption();
}
return "";
}
void TFitEditor::DoLibrary(Bool_t on)
{
TGButton *bt = (TGButton *)gTQSender;
Int_t id = bt->WidgetId();
switch (id) {
case kFP_LMIN:
{
if (on) {
fLibMinuit->SetState(kButtonDown);
fLibMinuit2->SetState(kButtonUp);
fLibFumili->SetState(kButtonUp);
if (fFumili->GetState() != kButtonDisabled) {
fFumili->SetDisabledAndSelected(kFALSE);
}
fMigrad->SetState(kButtonDown);
fStatusBar->SetText("MIGRAD", 1);
TVirtualFitter::SetDefaultFitter("Minuit");
fStatusBar->SetText("LIB Minuit", 0);
}
}
break;
case kFP_LMIN2:
{
if (on) {
fLibMinuit->SetState(kButtonUp);
fLibMinuit2->SetState(kButtonDown);
fLibFumili->SetState(kButtonUp);
if (fMigrad->GetState() == kButtonDisabled)
fMigrad->SetState(kButtonUp);
if (fFumili->GetState() == kButtonDisabled)
fFumili->SetState(kButtonUp);
if (fMigrad->GetState() == kButtonDown)
TVirtualFitter::SetDefaultFitter("Minuit2");
else if (fFumili->GetState() == kButtonDown)
TVirtualFitter::SetDefaultFitter("Fumili2");
fStatusBar->SetText("LIB Minuit2", 0);
}
}
break;
case kFP_LFUM:
{
if (on) {
if (fFumili->GetState() != kButtonDown) {
fFumili->SetState(kButtonDown);
fStatusBar->SetText("FUMILI", 1);
}
fLibMinuit->SetState(kButtonUp);
fLibMinuit2->SetState(kButtonUp);
fLibFumili->SetState(kButtonDown);
TVirtualFitter::SetDefaultFitter("Fumili");
fMigrad->SetDisabledAndSelected(kFALSE);
fStatusBar->SetText("LIB Fumili", 0);
}
}
default:
break;
}
}
void TFitEditor::DoMinMethod(Bool_t on)
{
TGButton *bt = (TGButton *)gTQSender;
Int_t id = bt->WidgetId();
switch (id) {
case kFP_MIGRAD:
{
if (on) {
if (fLibMinuit->GetState() == kButtonDown)
fFumili->SetState(kButtonDisabled);
else
fFumili->SetState(kButtonUp);
fMigrad->SetState(kButtonDown);
fStatusBar->SetText("MIGRAD",1);
if (fLibMinuit2->GetState() == kButtonDown)
if (strncmp(TVirtualFitter::GetDefaultFitter(),"Minuit2",7) != 0)
TVirtualFitter::SetDefaultFitter("Minuit2");
}
}
break;
case kFP_SIMPLX:
{
if (on) {
}
}
break;
case kFP_FUMILI:
{
if (on) {
fMigrad->SetState(kButtonUp);
fFumili->SetState(kButtonDown);
fStatusBar->SetText("FUMILI",1);
if (fLibMinuit2->GetState() == kButtonDown)
TVirtualFitter::SetDefaultFitter("Fumili2");
else
TVirtualFitter::SetDefaultFitter("Fumili");
}
}
break;
}
}
void TFitEditor::DoErrorsDef()
{
Double_t err = fErrorScale->GetNumber();
TVirtualFitter::SetErrorDef(err);
}
void TFitEditor::DoMaxTolerance()
{
Double_t tol = fTolerance->GetNumber();
TVirtualFitter::SetPrecision(tol);
}
void TFitEditor::DoMaxIterations()
{
Long_t itr = fIterations->GetIntNumber();
TVirtualFitter::SetMaxIterations(itr);
fStatusBar->SetText(Form("Itr: %ld",itr),2);
}
void TFitEditor::MakeTitle(TGCompositeFrame *parent, const char *title)
{
TGCompositeFrame *ht = new TGCompositeFrame(parent, 350, 10,
kFixedWidth | kHorizontalFrame);
ht->AddFrame(new TGLabel(ht, title),
new TGLayoutHints(kLHintsLeft, 1, 1, 0, 0));
ht->AddFrame(new TGHorizontal3DLine(ht),
new TGLayoutHints(kLHintsExpandX | kLHintsCenterY, 5, 5, 2, 2));
parent->AddFrame(ht, new TGLayoutHints(kLHintsTop, 5, 0, 5, 0));
}
Bool_t TFitEditor::HasFitFunction(TObject *obj)
{
TF1 *func = 0;
TList *lf = 0;
switch (fType) {
case kObjectHisto: {
func =((TH1 *)obj)->GetFunction("fitFunc");
lf = ((TH1 *)obj)->GetListOfFunctions();
break;
}
case kObjectGraph: {
func =((TGraph *)obj)->GetFunction("fitFunc");
lf = ((TGraph *)obj)->GetListOfFunctions();
break;
}
case kObjectGraph2D: {
func =(TF1 *)((TGraph2D *)obj)->GetListOfFunctions()->FindObject("fitFunc");
lf = ((TGraph2D *)obj)->GetListOfFunctions();
break;
}
case kObjectHStack: {
break;
}
case kObjectTree: {
break;
}
}
if (func) {
if (lf) GetFunctionsFromList(lf);
fFitFunc = new TF1();
func->Copy(*fFitFunc);
CheckRange(func);
TGLBEntry *le = fFuncList->FindEntry(Form("%s",func->GetName()));
if (le) {
fFuncList->Select(le->EntryId(), kFALSE);
fSelLabel->SetText(le->GetTitle());
}
return kTRUE;
}
if (lf) {
GetFunctionsFromList(lf);
TObject *obj2;
TIter next(lf, kIterBackward);
while ((obj2 = next())) {
if (obj2->InheritsFrom(TF1::Class())) {
func = (TF1 *)obj2;
fFitFunc = new TF1();
func->Copy(*fFitFunc);
CheckRange(func);
TGLBEntry *le = fFuncList->FindEntry(Form("%s",func->GetName()));
if (le) {
fFuncList->Select(le->EntryId(), kTRUE);
}
return kTRUE;
}
}
}
return kFALSE;
}
void TFitEditor::GetFunctionsFromList(TList *list)
{
TF1 *f1 = 0;
TObject *obj;
Int_t newid = kFP_USER*30;
if (fLastEntryId != kFP_USER) {
fFuncList->RemoveEntries(newid, fLastEntryId);
TIter next(fFitFuncList);
while ((obj = next())) {
((TF1 *)obj)->SetParent(0);
fFitFuncList->Remove(obj);
delete obj;
}
}
TIter next(list, kIterBackward);
while ((obj = next())) {
if (obj->InheritsFrom(TF1::Class())) {
if (!strcmp(obj->GetName(), "fitFunc")) continue;
fFuncList->AddEntry(obj->GetName(), newid);
newid++;
f1 = new TF1();
TF1 *f2 = (TF1 *)obj;
f2->Copy(*f1);
f1->SetParent(fFitObject);
fFitFuncList->Add(f1);
}
}
fLastEntryId = newid - 1;
}
void TFitEditor::CheckRange(TF1 *f1)
{
Double_t fxmin=0, fxmax=0, xmin=0, xmax=0;
f1->GetRange(fxmin, fxmax);
xmin = fXaxis->GetXmin();
xmax = fXaxis->GetXmax();
fXrange = fXaxis->GetNbins();
if (fFitObject->InheritsFrom(TGraph::Class())) {
TGraph *gr = (TGraph *)fFitObject;
Int_t npoints = gr->GetN();
Double_t *gx = gr->GetX();
Double_t gxmin, gxmax;
const Int_t imin = TMath::LocMin(npoints,gx);
const Int_t imax = TMath::LocMax(npoints,gx);
gxmin = gx[imin];
gxmax = gx[imax];
if (xmin < gxmin) xmin = gxmin;
if (xmax > gxmax) xmax = gxmax;
if (xmin > xmax) {
Double_t tmp = gxmin;
xmin = xmax;
xmax = tmp;
}
}
if ((fxmin > xmin) || (fxmax < xmax)) {
fXmin = fXaxis->FindBin(fxmin);
fXmax = fXaxis->FindBin(fxmax);
fUseRange->SetState(kButtonDown);
} else {
fXmin = fXaxis->FindBin(xmin);
fXmax = fXaxis->FindBin(xmax);
fUseRange->SetState(kButtonUp);
}
if (fDim > 0) {
fSliderX->Disconnect("PositionChanged()");
fSliderX->Disconnect("Pressed()");
fSliderX->Disconnect("Released()");
fSliderX->SetRange(1,fXrange);
if (!fXmin && !fXmax)
fSliderX->SetPosition(1,fXrange);
else
fSliderX->SetPosition(fXmin,fXmax);
fSliderX->SetScale(5);
fSliderX->Connect("PositionChanged()","TFitEditor",this, "DoSliderXMoved()");
fSliderX->Connect("Pressed()","TFitEditor",this, "DoSliderXPressed()");
fSliderX->Connect("Released()","TFitEditor",this, "DoSliderXReleased()");
}
}
Last change: Wed Jun 25 08:39:07 2008
Last generated: 2008-06-25 08:39
This page has been automatically generated. If you have any comments or suggestions about the page layout send a mail to ROOT support, or contact the developers with any questions or problems regarding ROOT.