Add support for switch case statement

houshengyuan · Aug 25, 2016 · 93ca860 · 93ca860
1 parent 82aae37
commit 93ca860
Show file tree

Hide file tree

Showing 2 changed files with 50 additions and 1 deletion.
diff --git a/sjtuthesis.cls b/sjtuthesis.cls
@@ -261,6 +261,34 @@
   \newtheorem{bcor}[thm]{\sjtu@label@cor~}
   \renewcommand{\proofname}{\bf\sjtu@label@proof}
 
+%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
+% The following definitions add Switch statement to LaTeX algorithmicx package
+% It's based on Werner's answer on stackoverflow
+% http://tex.stackexchange.com/questions/53357/switch-cases-in-algorithmic  
+
+% New definitions
+\algnewcommand\algorithmicswitch{\textbf{switch}}
+\algnewcommand\algorithmiccase{\textbf{case}}
+\algnewcommand\algorithmicdefault{\textbf{default}}
+
+% New "environments"
+% using \algtext*{<env>} removes any typesetting of that command
+\algdef{SE}[SWITCH]{Switch}{EndSwitch}[1]{\algorithmicswitch\ (#1)}{\algorithmicend\ \algorithmicswitch}%
+%\algtext*{EndSwitch}%
+\algdef{SE}[CASE]{Case}{EndCase}[1]{\algorithmiccase\ #1:}{\algorithmicend\ \algorithmiccase}%
+\algtext*{EndCase}%
+\algdef{SE}[DEFAULT]{Default}{EndDefault}{\algorithmicdefault\ :}{\algorithmicend\ \algorithmicdefault}%
+\algtext*{EndDefault}%
+%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
+
+%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
+% The following add some extra commands to LaTeX algorithmicx package
+
+% Assert
+\algnewcommand\algorithmicassert{\texttt{assert}}
+\algnewcommand\Assert[1]{\State \algorithmicassert(#1)}%
+%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
+
 %==========
 % Segment 4. Draw the sjtuthesis
 %==========

diff --git a/tex/example.tex b/tex/example.tex
@@ -471,4 +471,25 @@ \section{用algorithm和algorithmicx宏包插入算法描述}
 
 对于算法的索引要注意\verb+\caption+和\verb+\label+的位置, 
 必须是先\verb+\caption+再\verb+\label+\footnote{http://tex.stackexchange.com/questions/65993/algorithm-numbering}，
-否则会出现\verb+\ref{algo:sum_100}+生成的编号跟对应算法上显示不一致的问题。
+否则会出现\verb+\ref{algo:sum_100}+生成的编号跟对应算法上显示不一致的问题。
+
+根据Werner的回答\footnote{http://tex.stackexchange.com/questions/53357/switch-cases-in-algorithmic}
+增加了\verb+Switch+和\verb+Case+的支持，见算法\ref{algo:switch_example}。
+
+\begin{algorithm}
+\caption{Switch示例}
+\label{algo:switch_example}
+\begin{algorithmic}[1]
+  \Switch{$s$}
+    \Case{$a$}
+      \Assert{0}
+    \EndCase
+    \Case{$b$}
+      \Assert{1}
+    \EndCase
+    \Default
+      \Assert{2}
+    \EndDefault
+  \EndSwitch
+\end{algorithmic}
+\end{algorithm}