There is no specific statute regarding treason other than "aiding and abetting enemies of the United States or other high crimes and misdemeanors."
Constitutionally, treason is decided by the House and Senate, with the House acting as the jury and the Senate as the attorneys.
The problem is that if you are basically arresting said house and senate, you must do so under the context of martial law, in which the trials for treason would be an empaneled jury determined by the military authority and prosecuted by the military officers.
So we are talking about military tribunal law, which basically boils down to: "here are the rules as we have determined them to be because we could just off you out by the latrine if we want, so we are giving you a trial because people don't like overly capricious systems."