/* ******************************************************************************
 *
 *
 * 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