TKSMultiRibbons.h

// TKSMultiRibbons
//    See TKSMultiRes for more information
// -- Note: For CINT, load using gROOT->LoadMacro("TKSMultiRibbons.cxx");
// ~ Mark J. Duvall ~ mjduvall@hawaii.edu ~ 08/2022 ~ //
 
//Copyright (C) 2022 Mark J. Duvall / T. Rocks Science
//
//    This program is free software: you can redistribute it and/or modify
//    it under the terms of the GNU General Public License as published by
//    the Free Software Foundation, either version 3 of the License, or
//    (at your option) any later version.
//
//    This program is distributed in the hope that it will be useful,
//    but WITHOUT ANY WARRANTY; without even the implied warranty of
//    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
//    GNU General Public License for more details.
//
//    You should have received a copy of the GNU General Public License
//    along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
#ifndef TKSMultiRibbons
#define TKSMultiRibbons
 
//#include "TKSMultiRes.cxx"
 
class TKSMultiRibbons : public TClass {
 
public:
  // graph-selection enum
  enum EPlotTypes                                   
  {
    kRibbons,                                       
    kHeatmap,                                       
    kBoth                                       
  };
 
private:
  // members
  /* TKSMultiRes*       fkMultiRes;                         ///< Invoking TKSMultiRes object (if any) */
  Bool_t        fkInit;                             
  Int_t         fkNFiles;                           
  Int_t         fkNAngles;                          
  EPlotTypes        fPlotTypes;                         
  EPlotTypes        fCurrentPlotType;                       
  TObjArray*        fFileList;                          
  TString       fOutFileName;                           
  TFile*        fOutFile;                           
  TCanvas*      fHeatmapCanvas;                         
  TCanvas*      fRibbonCanvas;                          
  TList*        fRibbonList;                            
  TList*        fCutList;                           
  TMatrixD*     fDeltas;                            
 
private:
  // internal methods
  /* void           SetMultiRes(TKSMultiRes* newMultiRes) { fkMultiRes=newMultiRes; } */
  void          SetInit(Bool_t newInit) { fkInit=newInit; }
  void          FillFileList();                         
  void          SetNFiles(Int_t newNFiles) { fkNFiles=newNFiles; }
  void          SetNAngles(Int_t newNAngles) { fkNAngles=newNAngles; }
  void          SetDeltas(TMatrixD *newDeltas);                 
  TMatrixD      RetrieveData();                         
  void          FillData();                         
 
public:
  // public methods
  TKSMultiRibbons();                                    
  Bool_t        IsInit() { return fkInit; }                 
  // setters and getters
  void          SetPlotTypes(EPlotTypes plotSelection) { fPlotTypes=plotSelection; }
  void          SetCurrentPlotType(EPlotTypes plotSelection) { fCurrentPlotType=plotSelection; }
  void          SetFileList(TObjArray* newFileList) { fFileList=newFileList; }
  void          SetOutFileName(TString fileName) { fOutFileName=fileName; }
  void          SetOutFileName(const char* filename) { fOutFileName=TString(filename); }
  void          SetOutFile(TFile *fNew) { fOutFile=fNew; }
  /* TKSMultiRes*       GetMultiRes() { return fkMultiRes; } */
  Int_t         GetNFiles() { return fkNFiles; }
  Int_t         GetNAngles() { return fkNAngles; }
  EPlotTypes        GetPlotTypes() { return fPlotTypes; }
  EPlotTypes        GetCurrentPlotType() { return fCurrentPlotType; }
  TObjArray*        GetFileList() { return fFileList; }
  TString       GetOutFileName() { return fOutFileName; }
  TFile*        GetOutFile() { return fOutFile; }
  TCanvas*      GetHeatmapCanvas() { return fHeatmapCanvas; }
  TCanvas*      GetRibbonCanvas() { return fRibbonCanvas; }
  TList*        GetRibbonList() { return fRibbonList; }
  TList*        GetCutList() { return fCutList; }
  TMatrixD*     GetDeltas() { return fDeltas; }
  // general
  void          Init();                             
  void          DrawPlots();                            
  void          Ribbons();                          
  void          Heatmap();                          
  void          Save();                             
  void          Run();                              
  void          PrintEnums();                           
 
// Integrating the TKSMultiRibbons class into ROOT
ClassDef(TKSMultiRibbons,2) 
 
}; //end class
 
// all pau!   )
#endif