92 lines
2.1 KiB
C++
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
|