JStyle.hh

Go to the documentation of this file.

#ifndef __JROOT__JSTYLE__
#define __JROOT__JSTYLE__

#include "TStyle.h"
#include "TCanvas.h"

#include "Jeep/JProperties.hh"


/**
 * \author mdejong
 */

namespace JROOT {}
namespace JPP { using namespace JROOT; }

namespace JROOT {

  /**
   * Wrapper class around ROOT TStyle.
   */
  class JStyle : 
    public TStyle 
  {
  public:
    struct JParameters {
      /**
       * Default constructor.
       */
      JParameters() :
        fontSize    (1.0),
        topMargin   (0.1),
        bottomMargin(0.15),
        leftMargin  (0.15),
        rightMargin (0.10),
        titleOffset (1.00),
        tickLength  (1.00)
      {}


      /**
       * Get properties of this class.
       */
      JProperties getProperties()
      {
        return JParametersHelper(*this);
      }


      /**
       * Get properties of this class.
       */
      JProperties getProperties() const
      {
        return JParametersHelper(*this);
      }
      
      Double_t fontSize;
      Double_t topMargin;
      Double_t bottomMargin;
      Double_t leftMargin;
      Double_t rightMargin;
      Double_t titleOffset;
      Double_t tickLength;

    private:
      /**
       * Auxiliary class for I/O of parameters.
       */
      class JParametersHelper :
        public JProperties
      {
      public:
        /**
         * Constructor.
         *
         * \param  object                  parameters
         */
        template<class JParameters_t>
        JParametersHelper(JParameters_t& object)
        {
          insert(gmake_property(object.fontSize));
          insert(gmake_property(object.topMargin));
          insert(gmake_property(object.bottomMargin));
          insert(gmake_property(object.leftMargin));
          insert(gmake_property(object.rightMargin));
          insert(gmake_property(object.titleOffset));
          insert(gmake_property(object.tickLength));
        }
      };
    };
    

    /**
     * Constructor.
     *
     * \param  title          title
     * \param  x              number of pixels in X
     * \param  y              number of pixels in Y
     * \param  parameters     parameters
     */
    JStyle(const char*  title,
           const UInt_t x,
           const UInt_t y,
           const JParameters& parameters = JParameters()) :
      TStyle(title, NULL)
    {
      const Double_t font_size    = (Double_t) y * 0.7e-4 * parameters.fontSize;
      const Double_t title_offset = 1.5                   * parameters.titleOffset;
      const Double_t tick_length  = 3.0e-5                * parameters.tickLength;

      SetPadTopMargin   (parameters.topMargin);
      SetPadBottomMargin(parameters.bottomMargin);
      SetPadLeftMargin  (parameters.leftMargin);
      SetPadRightMargin (parameters.rightMargin);
      
      SetCanvasColor(0);
      SetCanvasBorderMode(0);
      SetCanvasBorderSize(0);

      SetPadColor(0);
      SetPadBorderMode(0);
      SetPadBorderSize(0);

      for (const auto option : { "X", "Y", "Z"}) {
        SetTickLength (tick_length * (x < y ? x : y), option);
        SetLabelSize  (font_size,                     option);
        SetTitleOffset(title_offset,                  option);
        SetTitleSize  (font_size,                     option);
      }

      SetTitleX(0.05);
      SetTitleY(0.99);

      SetStatColor(0);
      SetStatBorderSize(1);
      SetStatFontSize(font_size);
      SetOptStat(111111);
      SetOptFit(1111);

      SetPaintTextFormat("3.0f");
      SetPalette(1);

      SetTitleFillColor(0);
      SetTitleAlign(23);
      SetTitleX(0.5);
      SetTitleBorderSize(0);
    }


    /**
     * Scale font sizes.
     *
     * \param  factor         scale factor
     */
    void setFontSize(const Double_t factor)
    {
      for (const auto option : { "X", "Y", "Z"}) {
        SetLabelSize(GetLabelSize(option) * factor, option);
        SetTitleSize(GetTitleSize(option) * factor, option);
      }
    }
  };
}

#endif