ortools.sat.cp_model_pb2

498  def init(self, **kwargs):
499    self._cached_byte_size = 0
500    self._cached_byte_size_dirty = len(kwargs) > 0
501    self._fields = {}
502    # Contains a mapping from oneof field descriptors to the descriptor
503    # of the currently set field in that oneof field.
504    self._oneofs = {}
505
506    # _unknown_fields is () when empty for efficiency, and will be turned into
507    # a list if fields are added.
508    self._unknown_fields = ()
509    self._is_present_in_parent = False
510    self._listener = message_listener_mod.NullMessageListener()
511    self._listener_for_children = _Listener(self)
512    for field_name, field_value in kwargs.items():
513      field = _GetFieldByName(message_descriptor, field_name)
514      if field is None:
515        raise TypeError('%s() got an unexpected keyword argument "%s"' %
516                        (message_descriptor.name, field_name))
517      if field_value is None:
518        # field=None is the same as no field at all.
519        continue
520      if field.label == _FieldDescriptor.LABEL_REPEATED:
521        field_copy = field._default_constructor(self)
522        if field.cpp_type == _FieldDescriptor.CPPTYPE_MESSAGE:  # Composite
523          if _IsMapField(field):
524            if _IsMessageMapField(field):
525              for key in field_value:
526                field_copy[key].MergeFrom(field_value[key])
527            else:
528              field_copy.update(field_value)
529          else:
530            for val in field_value:
531              if isinstance(val, dict):
532                field_copy.add(**val)
533              else:
534                field_copy.add().MergeFrom(val)
535        else:  # Scalar
536          if field.cpp_type == _FieldDescriptor.CPPTYPE_ENUM:
537            field_value = [_GetIntegerEnumValue(field.enum_type, val)
538                           for val in field_value]
539          field_copy.extend(field_value)
540        self._fields[field] = field_copy
541      elif field.cpp_type == _FieldDescriptor.CPPTYPE_MESSAGE:
542        field_copy = field._default_constructor(self)
543        new_val = None
544        if isinstance(field_value, message_mod.Message):
545          new_val = field_value
546        elif isinstance(field_value, dict):
547          if field.message_type.full_name == _StructFullTypeName:
548            field_copy.Clear()
549            if len(field_value) == 1 and 'fields' in field_value:
550              try:
551                field_copy.update(field_value)
552              except:
553                # Fall back to init normal message field
554                field_copy.Clear()
555                new_val = field.message_type._concrete_class(**field_value)
556            else:
557              field_copy.update(field_value)
558          else:
559            new_val = field.message_type._concrete_class(**field_value)
560        elif hasattr(field_copy, '_internal_assign'):
561          field_copy._internal_assign(field_value)
562        else:
563          raise TypeError(
564              'Message field {0}.{1} must be initialized with a '
565              'dict or instance of same class, got {2}.'.format(
566                  message_descriptor.name,
567                  field_name,
568                  type(field_value).__name__,
569              )
570          )
571
572        if new_val != None:
573          try:
574            field_copy.MergeFrom(new_val)
575          except TypeError:
576            _ReraiseTypeErrorWithFieldName(message_descriptor.name, field_name)
577        self._fields[field] = field_copy
578      else:
579        if field.cpp_type == _FieldDescriptor.CPPTYPE_ENUM:
580          field_value = _GetIntegerEnumValue(field.enum_type, field_value)
581        try:
582          setattr(self, field_name, field_value)
583        except TypeError:
584          _ReraiseTypeErrorWithFieldName(message_descriptor.name, field_name)

