Как мне создать перечисляемые типы в MATLAB?
Есть ли перечисленные типы в MATLAB? Если нет, каковы альтернативы?
10 ответов
Вы можете получить некоторые функциональные возможности с классами MATLAB нового стиля:
classdef (Sealed) Colors
properties (Constant)
RED = 1;
GREEN = 2;
BLUE = 3;
end
methods (Access = private) % private so that you cant instantiate
function out = Colors
end
end
end
На самом деле это не тип, но, поскольку MATLAB свободно набран, если вы используете целые числа, вы можете делать вещи, которые приближают его:
line1 = Colors.RED;
...
if Colors.BLUE == line1
end
В этом случае "перечисления" MATLAB близки к перечислениям в стиле C- заменяют синтаксис для целых чисел.
С осторожным использованием статических методов вы можете даже сделать перечисления MATLAB похожими на Ada в изощренности, но, к сожалению, с неуклюжим синтаксисом.
Начиная с R2010b, MATLAB поддерживает перечисления.
Пример из документации:
classdef Colors
properties
R = 0;
G = 0;
B = 0;
end
methods
function c = Colors(r, g, b)
c.R = r; c.G = g; c.B = b;
end
end
enumeration
Red (1, 0, 0)
Green (0, 1, 0)
Blue (0, 0, 1)
end
end
Если вы хотите сделать что-то похожее на то, что предлагал Марк, вы можете просто создать структуру для представления ваших перечисляемых типов вместо целого нового класса:
colors = struct('RED', 1, 'GREEN', 2, 'BLUE', 3);
Одним из преимуществ является то, что вы можете легко получить доступ к структурам двумя различными способами. Вы можете указать поле напрямую, используя имя поля:
a = colors.RED;
или вы можете использовать динамические имена полей, если у вас есть имя поля в строке:
a = colors.('RED');
По правде говоря, есть несколько преимуществ для выполнения того, что предложил Марк, и создания целого нового класса для представления объекта "enum":
- Вы можете контролировать, как объект изменяется.
- Вы можете хранить определение в одном месте и легко использовать его в нескольких местах.
- Вы можете контролировать сбои и делать их более "изящными", например, возвращая пустую матрицу, если вы пытаетесь получить доступ к несуществующему полю (в отличие от выдачи ошибки).
Однако, если вам не нужна такая сложность и вам нужно просто что-то сделать быстро, структура, вероятно, является самой простой и понятной реализацией. Он также будет работать с более старыми версиями MATLAB, которые не используют новейшую структуру ООП.
На самом деле в MATLAB R2009b есть ключевое слово, называемое "перечисление". Это кажется недокументированным, и я не могу сказать, что знаю, как его использовать, но функциональность, вероятно, есть.
Вы можете найти это в matlabroot\toolbox\distcomp\examples\+examples
classdef(Enumeration) DmatFileMode < int32
enumeration
ReadMode(0)
ReadCompatibilityMode(1)
WriteMode(2)
end
<snip>
end
Вы также можете использовать Java enum классы из вашего кода Matlab. Определите их в Java и поместите их в javaclasspath вашего Matlab.
// Java class definition
package test;
public enum ColorEnum {
RED, GREEN, BLUE
}
Вы можете ссылаться на них по имени в М-коде.
mycolor = test.ColorEnum.RED
if mycolor == test.ColorEnum.RED
disp('got red');
else
disp('got other color');
end
% Use ordinal() to get a primitive you can use in a switch statement
switch mycolor.ordinal
case test.ColorEnum.BLUE.ordinal
disp('blue');
otherwise
disp(sprintf('other color: %s', char(mycolor.toString())))
end
Это не поймает сравнения с другими типами, все же. И сравнение со строкой имеет нечетный размер возврата.
>> test.ColorEnum.RED == 'GREEN'
ans =
0
>> test.ColorEnum.RED == 'RED'
ans =
1 1 1
Вы можете создать класс Matlab, который будет вести себя подобно старому типу безопасного перечисления в Java. Модификация решения Марка может принять его от typedefs в стиле C до более похожих на Java перечислений. В этой версии значения в константах являются типизированными объектами Color.
Плюсы:
- Тип можно проверить (во время выполнения) с помощью == и других операций, чтобы предотвратить случайное сравнение с необработанными числами или другими типами перечислений.
- Вы можете явно проверить тип ваших переменных (во время выполнения).
- Значения отображаются с читаемыми именами вместо непрозрачных кодов.
- Такие операции, как mean() и std(), которые не имеют смысла для перечислений, запрещены.
Недостатки:
- Более длинное определение класса. Но это все типовой пример, и его можно использовать повторно для любого другого класса enum, изменяя только имя класса и свойства Constant.
- Эти перечисления нельзя использовать непосредственно в блоках переключателей. Нужно выдать код, который теряет некоторую безопасность типов.
- Объекты будут медленнее, чем примитивы. Уместно, если вы используете константы внутри циклов.
В общем, я не знаю, какой подход лучше. Не использовался ни на практике.
classdef (Sealed) Color
%COLOR Example of Java-style typesafe enum for Matlab
properties (Constant)
RED = Color(1, 'RED');
GREEN = Color(2, 'GREEN');
BLUE = Color(3, 'BLUE');
end
properties (SetAccess=private)
% All these properties are immutable.
Code;
Name;
end
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
methods (Access = private)
%private so that you can't instatiate directly
function out = Color(InCode, InName)
out.Code = InCode;
out.Name = InName;
end
end
methods (Static = true)
function needa(obj)
%NEEDA Asserts that obj must be a Color
if ~isa(obj, mfilename)
error('Input must be a %s; got a %s', mfilename, class(obj));
end
end
end
methods (Access = public)
function display(obj)
disp([inputname(1) ' =']);
disp(obj);
end
function disp(obj)
if isscalar(obj)
disp(sprintf('%s: %s (%d)', class(obj), obj.Name, obj.Code));
else
disp(sprintf('%s array: size %s', class(obj), mat2str(size(obj))));
end
end
function out = eq(a, b)
%EQ Basic "type-safe" eq
check_type_safety(a, b);
out = [a.Code] == [b.Code];
end
function [tf,loc] = ismember(a, b)
check_type_safety(a, b);
[tf,loc] = ismember([a.Code], [b.Code]);
end
function check_type_safety(varargin)
%CHECK_TYPE_SAFETY Check that all inputs are of this enum type
for i = 1:nargin
if ~isa(varargin{i}, mfilename)
error('Non-typesafe comparison of %s vs. %s', mfilename, class(varargin{i}));
end
end
end
end
end
Вот функция для этого.
function do_stuff_with_color(c)
%DO_STUFF_WITH_COLOR Demo use of the Color typesafe enum
Color.needa(c); % Make sure input was a color
if (c == Color.BLUE)
disp('color was blue');
else
disp('color was not blue');
end
% To work with switch statements, you have to explicitly pop the code out
switch c.Code
case Color.BLUE.Code
disp('blue');
otherwise
disp(sprintf('some other color: %s', c.Name));
end
Пример использования:
>> Color.RED == Color.RED
ans =
1
>> Color.RED == 1
??? Error using ==> Color>Color.check_type_safety at 55
Non-typesafe comparison of Color vs. double
Error in ==> Color>Color.eq at 44
check_type_safety(a, b);
>> do_stuff_with_color(Color.BLUE)
color was blue
blue
>> do_stuff_with_color(Color.GREEN)
color was not blue
some other color: GREEN
>> do_stuff_with_color(1+1) % oops - passing the wrong type, should error
??? Error using ==> Color>Color.needa at 26
Input must be a Color; got a double
Error in ==> do_stuff_with_color at 4
Color.needa(c); % Make sure input was a color
>>
Незначительная изюминка в обоих подходах: соглашение C о размещении константы в левой части символа "==" для предотвращения неправильного присваивания здесь не сильно помогает. В Matlab, если вы случайно используете "=" с этой константой на LHS, вместо ошибки, он просто создаст новую локальную переменную структуры с именем Colors и замаскирует класс enum.
>> Colors.BLUE = 42
Colors =
BLUE: 42
>> Color.BLUE = 42
Color =
BLUE: 42
>> Color.RED
??? Reference to non-existent field 'RED'.
Если у вас есть доступ к панели инструментов статистики, вы можете рассмотреть возможность использования категориального объекта.
Попробовав другие предложения на этой странице, я наткнулся на полностью объектно-ориентированный подход Эндрю. Очень приятно - спасибо, Андрей.
Однако, если кому-то интересно, я внес (как мне кажется, некоторые) некоторые улучшения. В частности, я убрал необходимость двойного указания имени объекта enum. Имена теперь получены с использованием отражения и системы метаклассов. Кроме того, функции eq() и ismember() были переписаны, чтобы вернуть возвращаемые значения правильной формы для матриц объектов enum. И наконец, функция check_type_safety() была изменена, чтобы сделать ее совместимой с каталогами пакетов (например, пространствами имен).
Кажется, работает хорошо, но дайте мне знать, что вы думаете:
classdef (Sealed) Color
%COLOR Example of Java-style typesafe enum for Matlab
properties (Constant)
RED = Color(1);
GREEN = Color(2);
BLUE = Color(3);
end
methods (Access = private) % private so that you can''t instatiate directly
function out = Color(InCode)
out.Code = InCode;
end
end
% ============================================================================
% Everything from here down is completely boilerplate - no need to change anything.
% ============================================================================
properties (SetAccess=private) % All these properties are immutable.
Code;
end
properties (Dependent, SetAccess=private)
Name;
end
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
methods
function out = eq(a, b) %EQ Basic "type-safe" eq
check_type_safety(a, b);
out = reshape([a.Code],size(a)) == reshape([b.Code],size(b));
end
function [tf,loc] = ismember(a, b)
check_type_safety(a, b);
[tf,loc] = ismember(reshape([a.Code],size(a)), [b.Code]);
end
function check_type_safety(varargin) %CHECK_TYPE_SAFETY Check that all inputs are of this enum type
theClass = class(varargin{1});
for ii = 2:nargin
if ~isa(varargin{ii}, theClass)
error('Non-typesafe comparison of %s vs. %s', theClass, class(varargin{ii}));
end
end
end
% Display stuff:
function display(obj)
disp([inputname(1) ' =']);
disp(obj);
end
function disp(obj)
if isscalar(obj)
fprintf('%s: %s (%d)\n', class(obj), obj.Name, obj.Code);
else
fprintf('%s array: size %s\n', class(obj), mat2str(size(obj)));
end
end
function name=get.Name(obj)
mc=metaclass(obj);
mp=mc.Properties;
for ii=1:length(mp)
if (mp{ii}.Constant && isequal(obj.(mp{ii}.Name).Code,obj.Code))
name = mp{ii}.Name;
return;
end;
end;
error('Unable to find a %s value of %d',class(obj),obj.Code);
end;
end
end
Спасибо масон
Если вам нужны перечислимые типы только для передачи в сборку C# или.NET, вы можете создавать и передавать перечисления с помощью MATLAB 2010:
A = NET.addAssembly(MyName.dll)
% suppose you have enum called "MyAlerts" in your assembly
myvar = MyName.MyAlerts.('value_1');
Вы также можете проверить официальный ответ MathWorks на
Как использовать перечисляемые значения.NET в MATLAB 7.8 (R2009a)?
// the enum "MyAlerts" in c# will look something like this
public enum MyAlerts
{
value_1 = 0,
value_2 = 1,
MyAlerts_Count = 2,
}