language-servers/roc
1
0
Fork 0
mirror of https://github.com/roc-lang/roc.git synced 2025-09-28 06:14:46 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 6

Many functions taken an allocator

Browse source
This commit is contained in:
Chadtech 2021-01-02 01:45:22 -05:00
parent 00ec961bfe
commit 12d836bfb0
1 changed files with 22 additions and 24 deletions
Download patch file Download diff file Expand all files Collapse all files

46
compiler/builtins/bitcode/src/dict.zig
View file

@ -34,19 +34,19 @@ pub fn RocDict(
// two entries, and there is // two entries, and there is
// therefore an extra entry // therefore an extra entry
// that needs to be stored // that needs to be stored
pub fn new(index: u64, key_ptr: *Key, value_ptr: *Value) Overflow { pub fn new(allocator: *Allocator, index: u64, key_ptr: *Key, value_ptr: *Value) Overflow {
var init_entries: [size]?*Entry = undefined; var init_entries: [size]?*Entry = undefined;
var new_overflow = Overflow{ .entries = &init_entries }; var new_overflow = Overflow{ .entries = &init_entries };
new_overflow.setAtIndex(index, key_ptr, value_ptr); new_overflow.setAtIndex(allocator, index, key_ptr, value_ptr);
return new_overflow; return new_overflow;
} }
pub fn setAtIndex(self: *Overflow, index: u64, key_ptr: *Key, value_ptr: *Value) void { pub fn setAtIndex(self: *Overflow, allocator: *Allocator, index: u64, key_ptr: *Key, value_ptr: *Value) void {
var overflow = self.*; var overflow = self.*;
var new_entry = Entry.new(key_ptr, value_ptr); var new_entry = Entry.new(allocator, key_ptr, value_ptr);
overflow.entries.*[index] = &new_entry; overflow.entries.*[index] = &new_entry;
} }
@ -59,7 +59,7 @@ pub fn RocDict(
} }
}; };
pub fn new(key_ptr: *Key, value_ptr: *Value) Entry { pub fn new(allocator: *Allocator, key_ptr: *Key, value_ptr: *Value) Entry {
return Entry{ return Entry{
.value_ptr = value_ptr, .value_ptr = value_ptr,
.key_ptr = key_ptr, .key_ptr = key_ptr,
@ -109,7 +109,7 @@ pub fn RocDict(
// represents if an new entry was // represents if an new entry was
// added (true), or if an old one // added (true), or if an old one
// was updated (false); // was updated (false);
pub fn insertIntoOverflow(self: *Entry, key_ptr: *Key, value_ptr: *Value, level: u64) bool { pub fn insertIntoOverflow(self: *Entry, allocator: *Allocator, key_ptr: *Key, value_ptr: *Value, level: u64) bool {
const key = key_ptr.*; const key = key_ptr.*;
const index = keyToIndexAtLevel(key, level); const index = keyToIndexAtLevel(key, level);
@ -118,7 +118,7 @@ pub fn RocDict(
// If there is no overflow, make a new OverFlow // If there is no overflow, make a new OverFlow
if (entry.overflow == null) { if (entry.overflow == null) {
var new_overflow: Overflow = Overflow.new(index, key_ptr, value_ptr); var new_overflow: Overflow = Overflow.new(allocator, index, key_ptr, value_ptr);
entry.overflow = new_overflow; entry.overflow = new_overflow;
@ -131,9 +131,7 @@ pub fn RocDict(
// If there is overflow, check if this index is taken // If there is overflow, check if this index is taken
// and if it isnt, than insert it // and if it isnt, than insert it
if (maybe_overflow_entry_ptr == null) { if (maybe_overflow_entry_ptr == null) {
var new_entry = Entry.new(key_ptr, value_ptr); overflow.setAtIndex(allocator, index, key_ptr, value_ptr);
overflow.setAtIndex(index, key_ptr, value_ptr);
return true; return true;
} else { } else {
var overflow_entry_ptr = maybe_overflow_entry_ptr.?; var overflow_entry_ptr = maybe_overflow_entry_ptr.?;
@ -148,7 +146,7 @@ pub fn RocDict(
overflow_entry.setValue(value_ptr); overflow_entry.setValue(value_ptr);
return false; return false;
} else { } else {
return overflow_entry.insertIntoOverflow(key_ptr, value_ptr, level + 1); return overflow_entry.insertIntoOverflow(allocator, key_ptr, value_ptr, level + 1);
} }
} }
} }
@ -219,7 +217,7 @@ pub fn RocDict(
} }
} }
pub fn insert(self: *Self, key_ptr: *Key, value_ptr: *Value) void { pub fn insert(self: *Self, allocator: *Allocator, key_ptr: *Key, value_ptr: *Value) void {
const level = 0; const level = 0;
const key = key_ptr.*; const key = key_ptr.*;
@ -229,7 +227,7 @@ pub fn RocDict(
const index = keyToIndexAtLevel(key, level); const index = keyToIndexAtLevel(key, level);
if (q.maybe_entry == null) { if (q.maybe_entry == null) {
var new_entry = Entry.new(value_ptr, key_ptr); var new_entry = Entry.new(allocator, value_ptr, key_ptr);
self.entries[index] = new_entry; self.entries[index] = new_entry;
@ -242,7 +240,7 @@ pub fn RocDict(
entry_ptr.setValue(value_ptr); entry_ptr.setValue(value_ptr);
self.entries[index] = entry; self.entries[index] = entry;
} else { } else {
const inserted_new_entry = entry.insertIntoOverflow(key_ptr, value_ptr, 0); const inserted_new_entry = entry.insertIntoOverflow(allocator, key_ptr, value_ptr, 0);
if (inserted_new_entry) { if (inserted_new_entry) {
self.len += 1; self.len += 1;
@ -300,7 +298,7 @@ test "RocDict.insert with hash collisions" {
var i: u64 = 0; var i: u64 = 0;
while (i < (size * 2)) { while (i < (size * 2)) {
dict.insert(&i, &i); dict.insert(testing.allocator, &i, &i);
i += 1; i += 1;
} }
@ -322,8 +320,8 @@ test "repeated RocDict.insert" {
var fst_val: u64 = 17; var fst_val: u64 = 17;
var snd_val: u64 = 49; var snd_val: u64 = 49;
dict.insert(&index, &fst_val); dict.insert(testing.allocator, &index, &fst_val);
dict.insert(&index, &snd_val); dict.insert(testing.allocator, &index, &snd_val);
var value_ptr: ?*u64 = dict.get(index); var value_ptr: ?*u64 = dict.get(index);
@ -342,8 +340,8 @@ test "RocDict.eq" {
var key: u64 = 0; var key: u64 = 0;
var value: u64 = 30; var value: u64 = 30;
fst.insert(&key, &value); fst.insert(testing.allocator, &key, &value);
snd.insert(&key, &value); snd.insert(testing.allocator, &key, &value);
assert(fst.eq(snd)); assert(fst.eq(snd));
var empty = RocDict(u64, u64).init(testing.allocator); var empty = RocDict(u64, u64).init(testing.allocator);
@ -351,7 +349,7 @@ test "RocDict.eq" {
var trd = RocDict(u64, u64).init(testing.allocator); var trd = RocDict(u64, u64).init(testing.allocator);
var new_value: u64 = value + 1; var new_value: u64 = value + 1;
trd.insert(&key, &new_value); trd.insert(testing.allocator, &key, &new_value);
assert(!fst.eq(trd)); assert(!fst.eq(trd));
} }
@ -362,18 +360,18 @@ test "RocDict.getLen" {
var key: u64 = 0; var key: u64 = 0;
var value: u64 = 16; var value: u64 = 16;
dict.insert(&key, &value); dict.insert(testing.allocator, &key, &value);
const expect_len: u64 = 1; const expect_len: u64 = 1;
expectEqual(dict.getLen(), expect_len); expectEqual(dict.getLen(), expect_len);
dict.insert(&key, &value); dict.insert(testing.allocator, &key, &value);
expectEqual(dict.getLen(), expect_len); expectEqual(dict.getLen(), expect_len);
var next_key: u64 = key + 1; var next_key: u64 = key + 1;
var next_value: u64 = 3; var next_value: u64 = 3;
dict.insert(&next_key, &next_value); dict.insert(testing.allocator, &next_key, &next_value);
var next_expected_len: u64 = 2; var next_expected_len: u64 = 2;
expectEqual(dict.getLen(), next_expected_len); expectEqual(dict.getLen(), next_expected_len);
@ -385,7 +383,7 @@ test "RocDict.insert" {
var index: u64 = 0; var index: u64 = 0;
var value: u64 = 30; var value: u64 = 30;
dict.insert(&index, &value); dict.insert(testing.allocator, &index, &value);
var result_value: ?*u64 = dict.get(index); var result_value: ?*u64 = dict.get(index);
expectEqual(result_value.?.*, value); expectEqual(result_value.?.*, value);