add bits lib

This commit is contained in:
ValKmjolnir 2021-10-28 21:49:08 +08:00
parent fd0d836c03
commit 4bfce37f40
5 changed files with 209 additions and 244 deletions

152
lib.nas
View File

@ -1,125 +1,42 @@
var import=func(filename)
{
return __builtin_import(filename);
}
var print=func(elems...)
{
return __builtin_print(elems);
};
var println=func(elems...)
{
var import= func(filename){return __builtin_import(filename);}
var print= func(elems...){return __builtin_print(elems);}
var append= func(vec,elems...){return __builtin_append(vec,elems);}
var setsize= func(vec,size){return __builtin_setsize(vec,size);}
var system= func(str){return __builtin_system(str);}
var input= func(){return __builtin_input();}
var sleep= func(duration){return __builtin_sleep(duration);}
var split= func(deli,str){return __builtin_split(deli,str);}
var rand= func(seed=nil){return __builtin_rand(seed);}
var id= func(object){return __builtin_id(object);}
var int= func(val){return __builtin_int(val);}
var num= func(val){return __builtin_num(val);}
var pop= func(vec){return __builtin_pop(vec);}
var str= func(num){return __builtin_str(num);}
var size= func(object){return __builtin_size(object);}
var contains= func(hash,key){return __builtin_contains(hash,key);}
var delete= func(hash,key){return __builtin_delete(hash,key);}
var keys= func(hash){return __builtin_keys(hash);}
var time= func(begin){return __builtin_time(begin);}
var die= func(str){return __builtin_die(str);}
var typeof= func(object){return __builtin_type(object);}
var substr= func(str,begin,len){return __builtin_substr(str,begin,len);}
var streq= func(a,b){return __builtin_streq(a,b);}
var left= func(str,len){return __builtin_left(str,len);}
var right= func(str,len){return __builtin_right(str,len);}
var cmp= func(a,b){return __builtin_cmp(a,b);}
var chr= func(code){return __builtin_chr(code);}
var println=func(elems...){
__builtin_print(elems);
elems=['\n'];
return __builtin_print(elems);
}
var append=func(vec,elems...)
{
return __builtin_append(vec,elems);
}
var setsize=func(vec,size)
{
return __builtin_setsize(vec,size);
}
var system=func(str)
{
return __builtin_system(str);
}
var input=func()
{
return __builtin_input();
}
var sleep=func(duration)
{
return __builtin_sleep(duration);
}
var split=func(deli,str)
{
return __builtin_split(deli,str);
}
var rand=func(seed=nil)
{
return __builtin_rand(seed);
}
var id=func(object)
{
return __builtin_id(object);
}
var int=func(val)
{
return __builtin_int(val);
}
var num=func(val)
{
return __builtin_num(val);
}
var pop=func(vec)
{
return __builtin_pop(vec);
}
var str=func(num)
{
return __builtin_str(num);
}
var size=func(object)
{
return __builtin_size(object);
}
var contains=func(hash,key)
{
return __builtin_contains(hash,key);
}
var delete=func(hash,key)
{
return __builtin_delete(hash,key);
}
var keys=func(hash)
{
return __builtin_keys(hash);
}
var time=func(begin_time)
{
return __builtin_time(begin_time);
}
var die=func(str)
{
return __builtin_die(str);
}
var typeof=func(object)
{
return __builtin_type(object);
}
var substr=func(str,begin,len)
{
return __builtin_substr(str,begin,len);
}
var streq=func(a,b)
{
return __builtin_streq(a,b);
}
var left=func(str,len)
{
return __builtin_left(str,len);
}
var right=func(str,len)
{
return __builtin_right(str,len);
}
var cmp=func(a,b)
{
return __builtin_cmp(a,b);
}
var chr=func(code)
{
return __builtin_chr(code);
}
var io=
{
fin: func(filename){return __builtin_fin(filename);},
fout:func(filename,str){return __builtin_fout(filename,str);},
SEEK_SET:0,
SEEK_CUR:1,
SEEK_END:2,
fin: func(filename){return __builtin_fin(filename);},
fout: func(filename,str){return __builtin_fout(filename,str);},
open: func(filename,mode="r"){return __builtin_open(filename,mode);},
close: func(filehandle){return __builtin_close(filehandle);},
read: func(filehandle,buf,len){return __builtin_read(filehandle,buf,len);},
@ -127,7 +44,8 @@ var io=
seek: func(filehandle,pos,whence){return __builtin_seek(filehandle,pos,whence);},
tell: func(filehandle){return __builtin_tell(filehandle);},
readln:func(filehandle){return __builtin_readln(filehandle);},
stat:func(filename){return __builtin_stat(filename);}
stat: func(filename){return __builtin_stat(filename);},
eof: func(filehandle){return __builtin_eof(filehandle);}
};
var bits=
@ -136,7 +54,11 @@ var bits=
bitand: func(a,b){return __builtin_and(a,b); },
bitor: func(a,b){return __builtin_or(a,b); },
bitnand: func(a,b){return __builtin_nand(a,b);},
bitnot: func(a) {return __builtin_not(a); }
bitnot: func(a) {return __builtin_not(a); },
fld: func(str,startbit,len){return __builtin_fld;},
sfld: func(str,startbit,len){return __builtin_sfld;},
setfld: func(str,startbit,len,val){return __builtin_setfld;},
buf: func(len){return __builtin_buf;}
};
var math=

