chore: bump version to 0.4.1

This also fixes some bugs with referencing figures in elements such as align,
this also generalizes the logic to handle nesting.

#__format-ref show rule now goes through #custom-ref most of the time,
so builtin #ref can be used instead.

Equations are now referenceable and respect numbering per sections.

README.md

@@ -1,3 +1,12 @@
 # ktu-paper
 
 This repository contains a Typst template for academic papers at the Kaunas University of Technology, Faculty of Informatics. A final Bachelor's thesis written with this template has successfully passed the formal review.
+
+## Initiating the template
+Install the downloaded template under the following directory:
+- Linux: `~/.local/share/typst/packages/local/ktu-paper/0.4.1/`
+
+And then initiate a new project:
+```sh
+typst init @local/ktu-paper:0.4.1
+```

src/config.typ

@@ -6,6 +6,11 @@
 /// Appendix and other non-standard sections.
 #let __FIG_PER_SECTION = state("svetikas.lt/ktu-paper/figure-numbering-per-section", false)
 
+/// Whether to uppercase section titles or not.
+/// If true, section titles will be uppercased.
+/// If false, section titles will be displayed as-is.
+#let __UPPERCASE_SECTION_TITLES = state("svetikas.lt/ktu-paper/uppercase-section-titles", false)
+
 #let __DEBUG_AUTHOR_TABLE = state("$ktu-template-DEBUG_AUTHOR_TABLE", false)
 #let debug-author-table() = {
   __DEBUG_AUTHOR_TABLE.update(true)

src/figures.typ

@@ -1,135 +1,354 @@
 // This file includes logic for displaying figures in the document, including numbering, captions, and references. It also includes logic for handling figure numbering per section or globally, based on the configuration set in `config.typ`.
 #import "config.typ": __FIG_PER_SECTION
 
-/// Localizations for various recognized referencable figure types.
-/// This is used to provide the correct supplement text for each type of figure
-/// when generating references, captions and outlines.
-#let __figureNames = (
-    table: (
-        supplement: "lentelė",
-    ),
-    image: (
-        supplement: "pav",
-    ),
-    raw: (
-        supplement: "kodo frag",
-    ),
-    equation: (
-        supplement: "lygtis",
-    ),
+// ---------------------------------------------------------------------------
+// Figure kinds
+// ---------------------------------------------------------------------------
+
+/// Localizations for the recognized referenceable figure kinds, keyed by `repr(func)`.
+/// - supplement: used in captions (the caption adds the separator, e.g. "1 pav.").
+/// - ref-name: used in references, e.g. "(žr. 1 pav.)".
+/// - single-name: full word used when the figure is the only one of its kind and
+///   therefore unnumbered, e.g. "Paveikslas. ..." and "(žr. paveikslas)".
+#let __figure-kinds = (
+  table: (supplement: "lentelė", ref-name: "lentelė", single-name: "lentelė"),
+  image: (supplement: "pav", ref-name: "pav.", single-name: "paveikslas"),
+  raw: (supplement: "kodo frag", ref-name: "kodo frag.", single-name: "kodo fragmentas"),
+  equation: (supplement: "lygtis", ref-name: "formulė", single-name: "lygtis"),
 )
 
-/// Helper function to get the supplement text for a given figure type.
-#let __fig-supplement(func) = {
-    return __figureNames.at(repr(func)).supplement
+/// Figure kinds registered by the document author via `ktu-paper(figureKinds: ...)`.
+#let __USER_FIGURE_KINDS = state("ktu-user-figure-kinds", (:))
+
+/// Validates author-supplied kinds and fills in defaults (`ref-name` falls back
+/// to `supplement`).
+#let __normalize-figure-kinds(kinds) = {
+  let normalized = (:)
+  for (key, spec) in kinds {
+    assert(
+      "supplement" in spec,
+      message: "figure kind `" + key + "` needs a `supplement`",
+    )
+    normalized.insert(key, (
+      supplement: spec.supplement,
+      ref-name: spec.at("ref-name", default: spec.supplement),
+      single-name: spec.at("single-name", default: spec.supplement),
+    ))
+  }
+  normalized
 }
 
-#let __fig-func-to-name(func) = {
-    if func == table {
-        return "lentelė"
+/// Recognized kinds merged with the author's. Must be called in a context.
+#let __all-figure-kinds() = __figure-kinds + __USER_FIGURE_KINDS.get()
+
+/// Resets all figure and equation counters (used when numbering per section).
+/// Must be called in a context.
+#let __reset-figure-counters() = {
+  let custom = (..__USER_FIGURE_KINDS.get().keys(),)
+  for kind in (image, table, raw, ..custom) {
+    counter(figure.where(kind: kind)).update(0)
   }
-    if func == image {
-        return "pav."
-    }
-    if func == grid {
-        return "pav."
-    }
-    return func
+  counter(math.equation).update(0)
 }
 
