diff mbox series

[v4] cmd: Add a pwm command

Message ID 20201222060006.25749-1-pragnesh.patel@sifive.com
State Accepted
Delegated to: Tom Rini
Headers show
Series [v4] cmd: Add a pwm command | expand

Commit Message

Pragnesh Patel Dec. 22, 2020, 6 a.m. UTC
Add the command "pwm" for controlling the pwm channels. This
command provides pwm invert/config/enable/disable functionalities
via PWM uclass drivers

Signed-off-by: Pragnesh Patel <pragnesh.patel@sifive.com>
Reviewed-by: Simon Glass <sjg@chromium.org>
---

Changes in v4:
- Add ut_assertok() for every run_command()

Changes in v3:
- Replace goto with return
- Print return value for error
- Change the assert condition for success

Changes in v2:
- Add test for pwm command

 README                    |   1 +
 cmd/Kconfig               |   6 ++
 cmd/Makefile              |   1 +
 cmd/pwm.c                 | 117 ++++++++++++++++++++++++++++++++++++++
 configs/sandbox_defconfig |   1 +
 test/cmd/Makefile         |   1 +
 test/cmd/pwm.c            |  47 +++++++++++++++
 7 files changed, 174 insertions(+)
 create mode 100644 cmd/pwm.c
 create mode 100644 test/cmd/pwm.c

Comments

Pragnesh Patel Jan. 17, 2021, 12:05 p.m. UTC | #1
Hi Tom,

Any comment on this ?

