//===- Main.cpp -----------------------------------------------------------===//
//
// 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.
//
//===----------------------------------------------------------------------===//
//
// This is the main file of the image processing benchmark.
//
//===----------------------------------------------------------------------===//

#include <benchmark/benchmark.h>

void initializeMLIRConv2D(int, char **);
void initializeBuddyConv2D(int, char **);
void initializeBuddyCorr2D(int, char **);
void initializeOpenCVFilter2D(int, char **);

void generateResultBuddyConv2D();
void generateResultBuddyCorr2D();

// Run benchmarks.
int main(int argc, char **argv) {
  if (argc != 3 && argc != 5) {
    throw std::invalid_argument(
        "Wrong format of command line arguments.\n"
        "Correct format is ./image-processing-benchmark <image path> <kernel "
        "name>\n where "
        "image path provides path of the image to be processed and kernel name "
        "denotes the name "
        "of desired kernel as specified in "
        "include/ImageProcessing/Kernels.h\n");
  }

  initializeMLIRConv2D(argc, argv);
  initializeBuddyConv2D(argc, argv);
  initializeBuddyCorr2D(argc, argv);
  initializeOpenCVFilter2D(argc, argv);

  ::benchmark::Initialize(&argc, argv);
  ::benchmark::RunSpecifiedBenchmarks();
  // Generate result image.
  generateResultBuddyConv2D();
  generateResultBuddyCorr2D();

  return 0;
}