Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add Testcases #9

Merged
merged 6 commits into from
Nov 25, 2024
Merged

Add Testcases #9

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
1beff6a
Add child, depth to language
glorialeezero Nov 22, 2024
939ac5d
Modify test cases
glorialeezero Nov 22, 2024
c7ceb52
Add transition
glorialeezero Nov 22, 2024
525a5c9
Add transition testcases
glorialeezero Nov 22, 2024
e76776e
Modify language function
glorialeezero Nov 25, 2024
f48bc1b
Add testcases
glorialeezero Nov 25, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
198 changes: 158 additions & 40 deletions qupsy/language.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -23,9 +23,70 @@ def cost(self) -> int:

@property
@abstractmethod
def depth(self) -> int:
def children(self) -> list[Aexp]:
pass

@property
def depth(self) -> int:
children = self.children
return 1 + (
max([child.depth for child in children]) if len(children) > 0 else 0
)

@property
def filled(self) -> bool:
for aexp in self.children:
if not aexp.filled:
return False
return True

def copy(self) -> Aexp:
return self.__class__(*[child.copy() for child in self.children])


class Integer(Aexp):
def __init__(self, value: int) -> None:
self.value = value

def __str__(self) -> str:
return str(self.value)

def __repr__(self) -> str:
return f"{self.value}"

@property
def cost(self) -> int:
return 0

@property
def children(self) -> list[Aexp]:
return []

def copy(self) -> Integer:
return Integer(self.value)


class Var(Aexp):
def __init__(self, name: str) -> None:
self.name = name

def __str__(self) -> str:
return self.name

def __repr__(self) -> str:
return f"{self.name}"

@property
def cost(self) -> int:
return 0

@property
def children(self) -> list[Aexp]:
return []

def copy(self) -> Var:
return Var(self.name)


class HoleAexp(Aexp):
def __str__(self) -> str:
Expand All @@ -39,8 +100,12 @@ def cost(self) -> int:
return 3

@property
def depth(self) -> int:
return 1
def children(self) -> list[Aexp]:
return []

@property
def filled(self) -> bool:
return False


class Add(Aexp):
Expand All @@ -62,8 +127,8 @@ def cost(self) -> int:
return self.a.cost + self.b.cost + 3

@property
def depth(self) -> int:
return max(self.a.depth, self.b.depth) + 1
def children(self) -> list[Aexp]:
return [self.a, self.b]


class Sub(Aexp):
Expand All @@ -85,8 +150,8 @@ def cost(self) -> int:
return self.a.cost + self.b.cost + 3

@property
def depth(self) -> int:
return max(self.a.depth, self.b.depth) + 1
def children(self) -> list[Aexp]:
return [self.a, self.b]


class Mul(Aexp):
Expand All @@ -108,8 +173,8 @@ def cost(self) -> int:
return self.a.cost + self.b.cost + 3

@property
def depth(self) -> int:
return max(self.a.depth, self.b.depth) + 1
def children(self) -> list[Aexp]:
return [self.a, self.b]


class Div(Aexp):
Expand All @@ -131,8 +196,8 @@ def cost(self) -> int:
return self.a.cost + self.b.cost + 3

@property
def depth(self) -> int:
return max(self.a.depth, self.b.depth) + 1
def children(self) -> list[Aexp]:
return [self.a, self.b]


class Gate(ABC):
Expand All @@ -151,9 +216,26 @@ def cost(self) -> int:

@property
@abstractmethod
def depth(self) -> int:
def children(self) -> list[Aexp]:
pass

@property
def depth(self) -> int:
children = self.children
return 1 + (
max([child.depth for child in children]) if len(children) > 0 else 0
)

@property
def filled(self) -> bool:
for aexp in self.children:
if not aexp.filled:
return False
return True

def copy(self) -> Gate:
return self.__class__(*[child.copy() for child in self.children])


class HoleGate(Gate):
def __str__(self) -> str:
Expand All @@ -167,8 +249,12 @@ def cost(self) -> int:
return 3

@property
def depth(self) -> int:
return 1
def children(self) -> list[Aexp]:
return []

@property
def filled(self) -> bool:
return False


class H(Gate):
Expand All @@ -188,8 +274,8 @@ def cost(self) -> int:
return self.qreg.cost + 2

@property
def depth(self) -> int:
return self.qreg.depth + 1
def children(self) -> list[Aexp]:
return [self.qreg]


class X(Gate):
Expand All @@ -209,8 +295,8 @@ def cost(self) -> int:
return self.qreg.cost + 2

@property
def depth(self) -> int:
return self.qreg.depth + 1
def children(self) -> list[Aexp]:
return [self.qreg]


