/*
 * Copyright (C) 2011 The Android Open Source Project
 *
 * 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.
 */

#ifndef ART_DEX2OAT_DEX_QUICK_COMPILER_CALLBACKS_H_
#define ART_DEX2OAT_DEX_QUICK_COMPILER_CALLBACKS_H_

#include "compiler_callbacks.h"
#include "verifier/verifier_deps.h"

namespace art {

class CompilerDriver;
class DexFile;
class VerificationResults;

class QuickCompilerCallbacks final : public CompilerCallbacks {
 public:
  explicit QuickCompilerCallbacks(CompilerCallbacks::CallbackMode mode)
      : CompilerCallbacks(mode), dex_files_(nullptr) {}

  ~QuickCompilerCallbacks() { }

  ClassLinker* CreateAotClassLinker(InternTable* intern_table) override;

  void AddUncompilableMethod(MethodReference ref) override;
  void AddUncompilableClass(ClassReference ref) override;
  bool IsUncompilableMethod(MethodReference ref) override;

  void ClassRejected(ClassReference ref) override;

  verifier::VerifierDeps* GetVerifierDeps() const override {
    return verifier_deps_.get();
  }

  void SetVerifierDeps(verifier::VerifierDeps* deps) override {
    verifier_deps_.reset(deps);
  }

  void SetVerificationResults(VerificationResults* verification_results) {
    verification_results_ = verification_results;
  }

  ClassStatus GetPreviousClassState(ClassReference ref) override;

  void SetDoesClassUnloading(bool does_class_unloading, CompilerDriver* compiler_driver)
      override {
    does_class_unloading_ = does_class_unloading;
    compiler_driver_ = compiler_driver;
    DCHECK(!does_class_unloading || compiler_driver_ != nullptr);
  }

  void UpdateClassState(ClassReference ref, ClassStatus state) override;

  bool CanUseOatStatusForVerification(mirror::Class* klass) override
      REQUIRES_SHARED(Locks::mutator_lock_);

  void SetDexFiles(const std::vector<const DexFile*>* dex_files) {
    dex_files_ = dex_files;
  }

  bool ShouldEnableProfileCode() override { return should_enable_profile_code_; }

  void SetShouldEnableProfileCode(bool flag) { should_enable_profile_code_ = flag; }

 private:
  VerificationResults* verification_results_ = nullptr;
  bool does_class_unloading_ = false;
  CompilerDriver* compiler_driver_ = nullptr;
  std::unique_ptr<verifier::VerifierDeps> verifier_deps_;
  const std::vector<const DexFile*>* dex_files_;
  bool should_enable_profile_code_ = false;
};

}  // namespace art

#endif  // ART_DEX2OAT_DEX_QUICK_COMPILER_CALLBACKS_H_