Test for pro

1 files changed, 13 insertions, 4 deletions

diff --git a/mert/Data.cpp b/mert/Data.cpp
index e4bef6707..0acfbeac3 100644
--- a/mert/Data.cpp
+++ b/mert/Data.cpp
@@ -170,9 +170,15 @@ public:
 void Data::sampleRankedPairs( const std::string &rankedpairfile ) {
 	cout << "Sampling ranked pairs." << endl;
 
-	ofstream *outFile = new ofstream();
-	outFile->open( rankedpairfile.c_str() );
-	ostream *out = outFile;
+  ostream* out = NULL;
+  ofstream*  outFile = NULL;
+  if (rankedpairfile == "stdout") {
+    out = &cout;
+  } else {
+    outFile = new ofstream();
+    outFile->open( rankedpairfile.c_str() );
+    out = outFile;
+  }
 
 	const unsigned int n_samplings = 5000;
 	const unsigned int n_samples = 50;
@@ -225,7 +231,10 @@ void Data::sampleRankedPairs( const std::string &rankedpairfile ) {
 		//cerr << "collected " << collected << endl;
 	}
 	out->flush();
-	outFile->close();
+  if (outFile) {
+    outFile->close();
+    delete outFile;
+  }
 }
 
 void Data::outputSample( ostream &out, const FeatureStats &f1, const FeatureStats &f2 )