commit
37b85a90f5
@ -0,0 +1,7 @@ |
||||
CC=gcc
|
||||
CFLAGS=-Wall -Wextra -pedantic -Werror
|
||||
OFLAGS=-O3
|
||||
LFLAGS=-lncurses
|
||||
|
||||
ste: ste.c |
||||
$(CC) $(CFLAGS) $(OFLAGS) $(LFLAGS) -o ste $^
|
@ -0,0 +1,464 @@ |
||||
#include <ncurses.h> |
||||
#include <stdlib.h> |
||||
#include <string.h> |
||||
|
||||
/* defines */ |
||||
#define CTRL(k) ((k) & 0x1f) // Control mask modifier
|
||||
#define TABSIZE 4 // Tab size as used in render
|
||||
|
||||
/* main data structure containing:
|
||||
* -cursor position |
||||
* -real (matching memory) |
||||
* -offset (memory and screen offset) |
||||
* -render (drawn on the screen) |
||||
* -window size |
||||
* -statusbar message */ |
||||
struct term { |
||||
struct { |
||||
int x; |
||||
int y; |
||||
int off_x; |
||||
int off_y; |
||||
int r_x; |
||||
int r_y; |
||||
} cur; |
||||
|
||||
struct { |
||||
int x; |
||||
int y; |
||||
} dim; |
||||
|
||||
char statusbar[30]; |
||||
int ln_s; |
||||
} t; |
||||
|
||||
/* Row structure, defines actual and
|
||||
* render chars, actual and render size |
||||
* and difference between render and |
||||
* real size of the row */ |
||||
typedef struct row { |
||||
int size; |
||||
char *chars; |
||||
int r_size; |
||||
char *render; |
||||
int delta; |
||||
} row; |
||||
|
||||
/* Rows structure (or file buffer)
|
||||
* defines rows and teh number of rows */ |
||||
struct { |
||||
row *rw; |
||||
int rownum; |
||||
} rows; |
||||
|
||||
/* Prototypes */ |
||||
/* draw operations */ |
||||
static void drawBar (char *s); |
||||
static void drawScreen (); |
||||
static void drawLines (void); |
||||
static void updateRender (row *rw); |
||||
static void updateScroll (void); |
||||
static void cursorMove(int a); |
||||
static int getLineNumberSize (void); |
||||
static void lnMove (int y, int x); |
||||
static int curRealToRender (row *rw, int c_x); |
||||
|
||||
/* Row operations */ |
||||
static inline void rowInit (void); |
||||
|
||||
/* Terminal operations */ |
||||
static void termInit (void); |
||||
static void termExit (void); |
||||
static void termDie (char *s); |
||||
|
||||
/* file operations */ |
||||
static void fileOpen (char *filename); |
||||
void fileSave (char *filename); |
||||
|
||||
/* buffer operations */ |
||||
static void rowAdd (char *s, int len); |
||||
|
||||
/* garbage */ |
||||
//inline void bufferFree (void);
|
||||
static int whatsThat (void); |
||||
|
||||
/* --------------------------------- main ------------------------------------ */ |
||||
int main (int argc, char *argv[]) |
||||
{ |
||||
/* Initialize the first row */ |
||||
rowInit(); |
||||
|
||||
/* Try to open the file */ |
||||
if (argc != 2) { |
||||
perror("File not found"); |
||||
exit(1); |
||||
} else fileOpen(argv[1]); |
||||
|
||||
/* Initialize the terminal in raw mode,
|
||||
* start curses and initialize the term struct */ |
||||
termInit(); |
||||
|
||||
/* Set the statusbar left (static) message */ |
||||
sprintf(t.statusbar, "%s : %s %d lines %dx%d", argv[0], argv[1], rows.rownum, t.dim.y, t.dim.y); |
||||
|
||||
/* Main event loop */ |
||||
int c; |
||||
while (1) { |
||||
/* Redraw the screen */ |
||||
drawScreen(); |
||||
|
||||
/* Wait for an event (keypress) */ |
||||
switch (c = getch()) { |
||||
case (CTRL('q')): |
||||
termExit(); |
||||
break; |
||||
case (KEY_LEFT): |
||||
case (KEY_RIGHT): |
||||
case (KEY_UP): |
||||
case (KEY_DOWN): |
||||
cursorMove(c); |
||||
break; |
||||
} |
||||
} |
||||
|
||||
/* If by chance i find myself here be sure
|
||||
* end curses mode and clenaup */ |
||||
termExit(); |
||||
return 0; |
||||
} |
||||
|
||||
/* ----------------------------- end of main ---------------------------------- */ |
||||
|
||||
void termInit (void) |
||||
{ |
||||
/* Init the screen and refresh */ |
||||
initscr(); |
||||
refresh(); |
||||
|
||||
/* Enable raw mode, this makes the termianl ignore
|
||||
* interrupt signals like CTRL+C and CTRL+Z |
||||
* allowing us to make our own bindings */ |
||||
raw(); |
||||
|
||||
/* Allow use of function keys */ |
||||
keypad(stdscr, TRUE); |
||||
|
||||
/* Turn off echoing */ |
||||
noecho(); |
||||
|
||||
/* Set the tab size */ |
||||
set_tabsize(TABSIZE); |
||||
|
||||
/* Start color mode */ |
||||
start_color(); |
||||
init_pair(1, COLOR_BLUE, COLOR_BLACK); |
||||
|
||||
/* Populate the main data structure */ |
||||
getmaxyx(stdscr, t.dim.y, t.dim.x); |
||||
t.dim.y -= 1; |
||||
t.ln_s = getLineNumberSize(); |
||||
t.dim.x -= t.ln_s + 1; |
||||
|
||||
/* Initialize the data staructure */ |
||||
t.cur.x = t.cur.off_x = 0; |
||||
t.cur.y = t.cur.off_y = 0; |
||||
} |
||||
|
||||
/* Calculate the correct spacing for the line numbers
|
||||
* based on the size of the file */ |
||||
int getLineNumberSize (void) |
||||
{ |
||||
int n = rows.rownum, l = 0; |
||||
for (l = 0; n > 0; l++) n /= 10; |
||||
return l + 1; |
||||
} |
||||
|
||||
void termExit (void) |
||||
{ |
||||
// bufferFree();
|
||||
erase(); |
||||
refresh(); |
||||
endwin(); |
||||
exit(0); |
||||
} |
||||
|
||||
void termDie (char *s) |
||||
{ |
||||
refresh(); |
||||
endwin(); |
||||
perror(s); |
||||
exit(1); |
||||
} |
||||
|
||||
/* ----------------------------- term operations -------------------------------- */ |
||||
|
||||
void drawScreen () |
||||
{ |
||||
/* Clear the screen */ |
||||
erase(); |
||||
/* Update Scroll */ |
||||
updateScroll(); |
||||
/* draw the lines */ |
||||
drawLines(); |
||||
/* draw the bar */ |
||||
drawBar(t.statusbar); |
||||
/* move back to the cursor position */ |
||||
lnMove(t.cur.y, t.cur.r_x); |
||||
/* refresh the screen */ |
||||
refresh(); |
||||
} |
||||
|
||||
/* Draw all the appropriate lines (following cursor) to the screen */ |
||||
void drawLines (void) |
||||
{ |
||||
int line = 0, ln; |
||||
/* move to the beginning of the screen */ |
||||
lnMove(0, 0); |
||||
|
||||
for (int i = 0; i < t.dim.y; i++) { |
||||
if (i >= rows.rownum) break; |
||||
ln = i + t.cur.off_y; |
||||
|
||||
/* Draw the line number */ |
||||
attron(COLOR_PAIR(1)); |
||||
mvprintw(i, 0, "%d", ln + 1); |
||||
attroff(COLOR_PAIR(1)); |
||||
lnMove(i, 0); |
||||
|
||||
/* Draw the line matcing render memory */ |
||||
if (rows.rw[ln].r_size >= t.cur.off_x) { |
||||
addnstr(&rows.rw[ln].render[t.cur.off_x], t.dim.x + 1 - rows.rw[ln].delta); |
||||
} |
||||
lnMove(++line, 0); |
||||
} |
||||
lnMove(t.cur.y, t.cur.x); |
||||
} |
||||
|
||||
/* Move avoiding the space allocated for line numbers */ |
||||
void lnMove (int y, int x) |
||||
{ |
||||
x += t.ln_s; |
||||
move(y, x); |
||||
} |
||||
|
||||
/* Draw the status bar at the bottom of the screen */ |
||||
void drawBar (char *s) |
||||
{ |
||||
/* Set maximum contrast for bar */ |
||||
attron(A_STANDOUT); |
||||
/* get the length of the statusbar text */ |
||||
int len = strlen(s); |
||||
/* Print the message */ |
||||
mvprintw(t.dim.y, 0, s); |
||||
/* Fill everything else with spaces */ |
||||
for (int i = len; i <= t.dim.x + t.ln_s; i++) |
||||
mvaddch(t.dim.y, i, ' '); |
||||
|
||||
char m[10]; |
||||
sprintf(m, "Zoom:\t%c", whatsThat()); |
||||
mvaddstr(t.dim.y, t.dim.x + t.ln_s - strlen(m), m); |
||||
|
||||
/* Return to normal contrast mode */ |
||||
attroff(A_STANDOUT); |
||||
} |
||||
|
||||
/* convert the cursor matchoing the memory to the drawn one */ |
||||
int curRealToRender (row *rw, int c_x) |
||||
{ |
||||
int r_x = 0; |
||||
for (int i = 0; i < c_x; i++) { |
||||
if (rw->chars[i] == '\t') r_x += (TABSIZE - 1) - (r_x % TABSIZE); |
||||
r_x++; |
||||
} |
||||
return r_x; |
||||
} |
||||
|
||||
/* -------------------------------- draw operations -------------------------------- */ |
||||
|
||||
/* Open a file and put it into a buffer line by line */ |
||||
void fileOpen (char *filename) |
||||
{ |
||||
FILE *fp = fopen(filename, "r"); |
||||
if (!fp) termDie("Cannot open file"); |
||||
|
||||
/* Init the linebuffer */ |
||||
char *line = NULL; |
||||
/* Set linecap to 0 so getline will atumatically allocate
|
||||
* memory for the line buffer*/ |
||||
size_t linecap = 0; |
||||
ssize_t linelen; |
||||
|
||||
/* getline returns -1 if no new line is present */ |
||||
while ((linelen = getline(&line, &linecap, fp)) != -1) { |
||||
while (linelen > 0 && (line[linelen - 1] == '\n' || line[linelen - 1] == '\r')) |
||||
linelen--; |
||||
rowAdd(line, linelen); |
||||
} |
||||
/* free the line buffer */ |
||||
free(line); |
||||
/* close the file */ |
||||
fclose(fp); |
||||
} |
||||
|
||||
/* Add a row to the file buffer */ |
||||
void rowAdd (char *s, int len) |
||||
{ |
||||
/* Extend the block of memory containing the lines */ |
||||
rows.rw = realloc(rows.rw, sizeof(row) * (rows.rownum + 1)); |
||||
|
||||
/* Allocate memory for the line and copy it
|
||||
* at the current row number */ |
||||
rows.rw[rows.rownum].chars = malloc(len + 1); |
||||
memcpy(rows.rw[rows.rownum].chars, s, len); |
||||
rows.rw[rows.rownum].chars[len] = '\0'; |
||||
rows.rw[rows.rownum].size = len; |
||||
updateRender(&rows.rw[rows.rownum]); |
||||
rows.rownum++; |
||||
} |
||||
|
||||
void updateRender (row *rw) |
||||
{ |
||||
/* count the special characters (only tabs for now) */ |
||||
int tabs = 0, i; |
||||
for (i = 0; i < rw->size; i++) { |
||||
if (rw->chars[i] == '\t') tabs++; |
||||
} |
||||
rw->render = NULL; |
||||
free(rw->render); |
||||
|
||||
/* Render is long as size with the added tab spaces - 1
|
||||
* (we already count for the \t as a char) */ |
||||
rw->render = malloc(rw->size + tabs * (TABSIZE - 1) + 1); |
||||
|
||||
if (rw->render == NULL) termDie ("malloc in updateRender"); |
||||
|
||||
/* put all the characters (substituing all special chars)
|
||||
* into the render buffer */ |
||||
int off = 0; |
||||
for (i = 0; i < rw->size; i++) { |
||||
if (rw->chars[i] == '\t') { |
||||
for (int j = 0; j < TABSIZE; j++) |
||||
rw->render[off++] = ' '; |
||||
} else { |
||||
rw->render[off++] = rw->chars[i]; |
||||
} |
||||
} |
||||
rw->render[off] = '\0'; |
||||
rw->r_size = off; |
||||
} |
||||
|
||||
void rowInit (void) |
||||
{ |
||||
rows.rw = NULL; |
||||
rows.rownum = 0; |
||||
} |
||||
|
||||
/* ----------------------------- file operations --------------------------- */ |
||||
|
||||
/* take care of the cursor movement */ |
||||
void cursorMove (int a) |
||||
{ |
||||
switch (a) { |
||||
case (KEY_LEFT): |
||||
if (t.cur.x <= 0 && !t.cur.off_x) { |
||||
if (t.cur.y) { |
||||
t.cur.y--; |
||||
t.cur.x = rows.rw[t.cur.y + t.cur.off_y].size; |
||||
} |
||||
} else t.cur.x--; |
||||
break; |
||||
|
||||
case (KEY_RIGHT): |
||||
if (t.cur.x + t.cur.off_x >= rows.rw[t.cur.y + t.cur.off_y].size) { |
||||
if (t.cur.y + t.cur.off_y < rows.rownum - 1) { |
||||
t.cur.y++; |
||||
if (t.cur.off_x) t.cur.off_x = 0; |
||||
t.cur.x = rows.rw[t.cur.y + t.cur.off_y].size; |
||||
} |
||||
} else t.cur.x++; |
||||
break; |
||||
|
||||
case (KEY_UP): |
||||
if (t.cur.y + t.cur.off_y > 0) { |
||||
if (t.cur.y) t.cur.y--; |
||||
if (t.cur.x + t.cur.off_x > rows.rw[t.cur.y + t.cur.off_y].size) { |
||||
if (t.cur.off_x) t.cur.off_x = 0; |
||||
t.cur.x = rows.rw[t.cur.y + t.cur.off_y].size; |
||||
} |
||||
} |
||||
break; |
||||
|
||||
case (KEY_DOWN): |
||||
if (t.cur.y + t.cur.off_y < rows.rownum - 1) { |
||||
t.cur.y++; |
||||
if (t.cur.x + t.cur.off_x > rows.rw[t.cur.y + t.cur.off_y].size) { |
||||
if (t.cur.off_x) t.cur.off_x = 0; |
||||
t.cur.x = rows.rw[t.cur.y + t.cur.off_y].size; |
||||
} |
||||
} |
||||
break; |
||||
} |
||||
} |
||||
|
||||
void updateScroll (void) |
||||
{ |
||||
/* Set y offset */ |
||||
if (t.cur.y >= t.dim.y) { |
||||
if (t.cur.y == t.dim.y) t.cur.off_y++; |
||||
else t.cur.off_y += t.cur.y - t.dim.y; |
||||
|
||||
t.cur.y = t.dim.y - 1; |
||||
|
||||
} else if (t.cur.y <= 0 && t.cur.off_y > 0) { |
||||
t.cur.off_y--; |
||||
t.cur.y = 0; |
||||
} |
||||
|
||||
/* Set x offeset */ |
||||
if (t.cur.x >= t.dim.x) { |
||||
if (t.cur.x == t.dim.x) t.cur.off_x++; |
||||
else t.cur.off_x += t.cur.x - t.dim.x; |
||||
|
||||
t.cur.x = t.dim.x; |
||||
|
||||
} else if (t.cur.x <= 0 && t.cur.off_x > 0) { |
||||
t.cur.off_x--; |
||||
t.cur.x = 0; |
||||
} |
||||
/* convert the cursor from real to render */ |
||||
int ln = t.cur.y + t.cur.off_y; |
||||
t.cur.r_x = curRealToRender(&rows.rw[ln], t.cur.x); |
||||
} |
||||
|
||||
/*---------------------------------- scroll ------------------------------------*/ |
||||
|
||||
/*
|
||||
void bufferFree (void) |
||||
{ |
||||
for (int i = 0; i < rows.rownum; i++) { |
||||
free(&rows.rw[i].chars); |
||||
free(&rows.rw[i].render); |
||||
} |
||||
free(rows.rw); |
||||
} |
||||
*/ |
||||
|
||||
/* See whats under the cursor (memory) */ |
||||
int whatsThat (void) { |
||||
int ln = t.cur.y + t.cur.off_y; |
||||
int c = rows.rw[ln].chars[t.cur.x + t.cur.off_x]; |
||||
switch (c) { |
||||
case ('\t'): |
||||
return '^'; |
||||
break; |
||||
case (' '): |
||||
return '~'; |
||||
break; |
||||
default: |
||||
return c; |
||||
break; |
||||
} |
||||
return 0; |
||||
} |
||||
|
||||
/*--------------------------------- garbage ------------------------------------*/ |
Loading…
Reference in new issue