Skip to content

Commit 0160cd4

Browse files
niavokniavok
committed
add SetNextColorsData to give per point colors
Implementation and demo for lines, scatter, infinite lines, shaded plots and stairstep
1 parent 626e391 commit 0160cd4

File tree

4 files changed

+466
-149
lines changed

4 files changed

+466
-149
lines changed

‎implot.h

+2
Original file line numberDiff line numberDiff line change
@@ -1117,6 +1117,8 @@ IMPLOT_API void SetNextFillStyle(const ImVec4& col = IMPLOT_AUTO_COL, float alph
11171117
IMPLOT_API void SetNextMarkerStyle(ImPlotMarker marker = IMPLOT_AUTO, float size = IMPLOT_AUTO, const ImVec4& fill = IMPLOT_AUTO_COL, float weight = IMPLOT_AUTO, const ImVec4& outline = IMPLOT_AUTO_COL);
11181118
// Set the error bar style for the next item only.
11191119
IMPLOT_API void SetNextErrorBarStyle(const ImVec4& col = IMPLOT_AUTO_COL, float size = IMPLOT_AUTO, float weight = IMPLOT_AUTO);
1120+
// Set color buffer for the next item only. The colors array must be keep alive until the next item plot.
1121+
IMPLOT_API void SetNextColorsData(ImPlotCol_ target, ImU32* const& colors, int stride = sizeof(ImU32));
11201122

11211123
// Gets the last item primary color (i.e. its legend icon color)
11221124
IMPLOT_API ImVec4 GetLastItemColor();

‎implot_demo.cpp

+182
Original file line numberDiff line numberDiff line change
@@ -960,6 +960,187 @@ void Demo_MarkersAndText() {
960960
}
961961
}
962962