1364  def MergeFrom(self, msg):
1365    if not isinstance(msg, cls):
1366      raise TypeError(
1367          'Parameter to MergeFrom() must be instance of same class: '
1368          'expected %s got %s.' % (_FullyQualifiedClassName(cls),
1369                                   _FullyQualifiedClassName(msg.__class__)))
1370
1371    assert msg is not self
1372    self._Modified()
1373
1374    fields = self._fields
1375
1376    for field, value in msg._fields.items():
1377      if field.label == LABEL_REPEATED:
1378        field_value = fields.get(field)
1379        if field_value is None:
1380          # Construct a new object to represent this field.
1381          field_value = field._default_constructor(self)
1382          fields[field] = field_value
1383        field_value.MergeFrom(value)
1384      elif field.cpp_type == CPPTYPE_MESSAGE:
1385        if value._is_present_in_parent:
1386          field_value = fields.get(field)
1387          if field_value is None:
1388            # Construct a new object to represent this field.
1389            field_value = field._default_constructor(self)
1390            fields[field] = field_value
1391          field_value.MergeFrom(value)
1392      else:
1393        self._fields[field] = value
1394        if field.containing_oneof:
1395          self._UpdateOneofState(field)
1396
1397    if msg._unknown_fields:
1398      if not self._unknown_fields:
1399        self._unknown_fields = []
1400      self._unknown_fields.extend(msg._unknown_fields)

1423def _Clear(self):
1424  # Clear fields.
1425  self._fields = {}
1426  self._unknown_fields = ()
1427
1428  self._oneofs = {}
1429  self._Modified()

1489  def Modified(self):
1490    """Sets the _cached_byte_size_dirty bit to true,
1491    and propagates this to our listener iff this was a state change.
1492    """
1493
1494    # Note:  Some callers check _cached_byte_size_dirty before calling
1495    #   _Modified() as an extra optimization.  So, if this method is ever
1496    #   changed such that it does stuff even when _cached_byte_size_dirty is
1497    #   already true, the callers need to be updated.
1498    if not self._cached_byte_size_dirty:
1499      self._cached_byte_size_dirty = True
1500      self._listener_for_children.dirty = True
1501      self._is_present_in_parent = True
1502      self._listener.Modified()

1265  def IsInitialized(self, errors=None):
1266    """Checks if all required fields of a message are set.
1267
1268    Args:
1269      errors:  A list which, if provided, will be populated with the field
1270               paths of all missing required fields.
1271
1272    Returns:
1273      True iff the specified message has all required fields set.
1274    """
1275
1276    # Performance is critical so we avoid HasField() and ListFields().
1277
1278    for field in required_fields:
1279      if (field not in self._fields or
1280          (field.cpp_type == _FieldDescriptor.CPPTYPE_MESSAGE and
1281           not self._fields[field]._is_present_in_parent)):
1282        if errors is not None:
1283          errors.extend(self.FindInitializationErrors())
1284        return False
1285
1286    for field, value in list(self._fields.items()):  # dict can change size!
1287      if field.cpp_type == _FieldDescriptor.CPPTYPE_MESSAGE:
1288        if field.label == _FieldDescriptor.LABEL_REPEATED:
1289          if (field.message_type._is_map_entry):
1290            continue
1291          for element in value:
1292            if not element.IsInitialized():
1293              if errors is not None:
1294                errors.extend(self.FindInitializationErrors())
1295              return False
1296        elif value._is_present_in_parent and not value.IsInitialized():
1297          if errors is not None:
1298            errors.extend(self.FindInitializationErrors())
1299          return False
1300
1301    return True

1180  def MergeFromString(self, serialized):
1181    serialized = memoryview(serialized)
1182    length = len(serialized)
1183    try:
1184      if self._InternalParse(serialized, 0, length) != length:
1185        # The only reason _InternalParse would return early is if it
1186        # encountered an end-group tag.
1187        raise message_mod.DecodeError('Unexpected end-group tag.')
1188    except (IndexError, TypeError):
1189      # Now ord(buf[p:p+1]) == ord('') gets TypeError.
1190      raise message_mod.DecodeError('Truncated message.')
1191    except struct.error as e:
1192      raise message_mod.DecodeError(e)
1193    return length   # Return this for legacy reasons.

1133  def SerializeToString(self, **kwargs):
1134    # Check if the message has all of its required fields set.
1135    if not self.IsInitialized():
1136      raise message_mod.EncodeError(
1137          'Message %s is missing required fields: %s' % (
1138          self.DESCRIPTOR.full_name, ','.join(self.FindInitializationErrors())))
1139    return self.SerializePartialToString(**kwargs)

1146  def SerializePartialToString(self, **kwargs):
1147    out = BytesIO()
1148    self._InternalSerialize(out.write, **kwargs)
1149    return out.getvalue()

848  def ListFields(self):
849    all_fields = [item for item in self._fields.items() if _IsPresent(item)]
850    all_fields.sort(key = lambda item: item[0].number)
851    return all_fields

