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_PROCESSOR_XLSNNMODELPROCESSOR_H_
      24             : #define SRC_PROCESSOR_XLSNNMODELPROCESSOR_H_
      25             : 
      26             : #include "XLSnnModel.h"
      27             : #include "XLCoreQueue.h"
      28             : 
      29             : namespace stappler::xenolith::shadernn {
      30             : 
      31             : struct ModelSpecialization {
      32             :         Map<const Layer *, Extent3> inputs;
      33             :         Map<const Attachment *, Extent3> attachments;
      34             : };
      35             : 
      36             : class ModelProcessor : public Ref {
      37             : public:
      38             :         using LayerConstructor = Rc<Layer> (*) (Model *, StringView tag, size_t idx, const Value &);
      39             : 
      40           2 :         virtual ~ModelProcessor() = default;
      41             : 
      42             :         bool init();
      43             : 
      44             :         Rc<Model> load(FilePath modelPath, ModelFlags);
      45             : 
      46             :         ModelSpecialization specializeModel(Model *, Extent3);
      47             :         ModelSpecialization specializeModel(Model *, Map<const Layer *, Extent3> &&inputs);
      48             : 
      49             : protected:
      50             :         bool loadFromJson(Model *, Value &&) const;
      51             : 
      52             :         Rc<Layer> makeLayer(Model *m, StringView tag, size_t idx, Value &&) const;
      53             : 
      54             :         Map<String, LayerConstructor> _layers;
      55             : };
      56             : 
      57             : }
      58             : 
      59             : #endif /* SRC_PROCESSOR_XLSNNMODELPROCESSOR_H_ */