963+
void Demo_PlotColors() {
964+
// Lines
965+
{
966+
static float xs1[1001], ys1[1001];
967+
ImU32 cs1[1001];
968+
for (int i = 0; i < 1001; ++i) {
969+
xs1[i] = i * 0.001f;
970+
ys1[i] = 0.5f + 0.5f * sinf(50 * (xs1[i] + (float)ImGui::GetTime() / 10));
971+
ImVec4 lineColor = ImPlot::SampleColormap(ys1[i], ImPlotColormap_Spectral);
972+
lineColor.w = 1 - powf(1 - (i * 0.001f), 4);
973+
cs1[i] = ImGui::GetColorU32(lineColor);
974+
}
975+
static double xs2[11], ys2[11];
976+
ImU32 cls2[11], cms2[11];
977+
for (int i = 0; i < 11; ++i) {
978+
xs2[i] = i * 0.1f;
979+
ys2[i] = xs2[i] * xs2[i];
980+
cls2[i] = ImGui::GetColorU32(ImPlot::RandomColor());
981+
cms2[i] = ImGui::GetColorU32(ImPlot::RandomColor());
982+
}
983+
if (ImPlot::BeginPlot("Line Plot with colors")) {
984+
ImPlot::SetupAxes("x", "f(x)");
985+
ImVec4 s1Color = ImPlot::SampleColormap(0.5f, ImPlotColormap_Spectral);
986+
ImPlot::SetNextLineStyle(s1Color);
987+
ImPlot::SetNextColorsData(ImPlotCol_Line, cs1);
988+
ImPlot::PlotLine("sin(x)", xs1, ys1, 1001);
989+
ImVec4 s2Color = ImPlot::SampleColormap(0.5f, ImPlotColormap_Twilight);
990+
ImPlot::SetNextLineStyle(s2Color);
991+
ImPlot::SetNextMarkerStyle(ImPlotMarker_Circle);
992+
ImPlot::SetNextColorsData(ImPlotCol_Line, cms2);
993+
ImPlot::SetNextColorsData(ImPlotCol_MarkerFill, cms2);
994+
ImPlot::SetNextColorsData(ImPlotCol_MarkerOutline, cms2);
995+
ImPlot::PlotLine("x^2", xs2, ys2, 11);
996+
ImPlot::EndPlot();
997+
}
998+
}
999+
// Scatter
1000+
{
1001+
srand(0);
1002+
static float xs1[100], ys1[100];
1003+
static ImU32 cs1[1001];
1004+
for (int i = 0; i < 100; ++i) {
1005+
xs1[i] = i * 0.01f;
1006+
ys1[i] = xs1[i] + 0.1f * ((float)rand() / (float)RAND_MAX);
1007+
cs1[i] = ImGui::GetColorU32(ImPlot::SampleColormap(10 * (ys1[i] - xs1[i]), ImPlotColormap_Viridis));
1008+
}
1009+
static float xs2[50], ys2[50];
1010+
ImU32 cmos2[50], cmfs2[50];
1011+
1012+
for (int i = 0; i < 50; i++) {
1013+
xs2[i] = 0.25f + 0.2f * ((float)rand() / (float)RAND_MAX);
1014+
ys2[i] = 0.75f + 0.2f * ((float)rand() / (float)RAND_MAX);
1015+
1016+
ImVec4 markerOutlineColor = ImPlot::SampleColormap(((float)rand() / (float)RAND_MAX), ImPlotColormap_Plasma);
1017+
ImVec4 markerFillColor = ImVec4(markerOutlineColor.x, markerOutlineColor.x, markerOutlineColor.z, 0.5);
1018+
cmfs2[i] = ImGui::GetColorU32(markerFillColor);
1019+
cmos2[i] = ImGui::GetColorU32(markerOutlineColor);
1020+
}
1021+
1022+
if (ImPlot::BeginPlot("Scatter Plot with colors")) {
1023+
ImVec4 s1Color = ImPlot::SampleColormap(0.5f, ImPlotColormap_Viridis);
1024+
ImPlot::SetNextMarkerStyle(ImPlotMarker_Circle, 4, s1Color, IMPLOT_AUTO, s1Color);
1025+
ImPlot::SetNextColorsData(ImPlotCol_MarkerOutline, cs1);
1026+
ImPlot::SetNextColorsData(ImPlotCol_MarkerFill, cs1);
1027+
ImPlot::PlotScatter("Data 1", xs1, ys1, 100);
1028+
1029+
ImVec4 s2Color = ImPlot::SampleColormap(0.5f, ImPlotColormap_Plasma);
1030+
ImPlot::SetNextMarkerStyle(ImPlotMarker_Square, 6, s2Color, IMPLOT_AUTO, s2Color);
1031+
ImPlot::SetNextColorsData(ImPlotCol_MarkerOutline, cmos2);
1032+
ImPlot::SetNextColorsData(ImPlotCol_MarkerFill, cmfs2);
1033+
ImPlot::PlotScatter("Data 2", xs2, ys2, 50);
1034+
1035+
ImPlot::EndPlot();
1036+
}
1037+
}
1038+
1039+
// Infinite lines
1040+
{
1041+
static double vals[] = { 0.25, 0.5, 0.75 };
1042+
static ImU32 colors[] = { ImGui::GetColorU32(ImPlot::RandomColor()), ImGui::GetColorU32(ImPlot::RandomColor()), ImGui::GetColorU32(ImPlot::RandomColor()) };
1043+
if (ImPlot::BeginPlot("##Infinite with colors")) {
1044+
ImPlot::SetupAxes(NULL, NULL, ImPlotAxisFlags_NoInitialFit, ImPlotAxisFlags_NoInitialFit);
1045+
ImPlot::SetNextLineStyle(ImVec4(1, 1, 1, 0.5));
1046+
ImPlot::SetNextColorsData(ImPlotCol_Line, colors);
1047+
ImPlot::PlotInfLines("VLines", vals, 3);
1048+
ImPlot::SetNextLineStyle(ImVec4(1, 1, 1, 0.5));
1049+
ImPlot::SetNextColorsData(ImPlotCol_Line, colors);
1050+
ImPlot::PlotInfLines("HLines", vals, 3, ImPlotInfLinesFlags_Horizontal);
1051+
ImPlot::EndPlot();
1052+
}
1053+
}
1054+
1055+
// Shaded plots
1056+
{
1057+
static float xs[1001], ys[1001], ys1[1001], ys2[1001], ys3[1001], ys4[1001];
1058+
static ImU32 cl1[1001];
1059+
static ImU32 cf1[1001];
1060+
static ImU32 cf2[1001];
1061+
srand(0);
1062+
1063+
for (int i = 0; i < 1001; ++i) {
1064+
xs[i] = i * 0.001f;
1065+
ys[i] = 0.25f + 0.25f * sinf(25 * xs[i]) * sinf(5 * xs[i]) + RandomRange(-0.01f, 0.01f);
1066+
ys1[i] = ys[i] + RandomRange(0.1f, 0.12f);
1067+
ys2[i] = ys[i] - RandomRange(0.1f, 0.12f);
1068+
ys3[i] = 0.75f + 0.2f * sinf(37 * xs[i]);
1069+
ys4[i] = 0.75f + 0.1f * cosf(29 * xs[i]);
1070+
1071+
ImVec4 lineColor = ImPlot::SampleColormap(i/1001.f, ImPlotColormap_Hot);
1072+
ImVec4 fillColor = lineColor;
1073+
fillColor.w = 0.25f;
1074+
cl1[i] = ImGui::GetColorU32(lineColor);
1075+
cf1[i] = ImGui::GetColorU32(fillColor);
1076+
1077+
float diffNormalized = (float) fabs(ys3[i] - ys4[i]) / (0.3f);
1078+
ImVec4 fillColor2 = ImPlot::SampleColormap(diffNormalized, ImPlotColormap_Plasma);
1079+
fillColor2.w = 0.8f;
1080+
cf2[i] = ImGui::GetColorU32(fillColor2);
1081+
}
1082+
1083+
if (ImPlot::BeginPlot("Shaded Plots with colors")) {
1084+
ImPlot::SetNextColorsData(ImPlotCol_Fill, cf1);
1085+
ImPlot::PlotShaded("Uncertain Data", xs, ys1, ys2, 1001);
1086+
ImPlot::SetNextColorsData(ImPlotCol_Line, cl1);
1087+
ImPlot::PlotLine("Uncertain Data", xs, ys, 1001);
1088+
ImPlot::SetNextColorsData(ImPlotCol_Fill, cf2);
1089+
ImPlot::PlotShaded("Overlapping", xs, ys3, ys4, 1001);
1090+
ImPlot::PlotLine("Overlapping", xs, ys3, 1001);
1091+
ImPlot::PlotLine("Overlapping", xs, ys4, 1001);
1092+
ImPlot::EndPlot();
1093+
}
1094+
}
1095+
1096+
// Stairstep
1097+
{
1098+
static float ys1[21], ys2[21], s;
1099+
static ImU32 clys1[21], cfys1[21], clys2[21], cfys2[21];
1100+
for (int i = 0; i < 21; ++i) {
1101+
ys1[i] = 0.75f + 0.2f * sinf(10 * i * 0.05f);
1102+
ys2[i] = 0.25f + 0.2f * sinf(10 * i * 0.05f);
1103+
ImVec4 cl1 = ImPlot::RandomColor();
1104+
ImVec4 cl2 = ImPlot::RandomColor();
1105+
ImVec4 cf1 = cl1;
1106+
ImVec4 cf2 = cl2;
1107+
cf1.w = 0.5;
1108+
cf2.w = 0.5;
1109+
1110+
clys1[i] = ImGui::GetColorU32(cl1);
1111+
clys2[i] = ImGui::GetColorU32(cl2);
1112+
cfys1[i] = ImGui::GetColorU32(cf1);
1113+
cfys2[i] = ImGui::GetColorU32(cf2);
1114+
}
1115+
static ImPlotStairsFlags flags = 0;
1116+
CHECKBOX_FLAG(flags, ImPlotStairsFlags_Shaded);
1117+
if (ImPlot::BeginPlot("Stairstep Plot with colors")) {
1118+
ImPlot::SetupAxes("x", "f(x)");
1119+
ImPlot::SetupAxesLimits(0, 1, 0, 1);
1120+
1121+
ImPlot::PushStyleColor(ImPlotCol_Line, ImVec4(0.5f, 0.5f, 0.5f, 1.0f));
1122+
ImPlot::PlotLine("##1", ys1, 21, 0.05f);
1123+
ImPlot::PlotLine("##2", ys2, 21, 0.05f);
1124+
ImPlot::PopStyleColor();
1125+
1126+
ImPlot::SetNextMarkerStyle(ImPlotMarker_Circle);
1127+
ImPlot::SetNextFillStyle(IMPLOT_AUTO_COL, 0.25f);
1128+
ImPlot::SetNextColorsData(ImPlotCol_Line, clys1);
1129+
ImPlot::SetNextColorsData(ImPlotCol_Fill, cfys1);
1130+
ImPlot::SetNextColorsData(ImPlotCol_MarkerOutline, clys1);
1131+
ImPlot::PlotStairs("Post Step (default)", ys1, 21, 0.05f, 0, flags);
1132+
ImPlot::SetNextMarkerStyle(ImPlotMarker_Circle);
1133+
ImPlot::SetNextFillStyle(IMPLOT_AUTO_COL, 0.25f);
1134+
ImPlot::SetNextColorsData(ImPlotCol_Line, clys2);
1135+
ImPlot::SetNextColorsData(ImPlotCol_Fill, cfys2);
1136+
ImPlot::SetNextColorsData(ImPlotCol_MarkerOutline, clys2);
1137+
ImPlot::PlotStairs("Pre Step", ys2, 21, 0.05f, 0, flags | ImPlotStairsFlags_PreStep);
1138+
1139+
ImPlot::EndPlot();
1140+
}
1141+
}
1142+
}
1143+
9631144
//-----------------------------------------------------------------------------
9641145

