TRestMetadataPlot.h

/*************************************************************************
 * This file is part of the REST software framework.                     *
 *                                                                       *
 * Copyright (C) 2016 GIFNA/TREX (University of Zaragoza)                *
 * For more information see http://gifna.unizar.es/trex                  *
 *                                                                       *
 * REST 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.                                   *
 *                                                                       *
 * REST 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 a copy of the GNU General Public License along with   *
 * REST in $REST_PATH/LICENSE.                                           *
 * If not, see http://www.gnu.org/licenses/.                             *
 * For the list of contributors see $REST_PATH/CREDITS.                  *
 *************************************************************************/
 
#ifndef RestCore_TRestMetadataPlot
#define RestCore_TRestMetadataPlot
 
#include <TCanvas.h>
#include <TLatex.h>
 
#include "TRestRun.h"
 
class TRestMetadataPlot : public TRestMetadata {
   public:
    struct Graph_Info_Set {
        std::string name;
        std::string title;
        std::string yVariable;
        std::string metadataRule;
        std::string drawOption;
        Int_t lineColor;
        Int_t lineWidth;
        Int_t lineStyle;
        Int_t fillColor;
        Int_t fillStyle;
        Int_t markerStyle;
        Int_t markerSize;
        Int_t markerColor;
    };
 
    struct Plot_Info_Set {
        std::string name;
        std::string title;
        std::string xVariable;
        Bool_t logX;
        Bool_t logY;
        Bool_t legendOn;
        Bool_t timeDisplay;
        std::string labelX;
        std::string labelY;
        std::string save;
        TVector2 xRange;
        TVector2 yRange;
        std::vector<Graph_Info_Set> graphs;
    };
 
    struct Panel_Info {
        Float_t font_size;
        std::vector<Float_t> posX;
        std::vector<Float_t> posY;
        std::vector<std::string> label;
    };
 
   private:
    void InitFromConfigFile() override;
    Graph_Info_Set SetupGraphFromConfigFile(TiXmlElement* ele, Plot_Info_Set info);
 
    Int_t fNFiles;
 
    TVector2 fCanvasSize;
 
    TVector2 fCanvasDivisions;
 
    TString fCanvasSave;
 
    Int_t fLabelFont;
 
    Double_t fLabelOffsetX = 1.1;
    Double_t fLabelOffsetY = 1.3;
 
    Double_t fLabelScaleX = 1.2;
    Double_t fLabelScaleY = 1.3;
 
    Double_t fTicksScaleX = 1.5;
    Double_t fTicksScaleY = 1.5;
 
    Double_t fLegendX1 = 0.7;
    Double_t fLegendY1 = 0.75;
    Double_t fLegendX2 = 0.88;
    Double_t fLegendY2 = 0.88;
 
    TString fLegendOption = "lp";
 
    std::vector<Plot_Info_Set> fPlots;
 
    std::vector<Panel_Info> fPanels;
 
    std::vector<std::string> fPlotNamesCheck;  
 
    TRestRun* fRun;  
 
    std::vector<std::string> fRunInputFileName;  
 
    TCanvas* fCombinedCanvas;  
 
    void AddFile(TString fileName);
    void AddFileFromExternalRun();
    void AddFileFromEnv();
 
   protected:
   public:
    void Initialize() override;
 
    void PrintMetadata() override {}
 
    void SaveCanvasToPDF(TString fileName);
    void SavePlotToPDF(TString fileName, Int_t n = 0);
    void SaveGraphToPDF(TString fileName, Int_t nPlot = 0, Int_t nGraph = 0);
 
    void SetOutputPlotsFilename(TString fname) { fCanvasSave = fname; }
 
    Int_t GetPlotIndex(TString plotName);
 
    inline TVector2 GetCanvasSize() const { return fCanvasSize; }
 
    inline TVector2 GetCanvasDivisions() const { return fCanvasDivisions; }
 
    void GenerateCanvas();
 
    TRestMetadataPlot();
    TRestMetadataPlot(const char* configFilename, const char* name = "");
 
    virtual ~TRestMetadataPlot();
 
    ClassDefOverride(TRestMetadataPlot, 3);
};
#endif