872  def HasField(self, field_name):
873    try:
874      field = hassable_fields[field_name]
875    except KeyError as exc:
876      raise ValueError('Protocol message %s has no non-repeated field "%s" '
877                       'nor has presence is not available for this field.' % (
878                           message_descriptor.full_name, field_name)) from exc
879
880    if isinstance(field, descriptor_mod.OneofDescriptor):
881      try:
882        return HasField(self, self._oneofs[field].name)
883      except KeyError:
884        return False
885    else:
886      if field.cpp_type == _FieldDescriptor.CPPTYPE_MESSAGE:
887        value = self._fields.get(field)
888        return value is not None and value._is_present_in_parent
889      else:
890        return field in self._fields

897  def ClearField(self, field_name):
898    try:
899      field = message_descriptor.fields_by_name[field_name]
900    except KeyError:
901      try:
902        field = message_descriptor.oneofs_by_name[field_name]
903        if field in self._oneofs:
904          field = self._oneofs[field]
905        else:
906          return
907      except KeyError:
908        raise ValueError('Protocol message %s has no "%s" field.' %
909                         (message_descriptor.name, field_name))
910
911    if field in self._fields:
912      # To match the C++ implementation, we need to invalidate iterators
913      # for map fields when ClearField() happens.
914      if hasattr(self._fields[field], 'InvalidateIterators'):
915        self._fields[field].InvalidateIterators()
916
917      # Note:  If the field is a sub-message, its listener will still point
918      #   at us.  That's fine, because the worst than can happen is that it
919      #   will call _Modified() and invalidate our byte size.  Big deal.
920      del self._fields[field]
921
922      if self._oneofs.get(field.containing_oneof, None) is field:
923        del self._oneofs[field.containing_oneof]
924
925    # Always call _Modified() -- even if nothing was changed, this is
926    # a mutating method, and thus calling it should cause the field to become
927    # present in the parent message.
928    self._Modified()

1406  def WhichOneof(self, oneof_name):
1407    """Returns the name of the currently set field inside a oneof, or None."""
1408    try:
1409      field = message_descriptor.oneofs_by_name[oneof_name]
1410    except KeyError:
1411      raise ValueError(
1412          'Protocol message has no oneof "%s" field.' % oneof_name)
1413
1414    nested_field = self._oneofs.get(field, None)
1415    if nested_field is not None and self.HasField(nested_field.name):
1416      return nested_field.name
1417    else:
1418      return None

1438def _DiscardUnknownFields(self):
1439  self._unknown_fields = []
1440  for field, value in self.ListFields():
1441    if field.cpp_type == _FieldDescriptor.CPPTYPE_MESSAGE:
1442      if _IsMapField(field):
1443        if _IsMessageMapField(field):
1444          for key in value:
1445            value[key].DiscardUnknownFields()
1446      elif field.label == _FieldDescriptor.LABEL_REPEATED:
1447        for sub_message in value:
1448          sub_message.DiscardUnknownFields()
1449      else:
1450        value.DiscardUnknownFields()

1101  def ByteSize(self):
1102    if not self._cached_byte_size_dirty:
1103      return self._cached_byte_size
1104
1105    size = 0
1106    descriptor = self.DESCRIPTOR
1107    if descriptor._is_map_entry:
1108      # Fields of map entry should always be serialized.
1109      key_field = descriptor.fields_by_name['key']
1110      _MaybeAddEncoder(cls, key_field)
1111      size = key_field._sizer(self.key)
1112      value_field = descriptor.fields_by_name['value']
1113      _MaybeAddEncoder(cls, value_field)
1114      size += value_field._sizer(self.value)
1115    else:
1116      for field_descriptor, field_value in self.ListFields():
1117        _MaybeAddEncoder(cls, field_descriptor)
1118        size += field_descriptor._sizer(field_value)
1119      for tag_bytes, value_bytes in self._unknown_fields:
1120        size += len(tag_bytes) + len(value_bytes)
1121
1122    self._cached_byte_size = size
1123    self._cached_byte_size_dirty = False
1124    self._listener_for_children.dirty = False
1125    return size

826  def FromString(s):
827    message = cls()
828    message.MergeFromString(s)
829    return message

