def METHOD_NAME(self, address):
    split = self.colbits - self.address_align
    if isinstance(address, int):
        return (address & (2**split - 1)) << self.address_align
    else:
        return Cat(Replicate(0, self.address_align), address[:split])