-/// Helper function to resolve figure numbers (either global "1" or section-based "1.1").
-/// It takes a figure item and referenced location and returns the appropriate number as a string.
-#let __get-figure-number(fig-item, loc) = context {
-    if type(fig-item) != content {
-        panic("can only get figure number for content elements")
+/// Returns the caption supplement text for a figure kind (e.g. `table`, `math.equation`).
+#let __fig-supplement(func) = __figure-kinds.at(repr(func)).supplement
+
+/// Finds the first recognized kind ("table", "image", "raw" or "equation") inside
+/// `node`, descending through wrappers (align, block, grid, stack, ...).
+/// Returns the key into `__figure-kinds`, or `none` if nothing is recognized.
+#let __find-figure-kind(node) = {
+  if type(node) != content { return none }
+
+  let key = repr(node.func())
+  if key in __figure-kinds { return key }
+
+  // Single-child wrappers
+  for field in ("body", "child") {
+    if node.has(field) {
+      let found = __find-figure-kind(node.at(field))
+      if found != none { return found }
+    }
   }
 
-    if fig-item.func() != figure {
-        panic("can only get figure number for figure elements")
+  // Multi-child containers
+  if node.has("children") {
+    for child in node.children {
+      let found = __find-figure-kind(child)
+      if found != none { return found }
+    }
   }
 
-    let perSection = __FIG_PER_SECTION.get()
-    if perSection {
+  none
+}
+
+/// Resolves the `__figure-kinds` key of a figure element. An explicit custom
+/// (string) kind always wins; otherwise the kind is detected from the body.
+#let __figure-key(fig) = {
+  if type(fig.kind) == str { return fig.kind }
+  let found = __find-figure-kind(fig.body)
+  if found != none { found } else { repr(fig.kind) }
+}
+
+/// Returns the caption supplement of a figure element.
+#let __figure-supplement-of(fig) = {
+  let key = __figure-key(fig)
+  let kinds = __all-figure-kinds()
+  assert(
+    key in kinds,
+    message: "unregistered figure kind `" + key + "`: register it with ktu-paper(figureKinds: ...)",
+  )
+  kinds.at(key).supplement
+}
+
+// ---------------------------------------------------------------------------
+// Numbering
+// ---------------------------------------------------------------------------
+
+/// Resolves a figure's number (global "1" or section-based "1.1") at `loc`,
+/// returned as a string.
+#let __get-figure-number(fig-item, loc) = {
+  assert(
+    type(fig-item) == content and fig-item.func() == figure,
+    message: "can only get figure number for figure elements",
+  )
+
+  let fig-counter = counter(figure.where(kind: fig-item.kind))
+
+  if __FIG_PER_SECTION.at(loc) {
     let chapter = counter(heading).at(loc).at(0, default: 0)
-        let figNum = counter(figure.where(kind: fig-item.kind)).at(loc).at(0, default: 1)
-        [#chapter.#figNum]
+    let fig-num = fig-counter.at(loc).at(0, default: 1)
+    str(chapter) + "." + str(fig-num)
   } else {
-        let caption = fig-item.caption
-        if caption != none and caption.numbering != none {
-            caption.counter.at(loc).at(0)
-        } else {
-            [#counter(figure.where(kind: fig-item.kind)).at(loc).at(0)]
-        }
+    str(fig-counter.at(loc).at(0))
   }
 }
 
-/// Helper function to get the number of a locatable element (figure, table, etc.)
-/// based on its location in the document. It returns the number as a string, formatted according to the current numbering scheme.
+/// Resolves an equation's number (global "2" or section-based "1.2") at `loc` without parentheses.
+/// `loc` is the location of the equation.
+#let __get-equation-number(loc) = {
+  let number = counter(math.equation).at(loc).at(0)
+  if __FIG_PER_SECTION.at(loc) {
+    let chapter = counter(heading).at(loc).at(0, default: 0)
+    str(chapter) + "." + str(number)
+  } else {
+    str(number)
+  }
+}
+
+/// Numbering function for `math.equation`: "(2)" or, when numbering per
+/// section, "(1.2)".
+#let __equation-numbering(..nums) = context {
+  let number = nums.pos().first()
+  if __FIG_PER_SECTION.get() {
+    let chapter = counter(heading).get().at(0, default: 0)
+    // TODO: this is a hack to prevent the numbers from being completely
+    // part of a regex match against math.equation.
+    // It would be better to render math equation numbers as a separate element, so that regexes don't see them at all.
+    [(#box[#chapter].#box[#number])]
+  } else {
+    [(#number)]
+  }
+}
+
+/// Returns the number of a located figure element as a string, formatted
+/// according to the current numbering scheme.
 #let __get-locatable-element-number(element) = context {
-    if type(element) != content {
-        panic("can only reference elements of type content")
-    }
-    let loc = element.location()
-    let elem = query(loc)
-    let fig = elem.at(0)
-    if fig.func() != figure {
-        panic("referenced element does not have a figure function")
-    }
-    __get-figure-number(fig, loc)
+  assert(type(element) == content, message: "can only reference elements of type content")
+  assert(element.func() == figure, message: "referenced element does not have a figure function")
+  __get-figure-number(element, element.location())
 }
 
-/// Formats a reference to a figure, table, or other numbered element.
+/// Appendix headings carry an "appendix-heading" metadata marker as their
+/// first child and are numbered differently from normal headings.
+#let __is-appendix-heading(item) = {
+  let body = item.body
+  body.has("children") and body.children.at(0, default: none) == metadata("appendix-heading")
+}
+
+/// Parses a dotted number such as "12" or "1.12.5" into an array of ints.
+/// Returns `none` if any part is not purely numeric.
+#let __parse-number(value) = {
+  let parts = value.split(".")
+  if parts.all(p => p.match(regex("^\\d+$")) != none) {
+    parts.map(int)
+  } else {
+    none
+  }
+}
+
+/// Sorts `(location, number)` entries. If every number is numeric (optionally
+/// dotted, e.g. "1.12.5"), sorts component-wise numerically; otherwise falls
+/// back to plain string ordering.
+#let __sort-entries(entries) = {
+  let keys = entries.map(e => __parse-number(e.at(1)))
+  if keys.all(k => k != none) {
+    entries.zip(keys).sorted(key: pair => pair.at(1)).map(pair => pair.at(0))
+  } else {
+    entries.sorted(key: e => e.at(1))
+  }
+}
+
+// ---------------------------------------------------------------------------
+// Single-figure handling
+// ---------------------------------------------------------------------------
+
+/// Figure kinds that are left unnumbered when the document contains only one
+/// captioned figure of that kind: "Lentelė. Aprašas" and "(žr. lentelė)"
+/// instead of "1 lentelė. Aprašas" and "(žr. 1 lentelė)". The word used comes
+/// from the kind's `single-name`.
+/// Keys are `__figure-key` values.
+#let __UNNUMBERED_WHEN_SINGLE = ("table", "image")
+
+/// Whether `fig` is the only captioned figure of its kind in the document and
+/// its kind is listed in `__UNNUMBERED_WHEN_SINGLE`. Must be called in a context.
+#let __is-single-unnumbered(fig) = {
+  if __figure-key(fig) not in __UNNUMBERED_WHEN_SINGLE { return false }
+  query(figure.where(kind: fig.kind)).filter(f => f.caption != none).len() == 1
+}
+
+/// Capitalizes the first letter of `value` ("lentelė" -> "Lentelė").
+#let __capitalize(value) = {
+  let first = value.first()
+  upper(first) + value.slice(first.len())
+}
+
+/// Caption label of a figure element: "1 lentelė." normally, or
+/// "Lentelė." for a single unnumbered figure.
+#let __caption-label(fig) = context {
+  let supplement = __figure-supplement-of(fig)
+  if __is-single-unnumbered(fig) {
+    let name = __all-figure-kinds().at(__figure-key(fig)).single-name
+    [#__capitalize(name).]
+  } else {
+    [#__get-figure-number(fig, fig.location()) #supplement.]
+  }
+}
+
+// ---------------------------------------------------------------------------
+// References
+// ---------------------------------------------------------------------------
+
+/// Returns a number for the given reference target.
+/// Panics for a figure that is the only one of its kind,
+/// since such figures have no number.
+/// - target (label): Label to get the reference number for.
+#let ref-no(target) = context {
+  if type(target) != label {
+    panic("ref-no expects a label as input")
+  }
+
+  let item = query(target).first()
+  let loc = item.location()
+  let kind = item.func()
+
+  let number = if kind == figure {
+    if __is-single-unnumbered(item) {
+      panic(
+        "ref-no: " + str(target) + " is the only " + __figure-key(item)
+          + " in the document and is therefore unnumbered; "
+          + "refer to it with @" + str(target) + " (or custom-ref) instead",
+      )
+    }
+    __get-figure-number(item, loc)
+  } else if kind == math.equation {
+    "(" + __get-equation-number(loc) + ")"
+  } else if kind == heading {
+    let numbers = counter(heading).at(loc)
+    if __is-appendix-heading(item) {
+      str(numbers.at(1))
+    } else {
+      numbers.map(str).join(".")
+    }
+  } else {
+    panic("referencing unknown type: " + repr(kind))
+  }
+
+  link(loc, number)
+}
+
+/// Create a custom reference for multiple targets of the same type.
+/// - targets (label...): Labels to create a custom reference for.
+#let custom-ref(..targets) = context {
+  let group-name = none
+  let single-name = none // set when the target is a lone, unnumbered figure
+  let entries = () // (location, number) pairs
+
+  for target in targets.pos() {
+    let item = query(target).first()
+    let loc = item.location()
+
+    let (name, number) = if item.func() == math.equation {
+      (__figure-kinds.equation.ref-name, __get-equation-number(loc))
+    } else if item.func() == figure {
+      let kind = __figure-key(item)
+      let kinds = __all-figure-kinds()
+      if kind not in kinds {
+        panic("unregistered figure kind `" + kind + "`: register it with ktu-paper(figureKinds: ...)")
+      }
+      if __is-single-unnumbered(item) { single-name = kinds.at(kind).single-name }
+      (kinds.at(kind).ref-name, __get-figure-number(item, loc))
+    } else {
+      panic("referencing unknown type: " + repr(item.func()))
+    }
+
+    if group-name != none and group-name != name {
+      panic("custom-ref does not support grouping different elements together")
+    }
+    group-name = name
+    entries.push((loc, number))
+  }
+
+  if entries.len() > 0 {
+    if single-name != none {
+      // The only figure of its kind: no number, e.g. "(žr. lentelė)"
+      [(žr. #link(entries.first().at(0), single-name))]
+    } else {
+      let links = __sort-entries(entries).map(e => link(e.at(0), e.at(1)))
+      [(žr. #links.join(", ") #group-name)]
+    }
+  }
+}
+
+/// Show rule for `@label` references. Equations render as a bare "(1.2)";
+/// figures delegate to `custom-ref`, so "(žr. 1 pav.)" is built in one place.
+/// Anything else (headings, citations, unregistered figure kinds) keeps
+/// Typst's default rendering.
 #let __format-ref(it) = {
   let el = it.element
-    if el == none {
-        return it
-    }
+  if el == none { return it }
 
-    // Override equation references
   if el.func() == math.equation {
-        link(
-            el.location(),
-            numbering(el.numbering, ..counter(math.equation).at(el.location()))
-        )
-    // Override references to figures
-    } else if el.func() == figure {
-        let fig = el.body
-        let figType = fig.func()
-        let numStr = __get-figure-number(el, el.location())
-        link(el.location(), [
-            (žr. #numStr #__fig-func-to-name(figType))
-        ])
+    let loc = el.location()
+    link(loc, "(" + __get-equation-number(loc) + ")")
+  } else if el.func() == figure and __figure-key(el) in __all-figure-kinds() {
+    custom-ref(it.target)
   } else {
     it
   }
 }
 
+// ---------------------------------------------------------------------------
+// Figure layout
+// ---------------------------------------------------------------------------
+
 /// Formats a figure with a caption and number.
 #let __format-figure(it) = {
-    let separator = [.]
   let caption = it.caption
 
   // If caption is not provided, just show the figure body
-    if caption == none {
-        return it.body
-    }
+  if caption == none { return it.body }
 
-    let counter = caption.counter  
-    let supplement = caption.supplement
-
-    if it.body.func() == math.equation {
-        supplement = __fig-supplement("equation")
-    } else {
-        supplement = __fig-supplement(it.kind)
-    }
   set text(size: 11pt)
 
+  let caption-line = [*#__caption-label(it)* #caption.body]
+
   // Tables have captions at the top
   if it.kind == table {
-        align(left)[*#__get-figure-number(it, here()) #supplement#separator* #caption.body]
+    align(left, caption-line)
     it.body
   } else {
     it.body
-        [*#__get-figure-number(it, here()) #supplement#separator* #caption.body]
+    align(center, caption-line)
   }
 }

src/fragments.typ

@@ -17,6 +17,7 @@
 #let ktu-author-table(
   authors: (),
   alignment: right,
+  namesFirst: true, // if false, roles will be displayed first
 ) = context {
   // The layout should be a list of lists whose elements are (name, role) pairs.
   assert(authors.len() > 0, message: "author table must have at least one author")
@@ -51,29 +52,43 @@
       table.header([]),
       ..for author in authors {
         let (name, role) = author
-                (align(center)[
+        name = [*#name*]
+        if not namesFirst {
+          (role, name) = (name, role)
+        }
+        (
+          align(center)[
             #set block(below: 8pt)
-                    #block[*#name*]
+            #block[#name]
             #block[#role]
 
-                ], [])
+          ],
+          [],
+        )
       }
     ))
   } else {
     align(alignment, table(
       columns: (198.5pt, 56.7pt),
-            table.header(
-                [], [],
-            ),
+      table.header([], []),
       ..for author in authors {
         let (name, role) = author
-                (align(left)[
+        name = [*#name*]
+        if not namesFirst {
+          (role, name) = (name, role)
+        }
+        (
+          align(left)[
             #set block(below: 8pt)
-                    #block[*#name*]
+            #block[#name]
             #block[#role]
 
-                ], [], [], [])
-            }
+          ],
+          [],
+          [],
+          [],
+        )
+      },
     ))
   }
 }
@@ -128,8 +143,9 @@
   ]
 }
 