711  def getter(self):
712    # TODO: This may be broken since there may not be
713    # default_value.  Combine with has_default_value somehow.
714    return self._fields.get(field, default_value)

668  def getter(self):
669    field_value = self._fields.get(field)
670    if field_value is None:
671      # Construct a new object to represent this field.
672      field_value = field._default_constructor(self)
673
674      # Atomically check if another thread has preempted us and, if not, swap
675      # in the new object we just created.  If someone has preempted us, we
676      # take that object and discard ours.
677      # WARNING:  We are relying on setdefault() being atomic.  This is true
678      #   in CPython but we haven't investigated others.  This warning appears
679      #   in several other locations in this file.
680      field_value = self._fields.setdefault(field, field_value)
681    return field_value

1305  def FindInitializationErrors(self):
1306    """Finds required fields which are not initialized.
1307
1308    Returns:
1309      A list of strings.  Each string is a path to an uninitialized field from
1310      the top-level message, e.g. "foo.bar[5].baz".
1311    """
1312
1313    errors = []  # simplify things
1314
1315    for field in required_fields:
1316      if not self.HasField(field.name):
1317        errors.append(field.name)
1318
1319    for field, value in self.ListFields():
1320      if field.cpp_type == _FieldDescriptor.CPPTYPE_MESSAGE:
1321        if field.is_extension:
1322          name = '(%s)' % field.full_name
1323        else:
1324          name = field.name
1325
1326        if _IsMapField(field):
1327          if _IsMessageMapField(field):
1328            for key in value:
1329              element = value[key]
1330              prefix = '%s[%s].' % (name, key)
1331              sub_errors = element.FindInitializationErrors()
1332              errors += [prefix + error for error in sub_errors]
1333          else:
1334            # ScalarMaps can't have any initialization errors.
1335            pass
1336        elif field.label == _FieldDescriptor.LABEL_REPEATED:
1337          for i in range(len(value)):
1338            element = value[i]
1339            prefix = '%s[%d].' % (name, i)
1340            sub_errors = element.FindInitializationErrors()
1341            errors += [prefix + error for error in sub_errors]
1342        else:
1343          prefix = name + '.'
1344          sub_errors = value.FindInitializationErrors()
1345          errors += [prefix + error for error in sub_errors]
1346
1347    return errors

498  def init(self, **kwargs):
499    self._cached_byte_size = 0
500    self._cached_byte_size_dirty = len(kwargs) > 0
501    self._fields = {}
502    # Contains a mapping from oneof field descriptors to the descriptor
503    # of the currently set field in that oneof field.
504    self._oneofs = {}
505
506    # _unknown_fields is () when empty for efficiency, and will be turned into
507    # a list if fields are added.
508    self._unknown_fields = ()
509    self._is_present_in_parent = False
510    self._listener = message_listener_mod.NullMessageListener()
511    self._listener_for_children = _Listener(self)
512    for field_name, field_value in kwargs.items():
513      field = _GetFieldByName(message_descriptor, field_name)
514      if field is None:
515        raise TypeError('%s() got an unexpected keyword argument "%s"' %
516                        (message_descriptor.name, field_name))
517      if field_value is None:
518        # field=None is the same as no field at all.
519        continue
520      if field.label == _FieldDescriptor.LABEL_REPEATED:
521        field_copy = field._default_constructor(self)
522        if field.cpp_type == _FieldDescriptor.CPPTYPE_MESSAGE:  # Composite
523          if _IsMapField(field):
524            if _IsMessageMapField(field):
525              for key in field_value:
526                field_copy[key].MergeFrom(field_value[key])
527            else:
528              field_copy.update(field_value)
529          else:
530            for val in field_value:
531              if isinstance(val, dict):
532                field_copy.add(**val)
533              else:
534                field_copy.add().MergeFrom(val)
535        else:  # Scalar
536          if field.cpp_type == _FieldDescriptor.CPPTYPE_ENUM:
537            field_value = [_GetIntegerEnumValue(field.enum_type, val)
538                           for val in field_value]
539          field_copy.extend(field_value)
540        self._fields[field] = field_copy
541      elif field.cpp_type == _FieldDescriptor.CPPTYPE_MESSAGE:
542        field_copy = field._default_constructor(self)
543        new_val = None
544        if isinstance(field_value, message_mod.Message):
545          new_val = field_value
546        elif isinstance(field_value, dict):
547          if field.message_type.full_name == _StructFullTypeName:
548            field_copy.Clear()
549            if len(field_value) == 1 and 'fields' in field_value:
550              try:
551                field_copy.update(field_value)
552              except:
553                # Fall back to init normal message field
554                field_copy.Clear()
555                new_val = field.message_type._concrete_class(**field_value)
556            else:
557              field_copy.update(field_value)
558          else:
559            new_val = field.message_type._concrete_class(**field_value)
560        elif hasattr(field_copy, '_internal_assign'):
561          field_copy._internal_assign(field_value)
562        else:
563          raise TypeError(
564              'Message field {0}.{1} must be initialized with a '
565              'dict or instance of same class, got {2}.'.format(
566                  message_descriptor.name,
567                  field_name,
568                  type(field_value).__name__,
569              )
570          )
571
572        if new_val != None:
573          try:
574            field_copy.MergeFrom(new_val)
575          except TypeError:
576            _ReraiseTypeErrorWithFieldName(message_descriptor.name, field_name)
577        self._fields[field] = field_copy
578      else:
579        if field.cpp_type == _FieldDescriptor.CPPTYPE_ENUM:
580          field_value = _GetIntegerEnumValue(field.enum_type, field_value)
581        try:
582          setattr(self, field_name, field_value)
583        except TypeError:
584          _ReraiseTypeErrorWithFieldName(message_descriptor.name, field_name)

