/*

  KLayout Layout Viewer
  Copyright (C) 2006-2025 Matthias Koefferlein

  This program is free software; you can redistribute it and/or modify
  it under the terms of the GNU General Public License as published by
  the Free Software Foundation; either version 2 of the License, or
  (at your option) any later version.

  This program is distributed in the hope that it will be useful,
  but WITHOUT ANY WARRANTY; without even the implied warranty of
  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  GNU General Public License for more details.

  You should have received a copy of the GNU General Public License
  along with this program; if not, write to the Free Software
  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA

*/


#include "pexRNetExtractor.h"
#include "pexRExtractorTech.h"
#include "pexRNetwork.h"
#include "dbReader.h"
#include "dbLayout.h"
#include "tlUnitTest.h"

class TestableRNetExtractor
  : public pex::RNetExtractor
{
public:
  TestableRNetExtractor (double dbu) : pex::RNetExtractor (dbu) { }

  using pex::RNetExtractor::create_via_ports;
};

TEST(netex_viagen1)
{
  db::Layout ly;

  {
    std::string fn = tl::testdata () + "/pex/netex_viagen1.gds";
    tl::InputStream is (fn);
    db::Reader reader (is);
    reader.read (ly);
  }

  TestableRNetExtractor rex (ly.dbu ());

  auto tc = ly.cell_by_name ("TOP");
  tl_assert (tc.first);

  unsigned int l1 = ly.get_layer (db::LayerProperties (1, 0));
  unsigned int l2 = ly.get_layer (db::LayerProperties (2, 0));
  unsigned int l3 = ly.get_layer (db::LayerProperties (3, 0));

  std::map<unsigned int, db::Region> geo;
  geo.insert (std::make_pair (l2, db::Region (db::RecursiveShapeIterator (ly, ly.cell (tc.second), l2))));

  pex::RNetwork network;

  pex::RExtractorTech tech;

  pex::RExtractorTechVia via1;
  via1.bottom_conductor = l1;
  via1.cut_layer = l2;
  via1.top_conductor = l3;
  via1.resistance = 2.0;
  tech.vias.push_back (via1);

  std::map<unsigned int, std::vector<pex::RNetExtractor::ViaPort> > via_ports;
  rex.create_via_ports (tech, geo, via_ports, network);

  EXPECT_EQ (via_ports [l1].size (), size_t (4));
  EXPECT_EQ (via_ports [l2].size (), size_t (0));
  EXPECT_EQ (via_ports [l3].size (), size_t (4));

  EXPECT_EQ (network.to_string (true),
    "R $0(1.7,0.1;1.9,0.3) $1(1.7,0.1;1.9,0.3) 50\n"
    "R $2(0.4,0.5;0.6,0.7) $3(0.4,0.5;0.6,0.7) 50\n"
    "R $4(0.8,0.5;1,0.7) $5(0.8,0.5;1,0.7) 50\n"
    "R $6(2.9,0.5;3.1,0.7) $7(2.9,0.5;3.1,0.7) 50"
  );
}

TEST(netex_viagen2)
{
  db::Layout ly;

  {
    std::string fn = tl::testdata () + "/pex/netex_viagen2.gds";
    tl::InputStream is (fn);
    db::Reader reader (is);
    reader.read (ly);
  }

  TestableRNetExtractor rex (ly.dbu ());

  auto tc = ly.cell_by_name ("TOP");
  tl_assert (tc.first);

  unsigned int l1 = ly.get_layer (db::LayerProperties (1, 0));
  unsigned int l2 = ly.get_layer (db::LayerProperties (2, 0));
  unsigned int l3 = ly.get_layer (db::LayerProperties (3, 0));

  std::map<unsigned int, db::Region> geo;
  geo.insert (std::make_pair (l2, db::Region (db::RecursiveShapeIterator (ly, ly.cell (tc.second), l2))));

  pex::RNetwork network;

  pex::RExtractorTech tech;

  pex::RExtractorTechVia via1;
  via1.bottom_conductor = l1;
  via1.cut_layer = l2;
  via1.top_conductor = l3;
  via1.resistance = 2.0;
  via1.merge_distance = 0.2;
  tech.vias.push_back (via1);

  std::map<unsigned int, std::vector<pex::RNetExtractor::ViaPort> > via_ports;
  rex.create_via_ports (tech, geo, via_ports, network);

  EXPECT_EQ (via_ports [l1].size (), size_t (6));
  EXPECT_EQ (via_ports [l2].size (), size_t (0));
  EXPECT_EQ (via_ports [l3].size (), size_t (6));

  EXPECT_EQ (network.to_string (true),
    "R $0(4.6,2.8;4.8,3) $1(4.6,2.8;4.8,3) 50\n"
    "R $2(2.5,3.7;2.7,3.9) $3(2.5,3.7;2.7,3.9) 50\n"
    "R $4(3,3.7;3.2,3.9) $5(3,3.7;3.2,3.9) 50\n"
    "R $6(2.2,1.2;3.4,3.4) $7(2.2,1.2;3.4,3.4) 2.77778\n"
    "R $8(0.4,0.4;2.2,4.2) $9(0.4,0.4;2.2,4.2) 1\n"
    "R $10(0.6,4.9;1.2,5.1) $11(0.6,4.9;1.2,5.1) 25"
  );
}