491 lines
16 KiB
C++
491 lines
16 KiB
C++
/* ----------------------------------------------------------------------
|
|
LAMMPS - Large-scale Atomic/Molecular Massively Parallel Simulator
|
|
https://www.lammps.org/, Sandia National Laboratories
|
|
LAMMPS development team: developers@lammps.org
|
|
|
|
Copyright (2003) Sandia Corporation. Under the terms of Contract
|
|
DE-AC04-94AL85000 with Sandia Corporation, the U.S. Government retains
|
|
certain rights in this software. This software is distributed under
|
|
the GNU General Public License.
|
|
|
|
See the README file in the top-level LAMMPS directory.
|
|
------------------------------------------------------------------------- */
|
|
|
|
#include "codeeditor.h"
|
|
#include "lammpsgui.h"
|
|
#include "linenumberarea.h"
|
|
|
|
#include <QAction>
|
|
#include <QDesktopServices>
|
|
#include <QDragEnterEvent>
|
|
#include <QDropEvent>
|
|
#include <QIcon>
|
|
#include <QKeySequence>
|
|
#include <QMenu>
|
|
#include <QMimeData>
|
|
#include <QPainter>
|
|
#include <QRegularExpression>
|
|
#include <QSettings>
|
|
#include <QTextBlock>
|
|
#include <QUrl>
|
|
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
// Convert string into words on whitespace while handling single and double
|
|
// quotes. Adapted from LAMMPS_NS::utils::split_words() to preserve quotes.
|
|
|
|
static std::vector<std::string> split_line(const std::string &text)
|
|
{
|
|
std::vector<std::string> list;
|
|
const char *buf = text.c_str();
|
|
std::size_t beg = 0;
|
|
std::size_t len = 0;
|
|
std::size_t add = 0;
|
|
char c = *buf;
|
|
|
|
while (c) { // leading whitespace
|
|
if (c == ' ' || c == '\t' || c == '\r' || c == '\n' || c == '\f') {
|
|
c = *++buf;
|
|
++beg;
|
|
continue;
|
|
};
|
|
len = 0;
|
|
|
|
// handle escaped/quoted text.
|
|
quoted:
|
|
|
|
if (c == '\'') { // handle single quote
|
|
add = 0;
|
|
len = 1;
|
|
c = *++buf;
|
|
while (((c != '\'') && (c != '\0')) || ((c == '\\') && (buf[1] == '\''))) {
|
|
if ((c == '\\') && (buf[1] == '\'')) {
|
|
++buf;
|
|
++len;
|
|
}
|
|
c = *++buf;
|
|
++len;
|
|
}
|
|
++len;
|
|
c = *++buf;
|
|
|
|
// handle triple double quotation marks
|
|
} else if ((c == '"') && (buf[1] == '"') && (buf[2] == '"') && (buf[3] != '"')) {
|
|
len = 3;
|
|
add = 1;
|
|
buf += 3;
|
|
c = *buf;
|
|
|
|
} else if (c == '"') { // handle double quote
|
|
add = 0;
|
|
len = 1;
|
|
c = *++buf;
|
|
while (((c != '"') && (c != '\0')) || ((c == '\\') && (buf[1] == '"'))) {
|
|
if ((c == '\\') && (buf[1] == '"')) {
|
|
++buf;
|
|
++len;
|
|
}
|
|
c = *++buf;
|
|
++len;
|
|
}
|
|
++len;
|
|
c = *++buf;
|
|
}
|
|
|
|
while (true) { // unquoted
|
|
if ((c == '\'') || (c == '"')) goto quoted;
|
|
// skip escaped quote
|
|
if ((c == '\\') && ((buf[1] == '\'') || (buf[1] == '"'))) {
|
|
++buf;
|
|
++len;
|
|
c = *++buf;
|
|
++len;
|
|
}
|
|
if ((c == ' ') || (c == '\t') || (c == '\r') || (c == '\n') || (c == '\f') ||
|
|
(c == '\0')) {
|
|
list.push_back(text.substr(beg, len));
|
|
beg += len + add;
|
|
break;
|
|
}
|
|
c = *++buf;
|
|
++len;
|
|
}
|
|
}
|
|
return list;
|
|
}
|
|
|
|
CodeEditor::CodeEditor(QWidget *parent) : QPlainTextEdit(parent), highlight(NO_HIGHLIGHT)
|
|
{
|
|
help_action = new QShortcut(QKeySequence::fromString("Ctrl+?"), parent);
|
|
connect(help_action, &QShortcut::activated, this, &CodeEditor::get_help);
|
|
|
|
// initialize help system
|
|
QFile help_index(":/help_index.table");
|
|
if (help_index.open(QIODevice::ReadOnly | QIODevice::Text)) {
|
|
while (!help_index.atEnd()) {
|
|
auto line = QString(help_index.readLine());
|
|
auto words = line.trimmed().split(' ');
|
|
if (words.size() > 2) {
|
|
if (words.at(1) == "pair_style") {
|
|
pair_map[words.at(2)] = words.at(0);
|
|
} else if (words.at(1) == "bond_style") {
|
|
bond_map[words.at(2)] = words.at(0);
|
|
} else if (words.at(1) == "angle_style") {
|
|
angle_map[words.at(2)] = words.at(0);
|
|
} else if (words.at(1) == "dihedral_style") {
|
|
dihedral_map[words.at(2)] = words.at(0);
|
|
} else if (words.at(1) == "improper_style") {
|
|
improper_map[words.at(2)] = words.at(0);
|
|
} else if (words.at(1) == "fix") {
|
|
fix_map[words.at(2)] = words.at(0);
|
|
} else if (words.at(1) == "compute") {
|
|
compute_map[words.at(2)] = words.at(0);
|
|
} else if (words.at(1) == "kspace_style") {
|
|
cmd_map["kspace_style"] = "kspace_style.html";
|
|
}
|
|
// ignoring: dump, fix_modify ATC
|
|
} else if (words.size() == 2) {
|
|
cmd_map[words.at(1)] = words.at(0);
|
|
} else {
|
|
fprintf(stderr, "unhandled: %s", line.toStdString().c_str());
|
|
}
|
|
}
|
|
help_index.close();
|
|
}
|
|
|
|
lineNumberArea = new LineNumberArea(this);
|
|
connect(this, &CodeEditor::blockCountChanged, this, &CodeEditor::updateLineNumberAreaWidth);
|
|
connect(this, &CodeEditor::updateRequest, this, &CodeEditor::updateLineNumberArea);
|
|
updateLineNumberAreaWidth(0);
|
|
setCursorWidth(2);
|
|
}
|
|
|
|
int CodeEditor::lineNumberAreaWidth()
|
|
{
|
|
int digits = 1;
|
|
int max = qMax(1, blockCount());
|
|
while (max >= 10) {
|
|
max /= 10;
|
|
++digits;
|
|
}
|
|
|
|
int space = 3 + fontMetrics().horizontalAdvance(QLatin1Char('9')) * (digits + 2);
|
|
|
|
return space;
|
|
}
|
|
|
|
void CodeEditor::setFont(const QFont &newfont)
|
|
{
|
|
lineNumberArea->setFont(newfont);
|
|
document()->setDefaultFont(newfont);
|
|
}
|
|
|
|
void CodeEditor::setCursor(int block)
|
|
{
|
|
// move cursor to given position
|
|
auto cursor = textCursor();
|
|
int moves = block - cursor.blockNumber();
|
|
if (moves < 0)
|
|
cursor.movePosition(QTextCursor::Up, QTextCursor::MoveAnchor, -moves);
|
|
else
|
|
cursor.movePosition(QTextCursor::Down, QTextCursor::MoveAnchor, moves);
|
|
setTextCursor(cursor);
|
|
}
|
|
|
|
void CodeEditor::setHighlight(int block, bool error)
|
|
{
|
|
if (error)
|
|
highlight = -block;
|
|
else
|
|
highlight = block;
|
|
|
|
// also reset the cursor
|
|
setCursor(block);
|
|
|
|
// update graphics
|
|
repaint();
|
|
}
|
|
|
|
// reformat line
|
|
|
|
QString CodeEditor::reformatLine(const QString &line)
|
|
{
|
|
auto words = split_line(line.toStdString());
|
|
QString newtext;
|
|
QSettings settings;
|
|
settings.beginGroup("reformat");
|
|
int cmdsize = settings.value("command", "16").toInt();
|
|
int typesize = settings.value("type", "4").toInt();
|
|
int idsize = settings.value("id", "4").toInt();
|
|
int namesize = settings.value("name", "8").toInt();
|
|
settings.endGroup();
|
|
|
|
if (words.size()) {
|
|
// commented line. do nothing
|
|
if (words[0][0] == '#') return line;
|
|
|
|
// start with LAMMPS command plus padding if another word follows
|
|
newtext = words[0].c_str();
|
|
if (words.size() > 1) {
|
|
for (int i = words[0].size() + 1; i < cmdsize; ++i)
|
|
newtext += ' ';
|
|
}
|
|
|
|
// append remaining words with just a single blank added.
|
|
for (int i = 1; i < words.size(); ++i) {
|
|
newtext += ' ';
|
|
newtext += words[i].c_str();
|
|
|
|
// special cases
|
|
|
|
if (i < 3) {
|
|
// additional space for types or type ranges
|
|
if (words[0] == "pair_coeff")
|
|
for (int j = words[i].size(); j < typesize; ++j)
|
|
newtext += ' ';
|
|
|
|
// pad 4 for IDs and 8 for groups
|
|
if ((words[0] == "fix") || (words[0] == "compute") || (words[0] == "dump")) {
|
|
if (i == 1) {
|
|
for (int j = words[i].size(); j < idsize; ++j)
|
|
newtext += ' ';
|
|
} else if (i == 2) {
|
|
for (int j = words[i].size(); j < namesize; ++j)
|
|
newtext += ' ';
|
|
}
|
|
}
|
|
}
|
|
|
|
if (i < 2) {
|
|
if ((words[0] == "bond_coeff") || (words[0] == "angle_coeff") ||
|
|
(words[0] == "dihedral_coeff") || (words[0] == "improper_coeff") ||
|
|
(words[0] == "mass"))
|
|
for (int j = words[i].size(); j < typesize; ++j)
|
|
newtext += ' ';
|
|
}
|
|
}
|
|
}
|
|
return newtext;
|
|
}
|
|
|
|
void CodeEditor::keyPressEvent(QKeyEvent *event)
|
|
{
|
|
if (event->key() == Qt::Key_Tab) {
|
|
reformatCurrentLine();
|
|
return;
|
|
}
|
|
if (event->key() == Qt::Key_Backtab) {
|
|
fprintf(stderr, "Shift + Tab key hit\n");
|
|
return;
|
|
}
|
|
QPlainTextEdit::keyPressEvent(event);
|
|
}
|
|
|
|
void CodeEditor::updateLineNumberAreaWidth(int /* newBlockCount */)
|
|
{
|
|
setViewportMargins(lineNumberAreaWidth(), 0, 0, 0);
|
|
}
|
|
|
|
void CodeEditor::updateLineNumberArea(const QRect &rect, int dy)
|
|
{
|
|
if (dy)
|
|
lineNumberArea->scroll(0, dy);
|
|
else
|
|
lineNumberArea->update(0, rect.y(), lineNumberArea->width(), rect.height());
|
|
|
|
if (rect.contains(viewport()->rect())) updateLineNumberAreaWidth(0);
|
|
}
|
|
|
|
void CodeEditor::dragEnterEvent(QDragEnterEvent *event)
|
|
{
|
|
event->acceptProposedAction();
|
|
}
|
|
|
|
bool CodeEditor::canInsertFromMimeData(const QMimeData *source) const
|
|
{
|
|
return source->hasUrls(); // || source->hasText();
|
|
}
|
|
|
|
void CodeEditor::dropEvent(QDropEvent *event)
|
|
{
|
|
if (event->mimeData()->hasUrls()) {
|
|
event->accept();
|
|
auto file = event->mimeData()->urls()[0].toLocalFile();
|
|
auto gui = dynamic_cast<LammpsGui *>(parent());
|
|
if (gui) {
|
|
moveCursor(QTextCursor::Start, QTextCursor::MoveAnchor);
|
|
gui->open_file(file);
|
|
}
|
|
} else if (event->mimeData()->hasText()) {
|
|
event->accept();
|
|
fprintf(stderr, "Drag - Drop for text block not yet implemented: text=%s\n",
|
|
event->mimeData()->text().toStdString().c_str());
|
|
} else
|
|
event->ignore();
|
|
}
|
|
|
|
void CodeEditor::resizeEvent(QResizeEvent *e)
|
|
{
|
|
QPlainTextEdit::resizeEvent(e);
|
|
|
|
QRect cr = contentsRect();
|
|
lineNumberArea->setGeometry(QRect(cr.left(), cr.top(), lineNumberAreaWidth(), cr.height()));
|
|
}
|
|
|
|
void CodeEditor::lineNumberAreaPaintEvent(QPaintEvent *event)
|
|
{
|
|
QPainter painter(lineNumberArea);
|
|
QTextBlock block = firstVisibleBlock();
|
|
int blockNumber = block.blockNumber();
|
|
|
|
int top = qRound(blockBoundingGeometry(block).translated(contentOffset()).top());
|
|
int bottom = top + qRound(blockBoundingRect(block).height());
|
|
while (block.isValid() && top <= event->rect().bottom()) {
|
|
if (block.isVisible() && bottom >= event->rect().top()) {
|
|
QString number = QString::number(blockNumber + 1) + " ";
|
|
if ((highlight == NO_HIGHLIGHT) || (blockNumber != std::abs(highlight))) {
|
|
painter.setPen(Qt::black);
|
|
} else {
|
|
number = QString(">") + QString::number(blockNumber + 1) + "<";
|
|
if (highlight < 0)
|
|
painter.fillRect(0, top, lineNumberArea->width(), fontMetrics().height(),
|
|
Qt::darkRed);
|
|
else
|
|
painter.fillRect(0, top, lineNumberArea->width(), fontMetrics().height(),
|
|
Qt::darkGreen);
|
|
|
|
painter.setPen(Qt::white);
|
|
}
|
|
painter.drawText(0, top, lineNumberArea->width(), fontMetrics().height(),
|
|
Qt::AlignRight, number);
|
|
}
|
|
|
|
block = block.next();
|
|
top = bottom;
|
|
bottom = top + qRound(blockBoundingRect(block).height());
|
|
++blockNumber;
|
|
}
|
|
}
|
|
|
|
void CodeEditor::contextMenuEvent(QContextMenuEvent *event)
|
|
{
|
|
// reposition the cursor here?
|
|
QString page, help;
|
|
find_help(page, help);
|
|
|
|
// print augmented context menu if an entry was found
|
|
auto *menu = createStandardContextMenu();
|
|
if (!page.isEmpty()) {
|
|
menu->addSeparator();
|
|
auto action = menu->addAction(QString("Reformat '%1' command").arg(help));
|
|
action->setIcon(QIcon(":/format-indent-less-3.png"));
|
|
connect(action, &QAction::triggered, this, &CodeEditor::reformatCurrentLine);
|
|
|
|
menu->addSeparator();
|
|
action = menu->addAction(QString("View Documentation for '%1'").arg(help));
|
|
action->setIcon(QIcon(":/system-help.png"));
|
|
action->setData(page);
|
|
connect(action, &QAction::triggered, this, &CodeEditor::open_help);
|
|
// if we link to help with specific styles (fix, compute, pair, bond, ...)
|
|
// also link to the docs for the primary command
|
|
auto words = help.split(' ');
|
|
if (words.size() > 1) {
|
|
help = words.at(0);
|
|
page = words.at(0);
|
|
page += ".html";
|
|
auto action2 = menu->addAction(QString("View Documentation for '%1'").arg(help));
|
|
action2->setIcon(QIcon(":/system-help.png"));
|
|
action2->setData(page);
|
|
connect(action2, &QAction::triggered, this, &CodeEditor::open_help);
|
|
}
|
|
}
|
|
auto action3 = menu->addAction(QString("LAMMPS Manual"));
|
|
action3->setIcon(QIcon(":/help-browser.png"));
|
|
action3->setData(QString());
|
|
connect(action3, &QAction::triggered, this, &CodeEditor::open_help);
|
|
|
|
menu->exec(event->globalPos());
|
|
delete menu;
|
|
}
|
|
|
|
void CodeEditor::reformatCurrentLine()
|
|
{
|
|
auto cursor = textCursor();
|
|
auto text = cursor.block().text();
|
|
auto newtext = reformatLine(text);
|
|
|
|
// perform edit but only if text has changed
|
|
if (QString::compare(text, newtext)) {
|
|
cursor.beginEditBlock();
|
|
cursor.movePosition(QTextCursor::StartOfLine);
|
|
cursor.movePosition(QTextCursor::EndOfLine, QTextCursor::KeepAnchor, 1);
|
|
cursor.insertText(newtext);
|
|
cursor.endEditBlock();
|
|
}
|
|
}
|
|
|
|
void CodeEditor::get_help()
|
|
{
|
|
QString page, help;
|
|
find_help(page, help);
|
|
if (!page.isEmpty())
|
|
QDesktopServices::openUrl(QUrl(QString("https://docs.lammps.org/%1").arg(page)));
|
|
}
|
|
|
|
void CodeEditor::find_help(QString &page, QString &help)
|
|
{
|
|
// process line of text where the cursor is
|
|
auto text = textCursor().block().text().replace('\t', ' ').trimmed();
|
|
auto style =
|
|
QRegularExpression("^(pair|bond|angle|dihedral|improper)_style\\s+(\\S+)").match(text);
|
|
help.clear();
|
|
page.clear();
|
|
if (style.hasMatch()) {
|
|
if (style.captured(1) == "pair") {
|
|
page = pair_map.value(style.captured(2), QString());
|
|
help = QString("pair_style %1").arg(style.captured(2));
|
|
} else if (style.captured(1) == "bond") {
|
|
page = bond_map.value(style.captured(2), QString());
|
|
help = QString("bond_style %1").arg(style.captured(2));
|
|
} else if (style.captured(1) == "angle") {
|
|
page = angle_map.value(style.captured(2), QString());
|
|
help = QString("angle_style %1").arg(style.captured(2));
|
|
} else if (style.captured(1) == "dihedral") {
|
|
page = dihedral_map.value(style.captured(2), QString());
|
|
help = QString("dihedral_style %1").arg(style.captured(2));
|
|
} else if (style.captured(1) == "improper") {
|
|
page = improper_map.value(style.captured(2), QString());
|
|
help = QString("improper_style %1").arg(style.captured(2));
|
|
}
|
|
}
|
|
|
|
style = QRegularExpression("^(fix|compute)\\s+\\w+\\s+\\w+\\s+(\\S+)").match(text);
|
|
if (style.hasMatch()) {
|
|
help = QString("%1 %2").arg(style.captured(1), style.captured(2));
|
|
if (style.captured(1) == "fix") {
|
|
page = fix_map.value(style.captured(2), QString());
|
|
} else if (style.captured(1) == "compute") {
|
|
page = compute_map.value(style.captured(2), QString());
|
|
}
|
|
}
|
|
|
|
// could not find a matching "style", now try the plain command
|
|
if (page.isEmpty() && !text.isEmpty()) {
|
|
auto cmd = text.split(' ').at(0);
|
|
help = cmd;
|
|
page = cmd_map.value(cmd, QString());
|
|
}
|
|
}
|
|
|
|
void CodeEditor::open_help()
|
|
{
|
|
QAction *act = qobject_cast<QAction *>(sender());
|
|
QDesktopServices::openUrl(
|
|
QUrl(QString("https://docs.lammps.org/%1").arg(act->data().toString())));
|
|
}
|
|
|
|
// Local Variables:
|
|
// c-basic-offset: 4
|
|
// End:
|