corx

---

A statically typed, general-purpose programming language, crafted for simplicity and performance.

"Trust the programmer" - corx

---

---

module main;

import "std:io";

# Forward declarations
class circle;
class rectangle;
class square;

interface visitor {
    float visit(circle &c);
    float visit(rectangle &r);
    float visit(square &s);
};

interface shape {
    void accept(visitor &v);
};

class circle {
    float radius = 10.0;
    string name = "Circle";
} shape;

circle : accept(visitor &v) {
    return v.visit(this);
}

class rectangle {
    float width = 20.0;
    float height = 15.0;
    string name = "Rectangle";
} shape;

rectangle : accept(visitor &v) {
    return v.visit(this);
}

class square : rectangle {
    string name = "Square";
};

square : accept(visitor &v) {
    return v.visit(this);
}

class draw_visitor {} visitor;

draw_visitor : visit(circle &c) {
    float area = 3.1415 * c.radius * c.radius;
    printn("Drawing a " + c.name + " with area: " + area);
    return area;
}

draw_visitor : visit(rectangle &r) {
    float area = r.width * r.height;
    printn("Drawing a " + r.name + " with area: " + area);
    return area;
}

draw_visitor : visit(square &s) {
    float area = s.width * s.width;
    printn("Drawing a " + s.name + " with area: " + area);
    return area;
}

int main(void) {
    shape **shapes = new *shape[3];  # allocates an array of 3 pointers
    shapes[0] = new circle();        # allocates on the heap
    shapes[1] = new rectangle();
    shapes[2] = new square();

    draw_visitor dv;
    float total_area = 0.0;

    for (int i = 0; i < 3; i++) {
        total_area += shapes[i]->accept(dv);
    }

    printn("Total area of all shapes: " + total_area);

    # clean up memory
    for (int i = 0; i < 3; i++) {
        purge shapes[i];
    }

    purge shapes;
    return 0;
}