On Tue, Dec 22, 2020 at 11:30 AM Pragnesh Patel
<pragnesh.patel@sifive.com> wrote:
>
> Add the command "pwm" for controlling the pwm channels. This
> command provides pwm invert/config/enable/disable functionalities
> via PWM uclass drivers
>
> Signed-off-by: Pragnesh Patel <pragnesh.patel@sifive.com>
> Reviewed-by: Simon Glass <sjg@chromium.org>
> ---
>
> Changes in v4:
> - Add ut_assertok() for every run_command()
>
> Changes in v3:
> - Replace goto with return
> - Print return value for error
> - Change the assert condition for success
>
> Changes in v2:
> - Add test for pwm command
>
>  README                    |   1 +
>  cmd/Kconfig               |   6 ++
>  cmd/Makefile              |   1 +
>  cmd/pwm.c                 | 117 ++++++++++++++++++++++++++++++++++++++
>  configs/sandbox_defconfig |   1 +
>  test/cmd/Makefile         |   1 +
>  test/cmd/pwm.c            |  47 +++++++++++++++
>  7 files changed, 174 insertions(+)
>  create mode 100644 cmd/pwm.c
>  create mode 100644 test/cmd/pwm.c
>
> diff --git a/README b/README
> index cb49aa15da..dab291e0d0 100644
> --- a/README
> +++ b/README
> @@ -3160,6 +3160,7 @@ i2c       - I2C sub-system
>  sspi   - SPI utility commands
>  base   - print or set address offset
>  printenv- print environment variables
> +pwm    - control pwm channels
>  setenv - set environment variables
>  saveenv - save environment variables to persistent storage
>  protect - enable or disable FLASH write protection
> diff --git a/cmd/Kconfig b/cmd/Kconfig
> index 1595de999b..0d085108f4 100644
> --- a/cmd/Kconfig
> +++ b/cmd/Kconfig
> @@ -918,6 +918,12 @@ config CMD_GPIO
>         help
>           GPIO support.
>
> +config CMD_PWM
> +       bool "pwm"
> +       depends on DM_PWM
> +       help
> +         Control PWM channels, this allows invert/config/enable/disable PWM channels.
> +
>  config CMD_GPT
>         bool "GPT (GUID Partition Table) command"
>         select EFI_PARTITION
> diff --git a/cmd/Makefile b/cmd/Makefile
> index dd86675bf2..75df3c136c 100644
> --- a/cmd/Makefile
> +++ b/cmd/Makefile
> @@ -120,6 +120,7 @@ endif
>  obj-$(CONFIG_CMD_PINMUX) += pinmux.o
>  obj-$(CONFIG_CMD_PMC) += pmc.o
>  obj-$(CONFIG_CMD_PSTORE) += pstore.o
> +obj-$(CONFIG_CMD_PWM) += pwm.o
>  obj-$(CONFIG_CMD_PXE) += pxe.o pxe_utils.o
>  obj-$(CONFIG_CMD_WOL) += wol.o
>  obj-$(CONFIG_CMD_QFW) += qfw.o
> diff --git a/cmd/pwm.c b/cmd/pwm.c
> new file mode 100644
> index 0000000000..5849fc57b6
> --- /dev/null
> +++ b/cmd/pwm.c
> @@ -0,0 +1,117 @@
> +// SPDX-License-Identifier: GPL-2.0+
> +/*
> + * Control PWM channels
> + *
> + * Copyright (c) 2020 SiFive, Inc
> + * author: Pragnesh Patel <pragnesh.patel@sifive.com>
> + */
> +
> +#include <command.h>
> +#include <dm.h>
> +#include <pwm.h>
> +
> +enum pwm_cmd {
> +       PWM_SET_INVERT,
> +       PWM_SET_CONFIG,
> +       PWM_SET_ENABLE,
> +       PWM_SET_DISABLE,
> +};
> +
> +static int do_pwm(struct cmd_tbl *cmdtp, int flag, int argc,
> +                 char *const argv[])
> +{
> +       const char *str_cmd, *str_channel = NULL, *str_enable = NULL;
> +       const char *str_pwm = NULL, *str_period = NULL, *str_duty = NULL;
> +       enum pwm_cmd sub_cmd;
> +       struct udevice *dev;
> +       u32 channel, pwm_enable, pwm_dev, period_ns = 0, duty_ns = 0;
> +       int ret;
> +
> +       if (argc < 4)
> +               return CMD_RET_USAGE;
> +
> +       str_cmd = argv[1];
> +       argc -= 2;
> +       argv += 2;
> +
> +       if (argc > 0) {
> +               str_pwm = *argv;
> +               argc--;
> +               argv++;
> +       }
> +
> +       if (!str_pwm)
> +               return CMD_RET_USAGE;
> +
> +       switch (*str_cmd) {
> +       case 'i':
> +               sub_cmd = PWM_SET_INVERT;
> +               break;
> +       case 'c':
> +               sub_cmd = PWM_SET_CONFIG;
> +               break;
> +       case 'e':
> +               sub_cmd = PWM_SET_ENABLE;
> +               break;
> +       case 'd':
> +               sub_cmd = PWM_SET_DISABLE;
> +               break;
> +       default:
> +               return CMD_RET_USAGE;
> +       }
> +
> +       pwm_dev = simple_strtoul(str_pwm, NULL, 10);
> +       ret = uclass_get_device(UCLASS_PWM, pwm_dev, &dev);
> +       if (ret) {
> +               printf("pwm: '%s' not found\n", str_pwm);
> +               return cmd_process_error(cmdtp, ret);
> +       }
> +
> +       if (argc > 0) {
> +               str_channel = *argv;
> +               channel = simple_strtoul(str_channel, NULL, 10);
> +               argc--;
> +               argv++;
> +       } else {
> +               return CMD_RET_USAGE;
> +       }
> +
> +       if (sub_cmd == PWM_SET_INVERT && argc > 0) {
> +               str_enable = *argv;
> +               pwm_enable = simple_strtoul(str_enable, NULL, 10);
> +               ret = pwm_set_invert(dev, channel, pwm_enable);
> +       } else if (sub_cmd == PWM_SET_CONFIG && argc == 2) {
> +               str_period = *argv;
> +               argc--;
> +               argv++;
> +               period_ns = simple_strtoul(str_period, NULL, 10);
> +
> +               if (argc > 0) {
> +                       str_duty = *argv;
> +                       duty_ns = simple_strtoul(str_duty, NULL, 10);
> +               }
> +
> +               ret = pwm_set_config(dev, channel, period_ns, duty_ns);
> +       } else if (sub_cmd == PWM_SET_ENABLE) {
> +               ret = pwm_set_enable(dev, channel, 1);
> +       } else if (sub_cmd == PWM_SET_DISABLE) {
> +               ret = pwm_set_enable(dev, channel, 0);
> +       } else {
> +               printf("PWM arguments missing\n");
> +               return CMD_RET_FAILURE;
> +       }
> +
> +       if (ret) {
> +               printf("error(%d)\n", ret);
> +               return CMD_RET_FAILURE;
> +       }
> +
> +       return CMD_RET_SUCCESS;
> +}
> +
> +U_BOOT_CMD(pwm, 6, 0, do_pwm,
> +          "control pwm channels",
> +          "pwm <invert> <pwm_dev_num> <channel> <polarity>\n"
> +          "pwm <config> <pwm_dev_num> <channel> <period_ns> <duty_ns>\n"
> +          "pwm <enable/disable> <pwm_dev_num> <channel>\n"
> +          "Note: All input values are in decimal");
> diff --git a/configs/sandbox_defconfig b/configs/sandbox_defconfig
> index f2a767a4cd..7a16603461 100644
> --- a/configs/sandbox_defconfig
> +++ b/configs/sandbox_defconfig
> @@ -58,6 +58,7 @@ CONFIG_CMD_LSBLK=y
>  CONFIG_CMD_MUX=y
>  CONFIG_CMD_OSD=y
>  CONFIG_CMD_PCI=y
> +CONFIG_CMD_PWM=y
>  CONFIG_CMD_READ=y
>  CONFIG_CMD_REMOTEPROC=y
>  CONFIG_CMD_SPI=y
> diff --git a/test/cmd/Makefile b/test/cmd/Makefile
> index 859dcda239..dde98dd371 100644
> --- a/test/cmd/Makefile
> +++ b/test/cmd/Makefile
> @@ -4,3 +4,4 @@
>
>  obj-y += mem.o
>  obj-$(CONFIG_CMD_MEM_SEARCH) += mem_search.o
> +obj-$(CONFIG_CMD_PWM) += pwm.o
> diff --git a/test/cmd/pwm.c b/test/cmd/pwm.c
> new file mode 100644
> index 0000000000..5343af83fa
> --- /dev/null
> +++ b/test/cmd/pwm.c
> @@ -0,0 +1,47 @@
> +// SPDX-License-Identifier: GPL-2.0+
> +/*
> + * Test for pwm command
> + *
> + * Copyright 2020 SiFive, Inc
> + *
> + * Authors:
> + *   Pragnesh Patel <pragnesh.patel@sifive.com>
> + */
> +
> +#include <dm.h>
> +#include <dm/test.h>
> +#include <test/test.h>
> +#include <test/ut.h>
> +
> +/* Basic test of 'pwm' command */
> +static int dm_test_pwm_cmd(struct unit_test_state *uts)
> +{
> +       struct udevice *dev;
> +
> +       ut_assertok(uclass_get_device(UCLASS_PWM, 0, &dev));
> +       ut_assertnonnull(dev);
> +
> +       ut_assertok(console_record_reset_enable());
> +
> +       /* pwm <invert> <pwm_dev_num> <channel> <polarity> */
> +       ut_assertok(run_command("pwm invert 0 0 1", 0));
> +       ut_assert_console_end();
> +
> +       ut_assertok(run_command("pwm invert 0 0 0", 0));
> +       ut_assert_console_end();
> +
> +       /* pwm <config> <pwm_dev_num> <channel> <period_ns> <duty_ns> */
> +       ut_assertok(run_command("pwm config 0 0 10 50", 0));
> +       ut_assert_console_end();
> +
> +       /* pwm <enable/disable> <pwm_dev_num> <channel> */
> +       ut_assertok(run_command("pwm enable 0 0", 0));
> +       ut_assert_console_end();
> +
> +       ut_assertok(run_command("pwm disable 0 0", 0));
> +       ut_assert_console_end();
> +
> +       return 0;
> +}
> +
> +DM_TEST(dm_test_pwm_cmd, UT_TESTF_SCAN_PDATA | UT_TESTF_SCAN_FDT | UT_TESTF_CONSOLE_REC);
> --
> 2.17.1
>
Heinrich Schuchardt Jan. 17, 2021, 12:29 p.m. UTC | #2
On 12/22/20 7:00 AM, Pragnesh Patel wrote:
> Add the command "pwm" for controlling the pwm channels. This
> command provides pwm invert/config/enable/disable functionalities
> via PWM uclass drivers
>
> Signed-off-by: Pragnesh Patel <pragnesh.patel@sifive.com>
> Reviewed-by: Simon Glass <sjg@chromium.org>

