103 lines
2.7 KiB
Python
103 lines
2.7 KiB
Python
with open('test.text', 'r') as file:
|
|
data: str = file.read().strip()
|
|
|
|
# If block has no ID then it is free
|
|
class FreeBlock:
|
|
size: int
|
|
|
|
def __init__(self, s: int):
|
|
self.size = s
|
|
|
|
def __str__(self) -> str:
|
|
return '.' * self.size
|
|
|
|
class FileBlock:
|
|
size: int
|
|
id: int
|
|
|
|
def __init__(self, size: int, id: int):
|
|
self.size = size
|
|
self.id = id
|
|
|
|
def __str__(self) -> str:
|
|
return str(self.id) * self.size
|
|
|
|
class DiskMap:
|
|
blocks: list[FreeBlock | FileBlock]
|
|
|
|
def __init__(self, s: str):
|
|
self.blocks = []
|
|
self.parse(s)
|
|
|
|
def parse(self, s: str) -> None:
|
|
is_file = True
|
|
id = 0
|
|
for char in s:
|
|
b = FreeBlock(int(char))
|
|
if is_file:
|
|
b = FileBlock(int(char), id)
|
|
id += 1
|
|
is_file = not is_file
|
|
|
|
self.blocks.append(b)
|
|
|
|
# Returns index of first free block or None
|
|
def findFreeBlock(self) -> int | None:
|
|
for i, block in enumerate(self.blocks):
|
|
if type(block) == FreeBlock:
|
|
return i
|
|
|
|
return None
|
|
|
|
# Returns index of last file block
|
|
def findLastFile(self) -> int | None:
|
|
for i, block in reversed(list(enumerate(self.blocks))):
|
|
if type(block) == FileBlock:
|
|
return i
|
|
|
|
return None
|
|
|
|
def compact(self) -> None:
|
|
while True:
|
|
free_index = self.findFreeBlock()
|
|
file_index = self.findLastFile()
|
|
|
|
if free_index > file_index:
|
|
break
|
|
|
|
free_size: int = self.blocks[free_index].size
|
|
file_size: int = self.blocks[file_index].size
|
|
file_id: int = self.blocks[file_index].id
|
|
|
|
remaining_file_size = file_size - free_size
|
|
remaining_free_size = 0
|
|
if remaining_file_size < 0:
|
|
remaining_free_size = abs(remaining_file_size)
|
|
remaining_file_size = 0
|
|
|
|
if remaining_free_size != 0:
|
|
self.blocks[free_index] = FileBlock(free_size - remaining_free_size, file_id)
|
|
self.blocks.insert(free_index + 1, FreeBlock(remaining_file_size))
|
|
self.blocks[file_id] = FreeBlock(free_size)
|
|
else:
|
|
self.blocks[free_index] = FileBlock(free_size, file_id)
|
|
self.blocks[file_index] = FreeBlock(free_size)
|
|
|
|
if remaining_file_size:
|
|
self.blocks.insert(file_index, FileBlock(remaining_file_size, file_id))
|
|
|
|
print(free_index)
|
|
self.print()
|
|
|
|
def print(self) -> None:
|
|
for block in self.blocks:
|
|
print(block, end='')
|
|
print()
|
|
|
|
# Parse Data
|
|
diskmap = DiskMap(data)
|
|
|
|
diskmap.print()
|
|
diskmap.compact()
|
|
diskmap.print()
|