object.h

Go to the documentation of this file.

#pragma once
#include <stdio.h>
#include "kuroko.h"
#include "value.h"
#include "chunk.h"
#include "table.h"
 
#ifndef KRK_DISABLE_THREADS
#include <pthread.h>
#endif
 
typedef enum {
    KRK_OBJ_CODEOBJECT,
    KRK_OBJ_NATIVE,
    KRK_OBJ_CLOSURE,
    KRK_OBJ_STRING,
    KRK_OBJ_UPVALUE,
    KRK_OBJ_CLASS,
    KRK_OBJ_INSTANCE,
    KRK_OBJ_BOUND_METHOD,
    KRK_OBJ_TUPLE,
    KRK_OBJ_BYTES,
} KrkObjType;
 
#undef KrkObj
typedef struct KrkObj {
    uint16_t type;        
    uint16_t flags;       
    uint32_t hash;        
    struct KrkObj * next; 
} KrkObj;
 
#define KRK_OBJ_FLAGS_STRING_MASK   0x0003
#define KRK_OBJ_FLAGS_STRING_ASCII  0x0000
#define KRK_OBJ_FLAGS_STRING_UCS1   0x0001
#define KRK_OBJ_FLAGS_STRING_UCS2   0x0002
#define KRK_OBJ_FLAGS_STRING_UCS4   0x0003
 
#define KRK_OBJ_FLAGS_CODEOBJECT_COLLECTS_ARGS 0x0001
#define KRK_OBJ_FLAGS_CODEOBJECT_COLLECTS_KWS  0x0002
#define KRK_OBJ_FLAGS_CODEOBJECT_IS_GENERATOR  0x0004
#define KRK_OBJ_FLAGS_CODEOBJECT_IS_COROUTINE  0x0008
 
#define KRK_OBJ_FLAGS_FUNCTION_MASK                0x0003
#define KRK_OBJ_FLAGS_FUNCTION_IS_CLASS_METHOD     0x0001
#define KRK_OBJ_FLAGS_FUNCTION_IS_STATIC_METHOD    0x0002
 
#define KRK_OBJ_FLAGS_NO_INHERIT    0x0200
#define KRK_OBJ_FLAGS_SECOND_CHANCE 0x0100
#define KRK_OBJ_FLAGS_IS_MARKED     0x0010
#define KRK_OBJ_FLAGS_IN_REPR       0x0020
#define KRK_OBJ_FLAGS_IMMORTAL      0x0040
#define KRK_OBJ_FLAGS_VALID_HASH    0x0080
 
 
typedef enum {
    /* For compatibility */
    KRK_STRING_ASCII   = KRK_OBJ_FLAGS_STRING_ASCII,  
    KRK_STRING_UCS1    = KRK_OBJ_FLAGS_STRING_UCS1,   
    KRK_STRING_UCS2    = KRK_OBJ_FLAGS_STRING_UCS2,   
    KRK_STRING_UCS4    = KRK_OBJ_FLAGS_STRING_UCS4,   
} KrkStringType;
 
typedef struct KrkString {
    KrkObj obj;          
    size_t length;       
    size_t codesLength;  
    char * chars;        
    void * codes;        
} KrkString;
 
typedef struct {
    KrkObj obj;       
    size_t length;    
    uint8_t * bytes;  
} KrkBytes;
 
typedef struct KrkUpvalue {
    KrkObj obj;                    
    int location;                  
    KrkValue   closed;             
    struct KrkUpvalue * next;      
    struct KrkThreadState * owner; 
} KrkUpvalue;
 
typedef struct {
    size_t id;        
    size_t birthday;  
    size_t deathday;  
    KrkString * name; 
} KrkLocalEntry;
 
typedef struct {
    uint32_t bytecodeOffset;
    uint8_t start;
    uint8_t midStart;
    uint8_t midEnd;
    uint8_t end;
} KrkExpressionsMap;
 
struct KrkInstance;
 
typedef struct {
    uint32_t instructionOffset;   
    uint16_t intendedTarget;      
    uint8_t  originalOpcode;      
} KrkOverlongJump;
 
