Refactor editor state/utils and split generators
This commit is contained in:
Bill
545
editor/src/ast/ElispGenerator.h
Normal file
545
editor/src/ast/ElispGenerator.h
Normal file
@@ -0,0 +1,545 @@
|
||||
#pragma once
|
||||
#include "ProjectionGenerator.h"
|
||||
|
||||
class ElispGenerator : public ProjectionGenerator {
|
||||
public:
|
||||
std::string generate(const ASTNode* node) override {
|
||||
return dispatchGenerate(this, node, "; Unknown concept: ");
|
||||
}
|
||||
|
||||
std::string visitModule(const Module* module) override {
|
||||
std::ostringstream oss;
|
||||
|
||||
oss << "; Module: " << module->name << "\n\n";
|
||||
|
||||
// Process variables first
|
||||
auto variables = module->getChildren("variables");
|
||||
for (const auto* var : variables) {
|
||||
oss << visitVariable(static_cast<const Variable*>(var)) << "\n";
|
||||
}
|
||||
|
||||
if (!variables.empty()) {
|
||||
oss << "\n";
|
||||
}
|
||||
|
||||
// Process functions
|
||||
auto functions = module->getChildren("functions");
|
||||
for (size_t i = 0; i < functions.size(); ++i) {
|
||||
if (i > 0) oss << "\n"; // Add blank line between functions
|
||||
oss << visitFunction(static_cast<const Function*>(functions[i]));
|
||||
}
|
||||
|
||||
return oss.str();
|
||||
}
|
||||
|
||||
std::string visitFunction(const Function* function) override {
|
||||
std::ostringstream oss;
|
||||
|
||||
// Process annotations first
|
||||
auto annotations = function->getChildren("annotations");
|
||||
for (const auto* annotation : annotations) {
|
||||
std::string annotationCode = generate(annotation);
|
||||
if (!annotationCode.empty()) {
|
||||
oss << annotationCode << "\n";
|
||||
}
|
||||
}
|
||||
|
||||
// Generate function definition in Elisp
|
||||
oss << "(defun " << function->name << " (";
|
||||
|
||||
auto parameters = function->getChildren("parameters");
|
||||
for (size_t i = 0; i < parameters.size(); ++i) {
|
||||
if (i > 0) oss << " ";
|
||||
oss << visitParameter(static_cast<const Parameter*>(parameters[i]));
|
||||
}
|
||||
|
||||
oss << ")\n";
|
||||
|
||||
// Process function body
|
||||
auto body = function->getChildren("body");
|
||||
if (body.empty()) {
|
||||
oss << " nil)\n"; // Return nil if no body
|
||||
} else {
|
||||
for (const auto* stmt : body) {
|
||||
std::string stmtCode = generate(stmt);
|
||||
// Indent each line of the statement
|
||||
size_t pos = 0;
|
||||
while (pos < stmtCode.length()) {
|
||||
size_t newlinePos = stmtCode.find('\n', pos);
|
||||
if (newlinePos == std::string::npos) {
|
||||
oss << " " << stmtCode.substr(pos) << "\n";
|
||||
break;
|
||||
} else {
|
||||
oss << " " << stmtCode.substr(pos, newlinePos - pos) << "\n";
|
||||
pos = newlinePos + 1;
|
||||
if (pos >= stmtCode.length()) break;
|
||||
}
|
||||
}
|
||||
}
|
||||
oss << ")\n"; // Close the function
|
||||
}
|
||||
|
||||
return oss.str();
|
||||
}
|
||||
|
||||
std::string visitVariable(const Variable* variable) override {
|
||||
std::ostringstream oss;
|
||||
oss << "(defvar " << variable->name << " ";
|
||||
|
||||
auto initializer = variable->getChild("initializer");
|
||||
if (initializer) {
|
||||
oss << generate(initializer);
|
||||
} else {
|
||||
oss << "nil";
|
||||
}
|
||||
|
||||
oss << ") ; Variable declaration\n";
|
||||
return oss.str();
|
||||
}
|
||||
|
||||
std::string visitParameter(const Parameter* parameter) override {
|
||||
// In Elisp, parameters are just symbols
|
||||
return parameter->name;
|
||||
}
|
||||
|
||||
std::string visitAssignment(const Assignment* assignment) override {
|
||||
std::ostringstream oss;
|
||||
|
||||
auto target = assignment->getChild("target");
|
||||
auto value = assignment->getChild("value");
|
||||
|
||||
if (target && value) {
|
||||
oss << "(setq " << generate(target) << " " << generate(value) << ")";
|
||||
} else if (target) {
|
||||
oss << "(setq " << generate(target) << " nil)";
|
||||
} else {
|
||||
oss << "nil";
|
||||
}
|
||||
|
||||
return oss.str();
|
||||
}
|
||||
|
||||
std::string visitReturn(const Return* ret) override {
|
||||
// In Elisp, the last expression in a function is the return value
|
||||
// So we just return the value expression
|
||||
auto value = ret->getChild("value");
|
||||
if (value) {
|
||||
return generate(value);
|
||||
} else {
|
||||
return "nil";
|
||||
}
|
||||
}
|
||||
|
||||
std::string visitBinaryOperation(const BinaryOperation* binOp) override {
|
||||
std::ostringstream oss;
|
||||
|
||||
auto left = binOp->getChild("left");
|
||||
auto right = binOp->getChild("right");
|
||||
|
||||
// Map operators to Elisp equivalents
|
||||
std::string op = binOp->op;
|
||||
if (op == "+") op = "+";
|
||||
else if (op == "-") op = "-";
|
||||
else if (op == "*") op = "*";
|
||||
else if (op == "/") op = "/";
|
||||
else if (op == "==") op = "=";
|
||||
else if (op == "!=") op = "/=";
|
||||
else if (op == "<") op = "<";
|
||||
else if (op == ">") op = ">";
|
||||
else if (op == "<=") op = "<=";
|
||||
else if (op == ">=") op = ">=";
|
||||
else if (op == "and") op = "and";
|
||||
else if (op == "or") op = "or";
|
||||
|
||||
oss << "(" << op << " ";
|
||||
if (left) {
|
||||
oss << generate(left);
|
||||
} else {
|
||||
oss << "nil";
|
||||
}
|
||||
oss << " ";
|
||||
if (right) {
|
||||
oss << generate(right);
|
||||
} else {
|
||||
oss << "nil";
|
||||
}
|
||||
oss << ")";
|
||||
|
||||
return oss.str();
|
||||
}
|
||||
|
||||
std::string visitVariableReference(const VariableReference* varRef) override {
|
||||
return varRef->variableName;
|
||||
}
|
||||
|
||||
std::string visitIntegerLiteral(const IntegerLiteral* lit) override {
|
||||
return std::to_string(lit->value);
|
||||
}
|
||||
|
||||
std::string visitFloatLiteral(const FloatLiteral* lit) override {
|
||||
return lit->value;
|
||||
}
|
||||
|
||||
std::string visitStringLiteral(const StringLiteral* lit) override {
|
||||
// In Elisp, strings are enclosed in double quotes
|
||||
return "\"" + lit->value + "\"";
|
||||
}
|
||||
|
||||
std::string visitBooleanLiteral(const BooleanLiteral* lit) override {
|
||||
return lit->value ? "t" : "nil";
|
||||
}
|
||||
|
||||
std::string visitNullLiteral(const NullLiteral* lit) override {
|
||||
return "nil";
|
||||
}
|
||||
|
||||
std::string visitIfStatement(const IfStatement* stmt) override {
|
||||
std::ostringstream oss;
|
||||
|
||||
auto condition = stmt->getChild("condition");
|
||||
auto thenBranch = stmt->getChildren("thenBranch");
|
||||
auto elseBranch = stmt->getChildren("elseBranch");
|
||||
|
||||
oss << "(if ";
|
||||
if (condition) {
|
||||
oss << generate(condition);
|
||||
} else {
|
||||
oss << "nil"; // fallback
|
||||
}
|
||||
oss << "\n";
|
||||
|
||||
// Then branch
|
||||
if (!thenBranch.empty()) {
|
||||
for (size_t i = 0; i < thenBranch.size(); ++i) {
|
||||
if (i > 0) oss << "\n";
|
||||
oss << " " << generate(thenBranch[i]);
|
||||
}
|
||||
} else {
|
||||
oss << " nil";
|
||||
}
|
||||
|
||||
// Else branch
|
||||
if (!elseBranch.empty()) {
|
||||
oss << "\n";
|
||||
for (size_t i = 0; i < elseBranch.size(); ++i) {
|
||||
if (i > 0) oss << "\n";
|
||||
oss << " " << generate(elseBranch[i]);
|
||||
}
|
||||
} else {
|
||||
oss << "\n nil";
|
||||
}
|
||||
|
||||
oss << ")";
|
||||
return oss.str();
|
||||
}
|
||||
|
||||
std::string visitWhileLoop(const WhileLoop* loop) override {
|
||||
std::ostringstream oss;
|
||||
|
||||
auto condition = loop->getChild("condition");
|
||||
auto body = loop->getChildren("body");
|
||||
|
||||
oss << "(while ";
|
||||
if (condition) {
|
||||
oss << generate(condition);
|
||||
} else {
|
||||
oss << "t"; // fallback to infinite loop
|
||||
}
|
||||
oss << "\n";
|
||||
|
||||
for (size_t i = 0; i < body.size(); ++i) {
|
||||
if (i > 0) oss << "\n";
|
||||
oss << " " << generate(body[i]);
|
||||
}
|
||||
|
||||
oss << ")";
|
||||
return oss.str();
|
||||
}
|
||||
|
||||
std::string visitForLoop(const ForLoop* loop) override {
|
||||
std::ostringstream oss;
|
||||
|
||||
auto iterable = loop->getChild("iterable");
|
||||
auto body = loop->getChildren("body");
|
||||
|
||||
std::string iterator = loop->iteratorName.empty() ? "item" : loop->iteratorName;
|
||||
|
||||
oss << "(dolist (" << iterator << " ";
|
||||
if (iterable) {
|
||||
oss << generate(iterable);
|
||||
} else {
|
||||
oss << "nil"; // fallback
|
||||
}
|
||||
oss << ")\n";
|
||||
|
||||
for (size_t i = 0; i < body.size(); ++i) {
|
||||
if (i > 0) oss << "\n";
|
||||
oss << " " << generate(body[i]);
|
||||
}
|
||||
|
||||
oss << ")";
|
||||
return oss.str();
|
||||
}
|
||||
|
||||
std::string visitExpressionStatement(const ExpressionStatement* stmt) override {
|
||||
std::ostringstream oss;
|
||||
|
||||
auto expr = stmt->getChild("expression");
|
||||
if (expr) {
|
||||
oss << generate(expr);
|
||||
} else {
|
||||
oss << "nil";
|
||||
}
|
||||
|
||||
return oss.str();
|
||||
}
|
||||
|
||||
std::string visitUnaryOperation(const UnaryOperation* unOp) override {
|
||||
std::ostringstream oss;
|
||||
|
||||
auto operand = unOp->getChild("operand");
|
||||
std::string op = unOp->op;
|
||||
|
||||
// Map unary operators to Elisp equivalents
|
||||
if (op == "!") op = "not";
|
||||
else if (op == "-") op = "-";
|
||||
else if (op == "+") op = "+";
|
||||
|
||||
oss << "(" << op << " ";
|
||||
if (operand) {
|
||||
oss << generate(operand);
|
||||
} else {
|
||||
oss << "nil";
|
||||
}
|
||||
oss << ")";
|
||||
|
||||
return oss.str();
|
||||
}
|
||||
|
||||
std::string visitFunctionCall(const FunctionCall* call) override {
|
||||
std::ostringstream oss;
|
||||
oss << "(" << call->functionName;
|
||||
|
||||
auto arguments = call->getChildren("arguments");
|
||||
for (const auto* arg : arguments) {
|
||||
oss << " " << generate(arg);
|
||||
}
|
||||
|
||||
oss << ")";
|
||||
|
||||
return oss.str();
|
||||
}
|
||||
|
||||
std::string visitBlock(const Block* block) override {
|
||||
std::ostringstream oss;
|
||||
|
||||
auto statements = block->getChildren("statements");
|
||||
if (statements.empty()) {
|
||||
oss << "nil";
|
||||
} else {
|
||||
oss << "(progn\n";
|
||||
for (size_t i = 0; i < statements.size(); ++i) {
|
||||
oss << " " << generate(statements[i]);
|
||||
if (i < statements.size() - 1) oss << "\n";
|
||||
}
|
||||
oss << ")";
|
||||
}
|
||||
|
||||
return oss.str();
|
||||
}
|
||||
|
||||
std::string visitListLiteral(const ListLiteral* lit) override {
|
||||
std::ostringstream oss;
|
||||
oss << "(list";
|
||||
|
||||
auto elements = lit->getChildren("elements");
|
||||
for (const auto* elem : elements) {
|
||||
oss << " " << generate(elem);
|
||||
}
|
||||
|
||||
oss << ")";
|
||||
|
||||
return oss.str();
|
||||
}
|
||||
|
||||
std::string visitIndexAccess(const IndexAccess* access) override {
|
||||
std::ostringstream oss;
|
||||
|
||||
auto target = access->getChild("target");
|
||||
auto index = access->getChild("index");
|
||||
|
||||
if (target && index) {
|
||||
// In Elisp, accessing list/array elements
|
||||
oss << "(nth " << generate(index) << " " << generate(target) << ")";
|
||||
} else if (target) {
|
||||
oss << generate(target);
|
||||
} else {
|
||||
oss << "nil";
|
||||
}
|
||||
|
||||
return oss.str();
|
||||
}
|
||||
|
||||
std::string visitMemberAccess(const MemberAccess* access) override {
|
||||
std::ostringstream oss;
|
||||
|
||||
auto target = access->getChild("target");
|
||||
if (target) {
|
||||
// In Elisp, member access depends on the data structure
|
||||
// For now, we'll represent it as a function call
|
||||
oss << "(." << access->memberName << " " << generate(target) << ")";
|
||||
} else {
|
||||
oss << access->memberName;
|
||||
}
|
||||
|
||||
return oss.str();
|
||||
}
|
||||
|
||||
std::string visitPrimitiveType(const PrimitiveType* type) override {
|
||||
// In Elisp, types are more dynamic, but we can represent them as symbols
|
||||
std::string kind = type->kind;
|
||||
if (kind == "int") return "integer";
|
||||
if (kind == "float") return "float";
|
||||
if (kind == "string") return "string";
|
||||
if (kind == "bool") return "boolean";
|
||||
if (kind == "char") return "character";
|
||||
if (kind == "double") return "float";
|
||||
if (kind == "long") return "integer";
|
||||
if (kind == "short") return "integer";
|
||||
if (kind == "byte") return "integer";
|
||||
if (kind == "void") return "null";
|
||||
|
||||
return kind; // Return as-is if not a common type
|
||||
}
|
||||
|
||||
std::string visitListType(const ListType* type) override {
|
||||
std::ostringstream oss;
|
||||
oss << "(list ";
|
||||
|
||||
auto elementType = type->getChild("elementType");
|
||||
if (elementType) {
|
||||
oss << generate(elementType);
|
||||
} else {
|
||||
oss << "t"; // t means any type in Elisp
|
||||
}
|
||||
|
||||
oss << ")";
|
||||
return oss.str();
|
||||
}
|
||||
|
||||
std::string visitSetType(const SetType* type) override {
|
||||
std::ostringstream oss;
|
||||
oss << "(hash-table :test 'equal)";
|
||||
return oss.str(); // Represent sets as hash tables with equal test
|
||||
}
|
||||
|
||||
std::string visitMapType(const MapType* type) override {
|
||||
std::ostringstream oss;
|
||||
oss << "(hash-table :test 'equal)";
|
||||
return oss.str(); // Represent maps as hash tables
|
||||
}
|
||||
|
||||
std::string visitTupleType(const TupleType* type) override {
|
||||
std::ostringstream oss;
|
||||
oss << "(vector"; // Represent tuples as vectors
|
||||
|
||||
auto elementTypes = type->getChildren("elementTypes");
|
||||
for (const auto* elemType : elementTypes) {
|
||||
oss << " " << generate(elemType);
|
||||
}
|
||||
|
||||
oss << ")";
|
||||
return oss.str();
|
||||
}
|
||||
|
||||
std::string visitArrayType(const ArrayType* type) override {
|
||||
std::ostringstream oss;
|
||||
oss << "(vector ";
|
||||
|
||||
auto elementType = type->getChild("elementType");
|
||||
if (elementType) {
|
||||
oss << generate(elementType);
|
||||
} else {
|
||||
oss << "t";
|
||||
}
|
||||
|
||||
oss << ")";
|
||||
return oss.str();
|
||||
}
|
||||
|
||||
std::string visitOptionalType(const OptionalType* type) override {
|
||||
std::ostringstream oss;
|
||||
oss << "(or null ";
|
||||
|
||||
auto innerType = type->getChild("innerType");
|
||||
if (innerType) {
|
||||
oss << generate(innerType);
|
||||
} else {
|
||||
oss << "t";
|
||||
}
|
||||
|
||||
oss << ")";
|
||||
return oss.str();
|
||||
}
|
||||
|
||||
std::string visitCustomType(const CustomType* type) override {
|
||||
return type->typeName;
|
||||
}
|
||||
|
||||
std::string visitDerefStrategy(const DerefStrategy* annotation) override {
|
||||
// Convert deref strategy to Elisp annotation/comment
|
||||
if (annotation->strategy == "batched") {
|
||||
return "; @deref(batched) - Process in batches for efficiency";
|
||||
} else if (annotation->strategy == "streamed") {
|
||||
return "; @deref(streamed) - Stream processing";
|
||||
} else if (annotation->strategy == "manual") {
|
||||
return "; @deref(manual) - Manual memory management";
|
||||
} else {
|
||||
return "; @deref(" + annotation->strategy + ") - Memory dereference strategy";
|
||||
}
|
||||
}
|
||||
|
||||
std::string visitOptimizationLock(const OptimizationLock* annotation) override {
|
||||
return "; @lock(" + annotation->lockedBy + ") - Optimization locked by " + annotation->lockedBy;
|
||||
}
|
||||
|
||||
std::string visitLangSpecific(const LangSpecific* annotation) override {
|
||||
return "; @lang_specific(" + annotation->language + ", " + annotation->idiomType + ")";
|
||||
}
|
||||
|
||||
std::string visitDeallocateAnnotation(const DeallocateAnnotation* annotation) override {
|
||||
return "; @dealloc(" + annotation->strategy + ")";
|
||||
}
|
||||
|
||||
std::string visitLifetimeAnnotation(const LifetimeAnnotation* annotation) override {
|
||||
return "; @lifetime(" + annotation->strategy + ")";
|
||||
}
|
||||
|
||||
std::string visitReclaimAnnotation(const ReclaimAnnotation* annotation) override {
|
||||
return "; @reclaim(" + annotation->strategy + ")";
|
||||
}
|
||||
|
||||
std::string visitOwnerAnnotation(const OwnerAnnotation* annotation) override {
|
||||
return "; @owner(" + annotation->strategy + ")";
|
||||
}
|
||||
|
||||
std::string visitAllocateAnnotation(const AllocateAnnotation* annotation) override {
|
||||
return "; @allocate(" + annotation->strategy + ")";
|
||||
}
|
||||
|
||||
std::string visitHotColdAnnotation(const HotColdAnnotation* annotation) override {
|
||||
return "; @" + annotation->hint + " - Optimization hint";
|
||||
}
|
||||
|
||||
std::string visitInlineAnnotation(const InlineAnnotation* annotation) override {
|
||||
return "; @inline(" + annotation->mode + ")";
|
||||
}
|
||||
|
||||
std::string visitPureAnnotation(const PureAnnotation*) override {
|
||||
return "; @pure - No side effects";
|
||||
}
|
||||
|
||||
std::string visitConstExprAnnotation(const ConstExprAnnotation*) override {
|
||||
return "; @constexpr - Compile-time evaluable";
|
||||
}
|
||||
};
|
||||
Reference in New Issue
Block a user