Format .jl files (Runic)

Author: mtfishman

Project.toml

@@ -1,7 +1,7 @@
 name = "QuantumOperatorDefinitions"
 uuid = "826dd319-6fd5-459a-a990-3a4f214664bf"
+version = "0.2.8"
 authors = ["ITensor developers <[email protected]> and contributors"]
-version = "0.2.7"
 
 [deps]
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
@@ -14,8 +14,8 @@ ITensorBase = "4795dd04-0d67-49bb-8f44-b89c448a1dc7"
 NamedDimsArrays = "60cbd0c0-df58-4cb7-918c-6f5607b73fde"
 
 [extensions]
-QuantumOperatorDefinitionsITensorBaseExt = ["ITensorBase", "GradedArrays", "NamedDimsArrays"]
 QuantumOperatorDefinitionsGradedArraysExt = ["BlockArrays", "GradedArrays"]
+QuantumOperatorDefinitionsITensorBaseExt = ["ITensorBase", "GradedArrays", "NamedDimsArrays"]
 
 [compat]
 BlockArrays = "1.3.0"

docs/make.jl

@@ -2,28 +2,28 @@ using QuantumOperatorDefinitions: QuantumOperatorDefinitions
 using Documenter: Documenter, DocMeta, deploydocs, makedocs
 
 DocMeta.setdocmeta!(
-  QuantumOperatorDefinitions,
-  :DocTestSetup,
-  :(using QuantumOperatorDefinitions);
-  recursive=true,
+    QuantumOperatorDefinitions,
+    :DocTestSetup,
+    :(using QuantumOperatorDefinitions);
+    recursive = true,
 )
 
 include("make_index.jl")
 
 makedocs(;
-  modules=[QuantumOperatorDefinitions],
-  authors="ITensor developers <[email protected]> and contributors",
-  sitename="QuantumOperatorDefinitions.jl",
-  format=Documenter.HTML(;
-    canonical="https://itensor.github.io/QuantumOperatorDefinitions.jl",
-    edit_link="main",
-    assets=["assets/favicon.ico", "assets/extras.css"],
-  ),
-  pages=["Home" => "index.md", "Reference" => "reference.md"],
+    modules = [QuantumOperatorDefinitions],
+    authors = "ITensor developers <[email protected]> and contributors",
+    sitename = "QuantumOperatorDefinitions.jl",
+    format = Documenter.HTML(;
+        canonical = "https://itensor.github.io/QuantumOperatorDefinitions.jl",
+        edit_link = "main",
+        assets = ["assets/favicon.ico", "assets/extras.css"],
+    ),
+    pages = ["Home" => "index.md", "Reference" => "reference.md"],
 )
 
 deploydocs(;
-  repo="github.com/ITensor/QuantumOperatorDefinitions.jl",
-  devbranch="main",
-  push_preview=true,
+    repo = "github.com/ITensor/QuantumOperatorDefinitions.jl",
+    devbranch = "main",
+    push_preview = true,
 )

docs/make_index.jl

@@ -2,20 +2,20 @@ using Literate: Literate
 using QuantumOperatorDefinitions: QuantumOperatorDefinitions
 
 function ccq_logo(content)
