KVIDGridEditor.h

 
#ifndef __KVIDGridEditor_H
#define __KVIDGridEditor_H
 
#include "TCanvas.h"
#include "TPaveLabel.h"
#include "TPaveText.h"
#include "TStyle.h"
#include "TH2F.h"
#include "TF1.h"
#include "TF2.h"
#include "TROOT.h"
#include "TView.h"
#include "TKey.h"
#include "TSpectrum.h"
 
#include "TMethod.h"
#include "TContextMenu.h"
#include "TGMsgBox.h"
#include "TGFileDialog.h"
 
#include "KVIDGridEditorCanvas.h"
#include "KVBase.h"
#include "KVHashList.h"
#include "KVUnownedList.h"
#include "KVIDGraph.h"
#include "KVIDGridManager.h"
#include "KVIDZALine.h"
#include "KVIDZAGrid.h"
#include "KVIDentifier.h"
#include "KVDropDownDialog.h"
#include "KVVirtualIDFitter.h"
#include "TMethod.h"
#include "TContextMenu.h"
#include "TGMsgBox.h"
#include "KVInputDialog.h"
#include "KVHistoManipulator.h"
 
#include "KVSpiderIdentificator.h"
#include "KVZALineFinder.h"
 
#define FACTOR -1.
 
class KVIDGridEditor : public KVBase {
   friend class KVIDGridEditorCanvas;
 
public:
   enum MatrixType {
      kSiCsI,
      kSiSi,
      kChIoSi,
      kPSA
   };
 
protected:
 
   Bool_t fDebug;
   Bool_t fSVGMode;
   Bool_t fBlackMode;
   Bool_t fJoelMode;
   Int_t  fSVGIndex;
   TPaveText* fKeyShow;
 
   KVIDGridEditorCanvas* fCanvas;
   TVirtualPad*          fPad;
   TGraph*               fPivot;
   TGraph*               fPiedestal;
   KVString              fListOfMethods;
   KVString              fDefaultMethod;
 
    KVHashList* lplabel;
    KVHashList* lplabel2;
    KVHashList* lplabel3;
    KVHashList* lplabel4;
    KVHashList* lplabel5;
 
   TPaveLabel* modulator;
 
    TH2* TheHisto;
   TH2*        TheHistoChoice;
    KVIDGraph* TheGrid;
    KVUnownedList* ListOfLines;
 
    Int_t SelectedColor;
 
    Double_t x0;
    Double_t y0;
    Int_t itrans, iact, iopt, imod;
 
    TF1* ft;
    TF1* fs;
    TF1* fsy;
    TF2* frx;
    TF2* fry;
 
    Double_t fAx, fBx, fAy, fBy;
 
    Bool_t ownhisto;
    Bool_t dlmode;
    Bool_t drawmode;
    Bool_t aoemode;
    Bool_t selectmode;
    Bool_t venermode;
   Bool_t ready;
 
   Bool_t is_col;
 
    Double_t xmin, xmax, ymin, ymax;
    Double_t oldx, oldy;
    Bool_t moved;
 
   Double_t fSpiderFactor;
   Int_t    fSpiderZp;
   TString  fSpiderOption;
   Bool_t   fAutoZoomMode;
 
   Int_t fPointStyle;
   Double_t fPointSize;
 
   std::vector<Color_t> cut_line_colors = {kRed,kYellow};
   std::vector<Color_t> cut_marker_colors = {kBlack,kGreen-7};
   std::vector<Color_t> identifier_line_colors_no_mass = {kBlack,kBlue};
   std::vector<Color_t> identifier_marker_colors_no_mass = {kBlack,kRed};
   std::vector<Color_t> identifier_line_colors_with_mass = {kAzure+7,kOrange-3};
   std::vector<Color_t> identifier_marker_colors_with_mass = {kPink-9,kYellow};
 
protected:
 
   Double_t GetX0();
   Double_t GetY0();
   Double_t GetPdx();
   Double_t GetPdy();
 
   Bool_t HandleKey(Int_t px, Int_t py);
   void MoveHor(Int_t sign, Double_t speed = 1, Bool_t update = true);
   void MoveVert(Int_t sign, Double_t speed = 1, Bool_t update = true);
 
   void ZoomOnMouse();
   void AddMethod(const char* theMethod);
   void SelectTrans(TPaveLabel* label);
 
   void SetDefault();                  //
   void init();                     //
   void AddTransformation(TString label);          //construction de l'interface
   void AddAction(TString label);                  //graphique et initialisation
   void AddGridOption(TString label, KVHashList* thelist);        //
   void ConstructModulator();             //
 
   void TranslateX(Int_t Sign);
   void TranslateY(Int_t Sign);
   void RotateZ(Int_t Sign);
   void ScaleX(Int_t Sign);
   void ScaleY(Int_t Sign);
   void ScaleXY(Int_t Sign);
   void ScaleCurvature(Int_t Sign);
 
   void DynamicZoom(Int_t Sign, Int_t px = 0, Int_t py = 0, Double_t speed = .5); //zoom molette
   void Unzoom();                      //unzoom histo (accessible via l'interface : I)
   void ZoomSelected();                //zoom 'gnuplot'
 