-#let ktu-heading-page-centered(
+#let ktu-heading-page(
   faculty: highlight[Fakulteto pavadinimas],
+  department: none, // optional
   title: highlight[Rašto darbo pavadinimas],
   subtitles: highlight[Praktikos ataskaita / referatas ir pan. (įrašyti rašto darbo tipą)],
   authors: (
@@ -138,44 +154,80 @@
   ),
   place: highlight[Kaunas],
   year: highlight[#datetime.today().year()],
+  centerAuthors: false,
+  authorNamesFirst: true, // if false, roles will be displayed first
 ) = {
-    let isArr = type(subtitles) == array
-    if not isArr {
+  let subHeaders = (faculty,)
+  if department != none {
+    subHeaders = (faculty, department)
+  }
+  if type(subtitles) != array {
     subtitles = (subtitles,)
   }
   ktu-title-header-block(
-        subHeaders: (faculty,),
+    subHeaders: subHeaders,
     bigTitle: title,
     subTitles: subtitles,
   )
   v(94pt)
-    ktu-author-table(authors: authors, alignment: center)
-    align(center+bottom, block([*#place, #year*]))
+  if centerAuthors {
+    ktu-author-table(authors: authors, namesFirst: authorNamesFirst, alignment: center)
+  } else {
+    ktu-author-table(authors: authors, namesFirst: authorNamesFirst)
+  }
+  align(center + bottom, block([*#place, #year*]))
+}
+
+#let ktu-heading-page-centered(
+  faculty: highlight[Fakulteto pavadinimas],
+  department: none, // optional
+  title: highlight[Rašto darbo pavadinimas],
+  subtitles: highlight[Praktikos ataskaita / referatas ir pan. (įrašyti rašto darbo tipą)],
+  authors: (
+    (highlight[Vardenis Pavardenis], highlight[Projekto autorius]),
+    (highlight[Vardenis Pavardenis], highlight[Vadovas]),
+  ),
+  authorNamesFirst: true, // if false, roles will be displayed first
+  place: highlight[Kaunas],
+  year: highlight[#datetime.today().year()],
+) = {
+  ktu-heading-page(
+    faculty: faculty,
+    department: department,
+    title: title,
+    subtitles: subtitles,
+    authors: authors,
+    place: place,
+    year: year,
+    centerAuthors: true,
+    authorNamesFirst: authorNamesFirst,
+  )
 }
 
 #let ktu-heading-page-normal(
   faculty: highlight[Fakulteto pavadinimas],
+  department: none, // optional
   title: highlight[Rašto darbo pavadinimas],
   subtitles: highlight[Praktikos ataskaita / referatas ir pan. (įrašyti rašto darbo tipą)],
   authors: (
     (highlight[Vardenis Pavardenis], highlight[Projekto autorius]),
     (highlight[Vardenis Pavardenis], highlight[Vadovas]),
   ),
+  authorNamesFirst: true, // if false, roles will be displayed first
   place: highlight[Kaunas],
   year: highlight[#datetime.today().year()],
 ) = {
-    let isArr = type(subtitles) == array
-    if not isArr {
-        subtitles = (subtitles,)
-    }
-    ktu-title-header-block(
-        subHeaders: (faculty,),
-        bigTitle: title,
-        subTitles: subtitles,
+  ktu-heading-page(
+    faculty: faculty,
+    department: department,
+    title: title,
+    subtitles: subtitles,
+    authors: authors,
+    place: place,
+    year: year,
+    centerAuthors: false,
+    authorNamesFirst: authorNamesFirst,
   )
-    v(94pt)
-    ktu-author-table(authors: authors)
-    align(center+bottom, block([*#place, #year*]))
 }
 
 #let ktu-academic-honestly-declaration-page(
@@ -186,7 +238,7 @@
   ktu-title-header-block(
     subHeaders: (faculty, author),
     bigTitle: projectTitle,
-        subTitles: ("Akademinio sąžiningumo deklaracija",) 
+    subTitles: ("Akademinio sąžiningumo deklaracija",),
   )
   v(20pt)
   [

src/lib.typ

@@ -1,14 +1,22 @@
 #import "@preview/codly:1.3.0": codly, codly-init
-#import "config.typ": debug-author-table, enable-heading-experiment, __HEADING_EXPERIMENT, __FIG_PER_SECTION
+#import "config.typ": debug-author-table, enable-heading-experiment, __HEADING_EXPERIMENT, __FIG_PER_SECTION, __UPPERCASE_SECTION_TITLES
 #import "fragments.typ": (
   ktu-table,
-    ktu-heading-page-centered, ktu-heading-page-normal, ktu-academic-honestly-declaration-page
+
+  // building blocks
+  ktu-author-table, ktu-title-header-block,
+  
+  ktu-heading-page, ktu-heading-page-centered, ktu-heading-page-normal,
+  ktu-academic-honestly-declaration-page
 )
 #import "figures.typ": (
-    __fig-supplement,
+  ref-no, custom-ref,
 
-    __get-figure-number,
-    __get-locatable-element-number,
+  __caption-label,
+  __reset-figure-counters,
+  __equation-numbering,
+  __normalize-figure-kinds,
+  __USER_FIGURE_KINDS,
 
   __format-ref,
   __format-figure,
@@ -16,23 +24,33 @@
 
 /**
  * Paruošta pagal "Rašto ..."
-*/
+ */
 
 /*
  * Further reading required:
  * - Double line spacing | https://github.com/typst/typst/issues/106
  * - Proposal: change `leading` option to `line-height` | https://github.com/typst/typst/issues/4224
-*/
+ */
+
+#let __METADATA_HEADING_ALIGN_LEFT = metadata("heading-align-left");
 
 ///
 /// Insert a heading without a number.
 ///
 /// Usually used for introductory or summary sections at the start and end.
-/// - text (str): The heading's title.
+/// - headingElement (heading): The heading's title.
+/// - alignLeft (bool): Whether the heading should be left-aligned or centered.
 /// ->
-#let unnumbered-heading(text) = {
-    set heading(numbering: none)
-    [#text]
+#let unnumbered-heading(headingElement, alignLeft: false) = {
+  let fields = headingElement.fields()
+  let body = fields.body
+  let level = fields.at("level", default: 1)
+
+  if alignLeft {
+    body = body + __METADATA_HEADING_ALIGN_LEFT
+  }
+
+  heading(level: level, numbering: none, body)
 }
 
 #let appendix-item(body) = {
@@ -49,8 +67,7 @@
   let s = str(rounded)
   let parts = s.split(".")
   let integer-part = parts.at(0)
-  let decimal-part = if parts.len() == 2 { parts.at(1) } else { ""}
-
+  let decimal-part = if parts.len() == 2 { parts.at(1) } else { "" }
 
   let padded-decimal = if decimals > 0 {
     let current-decimal-length = decimal-part.len()
@@ -70,109 +87,6 @@
   integer-part + padded-decimal
 }
 
-/// Returns a number for the given reference target.
-/// - target (label): Label to get the reference number for.
-#let ref-no(target) = context {
-    if type(target) != label {
-        panic("ref-no expects a label as input")
-    }
-
-    let loc = locate(target)
-    let elem = query(target)
-    let item = elem.at(0)
-    
-    // If it's a figure, return the figure number
-    if item.func() == figure {
-        link(loc, [#__get-figure-number(item, loc)])
-    } else if item.func() == heading {
-        let fullValue = counter(heading).at(loc)
-
-        // If the heading has the "appendix-heading" metadata attached,
-        // it means that it's an appendix heading and should be numbered differently
-        if item.body.has("children") and item.body.children.at(0) == metadata("appendix-heading") {
-            link(loc, [#fullValue.at(1)])
-        // Otherwise, show the fully qualified heading numbering
-        } else {
-            link(loc, [#fullValue.map((x) => str(x)).join(".")])
-        }
-    } else {
-        panic("referencing unknown type: ", item.func())
-    }
-}
-
-// Constructs a reference from multiple targets
-#let custom-ref(..targets) = context {
-  let collected = (:)
-
-  for target in targets.pos() {
-    let loc = locate(target)
-    let elem = query(target)
-
-    let fig = elem.at(0)
-    if fig.func() != figure {
-      panic("referencing unknown type")
-    }
-
-    // determine what we...
-    let element = fig.body;
-    let elementName = "unknown";
-    let found = false;
-    while not found {
-
-      // If it's a grid
-      if (element.func() == grid) {
-        for cell in element.children {
-          if cell.body.func() == image {
-            elementName = "pav."
-            found = true
-            break
-          }
-        }
-        break
-      }
-
-      if (element.func() == table) {
-        elementName = "lentelė"
-        found = true
-        break
-      }
-
-    if (element.func() == raw) {
-        elementName = "kodo frag."
-        found = true
-        break
-      }
-
-      if (element.func() == image) {
-        elementName = "pav."
-        found = true
-        break
-      }
-
-      panic("unknown element", element.func())
-    }
-
-    if elementName not in collected {
-      collected.insert(elementName, ())
-    }
-
-    collected.at(elementName).push(
-      (loc, __get-figure-number(fig, loc))
-    )
-  }
-
-  if collected.len() > 1 {
-    panic("multiple types detected")
-  }
-
-  for type in collected {
-    let name = type.at(0)
-    let arr = type.at(1).sorted(key: k => k.at(1))
-
-    [(žr. #arr.map(x =>link(x.at(0), [#x.at(1)])).join(", ") #name)]
-  }
-}
-
 
 /// Returns a KTU template-conformant outline.
 /// - depth (int | none): Same as outline depth.
@@ -184,7 +98,7 @@
   depth: none,
   indent: 0em, // Disable the default outline indenting
   target: heading,
-    title: auto
+  title: auto,
 ) = {
   // TODO: įtrauka
   set block(spacing: 0em)
@@ -192,6 +106,9 @@
   show outline: o => context {
     // If it's a heading, show the default outline with bold text
     if (o.target == selector(heading)) {
+      // Determine whether to uppercase section titles based on the global setting
+      let isUpper = __UPPERCASE_SECTION_TITLES.get()
+
       // Define known target widths based on KTU template
       let targets = (0.64cm, 0.96cm, 1.28cm, 1.6cm, 1.92cm)
 
@@ -207,9 +124,25 @@
           target = 0.64cm
         }
         let newGap = target - measurement.width
+
+        // Uppercase body if need be
+        let bodyContent = it.element.body
+        if isUpper and (it.level == 1 or it.element.numbering == none) {
+          bodyContent = upper(bodyContent)
+        }
+
+        // Reconstruct the it.inner content with the new body
+        let formattedInner = {
+          bodyContent
+          [ ] // whitespace is required for backwards compat
+          box(width: 1fr, it.fill)
+          [ ] // whitespace is required for backwards compat
+          it.page()
+        }
+
         link(
           it.element.location(),
-                    it.indented(prefixContent, it.inner(), gap: newGap),
+          it.indented(prefixContent, formattedInner, gap: newGap),
         )
       }
 
@@ -223,7 +156,7 @@
       show outline.entry: it => link(
         it.element.location(),
         it.indented(
-                    [*#__get-locatable-element-number(it.element) #__fig-supplement(image).*],
+          [*#__caption-label(it.element)*],
           [ #it.inner()], gap: 0pt
         ),
       )
@@ -234,7 +167,7 @@
       show outline.entry: it => link(
         it.element.location(),
         it.indented(
-                    [*#__get-locatable-element-number(it.element) #__fig-supplement(table).*],
+          [*#__caption-label(it.element)*],
           [ #it.inner()], gap: 0pt
         ),
       )
@@ -245,7 +178,7 @@
       show outline.entry: it => link(
         it.element.location(),
         it.indented(
-                    [*#__get-locatable-element-number(it.element) #__fig-supplement(raw).*],
+          [*#__caption-label(it.element)*],
           [ #it.inner()], gap: 0pt
         ),
       )
@@ -263,11 +196,12 @@
         #(o.target == selector(heading))
       ]*/
 
+      // Any other figure kind (generally, user defined kinds)
       show outline.entry: it => link(
         it.element.location(),
         it.indented(
-                    [*#__get-locatable-element-number(it.element) unknown.*],
-                    it.inner(), gap: 0pt
+          [*#__caption-label(it.element)*],
+          [ #it.inner()], gap: 0pt
         ),
       )
       o
@@ -291,7 +225,7 @@
   depth: none,
   indent: 0em, // Disable the default outline indenting
   target: heading,
-    title: auto
+  title: auto,
 ) = {
   __ktu_outline(depth: depth, indent: indent, target: target, title: title)
 }
@@ -308,6 +242,10 @@
   ktu-outline(title: heading("Paveikslų sąrašas", outlined: true), target: figure.where(kind: image))
 }
 
+#let ktu-code-list() = {
+  ktu-outline(title: heading("Kodo fragmentų sąrašas", outlined: true), target: figure.where(kind: raw))
+}
+
 /// Define a custom outline for a specific figure type.
 /// This function creates an outline entry with a specified title and figure type.
 /// The outline entry will link to the figure of the specified type.
@@ -340,8 +278,8 @@
       left: 30mm,
       right: 10mm,
       bottom: 20mm,
-            top: 20mm
-        )
+      top: 20mm,
+    ),
   )
   set page(width: 595pt, height: 842pt)
 
@@ -355,7 +293,7 @@
   set heading(numbering: "1.")
 
   // Number equations
-    set math.equation(numbering: "(1)")
+  set math.equation(numbering: __equation-numbering)
 
   // Set text font
   set text(
@@ -382,7 +320,9 @@
 
 #let setup-page(
   font: "Times New Roman",
-    body
+  uppercaseSectionTitles: false,
+  unnumberedHeadingAlignment: center,
+  body,
 ) = {
   show: __page_rules.with(font: font)
 
@@ -396,6 +336,14 @@
   set par(justify: true)
   set linebreak(justify: true)
 
+  let maybe-uppercase(body) = {
+    if uppercaseSectionTitles {
+      upper(body)
+    } else {
+      body
+    }
+  }
+
   // Antraštės
   // Force them into blocks so they don't count as paragraphs
   show heading: it => {
@@ -413,17 +361,25 @@
 
     // Antraštė be nr.
     if it.numbering == none {
-            // Centruota lygiuotė
-            set align(center)
+      // Centruota lygiuotė by default, but can be overridden by the user
+      let explicitlyAlignLeft = (
+        it.body.has("children")
+          and it.body.children.len() > 1
+          and it.body.children.at(1) == __METADATA_HEADING_ALIGN_LEFT
+      )
+
+      // If the user explicitly set alignLeft to true, we should align left regardless of the global setting
+      let targetAlign = if explicitlyAlignLeft { left } else { unnumberedHeadingAlignment }
+
+      set align(targetAlign)
 
       // atstumas prieš ir po antraštės - 10 pt
-            block[#it.body]
+      block[#maybe-uppercase(it.body)]
       if useExperiment {
         v(10pt)
       }
     }
 
-
     // Skyrius
     if it.numbering != none and it.level == 1 {
       // abipusė lygiuotė
@@ -431,13 +387,11 @@
 
       // Reset figure counters if numbering per section is enabled
       if __FIG_PER_SECTION.get() {
-                counter(figure.where(kind: image)).update(0)
-                counter(figure.where(kind: table)).update(0)
-                counter(figure.where(kind: raw)).update(0)
+        __reset-figure-counters()
       }
 
       // po antraštės - 10 pt
-            block[#counter(heading).display() #it.body]
+      block[#counter(heading).display() #maybe-uppercase(it.body)]
       if useExperiment {
         v(10pt)
       }
@@ -474,18 +428,45 @@
   // Bibliography citing style
   set bibliography(style: "assets/iso690-numeric-lt.csl")
 
+  // Block equations are left-aligned
+  show math.equation.where(block: true): it => {
+    align(left, it)
+  }
+
   body
 }
 
 /// Configure the template for a KTU paper.
-/// - font (str): Font to use for the paper.
+/// - font (str): Text font to use for the paper.
 /// - figureNumberingPerSection (bool): Whether to number figures per section rather than globally.
+/// - uppercaseSectionTitles (bool): Whether to uppercase section titles.
+/// - unnumberedHeadingAlignment (alignment): Alignment for unnumbered headings.
+/// - figureKinds (dictionary): Extra figure kinds, e.g.
+///   `(command: (supplement: "komanda", ref-name: "komanda", single-name: "komanda"))`.
+///   Use them with `figure(kind: "command", ...)`. Keys:
+///   - `supplement` (required): text after the number in captions, e.g. "1 komanda.".
+///   - `ref-name`: text used in references, e.g. "(žr. 1 komanda)". Defaults to `supplement`.
+///   - `single-name`: full word used when the figure is the only one of its kind
+///     and is left unnumbered, e.g. "Komanda." and "(žr. komanda)". Defaults to `supplement`.
+///     Currently only tables and images are left unnumbered when single, so this only matters for kinds added there.
 #let ktu-paper(
   font: "Times New Roman",
   figureNumberingPerSection: false,
-    body
+  figureKinds: (:),
+  // Not sure about formal conformity of deviating
+  // from the defaults of the next options,
+  // but some modules may require them
+  uppercaseSectionTitles: false,
+  unnumberedHeadingAlignment: center,
+  body,
 ) = context {
-    show: setup-page.with(font: font)
+  show: setup-page.with(
+    font: font,
+    uppercaseSectionTitles: uppercaseSectionTitles,
+    unnumberedHeadingAlignment: unnumberedHeadingAlignment,
+  )
   __FIG_PER_SECTION.update(figureNumberingPerSection)
+  __USER_FIGURE_KINDS.update(__normalize-figure-kinds(figureKinds))
+  __UPPERCASE_SECTION_TITLES.update(uppercaseSectionTitles)
   body
 }

template/ataskaita.typ

@@ -0,0 +1,150 @@
+#import "../src/lib.typ": (
+  ktu-paper, setup-code,
+  ktu-table-of-contents, ktu-picture-list, ktu-table-list, ktu-code-list,
+  ktu-heading-page-normal,
+  ktu-table,
+  unnumbered-heading,
+  custom-ref, ref-no
+)
+
+#show: ktu-paper.with(
+  font: "Times New Roman", // or Arial
+  figureNumberingPerSection: true,
+  figureKinds: (
+    veryCustomKind: (supplement: "komanda", ref-name: "komanda"),
+  ),
+)
+#show: setup-code.with()
+
+#show math.equation: it => {
+  show regex("\d+\.\d+"): it => {show ".": {","+h(0pt)}
+    it}
+  it
+}
+
+#ktu-heading-page-normal(
+  faculty: highlight[Informatikos fakultetas],
+  title: highlight[Laboratorinio darbo ataskaitos pavadinimas],
+  subtitles: (
+    "Laboratorinio darbo ataskaita",
+    highlight[Studijų modulio pavadinimas (kodas)]
+  ),
+  authors: (
+    (highlight[Vardenis Pavardenis], "Ataskaitą parengė"),
+    (highlight[Pareigų ir mokslo laipsnio sutrumpinimas Vardenis Pavardenis], "Laboratoriniam darbui vadovavo"),
+  ),
+  authorNamesFirst: false,
+  place: "Kaunas",
+  year: datetime.today().year(),
+)
+#pagebreak()
+#set page(numbering: "1")
+
+#ktu-table-of-contents()
+#pagebreak()
+
+#ktu-picture-list()
+#pagebreak()
+
+#ktu-table-list()
+#pagebreak()
+
+#ktu-code-list()
+#pagebreak()
+
+#unnumbered-heading[= Įvadas]
+#lorem(50)
+
+#pagebreak()
+= Skyrius
+#lorem(50)
+
+== Poskyrius
+#lorem(50)
+
+=== Dalis
+#lorem(50)
+
+#pagebreak()
+= Komponentai
+
+== Paveikslai
+Galima gauti paveikslo numerį #ref-no(<image-label>) arba visą paveikslo nuorodą #custom-ref(<image-label>).
+
+#figure(
+  image("images/santaka.png", width: 70%),
+  caption: [
+    Paveikslo pavadinimas
+  ],
+) <image-label>
+
+
+== Paveikslų "grid" išdėstymas
+Paveikslus taip pat galima sudėti į "grid" išdėstymą, į jį kreiptis kaip numerį #ref-no(<image-grid-label>) arba visą nuorodą #custom-ref(<image-grid-label>).
+
+#figure(grid(
+  columns: 2,
+  gutter: 2mm,
+  image("images/santaka.png", width: 100%),
+  image("images/santaka.png", width: 100%),
+), caption: [
+  Paveikslėliai "grid'e"
+]) <image-grid-label>
+
+
+== Lentelės
+
+Galima gauti lentelės numerį /*#ref-no(<table-label>)*/ arba visą lentelės nuorodą #custom-ref(<table-label>).
+
+#figure(
+  ktu-table(
+    columns: (auto, 1fr, 1fr, 1fr),
+    align: (x, y) => { horizon + left },
+    table.header([*Stulpelio pavadinimas*], [*Stulpelio pavadinimas*], [*Stulpelio pavadinimas*], [*Stulpelio pavadinimas*]),
+    [Eilutė 1], [...], [...], [...],
+    [Eilutė 2], [...], [...], [...],
+    [Eilutė 3], [...], [...], [...],
+    [Eilutė 4], [...], [...], [...],
+  ),
+  caption: [Lentelės pavadinimas],
+) <table-label>
+
+
+== Formulės
+Formules galima išreikšti eilutėje, pavyzdžiui $E = m c^2$ arba atskiroje eilutėje, pavyzdžiui:
+$
+  E = m c^2
+$ <formula-label>
+
+// TODO: fix
+Formulės atskiroje eilutėje numeruojamos besąlygiškai, galima gauti formulės numerį #ref-no(<formula-label>) arba visą formulės nuorodą #custom-ref(<formula-label>).
+
+== Kodo fragmentai
+Kodo fragmentus galima išreikšti eilutėje, pavyzdžiui `print("Hello, world!")` arba atskiroje eilutėje, pavyzdžiui:
+#figure(```python
+def main():
+  print("Hello, world!")
+if __name__ == "__main__":
+  main()
+```, caption: [
+  Python kodas
+]) <code-label>
+
+Galima gauti kodo fragmento numerį #ref-no(<code-label>) arba visą kodo fragmento nuorodą #custom-ref(<code-label>).
+
+== Naudotojo apibrėžtos figūros
+
+Naudotojai tai pat gali apibrėžti savo figūras, pavyzdžiui, komandų fragmentus. Pavyzdžiui, galima apibrėžti komandų fragmentą su papildoma informacija ir naudoti jį dokumente. Galima gauti kodo fragmento numerį #ref-no(<custom-figure-label>) arba visą kodo fragmento nuorodą #custom-ref(<custom-figure-label>).
+
+#figure(
+  ```sh
+  npm install foo
+  ```,
+  kind: "veryCustomKind",
+  supplement: "...",
+  caption: [Diegimo komanda],
+) <custom-figure-label>
+
+#pagebreak()
+= Išvados
+#lorem(50)

template/main.typ

@@ -1,4 +1,4 @@
-#import "@local/ktu-paper:0.2.0": (
+#import "@local/ktu-paper:0.4.1": (
   ktu-paper, setup-code,
   ktu-table-of-contents, ktu-picture-list, ktu-table-list,
   ktu-heading-page-normal,
@@ -94,7 +94,7 @@ show table.cell: set text(size: 8pt)
 Duomenų sklaidą yra patogu pavaizduoti ir stačiakampe diagrama:
 
 #figure(
-  image("image.jpeg", width: 70%),
+  image("images/santaka.png", width: 70%),
   caption: [
     Skaitinių atributų stačiakampė diagrama
   ],
@@ -136,8 +136,8 @@ Paveikslėlius galima grupuoti į "grid" #custom-ref(<image-grid-2>, <image-grid
 #figure(grid(
   columns: 2,
   gutter: 2mm,
-  image("image.jpeg", width: 100%),
-  image("image.jpeg", width: 100%),
+  image("images/santaka.png", width: 100%),
+  image("images/santaka.png", width: 100%),
 ), caption: [
   Paveikslėliai "grid'e" 1
 ]) <image-grid-2>
@@ -145,11 +145,11 @@ Paveikslėlius galima grupuoti į "grid" #custom-ref(<image-grid-2>, <image-grid
 #figure(grid(
   columns: 3,
   gutter: 2mm,
-  image("image.jpeg", width: 100%),
-  image("image.jpeg", width: 100%),
-  image("image.jpeg", width: 100%),
-  image("image.jpeg", width: 100%),
-  image("image.jpeg", width: 100%),
+  image("images/santaka.png", width: 100%),
+  image("images/santaka.png", width: 100%),
+  image("images/santaka.png", width: 100%),
+  image("images/santaka.png", width: 100%),
+  image("images/santaka.png", width: 100%),
 ), caption: [
   Paveikslėliai "grid'e" 2
 ]) <image-grid-3>
@@ -190,7 +190,7 @@ $
 čia $X$ -- imtis, $mu$ -- imties vidurkis, $sigma$ -- imties standartinis nuokrypis. Fragmentas pateikiamas #ref-no(<image-1>) paveikslėlyje.
 
 #figure(
-  image("image.jpeg", width: 90%),
+  image("images/santaka.png", width: 90%),
   caption: [
     Paveikslėlio pavadinimas
   ],

template/test.typ

@@ -1,4 +1,4 @@
-#import "@local/ktu-paper:0.2.0": (
+#import "@local/ktu-paper:0.4.0": (
   ktu-paper, setup-code,
   ktu-table-of-contents, ktu-picture-list, ktu-table-list,
   ktu-heading-page-normal,
@@ -191,7 +191,7 @@ Eu eum decore inimicus consetetur, cu usu habeo corpora intellegam. Ut antiopam
 Lorem ipsum dolor sit amet, eam ex decore persequeris, sit at illud lobortis atomorum. Sed dolorem quaerendum ne, prompta instructior ne pri. Et mel partiendo suscipiantur, docendi abhorreant ea sit. Recteque imperdiet eum te.
 
 #figure(
-  image("Picture.png", width: 11.2cm, height: 8cm),
+  image("images/santaka.png", width: 11.2cm, height: 8cm),
   caption: [
     Kauno technologijos universiteto „Santakos“ slėnio fasadas
   ],

typst.toml

@@ -1,7 +1,7 @@
 [package]
 name = "ktu-paper"
-version = "0.2.0"
-compiler = "0.15.1"
+version = "0.4.1"
+compiler = "0.15.0"
 entrypoint = "src/lib.typ"
 repository = "https://github.com/svetikas/ktu-paper"
 authors = ["Dominykas Svetikas <https://svetikas.lt>"]