Mirrors/crazydiskinfo
1
0
Fork 0
mirror of https://github.com/otakuto/crazydiskinfo.git synced 2025-07-26 02:48:32 +00:00
Code Issues Releases Wiki Activity

Add color name

Browse source
This commit is contained in:
Tact Yoshida 2019-01-02 21:42:02 +09:00
parent 8e9c9b298e
commit f366d8d22c
1 changed files with 60 additions and 55 deletions
Download patch file Download diff file Expand all files Collapse all files

115
main.cpp
View file

@ -18,6 +18,11 @@ constexpr int const DEVICE_BAR_HEIGHT = 4;
constexpr int const VERSION_HEIGHT = 1; constexpr int const VERSION_HEIGHT = 1;
constexpr int const HEALTH_COLOR = 1;
constexpr int const HEALTH_INV_COLOR = 4;
constexpr int const ATTRIBUTE_LEGEND_COLOR = 7;
constexpr int const TITLE_COLOR = 8;
int width; int width;
int height; int height;
@ -200,15 +205,15 @@ void drawVersion(WINDOW * window)
{ {
wresize(window, VERSION_HEIGHT, width); wresize(window, VERSION_HEIGHT, width);
wattrset(window, COLOR_PAIR(4)); wattrset(window, COLOR_PAIR(HEALTH_COLOR));
mvwhline(window, 0, 0, '-', width); mvwhline(window, 0, 0, '-', width);
wattroff(window, COLOR_PAIR(4)); wattroff(window, COLOR_PAIR(HEALTH_COLOR));
auto title = " " + TITLE + "-" + VERSION + " "; auto title = " " + TITLE + "-" + VERSION + " ";
wattrset(window, COLOR_PAIR(8)); wattrset(window, COLOR_PAIR(TITLE_COLOR));
mvwprintw(window, 0, (width - title.length()) / 2, title.c_str()); mvwprintw(window, 0, (width - title.length()) / 2, title.c_str());
wattroff(window, COLOR_PAIR(8)); wattroff(window, COLOR_PAIR(TITLE_COLOR));
wnoutrefresh(window); wnoutrefresh(window);
} }
@ -218,17 +223,17 @@ void drawDeviceBar(WINDOW * window, std::vector<SMART> const & smartList, int se
int x = 0; int x = 0;
for (int i = 0; i < static_cast<int>(smartList.size()); ++i) for (int i = 0; i < static_cast<int>(smartList.size()); ++i)
{ {
wattrset(window, COLOR_PAIR(1 + static_cast<int>(smartToHealth(smartList[i])))); wattrset(window, COLOR_PAIR(HEALTH_INV_COLOR + static_cast<int>(smartToHealth(smartList[i]))));
mvwprintw(window, 0, x, "%-7s", healthToString(smartToHealth(smartList[i])).c_str()); mvwprintw(window, 0, x, "%-7s", healthToString(smartToHealth(smartList[i])).c_str());
wattroff(window, COLOR_PAIR(1 + static_cast<int>(smartToHealth(smartList[i])))); wattroff(window, COLOR_PAIR(HEALTH_INV_COLOR + static_cast<int>(smartToHealth(smartList[i]))));
if (std::get<0>(smartList[i].temperature)) if (std::get<0>(smartList[i].temperature))
{ {
wattrset(window, COLOR_PAIR(1 + static_cast<int>(temperatureToHealth(std::get<1>(smartList[i].temperature))))); wattrset(window, COLOR_PAIR(HEALTH_INV_COLOR + static_cast<int>(temperatureToHealth(std::get<1>(smartList[i].temperature)))));
mvwprintw(window, 1, x, "%.1f ", smartList[i].temperature); mvwprintw(window, 1, x, "%.1f ", smartList[i].temperature);
waddch(window, ACS_DEGREE); waddch(window, ACS_DEGREE);
waddstr(window, "C"); waddstr(window, "C");
wattroff(window, COLOR_PAIR(1 + static_cast<int>(temperatureToHealth(std::get<1>(smartList[i].temperature))))); wattroff(window, COLOR_PAIR(HEALTH_INV_COLOR + static_cast<int>(temperatureToHealth(std::get<1>(smartList[i].temperature)))));
} }
else else
{ {
@ -239,13 +244,13 @@ void drawDeviceBar(WINDOW * window, std::vector<SMART> const & smartList, int se
if (i == select) if (i == select)
{ {
wattrset(window, COLOR_PAIR(4) | A_BOLD); wattrset(window, COLOR_PAIR(HEALTH_COLOR) | A_BOLD);
mvwprintw(window, 2, x, smartList[i].deviceName.c_str()); mvwprintw(window, 2, x, smartList[i].deviceName.c_str());
wattroff(window, COLOR_PAIR(4) | A_BOLD); wattroff(window, COLOR_PAIR(HEALTH_COLOR) | A_BOLD);
wattrset(window, COLOR_PAIR(4)); wattrset(window, COLOR_PAIR(HEALTH_COLOR));
mvwhline(window, 3, x, '-', smartList[i].deviceName.length()); mvwhline(window, 3, x, '-', smartList[i].deviceName.length());
wattroff(window, COLOR_PAIR(4)); wattroff(window, COLOR_PAIR(HEALTH_COLOR));
} }
else else
{ {
@ -279,30 +284,30 @@ void drawStatus(WINDOW * window, SMART const & smart, Option const & option)
} }
char s[STATUS_WIDTH]; char s[STATUS_WIDTH];
int len = snprintf(s, STATUS_WIDTH, " %s [%.1f %s] ", smart.model.c_str(), size, unit[u].c_str()); int len = snprintf(s, STATUS_WIDTH, " %s [%.1f %s] ", smart.model.c_str(), size, unit[u].c_str());
wattrset(window, COLOR_PAIR(4) | A_BOLD); wattrset(window, COLOR_PAIR(HEALTH_COLOR) | A_BOLD);
mvwprintw(window, 0, (STATUS_WIDTH - len) / 2, "%s", s); mvwprintw(window, 0, (STATUS_WIDTH - len) / 2, "%s", s);
wattroff(window, COLOR_PAIR(4) | A_BOLD); wattroff(window, COLOR_PAIR(HEALTH_COLOR) | A_BOLD);
} }
else else
{ {
char s[STATUS_WIDTH]; char s[STATUS_WIDTH];
int len = snprintf(s, STATUS_WIDTH, " %s [--] ", smart.model.c_str()); int len = snprintf(s, STATUS_WIDTH, " %s [--] ", smart.model.c_str());
wattrset(window, COLOR_PAIR(4) | A_BOLD); wattrset(window, COLOR_PAIR(HEALTH_COLOR) | A_BOLD);
mvwprintw(window, 0, (STATUS_WIDTH - len) / 2, "%s", s); mvwprintw(window, 0, (STATUS_WIDTH - len) / 2, "%s", s);
wattroff(window, COLOR_PAIR(4) | A_BOLD); wattroff(window, COLOR_PAIR(HEALTH_COLOR) | A_BOLD);
} }
wattrset(window, COLOR_PAIR(4)); wattrset(window, COLOR_PAIR(HEALTH_COLOR));
mvwprintw(window, 2, (int)(STATUS_WIDTH * (1.0 / 5)), "Firmware:"); mvwprintw(window, 2, (int)(STATUS_WIDTH * (1.0 / 5)), "Firmware:");
wattroff(window, COLOR_PAIR(4)); wattroff(window, COLOR_PAIR(HEALTH_COLOR));
wattrset(window, COLOR_PAIR(4) | A_BOLD); wattrset(window, COLOR_PAIR(HEALTH_COLOR) | A_BOLD);
wprintw(window, " %s", smart.firmware.c_str()); wprintw(window, " %s", smart.firmware.c_str());
wattroff(window, COLOR_PAIR(4) | A_BOLD); wattroff(window, COLOR_PAIR(HEALTH_COLOR) | A_BOLD);
wattrset(window, COLOR_PAIR(4)); wattrset(window, COLOR_PAIR(HEALTH_COLOR));
mvwprintw(window, 3, (int)(STATUS_WIDTH * (1.0 / 5)), "Serial: "); mvwprintw(window, 3, (int)(STATUS_WIDTH * (1.0 / 5)), "Serial: ");
wattroff(window, COLOR_PAIR(4)); wattroff(window, COLOR_PAIR(HEALTH_COLOR));
wattrset(window, COLOR_PAIR(4) | A_BOLD); wattrset(window, COLOR_PAIR(HEALTH_COLOR) | A_BOLD);
if (option.hideSerial) if (option.hideSerial)
{ {
wprintw(window, " ********************"); wprintw(window, " ********************");
@ -311,28 +316,28 @@ void drawStatus(WINDOW * window, SMART const & smart, Option const & option)
{ {
wprintw(window, " %s", smart.serial.c_str()); wprintw(window, " %s", smart.serial.c_str());
} }
wattroff(window, COLOR_PAIR(4) | A_BOLD); wattroff(window, COLOR_PAIR(HEALTH_COLOR) | A_BOLD);
wattrset(window, COLOR_PAIR(4)); wattrset(window, COLOR_PAIR(HEALTH_COLOR));
mvwprintw(window, 1, 1, "Status"); mvwprintw(window, 1, 1, "Status");
wattroff(window, COLOR_PAIR(4)); wattroff(window, COLOR_PAIR(HEALTH_COLOR));
wattrset(window, COLOR_PAIR(1 + static_cast<int>(smartToHealth(smart)))); wattrset(window, COLOR_PAIR(HEALTH_INV_COLOR + static_cast<int>(smartToHealth(smart))));
mvwprintw(window, 2, 2, "+--------+"); mvwprintw(window, 2, 2, "+--------+");
mvwprintw(window, 3, 2, "| |"); mvwprintw(window, 3, 2, "| |");
mvwprintw(window, 4, 2, "+--------+"); mvwprintw(window, 4, 2, "+--------+");
mvwprintw(window, 3, 2 + ((sizeof("| |") - healthToString(smartToHealth(smart)).length()) / 2), "%s", healthToString(smartToHealth(smart)).c_str()); mvwprintw(window, 3, 2 + ((sizeof("| |") - healthToString(smartToHealth(smart)).length()) / 2), "%s", healthToString(smartToHealth(smart)).c_str());
wattroff(window, COLOR_PAIR(1 + static_cast<int>(smartToHealth(smart)))); wattroff(window, COLOR_PAIR(HEALTH_INV_COLOR + static_cast<int>(smartToHealth(smart))));
if (std::get<0>(smart.temperature)) if (std::get<0>(smart.temperature))
{ {
wattrset(window, COLOR_PAIR(4)); wattrset(window, COLOR_PAIR(HEALTH_COLOR));
mvwprintw(window, 5, 1, "Temperature"); mvwprintw(window, 5, 1, "Temperature");
wattroff(window, COLOR_PAIR(4)); wattroff(window, COLOR_PAIR(HEALTH_COLOR));
wattrset(window, COLOR_PAIR(1 + static_cast<int>(temperatureToHealth(std::get<1>(smart.temperature))))); wattrset(window, COLOR_PAIR(HEALTH_INV_COLOR + static_cast<int>(temperatureToHealth(std::get<1>(smart.temperature)))));
mvwprintw(window, 6, 2, " %0.1f ", std::get<1>(smart.temperature)); mvwprintw(window, 6, 2, " %0.1f ", std::get<1>(smart.temperature));
waddch(window, ACS_DEGREE); waddch(window, ACS_DEGREE);
waddstr(window, "C "); waddstr(window, "C ");
wattroff(window, COLOR_PAIR(1 + static_cast<int>(temperatureToHealth(std::get<1>(smart.temperature))))); wattroff(window, COLOR_PAIR(HEALTH_INV_COLOR + static_cast<int>(temperatureToHealth(std::get<1>(smart.temperature)))));
} }
else else
{ {
@ -344,13 +349,13 @@ void drawStatus(WINDOW * window, SMART const & smart, Option const & option)
if (std::get<0>(smart.powerOnCount)) if (std::get<0>(smart.powerOnCount))
{ {
wattrset(window, COLOR_PAIR(4)); wattrset(window, COLOR_PAIR(HEALTH_COLOR));
mvwprintw(window, 2, (int)(STATUS_WIDTH * (3.0 / 5)), "Power On Count:"); mvwprintw(window, 2, (int)(STATUS_WIDTH * (3.0 / 5)), "Power On Count:");
wattroff(window, COLOR_PAIR(4)); wattroff(window, COLOR_PAIR(HEALTH_COLOR));
wattrset(window, COLOR_PAIR(4) | A_BOLD); wattrset(window, COLOR_PAIR(HEALTH_COLOR) | A_BOLD);
wprintw(window, " %llu ", std::get<1>(smart.powerOnCount)); wprintw(window, " %llu ", std::get<1>(smart.powerOnCount));
wattroff(window, COLOR_PAIR(4) | A_BOLD); wattroff(window, COLOR_PAIR(HEALTH_COLOR) | A_BOLD);
wattrset(window, COLOR_PAIR(4)); wattrset(window, COLOR_PAIR(HEALTH_COLOR));
wprintw(window, "count"); wprintw(window, "count");
} }
else else
@ -361,15 +366,15 @@ void drawStatus(WINDOW * window, SMART const & smart, Option const & option)
if (std::get<0>(smart.powerOnHour)) if (std::get<0>(smart.powerOnHour))
{ {
wattrset(window, COLOR_PAIR(4)); wattrset(window, COLOR_PAIR(HEALTH_COLOR));
mvwprintw(window, 3, (int)(STATUS_WIDTH * (3.0 / 5)), "Power On Hours:"); mvwprintw(window, 3, (int)(STATUS_WIDTH * (3.0 / 5)), "Power On Hours:");
wattroff(window, COLOR_PAIR(4)); wattroff(window, COLOR_PAIR(HEALTH_COLOR));
wattrset(window, COLOR_PAIR(4) | A_BOLD); wattrset(window, COLOR_PAIR(HEALTH_COLOR) | A_BOLD);
wprintw(window, " %llu ", std::get<1>(smart.powerOnHour)); wprintw(window, " %llu ", std::get<1>(smart.powerOnHour));
wattroff(window, COLOR_PAIR(4) | A_BOLD); wattroff(window, COLOR_PAIR(HEALTH_COLOR) | A_BOLD);
wattrset(window, COLOR_PAIR(4)); wattrset(window, COLOR_PAIR(HEALTH_COLOR));
wprintw(window, "hours"); wprintw(window, "hours");
wattroff(window, COLOR_PAIR(4)); wattroff(window, COLOR_PAIR(HEALTH_COLOR));
} }
else else
{ {
@ -377,12 +382,12 @@ void drawStatus(WINDOW * window, SMART const & smart, Option const & option)
wprintw(window, " -- hours"); wprintw(window, " -- hours");
} }
wattrset(window, COLOR_PAIR(7)); wattrset(window, COLOR_PAIR(ATTRIBUTE_LEGEND_COLOR));
mvwprintw(window, 8, 1, " Status ID AttributeName Current Worst Threshold Raw Values "); mvwprintw(window, 8, 1, " Status ID AttributeName Current Worst Threshold Raw Values ");
wattroff(window, COLOR_PAIR(7)); wattroff(window, COLOR_PAIR(ATTRIBUTE_LEGEND_COLOR));
for (int i = 0; i < static_cast<int>(smart.attribute.size()); ++i) for (int i = 0; i < static_cast<int>(smart.attribute.size()); ++i)
{ {
wattrset(window, COLOR_PAIR(4 + static_cast<int>(attributeToHealth(smart.attribute[i])))); wattrset(window, COLOR_PAIR(HEALTH_COLOR + static_cast<int>(attributeToHealth(smart.attribute[i]))));
#ifndef RAWDEC #ifndef RAWDEC
mvwprintw(window, 9 + i, 1, " %-7s %02X %-28s %7d %5d %9d %012X ", mvwprintw(window, 9 + i, 1, " %-7s %02X %-28s %7d %5d %9d %012X ",
#else #else
@ -395,7 +400,7 @@ void drawStatus(WINDOW * window, SMART const & smart, Option const & option)
smart.attribute[i].worst, smart.attribute[i].worst,
smart.attribute[i].threshold, smart.attribute[i].threshold,
smart.attribute[i].raw); smart.attribute[i].raw);
wattroff(window, COLOR_PAIR(4 + static_cast<int>(attributeToHealth(smart.attribute[i])))); wattroff(window, COLOR_PAIR(HEALTH_COLOR + static_cast<int>(attributeToHealth(smart.attribute[i]))));
} }
pnoutrefresh(window, 0, 0, pnoutrefresh(window, 0, 0,
5, std::max(0, (width - STATUS_WIDTH) / 2), 5, std::max(0, (width - STATUS_WIDTH) / 2),
@ -421,14 +426,14 @@ int main()
getmaxyx(stdscr, height, width); getmaxyx(stdscr, height, width);
start_color(); start_color();
init_pair(1, COLOR_BLACK, COLOR_CYAN); init_pair(HEALTH_COLOR, COLOR_CYAN, COLOR_BLACK);
init_pair(2, COLOR_BLACK, COLOR_YELLOW); init_pair(HEALTH_COLOR + 1, COLOR_BLACK, COLOR_YELLOW);
init_pair(3, COLOR_WHITE, COLOR_RED); init_pair(HEALTH_COLOR + 2, COLOR_WHITE, COLOR_RED);
init_pair(4, COLOR_CYAN, COLOR_BLACK); init_pair(HEALTH_INV_COLOR, COLOR_BLACK, COLOR_CYAN);
init_pair(5, COLOR_BLACK, COLOR_YELLOW); init_pair(HEALTH_INV_COLOR + 1, COLOR_BLACK, COLOR_YELLOW);
init_pair(6, COLOR_WHITE, COLOR_RED); init_pair(HEALTH_INV_COLOR + 2, COLOR_WHITE, COLOR_RED);
init_pair(7, COLOR_BLACK, COLOR_GREEN); init_pair(ATTRIBUTE_LEGEND_COLOR, COLOR_BLACK, COLOR_GREEN);
init_pair(8, COLOR_YELLOW, COLOR_BLACK); init_pair(TITLE_COLOR, COLOR_YELLOW, COLOR_BLACK);
std::vector<SMART> smartList; std::vector<SMART> smartList;
auto dir = opendir("/sys/block"); auto dir = opendir("/sys/block");