diff options
Diffstat (limited to 'compiler/circle2circle/src/Model.cpp')
-rw-r--r-- | compiler/circle2circle/src/Model.cpp | 78 |
1 files changed, 78 insertions, 0 deletions
diff --git a/compiler/circle2circle/src/Model.cpp b/compiler/circle2circle/src/Model.cpp new file mode 100644 index 000000000..20d55a131 --- /dev/null +++ b/compiler/circle2circle/src/Model.cpp @@ -0,0 +1,78 @@ +/* + * Copyright (c) 2020 Samsung Electronics Co., Ltd. All Rights Reserved + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +#include "Model.h" + +#include <fstream> +#include <vector> + +#include <fcntl.h> +#include <unistd.h> +#include <sys/stat.h> + +namespace +{ + +class FileModel final : public luci::Model +{ +public: + explicit FileModel(const std::string &filename) : _filename(filename) {} + +public: + FileModel(const FileModel &) = delete; + FileModel(FileModel &&) = delete; + +public: + const ::circle::Model *model(void) override + { + std::ifstream file(_filename, std::ios::binary | std::ios::in); + if (!file.good()) + return nullptr; + + file.unsetf(std::ios::skipws); + + std::streampos fileSize; + file.seekg(0, std::ios::end); + fileSize = file.tellg(); + file.seekg(0, std::ios::beg); + + // reserve capacity + _data.reserve(fileSize); + + // read the data + file.read(_data.data(), fileSize); + if (file.fail()) + return nullptr; + + return ::circle::GetModel(_data.data()); + } + +private: + const std::string _filename; + std::vector<char> _data; +}; + +} // namespace + +namespace luci +{ + +std::unique_ptr<Model> load_model(const std::string &path) +{ + return std::unique_ptr<Model>{new FileModel(path)}; +} + +} // namespace luci |