////////////////////////////////////////////////////////////////////////
//
// Copyright (C) 1996-2023 The Octave Project Developers
//
// See the file COPYRIGHT.md in the top-level directory of this
// distribution or <https://octave.org/copyright/>.
//
// This file is part of Octave.
//
// Octave is free software: you can redistribute it and/or modify it
// under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// Octave is distributed in the hope that it will be useful, but
// WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with Octave; see the file COPYING.  If not, see
// <https://www.gnu.org/licenses/>.
//
////////////////////////////////////////////////////////////////////////

#if ! defined (octave_pt_loop_h)
#define octave_pt_loop_h 1

#include "octave-config.h"

class octave_value;

#include "pt-cmd.h"
#include "pt-walk.h"

OCTAVE_BEGIN_NAMESPACE(octave)

class tree_argument_list;
class tree_expression;
class tree_statement_list;

// While.

class tree_while_command : public tree_command
{
public:

  tree_while_command (int l = -1, int c = -1)
    : tree_command (l, c), m_expr (nullptr), m_list (nullptr),
      m_lead_comm (nullptr), m_trail_comm (nullptr)
  { }

  tree_while_command (tree_expression *e,
                      comment_list *lc = nullptr,
                      comment_list *tc = nullptr,
                      int l = -1, int c = -1)
    : tree_command (l, c), m_expr (e), m_list (nullptr),
      m_lead_comm (lc), m_trail_comm (tc)
  { }

  tree_while_command (tree_expression *e, tree_statement_list *lst,
                      comment_list *lc = nullptr,
                      comment_list *tc = nullptr,
                      int l = -1, int c = -1)
    : tree_command (l, c), m_expr (e), m_list (lst), m_lead_comm (lc),
      m_trail_comm (tc)
  { }

  OCTAVE_DISABLE_COPY_MOVE (tree_while_command)

  ~tree_while_command ();

  tree_expression * condition () { return m_expr; }

  tree_statement_list * body () { return m_list; }

  comment_list * leading_comment () { return m_lead_comm; }

  comment_list * trailing_comment () { return m_trail_comm; }

  void accept (tree_walker& tw)
  {
    tw.visit_while_command (*this);
  }

protected:

  // Expression to test.
  tree_expression *m_expr;

  // List of commands to execute.
  tree_statement_list *m_list;

  // Comment preceding WHILE token.
  comment_list *m_lead_comm;

  // Comment preceding ENDWHILE token.
  comment_list *m_trail_comm;
};

// Do-Until.

class tree_do_until_command : public tree_while_command
{
public:

  tree_do_until_command (int l = -1, int c = -1)
    : tree_while_command (l, c)
  { }

  tree_do_until_command (tree_expression *e,
                         comment_list *lc = nullptr,
                         comment_list *tc = nullptr,
                         int l = -1, int c = -1)
    : tree_while_command (e, lc, tc, l, c)
  { }

  tree_do_until_command (tree_expression *e, tree_statement_list *lst,
                         comment_list *lc = nullptr,
                         comment_list *tc = nullptr,
                         int l = -1, int c = -1)
    : tree_while_command (e, lst, lc, tc, l, c)
  { }

  OCTAVE_DISABLE_COPY_MOVE (tree_do_until_command)

  ~tree_do_until_command () = default;

  void accept (tree_walker& tw)
  {
    tw.visit_do_until_command (*this);
  }
};

// For.

class tree_simple_for_command : public tree_command
{
public:

  tree_simple_for_command (int l = -1, int c = -1)
    : tree_command (l, c), m_parallel (false), m_lhs (nullptr),
      m_expr (nullptr), m_maxproc (nullptr), m_list (nullptr),
      m_lead_comm (nullptr), m_trail_comm (nullptr)
  { }

  tree_simple_for_command (bool parallel_arg, tree_expression *le,
                           tree_expression *re,
                           tree_expression *maxproc_arg,
                           tree_statement_list *lst,
                           comment_list *lc = nullptr,
                           comment_list *tc = nullptr,
                           int l = -1, int c = -1)
    : tree_command (l, c), m_parallel (parallel_arg), m_lhs (le),
      m_expr (re), m_maxproc (maxproc_arg), m_list (lst),
      m_lead_comm (lc), m_trail_comm (tc)
  { }

  OCTAVE_DISABLE_COPY_MOVE (tree_simple_for_command)

  ~tree_simple_for_command ();

  bool in_parallel () { return m_parallel; }

  tree_expression * left_hand_side () { return m_lhs; }

  tree_expression * control_expr () { return m_expr; }

  tree_expression * maxproc_expr () { return m_maxproc; }

  tree_statement_list * body () { return m_list; }

  comment_list * leading_comment () { return m_lead_comm; }

  comment_list * trailing_comment () { return m_trail_comm; }

  void accept (tree_walker& tw)
  {
    tw.visit_simple_for_command (*this);
  }

private:
  // TRUE means operate in parallel (subject to the value of the
  // maxproc expression).
  bool m_parallel;

  // Expression to modify.
  tree_expression *m_lhs;

  // Expression to evaluate.
  tree_expression *m_expr;

  // Expression to tell how many processors should be used (only valid
  // if parallel is TRUE).
  tree_expression *m_maxproc;

  // List of commands to execute.
  tree_statement_list *m_list;

  // Comment preceding FOR token.
  comment_list *m_lead_comm;

  // Comment preceding ENDFOR token.
  comment_list *m_trail_comm;
};

class tree_complex_for_command : public tree_command
{
public:

  tree_complex_for_command (int l = -1, int c = -1)
    : tree_command (l, c), m_lhs (nullptr), m_expr (nullptr),
      m_list (nullptr), m_lead_comm (nullptr), m_trail_comm (nullptr)
  { }

  tree_complex_for_command (tree_argument_list *le, tree_expression *re,
                            tree_statement_list *lst,
                            comment_list *lc = nullptr,
                            comment_list *tc = nullptr,
                            int l = -1, int c = -1)
    : tree_command (l, c), m_lhs (le), m_expr (re), m_list (lst),
      m_lead_comm (lc), m_trail_comm (tc)
  { }

  OCTAVE_DISABLE_COPY_MOVE (tree_complex_for_command)

  ~tree_complex_for_command ();

  tree_argument_list * left_hand_side () { return m_lhs; }

  tree_expression * control_expr () { return m_expr; }

  tree_statement_list * body () { return m_list; }

  comment_list * leading_comment () { return m_lead_comm; }

  comment_list * trailing_comment () { return m_trail_comm; }

  void accept (tree_walker& tw)
  {
    tw.visit_complex_for_command (*this);
  }

private:

  // Expression to modify.
  tree_argument_list *m_lhs;

  // Expression to evaluate.
  tree_expression *m_expr;

  // List of commands to execute.
  tree_statement_list *m_list;

  // Comment preceding FOR token.
  comment_list *m_lead_comm;

  // Comment preceding ENDFOR token.
  comment_list *m_trail_comm;
};

OCTAVE_END_NAMESPACE(octave)

#endif