/*
 * MIT License
 *
 * Copyright (c) 2020 International Business Machines
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 */

#include "HelibCkksPlaintext.h"
#include "helayers/hebase/utils/BinIoUtils.h"

using namespace std;
using namespace helib;

namespace helayers {

shared_ptr<AbstractPlaintext> HelibCkksPlaintext::doClone() const
{
  return shared_ptr<HelibCkksPlaintext>(new HelibCkksPlaintext(*this));
}

streamoff HelibCkksPlaintext::save(ostream& stream) const
{
  streampos streamStartPos = stream.tellp();

  stringstream out;
  pt.writeToJSON(out);

  BinIoUtils::writeString(stream, out.str());

  streampos streamEndPos = stream.tellp();

  return streamEndPos - streamStartPos;
}

streamoff HelibCkksPlaintext::load(istream& stream)
{
  streampos streamStartPos = stream.tellg();

  string data = BinIoUtils::readString(stream);

  stringstream in(data);
  pt = Ptxt<CKKS>::readFromJSON(in, heContext.getContext());

  streampos streamEndPos = stream.tellg();

  return streamEndPos - streamStartPos;
}

void HelibCkksPlaintext::reduceChainIndex() {}

void HelibCkksPlaintext::setChainIndex(const AbstractPlaintext& other) {}

void HelibCkksPlaintext::setChainIndex(int givenChainIndex) {}

int HelibCkksPlaintext::getChainIndex() const { return -1; }

int HelibCkksPlaintext::slotCount() const { return heContext.slotCount(); }

const helib::Ptxt<helib::CKKS>& HelibCkksPlaintext::getPlaintext() const
{

  return pt;
}
} // namespace helayers
