package net.vulkanmod.vulkan.shader.descriptor;

import net.vulkanmod.vulkan.texture.VTextureSelector;
import net.vulkanmod.vulkan.texture.VulkanImage;

public class ImageDescriptor implements Descriptor {
   private final int descriptorType;
   private final int binding;
   public final String qualifier;
   public final String name;
   public final int imageIdx;
   public boolean useSampler;
   public boolean isReadOnlyLayout;
   private int layout;
   private int mipLevel;

   public ImageDescriptor(int binding, String type, String name, int imageIdx) {
      this(binding, type, name, imageIdx, false);
   }

   public ImageDescriptor(int binding, String type, String name, int imageIdx, boolean isStorageImage) {
      this(binding, type, name, imageIdx, isStorageImage ? 3 : 1);
   }

   public ImageDescriptor(int binding, String type, String name, int imageIdx, int descriptorType) {
      this.mipLevel = -1;
      this.binding = binding;
      this.qualifier = type;
      this.name = name;
      this.imageIdx = imageIdx;
      if (this.imageIdx == -1) {
         throw new IllegalArgumentException();
      } else {
         this.descriptorType = descriptorType;
         boolean isStorageImage = this.isStorageImage();
         this.useSampler = !isStorageImage;
         this.setLayout(isStorageImage ? 1 : 5);
      }
   }

   public int getBinding() {
      return this.binding;
   }

   public int getType() {
      return this.descriptorType;
   }

   public int getStages() {
      return 63;
   }

   public void setLayout(int layout) {
      this.layout = layout;
      this.isReadOnlyLayout = layout == 5;
   }

   public int getLayout() {
      return this.layout;
   }

   public void setMipLevel(int mipLevel) {
      this.mipLevel = mipLevel;
   }

   public int getMipLevel() {
      return this.mipLevel;
   }

   public VulkanImage getImage() {
      return VTextureSelector.getImage(this.imageIdx);
   }

   public long getImageView(VulkanImage image) {
      long view;
      if (this.mipLevel == -1) {
         view = image.getImageView();
      } else {
         view = image.getLevelImageView(this.mipLevel);
      }

      return view;
   }

   public boolean isStorageImage() {
      return this.descriptorType == 3;
   }

   public static class State {
      long imageView;
      long sampler;

      public State(long imageView, long sampler) {
         this.set(imageView, sampler);
      }

      public void set(long imageView, long sampler) {
         this.imageView = imageView;
         this.sampler = sampler;
      }

      public boolean isCurrentState(long imageView, long sampler) {
         return this.imageView == imageView && this.sampler == sampler;
      }
   }
}