[libstd] getopts, now with fewer copies

Change the opt_ functions in getopts to take a reference to a
Matches, instead of taking a Matches by-value, as suggested in
This commit is contained in:
Tim Chevalier 2012-11-24 12:49:31 -08:00
parent e454a47bf2
commit f74fe894fc
10 changed files with 68 additions and 53 deletions

View File

@ -35,12 +35,12 @@ fn parse_config(args: ~[~str]) -> config {
assert (vec::is_not_empty(args)); assert (vec::is_not_empty(args));
let args_ = vec::tail(args); let args_ = vec::tail(args);
let matches = let matches =
match getopts::getopts(args_, opts) { &match getopts::getopts(args_, opts) {
Ok(m) => m, Ok(m) => m,
Err(f) => fail getopts::fail_str(f) Err(f) => fail getopts::fail_str(f)
}; };
fn opt_path(m: getopts::Matches, nm: ~str) -> Path { fn opt_path(m: &getopts::Matches, nm: ~str) -> Path {
Path(getopts::opt_str(m, nm)) Path(getopts::opt_str(m, nm))
} }

View File

@ -661,7 +661,7 @@ fn load_source_packages(c: &Cargo, src: @Source) {
} }
fn build_cargo_options(argv: ~[~str]) -> Options { fn build_cargo_options(argv: ~[~str]) -> Options {
let matches = match getopts::getopts(argv, opts()) { let matches = &match getopts::getopts(argv, opts()) {
result::Ok(m) => m, result::Ok(m) => m,
result::Err(f) => { result::Err(f) => {
fail fmt!("%s", getopts::fail_str(f)); fail fmt!("%s", getopts::fail_str(f));

View File

@ -458,7 +458,7 @@ fn host_triple() -> ~str {
} }
fn build_session_options(binary: ~str, fn build_session_options(binary: ~str,
matches: getopts::Matches, matches: &getopts::Matches,
demitter: diagnostic::emitter) -> @session::options { demitter: diagnostic::emitter) -> @session::options {
let crate_type = if opt_present(matches, ~"lib") { let crate_type = if opt_present(matches, ~"lib") {
session::lib_crate session::lib_crate
@ -807,7 +807,7 @@ mod test {
#[test] #[test]
fn test_switch_implies_cfg_test() { fn test_switch_implies_cfg_test() {
let matches = let matches =
match getopts(~[~"--test"], optgroups()) { &match getopts(~[~"--test"], optgroups()) {
Ok(m) => m, Ok(m) => m,
Err(f) => fail ~"test_switch_implies_cfg_test: " + Err(f) => fail ~"test_switch_implies_cfg_test: " +
getopts::fail_str(f) getopts::fail_str(f)
@ -824,7 +824,7 @@ mod test {
#[test] #[test]
fn test_switch_implies_cfg_test_unless_cfg_test() { fn test_switch_implies_cfg_test_unless_cfg_test() {
let matches = let matches =
match getopts(~[~"--test", ~"--cfg=test"], optgroups()) { &match getopts(~[~"--test", ~"--cfg=test"], optgroups()) {
Ok(m) => m, Ok(m) => m,
Err(f) => { Err(f) => {
fail ~"test_switch_implies_cfg_test_unless_cfg_test: " + fail ~"test_switch_implies_cfg_test_unless_cfg_test: " +

View File

@ -34,7 +34,7 @@ struct RH {
} }
fn setup_env(test_name: &str, source_string: &str) -> Env { fn setup_env(test_name: &str, source_string: &str) -> Env {
let matches = getopts(~[~"-Z", ~"verbose"], optgroups()).get(); let matches = &getopts(~[~"-Z", ~"verbose"], optgroups()).get();
let sessopts = build_session_options(~"rustc", matches, diagnostic::emit); let sessopts = build_session_options(~"rustc", matches, diagnostic::emit);
let sess = build_session(sessopts, diagnostic::emit); let sess = build_session(sessopts, diagnostic::emit);
let cfg = build_configuration(sess, ~"whatever", str_input(~"")); let cfg = build_configuration(sess, ~"whatever", str_input(~""));

View File

@ -85,7 +85,7 @@ fn run_compiler(args: &~[~str], demitter: diagnostic::emitter) {
if args.is_empty() { usage(binary); return; } if args.is_empty() { usage(binary); return; }
let matches = let matches =
match getopts::groups::getopts(args, optgroups()) { &match getopts::groups::getopts(args, optgroups()) {
Ok(m) => m, Ok(m) => m,
Err(f) => { Err(f) => {
early_error(demitter, getopts::fail_str(f)) early_error(demitter, getopts::fail_str(f))

View File

@ -124,14 +124,14 @@ fn parse_config_(
args: &[~str], args: &[~str],
+program_output: ProgramOutput +program_output: ProgramOutput
) -> Result<Config, ~str> { ) -> Result<Config, ~str> {
let args = vec::tail(args); let args = args.tail();
let opts = vec::unzip(opts()).first(); let opts = vec::unzip(opts()).first();
match getopts::getopts(args, opts) { match getopts::getopts(args, opts) {
result::Ok(matches) => { result::Ok(matches) => {
if vec::len(matches.free) == 1u { if matches.free.len() == 1 {
let input_crate = Path(vec::head(matches.free)); let input_crate = Path(vec::head(matches.free));
config_from_opts(&input_crate, matches, move program_output) config_from_opts(&input_crate, &matches, move program_output)
} else if vec::is_empty(matches.free) { } else if matches.free.is_empty() {
result::Err(~"no crates specified") result::Err(~"no crates specified")
} else { } else {
result::Err(~"multiple crates specified") result::Err(~"multiple crates specified")
@ -145,7 +145,7 @@ fn parse_config_(
fn config_from_opts( fn config_from_opts(
input_crate: &Path, input_crate: &Path,
+matches: getopts::Matches, matches: &getopts::Matches,
+program_output: ProgramOutput +program_output: ProgramOutput
) -> Result<Config, ~str> { ) -> Result<Config, ~str> {

View File

@ -518,7 +518,7 @@ pub fn getopts(args: &[~str], opts: &[Opt]) -> Result unsafe {
free: free}); free: free});
} }
fn opt_vals(mm: Matches, nm: &str) -> ~[Optval] { fn opt_vals(mm: &Matches, nm: &str) -> ~[Optval] {
return match find_opt(mm.opts, mkname(nm)) { return match find_opt(mm.opts, mkname(nm)) {
Some(id) => mm.vals[id], Some(id) => mm.vals[id],
None => { None => {
@ -528,27 +528,27 @@ fn opt_vals(mm: Matches, nm: &str) -> ~[Optval] {
}; };
} }
fn opt_val(mm: Matches, nm: &str) -> Optval { return opt_vals(mm, nm)[0]; } fn opt_val(mm: &Matches, nm: &str) -> Optval { return opt_vals(mm, nm)[0]; }
/// Returns true if an option was matched /// Returns true if an option was matched
pub fn opt_present(mm: Matches, nm: &str) -> bool { pub fn opt_present(mm: &Matches, nm: &str) -> bool {
return vec::len::<Optval>(opt_vals(mm, nm)) > 0u; opt_vals(mm, nm).is_not_empty()
} }
/// Returns the number of times an option was matched /// Returns the number of times an option was matched
pub fn opt_count(mm: Matches, nm: &str) -> uint { pub fn opt_count(mm: &Matches, nm: &str) -> uint {
return vec::len::<Optval>(opt_vals(mm, nm)); opt_vals(mm, nm).len()
} }
/// Returns true if any of several options were matched /// Returns true if any of several options were matched
pub fn opts_present(mm: Matches, names: &[~str]) -> bool { pub fn opts_present(mm: &Matches, names: &[~str]) -> bool {
for vec::each(names) |nm| { for vec::each(names) |nm| {
match find_opt(mm.opts, mkname(*nm)) { match find_opt(mm.opts, mkname(*nm)) {
Some(_) => return true, Some(_) => return true,
None => () None => ()
} }
} }
return false; false
} }
@ -558,7 +558,7 @@ pub fn opts_present(mm: Matches, names: &[~str]) -> bool {
* Fails if the option was not matched or if the match did not take an * Fails if the option was not matched or if the match did not take an
* argument * argument
*/ */
pub fn opt_str(mm: Matches, nm: &str) -> ~str { pub fn opt_str(mm: &Matches, nm: &str) -> ~str {
return match opt_val(mm, nm) { Val(copy s) => s, _ => fail }; return match opt_val(mm, nm) { Val(copy s) => s, _ => fail };
} }
@ -568,7 +568,7 @@ pub fn opt_str(mm: Matches, nm: &str) -> ~str {
* Fails if the no option was provided from the given list, or if the no such * Fails if the no option was provided from the given list, or if the no such
* option took an argument * option took an argument
*/ */
pub fn opts_str(mm: Matches, names: &[~str]) -> ~str { pub fn opts_str(mm: &Matches, names: &[~str]) -> ~str {
for vec::each(names) |nm| { for vec::each(names) |nm| {
match opt_val(mm, *nm) { match opt_val(mm, *nm) {
Val(copy s) => return s, Val(copy s) => return s,
@ -585,7 +585,7 @@ pub fn opts_str(mm: Matches, names: &[~str]) -> ~str {
* *
* Used when an option accepts multiple values. * Used when an option accepts multiple values.
*/ */
pub fn opt_strs(mm: Matches, nm: &str) -> ~[~str] { pub fn opt_strs(mm: &Matches, nm: &str) -> ~[~str] {
let mut acc: ~[~str] = ~[]; let mut acc: ~[~str] = ~[];
for vec::each(opt_vals(mm, nm)) |v| { for vec::each(opt_vals(mm, nm)) |v| {
match *v { Val(copy s) => acc.push(s), _ => () } match *v { Val(copy s) => acc.push(s), _ => () }
@ -594,7 +594,7 @@ pub fn opt_strs(mm: Matches, nm: &str) -> ~[~str] {
} }
/// Returns the string argument supplied to a matching option or none /// Returns the string argument supplied to a matching option or none
pub fn opt_maybe_str(mm: Matches, nm: &str) -> Option<~str> { pub fn opt_maybe_str(mm: &Matches, nm: &str) -> Option<~str> {
let vals = opt_vals(mm, nm); let vals = opt_vals(mm, nm);
if vec::len::<Optval>(vals) == 0u { return None::<~str>; } if vec::len::<Optval>(vals) == 0u { return None::<~str>; }
return match vals[0] { return match vals[0] {
@ -611,7 +611,7 @@ pub fn opt_maybe_str(mm: Matches, nm: &str) -> Option<~str> {
* present but no argument was provided, and the argument if the option was * present but no argument was provided, and the argument if the option was
* present and an argument was provided. * present and an argument was provided.
*/ */
pub fn opt_default(mm: Matches, nm: &str, def: &str) -> Option<~str> { pub fn opt_default(mm: &Matches, nm: &str, def: &str) -> Option<~str> {
let vals = opt_vals(mm, nm); let vals = opt_vals(mm, nm);
if vec::len::<Optval>(vals) == 0u { return None::<~str>; } if vec::len::<Optval>(vals) == 0u { return None::<~str>; }
return match vals[0] { Val(copy s) => Some::<~str>(s), return match vals[0] { Val(copy s) => Some::<~str>(s),
@ -870,7 +870,7 @@ mod tests {
let opts = ~[reqopt(~"test")]; let opts = ~[reqopt(~"test")];
let rs = getopts(args, opts); let rs = getopts(args, opts);
match rs { match rs {
Ok(copy m) => { Ok(ref m) => {
assert (opt_present(m, ~"test")); assert (opt_present(m, ~"test"));
assert (opt_str(m, ~"test") == ~"20"); assert (opt_str(m, ~"test") == ~"20");
} }
@ -917,7 +917,7 @@ mod tests {
let opts = ~[reqopt(~"t")]; let opts = ~[reqopt(~"t")];
let rs = getopts(args, opts); let rs = getopts(args, opts);
match rs { match rs {
Ok(copy m) => { Ok(ref m) => {
assert (opt_present(m, ~"t")); assert (opt_present(m, ~"t"));
assert (opt_str(m, ~"t") == ~"20"); assert (opt_str(m, ~"t") == ~"20");
} }
@ -966,7 +966,7 @@ mod tests {
let opts = ~[optopt(~"test")]; let opts = ~[optopt(~"test")];
let rs = getopts(args, opts); let rs = getopts(args, opts);
match rs { match rs {
Ok(copy m) => { Ok(ref m) => {
assert (opt_present(m, ~"test")); assert (opt_present(m, ~"test"));
assert (opt_str(m, ~"test") == ~"20"); assert (opt_str(m, ~"test") == ~"20");
} }
@ -980,7 +980,7 @@ mod tests {
let opts = ~[optopt(~"test")]; let opts = ~[optopt(~"test")];
let rs = getopts(args, opts); let rs = getopts(args, opts);
match rs { match rs {
Ok(copy m) => assert (!opt_present(m, ~"test")), Ok(ref m) => assert (!opt_present(m, ~"test")),
_ => fail _ => fail
} }
} }
@ -1013,7 +1013,7 @@ mod tests {
let opts = ~[optopt(~"t")]; let opts = ~[optopt(~"t")];
let rs = getopts(args, opts); let rs = getopts(args, opts);
match rs { match rs {
Ok(copy m) => { Ok(ref m) => {
assert (opt_present(m, ~"t")); assert (opt_present(m, ~"t"));
assert (opt_str(m, ~"t") == ~"20"); assert (opt_str(m, ~"t") == ~"20");
} }
@ -1027,7 +1027,7 @@ mod tests {
let opts = ~[optopt(~"t")]; let opts = ~[optopt(~"t")];
let rs = getopts(args, opts); let rs = getopts(args, opts);
match rs { match rs {
Ok(copy m) => assert (!opt_present(m, ~"t")), Ok(ref m) => assert (!opt_present(m, ~"t")),
_ => fail _ => fail
} }
} }
@ -1062,7 +1062,7 @@ mod tests {
let opts = ~[optflag(~"test")]; let opts = ~[optflag(~"test")];
let rs = getopts(args, opts); let rs = getopts(args, opts);
match rs { match rs {
Ok(copy m) => assert (opt_present(m, ~"test")), Ok(ref m) => assert (opt_present(m, ~"test")),
_ => fail _ => fail
} }
} }
@ -1073,7 +1073,7 @@ mod tests {
let opts = ~[optflag(~"test")]; let opts = ~[optflag(~"test")];
let rs = getopts(args, opts); let rs = getopts(args, opts);
match rs { match rs {
Ok(copy m) => assert (!opt_present(m, ~"test")), Ok(ref m) => assert (!opt_present(m, ~"test")),
_ => fail _ => fail
} }
} }
@ -1109,7 +1109,7 @@ mod tests {
let opts = ~[optflag(~"t")]; let opts = ~[optflag(~"t")];
let rs = getopts(args, opts); let rs = getopts(args, opts);
match rs { match rs {
Ok(copy m) => assert (opt_present(m, ~"t")), Ok(ref m) => assert (opt_present(m, ~"t")),
_ => fail _ => fail
} }
} }
@ -1120,7 +1120,7 @@ mod tests {
let opts = ~[optflag(~"t")]; let opts = ~[optflag(~"t")];
let rs = getopts(args, opts); let rs = getopts(args, opts);
match rs { match rs {
Ok(copy m) => assert (!opt_present(m, ~"t")), Ok(ref m) => assert (!opt_present(m, ~"t")),
_ => fail _ => fail
} }
} }
@ -1158,7 +1158,7 @@ mod tests {
let opts = ~[optflagmulti(~"v")]; let opts = ~[optflagmulti(~"v")];
let rs = getopts(args, opts); let rs = getopts(args, opts);
match rs { match rs {
Ok(copy m) => { Ok(ref m) => {
assert (opt_count(m, ~"v") == 1); assert (opt_count(m, ~"v") == 1);
} }
_ => fail _ => fail
@ -1171,7 +1171,7 @@ mod tests {
let opts = ~[optflagmulti(~"v")]; let opts = ~[optflagmulti(~"v")];
let rs = getopts(args, opts); let rs = getopts(args, opts);
match rs { match rs {
Ok(copy m) => { Ok(ref m) => {
assert (opt_count(m, ~"v") == 2); assert (opt_count(m, ~"v") == 2);
} }
_ => fail _ => fail
@ -1184,7 +1184,7 @@ mod tests {
let opts = ~[optflagmulti(~"v")]; let opts = ~[optflagmulti(~"v")];
let rs = getopts(args, opts); let rs = getopts(args, opts);
match rs { match rs {
Ok(copy m) => { Ok(ref m) => {
assert (opt_count(m, ~"v") == 2); assert (opt_count(m, ~"v") == 2);
} }
_ => fail _ => fail
@ -1197,7 +1197,7 @@ mod tests {
let opts = ~[optflagmulti(~"verbose")]; let opts = ~[optflagmulti(~"verbose")];
let rs = getopts(args, opts); let rs = getopts(args, opts);
match rs { match rs {
Ok(copy m) => { Ok(ref m) => {
assert (opt_count(m, ~"verbose") == 1); assert (opt_count(m, ~"verbose") == 1);
} }
_ => fail _ => fail
@ -1210,7 +1210,7 @@ mod tests {
let opts = ~[optflagmulti(~"verbose")]; let opts = ~[optflagmulti(~"verbose")];
let rs = getopts(args, opts); let rs = getopts(args, opts);
match rs { match rs {
Ok(copy m) => { Ok(ref m) => {
assert (opt_count(m, ~"verbose") == 2); assert (opt_count(m, ~"verbose") == 2);
} }
_ => fail _ => fail
@ -1224,7 +1224,7 @@ mod tests {
let opts = ~[optmulti(~"test")]; let opts = ~[optmulti(~"test")];
let rs = getopts(args, opts); let rs = getopts(args, opts);
match rs { match rs {
Ok(copy m) => { Ok(ref m) => {
assert (opt_present(m, ~"test")); assert (opt_present(m, ~"test"));
assert (opt_str(m, ~"test") == ~"20"); assert (opt_str(m, ~"test") == ~"20");
} }
@ -1238,7 +1238,7 @@ mod tests {
let opts = ~[optmulti(~"test")]; let opts = ~[optmulti(~"test")];
let rs = getopts(args, opts); let rs = getopts(args, opts);
match rs { match rs {
Ok(copy m) => assert (!opt_present(m, ~"test")), Ok(ref m) => assert (!opt_present(m, ~"test")),
_ => fail _ => fail
} }
} }
@ -1260,7 +1260,7 @@ mod tests {
let opts = ~[optmulti(~"test")]; let opts = ~[optmulti(~"test")];
let rs = getopts(args, opts); let rs = getopts(args, opts);
match rs { match rs {
Ok(copy m) => { Ok(ref m) => {
assert (opt_present(m, ~"test")); assert (opt_present(m, ~"test"));
assert (opt_str(m, ~"test") == ~"20"); assert (opt_str(m, ~"test") == ~"20");
let pair = opt_strs(m, ~"test"); let pair = opt_strs(m, ~"test");
@ -1277,7 +1277,7 @@ mod tests {
let opts = ~[optmulti(~"t")]; let opts = ~[optmulti(~"t")];
let rs = getopts(args, opts); let rs = getopts(args, opts);
match rs { match rs {
Ok(copy m) => { Ok(ref m) => {
assert (opt_present(m, ~"t")); assert (opt_present(m, ~"t"));
assert (opt_str(m, ~"t") == ~"20"); assert (opt_str(m, ~"t") == ~"20");
} }
@ -1291,7 +1291,7 @@ mod tests {
let opts = ~[optmulti(~"t")]; let opts = ~[optmulti(~"t")];
let rs = getopts(args, opts); let rs = getopts(args, opts);
match rs { match rs {
Ok(copy m) => assert (!opt_present(m, ~"t")), Ok(ref m) => assert (!opt_present(m, ~"t")),
_ => fail _ => fail
} }
} }
@ -1313,7 +1313,7 @@ mod tests {
let opts = ~[optmulti(~"t")]; let opts = ~[optmulti(~"t")];
let rs = getopts(args, opts); let rs = getopts(args, opts);
match rs { match rs {
Ok(copy m) => { Ok(ref m) => {
assert (opt_present(m, ~"t")); assert (opt_present(m, ~"t"));
assert (opt_str(m, ~"t") == ~"20"); assert (opt_str(m, ~"t") == ~"20");
let pair = opt_strs(m, ~"t"); let pair = opt_strs(m, ~"t");
@ -1358,7 +1358,7 @@ mod tests {
optopt(~"notpresent")]; optopt(~"notpresent")];
let rs = getopts(args, opts); let rs = getopts(args, opts);
match rs { match rs {
Ok(copy m) => { Ok(ref m) => {
assert (m.free[0] == ~"prog"); assert (m.free[0] == ~"prog");
assert (m.free[1] == ~"free1"); assert (m.free[1] == ~"free1");
assert (opt_str(m, ~"s") == ~"20"); assert (opt_str(m, ~"s") == ~"20");
@ -1382,7 +1382,7 @@ mod tests {
fn test_multi() { fn test_multi() {
let args = ~[~"-e", ~"foo", ~"--encrypt", ~"foo"]; let args = ~[~"-e", ~"foo", ~"--encrypt", ~"foo"];
let opts = ~[optopt(~"e"), optopt(~"encrypt")]; let opts = ~[optopt(~"e"), optopt(~"encrypt")];
let matches = match getopts(args, opts) { let matches = &match getopts(args, opts) {
result::Ok(move m) => m, result::Ok(move m) => m,
result::Err(_) => fail result::Err(_) => fail
}; };
@ -1403,7 +1403,7 @@ mod tests {
fn test_nospace() { fn test_nospace() {
let args = ~[~"-Lfoo"]; let args = ~[~"-Lfoo"];
let opts = ~[optmulti(~"L")]; let opts = ~[optmulti(~"L")];
let matches = match getopts(args, opts) { let matches = &match getopts(args, opts) {
result::Ok(move m) => m, result::Ok(move m) => m,
result::Err(_) => fail result::Err(_) => fail
}; };

View File

@ -73,8 +73,8 @@ fn parse_opts(args: &[~str]) -> OptRes {
option::Some(matches.free[0]) option::Some(matches.free[0])
} else { option::None }; } else { option::None };
let run_ignored = getopts::opt_present(matches, ~"ignored"); let run_ignored = getopts::opt_present(&matches, ~"ignored");
let logfile = getopts::opt_maybe_str(matches, ~"logfile"); let logfile = getopts::opt_maybe_str(&matches, ~"logfile");
let test_opts = {filter: filter, run_ignored: run_ignored, let test_opts = {filter: filter, run_ignored: run_ignored,
logfile: logfile}; logfile: logfile};

View File

@ -53,7 +53,7 @@ fn parse_opts(argv: ~[~str]) -> config {
let opt_args = vec::slice(argv, 1u, vec::len(argv)); let opt_args = vec::slice(argv, 1u, vec::len(argv));
match getopts::getopts(opt_args, opts) { match getopts::getopts(opt_args, opts) {
Ok(m) => { return {stress: getopts::opt_present(m, ~"stress")} } Ok(ref m) => { return {stress: getopts::opt_present(m, ~"stress")} }
Err(_) => { fail; } Err(_) => { fail; }
} }
} }

View File

@ -0,0 +1,15 @@
extern mod std;
use std::getopts::*;
fn main() {
let args = ~[];
let opts = ~[optopt(~"b")];
match getopts(args, opts) {
result::Ok(ref m) =>
assert !opt_present(m, "b"),
result::Err(f) => fail fail_str(f)
};
}