-  include_ccq_logo = """
+    include_ccq_logo = """
     ```@raw html
     <img class="display-light-only" src="assets/CCQ.png" width="20%" alt="Flatiron Center for Computational Quantum Physics logo."/>
     <img class="display-dark-only" src="assets/CCQ-dark.png" width="20%" alt="Flatiron Center for Computational Quantum Physics logo."/>
     ```
     """
-  content = replace(content, "{CCQ_LOGO}" => include_ccq_logo)
-  return content
+    content = replace(content, "{CCQ_LOGO}" => include_ccq_logo)
+    return content
 end
 
 Literate.markdown(
-  joinpath(pkgdir(QuantumOperatorDefinitions), "examples", "README.jl"),
-  joinpath(pkgdir(QuantumOperatorDefinitions), "docs", "src");
-  flavor=Literate.DocumenterFlavor(),
-  name="index",
-  postprocess=ccq_logo,
+    joinpath(pkgdir(QuantumOperatorDefinitions), "examples", "README.jl"),
+    joinpath(pkgdir(QuantumOperatorDefinitions), "docs", "src");
+    flavor = Literate.DocumenterFlavor(),
+    name = "index",
+    postprocess = ccq_logo,
 )

docs/make_readme.jl

@@ -2,20 +2,20 @@ using Literate: Literate
 using QuantumOperatorDefinitions: QuantumOperatorDefinitions
 
 function ccq_logo(content)
-  include_ccq_logo = """
+    include_ccq_logo = """
     <picture>
       <source media="(prefers-color-scheme: dark)" width="20%" srcset="docs/src/assets/CCQ-dark.png">
       <img alt="Flatiron Center for Computational Quantum Physics logo." width="20%" src="docs/src/assets/CCQ.png">
     </picture>
     """
-  content = replace(content, "{CCQ_LOGO}" => include_ccq_logo)
-  return content
+    content = replace(content, "{CCQ_LOGO}" => include_ccq_logo)
+    return content
 end
 
 Literate.markdown(
-  joinpath(pkgdir(QuantumOperatorDefinitions), "examples", "README.jl"),
-  joinpath(pkgdir(QuantumOperatorDefinitions));
-  flavor=Literate.CommonMarkFlavor(),
-  name="README",
-  postprocess=ccq_logo,
+    joinpath(pkgdir(QuantumOperatorDefinitions), "examples", "README.jl"),
+    joinpath(pkgdir(QuantumOperatorDefinitions));
+    flavor = Literate.CommonMarkFlavor(),
+    name = "README",
+    postprocess = ccq_logo,
 )

examples/README.jl

@@ -1,5 +1,5 @@
 # # QuantumOperatorDefinitions.jl
-# 
+#
 # [![Stable](https://img.shields.io/badge/docs-stable-blue.svg)](https://itensor.github.io/QuantumOperatorDefinitions.jl/stable/)
 # [![Dev](https://img.shields.io/badge/docs-dev-blue.svg)](https://itensor.github.io/QuantumOperatorDefinitions.jl/dev/)
 # [![Build Status](https://github.com/ITensor/QuantumOperatorDefinitions.jl/actions/workflows/Tests.yml/badge.svg?branch=main)](https://github.com/ITensor/QuantumOperatorDefinitions.jl/actions/workflows/Tests.yml?query=branch%3Amain)
@@ -83,4 +83,4 @@ using Test: @test
 @test Matrix(OpName("Y")) == [0 -im; im 0]
 @test Matrix(OpName("Z")) == [1 0; 0 -1]
 
-@test Matrix(OpName("Rx"; θ=π / 3)) ≈ [sin(π / 3) -cos(π / 3)*im; -cos(π / 3)*im sin(π / 3)]
+@test Matrix(OpName("Rx"; θ = π / 3)) ≈ [sin(π / 3) -cos(π / 3) * im; -cos(π / 3) * im sin(π / 3)]

ext/QuantumOperatorDefinitionsGradedArraysExt/QuantumOperatorDefinitionsGradedArraysExt.jl

@@ -2,97 +2,103 @@ module QuantumOperatorDefinitionsGradedArraysExt
 
 using BlockArrays: blocklasts, blocklength, blocklengths
 using GradedArrays:
-  AbstractGradedUnitRange, GradedOneTo, SectorProduct, U1, Z, ×, dual, gradedrange, sectors
+    AbstractGradedUnitRange, GradedOneTo, SectorProduct, U1, Z, ×, dual, gradedrange, sectors
 using QuantumOperatorDefinitions:
-  QuantumOperatorDefinitions,
-  @GradingType_str,
-  @SiteType_str,
-  GradingType,
-  OpName,
-  SiteType,
-  name
+    QuantumOperatorDefinitions,
+    @GradingType_str,
+    @SiteType_str,
+    GradingType,
+    OpName,
+    SiteType,
+    name
 
 function Base.axes(::OpName, domain::Tuple{Vararg{AbstractGradedUnitRange}})
-  return (domain..., dual.(domain)...)
+    return (domain..., dual.(domain)...)
 end
 
 sortedunion(a, b) = sort(union(a, b))
 function QuantumOperatorDefinitions.combine_axes(a1::GradedOneTo, a2::GradedOneTo)
-  blocklength(a1) == blocklength(a2) ||
-    throw(ArgumentError("Axes must have the same number of blocks."))
-  nblocks = blocklength(a1)
-  return gradedrange(
-    map(Base.OneTo(nblocks)) do i
-      l1 = blocklengths(a1)[i]
-      l2 = blocklengths(a2)[i]
-      l1 == l2 || throw(ArgumentError("Blocks must have the same length."))
-      return sectors(a1)[i] × sectors(a2)[i] => l1
-    end,
-  )
+    blocklength(a1) == blocklength(a2) ||
+        throw(ArgumentError("Axes must have the same number of blocks."))
+    nblocks = blocklength(a1)
+    return gradedrange(
+        map(Base.OneTo(nblocks)) do i
+            l1 = blocklengths(a1)[i]
+            l2 = blocklengths(a2)[i]
+            l1 == l2 || throw(ArgumentError("Blocks must have the same length."))
+            return sectors(a1)[i] × sectors(a2)[i] => l1
+        end,
+    )
 end
 QuantumOperatorDefinitions.combine_axes(a::GradedOneTo, b::Base.OneTo) = a
 QuantumOperatorDefinitions.combine_axes(a::Base.OneTo, b::GradedOneTo) = b
 
 function Base.AbstractUnitRange(::GradingType"N", t::SiteType)
-  return gradedrange(map(i -> SectorProduct((; N=U1(i - 1))) => 1, 1:length(t)))
+    return gradedrange(map(i -> SectorProduct((; N = U1(i - 1))) => 1, 1:length(t)))
 end
 function Base.AbstractUnitRange(::GradingType"Sz", t::SiteType)
-  return gradedrange(map(i -> SectorProduct((; Sz=U1(i - 1))) => 1, 1:length(t)))
+    return gradedrange(map(i -> SectorProduct((; Sz = U1(i - 1))) => 1, 1:length(t)))
 end
 function Base.AbstractUnitRange(::GradingType"Sz↑", t::SiteType)
-  return AbstractUnitRange(GradingType"Sz"(), t)
+    return AbstractUnitRange(GradingType"Sz"(), t)
 end
 function Base.AbstractUnitRange(::GradingType"Sz↓", t::SiteType)
-  return gradedrange(map(i -> SectorProduct((; Sz=U1(-(i - 1)))) => 1, 1:length(t)))
+    return gradedrange(map(i -> SectorProduct((; Sz = U1(-(i - 1)))) => 1, 1:length(t)))
 end
 
 function sector(gradingtype::GradingType, sec)
-  sectorname = Symbol(get(gradingtype, :name, name(gradingtype)))
-  return SectorProduct(NamedTuple{(sectorname,)}((sec,)))
+    sectorname = Symbol(get(gradingtype, :name, name(gradingtype)))
+    return SectorProduct(NamedTuple{(sectorname,)}((sec,)))
 end
 
 function Base.AbstractUnitRange(s::GradingType"Nf", t::SiteType"Fermion")
-  return gradedrange([sector(s, U1(0)) => 1, sector(s, U1(1)) => 1])
+    return gradedrange([sector(s, U1(0)) => 1, sector(s, U1(1)) => 1])
 end
 # TODO: Write in terms of `GradingType"Nf"` definition.
 function Base.AbstractUnitRange(s::GradingType"NfParity", t::SiteType"Fermion")