Dear Pragnesh,

Users should be pointed to the existence of the command.

Could you, please, provide a further patch to add a documentation of the
pwm command under /doc/usage/. You can use /doc/usage/button.rst as
template.

Best regards

Heinrich

> ---
>
> Changes in v4:
> - Add ut_assertok() for every run_command()
>
> Changes in v3:
> - Replace goto with return
> - Print return value for error
> - Change the assert condition for success
>
> Changes in v2:
> - Add test for pwm command
>
>   README                    |   1 +
>   cmd/Kconfig               |   6 ++
>   cmd/Makefile              |   1 +
>   cmd/pwm.c                 | 117 ++++++++++++++++++++++++++++++++++++++
>   configs/sandbox_defconfig |   1 +
>   test/cmd/Makefile         |   1 +
>   test/cmd/pwm.c            |  47 +++++++++++++++
>   7 files changed, 174 insertions(+)
>   create mode 100644 cmd/pwm.c
>   create mode 100644 test/cmd/pwm.c
>
> diff --git a/README b/README
> index cb49aa15da..dab291e0d0 100644
> --- a/README
> +++ b/README
> @@ -3160,6 +3160,7 @@ i2c	- I2C sub-system
>   sspi	- SPI utility commands
>   base	- print or set address offset
>   printenv- print environment variables
> +pwm	- control pwm channels
>   setenv	- set environment variables
>   saveenv - save environment variables to persistent storage
>   protect - enable or disable FLASH write protection
> diff --git a/cmd/Kconfig b/cmd/Kconfig
> index 1595de999b..0d085108f4 100644
> --- a/cmd/Kconfig
> +++ b/cmd/Kconfig
> @@ -918,6 +918,12 @@ config CMD_GPIO
>   	help
>   	  GPIO support.
>
> +config CMD_PWM
> +	bool "pwm"
> +	depends on DM_PWM
> +	help
> +	  Control PWM channels, this allows invert/config/enable/disable PWM channels.
> +
>   config CMD_GPT
>   	bool "GPT (GUID Partition Table) command"
>   	select EFI_PARTITION
> diff --git a/cmd/Makefile b/cmd/Makefile
> index dd86675bf2..75df3c136c 100644
> --- a/cmd/Makefile
> +++ b/cmd/Makefile
> @@ -120,6 +120,7 @@ endif
>   obj-$(CONFIG_CMD_PINMUX) += pinmux.o
>   obj-$(CONFIG_CMD_PMC) += pmc.o
>   obj-$(CONFIG_CMD_PSTORE) += pstore.o
> +obj-$(CONFIG_CMD_PWM) += pwm.o
>   obj-$(CONFIG_CMD_PXE) += pxe.o pxe_utils.o
>   obj-$(CONFIG_CMD_WOL) += wol.o
>   obj-$(CONFIG_CMD_QFW) += qfw.o
> diff --git a/cmd/pwm.c b/cmd/pwm.c
> new file mode 100644
> index 0000000000..5849fc57b6
> --- /dev/null
> +++ b/cmd/pwm.c
> @@ -0,0 +1,117 @@
> +// SPDX-License-Identifier: GPL-2.0+
> +/*
> + * Control PWM channels
> + *
> + * Copyright (c) 2020 SiFive, Inc
> + * author: Pragnesh Patel <pragnesh.patel@sifive.com>
> + */
> +
> +#include <command.h>
> +#include <dm.h>
> +#include <pwm.h>
> +
> +enum pwm_cmd {
> +	PWM_SET_INVERT,
> +	PWM_SET_CONFIG,
> +	PWM_SET_ENABLE,
> +	PWM_SET_DISABLE,
> +};
> +
> +static int do_pwm(struct cmd_tbl *cmdtp, int flag, int argc,
> +		  char *const argv[])
> +{
> +	const char *str_cmd, *str_channel = NULL, *str_enable = NULL;
> +	const char *str_pwm = NULL, *str_period = NULL, *str_duty = NULL;
> +	enum pwm_cmd sub_cmd;
> +	struct udevice *dev;
> +	u32 channel, pwm_enable, pwm_dev, period_ns = 0, duty_ns = 0;
> +	int ret;
> +
> +	if (argc < 4)
> +		return CMD_RET_USAGE;
> +
> +	str_cmd = argv[1];
> +	argc -= 2;
> +	argv += 2;
> +
> +	if (argc > 0) {
> +		str_pwm = *argv;
> +		argc--;
> +		argv++;
> +	}
> +
> +	if (!str_pwm)
> +		return CMD_RET_USAGE;
> +
> +	switch (*str_cmd) {
> +	case 'i':
> +		sub_cmd = PWM_SET_INVERT;
> +		break;
> +	case 'c':
> +		sub_cmd = PWM_SET_CONFIG;
> +		break;
> +	case 'e':
> +		sub_cmd = PWM_SET_ENABLE;
> +		break;
> +	case 'd':
> +		sub_cmd = PWM_SET_DISABLE;
> +		break;
> +	default:
> +		return CMD_RET_USAGE;
> +	}
> +
> +	pwm_dev = simple_strtoul(str_pwm, NULL, 10);
> +	ret = uclass_get_device(UCLASS_PWM, pwm_dev, &dev);
> +	if (ret) {
> +		printf("pwm: '%s' not found\n", str_pwm);
> +		return cmd_process_error(cmdtp, ret);
> +	}
> +
> +	if (argc > 0) {
> +		str_channel = *argv;
> +		channel = simple_strtoul(str_channel, NULL, 10);
> +		argc--;
> +		argv++;
> +	} else {
> +		return CMD_RET_USAGE;
> +	}
> +
> +	if (sub_cmd == PWM_SET_INVERT && argc > 0) {
> +		str_enable = *argv;
> +		pwm_enable = simple_strtoul(str_enable, NULL, 10);
> +		ret = pwm_set_invert(dev, channel, pwm_enable);
> +	} else if (sub_cmd == PWM_SET_CONFIG && argc == 2) {
> +		str_period = *argv;
> +		argc--;
> +		argv++;
> +		period_ns = simple_strtoul(str_period, NULL, 10);
> +
> +		if (argc > 0) {
> +			str_duty = *argv;
> +			duty_ns = simple_strtoul(str_duty, NULL, 10);
> +		}
> +
> +		ret = pwm_set_config(dev, channel, period_ns, duty_ns);
> +	} else if (sub_cmd == PWM_SET_ENABLE) {
> +		ret = pwm_set_enable(dev, channel, 1);
> +	} else if (sub_cmd == PWM_SET_DISABLE) {
> +		ret = pwm_set_enable(dev, channel, 0);
> +	} else {
> +		printf("PWM arguments missing\n");
> +		return CMD_RET_FAILURE;
> +	}
> +
> +	if (ret) {
> +		printf("error(%d)\n", ret);
> +		return CMD_RET_FAILURE;
> +	}
> +
> +	return CMD_RET_SUCCESS;
> +}
> +
> +U_BOOT_CMD(pwm, 6, 0, do_pwm,
> +	   "control pwm channels",
> +	   "pwm <invert> <pwm_dev_num> <channel> <polarity>\n"
> +	   "pwm <config> <pwm_dev_num> <channel> <period_ns> <duty_ns>\n"
> +	   "pwm <enable/disable> <pwm_dev_num> <channel>\n"
> +	   "Note: All input values are in decimal");
> diff --git a/configs/sandbox_defconfig b/configs/sandbox_defconfig
> index f2a767a4cd..7a16603461 100644
> --- a/configs/sandbox_defconfig
> +++ b/configs/sandbox_defconfig
> @@ -58,6 +58,7 @@ CONFIG_CMD_LSBLK=y
>   CONFIG_CMD_MUX=y
>   CONFIG_CMD_OSD=y
>   CONFIG_CMD_PCI=y
> +CONFIG_CMD_PWM=y
>   CONFIG_CMD_READ=y
>   CONFIG_CMD_REMOTEPROC=y
>   CONFIG_CMD_SPI=y
> diff --git a/test/cmd/Makefile b/test/cmd/Makefile
> index 859dcda239..dde98dd371 100644
> --- a/test/cmd/Makefile
> +++ b/test/cmd/Makefile
> @@ -4,3 +4,4 @@
>
>   obj-y += mem.o
>   obj-$(CONFIG_CMD_MEM_SEARCH) += mem_search.o
> +obj-$(CONFIG_CMD_PWM) += pwm.o
> diff --git a/test/cmd/pwm.c b/test/cmd/pwm.c
> new file mode 100644
> index 0000000000..5343af83fa
> --- /dev/null
> +++ b/test/cmd/pwm.c
> @@ -0,0 +1,47 @@
> +// SPDX-License-Identifier: GPL-2.0+
> +/*
> + * Test for pwm command
> + *
> + * Copyright 2020 SiFive, Inc
> + *
> + * Authors:
> + *   Pragnesh Patel <pragnesh.patel@sifive.com>
> + */
> +
> +#include <dm.h>
> +#include <dm/test.h>
> +#include <test/test.h>
> +#include <test/ut.h>
> +
> +/* Basic test of 'pwm' command */
> +static int dm_test_pwm_cmd(struct unit_test_state *uts)
> +{
> +	struct udevice *dev;
> +
> +	ut_assertok(uclass_get_device(UCLASS_PWM, 0, &dev));
> +	ut_assertnonnull(dev);
> +
> +	ut_assertok(console_record_reset_enable());
> +
> +	/* pwm <invert> <pwm_dev_num> <channel> <polarity> */
> +	ut_assertok(run_command("pwm invert 0 0 1", 0));
> +	ut_assert_console_end();
> +
> +	ut_assertok(run_command("pwm invert 0 0 0", 0));
> +	ut_assert_console_end();
> +
> +	/* pwm <config> <pwm_dev_num> <channel> <period_ns> <duty_ns> */
> +	ut_assertok(run_command("pwm config 0 0 10 50", 0));
> +	ut_assert_console_end();
> +
> +	/* pwm <enable/disable> <pwm_dev_num> <channel> */
> +	ut_assertok(run_command("pwm enable 0 0", 0));
> +	ut_assert_console_end();
> +
> +	ut_assertok(run_command("pwm disable 0 0", 0));
> +	ut_assert_console_end();
> +
> +	return 0;
> +}
> +
> +DM_TEST(dm_test_pwm_cmd, UT_TESTF_SCAN_PDATA | UT_TESTF_SCAN_FDT | UT_TESTF_CONSOLE_REC);
>
Tom Rini Jan. 18, 2021, 4:22 p.m. UTC | #3
On Sun, Jan 17, 2021 at 05:35:17PM +0530, Pragnesh Patel wrote:
> Hi Tom,
> 
> Any comment on this ?

