// Copyright (c) 2021 CINN Authors. All Rights Reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

#pragma once

#include <string_view>

#include "paddle/cinn/common/axis.h"
#include "paddle/cinn/common/cinn_value.h"
#include "paddle/cinn/common/context.h"
#include "paddle/cinn/common/graph_utils.h"
#include "paddle/cinn/common/macros.h"
#include "paddle/cinn/common/shared.h"
#include "paddle/cinn/common/target.h"
#include "paddle/cinn/common/type.h"
#include "paddle/cinn/utils/error.h"
#include "paddle/common/enforce.h"

namespace cinn {

// export some general concepts.
using cinn::common::Context;
using cinn::common::make_shared;
using cinn::common::Object;
using cinn::common::ref_count;
using cinn::common::Shared;
using cinn::common::UniqName;

// Type related.
using cinn::common::Bool;
using cinn::common::Float;
using cinn::common::Int;
using cinn::common::UInt;
using cinn::common::Void;

using cinn::common::type_of;

using cinn::common::Target;
using cinn::common::Type;
using cinn::common::UnkTarget;

template <typename T>
T& Reference(const T* x) {
  return *const_cast<T*>(x);
}

static void CheckVarNameValid(const std::string_view name) {
  PADDLE_ENFORCE_EQ(name.empty(),
                    false,
                    ::common::errors::InvalidArgument(
                        "Var name is empty. Please check your input"));
  PADDLE_ENFORCE_EQ(
      name.find(' '),
      std::string::npos,
      ::common::errors::InvalidArgument("Var name contains space. Received: "
                                        "%s, which is invalid for var name.",
                                        name));
  PADDLE_ENFORCE_EQ(
      name.find('.'),
      std::string::npos,
      ::common::errors::InvalidArgument(
          "Var name contains dot. Received: %s, which is invalid for var name.",
          name));
  PADDLE_ENFORCE_EQ(
      name.find('@'),
      std::string::npos,
      ::common::errors::InvalidArgument(
          "Var name contains at. Received: %s, which is invalid for var name.",
          name));
  PADDLE_ENFORCE_EQ(
      name.find('/'),
      std::string::npos,
      ::common::errors::InvalidArgument("Var name contains slash. Received: "
                                        "%s, which is invalid for var name.",
                                        name));
  PADDLE_ENFORCE_EQ(
      name.find('\t'),
      std::string::npos,
      ::common::errors::InvalidArgument(
          "Var name contains tab. Received: %s, which is invalid for var name.",
          name));
  PADDLE_ENFORCE_EQ(
      name.find('\n'),
      std::string::npos,
      ::common::errors::InvalidArgument("Var name contains newline. Received: "
                                        "%s, which is invalid for var name.",
                                        name));
  PADDLE_ENFORCE_EQ(name.find('\r'),
                    std::string::npos,
                    ::common::errors::InvalidArgument(
                        "Var name contains carriage return. Received: %s, "
                        "which is invalid for var name.",
                        name));
  PADDLE_ENFORCE_EQ(cinn::common::IsAxisNameReserved(std::string(name)),
                    false,
                    ::common::errors::InvalidArgument(
                        "The name [%s] is reserved for internal axis", name));
}

}  // namespace cinn