////////////////////////////////////////////////////////////////////////
//
// 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_select_h)
#define octave_pt_select_h 1

#include "octave-config.h"

#include "base-list.h"
#include "comment-list.h"
#include "pt-cmd.h"
#include "pt-walk.h"

OCTAVE_BEGIN_NAMESPACE(octave)

class tree_expression;
class tree_statement_list;

// If.

class tree_if_clause : public tree
{
public:

  tree_if_clause (int l = -1, int c = -1)
    : tree (l, c), m_expr (nullptr), m_list (nullptr), m_lead_comm (nullptr)
  { }

  tree_if_clause (tree_statement_list *sl, comment_list *lc = nullptr,
                  int l = -1, int c = -1)
    : tree (l, c), m_expr (nullptr), m_list (sl), m_lead_comm (lc) { }

  tree_if_clause (tree_expression *e, tree_statement_list *sl,
                  comment_list *lc = nullptr,
                  int l = -1, int c = -1)
    : tree (l, c), m_expr (e), m_list (sl), m_lead_comm (lc) { }

  OCTAVE_DISABLE_COPY_MOVE (tree_if_clause)

  ~tree_if_clause ();

  bool is_else_clause () { return ! m_expr; }

  tree_expression * condition () { return m_expr; }

  tree_statement_list * commands () { return m_list; }

  comment_list * leading_comment () { return m_lead_comm; }

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

private:

  // The condition to test.
  tree_expression *m_expr;

  // The list of statements to evaluate if expr is true.
  tree_statement_list *m_list;

  // Comment preceding ELSE or ELSEIF token.
  comment_list *m_lead_comm;
};

class tree_if_command_list : public base_list<tree_if_clause *>
{
public:

  tree_if_command_list () { }

  tree_if_command_list (tree_if_clause *t) { append (t); }

  OCTAVE_DISABLE_COPY_MOVE (tree_if_command_list)

  ~tree_if_command_list ()
  {
    while (! empty ())
      {
        auto p = begin ();
        delete *p;
        erase (p);
      }
  }

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

class tree_if_command : public tree_command
{
public:

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

  tree_if_command (tree_if_command_list *lst, comment_list *lc,
                   comment_list *tc, int l = -1, int c = -1)
    : tree_command (l, c), m_list (lst), m_lead_comm (lc), m_trail_comm (tc)
  { }

  OCTAVE_DISABLE_COPY_MOVE (tree_if_command)

  ~tree_if_command ();

  tree_if_command_list * cmd_list () { 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_if_command (*this);
  }

private:

  // List of if commands (if, elseif, elseif, ... else, endif)
  tree_if_command_list *m_list;

  // Comment preceding IF token.
  comment_list *m_lead_comm;

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

// Switch.

class tree_switch_case : public tree
{
public:

  tree_switch_case (int l = -1, int c = -1)
    : tree (l, c), m_label (nullptr), m_list (nullptr), m_lead_comm (nullptr)
  { }

  tree_switch_case (tree_statement_list *sl, comment_list *lc = nullptr,
                    int l = -1, int c = -1)
    : tree (l, c), m_label (nullptr), m_list (sl), m_lead_comm (lc) { }

  tree_switch_case (tree_expression *e, tree_statement_list *sl,
                    comment_list *lc = nullptr,
                    int l = -1, int c = -1)
    : tree (l, c), m_label (e), m_list (sl), m_lead_comm (lc) { }

  OCTAVE_DISABLE_COPY_MOVE (tree_switch_case)

  ~tree_switch_case ();

  bool is_default_case () { return ! m_label; }

  tree_expression * case_label () { return m_label; }

  tree_statement_list * commands () { return m_list; }

  comment_list * leading_comment () { return m_lead_comm; }

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

private:

  // The case label.
  tree_expression *m_label;

  // The list of statements to evaluate if the label matches.
  tree_statement_list *m_list;

  // Comment preceding CASE or OTHERWISE token.
  comment_list *m_lead_comm;
};

class tree_switch_case_list : public base_list<tree_switch_case *>
{
public:

  tree_switch_case_list () { }

  tree_switch_case_list (tree_switch_case *t) { append (t); }

  OCTAVE_DISABLE_COPY_MOVE (tree_switch_case_list)

  ~tree_switch_case_list ()
  {
    while (! empty ())
      {
        auto p = begin ();
        delete *p;
        erase (p);
      }
  }

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

class tree_switch_command : public tree_command
{
public:

  tree_switch_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_switch_command (tree_expression *e, tree_switch_case_list *lst,
                       comment_list *lc, comment_list *tc,
                       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_switch_command)

  ~tree_switch_command ();

  tree_expression * switch_value () { return m_expr; }

  tree_switch_case_list * case_list () { 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_switch_command (*this);
  }

private:

  // Value on which to switch.
  tree_expression *m_expr;

  // List of cases (case 1, case 2, ..., default)
  tree_switch_case_list *m_list;

  // Comment preceding SWITCH token.
  comment_list *m_lead_comm;

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

OCTAVE_END_NAMESPACE(octave)

#endif