- check result of fscanf

git-svn-id: http://moon:8086/svn/software/trunk/projects/RBM@548 b431acfa-c32f-4a4a-93f1-934dc6c82436
2019-10-14 19:00:51 +00:00
parent 82e7286848
commit 77a3bca92d
1 changed files with 15 additions and 4 deletions
@@ -128,8 +128,16 @@ public:
 		if (!pFile)
 			return;

-		fscanf(pFile, "%d\n", &size);
-		fscanf(pFile, "%d\n", &numUnits);
+		int result = fscanf(pFile, "%d\n", &size);
+		if (result < 0)
+		{
+			return;
+		}
+		result = fscanf(pFile, "%d\n", &numUnits);
+		if (result < 0)
+		{
+			return;
+		}
 		m_data.resize(size, numUnits);
 		RowVectorXd data(numUnits);

@@ -139,9 +147,12 @@ public:
 			for (j=0; j < data.cols(); j++)
 			{
 				float v;
-				fscanf(pFile, "%f", &v);
+				int result = fscanf(pFile, "%f", &v);
+				if (result > 0)
+				{
 					data(j) = v;
 				}
+			}
 //			fscanf(pFile, "%d", &numUnits);
 			m_data.row(i) = data;
 			if (!(i%100) || (i == (size-1)))