summary refs log tree commit diff stats
path: root/compiler/lists.nim
blob: 27e3733425ccef308ad853212343b46d09d09a0b (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
#
#
#           The Nim Compiler
#        (c) Copyright 2012 Andreas Rumpf
#
#    See the file "copying.txt", included in this
#    distribution, for details about the copyright.
#

# This module implements a generic doubled linked list.
# TODO Remove this and replace it with something sensible
import os
type
  PListEntry* = ref TListEntry
  TListEntry* = object of RootObj
    prev*, next*: PListEntry

  TStrEntry* = object of TListEntry
    data*: string

  PStrEntry* = ref TStrEntry
  TLinkedList* = object       # for the "find" operation:
    head*, tail*: PListEntry
    counter*: int

  TCompareProc* = proc (entry: PListEntry, closure: pointer): bool {.nimcall.}

proc initLinkedList*(list: var TLinkedList) =
  list.counter = 0
  list.head = nil
  list.tail = nil

proc append*(list: var TLinkedList, entry: PListEntry) =
  inc(list.counter)
  entry.next = nil
  entry.prev = list.tail
  if list.tail != nil:
    assert(list.tail.next == nil)
    list.tail.next = entry
  list.tail = entry
  if list.head == nil: list.head = entry

proc contains*(list: TLinkedList, data: string): bool =
  var it = list.head
  while it != nil:
    if PStrEntry(it).data == data:
      return true
    it = it.next

proc newStrEntry(data: string): PStrEntry =
  new(result)
  result.data = data

proc appendStr*(list: var TLinkedList, data: string) =
  append(list, newStrEntry(data))

proc includeStr*(list: var TLinkedList, data: string): bool =
  if contains(list, data): return true
  appendStr(list, data)       # else: add to list

proc prepend*(list: var TLinkedList, entry: PListEntry) =
  inc(list.counter)
  entry.prev = nil
  entry.next = list.head
  if list.head != nil:
    assert(list.head.prev == nil)
    list.head.prev = entry
  list.head = entry
  if list.tail == nil: list.tail = entry

proc prependStr*(list: var TLinkedList, data: string) =
  prepend(list, newStrEntry(data))

proc insertBefore*(list: var TLinkedList, pos, entry: PListEntry) =
  assert(pos != nil)
  if pos == list.head:
    prepend(list, entry)
  else:
    inc(list.counter)
    entry.next = pos
    entry.prev = pos.prev
    if pos.prev != nil: pos.prev.next = entry
    pos.prev = entry

proc remove*(list: var TLinkedList, entry: PListEntry) =
  dec(list.counter)
  if entry == list.tail:
    list.tail = entry.prev
  if entry == list.head:
    list.head = entry.next
  if entry.next != nil: entry.next.prev = entry.prev
  if entry.prev != nil: entry.prev.next = entry.next

proc bringToFront*(list: var TLinkedList, entry: PListEntry) =
  when true:
    list.remove entry
    list.prepend entry
  else:
    if entry == list.head: return
    if entry == list.tail: list.tail = entry.prev
    if entry.next != nil: entry.next.prev = entry.prev
    if entry.prev != nil: entry.prev.next = entry.next
    entry.prev = nil
    entry.next = list.head
    list.head = entry

proc excludePath*(list: var TLinkedList, data: string) =
  var it = list.head
  while it != nil:
    let nxt = it.next
    if cmpPaths(PStrEntry(it).data, data) == 0:
      remove(list, it)
    it = nxt

proc find*(list: TLinkedList, fn: TCompareProc, closure: pointer): PListEntry =
  result = list.head
  while result != nil:
    if fn(result, closure): return
    result = result.next