palette.cpp

Go to the documentation of this file.

#include "core/palette.hpp"

#include <fstream>
#include <sstream>

#include "TColor.h"

#include "core/utilities.hpp"

using namespace std;

Palette::Palette(const string &file,
                 const string &palette):
  file_(file),
  palette_(palette){
  }

Palette::Palette(const string &palette):
  Palette("txt/colors.txt", palette){
}

const string & Palette::File() const{
  return file_;
}

Palette & Palette::File(const string &file){
  file_ = file;
  return *this;
}

const string & Palette::PaletteName() const{
  return palette_;
}

Palette & Palette::PaletteName(const string &palette){
  return PaletteName(file_, palette);
}

Palette & Palette::PaletteName(const string &file,
                               const string &palette){
  file_ = file;
  palette_ = palette;
  return *this;
}

Int_t Palette::operator()(const string &color_name) const{
  ifstream file(file_);
  string line;
  string current_palette_ = "";
  int line_num = 0;
  while(getline(file, line)){
    ++line_num;
    ReplaceAll(line, "=", " ");
    ReplaceAll(line, "\t", " ");
    auto start  = line.find('[');
    auto end = line.find(']');
    if(start==string::npos && end!=string::npos){
      ERROR("Could not find opening brace in line "+to_string(line_num));
    }
    if(start!=string::npos && end==string::npos){
      ERROR("Could not find closing brace in line "+to_string(line_num));
    }
    if(start<end && start != string::npos && end != string::npos){
      current_palette_ = line.substr(start+1, end-start-1);
    }else if(current_palette_ == palette_
             && line.size()
             && line.at(0)!='#'){
      istringstream iss(line);
      string color;
      iss >> color;
      if(color != color_name) continue;
      Int_t r, g, b;
      iss >> r >> g >> b;
      return RGB(r, g, b);
    }
  }
  DBG("No color " << color_name << " in palette " << palette_ << " in file " << file_);
  return 0;
}

Int_t Palette::RGB(Int_t r, Int_t g, Int_t b){
  return TColor::GetColor(r, g, b);
}

Int_t Palette::RGB(Float_t r, Float_t g, Float_t b){
  return TColor::GetColor(r, g, b);
}

Int_t Palette::HSV(Float_t h, Float_t s, Float_t v){
  Float_t r, g, b;
  TColor::HSV2RGB(h, s, v, r, g, b);
  return RGB(r, g, b);
}

Int_t Palette::HLS(Int_t h, Int_t l, Int_t s){
  Int_t r, g, b;
  TColor::HLS2RGB(h, l, s, r, g, b);
  return RGB(r, g, b);
}

Int_t Palette::HLS(Float_t h, Float_t l, Float_t s){
  Float_t r, g, b;
  TColor::HLS2RGB(h, l, s, r, g, b);
  return RGB(r, g, b);
}