#include <algorithm>

#include "vtr_assert.h"
#include "vtr_log.h"

#include "repack_design_constraints.h"

/************************************************************************
 * Member functions for class RepackDesignConstraints
 ***********************************************************************/

/************************************************************************
 * Constructors
 ***********************************************************************/
RepackDesignConstraints::RepackDesignConstraints() {
  return;
}

/************************************************************************
 * Public Accessors : aggregates
 ***********************************************************************/
RepackDesignConstraints::repack_design_constraint_range RepackDesignConstraints::design_constraints() const {
  return vtr::make_range(repack_design_constraint_ids_.begin(), repack_design_constraint_ids_.end());
}

/************************************************************************
 * Public Accessors : Basic data query 
 ***********************************************************************/
RepackDesignConstraints::e_design_constraint_type RepackDesignConstraints::type(const RepackDesignConstraintId& repack_design_constraint_id) const {
  /* validate the design_constraint_id */
  VTR_ASSERT(valid_design_constraint_id(repack_design_constraint_id));
  return repack_design_constraint_types_[repack_design_constraint_id]; 
}

std::string RepackDesignConstraints::pb_type(const RepackDesignConstraintId& repack_design_constraint_id) const {
  /* validate the design_constraint_id */
  VTR_ASSERT(valid_design_constraint_id(repack_design_constraint_id));
  return repack_design_constraint_pb_types_[repack_design_constraint_id]; 
}

openfpga::BasicPort RepackDesignConstraints::pin(const RepackDesignConstraintId& repack_design_constraint_id) const {
  /* validate the design_constraint_id */
  VTR_ASSERT(valid_design_constraint_id(repack_design_constraint_id));
  return repack_design_constraint_pins_[repack_design_constraint_id]; 
}

std::string RepackDesignConstraints::net(const RepackDesignConstraintId& repack_design_constraint_id) const {
  /* validate the design_constraint_id */
  VTR_ASSERT(valid_design_constraint_id(repack_design_constraint_id));
  return repack_design_constraint_nets_[repack_design_constraint_id]; 
}

std::string RepackDesignConstraints::find_constrained_pin_net(const std::string& pb_type,
                                                              const openfpga::BasicPort& pin) const {
  std::string constrained_net_name;
  for (const RepackDesignConstraintId& design_constraint : design_constraints()) {
    /* If found a constraint, record the net name */
    if ( (pb_type == repack_design_constraint_pb_types_[design_constraint]) 
       && (pin == repack_design_constraint_pins_[design_constraint])) {
      constrained_net_name = repack_design_constraint_nets_[design_constraint];
      break;
    }
  }
  return constrained_net_name;
}

bool RepackDesignConstraints::empty() const {
  return 0 == repack_design_constraint_ids_.size();
}

/************************************************************************
 * Public Mutators
 ***********************************************************************/
void RepackDesignConstraints::reserve_design_constraints(const size_t& num_design_constraints) {
  repack_design_constraint_ids_.reserve(num_design_constraints);
  repack_design_constraint_types_.reserve(num_design_constraints);
  repack_design_constraint_pb_types_.reserve(num_design_constraints);
  repack_design_constraint_pins_.reserve(num_design_constraints);
  repack_design_constraint_nets_.reserve(num_design_constraints);
}

RepackDesignConstraintId RepackDesignConstraints::create_design_constraint(const RepackDesignConstraints::e_design_constraint_type& repack_design_constraint_type) {
  /* Create a new id */
  RepackDesignConstraintId repack_design_constraint_id = RepackDesignConstraintId(repack_design_constraint_ids_.size());
  
  repack_design_constraint_ids_.push_back(repack_design_constraint_id);
  repack_design_constraint_types_.push_back(repack_design_constraint_type);
  repack_design_constraint_pb_types_.emplace_back();
  repack_design_constraint_pins_.emplace_back();
  repack_design_constraint_nets_.emplace_back();
  
  return repack_design_constraint_id;
}

void RepackDesignConstraints::set_pb_type(const RepackDesignConstraintId& repack_design_constraint_id,
                                          const std::string& pb_type) {
  /* validate the design_constraint_id */
  VTR_ASSERT(valid_design_constraint_id(repack_design_constraint_id));
  repack_design_constraint_pb_types_[repack_design_constraint_id] = pb_type;
}

void RepackDesignConstraints::set_pin(const RepackDesignConstraintId& repack_design_constraint_id,
                                      const openfpga::BasicPort& pin) {
  /* validate the design_constraint_id */
  VTR_ASSERT(valid_design_constraint_id(repack_design_constraint_id));
  repack_design_constraint_pins_[repack_design_constraint_id] = pin;
}

void RepackDesignConstraints::set_net(const RepackDesignConstraintId& repack_design_constraint_id,
                                      const std::string& net) {
  /* validate the design_constraint_id */
  VTR_ASSERT(valid_design_constraint_id(repack_design_constraint_id));
  repack_design_constraint_nets_[repack_design_constraint_id] = net;
}

/************************************************************************
 * Internal invalidators/validators 
 ***********************************************************************/
/* Validators */
bool RepackDesignConstraints::valid_design_constraint_id(const RepackDesignConstraintId& design_constraint_id) const {
  return ( size_t(design_constraint_id) < repack_design_constraint_ids_.size() ) && ( design_constraint_id == repack_design_constraint_ids_[design_constraint_id] ); 
}

bool RepackDesignConstraints::unconstrained_net(const std::string& net) const {
  return net.empty();
}

bool RepackDesignConstraints::unmapped_net(const std::string& net) const {
  return std::string(REPACK_DESIGN_CONSTRAINT_OPEN_NET) == net;
}