2 курс. Программирование на C++. Турнир 2, осень 2021

вход курсы / турниры задачи участники консоль модули компиляторы ключевые слова результаты

ru en cn

5777092

Пользователь

Травка Евгения Александровна

Время

11.11.2021 18:29<< 15.11.2021 11:27 (26 д. 09:27)

Обработка

11:27 ... 11:27 (3,18 сек.)

Файл

5777092.cpp (CLang++ 5.0.0 + Address Sanitizer)

Турнир

2 курс. Программирование на C++. Турнир 2, осень 2021

Задача

F: Code format visitor

Набор тестов

Ограничения

Время: 1с Память: 4000Мб

Состояние [?]

Принято

код | результаты

[печатать | | ]

≡

#pragma once

#include <vector>
#include <string>

class FormatVisitor: public BaseVisitor {
 public:
  void Visit(const BaseNode* node) override;
  void Visit(const ClassDeclarationNode* node) override;
  void Visit(const VarDeclarationNode* node) override;
  void Visit(const MethodDeclarationNode* node) override;
  const std::vector<std::string>& GetFormattedCode() const;

private:
  std::string tab_;
  std::vector<std::string> formatted_code_;
};

void FormatVisitor::Visit(const BaseNode *node) {
  node->Visit(this);
}

void FormatVisitor::Visit(const ClassDeclarationNode *node) {
  formatted_code_.push_back(tab_ + "class " + node->ClassName() + " {");

if (!node->PublicFields().empty()) {
    formatted_code_.push_back(tab_ + "  public:");
    tab_ += "    ";
    for (size_t i = 0; i < node->PublicFields().size(); i++) {
      node->PublicFields()[i]->Visit(this);
      formatted_code_[formatted_code_.size() - 1] += ";";
    }
    tab_ = tab_.substr(0, tab_.size() - 4);
  }
  if (!node->ProtectedFields().empty()) {
    formatted_code_.emplace_back("");
    formatted_code_.push_back(tab_ + "  protected:");
    tab_ += "    ";
    for (size_t i = 0; i < node->ProtectedFields().size(); i++) {
      node->ProtectedFields()[i]->Visit(this);
      formatted_code_[formatted_code_.size() - 1] += ";";
    }
    tab_ = tab_.substr(0, tab_.size() - 4);
  }
  if (!node->PrivateFields().empty()) {
    formatted_code_.emplace_back("");
    formatted_code_.push_back(tab_ + "  private:");
    tab_ += "    ";
    for (size_t i = 0; i < node->PrivateFields().size(); i++) {
      node->PrivateFields()[i]->Visit(this);
      formatted_code_[formatted_code_.size() - 1] += ";";
    }
    tab_ = tab_.substr(0, tab_.size() - 4);
  }
  formatted_code_.push_back(tab_ + "}");
  if (tab_.empty()) {
    formatted_code_[formatted_code_.size() - 1] += ";";
  }
}

void FormatVisitor::Visit(const VarDeclarationNode* node) {
formatted_code_.push_back(tab_ + node->TypeName() + " " + node->VarName());
}

void FormatVisitor::Visit(const MethodDeclarationNode* node) {
  formatted_code_.push_back(tab_ + node->ReturnTypeName() + " " \
                                                 + node->MethodName() + "(");
  for (size_t i = 0; i < node->Arguments().size(); i++) {
    Visit(node->Arguments()[i]);
    formatted_code_[formatted_code_.size() - 2] +=
    formatted_code_[formatted_code_.size() - 1].substr(tab_.size());
    if (i < node->Arguments().size() - 1) {
      formatted_code_[formatted_code_.size() - 2] += ", ";
    }
    formatted_code_.pop_back();
  }
  formatted_code_[formatted_code_.size() - 1] += ")";
}

const std::vector<std::string> &FormatVisitor::GetFormattedCode() const {
  return formatted_code_;
}

(2 732 b)

0.049s 0.009s 9