mirror of
https://github.com/open-goal/jak-project
synced 2026-05-26 15:46:14 -04:00
Tyler Wilding
104 lines
3.5 KiB
C++
104 lines
3.5 KiB
C++
#include "rule_config.h"
|
|
|
|
#include "common/formatter/formatter_tree.h"
|
|
|
|
namespace formatter_rules {
|
|
namespace config {
|
|
|
|
// TODO - this could be greatly simplified with C++20's designated initialization
|
|
FormFormattingConfig new_permissive_flow_rule() {
|
|
FormFormattingConfig cfg;
|
|
cfg.hang_forms = false;
|
|
cfg.combine_first_two_lines = true;
|
|
return cfg;
|
|
}
|
|
|
|
FormFormattingConfig new_flow_rule(int start_index) {
|
|
FormFormattingConfig cfg;
|
|
cfg.hang_forms = false;
|
|
cfg.inline_until_index = [start_index](const std::vector<std::string>& /*curr_lines*/) {
|
|
return start_index;
|
|
};
|
|
return cfg;
|
|
}
|
|
|
|
FormFormattingConfig new_flow_rule_prevent_inlining_indexes(
|
|
int start_index,
|
|
const std::vector<int>& inlining_preventation_indices) {
|
|
FormFormattingConfig cfg;
|
|
cfg.hang_forms = false;
|
|
cfg.inline_until_index = [start_index](std::vector<std::string> curr_lines) {
|
|
if (curr_lines.size() >= 4 && curr_lines.at(3) == "()") {
|
|
return 4;
|
|
}
|
|
return start_index;
|
|
};
|
|
for (const auto& index : inlining_preventation_indices) {
|
|
auto temp_config = std::make_shared<FormFormattingConfig>();
|
|
temp_config->prevent_inlining = true;
|
|
temp_config->hang_forms = false;
|
|
temp_config->indentation_width = 1;
|
|
cfg.index_configs.emplace(index, temp_config);
|
|
}
|
|
return cfg;
|
|
}
|
|
|
|
FormFormattingConfig new_binding_rule(int form_head_width) {
|
|
FormFormattingConfig cfg;
|
|
cfg.hang_forms = false;
|
|
cfg.combine_first_two_lines = true;
|
|
auto binding_list_config = std::make_shared<FormFormattingConfig>();
|
|
binding_list_config->hang_forms = false;
|
|
binding_list_config->indentation_width = 1;
|
|
binding_list_config->indentation_width_for_index = [form_head_width](FormFormattingConfig /*cfg*/,
|
|
int index) {
|
|
if (index == 0) {
|
|
return 0;
|
|
}
|
|
return form_head_width;
|
|
};
|
|
binding_list_config->should_prevent_inlining = [](FormFormattingConfig /*config*/, int num_refs) {
|
|
// Only prevent inlining a binding list, if there are more than 1 bindings
|
|
if (num_refs > 1) {
|
|
return true;
|
|
}
|
|
return false;
|
|
};
|
|
binding_list_config->prevent_inlining =
|
|
true; // TODO - we only want to prevent inlining if there are more than 2 elements
|
|
cfg.index_configs.emplace(1, binding_list_config);
|
|
return cfg;
|
|
}
|
|
|
|
FormFormattingConfig new_pair_rule(bool combine_first_two_expr) {
|
|
FormFormattingConfig cfg;
|
|
cfg.hang_forms = false;
|
|
cfg.prevent_inlining = true;
|
|
cfg.combine_first_two_lines = combine_first_two_expr;
|
|
auto pair_config = std::make_shared<FormFormattingConfig>();
|
|
pair_config->hang_forms = false;
|
|
pair_config->indentation_width = 1;
|
|
cfg.default_index_config = pair_config;
|
|
return cfg;
|
|
}
|
|
|
|
const std::unordered_map<std::string, FormFormattingConfig> opengoal_form_config = {
|
|
{"case", new_pair_rule(true)},
|
|
{"cond", new_pair_rule(false)},
|
|
{"defmethod", new_flow_rule(3)},
|
|
{"deftype", new_flow_rule_prevent_inlining_indexes(3, {3, 4, 5})},
|
|
{"defun", new_flow_rule(3)},
|
|
{"defun-debug", new_flow_rule(3)},
|
|
{"defbehavior", new_flow_rule(4)},
|
|
{"if", new_permissive_flow_rule()},
|
|
{"define", new_permissive_flow_rule()},
|
|
{"define-extern", new_permissive_flow_rule()},
|
|
{"defmacro", new_flow_rule(3)},
|
|
{"dotimes", new_flow_rule(2)},
|
|
{"let", new_binding_rule(4)},
|
|
{"rlet", new_binding_rule(5)},
|
|
{"when", new_flow_rule(2)},
|
|
{"with-dma-buffer-add-bucket", new_flow_rule(2)}};
|
|
} // namespace config
|
|
} // namespace formatter_rules
|