typedef struct {
    KrkObj obj;                            
    unsigned short requiredArgs;           
    unsigned short keywordArgs;            
    unsigned short potentialPositionals;   
    unsigned short totalArguments;         
    size_t upvalueCount;                   
    KrkChunk chunk;                        
    KrkString * name;                      
    KrkString * docstring;                 
    KrkValueArray positionalArgNames;      
    KrkValueArray keywordArgNames;         
    size_t localNameCapacity;              
    size_t localNameCount;                 
    KrkLocalEntry * localNames;            
    KrkString * qualname;                  
    size_t expressionsCapacity;            
    size_t expressionsCount;               
    KrkExpressionsMap * expressions;       
    KrkValue jumpTargets;                  
    KrkOverlongJump * overlongJumps;       
    size_t overlongJumpsCapacity;          
    size_t overlongJumpsCount;             
} KrkCodeObject;
 
 
typedef struct {
    KrkObj obj;                
    KrkCodeObject * function;  
    KrkUpvalue ** upvalues;    
    size_t upvalueCount;       
    KrkValue annotations;      
    KrkTable fields;           
    KrkValue globalsOwner;     
    KrkTable * globalsTable;   
} KrkClosure;
 
typedef void (*KrkCleanupCallback)(struct KrkInstance *);
 
typedef struct KrkClass {
    KrkObj obj;               
    struct KrkClass * _class; 
    KrkTable methods;         
    KrkString * name;         
    KrkString * filename;     
    struct KrkClass * base;   
    size_t allocSize;         
    KrkCleanupCallback _ongcscan;   
    KrkCleanupCallback _ongcsweep;  
    KrkTable subclasses;      
    KrkObj * _getter;         
    KrkObj * _setter;         
    KrkObj * _reprer;         
    KrkObj * _tostr;          
    KrkObj * _call;           
    KrkObj * _init;           
    KrkObj * _eq;             
    KrkObj * _len;            
    KrkObj * _enter;          
    KrkObj * _exit;           
    KrkObj * _delitem;        
    KrkObj * _iter;           
    KrkObj * _getattr;        
    KrkObj * _dir;            
    KrkObj * _contains;       
    KrkObj * _descget;        
    KrkObj * _descset;        
    KrkObj * _classgetitem;   
    KrkObj * _hash;           
    KrkObj * _add, * _radd, * _iadd;
    KrkObj * _sub, * _rsub, * _isub;
    KrkObj * _mul, * _rmul, * _imul;
    KrkObj * _or,  * _ror,  * _ior;
    KrkObj * _xor, * _rxor, * _ixor;
    KrkObj * _and, * _rand, * _iand;
    KrkObj * _mod, * _rmod, * _imod;
    KrkObj * _pow, * _rpow, * _ipow;
    KrkObj * _lshift, * _rlshift, * _ilshift;
    KrkObj * _rshift, * _rrshift, * _irshift;
    KrkObj * _truediv, * _rtruediv, * _itruediv;
    KrkObj * _floordiv, * _rfloordiv, * _ifloordiv;
 
    KrkObj * _lt, * _gt, * _le, * _ge;
    KrkObj * _invert, * _negate;
    KrkObj * _set_name;
    KrkObj * _matmul, * _rmatmul, * _imatmul;
    KrkObj * _pos;
    KrkObj * _setattr;
    KrkObj * _format;
    KrkObj * _new;
    KrkObj * _bool;
 
    size_t cacheIndex;
} KrkClass;
 
typedef struct KrkInstance {
    KrkObj obj;         
    KrkClass * _class;  
    KrkTable fields;    
} KrkInstance;
 
typedef struct {
    KrkObj obj;         
    KrkValue receiver;  
    KrkObj * method;    
} KrkBoundMethod;
 
typedef KrkValue (*NativeFn)(int argCount, const KrkValue* args, int hasKwargs);
 
typedef struct {
    KrkObj obj;           
    NativeFn function;    
    const char * name;    
    const char * doc;     
} KrkNative;
 
typedef struct {
    KrkObj obj;            
    KrkValueArray values;  
} KrkTuple;
 
typedef struct {
    KrkInstance inst;     
    KrkValueArray values; 
#ifndef KRK_DISABLE_THREADS
    pthread_rwlock_t rwlock;
#endif
} KrkList;
 
typedef struct {
    KrkInstance inst;  
    KrkTable entries;  
} KrkDict;
 
