ptx-90-parser 0.4.4

Parse NVIDIA PTX 9.0 assembly into a structured AST and explore modules via a CLI.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73

use crate::lexer::PtxToken;
use std::fmt::{self, Write};

/// Utility that performs the inverse of the lexer: it writes textual PTX for a
/// sequence of [`PtxSpecToken`] values.
pub struct PtxUnlexer;

impl PtxUnlexer {
    /// Write the textual PTX representation of `tokens` to `writer`.
    pub fn write_tokens<W>(writer: &mut W, tokens: &[PtxToken]) -> fmt::Result
    where
        W: Write,
    {
        for token in tokens {
            write_token(writer, token)?;
        }
        Ok(())
    }

    /// Convenience helper that produces a PTX string for `tokens`.
    pub fn to_string(tokens: &[PtxToken]) -> Result<String, fmt::Error> {
        let mut buffer = String::new();
        Self::write_tokens(&mut buffer, tokens)?;
        Ok(buffer)
    }
}

fn write_token<W: Write>(writer: &mut W, token: &PtxToken) -> fmt::Result {
    match token {
        PtxToken::Identifier(name)
        | PtxToken::DecimalInteger(name)
        | PtxToken::HexInteger(name)
        | PtxToken::BinaryInteger(name)
        | PtxToken::OctalInteger(name)
        | PtxToken::Float(name)
        | PtxToken::FloatExponent(name)
        | PtxToken::HexFloatSingle(name)
        | PtxToken::HexFloatDouble(name)
        | PtxToken::Register(name) => writer.write_str(name),
        PtxToken::StringLiteral(name) => {
            writer.write_char('"')?;
            writer.write_str(name)?;
            writer.write_char('"')
        }
        PtxToken::Dot => writer.write_char('.'),
        PtxToken::Comma => writer.write_char(','),
        PtxToken::Semicolon => writer.write_char(';'),
        PtxToken::Colon => writer.write_char(':'),
        PtxToken::DoubleColon => writer.write_str("::"),
        PtxToken::LParen => writer.write_char('('),
        PtxToken::RParen => writer.write_char(')'),
        PtxToken::LBracket => writer.write_char('['),
        PtxToken::RBracket => writer.write_char(']'),
        PtxToken::LBrace => writer.write_char('{'),
        PtxToken::RBrace => writer.write_char('}'),
        PtxToken::Plus => writer.write_char('+'),
        PtxToken::Minus => writer.write_char('-'),
        PtxToken::Star => writer.write_char('*'),
        PtxToken::Slash => writer.write_char('/'),
        PtxToken::LAngle => writer.write_char('<'),
        PtxToken::RAngle => writer.write_char('>'),
        PtxToken::Equals => writer.write_char('='),
        PtxToken::Percent => writer.write_char('%'),
        PtxToken::Exclaim => writer.write_char('!'),
        PtxToken::Pipe => writer.write_char('|'),
        PtxToken::Ampersand => writer.write_char('&'),
        PtxToken::Caret => writer.write_char('^'),
        PtxToken::Tilde => writer.write_char('~'),
        PtxToken::At => writer.write_char('@'),
        PtxToken::Space => writer.write_char(' '),
        PtxToken::Newline => writer.write_char('\n'),
    }
}