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             : #include "XLSnnSubpixelLayer.h"
      24             : #include "XLSnnVkSubpixelLayer.h"
      25             : 
      26             : namespace stappler::xenolith::shadernn {
      27             : 
      28           0 : bool SubpixelLayer::init(Model *m, StringView tag, size_t idx, const Value &data) {
      29           0 :         if (!Layer::init(m, tag, idx, data)) {
      30           0 :                 return false;
      31             :         }
      32             : 
      33           0 :         return true;
      34             : }
      35             : 
      36           0 : const core::QueuePassData *SubpixelLayer::prepare(core::Queue::Builder &builder,
      37             :                 Map<Layer *, const core::AttachmentData *> inputs,
      38             :                 Map<Attachment *, const core::AttachmentData *> attachments) {
      39             : 
      40           0 :         auto inputIt = attachments.find(_inputs.front().attachment);
      41           0 :         auto outputIt = attachments.find(getOutput());
      42             : 
      43           0 :         if (inputIt == attachments.end() || outputIt == attachments.end()) {
      44           0 :                 log::error("snn::InputLayer", "No attachments specified");
      45           0 :                 return nullptr;
      46             :         }
      47             : 
      48           0 :         return builder.addPass(getName(), core::PassType::Compute, core::RenderOrdering(_inputIndex),
      49           0 :                         [&] (core::QueuePassBuilder &passBuilder) -> Rc<core::QueuePass> {
      50           0 :                 return Rc<vk::shadernn::SubpixelLayer>::create(builder, passBuilder, this, inputIt->second, outputIt->second);
      51           0 :         });
      52             : }
      53             : 
      54             : }