I'll likely be picking this up soon, thanks for your patience.
Tom Rini Jan. 19, 2021, 1:05 p.m. UTC | #4
On Tue, Dec 22, 2020 at 11:30:05AM +0530, Pragnesh Patel wrote:

> Add the command "pwm" for controlling the pwm channels. This
> command provides pwm invert/config/enable/disable functionalities
> via PWM uclass drivers
> 
> Signed-off-by: Pragnesh Patel <pragnesh.patel@sifive.com>
> Reviewed-by: Simon Glass <sjg@chromium.org>

Applied to u-boot/master, thanks!
diff mbox series

Patch

diff --git a/README b/README
index cb49aa15da..dab291e0d0 100644
--- a/README
+++ b/README
@@ -3160,6 +3160,7 @@  i2c	- I2C sub-system
 sspi	- SPI utility commands
 base	- print or set address offset
 printenv- print environment variables
+pwm	- control pwm channels
 setenv	- set environment variables
 saveenv - save environment variables to persistent storage
 protect - enable or disable FLASH write protection
diff --git a/cmd/Kconfig b/cmd/Kconfig
index 1595de999b..0d085108f4 100644
--- a/cmd/Kconfig
+++ b/cmd/Kconfig
@@ -918,6 +918,12 @@  config CMD_GPIO
 	help
 	  GPIO support.
 