   void SetLogz();                     //echelle log sur Oz
   void SetLogy();                     //echelle log sur Oy
   void SetLogx();                     //echelle log sur Ox
   void Undo();                        //annule toute les operation sur la grille courante (I)
 
   void SetPivot(Double_t xx0, Double_t yy0);       // set pivot position (double click on the histo)
   void SetPiedestal(Double_t ppdx, Double_t ppdy); // set piedestal position (shift click on the histo)
 
   void SetEditable(TPaveLabel* label);
   void SelectLines(const Char_t* label);
 
   void NewLine();                  //'Line'
   void NewCut();                   //'Cut'
   void NewInfo();                  //'Info'
   void FitGrid();                  //'Fit' avec fonctionnelle de T.G
   void TestGrid();                 //'Test' linearisation de l'histo avec la grille courante
 
   void DeleteObject(KVIDentifier* obj);           //'Delete'
 
   void DrawAtt(Bool_t piv);              //affiche les bouton sur le canvas
   void ResetColor(KVIDentifier* Ident);        //redonne la couleur d'origine a une ligne
   void ResetColor(KVSeqCollection* IdentList);          //redonne la couleur d'origine a une liste de ligne
 
   void SuggestMoreAction();              //'More', modifier cette methode pour ajouter des fonctionnalites !!!
   void ChooseSelectedColor();               //'More' -> SetSelectedColor change la couleur des lignes selectionnes
   void OpenRootFile();                //pas implemente
 
   void DispatchOrder(TPaveLabel* label);       //methode qui distribue les ordres quand on clic sur un bouton
   void ChangeStep(const char* title, Int_t dstep = 1);
   const char* WhoIsSelected();              //methode qui regarde quelle transformation est selectionnee
 
   TString ListOfHistogramInMemory();           //revoie la liste des histo dans gFile
   TString PreselectHistogram(TString ListOfName, Int_t ipert = 0); //revoie le nom de l' histo qui as le mm nom que la grille (si existe)
   TH2* FindInCanvases(const Char_t* name);
   void RecurseFileStructureFindHistos(TString& hist_names, TDirectory* the_directory);
 
   void ChangeZoomRatio(Int_t sign, Double_t speed = .5);
public:
 
   KVIDGridEditor();
   KVIDGridEditor(const KVIDGridEditor&) ;
   ROOT_COPY_ASSIGN_OP(KVIDGridEditor)
   virtual ~KVIDGridEditor();
   void Copy(TObject&) const override;
 
   void StartViewer();                 //ouvre l'interface graphique, utilise quand double clic sur une grille dans le gIDGridManagerGUI
   void Clear(const Option_t* opt = "") override;        //nettoi l'interface ('all' enleve la grille et l'histo)
   void Close();                 //ferme l'interface graphique
   Bool_t IsClosed();                  //true si l'interface est fermee
   void CanvasWasClosed();     // called when GUI is used to close the canvas
   void UpdateViewer();                //rafraichit l'interface
   void ForceUpdate();                 //rafraichit toute l'interface de force
   void SaveCurrentGrid();             //'More' -> SaveCurrentGrid ouvre une boite de dialogue pour sauver la grille
 
   void AutoFit();            //'More' -> AutoFit try to autofit the selected lines
 
   void   SetPointStyle(int pstyle);
   int    GetPointStyle()
   {
      return fPointStyle;
   }
   void   SetPointSize(double psize);
   double GetPointSize()
   {
      return fPointSize;
   }
 
   void SetHisto(TH2* hh);             //donne l'histo a l'editeur
   void SetGrid(KVIDGraph* gg, Bool_t histo = true);  //donne la grille a l'editeur
   void SetGrid(TString GridName);           //donne la grille a l'editeur par son nom
 
   KVIDGraph* GetGrid()
   {
      return TheGrid;
   }
   TH2*       GetHisto()
   {
      return TheHisto;
   }
 
   TCanvas* GetEditorCanvas()
   {
      return fCanvas;
   }
 
   void SelectLabel();                 //methode ajouter a la liste des auto-exec du canvas : gere les bouton
   void MakeTransformation();             //methode ajouter a la liste des auto-exec du canvas : gere le reste
 
   void SetSelectedColor(Int_t color)
   {
      SelectedColor = color;
   }
   void SelectLinesByZ(const Char_t* ListOfZ);        //'More' -> 'SelectLinesByZ'
   void SetDebug(Bool_t debug)
   {
      fDebug = debug;
   }
   void SetSVGMode()
   {
      fSVGMode = !fSVGMode;
   }
 
   Int_t GetSpiderZp()
   {
      return fSpiderZp;
   }
   Double_t GetSpiderFactor()
   {
      return fSpiderFactor;
   }
 
   void SetSpiderOption(const char* opt = "DRLF")
   {
      fSpiderOption = opt;
   }
   void SpiderIdentification();
   void FindZALines();
   void ChangeMasses(const Char_t* Zl, Int_t dA);
   void ChangeCharges(const Char_t* Zl, Int_t dZ);
 
   void ResetScalingRecap()
   {
      fAx = fAy = 1;
      fBx = fBy = 0;
   }
   void PrintScalingRecap();
 
   void ResetGridColors();
 
   ClassDefOverride(KVIDGridEditor, 1) // outil de modification de grille.
};
 
R__EXTERN KVIDGridEditor* gIDGridEditor;
 
#endif