Refactor editor state/utils and split generators

This commit is contained in:
Bill
2026-02-09 14:47:51 -07:00
parent 53e9154b25
commit eca19503ae
12 changed files with 4025 additions and 4032 deletions

View File

@@ -0,0 +1,702 @@
#pragma once
#include "ProjectionGenerator.h"
class CppGenerator : 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;
// Generate namespace wrapper
oss << "namespace " << 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]));
}
oss << "\n} // namespace " << module->name << "\n";
return oss.str();
}
std::string visitFunction(const Function* function) override {
std::ostringstream oss;
// Process annotations first (these are in the "annotations" role)
auto annotations = function->getChildren("annotations");
for (const auto* annotation : annotations) {
std::string annotationCode = generate(annotation);
if (!annotationCode.empty() && annotationCode != "// Unknown concept: Annotation") {
oss << annotationCode << "\n";
}
}
// Generate function return type
std::string returnTypeStr = "void"; // Default
auto returnType = function->getChild("returnType");
if (returnType) {
returnTypeStr = generate(returnType);
}
// Generate function signature
oss << returnTypeStr << " " << 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 << " // Function body is empty\n";
oss << " return;";
if (returnTypeStr != "void") {
oss << " // TODO: return appropriate value";
}
oss << "\n";
} 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";
return oss.str();
}
std::string visitVariable(const Variable* variable) override {
std::ostringstream oss;
auto type = variable->getChild("type");
std::string typeStr = "auto"; // Default
if (type) {
typeStr = generate(type);
}
// Check enclosing function for memory annotations that affect variable types
std::string wrapper = getMemoryTypeWrapper(variable);
if (!wrapper.empty()) {
oss << wrapper << "<" << typeStr << "> " << variable->name;
} else {
oss << typeStr << " " << variable->name;
}
auto initializer = variable->getChild("initializer");
if (initializer) {
oss << " = " << generate(initializer);
}
oss << ";";
return oss.str();
}
std::string visitParameter(const Parameter* parameter) override {
std::ostringstream oss;
auto type = parameter->getChild("type");
std::string typeStr = "auto"; // Default
if (type) {
typeStr = generate(type);
}
oss << typeStr << " " << parameter->name;
// Add default value if present
auto defaultValue = parameter->getChild("defaultValue");
if (defaultValue) {
oss << " = " << generate(defaultValue);
}
return oss.str();
}
std::string visitAssignment(const Assignment* assignment) override {
std::ostringstream oss;
auto target = assignment->getChild("target");
auto value = assignment->getChild("value");
if (target) {
oss << generate(target) << " = ";
} else {
oss << "/* missing target */ ";
}
if (value) {
oss << generate(value);
} else {
oss << "/* missing value */";
}
oss << ";";
return oss.str();
}
std::string visitReturn(const Return* ret) override {
std::ostringstream oss;
oss << "return ";
auto value = ret->getChild("value");
if (value) {
oss << generate(value);
} else {
// For void returns
oss << ";";
return oss.str();
}
oss << ";";
return oss.str();
}
std::string visitBinaryOperation(const BinaryOperation* binOp) override {
std::ostringstream oss;
auto left = binOp->getChild("left");
auto right = binOp->getChild("right");
if (left) {
oss << generate(left);
} else {
oss << "/* missing left */";
}
oss << " " << binOp->op << " ";
if (right) {
oss << generate(right);
} else {
oss << "/* missing right */";
}
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 C++, strings need to be enclosed in double quotes
return "\"" + lit->value + "\"";
}
std::string visitBooleanLiteral(const BooleanLiteral* lit) override {
return lit->value ? "true" : "false";
}
std::string visitNullLiteral(const NullLiteral* lit) override {
return "nullptr";
}
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 << "true"; // fallback
}
oss << ") {\n";
// Then branch
for (const auto* thenStmt : thenBranch) {
std::string stmtCode = generate(thenStmt);
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 << "}";
// Else branch
if (!elseBranch.empty()) {
oss << " else {\n";
for (const auto* elseStmt : elseBranch) {
std::string stmtCode = generate(elseStmt);
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 << "}";
}
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 << "true"; // fallback to infinite loop
}
oss << ") {\n";
for (const auto* stmt : body) {
std::string stmtCode = generate(stmt);
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 << "}";
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 << "for (auto& " << iterator << " : ";
if (iterable) {
oss << generate(iterable);
} else {
oss << "/* missing iterable */"; // fallback
}
oss << ") {\n";
for (const auto* stmt : body) {
std::string stmtCode = generate(stmt);
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 << "}";
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 << ";"; // Empty statement
}
return oss.str();
}
std::string visitUnaryOperation(const UnaryOperation* unOp) override {
std::ostringstream oss;
auto operand = unOp->getChild("operand");
std::string op = unOp->op;
oss << op << " ";
if (operand) {
oss << generate(operand);
} else {
oss << "/* missing operand */";
}
return oss.str();
}
std::string visitFunctionCall(const FunctionCall* call) override {
std::ostringstream oss;
oss << call->functionName << "(";
auto arguments = call->getChildren("arguments");
for (size_t i = 0; i < arguments.size(); ++i) {
if (i > 0) oss << ", ";
oss << generate(arguments[i]);
}
oss << ")";
return oss.str();
}
std::string visitBlock(const Block* block) override {
std::ostringstream oss;
auto statements = block->getChildren("statements");
if (statements.empty()) {
oss << "{}";
} else {
oss << "{\n";
for (size_t i = 0; i < statements.size(); ++i) {
std::string stmtCode = generate(statements[i]);
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;
}
}
if (i < statements.size() - 1) oss << "\n";
}
oss << "}";
}
return oss.str();
}
std::string visitListLiteral(const ListLiteral* lit) override {
std::ostringstream oss;
oss << "{";
auto elements = lit->getChildren("elements");
for (size_t i = 0; i < elements.size(); ++i) {
if (i > 0) oss << ", ";
oss << generate(elements[i]);
}
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) {
oss << generate(target);
} else {
oss << "/* missing target */";
}
oss << "[";
if (index) {
oss << generate(index);
} else {
oss << "/* missing index */";
}
oss << "]";
return oss.str();
}
std::string visitMemberAccess(const MemberAccess* access) override {
std::ostringstream oss;
auto target = access->getChild("target");
if (target) {
oss << generate(target);
} else {
oss << "/* missing target */";
}
oss << "." << access->memberName;
return oss.str();
}
std::string visitPrimitiveType(const PrimitiveType* type) override {
std::string kind = type->kind;
// Map common types to C++ equivalents
if (kind == "int") return "int";
if (kind == "float") return "float";
if (kind == "string") return "std::string";
if (kind == "bool") return "bool";
if (kind == "char") return "char";
if (kind == "double") return "double";
if (kind == "long") return "long";
if (kind == "short") return "short";
if (kind == "byte") return "char"; // C++ doesn't have byte, use char
if (kind == "void") return "void";
return kind; // Return as-is if not a common type
}
std::string visitListType(const ListType* type) override {
std::ostringstream oss;
oss << "std::vector<";
auto elementType = type->getChild("elementType");
if (elementType) {
oss << generate(elementType);
} else {
oss << "auto"; // Default if no element type specified
}
oss << ">";
return oss.str();
}
std::string visitSetType(const SetType* type) override {
std::ostringstream oss;
oss << "std::set<";
auto elementType = type->getChild("elementType");
if (elementType) {
oss << generate(elementType);
} else {
oss << "auto"; // Default if no element type specified
}
oss << ">";
return oss.str();
}
std::string visitMapType(const MapType* type) override {
std::ostringstream oss;
oss << "std::map<";
auto keyType = type->getChild("keyType");
auto valueType = type->getChild("valueType");
if (keyType) {
oss << generate(keyType);
} else {
oss << "auto"; // Default if no key type specified
}
oss << ", ";
if (valueType) {
oss << generate(valueType);
} else {
oss << "auto"; // Default if no value type specified
}
oss << ">";
return oss.str();
}
std::string visitTupleType(const TupleType* type) override {
std::ostringstream oss;
oss << "std::tuple<";
auto elementTypes = type->getChildren("elementTypes");
for (size_t i = 0; i < elementTypes.size(); ++i) {
if (i > 0) oss << ", ";
oss << generate(elementTypes[i]);
}
oss << ">";
return oss.str();
}
std::string visitArrayType(const ArrayType* type) override {
std::ostringstream oss;
oss << "std::array<";
auto elementType = type->getChild("elementType");
if (elementType) {
oss << generate(elementType);
} else {
oss << "auto"; // Default if no element type specified
}
oss << ", /* size unknown */>";
return oss.str();
}
std::string visitOptionalType(const OptionalType* type) override {
std::ostringstream oss;
oss << "std::optional<";
auto innerType = type->getChild("innerType");
if (innerType) {
oss << generate(innerType);
} else {
oss << "auto"; // Default if no inner type specified
}
oss << ">";
return oss.str();
}
std::string visitCustomType(const CustomType* type) override {
return type->typeName;
}
std::string visitDerefStrategy(const DerefStrategy* annotation) override {
// Generate C++-style comment for deref strategy
if (annotation->strategy == "batched") {
return "// @deref(batched) - Use batched memory management (smart pointers)";
} else if (annotation->strategy == "streamed") {
return "// @deref(streamed) - Use streamed memory management (RAII)";
} else if (annotation->strategy == "manual") {
return "// @deref(manual) - Use manual memory management (new/delete)";
} else {
return "// @deref(" + annotation->strategy + ") - Memory management strategy";
}
}
std::string visitOptimizationLock(const OptimizationLock* annotation) override {
return "// @lock(" + annotation->lockedBy + ") - Optimization locked: " + annotation->lockReason;
}
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 + ") - Memory deallocation strategy";
}
std::string visitLifetimeAnnotation(const LifetimeAnnotation* annotation) override {
return "// @lifetime(" + annotation->strategy + ") - Object lifetime management";
}
std::string visitReclaimAnnotation(const ReclaimAnnotation* annotation) override {
return "// @reclaim(" + annotation->strategy + ") - Memory reclamation strategy";
}
std::string visitOwnerAnnotation(const OwnerAnnotation* annotation) override {
return "// @owner(" + annotation->strategy + ") - Ownership management strategy";
}
std::string visitAllocateAnnotation(const AllocateAnnotation* annotation) override {
return "// @allocate(" + annotation->strategy + ") - Memory allocation strategy";
}
std::string visitHotColdAnnotation(const HotColdAnnotation* annotation) override {
if (annotation->hint == "Hot")
return "__attribute__((hot))";
if (annotation->hint == "Cold")
return "__attribute__((cold))";
return "// @hotcold(" + annotation->hint + ")";
}
std::string visitInlineAnnotation(const InlineAnnotation* annotation) override {
if (annotation->mode == "Always")
return "[[gnu::always_inline]] inline";
if (annotation->mode == "Never")
return "__attribute__((noinline))";
return "inline";
}
std::string visitPureAnnotation(const PureAnnotation*) override {
return "[[nodiscard]]";
}
std::string visitConstExprAnnotation(const ConstExprAnnotation*) override {
return "constexpr";
}
private:
// Check enclosing function's memory annotations to determine smart-pointer wrapper
std::string getMemoryTypeWrapper(const Variable* variable) const {
const ASTNode* cur = variable->parent;
while (cur && cur->conceptType != "Function") {
cur = cur->parent;
}
if (!cur) return "";
for (auto* anno : cur->getChildren("annotations")) {
if (anno->conceptType == "ReclaimAnnotation") {
auto* ra = static_cast<const ReclaimAnnotation*>(anno);
if (ra->strategy == "Tracing" || ra->strategy == "Cycle")
return "std::shared_ptr";
}
if (anno->conceptType == "LifetimeAnnotation") {
auto* la = static_cast<const LifetimeAnnotation*>(anno);
if (la->strategy == "RAII")
return "std::unique_ptr";
}
if (anno->conceptType == "OwnerAnnotation") {
auto* oa = static_cast<const OwnerAnnotation*>(anno);
if (oa->strategy == "Shared_ARC") return "std::shared_ptr";
if (oa->strategy == "Single") return "std::unique_ptr";
}
}
return "";
}
};