feature/treesitter-tag-queries #2
3 changed files with 9 additions and 10 deletions
|
|
@ -51,8 +51,8 @@ def _match_glob(filepath, pattern):
|
|||
IMPORT_PATTERNS = [
|
||||
re.compile(r"^\s*import\s+"),
|
||||
re.compile(r"^\s*from\s+"),
|
||||
re.compile(r"^\s*require\s+"),
|
||||
re.compile(r"^\s*require_relative\s+"),
|
||||
re.compile(r"^\s*require\s+"),
|
||||
re.compile(r"^\s*#\s*include\s+"),
|
||||
re.compile(r"^\s*use\s+"),
|
||||
re.compile(r"^\s*include\s+"),
|
||||
|
|
|
|||
|
|
@ -51,7 +51,7 @@ BEHAVIORAL_PREFIXES = (
|
|||
"module",
|
||||
)
|
||||
|
||||
CALL_PREFIXES = ("function", "reference.call", "reference.class")
|
||||
CALL_PREFIXES = ("function.call", "function.method.call", "reference.call", "reference.class", "constructor", "function")
|
||||
|
||||
def _extract_call_tokens(seq):
|
||||
"""Extract ordered call-like tokens from a method sequence.
|
||||
|
|
@ -107,8 +107,6 @@ def _find_arglist_node(parent):
|
|||
for child in parent.children:
|
||||
if child.type in ("argument_list", "arguments"):
|
||||
return child
|
||||
if child.type == "template_string":
|
||||
return child
|
||||
return None
|
||||
|
||||
|
||||
|
|
|
|||
|
|
@ -4,7 +4,7 @@
|
|||
│ (.py .js .ts .kt .rb .go │
|
||||
│ .rs .java .c .cpp .h) │
|
||||
└──────────┬────────────────────┘
|
||||
│ scan_directory() + .gitignore
|
||||
│ scan_directory()
|
||||
▼
|
||||
┌───────────────────────────────┐
|
||||
│ Files grouped by extension │
|
||||
|
|
@ -68,11 +68,12 @@
|
|||
│ │ └───────────┬────────────┘ │
|
||||
│ ▼ ▼ │
|
||||
│ ┌──────────────────────────────────────────────────┐ │
|
||||
│ │ infer_ensemble(symbol_seqs) │ │
|
||||
│ │ ├── CRX (fast, unordered) │ │
|
||||
│ │ ├── iDRegEx (ordered regex) │ │
|
||||
│ │ └── kOREInference (noisy, probabilistic) │ │
|
||||
│ │ └── pick best by MDL score │ │
|
||||
│ │ infer_ensemble(symbol_seqs, min_coverage=0.8) │ │
|
||||
│ │ ├── CRX (fast, unordered) │ │
|
||||
│ │ ├── iDRegEx (ordered regex) │ │
|
||||
│ │ └── kOREInference (noisy, probabilistic) │ │
|
||||
│ │ └── pick best by MDL score │ │
|
||||
│ │ └── core/outlier split via _find_core(0.8) │ │
|
||||
│ └──────────────────────┬───────────────────────────┘ │
|
||||
│ ▼ │
|
||||
│ (label, result, count, meta) │
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue