Simple Moving Average (SMA)

/**

 * Calculates the Simple Moving Average (SMA) for a given dataset and period.

 *

 * @param {array} data - an array where the first column is the  dates, and subsequent columns for  open, high, low, close and volume.  

 * @param {number} period The number of periods to calculate the SMA, e.g., 14 for a 14-day SMA.

 * @returns {Array<Array>} The Simple Moving Average (SMA) of the range.

 * @customfunction

 */

  

function SMA(data, period) {

  

  //Check number of arguments

  if (arguments.length !== SMA.length) {

    throw new Error(`Wrong number of arguments. Expected ${SMA.length} arguments, but got ${arguments.length} arguments.`);

  }

  const processedData = getData(data);

  const dates = processedData.slice(1).map(row => row[0]); // Extract dates

  const closePrices = processedData[0].length === 2

    ? processedData.slice(1).map(row => row[1]) // For 2-column arrays, use the second column

    : processedData.slice(1).map(row => row[4]); // Default: use index 4 for close price

  

  // Check if the period is greater than the length of the data

  if (period > closePrices.length) {

    throw new Error("Error: The specified period exceeds the length of the data.");

  }

  

  // Initialize an array to store SMA values with corresponding dates

  let results = [["Date", `SMA(${period})`]]; // Include headers

  

  // Calculate SMA for the given period

  for (let i = period - 1; i < closePrices.length; i++) {

    const periodPrices = closePrices.slice(i - period + 1, i + 1); // Get the slice for the period

    const sma = periodPrices.reduce((acc, val) => acc + val, 0) / period; // Calculate average

  

    // Append the current date and SMA to the results array

    results.push([dates[i], sma]);

  }

  

  return results;

}

  
  

/**

 *

 * @param {Array<Array>} data - A 2D array of cell values to process. The first row should contain headers.

 * @returns {Array<Array>} - A 2D array where the first row contains headers and subsequent rows contain processed data.

 *

 * @customfunction

 *

 */

  

function getData(data) {

  if (!Array.isArray(data) || data.length === 0) {

    throw new Error("Input data must be a non-empty array.");

  }

  
  

  // Check if the first row contains headers

  const headers = Array.isArray(data[0]) && data[0].every(item => typeof item === "string")

    ? data[0]

    : null;

  

  if (!headers) {

    // If no headers, assume the first column is the date column and process data

    return data.map(row => row.map((value, index) => {

      return index === 0 ? new Date(value) : parseFloat(value);

    }));

  }

  

  // Process rows with headers

  return [headers, ...data.slice(1).map(row => headers.map((header, index) => {

    const key = header.toLowerCase();

    return key === "date" ? new Date(row[index]) : parseFloat(row[index]) || row[index];

  }))];

}