/*
 * Copyright 2019 WebAssembly Community Group participants
 *
 * 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 "ir/memory-utils.h"
#include "pass.h"
#include "wasm.h"

//
// Attempt to merge segments to fit within a specified limit.
//
// By default this limit is equal to the one commonly used by wasm VMs
// (see wasm-limits.h), but it can be changed with the option below:
//
//   --pass-arg=limit-segments@max-data-segments
//
//       Specify a custom maximum number of data segments.
//

namespace wasm {

struct LimitSegments : public Pass {
  void run(Module* module) override {
    Index maxDataSegments;
    if (hasArgument("limit-segments")) {
      maxDataSegments = std::stoul(getArgument("limit-segments", ""));
    } else {
      maxDataSegments = WebLimitations::MaxDataSegments;
    }
    if (!MemoryUtils::ensureLimitedSegments(*module, maxDataSegments)) {
      std::cerr << "Unable to merge segments. "
                << "wasm VMs may not accept this binary" << std::endl;
    }
  }
};

Pass* createLimitSegmentsPass() { return new LimitSegments(); }

} // namespace wasm