// -*- coding: utf-8 -*-
// Copyright (C) 2015 Laboratoire de Recherche et
// Développement de l'Epita.
//
// This file is part of Spot, a model checking library.
//
// Spot 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.
//
// Spot 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 this program.  If not, see <http://www.gnu.org/licenses/>.

#include <algorithm>
#include <deque>
#include <stack>
#include <utility>
#include <unordered_map>

#include "safra.hh"
#include "twaalgos/degen.hh"
#include "twaalgos/simulation.hh"

namespace spot
{
  namespace
  {
    using power_set = std::map<safra_state, int>;
    const char* const sub[10] =
    {
      "\u2080",
      "\u2081",
      "\u2082",
      "\u2083",
      "\u2084",
      "\u2085",
      "\u2086",
      "\u2087",
      "\u2088",
      "\u2089",
    };

    std::string subscript(unsigned start)
    {
      std::string res;
      do
        {
          res = sub[start % 10] + res;
          start /= 10;
        }
      while (start);
      return res;
    }

    // Returns true if lhs has a smaller nesting pattern than rhs
    // If lhs and rhs are the same, return false.
    bool nesting_cmp(const std::vector<node_helper::brace_t>& lhs,
                     const std::vector<node_helper::brace_t>& rhs)
    {
      size_t m = std::min(lhs.size(), rhs.size());
      size_t i = 0;
      for (; i < m; ++i)
        {
          if (lhs[i] != rhs[i])
            return lhs[i] < rhs[i];
        }
      return lhs.size() > rhs.size();
    }

    // Used to remove all acceptance whos value is above and equal max_acc
    void remove_dead_acc(twa_graph_ptr& aut, unsigned max_acc)
    {
      assert(max_acc < 32);
      unsigned mask = (1 << max_acc) - 1;
      for (auto& t: aut->edges())
        {
          t.acc &= mask;
        }
    }

    struct compare
    {
      bool
      operator() (const safra_state::safra_node_t& lhs,
                  const safra_state::safra_node_t& rhs)
      {
        return lhs.second < rhs.second;
      }
    };

    // Return the sorteds nodes in ascending order
    std::vector<safra_state::safra_node_t>
    sorted_nodes(const safra_state::nodes_t& nodes)
    {
      std::vector<safra_state::safra_node_t> res;
      for (auto& n: nodes)
        res.emplace_back(n.first, n.second);
      std::sort(res.begin(), res.end(), compare());
      return res;
    }

    std::string
    nodes_to_string(const safra_state::nodes_t& states)
    {
      auto copy = sorted_nodes(states);
      std::ostringstream os;
      std::stack<unsigned> s;
      bool first = true;
      for (auto& n: copy)
        {
          auto it = n.second.begin();
          // Find brace on top of stack in vector
          // If brace is not present, then we close it as no other ones of that
          // type will be found since we ordered our vector
          while (!s.empty())
            {
              it = std::lower_bound(n.second.begin(), n.second.end(),
                                         s.top());
              if (it == n.second.end() || *it != s.top())
                {
                  os << subscript(s.top()) << '}';
                  s.pop();
                }
              else
                {
                  if (*it == s.top())
                    ++it;
                  break;
                }
            }
          // Add new braces
          while (it != n.second.end())
            {
              os << '{' << subscript(*it);
              s.push(*it);
              ++it;
              first = true;
            }
          if (!first)
            os << ' ';
          os << n.first;
          first = false;
        }
      // Finish unwinding stack to print last braces
      while (!s.empty())
        {
          os << subscript(s.top()) << '}';
          s.pop();
        }
      return os.str();
    }

    std::vector<std::string>*
    print_debug(std::map<safra_state, int>& states)
    {
      auto res = new std::vector<std::string>(states.size());
      for (auto& p: states)
        (*res)[p.second] = nodes_to_string(p.first.nodes_);
      return res;
    }

  }

  void
  safra_state::compute_succs(const const_twa_graph_ptr& aut,
                             const std::vector<bdd_id_t>& bddnums,
                             std::unordered_map<bdd,
                                             std::pair<unsigned, unsigned>,
                                             bdd_hash>& deltas,
                              succs_t& res,
                              const scc_info& scc,
                              bool track_scc) const
  {
    // Given a bdd returns index of associated safra_state in res
    std::map<unsigned, unsigned> bdd2num;
    for (auto& node: nodes_)
      {
        for (auto& t: aut->out(node.first))
          {
            // deltas precalculate all transitions in edge t
            auto p = deltas[t.cond];
            for (unsigned j = p.first; j < p.second; ++j)
              {
                auto i = bdd2num.insert(std::make_pair(bddnums[j], res.size()));
                unsigned idx;
                if (!i.second)
                  // No new state added, so get pre-existing state whichi is an
                  // index in the vector of safra states (res)
                  idx = i.first->second;
                else
                  {
                    // Each new node starts out with same number of nodes as src
                    idx = res.size();
                    res.emplace_back(safra_state(nb_braces_.size()),
                                     bddnums[j]);
                  }
                safra_state& ss = res[idx].first;
                // Check if we are leaving the SCC, if so we delete all the
                // braces as no cycles can be found with that node
                if (track_scc && scc.scc_of(node.first) != scc.scc_of(t.dst))
                  ss.update_succ({ /* no braces */ }, t.dst, t.acc);
                else
                  ss.update_succ(node.second, t.dst, t.acc);
                assert(ss.nb_braces_.size() == ss.is_green_.size());
              }
          }
      }
    for (auto& s: res)
      {
        safra_state& tmp = s.first;
        tmp.ungreenify_last_brace();
        s.first.color_ = tmp.finalize_construction();
      }
  }