struct DictItems {
    KrkInstance inst;
    KrkValue dict;
    size_t i;
};
 
struct DictKeys {
    KrkInstance inst;
    KrkValue dict;
    size_t i;
};
 
struct DictValues {
    KrkInstance inst;
    KrkValue dict;
    size_t i;
};
 
struct KrkModule {
    KrkInstance inst;
#ifndef KRK_STATIC_ONLY
    krk_dlRefType libHandle;
#endif
};
 
struct KrkSlice {
    KrkInstance inst;
    KrkValue    start;
    KrkValue    end;
    KrkValue    step;
};
 
extern KrkString * krk_takeString(char * chars, size_t length);
 
extern KrkString * krk_takeStringVetted(char * chars, size_t length, size_t codesLength, KrkStringType type, uint32_t hash);
 
extern KrkString * krk_copyString(const char * chars, size_t length);
 
extern void * krk_unicodeString(KrkString * string);
 
extern uint32_t krk_unicodeCodepoint(KrkString * string, size_t index);
 
extern size_t krk_codepointToBytes(krk_integer_type value, unsigned char * out);
 
extern KrkInstance * krk_buildGenerator(KrkClosure * function, KrkValue * arguments, size_t argCount);
 
extern int krk_getAwaitable(void);
 
extern NativeFn krk_GenericAlias;
 
extern KrkCodeObject *    krk_newCodeObject(void);
 
extern KrkNative *      krk_newNative(NativeFn function, const char * name, int type);
 
extern KrkClosure *     krk_newClosure(KrkCodeObject * function, KrkValue globals);
 
extern KrkUpvalue *     krk_newUpvalue(int slot);
 
extern KrkClass *       krk_newClass(KrkString * name, KrkClass * base);
 
extern KrkInstance *    krk_newInstance(KrkClass * _class);
 
extern KrkBoundMethod * krk_newBoundMethod(KrkValue receiver, KrkObj * method);
 
extern KrkTuple *       krk_newTuple(size_t length);
 
extern KrkBytes *       krk_newBytes(size_t length, uint8_t * source);
 
#define krk_isObjType(v,t) (IS_OBJECT(v) && (AS_OBJECT(v)->type == (t)))
#define OBJECT_TYPE(value) (AS_OBJECT(value)->type)
#define IS_STRING(value)   krk_isObjType(value, KRK_OBJ_STRING)
#define AS_STRING(value)   ((KrkString *)AS_OBJECT(value))
#define AS_CSTRING(value)  (((KrkString *)AS_OBJECT(value))->chars)
#define IS_BYTES(value)    krk_isObjType(value, KRK_OBJ_BYTES)
#define AS_BYTES(value)    ((KrkBytes*)AS_OBJECT(value))
#define IS_NATIVE(value)   krk_isObjType(value, KRK_OBJ_NATIVE)
#define AS_NATIVE(value)   ((KrkNative *)AS_OBJECT(value))
#define IS_CLOSURE(value)  krk_isObjType(value, KRK_OBJ_CLOSURE)
#define AS_CLOSURE(value)  ((KrkClosure *)AS_OBJECT(value))
#define IS_CLASS(value)    krk_isObjType(value, KRK_OBJ_CLASS)
#define AS_CLASS(value)    ((KrkClass *)AS_OBJECT(value))
#define IS_INSTANCE(value) krk_isObjType(value, KRK_OBJ_INSTANCE)
#define AS_INSTANCE(value) ((KrkInstance *)AS_OBJECT(value))
#define IS_BOUND_METHOD(value) krk_isObjType(value, KRK_OBJ_BOUND_METHOD)
#define AS_BOUND_METHOD(value) ((KrkBoundMethod*)AS_OBJECT(value))
#define IS_TUPLE(value)    krk_isObjType(value, KRK_OBJ_TUPLE)
#define AS_TUPLE(value)    ((KrkTuple *)AS_OBJECT(value))
#define AS_LIST(value) (&((KrkList *)AS_OBJECT(value))->values)
#define AS_DICT(value) (&((KrkDict *)AS_OBJECT(value))->entries)
 
#define IS_codeobject(value) krk_isObjType(value, KRK_OBJ_CODEOBJECT)
#define AS_codeobject(value) ((KrkCodeObject *)AS_OBJECT(value))