normalMode.c (12577B)
1 #include <X11/keysym.h> 2 #include <X11/XKBlib.h> 3 4 #include "normalMode.h" 5 #include "utils.h" 6 7 extern Glyph const styleSearch, style[]; 8 extern char const wDelS[], wDelL[], *nmKeys[]; 9 extern unsigned int bg[], fg, currentBg, highlightBg, highlightFg, amountNmKeys; 10 11 typedef struct { int p[3]; } Pos; 12 struct NormalModeState { 13 struct OperationState { 14 enum Op {noop=0, visual='v', visualLine='V', yank = 'y'} op; 15 enum Infix {infix_none=0, infix_i='i', infix_a='a'} infix; 16 } cmd; 17 struct MotionState { 18 uint32_t c; int active; Pos searchPos; 19 enum Search {none=0, fw='/', bw='?'} search; 20 } m; 21 } defaultNormalMode, state; 22 DynamicArray searchStr=UTF8_ARRAY, cCmd=UTF8_ARRAY, lCmd=UTF8_ARRAY; 23 Glyph styleCmd; 24 char posBuffer[10], brack[6][2] = { {"()"}, {"<>"}, {"{}"}, {"[]"}, {"\"\""}, {"''"}}; 25 int exited=1, overlay=1; 26 static inline uint32_t cchar() { return term.line[term.c.y][term.c.x].u; } 27 static inline int pos(int p, int h) {return IS_SET(MODE_ALTSCREEN)?p:rangeY(p+h*histOff-insertOff);} 28 static inline int contains(char c, char const * values, uint32_t memSize) { 29 for (uint32_t i = 0; i < memSize; ++i) if (c == values[i]) return 1; 30 return 0; 31 } 32 static inline void decodeTo(char const *cs, int len, DynamicArray *darr) { 33 char *var = expand(darr); 34 if (!var) empty(darr); else utf8decode(cs, (Rune*)(var), len); 35 } 36 static inline void applyPos(Pos p) { 37 term.c.x = p.p[0], term.c.y = p.p[1]; 38 if (!IS_SET(MODE_ALTSCREEN) && histOp) term.line = &buf[histOff = p.p[2]]; 39 } 40 /// Find string in history buffer, and provide string-match-lookup for highlighting matches 41 static int highlighted(int x, int y) { 42 int const s=term.row*term.col, i=y*term.col+x, sz=size(&searchStr); 43 return sz && i<s && mark[i]!=sz && i+mark[i]<s && !mark[i+mark[i]]; 44 } 45 static void markSearchMatches(int all) { 46 int sz = size(&searchStr), ox = 0, oy = 0, oi=0; 47 for (int y=0; sz && all && y<term.row; ++y) 48 for (int x=0; x<term.col; ++x) term.dirty[y] |= highlighted(x, y); 49 for (int y = 0, wi=0, owi=0, i=0; sz && y < term.row; ++y) 50 for (int x=0; x<term.col; ++x, wi%=sz, ++i, owi=wi) 51 if (all || term.dirty[y]) { 52 mark[i]=sz-(wi=(getU32(&searchStr,wi,1)==term.line[y][x].u?wi+1:0)); 53 if (wi==1) ox=x, oy=y, oi=i; else if (!wi && owi) x=ox, y=oy, i=oi; 54 } 55 for (int y=0; sz &&all &&y<term.row; ++y) 56 for (int x=0; x<term.col; ++x) term.dirty[y] |= highlighted(x, y); 57 } 58 static int findString(int8_t s, int all) { 59 Pos p = (Pos) {.p={term.c.x, term.c.y, IS_SET(MODE_ALTSCREEN) ? 0 : histOff}}; 60 historyMove(s, 0, 0); 61 uint32_t strSz=size(&searchStr), maxIter=rows()*term.col+strSz, widx=0; 62 for (uint32_t i=0, wi = 0; widx<strSz && ++i<=maxIter; historyMove(s, 0, 0), wi=widx) { 63 widx = (getU32(&searchStr, widx, s>0)==cchar())?widx+1:0; 64 if (wi && !widx) historyMove(-s*wi, 0, 0); 65 } 66 if (widx == strSz && widx) historyMove(-s * strSz, 0, 0); 67 else applyPos(p); 68 markSearchMatches(all); 69 return widx == strSz; 70 } 71 /// Execute series of normal-mode commands from char array / decoded from dynamic array 72 static ExitState pressKeys(char const* s, size_t e) { 73 ExitState x=succ; 74 for (size_t i=0; i<e && (x=(!s[i] ? x : kpressHist(&s[i], 1, 0, NULL))); ++i); 75 return x; 76 } 77 static ExitState executeCommand(uint32_t *c, size_t z) { 78 ExitState x=succ; 79 char dc [32]; 80 for (size_t i=0; i<z && (x=kpressHist(dc,utf8encode(c[i],dc),0,NULL));++i); 81 return x; 82 } 83 /// Get character for overlay, if the overlay (st) has something to show, else normal char. 84 static void getChar(DynamicArray *st, Glyph *glyphChange, int y, int xEnd, int width, int x) { 85 if (x < xEnd - min(width=min(width,xEnd), size(st))) *glyphChange = term.line[y][x]; 86 else if (x<xEnd) glyphChange->u = *((Rune*)(st->content + (size(st)+x-xEnd)*st->elSize)); 87 } 88 /// Expand "infix" expression: for instance (w =>) l b | | v e | | y 89 static ExitState expandExpression(char c) { // ({ =>) l ? { \n | l | v / } \n | h | y 90 int a=state.cmd.infix==infix_a, yank=state.cmd.op=='y', lc=tolower(c), found=1; 91 state.cmd.infix = infix_none; 92 if(!yank && state.cmd.op!=visual && state.cmd.op!=visualLine) return failed; 93 char mot[11] = {'l', 0, 'b', 0, 0, 'v', 0, 'e', 0, 0, yank ? 'y' : 0}; 94 if (lc == 'w') mot[2] = 'b' - lc + c, mot[7] = (a ? 'w' : 'e') - lc + c, mot[9]=a?'h':0; 95 else { 96 mot[1]='?', mot[3]=mot[8]='\n', mot[6]='/', mot[4]=a?0:'l', mot[9]=a?0:'h'; 97 for (int i=found=0; !found && i < 6; ++i) 98 if ((found=contains(c,brack[i],2))) mot[2]=brack[i][0], mot[7]=brack[i][1]; 99 } 100 if (!found) return failed; 101 assign(&lCmd, &cCmd); 102 empty(&cCmd); 103 state.cmd = defaultNormalMode.cmd; 104 return pressKeys(mot, 11); 105 } 106 107 int executeMotion(char const cs, int len, KeySym const *const ks) { 108 state.m.c = max(state.m.c, 1); 109 if (ks && *ks == XK_d) historyMove(0, 0, term.row / 2); 110 else if (ks && *ks == XK_u) historyMove(0, 0, -term.row / 2); 111 else if (ks && *ks == XK_f) historyMove(0, 0, term.row-1+(term.c.y=0)); 112 else if (ks && *ks == XK_b) historyMove(0, 0, -(term.c.y=term.row-1)); 113 else if (ks && *ks == XK_h) overlay = !overlay; 114 else if (!len) return failed; 115 else if (cs == 'K') historyMove(0, 0, -state.m.c); 116 else if (cs == 'J') historyMove(0, 0, state.m.c); 117 else if (cs == 'k') historyMove(0, -state.m.c, 0); 118 else if (cs == 'j') historyMove(0, state.m.c, 0); 119 else if (cs == 'h') historyMove(-state.m.c, 0, 0); 120 else if (cs == 'l') historyMove( state.m.c, 0, 0); 121 else if (cs == 'H') term.c.y = 0; 122 else if (cs == 'M') term.c.y = term.bot / 2; 123 else if (cs == 'L') term.c.y = term.bot; 124 else if (cs == 's' || cs == 'S') altToggle = cs == 's' ? !altToggle : 1; 125 else if (cs == 'G' || cs == 'g') { 126 if (cs == 'G') term.c = c[0] = c[IS_SET(MODE_ALTSCREEN)+1]; 127 if (!IS_SET(MODE_ALTSCREEN)) term.line = &buf[histOff=insertOff]; 128 } else if (cs == '0') term.c.x = 0; 129 else if (cs == '$') term.c.x = term.col-1; 130 else if (cs == 't') sel.type = sel.type==SEL_REGULAR ? SEL_RECTANGULAR : SEL_REGULAR; 131 else if (cs == 'n' || cs == 'N') { 132 int const d = ((cs=='N')!=(state.m.search==bw))?-1:1; 133 for (int i = state.m.c; i && findString(d, 0); --i); 134 } else if (contains(cs, "wWeEbB", 6)) { 135 int const low=cs<=90, off=tolower(cs)!='w', sgn=(tolower(cs)=='b')?-1:1, 136 l=strlen(wDelL), s=strlen(wDelS), mit=rows()*term.col; 137 for (int it=0, on=0; state.m.c > 0; ++it) { 138 if (off || it) if (!historyMove(sgn, 0, 0)) it = mit; //< offset move 139 int n = 1<<(contains(cchar(),wDelS,s) ?(2-low) :!contains(cchar(),wDelL,l)), 140 found = (on|=n)^n && ((off ?on^n :n)!=1); //< state change &letter state 141 if (found && off) historyMove(-sgn, 0, 0); //< offset move if required 142 if (found || it>mit) it=-1, on=0, --state.m.c; //< terminate iteration 143 } 144 } else return failed; 145 state.m.c = 0; 146 return state.cmd.op == yank ? exitMotion : succ; 147 } 148 149 ExitState kpressHist(char const *cs, int len, int ctrl, KeySym const *ksym) { 150 historyOpToggle(1, 1); 151 int const prevYOff=IS_SET(MODE_ALTSCREEN)?0:histOff, search=state.m.search&&state.m.active, 152 prevAltToggle=altToggle, prevOverlay=overlay; 153 int const noOp=!state.cmd.op&&!state.cmd.infix, num=len==1&&BETWEEN(cs[0],48,57), 154 esc=ksym&&*ksym==XK_Escape, ret=(ksym&&*ksym==XK_Return)||(len==1&&cs[0]=='\n'), 155 quantifier=num&&(cs[0]!='0'||state.m.c), ins=!search &&noOp &&len &&cs[0]=='i'; 156 exited = 0; 157 ExitState result = succ; 158 if (esc || ret || ins) { result = exitMotion, len = 0; 159 } else if (ksym && *ksym == XK_BackSpace) { 160 if ((search || state.m.c) && size(&cCmd)) pop(&cCmd); 161 if (search) { 162 if (size(&searchStr)) pop(&searchStr); 163 else result = exitMotion; 164 if (!size(&searchStr)) tfulldirt(); 165 applyPos(state.m.searchPos); 166 findString(state.m.search==fw ? 1 : -1, 1); 167 } else if (state.m.c) state.m.c /= 10; 168 len = 0; 169 } else if (search) { 170 if (len >= 1) decodeTo(cs, len, &searchStr); 171 applyPos(state.m.searchPos); 172 findString(state.m.search==fw ? 1 : -1, 1); 173 } else if (len == 0) { result = failed; 174 } else if (quantifier) { state.m.c = min(SHRT_MAX, state.m.c*10+cs[0]-48); 175 } else if (state.cmd.infix && state.cmd.op && (result = expandExpression(cs[0]), len=0)) { 176 } else if (cs[0] == '.') { 177 if (size(&cCmd)) assign(&lCmd, &cCmd); 178 empty(&cCmd); 179 executeCommand((uint32_t*) lCmd.content, size(&lCmd)); 180 empty(&cCmd); 181 len = 0; 182 } else if (cs[0] == 'r') { tfulldirt(); 183 } else if (cs[0] == 'c') { 184 empty(&lCmd); 185 empty(&cCmd); 186 empty(&searchStr); 187 tfulldirt(); 188 len = 0; 189 } else if (cs[0] == fw || cs[0] == bw) { 190 empty(&searchStr); 191 state.m.search = cs[0]; 192 state.m.searchPos = (Pos){.p={term.c.x, term.c.y, prevYOff}}; 193 state.m.active = 1; 194 } else if (cs[0]==infix_i || cs[0]==infix_a) { state.cmd.infix=cs[0]; 195 } else if (cs[0] == 'y') { 196 if (state.cmd.op) { 197 result = (state.cmd.op == yank) ? exitOp : exitMotion; 198 if (state.cmd.op == yank) selstart(0, term.c.y, 0); 199 } else selstart(term.c.x, term.c.y, 0); 200 state.cmd.op = yank; 201 } else if (cs[0] == visual || cs[0] == visualLine) { 202 if (state.cmd.op != (unsigned char) cs[0]) { 203 state.cmd = defaultNormalMode.cmd; 204 state.cmd.op = cs[0]; 205 selstart(cs[0] == visualLine ?0 :term.c.x, term.c.y, 0); 206 } else result = exitOp; 207 } else if (!(result =executeMotion(len?cs[0]:0, len, ctrl?ksym:NULL))) { 208 result=failed; 209 for (size_t i = 0; !ctrl && i < amountNmKeys; ++i) 210 if (cs[0]==nmKeys[i][0] && 211 failed!=(result=pressKeys(&nmKeys[i][1], strlen(nmKeys[i])-1))) goto end; 212 } // Operation/Motion finished if valid: update cmd string, extend selection, update search 213 if (result != failed) { 214 if (len == 1 && !ctrl) decodeTo(cs, len, &cCmd); 215 if ((state.cmd.op == visualLine) || ((state.cmd.op == yank) && (result == exitOp))) { 216 int const off = pos(term.c.y, 1) < pos(sel.ob.y, 0); 217 sel.ob.x = off ? term.col - 1 : 0; 218 selextend(off ? 0 : term.col-1, term.c.y, sel.type, 0); 219 } else if (sel.oe.x != -1) selextend(term.c.x, term.c.y, sel.type, 0); 220 } // Set repaint for motion or status bar 221 if (!IS_SET(MODE_ALTSCREEN) && prevYOff != histOff) tfulldirt(); 222 // Terminate Motion / operation if thus indicated 223 if (result == exitMotion) { 224 if (!state.m.active) result = (exited=noOp) ? finished : exitOp; 225 state.m.active = state.m.c = 0; 226 } 227 if (result == exitOp || result == finished) { 228 if (state.cmd.op == yank) { 229 xsetsel(getsel()); 230 xclipcopy(); 231 } 232 state = defaultNormalMode; 233 selclear(); 234 if (!esc) assign(&lCmd, &cCmd); 235 empty(&cCmd); 236 } // Update the content displayed in the history overlay 237 styleCmd = style[state.cmd.op==yank ? 1 : (state.cmd.op==visual ? 2 : 238 (state.cmd.op==visualLine ? 3 :0))]; 239 int const posLin = !IS_SET(MODE_ALTSCREEN) ? rangeY(insertOff-histOff):0, h=rows()-term.row; 240 if (!posLin || posLin==h || !h) strcpy(posBuffer, posLin ? " [BOT] " : " [TOP] "); 241 else sprintf(posBuffer, " % 3d%c ", min(100, max(0, .5 + posLin * 100. / h)),'%'); 242 if ((overlay || overlay!=prevOverlay) && term.col>9 && term.row>4) { 243 if (!term.dirty[term.row-1]) xdrawline(term.line[term.row-1], term.col*2/3, term.row-1, term.col-1); 244 if (!term.dirty[term.row-2]) xdrawline(term.line[term.row-2], term.col*2/3, term.row-2, term.col-1); 245 } 246 if (result==finished) altToggle = 0; 247 if (altToggle != prevAltToggle) tswapscreen(); 248 end: 249 historyOpToggle(-1, 1); 250 return result; 251 } 252 253 void historyOverlay(int x, int y, Glyph* g) { 254 if (!histMode) return; 255 TCursor const *cHist = histOp ? &term.c : &c[0]; 256 if(overlay && term.col > 9 && term.row > 4 && (x > (2*term.col/3)) && (y >= (term.row-2))) { 257 *g = (y == term.row - 2) ? styleSearch : styleCmd; 258 if (y == term.row-2) getChar(&searchStr, g, term.row-2, term.col-2, term.col/3, x); 259 else if (x > term.col - 7) g->u = posBuffer[x - term.col + 7]; 260 else getChar(size(&cCmd) ?&cCmd :&lCmd, g, term.row-1, term.col-7, term.col/3-6, x); 261 } else if (highlighted(x, y)) g->bg = highlightBg, g->fg = highlightFg; 262 else if ((x==cHist->x) ^ (y==cHist->y)) g->bg = currentBg; 263 else if (x==cHist->x) g->mode^=ATTR_REVERSE; 264 } 265 void historyPreDraw() { 266 static Pos op = {.p={0, 0, 0}}; 267 historyOpToggle(1, 0); 268 // Draw the cursor cross if changed 269 if (term.c.y >= term.row || op.p[1] >= term.row) tfulldirt(); 270 else if (exited || (op.p[1] != term.c.y)) term.dirty[term.c.y] = term.dirty[op.p[1]] = 1; 271 for (int i=0; (exited || term.c.x != op.p[0]) && i<term.row; ++i) if (!term.dirty[i]) { 272 xdrawline(term.line[i], term.c.x, i, term.c.x + 1); 273 xdrawline(term.line[i], op.p[0], i, op.p[0] + 1); 274 } 275 // Update search results either only for lines with new content or all results if exiting 276 markSearchMatches(exited); 277 op = (Pos){.p = {term.c.x, term.c.y, 0}}; 278 historyOpToggle(-1, 0); 279 }