View File

@ -64,6 +64,11 @@ nas_native(builtin_seek);
nas_native(builtin_tell);
nas_native(builtin_readln);
nas_native(builtin_stat);
nas_native(builtin_eof);
nas_native(builtin_fld);
nas_native(builtin_sfld);
nas_native(builtin_setfld);
nas_native(builtin_buf);
nasal_ref builtin_err(const char* func_name,std::string info)
{
@ -129,6 +134,11 @@ struct
{"__builtin_tell", builtin_tell },
{"__builtin_readln", builtin_readln },
{"__builtin_stat", builtin_stat },
{"__builtin_eof", builtin_eof },
{"__builtin_fld", builtin_fld },
{"__builtin_sfld", builtin_sfld },
{"__builtin_setfld", builtin_setfld },
{"__builtin_buf", builtin_buf },
{nullptr, nullptr }
};
@ -745,7 +755,19 @@ nasal_ref builtin_readln(std::vector<nasal_ref>& local,nasal_gc& gc)
nasal_ref filehandle=local[1];
if(filehandle.type!=vm_obj || filehandle.obj()->type!=obj_file)
return builtin_err("readln","not a correct filehandle");
/* unfinished */
nasal_ref str=gc.alloc(vm_str);
auto& s=*str.str();
char c;
while((c=fgetc((FILE*)filehandle.obj()->ptr))!=EOF)
{
if(c=='\r')
continue;
if(c=='\n')
return str;
s+=c;
}
if(s.length())
return str;
return gc.nil;
}
nasal_ref builtin_stat(std::vector<nasal_ref>& local,nasal_gc& gc)
@ -772,4 +794,100 @@ nasal_ref builtin_stat(std::vector<nasal_ref>& local,nasal_gc& gc)
};
return ret;
}
nasal_ref builtin_eof(std::vector<nasal_ref>& local,nasal_gc& gc)
{
nasal_ref filehandle=local[1];
if(filehandle.type!=vm_obj || filehandle.obj()->type!=obj_file)
return builtin_err("readln","not a correct filehandle");
double res=feof((FILE*)filehandle.obj()->ptr);
return {vm_num,res};
}
nasal_ref builtin_fld(std::vector<nasal_ref>& local,nasal_gc& gc)
{
// bits.fld(s,0,3);
// if s stores 10100010(162)
// will get 101(5)
nasal_ref str=local[1];
nasal_ref startbit=local[2];
nasal_ref length=local[3];
if(str.type!=vm_str || str.value.gcobj->unmut)
return builtin_err("fld","\"str\" must be mutable string");
if(startbit.type!=vm_num || length.type!=vm_num)
return builtin_err("fld","\"startbit\",\"len\" must be number");
uint32_t bit=(uint32_t)startbit.num();
uint32_t len=(uint32_t)length.num();
if(bit+len>8*str.str()->length())
return builtin_err("fld","bitfield out of bounds");
uint32_t res=0;
auto& s=*str.str();
for(uint32_t i=bit;i<bit+len;++i)
if(s[i>>3]&(1<<(7-(i&7))))
res|=1<<(bit+len-i-1);
return {vm_num,(double)res};
}
nasal_ref builtin_sfld(std::vector<nasal_ref>& local,nasal_gc& gc)
{
// bits.sfld(s,0,3);
// if s stores 10100010(162)
// will get 101(5) then this will be signed extended to
// 11111101(-3)
nasal_ref str=local[1];
nasal_ref startbit=local[2];
nasal_ref length=local[3];
if(str.type!=vm_str || str.value.gcobj->unmut)
return builtin_err("sfld","\"str\" must be mutable string");
if(startbit.type!=vm_num || length.type!=vm_num)
return builtin_err("sfld","\"startbit\",\"len\" must be number");
uint32_t bit=(uint32_t)startbit.num();
uint32_t len=(uint32_t)length.num();
if(bit+len>8*str.str()->length())
return builtin_err("sfld","bitfield out of bounds");
uint32_t res=0;
auto& s=*str.str();
for(uint32_t i=bit;i<bit+len;++i)
if(s[i>>3]&(1<<(7-(i&7))))
res|=1<<(bit+len-i-1);
if(res&(1<<(len-1)))
res|=~((1<<len)-1);
return {vm_num,(double)((int32_t)res)};
}
nasal_ref builtin_setfld(std::vector<nasal_ref>& local,nasal_gc& gc)
{
// bits.setfld(s,0,8,69);
// set 1000101(69) to string will get this:
// 10100010(162)
// so s[0]=162
nasal_ref str=local[1];
nasal_ref startbit=local[2];
nasal_ref length=local[3];
nasal_ref value=local[4];
if(str.type!=vm_str || str.value.gcobj->unmut)
return builtin_err("setfld","\"str\" must be mutable string");
if(startbit.type!=vm_num || length.type!=vm_num || value.type!=vm_num)
return builtin_err("setfld","\"startbit\",\"len\",\"val\" must be number");
uint32_t bit=(uint32_t)startbit.num();
uint32_t len=(uint32_t)length.num();
uint64_t val=(uint64_t)value.num();
if(bit+len>8*str.str()->length())
return builtin_err("setfld","bitfield out of bounds");
auto& s=*str.str();
for(uint32_t i=bit;i<bit+len;++i)
{
if(val&(1<<(i-bit)))
s[i>>3]|=(1<<(7-(i&7)));
else
s[i>>3]&=~(1<<(7-(i&7)));
}
return gc.nil;
}
nasal_ref builtin_buf(std::vector<nasal_ref>& local,nasal_gc& gc)
{
nasal_ref length=local[1];
if(length.type!=vm_num || length.num()<=0)
return builtin_err("buf","\"len\" must be a number greater than 9");
nasal_ref str=gc.alloc(vm_str);
auto& s=*str.str();
s.resize(length.num(),'\0');
return str;
}
#endif

