Files
2026-07-13 12:47:05 +08:00

92 lines
2.1 KiB
C++

/* ******************************************************************************
*
*
* This program and the accompanying materials are made available under the
* terms of the Apache License, Version 2.0 which is available at
* https://www.apache.org/licenses/LICENSE-2.0.
*
* See the NOTICE file distributed with this work for additional
* information regarding copyright ownership.
* 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.
*
* SPDX-License-Identifier: Apache-2.0
******************************************************************************/
//
// @author raver119@gmail.com
//
#include <array/ShapeList.h>
namespace sd {
ShapeList::ShapeList(LongType* shape) {
if (shape != nullptr) push_back(shape);
}
ShapeList::~ShapeList() {
if (_autoremovable) destroy();
}
ShapeList::ShapeList(const std::vector< LongType*>& shapes, bool isWorkspace)
{
for (size_t i = 0; i < shapes.size(); i++) {
push_back(shapes[i]);
}
_workspace = isWorkspace;
}
ShapeList::ShapeList(const std::vector< LongType*>& shapes) {
_shapes = shapes;
}
void ShapeList::destroy() {
if (_destroyed) return;
if (!_workspace) {
for (int i = 0; i < size(); i++){
if (_shapes[i] != nullptr) delete[] _shapes[i];
}
}
_destroyed = true;
}
int ShapeList::size() const {
return (int)_shapes.size();
}
LongType* ShapeList::at(int idx) {
if(idx < 0) {
idx += _shapes.size();
}
if (size() <= idx) {
std::string errorMessage;
errorMessage += "Can't find requested variable by index: ";
errorMessage += std::to_string(idx);
THROW_EXCEPTION(errorMessage.c_str());
}
return _shapes[idx];
}
void ShapeList::push_back( LongType* shape) {
_shapes.push_back(shape);
}
void ShapeList::detach() {
for (int e = 0; e < size(); e++) {
_shapes[e] = shape::detachShape(_shapes[e]);
}
_autoremovable = true;
_workspace = false;
}
} // namespace sd