pqPythonCalculatorCompleter.h

Go to the documentation of this file.

// SPDX-FileCopyrightText: Copyright (c) Kitware Inc.
// SPDX-License-Identifier: BSD-3-Clause
#ifndef pqPythonCalculatorCompleter_h
#define pqPythonCalculatorCompleter_h

#include <vtkPython.h>

#include "pqPythonCompleter.h"
#include "vtkSMSourceProxy.h"

class PQPYTHON_EXPORT pqPythonCalculatorCompleter : public pqPythonCompleter
{
public:
  pqPythonCalculatorCompleter(QWidget* parent, vtkSMSourceProxy* input)
    : pqPythonCompleter(parent)
    , Input(input)
  {
    this->setCompleteEmptyPrompts(true); // Ovrerride pqPythonCompleter default
  }

protected:
  QStringList getPythonCompletions(const QString& pythonObjectName) override;

  void AddArrayNamesToResults(
    const vtkPVDataSetAttributesInformation* attributesInformation, QStringList& results);

private:
  vtkSMSourceProxy* Input; // Needed to retrieve input field names
};
#endif