+config CMD_PWM
+	bool "pwm"
+	depends on DM_PWM
+	help
+	  Control PWM channels, this allows invert/config/enable/disable PWM channels.
+
 config CMD_GPT
 	bool "GPT (GUID Partition Table) command"
 	select EFI_PARTITION
diff --git a/cmd/Makefile b/cmd/Makefile
index dd86675bf2..75df3c136c 100644
--- a/cmd/Makefile
+++ b/cmd/Makefile
@@ -120,6 +120,7 @@  endif
 obj-$(CONFIG_CMD_PINMUX) += pinmux.o
 obj-$(CONFIG_CMD_PMC) += pmc.o
 obj-$(CONFIG_CMD_PSTORE) += pstore.o
+obj-$(CONFIG_CMD_PWM) += pwm.o
 obj-$(CONFIG_CMD_PXE) += pxe.o pxe_utils.o
 obj-$(CONFIG_CMD_WOL) += wol.o
 obj-$(CONFIG_CMD_QFW) += qfw.o
diff --git a/cmd/pwm.c b/cmd/pwm.c
new file mode 100644
index 0000000000..5849fc57b6
--- /dev/null
+++ b/cmd/pwm.c
@@ -0,0 +1,117 @@ 
+// SPDX-License-Identifier: GPL-2.0+
+/*
+ * Control PWM channels
+ *
+ * Copyright (c) 2020 SiFive, Inc
+ * author: Pragnesh Patel <pragnesh.patel@sifive.com>
+ */
+
+#include <command.h>
+#include <dm.h>
+#include <pwm.h>
+
+enum pwm_cmd {
+	PWM_SET_INVERT,
+	PWM_SET_CONFIG,
+	PWM_SET_ENABLE,
+	PWM_SET_DISABLE,
+};
+
+static int do_pwm(struct cmd_tbl *cmdtp, int flag, int argc,
+		  char *const argv[])
+{
+	const char *str_cmd, *str_channel = NULL, *str_enable = NULL;
+	const char *str_pwm = NULL, *str_period = NULL, *str_duty = NULL;
+	enum pwm_cmd sub_cmd;
+	struct udevice *dev;
+	u32 channel, pwm_enable, pwm_dev, period_ns = 0, duty_ns = 0;
+	int ret;
+
+	if (argc < 4)
+		return CMD_RET_USAGE;
+
+	str_cmd = argv[1];
+	argc -= 2;
+	argv += 2;
+
+	if (argc > 0) {
+		str_pwm = *argv;
+		argc--;
+		argv++;
+	}
+
+	if (!str_pwm)
+		return CMD_RET_USAGE;
+
+	switch (*str_cmd) {
+	case 'i':
+		sub_cmd = PWM_SET_INVERT;
+		break;
+	case 'c':
+		sub_cmd = PWM_SET_CONFIG;
+		break;
+	case 'e':
+		sub_cmd = PWM_SET_ENABLE;
+		break;
+	case 'd':
+		sub_cmd = PWM_SET_DISABLE;
+		break;
+	default:
+		return CMD_RET_USAGE;
+	}
+
+	pwm_dev = simple_strtoul(str_pwm, NULL, 10);
+	ret = uclass_get_device(UCLASS_PWM, pwm_dev, &dev);
+	if (ret) {
+		printf("pwm: '%s' not found\n", str_pwm);
+		return cmd_process_error(cmdtp, ret);
+	}
+
+	if (argc > 0) {
+		str_channel = *argv;
+		channel = simple_strtoul(str_channel, NULL, 10);
+		argc--;
+		argv++;
+	} else {
+		return CMD_RET_USAGE;
+	}
+
+	if (sub_cmd == PWM_SET_INVERT && argc > 0) {
+		str_enable = *argv;
+		pwm_enable = simple_strtoul(str_enable, NULL, 10);
+		ret = pwm_set_invert(dev, channel, pwm_enable);
+	} else if (sub_cmd == PWM_SET_CONFIG && argc == 2) {
+		str_period = *argv;
+		argc--;
+		argv++;
+		period_ns = simple_strtoul(str_period, NULL, 10);
+
+		if (argc > 0) {
+			str_duty = *argv;
+			duty_ns = simple_strtoul(str_duty, NULL, 10);
+		}
+
+		ret = pwm_set_config(dev, channel, period_ns, duty_ns);
+	} else if (sub_cmd == PWM_SET_ENABLE) {
+		ret = pwm_set_enable(dev, channel, 1);
+	} else if (sub_cmd == PWM_SET_DISABLE) {
+		ret = pwm_set_enable(dev, channel, 0);
+	} else {
+		printf("PWM arguments missing\n");
+		return CMD_RET_FAILURE;
+	}
+
+	if (ret) {
+		printf("error(%d)\n", ret);
+		return CMD_RET_FAILURE;
+	}
+
+	return CMD_RET_SUCCESS;
+}
+
+U_BOOT_CMD(pwm, 6, 0, do_pwm,
+	   "control pwm channels",
+	   "pwm <invert> <pwm_dev_num> <channel> <polarity>\n"
+	   "pwm <config> <pwm_dev_num> <channel> <period_ns> <duty_ns>\n"
+	   "pwm <enable/disable> <pwm_dev_num> <channel>\n"
+	   "Note: All input values are in decimal");
diff --git a/configs/sandbox_defconfig b/configs/sandbox_defconfig
index f2a767a4cd..7a16603461 100644
--- a/configs/sandbox_defconfig
+++ b/configs/sandbox_defconfig
@@ -58,6 +58,7 @@  CONFIG_CMD_LSBLK=y
 CONFIG_CMD_MUX=y
 CONFIG_CMD_OSD=y
 CONFIG_CMD_PCI=y