1364  def MergeFrom(self, msg):
1365    if not isinstance(msg, cls):
1366      raise TypeError(
1367          'Parameter to MergeFrom() must be instance of same class: '
1368          'expected %s got %s.' % (_FullyQualifiedClassName(cls),
1369                                   _FullyQualifiedClassName(msg.__class__)))
1370
1371    assert msg is not self
1372    self._Modified()
1373
1374    fields = self._fields
1375
1376    for field, value in msg._fields.items():
1377      if field.label == LABEL_REPEATED:
1378        field_value = fields.get(field)
1379        if field_value is None:
1380          # Construct a new object to represent this field.
1381          field_value = field._default_constructor(self)
1382          fields[field] = field_value
1383        field_value.MergeFrom(value)
1384      elif field.cpp_type == CPPTYPE_MESSAGE:
1385        if value._is_present_in_parent:
1386          field_value = fields.get(field)
1387          if field_value is None:
1388            # Construct a new object to represent this field.
1389            field_value = field._default_constructor(self)
1390            fields[field] = field_value
1391          field_value.MergeFrom(value)
1392      else:
1393        self._fields[field] = value
1394        if field.containing_oneof:
1395          self._UpdateOneofState(field)
1396
1397    if msg._unknown_fields:
1398      if not self._unknown_fields:
1399        self._unknown_fields = []
1400      self._unknown_fields.extend(msg._unknown_fields)

1423def _Clear(self):
1424  # Clear fields.
1425  self._fields = {}
1426  self._unknown_fields = ()
1427
1428  self._oneofs = {}
1429  self._Modified()

1489  def Modified(self):
1490    """Sets the _cached_byte_size_dirty bit to true,
1491    and propagates this to our listener iff this was a state change.
1492    """
1493
1494    # Note:  Some callers check _cached_byte_size_dirty before calling
1495    #   _Modified() as an extra optimization.  So, if this method is ever
1496    #   changed such that it does stuff even when _cached_byte_size_dirty is
1497    #   already true, the callers need to be updated.
1498    if not self._cached_byte_size_dirty:
1499      self._cached_byte_size_dirty = True
1500      self._listener_for_children.dirty = True
1501      self._is_present_in_parent = True
1502      self._listener.Modified()

