src/graphics/lib/compute/common/vk/cache.c

// Copyright 2019 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

//
//
//

#if defined(_MSC_VER)

#define _CRT_SECURE_NO_WARNINGS  // squelch fopen()

#endif

//
//
//

#include <stdio.h>
#include <stdlib.h>

//
//
//

#include "assert.h"
#include "cache.h"

//
//
//

VkResult
vk_pipeline_cache_create(VkDevice                      device,
                         VkAllocationCallbacks const * allocator,
                         char const * const            name,
                         VkPipelineCache *             pipeline_cache)
{
  VkPipelineCacheCreateInfo pipeline_cache_info = {

    .sType           = VK_STRUCTURE_TYPE_PIPELINE_CACHE_CREATE_INFO,
    .pNext           = NULL,
    .flags           = 0,
    .initialDataSize = 0,
    .pInitialData    = NULL
  };

  FILE * f    = fopen(name, "rb");
  void * data = NULL;

  if (f != NULL)
    {
      if (fseek(f, 0L, SEEK_END) == 0)
        {
          size_t const data_size = ftell(f);

          if (data_size > 0)
            {
              fseek(f, 0L, SEEK_SET);

              data = malloc(data_size);

              if (data != NULL)
                {
                  fread(data, data_size, 1, f);

                  pipeline_cache_info.initialDataSize = data_size;
                  pipeline_cache_info.pInitialData    = data;
                }
            }
        }

      fclose(f);
    }

  VkResult vk_res = vkCreatePipelineCache(device, &pipeline_cache_info, allocator, pipeline_cache);

  free(data);

  return vk_res;
}

//
//
//

VkResult
vk_pipeline_cache_destroy(VkDevice                      device,
                          VkAllocationCallbacks const * allocator,
                          char const * const            name,
                          VkPipelineCache               pipeline_cache)
{
  size_t   data_size;
  VkResult vk_res = vkGetPipelineCacheData(device, pipeline_cache, &data_size, NULL);

  if (vk_res != VK_SUCCESS)
    return vk_res;

  if (data_size > 0)
    {
      void * data = malloc(data_size);

      if (data != NULL)
        {
          vk_res = vkGetPipelineCacheData(device, pipeline_cache, &data_size, data);

          if (vk_res == VK_SUCCESS)
            {
              FILE * f = fopen(name, "wb");

              if (f != NULL)
                {
                  fwrite(data, data_size, 1, f);
                  fclose(f);
                }

              free(data);
            }
        }
    }

  vkDestroyPipelineCache(device, pipeline_cache, allocator);

  return vk_res;
}

//
//
//