package main
import (
- "strings"
-)
-
-// FromCharPos converts from a character position to an x, y position
-func FromCharPos(loc int, buf *Buffer) (int, int) {
- return FromCharPosStart(0, 0, 0, loc, buf)
-}
+ "unicode/utf8"
-// FromCharPosStart converts from a character position to an x, y position, starting at the specified character location
-func FromCharPosStart(startLoc, startX, startY, loc int, buf *Buffer) (int, int) {
- charNum := startLoc
- x, y := startX, startY
+ runewidth "github.com/mattn/go-runewidth"
+ "github.com/zyedidia/clipboard"
+)
- lineLen := Count(buf.lines[y]) + 1
- for charNum+lineLen <= loc {
- charNum += lineLen
- y++
- lineLen = Count(buf.lines[y]) + 1
+// InBounds returns whether the given location is a valid character position in the given buffer
+func InBounds(pos Loc, buf *Buffer) bool {
+ if pos.Y < 0 || pos.Y >= len(buf.lines) || pos.X < 0 || pos.X > utf8.RuneCount(buf.LineBytes(pos.Y)) {
+ return false
}
- x = loc - charNum
- return x, y
-}
-
-// ToCharPos converts from an x, y position to a character position
-func ToCharPos(x, y int, buf *Buffer) int {
- loc := 0
- for i := 0; i < y; i++ {
- // + 1 for the newline
- loc += Count(buf.lines[i]) + 1
- }
- loc += x
- return loc
+ return true
}
-// The Cursor struct stores the location of the cursor in the view
-// The complicated part about the cursor is storing its location.
-// The cursor must be displayed at an x, y location, but since the buffer
-// uses a rope to store text, to insert text we must have an index. It
-// is also simpler to use character indicies for other tasks such as
-// selection.
+// The Cursor struct stores the location of the cursor in the buffer
+// as well as the selection
type Cursor struct {
- v *View
-
- // The cursor display location
- x int
- y int
+ buf *Buffer
+ Loc
// Last cursor x position
- lastVisualX int
+ LastVisualX int
// The current selection as a range of character numbers (inclusive)
- curSelection [2]int
+ CurSelection [2]Loc
// The original selection as a range of character numbers
// This is used for line and word selection where it is necessary
// to know what the original selection was
- origSelection [2]int
+ OrigSelection [2]Loc
+
+ // Which cursor index is this (for multiple cursors)
+ Num int
+}
+
+// Goto puts the cursor at the given cursor's location and gives
+// the current cursor its selection too
+func (c *Cursor) Goto(b Cursor) {
+ c.X, c.Y, c.LastVisualX = b.X, b.Y, b.LastVisualX
+ c.OrigSelection, c.CurSelection = b.OrigSelection, b.CurSelection
+}
+
+// GotoLoc puts the cursor at the given cursor's location and gives
+// the current cursor its selection too
+func (c *Cursor) GotoLoc(l Loc) {
+ c.X, c.Y = l.X, l.Y
+ c.LastVisualX = c.GetVisualX()
+}
+
+// GetVisualX returns the x value of the cursor in visual spaces
+func (c *Cursor) GetVisualX() int {
+ if c.X <= 0 {
+ c.X = 0
+ return 0
+ }
+
+ bytes := c.buf.LineBytes(c.Y)
+ tabsize := int(c.buf.Settings["tabsize"].(float64))
+ if c.X > utf8.RuneCount(bytes) {
+ c.X = utf8.RuneCount(bytes) - 1
+ }
+
+ return StringWidth(bytes, c.X, tabsize)
+}
+
+// GetCharPosInLine gets the char position of a visual x y
+// coordinate (this is necessary because tabs are 1 char but
+// 4 visual spaces)
+func (c *Cursor) GetCharPosInLine(b []byte, visualPos int) int {
+ tabsize := int(c.buf.Settings["tabsize"].(float64))
+
+ // Scan rune by rune until we exceed the visual width that we are
+ // looking for. Then we can return the character position we have found
+ i := 0 // char pos
+ width := 0 // string visual width
+ for len(b) > 0 {
+ r, size := utf8.DecodeRune(b)
+ b = b[size:]
+
+ switch r {
+ case '\t':
+ ts := tabsize - (width % tabsize)
+ width += ts
+ default:
+ width += runewidth.RuneWidth(r)
+ }
+
+ i++
+
+ if width >= visualPos {
+ break
+ }
+ }
+
+ return i
}
-// SetLoc sets the location of the cursor in terms of character number
-// and not x, y location
-// It's just a simple wrapper of FromCharPos
-func (c *Cursor) SetLoc(loc int) {
- c.x, c.y = FromCharPos(loc, c.v.buf)
- c.lastVisualX = c.GetVisualX()
+// Start moves the cursor to the start of the line it is on
+func (c *Cursor) Start() {
+ c.X = 0
+ c.LastVisualX = c.GetVisualX()
}
-// Loc gets the cursor location in terms of character number instead
-// of x, y location
-// It's just a simple wrapper of ToCharPos
-func (c *Cursor) Loc() int {
- return ToCharPos(c.x, c.y, c.v.buf)
+// End moves the cursor to the end of the line it is on
+func (c *Cursor) End() {
+ c.X = utf8.RuneCount(c.buf.LineBytes(c.Y))
+ c.LastVisualX = c.GetVisualX()
+}
+
+// CopySelection copies the user's selection to either "primary"
+// or "clipboard"
+func (c *Cursor) CopySelection(target string) {
+ if c.HasSelection() {
+ if target != "primary" || c.buf.Settings["useprimary"].(bool) {
+ clipboard.WriteAll(string(c.GetSelection()), target)
+ }
+ }
}
// ResetSelection resets the user's selection
func (c *Cursor) ResetSelection() {
- c.curSelection[0] = 0
- c.curSelection[1] = 0
+ c.CurSelection[0] = c.buf.Start()
+ c.CurSelection[1] = c.buf.Start()
+}
+
+// SetSelectionStart sets the start of the selection
+func (c *Cursor) SetSelectionStart(pos Loc) {
+ c.CurSelection[0] = pos
+}
+
+// SetSelectionEnd sets the end of the selection
+func (c *Cursor) SetSelectionEnd(pos Loc) {
+ c.CurSelection[1] = pos
}
// HasSelection returns whether or not the user has selected anything
func (c *Cursor) HasSelection() bool {
- return c.curSelection[0] != c.curSelection[1]
+ return c.CurSelection[0] != c.CurSelection[1]
}
// DeleteSelection deletes the currently selected text
func (c *Cursor) DeleteSelection() {
- if c.curSelection[0] > c.curSelection[1] {
- c.v.eh.Remove(c.curSelection[1], c.curSelection[0])
- c.SetLoc(c.curSelection[1])
- } else if c.GetSelection() == "" {
+ if c.CurSelection[0].GreaterThan(c.CurSelection[1]) {
+ c.buf.Remove(c.CurSelection[1], c.CurSelection[0])
+ c.Loc = c.CurSelection[1]
+ } else if !c.HasSelection() {
return
} else {
- c.v.eh.Remove(c.curSelection[0], c.curSelection[1])
- c.SetLoc(c.curSelection[0])
+ c.buf.Remove(c.CurSelection[0], c.CurSelection[1])
+ c.Loc = c.CurSelection[0]
}
}
// GetSelection returns the cursor's selection
-func (c *Cursor) GetSelection() string {
- if c.curSelection[0] > c.curSelection[1] {
- return string([]rune(c.v.buf.text)[c.curSelection[1]:c.curSelection[0]])
+func (c *Cursor) GetSelection() []byte {
+ if InBounds(c.CurSelection[0], c.buf) && InBounds(c.CurSelection[1], c.buf) {
+ if c.CurSelection[0].GreaterThan(c.CurSelection[1]) {
+ return c.buf.Substr(c.CurSelection[1], c.CurSelection[0])
+ }
+ return c.buf.Substr(c.CurSelection[0], c.CurSelection[1])
}
- return string([]rune(c.v.buf.text)[c.curSelection[0]:c.curSelection[1]])
+ return []byte{}
}
// SelectLine selects the current line
func (c *Cursor) SelectLine() {
c.Start()
- c.curSelection[0] = c.Loc()
+ c.SetSelectionStart(c.Loc)
c.End()
- if len(c.v.buf.lines)-1 > c.y {
- c.curSelection[1] = c.Loc() + 1
+ if len(c.buf.lines)-1 > c.Y {
+ c.SetSelectionEnd(c.Loc.Move(1, c.buf))
} else {
- c.curSelection[1] = c.Loc()
+ c.SetSelectionEnd(c.Loc)
}
- c.origSelection = c.curSelection
+ c.OrigSelection = c.CurSelection
}
// AddLineToSelection adds the current line to the selection
func (c *Cursor) AddLineToSelection() {
- loc := c.Loc()
-
- if loc < c.origSelection[0] {
+ if c.Loc.LessThan(c.OrigSelection[0]) {
c.Start()
- c.curSelection[0] = c.Loc()
- c.curSelection[1] = c.origSelection[1]
+ c.SetSelectionStart(c.Loc)
+ c.SetSelectionEnd(c.OrigSelection[1])
}
- if loc > c.origSelection[1] {
+ if c.Loc.GreaterThan(c.OrigSelection[1]) {
c.End()
- c.curSelection[1] = c.Loc()
- c.curSelection[0] = c.origSelection[0]
+ c.SetSelectionEnd(c.Loc.Move(1, c.buf))
+ c.SetSelectionStart(c.OrigSelection[0])
}
- if loc < c.origSelection[1] && loc > c.origSelection[0] {
- c.curSelection = c.origSelection
+ if c.Loc.LessThan(c.OrigSelection[1]) && c.Loc.GreaterThan(c.OrigSelection[0]) {
+ c.CurSelection = c.OrigSelection
}
}
-// SelectWord selects the word the cursor is currently on
-func (c *Cursor) SelectWord() {
- if len(c.v.buf.lines[c.y]) == 0 {
- return
+// UpN moves the cursor up N lines (if possible)
+func (c *Cursor) UpN(amount int) {
+ proposedY := c.Y - amount
+ if proposedY < 0 {
+ proposedY = 0
+ } else if proposedY >= len(c.buf.lines) {
+ proposedY = len(c.buf.lines) - 1
}
- if !IsWordChar(string(c.RuneUnder(c.x))) {
- loc := c.Loc()
- c.curSelection[0] = loc
- c.curSelection[1] = loc + 1
- c.origSelection = c.curSelection
- return
- }
+ bytes := c.buf.LineBytes(proposedY)
+ c.X = c.GetCharPosInLine(bytes, c.LastVisualX)
- forward, backward := c.x, c.x
-
- for backward > 0 && IsWordChar(string(c.RuneUnder(backward-1))) {
- backward--
- }
-
- c.curSelection[0] = ToCharPos(backward, c.y, c.v.buf)
- c.origSelection[0] = c.curSelection[0]
-
- for forward < Count(c.v.buf.lines[c.y])-1 && IsWordChar(string(c.RuneUnder(forward+1))) {
- forward++
+ if c.X > utf8.RuneCount(bytes) || (amount < 0 && proposedY == c.Y) {
+ c.X = utf8.RuneCount(bytes)
}
- c.curSelection[1] = ToCharPos(forward, c.y, c.v.buf) + 1
- c.origSelection[1] = c.curSelection[1]
+ c.Y = proposedY
}
-// AddWordToSelection adds the word the cursor is currently on to the selection
-func (c *Cursor) AddWordToSelection() {
- loc := c.Loc()
-
- if loc > c.origSelection[0] && loc < c.origSelection[1] {
- c.curSelection = c.origSelection
- return
- }
-
- if loc < c.origSelection[0] {
- backward := c.x
-
- for backward > 0 && IsWordChar(string(c.RuneUnder(backward-1))) {
- backward--
- }
-
- c.curSelection[0] = ToCharPos(backward, c.y, c.v.buf)
- c.curSelection[1] = c.origSelection[1]
- }
-
- if loc > c.origSelection[1] {
- forward := c.x
-
- for forward < Count(c.v.buf.lines[c.y])-1 && IsWordChar(string(c.RuneUnder(forward+1))) {
- forward++
- }
-
- c.curSelection[1] = ToCharPos(forward, c.y, c.v.buf) + 1
- c.curSelection[0] = c.origSelection[0]
- }
-}
-
-func (c *Cursor) SelectTo(loc int) {
- if loc > c.origSelection[0] {
- c.curSelection[0] = c.origSelection[0]
- c.curSelection[1] = loc + 1
- } else {
- c.curSelection[0] = loc
- c.curSelection[1] = c.origSelection[0] + 1
- }
-}
-
-func (c *Cursor) WordRight() {
- for IsWordChar(string(c.RuneUnder(c.x))) {
- c.Right()
- }
- c.Right()
-}
-
-func (c *Cursor) WordLeft() {
- for IsWordChar(string(c.RuneUnder(c.x))) {
- c.Left()
- }
- c.Left()
-}
-
-// RuneUnder returns the rune under the given x position
-func (c *Cursor) RuneUnder(x int) rune {
- line := []rune(c.v.buf.lines[c.y])
- if len(line) == 0 {
- return '\n'
- }
- if x >= len(line) {
- x = len(line) - 1
- } else if x < 0 {
- x = 0
- }
- return line[x]
+// DownN moves the cursor down N lines (if possible)
+func (c *Cursor) DownN(amount int) {
+ c.UpN(-amount)
}
// Up moves the cursor up one line (if possible)
func (c *Cursor) Up() {
- if c.y > 0 {
- c.y--
-
- runes := []rune(c.v.buf.lines[c.y])
- c.x = c.GetCharPosInLine(c.y, c.lastVisualX)
- if c.x > len(runes) {
- c.x = len(runes)
- }
- }
+ c.UpN(1)
}
// Down moves the cursor down one line (if possible)
func (c *Cursor) Down() {
- if c.y < len(c.v.buf.lines)-1 {
- c.y++
-
- runes := []rune(c.v.buf.lines[c.y])
- c.x = c.GetCharPosInLine(c.y, c.lastVisualX)
- if c.x > len(runes) {
- c.x = len(runes)
- }
- }
+ c.DownN(1)
}
-// Left moves the cursor left one cell (if possible) or to the last line if it is at the beginning
+// Left moves the cursor left one cell (if possible) or to
+// the previous line if it is at the beginning
func (c *Cursor) Left() {
- if c.Loc() == 0 {
+ if c.Loc == c.buf.Start() {
return
}
- if c.x > 0 {
- c.x--
+ if c.X > 0 {
+ c.X--
} else {
c.Up()
c.End()
}
- c.lastVisualX = c.GetVisualX()
+ c.LastVisualX = c.GetVisualX()
}
-// Right moves the cursor right one cell (if possible) or to the next line if it is at the end
+// Right moves the cursor right one cell (if possible) or
+// to the next line if it is at the end
func (c *Cursor) Right() {
- if c.Loc() == c.v.buf.Len() {
+ if c.Loc == c.buf.End() {
return
}
- if c.x < Count(c.v.buf.lines[c.y]) {
- c.x++
+ if c.X < utf8.RuneCount(c.buf.LineBytes(c.Y)) {
+ c.X++
} else {
c.Down()
c.Start()
}
- c.lastVisualX = c.GetVisualX()
-}
-
-// End moves the cursor to the end of the line it is on
-func (c *Cursor) End() {
- c.x = Count(c.v.buf.lines[c.y])
- c.lastVisualX = c.GetVisualX()
-}
-
-// Start moves the cursor to the start of the line it is on
-func (c *Cursor) Start() {
- c.x = 0
- c.lastVisualX = c.GetVisualX()
+ c.LastVisualX = c.GetVisualX()
}
-// GetCharPosInLine gets the char position of a visual x y coordinate (this is necessary because tabs are 1 char but 4 visual spaces)
-func (c *Cursor) GetCharPosInLine(lineNum, visualPos int) int {
- // Get the tab size
- tabSize := settings.TabSize
- // This is the visual line -- every \t replaced with the correct number of spaces
- visualLine := strings.Replace(c.v.buf.lines[lineNum], "\t", "\t"+Spaces(tabSize-1), -1)
- if visualPos > Count(visualLine) {
- visualPos = Count(visualLine)
- }
- numTabs := NumOccurences(visualLine[:visualPos], '\t')
- if visualPos >= (tabSize-1)*numTabs {
- return visualPos - (tabSize-1)*numTabs
- }
- return visualPos / tabSize
-}
-
-// GetVisualX returns the x value of the cursor in visual spaces
-func (c *Cursor) GetVisualX() int {
- runes := []rune(c.v.buf.lines[c.y])
- tabSize := settings.TabSize
- return c.x + NumOccurences(string(runes[:c.x]), '\t')*(tabSize-1)
-}
-
-// Relocate makes sure that the cursor is inside the bounds of the buffer
-// If it isn't, it moves it to be within the buffer's lines
+// Relocate makes sure that the cursor is inside the bounds
+// of the buffer If it isn't, it moves it to be within the
+// buffer's lines
func (c *Cursor) Relocate() {
- if c.y < 0 {
- c.y = 0
- } else if c.y >= len(c.v.buf.lines) {
- c.y = len(c.v.buf.lines) - 1
- }
-
- if c.x < 0 {
- c.x = 0
- } else if c.x > Count(c.v.buf.lines[c.y]) {
- c.x = Count(c.v.buf.lines[c.y])
+ if c.Y < 0 {
+ c.Y = 0
+ } else if c.Y >= len(c.buf.lines) {
+ c.Y = len(c.buf.lines) - 1
}
-}
-// Display draws the cursor to the screen at the correct position
-func (c *Cursor) Display() {
- // Don't draw the cursor if it is out of the viewport or if it has a selection
- if (c.y-c.v.topline < 0 || c.y-c.v.topline > c.v.height-1) || c.HasSelection() {
- screen.HideCursor()
- } else {
- screen.ShowCursor(c.GetVisualX()+c.v.lineNumOffset-c.v.leftCol, c.y-c.v.topline)
+ if c.X < 0 {
+ c.X = 0
+ } else if c.X > utf8.RuneCount(c.buf.LineBytes(c.Y)) {
+ c.X = utf8.RuneCount(c.buf.LineBytes(c.Y))
}
}