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_XLSNNVKSTATPERCENTLAYER_H_
      24             : #define SRC_BACKEND_VK_XLSNNVKSTATPERCENTLAYER_H_
      25             : 
      26             : #include "XLVkRenderPass.h"
      27             : #include "XLVkQueuePass.h"
      28             : #include "XLVkAttachment.h"
      29             : #include "XLSnnStatPercentLayer.h"
      30             : 
      31             : namespace stappler::xenolith::vk::shadernn {
      32             : 
      33             : class StatPercentLayer : public vk::QueuePass {
      34             : public:
      35             :         using Front = xenolith::shadernn::StatPercentLayer;
      36             : 
      37             :         static constexpr auto StatPercentLayerClassesPipeline = "StatPercentLayerClassesPipeline";
      38             :         static constexpr auto StatPercentLayerPercentPipeline = "StatPercentLayerPercentPipeline";
      39             : 
      40             :         virtual ~StatPercentLayer();
      41             : 
      42             :         virtual bool init(Queue::Builder &queueBuilder, QueuePassBuilder &, Front *,
      43             :                         const AttachmentData *input, const AttachmentData *output);
      44             : 
      45           0 :         const AttachmentData *getInputAttachment() const { return _inputAttachment; }
      46           0 :         const AttachmentData *getOutputAttachment() const { return _outputAttachment; }
      47           0 :         const AttachmentData *getClassesAttachment() const { return _classesAttachment; }
      48             : 
      49           0 :         const Front *getFront() const { return _front; }
      50             : 
      51             :         class LayerHandle : public vk::QueuePassHandle {
      52             :         public:
      53           0 :                 virtual ~LayerHandle() = default;
      54             : 
      55             :                 virtual bool prepare(FrameQueue &q, Function<void(bool)> &&cb) override;
      56             : 
      57             :         protected:
      58             :                 virtual Vector<const vk::CommandBuffer *> doPrepareCommands(FrameHandle &) override;
      59             : 
      60             :                 virtual void doSubmitted(FrameHandle &, Function<void(bool)> &&, bool, Rc<Fence> &&) override;
      61             : 
      62             :                 vk::BufferAttachmentHandle *_inputBuffer = nullptr;
      63             :                 vk::BufferAttachmentHandle *_outputBuffer = nullptr;
      64             :                 vk::BufferAttachmentHandle *_classesBuffer = nullptr;
      65             :                 const Front *_front = nullptr;
      66             : 
      67             :                 Rc<Buffer> _classesSizes;
      68             :                 Rc<Buffer> _classesIndexes;
      69             :                 Rc<Buffer> _output;
      70             :         };
      71             : 
      72             : protected:
      73             :         using QueuePass::init;
      74             : 
      75             :         const AttachmentData *_inputAttachment = nullptr;
      76             :         const AttachmentData *_outputAttachment = nullptr;
      77             :         const AttachmentData *_classesAttachment = nullptr;
      78             : 
      79             :         Rc<Front> _front;
      80             : };
      81             : 
      82             : class StatAnalysisLayer : public vk::QueuePass {
      83             : public:
      84             :         using Front = xenolith::shadernn::StatAnalysisLayer;
      85             : 
      86             :         virtual ~StatAnalysisLayer();
      87             : 
      88             :         virtual bool init(Queue::Builder &queueBuilder, QueuePassBuilder &, Front *,
      89             :                         const AttachmentData *inputData, const AttachmentData *inputClasses, const AttachmentData *output);
      90             : 
      91           0 :         const AttachmentData *getInputDataAttachment() const { return _inputDataAttachment; }
      92           0 :         const AttachmentData *getInputClassesAttachment() const { return _inputClassesAttachment; }
      93           0 :         const AttachmentData *getOutputAttachment() const { return _outputAttachment; }
      94             : 
      95           0 :         const Front *getFront() const { return _front; }
      96             : 
      97             :         class LayerHandle : public vk::QueuePassHandle {
      98             :         public:
      99           0 :                 virtual ~LayerHandle() = default;
     100             : 
     101             :                 virtual bool prepare(FrameQueue &q, Function<void(bool)> &&cb) override;
     102             : 
     103             :         protected:
     104             :                 virtual Vector<const vk::CommandBuffer *> doPrepareCommands(FrameHandle &) override;
     105             : 
     106             :                 virtual void doSubmitted(FrameHandle &, Function<void(bool)> &&, bool, Rc<Fence> &&) override;
     107             : 
     108             :                 vk::BufferAttachmentHandle *_inputDataBuffer = nullptr;
     109             :                 vk::BufferAttachmentHandle *_inputClassesBuffer = nullptr;
     110             :                 vk::BufferAttachmentHandle *_outputBuffer = nullptr;
     111             :                 const Front *_front = nullptr;
     112             : 
     113             :                 Rc<Buffer> _output;
     114             :         };
     115             : 
     116             : protected:
     117             :         using QueuePass::init;
     118             : 
     119             :         const AttachmentData *_inputDataAttachment = nullptr;
     120             :         const AttachmentData *_inputClassesAttachment = nullptr;
     121             :         const AttachmentData *_outputAttachment = nullptr;
     122             : 
     123             :         Rc<Front> _front;
     124             : };
     125             : 
     126             : }
     127             : 
     128             : #endif /* SRC_BACKEND_VK_XLSNNVKSTATPERCENTLAYER_H_ */