ColorAxis.cxx

Go to the documentation of this file.

#include "ColorAxis.h"
#include "Rainbow.h"

#include "TVirtualPad.h"
#include "TBox.h"


#include <cstring>
#include <iostream>
using namespace std;

ClassImp(ColorAxis)

ColorAxis::ColorAxis(int ncolors, double width, double x1, double y1, double x2, double y2, int ndiv, 
                     Option_t* option)
    : RangeAxis(x1,y1,x2,y2,ndiv,option)
    , fNcolors(ncolors)
    , fWidth(width)
    , fScale(1.0)
{
}
ColorAxis::ColorAxis(int ncolors, double width, RangeDouble* range, 
                     double x1, double y1, double x2, double y2, int ndiv,
                     Option_t* option)
    : RangeAxis(range,x1,y1,x2,y2,ndiv,option)
    , fNcolors(ncolors)
    , fWidth(width)
    , fScale(1.0)
{
}
ColorAxis::~ColorAxis()
{
}


// override this so as to add color
void ColorAxis::PaintAxis(double xmin, double ymin, double xmax, double ymax,
                          double &wmin, double &wmax, int &ndiv, Option_t *chopt,
                          double, bool)
{
    // First paint colors
    this->PaintPalette();

    // Then the TGaxis
#if 0
    cerr << "(" << xmin << "," << ymin << ") --> "
         << "(" << xmax << "," << ymax << "), "
         << "[" << wmin << "," << wmax << "]\n";
#endif

    if (wmax < wmin) {
        cerr << "Warning: wmax(" << wmax << ") < wmin(" << wmin <<"), reversing\n";
        std::swap(wmin,wmax);
    }
    if (wmin == wmax) {
        wmin /= 2.0;
        if (wmax == 0.0) wmax = 1.0;
        else wmax *= 2.0;
    }

    wmin /= fScale;
    wmax /= fScale;

    this->RangeAxis::PaintAxis(xmin,ymin,xmax,ymax,wmin,wmax,ndiv,chopt);
}

void ColorAxis::PaintPalette()
{
    static Rainbow rainbow;

    const char* opt = this->GetOption();
    int sign = 1;

    // Attempt to deal with TGaxis inconsistent options
    if (strstr(opt,"-")) sign = -1;
    if (strstr(opt,"+") && strstr(opt,"L")) sign = -1;
    
    double dx=0,dy=0, width=0, height=0;
    if (fX1 == fX2) {           // vertical
        dy = (fY2-fY1)/(1.0*fNcolors);
        width = sign*fWidth;
    }
    else {                      // horizontal
        dx = (fX2-fX1)/(1.0*fNcolors);
        height = sign*fWidth;
    }

    
//    cerr << "[" << dx << "," << dy << "] --> [" << width << "," << height << "]\n";

    TBox box;
    for (int i = 0; i < fNcolors; ++i) {
        double x1 = fX1 + i*dx;
        double x2 = fX1 + (i+1)*dx + width;
        double y1 = fY1 + i*dy;
        double y2 = fY1 + (i+1)*dy + height;

//        cerr << x1 << "," << y1 << " --> " << x2 << "," << y2 << endl;

        double rel = (1.0*i)/(1.0*fNcolors);
        int col = rainbow.GetRelativeColor(rel);

        box.SetFillColor(col);
        box.TAttFill::Modify();

        gPad->PaintBox(x1,y1,x2,y2);
   }
}

---