atcoder_include = re.compile(

r'#include\s*["<](atcoder/[a-z_]*(|.hpp))[">]\s*')

include_guard = re.compile(r'#.*ATCODER_[A-Z_]*_HPP')

def is_ignored_line(self, line) -> bool:

if self.include_guard.match(line):

return True

if line.strip() == "#pragma once":

return True

if line.strip().startswith('//'):

return True

return False

def __init__(self, lib_paths: List[Path]):

self.lib_paths = lib_paths

included = set() # type: Set[Path]

def find_acl(self, acl_name: str) -> Path:

for lib_path in self.lib_paths:

path = lib_path / acl_name

if path.exists():

return path

logger.error('cannot find: {}'.format(acl_name))

raise FileNotFoundError()

def expand_acl(self, acl_file_path: Path) -> List[str]:

if acl_file_path in self.included:

logger.info('already included: {}'.format(acl_file_path.name))

return []

self.included.add(acl_file_path)

logger.info('include: {}'.format(acl_file_path.name))

acl_source = open(str(acl_file_path)).read()

result = [] # type: List[str]

for line in acl_source.splitlines():

if self.is_ignored_line(line):

continue

m = self.atcoder_include.match(line)

if m:

name = m.group(1)

result.extend(self.expand_acl(self.find_acl(name)))

continue

result.append(line)

return result

def expand(self, source: str, origname) -> str:

self.included = set()

result = [] # type: List[str]

linenum = 0

for line in source.splitlines():

linenum = 1

m = self.atcoder_include.match(line)

if m:

acl_path = self.find_acl(m.group(1))

result.extend(self.expand_acl(acl_path))

if origname:

result.append('#line ' str(linenum 1) ' "' origname '"')

continue

result.append(line)