  void safra_state::ungreenify_last_brace()
  {
    // Step A4: For a brace to emit green it must surround other braces.
    // Hence, the last brace cannot emit green as it is the most inside brace.
    for (auto& n: nodes_)
      {
        if (!n.second.empty())
          is_green_[n.second.back()] = false;
      }
  }

  safra_state::color_t safra_state::finalize_construction()
  {
    unsigned red = -1U;
    unsigned green = -1U;
    std::vector<unsigned> rem_succ_of;
    assert(is_green_.size() == nb_braces_.size());
    for (unsigned i = 0; i < is_green_.size(); ++i)
      {
        if (nb_braces_[i] == 0)
          {
            // Step A3: Brackets that do not contain any nodes emit red
            is_green_[i] = false;

            // First brace can now be zero with new optim making it possible to
            // emit red 0
            red = std::min(red, 2 * i);
          }
        else if (is_green_[i])
          {
            green = std::min(green, 2 * i + 1);
            // Step A4 Emit green
            rem_succ_of.emplace_back(i);
          }
      }
    for (auto& n: nodes_)
      {
        // Step A4 Remove all brackets inside each green pair
        node_helper::truncate_braces(n.second, rem_succ_of, nb_braces_);
      }

    // Step A5 define the rem variable
    std::vector<unsigned> decr_by(nb_braces_.size());
    unsigned decr = 0;
    for (unsigned i = 0; i < nb_braces_.size(); ++i)
      {
        // Step A5 renumber braces
        nb_braces_[i - decr] = nb_braces_[i];
        if (nb_braces_[i] == 0)
          {
            ++decr;
          }
        // Step A5, renumber braces
        decr_by[i] = decr;
      }
    nb_braces_.resize(nb_braces_.size() - decr);
    for (auto& n: nodes_)
      {
        node_helper::renumber(n.second, decr_by);
      }
    return std::min(red, green);
  }

  void node_helper::renumber(std::vector<brace_t>& braces,
                             const std::vector<unsigned>& decr_by)
  {
    for (unsigned idx = 0; idx < braces.size(); ++idx)
      {
        braces[idx] -= decr_by[braces[idx]];
      }
  }

  void
  node_helper::truncate_braces(std::vector<brace_t>& braces,
                                const std::vector<unsigned>& rem_succ_of,
                                std::vector<size_t>& nb_braces)
  {
    for (unsigned idx = 0; idx < braces.size(); ++idx)
      {
        bool found = false;
        // find first brace that matches rem_succ_of
        for (auto s: rem_succ_of)
          {
            found |= braces[idx] == s;
          }
        if (found)
          {
            assert(idx < braces.size() - 1);
            // For each deleted brace, decrement elements of nb_braces
            // This corresponds to A4 step
            for (unsigned i = idx + 1; i < braces.size(); ++i)
              {
                --nb_braces[braces[i]];
              }
            braces.resize(idx + 1);
            break;
          }
      }
  }

  void safra_state::update_succ(const std::vector<node_helper::brace_t>& braces,
                                state_t dst, const acc_cond::mark_t acc)
  {
    std::vector<node_helper::brace_t> copy = braces;
    if (acc.count())
      {
        assert(acc.has(0) && acc.count() == 1 && "Only one TBA are accepted");
        // Accepting edge generate new braces: step A1
        copy.emplace_back(nb_braces_.size());
        // nb_braces_ gets updated later so put 0 for now
        nb_braces_.emplace_back(0);
        // Newly created braces cannot emit green as they won't have
        // any braces inside them (by construction)
        is_green_.push_back(false);
      }
    auto i = nodes_.emplace(dst, copy);
    if (!i.second)
      {
        // Step A2: Only keep the smallest nesting pattern (i-e  braces_) for
        // identical nodes.  Nesting_cmp returnes true if copy is smaller
        if (nesting_cmp(copy, i.first->second))
          {
            // Remove brace count of replaced node
            for (auto b: i.first->second)
              --nb_braces_[b];
            i.first->second = std::move(copy);
          }
        else
          // Node already exists and has bigger nesting pattern value
          return;
      }
    // After inserting new node, update the brace count per node
    for (auto b: i.first->second)
      ++nb_braces_[b];
  }

