Line data    Source code

       1             : /**
       2             :  Copyright (c) 2023 Stappler LLC <admin@stappler.dev>
       3             : 
       4             :  Permission is hereby granted, free of charge, to any person obtaining a copy
       5             :  of this software and associated documentation files (the "Software"), to deal
       6             :  in the Software without restriction, including without limitation the rights
       7             :  to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
       8             :  copies of the Software, and to permit persons to whom the Software is
       9             :  furnished to do so, subject to the following conditions:
      10             : 
      11             :  The above copyright notice and this permission notice shall be included in
      12             :  all copies or substantial portions of the Software.
      13             : 
      14             :  THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
      15             :  IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
      16             :  FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
      17             :  AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
      18             :  LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
      19             :  OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
      20             :  THE SOFTWARE.
      21             :  **/
      22             : 
      23             : #ifndef SRC_BACKEND_VK_XLSNNVKGENERATIONLAYER_H_
      24             : #define SRC_BACKEND_VK_XLSNNVKGENERATIONLAYER_H_
      25             : 
      26             : #include "XLVkRenderPass.h"
      27             : #include "XLVkQueuePass.h"
      28             : #include "XLVkAttachment.h"
      29             : 
      30             : namespace stappler::xenolith::vk::shadernn {
      31             : 
      32             : struct GenerationData {
      33             :         UVec3 seed;
      34             :         uint32_t hashMax;
      35             :         float limitsMin;
      36             :         float limitsMax;
      37             : };
      38             : 
      39             : struct GenerationDataInput : core::AttachmentInputData {
      40             :         GenerationData data;
      41             : };
      42             : 
      43             : class GenerationLayer : public vk::QueuePass {
      44             : public:
      45             :         virtual ~GenerationLayer();
      46             : 
      47             :         virtual bool init(Queue::Builder &queueBuilder, QueuePassBuilder &, const AttachmentData *output);
      48             : 
      49           0 :         const AttachmentData *getOutputAttachment() const { return _outputAttachment; }
      50           0 :         const AttachmentData *getDataAttachment() const { return _dataAttachment; }
      51             : 
      52             :         class LayerHandle : public vk::QueuePassHandle {
      53             :         public:
      54             :                 virtual ~LayerHandle();
      55             : 
      56             :                 virtual bool prepare(FrameQueue &q, Function<void(bool)> &&cb) override;
      57             : 
      58             :         protected:
      59             :                 virtual Vector<const vk::CommandBuffer *> doPrepareCommands(FrameHandle &) override;
      60             : 
      61             :                 const vk::ImageAttachmentHandle *_outputImage = nullptr;
      62             :                 const core::AttachmentHandle *_dataBuffer = nullptr;
      63             :         };
      64             : 
      65             : protected:
      66             :         using QueuePass::init;
      67             : 
      68             :         const AttachmentData *_outputAttachment = nullptr;
      69             :         const AttachmentData *_dataAttachment = nullptr;
      70             : };
      71             : 
      72             : }
      73             : 
      74             : #endif /* SRC_BACKEND_VK_XLSNNVKGENERATIONLAYER_H_ */