-  return gradedrange([sector(s, Z{2}(0)) => 1, sector(s, Z{2}(1)) => 1])
+    return gradedrange([sector(s, Z{2}(0)) => 1, sector(s, Z{2}(1)) => 1])
 end
 function Base.AbstractUnitRange(s::GradingType"Sz", t::SiteType"Fermion")
-  return gradedrange([sector(s, U1(0)) => 1, sector(s, U1(1)) => 1])
+    return gradedrange([sector(s, U1(0)) => 1, sector(s, U1(1)) => 1])
 end
 function Base.AbstractUnitRange(s::GradingType"Sz↑", t::SiteType"Fermion")
-  return gradedrange([sector(s, U1(0)) => 1, sector(s, U1(1)) => 1])
+    return gradedrange([sector(s, U1(0)) => 1, sector(s, U1(1)) => 1])
 end
 function Base.AbstractUnitRange(s::GradingType"Sz↓", t::SiteType"Fermion")
-  return gradedrange([sector(s, U1(0)) => 1, sector(s, U1(-1)) => 1])
+    return gradedrange([sector(s, U1(0)) => 1, sector(s, U1(-1)) => 1])
 end
 
 # TODO: Write in terms of `SiteType"Fermion"` definitions.
 function Base.AbstractUnitRange(s::GradingType"Nf", t::SiteType"Electron")
-  return gradedrange([
-    sector(s, U1(0)) => 1,
-    sector(s, U1(1)) => 1,
-    sector(s, U1(1)) => 1,
-    sector(s, U1(2)) => 1,
-  ])
+    return gradedrange(
+        [
+            sector(s, U1(0)) => 1,
+            sector(s, U1(1)) => 1,
+            sector(s, U1(1)) => 1,
+            sector(s, U1(2)) => 1,
+        ]
+    )
 end
 # TODO: Write in terms of `GradingType"Nf"` definition.
 function Base.AbstractUnitRange(s::GradingType"NfParity", t::SiteType"Electron")
-  return gradedrange([
-    sector(s, Z{2}(0)) => 1,
-    sector(s, Z{2}(1)) => 1,
-    sector(s, Z{2}(1)) => 1,
-    sector(s, Z{2}(0)) => 1,
-  ])
+    return gradedrange(
+        [
+            sector(s, Z{2}(0)) => 1,
+            sector(s, Z{2}(1)) => 1,
+            sector(s, Z{2}(1)) => 1,
+            sector(s, Z{2}(0)) => 1,
+        ]
+    )
 end
 function Base.AbstractUnitRange(s::GradingType"Sz", t::SiteType"Electron")
-  return gradedrange([
-    sector(s, U1(0)) => 1,
-    sector(s, U1(1)) => 1,
-    sector(s, U1(-1)) => 1,
-    sector(s, U1(0)) => 1,
-  ])
+    return gradedrange(
+        [
+            sector(s, U1(0)) => 1,
+            sector(s, U1(1)) => 1,
+            sector(s, U1(-1)) => 1,
+            sector(s, U1(0)) => 1,
+        ]
+    )
 end
 
 end

ext/QuantumOperatorDefinitionsITensorBaseExt/QuantumOperatorDefinitionsITensorBaseExt.jl

@@ -4,61 +4,61 @@ using ITensorBase: ITensorBase, ITensor, Index, gettag, prime, settag
 using GradedArrays: dual
 using NamedDimsArrays: dename
 using QuantumOperatorDefinitions:
-  QuantumOperatorDefinitions,
-  @OpName_str,
-  OpName,
-  SiteType,
-  StateName,
-  has_fermion_string,
-  name
+    QuantumOperatorDefinitions,
+    @OpName_str,
+    OpName,
+    SiteType,
+    StateName,
+    has_fermion_string,
+    name
 
 function QuantumOperatorDefinitions.SiteType(r::Index)
