diff options
| author | Albert Magyar | 2020-04-10 12:59:17 -0700 |
|---|---|---|
| committer | Albert Magyar | 2020-04-13 13:02:34 -0700 |
| commit | 8cb5aa96983210942336beb05a17ae1cade8b11a (patch) | |
| tree | 1947e7deec6ad3b20254a61a56261b4a49463dda /src | |
| parent | 319e2fcff501ac0e8c34625b35b154c5f6c0c4c3 (diff) | |
Split Resolves into separate files
* Remove unused imports
Diffstat (limited to 'src')
| -rw-r--r-- | src/main/scala/firrtl/passes/CInferMDir.scala (renamed from src/main/scala/firrtl/passes/Resolves.scala) | 80 | ||||
| -rw-r--r-- | src/main/scala/firrtl/passes/ResolveFlows.scala | 46 | ||||
| -rw-r--r-- | src/main/scala/firrtl/passes/ResolveKinds.scala | 49 |
3 files changed, 95 insertions, 80 deletions
diff --git a/src/main/scala/firrtl/passes/Resolves.scala b/src/main/scala/firrtl/passes/CInferMDir.scala index de54a5c1..33587741 100644 --- a/src/main/scala/firrtl/passes/Resolves.scala +++ b/src/main/scala/firrtl/passes/CInferMDir.scala @@ -8,86 +8,6 @@ import firrtl.Mappers._ import firrtl.options.{Dependency, PreservesAll} import Utils.throwInternalError - -object ResolveKinds extends Pass with PreservesAll[Transform] { - - override val prerequisites = firrtl.stage.Forms.WorkingIR - - type KindMap = collection.mutable.LinkedHashMap[String, Kind] - - def find_port(kinds: KindMap)(p: Port): Port = { - kinds(p.name) = PortKind ; p - } - - def find_stmt(kinds: KindMap)(s: Statement):Statement = { - s match { - case sx: DefWire => kinds(sx.name) = WireKind - case sx: DefNode => kinds(sx.name) = NodeKind - case sx: DefRegister => kinds(sx.name) = RegKind - case sx: WDefInstance => kinds(sx.name) = InstanceKind - case sx: DefMemory => kinds(sx.name) = MemKind - case _ => - } - s map find_stmt(kinds) - } - - def resolve_expr(kinds: KindMap)(e: Expression): Expression = e match { - case ex: WRef => ex copy (kind = kinds(ex.name)) - case _ => e map resolve_expr(kinds) - } - - def resolve_stmt(kinds: KindMap)(s: Statement): Statement = - s map resolve_stmt(kinds) map resolve_expr(kinds) - - def resolve_kinds(m: DefModule): DefModule = { - val kinds = new KindMap - (m map find_port(kinds) - map find_stmt(kinds) - map resolve_stmt(kinds)) - } - - def run(c: Circuit): Circuit = - c copy (modules = c.modules map resolve_kinds) -} - -object ResolveFlows extends Pass with PreservesAll[Transform] { - - override val prerequisites = - Seq( Dependency(passes.ResolveKinds), - Dependency(passes.InferTypes), - Dependency(passes.Uniquify) ) ++ firrtl.stage.Forms.WorkingIR - - def resolve_e(g: Flow)(e: Expression): Expression = e match { - case ex: WRef => ex copy (flow = g) - case WSubField(exp, name, tpe, _) => WSubField( - Utils.field_flip(exp.tpe, name) match { - case Default => resolve_e(g)(exp) - case Flip => resolve_e(Utils.swap(g))(exp) - }, name, tpe, g) - case WSubIndex(exp, value, tpe, _) => - WSubIndex(resolve_e(g)(exp), value, tpe, g) - case WSubAccess(exp, index, tpe, _) => - WSubAccess(resolve_e(g)(exp), resolve_e(SourceFlow)(index), tpe, g) - case _ => e map resolve_e(g) - } - - def resolve_s(s: Statement): Statement = s match { - //TODO(azidar): pretty sure don't need to do anything for Attach, but not positive... - case IsInvalid(info, expr) => - IsInvalid(info, resolve_e(SinkFlow)(expr)) - case Connect(info, loc, expr) => - Connect(info, resolve_e(SinkFlow)(loc), resolve_e(SourceFlow)(expr)) - case PartialConnect(info, loc, expr) => - PartialConnect(info, resolve_e(SinkFlow)(loc), resolve_e(SourceFlow)(expr)) - case sx => sx map resolve_e(SourceFlow) map resolve_s - } - - def resolve_flow(m: DefModule): DefModule = m map resolve_s - - def run(c: Circuit): Circuit = - c copy (modules = c.modules map resolve_flow) -} - object CInferMDir extends Pass with PreservesAll[Transform] { override val prerequisites = firrtl.stage.Forms.ChirrtlForm :+ Dependency(CInferTypes) diff --git a/src/main/scala/firrtl/passes/ResolveFlows.scala b/src/main/scala/firrtl/passes/ResolveFlows.scala new file mode 100644 index 00000000..8f413082 --- /dev/null +++ b/src/main/scala/firrtl/passes/ResolveFlows.scala @@ -0,0 +1,46 @@ +// See LICENSE for license details. + +package firrtl.passes + +import firrtl._ +import firrtl.ir._ +import firrtl.Mappers._ +import firrtl.options.{Dependency, PreservesAll} + +object ResolveFlows extends Pass with PreservesAll[Transform] { + + override val prerequisites = + Seq( Dependency(passes.ResolveKinds), + Dependency(passes.InferTypes), + Dependency(passes.Uniquify) ) ++ firrtl.stage.Forms.WorkingIR + + def resolve_e(g: Flow)(e: Expression): Expression = e match { + case ex: WRef => ex copy (flow = g) + case WSubField(exp, name, tpe, _) => WSubField( + Utils.field_flip(exp.tpe, name) match { + case Default => resolve_e(g)(exp) + case Flip => resolve_e(Utils.swap(g))(exp) + }, name, tpe, g) + case WSubIndex(exp, value, tpe, _) => + WSubIndex(resolve_e(g)(exp), value, tpe, g) + case WSubAccess(exp, index, tpe, _) => + WSubAccess(resolve_e(g)(exp), resolve_e(SourceFlow)(index), tpe, g) + case _ => e map resolve_e(g) + } + + def resolve_s(s: Statement): Statement = s match { + //TODO(azidar): pretty sure don't need to do anything for Attach, but not positive... + case IsInvalid(info, expr) => + IsInvalid(info, resolve_e(SinkFlow)(expr)) + case Connect(info, loc, expr) => + Connect(info, resolve_e(SinkFlow)(loc), resolve_e(SourceFlow)(expr)) + case PartialConnect(info, loc, expr) => + PartialConnect(info, resolve_e(SinkFlow)(loc), resolve_e(SourceFlow)(expr)) + case sx => sx map resolve_e(SourceFlow) map resolve_s + } + + def resolve_flow(m: DefModule): DefModule = m map resolve_s + + def run(c: Circuit): Circuit = + c copy (modules = c.modules map resolve_flow) +} diff --git a/src/main/scala/firrtl/passes/ResolveKinds.scala b/src/main/scala/firrtl/passes/ResolveKinds.scala new file mode 100644 index 00000000..fb36ccd5 --- /dev/null +++ b/src/main/scala/firrtl/passes/ResolveKinds.scala @@ -0,0 +1,49 @@ +// See LICENSE for license details. + +package firrtl.passes + +import firrtl._ +import firrtl.ir._ +import firrtl.Mappers._ +import firrtl.options.PreservesAll + +object ResolveKinds extends Pass with PreservesAll[Transform] { + + override val prerequisites = firrtl.stage.Forms.WorkingIR + + type KindMap = collection.mutable.LinkedHashMap[String, Kind] + + def find_port(kinds: KindMap)(p: Port): Port = { + kinds(p.name) = PortKind ; p + } + + def find_stmt(kinds: KindMap)(s: Statement):Statement = { + s match { + case sx: DefWire => kinds(sx.name) = WireKind + case sx: DefNode => kinds(sx.name) = NodeKind + case sx: DefRegister => kinds(sx.name) = RegKind + case sx: WDefInstance => kinds(sx.name) = InstanceKind + case sx: DefMemory => kinds(sx.name) = MemKind + case _ => + } + s map find_stmt(kinds) + } + + def resolve_expr(kinds: KindMap)(e: Expression): Expression = e match { + case ex: WRef => ex copy (kind = kinds(ex.name)) + case _ => e map resolve_expr(kinds) + } + + def resolve_stmt(kinds: KindMap)(s: Statement): Statement = + s map resolve_stmt(kinds) map resolve_expr(kinds) + + def resolve_kinds(m: DefModule): DefModule = { + val kinds = new KindMap + (m map find_port(kinds) + map find_stmt(kinds) + map resolve_stmt(kinds)) + } + + def run(c: Circuit): Circuit = + c copy (modules = c.modules map resolve_kinds) +} |
