########################################################################################################################
# Copyright 2022 Mabrains Company LLC
#
# Licensed under the LGPL v2.1 License (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#      https://www.gnu.org/licenses/old-licenses/lgpl-2.1.en.html
#
# 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 file is authored by:
#           - <Mina Maksimous> <mina_maksimous@mabrains.com>
##
########################################################################################################################
from .layers_definiations import *
import pya


class mimcap_m4():
    mimcap_drawing_offest = 0.5
    metal4_margin_right = 2.51
    mimcap_enc_metal5 = 0.08
    met5_side_pr_spacing = 1.02
    met5_side_width = 1.6

    def number_spc_contacts(self, box_width, min_enc, cont_spc, cont_width):
        """ Calculate number of cantacts in a given dimensions and the free space for symmetry.

            By getting the min enclosure,the width of the box,the width of the cont. or via
            and the spacing between cont. or via

            Parameters
            ----------
            box_width : double
                The length you place the via or cont. in

            min_enc : double
                the spacing between the edge of the box and the first via or cont.

            cont_spc : double
                the spacing between different via's or cont

            cont_width: double
                the cont. or via width in the same direction

        """

        spc_cont = box_width - 2 * min_enc
        num_cont = int((spc_cont + cont_spc) / (cont_width + cont_spc))
        free_spc = box_width - (num_cont * cont_width +
                                (num_cont - 1) * cont_spc)
        return num_cont, free_spc

    def __init__(self, layout, w, l, pin0 = 'p0',pin1 = 'n0',connection_labels = 1):
        self.layout = layout
        self.l_cap2m = self.layout.layer(cap2m_lay_num, cap2m_lay_dt)
        self.l_met4 = self.layout.layer(met4_lay_num, met4_lay_dt)
        self.l_met5 = self.layout.layer(met5_lay_num, met5_lay_dt)
        self.l_via4 = self.layout.layer(via4_lay_num, via4_lay_dt)
        self.l_met5_label = self.layout.layer(met5_label_lay_num,met5_label_lay_dt)
        self.l_prbndry = self.layout.layer(prbndry_lay_num, prbndry_lay_dt)
        self.percision = 1/self.layout.dbu
        self.pin0 = pin0
        self.pin1 = pin1
        self.cell = self.layout.create_cell(
            "sky130_fd_pr__cap_mim_m3_2_w"+str(w)+"_l"+str(l))
        self.w = w * self.percision
        self.l = l * self .percision
        self.connection_labels=connection_labels

    def draw_vias(self, box, via4_cell):

        via4_size = 0.8*self.percision
        via4_spc = 0.8*self.percision
        met_via4_enc = 0.310*self.percision
        AL_via4 = pya.Box(0, 0, via4_size, via4_size)
        via4_cell.shapes(self.l_via4).insert(AL_via4)
        num_via4_1, via4_free_spc_1 = self.number_spc_contacts(
            box.width(), met_via4_enc, via4_spc, via4_size)
        num_via4_2, via4_free_spc_2 = self.number_spc_contacts(
            box.height(), met_via4_enc, via4_spc, via4_size)
        via4_arr = pya.CellInstArray(via4_cell.cell_index(), pya.Trans(
            pya.Point(box.p1.x+via4_free_spc_1 / 2, box.p1.y + via4_free_spc_2 / 2)),
            pya.Vector(via4_spc + via4_size, 0),
            pya.Vector(0, via4_spc + via4_size),
            num_via4_1, num_via4_2)
        return via4_arr

    def draw_cap(self):
        
        mimcap_box = pya.Box(mimcap_m4.mimcap_drawing_offest*self.percision,
                             mimcap_m4.mimcap_drawing_offest*self.percision,
                             mimcap_m4.mimcap_drawing_offest*self.percision+self.w,
                             mimcap_m4.mimcap_drawing_offest*self.percision+self.l)

        self.cell.shapes(self.l_cap2m).insert(mimcap_box)
        met5_center_box = mimcap_box.enlarge(-1*mimcap_m4.mimcap_enc_metal5*self.percision,

                                             -1*mimcap_m4.mimcap_enc_metal5*self.percision)
        pin0_text = pya.Text(self.pin0, met5_center_box.center().x,
                                             met5_center_box.center().y)
        self.cell.shapes(self.l_met5).insert(met5_center_box)
        met4_box = pya.Box(0, 0,
                           2*mimcap_m4.mimcap_drawing_offest*self.percision +
                           self.w+mimcap_m4.metal4_margin_right*self.percision,
                           2*mimcap_m4.mimcap_drawing_offest*self.percision+self.l)

        self.cell.shapes(self.l_met4).insert(met4_box)

        prbndry_box = pya.Box(0, 0,
                              2*mimcap_m4.mimcap_drawing_offest*self.percision+self.w,
                              2*mimcap_m4.mimcap_drawing_offest*self.percision+self.l)

        self.cell.shapes(self.l_prbndry).insert(prbndry_box)

        met5_side_box = pya.Box(prbndry_box.p2.x+mimcap_m4.met5_side_pr_spacing*self.percision,
                                prbndry_box.p1.y,
                                prbndry_box.p2.x+mimcap_m4.met5_side_pr_spacing *
                                self.percision+mimcap_m4.met5_side_width*self.percision,
                                prbndry_box.p2.y)

        pin1_text = pya.Text(self.pin1, met5_side_box.center().x,
                                             met5_side_box.center().y)
        if self.connection_labels:
            self.cell.shapes(self.l_met5_label).insert(pin0_text)
            self.cell.shapes(self.l_met5_label).insert(pin1_text)
        self.cell.shapes(self.l_met5).insert(met5_side_box)

        print('--->', self.layout.cell("via4"))
        if self.layout.cell("via4") == None:
            via4_cell = self.layout.create_cell("via4")
            print('--->', self.layout.cell("via4"))
        else:
            via4_cell = self.layout.cell("via4")

        via4_arr = self.draw_vias(met5_center_box, via4_cell)

        self.cell.insert(via4_arr)

        via4_arr = self.draw_vias(met5_side_box, via4_cell)
        self.cell.insert(via4_arr)
        return self.cell