ref: 20bca0c2029c53de3d4149d957240b17d44684a8
dir: /sys/src/cmd/ktrans/main.c/
#include <u.h> #include <libc.h> #include <ctype.h> #include <bio.h> #include <plumb.h> #include <thread.h> #include <draw.h> #include <mouse.h> #include <keyboard.h> #include "hash.h" char* pushutf(char *dst, char *e, char *u, int nrune) { Rune r; char *p; char *d; if(dst >= e) return dst; d = dst; p = u; while(d < e-1){ if(isascii(*p)){ if((*d = *p) == '\0') return d; p++; d++; } else { p += chartorune(&r, p); if(r == Runeerror){ *d = '\0'; return d; } d += runetochar(d, &r); } if(nrune > 0 && --nrune == 0) break; } if(d > e-1) d = e-1; *d = '\0'; return d; } char* peekstr(char *s, char *b) { while(s > b && (*--s & 0xC0)==Runesync) ; return s; } typedef struct Str Str; struct Str { char b[128]; char *p; }; #define strend(s) ((s)->b + sizeof (s)->b) void resetstr(Str *s, ...) { va_list args; va_start(args, s); do { s->p = s->b; s->p[0] = '\0'; s = va_arg(args, Str*); } while(s != nil); va_end(args); } void popstr(Str *s) { s->p = peekstr(s->p, s->b); s->p[0] = '\0'; } typedef struct Map Map; struct Map { char *roma; char *kana; char leadstomore; }; Hmap* openmap(char *file) { Biobuf *b; char *s; Map map; Hmap *h; char *key, *val; Str partial; Rune r; h = hmapalloc(64, sizeof(Map)); b = Bopen(file, OREAD); if(b == nil) return nil; while(key = Brdstr(b, '\n', 1)){ if(key[0] == '\0'){ Err: free(key); continue; } val = strchr(key, '\t'); if(val == nil || val[1] == '\0') goto Err; *val = '\0'; val++; resetstr(&partial, nil); for(s = key; *s; s += chartorune(&r, s)){ partial.p = pushutf(partial.p, strend(&partial), s, 1); map.leadstomore = 0; if(hmapget(h, partial.b, &map) == 0){ if(map.leadstomore == 1 && s[1] == '\0') map.leadstomore = 1; } if(s[1] == '\0'){ map.roma = key; map.kana = val; hmaprepl(&h, strdup(map.roma), &map, nil, 1); } else { map.roma = strdup(partial.b); map.leadstomore = 1; map.kana = nil; hmaprepl(&h, strdup(partial.b), &map, nil, 1); } } } Bterm(b); return h; } enum{ Maxkouho=32, }; Hmap* opendict(Hmap *h, char *name) { Biobuf *b; char *p; char *dot, *rest; char *kouho[Maxkouho]; int i; b = Bopen(name, OREAD); if(b == nil) return nil; if(h == nil) h = hmapalloc(8192, sizeof(kouho)); else hmapreset(h, 1); while(p = Brdstr(b, '\n', 1)){ if(p[0] == '\0' || p[0] == ';'){ Err: free(p); continue; } dot = strchr(p, '\t'); if(dot == nil) goto Err; *dot = '\0'; rest = dot+1; if(*rest == '\0') goto Err; memset(kouho, 0, sizeof kouho); i = 0; while(i < nelem(kouho)-1 && (dot = utfrune(rest, ' '))){ *dot = '\0'; kouho[i++] = rest; rest = dot+1; } if(i < nelem(kouho)-1) kouho[i] = rest; /* key is the base pointer; overwrites clean up for us */ hmaprepl(&h, p, kouho, nil, 1); } Bterm(b); return h; } enum{ LangEN = '', // ^t LangJP = '', // ^n LangJPK = '', // ^k LangRU = '', // ^r LangEL = '', // ^o LangKO = '', // ^s LangZH = '', // ^c LangVN = '', // ^v }; int deflang; Hmap *natural; Hmap *hira, *kata, *jisho; Hmap *cyril; Hmap *greek; Hmap *hangul; Hmap *hanzi, *zidian; Hmap *telex; Hmap **langtab[] = { [LangEN] &natural, [LangJP] &hira, [LangJPK] &kata, [LangRU] &cyril, [LangEL] &greek, [LangKO] &hangul, [LangZH] &hanzi, [LangVN] &telex, }; char *langcodetab[] = { [LangEN] "en", [LangJP] "jp", [LangJPK] "jpk", [LangRU] "ru", [LangEL] "el", [LangKO] "ko", [LangZH] "zh", [LangVN] "vn", }; int parselang(char *s) { int i; for(i = 0; i < nelem(langcodetab); i++){ if(langcodetab[i] == nil) continue; if(strcmp(langcodetab[i], s) == 0) return i; } return -1; } int checklang(int *dst, int c) { Hmap **p; if(c >= nelem(langtab)) return 0; p = langtab[c]; if(p == nil) return 0; *dst = c; return c; } int maplkup(int lang, char *s, Map *m) { Hmap **h; if(lang >= nelem(langtab)) return -1; h = langtab[lang]; if(h == nil || *h == nil) return -1; return hmapget(*h, s, m); } enum { Msgsize = 64 }; static Channel *dictch; static Channel *output; static Channel *input; static char backspace[Msgsize]; static Channel *displaych; static Channel *selectch; static void displaythread(void*) { Mousectl *mctl; Mouse m; Keyboardctl *kctl; Rune key; char *kouho[Maxkouho+1], **s; Image *back, *text, *board, *high; Font *f; Point p; Rectangle r, exitr, selr; int selected; enum { Adisp, Aresize, Amouse, Asel, Akbd, Aend }; Alt a[] = { [Adisp] { nil, kouho+1, CHANRCV }, [Aresize] { nil, nil, CHANRCV }, [Amouse] { nil, &m, CHANRCV }, [Asel] { nil, &selected, CHANRCV }, [Akbd] { nil, &key, CHANRCV }, [Aend] { nil, nil, CHANEND }, }; if(initdraw(nil, nil, "ktrans") < 0) sysfatal("failed to initdraw: %r"); mctl = initmouse(nil, screen); if(mctl == nil) sysfatal("failed to get mouse: %r"); /* * For keys coming in to our specific window. * We've already transliterated these, but should * consume keys and exit on del to avoid artifacts. */ kctl = initkeyboard(nil); if(kctl == nil) sysfatal("failed to get keyboard: %r"); memset(kouho, 0, sizeof kouho); kouho[0] = "候補"; selected = -1; f = display->defaultfont; high = allocimagemix(display, DYellowgreen, DWhite); text = display->black; back = allocimagemix(display, DPaleyellow, DWhite); board = allocimagemix(display, DBlack, DWhite); a[Adisp].c = displaych; a[Aresize].c = mctl->resizec; a[Amouse].c = mctl->c; a[Asel].c = selectch; a[Akbd].c = kctl->c; threadsetname("display"); goto Redraw; for(;;) switch(alt(a)){ case Akbd: if(key != Kdel) break; closedisplay(display); threadexitsall(nil); case Amouse: if(!m.buttons) break; if(!ptinrect(m.xy, exitr)) break; closedisplay(display); threadexitsall(nil); case Aresize: getwindow(display, Refnone); case Adisp: Redraw: r = screen->r; draw(screen, r, back, nil, ZP); r.max.y = r.min.y + f->height; draw(screen, r, board, nil, ZP); if(selected+1 > 0 && kouho[selected+1] != nil){ selr = screen->r; selr.min.y += f->height*(selected+1); selr.max.y = selr.min.y + f->height; draw(screen, selr, high, nil, ZP); } r.min.x += Dx(r)/2; p.y = r.min.y; for(s = kouho; *s != nil; s++){ p.x = r.min.x - stringwidth(f, *s)/2; string(screen, p, text, ZP, f, *s); p.y += f->height; } p.x = r.min.x - stringwidth(f, "終了")/2; p.y = screen->r.max.y - f->height; exitr = Rpt(Pt(0, p.y), screen->r.max); draw(screen, exitr, board, nil, ZP); string(screen, p, text, ZP, f, "終了"); flushimage(display, 1); break; } } static int emitutf(Channel *out, char *u, int nrune) { char b[Msgsize]; char *e; b[0] = 'c'; e = pushutf(b+1, b + Msgsize - 1, u, nrune); send(out, b); return e - b; } static int compacting = 0; static void dictthread(void*) { char m[Msgsize]; Rune r; int n; char *p; Hmap *dict; char *kouho[Maxkouho]; Str line; Str last; Str okuri; int selected; enum{ Kanji, Okuri, Joshi, }; int mode; dict = jisho; selected = -1; mode = Kanji; memset(kouho, 0, sizeof kouho); resetstr(&last, &line, &okuri, nil); threadsetname("dict"); while(recv(dictch, m) != -1){ compacting = 1; for(p = m+1; *p; p += n){ n = chartorune(&r, p); switch(r){ case LangJP: dict = jisho; break; case LangZH: dict = zidian; break; case '': if(line.b == line.p){ emitutf(output, "", 1); break; } emitutf(output, backspace, utflen(line.b)); /* fallthrough */ case '': case ' ': case '\n': mode = Kanji; resetstr(&line, &okuri, &last, nil); break; case '\b': if(mode != Kanji){ if(okuri.p == okuri.b){ mode = Kanji; popstr(&line); }else popstr(&okuri); break; } popstr(&line); break; case '': selected++; if(selected == 0){ if(hmapget(dict, line.b, kouho) < 0) break; if(dict == jisho && line.p > line.b && isascii(line.p[-1])) line.p[-1] = '\0'; } if(kouho[selected] == nil){ /* cycled through all matches; bail */ if(utflen(okuri.b) != 0) emitutf(output, backspace, utflen(okuri.b)); emitutf(output, backspace, utflen(last.b)); emitutf(output, line.b, 0); emitutf(output, okuri.b, 0); break; } send(selectch, &selected); send(displaych, kouho); if(okuri.p != okuri.b) emitutf(output, backspace, utflen(okuri.b)); if(selected == 0) emitutf(output, backspace, utflen(line.b)); else emitutf(output, backspace, utflen(last.b)); emitutf(output, kouho[selected], 0); last.p = pushutf(last.b, strend(&last), kouho[selected], 0); emitutf(output, okuri.b, 0); mode = Kanji; continue; case ',': case '.': case L'。': case L'、': if(dict == zidian || line.p == line.b){ selected = 0; //hit cleanup below break; } mode = Joshi; okuri.p = pushutf(okuri.p, strend(&okuri), p, 1); break; default: if(dict == zidian) goto Line; if(mode == Joshi) goto Okuri; if(isupper(*p)){ if(mode == Okuri){ popstr(&line); mode = Joshi; goto Okuri; } mode = Okuri; *p = tolower(*p); okuri.p = pushutf(okuri.b, strend(&okuri), p, 1); goto Line; } switch(mode){ case Kanji: Line: line.p = pushutf(line.p, strend(&line), p, 1); break; default: Okuri: okuri.p = pushutf(okuri.p, strend(&okuri), p, 1); break; } } if(selected >= 0){ resetstr(&okuri, &last, &line, nil); selected = -1; send(selectch, &selected); } memset(kouho, 0, sizeof kouho); hmapget(dict, line.b, kouho); send(displaych, kouho); } compacting = 0; } } static void telexlkup(Str *line) { Map lkup; char buf[UTFmax*3], *p, *e; Str out; int n, ln; Again: ln = utflen(line->b); p = pushutf(buf, buf+sizeof buf, line->b, 1); n = p-buf; if(hmapget(telex, buf, &lkup) < 0){ resetstr(line, nil); return; } if(utflen(line->b) < 2) return; e = peekstr(line->p, line->b); pushutf(p, buf+sizeof buf, e, 1); if(hmapget(telex, buf, &lkup) < 0){ /* not correct; matches should be allowed to span vowels */ if(hmapget(telex, buf+n, &lkup) == 0) line->p = pushutf(line->b, strend(line), buf+n, 0); return; } out.p = pushutf(out.b, strend(&out), lkup.kana, 0); out.p = pushutf(out.p, strend(&out), line->b+n, 0); popstr(&out); if(ln > 0) emitutf(output, backspace, ln); emitutf(output, out.b, 0); line->p = pushutf(line->b, strend(line), out.b, 0); if(utflen(lkup.kana) == 2) return; goto Again; } static void keythread(void*) { int lang; char m[Msgsize]; Map lkup; char *p; int n; Rune r; char peek[UTFmax+1]; Str line; peek[0] = lang = deflang; resetstr(&line, nil); if(lang == LangJP || lang == LangZH) emitutf(dictch, peek, 1); threadsetname("keytrans"); while(recv(input, m) != -1){ if(m[0] == 'z'){ emitutf(dictch, "", 1); resetstr(&line, nil); continue; } if(m[0] != 'c'){ send(output, m); continue; } for(p = m+1; *p; p += n){ while(compacting) yield(); n = chartorune(&r, p); if(checklang(&lang, r)){ emitutf(dictch, "", 1); if(lang == LangJP || lang == LangZH) emitutf(dictch, p, 1); resetstr(&line, nil); continue; } if(lang == LangEN){ emitutf(output, p, 1); continue; } if(utfrune("", r) != nil){ resetstr(&line, nil); emitutf(dictch, p, 1); continue; } emitutf(output, p, 1); switch(lang){ case LangZH: emitutf(dictch, p, 1); continue; case LangJP: emitutf(dictch, p, 1); if(isupper(*p)) *p = tolower(*p); break; } if(utfrune("\n\t ", r) != nil){ resetstr(&line, nil); continue; } else if(r == '\b'){ popstr(&line); continue; } line.p = pushutf(line.p, strend(&line), p, 1); if(lang == LangVN){ telexlkup(&line); continue; } if(maplkup(lang, line.b, &lkup) < 0){ resetstr(&line, nil); pushutf(peek, peek + sizeof peek, p, 1); if(maplkup(lang, peek, &lkup) == 0) line.p = pushutf(line.p, strend(&line), p, 1); continue; } if(lkup.kana == nil) continue; if(!lkup.leadstomore) resetstr(&line, nil); if(lang == LangJP){ emitutf(dictch, backspace, utflen(lkup.roma)); emitutf(dictch, lkup.kana, 0); } emitutf(output, backspace, utflen(lkup.roma)); emitutf(output, lkup.kana, 0); } } } static int kbdin; static int kbdout; static void kbdtap(void*) { char m[Msgsize]; char buf[128]; char *p; int n; threadsetname("kbdtap"); for(;;){ Drop: n = read(kbdin, buf, sizeof buf); if(n < 0) break; for(p = buf; p < buf+n; p += strlen(p) + 1){ switch(*p){ case 'c': case 'k': case 'K': case 'z': break; default: goto Drop; } strcpy(m, p); if(send(input, m) == -1) return; } } threadexitsall(nil); } static void kbdsink(void*) { char in[Msgsize]; char out[Msgsize]; char *p; int n; Rune rn; out[0] = 'c'; threadsetname("kbdsink"); while(recv(output, in) != -1){ if(in[0] != 'c'){ if(write(kbdout, in, strlen(in)+1) < 0) break; continue; } for(p = in+1; *p; p += n){ n = chartorune(&rn, p); if(rn == Runeerror || rn == '\0') break; memmove(out+1, p, n); out[1+n] = '\0'; if(write(kbdout, out, 1+n+1) < 0) break; } } } static int plumbfd; static void plumbproc(void*) { char m[Msgsize]; Plumbmsg *p; threadsetname("plumbproc"); for(; p = plumbrecv(plumbfd); plumbfree(p)){ if(p->ndata > sizeof m - 1) continue; memmove(m, p->data, p->ndata); m[p->ndata] = '\0'; m[1] = parselang(m); if(m[1] == -1) continue; m[0] = 'c'; m[2] = '\0'; if(send(input, m) == -1) break; } plumbfree(p); } void usage(void) { fprint(2, "usage: %s [ -G ] [ -l lang ] [ kbdtap ]\n", argv0); threadexits("usage"); } mainstacksize = 8192*2; void threadmain(int argc, char *argv[]) { int nogui; char *jishoname, *zidianname; deflang = LangEN; nogui = 0; ARGBEGIN{ case 'l': deflang = parselang(EARGF(usage())); if(deflang < 0) usage(); break; case 'G': nogui++; break; default: usage(); }ARGEND; switch(argc){ case 0: kbdin = 0; kbdout = 1; break; case 1: kbdin = kbdout = open(argv[0], ORDWR); if(kbdin < 0) sysfatal("failed to open kbdtap: %r"); break; default: usage(); } /* allow gui to warm up while we're busy reading maps */ if(nogui || access("/dev/winid", AEXIST) < 0){ displaych = nil; selectch = nil; } else { selectch = chancreate(sizeof(int), 1); displaych = chancreate(sizeof(char*)*Maxkouho, 1); proccreate(displaythread, nil, mainstacksize); } memset(backspace, '\b', sizeof backspace-1); backspace[sizeof backspace-1] = '\0'; if((jishoname = getenv("jisho")) == nil) jishoname = "/lib/ktrans/kanji.dict"; jisho = opendict(nil, jishoname); if((zidianname = getenv("zidian")) == nil) zidianname = "/lib/ktrans/wubi.dict"; zidian = opendict(nil, zidianname); natural = hanzi = nil; hira = openmap("/lib/ktrans/hira.map"); kata = openmap("/lib/ktrans/kata.map"); greek = openmap("/lib/ktrans/greek.map"); cyril = openmap("/lib/ktrans/cyril.map"); hangul = openmap("/lib/ktrans/hangul.map"); telex = openmap("/lib/ktrans/telex.map"); dictch = chancreate(Msgsize, 0); input = chancreate(Msgsize, 0); output = chancreate(Msgsize, 0); plumbfd = plumbopen("lang", OREAD); if(plumbfd >= 0) proccreate(plumbproc, nil, mainstacksize); proccreate(kbdtap, nil, mainstacksize); proccreate(kbdsink, nil, mainstacksize); threadcreate(dictthread, nil, mainstacksize); threadcreate(keythread, nil, mainstacksize); threadexits(nil); }