Read the column names with the reader if the user didn't manually set the columns.

Add necessary setter for columnNames in the reader.

Author: Hernán Morales Durand

src/DataFrame-IO/DataFrame.extension.st

@@ -37,6 +37,19 @@ DataFrame class >> readFromCsv: aFileReference [
 	^ self readFrom: aFileReference using: reader
 ]
 
+{ #category : #'*DataFrame-IO' }
+DataFrame class >> readFromCsv: aFileReference withColumnNames: anArrayOfColumnNames separator: aSeparator [
+
+	| dataFrame |
+
+	dataFrame := DataFrameCsvReader new
+		separator: aSeparator;
+		columnNames: anArrayOfColumnNames;
+		readFrom: aFileReference.
+	dataFrame calculateDataTypes.
+	^ dataFrame.
+]
+
 { #category : #'*DataFrame-IO' }
 DataFrame class >> readFromCsv: aFileReference withSeparator: aSeparator [
 	| reader |

src/DataFrame-IO/DataFrameCsvReader.class.st

@@ -42,6 +42,13 @@ Class {
 	#category : #'DataFrame-IO-Core'
 }
 
+{ #category : #reading }
+DataFrameCsvReader >> columnNames: aCollectionOfString [
+	"Set the receiver' s column names"
+	
+	columnNames := aCollectionOfString
+]
+
 { #category : #reading }
 DataFrameCsvReader >> createDataFrame [
 	| df |
@@ -91,7 +98,9 @@ DataFrameCsvReader >> initialize [
 
 { #category : #reading }
 DataFrameCsvReader >> readColumnNamesWith: aReader [
-	columnNames := aReader readHeader.
+	"Set the receiver's column names if they were not manually set"
+
+	columnNames ifNil: [ columnNames := aReader readHeader ].
 
 	self includeRowNames ifTrue: [
 		columnNames := columnNames copyWithoutFirst ]