bargraph.c 5.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345
  1. #include <u.h>
  2. #include <libc.h>
  3. #include <draw.h>
  4. #include <bio.h>
  5. #include <event.h>
  6. enum {PNCTL=3};
  7. static char* rdenv(char*);
  8. int newwin(char*);
  9. Rectangle screenrect(void);
  10. int nokill;
  11. int textmode;
  12. char *title;
  13. Image *light;
  14. Image *dark;
  15. Image *text;
  16. void
  17. initcolor(void)
  18. {
  19. text = display->black;
  20. light = allocimagemix(display, DPalegreen, DWhite);
  21. dark = allocimage(display, Rect(0,0,1,1), CMAP8, 1, DDarkgreen);
  22. }
  23. Rectangle rbar;
  24. Point ptext;
  25. vlong n, d;
  26. int last;
  27. int lastp;
  28. int first = 1;
  29. char backup[80];
  30. void
  31. drawbar(void)
  32. {
  33. int i, j;
  34. int p;
  35. char buf[10], bar[100];
  36. if(n > d || n < 0 || d <= 0)
  37. return;
  38. i = (Dx(rbar)*n)/d;
  39. p = (n*100LL)/d;
  40. if(lastp == p && last == i)
  41. return;
  42. if(textmode){
  43. bar[0] = '|';
  44. for(j=0; j<i; j++)
  45. bar[j+1] = '#';
  46. for(; j<60; j++)
  47. bar[j+1] = '-';
  48. bar[61] = '|';
  49. bar[62] = ' ';
  50. sprint(bar+63, "%3d%%", p);
  51. if(first)
  52. first = 0;
  53. else{
  54. for(i=0; i<strlen(bar); i++)
  55. backup[i] = '\b';
  56. write(1, backup, i);
  57. }
  58. write(1, bar, strlen(bar));
  59. lastp = p;
  60. last = i;
  61. return;
  62. }
  63. if(lastp != p){
  64. sprint(buf, "%d%%", p);
  65. stringbg(screen, addpt(screen->r.min, Pt(Dx(rbar)-30, 4)), text, ZP, display->defaultfont, buf, light, ZP);
  66. lastp = p;
  67. }
  68. if(last != i){
  69. draw(screen, Rect(rbar.min.x+last, rbar.min.y, rbar.min.x+i, rbar.max.y),
  70. dark, nil, ZP);
  71. last = i;
  72. }
  73. flushimage(display, 1);
  74. }
  75. void
  76. eresized(int new)
  77. {
  78. Point p, q;
  79. Rectangle r;
  80. if(new && getwindow(display, Refnone) < 0)
  81. fprint(2,"can't reattach to window");
  82. r = screen->r;
  83. draw(screen, r, light, nil, ZP);
  84. p = string(screen, addpt(r.min, Pt(4,4)), text, ZP,
  85. display->defaultfont, title);
  86. p.x = r.min.x+4;
  87. p.y += display->defaultfont->height+4;
  88. q = subpt(r.max, Pt(4,4));
  89. rbar = Rpt(p, q);
  90. ptext = Pt(r.max.x-4-stringwidth(display->defaultfont, "100%"), r.min.x+4);
  91. border(screen, rbar, -2, dark, ZP);
  92. last = 0;
  93. lastp = -1;
  94. drawbar();
  95. }
  96. void
  97. bar(Biobuf *b)
  98. {
  99. char *p, *f[2];
  100. Event e;
  101. int k, die, parent, child;
  102. parent = getpid();
  103. die = 0;
  104. if(!textmode) switch(child = rfork(RFMEM|RFPROC)) {
  105. case 0:
  106. sleep(1000);
  107. while(!die && (k = eread(Ekeyboard|Emouse, &e))) {
  108. if(nokill==0 && k == Ekeyboard && (e.kbdc == 0x7F || e.kbdc == 0x03)) { /* del, ctl-c */
  109. die = 1;
  110. postnote(PNPROC, parent, "interrupt");
  111. _exits("interrupt");
  112. }
  113. }
  114. _exits(0);
  115. }
  116. while(!die && (p = Brdline(b, '\n'))) {
  117. p[Blinelen(b)-1] = '\0';
  118. if(tokenize(p, f, 2) != 2)
  119. continue;
  120. n = strtoll(f[0], 0, 0);
  121. d = strtoll(f[1], 0, 0);
  122. drawbar();
  123. }
  124. postnote(PNCTL, child, "kill");
  125. die = 1;
  126. }
  127. void
  128. usage(void)
  129. {
  130. fprint(2, "usage: bargraph [-kt] [-w minx,miny,maxx,maxy] 'title'\n");
  131. exits("usage");
  132. }
  133. void
  134. main(int argc, char **argv)
  135. {
  136. Biobuf b;
  137. char *p, *q;
  138. int lfd;
  139. p = "0,0,200,60";
  140. ARGBEGIN{
  141. case 'w':
  142. p = ARGF();
  143. break;
  144. case 't':
  145. textmode = 1;
  146. break;
  147. case 'k':
  148. nokill = 1;
  149. break;
  150. default:
  151. usage();
  152. }ARGEND;
  153. if(argc != 1)
  154. usage();
  155. title = argv[0];
  156. lfd = dup(0, -1);
  157. while(q = strchr(p, ','))
  158. *q = ' ';
  159. Binit(&b, lfd, OREAD);
  160. if(textmode || newwin(p) < 0){
  161. textmode = 1;
  162. rbar = Rect(0, 0, 60, 1);
  163. }else{
  164. initdraw(0, 0, "bar");
  165. initcolor();
  166. einit(Emouse|Ekeyboard);
  167. eresized(0);
  168. }
  169. bar(&b);
  170. }
  171. /* all code below this line should be in the library, but is stolen from colors instead */
  172. static char*
  173. rdenv(char *name)
  174. {
  175. char *v;
  176. int fd, size;
  177. fd = open(name, OREAD);
  178. if(fd < 0)
  179. return 0;
  180. size = seek(fd, 0, 2);
  181. v = malloc(size+1);
  182. if(v == 0){
  183. fprint(2, "page: can't malloc: %r\n");
  184. exits("no mem");
  185. }
  186. seek(fd, 0, 0);
  187. read(fd, v, size);
  188. v[size] = 0;
  189. close(fd);
  190. return v;
  191. }
  192. int
  193. newwin(char *win)
  194. {
  195. char *srv, *mntsrv;
  196. char spec[100];
  197. int srvfd, cons, pid;
  198. switch(rfork(RFFDG|RFPROC|RFNAMEG|RFENVG|RFNOTEG|RFNOWAIT)){
  199. case -1:
  200. fprint(2, "bargraph: can't fork: %r\n");
  201. return -1;
  202. case 0:
  203. break;
  204. default:
  205. exits(0);
  206. }
  207. srv = rdenv("/env/wsys");
  208. if(srv == 0){
  209. mntsrv = rdenv("/mnt/term/env/wsys");
  210. if(mntsrv == 0){
  211. fprint(2, "bargraph: can't find $wsys\n");
  212. return -1;
  213. }
  214. srv = malloc(strlen(mntsrv)+10);
  215. sprint(srv, "/mnt/term%s", mntsrv);
  216. free(mntsrv);
  217. pid = 0; /* can't send notes to remote processes! */
  218. }else
  219. pid = getpid();
  220. srvfd = open(srv, ORDWR);
  221. free(srv);
  222. if(srvfd == -1){
  223. fprint(2, "bargraph: can't open %s: %r\n", srv);
  224. return -1;
  225. }
  226. sprint(spec, "new -r %s", win);
  227. if(mount(srvfd, -1, "/mnt/wsys", 0, spec) == -1){
  228. fprint(2, "bargraph: can't mount /mnt/wsys: %r (spec=%s)\n", spec);
  229. return -1;
  230. }
  231. close(srvfd);
  232. unmount("/mnt/acme", "/dev");
  233. bind("/mnt/wsys", "/dev", MBEFORE);
  234. cons = open("/dev/cons", OREAD);
  235. if(cons==-1){
  236. NoCons:
  237. fprint(2, "bargraph: can't open /dev/cons: %r");
  238. return -1;
  239. }
  240. dup(cons, 0);
  241. close(cons);
  242. cons = open("/dev/cons", OWRITE);
  243. if(cons==-1)
  244. goto NoCons;
  245. dup(cons, 1);
  246. dup(cons, 2);
  247. close(cons);
  248. // wctlfd = open("/dev/wctl", OWRITE);
  249. return 0;
  250. }
  251. Rectangle
  252. screenrect(void)
  253. {
  254. int fd;
  255. char buf[12*5];
  256. fd = open("/dev/screen", OREAD);
  257. if(fd == -1)
  258. fd=open("/mnt/term/dev/screen", OREAD);
  259. if(fd == -1){
  260. fprint(2, "page: can't open /dev/screen: %r\n");
  261. exits("window read");
  262. }
  263. if(read(fd, buf, sizeof buf) != sizeof buf){
  264. fprint(2, "page: can't read /dev/screen: %r\n");
  265. exits("screen read");
  266. }
  267. close(fd);
  268. return Rect(atoi(buf+12), atoi(buf+24), atoi(buf+36), atoi(buf+48));
  269. }
  270. int
  271. postnote(int group, int pid, char *note)
  272. {
  273. char file[128];
  274. int f, r;
  275. switch(group) {
  276. case PNPROC:
  277. sprint(file, "/proc/%d/note", pid);
  278. break;
  279. case PNGROUP:
  280. sprint(file, "/proc/%d/notepg", pid);
  281. break;
  282. case PNCTL:
  283. sprint(file, "/proc/%d/ctl", pid);
  284. break;
  285. default:
  286. return -1;
  287. }
  288. f = open(file, OWRITE);
  289. if(f < 0)
  290. return -1;
  291. r = strlen(note);
  292. if(write(f, note, r) != r) {
  293. close(f);
  294. return -1;
  295. }
  296. close(f);
  297. return 0;
  298. }