+CONFIG_CMD_PWM=y
 CONFIG_CMD_READ=y
 CONFIG_CMD_REMOTEPROC=y
 CONFIG_CMD_SPI=y
diff --git a/test/cmd/Makefile b/test/cmd/Makefile
index 859dcda239..dde98dd371 100644
--- a/test/cmd/Makefile
+++ b/test/cmd/Makefile
@@ -4,3 +4,4 @@ 
 
 obj-y += mem.o
 obj-$(CONFIG_CMD_MEM_SEARCH) += mem_search.o
+obj-$(CONFIG_CMD_PWM) += pwm.o
diff --git a/test/cmd/pwm.c b/test/cmd/pwm.c
new file mode 100644
index 0000000000..5343af83fa
--- /dev/null
+++ b/test/cmd/pwm.c
@@ -0,0 +1,47 @@ 
+// SPDX-License-Identifier: GPL-2.0+
+/*
+ * Test for pwm command
+ *
+ * Copyright 2020 SiFive, Inc
+ *
+ * Authors:
+ *   Pragnesh Patel <pragnesh.patel@sifive.com>
+ */
+
+#include <dm.h>
+#include <dm/test.h>
+#include <test/test.h>
+#include <test/ut.h>
+
+/* Basic test of 'pwm' command */
+static int dm_test_pwm_cmd(struct unit_test_state *uts)
+{
+	struct udevice *dev;
+
+	ut_assertok(uclass_get_device(UCLASS_PWM, 0, &dev));
+	ut_assertnonnull(dev);
+
+	ut_assertok(console_record_reset_enable());
+
+	/* pwm <invert> <pwm_dev_num> <channel> <polarity> */
+	ut_assertok(run_command("pwm invert 0 0 1", 0));
+	ut_assert_console_end();
+
+	ut_assertok(run_command("pwm invert 0 0 0", 0));
+	ut_assert_console_end();
+
+	/* pwm <config> <pwm_dev_num> <channel> <period_ns> <duty_ns> */
+	ut_assertok(run_command("pwm config 0 0 10 50", 0));
+	ut_assert_console_end();
+
+	/* pwm <enable/disable> <pwm_dev_num> <channel> */
+	ut_assertok(run_command("pwm enable 0 0", 0));
+	ut_assert_console_end();
+
+	ut_assertok(run_command("pwm disable 0 0", 0));
+	ut_assert_console_end();
+
+	return 0;
+}
+
+DM_TEST(dm_test_pwm_cmd, UT_TESTF_SCAN_PDATA | UT_TESTF_SCAN_FDT | UT_TESTF_CONSOLE_REC);