generate_objc_conversion_helpers.py [plain text]
import logging
import string
from string import Template
from generator import Generator
from models import EnumType
from objc_generator import ObjCGenerator
from objc_generator_templates import ObjCGeneratorTemplates as ObjCTemplates
log = logging.getLogger('global')
def add_newline(lines):
if lines and lines[-1] == '':
return
lines.append('')
class ObjCConversionHelpersGenerator(Generator):
def __init__(self, model, input_filepath):
Generator.__init__(self, model, input_filepath)
def output_filename(self):
return '%sEnumConversionHelpers.h' % ObjCGenerator.OBJC_PREFIX
def domains_to_generate(self):
return filter(ObjCGenerator.should_generate_domain_types_filter(self.model()), Generator.domains_to_generate(self))
def generate_output(self):
headers = [
'"%sArrayConversionHelpers.h"' % ObjCGenerator.OBJC_PREFIX,
]
header_args = {
'includes': '\n'.join(['#import ' + header for header in headers]),
}
domains = self.domains_to_generate()
sections = []
sections.append(self.generate_license())
sections.append(Template(ObjCTemplates.ConversionHelpersPrelude).substitute(None, **header_args))
sections.append(Template(ObjCTemplates.ConversionHelpersStandard).substitute(None))
sections.extend(map(self._generate_enum_conversion_functions, domains))
sections.append(Template(ObjCTemplates.ConversionHelpersPostlude).substitute(None, **header_args))
return '\n\n'.join(sections)
def _generate_enum_conversion_functions(self, domain):
lines = []
for declaration in domain.type_declarations:
if isinstance(declaration.type, EnumType):
add_newline(lines)
lines.append(self._generate_anonymous_enum_conversion_for_declaration(domain, declaration))
else:
for member in declaration.type_members:
if (isinstance(member.type, EnumType) and member.type.is_anonymous):
add_newline(lines)
lines.append(self._generate_anonymous_enum_conversion_for_member(domain, declaration, member))
for command in domain.commands:
for parameter in command.call_parameters:
if (isinstance(parameter.type, EnumType) and parameter.type.is_anonymous):
add_newline(lines)
lines.append(self._generate_anonymous_enum_conversion_for_parameter(domain, command.command_name, parameter))
for parameter in command.return_parameters:
if (isinstance(parameter.type, EnumType) and parameter.type.is_anonymous):
add_newline(lines)
lines.append(self._generate_anonymous_enum_conversion_for_parameter(domain, command.command_name, parameter))
for event in domain.events:
for parameter in event.event_parameters:
if (isinstance(parameter.type, EnumType) and parameter.type.is_anonymous):
add_newline(lines)
lines.append(self._generate_anonymous_enum_conversion_for_parameter(domain, event.event_name, parameter))
return '\n'.join(lines)
def _generate_anonymous_enum_conversion_for_declaration(self, domain, declaration):
objc_enum_name = ObjCGenerator.objc_enum_name_for_anonymous_enum_declaration(declaration)
enum_values = declaration.type.enum_values()
lines = []
lines.append(self._generate_enum_objc_to_protocol_string(objc_enum_name, enum_values))
lines.append(self._generate_enum_from_protocol_string(objc_enum_name, enum_values))
return '\n\n'.join(lines)
def _generate_anonymous_enum_conversion_for_member(self, domain, declaration, member):
objc_enum_name = ObjCGenerator.objc_enum_name_for_anonymous_enum_member(declaration, member)
enum_values = member.type.enum_values()
lines = []
lines.append(self._generate_enum_objc_to_protocol_string(objc_enum_name, enum_values))
lines.append(self._generate_enum_from_protocol_string(objc_enum_name, enum_values))
return '\n\n'.join(lines)
def _generate_anonymous_enum_conversion_for_parameter(self, domain, event_or_command_name, parameter):
objc_enum_name = ObjCGenerator.objc_enum_name_for_anonymous_enum_parameter(domain, event_or_command_name, parameter)
enum_values = parameter.type.enum_values()
lines = []
lines.append(self._generate_enum_objc_to_protocol_string(objc_enum_name, enum_values))
lines.append(self._generate_enum_from_protocol_string(objc_enum_name, enum_values))
return '\n\n'.join(lines)
def _generate_enum_objc_to_protocol_string(self, objc_enum_name, enum_values):
lines = []
lines.append('inline String toProtocolString(%s value)' % objc_enum_name)
lines.append('{')
lines.append(' switch(value) {')
for enum_value in enum_values:
lines.append(' case %s%s:' % (objc_enum_name, Generator.stylized_name_for_enum_value(enum_value)))
lines.append(' return ASCIILiteral("%s");' % enum_value)
lines.append(' }')
lines.append('}')
return '\n'.join(lines)
def _generate_enum_from_protocol_string(self, objc_enum_name, enum_values):
lines = []
lines.append('template<>')
lines.append('inline %s fromProtocolString(const String& value)' % objc_enum_name)
lines.append('{')
for enum_value in enum_values:
lines.append(' if (value == "%s")' % enum_value)
lines.append(' return %s%s;' % (objc_enum_name, Generator.stylized_name_for_enum_value(enum_value)))
lines.append(' ASSERT_NOT_REACHED();')
lines.append(' return %s%s;' % (objc_enum_name, Generator.stylized_name_for_enum_value(enum_values[0])))
lines.append('}')
return '\n'.join(lines)