ICODE-ADC/lammps
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add completion for compute and fix ID references

Browse Source
This commit is contained in:
Axel Kohlmeyer
2023-09-08 10:43:31 -04:00
parent 778263673c
commit f8e65b19a2
3 changed files with 100 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

97
tools/lammps-gui/codeeditor.cpp
View File

@ -296,6 +296,8 @@ QString CodeEditor::reformatLine(const QString &line)
bool rebuildGroupComp = false;
bool rebuildVarNameComp = false;
bool rebuildComputeIDComp = false;
bool rebuildFixIDComp = false;
if (words.size()) {
// commented line. do nothing
@ -310,6 +312,10 @@ QString CodeEditor::reformatLine(const QString &line)
if (words[0] == "group") rebuildGroupComp = true;
// new/updated variable command -> update completer
if (words[0] == "variable") rebuildVarNameComp = true;
// new/updated compute command -> update completer
if (words[0] == "compute") rebuildComputeIDComp = true;
// new/updated fix command -> update completer
if (words[0] == "fix") rebuildFixIDComp = true;
}
// append remaining words with just a single blank added.
@ -348,6 +354,8 @@ QString CodeEditor::reformatLine(const QString &line)
}
if (rebuildGroupComp) setGroupList();
if (rebuildVarNameComp) setVarNameList();
if (rebuildComputeIDComp) setComputeIDList();
if (rebuildFixIDComp) setFixIDList();
return newtext;
}
@ -434,6 +442,54 @@ void CodeEditor::setVarNameList()
varname_comp->setModel(new QStringListModel(vars, varname_comp));
}
void CodeEditor::setComputeIDList()
{
QStringList compid;
QRegularExpression compcmd(QStringLiteral("^\\s*compute\\s+(\\S+)\\s+"));
auto saved = textCursor();
// reposition cursor to beginning of text and search for group commands
auto cursor = textCursor();
cursor.movePosition(QTextCursor::Start);
setTextCursor(cursor);
while (find(compcmd)) {
auto words = textCursor().block().text().replace('\t', ' ').split(' ', Qt::SkipEmptyParts);
if ((words.size() > 1)) {
QString w = QString("c_%1").arg(words[1]);
if (!compid.contains(w)) compid << w;
w = QString("C_%1").arg(words[1]);
if (!compid.contains(w)) compid << w;
}
}
compid.sort();
setTextCursor(saved);
compid_comp->setModel(new QStringListModel(compid, compid_comp));
}
void CodeEditor::setFixIDList()
{
QStringList fixid;
QRegularExpression fixcmd(QStringLiteral("^\\s*fix\\s+(\\S+)\\s+"));
auto saved = textCursor();
// reposition cursor to beginning of text and search for group commands
auto cursor = textCursor();
cursor.movePosition(QTextCursor::Start);
setTextCursor(cursor);
while (find(fixcmd)) {
auto words = textCursor().block().text().replace('\t', ' ').split(' ', Qt::SkipEmptyParts);
if ((words.size() > 1)) {
QString w = QString("f_%1").arg(words[1]);
if (!fixid.contains(w)) fixid << w;
w = QString("F_%1").arg(words[1]);
if (!fixid.contains(w)) fixid << w;
}
}
fixid.sort();
setTextCursor(saved);
fixid_comp->setModel(new QStringListModel(fixid, fixid_comp));
}
void CodeEditor::keyPressEvent(QKeyEvent *event)
{
const auto key = event->key();
@ -731,6 +787,14 @@ void CodeEditor::runCompletion()
current_comp = group_comp;
else if (selected.startsWith("v_"))
current_comp = varname_comp;
else if (selected.startsWith("c_"))
current_comp = compid_comp;
else if (selected.startsWith("C_"))
current_comp = compid_comp;
else if (selected.startsWith("f_"))
current_comp = fixid_comp;
else if (selected.startsWith("F_"))
current_comp = fixid_comp;
if (current_comp) {
current_comp->setCompletionPrefix(words[1].c_str());
@ -764,6 +828,14 @@ void CodeEditor::runCompletion()
current_comp = group_comp;
else if (selected.startsWith("v_"))
current_comp = varname_comp;
else if (selected.startsWith("c_"))
current_comp = compid_comp;
else if (selected.startsWith("C_"))
current_comp = compid_comp;
else if (selected.startsWith("f_"))
current_comp = fixid_comp;
else if (selected.startsWith("F_"))
current_comp = fixid_comp;
if (current_comp) {
current_comp->setCompletionPrefix(words[2].c_str());
@ -793,6 +865,14 @@ void CodeEditor::runCompletion()
current_comp = dump_comp;
else if (selected.startsWith("v_"))
current_comp = varname_comp;
else if (selected.startsWith("c_"))
current_comp = compid_comp;
else if (selected.startsWith("C_"))
current_comp = compid_comp;
else if (selected.startsWith("f_"))
current_comp = fixid_comp;
else if (selected.startsWith("F_"))
current_comp = fixid_comp;
if (current_comp) {
current_comp->setCompletionPrefix(words[3].c_str());
@ -809,15 +889,24 @@ void CodeEditor::runCompletion()
current_comp->complete(cr);
}
// reference located anywhere further right in the line
} else if (words.size() > 3) {
} else if (words.size() > 4) {
current_comp = nullptr;
if (selected.startsWith("v_")) current_comp = varname_comp;
if (selected.startsWith("v_"))
current_comp = varname_comp;
else if (selected.startsWith("c_"))
current_comp = compid_comp;
else if (selected.startsWith("C_"))
current_comp = compid_comp;
else if (selected.startsWith("f_"))
current_comp = fixid_comp;
else if (selected.startsWith("F_"))
current_comp = fixid_comp;
if (current_comp) {
current_comp->setCompletionPrefix(words[3].c_str());
current_comp->setCompletionPrefix(selected);
auto popup = current_comp->popup();
// if the command is already a complete command, remove existing popup
if (words[3] == current_comp->currentCompletion().toStdString()) {
if (selected == current_comp->currentCompletion()) {
if (popup->isVisible()) popup->hide();
return;
}

2
tools/lammps-gui/codeeditor.h
View File

@ -58,6 +58,8 @@ public:
void setUnitsList(const QStringList &words);
void setGroupList();
void setVarNameList();
void setComputeIDList();
void setFixIDList();
static constexpr int NO_HIGHLIGHT = 1 << 30;

2
tools/lammps-gui/lammpsgui.cpp
View File

@ -629,6 +629,8 @@ void LammpsGui::open_file(const QString &fileName)
ui->textEdit->document()->setModified(false);
ui->textEdit->setGroupList();
ui->textEdit->setVarNameList();
ui->textEdit->setComputeIDList();
ui->textEdit->setFixIDList();
file.close();
dirstatus->setText(QString(" Directory: ") + current_dir);
status->setText("Ready.");