class Ry(Gate):
Expand All @@ -236,8 +322,8 @@ def cost(self) -> int:
return self.qreg.cost + self.p.cost + self.q.cost + 2

@property
def depth(self) -> int:
return max(self.qreg.depth, self.p.depth, self.q.depth) + 1
def children(self) -> list[Aexp]:
return [self.qreg, self.p, self.q]


class CX(Gate):
Expand All @@ -259,8 +345,8 @@ def cost(self) -> int:
return self.qreg1.cost + self.qreg2.cost + 2

@property
def depth(self) -> int:
return max(self.qreg1.depth, self.qreg2.depth) + 1
def children(self) -> list[Aexp]:
return [self.qreg1, self.qreg2]


class CRy(Gate):
Expand Down Expand Up @@ -292,8 +378,8 @@ def cost(self) -> int:
return self.qreg1.cost + self.qreg2.cost + self.p.cost + self.q.cost + 2

@property
def depth(self) -> int:
return max(self.qreg1.depth, self.qreg2.depth, self.p.depth, self.q.depth) + 1
def children(self) -> list[Aexp]:
return [self.qreg1, self.qreg2, self.p, self.q]


class Cmd(ABC):
Expand All @@ -313,9 +399,26 @@ def cost(self) -> int:

@property
@abstractmethod
def depth(self) -> int:
def children(self) -> list[Cmd | Gate | Aexp]:
pass

@property
def depth(self) -> int:
children = self.children
return 1 + (
max([child.depth for child in children]) if len(children) > 0 else 0
)

@property
def filled(self) -> bool:
for child in self.children:
if not child.filled:
return False
return True

def copy(self) -> Cmd:
return self.__class__(*[child.copy() for child in self.children])


class HoleCmd(Cmd):
def __str__(self) -> str:
Expand All @@ -329,8 +432,12 @@ def cost(self) -> int:
return 5

@property
def depth(self) -> int:
return 1
def children(self) -> list[Cmd | Gate | Aexp]:
return []

@property
def filled(self) -> bool:
return False


class SeqCmd(Cmd):
Expand All @@ -345,15 +452,15 @@ def __str__(self) -> str:
return f"{self.pre}\n{self.post}"

def __repr__(self) -> str:
return f"SeqCmd({self.pre:!r}, {self.post:!r})"
return f"SeqCmd({repr(self.pre)}, {repr(self.post)})"

@property
def cost(self) -> int:
return self.pre.cost + self.post.cost + 5

@property
def depth(self) -> int:
return max(self.pre.depth, self.post.depth) + 1
def children(self) -> list[Cmd | Gate | Aexp]:
return [self.pre, self.post]


class ForCmd(Cmd):
Expand All @@ -380,13 +487,16 @@ def __str__(self) -> str:
def __repr__(self) -> str:
return f"For({self.var!r}, {self.start!r}, {self.end!r}, {self.body!r})"

def copy(self) -> ForCmd:
return ForCmd(self.var, self.start.copy(), self.end.copy(), self.body.copy())

@property
def cost(self) -> int:
return self.start.cost + self.end.cost + self.body.cost + 3

@property
def depth(self) -> int:
return self.body.depth + 1
def children(self) -> list[Cmd | Gate | Aexp]:
return [self.start, self.end, self.body]


class GateCmd(Cmd):
Expand All @@ -406,21 +516,23 @@ def cost(self) -> int:
return self.gate.cost

@property
def depth(self) -> int:
return self.gate.depth
def children(self) -> list[Cmd | Gate | Aexp]:
return [self.gate]


class Pgm:
n: str
body: Cmd

def __init__(self, body: Cmd | None = None) -> None:
def __init__(self, n: str, body: Cmd | None = None) -> None:
self.n = n
self.body = body or HoleCmd()

def __str__(self) -> str:
return str(self.body)

def __repr__(self) -> str:
return f"Pgm({self.body:!r})"
return f"Pgm({repr(self.body)})"

def __lt__(self, other: Pgm) -> bool:
return self.cost < other.cost
Expand All @@ -434,8 +546,14 @@ def depth(self) -> int:
return self.body.depth


GATE_MAP: dict[str, type[Gate]] = {
g.__name__: g
ALL_AEXPS: list[type[Aexp]] = [
a
for a in globals().values()
if isinstance(a, type) and issubclass(a, Aexp) and a != Aexp
]
ALL_GATES: list[type[Gate]] = [
g
for g in globals().values()
if isinstance(g, type) and issubclass(g, Gate)
}
if isinstance(g, type) and issubclass(g, Gate) and g != Gate
]
GATE_MAP: dict[str, type[Gate]] = {g.__name__: g for g in ALL_GATES}
Loading