9651146
void Demo_NaNValues() {
@@ -2214,6 +2395,7 @@ void ShowDemoWindow(bool* p_open) {
22142395
DemoHeader("Images", Demo_Images);
22152396
DemoHeader("Markers and Text", Demo_MarkersAndText);
22162397
DemoHeader("NaN Values", Demo_NaNValues);
2398+
DemoHeader("Plot Colors", Demo_PlotColors);
22172399
ImGui::EndTabItem();
22182400
}
22192401
if (ImGui::BeginTabItem("Subplots")) {

‎implot_internal.h

+5
Original file line numberDiff line numberDiff line change
@@ -1175,6 +1175,8 @@ struct ImPlotNextPlotData
11751175
// Temporary data storage for upcoming item
11761176
struct ImPlotNextItemData {
11771177
ImVec4 Colors[5]; // ImPlotCol_Line, ImPlotCol_Fill, ImPlotCol_MarkerOutline, ImPlotCol_MarkerFill, ImPlotCol_ErrorBar
1178+
ImU32 const* ColorsData[5];
1179+
int ColorsDataStride[5];
11781180
float LineWeight;
11791181
ImPlotMarker Marker;
11801182
float MarkerSize;
@@ -1194,7 +1196,10 @@ struct ImPlotNextItemData {
11941196
ImPlotNextItemData() { Reset(); }
11951197
void Reset() {
11961198
for (int i = 0; i < 5; ++i)
1199+
{
11971200
Colors[i] = IMPLOT_AUTO_COL;
1201+
ColorsData[i] = nullptr;
1202+
}
11981203
LineWeight = MarkerSize = MarkerWeight = FillAlpha = ErrorBarSize = ErrorBarWeight = DigitalBitHeight = DigitalBitGap = IMPLOT_AUTO;
11991204
Marker = IMPLOT_AUTO;
12001205
HasHidden = Hidden = false;

0 commit comments

Comments
 (0)