-  # We pass the axis of the (unnamed) Index because
-  # the Index may have originated from a slice, in which
-  # case the start may not be 1 (for NonContiguousIndex,
-  # which we need to add support for, it may not even
-  # be a unit range).
-  return SiteType(
-    gettag(r, "sitetype", "Qudit"); dim=Int.(length(r)), range=only(axes(dename(r)))
-  )
+    # We pass the axis of the (unnamed) Index because
+    # the Index may have originated from a slice, in which
+    # case the start may not be 1 (for NonContiguousIndex,
+    # which we need to add support for, it may not even
+    # be a unit range).
+    return SiteType(
+        gettag(r, "sitetype", "Qudit"); dim = Int.(length(r)), range = only(axes(dename(r)))
+    )
 end
 
 function (rangetype::Type{<:Index})(t::SiteType)
-  i = rangetype(AbstractUnitRange(t))
-  i = settag(i, "sitetype", String(name(t)))
-  if haskey(t, :site)
-    i = settag(i, "site", string(t.site))
-  end
-  return i
+    i = rangetype(AbstractUnitRange(t))
+    i = settag(i, "sitetype", String(name(t)))
+    if haskey(t, :site)
+        i = settag(i, "site", string(t.site))
+    end
+    return i
 end
 
 # TODO: Define in terms of `OpName` directly, and define a generic
 # forwarding method `has_fermion_string(n::String, t) = has_fermion_string(OpName(n), t)`.
 function QuantumOperatorDefinitions.has_fermion_string(n::String, r::Index)
-  return has_fermion_string(OpName(n), SiteType(r))
+    return has_fermion_string(OpName(n), SiteType(r))
 end
 
 function Base.axes(::OpName, domain::Tuple{Vararg{Index}})
-  return (prime.(domain)..., dual.(domain)...)
+    return (prime.(domain)..., dual.(domain)...)
 end
 ## function Base.axes(::OpName"SWAP", domain::Tuple{Vararg{Index}})
 ##   return (prime.(reverse(domain))..., dag.(domain)...)
 ## end
 
 # Fix ambiguity error with generic `AbstractArray` version.
-function ITensorBase.ITensor(n::Union{OpName,StateName}, domain::Index...)
-  return ITensor(n, domain)
+function ITensorBase.ITensor(n::Union{OpName, StateName}, domain::Index...)
+    return ITensor(n, domain)
 end
 # Fix ambiguity error with generic `AbstractArray` version.
-function ITensorBase.ITensor(n::Union{OpName,StateName}, domain::Tuple{Vararg{Index}})
-  return ITensor(AbstractArray(n, domain), axes(n, domain))
+function ITensorBase.ITensor(n::Union{OpName, StateName}, domain::Tuple{Vararg{Index}})
+    return ITensor(AbstractArray(n, domain), axes(n, domain))
 end
 function (arrtype::Type{<:AbstractArray})(
-  n::Union{OpName,StateName}, domain::Tuple{Vararg{Index}}
-)
-  # Convert to `SiteType` in case the Index specifies a `"sitetype"` tag.
-  # TODO: Try to build this into the generic codepath.
-  return ITensor(arrtype(n, SiteType.(domain)), axes(n, domain))
+        n::Union{OpName, StateName}, domain::Tuple{Vararg{Index}}
+    )
+    # Convert to `SiteType` in case the Index specifies a `"sitetype"` tag.
+    # TODO: Try to build this into the generic codepath.
+    return ITensor(arrtype(n, SiteType.(domain)), axes(n, domain))
 end
 
 end

src/has_fermion_string.jl

@@ -1,7 +1,7 @@
 function has_fermion_string(n::OpName, t::SiteType)
-  n′ = alias(n)
-  if n == n′
-    return false
-  end
-  return has_fermion_string(n′, t)
+    n′ = alias(n)
+    if n == n′
+        return false
+    end
+    return has_fermion_string(n′, t)
 end