Will Hawkins
35b2537754
Add support for directions on parameters and attributed types. The latter is necessary to support the former, but is not limited to the direction use case. An attributed type is a P4 type with attributes (like its direction, whether it is read only, etc.) Other attributes will be added in the future. Changes necessary to support attributed types include: 1. Global instances tracked during compilation are not attributed types and not simply types. 2. (future) Type checking will have to make sure that a types attributes do not affect type compatibility. Signed-off-by: Will Hawkins <hawkinsw@obs.cr>
100 lines
3.3 KiB
Swift
100 lines
3.3 KiB
Swift
// p4rse, Copyright 2026, Will Hawkins
|
|
//
|
|
// This file is part of p4rse.
|
|
//
|
|
// This file is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// This program is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU General Public License
|
|
// along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
import Common
|
|
|
|
public struct ExpressionStatement {
|
|
public init() {}
|
|
}
|
|
|
|
public struct Program {
|
|
public var types: [P4Type] = Array()
|
|
public var instances: [P4TypeAttributed] = Array()
|
|
|
|
/// Type of closure for filtering results from ``Program/InstancesWithTypes(_:)``
|
|
public typealias AttributedTypeFilter = (P4TypeAttributed) -> Bool
|
|
/// Type of closure for filtering results from ``Program/TypesWithTypes(_:)``
|
|
public typealias TypeFilter = (P4Type) -> Bool
|
|
|
|
/// Retrieve global instances in the compiled P4 program.
|
|
public func InstancesWithTypes() -> [P4TypeAttributed] {
|
|
return self.instances
|
|
}
|
|
|
|
/// Retrieve global instances in the compiled P4 program.
|
|
///
|
|
/// Use the given filter to select which of the global instances
|
|
/// from the compiled P4 program to retrieve.
|
|
///
|
|
/// If the compiled P4 program (from the source in the
|
|
/// string `p4_program_with_control_decl`) has two Control
|
|
/// instances and you only want to select the one named simple,
|
|
/// you could use a filter like
|
|
///
|
|
/// @Snippet(path: "use-program-instanceswithtypes", slice: "include")
|
|
///
|
|
public func InstancesWithTypes(_ filter: AttributedTypeFilter) -> [P4TypeAttributed] {
|
|
return self.instances.filter { instance in
|
|
filter(instance)
|
|
}
|
|
}
|
|
|
|
/// Retrieve global types in the compiled P4 program.
|
|
public func TypesWithTypes() -> [P4Type] {
|
|
return self.types
|
|
}
|
|
|
|
/// Retrieve global types declared in the compiled P4 program.
|
|
///
|
|
/// Use the given filter to select which of the global types
|
|
/// declared in the compiled P4 program to retrieve.
|
|
///
|
|
/// If the compiled P4 program (from the source in the
|
|
/// string `p4_program_with_struct_decl`) has two structs declared and
|
|
/// you only want to select the one named `agg`, you could
|
|
/// use a filter like
|
|
///
|
|
/// @Snippet(path: "use-program-typeswithtypes", slice: "include")
|
|
///
|
|
public func TypesWithTypes(_ filter: TypeFilter) -> [P4Type] {
|
|
return self.types.filter { instance in
|
|
filter(instance)
|
|
}
|
|
}
|
|
|
|
/// Find the program's main parser
|
|
///
|
|
/// Note: For now, the main parser is expected to be named main_parser.
|
|
public func starting_parser() -> Result<Parser> {
|
|
return self.find_parser(withName: Identifier(name: "main_parser"))
|
|
}
|
|
|
|
public func find_parser(withName name: Identifier) -> Result<Parser> {
|
|
for instance in self.instances {
|
|
guard let parser = instance.type as? Parser else {
|
|
continue
|
|
}
|
|
if parser.name == name {
|
|
return .Ok(parser)
|
|
}
|
|
}
|
|
return .Error(Error(withMessage: "Could not find parser named \(name)"))
|
|
}
|
|
|
|
public init() {}
|
|
}
|