quakeforge/tools/qfcc/source/class.c
Bill Currie e1d7854af5 [qfcc] Rename G_* macros to Q_*
They clash with those in progs.h, which is needed for using progs code
in the compiler.
2022-11-16 17:53:21 +09:00

2027 lines
49 KiB
C
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

/*
class.c
QC class support code
Copyright (C) 2002 Bill Currie
Author: Bill Currie <bill@taniwha.org>
Date: 2002/5/7
This program is free software; you can redistribute it and/or
modify it under the terms of the GNU General Public License
as published by the Free Software Foundation; either version 2
of the License, or (at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
See the GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program; if not, write to:
Free Software Foundation, Inc.
59 Temple Place - Suite 330
Boston, MA 02111-1307, USA
*/
#ifdef HAVE_CONFIG_H
# include "config.h"
#endif
#ifdef HAVE_STRING_H
# include <string.h>
#endif
#ifdef HAVE_STRINGS_H
# include <strings.h>
#endif
#include <stdlib.h>
#include "QF/dstring.h"
#include "QF/hash.h"
#include "QF/va.h"
#include "QF/progs/pr_obj.h"
#include "tools/qfcc/include/qfcc.h"
#include "tools/qfcc/include/codespace.h"
#include "tools/qfcc/include/class.h"
#include "tools/qfcc/include/def.h"
#include "tools/qfcc/include/defspace.h"
#include "tools/qfcc/include/diagnostic.h"
#include "tools/qfcc/include/emit.h"
#include "tools/qfcc/include/expr.h"
#include "tools/qfcc/include/method.h"
#include "tools/qfcc/include/options.h"
#include "tools/qfcc/include/reloc.h"
#include "tools/qfcc/include/shared.h"
#include "tools/qfcc/include/strpool.h"
#include "tools/qfcc/include/struct.h"
#include "tools/qfcc/include/symtab.h"
#include "tools/qfcc/include/type.h"
#include "tools/qfcc/include/value.h"
static hashtab_t *class_hash;
static hashtab_t *category_hash;
static hashtab_t *protocol_hash;
static hashtab_t *static_instances;
static hashtab_t *static_instance_classes;
// these will be built up further
type_t type_selector = {
.type = ev_invalid,
.meta = ty_struct,
};
type_t type_SEL = {
.type = ev_ptr,
.name = "SEL",
.alignment = 1,
.width = 1,
.meta = ty_basic,
{{&type_selector}},
};
type_t *IMP_params[] = { &type_id, &type_SEL };
type_t type_IMP = {
.type = ev_func,
.name = "IMP",
.alignment = 1,
.width = 1,
.meta = ty_basic,
{{&type_id, -3, IMP_params, .no_va_list = 1}},
};
type_t type_super = {
.type = ev_invalid,
};
type_t type_SuperPtr = {
.type = ev_ptr,
.alignment = 1,
.width = 1,
.meta = ty_basic,
{{&type_super}},
};
type_t *supermsg_params[] = { &type_SuperPtr, &type_SEL };
type_t type_supermsg = {
.type = ev_func,
.name = ".supermsg",
.alignment = 1,
.width = 1,
.meta = ty_basic,
{{&type_id, -3, supermsg_params}},
};
type_t type_method = {
.type = ev_invalid,
.meta = ty_struct,
};
type_t type_method_description = {
.type = ev_invalid,
.meta = ty_struct,
};
type_t type_category = {
.type = ev_invalid,
.meta = ty_struct,
};
type_t type_ivar = {
.type = ev_invalid,
.meta = ty_struct,
};
type_t type_module = {
.type = ev_invalid,
.meta = ty_struct,
};
type_t type_moduleptr = {
.type = ev_ptr,
.alignment = 1,
.width = 1,
.meta = ty_basic,
{{&type_module}},
};
type_t *obj_exec_class_params[] = {
&type_moduleptr,
};
type_t type_exec_class = {
.type = ev_func,
.alignment = 1,
.width = 1,
.meta = ty_basic,
{{&type_void, 1, obj_exec_class_params}},
};
// the cast of 1 in the init is to ensure pointers to incomplete types
// are never misidentified as id. It will be set to the correct value
// when the obj system is initialized.
type_t type_object = {
.type = ev_invalid,
.meta = ty_struct,
{{(type_t *)1}},
};
type_t type_id = {
.type = ev_ptr,
.name = "id",
.alignment = 1,
.width = 1,
.meta = ty_basic,
{{&type_object}},
};
type_t type_class = {
.type = ev_invalid,
.meta = ty_struct,
};
type_t type_Class = {
.type = ev_ptr,
.alignment = 1,
.width = 1,
.meta = ty_basic,
{{&type_class}},
};
type_t type_protocol = {
.type = ev_invalid,
.meta = ty_struct,
};
int obj_initialized = 0;
static struct_def_t sel_struct[] = {
{"sel_id", &type_string},
{"sel_types", &type_string},
{0, 0}
};
static struct_def_t method_struct[] = {
{"method_name", &type_SEL},
{"method_types", &type_string},
{"method_imp", &type_IMP},
{0, 0}
};
static struct_def_t method_desc_struct[] = {
{"name", &type_string},
{"types", &type_string},
{0, 0}
};
static struct_def_t category_struct[] = {
{"category_name", &type_string},
{"class_name", &type_string},
{"instance_methods", &type_ptr},
{"class_methods", &type_ptr},
{"protocols", &type_ptr},
{0, 0}
};
static struct_def_t ivar_struct[] = {
{"ivar_name", &type_string},
{"ivar_type", &type_string},
{"ivar_offset", &type_int},
{0, 0}
};
static struct_def_t super_struct[] = {
{"self", &type_id},
{"class", &type_Class},
{0, 0}
};
static struct_def_t module_struct[] = {
{"version", &type_int},
{"size", &type_int},
{"name", &type_string},
{"symtab", &type_ptr},
{0, 0}
};
static struct_def_t class_struct[] = {
{"class_pointer", &type_Class},
{"super_class", &type_Class},
{"name", &type_string},
{"version", &type_int},
{"info", &type_int},
{"instance_size", &type_int},
{"ivars", &type_ptr},
{"methods", &type_ptr},
{"dtable", &type_ptr},
{"subclass_list", &type_ptr},
{"sibling_class", &type_ptr},
{"protocols", &type_ptr},
{"gc_object_type", &type_ptr},
{0, 0}
};
static struct_def_t protocol_struct[] = {
{"class_pointer", &type_Class},
{"protocol_name", &type_string},
{"protocol_list", &type_ptr},
{"instance_methods", &type_ptr},
{"class_methods", &type_ptr},
{0, 0}
};
static struct_def_t object_struct[] = {
{"class_pointer", &type_Class},
{0, 0}
};
static const char *
static_instance_get_key (const void *instance, void *unused)
{
return ((static_instance_t *) instance)->class;
}
static void
add_static_instance (const char *class, def_t *instance_def)
{
static_instance_t *instance = malloc (sizeof (*instance));
if (!static_instances) {
static_instances = Hash_NewTable (1021, static_instance_get_key,
0, 0, 0);
static_instance_classes = Hash_NewTable (1021, static_instance_get_key,
0, 0, 0);
}
instance->class = save_string (class);
instance->instance = instance_def;
Hash_Add (static_instances, instance);
// uniqued set of class names for all static instances
if (!Hash_Find (static_instance_classes, class)) {
Hash_Add (static_instance_classes, instance);
}
}
typedef struct {
const char *class_name;
int num_instances;
static_instance_t **instances;
} obj_static_instances_data_t;
static void
emit_instance_classname (def_t *def, void *data, int index)
{
obj_static_instances_data_t *da = (obj_static_instances_data_t *)data;
if (!is_string(def->type))
internal_error (0, "%s: expected string def", __FUNCTION__);
EMIT_STRING (def->space, D_STRING (def), da->class_name);
}
static void
emit_instance_defs (def_t *def, void *data, int index)
{
obj_static_instances_data_t *da = (obj_static_instances_data_t *)data;
if (!is_array (def->type) || def->type->t.array.type->type != ev_ptr)
internal_error (0, "%s: expected array of pointers def", __FUNCTION__);
if (index < 0 || index >= da->num_instances + 1)
internal_error (0, "%s: out of bounds index: %d %d",
__FUNCTION__, index, da->num_instances + 1);
D_INT (def) = 0;
if (index < da->num_instances) {
EMIT_DEF (def->space, D_INT (def), da->instances[index]->instance);
}
}
static def_t *
emit_static_instances (const char *classname)
{
static struct_def_t instances_struct[] = {
{"class_name", &type_string, emit_instance_classname},
{"instances", 0, emit_instance_defs},
{0, 0}
};
obj_static_instances_data_t data = {};
def_t *instances_def;
data.class_name = classname;
data.instances = (static_instance_t **) Hash_FindList (static_instances,
classname);
for (static_instance_t **inst = data.instances; *inst; inst++) {
data.num_instances++;
}
instances_struct[1].type = array_type (&type_ptr, data.num_instances + 1);
instances_def = emit_structure (va (0, "_OBJ_STATIC_INSTANCES_%s",
classname),
's', instances_struct, 0, &data,
0, sc_static);
free (data.instances);
return instances_def;
}
static def_t *
emit_static_instances_list (void)
{
static_instance_t **classes;
int num_classes = 0;
def_t **instance_lists;
type_t *instance_lists_type;
symbol_t *instance_lists_sym;
def_t *instance_lists_def;
pr_ptr_t *list;
defspace_t *space;
if (!static_instance_classes || !static_instances) {
return 0;
}
classes = (static_instance_t **) Hash_GetList (static_instance_classes);
for (static_instance_t **c = classes; *c; c++) {
num_classes++;
}
if (!num_classes) {
free (classes);
return 0;
}
instance_lists = alloca (num_classes * sizeof (*instance_lists));
for (int i = 0; i < num_classes; i++) {
instance_lists[i] = emit_static_instances (classes[i]->class);
}
free (classes);
// +1 for terminating null
instance_lists_type = array_type (&type_ptr, num_classes + 1);
instance_lists_sym = make_symbol ("_OBJ_STATIC_INSTANCES",
instance_lists_type,
pr.far_data, sc_static);
if (!instance_lists_sym->table) {
symtab_addsymbol (pr.symtab, instance_lists_sym);
}
instance_lists_def = instance_lists_sym->s.def;
instance_lists_def->initialized = instance_lists_def->constant = 1;
instance_lists_def->nosave = 1;
list = D_POINTER (pr_ptr_t, instance_lists_def);
space = instance_lists_def->space;
for (int i = 0; i < num_classes; i++, list++) {
EMIT_DEF (space, *list, instance_lists[i]);
}
*list = 0;
return instance_lists_def;
}
int
is_id (const type_t *type)
{
if (type == &type_id)
return 1;
// type may be a qualified id, in which case it will be a pointer to
// a qualified obj_object struct
if (type->type != ev_ptr)
return 0;
if (!is_struct (type->t.fldptr.type))
return 0;
// if the the symtabs match, then type is id in disguise
if (type->t.fldptr.type->t.symtab == type_object.t.symtab)
return 1;
return 0;
}
int
is_class (const type_t *type)
{
if (type->type == ev_invalid && type->meta == ty_class)
return 1;
return 0;
}
int
is_Class (const type_t *type)
{
if (type == &type_Class)
return 1;
return 0;
}
int
is_classptr (const type_t *type)
{
// easy cases first :)
if (is_id (type) || is_Class (type))
return 1;
if (type->type != ev_ptr)
return 0;
type = type->t.fldptr.type;
if (is_class (type))
return 1;
return 0;
}
int
is_SEL (const type_t *type)
{
return type == &type_SEL;
}
int
is_object (const type_t *type)
{
return type == &type_object;
}
int
is_method (const type_t *type)
{
return type == &type_method;
}
int
is_method_description (const type_t *type)
{
return type == &type_method_description;
}
static protocollist_t *
obj_get_class_protos (const type_t *type)
{
if (is_ptr (type))
type = type->t.fldptr.type;
if (is_class (type))
return type->t.class->protocols;
return 0;
}
static protocollist_t *
obj_get_protos (const type_t *type)
{
if (is_ptr (type))
type = type->t.fldptr.type;
return type->protos;
}
static category_t *
obj_get_categories (const type_t *type)
{
if (is_ptr (type))
type = type->t.fldptr.type;
if (is_class (type))
return type->t.class->categories;
return 0;
}
static const char *
obj_classname (const type_t *type)
{
static dstring_t *str;
protocollist_t *protos;
if (!str)
str = dstring_new ();
dstring_clearstr (str);
if (is_id (type)) {
dstring_copystr (str, "id");
} else if (is_Class (type)) {
dstring_copystr (str, "Class");
} else {
if (is_ptr (type))
type = type->t.fldptr.type;
if (is_class (type))
dstring_copystr (str, type->t.class->name);
}
if ((protos = obj_get_protos (type)))
print_protocollist (str, protos);
return str->str;
}
static __attribute__((pure)) int
category_implements (category_t *cat, protocol_t *protocol)
{
for (; cat; cat = cat->next) {
if (!cat->protocols)
continue;
if (procollist_find_protocol (cat->protocols, protocol))
return 1;
}
return 0;
}
int
obj_types_assignable (const type_t *dst, const type_t *src)
{
class_t *dst_class, *src_class = 0;
category_t *cat;
int dst_is_proto, src_is_proto;
protocollist_t *dst_protos = 0, *src_protos = 0;
int i;
//puts ("%$$\"$#%");
if (!is_classptr (src)) {
// if dst is a class pointer, then the types are not compatible,
// otherwise unknown
return is_classptr (dst) - 1;
}
if (!is_classptr (dst)) {
return -1;
}
dst_is_proto = is_id (dst) && (dst_protos = obj_get_protos (dst));
src_is_proto = is_id (src) && (src_protos = obj_get_protos (src));
if (dst_is_proto) {
if (src_is_proto) {
// id <protos> = id <protos>
for (i = 0; i < dst_protos->count; i++) {
if (!procollist_find_protocol (src_protos,
dst_protos->list[i])) {
warning (0, "type %s does not conform to the %s protocol",
obj_classname (src), dst_protos->list[i]->name);
return 1;
}
}
} else if (!is_id (src)) {
src_protos = obj_get_class_protos (src);
for (i = 0; i < dst_protos->count; i++) {
if (procollist_find_protocol (src_protos, dst_protos->list[i]))
continue;
cat = obj_get_categories (src);
if (cat && category_implements (cat, dst_protos->list[i]))
continue;
warning (0, "class %s does not implement to the %s protocol",
obj_classname (src), dst_protos->list[i]->name);
return 1;
}
return 1;
}
} else if (src_is_proto) {
} else {
}
if (is_id (dst) || is_id (src))
return 1;
// check dst is a base class of src
dst_class = dst->t.fldptr.type->t.class;
if (src->t.fldptr.type->meta == ty_class) {
src_class = src->t.fldptr.type->t.class;
}
//printf ("%s %s\n", dst_class->name, src_class->name);
while (dst_class != src_class && src_class) {
src_class = src_class->super_class;
//if (src_class)
// printf ("%s %s\n", dst_class->name, src_class->name);
}
if (dst_class == src_class)
return 1;
return 0;
}
static const char *
class_get_key (const void *class, void *unused)
{
return ((class_t *) class)->name;
}
static const char *
protocol_get_key (const void *protocol, void *unused)
{
return ((protocol_t *) protocol)->name;
}
const char *
get_class_name (class_type_t *class_type, int pretty)
{
switch (class_type->type) {
case ct_class:
if (pretty)
return class_type->c.class->name;
else
return va (0, "%s_", class_type->c.class->name);
case ct_category:
if (pretty)
return va (0, "%s (%s)", class_type->c.category->class->name,
class_type->c.category->name);
else
return va (0, "%s_%s", class_type->c.category->class->name,
class_type->c.category->name);
case ct_protocol:
return va (0, "<%s>", class_type->c.protocol->name);
}
return "???";
}
symbol_t *
class_symbol (class_type_t *class_type, int external)
{
const char *name = 0;
type_t *type = 0;
symbol_t *sym;
switch (class_type->type) {
case ct_category:
name = va (0, "_OBJ_CATEGORY_%s_%s",
class_type->c.category->class->name,
class_type->c.category->name);
type = &type_category;
break;
case ct_class:
name = va (0, "_OBJ_CLASS_%s", class_type->c.class->name);
type = &type_class;
break;
case ct_protocol:
return 0; // probably in error recovery
}
sym = make_symbol (name, type, pr.far_data,
external ? sc_extern : sc_global);
if (!sym->table)
symtab_addsymbol (pr.symtab, sym);
return sym;
}
static class_t *
_get_class (symbol_t *sym, int create)
{
class_t *c;
if (!class_hash)
class_hash = Hash_NewTable (1021, class_get_key, 0, 0, 0);
if (sym) {
c = Hash_Find (class_hash, sym->name);
if (c || !create)
return c;
}
c = calloc (sizeof (class_t), 1);
c->methods = new_methodlist ();
c->class_type.type = ct_class;
c->class_type.c.class = c;
if (sym) {
c->name = sym->name;
Hash_Add (class_hash, c);
sym->sy_type = sy_class;
}
sym = class_symbol (&c->class_type, 1);
c->def = sym->s.def;
return c;
}
class_t *
get_class (symbol_t *sym, int create)
{
class_t *c;
type_t new;
if (!(c = _get_class (sym, create)))
return c;
memset (&new, 0, sizeof (new));
new.type = ev_invalid;
new.name = c->name;
new.meta = ty_class;
new.t.class = c;
c->type = find_type (&new);
if (sym)
sym->type = c->type;
return c;
}
static void
set_self_type (class_t *class, method_t *method)
{
if (method->instance)
method->params->type = pointer_type (class->type);
else
method->params->type = &type_Class;
}
static void
methods_set_self_type (class_t *class, methodlist_t *methods)
{
method_t *method;
for (method = methods->head; method; method = method->next)
set_self_type (class, method);
}
void
class_add_methods (class_t *class, methodlist_t *methods)
{
if (!methods)
return;
merge_method_lists (class->methods, methods);
methods_set_self_type (class, class->methods);
}
void
class_add_protocols (class_t *class, protocollist_t *protocols)
{
int i;
protocol_t *p;
methodlist_t *methods;
methodset_t *except;
class_t *super;
if (!protocols)
return;
methods = class->methods;
except = new_methodset ();
for (super = class->super_class; super; super = super->super_class) {
methodset_add_methods (except, super->methods);
}
for (i = 0; i < protocols->count; i++) {
p = protocols->list[i];
if (p->methods) {
copy_methods (methods, p->methods, except);
} else {
warning (0, "definition of protocol `%s' not found", p->name);
}
if (p->protocols)
class_add_protocols (class, p->protocols);
}
class->protocols = protocols;
}
static void
begin_category (category_t *category)
{
pr_category_t *pr_category;
class_t *class = category->class;
symbol_t *sym;
def_t *def;
defspace_t *space;
current_class = &category->class_type;
sym = class_symbol (current_class, 0);
category->def = def = sym->s.def;
def->initialized = def->constant = def->nosave = 1;
space = def->space;
pr_category = &D_STRUCT (pr_category_t, def);
EMIT_STRING (space, pr_category->category_name, category->name);
EMIT_STRING (space, pr_category->class_name, class->name);
EMIT_DEF (space, pr_category->protocols,
emit_protocol_list (category->protocols,
va (0, "%s_%s", class->name,
category->name)));
}
typedef struct {
int count;
symbol_t *ivars;
dstring_t *encoding;
} ivar_data_t;
static void
emit_ivar_count (def_t *def, void *data, int index)
{
ivar_data_t *ivar_data = (ivar_data_t *) data;
if (!is_int(def->type))
internal_error (0, "%s: expected int def", __FUNCTION__);
D_INT (def) = ivar_data->count;
}
static void
emit_ivar_list_item (def_t *def, void *data, int index)
{
ivar_data_t *ivar_data = (ivar_data_t *) data;
symbol_t *ivar_sym;
pr_ivar_t *ivar;
defspace_t *space;
#if 0
//FIXME the type is dynamic, so need a way to pass it before it can be
//checked
if (def->type != &XXX)
internal_error (0, "%s: expected XXX def",
__FUNCTION__);
#endif
if (index < 0 || index >= ivar_data->count)
internal_error (0, "%s: out of bounds index: %d %d",
__FUNCTION__, index, ivar_data->count);
for (ivar_sym = ivar_data->ivars; ivar_sym; ivar_sym = ivar_sym->next) {
if (ivar_sym->sy_type != sy_var)
continue;
if (!index--)
break;
}
ivar = D_POINTER (pr_ivar_t, def);
space = def->space;
dstring_clearstr (ivar_data->encoding);
EMIT_STRING (space, ivar->ivar_name, ivar_sym->name);
encode_type (ivar_data->encoding, ivar_sym->type);
EMIT_STRING (space, ivar->ivar_type, ivar_data->encoding->str);
ivar->ivar_offset = ivar_sym->s.offset;
}
static def_t *
emit_ivars (symtab_t *ivars, const char *name)
{
static struct_def_t ivar_list_struct[] = {
{"ivar_count", &type_int, emit_ivar_count},
{"ivar_list", 0, emit_ivar_list_item},
{0, 0}
};
ivar_data_t ivar_data = {0, 0, 0};
symbol_t *s;
def_t *def;
ivar_data.encoding = dstring_newstr ();
if (ivars) {
ivar_data.ivars = ivars->symbols;
for (s = ivars->symbols; s; s = s->next)
if (s->sy_type == sy_var)
ivar_data.count++;
}
ivar_list_struct[1].type = array_type (&type_ivar, ivar_data.count);
def = emit_structure (va (0, "_OBJ_INSTANCE_VARIABLES_%s", name), 's',
ivar_list_struct, 0, &ivar_data, 0, sc_static);
dstring_delete (ivar_data.encoding);
return def;
}
static void
begin_class (class_t *class)
{
def_t *meta_def;
pr_class_t *meta;
pr_class_t *pr_class;
symbol_t *sym;
def_t *def;
defspace_t *space;
sym = make_symbol (va (0, "_OBJ_METACLASS_%s", class->name),
&type_class, pr.far_data, sc_static);
meta_def = sym->s.def;
meta_def->initialized = meta_def->constant = meta_def->nosave = 1;
space = meta_def->space;
meta = &D_STRUCT (pr_class_t, meta_def);
EMIT_STRING (space, meta->class_pointer, class->name);
if (class->super_class)
EMIT_STRING (space, meta->super_class, class->super_class->name);
EMIT_STRING (space, meta->name, class->name);
meta->info = _PR_CLS_META;
meta->instance_size = type_size (&type_class);
if (!class->super_class) {
// The ivars list for the meta class struct get emitted only for the
// root class of the hierachy.
// NOTE: type_class is not actually a class
EMIT_DEF (space, meta->ivars,
emit_ivars (type_class.t.symtab, "Class"));
} else {
meta->ivars = 0;
}
current_class = &class->class_type;
sym = class_symbol (current_class, 0);
class->def = def = sym->s.def;
def->initialized = def->constant = def->nosave = 1;
space = def->space;
pr_class = &D_STRUCT (pr_class_t, def);
EMIT_DEF (space, pr_class->class_pointer, meta_def);
if (class->super_class) {
class_type_t class_type = {ct_class, {0}};
class_type.c.class = class->super_class;
EMIT_STRING (space, pr_class->super_class, class->super_class->name);
class_symbol (&class_type, 1);
}
EMIT_STRING (space, pr_class->name, class->name);
pr_class->info = _PR_CLS_CLASS;
EMIT_DEF (space, pr_class->protocols,
emit_protocol_list (class->protocols, class->name));
}
void
class_begin (class_type_t *class_type)
{
if (current_class) {
warning (0, "‘@end’ missing in implementation context");
class_finish (current_class);
current_class = 0;
}
switch (class_type->type) {
case ct_category:
begin_category (class_type->c.category);
break;
case ct_class:
begin_class (class_type->c.class);
break;
case ct_protocol:
return; // probably error recovery
}
}
void
emit_class_ref (const char *class_name)
{
symbol_t *ref_sym;
symbol_t *name_sym;
def_t *ref_def;
def_t *name_def;
ref_sym = make_symbol (va (0, ".obj_class_ref_%s", class_name),
&type_ptr, pr.far_data, sc_static);
if (!ref_sym->table)
symtab_addsymbol (pr.symtab, ref_sym);
ref_def = ref_sym->s.def;
if (ref_def->initialized)
return;
ref_def->initialized = ref_def->constant = ref_def->nosave = 1;
name_sym = make_symbol (va (0, ".obj_class_name_%s", class_name),
&type_ptr, pr.far_data, sc_extern);
if (!name_sym->table)
symtab_addsymbol (pr.symtab, name_sym);
name_def = name_sym->s.def;
if (!name_def->external)
D_INT (ref_def) = name_def->offset;
reloc_def_def (name_def, ref_def);
}
static void
emit_class_name (const char *class_name)
{
symbol_t *name_sym;
def_t *name_def;
name_sym = make_symbol (va (0, ".obj_class_name_%s", class_name),
&type_ptr, pr.far_data, sc_global);
if (!name_sym->table)
symtab_addsymbol (pr.symtab, name_sym);
name_def = name_sym->s.def;
if (name_def->initialized)
return;
name_def->initialized = name_def->constant = 1;
name_def->nosave = 1;
D_INT (name_def) = 0;
}
void
emit_category_ref (const char *class_name, const char *category_name)
{
symbol_t *ref_sym;
symbol_t *name_sym;
def_t *ref_def;
def_t *name_def;
ref_sym = make_symbol (va (0, ".obj_category_ref_%s_%s",
class_name, category_name),
&type_ptr, pr.far_data, sc_static);
if (!ref_sym->table)
symtab_addsymbol (pr.symtab, ref_sym);
ref_def = ref_sym->s.def;
if (ref_def->initialized)
return;
ref_def->initialized = ref_def->constant = 1;
ref_def->nosave = 1;
name_sym = make_symbol (va (0, ".obj_category_name_%s_%s",
class_name, category_name),
&type_ptr, pr.far_data, sc_extern);
if (!name_sym->table)
symtab_addsymbol (pr.symtab, name_sym);
name_def = name_sym->s.def;
if (!name_def->external)
D_INT (ref_def) = name_def->offset;
reloc_def_def (name_def, ref_def);
}
static void
emit_category_name (const char *class_name, const char *category_name)
{
symbol_t *name_sym;
def_t *name_def;
name_sym = make_symbol (va (0, ".obj_category_name_%s_%s",
class_name, category_name),
&type_ptr, pr.far_data, sc_global);
if (!name_sym->table)
symtab_addsymbol (pr.symtab, name_sym);
name_def = name_sym->s.def;
if (name_def->initialized)
return;
name_def->initialized = name_def->constant = 1;
name_def->nosave = 1;
D_INT (name_def) = 0;
}
static void
finish_category (category_t *category)
{
pr_category_t *pr_category;
class_t *class = category->class;
char *name;
defspace_t *space;
if (!category->def) // probably in error recovery
return;
name = nva ("%s_%s", class->name, category->name);
pr_category = &D_STRUCT (pr_category_t, category->def);
space = category->def->space;
EMIT_DEF (space, pr_category->instance_methods,
emit_methods (category->methods, name, 1));
EMIT_DEF (space, pr_category->class_methods,
emit_methods (category->methods, name, 0));
free (name);
emit_class_ref (class->name);
emit_category_name (class->name, category->name);
}
static void
finish_class (class_t *class)
{
pr_class_t *meta;
pr_class_t *cls;
defspace_t *space;
if (pr.error_count) // probably in error recovery
return;
space = class->def->space;
cls = &D_STRUCT (pr_class_t, class->def);
meta = &Q_STRUCT (space, pr_class_t, cls->class_pointer);
EMIT_DEF (space, meta->methods, emit_methods (class->methods,
class->name, 0));
cls->instance_size = type_size (class->type);
EMIT_DEF (space, cls->ivars, emit_ivars (class->ivars, class->name));
EMIT_DEF (space, cls->methods, emit_methods (class->methods,
class->name, 1));
if (class->super_class)
emit_class_ref (class->super_class->name);
emit_class_name (class->name);
}
void
class_finish (class_type_t *class_type)
{
switch (class_type->type) {
case ct_category:
finish_category (class_type->c.category);
break;
case ct_class:
finish_class (class_type->c.class);
break;
case ct_protocol:
return; // probably in error recovery
}
}
class_t *
extract_class (class_type_t *class_type)
{
switch (class_type->type) {
case ct_class:
return current_class->c.class;
case ct_category:
return current_class->c.category->class;
case ct_protocol:
return 0; // probably in error recovery
}
return 0; // should not happen
}
int
class_access (class_type_t *class_type, class_t *class)
{
class_t *cur;
if (class_type) {
if (!(cur = extract_class (class_type)))
return vis_private;
if (cur == class)
return vis_private;
cur = cur->super_class;
while (cur) {
if (cur == class)
return vis_protected;
cur = cur->super_class;
}
}
return vis_public;
}
symbol_t *
class_find_ivar (class_t *class, int vis, const char *name)
{
symbol_t *ivar;
if (!class->ivars) {
if (!class->interface_declared) {
class->interface_declared = 1;
error (0, "accessing incomplete type %s", class->name);
}
return 0;
}
ivar = symtab_lookup (class->ivars, name);
if (ivar) {
if (ivar->visibility > (vis_t) vis
|| (ivar->table->class != class
&& ivar->visibility > vis_protected)) {
goto access_error;
}
return ivar;
}
error (0, "%s.%s does not exist", class->name, name);
return 0;
access_error:
error (0, "%s.%s is not accessable here", class->name, name);
return 0;
}
method_t *
class_find_method (class_type_t *class_type, method_t *method)
{
methodlist_t *methods = 0, *start_methods;
method_t *m;
dstring_t *sel;
class_t *class = 0, *start_class;
switch (class_type->type) {
case ct_category:
methods = class_type->c.category->methods;
class = class_type->c.category->class;
break;
case ct_class:
class = class_type->c.class;
methods = class->methods;
break;
case ct_protocol:
return 0; // probably in error recovery
}
start_methods = methods;
start_class = class;
while (class) {
for (m = methods->head; m; m = m->next) {
if (method_compare (method, m)) {
if (m->type != method->type)
error (0, "method type mismatch");
if (methods != start_methods) {
m = copy_method (m);
set_self_type (start_class, m);
add_method (start_methods, m);
}
method_set_param_names (m, method);
return m;
}
}
if (class->methods == methods)
class = class->super_class;
else
methods = class->methods;
}
sel = dstring_newstr ();
selector_name (sel, (keywordarg_t *)method->selector);
set_self_type (start_class, method);
add_method (start_methods, method);
dstring_delete (sel);
return method;
}
static method_t *
cls_find_method (methodlist_t *methodlist, selector_t *selector,
int class_msg, int is_root)
{
method_t *m = 0;
m = methodlist_find_method (methodlist, selector, !class_msg);
if (!m && is_root && class_msg
&& (m = methodlist_find_method (methodlist, selector, 1))) {
return m;
}
return m;
}
method_t *
class_message_response (type_t *clstype, int class_msg, expr_t *sel)
{
selector_t *selector;
method_t *m;
class_t *c;
class_t *class = 0;
category_t *cat;
dstring_t *dstr;
selector = get_selector (sel);
if (!selector)
return 0;
if (!is_classptr (clstype) && !is_class (clstype)) {
error (0, "neither class nor object");
return 0;
}
if (is_id (clstype)) {
protocollist_t *protos = clstype->t.fldptr.type->protos;
if (protos) {
if ((m = protocollist_find_method (protos, selector, !class_msg))) {
return m;
}
dstr = dstring_new ();
print_protocollist (dstr, protos);
warning (sel, "id%s may not respond to %c%s", dstr->str,
class_msg ? '+' : '-', selector->name);
dstring_delete (dstr);
}
} else {
if (is_class (clstype)) {
class = clstype->t.class;
} else if (is_class (clstype->t.fldptr.type)) {
class = clstype->t.fldptr.type->t.class;
}
if (class && !is_object(class->type)) {
if (!class->interface_declared) {
class->interface_declared = 1;
warning (0, "cannot find interface declaration for `%s'",
class->name);
}
c = class;
while (c) {
for (cat = c->categories; cat; cat = cat->next) {
if ((m = cls_find_method (cat->methods, selector,
class_msg,
!c->super_class))) {
return m;
}
}
if ((m = cls_find_method (c->methods, selector, class_msg,
!c->super_class))) {
return m;
}
c = c->super_class;
}
warning (sel, "%s may not respond to %c%s", class->name,
class_msg ? '+' : '-', selector->name);
}
}
m = find_method (selector->name);
if (!m && (!class || is_object(class->type))) {
warning (sel, "could not find method for %c%s",
class_msg ? '+' : '-', selector->name);
}
return m;
}
static uintptr_t
category_get_hash (const void *_c, void *unused)
{
category_t *c = (category_t *) _c;
return Hash_String (c->name) ^ Hash_String (c->class->name);
}
static int
category_compare (const void *_c1, const void *_c2, void *unused)
{
category_t *c1 = (category_t *) _c1;
category_t *c2 = (category_t *) _c2;
return strcmp (c1->name, c2->name) == 0
&& strcmp (c1->class->name, c2->class->name) == 0;
}
symtab_t *
class_new_ivars (class_t *class)
{
symtab_t *ivars;
symtab_t *super_ivars = 0;
if (class->super_class)
super_ivars = class->super_class->ivars;
ivars = new_symtab (super_ivars, stab_local);
ivars->class = class;
return ivars;
}
void
class_add_ivars (class_t *class, symtab_t *ivars)
{
int base = 0;
symbol_t *sym;
if (class->super_class)
base = type_size (class->super_class->type);
for (sym = ivars->symbols; sym; sym = sym->next)
sym->s.offset += base;
class->ivars = ivars;
}
static int
compare_symbols (symbol_t *s1, symbol_t *s2)
{
if (s1->visibility != s2->visibility)
return 0;
if (s1->name != s2->name) // works thanks to save_string
return 0;
if (s1->sy_type != s2->sy_type)
return 0;
if (s1->type != s2->type)
return 0;
return 1;
}
void
class_check_ivars (class_t *class, symtab_t *ivars)
{
symbol_t *civ, *iv;
int missmatch = 0;
if (!class->ivars != !ivars) {
missmatch = 1;
} else if (ivars) {
for (civ = class->ivars->symbols, iv = ivars->symbols;
civ && iv; civ = civ->next, iv = iv->next) {
if (!compare_symbols (civ, iv)) {
missmatch = 1;
break;
}
}
}
//FIXME right option?
if (missmatch && options.warnings.interface_check)
warning (0, "instance variable missmatch for %s", class->name);
class_add_ivars (class, ivars);
}
category_t *
get_category (symbol_t *class_name, const char *category_name, int create)
{
category_t *category;
class_t *class;
if (!category_hash) {
category_hash = Hash_NewTable (1021, 0, 0, 0, 0);
Hash_SetHashCompare (category_hash,
category_get_hash, category_compare);
}
class = get_class (class_name, 0);
if (!class) {
error (0, "undefined class %s", class_name->name);
return 0;
}
if (class_name && category_name) {
category_t _c = {0, category_name, class};
category = Hash_FindElement (category_hash, &_c);
if (category || !create)
return category;
}
category = calloc (sizeof (category_t), 1);
category->next = class->categories;
class->categories = category;
category->name = category_name;
category->class = class;
category->methods = new_methodlist ();
category->class_type.type = ct_category;
category->class_type.c.category = category;
if (class_name && category_name)
Hash_AddElement (category_hash, category);
return category;
}
void
category_add_methods (category_t *category, methodlist_t *methods)
{
if (!methods)
return;
merge_method_lists (category->methods, methods);
methods_set_self_type (category->class, category->methods);
}
void
category_add_protocols (category_t *category, protocollist_t *protocols)
{
int i;
protocol_t *p;
methodlist_t *methods;
methodset_t *except;
class_t *class;
if (!protocols)
return;
methods = category->methods;
except = new_methodset ();
for (class = category->class; class; class = class->super_class) {
methodset_add_methods (except, class->methods);
}
for (i = 0; i < protocols->count; i++) {
p = protocols->list[i];
copy_methods (methods, p->methods, except);
if (p->protocols)
category_add_protocols (category, p->protocols);
}
category->protocols = protocols;
}
symbol_t *
class_pointer_symbol (class_t *class)
{
def_t *def;
symbol_t *sym;
class_type_t class_type = {ct_class, {0}};
class_type.c.class = class;
sym = make_symbol (va (0, "_OBJ_CLASS_POINTER_%s", class->name),
&type_Class, pr.near_data, sc_static);
if (!sym->table)
symtab_addsymbol (pr.symtab, sym);
def = sym->s.def;
if (def->initialized)
return sym;
def->initialized = def->constant = 1;
def->nosave = 1;
if (!class->def)
class->def = class_symbol (&class_type, 1)->s.def;
if (!class->def->external)
D_INT (def) = class->def->offset;
reloc_def_def (class->def, def);
return sym;
}
typedef struct {
def_t *refs;
class_t **classes;
int cls_def_cnt;
category_t **categories;
int cat_def_cnt;
def_t *instances_list;
} obj_symtab_data_t;
static void
emit_symtab_ref_cnt (def_t *def, void *data, int index)
{
obj_symtab_data_t *da = (obj_symtab_data_t *)data;
if (!is_int(def->type))
internal_error (0, "%s: expected int def", __FUNCTION__);
D_INT (def) = 0;
if (da->refs)
D_INT (def) = da->refs->type->t.array.size;
}
static void
emit_symtab_refs (def_t *def, void *data, int index)
{
obj_symtab_data_t *da = (obj_symtab_data_t *)data;
if (!is_SEL(def->type))
internal_error (0, "%s: expected SEL def", __FUNCTION__);
D_INT (def) = 0;
if (da->refs)
EMIT_DEF (def->space, D_INT (def), da->refs);
}
static void
emit_symtab_cls_def_cnt (def_t *def, void *data, int index)
{
obj_symtab_data_t *da = (obj_symtab_data_t *)data;
if (!is_int(def->type))
internal_error (0, "%s: expected int def", __FUNCTION__);
D_INT (def) = da->cls_def_cnt;
}
static void
emit_symtab_cat_def_cnt (def_t *def, void *data, int index)
{
obj_symtab_data_t *da = (obj_symtab_data_t *)data;
if (!is_int(def->type))
internal_error (0, "%s: expected int def", __FUNCTION__);
D_INT (def) = da->cat_def_cnt;
}
static void
emit_symtab_defs (def_t *def, void *data, int index)
{
obj_symtab_data_t *da = (obj_symtab_data_t *)data;
if (!is_array (def->type) || def->type->t.array.type->type != ev_ptr)
internal_error (0, "%s: expected array of pointers def", __FUNCTION__);
if (index < 0 || index >= da->cls_def_cnt + da->cat_def_cnt + 1)
internal_error (0, "%s: out of bounds index: %d %d",
__FUNCTION__, index,
da->cls_def_cnt + da->cat_def_cnt + 1);
if (index < da->cls_def_cnt) {
class_t **cl;
for (cl = da->classes; *cl; cl++)
if ((*cl)->def && !(*cl)->def->external)
if (!index--)
break;
EMIT_DEF (def->space, D_INT (def), (*cl)->def);
} else if (index < da->cls_def_cnt + da->cat_def_cnt) {
category_t **ca;
index -= da->cls_def_cnt;
for (ca = da->categories; *ca; ca++)
if ((*ca)->def && !(*ca)->def->external)
if (!index--)
break;
EMIT_DEF (def->space, D_INT (def), (*ca)->def);
} else {
D_INT (def) = 0;
if (da->instances_list) {
EMIT_DEF (def->space, D_INT (def), da->instances_list);
}
}
}
void
class_finish_module (void)
{
static struct_def_t symtab_struct[] = {
{"sel_ref_cnt", &type_int, emit_symtab_ref_cnt},
{"refs", &type_SEL, emit_symtab_refs},
{"cls_def_cnt", &type_int, emit_symtab_cls_def_cnt},
{"cat_def_cnt", &type_int, emit_symtab_cat_def_cnt},
{"defs", 0, emit_symtab_defs},
{0, 0}
};
obj_symtab_data_t data = {};
class_t **cl;
category_t **ca;
def_t *symtab_def;
symbol_t *module_sym;
expr_t *module_expr;
pr_module_t *module;
symbol_t *exec_class_sym;
symbol_t *init_sym;
expr_t *init_expr;
data.refs = emit_selectors ();
if (class_hash) {
data.classes = (class_t **) Hash_GetList (class_hash);
for (cl = data.classes; *cl; cl++)
if ((*cl)->def && !(*cl)->def->external)
data.cls_def_cnt++;
}
if (category_hash) {
data.categories = (category_t **) Hash_GetList (category_hash);
for (ca = data.categories; *ca; ca++)
if ((*ca)->def && !(*ca)->def->external)
data.cat_def_cnt++;
}
data.instances_list = emit_static_instances_list ();
if (!data.refs && !data.cls_def_cnt && !data.cat_def_cnt
&& !data.instances_list)
return;
symtab_struct[4].type = array_type (&type_ptr,
data.cls_def_cnt
+ data.cat_def_cnt
+ 1);
symtab_def = emit_structure ("_OBJ_SYMTAB", 's', symtab_struct, 0, &data,
0, sc_static);
free (data.classes);
free (data.categories);
module_sym = make_symbol ("_OBJ_MODULE", &type_module, pr.far_data,
sc_static);
symtab_addsymbol (current_symtab, module_sym);
module = &D_STRUCT (pr_module_t, module_sym->s.def);
module->size = type_size (&type_module);
EMIT_STRING (module_sym->s.def->space, module->name,
GETSTR (pr.source_file));
EMIT_DEF (module_sym->s.def->space, module->symtab, symtab_def);
exec_class_sym = symtab_lookup (pr.symtab, "__obj_exec_class");
if (!exec_class_sym) {
exec_class_sym = new_symbol_type ("__obj_exec_class",
&type_exec_class);
exec_class_sym = function_symbol (exec_class_sym, 0, 1);
make_function (exec_class_sym, 0, exec_class_sym->table->space,
sc_extern);
}
init_sym = new_symbol_type (".ctor", &type_func);
init_sym = function_symbol (init_sym, 0, 1);
module_expr = address_expr (new_symbol_expr (module_sym), 0);
init_expr = new_block_expr ();
append_expr (init_expr,
build_function_call (new_symbol_expr (exec_class_sym),
exec_class_sym->type, module_expr));
current_func = begin_function (init_sym, 0, current_symtab, 1, sc_static);
build_code_function (init_sym, 0, init_expr);
}
protocol_t *
get_protocol (const char *name, int create)
{
protocol_t *p;
if (!protocol_hash)
protocol_hash = Hash_NewTable (1021, protocol_get_key, 0, 0, 0);
if (name) {
p = Hash_Find (protocol_hash, name);
if (p || !create)
return p;
}
p = calloc (sizeof (protocol_t), 1);
p->name = name;
p->methods = 0;
p->class_type.type = ct_protocol;
p->class_type.c.protocol = p;
if (name)
Hash_Add (protocol_hash, p);
return p;
}
void
protocol_add_methods (protocol_t *protocol, methodlist_t *methods)
{
if (!methods)
return;
merge_method_lists (protocol->methods, methods);
}
void
protocol_add_protocols (protocol_t *protocol, protocollist_t *protocols)
{
protocol->protocols = protocols;
}
def_t *
protocol_def (protocol_t *protocol)
{
if (!protocol->def) {
protocol->def = emit_protocol (protocol);
add_static_instance ("Protocol", protocol->def);
}
return protocol->def;
}
protocollist_t *
new_protocol_list (void)
{
protocollist_t *protocollist = malloc (sizeof (protocollist_t));
protocollist->count = 0;
protocollist->list = 0;
return protocollist;
}
protocollist_t *
add_protocol (protocollist_t *protocollist, const char *name)
{
protocol_t *protocol = get_protocol (name, 0);
if (!protocol) {
error (0, "undefined protocol `%s'", name);
return protocollist;
}
protocollist->count++;
protocollist->list = realloc (protocollist->list,
sizeof (protocol_t *) * protocollist->count);
protocollist->list[protocollist->count - 1] = protocol;
return protocollist;
}
int
procollist_find_protocol (protocollist_t *protocollist, protocol_t *proto)
{
int i;
if (!protocollist)
return 0;
for (i = 0; i < protocollist->count; i++)
if (protocollist->list[i] == proto)
return 1;
return 0;
}
static method_t *
protocol_find_method (protocol_t *protocol, selector_t *selector, int instance)
{
method_t *m = 0;
if (protocol->methods) {
m = methodlist_find_method (protocol->methods, selector, instance);
}
if (!m && protocol->protocols) {
return protocollist_find_method (protocol->protocols, selector,
instance);
}
return m;
}
method_t *
protocollist_find_method (protocollist_t *protocollist, selector_t *selector,
int instance)
{
method_t *m;
for (int i = 0; i < protocollist->count; i++) {
if ((m = protocol_find_method (protocollist->list[i], selector,
instance))) {
return m;
}
}
return 0;
}
int
compare_protocols (protocollist_t *protos1, protocollist_t *protos2)
{
int i;
if (protos1 == protos2)
return 1;
if (!protos1 || !protos2)
return 0;
if (protos1->count != protos2->count)
return 0;
for (i = 0; i < protos1->count; i++)
if (!procollist_find_protocol (protos2, protos1->list[i]))
return 0;
return 1;
}
void
print_protocollist (dstring_t *dstr, protocollist_t *protocollist)
{
int i;
dstring_appendstr (dstr, "<");
for (i = 0; i < protocollist->count; i++)
dasprintf (dstr, "%s%s", i ? "," : "", protocollist->list[i]->name);
dstring_appendstr (dstr, ">");
}
def_t *
emit_protocol (protocol_t *protocol)
{
def_t *proto_def;
pr_protocol_t *proto;
defspace_t *space;
proto_def = make_symbol (va (0, "_OBJ_PROTOCOL_%s", protocol->name),
&type_protocol, pr.far_data, sc_static)->s.def;
if (proto_def->initialized)
return proto_def;
proto_def->initialized = proto_def->constant = 1;
proto_def->nosave = 1;
space = proto_def->space;
proto = &D_STRUCT (pr_protocol_t, proto_def);
proto->class_pointer = 0;
EMIT_STRING (space, proto->protocol_name, protocol->name);
EMIT_DEF (space, proto->protocol_list,
emit_protocol_list (protocol->protocols,
va (0, "PROTOCOL_%s", protocol->name)));
EMIT_DEF (space, proto->instance_methods,
emit_method_descriptions (protocol->methods, protocol->name, 1));
EMIT_DEF (space, proto->class_methods,
emit_method_descriptions (protocol->methods, protocol->name, 0));
emit_class_ref ("Protocol");
return proto_def;
}
static void
emit_protocol_next (def_t *def, void *data, int index)
{
if (!is_ptr(def->type)) {
internal_error (0, "%s: expected pointer def", __FUNCTION__);
}
D_INT (def) = 0;
}
static void
emit_protocol_count (def_t *def, void *data, int index)
{
protocollist_t *protocols = (protocollist_t *) data;
if (!is_int(def->type)) {
internal_error (0, "%s: expected int def", __FUNCTION__);
}
D_INT (def) = protocols->count;
}
static void
emit_protocol_list_item (def_t *def, void *data, int index)
{
protocollist_t *protocols = (protocollist_t *) data;
protocol_t *protocol = protocols->list[index];
if (!is_array (def->type) || !is_ptr(def->type->t.array.type)) {
internal_error (0, "%s: expected array of pointer def", __FUNCTION__);
}
if (index < 0 || index >= protocols->count) {
internal_error (0, "%s: out of bounds index: %d %d",
__FUNCTION__, index, protocols->count);
}
EMIT_DEF (def->space, D_INT(def), protocol_def (protocol));
}
def_t *
emit_protocol_list (protocollist_t *protocols, const char *name)
{
static struct_def_t proto_list_struct[] = {
{"next", &type_ptr, emit_protocol_next},
{"count", &type_int, emit_protocol_count},
{"list", 0, emit_protocol_list_item},
{0, 0},
};
if (!protocols)
return 0;
proto_list_struct[2].type = array_type (&type_ptr, protocols->count);
return emit_structure (va (0, "_OBJ_PROTOCOLS_%s", name), 's',
proto_list_struct, 0, protocols, 0, sc_static);
}
void
clear_classes (void)
{
if (class_hash)
Hash_FlushTable (class_hash);
if (protocol_hash)
Hash_FlushTable (protocol_hash);
if (category_hash)
Hash_FlushTable (category_hash);
if (static_instances)
Hash_FlushTable (static_instances);
if (static_instance_classes)
Hash_FlushTable (static_instance_classes);
obj_initialized = 0;
}
symtab_t *
class_to_struct (class_t *class, symtab_t *symtab)
{
symtab_t *parent = symtab->parent;
symtab_t *ivars = class->ivars;
symtab_t *ancestor;
if (!ivars)
return symtab;
// disconnect the struct symbol table from the scope
symtab->parent = 0;
// find the ancestor of the ivars symbol table chain
for (ancestor = ivars; ancestor->parent; ancestor = ancestor->parent)
;
// connect the ivars symbol table chain to the struct symbol table
ancestor->parent = symtab;
// create a new struct symbol table from the ivars symbol table chain
symtab = symtab_flat_copy (ivars, 0);
// disconnect the ivars symbol table chain
ancestor->parent = 0;
// connect the new struct symbol table to the scope
symtab->parent = parent;
return symtab;
}
symtab_t *
class_ivar_scope (class_type_t *class_type, symtab_t *parent)
{
class_t *class = extract_class (class_type);
if (!class->ivars)
return 0;
return symtab_flat_copy (class->ivars, parent);
}
static expr_t *
class_dereference_ivar (symbol_t *sym, void *_self)
{
expr_t *self = (expr_t *) _self;
return field_expr (copy_expr (self),
new_symbol_expr (new_symbol (sym->name)));
}
void
class_finish_ivar_scope (class_type_t *class_type, symtab_t *ivar_scope,
symtab_t *param_scope)
{
class_t *class = extract_class (class_type);
type_t *class_ptr = pointer_type (class->type);
symbol_t *sym;
symbol_t *self;
expr_t *self_expr;
if (!ivar_scope)
return;
self = symtab_lookup (param_scope, "self");
if (!self) {
internal_error (0, "I've lost my self!");
}
self_expr = new_symbol_expr (self);
if (self->type != class_ptr) {
debug (0, "class method scope");
//FIXME should generate a warning on access
self_expr = cast_expr (class_ptr, self_expr);
}
for (sym = ivar_scope->symbols; sym; sym = sym->next) {
if (sym->sy_type != sy_var)
continue;
sym->sy_type = sy_convert;
sym->s.convert.conv = class_dereference_ivar;
sym->s.convert.data = self_expr;
}
}
static void
init_objective_structs (void)
{
make_structure ("obj_selector", 's', sel_struct, &type_selector);
chain_type (&type_selector);
chain_type (&type_SEL);
chain_type (&type_IMP);
make_structure ("obj_method", 's', method_struct, &type_method);
chain_type (&type_method);
make_structure ("obj_method_description", 's', method_desc_struct,
&type_method_description);
chain_type (&type_method_description);
make_structure ("obj_category", 's', category_struct, &type_category);
chain_type (&type_category);
make_structure ("obj_ivar", 's', ivar_struct, &type_ivar);
chain_type (&type_ivar);
make_structure ("obj_super", 's', super_struct, &type_super);
chain_type (&type_super);
chain_type (&type_SuperPtr);
chain_type (&type_supermsg);
symtab_addsymbol (current_symtab, new_symbol ("id"));
}
static void
init_classes (void)
{
make_structure ("obj_class", 's', class_struct, &type_class);
chain_type (&type_class);
chain_type (&type_Class);
make_structure ("obj_object", 's', object_struct, &type_object);
chain_type (&type_object);
chain_type (&type_id);
make_structure ("obj_protocol", 's', protocol_struct, &type_protocol);
chain_type (&type_protocol);
}
static void
class_init_obj_module (void)
{
make_structure ("obj_module", 's', module_struct, &type_module);
chain_type (&type_module);
chain_type (&type_moduleptr);
chain_type (&type_exec_class);
}
void
class_init (void)
{
symtab_t *cs = current_symtab;
current_symtab = pr.symtab;
class_init_obj_module ();
init_classes ();
init_objective_structs ();
obj_initialized = 1;
current_symtab = cs;
}