// Copyright (c) 2024 PaddlePaddle 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 <optional>
#include <set>
#include <string>
#include <vector>
#include "paddle/pir/include/core/program.h"
#include "paddle/pir/include/core/value.h"

namespace pir {
namespace utils {
namespace name_analysis {

std::vector<std::string> GetValueAllNames(Value value);
std::string GetValueFirstName(Value value);
std::optional<std::string> TryGetValueFirstName(Value value);
Value GetParameterValueByName(const Program &program, const std::string &name);
std::unordered_map<std::string, Value> GetAllParameterValues(
    const Program &program);

void SetValueName(Value value, const std::string name);

std::map<std::string, std::string> RenameValue(Value value,
                                               const std::string &new_name,
                                               Block *block);
std::optional<std::string> GetValueInputName(Value value);

std::vector<std::string> GetValueOutputNames(Value value);
Value GetOutputValueByName(const Program &program, const std::string &name);
Value GetValueByNameInPhiKernelProgram(const Program &program,
                                       const std::string &name);
inline bool HasOnlyOneValueName(Value value) {
  std::vector<std::string> names = GetValueAllNames(value);
  return std::set<std::string>(names.begin(), names.end()).size() <= 1;
}
std::unordered_map<std::string, Value> GetAllNamedValues(
    const Program &program);

}  // namespace name_analysis
}  // namespace utils
}  // namespace pir