source/opt/libspirv.cpp

// Copyright (c) 2016 Google Inc.
//
// 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.

#include "libspirv.hpp"

#include "ir_loader.h"
#include "make_unique.h"
#include "message.h"
#include "table.h"

namespace spvtools {

// Structs for holding the data members for SpvTools.
struct SpvTools::Impl {
  explicit Impl(spv_target_env env) : context(spvContextCreate(env)) {
    // The default consumer in spv_context_t is a null consumer, which provides
    // equivalent functionality (from the user's perspective) as a real consumer
    // does nothing.
  }
  ~Impl() { spvContextDestroy(context); }

  spv_context context;  // C interface context object.
};

SpvTools::SpvTools(spv_target_env env) : impl_(new Impl(env)) {}

SpvTools::~SpvTools() {}

void SpvTools::SetMessageConsumer(MessageConsumer consumer) {
  SetContextMessageConsumer(impl_->context, std::move(consumer));
}

bool SpvTools::Assemble(const std::string& text,
                        std::vector<uint32_t>* binary) const {
  spv_binary spvbinary = nullptr;
  spv_result_t status = spvTextToBinary(impl_->context, text.data(),
                                        text.size(), &spvbinary, nullptr);
  if (status == SPV_SUCCESS) {
    binary->assign(spvbinary->code, spvbinary->code + spvbinary->wordCount);
  }
  spvBinaryDestroy(spvbinary);
  return status == SPV_SUCCESS;
}

bool SpvTools::Disassemble(const std::vector<uint32_t>& binary,
                           std::string* text, uint32_t options) const {
  spv_text spvtext = nullptr;
  spv_result_t status = spvBinaryToText(
      impl_->context, binary.data(), binary.size(), options, &spvtext, nullptr);
  if (status == SPV_SUCCESS) {
    text->assign(spvtext->str, spvtext->str + spvtext->length);
  }
  spvTextDestroy(spvtext);
  return status == SPV_SUCCESS;
}

bool SpvTools::Validate(const std::vector<uint32_t>& binary) const {
  spv_const_binary_t b = {binary.data(), binary.size()};
  return spvValidate(impl_->context, &b, nullptr) == SPV_SUCCESS;
}

}  // namespace spvtools