File: plotLegend.cpp

package info (click to toggle)
groops 0%2Bgit20250907%2Bds-1
  • links: PTS, VCS
  • area: non-free
  • in suites: forky, sid
  • size: 11,140 kB
  • sloc: cpp: 135,607; fortran: 1,603; makefile: 20
file content (118 lines) | stat: -rw-r--r-- 3,593 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
/***********************************************/
/**
* @file plotLegend.cpp
*
* @brief Legend.
*
* @author Andreas Kvas
* @author Torsten Mayer-Guerr
* @date 2020-05-03
*
*/
/***********************************************/

#define DOCSTRING_PlotLegend

#include "base/import.h"
#include "base/string.h"
#include "config/configRegister.h"
#include "inputOutput/file.h"
#include "plot/plotMisc.h"
#include "plotLegend.h"

/***********************************************/

GROOPS_REGISTER_CLASS_WITHOUT_SUBS(PlotLegend, "plotLegendType")
GROOPS_READCONFIG_CLASS(PlotLegend, "plotLegendType")

/***********************************************/

PlotLegend::PlotLegend(Config &config, const std::string &name)
{
  try
  {
    height = 0;

    readConfigSequence(config, name, Config::MUSTSET, "", "");
    readConfig(config, "width",       width,     Config::DEFAULT,  "10",   "legend width [cm]");
    readConfig(config, "height",      height,    Config::OPTIONAL, "",     "legend height [cm] (default: estimated)");
    readConfig(config, "positionX",   positionX, Config::DEFAULT,  "1.05", "legend x-position in normalized (0-1) coordinates.");
    readConfig(config, "positionY",   positionY, Config::DEFAULT,  "1.0",  "legend y-position in normalized (0-1) coordinates.");
    readConfig(config, "anchorPoint", anchor,    Config::DEFAULT,  "TL",   "Two character combination of L, C, R (for left, center, or right) and T, M, B for top, middle, or bottom. e.g., TL for top left");
    readConfig(config, "columns",     ncolumns,  Config::DEFAULT,  "1",    "number of columns in legend");
    readConfig(config, "textColor",   textColor, Config::OPTIONAL, "",     "color of the legend text");
    readConfig(config, "fillColor",   fillColor, Config::OPTIONAL, "",     "fill color of the legend box");
    readConfig(config, "edgeLine",    edgeLine,  Config::OPTIONAL, "",     "style of the legend box edge");
    endSequence(config);
    if(isCreateSchema(config)) return;

    _hasEntries = FALSE;
  }
  catch(std::exception &e)
  {
    GROOPS_RETHROW(e)
  }
}

/***********************************************/

void PlotLegend::writeDataFile(const FileName &workingDirectory, const std::vector<std::string> &legendText)
{
  try
  {
    if(legendText.size()==0)
      return;

    _hasEntries = TRUE;
    OutFile legendFile(workingDirectory.append("legend.txt"));

    if(textColor)
      legendFile<<"C "<<textColor->str()<<std::endl;
    legendFile<<"N "<<ncolumns<<std::endl;
    for(UInt i=0; i<legendText.size(); i++)
    {
      std::vector<std::string> lines = String::split(legendText.at(i), '\n');
      for(const std::string &line : lines)
        if(!line.empty())
          legendFile<<line<<std::endl;
    }
  }
  catch(const std::exception& e)
  {
    GROOPS_RETHROW(e)
  }

}

/***********************************************/

std::string PlotLegend::scriptEntry() const
{
  try
  {
    if(!_hasEntries)
      return "";

    std::stringstream ss;
    ss<<"gmt pslegend legend.txt -R0/1/0/1 -JX"<<PlotBasics::scriptVariable("width")<<"/"<<PlotBasics::scriptVariable("height");
    ss<<" -Dn"<<positionX<<"/"<<positionY<<"+w"<<width<<"c/"<<height<<"c+j"<<anchor<<"+l1.5";
    if(fillColor || edgeLine)
    {
      ss<<" -F";
      if(fillColor)
        ss<<"+g"<<fillColor->str();
      if(edgeLine)
        ss<<"+p"<<edgeLine->str();
      else
        ss<<"+p0.0p,"<<fillColor->str();
    }
    ss<<" -O -K >> groopsPlot.ps"<<std::endl;
    return ss.str();
  }
  catch(std::exception &e)
  {
    GROOPS_RETHROW(e)
  }
}

/***********************************************/