  // Called only to initialize first state
  safra_state::safra_state(state_t val, bool init_state, bool accepting_scc)
  {
    if (init_state)
      {
        unsigned state_num = val;
        if (!accepting_scc)
          {
            std::vector<node_helper::brace_t> braces = { /* no braces */ };
            nodes_.emplace(state_num, std::move(braces));
          }
        else
          {
            std::vector<node_helper::brace_t> braces = { 0 };
            nodes_.emplace(state_num, std::move(braces));
            // First brace has init_state hence one state inside the first
            // braces.
            nb_braces_.push_back(1);
            // One brace set
            is_green_.push_back(true);
          }
      }
    else
      {
        unsigned nb_braces = val;
        // One brace set
        is_green_.assign(nb_braces, true);
        // Newly created states are the empty mocrstate
        nb_braces_.assign(nb_braces, 0);
      }
  }

  bool
  safra_state::operator<(const safra_state& other) const
  {
    return nodes_ < other.nodes_;
  }

  // All edge going into an accepting SCC are marked as accepting
  // This helps safra make fewer trees as we immediately see the accepting
  // edge and don't create useless states
  void
  emit_accepting_scc(twa_graph_ptr& aut, scc_info& scc)
  {
    for (auto& e: aut->edges())
      {
        unsigned src_scc = scc.scc_of(e.src);
        unsigned dst_scc = scc.scc_of(e.dst);
        if (src_scc == dst_scc)
          continue;
        if (scc.is_accepting_scc(dst_scc))
          e.acc |= 1;
      }
  }

  twa_graph_ptr
  tgba_determinisation(const const_twa_graph_ptr& a, bool bisimulation,
                       bool pretty_print, bool emit_scc, bool track_scc)
  {
    // Degeneralize
    twa_graph_ptr aut = spot::degeneralize_tba(a);
    scc_info scc = scc_info(aut);
    if (emit_scc)
      emit_accepting_scc(aut, scc);


    bdd allap = bddtrue;
    {
      typedef std::set<bdd, bdd_less_than> sup_map;
      sup_map sup;
      // Record occurrences of all guards
      for (auto& t: aut->edges())
        sup.emplace(t.cond);
      for (auto& i: sup)
        allap &= bdd_support(i);
    }

    // Preprocessing
    // Used to convert atomic bdd to id
    std::unordered_map<bdd, unsigned, bdd_hash> bdd2num;
    std::vector<bdd> num2bdd;
    // Nedded for compute succs
    // Used to convert large bdd to indexes
    std::unordered_map<bdd, std::pair<unsigned, unsigned>, bdd_hash> deltas;
    std::vector<safra_state::bdd_id_t> bddnums;
    for (auto& t: aut->edges())
      {
        auto it = deltas.find(t.cond);
        if (it == deltas.end())
          {
            bdd all = t.cond;
            unsigned prev = bddnums.size();
            while (all != bddfalse)
              {
                bdd one = bdd_satoneset(all, allap, bddfalse);
                all -= one;
                auto p = bdd2num.emplace(one, num2bdd.size());
                if (p.second)
                  num2bdd.push_back(one);
                bddnums.emplace_back(p.first->second);
              }
            deltas[t.cond] = std::make_pair(prev, bddnums.size());
          }
      }

    auto res = make_twa_graph(aut->get_dict());
    res->copy_ap_of(aut);
    res->prop_copy(aut,
                   { false, // state based
                   true, // inherently_weak
                   false, // deterministic
                   true // stutter inv
                   });

    // Given a safra_state get its associated state in output automata.
    // Required to create new edges from 2 safra-state
    power_set seen;
    auto init_state = aut->get_init_state_number();
    bool start_accepting = scc.is_accepting_scc(scc.scc_of(init_state)) ||
                           (!track_scc && !emit_scc);
    safra_state init(init_state, true, start_accepting);
    unsigned num = res->new_state();
    res->set_init_state(num);
    seen.insert(std::make_pair(init, num));
    std::deque<safra_state> todo;
    todo.push_back(init);
    unsigned sets = 0;
    using succs_t = safra_state::succs_t;
    succs_t succs;
    while (!todo.empty())
      {
        safra_state curr = todo.front();
        unsigned src_num = seen.find(curr)->second;
        todo.pop_front();
        curr.compute_succs(aut, bddnums, deltas, succs, scc, track_scc);
        for (auto s: succs)
          {
            auto i = seen.find(s.first);
	    unsigned dst_num;
	    if (i != seen.end())
	      {
		dst_num = i->second;
	      }
	    else
	      {
                dst_num = res->new_state();
                todo.push_back(s.first);
                seen.insert(std::make_pair(s.first, dst_num));
              }
            if (s.first.color_ != -1U)
              {
                res->new_edge(src_num, dst_num, num2bdd[s.second],
                                    {s.first.color_});
                // We only care about green acc which are odd
                if (s.first.color_ % 2 == 1)
                  sets = std::max(s.first.color_ + 1, sets);
              }
            else
              res->new_edge(src_num, dst_num, num2bdd[s.second]);
          }
        succs.clear();
      }
    remove_dead_acc(res, sets);
    // Acceptance is now min(odd) since we con emit Red on paths 0 with new opti
    res->set_acceptance(sets, acc_cond::acc_code::parity(false, true, sets));
    res->prop_deterministic(true);
    res->prop_state_based_acc(false);
    if (bisimulation)
      res = simulation(res);
    if (pretty_print)
      res->set_named_prop("state-names", print_debug(seen));
    return res;
  }
}