1265  def IsInitialized(self, errors=None):
1266    """Checks if all required fields of a message are set.
1267
1268    Args:
1269      errors:  A list which, if provided, will be populated with the field
1270               paths of all missing required fields.
1271
1272    Returns:
1273      True iff the specified message has all required fields set.
1274    """
1275
1276    # Performance is critical so we avoid HasField() and ListFields().
1277
1278    for field in required_fields:
1279      if (field not in self._fields or
1280          (field.cpp_type == _FieldDescriptor.CPPTYPE_MESSAGE and
1281           not self._fields[field]._is_present_in_parent)):
1282        if errors is not None:
1283          errors.extend(self.FindInitializationErrors())
1284        return False
1285
1286    for field, value in list(self._fields.items()):  # dict can change size!
1287      if field.cpp_type == _FieldDescriptor.CPPTYPE_MESSAGE:
1288        if field.label == _FieldDescriptor.LABEL_REPEATED:
1289          if (field.message_type._is_map_entry):
1290            continue
1291          for element in value:
1292            if not element.IsInitialized():
1293              if errors is not None:
1294                errors.extend(self.FindInitializationErrors())
1295              return False
1296        elif value._is_present_in_parent and not value.IsInitialized():
1297          if errors is not None:
1298            errors.extend(self.FindInitializationErrors())
1299          return False
1300
1301    return True

1180  def MergeFromString(self, serialized):
1181    serialized = memoryview(serialized)
1182    length = len(serialized)
1183    try:
1184      if self._InternalParse(serialized, 0, length) != length:
1185        # The only reason _InternalParse would return early is if it
1186        # encountered an end-group tag.
1187        raise message_mod.DecodeError('Unexpected end-group tag.')
1188    except (IndexError, TypeError):
1189      # Now ord(buf[p:p+1]) == ord('') gets TypeError.
1190      raise message_mod.DecodeError('Truncated message.')
1191    except struct.error as e:
1192      raise message_mod.DecodeError(e)
1193    return length   # Return this for legacy reasons.

1133  def SerializeToString(self, **kwargs):
1134    # Check if the message has all of its required fields set.
1135    if not self.IsInitialized():
1136      raise message_mod.EncodeError(
1137          'Message %s is missing required fields: %s' % (
1138          self.DESCRIPTOR.full_name, ','.join(self.FindInitializationErrors())))
1139    return self.SerializePartialToString(**kwargs)

1146  def SerializePartialToString(self, **kwargs):
1147    out = BytesIO()
1148    self._InternalSerialize(out.write, **kwargs)
1149    return out.getvalue()

848  def ListFields(self):
849    all_fields = [item for item in self._fields.items() if _IsPresent(item)]
850    all_fields.sort(key = lambda item: item[0].number)
851    return all_fields

872  def HasField(self, field_name):
873    try:
874      field = hassable_fields[field_name]
875    except KeyError as exc:
876      raise ValueError('Protocol message %s has no non-repeated field "%s" '
877                       'nor has presence is not available for this field.' % (
878                           message_descriptor.full_name, field_name)) from exc
879
880    if isinstance(field, descriptor_mod.OneofDescriptor):
881      try:
882        return HasField(self, self._oneofs[field].name)
883      except KeyError:
884        return False
885    else:
886      if field.cpp_type == _FieldDescriptor.CPPTYPE_MESSAGE:
887        value = self._fields.get(field)
888        return value is not None and value._is_present_in_parent
889      else:
890        return field in self._fields

897  def ClearField(self, field_name):
898    try:
899      field = message_descriptor.fields_by_name[field_name]
900    except KeyError:
901      try:
902        field = message_descriptor.oneofs_by_name[field_name]
903        if field in self._oneofs:
904          field = self._oneofs[field]
905        else:
906          return
907      except KeyError:
908        raise ValueError('Protocol message %s has no "%s" field.' %
909                         (message_descriptor.name, field_name))
910
911    if field in self._fields:
912      # To match the C++ implementation, we need to invalidate iterators
913      # for map fields when ClearField() happens.
914      if hasattr(self._fields[field], 'InvalidateIterators'):
915        self._fields[field].InvalidateIterators()
916
917      # Note:  If the field is a sub-message, its listener will still point
918      #   at us.  That's fine, because the worst than can happen is that it
919      #   will call _Modified() and invalidate our byte size.  Big deal.
920      del self._fields[field]
921
922      if self._oneofs.get(field.containing_oneof, None) is field:
923        del self._oneofs[field.containing_oneof]
924
925    # Always call _Modified() -- even if nothing was changed, this is
926    # a mutating method, and thus calling it should cause the field to become
927    # present in the parent message.
928    self._Modified()