View File

@ -115,6 +115,7 @@ struct nasal_val
{
uint8_t mark;
uint8_t type;
uint8_t unmut;
union
{
std::string* str;
@ -262,6 +263,7 @@ nasal_val::nasal_val(uint8_t val_type)
{
mark=GC_COLLECTED;
type=val_type;
unmut=0;
switch(val_type)
{
case vm_str: ptr.str=new std::string; break;
@ -400,6 +402,7 @@ void nasal_gc::init(const std::vector<std::string>& s)
for(uint32_t i=0;i<strs.size();++i)
{
strs[i]={vm_str,new nasal_val(vm_str)};
strs[i].value.gcobj->unmut=1;
*strs[i].str()=s[i];
}
}

View File

@ -607,7 +607,7 @@ inline void nasal_vm::opr_callv()
int str_size=str.length();
if(num<-str_size || num>=str_size)
die("callv: index out of range:"+std::to_string(val.to_number()));
gc.top[0]={vm_num,static_cast<double>(str[num>=0? num:num+str_size])};
gc.top[0]={vm_num,double((uint8_t)str[num>=0? num:num+str_size])};
}
else
die("callv: must call a vector/hash/string");

View File

@ -1,125 +1,42 @@
var import=func(filename)
{
return __builtin_import(filename);
}
var print=func(elems...)
{
return __builtin_print(elems);
};
var println=func(elems...)
{
var import= func(filename){return __builtin_import(filename);}
var print= func(elems...){return __builtin_print(elems);}
var append= func(vec,elems...){return __builtin_append(vec,elems);}
var setsize= func(vec,size){return __builtin_setsize(vec,size);}
var system= func(str){return __builtin_system(str);}
var input= func(){return __builtin_input();}
var sleep= func(duration){return __builtin_sleep(duration);}
var split= func(deli,str){return __builtin_split(deli,str);}
var rand= func(seed=nil){return __builtin_rand(seed);}
var id= func(object){return __builtin_id(object);}
var int= func(val){return __builtin_int(val);}
var num= func(val){return __builtin_num(val);}
var pop= func(vec){return __builtin_pop(vec);}
var str= func(num){return __builtin_str(num);}
var size= func(object){return __builtin_size(object);}
var contains= func(hash,key){return __builtin_contains(hash,key);}
var delete= func(hash,key){return __builtin_delete(hash,key);}
var keys= func(hash){return __builtin_keys(hash);}
var time= func(begin){return __builtin_time(begin);}
var die= func(str){return __builtin_die(str);}
var typeof= func(object){return __builtin_type(object);}
var substr= func(str,begin,len){return __builtin_substr(str,begin,len);}
var streq= func(a,b){return __builtin_streq(a,b);}
var left= func(str,len){return __builtin_left(str,len);}
var right= func(str,len){return __builtin_right(str,len);}
var cmp= func(a,b){return __builtin_cmp(a,b);}
var chr= func(code){return __builtin_chr(code);}
var println=func(elems...){
__builtin_print(elems);
elems=['\n'];
return __builtin_print(elems);
}
var append=func(vec,elems...)
{
return __builtin_append(vec,elems);
}
var setsize=func(vec,size)
{
return __builtin_setsize(vec,size);
}
var system=func(str)
{
return __builtin_system(str);
}
var input=func()
{
return __builtin_input();
}
var sleep=func(duration)
{
return __builtin_sleep(duration);
}
var split=func(deli,str)
{
return __builtin_split(deli,str);
}
var rand=func(seed=nil)
{
return __builtin_rand(seed);
}
var id=func(object)
{
return __builtin_id(object);
}
var int=func(val)
{
return __builtin_int(val);
}
var num=func(val)
{
return __builtin_num(val);
}
var pop=func(vec)
{
return __builtin_pop(vec);
}
var str=func(num)
{
return __builtin_str(num);
}
var size=func(object)
{
return __builtin_size(object);
}
var contains=func(hash,key)
{
return __builtin_contains(hash,key);
}
var delete=func(hash,key)
{
return __builtin_delete(hash,key);
}
var keys=func(hash)
{
return __builtin_keys(hash);
}
var time=func(begin_time)
{
return __builtin_time(begin_time);
}
var die=func(str)
{
return __builtin_die(str);
}
var typeof=func(object)
{
return __builtin_type(object);
}
var substr=func(str,begin,len)
{
return __builtin_substr(str,begin,len);
}
var streq=func(a,b)
{
return __builtin_streq(a,b);
}
var left=func(str,len)
{
return __builtin_left(str,len);
}
var right=func(str,len)
{
return __builtin_right(str,len);
}
var cmp=func(a,b)
{
return __builtin_cmp(a,b);
}
var chr=func(code)
{
return __builtin_chr(code);
}
var io=
{
fin: func(filename){return __builtin_fin(filename);},
fout:func(filename,str){return __builtin_fout(filename,str);},
SEEK_SET:0,
SEEK_CUR:1,
SEEK_END:2,
fin: func(filename){return __builtin_fin(filename);},
fout: func(filename,str){return __builtin_fout(filename,str);},
open: func(filename,mode="r"){return __builtin_open(filename,mode);},
close: func(filehandle){return __builtin_close(filehandle);},
read: func(filehandle,buf,len){return __builtin_read(filehandle,buf,len);},
@ -127,7 +44,8 @@ var io=
seek: func(filehandle,pos,whence){return __builtin_seek(filehandle,pos,whence);},
tell: func(filehandle){return __builtin_tell(filehandle);},
readln:func(filehandle){return __builtin_readln(filehandle);},
stat:func(filename){return __builtin_stat(filename);}
stat: func(filename){return __builtin_stat(filename);},
eof: func(filehandle){return __builtin_eof(filehandle);}
};
var bits=
@ -136,7 +54,11 @@ var bits=
bitand: func(a,b){return __builtin_and(a,b); },
bitor: func(a,b){return __builtin_or(a,b); },
bitnand: func(a,b){return __builtin_nand(a,b);},
bitnot: func(a) {return __builtin_not(a); }
bitnot: func(a) {return __builtin_not(a); },
fld: func(str,startbit,len){return __builtin_fld;},
sfld: func(str,startbit,len){return __builtin_sfld;},
setfld: func(str,startbit,len,val){return __builtin_setfld;},
buf: func(len){return __builtin_buf;}
};
var math=